x86isel.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 <set>
#include <unordered_map>
 
#include <llvm/ADT/DenseMap.h>
#include <llvm/Analysis/OptimizationRemarkEmitter.h>
#include <llvm/CodeGen/MachineBasicBlock.h>
#include <llvm/IR/Module.h>
#include <llvm/IR/Type.h>
#include <llvm/Pass.h>
#include <llvm/Target/X86/X86InstrInfo.h>
#include <llvm/Target/X86/X86ISelDAGToDAG.h>
#include <llvm/Target/X86/X86MachineFunctionInfo.h>
#include <llvm/Target/X86/X86RegisterInfo.h>
#include <llvm/Target/X86/X86Subtarget.h>
#include <llvm/Target/X86/X86TargetMachine.h>
 
#include "core/insts.h"
#include "emitter/isel.h"
#include "emitter/x86/x86call.h"
 
class Data;
class Func;
class Inst;
class Prog;
 
 
 
class X86Matcher : public llvm::X86DAGMatcher {
public:
  X86Matcher(
      llvm::X86TargetMachine &tm,
      llvm::CodeGenOpt::Level ol,
      llvm::MachineFunction &mf
  );
  ~X86Matcher();
 
  llvm::SelectionDAG &GetDAG() const { return *CurDAG; }
  const llvm::X86TargetMachine &getTargetMachine() const override { return tm_; }
 
private:
  llvm::X86TargetMachine &tm_;
};
 
class X86ISel final : public ISel {
public:
  static char ID;
 
  X86ISel(
      Target &target,
      llvm::X86TargetMachine &tm,
      llvm::TargetLibraryInfo &libInfo,
      const Prog &prog,
      llvm::CodeGenOpt::Level ol,
      bool shared
  );
 
private:
  void Lower(llvm::MachineFunction &mf) override
  {
    m_.reset(new X86Matcher(tm_, ol_, mf));
  }
 
  llvm::SDValue LowerCallee(ConstRef<Inst> inst);
 
  void LowerArch(const Inst *inst) override;
 
  void LowerSyscall(const SyscallInst *inst) override;
  void LowerClone(const CloneInst *inst) override;
  void LowerReturn(const ReturnInst *inst) override;
  void LowerRaise(const RaiseInst *inst) override;
  void LowerSpawn(const SpawnInst *inst) override;
  void LowerLandingPad(const LandingPadInst *inst) override;
  void LowerSet(const SetInst *inst) override;
 
  void LowerFPUControl(
      unsigned opcode,
      unsigned bytes,
      bool store,
      const Inst *inst
  );
 
  void Lower(const BarrierInst *inst, unsigned op);
  void Lower(const X86_CpuIdInst *inst);
  void Lower(const X86_XchgInst *inst);
  void Lower(const X86_CmpXchgInst *inst);
  void Lower(const X86_FnClExInst *inst);
  void Lower(const X86_RdTscInst *inst);
  void Lower(const X86_InInst *inst);
  void Lower(const X86_OutInst *inst);
  void Lower(const X86_WrMsrInst *inst);
  void Lower(const X86_RdMsrInst *inst);
  void Lower(const X86_PauseInst *inst);
  void Lower(const X86_YieldInst *inst);
  void Lower(const X86_StiInst *inst);
  void Lower(const X86_CliInst *inst);
  void Lower(const X86_HltInst *inst);
  void Lower(const X86_SpinInst *inst);
  void Lower(const X86_LgdtInst *inst);
  void Lower(const X86_LidtInst *inst);
  void Lower(const X86_LtrInst *inst);
  void Lower(const X86_XSaveOptInst *inst);
  void Lower(const X86_XRestoreInst *inst);
 
  void LowerArguments(bool hasVAStart) override;
  void LowerVASetup(const X86Call &lowering);
 
  virtual bool Finalize(llvm::MachineFunction &MF) override;
 
private:
  SDValue GetRegArch(Register reg) override;
  void LowerSetSP(SDValue value);
  void LowerSetReg(const char *code, MVT type, SDValue value);
  void LowerRaise(SDValue spVal, SDValue pcVal, SDValue glue);
  void LowerContextMask(bool store, unsigned op, SDValue addr, SDValue mask);
  void LowerSetCs(SDValue value);
 
private:
  llvm::SelectionDAG &GetDAG() const override { return m_->GetDAG(); }
  void PreprocessISelDAG() override { m_->PreprocessISelDAG(); }
  void PostprocessISelDAG() override { m_->PostprocessISelDAG(); }
  void Select(SDNode *node) override { m_->Select(node); }
 
  llvm::Register GetStackRegister() const override;
  llvm::Register GetBaseRegister() const;
 
private:
  void LowerCallSite(llvm::SDValue chain, const CallSite *call) override;
 
private:
  llvm::X86TargetMachine &tm_;
  std::unique_ptr<X86Matcher> m_;
  llvm::Function *trampoline_;
  bool shared_;
};