src/core/InformationManager.hxx

   1 #ifndef INCLUDE__InformationManager_hxx
   2 #define INCLUDE__InformationManager_hxx
   3
   4 #include <boost/signals2.hpp>
   5 #include <functional>
   6 #include <string>
   7 #include <map>
   8
   9 class Disassembler;
  10 class Function;
  11 class BasicBlock;
  12
  13 class QString;
  14
  15 class RenameFunctionEvent;
  16
  17 class InformationManager {
  18 public:
  19         ~InformationManager();
  20
  21         void reset(const std::string& filename);
  22         void save(const QString& filename);
  23
  24         void signal_new_function(Function* f);
  25         void signal_new_dyn_symbol(const std::string& f)
  26                 { new_dyn_symbol_signal(f);     }
  27
  28         boost::signals2::connection
  29         connect_new_function_signal(std::function<void(Function*)> f)
  30                 { return new_function_signal.connect(f); }
  31
  32         boost::signals2::connection
  33         connect_new_dyn_symbol_signal(std::function<void(const std::string& name)> f)
  34                 { return new_dyn_symbol_signal.connect(f); }
  35
  36         boost::signals2::connection
  37         connect_reset_signal(std::function<void ()> f)
  38                 { return reset_signal.connect(f); }
  39
  40         boost::signals2::connection
  41         connect_rename_function_signal(std::function<void (RenameFunctionEvent*)> f)
  42                 { return rename_function_signal.connect(f); }
  43
  44         Disassembler* getDisassembler()
  45                 { return disassembler.get(); }
  46
  47         void dispatch(RenameFunctionEvent* event)
  48                 { rename_function_signal(event); }
  49
  50         Function* getFunction(uint64_t address)
  51         { return functions[address]; }
  52
  53         BasicBlock* getBlock(uint64_t address)
  54         { return blocks[address]; }
  55
  56 private:
  57         boost::signals2::signal<void ()> reset_signal;
  58         boost::signals2::signal<void (Function*)> new_function_signal;
  59         boost::signals2::signal<void (const std::string& name)> new_dyn_symbol_signal;
  60         boost::signals2::signal<void (RenameFunctionEvent*)> rename_function_signal;
  61         std::unique_ptr<Disassembler> disassembler;
  62         std::map<uint64_t, Function*> functions;
  63         std::map<uint64_t, BasicBlock*> blocks;
  64 };
  65
  66 #endif /* INCLUDE__InformationManager_hxx */