x86call.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 <vector>
 
#include <llvm/ADT/ArrayRef.h>
#include <llvm/ADT/iterator_range.h>
 
#include "core/inst.h"
#include "emitter/call_lowering.h"
 
class CallInst;
class InvokeInst;
class ReturnInst;
class RaiseInst;
class Func;
 
 
class X86Call : public CallLowering {
public:
  X86Call(const Func *func) : CallLowering(func) {}
  X86Call(const CallSite *inst) : CallLowering(inst) {}
  X86Call(const ReturnInst *inst) : CallLowering(inst) {}
  X86Call(const RaiseInst *inst) : CallLowering(inst) {}
  X86Call(const LandingPadInst *inst) : CallLowering(inst) {}
 
  unsigned GetFrameSize() const override
  {
    return llvm::alignTo(stack_, maxAlign_);
  }
 
  llvm::ArrayRef<unsigned> GetUnusedGPRs() const;
  llvm::ArrayRef<unsigned> GetUsedGPRs() const;
  llvm::ArrayRef<unsigned> GetUnusedXMMs() const;
  llvm::ArrayRef<unsigned> GetUsedXMMs() const;
 
protected:
  void AssignArgReg(ArgLoc &loc, llvm::MVT t, llvm::Register reg);
  void AssignArgStack(ArgLoc &loc, llvm::MVT t, unsigned size);
  void AssignArgByVal(ArgLoc &loc, llvm::MVT t, unsigned size, llvm::Align a);
  void AssignRetReg(RetLoc &loc, llvm::MVT t, llvm::Register reg);
 
protected:
  virtual llvm::ArrayRef<unsigned> GetGPRs() const = 0;
  virtual llvm::ArrayRef<unsigned> GetXMMs() const = 0;
 
protected:
  unsigned argRegs_ = 0;
  unsigned argXMMs_ = 0;
  unsigned retRegs_ = 0;
  unsigned retXMMs_ = 0;
  unsigned retFPs_ = 0;
  unsigned stack_ = 0;
  llvm::Align maxAlign_ = llvm::Align(8);
};
 
class X86_32Call final : public X86Call {
public:
  X86_32Call(const Func *func) : X86Call(func) { AnalyseFunc(func); }
  X86_32Call(const CallSite *inst) : X86Call(inst) { AnalyseCall(inst); }
  X86_32Call(const ReturnInst *inst) : X86Call(inst) { AnalyseReturn(inst); }
  X86_32Call(const RaiseInst *inst) : X86Call(inst) { AnalyseRaise(inst); }
  X86_32Call(const LandingPadInst *inst) : X86Call(inst) { AnalysePad(inst); }
 
protected:
  llvm::ArrayRef<unsigned> GetGPRs() const override;
  llvm::ArrayRef<unsigned> GetXMMs() const override;
 
private:
  void AssignArgC(unsigned i, FlaggedType type) override;
  void AssignArgOCaml(unsigned i, FlaggedType type) override;
  void AssignArgOCamlAlloc(unsigned i, FlaggedType type) override;
  void AssignArgOCamlGc(unsigned i, FlaggedType type) override;
  void AssignArgXen(unsigned i, FlaggedType type) override;
  void AssignArgMultiboot(unsigned i, FlaggedType type) override;
  void AssignArgWin64(unsigned i, FlaggedType type) override;
 
  void AssignRetC(unsigned i, FlaggedType type) override;
  void AssignRetOCaml(unsigned i, FlaggedType type) override;
  void AssignRetOCamlAlloc(unsigned i, FlaggedType type) override;
  void AssignRetOCamlGc(unsigned i, FlaggedType type) override;
  void AssignRetXen(unsigned i, FlaggedType type) override;
  void AssignRetWin64(unsigned i, FlaggedType type) override;
};
 
class X86_64Call final : public X86Call {
public:
  X86_64Call(const Func *func) : X86Call(func) { AnalyseFunc(func); }
  X86_64Call(const CallSite *inst) : X86Call(inst) { AnalyseCall(inst); }
  X86_64Call(const ReturnInst *inst) : X86Call(inst) { AnalyseReturn(inst); }
  X86_64Call(const RaiseInst *inst) : X86Call(inst) { AnalyseRaise(inst); }
  X86_64Call(const LandingPadInst *inst) : X86Call(inst) { AnalysePad(inst); }
 
protected:
  llvm::ArrayRef<unsigned> GetGPRs() const override;
  llvm::ArrayRef<unsigned> GetXMMs() const override;
 
private:
  void AssignArgC(unsigned i, FlaggedType type) override;
  void AssignArgOCaml(unsigned i, FlaggedType type) override;
  void AssignArgOCamlAlloc(unsigned i, FlaggedType type) override;
  void AssignArgOCamlGc(unsigned i, FlaggedType type) override;
  void AssignArgXen(unsigned i, FlaggedType type) override;
  void AssignArgMultiboot(unsigned i, FlaggedType type) override;
  void AssignArgWin64(unsigned i, FlaggedType type) override;
 
  void AssignRetC(unsigned i, FlaggedType type) override;
  void AssignRetOCaml(unsigned i, FlaggedType type) override;
  void AssignRetOCamlAlloc(unsigned i, FlaggedType type) override;
  void AssignRetOCamlGc(unsigned i, FlaggedType type) override;
  void AssignRetXen(unsigned i, FlaggedType type) override;
  void AssignRetWin64(unsigned i, FlaggedType type) override;
};