Finish archive loading

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

diff --git a/src/core/InformationManager.hxx b/src/core/InformationManager.hxx
index dc781285f049de01b2200f6a2e852ee3f21d9353..557816877ec9ded8804bd0f88118191818573062 100644 (file)
--- a/src/core/InformationManager.hxx
+++ b/src/core/InformationManager.hxx
@@ -1,14 +1,18 @@
 #ifndef INCLUDE__InformationManager_hxx
 #define INCLUDE__InformationManager_hxx
 
+#include <log4cxx/logger.h>
 #include <boost/signals2.hpp>
 #include <functional>
 #include <string>
 #include <map>
 
-class Disassembler;
+#include "disassembler/Disassembler.hxx"
+#include <QTemporaryFile>
+
 class Function;
 class BasicBlock;
+class Comment;
 
 class QString;
 
@@ -16,10 +20,12 @@ class RenameFunctionEvent;
 
 class InformationManager {
 public:
+       InformationManager();
        ~InformationManager();
 
        void reset(const std::string& filename);
-       void save(const QString& filename);
+       void load(const std::string& filename);
+       void save(const std::string& filename);
 
        void signal_new_function(Function* f);
        void signal_new_dyn_symbol(const std::string& f)
@@ -47,12 +53,39 @@ public:
        void dispatch(RenameFunctionEvent* event)
                { rename_function_signal(event); }
 
-       Function* getFunction(uint64_t address)
-       { return functions[address]; }
-
-       BasicBlock* getBlock(uint64_t address)
-       { return blocks[address]; }
-
+       Function* getFunction(uint64_t address);
+       BasicBlock* getBasicBlock(uint64_t address);
+       bool hasFunctions() const {return functions.size() != 0;}
+
+       /* Protocoll:
+        *
+        * Users may allocate new Data containers with the new*()
+        * functions. Once they have populated the information they hand
+        * over the object to the information manager using the finish*()
+        * functions.
+        *
+        * if new*() returns NULL there already exists a function at the
+        * specified address. Users may then get the old object if they
+        * wish or (more likely) skip creating it. Uniqueness of the
+        * object is only guaranteed as compared to the finish()ed
+        * objects.
+        *
+        * Users are responsible for destroying functions iff they do not
+        * finish them using the delete*() functions. Once the objects are
+        * finished, the information manager is responsible for cleaning
+        * up the memory. If delete*() is called on a finished object, bad
+        * thingsmay happen.
+        */
+       Function* newFunction(uint64_t address);
+       BasicBlock* newBasicBlock(uint64_t address);
+       Comment* newGlobalComment(uint64_t address);
+       Comment* newLocalComment(uint64_t address, Function* f);
+       void finishFunction(Function* f);
+       void finishBasicBlock(BasicBlock* b);
+       void finnishComment(Comment* c);
+       void deleteFunction(Function* f);
+       void deleteBasicBlock(BasicBlock* b);
+       void deleteComment(Comment* c);
 private:
        boost::signals2::signal<void ()> reset_signal;
        boost::signals2::signal<void (Function*)> new_function_signal;
@@ -61,6 +94,10 @@ private:
        std::unique_ptr<Disassembler> disassembler;
        std::map<uint64_t, Function*> functions;
        std::map<uint64_t, BasicBlock*> blocks;
+       std::string filename;
+       std::unique_ptr<QTemporaryFile> tmpfile;
+
+       log4cxx::LoggerPtr logger;
 };
 
 #endif /* INCLUDE__InformationManager_hxx */