MyVisitor.h

#include <iostream>
#include "Visitor.h"
#include "Immediate.h"
#include "BinaryFunction.h"
#include "Dereference.h"
#include "Register.h"
#include "Result.h"
#include <string>
using namespace std;

class MyVisitor: public Dyninst::InstructionAPI::Visitor {
  public :
    MyVisitor ( bool _debug=false) { 
      isImmediate = 0; 
      regUsed = Dyninst::x86_64::rip; 
      immediateValue = -1;
      isDereference = false;
      isRegister = false;
      debug = _debug;
    };
    ~MyVisitor () {};

    bool debug;
    bool isDereference, isRegister, isBinaryFunction, isImmediate;

    int getImmediateValue() {
      assert(isImmediate);
      if(debug) cout<<" returning immediateValue ("<<immediateValue<<") ";
      return immediateValue; 
    }

    Dyninst::MachRegister regUsed;
    const Dyninst::MachRegister getRegUsed() {
      assert(isRegister);
      if(debug) cout<< " returning reg("<<regUsed.name()<<") ";
      return regUsed; 
    }

    int immediateValue;
    bool getIsImmediate() { return isImmediate; }
    
    virtual void visit ( Dyninst::InstructionAPI::BinaryFunction * b ) {
      if (debug) cout <<" \tVisiting BinaryFunction\n";
      isBinaryFunction = true;
      isImmediate = isRegister = isDereference = false;
    };

    virtual void visit ( Dyninst::InstructionAPI::Immediate * i ) {
      Dyninst::InstructionAPI::Result result = i->eval();
      if(debug) cout << " \t Immediate value = " << result.format()<<" ";
      //Dyninst::InstructionAPI::s32 ret = (Dyninst::InstructionAPI::s32) result.convert();
      //immediateValue = (int) ret;
      immediateValue = result.convert<int>();
      isImmediate = true;
      isDereference = isRegister = isBinaryFunction = false;
    };

    virtual void visit ( Dyninst::InstructionAPI::RegisterAST * r ) {
      if(debug) {
        cout <<" \tVisiting register " << r -> getID ().name ()<< " " << r->format()<<endl;
      }
      //Dyninst::InstructionAPI::RegisterAST::Ptr rPtr = Dyninst::InstructionAPI::RegisterAST::promote(r);
      regUsed = r->getID();
      if(r -> getID() == Dyninst::x86_64::eax) {
        if(debug) cout<<" visited eax ("<<regUsed.name()<<")";
      }
      isDereference = isImmediate = isBinaryFunction = false;
      isRegister = true;
    }
    
    virtual void visit ( Dyninst::InstructionAPI::Dereference * d ) {
      if (debug) cout <<" \tVisiting Dereference ";
      /*std::vector<Dyninst::InstructionAPI::InstructionAST::Ptr> children;
      d->getChildren(children);
      Dyninst::InstructionAPI::Result result = (*children[0]).eval();
      unsigned long long address = result.convert<unsigned long long>();
      cout<<" address = "<<address<<" result = "<<result.format()<<endl;*/
      isDereference = true;
      isRegister = isImmediate = isBinaryFunction = false;
    };
};