WebAssemblyCallIndirectFixup.cpp revision 341825
1311116Sdim//===-- WebAssemblyCallIndirectFixup.cpp - Fix call_indirects -------------===// 2311116Sdim// 3311116Sdim// The LLVM Compiler Infrastructure 4311116Sdim// 5311116Sdim// This file is distributed under the University of Illinois Open Source 6311116Sdim// License. See LICENSE.TXT for details. 7311116Sdim// 8311116Sdim//===----------------------------------------------------------------------===// 9311116Sdim/// 10311116Sdim/// \file 11341825Sdim/// This file converts pseudo call_indirect instructions into real 12311116Sdim/// call_indirects. 13311116Sdim/// 14311116Sdim/// The order of arguments for a call_indirect is the arguments to the function 15311116Sdim/// call, followed by the function pointer. There's no natural way to express 16311116Sdim/// a machineinstr with varargs followed by one more arg, so we express it as 17311116Sdim/// the function pointer followed by varargs, then rewrite it here. 18311116Sdim/// 19311116Sdim/// We need to rewrite the order of the arguments on the machineinstrs 20311116Sdim/// themselves so that register stackification knows the order they'll be 21311116Sdim/// executed in. 22311116Sdim/// 23311116Sdim//===----------------------------------------------------------------------===// 24311116Sdim 25321369Sdim#include "MCTargetDesc/WebAssemblyMCTargetDesc.h" // for WebAssembly::ARGUMENT_* 26311116Sdim#include "WebAssembly.h" 27311116Sdim#include "WebAssemblyMachineFunctionInfo.h" 28311116Sdim#include "WebAssemblySubtarget.h" 29311116Sdim#include "llvm/Analysis/AliasAnalysis.h" 30327952Sdim#include "llvm/CodeGen/LiveIntervals.h" 31311116Sdim#include "llvm/CodeGen/MachineBlockFrequencyInfo.h" 32311116Sdim#include "llvm/CodeGen/MachineDominators.h" 33311116Sdim#include "llvm/CodeGen/MachineInstrBuilder.h" 34311116Sdim#include "llvm/CodeGen/MachineRegisterInfo.h" 35311116Sdim#include "llvm/CodeGen/Passes.h" 36311116Sdim#include "llvm/Support/Debug.h" 37311116Sdim#include "llvm/Support/raw_ostream.h" 38311116Sdimusing namespace llvm; 39311116Sdim 40311116Sdim#define DEBUG_TYPE "wasm-call-indirect-fixup" 41311116Sdim 42311116Sdimnamespace { 43311116Sdimclass WebAssemblyCallIndirectFixup final : public MachineFunctionPass { 44311116Sdim StringRef getPassName() const override { 45311116Sdim return "WebAssembly CallIndirect Fixup"; 46311116Sdim } 47311116Sdim 48311116Sdim bool runOnMachineFunction(MachineFunction &MF) override; 49311116Sdim 50311116Sdimpublic: 51311116Sdim static char ID; // Pass identification, replacement for typeid 52311116Sdim WebAssemblyCallIndirectFixup() : MachineFunctionPass(ID) {} 53311116Sdim}; 54311116Sdim} // end anonymous namespace 55311116Sdim 56311116Sdimchar WebAssemblyCallIndirectFixup::ID = 0; 57341825SdimINITIALIZE_PASS(WebAssemblyCallIndirectFixup, DEBUG_TYPE, 58341825Sdim "Rewrite call_indirect argument orderings", false, false) 59341825Sdim 60311116SdimFunctionPass *llvm::createWebAssemblyCallIndirectFixup() { 61311116Sdim return new WebAssemblyCallIndirectFixup(); 62311116Sdim} 63311116Sdim 64311116Sdimstatic unsigned GetNonPseudoCallIndirectOpcode(const MachineInstr &MI) { 65311116Sdim switch (MI.getOpcode()) { 66311116Sdim using namespace WebAssembly; 67311116Sdim case PCALL_INDIRECT_VOID: return CALL_INDIRECT_VOID; 68311116Sdim case PCALL_INDIRECT_I32: return CALL_INDIRECT_I32; 69311116Sdim case PCALL_INDIRECT_I64: return CALL_INDIRECT_I64; 70311116Sdim case PCALL_INDIRECT_F32: return CALL_INDIRECT_F32; 71311116Sdim case PCALL_INDIRECT_F64: return CALL_INDIRECT_F64; 72311116Sdim case PCALL_INDIRECT_v16i8: return CALL_INDIRECT_v16i8; 73311116Sdim case PCALL_INDIRECT_v8i16: return CALL_INDIRECT_v8i16; 74311116Sdim case PCALL_INDIRECT_v4i32: return CALL_INDIRECT_v4i32; 75311116Sdim case PCALL_INDIRECT_v4f32: return CALL_INDIRECT_v4f32; 76311116Sdim default: return INSTRUCTION_LIST_END; 77311116Sdim } 78311116Sdim} 79311116Sdim 80311116Sdimstatic bool IsPseudoCallIndirect(const MachineInstr &MI) { 81311116Sdim return GetNonPseudoCallIndirectOpcode(MI) != 82311116Sdim WebAssembly::INSTRUCTION_LIST_END; 83311116Sdim} 84311116Sdim 85311116Sdimbool WebAssemblyCallIndirectFixup::runOnMachineFunction(MachineFunction &MF) { 86341825Sdim LLVM_DEBUG(dbgs() << "********** Fixing up CALL_INDIRECTs **********\n" 87341825Sdim << MF.getName() << '\n'); 88311116Sdim 89311116Sdim bool Changed = false; 90311116Sdim const WebAssemblyInstrInfo *TII = 91311116Sdim MF.getSubtarget<WebAssemblySubtarget>().getInstrInfo(); 92311116Sdim 93311116Sdim for (MachineBasicBlock &MBB : MF) { 94311116Sdim for (MachineInstr &MI : MBB) { 95311116Sdim if (IsPseudoCallIndirect(MI)) { 96341825Sdim LLVM_DEBUG(dbgs() << "Found call_indirect: " << MI << '\n'); 97311116Sdim 98311116Sdim // Rewrite pseudo to non-pseudo 99311116Sdim const MCInstrDesc &Desc = TII->get(GetNonPseudoCallIndirectOpcode(MI)); 100311116Sdim MI.setDesc(Desc); 101311116Sdim 102311116Sdim // Rewrite argument order 103321369Sdim SmallVector<MachineOperand, 8> Ops; 104311116Sdim 105321369Sdim // Set up a placeholder for the type signature immediate. 106321369Sdim Ops.push_back(MachineOperand::CreateImm(0)); 107321369Sdim 108311116Sdim // Set up the flags immediate, which currently has no defined flags 109311116Sdim // so it's always zero. 110321369Sdim Ops.push_back(MachineOperand::CreateImm(0)); 111311116Sdim 112321369Sdim for (const MachineOperand &MO : 113321369Sdim make_range(MI.operands_begin() + 114321369Sdim MI.getDesc().getNumDefs() + 1, 115321369Sdim MI.operands_begin() + 116321369Sdim MI.getNumExplicitOperands())) 117321369Sdim Ops.push_back(MO); 118321369Sdim Ops.push_back(MI.getOperand(MI.getDesc().getNumDefs())); 119311116Sdim 120321369Sdim // Replace the instructions operands. 121321369Sdim while (MI.getNumOperands() > MI.getDesc().getNumDefs()) 122321369Sdim MI.RemoveOperand(MI.getNumOperands() - 1); 123321369Sdim for (const MachineOperand &MO : Ops) 124321369Sdim MI.addOperand(MO); 125321369Sdim 126341825Sdim LLVM_DEBUG(dbgs() << " After transform: " << MI); 127311116Sdim Changed = true; 128311116Sdim } 129311116Sdim } 130311116Sdim } 131311116Sdim 132341825Sdim LLVM_DEBUG(dbgs() << "\nDone fixing up CALL_INDIRECTs\n\n"); 133311116Sdim 134311116Sdim return Changed; 135311116Sdim} 136311116Sdim 137