LibraryFunctions.hpp

/**
 * Copyright (C) 2014 - 2020 Map2Check tool
 * This file is part of the Map2Check tool, and is made available under
 * the terms of the GNU General Public License version 2.
 *
 * LLVM -> NCSA
 *
 * SPDX-License-Identifier: (GPL-2.0 AND NCSA)
 **/

#ifndef MODULES_BACKEND_PASS_LIBRARYFUNCTIONS_HPP_
#define MODULES_BACKEND_PASS_LIBRARYFUNCTIONS_HPP_

#include <llvm/IR/Constants.h>
#include <llvm/IR/Function.h>
#include <llvm/IR/IRBuilder.h>
#include <llvm/IR/Instructions.h>
#include <llvm/IR/Metadata.h>
#include <llvm/IR/Module.h>
#include <llvm/Pass.h>
#include <llvm/Support/raw_ostream.h>

#include <iostream>
#include <sstream>
#include <stdexcept>
#include <string>
#include <vector>

using llvm::Constant;
using llvm::FunctionCallee;
using llvm::PointerType;
using llvm::Function;

class LibraryFunctions {
  FunctionCallee map2check_init;
  FunctionCallee map2check_exit;
  FunctionCallee map2check_track_bb;

 public:
  FunctionCallee getInitFunction() { return this->map2check_init; }
  FunctionCallee getTrackBBFunction() { return this->map2check_track_bb; }
  FunctionCallee getExitFunction() { return this->map2check_exit; }

  LibraryFunctions(Function* F, LLVMContext* Ctx) {
    this->map2check_init = F->getParent()->getOrInsertFunction(
        "map2check_init", Type::getVoidTy(*Ctx), Type::getInt32Ty(*Ctx));

    this->map2check_exit = F->getParent()->getOrInsertFunction(
        "map2check_success", Type::getVoidTy(*Ctx));

    this->map2check_exit = F->getParent()->getOrInsertFunction(
        "map2check_success", Type::getVoidTy(*Ctx));

    this->map2check_track_bb = F->getParent()->getOrInsertFunction(
        "map2check_track_bb", Type::getVoidTy(*Ctx), Type::getInt32Ty(*Ctx),
        PointerType::get(*Ctx, 0));
  }
};

#endif  // MODULES_BACKEND_PASS_LIBRARYFUNCTIONS_HPP_