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 4e1b3187a52a081e5453aa9beb661af872170d55..a43b1181281cd1deea73b4d8cb595a1128bd108b 100644 (file)
--- a/src/core/InformationManager.hxx
+++ b/src/core/InformationManager.hxx
@@ -8,20 +8,24 @@
 
 class Disassembler;
 class Function;
+class BasicBlock;
 
 class QString;
 
 class InformationManager {
 public:
+       ~InformationManager();
+
        void reset(const std::string& filename);
        void save(const QString& filename);
 
+       void signal_new_function(Function* f);
+
        boost::signals2::connection
        connect_new_function_signal(std::function<void(Function*)> f)
                { return new_function_signal.connect(f); }
 
-       void signal_new_function(Function* f)
-               { functions.insert(f); new_function_signal(f); }
+
 
        boost::signals2::connection
        connect_new_dyn_symbol_signal(std::function<void(const std::string& name)> f)
@@ -42,6 +46,7 @@ private:
        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 */