driver.h

// This file if part of the llir-opt project.
// Licensing information can be found in the LICENSE file.
// (C) 2018 Nandor Licker. All rights reserved.
 
#pragma once
 
#include <optional>
 
#include <llvm/ADT/Triple.h>
#include <llvm/Option/ArgList.h>
#include <llvm/Support/Error.h>
 
#include "linker.h"
 
class Prog;
 
 
 
enum class OptLevel {
  O0,
  O1,
  O2,
  O3,
  O4,
  Os
};
 
llvm::Error MakeError(llvm::Twine msg);
 
class Driver final {
public:
  Driver(
      const llvm::Triple &triple,
      const llvm::Triple &base,
      llvm::opt::InputArgList &args
  );
 
  ~Driver();
 
  llvm::Error Link();
 
private:
  llvm::Expected<Linker::Archive>
  LoadArchive(llvm::MemoryBufferRef buffer);
 
  llvm::Expected<std::optional<Linker::Archive>>
  TryLoadArchive(const std::string &path);
 
private:
  enum class OutputType {
    EXE,
    OBJ,
    ASM,
    LLIR,
    LLBC
  };
  OutputType GetOutputType();
 
  llvm::Error Output(OutputType type, Prog &prog);
  // Run the optimiser on a binary.
  llvm::Error RunOpt(
      llvm::StringRef input,
      llvm::StringRef output,
      OutputType type
  );
  llvm::Error RunExecutable(
      llvm::StringRef exe,
      llvm::ArrayRef<llvm::StringRef> args
  );
 
private:
  const llvm::Triple &llirTriple_;
  const llvm::Triple &baseTriple_;
 
  llvm::opt::InputArgList &args_;
 
  const std::string output_;
  bool shared_;
  bool static_;
  bool noShared_;
  bool relocatable_;
  bool exportDynamic_;
  bool ehFrameHdr_;
  std::string targetCPU_;
  std::string targetABI_;
  std::string targetFS_;
  std::string entry_;
  OptLevel optLevel_;
  std::vector<std::string> libraryPaths_;
 
  std::vector<std::unique_ptr<llvm::MemoryBuffer>> buffers_;
  std::vector<llvm::sys::fs::TempFile> tempFiles_;
  std::vector<std::string> externLibs_;
  llvm::opt::ArgStringList forwarded_;
};