blob: 3149a7a494a99be36bfa6e8aa5730a75b156c4ff [file] [log] [blame]
Dan Gohmanf17a25c2007-07-18 16:29:46 +00001//===-- ToolRunner.cpp ----------------------------------------------------===//
2//
3// The LLVM Compiler Infrastructure
4//
Chris Lattner5f5a5732007-12-29 20:44:31 +00005// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
Dan Gohmanf17a25c2007-07-18 16:29:46 +00007//
8//===----------------------------------------------------------------------===//
9//
10// This file implements the interfaces described in the ToolRunner.h file.
11//
12//===----------------------------------------------------------------------===//
13
14#define DEBUG_TYPE "toolrunner"
15#include "ToolRunner.h"
Dan Gohmanf17a25c2007-07-18 16:29:46 +000016#include "llvm/System/Program.h"
17#include "llvm/Support/CommandLine.h"
18#include "llvm/Support/Debug.h"
19#include "llvm/Support/FileUtilities.h"
Chris Lattnerb1aa85b2009-08-23 22:45:37 +000020#include "llvm/Support/raw_ostream.h"
21#include "llvm/Config/config.h" // for HAVE_LINK_R
Dan Gohmanf17a25c2007-07-18 16:29:46 +000022#include <fstream>
23#include <sstream>
Dan Gohmanf17a25c2007-07-18 16:29:46 +000024using namespace llvm;
25
Anton Korobeynikov43cf4f12009-08-05 09:32:10 +000026namespace llvm {
27 cl::opt<bool>
28 SaveTemps("save-temps", cl::init(false), cl::desc("Save temporary files"));
29}
30
Dan Gohmanf17a25c2007-07-18 16:29:46 +000031namespace {
32 cl::opt<std::string>
Evan Cheng0646fff2008-09-09 06:11:26 +000033 RemoteClient("remote-client",
34 cl::desc("Remote execution client (rsh/ssh)"));
Dan Gohmanf17a25c2007-07-18 16:29:46 +000035
36 cl::opt<std::string>
Evan Cheng0646fff2008-09-09 06:11:26 +000037 RemoteHost("remote-host",
38 cl::desc("Remote execution (rsh/ssh) host"));
39
40 cl::opt<std::string>
David Goodwinec1edba2009-07-10 21:39:28 +000041 RemotePort("remote-port",
42 cl::desc("Remote execution (rsh/ssh) port"));
43
44 cl::opt<std::string>
Evan Cheng0646fff2008-09-09 06:11:26 +000045 RemoteUser("remote-user",
46 cl::desc("Remote execution (rsh/ssh) user id"));
47
48 cl::opt<std::string>
49 RemoteExtra("remote-extra-options",
50 cl::desc("Remote execution (rsh/ssh) extra options"));
Dan Gohmanf17a25c2007-07-18 16:29:46 +000051}
52
Viktor Kutuzov2743a2e2009-07-18 18:39:24 +000053/// RunProgramWithTimeout - This function provides an alternate interface
54/// to the sys::Program::ExecuteAndWait interface.
Nick Lewycky3cefdd32010-04-12 05:08:25 +000055/// @see sys::Program::ExecuteAndWait
Dan Gohmanf17a25c2007-07-18 16:29:46 +000056static int RunProgramWithTimeout(const sys::Path &ProgramPath,
57 const char **Args,
58 const sys::Path &StdInFile,
59 const sys::Path &StdOutFile,
60 const sys::Path &StdErrFile,
61 unsigned NumSeconds = 0,
62 unsigned MemoryLimit = 0) {
63 const sys::Path* redirects[3];
64 redirects[0] = &StdInFile;
65 redirects[1] = &StdOutFile;
66 redirects[2] = &StdErrFile;
Viktor Kutuzov2743a2e2009-07-18 18:39:24 +000067
68#if 0 // For debug purposes
69 {
Dan Gohmanf8b81bf2009-07-15 16:35:29 +000070 errs() << "RUN:";
Dan Gohmanf17a25c2007-07-18 16:29:46 +000071 for (unsigned i = 0; Args[i]; ++i)
Dan Gohmanf8b81bf2009-07-15 16:35:29 +000072 errs() << " " << Args[i];
73 errs() << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +000074 }
Viktor Kutuzov2743a2e2009-07-18 18:39:24 +000075#endif
Dan Gohmanf17a25c2007-07-18 16:29:46 +000076
77 return
78 sys::Program::ExecuteAndWait(ProgramPath, Args, 0, redirects,
79 NumSeconds, MemoryLimit);
80}
81
Viktor Kutuzov2743a2e2009-07-18 18:39:24 +000082/// RunProgramRemotelyWithTimeout - This function runs the given program
83/// remotely using the given remote client and the sys::Program::ExecuteAndWait.
84/// Returns the remote program exit code or reports a remote client error if it
85/// fails. Remote client is required to return 255 if it failed or program exit
86/// code otherwise.
Nick Lewycky3cefdd32010-04-12 05:08:25 +000087/// @see sys::Program::ExecuteAndWait
Viktor Kutuzov2743a2e2009-07-18 18:39:24 +000088static int RunProgramRemotelyWithTimeout(const sys::Path &RemoteClientPath,
89 const char **Args,
90 const sys::Path &StdInFile,
91 const sys::Path &StdOutFile,
92 const sys::Path &StdErrFile,
93 unsigned NumSeconds = 0,
94 unsigned MemoryLimit = 0) {
95 const sys::Path* redirects[3];
96 redirects[0] = &StdInFile;
97 redirects[1] = &StdOutFile;
98 redirects[2] = &StdErrFile;
Dan Gohmanf17a25c2007-07-18 16:29:46 +000099
Viktor Kutuzov2743a2e2009-07-18 18:39:24 +0000100#if 0 // For debug purposes
101 {
102 errs() << "RUN:";
103 for (unsigned i = 0; Args[i]; ++i)
104 errs() << " " << Args[i];
105 errs() << "\n";
106 }
107#endif
108
109 // Run the program remotely with the remote client
110 int ReturnCode = sys::Program::ExecuteAndWait(RemoteClientPath, Args,
111 0, redirects, NumSeconds, MemoryLimit);
112
113 // Has the remote client fail?
114 if (255 == ReturnCode) {
115 std::ostringstream OS;
116 OS << "\nError running remote client:\n ";
117 for (const char **Arg = Args; *Arg; ++Arg)
118 OS << " " << *Arg;
119 OS << "\n";
120
121 // The error message is in the output file, let's print it out from there.
122 std::ifstream ErrorFile(StdOutFile.c_str());
123 if (ErrorFile) {
124 std::copy(std::istreambuf_iterator<char>(ErrorFile),
125 std::istreambuf_iterator<char>(),
126 std::ostreambuf_iterator<char>(OS));
127 ErrorFile.close();
128 }
129
Nick Lewycky3cefdd32010-04-12 05:08:25 +0000130 errs() << OS;
Viktor Kutuzov2743a2e2009-07-18 18:39:24 +0000131 }
132
133 return ReturnCode;
134}
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000135
Duncan Sandsc40ce132010-05-24 07:49:55 +0000136static std::string ProcessFailure(sys::Path ProgPath, const char** Args,
137 unsigned Timeout = 0,
138 unsigned MemoryLimit = 0) {
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000139 std::ostringstream OS;
140 OS << "\nError running tool:\n ";
141 for (const char **Arg = Args; *Arg; ++Arg)
142 OS << " " << *Arg;
143 OS << "\n";
Chris Lattner51ff0582010-03-16 06:41:47 +0000144
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000145 // Rerun the compiler, capturing any error messages to print them.
Dan Gohman3c814b32009-02-12 20:53:27 +0000146 sys::Path ErrorFilename("bugpoint.program_error_messages");
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000147 std::string ErrMsg;
148 if (ErrorFilename.makeUnique(true, &ErrMsg)) {
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000149 errs() << "Error making unique filename: " << ErrMsg << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000150 exit(1);
151 }
152 RunProgramWithTimeout(ProgPath, Args, sys::Path(""), ErrorFilename,
Duncan Sandsc40ce132010-05-24 07:49:55 +0000153 ErrorFilename, Timeout, MemoryLimit);
154 // FIXME: check return code ?
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000155
156 // Print out the error messages generated by GCC if possible...
157 std::ifstream ErrorFile(ErrorFilename.c_str());
158 if (ErrorFile) {
159 std::copy(std::istreambuf_iterator<char>(ErrorFile),
160 std::istreambuf_iterator<char>(),
161 std::ostreambuf_iterator<char>(OS));
162 ErrorFile.close();
163 }
164
165 ErrorFilename.eraseFromDisk();
Nick Lewycky3cefdd32010-04-12 05:08:25 +0000166 return OS.str();
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000167}
168
169//===---------------------------------------------------------------------===//
170// LLI Implementation of AbstractIntepreter interface
171//
172namespace {
173 class LLI : public AbstractInterpreter {
174 std::string LLIPath; // The path to the LLI executable
175 std::vector<std::string> ToolArgs; // Args to pass to LLI
176 public:
177 LLI(const std::string &Path, const std::vector<std::string> *Args)
178 : LLIPath(Path) {
179 ToolArgs.clear ();
180 if (Args) { ToolArgs = *Args; }
181 }
182
183 virtual int ExecuteProgram(const std::string &Bitcode,
184 const std::vector<std::string> &Args,
185 const std::string &InputFile,
186 const std::string &OutputFile,
Nick Lewycky3cefdd32010-04-12 05:08:25 +0000187 std::string *Error,
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000188 const std::vector<std::string> &GCCArgs,
189 const std::vector<std::string> &SharedLibs =
190 std::vector<std::string>(),
191 unsigned Timeout = 0,
192 unsigned MemoryLimit = 0);
193 };
194}
195
196int LLI::ExecuteProgram(const std::string &Bitcode,
197 const std::vector<std::string> &Args,
198 const std::string &InputFile,
199 const std::string &OutputFile,
Nick Lewycky3cefdd32010-04-12 05:08:25 +0000200 std::string *Error,
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000201 const std::vector<std::string> &GCCArgs,
202 const std::vector<std::string> &SharedLibs,
203 unsigned Timeout,
204 unsigned MemoryLimit) {
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000205 std::vector<const char*> LLIArgs;
206 LLIArgs.push_back(LLIPath.c_str());
207 LLIArgs.push_back("-force-interpreter=true");
208
Duncan Sands18d02882009-11-17 10:20:22 +0000209 for (std::vector<std::string>::const_iterator i = SharedLibs.begin(), e = SharedLibs.end(); i != e; ++i) {
210 LLIArgs.push_back("-load");
211 LLIArgs.push_back((*i).c_str());
212 }
213
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000214 // Add any extra LLI args.
215 for (unsigned i = 0, e = ToolArgs.size(); i != e; ++i)
216 LLIArgs.push_back(ToolArgs[i].c_str());
217
218 LLIArgs.push_back(Bitcode.c_str());
219 // Add optional parameters to the running program from Argv
220 for (unsigned i=0, e = Args.size(); i != e; ++i)
221 LLIArgs.push_back(Args[i].c_str());
222 LLIArgs.push_back(0);
223
Dan Gohmandd7ca8e2009-07-15 17:29:42 +0000224 outs() << "<lli>"; outs().flush();
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000225 DEBUG(errs() << "\nAbout to run:\t";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000226 for (unsigned i=0, e = LLIArgs.size()-1; i != e; ++i)
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000227 errs() << " " << LLIArgs[i];
228 errs() << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000229 );
230 return RunProgramWithTimeout(sys::Path(LLIPath), &LLIArgs[0],
231 sys::Path(InputFile), sys::Path(OutputFile), sys::Path(OutputFile),
232 Timeout, MemoryLimit);
233}
234
235// LLI create method - Try to find the LLI executable
Dan Gohman8e0b7b9d2009-08-05 20:21:17 +0000236AbstractInterpreter *AbstractInterpreter::createLLI(const char *Argv0,
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000237 std::string &Message,
238 const std::vector<std::string> *ToolArgs) {
Dan Gohman8e0b7b9d2009-08-05 20:21:17 +0000239 std::string LLIPath =
Chris Lattnerb1aa85b2009-08-23 22:45:37 +0000240 FindExecutable("lli", Argv0, (void *)(intptr_t)&createLLI).str();
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000241 if (!LLIPath.empty()) {
242 Message = "Found lli: " + LLIPath + "\n";
243 return new LLI(LLIPath, ToolArgs);
244 }
245
246 Message = "Cannot find `lli' in executable directory or PATH!\n";
247 return 0;
248}
249
Anton Korobeynikovfc0116f2008-04-28 20:53:48 +0000250//===---------------------------------------------------------------------===//
251// Custom execution command implementation of AbstractIntepreter interface
252//
253// Allows using a custom command for executing the bitcode, thus allows,
254// for example, to invoke a cross compiler for code generation followed by
255// a simulator that executes the generated binary.
256namespace {
257 class CustomExecutor : public AbstractInterpreter {
258 std::string ExecutionCommand;
259 std::vector<std::string> ExecutorArgs;
260 public:
261 CustomExecutor(
262 const std::string &ExecutionCmd, std::vector<std::string> ExecArgs) :
263 ExecutionCommand(ExecutionCmd), ExecutorArgs(ExecArgs) {}
264
265 virtual int ExecuteProgram(const std::string &Bitcode,
266 const std::vector<std::string> &Args,
267 const std::string &InputFile,
268 const std::string &OutputFile,
Nick Lewycky3cefdd32010-04-12 05:08:25 +0000269 std::string *Error,
Anton Korobeynikovfc0116f2008-04-28 20:53:48 +0000270 const std::vector<std::string> &GCCArgs,
271 const std::vector<std::string> &SharedLibs =
Nick Lewycky3cefdd32010-04-12 05:08:25 +0000272 std::vector<std::string>(),
Anton Korobeynikovfc0116f2008-04-28 20:53:48 +0000273 unsigned Timeout = 0,
274 unsigned MemoryLimit = 0);
275 };
276}
277
278int CustomExecutor::ExecuteProgram(const std::string &Bitcode,
279 const std::vector<std::string> &Args,
280 const std::string &InputFile,
281 const std::string &OutputFile,
Nick Lewycky3cefdd32010-04-12 05:08:25 +0000282 std::string *Error,
Anton Korobeynikovfc0116f2008-04-28 20:53:48 +0000283 const std::vector<std::string> &GCCArgs,
284 const std::vector<std::string> &SharedLibs,
285 unsigned Timeout,
286 unsigned MemoryLimit) {
287
288 std::vector<const char*> ProgramArgs;
289 ProgramArgs.push_back(ExecutionCommand.c_str());
290
291 for (std::size_t i = 0; i < ExecutorArgs.size(); ++i)
292 ProgramArgs.push_back(ExecutorArgs.at(i).c_str());
293 ProgramArgs.push_back(Bitcode.c_str());
294 ProgramArgs.push_back(0);
295
296 // Add optional parameters to the running program from Argv
Nick Lewycky3cefdd32010-04-12 05:08:25 +0000297 for (unsigned i = 0, e = Args.size(); i != e; ++i)
Anton Korobeynikovfc0116f2008-04-28 20:53:48 +0000298 ProgramArgs.push_back(Args[i].c_str());
299
300 return RunProgramWithTimeout(
301 sys::Path(ExecutionCommand),
302 &ProgramArgs[0], sys::Path(InputFile), sys::Path(OutputFile),
303 sys::Path(OutputFile), Timeout, MemoryLimit);
304}
305
306// Custom execution environment create method, takes the execution command
307// as arguments
308AbstractInterpreter *AbstractInterpreter::createCustom(
Anton Korobeynikovfc0116f2008-04-28 20:53:48 +0000309 std::string &Message,
310 const std::string &ExecCommandLine) {
311
312 std::string Command = "";
313 std::vector<std::string> Args;
314 std::string delimiters = " ";
315
316 // Tokenize the ExecCommandLine to the command and the args to allow
317 // defining a full command line as the command instead of just the
318 // executed program. We cannot just pass the whole string after the command
319 // as a single argument because then program sees only a single
320 // command line argument (with spaces in it: "foo bar" instead
321 // of "foo" and "bar").
322
323 // code borrowed from:
324 // http://oopweb.com/CPP/Documents/CPPHOWTO/Volume/C++Programming-HOWTO-7.html
325 std::string::size_type lastPos =
326 ExecCommandLine.find_first_not_of(delimiters, 0);
327 std::string::size_type pos =
328 ExecCommandLine.find_first_of(delimiters, lastPos);
329
330 while (std::string::npos != pos || std::string::npos != lastPos) {
331 std::string token = ExecCommandLine.substr(lastPos, pos - lastPos);
332 if (Command == "")
333 Command = token;
334 else
335 Args.push_back(token);
336 // Skip delimiters. Note the "not_of"
337 lastPos = ExecCommandLine.find_first_not_of(delimiters, pos);
338 // Find next "non-delimiter"
339 pos = ExecCommandLine.find_first_of(delimiters, lastPos);
340 }
341
Chris Lattnerb1aa85b2009-08-23 22:45:37 +0000342 std::string CmdPath = sys::Program::FindProgramByName(Command).str();
Anton Korobeynikovfc0116f2008-04-28 20:53:48 +0000343 if (CmdPath.empty()) {
344 Message =
345 std::string("Cannot find '") + Command +
346 "' in executable directory or PATH!\n";
347 return 0;
348 }
349
350 Message = "Found command in: " + CmdPath + "\n";
351
352 return new CustomExecutor(CmdPath, Args);
353}
354
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000355//===----------------------------------------------------------------------===//
356// LLC Implementation of AbstractIntepreter interface
357//
358GCC::FileType LLC::OutputCode(const std::string &Bitcode,
Duncan Sandsc40ce132010-05-24 07:49:55 +0000359 sys::Path &OutputAsmFile, std::string &Error,
360 unsigned Timeout, unsigned MemoryLimit) {
Chris Lattner51ff0582010-03-16 06:41:47 +0000361 const char *Suffix = (UseIntegratedAssembler ? ".llc.o" : ".llc.s");
362 sys::Path uniqueFile(Bitcode + Suffix);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000363 std::string ErrMsg;
364 if (uniqueFile.makeUnique(true, &ErrMsg)) {
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000365 errs() << "Error making unique filename: " << ErrMsg << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000366 exit(1);
367 }
368 OutputAsmFile = uniqueFile;
369 std::vector<const char *> LLCArgs;
Chris Lattner51ff0582010-03-16 06:41:47 +0000370 LLCArgs.push_back(LLCPath.c_str());
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000371
372 // Add any extra LLC args.
373 for (unsigned i = 0, e = ToolArgs.size(); i != e; ++i)
374 LLCArgs.push_back(ToolArgs[i].c_str());
375
Chris Lattner51ff0582010-03-16 06:41:47 +0000376 LLCArgs.push_back("-o");
377 LLCArgs.push_back(OutputAsmFile.c_str()); // Output to the Asm file
378 LLCArgs.push_back(Bitcode.c_str()); // This is the input bitcode
379
380 if (UseIntegratedAssembler)
381 LLCArgs.push_back("-filetype=obj");
382
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000383 LLCArgs.push_back (0);
384
Chris Lattner51ff0582010-03-16 06:41:47 +0000385 outs() << (UseIntegratedAssembler ? "<llc-ia>" : "<llc>");
386 outs().flush();
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000387 DEBUG(errs() << "\nAbout to run:\t";
Nick Lewycky3cefdd32010-04-12 05:08:25 +0000388 for (unsigned i = 0, e = LLCArgs.size()-1; i != e; ++i)
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000389 errs() << " " << LLCArgs[i];
390 errs() << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000391 );
392 if (RunProgramWithTimeout(sys::Path(LLCPath), &LLCArgs[0],
Duncan Sandsc40ce132010-05-24 07:49:55 +0000393 sys::Path(), sys::Path(), sys::Path(),
394 Timeout, MemoryLimit))
395 Error = ProcessFailure(sys::Path(LLCPath), &LLCArgs[0],
396 Timeout, MemoryLimit);
Nick Lewycky3cefdd32010-04-12 05:08:25 +0000397 return UseIntegratedAssembler ? GCC::ObjectFile : GCC::AsmFile;
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000398}
399
Duncan Sandsc40ce132010-05-24 07:49:55 +0000400void LLC::compileProgram(const std::string &Bitcode, std::string *Error,
401 unsigned Timeout, unsigned MemoryLimit) {
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000402 sys::Path OutputAsmFile;
Duncan Sandsc40ce132010-05-24 07:49:55 +0000403 OutputCode(Bitcode, OutputAsmFile, *Error, Timeout, MemoryLimit);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000404 OutputAsmFile.eraseFromDisk();
405}
406
407int LLC::ExecuteProgram(const std::string &Bitcode,
408 const std::vector<std::string> &Args,
409 const std::string &InputFile,
410 const std::string &OutputFile,
Nick Lewycky3cefdd32010-04-12 05:08:25 +0000411 std::string *Error,
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000412 const std::vector<std::string> &ArgsForGCC,
413 const std::vector<std::string> &SharedLibs,
414 unsigned Timeout,
415 unsigned MemoryLimit) {
416
417 sys::Path OutputAsmFile;
Duncan Sandsc40ce132010-05-24 07:49:55 +0000418 GCC::FileType FileKind = OutputCode(Bitcode, OutputAsmFile, *Error, Timeout,
419 MemoryLimit);
Anton Korobeynikov43cf4f12009-08-05 09:32:10 +0000420 FileRemover OutFileRemover(OutputAsmFile, !SaveTemps);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000421
422 std::vector<std::string> GCCArgs(ArgsForGCC);
Bill Wendlingcd1389d2009-03-02 23:13:18 +0000423 GCCArgs.insert(GCCArgs.end(), SharedLibs.begin(), SharedLibs.end());
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000424
425 // Assuming LLC worked, compile the result with GCC and run it.
Chris Lattner51ff0582010-03-16 06:41:47 +0000426 return gcc->ExecuteProgram(OutputAsmFile.str(), Args, FileKind,
Nick Lewycky3cefdd32010-04-12 05:08:25 +0000427 InputFile, OutputFile, Error, GCCArgs,
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000428 Timeout, MemoryLimit);
429}
430
431/// createLLC - Try to find the LLC executable
432///
Dan Gohman8e0b7b9d2009-08-05 20:21:17 +0000433LLC *AbstractInterpreter::createLLC(const char *Argv0,
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000434 std::string &Message,
Kalle Raiskila2c448962010-05-10 07:38:37 +0000435 const std::string &GCCBinary,
Bill Wendlingcd1389d2009-03-02 23:13:18 +0000436 const std::vector<std::string> *Args,
Chris Lattner51ff0582010-03-16 06:41:47 +0000437 const std::vector<std::string> *GCCArgs,
438 bool UseIntegratedAssembler) {
Dan Gohman8e0b7b9d2009-08-05 20:21:17 +0000439 std::string LLCPath =
Chris Lattnerb1aa85b2009-08-23 22:45:37 +0000440 FindExecutable("llc", Argv0, (void *)(intptr_t)&createLLC).str();
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000441 if (LLCPath.empty()) {
442 Message = "Cannot find `llc' in executable directory or PATH!\n";
443 return 0;
444 }
445
446 Message = "Found llc: " + LLCPath + "\n";
Kalle Raiskila2c448962010-05-10 07:38:37 +0000447 GCC *gcc = GCC::create(Message, GCCBinary, GCCArgs);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000448 if (!gcc) {
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000449 errs() << Message << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000450 exit(1);
451 }
Kalle Raiskila2c448962010-05-10 07:38:37 +0000452 return new LLC(LLCPath, gcc, Args, UseIntegratedAssembler);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000453}
454
455//===---------------------------------------------------------------------===//
456// JIT Implementation of AbstractIntepreter interface
457//
458namespace {
459 class JIT : public AbstractInterpreter {
460 std::string LLIPath; // The path to the LLI executable
461 std::vector<std::string> ToolArgs; // Args to pass to LLI
462 public:
463 JIT(const std::string &Path, const std::vector<std::string> *Args)
464 : LLIPath(Path) {
465 ToolArgs.clear ();
466 if (Args) { ToolArgs = *Args; }
467 }
468
469 virtual int ExecuteProgram(const std::string &Bitcode,
470 const std::vector<std::string> &Args,
471 const std::string &InputFile,
472 const std::string &OutputFile,
Nick Lewycky3cefdd32010-04-12 05:08:25 +0000473 std::string *Error,
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000474 const std::vector<std::string> &GCCArgs =
475 std::vector<std::string>(),
476 const std::vector<std::string> &SharedLibs =
477 std::vector<std::string>(),
Nick Lewycky3cefdd32010-04-12 05:08:25 +0000478 unsigned Timeout = 0,
479 unsigned MemoryLimit = 0);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000480 };
481}
482
483int JIT::ExecuteProgram(const std::string &Bitcode,
484 const std::vector<std::string> &Args,
485 const std::string &InputFile,
486 const std::string &OutputFile,
Nick Lewycky3cefdd32010-04-12 05:08:25 +0000487 std::string *Error,
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000488 const std::vector<std::string> &GCCArgs,
489 const std::vector<std::string> &SharedLibs,
490 unsigned Timeout,
491 unsigned MemoryLimit) {
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000492 // Construct a vector of parameters, incorporating those from the command-line
493 std::vector<const char*> JITArgs;
494 JITArgs.push_back(LLIPath.c_str());
495 JITArgs.push_back("-force-interpreter=false");
496
497 // Add any extra LLI args.
498 for (unsigned i = 0, e = ToolArgs.size(); i != e; ++i)
499 JITArgs.push_back(ToolArgs[i].c_str());
500
501 for (unsigned i = 0, e = SharedLibs.size(); i != e; ++i) {
502 JITArgs.push_back("-load");
503 JITArgs.push_back(SharedLibs[i].c_str());
504 }
505 JITArgs.push_back(Bitcode.c_str());
506 // Add optional parameters to the running program from Argv
507 for (unsigned i=0, e = Args.size(); i != e; ++i)
508 JITArgs.push_back(Args[i].c_str());
509 JITArgs.push_back(0);
510
Dan Gohmandd7ca8e2009-07-15 17:29:42 +0000511 outs() << "<jit>"; outs().flush();
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000512 DEBUG(errs() << "\nAbout to run:\t";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000513 for (unsigned i=0, e = JITArgs.size()-1; i != e; ++i)
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000514 errs() << " " << JITArgs[i];
515 errs() << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000516 );
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000517 DEBUG(errs() << "\nSending output to " << OutputFile << "\n");
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000518 return RunProgramWithTimeout(sys::Path(LLIPath), &JITArgs[0],
519 sys::Path(InputFile), sys::Path(OutputFile), sys::Path(OutputFile),
520 Timeout, MemoryLimit);
521}
522
523/// createJIT - Try to find the LLI executable
524///
Dan Gohman8e0b7b9d2009-08-05 20:21:17 +0000525AbstractInterpreter *AbstractInterpreter::createJIT(const char *Argv0,
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000526 std::string &Message, const std::vector<std::string> *Args) {
Dan Gohman8e0b7b9d2009-08-05 20:21:17 +0000527 std::string LLIPath =
Chris Lattnerb1aa85b2009-08-23 22:45:37 +0000528 FindExecutable("lli", Argv0, (void *)(intptr_t)&createJIT).str();
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000529 if (!LLIPath.empty()) {
530 Message = "Found lli: " + LLIPath + "\n";
531 return new JIT(LLIPath, Args);
532 }
533
534 Message = "Cannot find `lli' in executable directory or PATH!\n";
535 return 0;
536}
537
538GCC::FileType CBE::OutputCode(const std::string &Bitcode,
Duncan Sandsc40ce132010-05-24 07:49:55 +0000539 sys::Path &OutputCFile, std::string &Error,
540 unsigned Timeout, unsigned MemoryLimit) {
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000541 sys::Path uniqueFile(Bitcode+".cbe.c");
542 std::string ErrMsg;
543 if (uniqueFile.makeUnique(true, &ErrMsg)) {
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000544 errs() << "Error making unique filename: " << ErrMsg << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000545 exit(1);
546 }
547 OutputCFile = uniqueFile;
548 std::vector<const char *> LLCArgs;
Nick Lewycky3cefdd32010-04-12 05:08:25 +0000549 LLCArgs.push_back(LLCPath.c_str());
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000550
551 // Add any extra LLC args.
552 for (unsigned i = 0, e = ToolArgs.size(); i != e; ++i)
553 LLCArgs.push_back(ToolArgs[i].c_str());
554
Nick Lewycky3cefdd32010-04-12 05:08:25 +0000555 LLCArgs.push_back("-o");
556 LLCArgs.push_back(OutputCFile.c_str()); // Output to the C file
557 LLCArgs.push_back("-march=c"); // Output C language
Nick Lewycky3cefdd32010-04-12 05:08:25 +0000558 LLCArgs.push_back(Bitcode.c_str()); // This is the input bitcode
559 LLCArgs.push_back(0);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000560
Dan Gohmandd7ca8e2009-07-15 17:29:42 +0000561 outs() << "<cbe>"; outs().flush();
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000562 DEBUG(errs() << "\nAbout to run:\t";
Nick Lewycky3cefdd32010-04-12 05:08:25 +0000563 for (unsigned i = 0, e = LLCArgs.size()-1; i != e; ++i)
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000564 errs() << " " << LLCArgs[i];
565 errs() << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000566 );
567 if (RunProgramWithTimeout(LLCPath, &LLCArgs[0], sys::Path(), sys::Path(),
Duncan Sandsc40ce132010-05-24 07:49:55 +0000568 sys::Path(), Timeout, MemoryLimit))
569 Error = ProcessFailure(LLCPath, &LLCArgs[0], Timeout, MemoryLimit);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000570 return GCC::CFile;
571}
572
Duncan Sandsc40ce132010-05-24 07:49:55 +0000573void CBE::compileProgram(const std::string &Bitcode, std::string *Error,
574 unsigned Timeout, unsigned MemoryLimit) {
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000575 sys::Path OutputCFile;
Duncan Sandsc40ce132010-05-24 07:49:55 +0000576 OutputCode(Bitcode, OutputCFile, *Error, Timeout, MemoryLimit);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000577 OutputCFile.eraseFromDisk();
578}
579
580int CBE::ExecuteProgram(const std::string &Bitcode,
581 const std::vector<std::string> &Args,
582 const std::string &InputFile,
583 const std::string &OutputFile,
Nick Lewycky3cefdd32010-04-12 05:08:25 +0000584 std::string *Error,
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000585 const std::vector<std::string> &ArgsForGCC,
586 const std::vector<std::string> &SharedLibs,
587 unsigned Timeout,
588 unsigned MemoryLimit) {
589 sys::Path OutputCFile;
Duncan Sandsc40ce132010-05-24 07:49:55 +0000590 OutputCode(Bitcode, OutputCFile, *Error, Timeout, MemoryLimit);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000591
Anton Korobeynikov43cf4f12009-08-05 09:32:10 +0000592 FileRemover CFileRemove(OutputCFile, !SaveTemps);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000593
594 std::vector<std::string> GCCArgs(ArgsForGCC);
Bill Wendlingcd1389d2009-03-02 23:13:18 +0000595 GCCArgs.insert(GCCArgs.end(), SharedLibs.begin(), SharedLibs.end());
596
Chris Lattnerb1aa85b2009-08-23 22:45:37 +0000597 return gcc->ExecuteProgram(OutputCFile.str(), Args, GCC::CFile,
Nick Lewycky3cefdd32010-04-12 05:08:25 +0000598 InputFile, OutputFile, Error, GCCArgs,
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000599 Timeout, MemoryLimit);
600}
601
602/// createCBE - Try to find the 'llc' executable
603///
Dan Gohman8e0b7b9d2009-08-05 20:21:17 +0000604CBE *AbstractInterpreter::createCBE(const char *Argv0,
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000605 std::string &Message,
Kalle Raiskila2c448962010-05-10 07:38:37 +0000606 const std::string &GCCBinary,
Bill Wendlingcd1389d2009-03-02 23:13:18 +0000607 const std::vector<std::string> *Args,
608 const std::vector<std::string> *GCCArgs) {
Dan Gohman8e0b7b9d2009-08-05 20:21:17 +0000609 sys::Path LLCPath =
Dan Gohman108e0ed2009-08-05 21:03:39 +0000610 FindExecutable("llc", Argv0, (void *)(intptr_t)&createCBE);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000611 if (LLCPath.isEmpty()) {
612 Message =
613 "Cannot find `llc' in executable directory or PATH!\n";
614 return 0;
615 }
616
Chris Lattnerb1aa85b2009-08-23 22:45:37 +0000617 Message = "Found llc: " + LLCPath.str() + "\n";
Kalle Raiskila2c448962010-05-10 07:38:37 +0000618 GCC *gcc = GCC::create(Message, GCCBinary, GCCArgs);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000619 if (!gcc) {
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000620 errs() << Message << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000621 exit(1);
622 }
623 return new CBE(LLCPath, gcc, Args);
624}
625
626//===---------------------------------------------------------------------===//
627// GCC abstraction
628//
David Goodwinec1edba2009-07-10 21:39:28 +0000629
Chris Lattner51ff0582010-03-16 06:41:47 +0000630static bool IsARMArchitecture(std::vector<std::string> Args) {
David Goodwinec1edba2009-07-10 21:39:28 +0000631 for (std::vector<std::string>::const_iterator
632 I = Args.begin(), E = Args.end(); I != E; ++I) {
Jakob Stoklund Olesen663f02b2010-05-13 17:58:15 +0000633 if (StringRef(*I).equals_lower("-arch")) {
David Goodwinec1edba2009-07-10 21:39:28 +0000634 ++I;
Jakob Stoklund Olesen663f02b2010-05-13 17:58:15 +0000635 if (I != E && StringRef(*I).substr(0, strlen("arm")).equals_lower("arm"))
David Goodwinec1edba2009-07-10 21:39:28 +0000636 return true;
David Goodwinec1edba2009-07-10 21:39:28 +0000637 }
638 }
639
640 return false;
641}
642
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000643int GCC::ExecuteProgram(const std::string &ProgramFile,
644 const std::vector<std::string> &Args,
645 FileType fileType,
646 const std::string &InputFile,
647 const std::string &OutputFile,
Nick Lewycky3cefdd32010-04-12 05:08:25 +0000648 std::string *Error,
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000649 const std::vector<std::string> &ArgsForGCC,
650 unsigned Timeout,
651 unsigned MemoryLimit) {
652 std::vector<const char*> GCCArgs;
653
654 GCCArgs.push_back(GCCPath.c_str());
655
Chris Lattner51ff0582010-03-16 06:41:47 +0000656 if (TargetTriple.getArch() == Triple::x86)
657 GCCArgs.push_back("-m32");
658
Bill Wendlingcd1389d2009-03-02 23:13:18 +0000659 for (std::vector<std::string>::const_iterator
660 I = gccArgs.begin(), E = gccArgs.end(); I != E; ++I)
661 GCCArgs.push_back(I->c_str());
662
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000663 // Specify -x explicitly in case the extension is wonky
Chris Lattner51ff0582010-03-16 06:41:47 +0000664 if (fileType != ObjectFile) {
665 GCCArgs.push_back("-x");
666 if (fileType == CFile) {
667 GCCArgs.push_back("c");
668 GCCArgs.push_back("-fno-strict-aliasing");
669 } else {
670 GCCArgs.push_back("assembler");
Daniel Dunbard8590af2009-08-18 03:35:57 +0000671
Chris Lattner51ff0582010-03-16 06:41:47 +0000672 // For ARM architectures we don't want this flag. bugpoint isn't
673 // explicitly told what architecture it is working on, so we get
674 // it from gcc flags
675 if ((TargetTriple.getOS() == Triple::Darwin) &&
676 !IsARMArchitecture(ArgsForGCC))
677 GCCArgs.push_back("-force_cpusubtype_ALL");
678 }
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000679 }
Chris Lattner51ff0582010-03-16 06:41:47 +0000680
681 GCCArgs.push_back(ProgramFile.c_str()); // Specify the input filename.
682
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000683 GCCArgs.push_back("-x");
684 GCCArgs.push_back("none");
685 GCCArgs.push_back("-o");
686 sys::Path OutputBinary (ProgramFile+".gcc.exe");
687 std::string ErrMsg;
688 if (OutputBinary.makeUnique(true, &ErrMsg)) {
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000689 errs() << "Error making unique filename: " << ErrMsg << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000690 exit(1);
691 }
692 GCCArgs.push_back(OutputBinary.c_str()); // Output to the right file...
693
694 // Add any arguments intended for GCC. We locate them here because this is
695 // most likely -L and -l options that need to come before other libraries but
696 // after the source. Other options won't be sensitive to placement on the
697 // command line, so this should be safe.
698 for (unsigned i = 0, e = ArgsForGCC.size(); i != e; ++i)
699 GCCArgs.push_back(ArgsForGCC[i].c_str());
700
701 GCCArgs.push_back("-lm"); // Hard-code the math library...
702 GCCArgs.push_back("-O2"); // Optimize the program a bit...
703#if defined (HAVE_LINK_R)
704 GCCArgs.push_back("-Wl,-R."); // Search this dir for .so files
705#endif
Daniel Dunbard8590af2009-08-18 03:35:57 +0000706 if (TargetTriple.getArch() == Triple::sparc)
707 GCCArgs.push_back("-mcpu=v9");
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000708 GCCArgs.push_back(0); // NULL terminator
709
Dan Gohmandd7ca8e2009-07-15 17:29:42 +0000710 outs() << "<gcc>"; outs().flush();
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000711 DEBUG(errs() << "\nAbout to run:\t";
Nick Lewycky3cefdd32010-04-12 05:08:25 +0000712 for (unsigned i = 0, e = GCCArgs.size()-1; i != e; ++i)
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000713 errs() << " " << GCCArgs[i];
714 errs() << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000715 );
716 if (RunProgramWithTimeout(GCCPath, &GCCArgs[0], sys::Path(), sys::Path(),
717 sys::Path())) {
Nick Lewycky3cefdd32010-04-12 05:08:25 +0000718 *Error = ProcessFailure(GCCPath, &GCCArgs[0]);
719 return -1;
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000720 }
721
722 std::vector<const char*> ProgramArgs;
723
Evan Cheng0646fff2008-09-09 06:11:26 +0000724 if (RemoteClientPath.isEmpty())
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000725 ProgramArgs.push_back(OutputBinary.c_str());
726 else {
Evan Cheng0646fff2008-09-09 06:11:26 +0000727 ProgramArgs.push_back(RemoteClientPath.c_str());
728 ProgramArgs.push_back(RemoteHost.c_str());
Anton Korobeynikov33f97a62009-08-05 09:32:53 +0000729 if (!RemoteUser.empty()) {
730 ProgramArgs.push_back("-l");
731 ProgramArgs.push_back(RemoteUser.c_str());
732 }
David Goodwinec1edba2009-07-10 21:39:28 +0000733 if (!RemotePort.empty()) {
734 ProgramArgs.push_back("-p");
735 ProgramArgs.push_back(RemotePort.c_str());
736 }
Evan Cheng0646fff2008-09-09 06:11:26 +0000737 if (!RemoteExtra.empty()) {
738 ProgramArgs.push_back(RemoteExtra.c_str());
739 }
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000740
David Goodwinfbb0faf2009-07-20 17:15:03 +0000741 // Full path to the binary. We need to cd to the exec directory because
742 // there is a dylib there that the exec expects to find in the CWD
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000743 char* env_pwd = getenv("PWD");
David Goodwinfbb0faf2009-07-20 17:15:03 +0000744 std::string Exec = "cd ";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000745 Exec += env_pwd;
David Goodwinfbb0faf2009-07-20 17:15:03 +0000746 Exec += "; ./";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000747 Exec += OutputBinary.c_str();
748 ProgramArgs.push_back(Exec.c_str());
749 }
750
751 // Add optional parameters to the running program from Argv
Nick Lewycky3cefdd32010-04-12 05:08:25 +0000752 for (unsigned i = 0, e = Args.size(); i != e; ++i)
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000753 ProgramArgs.push_back(Args[i].c_str());
754 ProgramArgs.push_back(0); // NULL terminator
755
756 // Now that we have a binary, run it!
Dan Gohmandd7ca8e2009-07-15 17:29:42 +0000757 outs() << "<program>"; outs().flush();
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000758 DEBUG(errs() << "\nAbout to run:\t";
Nick Lewycky3cefdd32010-04-12 05:08:25 +0000759 for (unsigned i = 0, e = ProgramArgs.size()-1; i != e; ++i)
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000760 errs() << " " << ProgramArgs[i];
761 errs() << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000762 );
763
Anton Korobeynikov43cf4f12009-08-05 09:32:10 +0000764 FileRemover OutputBinaryRemover(OutputBinary, !SaveTemps);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000765
Viktor Kutuzov8c3e63c2009-07-14 19:10:55 +0000766 if (RemoteClientPath.isEmpty()) {
Nick Lewycky3cefdd32010-04-12 05:08:25 +0000767 DEBUG(errs() << "<run locally>");
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000768 return RunProgramWithTimeout(OutputBinary, &ProgramArgs[0],
769 sys::Path(InputFile), sys::Path(OutputFile), sys::Path(OutputFile),
770 Timeout, MemoryLimit);
Viktor Kutuzov8c3e63c2009-07-14 19:10:55 +0000771 } else {
Dan Gohmandd7ca8e2009-07-15 17:29:42 +0000772 outs() << "<run remotely>"; outs().flush();
Viktor Kutuzov2743a2e2009-07-18 18:39:24 +0000773 return RunProgramRemotelyWithTimeout(sys::Path(RemoteClientPath),
Viktor Kutuzov8c3e63c2009-07-14 19:10:55 +0000774 &ProgramArgs[0], sys::Path(InputFile), sys::Path(OutputFile),
775 sys::Path(OutputFile), Timeout, MemoryLimit);
Viktor Kutuzov8c3e63c2009-07-14 19:10:55 +0000776 }
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000777}
778
779int GCC::MakeSharedObject(const std::string &InputFile, FileType fileType,
780 std::string &OutputFile,
Nick Lewycky3cefdd32010-04-12 05:08:25 +0000781 const std::vector<std::string> &ArgsForGCC,
782 std::string &Error) {
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000783 sys::Path uniqueFilename(InputFile+LTDL_SHLIB_EXT);
784 std::string ErrMsg;
785 if (uniqueFilename.makeUnique(true, &ErrMsg)) {
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000786 errs() << "Error making unique filename: " << ErrMsg << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000787 exit(1);
788 }
Chris Lattnerb1aa85b2009-08-23 22:45:37 +0000789 OutputFile = uniqueFilename.str();
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000790
791 std::vector<const char*> GCCArgs;
792
793 GCCArgs.push_back(GCCPath.c_str());
Evan Chengf94bcf62009-03-12 00:53:34 +0000794
Chris Lattner51ff0582010-03-16 06:41:47 +0000795 if (TargetTriple.getArch() == Triple::x86)
796 GCCArgs.push_back("-m32");
797
Evan Chengf94bcf62009-03-12 00:53:34 +0000798 for (std::vector<std::string>::const_iterator
799 I = gccArgs.begin(), E = gccArgs.end(); I != E; ++I)
800 GCCArgs.push_back(I->c_str());
801
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000802 // Compile the C/asm file into a shared object
Chris Lattner51ff0582010-03-16 06:41:47 +0000803 if (fileType != ObjectFile) {
804 GCCArgs.push_back("-x");
805 GCCArgs.push_back(fileType == AsmFile ? "assembler" : "c");
806 }
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000807 GCCArgs.push_back("-fno-strict-aliasing");
808 GCCArgs.push_back(InputFile.c_str()); // Specify the input filename.
809 GCCArgs.push_back("-x");
810 GCCArgs.push_back("none");
Daniel Dunbard8590af2009-08-18 03:35:57 +0000811 if (TargetTriple.getArch() == Triple::sparc)
812 GCCArgs.push_back("-G"); // Compile a shared library, `-G' for Sparc
813 else if (TargetTriple.getOS() == Triple::Darwin) {
814 // link all source files into a single module in data segment, rather than
815 // generating blocks. dynamic_lookup requires that you set
816 // MACOSX_DEPLOYMENT_TARGET=10.3 in your env. FIXME: it would be better for
817 // bugpoint to just pass that in the environment of GCC.
818 GCCArgs.push_back("-single_module");
819 GCCArgs.push_back("-dynamiclib"); // `-dynamiclib' for MacOS X/PowerPC
820 GCCArgs.push_back("-undefined");
821 GCCArgs.push_back("dynamic_lookup");
822 } else
823 GCCArgs.push_back("-shared"); // `-shared' for Linux/X86, maybe others
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000824
Daniel Dunbard8590af2009-08-18 03:35:57 +0000825 if ((TargetTriple.getArch() == Triple::alpha) ||
826 (TargetTriple.getArch() == Triple::x86_64))
827 GCCArgs.push_back("-fPIC"); // Requires shared objs to contain PIC
828
829 if (TargetTriple.getArch() == Triple::sparc)
830 GCCArgs.push_back("-mcpu=v9");
831
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000832 GCCArgs.push_back("-o");
833 GCCArgs.push_back(OutputFile.c_str()); // Output to the right filename.
834 GCCArgs.push_back("-O2"); // Optimize the program a bit.
835
836
837
838 // Add any arguments intended for GCC. We locate them here because this is
839 // most likely -L and -l options that need to come before other libraries but
840 // after the source. Other options won't be sensitive to placement on the
841 // command line, so this should be safe.
842 for (unsigned i = 0, e = ArgsForGCC.size(); i != e; ++i)
843 GCCArgs.push_back(ArgsForGCC[i].c_str());
844 GCCArgs.push_back(0); // NULL terminator
845
846
847
Dan Gohmandd7ca8e2009-07-15 17:29:42 +0000848 outs() << "<gcc>"; outs().flush();
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000849 DEBUG(errs() << "\nAbout to run:\t";
Nick Lewycky3cefdd32010-04-12 05:08:25 +0000850 for (unsigned i = 0, e = GCCArgs.size()-1; i != e; ++i)
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000851 errs() << " " << GCCArgs[i];
852 errs() << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000853 );
854 if (RunProgramWithTimeout(GCCPath, &GCCArgs[0], sys::Path(), sys::Path(),
855 sys::Path())) {
Nick Lewycky3cefdd32010-04-12 05:08:25 +0000856 Error = ProcessFailure(GCCPath, &GCCArgs[0]);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000857 return 1;
858 }
859 return 0;
860}
861
862/// create - Try to find the `gcc' executable
863///
Dan Gohman8e0b7b9d2009-08-05 20:21:17 +0000864GCC *GCC::create(std::string &Message,
Kalle Raiskila2c448962010-05-10 07:38:37 +0000865 const std::string &GCCBinary,
Bill Wendlingcd1389d2009-03-02 23:13:18 +0000866 const std::vector<std::string> *Args) {
Kalle Raiskila2c448962010-05-10 07:38:37 +0000867 sys::Path GCCPath = sys::Program::FindProgramByName(GCCBinary);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000868 if (GCCPath.isEmpty()) {
Kalle Raiskila2c448962010-05-10 07:38:37 +0000869 Message = "Cannot find `"+ GCCBinary +"' in executable directory or PATH!\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000870 return 0;
871 }
872
Evan Cheng0646fff2008-09-09 06:11:26 +0000873 sys::Path RemoteClientPath;
874 if (!RemoteClient.empty())
Dan Gohman8e0b7b9d2009-08-05 20:21:17 +0000875 RemoteClientPath = sys::Program::FindProgramByName(RemoteClient);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000876
Chris Lattnerb1aa85b2009-08-23 22:45:37 +0000877 Message = "Found gcc: " + GCCPath.str() + "\n";
Bill Wendlingcd1389d2009-03-02 23:13:18 +0000878 return new GCC(GCCPath, RemoteClientPath, Args);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000879}