xref: /openbsd-src/gnu/llvm/clang/lib/CodeGen/CodeGenAction.cpp (revision 12c855180aad702bbcca06e0398d774beeafb155)
1e5dd7070Spatrick //===--- CodeGenAction.cpp - LLVM Code Generation Frontend Action ---------===//
2e5dd7070Spatrick //
3e5dd7070Spatrick // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4e5dd7070Spatrick // See https://llvm.org/LICENSE.txt for license information.
5e5dd7070Spatrick // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6e5dd7070Spatrick //
7e5dd7070Spatrick //===----------------------------------------------------------------------===//
8e5dd7070Spatrick 
9e5dd7070Spatrick #include "clang/CodeGen/CodeGenAction.h"
10e5dd7070Spatrick #include "CodeGenModule.h"
11e5dd7070Spatrick #include "CoverageMappingGen.h"
12e5dd7070Spatrick #include "MacroPPCallbacks.h"
13e5dd7070Spatrick #include "clang/AST/ASTConsumer.h"
14e5dd7070Spatrick #include "clang/AST/ASTContext.h"
15e5dd7070Spatrick #include "clang/AST/DeclCXX.h"
16e5dd7070Spatrick #include "clang/AST/DeclGroup.h"
17e5dd7070Spatrick #include "clang/Basic/DiagnosticFrontend.h"
18e5dd7070Spatrick #include "clang/Basic/FileManager.h"
19e5dd7070Spatrick #include "clang/Basic/LangStandard.h"
20e5dd7070Spatrick #include "clang/Basic/SourceManager.h"
21e5dd7070Spatrick #include "clang/Basic/TargetInfo.h"
22e5dd7070Spatrick #include "clang/CodeGen/BackendUtil.h"
23e5dd7070Spatrick #include "clang/CodeGen/ModuleBuilder.h"
24e5dd7070Spatrick #include "clang/Driver/DriverDiagnostic.h"
25e5dd7070Spatrick #include "clang/Frontend/CompilerInstance.h"
26e5dd7070Spatrick #include "clang/Frontend/FrontendDiagnostic.h"
27e5dd7070Spatrick #include "clang/Lex/Preprocessor.h"
28*12c85518Srobert #include "llvm/ADT/Hashing.h"
29e5dd7070Spatrick #include "llvm/Bitcode/BitcodeReader.h"
30e5dd7070Spatrick #include "llvm/CodeGen/MachineOptimizationRemarkEmitter.h"
31*12c85518Srobert #include "llvm/Demangle/Demangle.h"
32e5dd7070Spatrick #include "llvm/IR/DebugInfo.h"
33e5dd7070Spatrick #include "llvm/IR/DiagnosticInfo.h"
34e5dd7070Spatrick #include "llvm/IR/DiagnosticPrinter.h"
35e5dd7070Spatrick #include "llvm/IR/GlobalValue.h"
36e5dd7070Spatrick #include "llvm/IR/LLVMContext.h"
37ec727ea7Spatrick #include "llvm/IR/LLVMRemarkStreamer.h"
38e5dd7070Spatrick #include "llvm/IR/Module.h"
39e5dd7070Spatrick #include "llvm/IRReader/IRReader.h"
40a9ac8606Spatrick #include "llvm/LTO/LTOBackend.h"
41e5dd7070Spatrick #include "llvm/Linker/Linker.h"
42e5dd7070Spatrick #include "llvm/Pass.h"
43e5dd7070Spatrick #include "llvm/Support/MemoryBuffer.h"
44e5dd7070Spatrick #include "llvm/Support/SourceMgr.h"
45e5dd7070Spatrick #include "llvm/Support/TimeProfiler.h"
46e5dd7070Spatrick #include "llvm/Support/Timer.h"
47e5dd7070Spatrick #include "llvm/Support/ToolOutputFile.h"
48e5dd7070Spatrick #include "llvm/Support/YAMLTraits.h"
49e5dd7070Spatrick #include "llvm/Transforms/IPO/Internalize.h"
50e5dd7070Spatrick 
51e5dd7070Spatrick #include <memory>
52*12c85518Srobert #include <optional>
53e5dd7070Spatrick using namespace clang;
54e5dd7070Spatrick using namespace llvm;
55e5dd7070Spatrick 
56*12c85518Srobert #define DEBUG_TYPE "codegenaction"
57*12c85518Srobert 
58e5dd7070Spatrick namespace clang {
59e5dd7070Spatrick   class BackendConsumer;
60e5dd7070Spatrick   class ClangDiagnosticHandler final : public DiagnosticHandler {
61e5dd7070Spatrick   public:
ClangDiagnosticHandler(const CodeGenOptions & CGOpts,BackendConsumer * BCon)62e5dd7070Spatrick     ClangDiagnosticHandler(const CodeGenOptions &CGOpts, BackendConsumer *BCon)
63e5dd7070Spatrick         : CodeGenOpts(CGOpts), BackendCon(BCon) {}
64e5dd7070Spatrick 
65e5dd7070Spatrick     bool handleDiagnostics(const DiagnosticInfo &DI) override;
66e5dd7070Spatrick 
isAnalysisRemarkEnabled(StringRef PassName) const67e5dd7070Spatrick     bool isAnalysisRemarkEnabled(StringRef PassName) const override {
68a9ac8606Spatrick       return CodeGenOpts.OptimizationRemarkAnalysis.patternMatches(PassName);
69e5dd7070Spatrick     }
isMissedOptRemarkEnabled(StringRef PassName) const70e5dd7070Spatrick     bool isMissedOptRemarkEnabled(StringRef PassName) const override {
71a9ac8606Spatrick       return CodeGenOpts.OptimizationRemarkMissed.patternMatches(PassName);
72e5dd7070Spatrick     }
isPassedOptRemarkEnabled(StringRef PassName) const73e5dd7070Spatrick     bool isPassedOptRemarkEnabled(StringRef PassName) const override {
74a9ac8606Spatrick       return CodeGenOpts.OptimizationRemark.patternMatches(PassName);
75e5dd7070Spatrick     }
76e5dd7070Spatrick 
isAnyRemarkEnabled() const77e5dd7070Spatrick     bool isAnyRemarkEnabled() const override {
78a9ac8606Spatrick       return CodeGenOpts.OptimizationRemarkAnalysis.hasValidPattern() ||
79a9ac8606Spatrick              CodeGenOpts.OptimizationRemarkMissed.hasValidPattern() ||
80a9ac8606Spatrick              CodeGenOpts.OptimizationRemark.hasValidPattern();
81e5dd7070Spatrick     }
82e5dd7070Spatrick 
83e5dd7070Spatrick   private:
84e5dd7070Spatrick     const CodeGenOptions &CodeGenOpts;
85e5dd7070Spatrick     BackendConsumer *BackendCon;
86e5dd7070Spatrick   };
87e5dd7070Spatrick 
reportOptRecordError(Error E,DiagnosticsEngine & Diags,const CodeGenOptions CodeGenOpts)88e5dd7070Spatrick   static void reportOptRecordError(Error E, DiagnosticsEngine &Diags,
89e5dd7070Spatrick                                    const CodeGenOptions CodeGenOpts) {
90e5dd7070Spatrick     handleAllErrors(
91e5dd7070Spatrick         std::move(E),
92ec727ea7Spatrick       [&](const LLVMRemarkSetupFileError &E) {
93e5dd7070Spatrick           Diags.Report(diag::err_cannot_open_file)
94e5dd7070Spatrick               << CodeGenOpts.OptRecordFile << E.message();
95e5dd7070Spatrick         },
96ec727ea7Spatrick       [&](const LLVMRemarkSetupPatternError &E) {
97e5dd7070Spatrick           Diags.Report(diag::err_drv_optimization_remark_pattern)
98e5dd7070Spatrick               << E.message() << CodeGenOpts.OptRecordPasses;
99e5dd7070Spatrick         },
100ec727ea7Spatrick       [&](const LLVMRemarkSetupFormatError &E) {
101e5dd7070Spatrick           Diags.Report(diag::err_drv_optimization_remark_format)
102e5dd7070Spatrick               << CodeGenOpts.OptRecordFormat;
103e5dd7070Spatrick         });
104e5dd7070Spatrick     }
105e5dd7070Spatrick 
106e5dd7070Spatrick   class BackendConsumer : public ASTConsumer {
107e5dd7070Spatrick     using LinkModule = CodeGenAction::LinkModule;
108e5dd7070Spatrick 
109e5dd7070Spatrick     virtual void anchor();
110e5dd7070Spatrick     DiagnosticsEngine &Diags;
111e5dd7070Spatrick     BackendAction Action;
112e5dd7070Spatrick     const HeaderSearchOptions &HeaderSearchOpts;
113e5dd7070Spatrick     const CodeGenOptions &CodeGenOpts;
114e5dd7070Spatrick     const TargetOptions &TargetOpts;
115e5dd7070Spatrick     const LangOptions &LangOpts;
116e5dd7070Spatrick     std::unique_ptr<raw_pwrite_stream> AsmOutStream;
117e5dd7070Spatrick     ASTContext *Context;
118e5dd7070Spatrick 
119e5dd7070Spatrick     Timer LLVMIRGeneration;
120e5dd7070Spatrick     unsigned LLVMIRGenerationRefCount;
121e5dd7070Spatrick 
122e5dd7070Spatrick     /// True if we've finished generating IR. This prevents us from generating
123e5dd7070Spatrick     /// additional LLVM IR after emitting output in HandleTranslationUnit. This
124e5dd7070Spatrick     /// can happen when Clang plugins trigger additional AST deserialization.
125e5dd7070Spatrick     bool IRGenFinished = false;
126e5dd7070Spatrick 
127a9ac8606Spatrick     bool TimerIsEnabled = false;
128a9ac8606Spatrick 
129e5dd7070Spatrick     std::unique_ptr<CodeGenerator> Gen;
130e5dd7070Spatrick 
131e5dd7070Spatrick     SmallVector<LinkModule, 4> LinkModules;
132e5dd7070Spatrick 
133*12c85518Srobert     // A map from mangled names to their function's source location, used for
134*12c85518Srobert     // backend diagnostics as the Clang AST may be unavailable. We actually use
135*12c85518Srobert     // the mangled name's hash as the key because mangled names can be very
136*12c85518Srobert     // long and take up lots of space. Using a hash can cause name collision,
137*12c85518Srobert     // but that is rare and the consequences are pointing to a wrong source
138*12c85518Srobert     // location which is not severe. This is a vector instead of an actual map
139*12c85518Srobert     // because we optimize for time building this map rather than time
140*12c85518Srobert     // retrieving an entry, as backend diagnostics are uncommon.
141*12c85518Srobert     std::vector<std::pair<llvm::hash_code, FullSourceLoc>>
142*12c85518Srobert         ManglingFullSourceLocs;
143*12c85518Srobert 
144e5dd7070Spatrick     // This is here so that the diagnostic printer knows the module a diagnostic
145e5dd7070Spatrick     // refers to.
146e5dd7070Spatrick     llvm::Module *CurLinkModule = nullptr;
147e5dd7070Spatrick 
148e5dd7070Spatrick   public:
BackendConsumer(BackendAction Action,DiagnosticsEngine & Diags,IntrusiveRefCntPtr<llvm::vfs::FileSystem> FS,const HeaderSearchOptions & HeaderSearchOpts,const PreprocessorOptions & PPOpts,const CodeGenOptions & CodeGenOpts,const TargetOptions & TargetOpts,const LangOptions & LangOpts,const std::string & InFile,SmallVector<LinkModule,4> LinkModules,std::unique_ptr<raw_pwrite_stream> OS,LLVMContext & C,CoverageSourceInfo * CoverageInfo=nullptr)149e5dd7070Spatrick     BackendConsumer(BackendAction Action, DiagnosticsEngine &Diags,
150*12c85518Srobert                     IntrusiveRefCntPtr<llvm::vfs::FileSystem> FS,
151e5dd7070Spatrick                     const HeaderSearchOptions &HeaderSearchOpts,
152e5dd7070Spatrick                     const PreprocessorOptions &PPOpts,
153e5dd7070Spatrick                     const CodeGenOptions &CodeGenOpts,
154e5dd7070Spatrick                     const TargetOptions &TargetOpts,
155a9ac8606Spatrick                     const LangOptions &LangOpts, const std::string &InFile,
156e5dd7070Spatrick                     SmallVector<LinkModule, 4> LinkModules,
157e5dd7070Spatrick                     std::unique_ptr<raw_pwrite_stream> OS, LLVMContext &C,
158e5dd7070Spatrick                     CoverageSourceInfo *CoverageInfo = nullptr)
159e5dd7070Spatrick         : Diags(Diags), Action(Action), HeaderSearchOpts(HeaderSearchOpts),
160e5dd7070Spatrick           CodeGenOpts(CodeGenOpts), TargetOpts(TargetOpts), LangOpts(LangOpts),
161e5dd7070Spatrick           AsmOutStream(std::move(OS)), Context(nullptr),
162e5dd7070Spatrick           LLVMIRGeneration("irgen", "LLVM IR Generation Time"),
163e5dd7070Spatrick           LLVMIRGenerationRefCount(0),
164*12c85518Srobert           Gen(CreateLLVMCodeGen(Diags, InFile, std::move(FS), HeaderSearchOpts,
165*12c85518Srobert                                 PPOpts, CodeGenOpts, C, CoverageInfo)),
166e5dd7070Spatrick           LinkModules(std::move(LinkModules)) {
167a9ac8606Spatrick       TimerIsEnabled = CodeGenOpts.TimePasses;
168a9ac8606Spatrick       llvm::TimePassesIsEnabled = CodeGenOpts.TimePasses;
169a9ac8606Spatrick       llvm::TimePassesPerRun = CodeGenOpts.TimePassesPerRun;
170e5dd7070Spatrick     }
171e5dd7070Spatrick 
172e5dd7070Spatrick     // This constructor is used in installing an empty BackendConsumer
173e5dd7070Spatrick     // to use the clang diagnostic handler for IR input files. It avoids
174e5dd7070Spatrick     // initializing the OS field.
BackendConsumer(BackendAction Action,DiagnosticsEngine & Diags,IntrusiveRefCntPtr<llvm::vfs::FileSystem> FS,const HeaderSearchOptions & HeaderSearchOpts,const PreprocessorOptions & PPOpts,const CodeGenOptions & CodeGenOpts,const TargetOptions & TargetOpts,const LangOptions & LangOpts,llvm::Module * Module,SmallVector<LinkModule,4> LinkModules,LLVMContext & C,CoverageSourceInfo * CoverageInfo=nullptr)175e5dd7070Spatrick     BackendConsumer(BackendAction Action, DiagnosticsEngine &Diags,
176*12c85518Srobert                     IntrusiveRefCntPtr<llvm::vfs::FileSystem> FS,
177e5dd7070Spatrick                     const HeaderSearchOptions &HeaderSearchOpts,
178e5dd7070Spatrick                     const PreprocessorOptions &PPOpts,
179e5dd7070Spatrick                     const CodeGenOptions &CodeGenOpts,
180e5dd7070Spatrick                     const TargetOptions &TargetOpts,
181*12c85518Srobert                     const LangOptions &LangOpts, llvm::Module *Module,
182e5dd7070Spatrick                     SmallVector<LinkModule, 4> LinkModules, LLVMContext &C,
183e5dd7070Spatrick                     CoverageSourceInfo *CoverageInfo = nullptr)
184e5dd7070Spatrick         : Diags(Diags), Action(Action), HeaderSearchOpts(HeaderSearchOpts),
185e5dd7070Spatrick           CodeGenOpts(CodeGenOpts), TargetOpts(TargetOpts), LangOpts(LangOpts),
186e5dd7070Spatrick           Context(nullptr),
187e5dd7070Spatrick           LLVMIRGeneration("irgen", "LLVM IR Generation Time"),
188e5dd7070Spatrick           LLVMIRGenerationRefCount(0),
189*12c85518Srobert           Gen(CreateLLVMCodeGen(Diags, "", std::move(FS), HeaderSearchOpts,
190*12c85518Srobert                                 PPOpts, CodeGenOpts, C, CoverageInfo)),
191*12c85518Srobert           LinkModules(std::move(LinkModules)), CurLinkModule(Module) {
192a9ac8606Spatrick       TimerIsEnabled = CodeGenOpts.TimePasses;
193a9ac8606Spatrick       llvm::TimePassesIsEnabled = CodeGenOpts.TimePasses;
194a9ac8606Spatrick       llvm::TimePassesPerRun = CodeGenOpts.TimePassesPerRun;
195e5dd7070Spatrick     }
getModule() const196e5dd7070Spatrick     llvm::Module *getModule() const { return Gen->GetModule(); }
takeModule()197e5dd7070Spatrick     std::unique_ptr<llvm::Module> takeModule() {
198e5dd7070Spatrick       return std::unique_ptr<llvm::Module>(Gen->ReleaseModule());
199e5dd7070Spatrick     }
200e5dd7070Spatrick 
getCodeGenerator()201e5dd7070Spatrick     CodeGenerator *getCodeGenerator() { return Gen.get(); }
202e5dd7070Spatrick 
HandleCXXStaticMemberVarInstantiation(VarDecl * VD)203e5dd7070Spatrick     void HandleCXXStaticMemberVarInstantiation(VarDecl *VD) override {
204e5dd7070Spatrick       Gen->HandleCXXStaticMemberVarInstantiation(VD);
205e5dd7070Spatrick     }
206e5dd7070Spatrick 
Initialize(ASTContext & Ctx)207e5dd7070Spatrick     void Initialize(ASTContext &Ctx) override {
208e5dd7070Spatrick       assert(!Context && "initialized multiple times");
209e5dd7070Spatrick 
210e5dd7070Spatrick       Context = &Ctx;
211e5dd7070Spatrick 
212a9ac8606Spatrick       if (TimerIsEnabled)
213e5dd7070Spatrick         LLVMIRGeneration.startTimer();
214e5dd7070Spatrick 
215e5dd7070Spatrick       Gen->Initialize(Ctx);
216e5dd7070Spatrick 
217a9ac8606Spatrick       if (TimerIsEnabled)
218e5dd7070Spatrick         LLVMIRGeneration.stopTimer();
219e5dd7070Spatrick     }
220e5dd7070Spatrick 
HandleTopLevelDecl(DeclGroupRef D)221e5dd7070Spatrick     bool HandleTopLevelDecl(DeclGroupRef D) override {
222e5dd7070Spatrick       PrettyStackTraceDecl CrashInfo(*D.begin(), SourceLocation(),
223e5dd7070Spatrick                                      Context->getSourceManager(),
224e5dd7070Spatrick                                      "LLVM IR generation of declaration");
225e5dd7070Spatrick 
226e5dd7070Spatrick       // Recurse.
227a9ac8606Spatrick       if (TimerIsEnabled) {
228e5dd7070Spatrick         LLVMIRGenerationRefCount += 1;
229e5dd7070Spatrick         if (LLVMIRGenerationRefCount == 1)
230e5dd7070Spatrick           LLVMIRGeneration.startTimer();
231e5dd7070Spatrick       }
232e5dd7070Spatrick 
233e5dd7070Spatrick       Gen->HandleTopLevelDecl(D);
234e5dd7070Spatrick 
235a9ac8606Spatrick       if (TimerIsEnabled) {
236e5dd7070Spatrick         LLVMIRGenerationRefCount -= 1;
237e5dd7070Spatrick         if (LLVMIRGenerationRefCount == 0)
238e5dd7070Spatrick           LLVMIRGeneration.stopTimer();
239e5dd7070Spatrick       }
240e5dd7070Spatrick 
241e5dd7070Spatrick       return true;
242e5dd7070Spatrick     }
243e5dd7070Spatrick 
HandleInlineFunctionDefinition(FunctionDecl * D)244e5dd7070Spatrick     void HandleInlineFunctionDefinition(FunctionDecl *D) override {
245e5dd7070Spatrick       PrettyStackTraceDecl CrashInfo(D, SourceLocation(),
246e5dd7070Spatrick                                      Context->getSourceManager(),
247e5dd7070Spatrick                                      "LLVM IR generation of inline function");
248a9ac8606Spatrick       if (TimerIsEnabled)
249e5dd7070Spatrick         LLVMIRGeneration.startTimer();
250e5dd7070Spatrick 
251e5dd7070Spatrick       Gen->HandleInlineFunctionDefinition(D);
252e5dd7070Spatrick 
253a9ac8606Spatrick       if (TimerIsEnabled)
254e5dd7070Spatrick         LLVMIRGeneration.stopTimer();
255e5dd7070Spatrick     }
256e5dd7070Spatrick 
HandleInterestingDecl(DeclGroupRef D)257e5dd7070Spatrick     void HandleInterestingDecl(DeclGroupRef D) override {
258e5dd7070Spatrick       // Ignore interesting decls from the AST reader after IRGen is finished.
259e5dd7070Spatrick       if (!IRGenFinished)
260e5dd7070Spatrick         HandleTopLevelDecl(D);
261e5dd7070Spatrick     }
262e5dd7070Spatrick 
263e5dd7070Spatrick     // Links each entry in LinkModules into our module.  Returns true on error.
LinkInModules()264e5dd7070Spatrick     bool LinkInModules() {
265e5dd7070Spatrick       for (auto &LM : LinkModules) {
266e5dd7070Spatrick         if (LM.PropagateAttrs)
267a9ac8606Spatrick           for (Function &F : *LM.Module) {
268a9ac8606Spatrick             // Skip intrinsics. Keep consistent with how intrinsics are created
269a9ac8606Spatrick             // in LLVM IR.
270a9ac8606Spatrick             if (F.isIntrinsic())
271a9ac8606Spatrick               continue;
272ec727ea7Spatrick             Gen->CGM().addDefaultFunctionDefinitionAttributes(F);
273a9ac8606Spatrick           }
274e5dd7070Spatrick 
275e5dd7070Spatrick         CurLinkModule = LM.Module.get();
276e5dd7070Spatrick 
277e5dd7070Spatrick         bool Err;
278e5dd7070Spatrick         if (LM.Internalize) {
279e5dd7070Spatrick           Err = Linker::linkModules(
280e5dd7070Spatrick               *getModule(), std::move(LM.Module), LM.LinkFlags,
281e5dd7070Spatrick               [](llvm::Module &M, const llvm::StringSet<> &GVS) {
282e5dd7070Spatrick                 internalizeModule(M, [&GVS](const llvm::GlobalValue &GV) {
283e5dd7070Spatrick                   return !GV.hasName() || (GVS.count(GV.getName()) == 0);
284e5dd7070Spatrick                 });
285e5dd7070Spatrick               });
286e5dd7070Spatrick         } else {
287e5dd7070Spatrick           Err = Linker::linkModules(*getModule(), std::move(LM.Module),
288e5dd7070Spatrick                                     LM.LinkFlags);
289e5dd7070Spatrick         }
290e5dd7070Spatrick 
291e5dd7070Spatrick         if (Err)
292e5dd7070Spatrick           return true;
293e5dd7070Spatrick       }
294e5dd7070Spatrick       return false; // success
295e5dd7070Spatrick     }
296e5dd7070Spatrick 
HandleTranslationUnit(ASTContext & C)297e5dd7070Spatrick     void HandleTranslationUnit(ASTContext &C) override {
298e5dd7070Spatrick       {
299e5dd7070Spatrick         llvm::TimeTraceScope TimeScope("Frontend");
300e5dd7070Spatrick         PrettyStackTraceString CrashInfo("Per-file LLVM IR generation");
301a9ac8606Spatrick         if (TimerIsEnabled) {
302e5dd7070Spatrick           LLVMIRGenerationRefCount += 1;
303e5dd7070Spatrick           if (LLVMIRGenerationRefCount == 1)
304e5dd7070Spatrick             LLVMIRGeneration.startTimer();
305e5dd7070Spatrick         }
306e5dd7070Spatrick 
307e5dd7070Spatrick         Gen->HandleTranslationUnit(C);
308e5dd7070Spatrick 
309a9ac8606Spatrick         if (TimerIsEnabled) {
310e5dd7070Spatrick           LLVMIRGenerationRefCount -= 1;
311e5dd7070Spatrick           if (LLVMIRGenerationRefCount == 0)
312e5dd7070Spatrick             LLVMIRGeneration.stopTimer();
313e5dd7070Spatrick         }
314e5dd7070Spatrick 
315e5dd7070Spatrick         IRGenFinished = true;
316e5dd7070Spatrick       }
317e5dd7070Spatrick 
318e5dd7070Spatrick       // Silently ignore if we weren't initialized for some reason.
319e5dd7070Spatrick       if (!getModule())
320e5dd7070Spatrick         return;
321e5dd7070Spatrick 
322e5dd7070Spatrick       LLVMContext &Ctx = getModule()->getContext();
323e5dd7070Spatrick       std::unique_ptr<DiagnosticHandler> OldDiagnosticHandler =
324e5dd7070Spatrick           Ctx.getDiagnosticHandler();
325e5dd7070Spatrick       Ctx.setDiagnosticHandler(std::make_unique<ClangDiagnosticHandler>(
326e5dd7070Spatrick         CodeGenOpts, this));
327e5dd7070Spatrick 
328e5dd7070Spatrick       Expected<std::unique_ptr<llvm::ToolOutputFile>> OptRecordFileOrErr =
329ec727ea7Spatrick           setupLLVMOptimizationRemarks(
330e5dd7070Spatrick               Ctx, CodeGenOpts.OptRecordFile, CodeGenOpts.OptRecordPasses,
331e5dd7070Spatrick               CodeGenOpts.OptRecordFormat, CodeGenOpts.DiagnosticsWithHotness,
332e5dd7070Spatrick               CodeGenOpts.DiagnosticsHotnessThreshold);
333e5dd7070Spatrick 
334e5dd7070Spatrick       if (Error E = OptRecordFileOrErr.takeError()) {
335e5dd7070Spatrick         reportOptRecordError(std::move(E), Diags, CodeGenOpts);
336e5dd7070Spatrick         return;
337e5dd7070Spatrick       }
338e5dd7070Spatrick 
339e5dd7070Spatrick       std::unique_ptr<llvm::ToolOutputFile> OptRecordFile =
340e5dd7070Spatrick           std::move(*OptRecordFileOrErr);
341e5dd7070Spatrick 
342e5dd7070Spatrick       if (OptRecordFile &&
343e5dd7070Spatrick           CodeGenOpts.getProfileUse() != CodeGenOptions::ProfileNone)
344e5dd7070Spatrick         Ctx.setDiagnosticsHotnessRequested(true);
345e5dd7070Spatrick 
346*12c85518Srobert       if (CodeGenOpts.MisExpect) {
347*12c85518Srobert         Ctx.setMisExpectWarningRequested(true);
348*12c85518Srobert       }
349*12c85518Srobert 
350*12c85518Srobert       if (CodeGenOpts.DiagnosticsMisExpectTolerance) {
351*12c85518Srobert         Ctx.setDiagnosticsMisExpectTolerance(
352*12c85518Srobert             CodeGenOpts.DiagnosticsMisExpectTolerance);
353*12c85518Srobert       }
354*12c85518Srobert 
355e5dd7070Spatrick       // Link each LinkModule into our module.
356e5dd7070Spatrick       if (LinkInModules())
357e5dd7070Spatrick         return;
358e5dd7070Spatrick 
359*12c85518Srobert       for (auto &F : getModule()->functions()) {
360*12c85518Srobert         if (const Decl *FD = Gen->GetDeclForMangledName(F.getName())) {
361*12c85518Srobert           auto Loc = FD->getASTContext().getFullLoc(FD->getLocation());
362*12c85518Srobert           // TODO: use a fast content hash when available.
363*12c85518Srobert           auto NameHash = llvm::hash_value(F.getName());
364*12c85518Srobert           ManglingFullSourceLocs.push_back(std::make_pair(NameHash, Loc));
365*12c85518Srobert         }
366*12c85518Srobert       }
367*12c85518Srobert 
368*12c85518Srobert       if (CodeGenOpts.ClearASTBeforeBackend) {
369*12c85518Srobert         LLVM_DEBUG(llvm::dbgs() << "Clearing AST...\n");
370*12c85518Srobert         // Access to the AST is no longer available after this.
371*12c85518Srobert         // Other things that the ASTContext manages are still available, e.g.
372*12c85518Srobert         // the SourceManager. It'd be nice if we could separate out all the
373*12c85518Srobert         // things in ASTContext used after this point and null out the
374*12c85518Srobert         // ASTContext, but too many various parts of the ASTContext are still
375*12c85518Srobert         // used in various parts.
376*12c85518Srobert         C.cleanup();
377*12c85518Srobert         C.getAllocator().Reset();
378*12c85518Srobert       }
379*12c85518Srobert 
380e5dd7070Spatrick       EmbedBitcode(getModule(), CodeGenOpts, llvm::MemoryBufferRef());
381e5dd7070Spatrick 
382e5dd7070Spatrick       EmitBackendOutput(Diags, HeaderSearchOpts, CodeGenOpts, TargetOpts,
383a9ac8606Spatrick                         LangOpts, C.getTargetInfo().getDataLayoutString(),
384e5dd7070Spatrick                         getModule(), Action, std::move(AsmOutStream));
385e5dd7070Spatrick 
386e5dd7070Spatrick       Ctx.setDiagnosticHandler(std::move(OldDiagnosticHandler));
387e5dd7070Spatrick 
388e5dd7070Spatrick       if (OptRecordFile)
389e5dd7070Spatrick         OptRecordFile->keep();
390e5dd7070Spatrick     }
391e5dd7070Spatrick 
HandleTagDeclDefinition(TagDecl * D)392e5dd7070Spatrick     void HandleTagDeclDefinition(TagDecl *D) override {
393e5dd7070Spatrick       PrettyStackTraceDecl CrashInfo(D, SourceLocation(),
394e5dd7070Spatrick                                      Context->getSourceManager(),
395e5dd7070Spatrick                                      "LLVM IR generation of declaration");
396e5dd7070Spatrick       Gen->HandleTagDeclDefinition(D);
397e5dd7070Spatrick     }
398e5dd7070Spatrick 
HandleTagDeclRequiredDefinition(const TagDecl * D)399e5dd7070Spatrick     void HandleTagDeclRequiredDefinition(const TagDecl *D) override {
400e5dd7070Spatrick       Gen->HandleTagDeclRequiredDefinition(D);
401e5dd7070Spatrick     }
402e5dd7070Spatrick 
CompleteTentativeDefinition(VarDecl * D)403e5dd7070Spatrick     void CompleteTentativeDefinition(VarDecl *D) override {
404e5dd7070Spatrick       Gen->CompleteTentativeDefinition(D);
405e5dd7070Spatrick     }
406e5dd7070Spatrick 
CompleteExternalDeclaration(VarDecl * D)407e5dd7070Spatrick     void CompleteExternalDeclaration(VarDecl *D) override {
408e5dd7070Spatrick       Gen->CompleteExternalDeclaration(D);
409e5dd7070Spatrick     }
410e5dd7070Spatrick 
AssignInheritanceModel(CXXRecordDecl * RD)411e5dd7070Spatrick     void AssignInheritanceModel(CXXRecordDecl *RD) override {
412e5dd7070Spatrick       Gen->AssignInheritanceModel(RD);
413e5dd7070Spatrick     }
414e5dd7070Spatrick 
HandleVTable(CXXRecordDecl * RD)415e5dd7070Spatrick     void HandleVTable(CXXRecordDecl *RD) override {
416e5dd7070Spatrick       Gen->HandleVTable(RD);
417e5dd7070Spatrick     }
418e5dd7070Spatrick 
419e5dd7070Spatrick     /// Get the best possible source location to represent a diagnostic that
420e5dd7070Spatrick     /// may have associated debug info.
421e5dd7070Spatrick     const FullSourceLoc
422e5dd7070Spatrick     getBestLocationFromDebugLoc(const llvm::DiagnosticInfoWithLocationBase &D,
423e5dd7070Spatrick                                 bool &BadDebugInfo, StringRef &Filename,
424e5dd7070Spatrick                                 unsigned &Line, unsigned &Column) const;
425e5dd7070Spatrick 
426*12c85518Srobert     std::optional<FullSourceLoc>
427*12c85518Srobert     getFunctionSourceLocation(const Function &F) const;
428*12c85518Srobert 
429e5dd7070Spatrick     void DiagnosticHandlerImpl(const llvm::DiagnosticInfo &DI);
430e5dd7070Spatrick     /// Specialized handler for InlineAsm diagnostic.
431e5dd7070Spatrick     /// \return True if the diagnostic has been successfully reported, false
432e5dd7070Spatrick     /// otherwise.
433e5dd7070Spatrick     bool InlineAsmDiagHandler(const llvm::DiagnosticInfoInlineAsm &D);
434a9ac8606Spatrick     /// Specialized handler for diagnostics reported using SMDiagnostic.
435a9ac8606Spatrick     void SrcMgrDiagHandler(const llvm::DiagnosticInfoSrcMgr &D);
436e5dd7070Spatrick     /// Specialized handler for StackSize diagnostic.
437e5dd7070Spatrick     /// \return True if the diagnostic has been successfully reported, false
438e5dd7070Spatrick     /// otherwise.
439e5dd7070Spatrick     bool StackSizeDiagHandler(const llvm::DiagnosticInfoStackSize &D);
440*12c85518Srobert     /// Specialized handler for ResourceLimit diagnostic.
441*12c85518Srobert     /// \return True if the diagnostic has been successfully reported, false
442*12c85518Srobert     /// otherwise.
443*12c85518Srobert     bool ResourceLimitDiagHandler(const llvm::DiagnosticInfoResourceLimit &D);
444*12c85518Srobert 
445e5dd7070Spatrick     /// Specialized handler for unsupported backend feature diagnostic.
446e5dd7070Spatrick     void UnsupportedDiagHandler(const llvm::DiagnosticInfoUnsupported &D);
447e5dd7070Spatrick     /// Specialized handlers for optimization remarks.
448e5dd7070Spatrick     /// Note that these handlers only accept remarks and they always handle
449e5dd7070Spatrick     /// them.
450e5dd7070Spatrick     void EmitOptimizationMessage(const llvm::DiagnosticInfoOptimizationBase &D,
451e5dd7070Spatrick                                  unsigned DiagID);
452e5dd7070Spatrick     void
453e5dd7070Spatrick     OptimizationRemarkHandler(const llvm::DiagnosticInfoOptimizationBase &D);
454e5dd7070Spatrick     void OptimizationRemarkHandler(
455e5dd7070Spatrick         const llvm::OptimizationRemarkAnalysisFPCommute &D);
456e5dd7070Spatrick     void OptimizationRemarkHandler(
457e5dd7070Spatrick         const llvm::OptimizationRemarkAnalysisAliasing &D);
458e5dd7070Spatrick     void OptimizationFailureHandler(
459e5dd7070Spatrick         const llvm::DiagnosticInfoOptimizationFailure &D);
460*12c85518Srobert     void DontCallDiagHandler(const DiagnosticInfoDontCall &D);
461*12c85518Srobert     /// Specialized handler for misexpect warnings.
462*12c85518Srobert     /// Note that misexpect remarks are emitted through ORE
463*12c85518Srobert     void MisExpectDiagHandler(const llvm::DiagnosticInfoMisExpect &D);
464e5dd7070Spatrick   };
465e5dd7070Spatrick 
anchor()466e5dd7070Spatrick   void BackendConsumer::anchor() {}
467e5dd7070Spatrick }
468e5dd7070Spatrick 
handleDiagnostics(const DiagnosticInfo & DI)469e5dd7070Spatrick bool ClangDiagnosticHandler::handleDiagnostics(const DiagnosticInfo &DI) {
470e5dd7070Spatrick   BackendCon->DiagnosticHandlerImpl(DI);
471e5dd7070Spatrick   return true;
472e5dd7070Spatrick }
473e5dd7070Spatrick 
474e5dd7070Spatrick /// ConvertBackendLocation - Convert a location in a temporary llvm::SourceMgr
475e5dd7070Spatrick /// buffer to be a valid FullSourceLoc.
ConvertBackendLocation(const llvm::SMDiagnostic & D,SourceManager & CSM)476e5dd7070Spatrick static FullSourceLoc ConvertBackendLocation(const llvm::SMDiagnostic &D,
477e5dd7070Spatrick                                             SourceManager &CSM) {
478e5dd7070Spatrick   // Get both the clang and llvm source managers.  The location is relative to
479e5dd7070Spatrick   // a memory buffer that the LLVM Source Manager is handling, we need to add
480e5dd7070Spatrick   // a copy to the Clang source manager.
481e5dd7070Spatrick   const llvm::SourceMgr &LSM = *D.getSourceMgr();
482e5dd7070Spatrick 
483e5dd7070Spatrick   // We need to copy the underlying LLVM memory buffer because llvm::SourceMgr
484e5dd7070Spatrick   // already owns its one and clang::SourceManager wants to own its one.
485e5dd7070Spatrick   const MemoryBuffer *LBuf =
486e5dd7070Spatrick   LSM.getMemoryBuffer(LSM.FindBufferContainingLoc(D.getLoc()));
487e5dd7070Spatrick 
488e5dd7070Spatrick   // Create the copy and transfer ownership to clang::SourceManager.
489e5dd7070Spatrick   // TODO: Avoid copying files into memory.
490e5dd7070Spatrick   std::unique_ptr<llvm::MemoryBuffer> CBuf =
491e5dd7070Spatrick       llvm::MemoryBuffer::getMemBufferCopy(LBuf->getBuffer(),
492e5dd7070Spatrick                                            LBuf->getBufferIdentifier());
493e5dd7070Spatrick   // FIXME: Keep a file ID map instead of creating new IDs for each location.
494e5dd7070Spatrick   FileID FID = CSM.createFileID(std::move(CBuf));
495e5dd7070Spatrick 
496e5dd7070Spatrick   // Translate the offset into the file.
497e5dd7070Spatrick   unsigned Offset = D.getLoc().getPointer() - LBuf->getBufferStart();
498e5dd7070Spatrick   SourceLocation NewLoc =
499e5dd7070Spatrick   CSM.getLocForStartOfFile(FID).getLocWithOffset(Offset);
500e5dd7070Spatrick   return FullSourceLoc(NewLoc, CSM);
501e5dd7070Spatrick }
502e5dd7070Spatrick 
503e5dd7070Spatrick #define ComputeDiagID(Severity, GroupName, DiagID)                             \
504e5dd7070Spatrick   do {                                                                         \
505e5dd7070Spatrick     switch (Severity) {                                                        \
506e5dd7070Spatrick     case llvm::DS_Error:                                                       \
507e5dd7070Spatrick       DiagID = diag::err_fe_##GroupName;                                       \
508e5dd7070Spatrick       break;                                                                   \
509e5dd7070Spatrick     case llvm::DS_Warning:                                                     \
510e5dd7070Spatrick       DiagID = diag::warn_fe_##GroupName;                                      \
511e5dd7070Spatrick       break;                                                                   \
512e5dd7070Spatrick     case llvm::DS_Remark:                                                      \
513e5dd7070Spatrick       llvm_unreachable("'remark' severity not expected");                      \
514e5dd7070Spatrick       break;                                                                   \
515e5dd7070Spatrick     case llvm::DS_Note:                                                        \
516e5dd7070Spatrick       DiagID = diag::note_fe_##GroupName;                                      \
517e5dd7070Spatrick       break;                                                                   \
518e5dd7070Spatrick     }                                                                          \
519e5dd7070Spatrick   } while (false)
520e5dd7070Spatrick 
521e5dd7070Spatrick #define ComputeDiagRemarkID(Severity, GroupName, DiagID)                       \
522e5dd7070Spatrick   do {                                                                         \
523e5dd7070Spatrick     switch (Severity) {                                                        \
524e5dd7070Spatrick     case llvm::DS_Error:                                                       \
525e5dd7070Spatrick       DiagID = diag::err_fe_##GroupName;                                       \
526e5dd7070Spatrick       break;                                                                   \
527e5dd7070Spatrick     case llvm::DS_Warning:                                                     \
528e5dd7070Spatrick       DiagID = diag::warn_fe_##GroupName;                                      \
529e5dd7070Spatrick       break;                                                                   \
530e5dd7070Spatrick     case llvm::DS_Remark:                                                      \
531e5dd7070Spatrick       DiagID = diag::remark_fe_##GroupName;                                    \
532e5dd7070Spatrick       break;                                                                   \
533e5dd7070Spatrick     case llvm::DS_Note:                                                        \
534e5dd7070Spatrick       DiagID = diag::note_fe_##GroupName;                                      \
535e5dd7070Spatrick       break;                                                                   \
536e5dd7070Spatrick     }                                                                          \
537e5dd7070Spatrick   } while (false)
538e5dd7070Spatrick 
SrcMgrDiagHandler(const llvm::DiagnosticInfoSrcMgr & DI)539a9ac8606Spatrick void BackendConsumer::SrcMgrDiagHandler(const llvm::DiagnosticInfoSrcMgr &DI) {
540a9ac8606Spatrick   const llvm::SMDiagnostic &D = DI.getSMDiag();
541a9ac8606Spatrick 
542a9ac8606Spatrick   unsigned DiagID;
543a9ac8606Spatrick   if (DI.isInlineAsmDiag())
544a9ac8606Spatrick     ComputeDiagID(DI.getSeverity(), inline_asm, DiagID);
545a9ac8606Spatrick   else
546a9ac8606Spatrick     ComputeDiagID(DI.getSeverity(), source_mgr, DiagID);
547a9ac8606Spatrick 
548a9ac8606Spatrick   // This is for the empty BackendConsumer that uses the clang diagnostic
549a9ac8606Spatrick   // handler for IR input files.
550a9ac8606Spatrick   if (!Context) {
551a9ac8606Spatrick     D.print(nullptr, llvm::errs());
552a9ac8606Spatrick     Diags.Report(DiagID).AddString("cannot compile inline asm");
553a9ac8606Spatrick     return;
554a9ac8606Spatrick   }
555a9ac8606Spatrick 
556a9ac8606Spatrick   // There are a couple of different kinds of errors we could get here.
557a9ac8606Spatrick   // First, we re-format the SMDiagnostic in terms of a clang diagnostic.
558a9ac8606Spatrick 
559a9ac8606Spatrick   // Strip "error: " off the start of the message string.
560a9ac8606Spatrick   StringRef Message = D.getMessage();
561a9ac8606Spatrick   (void)Message.consume_front("error: ");
562a9ac8606Spatrick 
563a9ac8606Spatrick   // If the SMDiagnostic has an inline asm source location, translate it.
564a9ac8606Spatrick   FullSourceLoc Loc;
565a9ac8606Spatrick   if (D.getLoc() != SMLoc())
566a9ac8606Spatrick     Loc = ConvertBackendLocation(D, Context->getSourceManager());
567a9ac8606Spatrick 
568a9ac8606Spatrick   // If this problem has clang-level source location information, report the
569a9ac8606Spatrick   // issue in the source with a note showing the instantiated
570a9ac8606Spatrick   // code.
571a9ac8606Spatrick   if (DI.isInlineAsmDiag()) {
572a9ac8606Spatrick     SourceLocation LocCookie =
573a9ac8606Spatrick         SourceLocation::getFromRawEncoding(DI.getLocCookie());
574a9ac8606Spatrick     if (LocCookie.isValid()) {
575a9ac8606Spatrick       Diags.Report(LocCookie, DiagID).AddString(Message);
576a9ac8606Spatrick 
577a9ac8606Spatrick       if (D.getLoc().isValid()) {
578a9ac8606Spatrick         DiagnosticBuilder B = Diags.Report(Loc, diag::note_fe_inline_asm_here);
579a9ac8606Spatrick         // Convert the SMDiagnostic ranges into SourceRange and attach them
580a9ac8606Spatrick         // to the diagnostic.
581a9ac8606Spatrick         for (const std::pair<unsigned, unsigned> &Range : D.getRanges()) {
582a9ac8606Spatrick           unsigned Column = D.getColumnNo();
583a9ac8606Spatrick           B << SourceRange(Loc.getLocWithOffset(Range.first - Column),
584a9ac8606Spatrick                            Loc.getLocWithOffset(Range.second - Column));
585a9ac8606Spatrick         }
586a9ac8606Spatrick       }
587a9ac8606Spatrick       return;
588a9ac8606Spatrick     }
589a9ac8606Spatrick   }
590a9ac8606Spatrick 
591a9ac8606Spatrick   // Otherwise, report the backend issue as occurring in the generated .s file.
592a9ac8606Spatrick   // If Loc is invalid, we still need to report the issue, it just gets no
593a9ac8606Spatrick   // location info.
594a9ac8606Spatrick   Diags.Report(Loc, DiagID).AddString(Message);
595a9ac8606Spatrick }
596a9ac8606Spatrick 
597e5dd7070Spatrick bool
InlineAsmDiagHandler(const llvm::DiagnosticInfoInlineAsm & D)598e5dd7070Spatrick BackendConsumer::InlineAsmDiagHandler(const llvm::DiagnosticInfoInlineAsm &D) {
599e5dd7070Spatrick   unsigned DiagID;
600e5dd7070Spatrick   ComputeDiagID(D.getSeverity(), inline_asm, DiagID);
601e5dd7070Spatrick   std::string Message = D.getMsgStr().str();
602e5dd7070Spatrick 
603e5dd7070Spatrick   // If this problem has clang-level source location information, report the
604e5dd7070Spatrick   // issue as being a problem in the source with a note showing the instantiated
605e5dd7070Spatrick   // code.
606e5dd7070Spatrick   SourceLocation LocCookie =
607e5dd7070Spatrick       SourceLocation::getFromRawEncoding(D.getLocCookie());
608e5dd7070Spatrick   if (LocCookie.isValid())
609e5dd7070Spatrick     Diags.Report(LocCookie, DiagID).AddString(Message);
610e5dd7070Spatrick   else {
611e5dd7070Spatrick     // Otherwise, report the backend diagnostic as occurring in the generated
612e5dd7070Spatrick     // .s file.
613e5dd7070Spatrick     // If Loc is invalid, we still need to report the diagnostic, it just gets
614e5dd7070Spatrick     // no location info.
615e5dd7070Spatrick     FullSourceLoc Loc;
616e5dd7070Spatrick     Diags.Report(Loc, DiagID).AddString(Message);
617e5dd7070Spatrick   }
618e5dd7070Spatrick   // We handled all the possible severities.
619e5dd7070Spatrick   return true;
620e5dd7070Spatrick }
621e5dd7070Spatrick 
622e5dd7070Spatrick bool
StackSizeDiagHandler(const llvm::DiagnosticInfoStackSize & D)623e5dd7070Spatrick BackendConsumer::StackSizeDiagHandler(const llvm::DiagnosticInfoStackSize &D) {
624e5dd7070Spatrick   if (D.getSeverity() != llvm::DS_Warning)
625e5dd7070Spatrick     // For now, the only support we have for StackSize diagnostic is warning.
626e5dd7070Spatrick     // We do not know how to format other severities.
627e5dd7070Spatrick     return false;
628e5dd7070Spatrick 
629*12c85518Srobert   auto Loc = getFunctionSourceLocation(D.getFunction());
630*12c85518Srobert   if (!Loc)
631*12c85518Srobert     return false;
632*12c85518Srobert 
633*12c85518Srobert   Diags.Report(*Loc, diag::warn_fe_frame_larger_than)
634*12c85518Srobert       << D.getStackSize()
635*12c85518Srobert       << D.getStackLimit()
636*12c85518Srobert       << llvm::demangle(D.getFunction().getName().str());
637e5dd7070Spatrick   return true;
638e5dd7070Spatrick }
639e5dd7070Spatrick 
ResourceLimitDiagHandler(const llvm::DiagnosticInfoResourceLimit & D)640*12c85518Srobert bool BackendConsumer::ResourceLimitDiagHandler(
641*12c85518Srobert     const llvm::DiagnosticInfoResourceLimit &D) {
642*12c85518Srobert   auto Loc = getFunctionSourceLocation(D.getFunction());
643*12c85518Srobert   if (!Loc)
644e5dd7070Spatrick     return false;
645*12c85518Srobert   unsigned DiagID = diag::err_fe_backend_resource_limit;
646*12c85518Srobert   ComputeDiagID(D.getSeverity(), backend_resource_limit, DiagID);
647*12c85518Srobert 
648*12c85518Srobert   Diags.Report(*Loc, DiagID)
649*12c85518Srobert       << D.getResourceName() << D.getResourceSize() << D.getResourceLimit()
650*12c85518Srobert       << llvm::demangle(D.getFunction().getName().str());
651*12c85518Srobert   return true;
652e5dd7070Spatrick }
653e5dd7070Spatrick 
getBestLocationFromDebugLoc(const llvm::DiagnosticInfoWithLocationBase & D,bool & BadDebugInfo,StringRef & Filename,unsigned & Line,unsigned & Column) const654e5dd7070Spatrick const FullSourceLoc BackendConsumer::getBestLocationFromDebugLoc(
655e5dd7070Spatrick     const llvm::DiagnosticInfoWithLocationBase &D, bool &BadDebugInfo,
656e5dd7070Spatrick     StringRef &Filename, unsigned &Line, unsigned &Column) const {
657e5dd7070Spatrick   SourceManager &SourceMgr = Context->getSourceManager();
658e5dd7070Spatrick   FileManager &FileMgr = SourceMgr.getFileManager();
659e5dd7070Spatrick   SourceLocation DILoc;
660e5dd7070Spatrick 
661e5dd7070Spatrick   if (D.isLocationAvailable()) {
662e5dd7070Spatrick     D.getLocation(Filename, Line, Column);
663e5dd7070Spatrick     if (Line > 0) {
664e5dd7070Spatrick       auto FE = FileMgr.getFile(Filename);
665e5dd7070Spatrick       if (!FE)
666e5dd7070Spatrick         FE = FileMgr.getFile(D.getAbsolutePath());
667e5dd7070Spatrick       if (FE) {
668e5dd7070Spatrick         // If -gcolumn-info was not used, Column will be 0. This upsets the
669e5dd7070Spatrick         // source manager, so pass 1 if Column is not set.
670e5dd7070Spatrick         DILoc = SourceMgr.translateFileLineCol(*FE, Line, Column ? Column : 1);
671e5dd7070Spatrick       }
672e5dd7070Spatrick     }
673e5dd7070Spatrick     BadDebugInfo = DILoc.isInvalid();
674e5dd7070Spatrick   }
675e5dd7070Spatrick 
676e5dd7070Spatrick   // If a location isn't available, try to approximate it using the associated
677e5dd7070Spatrick   // function definition. We use the definition's right brace to differentiate
678e5dd7070Spatrick   // from diagnostics that genuinely relate to the function itself.
679e5dd7070Spatrick   FullSourceLoc Loc(DILoc, SourceMgr);
680*12c85518Srobert   if (Loc.isInvalid()) {
681*12c85518Srobert     if (auto MaybeLoc = getFunctionSourceLocation(D.getFunction()))
682*12c85518Srobert       Loc = *MaybeLoc;
683*12c85518Srobert   }
684e5dd7070Spatrick 
685e5dd7070Spatrick   if (DILoc.isInvalid() && D.isLocationAvailable())
686e5dd7070Spatrick     // If we were not able to translate the file:line:col information
687e5dd7070Spatrick     // back to a SourceLocation, at least emit a note stating that
688e5dd7070Spatrick     // we could not translate this location. This can happen in the
689e5dd7070Spatrick     // case of #line directives.
690e5dd7070Spatrick     Diags.Report(Loc, diag::note_fe_backend_invalid_loc)
691e5dd7070Spatrick         << Filename << Line << Column;
692e5dd7070Spatrick 
693e5dd7070Spatrick   return Loc;
694e5dd7070Spatrick }
695e5dd7070Spatrick 
696*12c85518Srobert std::optional<FullSourceLoc>
getFunctionSourceLocation(const Function & F) const697*12c85518Srobert BackendConsumer::getFunctionSourceLocation(const Function &F) const {
698*12c85518Srobert   auto Hash = llvm::hash_value(F.getName());
699*12c85518Srobert   for (const auto &Pair : ManglingFullSourceLocs) {
700*12c85518Srobert     if (Pair.first == Hash)
701*12c85518Srobert       return Pair.second;
702*12c85518Srobert   }
703*12c85518Srobert   return std::nullopt;
704*12c85518Srobert }
705*12c85518Srobert 
UnsupportedDiagHandler(const llvm::DiagnosticInfoUnsupported & D)706e5dd7070Spatrick void BackendConsumer::UnsupportedDiagHandler(
707e5dd7070Spatrick     const llvm::DiagnosticInfoUnsupported &D) {
708ec727ea7Spatrick   // We only support warnings or errors.
709ec727ea7Spatrick   assert(D.getSeverity() == llvm::DS_Error ||
710ec727ea7Spatrick          D.getSeverity() == llvm::DS_Warning);
711e5dd7070Spatrick 
712e5dd7070Spatrick   StringRef Filename;
713e5dd7070Spatrick   unsigned Line, Column;
714e5dd7070Spatrick   bool BadDebugInfo = false;
715e5dd7070Spatrick   FullSourceLoc Loc;
716e5dd7070Spatrick   std::string Msg;
717e5dd7070Spatrick   raw_string_ostream MsgStream(Msg);
718e5dd7070Spatrick 
719e5dd7070Spatrick   // Context will be nullptr for IR input files, we will construct the diag
720e5dd7070Spatrick   // message from llvm::DiagnosticInfoUnsupported.
721e5dd7070Spatrick   if (Context != nullptr) {
722e5dd7070Spatrick     Loc = getBestLocationFromDebugLoc(D, BadDebugInfo, Filename, Line, Column);
723e5dd7070Spatrick     MsgStream << D.getMessage();
724e5dd7070Spatrick   } else {
725e5dd7070Spatrick     DiagnosticPrinterRawOStream DP(MsgStream);
726e5dd7070Spatrick     D.print(DP);
727e5dd7070Spatrick   }
728ec727ea7Spatrick 
729ec727ea7Spatrick   auto DiagType = D.getSeverity() == llvm::DS_Error
730ec727ea7Spatrick                       ? diag::err_fe_backend_unsupported
731ec727ea7Spatrick                       : diag::warn_fe_backend_unsupported;
732ec727ea7Spatrick   Diags.Report(Loc, DiagType) << MsgStream.str();
733e5dd7070Spatrick 
734e5dd7070Spatrick   if (BadDebugInfo)
735e5dd7070Spatrick     // If we were not able to translate the file:line:col information
736e5dd7070Spatrick     // back to a SourceLocation, at least emit a note stating that
737e5dd7070Spatrick     // we could not translate this location. This can happen in the
738e5dd7070Spatrick     // case of #line directives.
739e5dd7070Spatrick     Diags.Report(Loc, diag::note_fe_backend_invalid_loc)
740e5dd7070Spatrick         << Filename << Line << Column;
741e5dd7070Spatrick }
742e5dd7070Spatrick 
EmitOptimizationMessage(const llvm::DiagnosticInfoOptimizationBase & D,unsigned DiagID)743e5dd7070Spatrick void BackendConsumer::EmitOptimizationMessage(
744e5dd7070Spatrick     const llvm::DiagnosticInfoOptimizationBase &D, unsigned DiagID) {
745e5dd7070Spatrick   // We only support warnings and remarks.
746e5dd7070Spatrick   assert(D.getSeverity() == llvm::DS_Remark ||
747e5dd7070Spatrick          D.getSeverity() == llvm::DS_Warning);
748e5dd7070Spatrick 
749e5dd7070Spatrick   StringRef Filename;
750e5dd7070Spatrick   unsigned Line, Column;
751e5dd7070Spatrick   bool BadDebugInfo = false;
752e5dd7070Spatrick   FullSourceLoc Loc;
753e5dd7070Spatrick   std::string Msg;
754e5dd7070Spatrick   raw_string_ostream MsgStream(Msg);
755e5dd7070Spatrick 
756e5dd7070Spatrick   // Context will be nullptr for IR input files, we will construct the remark
757e5dd7070Spatrick   // message from llvm::DiagnosticInfoOptimizationBase.
758e5dd7070Spatrick   if (Context != nullptr) {
759e5dd7070Spatrick     Loc = getBestLocationFromDebugLoc(D, BadDebugInfo, Filename, Line, Column);
760e5dd7070Spatrick     MsgStream << D.getMsg();
761e5dd7070Spatrick   } else {
762e5dd7070Spatrick     DiagnosticPrinterRawOStream DP(MsgStream);
763e5dd7070Spatrick     D.print(DP);
764e5dd7070Spatrick   }
765e5dd7070Spatrick 
766e5dd7070Spatrick   if (D.getHotness())
767e5dd7070Spatrick     MsgStream << " (hotness: " << *D.getHotness() << ")";
768e5dd7070Spatrick 
769e5dd7070Spatrick   Diags.Report(Loc, DiagID)
770e5dd7070Spatrick       << AddFlagValue(D.getPassName())
771e5dd7070Spatrick       << MsgStream.str();
772e5dd7070Spatrick 
773e5dd7070Spatrick   if (BadDebugInfo)
774e5dd7070Spatrick     // If we were not able to translate the file:line:col information
775e5dd7070Spatrick     // back to a SourceLocation, at least emit a note stating that
776e5dd7070Spatrick     // we could not translate this location. This can happen in the
777e5dd7070Spatrick     // case of #line directives.
778e5dd7070Spatrick     Diags.Report(Loc, diag::note_fe_backend_invalid_loc)
779e5dd7070Spatrick         << Filename << Line << Column;
780e5dd7070Spatrick }
781e5dd7070Spatrick 
OptimizationRemarkHandler(const llvm::DiagnosticInfoOptimizationBase & D)782e5dd7070Spatrick void BackendConsumer::OptimizationRemarkHandler(
783e5dd7070Spatrick     const llvm::DiagnosticInfoOptimizationBase &D) {
784e5dd7070Spatrick   // Without hotness information, don't show noisy remarks.
785e5dd7070Spatrick   if (D.isVerbose() && !D.getHotness())
786e5dd7070Spatrick     return;
787e5dd7070Spatrick 
788e5dd7070Spatrick   if (D.isPassed()) {
789e5dd7070Spatrick     // Optimization remarks are active only if the -Rpass flag has a regular
790e5dd7070Spatrick     // expression that matches the name of the pass name in \p D.
791a9ac8606Spatrick     if (CodeGenOpts.OptimizationRemark.patternMatches(D.getPassName()))
792e5dd7070Spatrick       EmitOptimizationMessage(D, diag::remark_fe_backend_optimization_remark);
793e5dd7070Spatrick   } else if (D.isMissed()) {
794e5dd7070Spatrick     // Missed optimization remarks are active only if the -Rpass-missed
795e5dd7070Spatrick     // flag has a regular expression that matches the name of the pass
796e5dd7070Spatrick     // name in \p D.
797a9ac8606Spatrick     if (CodeGenOpts.OptimizationRemarkMissed.patternMatches(D.getPassName()))
798e5dd7070Spatrick       EmitOptimizationMessage(
799e5dd7070Spatrick           D, diag::remark_fe_backend_optimization_remark_missed);
800e5dd7070Spatrick   } else {
801e5dd7070Spatrick     assert(D.isAnalysis() && "Unknown remark type");
802e5dd7070Spatrick 
803e5dd7070Spatrick     bool ShouldAlwaysPrint = false;
804e5dd7070Spatrick     if (auto *ORA = dyn_cast<llvm::OptimizationRemarkAnalysis>(&D))
805e5dd7070Spatrick       ShouldAlwaysPrint = ORA->shouldAlwaysPrint();
806e5dd7070Spatrick 
807e5dd7070Spatrick     if (ShouldAlwaysPrint ||
808a9ac8606Spatrick         CodeGenOpts.OptimizationRemarkAnalysis.patternMatches(D.getPassName()))
809e5dd7070Spatrick       EmitOptimizationMessage(
810e5dd7070Spatrick           D, diag::remark_fe_backend_optimization_remark_analysis);
811e5dd7070Spatrick   }
812e5dd7070Spatrick }
813e5dd7070Spatrick 
OptimizationRemarkHandler(const llvm::OptimizationRemarkAnalysisFPCommute & D)814e5dd7070Spatrick void BackendConsumer::OptimizationRemarkHandler(
815e5dd7070Spatrick     const llvm::OptimizationRemarkAnalysisFPCommute &D) {
816e5dd7070Spatrick   // Optimization analysis remarks are active if the pass name is set to
817e5dd7070Spatrick   // llvm::DiagnosticInfo::AlwasyPrint or if the -Rpass-analysis flag has a
818e5dd7070Spatrick   // regular expression that matches the name of the pass name in \p D.
819e5dd7070Spatrick 
820e5dd7070Spatrick   if (D.shouldAlwaysPrint() ||
821a9ac8606Spatrick       CodeGenOpts.OptimizationRemarkAnalysis.patternMatches(D.getPassName()))
822e5dd7070Spatrick     EmitOptimizationMessage(
823e5dd7070Spatrick         D, diag::remark_fe_backend_optimization_remark_analysis_fpcommute);
824e5dd7070Spatrick }
825e5dd7070Spatrick 
OptimizationRemarkHandler(const llvm::OptimizationRemarkAnalysisAliasing & D)826e5dd7070Spatrick void BackendConsumer::OptimizationRemarkHandler(
827e5dd7070Spatrick     const llvm::OptimizationRemarkAnalysisAliasing &D) {
828e5dd7070Spatrick   // Optimization analysis remarks are active if the pass name is set to
829e5dd7070Spatrick   // llvm::DiagnosticInfo::AlwasyPrint or if the -Rpass-analysis flag has a
830e5dd7070Spatrick   // regular expression that matches the name of the pass name in \p D.
831e5dd7070Spatrick 
832e5dd7070Spatrick   if (D.shouldAlwaysPrint() ||
833a9ac8606Spatrick       CodeGenOpts.OptimizationRemarkAnalysis.patternMatches(D.getPassName()))
834e5dd7070Spatrick     EmitOptimizationMessage(
835e5dd7070Spatrick         D, diag::remark_fe_backend_optimization_remark_analysis_aliasing);
836e5dd7070Spatrick }
837e5dd7070Spatrick 
OptimizationFailureHandler(const llvm::DiagnosticInfoOptimizationFailure & D)838e5dd7070Spatrick void BackendConsumer::OptimizationFailureHandler(
839e5dd7070Spatrick     const llvm::DiagnosticInfoOptimizationFailure &D) {
840e5dd7070Spatrick   EmitOptimizationMessage(D, diag::warn_fe_backend_optimization_failure);
841e5dd7070Spatrick }
842e5dd7070Spatrick 
DontCallDiagHandler(const DiagnosticInfoDontCall & D)843*12c85518Srobert void BackendConsumer::DontCallDiagHandler(const DiagnosticInfoDontCall &D) {
844*12c85518Srobert   SourceLocation LocCookie =
845*12c85518Srobert       SourceLocation::getFromRawEncoding(D.getLocCookie());
846*12c85518Srobert 
847*12c85518Srobert   // FIXME: we can't yet diagnose indirect calls. When/if we can, we
848*12c85518Srobert   // should instead assert that LocCookie.isValid().
849*12c85518Srobert   if (!LocCookie.isValid())
850*12c85518Srobert     return;
851*12c85518Srobert 
852*12c85518Srobert   Diags.Report(LocCookie, D.getSeverity() == DiagnosticSeverity::DS_Error
853*12c85518Srobert                               ? diag::err_fe_backend_error_attr
854*12c85518Srobert                               : diag::warn_fe_backend_warning_attr)
855*12c85518Srobert       << llvm::demangle(D.getFunctionName().str()) << D.getNote();
856*12c85518Srobert }
857*12c85518Srobert 
MisExpectDiagHandler(const llvm::DiagnosticInfoMisExpect & D)858*12c85518Srobert void BackendConsumer::MisExpectDiagHandler(
859*12c85518Srobert     const llvm::DiagnosticInfoMisExpect &D) {
860*12c85518Srobert   StringRef Filename;
861*12c85518Srobert   unsigned Line, Column;
862*12c85518Srobert   bool BadDebugInfo = false;
863*12c85518Srobert   FullSourceLoc Loc =
864*12c85518Srobert       getBestLocationFromDebugLoc(D, BadDebugInfo, Filename, Line, Column);
865*12c85518Srobert 
866*12c85518Srobert   Diags.Report(Loc, diag::warn_profile_data_misexpect) << D.getMsg().str();
867*12c85518Srobert 
868*12c85518Srobert   if (BadDebugInfo)
869*12c85518Srobert     // If we were not able to translate the file:line:col information
870*12c85518Srobert     // back to a SourceLocation, at least emit a note stating that
871*12c85518Srobert     // we could not translate this location. This can happen in the
872*12c85518Srobert     // case of #line directives.
873*12c85518Srobert     Diags.Report(Loc, diag::note_fe_backend_invalid_loc)
874*12c85518Srobert         << Filename << Line << Column;
875*12c85518Srobert }
876*12c85518Srobert 
877e5dd7070Spatrick /// This function is invoked when the backend needs
878e5dd7070Spatrick /// to report something to the user.
DiagnosticHandlerImpl(const DiagnosticInfo & DI)879e5dd7070Spatrick void BackendConsumer::DiagnosticHandlerImpl(const DiagnosticInfo &DI) {
880e5dd7070Spatrick   unsigned DiagID = diag::err_fe_inline_asm;
881e5dd7070Spatrick   llvm::DiagnosticSeverity Severity = DI.getSeverity();
882e5dd7070Spatrick   // Get the diagnostic ID based.
883e5dd7070Spatrick   switch (DI.getKind()) {
884e5dd7070Spatrick   case llvm::DK_InlineAsm:
885e5dd7070Spatrick     if (InlineAsmDiagHandler(cast<DiagnosticInfoInlineAsm>(DI)))
886e5dd7070Spatrick       return;
887e5dd7070Spatrick     ComputeDiagID(Severity, inline_asm, DiagID);
888e5dd7070Spatrick     break;
889a9ac8606Spatrick   case llvm::DK_SrcMgr:
890a9ac8606Spatrick     SrcMgrDiagHandler(cast<DiagnosticInfoSrcMgr>(DI));
891a9ac8606Spatrick     return;
892e5dd7070Spatrick   case llvm::DK_StackSize:
893e5dd7070Spatrick     if (StackSizeDiagHandler(cast<DiagnosticInfoStackSize>(DI)))
894e5dd7070Spatrick       return;
895e5dd7070Spatrick     ComputeDiagID(Severity, backend_frame_larger_than, DiagID);
896e5dd7070Spatrick     break;
897*12c85518Srobert   case llvm::DK_ResourceLimit:
898*12c85518Srobert     if (ResourceLimitDiagHandler(cast<DiagnosticInfoResourceLimit>(DI)))
899e5dd7070Spatrick       return;
900*12c85518Srobert     ComputeDiagID(Severity, backend_resource_limit, DiagID);
901*12c85518Srobert     break;
902*12c85518Srobert   case DK_Linker:
903*12c85518Srobert     ComputeDiagID(Severity, linking_module, DiagID);
904e5dd7070Spatrick     break;
905e5dd7070Spatrick   case llvm::DK_OptimizationRemark:
906e5dd7070Spatrick     // Optimization remarks are always handled completely by this
907e5dd7070Spatrick     // handler. There is no generic way of emitting them.
908e5dd7070Spatrick     OptimizationRemarkHandler(cast<OptimizationRemark>(DI));
909e5dd7070Spatrick     return;
910e5dd7070Spatrick   case llvm::DK_OptimizationRemarkMissed:
911e5dd7070Spatrick     // Optimization remarks are always handled completely by this
912e5dd7070Spatrick     // handler. There is no generic way of emitting them.
913e5dd7070Spatrick     OptimizationRemarkHandler(cast<OptimizationRemarkMissed>(DI));
914e5dd7070Spatrick     return;
915e5dd7070Spatrick   case llvm::DK_OptimizationRemarkAnalysis:
916e5dd7070Spatrick     // Optimization remarks are always handled completely by this
917e5dd7070Spatrick     // handler. There is no generic way of emitting them.
918e5dd7070Spatrick     OptimizationRemarkHandler(cast<OptimizationRemarkAnalysis>(DI));
919e5dd7070Spatrick     return;
920e5dd7070Spatrick   case llvm::DK_OptimizationRemarkAnalysisFPCommute:
921e5dd7070Spatrick     // Optimization remarks are always handled completely by this
922e5dd7070Spatrick     // handler. There is no generic way of emitting them.
923e5dd7070Spatrick     OptimizationRemarkHandler(cast<OptimizationRemarkAnalysisFPCommute>(DI));
924e5dd7070Spatrick     return;
925e5dd7070Spatrick   case llvm::DK_OptimizationRemarkAnalysisAliasing:
926e5dd7070Spatrick     // Optimization remarks are always handled completely by this
927e5dd7070Spatrick     // handler. There is no generic way of emitting them.
928e5dd7070Spatrick     OptimizationRemarkHandler(cast<OptimizationRemarkAnalysisAliasing>(DI));
929e5dd7070Spatrick     return;
930e5dd7070Spatrick   case llvm::DK_MachineOptimizationRemark:
931e5dd7070Spatrick     // Optimization remarks are always handled completely by this
932e5dd7070Spatrick     // handler. There is no generic way of emitting them.
933e5dd7070Spatrick     OptimizationRemarkHandler(cast<MachineOptimizationRemark>(DI));
934e5dd7070Spatrick     return;
935e5dd7070Spatrick   case llvm::DK_MachineOptimizationRemarkMissed:
936e5dd7070Spatrick     // Optimization remarks are always handled completely by this
937e5dd7070Spatrick     // handler. There is no generic way of emitting them.
938e5dd7070Spatrick     OptimizationRemarkHandler(cast<MachineOptimizationRemarkMissed>(DI));
939e5dd7070Spatrick     return;
940e5dd7070Spatrick   case llvm::DK_MachineOptimizationRemarkAnalysis:
941e5dd7070Spatrick     // Optimization remarks are always handled completely by this
942e5dd7070Spatrick     // handler. There is no generic way of emitting them.
943e5dd7070Spatrick     OptimizationRemarkHandler(cast<MachineOptimizationRemarkAnalysis>(DI));
944e5dd7070Spatrick     return;
945e5dd7070Spatrick   case llvm::DK_OptimizationFailure:
946e5dd7070Spatrick     // Optimization failures are always handled completely by this
947e5dd7070Spatrick     // handler.
948e5dd7070Spatrick     OptimizationFailureHandler(cast<DiagnosticInfoOptimizationFailure>(DI));
949e5dd7070Spatrick     return;
950e5dd7070Spatrick   case llvm::DK_Unsupported:
951e5dd7070Spatrick     UnsupportedDiagHandler(cast<DiagnosticInfoUnsupported>(DI));
952e5dd7070Spatrick     return;
953*12c85518Srobert   case llvm::DK_DontCall:
954*12c85518Srobert     DontCallDiagHandler(cast<DiagnosticInfoDontCall>(DI));
955*12c85518Srobert     return;
956*12c85518Srobert   case llvm::DK_MisExpect:
957*12c85518Srobert     MisExpectDiagHandler(cast<DiagnosticInfoMisExpect>(DI));
958*12c85518Srobert     return;
959e5dd7070Spatrick   default:
960e5dd7070Spatrick     // Plugin IDs are not bound to any value as they are set dynamically.
961e5dd7070Spatrick     ComputeDiagRemarkID(Severity, backend_plugin, DiagID);
962e5dd7070Spatrick     break;
963e5dd7070Spatrick   }
964e5dd7070Spatrick   std::string MsgStorage;
965e5dd7070Spatrick   {
966e5dd7070Spatrick     raw_string_ostream Stream(MsgStorage);
967e5dd7070Spatrick     DiagnosticPrinterRawOStream DP(Stream);
968e5dd7070Spatrick     DI.print(DP);
969e5dd7070Spatrick   }
970e5dd7070Spatrick 
971*12c85518Srobert   if (DI.getKind() == DK_Linker) {
972*12c85518Srobert     assert(CurLinkModule && "CurLinkModule must be set for linker diagnostics");
973*12c85518Srobert     Diags.Report(DiagID) << CurLinkModule->getModuleIdentifier() << MsgStorage;
974e5dd7070Spatrick     return;
975e5dd7070Spatrick   }
976e5dd7070Spatrick 
977e5dd7070Spatrick   // Report the backend message using the usual diagnostic mechanism.
978e5dd7070Spatrick   FullSourceLoc Loc;
979e5dd7070Spatrick   Diags.Report(Loc, DiagID).AddString(MsgStorage);
980e5dd7070Spatrick }
981e5dd7070Spatrick #undef ComputeDiagID
982e5dd7070Spatrick 
CodeGenAction(unsigned _Act,LLVMContext * _VMContext)983e5dd7070Spatrick CodeGenAction::CodeGenAction(unsigned _Act, LLVMContext *_VMContext)
984e5dd7070Spatrick     : Act(_Act), VMContext(_VMContext ? _VMContext : new LLVMContext),
985e5dd7070Spatrick       OwnsVMContext(!_VMContext) {}
986e5dd7070Spatrick 
~CodeGenAction()987e5dd7070Spatrick CodeGenAction::~CodeGenAction() {
988e5dd7070Spatrick   TheModule.reset();
989e5dd7070Spatrick   if (OwnsVMContext)
990e5dd7070Spatrick     delete VMContext;
991e5dd7070Spatrick }
992e5dd7070Spatrick 
hasIRSupport() const993e5dd7070Spatrick bool CodeGenAction::hasIRSupport() const { return true; }
994e5dd7070Spatrick 
EndSourceFileAction()995e5dd7070Spatrick void CodeGenAction::EndSourceFileAction() {
996e5dd7070Spatrick   // If the consumer creation failed, do nothing.
997e5dd7070Spatrick   if (!getCompilerInstance().hasASTConsumer())
998e5dd7070Spatrick     return;
999e5dd7070Spatrick 
1000e5dd7070Spatrick   // Steal the module from the consumer.
1001e5dd7070Spatrick   TheModule = BEConsumer->takeModule();
1002e5dd7070Spatrick }
1003e5dd7070Spatrick 
takeModule()1004e5dd7070Spatrick std::unique_ptr<llvm::Module> CodeGenAction::takeModule() {
1005e5dd7070Spatrick   return std::move(TheModule);
1006e5dd7070Spatrick }
1007e5dd7070Spatrick 
takeLLVMContext()1008e5dd7070Spatrick llvm::LLVMContext *CodeGenAction::takeLLVMContext() {
1009e5dd7070Spatrick   OwnsVMContext = false;
1010e5dd7070Spatrick   return VMContext;
1011e5dd7070Spatrick }
1012e5dd7070Spatrick 
getCodeGenerator() const1013a9ac8606Spatrick CodeGenerator *CodeGenAction::getCodeGenerator() const {
1014a9ac8606Spatrick   return BEConsumer->getCodeGenerator();
1015a9ac8606Spatrick }
1016a9ac8606Spatrick 
1017e5dd7070Spatrick static std::unique_ptr<raw_pwrite_stream>
GetOutputStream(CompilerInstance & CI,StringRef InFile,BackendAction Action)1018e5dd7070Spatrick GetOutputStream(CompilerInstance &CI, StringRef InFile, BackendAction Action) {
1019e5dd7070Spatrick   switch (Action) {
1020e5dd7070Spatrick   case Backend_EmitAssembly:
1021e5dd7070Spatrick     return CI.createDefaultOutputFile(false, InFile, "s");
1022e5dd7070Spatrick   case Backend_EmitLL:
1023e5dd7070Spatrick     return CI.createDefaultOutputFile(false, InFile, "ll");
1024e5dd7070Spatrick   case Backend_EmitBC:
1025e5dd7070Spatrick     return CI.createDefaultOutputFile(true, InFile, "bc");
1026e5dd7070Spatrick   case Backend_EmitNothing:
1027e5dd7070Spatrick     return nullptr;
1028e5dd7070Spatrick   case Backend_EmitMCNull:
1029e5dd7070Spatrick     return CI.createNullOutputFile();
1030e5dd7070Spatrick   case Backend_EmitObj:
1031e5dd7070Spatrick     return CI.createDefaultOutputFile(true, InFile, "o");
1032e5dd7070Spatrick   }
1033e5dd7070Spatrick 
1034e5dd7070Spatrick   llvm_unreachable("Invalid action!");
1035e5dd7070Spatrick }
1036e5dd7070Spatrick 
1037e5dd7070Spatrick std::unique_ptr<ASTConsumer>
CreateASTConsumer(CompilerInstance & CI,StringRef InFile)1038e5dd7070Spatrick CodeGenAction::CreateASTConsumer(CompilerInstance &CI, StringRef InFile) {
1039e5dd7070Spatrick   BackendAction BA = static_cast<BackendAction>(Act);
1040e5dd7070Spatrick   std::unique_ptr<raw_pwrite_stream> OS = CI.takeOutputStream();
1041e5dd7070Spatrick   if (!OS)
1042e5dd7070Spatrick     OS = GetOutputStream(CI, InFile, BA);
1043e5dd7070Spatrick 
1044e5dd7070Spatrick   if (BA != Backend_EmitNothing && !OS)
1045e5dd7070Spatrick     return nullptr;
1046e5dd7070Spatrick 
1047*12c85518Srobert   VMContext->setOpaquePointers(CI.getCodeGenOpts().OpaquePointers);
1048*12c85518Srobert 
1049e5dd7070Spatrick   // Load bitcode modules to link with, if we need to.
1050e5dd7070Spatrick   if (LinkModules.empty())
1051e5dd7070Spatrick     for (const CodeGenOptions::BitcodeFileToLink &F :
1052e5dd7070Spatrick          CI.getCodeGenOpts().LinkBitcodeFiles) {
1053e5dd7070Spatrick       auto BCBuf = CI.getFileManager().getBufferForFile(F.Filename);
1054e5dd7070Spatrick       if (!BCBuf) {
1055e5dd7070Spatrick         CI.getDiagnostics().Report(diag::err_cannot_open_file)
1056e5dd7070Spatrick             << F.Filename << BCBuf.getError().message();
1057e5dd7070Spatrick         LinkModules.clear();
1058e5dd7070Spatrick         return nullptr;
1059e5dd7070Spatrick       }
1060e5dd7070Spatrick 
1061e5dd7070Spatrick       Expected<std::unique_ptr<llvm::Module>> ModuleOrErr =
1062e5dd7070Spatrick           getOwningLazyBitcodeModule(std::move(*BCBuf), *VMContext);
1063e5dd7070Spatrick       if (!ModuleOrErr) {
1064e5dd7070Spatrick         handleAllErrors(ModuleOrErr.takeError(), [&](ErrorInfoBase &EIB) {
1065e5dd7070Spatrick           CI.getDiagnostics().Report(diag::err_cannot_open_file)
1066e5dd7070Spatrick               << F.Filename << EIB.message();
1067e5dd7070Spatrick         });
1068e5dd7070Spatrick         LinkModules.clear();
1069e5dd7070Spatrick         return nullptr;
1070e5dd7070Spatrick       }
1071e5dd7070Spatrick       LinkModules.push_back({std::move(ModuleOrErr.get()), F.PropagateAttrs,
1072e5dd7070Spatrick                              F.Internalize, F.LinkFlags});
1073e5dd7070Spatrick     }
1074e5dd7070Spatrick 
1075e5dd7070Spatrick   CoverageSourceInfo *CoverageInfo = nullptr;
1076e5dd7070Spatrick   // Add the preprocessor callback only when the coverage mapping is generated.
1077a9ac8606Spatrick   if (CI.getCodeGenOpts().CoverageMapping)
1078a9ac8606Spatrick     CoverageInfo = CodeGen::CoverageMappingModuleGen::setUpCoverageCallbacks(
1079a9ac8606Spatrick         CI.getPreprocessor());
1080e5dd7070Spatrick 
1081e5dd7070Spatrick   std::unique_ptr<BackendConsumer> Result(new BackendConsumer(
1082*12c85518Srobert       BA, CI.getDiagnostics(), &CI.getVirtualFileSystem(),
1083*12c85518Srobert       CI.getHeaderSearchOpts(), CI.getPreprocessorOpts(), CI.getCodeGenOpts(),
1084*12c85518Srobert       CI.getTargetOpts(), CI.getLangOpts(), std::string(InFile),
1085*12c85518Srobert       std::move(LinkModules), std::move(OS), *VMContext, CoverageInfo));
1086e5dd7070Spatrick   BEConsumer = Result.get();
1087e5dd7070Spatrick 
1088e5dd7070Spatrick   // Enable generating macro debug info only when debug info is not disabled and
1089e5dd7070Spatrick   // also macro debug info is enabled.
1090e5dd7070Spatrick   if (CI.getCodeGenOpts().getDebugInfo() != codegenoptions::NoDebugInfo &&
1091e5dd7070Spatrick       CI.getCodeGenOpts().MacroDebugInfo) {
1092e5dd7070Spatrick     std::unique_ptr<PPCallbacks> Callbacks =
1093e5dd7070Spatrick         std::make_unique<MacroPPCallbacks>(BEConsumer->getCodeGenerator(),
1094e5dd7070Spatrick                                             CI.getPreprocessor());
1095e5dd7070Spatrick     CI.getPreprocessor().addPPCallbacks(std::move(Callbacks));
1096e5dd7070Spatrick   }
1097e5dd7070Spatrick 
1098e5dd7070Spatrick   return std::move(Result);
1099e5dd7070Spatrick }
1100e5dd7070Spatrick 
1101e5dd7070Spatrick std::unique_ptr<llvm::Module>
loadModule(MemoryBufferRef MBRef)1102e5dd7070Spatrick CodeGenAction::loadModule(MemoryBufferRef MBRef) {
1103e5dd7070Spatrick   CompilerInstance &CI = getCompilerInstance();
1104e5dd7070Spatrick   SourceManager &SM = CI.getSourceManager();
1105e5dd7070Spatrick 
1106*12c85518Srobert   VMContext->setOpaquePointers(CI.getCodeGenOpts().OpaquePointers);
1107*12c85518Srobert 
1108e5dd7070Spatrick   // For ThinLTO backend invocations, ensure that the context
1109e5dd7070Spatrick   // merges types based on ODR identifiers. We also need to read
1110e5dd7070Spatrick   // the correct module out of a multi-module bitcode file.
1111e5dd7070Spatrick   if (!CI.getCodeGenOpts().ThinLTOIndexFile.empty()) {
1112e5dd7070Spatrick     VMContext->enableDebugTypeODRUniquing();
1113e5dd7070Spatrick 
1114e5dd7070Spatrick     auto DiagErrors = [&](Error E) -> std::unique_ptr<llvm::Module> {
1115e5dd7070Spatrick       unsigned DiagID =
1116e5dd7070Spatrick           CI.getDiagnostics().getCustomDiagID(DiagnosticsEngine::Error, "%0");
1117e5dd7070Spatrick       handleAllErrors(std::move(E), [&](ErrorInfoBase &EIB) {
1118e5dd7070Spatrick         CI.getDiagnostics().Report(DiagID) << EIB.message();
1119e5dd7070Spatrick       });
1120e5dd7070Spatrick       return {};
1121e5dd7070Spatrick     };
1122e5dd7070Spatrick 
1123e5dd7070Spatrick     Expected<std::vector<BitcodeModule>> BMsOrErr = getBitcodeModuleList(MBRef);
1124e5dd7070Spatrick     if (!BMsOrErr)
1125e5dd7070Spatrick       return DiagErrors(BMsOrErr.takeError());
1126a9ac8606Spatrick     BitcodeModule *Bm = llvm::lto::findThinLTOModule(*BMsOrErr);
1127e5dd7070Spatrick     // We have nothing to do if the file contains no ThinLTO module. This is
1128e5dd7070Spatrick     // possible if ThinLTO compilation was not able to split module. Content of
1129e5dd7070Spatrick     // the file was already processed by indexing and will be passed to the
1130e5dd7070Spatrick     // linker using merged object file.
1131e5dd7070Spatrick     if (!Bm) {
1132e5dd7070Spatrick       auto M = std::make_unique<llvm::Module>("empty", *VMContext);
1133e5dd7070Spatrick       M->setTargetTriple(CI.getTargetOpts().Triple);
1134e5dd7070Spatrick       return M;
1135e5dd7070Spatrick     }
1136e5dd7070Spatrick     Expected<std::unique_ptr<llvm::Module>> MOrErr =
1137e5dd7070Spatrick         Bm->parseModule(*VMContext);
1138e5dd7070Spatrick     if (!MOrErr)
1139e5dd7070Spatrick       return DiagErrors(MOrErr.takeError());
1140e5dd7070Spatrick     return std::move(*MOrErr);
1141e5dd7070Spatrick   }
1142e5dd7070Spatrick 
1143e5dd7070Spatrick   llvm::SMDiagnostic Err;
1144e5dd7070Spatrick   if (std::unique_ptr<llvm::Module> M = parseIR(MBRef, Err, *VMContext))
1145e5dd7070Spatrick     return M;
1146e5dd7070Spatrick 
1147e5dd7070Spatrick   // Translate from the diagnostic info to the SourceManager location if
1148e5dd7070Spatrick   // available.
1149e5dd7070Spatrick   // TODO: Unify this with ConvertBackendLocation()
1150e5dd7070Spatrick   SourceLocation Loc;
1151e5dd7070Spatrick   if (Err.getLineNo() > 0) {
1152e5dd7070Spatrick     assert(Err.getColumnNo() >= 0);
1153e5dd7070Spatrick     Loc = SM.translateFileLineCol(SM.getFileEntryForID(SM.getMainFileID()),
1154e5dd7070Spatrick                                   Err.getLineNo(), Err.getColumnNo() + 1);
1155e5dd7070Spatrick   }
1156e5dd7070Spatrick 
1157e5dd7070Spatrick   // Strip off a leading diagnostic code if there is one.
1158e5dd7070Spatrick   StringRef Msg = Err.getMessage();
1159e5dd7070Spatrick   if (Msg.startswith("error: "))
1160e5dd7070Spatrick     Msg = Msg.substr(7);
1161e5dd7070Spatrick 
1162e5dd7070Spatrick   unsigned DiagID =
1163e5dd7070Spatrick       CI.getDiagnostics().getCustomDiagID(DiagnosticsEngine::Error, "%0");
1164e5dd7070Spatrick 
1165e5dd7070Spatrick   CI.getDiagnostics().Report(Loc, DiagID) << Msg;
1166e5dd7070Spatrick   return {};
1167e5dd7070Spatrick }
1168e5dd7070Spatrick 
ExecuteAction()1169e5dd7070Spatrick void CodeGenAction::ExecuteAction() {
1170a9ac8606Spatrick   if (getCurrentFileKind().getLanguage() != Language::LLVM_IR) {
1171a9ac8606Spatrick     this->ASTFrontendAction::ExecuteAction();
1172a9ac8606Spatrick     return;
1173a9ac8606Spatrick   }
1174a9ac8606Spatrick 
1175e5dd7070Spatrick   // If this is an IR file, we have to treat it specially.
1176e5dd7070Spatrick   BackendAction BA = static_cast<BackendAction>(Act);
1177e5dd7070Spatrick   CompilerInstance &CI = getCompilerInstance();
1178e5dd7070Spatrick   auto &CodeGenOpts = CI.getCodeGenOpts();
1179e5dd7070Spatrick   auto &Diagnostics = CI.getDiagnostics();
1180e5dd7070Spatrick   std::unique_ptr<raw_pwrite_stream> OS =
1181*12c85518Srobert       GetOutputStream(CI, getCurrentFileOrBufferName(), BA);
1182e5dd7070Spatrick   if (BA != Backend_EmitNothing && !OS)
1183e5dd7070Spatrick     return;
1184e5dd7070Spatrick 
1185e5dd7070Spatrick   SourceManager &SM = CI.getSourceManager();
1186e5dd7070Spatrick   FileID FID = SM.getMainFileID();
1187*12c85518Srobert   std::optional<MemoryBufferRef> MainFile = SM.getBufferOrNone(FID);
1188a9ac8606Spatrick   if (!MainFile)
1189e5dd7070Spatrick     return;
1190e5dd7070Spatrick 
1191e5dd7070Spatrick   TheModule = loadModule(*MainFile);
1192e5dd7070Spatrick   if (!TheModule)
1193e5dd7070Spatrick     return;
1194e5dd7070Spatrick 
1195e5dd7070Spatrick   const TargetOptions &TargetOpts = CI.getTargetOpts();
1196e5dd7070Spatrick   if (TheModule->getTargetTriple() != TargetOpts.Triple) {
1197a9ac8606Spatrick     Diagnostics.Report(SourceLocation(), diag::warn_fe_override_module)
1198e5dd7070Spatrick         << TargetOpts.Triple;
1199e5dd7070Spatrick     TheModule->setTargetTriple(TargetOpts.Triple);
1200e5dd7070Spatrick   }
1201e5dd7070Spatrick 
1202*12c85518Srobert   EmbedObject(TheModule.get(), CodeGenOpts, Diagnostics);
1203a9ac8606Spatrick   EmbedBitcode(TheModule.get(), CodeGenOpts, *MainFile);
1204e5dd7070Spatrick 
1205e5dd7070Spatrick   LLVMContext &Ctx = TheModule->getContext();
1206a9ac8606Spatrick 
1207a9ac8606Spatrick   // Restore any diagnostic handler previously set before returning from this
1208a9ac8606Spatrick   // function.
1209a9ac8606Spatrick   struct RAII {
1210a9ac8606Spatrick     LLVMContext &Ctx;
1211a9ac8606Spatrick     std::unique_ptr<DiagnosticHandler> PrevHandler = Ctx.getDiagnosticHandler();
1212a9ac8606Spatrick     ~RAII() { Ctx.setDiagnosticHandler(std::move(PrevHandler)); }
1213a9ac8606Spatrick   } _{Ctx};
1214e5dd7070Spatrick 
1215e5dd7070Spatrick   // Set clang diagnostic handler. To do this we need to create a fake
1216e5dd7070Spatrick   // BackendConsumer.
1217*12c85518Srobert   BackendConsumer Result(BA, CI.getDiagnostics(), &CI.getVirtualFileSystem(),
1218*12c85518Srobert                          CI.getHeaderSearchOpts(), CI.getPreprocessorOpts(),
1219*12c85518Srobert                          CI.getCodeGenOpts(), CI.getTargetOpts(),
1220*12c85518Srobert                          CI.getLangOpts(), TheModule.get(),
1221e5dd7070Spatrick                          std::move(LinkModules), *VMContext, nullptr);
1222e5dd7070Spatrick   // PR44896: Force DiscardValueNames as false. DiscardValueNames cannot be
1223e5dd7070Spatrick   // true here because the valued names are needed for reading textual IR.
1224e5dd7070Spatrick   Ctx.setDiscardValueNames(false);
1225e5dd7070Spatrick   Ctx.setDiagnosticHandler(
1226e5dd7070Spatrick       std::make_unique<ClangDiagnosticHandler>(CodeGenOpts, &Result));
1227e5dd7070Spatrick 
1228e5dd7070Spatrick   Expected<std::unique_ptr<llvm::ToolOutputFile>> OptRecordFileOrErr =
1229ec727ea7Spatrick       setupLLVMOptimizationRemarks(
1230e5dd7070Spatrick           Ctx, CodeGenOpts.OptRecordFile, CodeGenOpts.OptRecordPasses,
1231e5dd7070Spatrick           CodeGenOpts.OptRecordFormat, CodeGenOpts.DiagnosticsWithHotness,
1232e5dd7070Spatrick           CodeGenOpts.DiagnosticsHotnessThreshold);
1233e5dd7070Spatrick 
1234e5dd7070Spatrick   if (Error E = OptRecordFileOrErr.takeError()) {
1235e5dd7070Spatrick     reportOptRecordError(std::move(E), Diagnostics, CodeGenOpts);
1236e5dd7070Spatrick     return;
1237e5dd7070Spatrick   }
1238e5dd7070Spatrick   std::unique_ptr<llvm::ToolOutputFile> OptRecordFile =
1239e5dd7070Spatrick       std::move(*OptRecordFileOrErr);
1240e5dd7070Spatrick 
1241e5dd7070Spatrick   EmitBackendOutput(Diagnostics, CI.getHeaderSearchOpts(), CodeGenOpts,
1242e5dd7070Spatrick                     TargetOpts, CI.getLangOpts(),
1243a9ac8606Spatrick                     CI.getTarget().getDataLayoutString(), TheModule.get(), BA,
1244e5dd7070Spatrick                     std::move(OS));
1245e5dd7070Spatrick   if (OptRecordFile)
1246e5dd7070Spatrick     OptRecordFile->keep();
1247e5dd7070Spatrick }
1248e5dd7070Spatrick 
1249e5dd7070Spatrick //
1250e5dd7070Spatrick 
anchor()1251e5dd7070Spatrick void EmitAssemblyAction::anchor() { }
EmitAssemblyAction(llvm::LLVMContext * _VMContext)1252e5dd7070Spatrick EmitAssemblyAction::EmitAssemblyAction(llvm::LLVMContext *_VMContext)
1253e5dd7070Spatrick   : CodeGenAction(Backend_EmitAssembly, _VMContext) {}
1254e5dd7070Spatrick 
anchor()1255e5dd7070Spatrick void EmitBCAction::anchor() { }
EmitBCAction(llvm::LLVMContext * _VMContext)1256e5dd7070Spatrick EmitBCAction::EmitBCAction(llvm::LLVMContext *_VMContext)
1257e5dd7070Spatrick   : CodeGenAction(Backend_EmitBC, _VMContext) {}
1258e5dd7070Spatrick 
anchor()1259e5dd7070Spatrick void EmitLLVMAction::anchor() { }
EmitLLVMAction(llvm::LLVMContext * _VMContext)1260e5dd7070Spatrick EmitLLVMAction::EmitLLVMAction(llvm::LLVMContext *_VMContext)
1261e5dd7070Spatrick   : CodeGenAction(Backend_EmitLL, _VMContext) {}
1262e5dd7070Spatrick 
anchor()1263e5dd7070Spatrick void EmitLLVMOnlyAction::anchor() { }
EmitLLVMOnlyAction(llvm::LLVMContext * _VMContext)1264e5dd7070Spatrick EmitLLVMOnlyAction::EmitLLVMOnlyAction(llvm::LLVMContext *_VMContext)
1265e5dd7070Spatrick   : CodeGenAction(Backend_EmitNothing, _VMContext) {}
1266e5dd7070Spatrick 
anchor()1267e5dd7070Spatrick void EmitCodeGenOnlyAction::anchor() { }
EmitCodeGenOnlyAction(llvm::LLVMContext * _VMContext)1268e5dd7070Spatrick EmitCodeGenOnlyAction::EmitCodeGenOnlyAction(llvm::LLVMContext *_VMContext)
1269e5dd7070Spatrick   : CodeGenAction(Backend_EmitMCNull, _VMContext) {}
1270e5dd7070Spatrick 
anchor()1271e5dd7070Spatrick void EmitObjAction::anchor() { }
EmitObjAction(llvm::LLVMContext * _VMContext)1272e5dd7070Spatrick EmitObjAction::EmitObjAction(llvm::LLVMContext *_VMContext)
1273e5dd7070Spatrick   : CodeGenAction(Backend_EmitObj, _VMContext) {}
1274