SEGCallSiteReturnSummaryNode.h

/*
 * SEGCallSiteReturnSummaryNode.h
 *
 * This node is to model a summary node passed from callee to caller.
 *
 *  Created on: 2016.12.15
 *      Author: andyzhou
 */
 
#ifndef IR_SEG_SEGCALLSITERETURNSUMMARYNODE_H
#define IR_SEG_SEGCALLSITERETURNSUMMARYNODE_H
 
#include "IR/SEG/SEGCallSite.h"
#include "IR/SEG/SymbolicExprGraph.h"
 
using namespace llvm;
 
class SEGCallSiteReturnSummaryNode;
class PersistedSEGCallSiteReturnSummaryNode;
 
class SEGCallSiteReturnSummaryNode : public SEGOperandNode {
public:
  SEGCallSiteReturnSummaryNode(PersistedSEGCallSiteReturnSummaryNode *Node,
                               SymbolicExprGraph *SEG);
 
  virtual void assembleSEGObject(std::map<int, SEGObject *> &FuncSEGObjMap);
 
private:
  // The callsite that the summary node is created for
  Instruction *Callsite = nullptr;
  WrappedInstruction *WrappedCallSite = nullptr;
 
  // 0-1 confidence of that the node is valid
  float Confidence;
 
  SEGCallSiteReturnSummaryNode(Type *Ty, SymbolicExprGraph *SEG, BasicBlock *BB,
                               Instruction *Callsite, float Confidence);
 
public:
  virtual ~SEGCallSiteReturnSummaryNode();
 
  virtual PersistedSEGObject *createPersistedObject() const;
 
public:
  static bool classof(const SEGObject *N) {
    return N->getKind() == SEGOBJK_CallSiteSummaryReturn;
  }
 
  Instruction *getCallSiteInst() const { return Callsite; }
 
  WrappedInstruction *getWrappedCallSiteInst() const { return WrappedCallSite; }
 
  float getConfidence() const { return Confidence; }
 
  SEGCallSite *getSEGCallSite() const {
    SEGSiteBase *SEGSite =
        getParentGraph()->findSite<SEGSiteBase>(getCallSiteInst());
    return dyn_cast<SEGCallSite>(SEGSite);
  }
 
  friend class SymbolicExprGraph;
};
 
#endif /* IR_SEG_SEGCALLSITERETURNSUMMARYNODE_H */