src/monitor/TaintTracker.h

   1 #ifndef AESALON_MONITOR_TAINT_TRACKER_H
   2 #define AESALON_MONITOR_TAINT_TRACKER_H
   3
   4 #include <vector>
   5
   6 #include "asm/Instruction.h"
   7 #include "asm/InstructionList.h"
   8 #include "asm/Disassembler.h"
   9 #include "ptrace/BreakpointObserver.h"
  10 #include "TaintedData.h"
  11
  12 namespace Aesalon {
  13 namespace Monitor {
  14
  15 class TaintTracker : public PTrace::BreakpointObserver {
  16 protected:
  17     typedef std::vector<Misc::SmartPointer<TaintedData> > tainted_data_t;
  18 private:
  19     Misc::SmartPointer<ASM::Disassembler> disassembler;
  20     tainted_data_t tainted_data;
  21 public:
  22     TaintTracker(Misc::SmartPointer<ASM::Disassembler> disassembler) : disassembler(disassembler) {}
  23     virtual ~TaintTracker() {}
  24
  25     Misc::SmartPointer<TaintedData> get_tainted_data(Word address) const;
  26     void add_tainted_data(Misc::SmartPointer<TaintedData> new_tainted_data) {
  27         tainted_data.push_back(new_tainted_data);
  28     }
  29
  30     virtual void handle_breakpoint(const PTrace::BreakpointReference &breakpoint);
  31
  32     void place_breakpoints(std::string symbol);
  33 };
  34
  35 } // namespace Monitor
  36 } // namespace Aesalon
  37
  38 #endif