SummaryMapper.h

//
// Created by yongchao on 12/20/22.
//
 
#ifndef CLEARBLUE_SUMMARYMAPPER_H
#define CLEARBLUE_SUMMARYMAPPER_H
 
#include <map>
#include <set>
 
#include <llvm/Support/Debug.h>
 
#include "Checker/PSA/SymbolicSummary.h"
#include "Checker/PSA/TraceSummary.h"
 
class SummaryMapper;
 
typedef struct {
  std::set<TraceSummary *> ParamSummarySet;
 
  std::set<TraceSummary *> ErSrcSummarySet;
 
  std::set<TraceSummary *> SrcWpSummarySet;
 
  std::set<TraceSummary *> CSOutSummarySet;
 
  std::map<const SEGReturnNode *, SymbolicSummary *> RetValSummaryMap;
 
  SummaryMapper *SailfishSummaryMapper = nullptr;
} SailfishSummary;
 
class SummaryMapper {
private:
  std::map<std::pair<Instruction *, bool>,
           std::vector<std::set<TransferSummary *>>>
      TransferSmryMap;
 
  std::map<std::pair<Instruction *, bool>,
           std::vector<std::set<SourceSummary *>>>
      SourceSmryMap;
 
  std::map<std::pair<Instruction *, bool>,
           std::vector<std::set<InputSummary *>>>
      InputSmryMap;
 
  std::map<std::pair<Instruction *, bool>,
           std::vector<std::set<OutputSummary *>>>
      OutputSmryMap;
 
public:
  SummaryMapper(Function *F);
 
  ~SummaryMapper() {}
 
  void index(Instruction *CallSite, std::set<TraceSummary *> &ParamSummarySet);
 
public:
  const std::set<InputSummary *> *queryInputSummary(Instruction *, bool,
                                                    size_t) const;
  const std::set<TransferSummary *> *queryTransferSummary(Instruction *, bool,
                                                          size_t) const;
  const std::set<SourceSummary *> *querySourceSummary(Instruction *, bool,
                                                      size_t) const;
  const std::set<OutputSummary *> *queryOutputSummary(Instruction *, bool,
                                                      size_t) const;
 
  bool hasInputSummary(Instruction *, bool, size_t) const;
  bool hasTransferSummary(Instruction *, bool, size_t) const;
  bool hasSourceSummary(Instruction *, bool, size_t) const;
  bool hasOutputSummary(Instruction *, bool, size_t) const;
 
  std::map<std::pair<Instruction *, bool>,
           std::vector<std::set<SourceSummary *>>>::iterator
  source_summary_begin() {
    return SourceSmryMap.begin();
  }
 
  std::map<std::pair<Instruction *, bool>,
           std::vector<std::set<SourceSummary *>>>::iterator
  source_summary_end() {
    return SourceSmryMap.end();
  }
};
 
#endif // CLEARBLUE_SUMMARYMAPPER_H