SEGStoreMemNode.h

/*
 * SEGStoreMemNode.h
 *
 * Qingkai
 *
 * This node is to model a value stored in a memory.
 */
 
#ifndef IR_SEG_SEGSTOREMEMNODE_H
#define IR_SEG_SEGSTOREMEMNODE_H
 
#include <llvm/IR/BasicBlock.h>
#include <llvm/IR/Function.h>
#include <llvm/IR/Instructions.h>
#include <llvm/IR/Value.h>
 
#include "IR/SEG/SymbolicExprGraph.h"
 
class PersistedSEGStoreMemNode;
 
using namespace llvm;
 
class SEGStoreMemNode : public SEGOperandNode {
public:
  SEGStoreMemNode(PersistedSEGStoreMemNode *Node, SymbolicExprGraph *SEG);
 
  virtual void assembleSEGObject(std::map<int, SEGObject *> &FuncSEGObjMap) {
    SEGOperandNode::assembleSEGObject(FuncSEGObjMap);
  }
 
private:
  Instruction *StoreSite = nullptr;
  WrappedInstruction *WrappedStoreSite = nullptr;
  Value *StoreValue = nullptr;
  WrappedValue *WrappedStoreValue = nullptr;
  friend class SymbolicExprGraph;
 
  SEGStoreMemNode(Type *Ty, Instruction *StoreSite, Value *StoreVal,
                  BasicBlock *BB, SymbolicExprGraph *SEG);
 
public:
  ~SEGStoreMemNode() {}
 
  virtual PersistedSEGObject *createPersistedObject() const;
 
  Instruction *getStoreSite() const { return StoreSite; }
 
  WrappedInstruction *getWrappedStoreSite() const { return WrappedStoreSite; }
 
  StoreInst *getStoreSiteAsStoreInst() const {
    if (!StoreSite)
      return nullptr;
    return dyn_cast<StoreInst>(StoreSite);
  }
 
  WrappedStoreInst *getWrappedStoreSiteAsStoreInst(void) const {
    if (!WrappedStoreSite) {
      return nullptr;
    }
    return dyn_cast<WrappedStoreInst>(WrappedStoreSite);
  }
 
  Value *getStoreValue() const { return StoreValue; }
 
  WrappedValue *getWrappedStoreValue() const { return WrappedStoreValue; }
 
  void dot(raw_fd_ostream &O) const override;
 
  virtual Instruction *getLLVMDbgInstruction() const override;
 
  virtual std::string getSrcFile() const {
    if (WrappedStoreSite) {
      return WrappedStoreSite->getSrcFileName();
    }
    return SEGOperandNode::getSrcFile();
  }
 
  virtual std::int32_t getSrcLine() const {
    if (WrappedStoreSite) {
      return WrappedStoreSite->getSrcLine();
    }
    return SEGOperandNode::getSrcLine();
  }
 
public:
  static bool classof(const SEGObject *N) {
    return N->getKind() == SEGOBJK_StoreMem;
  }
};
 
#endif