llvm/lib/Target/WebAssembly/WebAssemblyMCLowerPrePass.cpp

   1 //===-- WebAssemblyMCLowerPrePass.cpp - Prepare for MC lower --------------===//
   2 //
   3 // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
   4 // See https://llvm.org/LICENSE.txt for license information.
   5 // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
   6 //
   7 //===----------------------------------------------------------------------===//
   8 ///
   9 /// \file
  10 /// Some information in MC lowering / asm printing gets generated as
  11 /// instructions get emitted, but may be necessary at the start, such as for
  12 /// .globaltype declarations. This pass collects this information.
  13 ///
  14 //===----------------------------------------------------------------------===//
  15
  16 #include "MCTargetDesc/WebAssemblyMCTargetDesc.h"
  17 #include "Utils/WebAssemblyUtilities.h"
  18 #include "WebAssembly.h"
  19 #include "WebAssemblyMachineFunctionInfo.h"
  20 #include "WebAssemblySubtarget.h"
  21 #include "llvm/ADT/SCCIterator.h"
  22 #include "llvm/CodeGen/MachineFrameInfo.h"
  23 #include "llvm/CodeGen/MachineFunction.h"
  24 #include "llvm/CodeGen/MachineInstrBuilder.h"
  25 #include "llvm/CodeGen/MachineLoopInfo.h"
  26 #include "llvm/CodeGen/MachineModuleInfoImpls.h"
  27 #include "llvm/CodeGen/MachineRegisterInfo.h"
  28 #include "llvm/CodeGen/Passes.h"
  29 #include "llvm/Support/Debug.h"
  30 #include "llvm/Support/raw_ostream.h"
  31 using namespace llvm;
  32
  33 #define DEBUG_TYPE "wasm-mclower-prepass"
  34
  35 namespace {
  36 class WebAssemblyMCLowerPrePass final : public MachineFunctionPass {
  37   StringRef getPassName() const override {
  38     return "WebAssembly MC Lower Pre Pass";
  39   }
  40
  41   void getAnalysisUsage(AnalysisUsage &AU) const override {
  42     AU.setPreservesCFG();
  43     MachineFunctionPass::getAnalysisUsage(AU);
  44   }
  45
  46   bool runOnMachineFunction(MachineFunction &MF) override;
  47
  48 public:
  49   static char ID; // Pass identification, replacement for typeid
  50   WebAssemblyMCLowerPrePass() : MachineFunctionPass(ID) {}
  51 };
  52 } // end anonymous namespace
  53
  54 char WebAssemblyMCLowerPrePass::ID = 0;
  55 INITIALIZE_PASS(
  56     WebAssemblyMCLowerPrePass, DEBUG_TYPE,
  57     "Collects information ahead of time for MC lowering",
  58     false, false)
  59
  60 FunctionPass *llvm::createWebAssemblyMCLowerPrePass() {
  61   return new WebAssemblyMCLowerPrePass();
  62 }
  63
  64 bool WebAssemblyMCLowerPrePass::runOnMachineFunction(MachineFunction &MF) {
  65   LLVM_DEBUG(dbgs() << "********** MC Lower Pre Pass **********\n"
  66                        "********** Function: "
  67                     << MF.getName() << '\n');
  68
  69   MachineModuleInfo &MMI = MF.getMMI();
  70   MachineModuleInfoWasm &MMIW = MMI.getObjFileInfo<MachineModuleInfoWasm>();
  71
  72   for (MachineBasicBlock &MBB : MF) {
  73     for (auto &MI : MBB) {
  74       // FIXME: what should all be filtered out beyond these?
  75       if (MI.isDebugInstr() || MI.isInlineAsm())
  76         continue;
  77       for (MachineOperand &MO : MI.uses()) {
  78         if (MO.isSymbol()) {
  79           MMIW.MachineSymbolsUsed.insert(MO.getSymbolName());
  80         }
  81       }
  82     }
  83   }
  84
  85   return true;
  86 }