Reoganize Function/BasicBlock creation
[frida/frida.git] / src / core / InformationManager.cxx
index a274c418091e40efcab87f1f29821cb67ab3c5f9..294dc2f53583b7e3a9c5ce6b8f33fd49427bf058 100644 (file)
@@ -2,6 +2,7 @@
 #include "disassembler/llvm/LLVMDisassembler.hxx"
 #include "core/Function.hxx"
 #include "core/BasicBlock.hxx"
+#include "core/Comment.hxx"
 
 #include "gui/qt.hxx"
 #include <quazip/quazip.h>
@@ -61,10 +62,68 @@ void InformationManager::save(const QString& filename) {
 }
 
 void InformationManager::signal_new_function(Function* fun) {
-       functions.insert(std::make_pair(fun->getStartAddress(), fun));
+}
+
+Function* InformationManager::getFunction(uint64_t address) {
+       auto it = functions.find(address);
+       if (it != functions.end())
+               return it->second;
+       else
+               return NULL;
+}
+
+BasicBlock* InformationManager::getBasicBlock(uint64_t address) {
+       auto it = blocks.find(address);
+       if (it != blocks.end())
+               return it->second;
+       else
+               return NULL;
+}
+
+Function* InformationManager::newFunction(uint64_t address) {
+       Function* fun = new Function(address, this);
+       functions.insert(std::make_pair(address, fun));
+       return fun;
+}
+
+BasicBlock* InformationManager::newBasicBlock(uint64_t address) {
+       BasicBlock* block = new BasicBlock(address, this);
+       blocks.insert(std::make_pair(address, block));
+       return block;
+}
+
+Comment* InformationManager::newGlobalComment(uint64_t address) {
+       return NULL;
+}
+
+Comment* InformationManager::newLocalComment(uint64_t address, Function* f) {
+       return NULL;
+}
+
+void InformationManager::finishFunction(Function* fun) {
        for (auto b : fun->blocks()) {
                BasicBlock* bl = b.second;
                blocks.insert(std::make_pair(bl->getStartAddress(), bl));
        }
        new_function_signal(fun);
 }
+
+void InformationManager::finishBasicBlock(BasicBlock* b) {
+}
+
+void InformationManager::finnishComment(Comment* c) {
+}
+
+void InformationManager::deleteFunction(Function* f) {
+       functions.erase(f->getStartAddress());
+       delete f;
+}
+
+void InformationManager::deleteBasicBlock(BasicBlock* b) {
+       blocks.erase(b->getStartAddress());
+       delete b;
+}
+
+void InformationManager::deleteComment(Comment* c) {
+       delete c;
+}