Pass the InformationManager inside the BasicBlock and Function classes

[frida/frida.git] / src / core / InformationManager.hxx

diff --git a/src/core/InformationManager.hxx b/src/core/InformationManager.hxx
index 5d2ef64c67342f98cfddd2805d36335c0cfc22b0..a43b1181281cd1deea73b4d8cb595a1128bd108b 100644 (file)
--- a/src/core/InformationManager.hxx
+++ b/src/core/InformationManager.hxx
@@ -3,41 +3,50 @@
 
 #include <boost/signals2.hpp>
 #include <functional>
+#include <string>
+#include <set>
 
 class Disassembler;
 class Function;
+class BasicBlock;
+
+class QString;
 
 class InformationManager {
 public:
-       boost::signals2::connection
-       connect_new_function_signal(std::function<void(Function*)> f) {
-               return new_function_signal.connect(f);
-       }
+       ~InformationManager();
 
-       void signal_new_function(Function* f) {
-               new_function_signal(f);
-       }
+       void reset(const std::string& filename);
+       void save(const QString& filename);
+
+       void signal_new_function(Function* f);
 
        boost::signals2::connection
-       connect_reset_signal(std::function<void ()> f) {
-               return reset_signal.connect(f);
-       }
+       connect_new_function_signal(std::function<void(Function*)> f)
+               { return new_function_signal.connect(f); }
 
-       // boost::signals2::connection
-       // connect_information_added_signal(uint64_t begin, uint64_t end,
-       //                                  std::function<void (uint64_t)>) {
 
-       // }
 
-       Disassembler* getDisassembler() {
-               return disassembler.get();
-       }
+       boost::signals2::connection
+       connect_new_dyn_symbol_signal(std::function<void(const std::string& name)> f)
+               { return new_dyn_symbol_signal.connect(f); }
 
-       void reset(const std::string& filename);
+       void signal_new_dyn_symbol(const std::string& f)
+               { new_dyn_symbol_signal(f);     }
+
+       boost::signals2::connection
+       connect_reset_signal(std::function<void ()> f)
+               { return reset_signal.connect(f); }
+
+       Disassembler* getDisassembler()
+               { return disassembler.get(); }
 private:
        boost::signals2::signal<void ()> reset_signal;
        boost::signals2::signal<void (Function*)> new_function_signal;
+       boost::signals2::signal<void (const std::string& name)> new_dyn_symbol_signal;
        std::unique_ptr<Disassembler> disassembler;
+       std::set<Function*> functions;
+       std::set<BasicBlock*> blocks;
 };
 
 #endif /* INCLUDE__InformationManager_hxx */