SEGCallSiteOutputNode.h

/*
 * SEGCallSiteOutputNode.h
 *
 * Qingkai
 *
 * This node is to model the output of a call site.
 */
 
#ifndef IR_SEG_SEGCALLSITEOUTPUTNODE_H
#define IR_SEG_SEGCALLSITEOUTPUTNODE_H
 
#include <llvm/IR/BasicBlock.h>
 
#include "CBAccessPath.h"
#include "IR/SEG/SymbolicExprGraph.h"
 
using namespace llvm;
 
class SEGCallSiteOutputNode : public SEGOperandNode {
public:
  friend class SEGSerializer;
 
protected:
  SEGCallSite *SEGCS;
 
  SEGCallSiteOutputNode(SEGObjectKind, SEGCallSite *CS, Type *Ty,
                        SymbolicExprGraph *SEG, BasicBlock *BB, bool fromDisk);
 
  SEGCallSiteOutputNode(SEGObjectKind, SEGCallSite *CS, Value *Val, Type *Ty,
                        SymbolicExprGraph *SEG, BasicBlock *BB, bool fromDisk);
 
  friend class SymbolicExprGraph;
 
public:
  ~SEGCallSiteOutputNode() {}
 
  virtual Instruction *getLLVMDbgInstruction() const override;
 
  const SEGCallSite *getCallSite() const { return SEGCS; }
 
  virtual int64_t getIndex() const = 0;
 
public:
  static bool classof(const SEGObject *N) {
    return N->getKind() >= SEGOBJK_CallSiteOutputBegin &&
           N->getKind() <= SEGOBJK_CallSiteOutputEnd;
  }
};
 
class SEGCallSiteCommonOutputNode : public SEGCallSiteOutputNode {
private:
  SEGCallSiteCommonOutputNode(SEGCallSite *CS, Type *Ty, SymbolicExprGraph *SEG,
                              BasicBlock *BB, bool fromDisk);
 
  SEGCallSiteCommonOutputNode(SEGCallSite *CS, Value *Val, Type *Ty,
                              SymbolicExprGraph *SEG, BasicBlock *BB,
                              bool fromDisk);
 
  friend class SymbolicExprGraph;
  friend class SEGSerializer;
 
public:
  ~SEGCallSiteCommonOutputNode() {}
 
  virtual int64_t getIndex() const {
    assert(false);
    return 0;
  }
 
public:
  static bool classof(const SEGObject *N) {
    return N->getKind() == SEGOBJK_CallSiteCommonOutput;
  }
};
 
class SEGCallSitePseudoOutputNode : public SEGCallSiteOutputNode {
private:
  SEGCallSitePseudoOutputNode(SEGCallSite *CS, Function *Callee, Value *Val,
                              Type *Ty, SymbolicExprGraph *SEG, BasicBlock *BB,
                              bool fromDisk);
 
  // If it is a pseudo call site return node, it has an index
  // It is set by the SEGCallSite
  int64_t Index = -1;
 
  CBAccessPath AP;
 
  Function *Callee = nullptr;
 
  friend class SymbolicExprGraph;
  friend class SEGCallSite;
  friend class SEGSerializer;
 
public:
  ~SEGCallSitePseudoOutputNode() {}
 
  virtual int64_t getIndex() const { return Index; }
 
  const CBAccessPath &getAccessPath() const { return AP; }
 
  CBAccessPath &getAccessPath() { return AP; }
 
  Function *getCallee() const { return Callee; }
 
public:
  static bool classof(const SEGObject *N) {
    return N->getKind() == SEGOBJK_CallSitePseudoOutput;
  }
};
 
#endif