SEGCallSitePseudoInputNode.h

/*
 * SEGCallSitePseudoInputNode.h
 *
 * Zhou Jinguo
 *
 * 2016.5.19
 *
 * This node is to model the pseudo input of a call site.
 */
 
#ifndef IR_SEG_SEGCALLSITEPSEUDOINPUTNODE_H
#define IR_SEG_SEGCALLSITEPSEUDOINPUTNODE_H
 
#include <llvm/IR/BasicBlock.h>
#include <llvm/IR/CallSite.h>
#include <llvm/IR/Constants.h>
#include <llvm/IR/Function.h>
#include <llvm/IR/Instructions.h>
#include <llvm/IR/Value.h>
#include <llvm/Support/FileSystem.h>
#include <llvm/Support/Format.h>
#include <llvm/Support/raw_ostream.h>
 
#include "CBAccessPath.h"
#include "IR/SEG/SymbolicExprGraph.h"
 
using namespace llvm;
 
class SEGCallSitePseudoInputNode : public SEGOperandNode {
protected:
  CallSite LLVMCS;
 
  Function *Callee = nullptr;
 
  SEGCallSitePseudoInputNode(CallSite CS, Function *Callee, Value *Val,
                             Type *Ty, SymbolicExprGraph *SEG, BasicBlock *BB,
                             bool fromDisk)
      : SEGOperandNode(SEGOBJK_CallSitePseudoInput, Val, Ty, SEG, BB, fromDisk),
        LLVMCS(CS), Callee(Callee), index(-1) {
    access_path.reset(Val);
  }
 
  friend class SymbolicExprGraph;
  friend class SEGCallSite;
  friend class SEGSerializer;
 
  CBAccessPath access_path;
 
  int64_t index;
 
public:
  ~SEGCallSitePseudoInputNode() {}
 
  CallSite getCallSite() const { return LLVMCS; }
 
  Function *getCallee() const { return Callee; }
 
  virtual int64_t getIndex() const { return index; }
 
  const CBAccessPath &getAccessPath() const { return access_path; }
 
  CBAccessPath &getAccessPath() { return access_path; }
 
public:
  static bool classof(const SEGObject *N) {
    return N->getKind() == SEGOBJK_CallSitePseudoInput;
  }
};
 
#endif