Reid Spencer | c0af3f0 | 2004-09-13 01:27:53 +0000 | [diff] [blame] | 1 | //===- llvm-ld.cpp - LLVM 'ld' compatible linker --------------------------===// |
Misha Brukman | 3da94ae | 2005-04-22 00:00:37 +0000 | [diff] [blame] | 2 | // |
Reid Spencer | c0af3f0 | 2004-09-13 01:27:53 +0000 | [diff] [blame] | 3 | // The LLVM Compiler Infrastructure |
| 4 | // |
Chris Lattner | 21c62da | 2007-12-29 20:44:31 +0000 | [diff] [blame] | 5 | // This file is distributed under the University of Illinois Open Source |
| 6 | // License. See LICENSE.TXT for details. |
Misha Brukman | 3da94ae | 2005-04-22 00:00:37 +0000 | [diff] [blame] | 7 | // |
Reid Spencer | c0af3f0 | 2004-09-13 01:27:53 +0000 | [diff] [blame] | 8 | //===----------------------------------------------------------------------===// |
| 9 | // |
| 10 | // This utility is intended to be compatible with GCC, and follows standard |
| 11 | // system 'ld' conventions. As such, the default output file is ./a.out. |
| 12 | // Additionally, this program outputs a shell script that is used to invoke LLI |
| 13 | // to execute the program. In this manner, the generated executable (a.out for |
Gabor Greif | a99be51 | 2007-07-05 17:07:56 +0000 | [diff] [blame] | 14 | // example), is directly executable, whereas the bitcode file actually lives in |
Dan Gohman | baa2639 | 2009-08-25 15:34:52 +0000 | [diff] [blame] | 15 | // the a.out.bc file generated by this program. |
Reid Spencer | c0af3f0 | 2004-09-13 01:27:53 +0000 | [diff] [blame] | 16 | // |
| 17 | // Note that if someone (or a script) deletes the executable program generated, |
| 18 | // the .bc file will be left around. Considering that this is a temporary hack, |
| 19 | // I'm not too worried about this. |
| 20 | // |
| 21 | //===----------------------------------------------------------------------===// |
| 22 | |
Reid Spencer | af303d5 | 2006-06-07 23:03:13 +0000 | [diff] [blame] | 23 | #include "llvm/LinkAllVMCore.h" |
Reid Spencer | 605b9e2 | 2004-11-14 23:00:08 +0000 | [diff] [blame] | 24 | #include "llvm/Linker.h" |
Owen Anderson | 8b477ed | 2009-07-01 16:58:40 +0000 | [diff] [blame] | 25 | #include "llvm/LLVMContext.h" |
Michael J. Spencer | 1f6efa3 | 2010-11-29 18:16:10 +0000 | [diff] [blame] | 26 | #include "llvm/Support/Program.h" |
Reid Spencer | c0af3f0 | 2004-09-13 01:27:53 +0000 | [diff] [blame] | 27 | #include "llvm/Module.h" |
| 28 | #include "llvm/PassManager.h" |
Chris Lattner | bb3f3d3 | 2007-05-06 05:56:58 +0000 | [diff] [blame] | 29 | #include "llvm/Bitcode/ReaderWriter.h" |
Reid Spencer | c0af3f0 | 2004-09-13 01:27:53 +0000 | [diff] [blame] | 30 | #include "llvm/Target/TargetData.h" |
Reid Spencer | aefd04b | 2004-09-25 16:00:07 +0000 | [diff] [blame] | 31 | #include "llvm/Target/TargetMachine.h" |
Reid Spencer | c0af3f0 | 2004-09-13 01:27:53 +0000 | [diff] [blame] | 32 | #include "llvm/Support/CommandLine.h" |
Dan Gohman | 132a994 | 2010-03-30 19:56:41 +0000 | [diff] [blame] | 33 | #include "llvm/Support/FileUtilities.h" |
Chris Lattner | c30598b | 2006-12-06 01:18:01 +0000 | [diff] [blame] | 34 | #include "llvm/Support/ManagedStatic.h" |
Chris Lattner | bb3f3d3 | 2007-05-06 05:56:58 +0000 | [diff] [blame] | 35 | #include "llvm/Support/MemoryBuffer.h" |
Chris Lattner | cc14d25 | 2009-03-06 05:34:10 +0000 | [diff] [blame] | 36 | #include "llvm/Support/PrettyStackTrace.h" |
Reid Spencer | c0af3f0 | 2004-09-13 01:27:53 +0000 | [diff] [blame] | 37 | #include "llvm/Support/SystemUtils.h" |
Dan Gohman | e4f1a9b | 2010-10-07 20:32:40 +0000 | [diff] [blame] | 38 | #include "llvm/Support/ToolOutputFile.h" |
Michael J. Spencer | 1f6efa3 | 2010-11-29 18:16:10 +0000 | [diff] [blame] | 39 | #include "llvm/Support/Signals.h" |
Chris Lattner | 5d5a897 | 2009-01-05 19:01:32 +0000 | [diff] [blame] | 40 | #include "llvm/Config/config.h" |
Reid Spencer | c0af3f0 | 2004-09-13 01:27:53 +0000 | [diff] [blame] | 41 | #include <memory> |
Anton Korobeynikov | ae9f3a3 | 2008-02-20 11:08:44 +0000 | [diff] [blame] | 42 | #include <cstring> |
Reid Spencer | c0af3f0 | 2004-09-13 01:27:53 +0000 | [diff] [blame] | 43 | using namespace llvm; |
| 44 | |
Dan Gohman | 197f728 | 2009-08-05 20:21:17 +0000 | [diff] [blame] | 45 | // Rightly this should go in a header file but it just seems such a waste. |
| 46 | namespace llvm { |
| 47 | extern void Optimize(Module*); |
| 48 | } |
| 49 | |
Reid Spencer | 445564a | 2004-11-20 20:02:56 +0000 | [diff] [blame] | 50 | // Input/Output Options |
| 51 | static cl::list<std::string> InputFilenames(cl::Positional, cl::OneOrMore, |
Gabor Greif | a99be51 | 2007-07-05 17:07:56 +0000 | [diff] [blame] | 52 | cl::desc("<input bitcode files>")); |
Reid Spencer | c0af3f0 | 2004-09-13 01:27:53 +0000 | [diff] [blame] | 53 | |
Reid Spencer | 445564a | 2004-11-20 20:02:56 +0000 | [diff] [blame] | 54 | static cl::opt<std::string> OutputFilename("o", cl::init("a.out"), |
Misha Brukman | 3da94ae | 2005-04-22 00:00:37 +0000 | [diff] [blame] | 55 | cl::desc("Override output filename"), |
Reid Spencer | 445564a | 2004-11-20 20:02:56 +0000 | [diff] [blame] | 56 | cl::value_desc("filename")); |
Reid Spencer | c0af3f0 | 2004-09-13 01:27:53 +0000 | [diff] [blame] | 57 | |
Sanjiv Gupta | d7de7bc | 2009-07-22 18:41:45 +0000 | [diff] [blame] | 58 | static cl::opt<std::string> BitcodeOutputFilename("b", cl::init(""), |
| 59 | cl::desc("Override bitcode output filename"), |
| 60 | cl::value_desc("filename")); |
| 61 | |
Misha Brukman | 3da94ae | 2005-04-22 00:00:37 +0000 | [diff] [blame] | 62 | static cl::opt<bool> Verbose("v", |
Reid Spencer | 445564a | 2004-11-20 20:02:56 +0000 | [diff] [blame] | 63 | cl::desc("Print information about actions taken")); |
Misha Brukman | 3da94ae | 2005-04-22 00:00:37 +0000 | [diff] [blame] | 64 | |
Reid Spencer | 445564a | 2004-11-20 20:02:56 +0000 | [diff] [blame] | 65 | static cl::list<std::string> LibPaths("L", cl::Prefix, |
Misha Brukman | 3da94ae | 2005-04-22 00:00:37 +0000 | [diff] [blame] | 66 | cl::desc("Specify a library search path"), |
Reid Spencer | 445564a | 2004-11-20 20:02:56 +0000 | [diff] [blame] | 67 | cl::value_desc("directory")); |
Reid Spencer | c0af3f0 | 2004-09-13 01:27:53 +0000 | [diff] [blame] | 68 | |
Chris Lattner | 3992f52 | 2008-01-27 22:58:59 +0000 | [diff] [blame] | 69 | static cl::list<std::string> FrameworkPaths("F", cl::Prefix, |
| 70 | cl::desc("Specify a framework search path"), |
| 71 | cl::value_desc("directory")); |
| 72 | |
Reid Spencer | 445564a | 2004-11-20 20:02:56 +0000 | [diff] [blame] | 73 | static cl::list<std::string> Libraries("l", cl::Prefix, |
Misha Brukman | 3da94ae | 2005-04-22 00:00:37 +0000 | [diff] [blame] | 74 | cl::desc("Specify libraries to link to"), |
Reid Spencer | 445564a | 2004-11-20 20:02:56 +0000 | [diff] [blame] | 75 | cl::value_desc("library prefix")); |
Reid Spencer | c0af3f0 | 2004-09-13 01:27:53 +0000 | [diff] [blame] | 76 | |
Chris Lattner | 3992f52 | 2008-01-27 22:58:59 +0000 | [diff] [blame] | 77 | static cl::list<std::string> Frameworks("framework", |
| 78 | cl::desc("Specify frameworks to link to"), |
| 79 | cl::value_desc("framework")); |
| 80 | |
Reid Spencer | 708585a | 2007-02-09 03:08:06 +0000 | [diff] [blame] | 81 | // Options to control the linking, optimization, and code gen processes |
Misha Brukman | 3da94ae | 2005-04-22 00:00:37 +0000 | [diff] [blame] | 82 | static cl::opt<bool> LinkAsLibrary("link-as-library", |
Reid Spencer | 445564a | 2004-11-20 20:02:56 +0000 | [diff] [blame] | 83 | cl::desc("Link the .bc files together as a library, not an executable")); |
Reid Spencer | c0af3f0 | 2004-09-13 01:27:53 +0000 | [diff] [blame] | 84 | |
Reid Spencer | 445564a | 2004-11-20 20:02:56 +0000 | [diff] [blame] | 85 | static cl::alias Relink("r", cl::aliasopt(LinkAsLibrary), |
| 86 | cl::desc("Alias for -link-as-library")); |
Reid Spencer | c0af3f0 | 2004-09-13 01:27:53 +0000 | [diff] [blame] | 87 | |
Reid Spencer | 445564a | 2004-11-20 20:02:56 +0000 | [diff] [blame] | 88 | static cl::opt<bool> Native("native", |
| 89 | cl::desc("Generate a native binary instead of a shell script")); |
Reid Spencer | c0af3f0 | 2004-09-13 01:27:53 +0000 | [diff] [blame] | 90 | |
Reid Spencer | 445564a | 2004-11-20 20:02:56 +0000 | [diff] [blame] | 91 | static cl::opt<bool>NativeCBE("native-cbe", |
| 92 | cl::desc("Generate a native binary with the C backend and GCC")); |
| 93 | |
Reid Spencer | 73a74be | 2005-12-21 05:03:23 +0000 | [diff] [blame] | 94 | static cl::list<std::string> PostLinkOpts("post-link-opts", |
Reid Spencer | af303d5 | 2006-06-07 23:03:13 +0000 | [diff] [blame] | 95 | cl::value_desc("path"), |
Reid Spencer | 73a74be | 2005-12-21 05:03:23 +0000 | [diff] [blame] | 96 | cl::desc("Run one or more optimization programs after linking")); |
| 97 | |
Reid Spencer | af303d5 | 2006-06-07 23:03:13 +0000 | [diff] [blame] | 98 | static cl::list<std::string> XLinker("Xlinker", cl::value_desc("option"), |
| 99 | cl::desc("Pass options to the system linker")); |
| 100 | |
Mikhail Glushenkov | 572ec1f | 2010-11-02 20:32:52 +0000 | [diff] [blame] | 101 | // Compatibility options that llvm-ld ignores but are supported for |
Reid Spencer | 708585a | 2007-02-09 03:08:06 +0000 | [diff] [blame] | 102 | // compatibility with LD |
Misha Brukman | 3da94ae | 2005-04-22 00:00:37 +0000 | [diff] [blame] | 103 | static cl::opt<std::string> CO3("soname", cl::Hidden, |
Reid Spencer | 445564a | 2004-11-20 20:02:56 +0000 | [diff] [blame] | 104 | cl::desc("Compatibility option: ignored")); |
| 105 | |
Misha Brukman | 3da94ae | 2005-04-22 00:00:37 +0000 | [diff] [blame] | 106 | static cl::opt<std::string> CO4("version-script", cl::Hidden, |
Reid Spencer | 445564a | 2004-11-20 20:02:56 +0000 | [diff] [blame] | 107 | cl::desc("Compatibility option: ignored")); |
| 108 | |
Misha Brukman | 3da94ae | 2005-04-22 00:00:37 +0000 | [diff] [blame] | 109 | static cl::opt<bool> CO5("eh-frame-hdr", cl::Hidden, |
Reid Spencer | 445564a | 2004-11-20 20:02:56 +0000 | [diff] [blame] | 110 | cl::desc("Compatibility option: ignored")); |
| 111 | |
Misha Brukman | 3da94ae | 2005-04-22 00:00:37 +0000 | [diff] [blame] | 112 | static cl::opt<std::string> CO6("h", cl::Hidden, |
Reid Spencer | 445564a | 2004-11-20 20:02:56 +0000 | [diff] [blame] | 113 | cl::desc("Compatibility option: ignored")); |
| 114 | |
Mikhail Glushenkov | 572ec1f | 2010-11-02 20:32:52 +0000 | [diff] [blame] | 115 | static cl::opt<bool> CO7("start-group", cl::Hidden, |
Reid Spencer | 98a030c | 2007-02-08 19:03:11 +0000 | [diff] [blame] | 116 | cl::desc("Compatibility option: ignored")); |
| 117 | |
Mikhail Glushenkov | 572ec1f | 2010-11-02 20:32:52 +0000 | [diff] [blame] | 118 | static cl::opt<bool> CO8("end-group", cl::Hidden, |
Reid Spencer | 98a030c | 2007-02-08 19:03:11 +0000 | [diff] [blame] | 119 | cl::desc("Compatibility option: ignored")); |
Reid Spencer | af303d5 | 2006-06-07 23:03:13 +0000 | [diff] [blame] | 120 | |
Mikhail Glushenkov | 572ec1f | 2010-11-02 20:32:52 +0000 | [diff] [blame] | 121 | static cl::opt<std::string> CO9("m", cl::Hidden, |
Andrew Lenharth | 0c6ba44 | 2008-11-19 17:00:08 +0000 | [diff] [blame] | 122 | cl::desc("Compatibility option: ignored")); |
| 123 | |
Reid Spencer | 445564a | 2004-11-20 20:02:56 +0000 | [diff] [blame] | 124 | /// This is just for convenience so it doesn't have to be passed around |
| 125 | /// everywhere. |
Reid Spencer | c406413 | 2004-12-13 03:01:14 +0000 | [diff] [blame] | 126 | static std::string progname; |
Reid Spencer | c0af3f0 | 2004-09-13 01:27:53 +0000 | [diff] [blame] | 127 | |
Dan Gohman | 132a994 | 2010-03-30 19:56:41 +0000 | [diff] [blame] | 128 | /// FileRemover objects to clean up output files in the event of an error. |
| 129 | static FileRemover OutputRemover; |
| 130 | static FileRemover BitcodeOutputRemover; |
| 131 | |
Reid Spencer | 708585a | 2007-02-09 03:08:06 +0000 | [diff] [blame] | 132 | /// PrintAndExit - Prints a message to standard error and exits with error code |
Reid Spencer | c0af3f0 | 2004-09-13 01:27:53 +0000 | [diff] [blame] | 133 | /// |
| 134 | /// Inputs: |
Reid Spencer | c0af3f0 | 2004-09-13 01:27:53 +0000 | [diff] [blame] | 135 | /// Message - The message to print to standard error. |
| 136 | /// |
Chris Lattner | 25c54c0 | 2010-03-23 21:59:43 +0000 | [diff] [blame] | 137 | static void PrintAndExit(const std::string &Message, Module *M, int errcode = 1) { |
Dan Gohman | ac95cc7 | 2009-07-16 15:30:09 +0000 | [diff] [blame] | 138 | errs() << progname << ": " << Message << "\n"; |
Chris Lattner | 25c54c0 | 2010-03-23 21:59:43 +0000 | [diff] [blame] | 139 | delete M; |
Reid Spencer | 708585a | 2007-02-09 03:08:06 +0000 | [diff] [blame] | 140 | llvm_shutdown(); |
| 141 | exit(errcode); |
Reid Spencer | c0af3f0 | 2004-09-13 01:27:53 +0000 | [diff] [blame] | 142 | } |
| 143 | |
Reid Spencer | 3b72639 | 2007-04-29 23:59:47 +0000 | [diff] [blame] | 144 | static void PrintCommand(const std::vector<const char*> &args) { |
Mikhail Glushenkov | 572ec1f | 2010-11-02 20:32:52 +0000 | [diff] [blame] | 145 | std::vector<const char*>::const_iterator I = args.begin(), E = args.end(); |
Reid Spencer | 3b72639 | 2007-04-29 23:59:47 +0000 | [diff] [blame] | 146 | for (; I != E; ++I) |
| 147 | if (*I) |
Dan Gohman | 4bb122e | 2010-08-20 01:02:14 +0000 | [diff] [blame] | 148 | errs() << "'" << *I << "'" << " "; |
| 149 | errs() << "\n"; |
Reid Spencer | 3b72639 | 2007-04-29 23:59:47 +0000 | [diff] [blame] | 150 | } |
| 151 | |
Reid Spencer | 445564a | 2004-11-20 20:02:56 +0000 | [diff] [blame] | 152 | /// CopyEnv - This function takes an array of environment variables and makes a |
| 153 | /// copy of it. This copy can then be manipulated any way the caller likes |
| 154 | /// without affecting the process's real environment. |
| 155 | /// |
| 156 | /// Inputs: |
| 157 | /// envp - An array of C strings containing an environment. |
| 158 | /// |
| 159 | /// Return value: |
| 160 | /// NULL - An error occurred. |
| 161 | /// |
| 162 | /// Otherwise, a pointer to a new array of C strings is returned. Every string |
| 163 | /// in the array is a duplicate of the one in the original array (i.e. we do |
| 164 | /// not copy the char *'s from one array to another). |
| 165 | /// |
| 166 | static char ** CopyEnv(char ** const envp) { |
| 167 | // Count the number of entries in the old list; |
| 168 | unsigned entries; // The number of entries in the old environment list |
| 169 | for (entries = 0; envp[entries] != NULL; entries++) |
| 170 | /*empty*/; |
| 171 | |
| 172 | // Add one more entry for the NULL pointer that ends the list. |
| 173 | ++entries; |
| 174 | |
| 175 | // If there are no entries at all, just return NULL. |
| 176 | if (entries == 0) |
| 177 | return NULL; |
| 178 | |
| 179 | // Allocate a new environment list. |
| 180 | char **newenv = new char* [entries]; |
| 181 | if ((newenv = new char* [entries]) == NULL) |
| 182 | return NULL; |
| 183 | |
| 184 | // Make a copy of the list. Don't forget the NULL that ends the list. |
| 185 | entries = 0; |
| 186 | while (envp[entries] != NULL) { |
Benjamin Kramer | 12ea66a | 2010-01-28 18:04:38 +0000 | [diff] [blame] | 187 | size_t len = strlen(envp[entries]) + 1; |
| 188 | newenv[entries] = new char[len]; |
| 189 | memcpy(newenv[entries], envp[entries], len); |
Reid Spencer | 445564a | 2004-11-20 20:02:56 +0000 | [diff] [blame] | 190 | ++entries; |
| 191 | } |
| 192 | newenv[entries] = NULL; |
| 193 | |
| 194 | return newenv; |
| 195 | } |
| 196 | |
| 197 | |
| 198 | /// RemoveEnv - Remove the specified environment variable from the environment |
| 199 | /// array. |
| 200 | /// |
| 201 | /// Inputs: |
| 202 | /// name - The name of the variable to remove. It cannot be NULL. |
| 203 | /// envp - The array of environment variables. It cannot be NULL. |
| 204 | /// |
| 205 | /// Notes: |
| 206 | /// This is mainly done because functions to remove items from the environment |
| 207 | /// are not available across all platforms. In particular, Solaris does not |
| 208 | /// seem to have an unsetenv() function or a setenv() function (or they are |
| 209 | /// undocumented if they do exist). |
| 210 | /// |
| 211 | static void RemoveEnv(const char * name, char ** const envp) { |
| 212 | for (unsigned index=0; envp[index] != NULL; index++) { |
| 213 | // Find the first equals sign in the array and make it an EOS character. |
| 214 | char *p = strchr (envp[index], '='); |
| 215 | if (p == NULL) |
| 216 | continue; |
| 217 | else |
| 218 | *p = '\0'; |
| 219 | |
| 220 | // Compare the two strings. If they are equal, zap this string. |
| 221 | // Otherwise, restore it. |
| 222 | if (!strcmp(name, envp[index])) |
| 223 | *envp[index] = '\0'; |
| 224 | else |
| 225 | *p = '='; |
| 226 | } |
| 227 | |
| 228 | return; |
| 229 | } |
| 230 | |
Gabor Greif | a99be51 | 2007-07-05 17:07:56 +0000 | [diff] [blame] | 231 | /// GenerateBitcode - generates a bitcode file from the module provided |
| 232 | void GenerateBitcode(Module* M, const std::string& FileName) { |
Reid Spencer | 445564a | 2004-11-20 20:02:56 +0000 | [diff] [blame] | 233 | |
Reid Spencer | 3b72639 | 2007-04-29 23:59:47 +0000 | [diff] [blame] | 234 | if (Verbose) |
Dan Gohman | 4bb122e | 2010-08-20 01:02:14 +0000 | [diff] [blame] | 235 | errs() << "Generating Bitcode To " << FileName << '\n'; |
Reid Spencer | 3b72639 | 2007-04-29 23:59:47 +0000 | [diff] [blame] | 236 | |
Reid Spencer | 445564a | 2004-11-20 20:02:56 +0000 | [diff] [blame] | 237 | // Create the output file. |
Dan Gohman | ac95cc7 | 2009-07-16 15:30:09 +0000 | [diff] [blame] | 238 | std::string ErrorInfo; |
Dan Gohman | f291401 | 2010-08-20 16:59:15 +0000 | [diff] [blame] | 239 | tool_output_file Out(FileName.c_str(), ErrorInfo, |
| 240 | raw_fd_ostream::F_Binary); |
| 241 | if (!ErrorInfo.empty()) { |
Chris Lattner | 25c54c0 | 2010-03-23 21:59:43 +0000 | [diff] [blame] | 242 | PrintAndExit(ErrorInfo, M); |
Dan Gohman | f291401 | 2010-08-20 16:59:15 +0000 | [diff] [blame] | 243 | return; |
| 244 | } |
Reid Spencer | 445564a | 2004-11-20 20:02:56 +0000 | [diff] [blame] | 245 | |
Reid Spencer | 445564a | 2004-11-20 20:02:56 +0000 | [diff] [blame] | 246 | // Write it out |
Dan Gohman | d4c4543 | 2010-09-01 14:20:41 +0000 | [diff] [blame] | 247 | WriteBitcodeToFile(M, Out.os()); |
Dan Gohman | f291401 | 2010-08-20 16:59:15 +0000 | [diff] [blame] | 248 | Out.keep(); |
Reid Spencer | 445564a | 2004-11-20 20:02:56 +0000 | [diff] [blame] | 249 | } |
| 250 | |
| 251 | /// GenerateAssembly - generates a native assembly language source file from the |
Gabor Greif | a99be51 | 2007-07-05 17:07:56 +0000 | [diff] [blame] | 252 | /// specified bitcode file. |
Reid Spencer | 445564a | 2004-11-20 20:02:56 +0000 | [diff] [blame] | 253 | /// |
| 254 | /// Inputs: |
Gabor Greif | a99be51 | 2007-07-05 17:07:56 +0000 | [diff] [blame] | 255 | /// InputFilename - The name of the input bitcode file. |
Reid Spencer | 445564a | 2004-11-20 20:02:56 +0000 | [diff] [blame] | 256 | /// OutputFilename - The name of the file to generate. |
| 257 | /// llc - The pathname to use for LLC. |
| 258 | /// envp - The environment to use when running LLC. |
| 259 | /// |
| 260 | /// Return non-zero value on error. |
| 261 | /// |
| 262 | static int GenerateAssembly(const std::string &OutputFilename, |
| 263 | const std::string &InputFilename, |
Reid Spencer | 8ea5ecb | 2006-08-21 06:04:45 +0000 | [diff] [blame] | 264 | const sys::Path &llc, |
| 265 | std::string &ErrMsg ) { |
Gabor Greif | a99be51 | 2007-07-05 17:07:56 +0000 | [diff] [blame] | 266 | // Run LLC to convert the bitcode file into assembly code. |
Reid Spencer | f6358c7 | 2004-12-19 18:00:56 +0000 | [diff] [blame] | 267 | std::vector<const char*> args; |
Chris Lattner | bf9add4 | 2005-04-10 20:59:38 +0000 | [diff] [blame] | 268 | args.push_back(llc.c_str()); |
Argyrios Kyrtzidis | ca29dff | 2008-06-27 15:08:59 +0000 | [diff] [blame] | 269 | // We will use GCC to assemble the program so set the assembly syntax to AT&T, |
| 270 | // regardless of what the target in the bitcode file is. |
| 271 | args.push_back("-x86-asm-syntax=att"); |
Chris Lattner | bf9add4 | 2005-04-10 20:59:38 +0000 | [diff] [blame] | 272 | args.push_back("-o"); |
| 273 | args.push_back(OutputFilename.c_str()); |
| 274 | args.push_back(InputFilename.c_str()); |
Chris Lattner | 7456e3c | 2005-02-13 23:10:45 +0000 | [diff] [blame] | 275 | args.push_back(0); |
Reid Spencer | 445564a | 2004-11-20 20:02:56 +0000 | [diff] [blame] | 276 | |
Reid Spencer | 3b72639 | 2007-04-29 23:59:47 +0000 | [diff] [blame] | 277 | if (Verbose) { |
Dan Gohman | 4bb122e | 2010-08-20 01:02:14 +0000 | [diff] [blame] | 278 | errs() << "Generating Assembly With: \n"; |
Reid Spencer | 3b72639 | 2007-04-29 23:59:47 +0000 | [diff] [blame] | 279 | PrintCommand(args); |
| 280 | } |
| 281 | |
Anton Korobeynikov | 9ba8a76 | 2007-02-16 19:11:07 +0000 | [diff] [blame] | 282 | return sys::Program::ExecuteAndWait(llc, &args[0], 0, 0, 0, 0, &ErrMsg); |
Reid Spencer | 445564a | 2004-11-20 20:02:56 +0000 | [diff] [blame] | 283 | } |
| 284 | |
Gabor Greif | a99be51 | 2007-07-05 17:07:56 +0000 | [diff] [blame] | 285 | /// GenerateCFile - generates a C source file from the specified bitcode file. |
Reid Spencer | 445564a | 2004-11-20 20:02:56 +0000 | [diff] [blame] | 286 | static int GenerateCFile(const std::string &OutputFile, |
| 287 | const std::string &InputFile, |
Reid Spencer | 8ea5ecb | 2006-08-21 06:04:45 +0000 | [diff] [blame] | 288 | const sys::Path &llc, |
| 289 | std::string& ErrMsg) { |
Gabor Greif | a99be51 | 2007-07-05 17:07:56 +0000 | [diff] [blame] | 290 | // Run LLC to convert the bitcode file into C. |
Reid Spencer | f6358c7 | 2004-12-19 18:00:56 +0000 | [diff] [blame] | 291 | std::vector<const char*> args; |
Chris Lattner | bf9add4 | 2005-04-10 20:59:38 +0000 | [diff] [blame] | 292 | args.push_back(llc.c_str()); |
| 293 | args.push_back("-march=c"); |
Chris Lattner | bf9add4 | 2005-04-10 20:59:38 +0000 | [diff] [blame] | 294 | args.push_back("-o"); |
| 295 | args.push_back(OutputFile.c_str()); |
| 296 | args.push_back(InputFile.c_str()); |
Chris Lattner | 7456e3c | 2005-02-13 23:10:45 +0000 | [diff] [blame] | 297 | args.push_back(0); |
Reid Spencer | 3b72639 | 2007-04-29 23:59:47 +0000 | [diff] [blame] | 298 | |
| 299 | if (Verbose) { |
Dan Gohman | 4bb122e | 2010-08-20 01:02:14 +0000 | [diff] [blame] | 300 | errs() << "Generating C Source With: \n"; |
Reid Spencer | 3b72639 | 2007-04-29 23:59:47 +0000 | [diff] [blame] | 301 | PrintCommand(args); |
| 302 | } |
| 303 | |
Anton Korobeynikov | 9ba8a76 | 2007-02-16 19:11:07 +0000 | [diff] [blame] | 304 | return sys::Program::ExecuteAndWait(llc, &args[0], 0, 0, 0, 0, &ErrMsg); |
Reid Spencer | 445564a | 2004-11-20 20:02:56 +0000 | [diff] [blame] | 305 | } |
| 306 | |
Devang Patel | e2f8ad8 | 2006-06-27 18:07:29 +0000 | [diff] [blame] | 307 | /// GenerateNative - generates a native object file from the |
Gabor Greif | a99be51 | 2007-07-05 17:07:56 +0000 | [diff] [blame] | 308 | /// specified bitcode file. |
Reid Spencer | 445564a | 2004-11-20 20:02:56 +0000 | [diff] [blame] | 309 | /// |
| 310 | /// Inputs: |
Gabor Greif | a99be51 | 2007-07-05 17:07:56 +0000 | [diff] [blame] | 311 | /// InputFilename - The name of the input bitcode file. |
Reid Spencer | c82a5da | 2007-04-04 06:34:22 +0000 | [diff] [blame] | 312 | /// OutputFilename - The name of the file to generate. |
| 313 | /// NativeLinkItems - The native libraries, files, code with which to link |
| 314 | /// LibPaths - The list of directories in which to find libraries. |
Chris Lattner | 3992f52 | 2008-01-27 22:58:59 +0000 | [diff] [blame] | 315 | /// FrameworksPaths - The list of directories in which to find frameworks. |
| 316 | /// Frameworks - The list of frameworks (dynamic libraries) |
Reid Spencer | c82a5da | 2007-04-04 06:34:22 +0000 | [diff] [blame] | 317 | /// gcc - The pathname to use for GGC. |
| 318 | /// envp - A copy of the process's current environment. |
Reid Spencer | 445564a | 2004-11-20 20:02:56 +0000 | [diff] [blame] | 319 | /// |
| 320 | /// Outputs: |
| 321 | /// None. |
| 322 | /// |
| 323 | /// Returns non-zero value on error. |
| 324 | /// |
| 325 | static int GenerateNative(const std::string &OutputFilename, |
| 326 | const std::string &InputFilename, |
Reid Spencer | 4952143 | 2006-11-11 11:54:25 +0000 | [diff] [blame] | 327 | const Linker::ItemList &LinkItems, |
Reid Spencer | 8ea5ecb | 2006-08-21 06:04:45 +0000 | [diff] [blame] | 328 | const sys::Path &gcc, char ** const envp, |
| 329 | std::string& ErrMsg) { |
Reid Spencer | 445564a | 2004-11-20 20:02:56 +0000 | [diff] [blame] | 330 | // Remove these environment variables from the environment of the |
| 331 | // programs that we will execute. It appears that GCC sets these |
| 332 | // environment variables so that the programs it uses can configure |
| 333 | // themselves identically. |
| 334 | // |
| 335 | // However, when we invoke GCC below, we want it to use its normal |
| 336 | // configuration. Hence, we must sanitize its environment. |
| 337 | char ** clean_env = CopyEnv(envp); |
| 338 | if (clean_env == NULL) |
| 339 | return 1; |
| 340 | RemoveEnv("LIBRARY_PATH", clean_env); |
| 341 | RemoveEnv("COLLECT_GCC_OPTIONS", clean_env); |
| 342 | RemoveEnv("GCC_EXEC_PREFIX", clean_env); |
| 343 | RemoveEnv("COMPILER_PATH", clean_env); |
| 344 | RemoveEnv("COLLECT_GCC", clean_env); |
| 345 | |
Reid Spencer | 445564a | 2004-11-20 20:02:56 +0000 | [diff] [blame] | 346 | |
| 347 | // Run GCC to assemble and link the program into native code. |
| 348 | // |
| 349 | // Note: |
| 350 | // We can't just assemble and link the file with the system assembler |
| 351 | // and linker because we don't know where to put the _start symbol. |
| 352 | // GCC mysteriously knows how to do it. |
Chris Lattner | 3f931b8 | 2007-06-19 16:46:48 +0000 | [diff] [blame] | 353 | std::vector<std::string> args; |
Chris Lattner | bf9add4 | 2005-04-10 20:59:38 +0000 | [diff] [blame] | 354 | args.push_back(gcc.c_str()); |
Reid Spencer | 6da1e0d | 2004-12-14 04:20:08 +0000 | [diff] [blame] | 355 | args.push_back("-fno-strict-aliasing"); |
| 356 | args.push_back("-O3"); |
| 357 | args.push_back("-o"); |
Chris Lattner | 3f931b8 | 2007-06-19 16:46:48 +0000 | [diff] [blame] | 358 | args.push_back(OutputFilename); |
| 359 | args.push_back(InputFilename); |
Reid Spencer | 445564a | 2004-11-20 20:02:56 +0000 | [diff] [blame] | 360 | |
Chris Lattner | 3992f52 | 2008-01-27 22:58:59 +0000 | [diff] [blame] | 361 | // Add in the library and framework paths |
Reid Spencer | af303d5 | 2006-06-07 23:03:13 +0000 | [diff] [blame] | 362 | for (unsigned index = 0; index < LibPaths.size(); index++) { |
Chris Lattner | 3992f52 | 2008-01-27 22:58:59 +0000 | [diff] [blame] | 363 | args.push_back("-L" + LibPaths[index]); |
| 364 | } |
| 365 | for (unsigned index = 0; index < FrameworkPaths.size(); index++) { |
| 366 | args.push_back("-F" + FrameworkPaths[index]); |
Reid Spencer | af303d5 | 2006-06-07 23:03:13 +0000 | [diff] [blame] | 367 | } |
| 368 | |
| 369 | // Add the requested options |
Chris Lattner | 03a1c7a | 2008-01-09 01:01:17 +0000 | [diff] [blame] | 370 | for (unsigned index = 0; index < XLinker.size(); index++) |
Chris Lattner | 3f931b8 | 2007-06-19 16:46:48 +0000 | [diff] [blame] | 371 | args.push_back(XLinker[index]); |
Reid Spencer | af303d5 | 2006-06-07 23:03:13 +0000 | [diff] [blame] | 372 | |
Reid Spencer | 445564a | 2004-11-20 20:02:56 +0000 | [diff] [blame] | 373 | // Add in the libraries to link. |
Reid Spencer | 4952143 | 2006-11-11 11:54:25 +0000 | [diff] [blame] | 374 | for (unsigned index = 0; index < LinkItems.size(); index++) |
| 375 | if (LinkItems[index].first != "crtend") { |
Chris Lattner | 3f931b8 | 2007-06-19 16:46:48 +0000 | [diff] [blame] | 376 | if (LinkItems[index].second) |
| 377 | args.push_back("-l" + LinkItems[index].first); |
| 378 | else |
| 379 | args.push_back(LinkItems[index].first); |
Reid Spencer | f6358c7 | 2004-12-19 18:00:56 +0000 | [diff] [blame] | 380 | } |
Reid Spencer | af303d5 | 2006-06-07 23:03:13 +0000 | [diff] [blame] | 381 | |
Chris Lattner | 3992f52 | 2008-01-27 22:58:59 +0000 | [diff] [blame] | 382 | // Add in frameworks to link. |
| 383 | for (unsigned index = 0; index < Frameworks.size(); index++) { |
| 384 | args.push_back("-framework"); |
| 385 | args.push_back(Frameworks[index]); |
| 386 | } |
Mikhail Glushenkov | 572ec1f | 2010-11-02 20:32:52 +0000 | [diff] [blame] | 387 | |
Chris Lattner | 3f931b8 | 2007-06-19 16:46:48 +0000 | [diff] [blame] | 388 | // Now that "args" owns all the std::strings for the arguments, call the c_str |
| 389 | // method to get the underlying string array. We do this game so that the |
| 390 | // std::string array is guaranteed to outlive the const char* array. |
| 391 | std::vector<const char *> Args; |
| 392 | for (unsigned i = 0, e = args.size(); i != e; ++i) |
| 393 | Args.push_back(args[i].c_str()); |
| 394 | Args.push_back(0); |
Reid Spencer | 445564a | 2004-11-20 20:02:56 +0000 | [diff] [blame] | 395 | |
Reid Spencer | 3b72639 | 2007-04-29 23:59:47 +0000 | [diff] [blame] | 396 | if (Verbose) { |
Dan Gohman | 4bb122e | 2010-08-20 01:02:14 +0000 | [diff] [blame] | 397 | errs() << "Generating Native Executable With:\n"; |
Chris Lattner | 3f931b8 | 2007-06-19 16:46:48 +0000 | [diff] [blame] | 398 | PrintCommand(Args); |
Reid Spencer | 3b72639 | 2007-04-29 23:59:47 +0000 | [diff] [blame] | 399 | } |
| 400 | |
Reid Spencer | 445564a | 2004-11-20 20:02:56 +0000 | [diff] [blame] | 401 | // Run the compiler to assembly and link together the program. |
Reid Spencer | 8ea5ecb | 2006-08-21 06:04:45 +0000 | [diff] [blame] | 402 | int R = sys::Program::ExecuteAndWait( |
Dan Gohman | 43bc70e | 2010-04-17 17:44:03 +0000 | [diff] [blame] | 403 | gcc, &Args[0], const_cast<const char **>(clean_env), 0, 0, 0, &ErrMsg); |
Chris Lattner | 2f86362 | 2006-05-14 18:38:13 +0000 | [diff] [blame] | 404 | delete [] clean_env; |
| 405 | return R; |
Reid Spencer | 445564a | 2004-11-20 20:02:56 +0000 | [diff] [blame] | 406 | } |
| 407 | |
Reid Spencer | c0af3f0 | 2004-09-13 01:27:53 +0000 | [diff] [blame] | 408 | /// EmitShellScript - Output the wrapper file that invokes the JIT on the LLVM |
Gabor Greif | a99be51 | 2007-07-05 17:07:56 +0000 | [diff] [blame] | 409 | /// bitcode file for the program. |
Chris Lattner | 25c54c0 | 2010-03-23 21:59:43 +0000 | [diff] [blame] | 410 | static void EmitShellScript(char **argv, Module *M) { |
Reid Spencer | 3b72639 | 2007-04-29 23:59:47 +0000 | [diff] [blame] | 411 | if (Verbose) |
Dan Gohman | 4bb122e | 2010-08-20 01:02:14 +0000 | [diff] [blame] | 412 | errs() << "Emitting Shell Script\n"; |
Reid Spencer | c0af3f0 | 2004-09-13 01:27:53 +0000 | [diff] [blame] | 413 | #if defined(_WIN32) || defined(__CYGWIN__) |
| 414 | // Windows doesn't support #!/bin/sh style shell scripts in .exe files. To |
| 415 | // support windows systems, we copy the llvm-stub.exe executable from the |
| 416 | // build tree to the destination file. |
Mikhail Glushenkov | 572ec1f | 2010-11-02 20:32:52 +0000 | [diff] [blame] | 417 | std::string ErrMsg; |
Mikhail Glushenkov | d66f2b7 | 2010-11-03 16:14:16 +0000 | [diff] [blame] | 418 | sys::Path llvmstub = PrependMainExecutablePath("llvm-stub", argv[0], |
| 419 | (void *)(intptr_t)&Optimize); |
Reid Spencer | 708585a | 2007-02-09 03:08:06 +0000 | [diff] [blame] | 420 | if (llvmstub.isEmpty()) |
Chris Lattner | 25c54c0 | 2010-03-23 21:59:43 +0000 | [diff] [blame] | 421 | PrintAndExit("Could not find llvm-stub.exe executable!", M); |
Anton Korobeynikov | 7d51544 | 2006-09-01 20:35:17 +0000 | [diff] [blame] | 422 | |
Argyrios Kyrtzidis | 1662183 | 2008-06-15 13:48:12 +0000 | [diff] [blame] | 423 | if (0 != sys::CopyFile(sys::Path(OutputFilename), llvmstub, &ErrMsg)) |
Chris Lattner | 25c54c0 | 2010-03-23 21:59:43 +0000 | [diff] [blame] | 424 | PrintAndExit(ErrMsg, M); |
Anton Korobeynikov | 7d51544 | 2006-09-01 20:35:17 +0000 | [diff] [blame] | 425 | |
Reid Spencer | c0af3f0 | 2004-09-13 01:27:53 +0000 | [diff] [blame] | 426 | return; |
| 427 | #endif |
| 428 | |
| 429 | // Output the script to start the program... |
Dan Gohman | ac95cc7 | 2009-07-16 15:30:09 +0000 | [diff] [blame] | 430 | std::string ErrorInfo; |
Dan Gohman | f291401 | 2010-08-20 16:59:15 +0000 | [diff] [blame] | 431 | tool_output_file Out2(OutputFilename.c_str(), ErrorInfo); |
Dan Gohman | ac95cc7 | 2009-07-16 15:30:09 +0000 | [diff] [blame] | 432 | if (!ErrorInfo.empty()) |
Chris Lattner | 25c54c0 | 2010-03-23 21:59:43 +0000 | [diff] [blame] | 433 | PrintAndExit(ErrorInfo, M); |
Reid Spencer | c0af3f0 | 2004-09-13 01:27:53 +0000 | [diff] [blame] | 434 | |
Dan Gohman | d4c4543 | 2010-09-01 14:20:41 +0000 | [diff] [blame] | 435 | Out2.os() << "#!/bin/sh\n"; |
Reid Spencer | c0af3f0 | 2004-09-13 01:27:53 +0000 | [diff] [blame] | 436 | // Allow user to setenv LLVMINTERP if lli is not in their PATH. |
Dan Gohman | d4c4543 | 2010-09-01 14:20:41 +0000 | [diff] [blame] | 437 | Out2.os() << "lli=${LLVMINTERP-lli}\n"; |
| 438 | Out2.os() << "exec $lli \\\n"; |
Reid Spencer | c0af3f0 | 2004-09-13 01:27:53 +0000 | [diff] [blame] | 439 | // gcc accepts -l<lib> and implicitly searches /lib and /usr/lib. |
| 440 | LibPaths.push_back("/lib"); |
| 441 | LibPaths.push_back("/usr/lib"); |
| 442 | LibPaths.push_back("/usr/X11R6/lib"); |
| 443 | // We don't need to link in libc! In fact, /usr/lib/libc.so may not be a |
| 444 | // shared object at all! See RH 8: plain text. |
Misha Brukman | 3da94ae | 2005-04-22 00:00:37 +0000 | [diff] [blame] | 445 | std::vector<std::string>::iterator libc = |
Reid Spencer | c0af3f0 | 2004-09-13 01:27:53 +0000 | [diff] [blame] | 446 | std::find(Libraries.begin(), Libraries.end(), "c"); |
| 447 | if (libc != Libraries.end()) Libraries.erase(libc); |
| 448 | // List all the shared object (native) libraries this executable will need |
| 449 | // on the command line, so that we don't have to do this manually! |
Misha Brukman | 3da94ae | 2005-04-22 00:00:37 +0000 | [diff] [blame] | 450 | for (std::vector<std::string>::iterator i = Libraries.begin(), |
Reid Spencer | c0af3f0 | 2004-09-13 01:27:53 +0000 | [diff] [blame] | 451 | e = Libraries.end(); i != e; ++i) { |
Chris Lattner | 5d5a897 | 2009-01-05 19:01:32 +0000 | [diff] [blame] | 452 | // try explicit -L arguments first: |
| 453 | sys::Path FullLibraryPath; |
| 454 | for (cl::list<std::string>::const_iterator P = LibPaths.begin(), |
| 455 | E = LibPaths.end(); P != E; ++P) { |
| 456 | FullLibraryPath = *P; |
| 457 | FullLibraryPath.appendComponent("lib" + *i); |
Mikhail Glushenkov | c8aef4b | 2010-11-02 20:32:59 +0000 | [diff] [blame] | 458 | FullLibraryPath.appendSuffix(sys::Path::GetDLLSuffix()); |
Chris Lattner | 5d5a897 | 2009-01-05 19:01:32 +0000 | [diff] [blame] | 459 | if (!FullLibraryPath.isEmpty()) { |
| 460 | if (!FullLibraryPath.isDynamicLibrary()) { |
| 461 | // Not a native shared library; mark as invalid |
| 462 | FullLibraryPath = sys::Path(); |
| 463 | } else break; |
| 464 | } |
| 465 | } |
| 466 | if (FullLibraryPath.isEmpty()) |
| 467 | FullLibraryPath = sys::Path::FindLibrary(*i); |
| 468 | if (!FullLibraryPath.isEmpty()) |
Dan Gohman | d4c4543 | 2010-09-01 14:20:41 +0000 | [diff] [blame] | 469 | Out2.os() << " -load=" << FullLibraryPath.str() << " \\\n"; |
Reid Spencer | c0af3f0 | 2004-09-13 01:27:53 +0000 | [diff] [blame] | 470 | } |
Dan Gohman | d4c4543 | 2010-09-01 14:20:41 +0000 | [diff] [blame] | 471 | Out2.os() << " " << BitcodeOutputFilename << " ${1+\"$@\"}\n"; |
Dan Gohman | f291401 | 2010-08-20 16:59:15 +0000 | [diff] [blame] | 472 | Out2.keep(); |
Reid Spencer | c0af3f0 | 2004-09-13 01:27:53 +0000 | [diff] [blame] | 473 | } |
| 474 | |
Reid Spencer | c406413 | 2004-12-13 03:01:14 +0000 | [diff] [blame] | 475 | // BuildLinkItems -- This function generates a LinkItemList for the LinkItems |
| 476 | // linker function by combining the Files and Libraries in the order they were |
| 477 | // declared on the command line. |
| 478 | static void BuildLinkItems( |
| 479 | Linker::ItemList& Items, |
| 480 | const cl::list<std::string>& Files, |
| 481 | const cl::list<std::string>& Libraries) { |
| 482 | |
Misha Brukman | 3da94ae | 2005-04-22 00:00:37 +0000 | [diff] [blame] | 483 | // Build the list of linkage items for LinkItems. |
Reid Spencer | c406413 | 2004-12-13 03:01:14 +0000 | [diff] [blame] | 484 | |
| 485 | cl::list<std::string>::const_iterator fileIt = Files.begin(); |
| 486 | cl::list<std::string>::const_iterator libIt = Libraries.begin(); |
| 487 | |
| 488 | int libPos = -1, filePos = -1; |
Reid Spencer | 05f7e79 | 2004-12-13 17:18:19 +0000 | [diff] [blame] | 489 | while ( libIt != Libraries.end() || fileIt != Files.end() ) { |
Reid Spencer | c406413 | 2004-12-13 03:01:14 +0000 | [diff] [blame] | 490 | if (libIt != Libraries.end()) |
| 491 | libPos = Libraries.getPosition(libIt - Libraries.begin()); |
| 492 | else |
| 493 | libPos = -1; |
| 494 | if (fileIt != Files.end()) |
| 495 | filePos = Files.getPosition(fileIt - Files.begin()); |
| 496 | else |
| 497 | filePos = -1; |
| 498 | |
| 499 | if (filePos != -1 && (libPos == -1 || filePos < libPos)) { |
| 500 | // Add a source file |
| 501 | Items.push_back(std::make_pair(*fileIt++, false)); |
| 502 | } else if (libPos != -1 && (filePos == -1 || libPos < filePos)) { |
| 503 | // Add a library |
| 504 | Items.push_back(std::make_pair(*libIt++, true)); |
Reid Spencer | c406413 | 2004-12-13 03:01:14 +0000 | [diff] [blame] | 505 | } |
| 506 | } |
| 507 | } |
| 508 | |
Reid Spencer | c0af3f0 | 2004-09-13 01:27:53 +0000 | [diff] [blame] | 509 | int main(int argc, char **argv, char **envp) { |
Chris Lattner | cc14d25 | 2009-03-06 05:34:10 +0000 | [diff] [blame] | 510 | // Print a stack trace if we signal out. |
| 511 | sys::PrintStackTraceOnErrorSignal(); |
| 512 | PrettyStackTraceProgram X(argc, argv); |
Owen Anderson | 8b477ed | 2009-07-01 16:58:40 +0000 | [diff] [blame] | 513 | |
Owen Anderson | 0d7c695 | 2009-07-15 22:16:10 +0000 | [diff] [blame] | 514 | LLVMContext &Context = getGlobalContext(); |
Chris Lattner | cc14d25 | 2009-03-06 05:34:10 +0000 | [diff] [blame] | 515 | llvm_shutdown_obj Y; // Call llvm_shutdown() on exit. |
Mikhail Glushenkov | 572ec1f | 2010-11-02 20:32:52 +0000 | [diff] [blame] | 516 | |
Owen Anderson | 081c34b | 2010-10-19 17:21:58 +0000 | [diff] [blame] | 517 | // Initialize passes |
| 518 | PassRegistry &Registry = *PassRegistry::getPassRegistry(); |
| 519 | initializeCore(Registry); |
| 520 | initializeScalarOpts(Registry); |
| 521 | initializeIPO(Registry); |
| 522 | initializeAnalysis(Registry); |
| 523 | initializeIPA(Registry); |
| 524 | initializeTransformUtils(Registry); |
| 525 | initializeInstCombine(Registry); |
| 526 | initializeTarget(Registry); |
Mikhail Glushenkov | 572ec1f | 2010-11-02 20:32:52 +0000 | [diff] [blame] | 527 | |
Chris Lattner | 316e326 | 2009-10-22 00:52:28 +0000 | [diff] [blame] | 528 | // Initial global variable above for convenience printing of program name. |
Owen Anderson | fffa863 | 2010-12-18 07:37:18 +0000 | [diff] [blame^] | 529 | progname = sys::Path(argv[0]).getBasename(); |
Misha Brukman | 3da94ae | 2005-04-22 00:00:37 +0000 | [diff] [blame] | 530 | |
Chris Lattner | 316e326 | 2009-10-22 00:52:28 +0000 | [diff] [blame] | 531 | // Parse the command line options |
| 532 | cl::ParseCommandLineOptions(argc, argv, "llvm linker\n"); |
Reid Spencer | c0af3f0 | 2004-09-13 01:27:53 +0000 | [diff] [blame] | 533 | |
Dan Gohman | 132a994 | 2010-03-30 19:56:41 +0000 | [diff] [blame] | 534 | #if defined(_WIN32) || defined(__CYGWIN__) |
| 535 | if (!LinkAsLibrary) { |
| 536 | // Default to "a.exe" instead of "a.out". |
| 537 | if (OutputFilename.getNumOccurrences() == 0) |
| 538 | OutputFilename = "a.exe"; |
| 539 | |
| 540 | // If there is no suffix add an "exe" one. |
Owen Anderson | fffa863 | 2010-12-18 07:37:18 +0000 | [diff] [blame^] | 541 | sys::Path ExeFile( OutputFilename ); |
| 542 | if (ExeFile.getSuffix() == "") { |
| 543 | ExeFile.appendSuffix("exe"); |
| 544 | OutputFilename = ExeFile.str(); |
| 545 | } |
Dan Gohman | 132a994 | 2010-03-30 19:56:41 +0000 | [diff] [blame] | 546 | } |
| 547 | #endif |
| 548 | |
| 549 | // Generate the bitcode for the optimized module. |
| 550 | // If -b wasn't specified, use the name specified |
| 551 | // with -o to construct BitcodeOutputFilename. |
| 552 | if (BitcodeOutputFilename.empty()) { |
| 553 | BitcodeOutputFilename = OutputFilename; |
| 554 | if (!LinkAsLibrary) BitcodeOutputFilename += ".bc"; |
| 555 | } |
| 556 | |
| 557 | // Arrange for the bitcode output file to be deleted on any errors. |
| 558 | BitcodeOutputRemover.setFile(sys::Path(BitcodeOutputFilename)); |
| 559 | sys::RemoveFileOnSignal(sys::Path(BitcodeOutputFilename)); |
| 560 | |
| 561 | // Arrange for the output file to be deleted on any errors. |
| 562 | if (!LinkAsLibrary) { |
| 563 | OutputRemover.setFile(sys::Path(OutputFilename)); |
| 564 | sys::RemoveFileOnSignal(sys::Path(OutputFilename)); |
| 565 | } |
| 566 | |
Chris Lattner | 316e326 | 2009-10-22 00:52:28 +0000 | [diff] [blame] | 567 | // Construct a Linker (now that Verbose is set) |
| 568 | Linker TheLinker(progname, OutputFilename, Context, Verbose); |
Reid Spencer | c82a5da | 2007-04-04 06:34:22 +0000 | [diff] [blame] | 569 | |
Chris Lattner | 316e326 | 2009-10-22 00:52:28 +0000 | [diff] [blame] | 570 | // Keep track of the native link items (versus the bitcode items) |
| 571 | Linker::ItemList NativeLinkItems; |
Reid Spencer | 4952143 | 2006-11-11 11:54:25 +0000 | [diff] [blame] | 572 | |
Chris Lattner | 316e326 | 2009-10-22 00:52:28 +0000 | [diff] [blame] | 573 | // Add library paths to the linker |
| 574 | TheLinker.addPaths(LibPaths); |
| 575 | TheLinker.addSystemPaths(); |
Reid Spencer | 78df5c3 | 2006-03-06 06:38:19 +0000 | [diff] [blame] | 576 | |
Chris Lattner | 316e326 | 2009-10-22 00:52:28 +0000 | [diff] [blame] | 577 | // Remove any consecutive duplicates of the same library... |
| 578 | Libraries.erase(std::unique(Libraries.begin(), Libraries.end()), |
| 579 | Libraries.end()); |
Reid Spencer | c0af3f0 | 2004-09-13 01:27:53 +0000 | [diff] [blame] | 580 | |
Chris Lattner | 316e326 | 2009-10-22 00:52:28 +0000 | [diff] [blame] | 581 | if (LinkAsLibrary) { |
| 582 | std::vector<sys::Path> Files; |
| 583 | for (unsigned i = 0; i < InputFilenames.size(); ++i ) |
| 584 | Files.push_back(sys::Path(InputFilenames[i])); |
| 585 | if (TheLinker.LinkInFiles(Files)) |
| 586 | return 1; // Error already printed |
Reid Spencer | 6b463b2 | 2004-12-08 05:17:40 +0000 | [diff] [blame] | 587 | |
Chris Lattner | 316e326 | 2009-10-22 00:52:28 +0000 | [diff] [blame] | 588 | // The libraries aren't linked in but are noted as "dependent" in the |
| 589 | // module. |
| 590 | for (cl::list<std::string>::const_iterator I = Libraries.begin(), |
| 591 | E = Libraries.end(); I != E ; ++I) { |
| 592 | TheLinker.getModule()->addLibrary(*I); |
Reid Spencer | c0af3f0 | 2004-09-13 01:27:53 +0000 | [diff] [blame] | 593 | } |
Chris Lattner | 316e326 | 2009-10-22 00:52:28 +0000 | [diff] [blame] | 594 | } else { |
| 595 | // Build a list of the items from our command line |
| 596 | Linker::ItemList Items; |
| 597 | BuildLinkItems(Items, InputFilenames, Libraries); |
Reid Spencer | c0af3f0 | 2004-09-13 01:27:53 +0000 | [diff] [blame] | 598 | |
Chris Lattner | 316e326 | 2009-10-22 00:52:28 +0000 | [diff] [blame] | 599 | // Link all the items together |
| 600 | if (TheLinker.LinkInItems(Items, NativeLinkItems) ) |
| 601 | return 1; // Error already printed |
| 602 | } |
Reid Spencer | 1ef8bda | 2004-12-30 05:36:08 +0000 | [diff] [blame] | 603 | |
Chris Lattner | 316e326 | 2009-10-22 00:52:28 +0000 | [diff] [blame] | 604 | std::auto_ptr<Module> Composite(TheLinker.releaseModule()); |
| 605 | |
| 606 | // Optimize the module |
| 607 | Optimize(Composite.get()); |
Reid Spencer | 1ef8bda | 2004-12-30 05:36:08 +0000 | [diff] [blame] | 608 | |
Dan Gohman | 132a994 | 2010-03-30 19:56:41 +0000 | [diff] [blame] | 609 | // Generate the bitcode output. |
Chris Lattner | 316e326 | 2009-10-22 00:52:28 +0000 | [diff] [blame] | 610 | GenerateBitcode(Composite.get(), BitcodeOutputFilename); |
Reid Spencer | 1ef8bda | 2004-12-30 05:36:08 +0000 | [diff] [blame] | 611 | |
Chris Lattner | 316e326 | 2009-10-22 00:52:28 +0000 | [diff] [blame] | 612 | // If we are not linking a library, generate either a native executable |
| 613 | // or a JIT shell script, depending upon what the user wants. |
| 614 | if (!LinkAsLibrary) { |
| 615 | // If the user wants to run a post-link optimization, run it now. |
| 616 | if (!PostLinkOpts.empty()) { |
| 617 | std::vector<std::string> opts = PostLinkOpts; |
| 618 | for (std::vector<std::string>::iterator I = opts.begin(), |
| 619 | E = opts.end(); I != E; ++I) { |
| 620 | sys::Path prog(*I); |
| 621 | if (!prog.canExecute()) { |
| 622 | prog = sys::Program::FindProgramByName(*I); |
| 623 | if (prog.isEmpty()) |
| 624 | PrintAndExit(std::string("Optimization program '") + *I + |
Chris Lattner | 25c54c0 | 2010-03-23 21:59:43 +0000 | [diff] [blame] | 625 | "' is not found or not executable.", Composite.get()); |
Chris Lattner | 316e326 | 2009-10-22 00:52:28 +0000 | [diff] [blame] | 626 | } |
| 627 | // Get the program arguments |
| 628 | sys::Path tmp_output("opt_result"); |
| 629 | std::string ErrMsg; |
| 630 | if (tmp_output.createTemporaryFileOnDisk(true, &ErrMsg)) |
Chris Lattner | 25c54c0 | 2010-03-23 21:59:43 +0000 | [diff] [blame] | 631 | PrintAndExit(ErrMsg, Composite.get()); |
Reid Spencer | 708585a | 2007-02-09 03:08:06 +0000 | [diff] [blame] | 632 | |
Chris Lattner | 316e326 | 2009-10-22 00:52:28 +0000 | [diff] [blame] | 633 | const char* args[4]; |
| 634 | args[0] = I->c_str(); |
| 635 | args[1] = BitcodeOutputFilename.c_str(); |
| 636 | args[2] = tmp_output.c_str(); |
| 637 | args[3] = 0; |
| 638 | if (0 == sys::Program::ExecuteAndWait(prog, args, 0,0,0,0, &ErrMsg)) { |
Dan Gohman | 92f5fcc | 2010-03-27 16:36:08 +0000 | [diff] [blame] | 639 | if (tmp_output.isBitcodeFile()) { |
Chris Lattner | 316e326 | 2009-10-22 00:52:28 +0000 | [diff] [blame] | 640 | sys::Path target(BitcodeOutputFilename); |
| 641 | target.eraseFromDisk(); |
| 642 | if (tmp_output.renamePathOnDisk(target, &ErrMsg)) |
Chris Lattner | 25c54c0 | 2010-03-23 21:59:43 +0000 | [diff] [blame] | 643 | PrintAndExit(ErrMsg, Composite.get(), 2); |
Chris Lattner | 316e326 | 2009-10-22 00:52:28 +0000 | [diff] [blame] | 644 | } else |
Chris Lattner | 25c54c0 | 2010-03-23 21:59:43 +0000 | [diff] [blame] | 645 | PrintAndExit("Post-link optimization output is not bitcode", |
| 646 | Composite.get()); |
Chris Lattner | 316e326 | 2009-10-22 00:52:28 +0000 | [diff] [blame] | 647 | } else { |
Chris Lattner | 25c54c0 | 2010-03-23 21:59:43 +0000 | [diff] [blame] | 648 | PrintAndExit(ErrMsg, Composite.get()); |
Reid Spencer | 73a74be | 2005-12-21 05:03:23 +0000 | [diff] [blame] | 649 | } |
| 650 | } |
Reid Spencer | 708585a | 2007-02-09 03:08:06 +0000 | [diff] [blame] | 651 | } |
Chris Lattner | 316e326 | 2009-10-22 00:52:28 +0000 | [diff] [blame] | 652 | |
| 653 | // If the user wants to generate a native executable, compile it from the |
| 654 | // bitcode file. |
| 655 | // |
| 656 | // Otherwise, create a script that will run the bitcode through the JIT. |
| 657 | if (Native) { |
| 658 | // Name of the Assembly Language output file |
| 659 | sys::Path AssemblyFile ( OutputFilename); |
| 660 | AssemblyFile.appendSuffix("s"); |
| 661 | |
Dan Gohman | 132a994 | 2010-03-30 19:56:41 +0000 | [diff] [blame] | 662 | // Mark the output files for removal. |
| 663 | FileRemover AssemblyFileRemover(AssemblyFile); |
Chris Lattner | 316e326 | 2009-10-22 00:52:28 +0000 | [diff] [blame] | 664 | sys::RemoveFileOnSignal(AssemblyFile); |
Chris Lattner | 316e326 | 2009-10-22 00:52:28 +0000 | [diff] [blame] | 665 | |
| 666 | // Determine the locations of the llc and gcc programs. |
Mikhail Glushenkov | d66f2b7 | 2010-11-03 16:14:16 +0000 | [diff] [blame] | 667 | sys::Path llc = PrependMainExecutablePath("llc", argv[0], |
| 668 | (void *)(intptr_t)&Optimize); |
Chris Lattner | 316e326 | 2009-10-22 00:52:28 +0000 | [diff] [blame] | 669 | if (llc.isEmpty()) |
Chris Lattner | 25c54c0 | 2010-03-23 21:59:43 +0000 | [diff] [blame] | 670 | PrintAndExit("Failed to find llc", Composite.get()); |
Chris Lattner | 316e326 | 2009-10-22 00:52:28 +0000 | [diff] [blame] | 671 | |
| 672 | sys::Path gcc = sys::Program::FindProgramByName("gcc"); |
| 673 | if (gcc.isEmpty()) |
Chris Lattner | 25c54c0 | 2010-03-23 21:59:43 +0000 | [diff] [blame] | 674 | PrintAndExit("Failed to find gcc", Composite.get()); |
Chris Lattner | 316e326 | 2009-10-22 00:52:28 +0000 | [diff] [blame] | 675 | |
| 676 | // Generate an assembly language file for the bitcode. |
| 677 | std::string ErrMsg; |
| 678 | if (0 != GenerateAssembly(AssemblyFile.str(), BitcodeOutputFilename, |
| 679 | llc, ErrMsg)) |
Chris Lattner | 25c54c0 | 2010-03-23 21:59:43 +0000 | [diff] [blame] | 680 | PrintAndExit(ErrMsg, Composite.get()); |
Chris Lattner | 316e326 | 2009-10-22 00:52:28 +0000 | [diff] [blame] | 681 | |
| 682 | if (0 != GenerateNative(OutputFilename, AssemblyFile.str(), |
| 683 | NativeLinkItems, gcc, envp, ErrMsg)) |
Chris Lattner | 25c54c0 | 2010-03-23 21:59:43 +0000 | [diff] [blame] | 684 | PrintAndExit(ErrMsg, Composite.get()); |
Chris Lattner | 316e326 | 2009-10-22 00:52:28 +0000 | [diff] [blame] | 685 | } else if (NativeCBE) { |
| 686 | sys::Path CFile (OutputFilename); |
| 687 | CFile.appendSuffix("cbe.c"); |
| 688 | |
Dan Gohman | 132a994 | 2010-03-30 19:56:41 +0000 | [diff] [blame] | 689 | // Mark the output files for removal. |
| 690 | FileRemover CFileRemover(CFile); |
Chris Lattner | 316e326 | 2009-10-22 00:52:28 +0000 | [diff] [blame] | 691 | sys::RemoveFileOnSignal(CFile); |
Chris Lattner | 316e326 | 2009-10-22 00:52:28 +0000 | [diff] [blame] | 692 | |
| 693 | // Determine the locations of the llc and gcc programs. |
Mikhail Glushenkov | d66f2b7 | 2010-11-03 16:14:16 +0000 | [diff] [blame] | 694 | sys::Path llc = PrependMainExecutablePath("llc", argv[0], |
| 695 | (void *)(intptr_t)&Optimize); |
Chris Lattner | 316e326 | 2009-10-22 00:52:28 +0000 | [diff] [blame] | 696 | if (llc.isEmpty()) |
Chris Lattner | 25c54c0 | 2010-03-23 21:59:43 +0000 | [diff] [blame] | 697 | PrintAndExit("Failed to find llc", Composite.get()); |
Chris Lattner | 316e326 | 2009-10-22 00:52:28 +0000 | [diff] [blame] | 698 | |
| 699 | sys::Path gcc = sys::Program::FindProgramByName("gcc"); |
| 700 | if (gcc.isEmpty()) |
Chris Lattner | 25c54c0 | 2010-03-23 21:59:43 +0000 | [diff] [blame] | 701 | PrintAndExit("Failed to find gcc", Composite.get()); |
Chris Lattner | 316e326 | 2009-10-22 00:52:28 +0000 | [diff] [blame] | 702 | |
| 703 | // Generate an assembly language file for the bitcode. |
| 704 | std::string ErrMsg; |
| 705 | if (GenerateCFile(CFile.str(), BitcodeOutputFilename, llc, ErrMsg)) |
Chris Lattner | 25c54c0 | 2010-03-23 21:59:43 +0000 | [diff] [blame] | 706 | PrintAndExit(ErrMsg, Composite.get()); |
Chris Lattner | 316e326 | 2009-10-22 00:52:28 +0000 | [diff] [blame] | 707 | |
Mikhail Glushenkov | 572ec1f | 2010-11-02 20:32:52 +0000 | [diff] [blame] | 708 | if (GenerateNative(OutputFilename, CFile.str(), |
Chris Lattner | 316e326 | 2009-10-22 00:52:28 +0000 | [diff] [blame] | 709 | NativeLinkItems, gcc, envp, ErrMsg)) |
Chris Lattner | 25c54c0 | 2010-03-23 21:59:43 +0000 | [diff] [blame] | 710 | PrintAndExit(ErrMsg, Composite.get()); |
Chris Lattner | 316e326 | 2009-10-22 00:52:28 +0000 | [diff] [blame] | 711 | } else { |
Chris Lattner | 25c54c0 | 2010-03-23 21:59:43 +0000 | [diff] [blame] | 712 | EmitShellScript(argv, Composite.get()); |
Chris Lattner | 316e326 | 2009-10-22 00:52:28 +0000 | [diff] [blame] | 713 | } |
| 714 | |
| 715 | // Make the script executable... |
| 716 | std::string ErrMsg; |
| 717 | if (sys::Path(OutputFilename).makeExecutableOnDisk(&ErrMsg)) |
Chris Lattner | 25c54c0 | 2010-03-23 21:59:43 +0000 | [diff] [blame] | 718 | PrintAndExit(ErrMsg, Composite.get()); |
Chris Lattner | 316e326 | 2009-10-22 00:52:28 +0000 | [diff] [blame] | 719 | |
| 720 | // Make the bitcode file readable and directly executable in LLEE as well |
| 721 | if (sys::Path(BitcodeOutputFilename).makeExecutableOnDisk(&ErrMsg)) |
Chris Lattner | 25c54c0 | 2010-03-23 21:59:43 +0000 | [diff] [blame] | 722 | PrintAndExit(ErrMsg, Composite.get()); |
Chris Lattner | 316e326 | 2009-10-22 00:52:28 +0000 | [diff] [blame] | 723 | |
| 724 | if (sys::Path(BitcodeOutputFilename).makeReadableOnDisk(&ErrMsg)) |
Chris Lattner | 25c54c0 | 2010-03-23 21:59:43 +0000 | [diff] [blame] | 725 | PrintAndExit(ErrMsg, Composite.get()); |
Reid Spencer | c0af3f0 | 2004-09-13 01:27:53 +0000 | [diff] [blame] | 726 | } |
Reid Spencer | 708585a | 2007-02-09 03:08:06 +0000 | [diff] [blame] | 727 | |
Dan Gohman | 132a994 | 2010-03-30 19:56:41 +0000 | [diff] [blame] | 728 | // Operations which may fail are now complete. |
| 729 | BitcodeOutputRemover.releaseFile(); |
| 730 | if (!LinkAsLibrary) |
| 731 | OutputRemover.releaseFile(); |
| 732 | |
Reid Spencer | 708585a | 2007-02-09 03:08:06 +0000 | [diff] [blame] | 733 | // Graceful exit |
| 734 | return 0; |
Reid Spencer | c0af3f0 | 2004-09-13 01:27:53 +0000 | [diff] [blame] | 735 | } |