1 //===-- llvm/CodeGen/MachineModuleInfo.cpp ----------------------*- C++ -*-===// 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 #include "llvm/CodeGen/MachineModuleInfo.h" 10 #include "llvm/ADT/DenseMap.h" 11 #include "llvm/ADT/StringRef.h" 12 #include "llvm/CodeGen/MachineFunction.h" 13 #include "llvm/CodeGen/Passes.h" 14 #include "llvm/IR/Constants.h" 15 #include "llvm/IR/DiagnosticInfo.h" 16 #include "llvm/IR/LLVMContext.h" 17 #include "llvm/IR/Module.h" 18 #include "llvm/InitializePasses.h" 19 #include "llvm/MC/MCContext.h" 20 #include "llvm/Pass.h" 21 #include "llvm/Support/CommandLine.h" 22 #include "llvm/Support/ErrorHandling.h" 23 #include "llvm/Target/TargetLoweringObjectFile.h" 24 #include "llvm/Target/TargetMachine.h" 25 #include <algorithm> 26 #include <cassert> 27 #include <memory> 28 #include <utility> 29 #include <vector> 30 31 using namespace llvm; 32 using namespace llvm::dwarf; 33 34 static cl::opt<bool> 35 DisableDebugInfoPrinting("disable-debug-info-print", cl::Hidden, 36 cl::desc("Disable debug info printing")); 37 38 // Out of line virtual method. 39 MachineModuleInfoImpl::~MachineModuleInfoImpl() = default; 40 41 void MachineModuleInfo::initialize() { 42 ObjFileMMI = nullptr; 43 CurCallSite = 0; 44 NextFnNum = 0; 45 UsesMSVCFloatingPoint = false; 46 DbgInfoAvailable = false; 47 } 48 49 void MachineModuleInfo::finalize() { 50 Context.reset(); 51 // We don't clear the ExternalContext. 52 53 delete ObjFileMMI; 54 ObjFileMMI = nullptr; 55 } 56 57 MachineModuleInfo::MachineModuleInfo(MachineModuleInfo &&MMI) 58 : TM(std::move(MMI.TM)), 59 Context(TM.getTargetTriple(), TM.getMCAsmInfo(), TM.getMCRegisterInfo(), 60 TM.getMCSubtargetInfo(), nullptr, &TM.Options.MCOptions, false), 61 MachineFunctions(std::move(MMI.MachineFunctions)) { 62 Context.setObjectFileInfo(TM.getObjFileLowering()); 63 ObjFileMMI = MMI.ObjFileMMI; 64 CurCallSite = MMI.CurCallSite; 65 ExternalContext = MMI.ExternalContext; 66 TheModule = MMI.TheModule; 67 } 68 69 MachineModuleInfo::MachineModuleInfo(const LLVMTargetMachine *TM) 70 : TM(*TM), Context(TM->getTargetTriple(), TM->getMCAsmInfo(), 71 TM->getMCRegisterInfo(), TM->getMCSubtargetInfo(), 72 nullptr, &TM->Options.MCOptions, false) { 73 Context.setObjectFileInfo(TM->getObjFileLowering()); 74 initialize(); 75 } 76 77 MachineModuleInfo::MachineModuleInfo(const LLVMTargetMachine *TM, 78 MCContext *ExtContext) 79 : TM(*TM), Context(TM->getTargetTriple(), TM->getMCAsmInfo(), 80 TM->getMCRegisterInfo(), TM->getMCSubtargetInfo(), 81 nullptr, &TM->Options.MCOptions, false), 82 ExternalContext(ExtContext) { 83 Context.setObjectFileInfo(TM->getObjFileLowering()); 84 initialize(); 85 } 86 87 MachineModuleInfo::~MachineModuleInfo() { finalize(); } 88 89 MachineFunction * 90 MachineModuleInfo::getMachineFunction(const Function &F) const { 91 auto I = MachineFunctions.find(&F); 92 return I != MachineFunctions.end() ? I->second.get() : nullptr; 93 } 94 95 MachineFunction &MachineModuleInfo::getOrCreateMachineFunction(Function &F) { 96 // Shortcut for the common case where a sequence of MachineFunctionPasses 97 // all query for the same Function. 98 if (LastRequest == &F) 99 return *LastResult; 100 101 auto I = MachineFunctions.insert( 102 std::make_pair(&F, std::unique_ptr<MachineFunction>())); 103 MachineFunction *MF; 104 if (I.second) { 105 // No pre-existing machine function, create a new one. 106 const TargetSubtargetInfo &STI = *TM.getSubtargetImpl(F); 107 MF = new MachineFunction(F, TM, STI, NextFnNum++, *this); 108 MF->initTargetMachineFunctionInfo(STI); 109 110 // MRI callback for target specific initializations. 111 TM.registerMachineRegisterInfoCallback(*MF); 112 113 // Update the set entry. 114 I.first->second.reset(MF); 115 } else { 116 MF = I.first->second.get(); 117 } 118 119 LastRequest = &F; 120 LastResult = MF; 121 return *MF; 122 } 123 124 void MachineModuleInfo::deleteMachineFunctionFor(Function &F) { 125 MachineFunctions.erase(&F); 126 LastRequest = nullptr; 127 LastResult = nullptr; 128 } 129 130 void MachineModuleInfo::insertFunction(const Function &F, 131 std::unique_ptr<MachineFunction> &&MF) { 132 auto I = MachineFunctions.insert(std::make_pair(&F, std::move(MF))); 133 assert(I.second && "machine function already mapped"); 134 (void)I; 135 } 136 137 namespace { 138 139 /// This pass frees the MachineFunction object associated with a Function. 140 class FreeMachineFunction : public FunctionPass { 141 public: 142 static char ID; 143 144 FreeMachineFunction() : FunctionPass(ID) {} 145 146 void getAnalysisUsage(AnalysisUsage &AU) const override { 147 AU.addRequired<MachineModuleInfoWrapperPass>(); 148 AU.addPreserved<MachineModuleInfoWrapperPass>(); 149 } 150 151 bool runOnFunction(Function &F) override { 152 MachineModuleInfo &MMI = 153 getAnalysis<MachineModuleInfoWrapperPass>().getMMI(); 154 MMI.deleteMachineFunctionFor(F); 155 return true; 156 } 157 158 StringRef getPassName() const override { 159 return "Free MachineFunction"; 160 } 161 }; 162 163 } // end anonymous namespace 164 165 char FreeMachineFunction::ID; 166 167 FunctionPass *llvm::createFreeMachineFunctionPass() { 168 return new FreeMachineFunction(); 169 } 170 171 MachineModuleInfoWrapperPass::MachineModuleInfoWrapperPass( 172 const LLVMTargetMachine *TM) 173 : ImmutablePass(ID), MMI(TM) { 174 initializeMachineModuleInfoWrapperPassPass(*PassRegistry::getPassRegistry()); 175 } 176 177 MachineModuleInfoWrapperPass::MachineModuleInfoWrapperPass( 178 const LLVMTargetMachine *TM, MCContext *ExtContext) 179 : ImmutablePass(ID), MMI(TM, ExtContext) { 180 initializeMachineModuleInfoWrapperPassPass(*PassRegistry::getPassRegistry()); 181 } 182 183 // Handle the Pass registration stuff necessary to use DataLayout's. 184 INITIALIZE_PASS(MachineModuleInfoWrapperPass, "machinemoduleinfo", 185 "Machine Module Information", false, false) 186 char MachineModuleInfoWrapperPass::ID = 0; 187 188 static unsigned getLocCookie(const SMDiagnostic &SMD, const SourceMgr &SrcMgr, 189 std::vector<const MDNode *> &LocInfos) { 190 // Look up a LocInfo for the buffer this diagnostic is coming from. 191 unsigned BufNum = SrcMgr.FindBufferContainingLoc(SMD.getLoc()); 192 const MDNode *LocInfo = nullptr; 193 if (BufNum > 0 && BufNum <= LocInfos.size()) 194 LocInfo = LocInfos[BufNum - 1]; 195 196 // If the inline asm had metadata associated with it, pull out a location 197 // cookie corresponding to which line the error occurred on. 198 unsigned LocCookie = 0; 199 if (LocInfo) { 200 unsigned ErrorLine = SMD.getLineNo() - 1; 201 if (ErrorLine >= LocInfo->getNumOperands()) 202 ErrorLine = 0; 203 204 if (LocInfo->getNumOperands() != 0) 205 if (const ConstantInt *CI = 206 mdconst::dyn_extract<ConstantInt>(LocInfo->getOperand(ErrorLine))) 207 LocCookie = CI->getZExtValue(); 208 } 209 210 return LocCookie; 211 } 212 213 bool MachineModuleInfoWrapperPass::doInitialization(Module &M) { 214 MMI.initialize(); 215 MMI.TheModule = &M; 216 // FIXME: Do this for new pass manager. 217 LLVMContext &Ctx = M.getContext(); 218 MMI.getContext().setDiagnosticHandler( 219 [&Ctx, &M](const SMDiagnostic &SMD, bool IsInlineAsm, 220 const SourceMgr &SrcMgr, 221 std::vector<const MDNode *> &LocInfos) { 222 unsigned LocCookie = 0; 223 if (IsInlineAsm) 224 LocCookie = getLocCookie(SMD, SrcMgr, LocInfos); 225 Ctx.diagnose( 226 DiagnosticInfoSrcMgr(SMD, M.getName(), IsInlineAsm, LocCookie)); 227 }); 228 MMI.DbgInfoAvailable = !DisableDebugInfoPrinting && 229 !M.debug_compile_units().empty(); 230 return false; 231 } 232 233 bool MachineModuleInfoWrapperPass::doFinalization(Module &M) { 234 MMI.finalize(); 235 return false; 236 } 237 238 AnalysisKey MachineModuleAnalysis::Key; 239 240 MachineModuleInfo MachineModuleAnalysis::run(Module &M, 241 ModuleAnalysisManager &) { 242 MachineModuleInfo MMI(TM); 243 MMI.TheModule = &M; 244 MMI.DbgInfoAvailable = !DisableDebugInfoPrinting && 245 !M.debug_compile_units().empty(); 246 return MMI; 247 } 248