Daniel Dunbar | 41b5b17 | 2010-05-20 17:49:16 +0000 | [diff] [blame] | 1 | //===-- cc1_main.cpp - Clang CC1 Compiler Frontend ------------------------===// |
Daniel Dunbar | 217acbf | 2009-11-19 07:37:51 +0000 | [diff] [blame] | 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 | // |
Daniel Dunbar | 2108577 | 2009-12-11 22:20:12 +0000 | [diff] [blame] | 10 | // This is the entry point to the clang -cc1 functionality, which implements the |
| 11 | // core compiler functionality along with a number of additional tools for |
| 12 | // demonstration and testing purposes. |
Daniel Dunbar | 217acbf | 2009-11-19 07:37:51 +0000 | [diff] [blame] | 13 | // |
| 14 | //===----------------------------------------------------------------------===// |
| 15 | |
Daniel Dunbar | 2108577 | 2009-12-11 22:20:12 +0000 | [diff] [blame] | 16 | #include "clang/Driver/Arg.h" |
| 17 | #include "clang/Driver/ArgList.h" |
| 18 | #include "clang/Driver/CC1Options.h" |
| 19 | #include "clang/Driver/DriverDiagnostic.h" |
| 20 | #include "clang/Driver/OptTable.h" |
| 21 | #include "clang/Frontend/CompilerInstance.h" |
| 22 | #include "clang/Frontend/CompilerInvocation.h" |
Daniel Dunbar | 2108577 | 2009-12-11 22:20:12 +0000 | [diff] [blame] | 23 | #include "clang/Frontend/FrontendDiagnostic.h" |
Daniel Dunbar | 2108577 | 2009-12-11 22:20:12 +0000 | [diff] [blame] | 24 | #include "clang/Frontend/TextDiagnosticBuffer.h" |
| 25 | #include "clang/Frontend/TextDiagnosticPrinter.h" |
Peter Collingbourne | 1b7255d | 2010-08-24 00:31:22 +0000 | [diff] [blame] | 26 | #include "clang/FrontendTool/Utils.h" |
Douglas Gregor | 95dd558 | 2010-03-30 17:33:59 +0000 | [diff] [blame] | 27 | #include "llvm/ADT/Statistic.h" |
Daniel Dunbar | 2108577 | 2009-12-11 22:20:12 +0000 | [diff] [blame] | 28 | #include "llvm/Support/ErrorHandling.h" |
| 29 | #include "llvm/Support/ManagedStatic.h" |
Chris Lattner | 30bc7e8 | 2010-03-30 05:39:52 +0000 | [diff] [blame] | 30 | #include "llvm/Support/Timer.h" |
Daniel Dunbar | 217acbf | 2009-11-19 07:37:51 +0000 | [diff] [blame] | 31 | #include "llvm/Support/raw_ostream.h" |
Daniel Dunbar | 2108577 | 2009-12-11 22:20:12 +0000 | [diff] [blame] | 32 | #include "llvm/Target/TargetSelect.h" |
| 33 | #include <cstdio> |
| 34 | using namespace clang; |
Daniel Dunbar | 217acbf | 2009-11-19 07:37:51 +0000 | [diff] [blame] | 35 | |
Daniel Dunbar | 2108577 | 2009-12-11 22:20:12 +0000 | [diff] [blame] | 36 | //===----------------------------------------------------------------------===// |
| 37 | // Main driver |
| 38 | //===----------------------------------------------------------------------===// |
| 39 | |
Daniel Dunbar | 41b5b17 | 2010-05-20 17:49:16 +0000 | [diff] [blame] | 40 | static void LLVMErrorHandler(void *UserData, const std::string &Message) { |
Daniel Dunbar | 2108577 | 2009-12-11 22:20:12 +0000 | [diff] [blame] | 41 | Diagnostic &Diags = *static_cast<Diagnostic*>(UserData); |
| 42 | |
| 43 | Diags.Report(diag::err_fe_error_backend) << Message; |
| 44 | |
| 45 | // We cannot recover from llvm errors. |
| 46 | exit(1); |
| 47 | } |
| 48 | |
Daniel Dunbar | 2108577 | 2009-12-11 22:20:12 +0000 | [diff] [blame] | 49 | // FIXME: Define the need for this testing away. |
| 50 | static int cc1_test(Diagnostic &Diags, |
Daniel Dunbar | 1e69fe3 | 2009-12-13 03:45:58 +0000 | [diff] [blame] | 51 | const char **ArgBegin, const char **ArgEnd) { |
Daniel Dunbar | 2108577 | 2009-12-11 22:20:12 +0000 | [diff] [blame] | 52 | using namespace clang::driver; |
| 53 | |
Daniel Dunbar | 217acbf | 2009-11-19 07:37:51 +0000 | [diff] [blame] | 54 | llvm::errs() << "cc1 argv:"; |
| 55 | for (const char **i = ArgBegin; i != ArgEnd; ++i) |
| 56 | llvm::errs() << " \"" << *i << '"'; |
| 57 | llvm::errs() << "\n"; |
| 58 | |
Daniel Dunbar | 2108577 | 2009-12-11 22:20:12 +0000 | [diff] [blame] | 59 | // Parse the arguments. |
| 60 | OptTable *Opts = createCC1OptTable(); |
| 61 | unsigned MissingArgIndex, MissingArgCount; |
| 62 | InputArgList *Args = Opts->ParseArgs(ArgBegin, ArgEnd, |
| 63 | MissingArgIndex, MissingArgCount); |
| 64 | |
| 65 | // Check for missing argument error. |
| 66 | if (MissingArgCount) |
| 67 | Diags.Report(clang::diag::err_drv_missing_argument) |
| 68 | << Args->getArgString(MissingArgIndex) << MissingArgCount; |
| 69 | |
| 70 | // Dump the parsed arguments. |
| 71 | llvm::errs() << "cc1 parsed options:\n"; |
| 72 | for (ArgList::const_iterator it = Args->begin(), ie = Args->end(); |
| 73 | it != ie; ++it) |
| 74 | (*it)->dump(); |
| 75 | |
| 76 | // Create a compiler invocation. |
| 77 | llvm::errs() << "cc1 creating invocation.\n"; |
| 78 | CompilerInvocation Invocation; |
Daniel Dunbar | 1e69fe3 | 2009-12-13 03:45:58 +0000 | [diff] [blame] | 79 | CompilerInvocation::CreateFromArgs(Invocation, ArgBegin, ArgEnd, Diags); |
Daniel Dunbar | 2108577 | 2009-12-11 22:20:12 +0000 | [diff] [blame] | 80 | |
| 81 | // Convert the invocation back to argument strings. |
| 82 | std::vector<std::string> InvocationArgs; |
| 83 | Invocation.toArgs(InvocationArgs); |
| 84 | |
| 85 | // Dump the converted arguments. |
| 86 | llvm::SmallVector<const char*, 32> Invocation2Args; |
| 87 | llvm::errs() << "invocation argv :"; |
| 88 | for (unsigned i = 0, e = InvocationArgs.size(); i != e; ++i) { |
| 89 | Invocation2Args.push_back(InvocationArgs[i].c_str()); |
| 90 | llvm::errs() << " \"" << InvocationArgs[i] << '"'; |
| 91 | } |
| 92 | llvm::errs() << "\n"; |
| 93 | |
| 94 | // Convert those arguments to another invocation, and check that we got the |
| 95 | // same thing. |
| 96 | CompilerInvocation Invocation2; |
| 97 | CompilerInvocation::CreateFromArgs(Invocation2, Invocation2Args.begin(), |
Daniel Dunbar | 1e69fe3 | 2009-12-13 03:45:58 +0000 | [diff] [blame] | 98 | Invocation2Args.end(), Diags); |
Daniel Dunbar | 2108577 | 2009-12-11 22:20:12 +0000 | [diff] [blame] | 99 | |
| 100 | // FIXME: Implement CompilerInvocation comparison. |
| 101 | if (true) { |
| 102 | //llvm::errs() << "warning: Invocations differ!\n"; |
| 103 | |
| 104 | std::vector<std::string> Invocation2Args; |
| 105 | Invocation2.toArgs(Invocation2Args); |
| 106 | llvm::errs() << "invocation2 argv:"; |
| 107 | for (unsigned i = 0, e = Invocation2Args.size(); i != e; ++i) |
| 108 | llvm::errs() << " \"" << Invocation2Args[i] << '"'; |
| 109 | llvm::errs() << "\n"; |
| 110 | } |
| 111 | |
Daniel Dunbar | 217acbf | 2009-11-19 07:37:51 +0000 | [diff] [blame] | 112 | return 0; |
| 113 | } |
Daniel Dunbar | 2108577 | 2009-12-11 22:20:12 +0000 | [diff] [blame] | 114 | |
| 115 | int cc1_main(const char **ArgBegin, const char **ArgEnd, |
| 116 | const char *Argv0, void *MainAddr) { |
Daniel Dunbar | 42444fb | 2010-03-23 05:09:16 +0000 | [diff] [blame] | 117 | llvm::OwningPtr<CompilerInstance> Clang(new CompilerInstance()); |
Argyrios Kyrtzidis | 33e4e70 | 2010-11-18 20:06:41 +0000 | [diff] [blame] | 118 | llvm::IntrusiveRefCntPtr<DiagnosticIDs> DiagID(new DiagnosticIDs()); |
Daniel Dunbar | 42e9f8e4 | 2010-02-16 01:54:47 +0000 | [diff] [blame] | 119 | |
Daniel Dunbar | 2108577 | 2009-12-11 22:20:12 +0000 | [diff] [blame] | 120 | // Run clang -cc1 test. |
| 121 | if (ArgBegin != ArgEnd && llvm::StringRef(ArgBegin[0]) == "-cc1test") { |
Argyrios Kyrtzidis | 33e4e70 | 2010-11-18 20:06:41 +0000 | [diff] [blame] | 122 | Diagnostic Diags(DiagID, new TextDiagnosticPrinter(llvm::errs(), |
| 123 | DiagnosticOptions())); |
Daniel Dunbar | 1e69fe3 | 2009-12-13 03:45:58 +0000 | [diff] [blame] | 124 | return cc1_test(Diags, ArgBegin + 1, ArgEnd); |
Daniel Dunbar | 2108577 | 2009-12-11 22:20:12 +0000 | [diff] [blame] | 125 | } |
| 126 | |
| 127 | // Initialize targets first, so that --version shows registered targets. |
| 128 | llvm::InitializeAllTargets(); |
| 129 | llvm::InitializeAllAsmPrinters(); |
Chris Lattner | 01ae93f | 2010-04-05 23:33:20 +0000 | [diff] [blame] | 130 | llvm::InitializeAllAsmParsers(); |
Daniel Dunbar | 2108577 | 2009-12-11 22:20:12 +0000 | [diff] [blame] | 131 | |
| 132 | // Buffer diagnostics from argument parsing so that we can output them using a |
| 133 | // well formed diagnostic object. |
Douglas Gregor | bdbb004 | 2010-08-18 22:29:43 +0000 | [diff] [blame] | 134 | TextDiagnosticBuffer *DiagsBuffer = new TextDiagnosticBuffer; |
Argyrios Kyrtzidis | 33e4e70 | 2010-11-18 20:06:41 +0000 | [diff] [blame] | 135 | Diagnostic Diags(DiagID, DiagsBuffer); |
Daniel Dunbar | 42444fb | 2010-03-23 05:09:16 +0000 | [diff] [blame] | 136 | CompilerInvocation::CreateFromArgs(Clang->getInvocation(), ArgBegin, ArgEnd, |
Daniel Dunbar | 1e69fe3 | 2009-12-13 03:45:58 +0000 | [diff] [blame] | 137 | Diags); |
| 138 | |
| 139 | // Infer the builtin include path if unspecified. |
Daniel Dunbar | 42444fb | 2010-03-23 05:09:16 +0000 | [diff] [blame] | 140 | if (Clang->getHeaderSearchOpts().UseBuiltinIncludes && |
| 141 | Clang->getHeaderSearchOpts().ResourceDir.empty()) |
| 142 | Clang->getHeaderSearchOpts().ResourceDir = |
Daniel Dunbar | 8b9adfe | 2009-12-15 00:06:45 +0000 | [diff] [blame] | 143 | CompilerInvocation::GetResourcesPath(Argv0, MainAddr); |
Daniel Dunbar | 2108577 | 2009-12-11 22:20:12 +0000 | [diff] [blame] | 144 | |
Daniel Dunbar | a6bf47f | 2010-08-12 02:53:07 +0000 | [diff] [blame] | 145 | // Create the actual diagnostics engine. |
| 146 | Clang->createDiagnostics(ArgEnd - ArgBegin, const_cast<char**>(ArgBegin)); |
| 147 | if (!Clang->hasDiagnostics()) |
| 148 | return 1; |
| 149 | |
| 150 | // Set an error handler, so that any LLVM backend diagnostics go through our |
| 151 | // error handler. |
| 152 | llvm::install_fatal_error_handler(LLVMErrorHandler, |
| 153 | static_cast<void*>(&Clang->getDiagnostics())); |
| 154 | |
Douglas Gregor | bdbb004 | 2010-08-18 22:29:43 +0000 | [diff] [blame] | 155 | DiagsBuffer->FlushDiagnostics(Clang->getDiagnostics()); |
Daniel Dunbar | a6bf47f | 2010-08-12 02:53:07 +0000 | [diff] [blame] | 156 | |
Daniel Dunbar | 8eb2b01 | 2010-08-12 02:53:12 +0000 | [diff] [blame] | 157 | // Execute the frontend actions. |
| 158 | bool Success = ExecuteCompilerInvocation(Clang.get()); |
Daniel Dunbar | 42444fb | 2010-03-23 05:09:16 +0000 | [diff] [blame] | 159 | |
Chris Lattner | 30bc7e8 | 2010-03-30 05:39:52 +0000 | [diff] [blame] | 160 | // If any timers were active but haven't been destroyed yet, print their |
| 161 | // results now. This happens in -disable-free mode. |
| 162 | llvm::TimerGroup::printAll(llvm::errs()); |
Daniel Dunbar | f56a488 | 2010-08-02 15:31:28 +0000 | [diff] [blame] | 163 | |
Dan Gohman | 726578c | 2010-08-18 21:23:17 +0000 | [diff] [blame] | 164 | // Our error handler depends on the Diagnostics object, which we're |
| 165 | // potentially about to delete. Uninstall the handler now so that any |
| 166 | // later errors use the default handling behavior instead. |
| 167 | llvm::remove_fatal_error_handler(); |
| 168 | |
Daniel Dunbar | 42444fb | 2010-03-23 05:09:16 +0000 | [diff] [blame] | 169 | // When running with -disable-free, don't do any destruction or shutdown. |
| 170 | if (Clang->getFrontendOpts().DisableFree) { |
Benjamin Kramer | 77d2c5f | 2011-02-27 18:07:41 +0000 | [diff] [blame] | 171 | if (llvm::AreStatisticsEnabled() || Clang->getFrontendOpts().ShowStats) |
Douglas Gregor | 95dd558 | 2010-03-30 17:33:59 +0000 | [diff] [blame] | 172 | llvm::PrintStatistics(); |
Daniel Dunbar | 42444fb | 2010-03-23 05:09:16 +0000 | [diff] [blame] | 173 | Clang.take(); |
| 174 | return !Success; |
Daniel Dunbar | 2108577 | 2009-12-11 22:20:12 +0000 | [diff] [blame] | 175 | } |
| 176 | |
Daniel Dunbar | 2108577 | 2009-12-11 22:20:12 +0000 | [diff] [blame] | 177 | // Managed static deconstruction. Useful for making things like |
| 178 | // -time-passes usable. |
| 179 | llvm::llvm_shutdown(); |
| 180 | |
Daniel Dunbar | 0397af2 | 2010-01-13 00:48:06 +0000 | [diff] [blame] | 181 | return !Success; |
Daniel Dunbar | 2108577 | 2009-12-11 22:20:12 +0000 | [diff] [blame] | 182 | } |