SEGLoadMemNode.h

/*
 * SEGLoadMemNode.h
 *
 * Qingkai
 *
 * This node is to model value loaded from a memory.
 */
 
#ifndef IR_SEG_SEGLOADMEMNODE_H
#define IR_SEG_SEGLOADMEMNODE_H
 
#include <llvm/IR/BasicBlock.h>
 
#include <sstream>
#include <unordered_map>
 
#include "IR/SEG/SymbolicExprGraph.h"
 
using namespace llvm;
 
class PersistedSEGLoadMemNode;
 
class SEGLoadMemNode : public SEGOperandNode {
public:
  SEGLoadMemNode(PersistedSEGLoadMemNode *Node, SymbolicExprGraph *SEG);
 
  virtual void assembleSEGObject(std::map<int, SEGObject *> &FuncSEGObjMap);
 
private:
  // Used when pointer analysis is applied
  // Loaded Value -> Conditions
  std::unordered_map<const SEGNodeBase *, SEGRegionNode *> MatchingConditions;
  friend class SymbolicExprGraph;
 
  SEGLoadMemNode(Type *Ty, SymbolicExprGraph *SEG, BasicBlock *BB);
 
public:
  ~SEGLoadMemNode() {}
 
  virtual PersistedSEGObject *createPersistedObject() const;
 
  void addMatchingCondition(const SEGNodeBase *Node, SEGRegionNode *Cond) {
    MatchingConditions[Node] = Cond;
  }
 
  SEGRegionNode *getMatchingCondition(const SEGNodeBase *Node) const {
    auto It = MatchingConditions.find(Node);
    if (It != MatchingConditions.end()) {
      return It->second;
    }
    // assert(false && "No matching condition for a mem node.");
    // It may return null when falcon is disabled.
    return nullptr;
  }
 
  void dot(raw_fd_ostream &O) const;
 
  virtual std::string getAuxiliaryDebugStr() const {
    std::stringstream Result;
    int Index = 1;
    for (auto &It : MatchingConditions) {
      if (It.first) {
        Result << Index << ". value is from file:" << It.first->getSrcFile()
               << ", the line is:" << It.first->getSrcLine() << ";";
      } else {
        Result << Index << ". value is from file: NA"
               << ", the line is: NA"
               << ";";
      }
      Index++;
    }
    return Result.str();
  }
 
public:
  static bool classof(const SEGObject *N) {
    return N->getKind() == SEGOBJK_LoadMem;
  }
};
 
#endif