contrib/llvm/tools/clang/lib/FrontendTool/ExecuteCompilerInvocation.cpp

   1 //===--- ExecuteCompilerInvocation.cpp ------------------------------------===//
   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 // This file holds ExecuteCompilerInvocation(). It is split into its own file to
  11 // minimize the impact of pulling in essentially everything else in Clang.
  12 //
  13 //===----------------------------------------------------------------------===//
  14
  15 #include "clang/FrontendTool/Utils.h"
  16 #include "clang/StaticAnalyzer/Frontend/FrontendActions.h"
  17 #include "clang/ARCMigrate/ARCMTActions.h"
  18 #include "clang/CodeGen/CodeGenAction.h"
  19 #include "clang/Driver/CC1Options.h"
  20 #include "clang/Driver/OptTable.h"
  21 #include "clang/Frontend/CompilerInvocation.h"
  22 #include "clang/Frontend/CompilerInstance.h"
  23 #include "clang/Frontend/FrontendActions.h"
  24 #include "clang/Frontend/FrontendDiagnostic.h"
  25 #include "clang/Frontend/FrontendPluginRegistry.h"
  26 #include "clang/Rewrite/FrontendActions.h"
  27 #include "llvm/Support/ErrorHandling.h"
  28 #include "llvm/Support/DynamicLibrary.h"
  29 using namespace clang;
  30
  31 static FrontendAction *CreateFrontendBaseAction(CompilerInstance &CI) {
  32   using namespace clang::frontend;
  33
  34   switch (CI.getFrontendOpts().ProgramAction) {
  35   default:
  36     llvm_unreachable("Invalid program action!");
  37
  38   case ASTDump:                return new ASTDumpAction();
  39   case ASTDumpXML:             return new ASTDumpXMLAction();
  40   case ASTPrint:               return new ASTPrintAction();
  41   case ASTView:                return new ASTViewAction();
  42   case CreateModule:           return 0;
  43   case DumpRawTokens:          return new DumpRawTokensAction();
  44   case DumpTokens:             return new DumpTokensAction();
  45   case EmitAssembly:           return new EmitAssemblyAction();
  46   case EmitBC:                 return new EmitBCAction();
  47   case EmitHTML:               return new HTMLPrintAction();
  48   case EmitLLVM:               return new EmitLLVMAction();
  49   case EmitLLVMOnly:           return new EmitLLVMOnlyAction();
  50   case EmitCodeGenOnly:        return new EmitCodeGenOnlyAction();
  51   case EmitObj:                return new EmitObjAction();
  52   case FixIt:                  return new FixItAction();
  53   case GeneratePCH:            return new GeneratePCHAction();
  54   case GeneratePTH:            return new GeneratePTHAction();
  55   case InitOnly:               return new InitOnlyAction();
  56   case ParseSyntaxOnly:        return new SyntaxOnlyAction();
  57
  58   case PluginAction: {
  59     for (FrontendPluginRegistry::iterator it =
  60            FrontendPluginRegistry::begin(), ie = FrontendPluginRegistry::end();
  61          it != ie; ++it) {
  62       if (it->getName() == CI.getFrontendOpts().ActionName) {
  63         llvm::OwningPtr<PluginASTAction> P(it->instantiate());
  64         if (!P->ParseArgs(CI, CI.getFrontendOpts().PluginArgs))
  65           return 0;
  66         return P.take();
  67       }
  68     }
  69
  70     CI.getDiagnostics().Report(diag::err_fe_invalid_plugin_name)
  71       << CI.getFrontendOpts().ActionName;
  72     return 0;
  73   }
  74
  75   case PrintDeclContext:       return new DeclContextPrintAction();
  76   case PrintPreamble:          return new PrintPreambleAction();
  77   case PrintPreprocessedInput: return new PrintPreprocessedAction();
  78   case RewriteMacros:          return new RewriteMacrosAction();
  79   case RewriteObjC:            return new RewriteObjCAction();
  80   case RewriteTest:            return new RewriteTestAction();
  81   case RunAnalysis:            return new ento::AnalysisAction();
  82   case RunPreprocessorOnly:    return new PreprocessOnlyAction();
  83   }
  84 }
  85
  86 static FrontendAction *CreateFrontendAction(CompilerInstance &CI) {
  87   // Create the underlying action.
  88   FrontendAction *Act = CreateFrontendBaseAction(CI);
  89   if (!Act)
  90     return 0;
  91
  92   // Potentially wrap the base FE action in an ARC Migrate Tool action.
  93   switch (CI.getFrontendOpts().ARCMTAction) {
  94   case FrontendOptions::ARCMT_None:
  95     break;
  96   case FrontendOptions::ARCMT_Check:
  97     Act = new arcmt::CheckAction(Act);
  98     break;
  99   case FrontendOptions::ARCMT_Modify:
 100     Act = new arcmt::ModifyAction(Act);
 101     break;
 102   case FrontendOptions::ARCMT_Migrate:
 103     Act = new arcmt::MigrateAction(Act, CI.getFrontendOpts().ARCMTMigrateDir);
 104     break;
 105   }
 106
 107   // If there are any AST files to merge, create a frontend action
 108   // adaptor to perform the merge.
 109   if (!CI.getFrontendOpts().ASTMergeFiles.empty())
 110     Act = new ASTMergeAction(Act, &CI.getFrontendOpts().ASTMergeFiles[0],
 111                              CI.getFrontendOpts().ASTMergeFiles.size());
 112
 113   return Act;
 114 }
 115
 116 bool clang::ExecuteCompilerInvocation(CompilerInstance *Clang) {
 117   // Honor -help.
 118   if (Clang->getFrontendOpts().ShowHelp) {
 119     llvm::OwningPtr<driver::OptTable> Opts(driver::createCC1OptTable());
 120     Opts->PrintHelp(llvm::outs(), "clang -cc1",
 121                     "LLVM 'Clang' Compiler: http://clang.llvm.org");
 122     return 0;
 123   }
 124
 125   // Honor -analyzer-checker-help.
 126   if (Clang->getAnalyzerOpts().ShowCheckerHelp) {
 127     ento::printCheckerHelp(llvm::outs());
 128     return 0;
 129   }
 130
 131   // Honor -version.
 132   //
 133   // FIXME: Use a better -version message?
 134   if (Clang->getFrontendOpts().ShowVersion) {
 135     llvm::cl::PrintVersionMessage();
 136     return 0;
 137   }
 138
 139   // Honor -mllvm.
 140   //
 141   // FIXME: Remove this, one day.
 142   if (!Clang->getFrontendOpts().LLVMArgs.empty()) {
 143     unsigned NumArgs = Clang->getFrontendOpts().LLVMArgs.size();
 144     const char **Args = new const char*[NumArgs + 2];
 145     Args[0] = "clang (LLVM option parsing)";
 146     for (unsigned i = 0; i != NumArgs; ++i)
 147       Args[i + 1] = Clang->getFrontendOpts().LLVMArgs[i].c_str();
 148     Args[NumArgs + 1] = 0;
 149     llvm::cl::ParseCommandLineOptions(NumArgs + 1, const_cast<char **>(Args));
 150   }
 151
 152   // Load any requested plugins.
 153   for (unsigned i = 0,
 154          e = Clang->getFrontendOpts().Plugins.size(); i != e; ++i) {
 155     const std::string &Path = Clang->getFrontendOpts().Plugins[i];
 156     std::string Error;
 157     if (llvm::sys::DynamicLibrary::LoadLibraryPermanently(Path.c_str(), &Error))
 158       Clang->getDiagnostics().Report(diag::err_fe_unable_to_load_plugin)
 159         << Path << Error;
 160   }
 161
 162   // If there were errors in processing arguments, don't do anything else.
 163   bool Success = false;
 164   if (!Clang->getDiagnostics().hasErrorOccurred()) {
 165     // Create and execute the frontend action.
 166     llvm::OwningPtr<FrontendAction> Act(CreateFrontendAction(*Clang));
 167     if (Act) {
 168       Success = Clang->ExecuteAction(*Act);
 169       if (Clang->getFrontendOpts().DisableFree)
 170         Act.take();
 171     }
 172   }
 173
 174   return Success;
 175 }