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 SEGLoadMemNode : public SEGOperandNode {
  friend class SEGSerializer;
  friend class SEGHash;
 
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,
                 bool fromDisk);
 
public:
  ~SEGLoadMemNode() {}
 
  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;
 
public:
  static bool classof(const SEGObject *N) {
    return N->getKind() == SEGOBJK_LoadMem;
  }
};
 
#endif