Home | History | Annotate | Download | only in CodeGen
      1 //===--- CodeGenAction.cpp - LLVM Code Generation Frontend Action ---------===//
      2 //
      3 //                     The LLVM Compiler Infrastructure
      4 //
      5 // This file is distributed under the University of Illinois Open Source
      6 // License. See LICENSE.TXT for details.
      7 //
      8 //===----------------------------------------------------------------------===//
      9 
     10 #include "clang/CodeGen/CodeGenAction.h"
     11 #include "clang/AST/ASTConsumer.h"
     12 #include "clang/AST/ASTContext.h"
     13 #include "clang/AST/DeclGroup.h"
     14 #include "clang/AST/DeclCXX.h"
     15 #include "clang/Basic/FileManager.h"
     16 #include "clang/Basic/SourceManager.h"
     17 #include "clang/Basic/TargetInfo.h"
     18 #include "clang/CodeGen/BackendUtil.h"
     19 #include "clang/CodeGen/ModuleBuilder.h"
     20 #include "clang/Frontend/CompilerInstance.h"
     21 #include "clang/Frontend/FrontendDiagnostic.h"
     22 #include "llvm/ADT/SmallString.h"
     23 #include "llvm/Bitcode/ReaderWriter.h"
     24 #include "llvm/IR/DebugInfo.h"
     25 #include "llvm/IR/DiagnosticInfo.h"
     26 #include "llvm/IR/DiagnosticPrinter.h"
     27 #include "llvm/IR/LLVMContext.h"
     28 #include "llvm/IR/Module.h"
     29 #include "llvm/IRReader/IRReader.h"
     30 #include "llvm/Linker/Linker.h"
     31 #include "llvm/Pass.h"
     32 #include "llvm/Support/MemoryBuffer.h"
     33 #include "llvm/Support/SourceMgr.h"
     34 #include "llvm/Support/Timer.h"
     35 #include <memory>
     36 using namespace clang;
     37 using namespace llvm;
     38 
     39 namespace clang {
     40   class BackendConsumer : public ASTConsumer {
     41     virtual void anchor();
     42     DiagnosticsEngine &Diags;
     43     BackendAction Action;
     44     const CodeGenOptions &CodeGenOpts;
     45     const TargetOptions &TargetOpts;
     46     const LangOptions &LangOpts;
     47     raw_ostream *AsmOutStream;
     48     ASTContext *Context;
     49 
     50     Timer LLVMIRGeneration;
     51 
     52     std::unique_ptr<CodeGenerator> Gen;
     53 
     54     std::unique_ptr<llvm::Module> TheModule, LinkModule;
     55 
     56   public:
     57     BackendConsumer(BackendAction action, DiagnosticsEngine &_Diags,
     58                     const CodeGenOptions &compopts,
     59                     const TargetOptions &targetopts,
     60                     const LangOptions &langopts, bool TimePasses,
     61                     const std::string &infile, llvm::Module *LinkModule,
     62                     raw_ostream *OS, LLVMContext &C)
     63         : Diags(_Diags), Action(action), CodeGenOpts(compopts),
     64           TargetOpts(targetopts), LangOpts(langopts), AsmOutStream(OS),
     65           Context(), LLVMIRGeneration("LLVM IR Generation Time"),
     66           Gen(CreateLLVMCodeGen(Diags, infile, compopts, targetopts, C)),
     67           LinkModule(LinkModule) {
     68       llvm::TimePassesIsEnabled = TimePasses;
     69     }
     70 
     71     llvm::Module *takeModule() { return TheModule.release(); }
     72     llvm::Module *takeLinkModule() { return LinkModule.release(); }
     73 
     74     void HandleCXXStaticMemberVarInstantiation(VarDecl *VD) override {
     75       Gen->HandleCXXStaticMemberVarInstantiation(VD);
     76     }
     77 
     78     void Initialize(ASTContext &Ctx) override {
     79       Context = &Ctx;
     80 
     81       if (llvm::TimePassesIsEnabled)
     82         LLVMIRGeneration.startTimer();
     83 
     84       Gen->Initialize(Ctx);
     85 
     86       TheModule.reset(Gen->GetModule());
     87 
     88       if (llvm::TimePassesIsEnabled)
     89         LLVMIRGeneration.stopTimer();
     90     }
     91 
     92     bool HandleTopLevelDecl(DeclGroupRef D) override {
     93       PrettyStackTraceDecl CrashInfo(*D.begin(), SourceLocation(),
     94                                      Context->getSourceManager(),
     95                                      "LLVM IR generation of declaration");
     96 
     97       if (llvm::TimePassesIsEnabled)
     98         LLVMIRGeneration.startTimer();
     99 
    100       Gen->HandleTopLevelDecl(D);
    101 
    102       if (llvm::TimePassesIsEnabled)
    103         LLVMIRGeneration.stopTimer();
    104 
    105       return true;
    106     }
    107 
    108     void HandleInlineMethodDefinition(CXXMethodDecl *D) override {
    109       PrettyStackTraceDecl CrashInfo(D, SourceLocation(),
    110                                      Context->getSourceManager(),
    111                                      "LLVM IR generation of inline method");
    112       if (llvm::TimePassesIsEnabled)
    113         LLVMIRGeneration.startTimer();
    114 
    115       Gen->HandleInlineMethodDefinition(D);
    116 
    117       if (llvm::TimePassesIsEnabled)
    118         LLVMIRGeneration.stopTimer();
    119     }
    120 
    121     void HandleTranslationUnit(ASTContext &C) override {
    122       {
    123         PrettyStackTraceString CrashInfo("Per-file LLVM IR generation");
    124         if (llvm::TimePassesIsEnabled)
    125           LLVMIRGeneration.startTimer();
    126 
    127         Gen->HandleTranslationUnit(C);
    128 
    129         if (llvm::TimePassesIsEnabled)
    130           LLVMIRGeneration.stopTimer();
    131       }
    132 
    133       // Silently ignore if we weren't initialized for some reason.
    134       if (!TheModule)
    135         return;
    136 
    137       // Make sure IR generation is happy with the module. This is released by
    138       // the module provider.
    139       llvm::Module *M = Gen->ReleaseModule();
    140       if (!M) {
    141         // The module has been released by IR gen on failures, do not double
    142         // free.
    143         TheModule.release();
    144         return;
    145       }
    146 
    147       assert(TheModule.get() == M &&
    148              "Unexpected module change during IR generation");
    149 
    150       // Link LinkModule into this module if present, preserving its validity.
    151       if (LinkModule) {
    152         std::string ErrorMsg;
    153         if (Linker::LinkModules(M, LinkModule.get(), Linker::PreserveSource,
    154                                 &ErrorMsg)) {
    155           Diags.Report(diag::err_fe_cannot_link_module)
    156             << LinkModule->getModuleIdentifier() << ErrorMsg;
    157           return;
    158         }
    159       }
    160 
    161       // Install an inline asm handler so that diagnostics get printed through
    162       // our diagnostics hooks.
    163       LLVMContext &Ctx = TheModule->getContext();
    164       LLVMContext::InlineAsmDiagHandlerTy OldHandler =
    165         Ctx.getInlineAsmDiagnosticHandler();
    166       void *OldContext = Ctx.getInlineAsmDiagnosticContext();
    167       Ctx.setInlineAsmDiagnosticHandler(InlineAsmDiagHandler, this);
    168 
    169       LLVMContext::DiagnosticHandlerTy OldDiagnosticHandler =
    170           Ctx.getDiagnosticHandler();
    171       void *OldDiagnosticContext = Ctx.getDiagnosticContext();
    172       Ctx.setDiagnosticHandler(DiagnosticHandler, this);
    173 
    174       EmitBackendOutput(Diags, CodeGenOpts, TargetOpts, LangOpts,
    175                         C.getTargetInfo().getTargetDescription(),
    176                         TheModule.get(), Action, AsmOutStream);
    177 
    178       Ctx.setInlineAsmDiagnosticHandler(OldHandler, OldContext);
    179 
    180       Ctx.setDiagnosticHandler(OldDiagnosticHandler, OldDiagnosticContext);
    181     }
    182 
    183     void HandleTagDeclDefinition(TagDecl *D) override {
    184       PrettyStackTraceDecl CrashInfo(D, SourceLocation(),
    185                                      Context->getSourceManager(),
    186                                      "LLVM IR generation of declaration");
    187       Gen->HandleTagDeclDefinition(D);
    188     }
    189 
    190     void HandleTagDeclRequiredDefinition(const TagDecl *D) override {
    191       Gen->HandleTagDeclRequiredDefinition(D);
    192     }
    193 
    194     void CompleteTentativeDefinition(VarDecl *D) override {
    195       Gen->CompleteTentativeDefinition(D);
    196     }
    197 
    198     void HandleVTable(CXXRecordDecl *RD, bool DefinitionRequired) override {
    199       Gen->HandleVTable(RD, DefinitionRequired);
    200     }
    201 
    202     void HandleLinkerOptionPragma(llvm::StringRef Opts) override {
    203       Gen->HandleLinkerOptionPragma(Opts);
    204     }
    205 
    206     void HandleDetectMismatch(llvm::StringRef Name,
    207                                       llvm::StringRef Value) override {
    208       Gen->HandleDetectMismatch(Name, Value);
    209     }
    210 
    211     void HandleDependentLibrary(llvm::StringRef Opts) override {
    212       Gen->HandleDependentLibrary(Opts);
    213     }
    214 
    215     static void InlineAsmDiagHandler(const llvm::SMDiagnostic &SM,void *Context,
    216                                      unsigned LocCookie) {
    217       SourceLocation Loc = SourceLocation::getFromRawEncoding(LocCookie);
    218       ((BackendConsumer*)Context)->InlineAsmDiagHandler2(SM, Loc);
    219     }
    220 
    221     static void DiagnosticHandler(const llvm::DiagnosticInfo &DI,
    222                                   void *Context) {
    223       ((BackendConsumer *)Context)->DiagnosticHandlerImpl(DI);
    224     }
    225 
    226     void InlineAsmDiagHandler2(const llvm::SMDiagnostic &,
    227                                SourceLocation LocCookie);
    228 
    229     void DiagnosticHandlerImpl(const llvm::DiagnosticInfo &DI);
    230     /// \brief Specialized handler for InlineAsm diagnostic.
    231     /// \return True if the diagnostic has been successfully reported, false
    232     /// otherwise.
    233     bool InlineAsmDiagHandler(const llvm::DiagnosticInfoInlineAsm &D);
    234     /// \brief Specialized handler for StackSize diagnostic.
    235     /// \return True if the diagnostic has been successfully reported, false
    236     /// otherwise.
    237     bool StackSizeDiagHandler(const llvm::DiagnosticInfoStackSize &D);
    238     /// \brief Specialized handlers for optimization remarks.
    239     /// Note that these handlers only accept remarks and they always handle
    240     /// them.
    241     void
    242     EmitOptimizationRemark(const llvm::DiagnosticInfoOptimizationRemarkBase &D,
    243                            unsigned DiagID);
    244     void
    245     OptimizationRemarkHandler(const llvm::DiagnosticInfoOptimizationRemark &D);
    246     void OptimizationRemarkHandler(
    247         const llvm::DiagnosticInfoOptimizationRemarkMissed &D);
    248     void OptimizationRemarkHandler(
    249         const llvm::DiagnosticInfoOptimizationRemarkAnalysis &D);
    250   };
    251 
    252   void BackendConsumer::anchor() {}
    253 }
    254 
    255 /// ConvertBackendLocation - Convert a location in a temporary llvm::SourceMgr
    256 /// buffer to be a valid FullSourceLoc.
    257 static FullSourceLoc ConvertBackendLocation(const llvm::SMDiagnostic &D,
    258                                             SourceManager &CSM) {
    259   // Get both the clang and llvm source managers.  The location is relative to
    260   // a memory buffer that the LLVM Source Manager is handling, we need to add
    261   // a copy to the Clang source manager.
    262   const llvm::SourceMgr &LSM = *D.getSourceMgr();
    263 
    264   // We need to copy the underlying LLVM memory buffer because llvm::SourceMgr
    265   // already owns its one and clang::SourceManager wants to own its one.
    266   const MemoryBuffer *LBuf =
    267   LSM.getMemoryBuffer(LSM.FindBufferContainingLoc(D.getLoc()));
    268 
    269   // Create the copy and transfer ownership to clang::SourceManager.
    270   // TODO: Avoid copying files into memory.
    271   llvm::MemoryBuffer *CBuf =
    272   llvm::MemoryBuffer::getMemBufferCopy(LBuf->getBuffer(),
    273                                        LBuf->getBufferIdentifier());
    274   // FIXME: Keep a file ID map instead of creating new IDs for each location.
    275   FileID FID = CSM.createFileID(CBuf);
    276 
    277   // Translate the offset into the file.
    278   unsigned Offset = D.getLoc().getPointer() - LBuf->getBufferStart();
    279   SourceLocation NewLoc =
    280   CSM.getLocForStartOfFile(FID).getLocWithOffset(Offset);
    281   return FullSourceLoc(NewLoc, CSM);
    282 }
    283 
    284 
    285 /// InlineAsmDiagHandler2 - This function is invoked when the backend hits an
    286 /// error parsing inline asm.  The SMDiagnostic indicates the error relative to
    287 /// the temporary memory buffer that the inline asm parser has set up.
    288 void BackendConsumer::InlineAsmDiagHandler2(const llvm::SMDiagnostic &D,
    289                                             SourceLocation LocCookie) {
    290   // There are a couple of different kinds of errors we could get here.  First,
    291   // we re-format the SMDiagnostic in terms of a clang diagnostic.
    292 
    293   // Strip "error: " off the start of the message string.
    294   StringRef Message = D.getMessage();
    295   if (Message.startswith("error: "))
    296     Message = Message.substr(7);
    297 
    298   // If the SMDiagnostic has an inline asm source location, translate it.
    299   FullSourceLoc Loc;
    300   if (D.getLoc() != SMLoc())
    301     Loc = ConvertBackendLocation(D, Context->getSourceManager());
    302 
    303   unsigned DiagID;
    304   switch (D.getKind()) {
    305   case llvm::SourceMgr::DK_Error:
    306     DiagID = diag::err_fe_inline_asm;
    307     break;
    308   case llvm::SourceMgr::DK_Warning:
    309     DiagID = diag::warn_fe_inline_asm;
    310     break;
    311   case llvm::SourceMgr::DK_Note:
    312     DiagID = diag::note_fe_inline_asm;
    313     break;
    314   }
    315   // If this problem has clang-level source location information, report the
    316   // issue in the source with a note showing the instantiated
    317   // code.
    318   if (LocCookie.isValid()) {
    319     Diags.Report(LocCookie, DiagID).AddString(Message);
    320 
    321     if (D.getLoc().isValid()) {
    322       DiagnosticBuilder B = Diags.Report(Loc, diag::note_fe_inline_asm_here);
    323       // Convert the SMDiagnostic ranges into SourceRange and attach them
    324       // to the diagnostic.
    325       for (unsigned i = 0, e = D.getRanges().size(); i != e; ++i) {
    326         std::pair<unsigned, unsigned> Range = D.getRanges()[i];
    327         unsigned Column = D.getColumnNo();
    328         B << SourceRange(Loc.getLocWithOffset(Range.first - Column),
    329                          Loc.getLocWithOffset(Range.second - Column));
    330       }
    331     }
    332     return;
    333   }
    334 
    335   // Otherwise, report the backend issue as occurring in the generated .s file.
    336   // If Loc is invalid, we still need to report the issue, it just gets no
    337   // location info.
    338   Diags.Report(Loc, DiagID).AddString(Message);
    339 }
    340 
    341 #define ComputeDiagID(Severity, GroupName, DiagID)                             \
    342   do {                                                                         \
    343     switch (Severity) {                                                        \
    344     case llvm::DS_Error:                                                       \
    345       DiagID = diag::err_fe_##GroupName;                                       \
    346       break;                                                                   \
    347     case llvm::DS_Warning:                                                     \
    348       DiagID = diag::warn_fe_##GroupName;                                      \
    349       break;                                                                   \
    350     case llvm::DS_Remark:                                                      \
    351       llvm_unreachable("'remark' severity not expected");                      \
    352       break;                                                                   \
    353     case llvm::DS_Note:                                                        \
    354       DiagID = diag::note_fe_##GroupName;                                      \
    355       break;                                                                   \
    356     }                                                                          \
    357   } while (false)
    358 
    359 #define ComputeDiagRemarkID(Severity, GroupName, DiagID)                       \
    360   do {                                                                         \
    361     switch (Severity) {                                                        \
    362     case llvm::DS_Error:                                                       \
    363       DiagID = diag::err_fe_##GroupName;                                       \
    364       break;                                                                   \
    365     case llvm::DS_Warning:                                                     \
    366       DiagID = diag::warn_fe_##GroupName;                                      \
    367       break;                                                                   \
    368     case llvm::DS_Remark:                                                      \
    369       DiagID = diag::remark_fe_##GroupName;                                    \
    370       break;                                                                   \
    371     case llvm::DS_Note:                                                        \
    372       DiagID = diag::note_fe_##GroupName;                                      \
    373       break;                                                                   \
    374     }                                                                          \
    375   } while (false)
    376 
    377 bool
    378 BackendConsumer::InlineAsmDiagHandler(const llvm::DiagnosticInfoInlineAsm &D) {
    379   unsigned DiagID;
    380   ComputeDiagID(D.getSeverity(), inline_asm, DiagID);
    381   std::string Message = D.getMsgStr().str();
    382 
    383   // If this problem has clang-level source location information, report the
    384   // issue as being a problem in the source with a note showing the instantiated
    385   // code.
    386   SourceLocation LocCookie =
    387       SourceLocation::getFromRawEncoding(D.getLocCookie());
    388   if (LocCookie.isValid())
    389     Diags.Report(LocCookie, DiagID).AddString(Message);
    390   else {
    391     // Otherwise, report the backend diagnostic as occurring in the generated
    392     // .s file.
    393     // If Loc is invalid, we still need to report the diagnostic, it just gets
    394     // no location info.
    395     FullSourceLoc Loc;
    396     Diags.Report(Loc, DiagID).AddString(Message);
    397   }
    398   // We handled all the possible severities.
    399   return true;
    400 }
    401 
    402 bool
    403 BackendConsumer::StackSizeDiagHandler(const llvm::DiagnosticInfoStackSize &D) {
    404   if (D.getSeverity() != llvm::DS_Warning)
    405     // For now, the only support we have for StackSize diagnostic is warning.
    406     // We do not know how to format other severities.
    407     return false;
    408 
    409   if (const Decl *ND = Gen->GetDeclForMangledName(D.getFunction().getName())) {
    410     Diags.Report(ND->getASTContext().getFullLoc(ND->getLocation()),
    411                  diag::warn_fe_frame_larger_than)
    412         << D.getStackSize() << Decl::castToDeclContext(ND);
    413     return true;
    414   }
    415 
    416   return false;
    417 }
    418 
    419 void BackendConsumer::EmitOptimizationRemark(
    420     const llvm::DiagnosticInfoOptimizationRemarkBase &D, unsigned DiagID) {
    421   // We only support remarks.
    422   assert(D.getSeverity() == llvm::DS_Remark);
    423 
    424   SourceManager &SourceMgr = Context->getSourceManager();
    425   FileManager &FileMgr = SourceMgr.getFileManager();
    426   StringRef Filename;
    427   unsigned Line, Column;
    428   D.getLocation(&Filename, &Line, &Column);
    429   SourceLocation DILoc;
    430   const FileEntry *FE = FileMgr.getFile(Filename);
    431   if (FE && Line > 0) {
    432     // If -gcolumn-info was not used, Column will be 0. This upsets the
    433     // source manager, so pass 1 if Column is not set.
    434     DILoc = SourceMgr.translateFileLineCol(FE, Line, Column ? Column : 1);
    435   }
    436 
    437   // If a location isn't available, try to approximate it using the associated
    438   // function definition. We use the definition's right brace to differentiate
    439   // from diagnostics that genuinely relate to the function itself.
    440   FullSourceLoc Loc(DILoc, SourceMgr);
    441   if (Loc.isInvalid())
    442     if (const Decl *FD = Gen->GetDeclForMangledName(D.getFunction().getName()))
    443       Loc = FD->getASTContext().getFullLoc(FD->getBodyRBrace());
    444 
    445   Diags.Report(Loc, DiagID) << AddFlagValue(D.getPassName())
    446                             << D.getMsg().str();
    447 
    448   if (DILoc.isInvalid())
    449     // If we were not able to translate the file:line:col information
    450     // back to a SourceLocation, at least emit a note stating that
    451     // we could not translate this location. This can happen in the
    452     // case of #line directives.
    453     Diags.Report(Loc, diag::note_fe_backend_optimization_remark_invalid_loc)
    454         << Filename << Line << Column;
    455 }
    456 
    457 void BackendConsumer::OptimizationRemarkHandler(
    458     const llvm::DiagnosticInfoOptimizationRemark &D) {
    459   // Optimization remarks are active only if the -Rpass flag has a regular
    460   // expression that matches the name of the pass name in \p D.
    461   if (CodeGenOpts.OptimizationRemarkPattern &&
    462       CodeGenOpts.OptimizationRemarkPattern->match(D.getPassName()))
    463     EmitOptimizationRemark(D, diag::remark_fe_backend_optimization_remark);
    464 }
    465 
    466 void BackendConsumer::OptimizationRemarkHandler(
    467     const llvm::DiagnosticInfoOptimizationRemarkMissed &D) {
    468   // Missed optimization remarks are active only if the -Rpass-missed
    469   // flag has a regular expression that matches the name of the pass
    470   // name in \p D.
    471   if (CodeGenOpts.OptimizationRemarkMissedPattern &&
    472       CodeGenOpts.OptimizationRemarkMissedPattern->match(D.getPassName()))
    473     EmitOptimizationRemark(D,
    474                            diag::remark_fe_backend_optimization_remark_missed);
    475 }
    476 
    477 void BackendConsumer::OptimizationRemarkHandler(
    478     const llvm::DiagnosticInfoOptimizationRemarkAnalysis &D) {
    479   // Optimization analysis remarks are active only if the -Rpass-analysis
    480   // flag has a regular expression that matches the name of the pass
    481   // name in \p D.
    482   if (CodeGenOpts.OptimizationRemarkAnalysisPattern &&
    483       CodeGenOpts.OptimizationRemarkAnalysisPattern->match(D.getPassName()))
    484     EmitOptimizationRemark(
    485         D, diag::remark_fe_backend_optimization_remark_analysis);
    486 }
    487 
    488 /// \brief This function is invoked when the backend needs
    489 /// to report something to the user.
    490 void BackendConsumer::DiagnosticHandlerImpl(const DiagnosticInfo &DI) {
    491   unsigned DiagID = diag::err_fe_inline_asm;
    492   llvm::DiagnosticSeverity Severity = DI.getSeverity();
    493   // Get the diagnostic ID based.
    494   switch (DI.getKind()) {
    495   case llvm::DK_InlineAsm:
    496     if (InlineAsmDiagHandler(cast<DiagnosticInfoInlineAsm>(DI)))
    497       return;
    498     ComputeDiagID(Severity, inline_asm, DiagID);
    499     break;
    500   case llvm::DK_StackSize:
    501     if (StackSizeDiagHandler(cast<DiagnosticInfoStackSize>(DI)))
    502       return;
    503     ComputeDiagID(Severity, backend_frame_larger_than, DiagID);
    504     break;
    505   case llvm::DK_OptimizationRemark:
    506     // Optimization remarks are always handled completely by this
    507     // handler. There is no generic way of emitting them.
    508     OptimizationRemarkHandler(cast<DiagnosticInfoOptimizationRemark>(DI));
    509     return;
    510   case llvm::DK_OptimizationRemarkMissed:
    511     // Optimization remarks are always handled completely by this
    512     // handler. There is no generic way of emitting them.
    513     OptimizationRemarkHandler(cast<DiagnosticInfoOptimizationRemarkMissed>(DI));
    514     return;
    515   case llvm::DK_OptimizationRemarkAnalysis:
    516     // Optimization remarks are always handled completely by this
    517     // handler. There is no generic way of emitting them.
    518     OptimizationRemarkHandler(
    519         cast<DiagnosticInfoOptimizationRemarkAnalysis>(DI));
    520     return;
    521   default:
    522     // Plugin IDs are not bound to any value as they are set dynamically.
    523     ComputeDiagRemarkID(Severity, backend_plugin, DiagID);
    524     break;
    525   }
    526   std::string MsgStorage;
    527   {
    528     raw_string_ostream Stream(MsgStorage);
    529     DiagnosticPrinterRawOStream DP(Stream);
    530     DI.print(DP);
    531   }
    532 
    533   // Report the backend message using the usual diagnostic mechanism.
    534   FullSourceLoc Loc;
    535   Diags.Report(Loc, DiagID).AddString(MsgStorage);
    536 }
    537 #undef ComputeDiagID
    538 
    539 CodeGenAction::CodeGenAction(unsigned _Act, LLVMContext *_VMContext)
    540   : Act(_Act), LinkModule(nullptr),
    541     VMContext(_VMContext ? _VMContext : new LLVMContext),
    542     OwnsVMContext(!_VMContext) {}
    543 
    544 CodeGenAction::~CodeGenAction() {
    545   TheModule.reset();
    546   if (OwnsVMContext)
    547     delete VMContext;
    548 }
    549 
    550 bool CodeGenAction::hasIRSupport() const { return true; }
    551 
    552 void CodeGenAction::EndSourceFileAction() {
    553   // If the consumer creation failed, do nothing.
    554   if (!getCompilerInstance().hasASTConsumer())
    555     return;
    556 
    557   // If we were given a link module, release consumer's ownership of it.
    558   if (LinkModule)
    559     BEConsumer->takeLinkModule();
    560 
    561   // Steal the module from the consumer.
    562   TheModule.reset(BEConsumer->takeModule());
    563 }
    564 
    565 llvm::Module *CodeGenAction::takeModule() { return TheModule.release(); }
    566 
    567 llvm::LLVMContext *CodeGenAction::takeLLVMContext() {
    568   OwnsVMContext = false;
    569   return VMContext;
    570 }
    571 
    572 static raw_ostream *GetOutputStream(CompilerInstance &CI,
    573                                     StringRef InFile,
    574                                     BackendAction Action) {
    575   switch (Action) {
    576   case Backend_EmitAssembly:
    577     return CI.createDefaultOutputFile(false, InFile, "s");
    578   case Backend_EmitLL:
    579     return CI.createDefaultOutputFile(false, InFile, "ll");
    580   case Backend_EmitBC:
    581     return CI.createDefaultOutputFile(true, InFile, "bc");
    582   case Backend_EmitNothing:
    583     return nullptr;
    584   case Backend_EmitMCNull:
    585     return CI.createNullOutputFile();
    586   case Backend_EmitObj:
    587     return CI.createDefaultOutputFile(true, InFile, "o");
    588   }
    589 
    590   llvm_unreachable("Invalid action!");
    591 }
    592 
    593 ASTConsumer *CodeGenAction::CreateASTConsumer(CompilerInstance &CI,
    594                                               StringRef InFile) {
    595   BackendAction BA = static_cast<BackendAction>(Act);
    596   std::unique_ptr<raw_ostream> OS(GetOutputStream(CI, InFile, BA));
    597   if (BA != Backend_EmitNothing && !OS)
    598     return nullptr;
    599 
    600   llvm::Module *LinkModuleToUse = LinkModule;
    601 
    602   // If we were not given a link module, and the user requested that one be
    603   // loaded from bitcode, do so now.
    604   const std::string &LinkBCFile = CI.getCodeGenOpts().LinkBitcodeFile;
    605   if (!LinkModuleToUse && !LinkBCFile.empty()) {
    606     std::string ErrorStr;
    607 
    608     llvm::MemoryBuffer *BCBuf =
    609       CI.getFileManager().getBufferForFile(LinkBCFile, &ErrorStr);
    610     if (!BCBuf) {
    611       CI.getDiagnostics().Report(diag::err_cannot_open_file)
    612         << LinkBCFile << ErrorStr;
    613       return nullptr;
    614     }
    615 
    616     ErrorOr<llvm::Module *> ModuleOrErr =
    617         getLazyBitcodeModule(BCBuf, *VMContext);
    618     if (std::error_code EC = ModuleOrErr.getError()) {
    619       CI.getDiagnostics().Report(diag::err_cannot_open_file)
    620         << LinkBCFile << EC.message();
    621       return nullptr;
    622     }
    623     LinkModuleToUse = ModuleOrErr.get();
    624   }
    625 
    626   BEConsumer = new BackendConsumer(BA, CI.getDiagnostics(), CI.getCodeGenOpts(),
    627                                    CI.getTargetOpts(), CI.getLangOpts(),
    628                                    CI.getFrontendOpts().ShowTimers, InFile,
    629                                    LinkModuleToUse, OS.release(), *VMContext);
    630   return BEConsumer;
    631 }
    632 
    633 void CodeGenAction::ExecuteAction() {
    634   // If this is an IR file, we have to treat it specially.
    635   if (getCurrentFileKind() == IK_LLVM_IR) {
    636     BackendAction BA = static_cast<BackendAction>(Act);
    637     CompilerInstance &CI = getCompilerInstance();
    638     raw_ostream *OS = GetOutputStream(CI, getCurrentFile(), BA);
    639     if (BA != Backend_EmitNothing && !OS)
    640       return;
    641 
    642     bool Invalid;
    643     SourceManager &SM = CI.getSourceManager();
    644     FileID FID = SM.getMainFileID();
    645     llvm::MemoryBuffer *MainFile = SM.getBuffer(FID, &Invalid);
    646     if (Invalid)
    647       return;
    648 
    649     llvm::SMDiagnostic Err;
    650     TheModule.reset(ParseIR(MainFile, Err, *VMContext));
    651     if (!TheModule) {
    652       // Translate from the diagnostic info to the SourceManager location if
    653       // available.
    654       // TODO: Unify this with ConvertBackendLocation()
    655       SourceLocation Loc;
    656       if (Err.getLineNo() > 0) {
    657         assert(Err.getColumnNo() >= 0);
    658         Loc = SM.translateFileLineCol(SM.getFileEntryForID(FID),
    659                                       Err.getLineNo(), Err.getColumnNo() + 1);
    660       }
    661 
    662       // Strip off a leading diagnostic code if there is one.
    663       StringRef Msg = Err.getMessage();
    664       if (Msg.startswith("error: "))
    665         Msg = Msg.substr(7);
    666 
    667       unsigned DiagID =
    668           CI.getDiagnostics().getCustomDiagID(DiagnosticsEngine::Error, "%0");
    669 
    670       CI.getDiagnostics().Report(Loc, DiagID) << Msg;
    671       return;
    672     }
    673     const TargetOptions &TargetOpts = CI.getTargetOpts();
    674     if (TheModule->getTargetTriple() != TargetOpts.Triple) {
    675       unsigned DiagID = CI.getDiagnostics().getCustomDiagID(
    676           DiagnosticsEngine::Warning,
    677           "overriding the module target triple with %0");
    678 
    679       CI.getDiagnostics().Report(SourceLocation(), DiagID) << TargetOpts.Triple;
    680       TheModule->setTargetTriple(TargetOpts.Triple);
    681     }
    682 
    683     EmitBackendOutput(CI.getDiagnostics(), CI.getCodeGenOpts(), TargetOpts,
    684                       CI.getLangOpts(), CI.getTarget().getTargetDescription(),
    685                       TheModule.get(), BA, OS);
    686     return;
    687   }
    688 
    689   // Otherwise follow the normal AST path.
    690   this->ASTFrontendAction::ExecuteAction();
    691 }
    692 
    693 //
    694 
    695 void EmitAssemblyAction::anchor() { }
    696 EmitAssemblyAction::EmitAssemblyAction(llvm::LLVMContext *_VMContext)
    697   : CodeGenAction(Backend_EmitAssembly, _VMContext) {}
    698 
    699 void EmitBCAction::anchor() { }
    700 EmitBCAction::EmitBCAction(llvm::LLVMContext *_VMContext)
    701   : CodeGenAction(Backend_EmitBC, _VMContext) {}
    702 
    703 void EmitLLVMAction::anchor() { }
    704 EmitLLVMAction::EmitLLVMAction(llvm::LLVMContext *_VMContext)
    705   : CodeGenAction(Backend_EmitLL, _VMContext) {}
    706 
    707 void EmitLLVMOnlyAction::anchor() { }
    708 EmitLLVMOnlyAction::EmitLLVMOnlyAction(llvm::LLVMContext *_VMContext)
    709   : CodeGenAction(Backend_EmitNothing, _VMContext) {}
    710 
    711 void EmitCodeGenOnlyAction::anchor() { }
    712 EmitCodeGenOnlyAction::EmitCodeGenOnlyAction(llvm::LLVMContext *_VMContext)
    713   : CodeGenAction(Backend_EmitMCNull, _VMContext) {}
    714 
    715 void EmitObjAction::anchor() { }
    716 EmitObjAction::EmitObjAction(llvm::LLVMContext *_VMContext)
    717   : CodeGenAction(Backend_EmitObj, _VMContext) {}
    718