InstructionInfoTable.h

Go to the documentation of this file.

//===-- InstructionInfoTable.h ----------------------------------*- C++ -*-===//
//
//                     The KLEE Symbolic Virtual Machine
//
// This file is distributed under the University of Illinois Open Source
// License. See LICENSE.TXT for details.
//
//===----------------------------------------------------------------------===//
 
#ifndef KLEE_INSTRUCTIONINFOTABLE_H
#define KLEE_INSTRUCTIONINFOTABLE_H
 
#include <memory>
#include <string>
#include <unordered_map>
#include <vector>
 
namespace llvm {
  class Function;
  class Instruction;
  class Module; 
}
 
namespace klee {
 
  struct InstructionInfo {
    unsigned id;
    unsigned line;
    unsigned column;
    unsigned assemblyLine;
    const std::string &file;
 
  public:
    InstructionInfo(unsigned id, const std::string &file, unsigned line,
                    unsigned column, unsigned assemblyLine)
        : id{id}, line{line}, column{column}, assemblyLine{assemblyLine}, file{file} {}
  };
 
  struct FunctionInfo {
    unsigned id;
    unsigned line;
    uint64_t assemblyLine;
    const std::string &file;
 
  public:
    FunctionInfo(unsigned id, const std::string &file, unsigned line, uint64_t assemblyLine)
        : id{id}, line{line}, assemblyLine{assemblyLine}, file{file} {}
 
    FunctionInfo(const FunctionInfo &) = delete;
    FunctionInfo &operator=(FunctionInfo const &) = delete;
 
    FunctionInfo(FunctionInfo &&) = default;
  };
 
  class InstructionInfoTable {
    std::unordered_map<const llvm::Instruction *,
                       std::unique_ptr<InstructionInfo>>
        infos;
    std::unordered_map<const llvm::Function *, std::unique_ptr<FunctionInfo>>
        functionInfos;
    std::vector<std::unique_ptr<std::string>> internedStrings;
 
  public:
    explicit InstructionInfoTable(const llvm::Module &m);
 
    unsigned getMaxID() const;
    const InstructionInfo &getInfo(const llvm::Instruction &) const;
    const FunctionInfo &getFunctionInfo(const llvm::Function &) const;
  };
 
}
 
#endif /* KLEE_INSTRUCTIONINFOTABLE_H */