SailfishStateBuilder.h

//
// Created by yongchao on 12/20/22.
//
 
#ifndef CLEARBLUE_YSPSASTATEBUILDER_H
#define CLEARBLUE_YSPSASTATEBUILDER_H
 
#include "Checker/PSA/SummaryBase.h"
#include "Checker/PSA/VulnerabilityTraceBuilder.h"
#include "IR/SEG/SymbolicExprGraphSolver.h"
 
class SailfishStateBuilder : public SymbolicExprGraphSolver,
                             public VulnerabilityTraceBuilder {
private:
  PushPopCache<const SEGObject *> SEGObjectCache;
  PushPopVector<SummaryCacheItem> NonSymDepsCache;
  PushPopVector<SummaryCacheItem> SymDepsCache;
 
public:
  SailfishStateBuilder(SMTFactory &Fcty, TSDataLayout &D,
                       ExternalMemorySpec *MemSepc, ExternalIOSpec *IOSpec)
      : SymbolicExprGraphSolver(Fcty, D, MemSepc, IOSpec),
        VulnerabilityTraceBuilder() {}
  virtual ~SailfishStateBuilder() {}
 
public:
  void pushState() {
    SymbolicExprGraphSolver::push();
    VulnerabilityTraceBuilder::push();
    SEGObjectCache.push();
    SymDepsCache.push();
    NonSymDepsCache.push();
  }
 
  void popState() {
    SymbolicExprGraphSolver::pop();
    VulnerabilityTraceBuilder::pop();
    SEGObjectCache.pop();
    SymDepsCache.pop();
    NonSymDepsCache.pop();
  }
 
  void resetState() {
    SymbolicExprGraphSolver::reset();
    VulnerabilityTraceBuilder::reset();
    SEGObjectCache.reset();
    SymDepsCache.reset();
    NonSymDepsCache.reset();
  }
 
  void recordState(const SEGObject *O) {
    VulnerabilityTraceBuilder::add(O);
    SEGObjectCache.add(O);
  }
 
  void recordState(const std::shared_ptr<VulnerabilityTrace> &T) {
    VulnerabilityTraceBuilder::add(T);
  }
 
  void recordState(const SMTExpr &E) { SymbolicExprGraphSolver::add(E); }
 
  void recordState(const SMTExprVec &EV) {
    SymbolicExprGraphSolver::addAll(EV);
  }
 
  void recordState(const SummaryCacheItem &SCI, bool Sym) {
    if (Sym)
      SymDepsCache.push_back(SCI);
    else
      NonSymDepsCache.push_back(SCI);
  }
 
public:
  bool hasState(const SEGObject *O) const { return SEGObjectCache.contains(O); }
 
public:
  std::vector<SummaryCacheItem>::const_iterator sym_dep_begin() const {
    return SymDepsCache.getCacheVector().begin();
  }
 
  std::vector<SummaryCacheItem>::const_iterator sym_dep_end() const {
    return SymDepsCache.getCacheVector().end();
  }
 
  std::vector<SummaryCacheItem>::const_iterator nonsym_dep_begin() const {
    return NonSymDepsCache.getCacheVector().begin();
  }
 
  std::vector<SummaryCacheItem>::const_iterator nonsym_dep_end() const {
    return NonSymDepsCache.getCacheVector().end();
  }
};
 
#endif // CLEARBLUE_YSPSASTATEBUILDER_H