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