SEGReturnSite.h

/*
 * SEGCallSite.h
 *
 * Qingkai
 *
 * This node is to model a call site.
 */
 
#ifndef IR_SEG_RETURNSITE_H
#define IR_SEG_RETURNSITE_H
 
#include <llvm/IR/Instructions.h>
 
#include "IR/SEG/SymbolicExprGraph.h"
#include "Persistence/PersistenceBasis.h"
 
using namespace llvm;
 
class SEGReturnSite : public SEGSiteBase {
public:
  SEGReturnSite(PersistedSEGReturnSite *Site, SymbolicExprGraph *SEG)
      : SEGSiteBase(Site, SEG) {}
 
  virtual void assembleSEGObject(std::map<int, SEGObject *> &FuncSEGObjMap) {
    SEGSiteBase::assembleSEGObject(FuncSEGObjMap);
  }
 
private:
  SEGReturnSite(Instruction *U, SymbolicExprGraph *G)
      : SEGSiteBase(SEGOBJK_ReturnSite, U, G) {}
 
  friend class SymbolicExprGraph;
  friend class IntraFalcon;
  friend class MantaIntraFalcon;
 
public:
  virtual PersistedSEGObject *createPersistedObject() const {
    PersistedSEGObject *PersistedObj = new PersistedSEGReturnSite;
    return PersistedObj;
  }
 
public:
  static bool classof(const SEGObject *N) {
    return N->getKind() == SEGOBJK_ReturnSite;
  }
};
 
#endif