#ifndef INCLUDE__InformationManager_hxx
#define INCLUDE__InformationManager_hxx
-#include <boost/signals2.hpp>
+#include <log4cxx/logger.h>
#include <functional>
#include <string>
-#include <set>
+#include <map>
+#include <vector>
+#include <memory>
+
+#include "qt.hxx"
+#include "disassembler/Disassembler.hxx"
+
+#include "core/events/NewFunctionEvent.hxx"
+
+class Interpreter;
-class Disassembler;
class Function;
class BasicBlock;
+class Comment;
+
+class RenameFunctionEvent;
+class NewFunctionEvent;
+class ChangeCommentEvent;
class QString;
+class QTemporaryFile;
+class QPluginLoader;
-class InformationManager {
+class InformationManager : public QObject {
+#ifndef SWIG
+ Q_OBJECT
+signals:
+#else
public:
+#endif
+ void renameFunctionEvent(RenameFunctionEvent* event);
+ void newFunctionEvent(NewFunctionEvent event);
+ void changeCommentEvent(ChangeCommentEvent* event);
+ void resetEvent();
+public:
+ InformationManager();
~InformationManager();
+ // Start working on a fresh binary
void reset(const std::string& filename);
- void save(const QString& filename);
- void signal_new_function(Function* f);
+ // Load a saved binary
+ void load(const std::string& filename);
- boost::signals2::connection
- connect_new_function_signal(std::function<void(Function*)> f)
- { return new_function_signal.connect(f); }
+ // Save current state to disk
+ void save(const std::string& filename);
+ Disassembler* getDisassembler()
+ { return disassembler.get(); }
+ // Accessors
+ /* Used by the disassembler to determine whether to use unsafe
+ * heuristics for finding an entry point
+ */
+ bool hasFunctions() const {return functions.size() != 0;}
- boost::signals2::connection
- connect_new_dyn_symbol_signal(std::function<void(const std::string& name)> f)
- { return new_dyn_symbol_signal.connect(f); }
+ uint64_t getEntryAddress() {return disassembler->entryAddress();}
- void signal_new_dyn_symbol(const std::string& f)
- { new_dyn_symbol_signal(f); }
+ Function* getFunction(uint64_t address);
+ std::map<uint64_t, Function*>::const_iterator beginFunctions();
+ std::map<uint64_t, Function*>::const_iterator endFunctions();
- boost::signals2::connection
- connect_reset_signal(std::function<void ()> f)
- { return reset_signal.connect(f); }
+ BasicBlock* getBasicBlock(uint64_t address);
+ std::map<uint64_t, BasicBlock*>::const_iterator beginBasicBlocks();
+ std::map<uint64_t, BasicBlock*>::const_iterator endBasicBlocks();
+
+ std::pair<
+ std::multimap<uint64_t, Comment*>::const_iterator,
+ std::multimap<uint64_t, Comment*>::const_iterator>
+ getComments(uint64_t address);
+ std::multimap<uint64_t,Comment*>::const_iterator beginComments();
+ std::multimap<uint64_t,Comment*>::const_iterator endComments();
+
+ Interpreter* getInterpreter(const std::string& name);
+ bool hasInterpreters() const {return interpreters.size() != 0;}
+ std::map<std::string, Interpreter*>::const_iterator beginInterpreters();
+ std::map<std::string, Interpreter*>::const_iterator endInterpreters();
+
+ /* 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);
+ Function* newDynamicFunction(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 finishComment(Comment* c);
+ void deleteFunction(Function* f);
+ void deleteBasicBlock(BasicBlock* b);
+ void deleteComment(Comment* c);
- 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;
+
+ std::map<std::string, Interpreter*> interpreters;
+ std::map<uint64_t, Function*> functions;
+ std::map<uint64_t, BasicBlock*> blocks;
+ std::multimap<uint64_t, Comment*> comments;
+
+ std::string filename;
+ std::unique_ptr<QTemporaryFile> tmpfile;
+ std::vector<QPluginLoader*> plugins;
+
+ QThread disassemblerThread;
+ log4cxx::LoggerPtr logger;
};
#endif /* INCLUDE__InformationManager_hxx */