blob: f6fc37eb1fff885a62ea081daee50bed9c7adb33 [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"
Chris Lattnerf5377682009-08-24 03:52:50 +000021#include "llvm/ADT/StringExtras.h"
Chris Lattnerb1aa85b2009-08-23 22:45:37 +000022#include "llvm/Config/config.h" // for HAVE_LINK_R
Dan Gohmanf17a25c2007-07-18 16:29:46 +000023#include <fstream>
24#include <sstream>
Dan Gohmanf17a25c2007-07-18 16:29:46 +000025using namespace llvm;
26
Anton Korobeynikov43cf4f12009-08-05 09:32:10 +000027namespace llvm {
28 cl::opt<bool>
29 SaveTemps("save-temps", cl::init(false), cl::desc("Save temporary files"));
30}
31
Dan Gohmanf17a25c2007-07-18 16:29:46 +000032namespace {
33 cl::opt<std::string>
Evan Cheng0646fff2008-09-09 06:11:26 +000034 RemoteClient("remote-client",
35 cl::desc("Remote execution client (rsh/ssh)"));
Dan Gohmanf17a25c2007-07-18 16:29:46 +000036
37 cl::opt<std::string>
Evan Cheng0646fff2008-09-09 06:11:26 +000038 RemoteHost("remote-host",
39 cl::desc("Remote execution (rsh/ssh) host"));
40
41 cl::opt<std::string>
David Goodwinec1edba2009-07-10 21:39:28 +000042 RemotePort("remote-port",
43 cl::desc("Remote execution (rsh/ssh) port"));
44
45 cl::opt<std::string>
Evan Cheng0646fff2008-09-09 06:11:26 +000046 RemoteUser("remote-user",
47 cl::desc("Remote execution (rsh/ssh) user id"));
48
49 cl::opt<std::string>
50 RemoteExtra("remote-extra-options",
51 cl::desc("Remote execution (rsh/ssh) extra options"));
Dan Gohmanf17a25c2007-07-18 16:29:46 +000052}
53
54ToolExecutionError::~ToolExecutionError() throw() { }
55
Viktor Kutuzov2743a2e2009-07-18 18:39:24 +000056/// RunProgramWithTimeout - This function provides an alternate interface
57/// to the sys::Program::ExecuteAndWait interface.
Dan Gohmanf17a25c2007-07-18 16:29:46 +000058/// @see sys:Program::ExecuteAndWait
59static int RunProgramWithTimeout(const sys::Path &ProgramPath,
60 const char **Args,
61 const sys::Path &StdInFile,
62 const sys::Path &StdOutFile,
63 const sys::Path &StdErrFile,
64 unsigned NumSeconds = 0,
65 unsigned MemoryLimit = 0) {
66 const sys::Path* redirects[3];
67 redirects[0] = &StdInFile;
68 redirects[1] = &StdOutFile;
69 redirects[2] = &StdErrFile;
Viktor Kutuzov2743a2e2009-07-18 18:39:24 +000070
71#if 0 // For debug purposes
72 {
Dan Gohmanf8b81bf2009-07-15 16:35:29 +000073 errs() << "RUN:";
Dan Gohmanf17a25c2007-07-18 16:29:46 +000074 for (unsigned i = 0; Args[i]; ++i)
Dan Gohmanf8b81bf2009-07-15 16:35:29 +000075 errs() << " " << Args[i];
76 errs() << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +000077 }
Viktor Kutuzov2743a2e2009-07-18 18:39:24 +000078#endif
Dan Gohmanf17a25c2007-07-18 16:29:46 +000079
80 return
81 sys::Program::ExecuteAndWait(ProgramPath, Args, 0, redirects,
82 NumSeconds, MemoryLimit);
83}
84
Viktor Kutuzov2743a2e2009-07-18 18:39:24 +000085/// RunProgramRemotelyWithTimeout - This function runs the given program
86/// remotely using the given remote client and the sys::Program::ExecuteAndWait.
87/// Returns the remote program exit code or reports a remote client error if it
88/// fails. Remote client is required to return 255 if it failed or program exit
89/// code otherwise.
90/// @see sys:Program::ExecuteAndWait
91static int RunProgramRemotelyWithTimeout(const sys::Path &RemoteClientPath,
92 const char **Args,
93 const sys::Path &StdInFile,
94 const sys::Path &StdOutFile,
95 const sys::Path &StdErrFile,
96 unsigned NumSeconds = 0,
97 unsigned MemoryLimit = 0) {
98 const sys::Path* redirects[3];
99 redirects[0] = &StdInFile;
100 redirects[1] = &StdOutFile;
101 redirects[2] = &StdErrFile;
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000102
Viktor Kutuzov2743a2e2009-07-18 18:39:24 +0000103#if 0 // For debug purposes
104 {
105 errs() << "RUN:";
106 for (unsigned i = 0; Args[i]; ++i)
107 errs() << " " << Args[i];
108 errs() << "\n";
109 }
110#endif
111
112 // Run the program remotely with the remote client
113 int ReturnCode = sys::Program::ExecuteAndWait(RemoteClientPath, Args,
114 0, redirects, NumSeconds, MemoryLimit);
115
116 // Has the remote client fail?
117 if (255 == ReturnCode) {
118 std::ostringstream OS;
119 OS << "\nError running remote client:\n ";
120 for (const char **Arg = Args; *Arg; ++Arg)
121 OS << " " << *Arg;
122 OS << "\n";
123
124 // The error message is in the output file, let's print it out from there.
125 std::ifstream ErrorFile(StdOutFile.c_str());
126 if (ErrorFile) {
127 std::copy(std::istreambuf_iterator<char>(ErrorFile),
128 std::istreambuf_iterator<char>(),
129 std::ostreambuf_iterator<char>(OS));
130 ErrorFile.close();
131 }
132
133 throw ToolExecutionError(OS.str());
134 }
135
136 return ReturnCode;
137}
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000138
139static void ProcessFailure(sys::Path ProgPath, const char** Args) {
140 std::ostringstream OS;
141 OS << "\nError running tool:\n ";
142 for (const char **Arg = Args; *Arg; ++Arg)
143 OS << " " << *Arg;
144 OS << "\n";
145
146 // Rerun the compiler, capturing any error messages to print them.
Dan Gohman3c814b32009-02-12 20:53:27 +0000147 sys::Path ErrorFilename("bugpoint.program_error_messages");
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000148 std::string ErrMsg;
149 if (ErrorFilename.makeUnique(true, &ErrMsg)) {
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000150 errs() << "Error making unique filename: " << ErrMsg << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000151 exit(1);
152 }
153 RunProgramWithTimeout(ProgPath, Args, sys::Path(""), ErrorFilename,
154 ErrorFilename); // FIXME: check return code ?
155
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();
166 throw ToolExecutionError(OS.str());
167}
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,
187 const std::vector<std::string> &GCCArgs,
188 const std::vector<std::string> &SharedLibs =
189 std::vector<std::string>(),
190 unsigned Timeout = 0,
191 unsigned MemoryLimit = 0);
192 };
193}
194
195int LLI::ExecuteProgram(const std::string &Bitcode,
196 const std::vector<std::string> &Args,
197 const std::string &InputFile,
198 const std::string &OutputFile,
199 const std::vector<std::string> &GCCArgs,
200 const std::vector<std::string> &SharedLibs,
201 unsigned Timeout,
202 unsigned MemoryLimit) {
203 if (!SharedLibs.empty())
204 throw ToolExecutionError("LLI currently does not support "
205 "loading shared libraries.");
206
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000207 std::vector<const char*> LLIArgs;
208 LLIArgs.push_back(LLIPath.c_str());
209 LLIArgs.push_back("-force-interpreter=true");
210
211 // Add any extra LLI args.
212 for (unsigned i = 0, e = ToolArgs.size(); i != e; ++i)
213 LLIArgs.push_back(ToolArgs[i].c_str());
214
215 LLIArgs.push_back(Bitcode.c_str());
216 // Add optional parameters to the running program from Argv
217 for (unsigned i=0, e = Args.size(); i != e; ++i)
218 LLIArgs.push_back(Args[i].c_str());
219 LLIArgs.push_back(0);
220
Dan Gohmandd7ca8e2009-07-15 17:29:42 +0000221 outs() << "<lli>"; outs().flush();
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000222 DEBUG(errs() << "\nAbout to run:\t";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000223 for (unsigned i=0, e = LLIArgs.size()-1; i != e; ++i)
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000224 errs() << " " << LLIArgs[i];
225 errs() << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000226 );
227 return RunProgramWithTimeout(sys::Path(LLIPath), &LLIArgs[0],
228 sys::Path(InputFile), sys::Path(OutputFile), sys::Path(OutputFile),
229 Timeout, MemoryLimit);
230}
231
232// LLI create method - Try to find the LLI executable
Dan Gohman8e0b7b9d2009-08-05 20:21:17 +0000233AbstractInterpreter *AbstractInterpreter::createLLI(const char *Argv0,
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000234 std::string &Message,
235 const std::vector<std::string> *ToolArgs) {
Dan Gohman8e0b7b9d2009-08-05 20:21:17 +0000236 std::string LLIPath =
Chris Lattnerb1aa85b2009-08-23 22:45:37 +0000237 FindExecutable("lli", Argv0, (void *)(intptr_t)&createLLI).str();
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000238 if (!LLIPath.empty()) {
239 Message = "Found lli: " + LLIPath + "\n";
240 return new LLI(LLIPath, ToolArgs);
241 }
242
243 Message = "Cannot find `lli' in executable directory or PATH!\n";
244 return 0;
245}
246
Anton Korobeynikovfc0116f2008-04-28 20:53:48 +0000247//===---------------------------------------------------------------------===//
248// Custom execution command implementation of AbstractIntepreter interface
249//
250// Allows using a custom command for executing the bitcode, thus allows,
251// for example, to invoke a cross compiler for code generation followed by
252// a simulator that executes the generated binary.
253namespace {
254 class CustomExecutor : public AbstractInterpreter {
255 std::string ExecutionCommand;
256 std::vector<std::string> ExecutorArgs;
257 public:
258 CustomExecutor(
259 const std::string &ExecutionCmd, std::vector<std::string> ExecArgs) :
260 ExecutionCommand(ExecutionCmd), ExecutorArgs(ExecArgs) {}
261
262 virtual int ExecuteProgram(const std::string &Bitcode,
263 const std::vector<std::string> &Args,
264 const std::string &InputFile,
265 const std::string &OutputFile,
266 const std::vector<std::string> &GCCArgs,
267 const std::vector<std::string> &SharedLibs =
268 std::vector<std::string>(),
269 unsigned Timeout = 0,
270 unsigned MemoryLimit = 0);
271 };
272}
273
274int CustomExecutor::ExecuteProgram(const std::string &Bitcode,
275 const std::vector<std::string> &Args,
276 const std::string &InputFile,
277 const std::string &OutputFile,
278 const std::vector<std::string> &GCCArgs,
279 const std::vector<std::string> &SharedLibs,
280 unsigned Timeout,
281 unsigned MemoryLimit) {
282
283 std::vector<const char*> ProgramArgs;
284 ProgramArgs.push_back(ExecutionCommand.c_str());
285
286 for (std::size_t i = 0; i < ExecutorArgs.size(); ++i)
287 ProgramArgs.push_back(ExecutorArgs.at(i).c_str());
288 ProgramArgs.push_back(Bitcode.c_str());
289 ProgramArgs.push_back(0);
290
291 // Add optional parameters to the running program from Argv
292 for (unsigned i=0, e = Args.size(); i != e; ++i)
293 ProgramArgs.push_back(Args[i].c_str());
294
295 return RunProgramWithTimeout(
296 sys::Path(ExecutionCommand),
297 &ProgramArgs[0], sys::Path(InputFile), sys::Path(OutputFile),
298 sys::Path(OutputFile), Timeout, MemoryLimit);
299}
300
301// Custom execution environment create method, takes the execution command
302// as arguments
303AbstractInterpreter *AbstractInterpreter::createCustom(
Anton Korobeynikovfc0116f2008-04-28 20:53:48 +0000304 std::string &Message,
305 const std::string &ExecCommandLine) {
306
307 std::string Command = "";
308 std::vector<std::string> Args;
309 std::string delimiters = " ";
310
311 // Tokenize the ExecCommandLine to the command and the args to allow
312 // defining a full command line as the command instead of just the
313 // executed program. We cannot just pass the whole string after the command
314 // as a single argument because then program sees only a single
315 // command line argument (with spaces in it: "foo bar" instead
316 // of "foo" and "bar").
317
318 // code borrowed from:
319 // http://oopweb.com/CPP/Documents/CPPHOWTO/Volume/C++Programming-HOWTO-7.html
320 std::string::size_type lastPos =
321 ExecCommandLine.find_first_not_of(delimiters, 0);
322 std::string::size_type pos =
323 ExecCommandLine.find_first_of(delimiters, lastPos);
324
325 while (std::string::npos != pos || std::string::npos != lastPos) {
326 std::string token = ExecCommandLine.substr(lastPos, pos - lastPos);
327 if (Command == "")
328 Command = token;
329 else
330 Args.push_back(token);
331 // Skip delimiters. Note the "not_of"
332 lastPos = ExecCommandLine.find_first_not_of(delimiters, pos);
333 // Find next "non-delimiter"
334 pos = ExecCommandLine.find_first_of(delimiters, lastPos);
335 }
336
Chris Lattnerb1aa85b2009-08-23 22:45:37 +0000337 std::string CmdPath = sys::Program::FindProgramByName(Command).str();
Anton Korobeynikovfc0116f2008-04-28 20:53:48 +0000338 if (CmdPath.empty()) {
339 Message =
340 std::string("Cannot find '") + Command +
341 "' in executable directory or PATH!\n";
342 return 0;
343 }
344
345 Message = "Found command in: " + CmdPath + "\n";
346
347 return new CustomExecutor(CmdPath, Args);
348}
349
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000350//===----------------------------------------------------------------------===//
351// LLC Implementation of AbstractIntepreter interface
352//
353GCC::FileType LLC::OutputCode(const std::string &Bitcode,
354 sys::Path &OutputAsmFile) {
355 sys::Path uniqueFile(Bitcode+".llc.s");
356 std::string ErrMsg;
357 if (uniqueFile.makeUnique(true, &ErrMsg)) {
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000358 errs() << "Error making unique filename: " << ErrMsg << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000359 exit(1);
360 }
361 OutputAsmFile = uniqueFile;
362 std::vector<const char *> LLCArgs;
363 LLCArgs.push_back (LLCPath.c_str());
364
365 // Add any extra LLC args.
366 for (unsigned i = 0, e = ToolArgs.size(); i != e; ++i)
367 LLCArgs.push_back(ToolArgs[i].c_str());
368
369 LLCArgs.push_back ("-o");
370 LLCArgs.push_back (OutputAsmFile.c_str()); // Output to the Asm file
371 LLCArgs.push_back ("-f"); // Overwrite as necessary...
372 LLCArgs.push_back (Bitcode.c_str()); // This is the input bitcode
373 LLCArgs.push_back (0);
374
Dan Gohmandd7ca8e2009-07-15 17:29:42 +0000375 outs() << "<llc>"; outs().flush();
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000376 DEBUG(errs() << "\nAbout to run:\t";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000377 for (unsigned i=0, e = LLCArgs.size()-1; i != e; ++i)
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000378 errs() << " " << LLCArgs[i];
379 errs() << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000380 );
381 if (RunProgramWithTimeout(sys::Path(LLCPath), &LLCArgs[0],
382 sys::Path(), sys::Path(), sys::Path()))
383 ProcessFailure(sys::Path(LLCPath), &LLCArgs[0]);
384
385 return GCC::AsmFile;
386}
387
388void LLC::compileProgram(const std::string &Bitcode) {
389 sys::Path OutputAsmFile;
390 OutputCode(Bitcode, OutputAsmFile);
391 OutputAsmFile.eraseFromDisk();
392}
393
394int LLC::ExecuteProgram(const std::string &Bitcode,
395 const std::vector<std::string> &Args,
396 const std::string &InputFile,
397 const std::string &OutputFile,
398 const std::vector<std::string> &ArgsForGCC,
399 const std::vector<std::string> &SharedLibs,
400 unsigned Timeout,
401 unsigned MemoryLimit) {
402
403 sys::Path OutputAsmFile;
404 OutputCode(Bitcode, OutputAsmFile);
Anton Korobeynikov43cf4f12009-08-05 09:32:10 +0000405 FileRemover OutFileRemover(OutputAsmFile, !SaveTemps);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000406
407 std::vector<std::string> GCCArgs(ArgsForGCC);
Bill Wendlingcd1389d2009-03-02 23:13:18 +0000408 GCCArgs.insert(GCCArgs.end(), SharedLibs.begin(), SharedLibs.end());
409 GCCArgs.insert(GCCArgs.end(), gccArgs.begin(), gccArgs.end());
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000410
411 // Assuming LLC worked, compile the result with GCC and run it.
Chris Lattnerb1aa85b2009-08-23 22:45:37 +0000412 return gcc->ExecuteProgram(OutputAsmFile.str(), Args, GCC::AsmFile,
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000413 InputFile, OutputFile, GCCArgs,
414 Timeout, MemoryLimit);
415}
416
417/// createLLC - Try to find the LLC executable
418///
Dan Gohman8e0b7b9d2009-08-05 20:21:17 +0000419LLC *AbstractInterpreter::createLLC(const char *Argv0,
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000420 std::string &Message,
Bill Wendlingcd1389d2009-03-02 23:13:18 +0000421 const std::vector<std::string> *Args,
422 const std::vector<std::string> *GCCArgs) {
Dan Gohman8e0b7b9d2009-08-05 20:21:17 +0000423 std::string LLCPath =
Chris Lattnerb1aa85b2009-08-23 22:45:37 +0000424 FindExecutable("llc", Argv0, (void *)(intptr_t)&createLLC).str();
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000425 if (LLCPath.empty()) {
426 Message = "Cannot find `llc' in executable directory or PATH!\n";
427 return 0;
428 }
429
430 Message = "Found llc: " + LLCPath + "\n";
Dan Gohman8e0b7b9d2009-08-05 20:21:17 +0000431 GCC *gcc = GCC::create(Message, GCCArgs);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000432 if (!gcc) {
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000433 errs() << Message << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000434 exit(1);
435 }
Bill Wendlingcd1389d2009-03-02 23:13:18 +0000436 return new LLC(LLCPath, gcc, Args, GCCArgs);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000437}
438
439//===---------------------------------------------------------------------===//
440// JIT Implementation of AbstractIntepreter interface
441//
442namespace {
443 class JIT : public AbstractInterpreter {
444 std::string LLIPath; // The path to the LLI executable
445 std::vector<std::string> ToolArgs; // Args to pass to LLI
446 public:
447 JIT(const std::string &Path, const std::vector<std::string> *Args)
448 : LLIPath(Path) {
449 ToolArgs.clear ();
450 if (Args) { ToolArgs = *Args; }
451 }
452
453 virtual int ExecuteProgram(const std::string &Bitcode,
454 const std::vector<std::string> &Args,
455 const std::string &InputFile,
456 const std::string &OutputFile,
457 const std::vector<std::string> &GCCArgs =
458 std::vector<std::string>(),
459 const std::vector<std::string> &SharedLibs =
460 std::vector<std::string>(),
461 unsigned Timeout =0,
462 unsigned MemoryLimit =0);
463 };
464}
465
466int JIT::ExecuteProgram(const std::string &Bitcode,
467 const std::vector<std::string> &Args,
468 const std::string &InputFile,
469 const std::string &OutputFile,
470 const std::vector<std::string> &GCCArgs,
471 const std::vector<std::string> &SharedLibs,
472 unsigned Timeout,
473 unsigned MemoryLimit) {
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000474 // Construct a vector of parameters, incorporating those from the command-line
475 std::vector<const char*> JITArgs;
476 JITArgs.push_back(LLIPath.c_str());
477 JITArgs.push_back("-force-interpreter=false");
478
479 // Add any extra LLI args.
480 for (unsigned i = 0, e = ToolArgs.size(); i != e; ++i)
481 JITArgs.push_back(ToolArgs[i].c_str());
482
483 for (unsigned i = 0, e = SharedLibs.size(); i != e; ++i) {
484 JITArgs.push_back("-load");
485 JITArgs.push_back(SharedLibs[i].c_str());
486 }
487 JITArgs.push_back(Bitcode.c_str());
488 // Add optional parameters to the running program from Argv
489 for (unsigned i=0, e = Args.size(); i != e; ++i)
490 JITArgs.push_back(Args[i].c_str());
491 JITArgs.push_back(0);
492
Dan Gohmandd7ca8e2009-07-15 17:29:42 +0000493 outs() << "<jit>"; outs().flush();
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000494 DEBUG(errs() << "\nAbout to run:\t";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000495 for (unsigned i=0, e = JITArgs.size()-1; i != e; ++i)
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000496 errs() << " " << JITArgs[i];
497 errs() << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000498 );
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000499 DEBUG(errs() << "\nSending output to " << OutputFile << "\n");
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000500 return RunProgramWithTimeout(sys::Path(LLIPath), &JITArgs[0],
501 sys::Path(InputFile), sys::Path(OutputFile), sys::Path(OutputFile),
502 Timeout, MemoryLimit);
503}
504
505/// createJIT - Try to find the LLI executable
506///
Dan Gohman8e0b7b9d2009-08-05 20:21:17 +0000507AbstractInterpreter *AbstractInterpreter::createJIT(const char *Argv0,
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000508 std::string &Message, const std::vector<std::string> *Args) {
Dan Gohman8e0b7b9d2009-08-05 20:21:17 +0000509 std::string LLIPath =
Chris Lattnerb1aa85b2009-08-23 22:45:37 +0000510 FindExecutable("lli", Argv0, (void *)(intptr_t)&createJIT).str();
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000511 if (!LLIPath.empty()) {
512 Message = "Found lli: " + LLIPath + "\n";
513 return new JIT(LLIPath, Args);
514 }
515
516 Message = "Cannot find `lli' in executable directory or PATH!\n";
517 return 0;
518}
519
520GCC::FileType CBE::OutputCode(const std::string &Bitcode,
521 sys::Path &OutputCFile) {
522 sys::Path uniqueFile(Bitcode+".cbe.c");
523 std::string ErrMsg;
524 if (uniqueFile.makeUnique(true, &ErrMsg)) {
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000525 errs() << "Error making unique filename: " << ErrMsg << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000526 exit(1);
527 }
528 OutputCFile = uniqueFile;
529 std::vector<const char *> LLCArgs;
530 LLCArgs.push_back (LLCPath.c_str());
531
532 // Add any extra LLC args.
533 for (unsigned i = 0, e = ToolArgs.size(); i != e; ++i)
534 LLCArgs.push_back(ToolArgs[i].c_str());
535
536 LLCArgs.push_back ("-o");
537 LLCArgs.push_back (OutputCFile.c_str()); // Output to the C file
538 LLCArgs.push_back ("-march=c"); // Output C language
539 LLCArgs.push_back ("-f"); // Overwrite as necessary...
540 LLCArgs.push_back (Bitcode.c_str()); // This is the input bitcode
541 LLCArgs.push_back (0);
542
Dan Gohmandd7ca8e2009-07-15 17:29:42 +0000543 outs() << "<cbe>"; outs().flush();
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000544 DEBUG(errs() << "\nAbout to run:\t";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000545 for (unsigned i=0, e = LLCArgs.size()-1; i != e; ++i)
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000546 errs() << " " << LLCArgs[i];
547 errs() << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000548 );
549 if (RunProgramWithTimeout(LLCPath, &LLCArgs[0], sys::Path(), sys::Path(),
550 sys::Path()))
551 ProcessFailure(LLCPath, &LLCArgs[0]);
552 return GCC::CFile;
553}
554
555void CBE::compileProgram(const std::string &Bitcode) {
556 sys::Path OutputCFile;
557 OutputCode(Bitcode, OutputCFile);
558 OutputCFile.eraseFromDisk();
559}
560
561int CBE::ExecuteProgram(const std::string &Bitcode,
562 const std::vector<std::string> &Args,
563 const std::string &InputFile,
564 const std::string &OutputFile,
565 const std::vector<std::string> &ArgsForGCC,
566 const std::vector<std::string> &SharedLibs,
567 unsigned Timeout,
568 unsigned MemoryLimit) {
569 sys::Path OutputCFile;
570 OutputCode(Bitcode, OutputCFile);
571
Anton Korobeynikov43cf4f12009-08-05 09:32:10 +0000572 FileRemover CFileRemove(OutputCFile, !SaveTemps);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000573
574 std::vector<std::string> GCCArgs(ArgsForGCC);
Bill Wendlingcd1389d2009-03-02 23:13:18 +0000575 GCCArgs.insert(GCCArgs.end(), SharedLibs.begin(), SharedLibs.end());
576
Chris Lattnerb1aa85b2009-08-23 22:45:37 +0000577 return gcc->ExecuteProgram(OutputCFile.str(), Args, GCC::CFile,
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000578 InputFile, OutputFile, GCCArgs,
579 Timeout, MemoryLimit);
580}
581
582/// createCBE - Try to find the 'llc' executable
583///
Dan Gohman8e0b7b9d2009-08-05 20:21:17 +0000584CBE *AbstractInterpreter::createCBE(const char *Argv0,
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000585 std::string &Message,
Bill Wendlingcd1389d2009-03-02 23:13:18 +0000586 const std::vector<std::string> *Args,
587 const std::vector<std::string> *GCCArgs) {
Dan Gohman8e0b7b9d2009-08-05 20:21:17 +0000588 sys::Path LLCPath =
Dan Gohman108e0ed2009-08-05 21:03:39 +0000589 FindExecutable("llc", Argv0, (void *)(intptr_t)&createCBE);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000590 if (LLCPath.isEmpty()) {
591 Message =
592 "Cannot find `llc' in executable directory or PATH!\n";
593 return 0;
594 }
595
Chris Lattnerb1aa85b2009-08-23 22:45:37 +0000596 Message = "Found llc: " + LLCPath.str() + "\n";
Dan Gohman8e0b7b9d2009-08-05 20:21:17 +0000597 GCC *gcc = GCC::create(Message, GCCArgs);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000598 if (!gcc) {
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000599 errs() << Message << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000600 exit(1);
601 }
602 return new CBE(LLCPath, gcc, Args);
603}
604
605//===---------------------------------------------------------------------===//
606// GCC abstraction
607//
David Goodwinec1edba2009-07-10 21:39:28 +0000608
609static bool
610IsARMArchitecture(std::vector<std::string> Args)
611{
612 for (std::vector<std::string>::const_iterator
613 I = Args.begin(), E = Args.end(); I != E; ++I) {
Benjamin Kramere97b56a2009-08-19 12:38:51 +0000614 if (!StringsEqualNoCase(*I, "-arch")) {
David Goodwinec1edba2009-07-10 21:39:28 +0000615 ++I;
Benjamin Kramere97b56a2009-08-19 12:38:51 +0000616 if ((I != E) && !StringsEqualNoCase(I->c_str(), "arm", strlen("arm"))) {
David Goodwinec1edba2009-07-10 21:39:28 +0000617 return true;
618 }
619 }
620 }
621
622 return false;
623}
624
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000625int GCC::ExecuteProgram(const std::string &ProgramFile,
626 const std::vector<std::string> &Args,
627 FileType fileType,
628 const std::string &InputFile,
629 const std::string &OutputFile,
630 const std::vector<std::string> &ArgsForGCC,
631 unsigned Timeout,
632 unsigned MemoryLimit) {
633 std::vector<const char*> GCCArgs;
634
635 GCCArgs.push_back(GCCPath.c_str());
636
Bill Wendlingcd1389d2009-03-02 23:13:18 +0000637 for (std::vector<std::string>::const_iterator
638 I = gccArgs.begin(), E = gccArgs.end(); I != E; ++I)
639 GCCArgs.push_back(I->c_str());
640
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000641 // Specify -x explicitly in case the extension is wonky
642 GCCArgs.push_back("-x");
643 if (fileType == CFile) {
644 GCCArgs.push_back("c");
645 GCCArgs.push_back("-fno-strict-aliasing");
646 } else {
647 GCCArgs.push_back("assembler");
Daniel Dunbard8590af2009-08-18 03:35:57 +0000648
David Goodwinec1edba2009-07-10 21:39:28 +0000649 // For ARM architectures we don't want this flag. bugpoint isn't
650 // explicitly told what architecture it is working on, so we get
651 // it from gcc flags
Daniel Dunbard8590af2009-08-18 03:35:57 +0000652 if ((TargetTriple.getOS() == Triple::Darwin) &&
653 !IsARMArchitecture(ArgsForGCC))
David Goodwinec1edba2009-07-10 21:39:28 +0000654 GCCArgs.push_back("-force_cpusubtype_ALL");
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000655 }
656 GCCArgs.push_back(ProgramFile.c_str()); // Specify the input filename...
657 GCCArgs.push_back("-x");
658 GCCArgs.push_back("none");
659 GCCArgs.push_back("-o");
660 sys::Path OutputBinary (ProgramFile+".gcc.exe");
661 std::string ErrMsg;
662 if (OutputBinary.makeUnique(true, &ErrMsg)) {
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000663 errs() << "Error making unique filename: " << ErrMsg << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000664 exit(1);
665 }
666 GCCArgs.push_back(OutputBinary.c_str()); // Output to the right file...
667
668 // Add any arguments intended for GCC. We locate them here because this is
669 // most likely -L and -l options that need to come before other libraries but
670 // after the source. Other options won't be sensitive to placement on the
671 // command line, so this should be safe.
672 for (unsigned i = 0, e = ArgsForGCC.size(); i != e; ++i)
673 GCCArgs.push_back(ArgsForGCC[i].c_str());
674
675 GCCArgs.push_back("-lm"); // Hard-code the math library...
676 GCCArgs.push_back("-O2"); // Optimize the program a bit...
677#if defined (HAVE_LINK_R)
678 GCCArgs.push_back("-Wl,-R."); // Search this dir for .so files
679#endif
Daniel Dunbard8590af2009-08-18 03:35:57 +0000680 if (TargetTriple.getArch() == Triple::sparc)
681 GCCArgs.push_back("-mcpu=v9");
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000682 GCCArgs.push_back(0); // NULL terminator
683
Dan Gohmandd7ca8e2009-07-15 17:29:42 +0000684 outs() << "<gcc>"; outs().flush();
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000685 DEBUG(errs() << "\nAbout to run:\t";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000686 for (unsigned i=0, e = GCCArgs.size()-1; i != e; ++i)
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000687 errs() << " " << GCCArgs[i];
688 errs() << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000689 );
690 if (RunProgramWithTimeout(GCCPath, &GCCArgs[0], sys::Path(), sys::Path(),
691 sys::Path())) {
692 ProcessFailure(GCCPath, &GCCArgs[0]);
693 exit(1);
694 }
695
696 std::vector<const char*> ProgramArgs;
697
Evan Cheng0646fff2008-09-09 06:11:26 +0000698 if (RemoteClientPath.isEmpty())
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000699 ProgramArgs.push_back(OutputBinary.c_str());
700 else {
Evan Cheng0646fff2008-09-09 06:11:26 +0000701 ProgramArgs.push_back(RemoteClientPath.c_str());
702 ProgramArgs.push_back(RemoteHost.c_str());
Anton Korobeynikov33f97a62009-08-05 09:32:53 +0000703 if (!RemoteUser.empty()) {
704 ProgramArgs.push_back("-l");
705 ProgramArgs.push_back(RemoteUser.c_str());
706 }
David Goodwinec1edba2009-07-10 21:39:28 +0000707 if (!RemotePort.empty()) {
708 ProgramArgs.push_back("-p");
709 ProgramArgs.push_back(RemotePort.c_str());
710 }
Evan Cheng0646fff2008-09-09 06:11:26 +0000711 if (!RemoteExtra.empty()) {
712 ProgramArgs.push_back(RemoteExtra.c_str());
713 }
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000714
David Goodwinfbb0faf2009-07-20 17:15:03 +0000715 // Full path to the binary. We need to cd to the exec directory because
716 // there is a dylib there that the exec expects to find in the CWD
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000717 char* env_pwd = getenv("PWD");
David Goodwinfbb0faf2009-07-20 17:15:03 +0000718 std::string Exec = "cd ";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000719 Exec += env_pwd;
David Goodwinfbb0faf2009-07-20 17:15:03 +0000720 Exec += "; ./";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000721 Exec += OutputBinary.c_str();
722 ProgramArgs.push_back(Exec.c_str());
723 }
724
725 // Add optional parameters to the running program from Argv
726 for (unsigned i=0, e = Args.size(); i != e; ++i)
727 ProgramArgs.push_back(Args[i].c_str());
728 ProgramArgs.push_back(0); // NULL terminator
729
730 // Now that we have a binary, run it!
Dan Gohmandd7ca8e2009-07-15 17:29:42 +0000731 outs() << "<program>"; outs().flush();
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000732 DEBUG(errs() << "\nAbout to run:\t";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000733 for (unsigned i=0, e = ProgramArgs.size()-1; i != e; ++i)
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000734 errs() << " " << ProgramArgs[i];
735 errs() << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000736 );
737
Anton Korobeynikov43cf4f12009-08-05 09:32:10 +0000738 FileRemover OutputBinaryRemover(OutputBinary, !SaveTemps);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000739
Viktor Kutuzov8c3e63c2009-07-14 19:10:55 +0000740 if (RemoteClientPath.isEmpty()) {
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000741 DEBUG(errs() << "<run locally>";);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000742 return RunProgramWithTimeout(OutputBinary, &ProgramArgs[0],
743 sys::Path(InputFile), sys::Path(OutputFile), sys::Path(OutputFile),
744 Timeout, MemoryLimit);
Viktor Kutuzov8c3e63c2009-07-14 19:10:55 +0000745 } else {
Dan Gohmandd7ca8e2009-07-15 17:29:42 +0000746 outs() << "<run remotely>"; outs().flush();
Viktor Kutuzov2743a2e2009-07-18 18:39:24 +0000747 return RunProgramRemotelyWithTimeout(sys::Path(RemoteClientPath),
Viktor Kutuzov8c3e63c2009-07-14 19:10:55 +0000748 &ProgramArgs[0], sys::Path(InputFile), sys::Path(OutputFile),
749 sys::Path(OutputFile), Timeout, MemoryLimit);
Viktor Kutuzov8c3e63c2009-07-14 19:10:55 +0000750 }
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000751}
752
753int GCC::MakeSharedObject(const std::string &InputFile, FileType fileType,
754 std::string &OutputFile,
755 const std::vector<std::string> &ArgsForGCC) {
756 sys::Path uniqueFilename(InputFile+LTDL_SHLIB_EXT);
757 std::string ErrMsg;
758 if (uniqueFilename.makeUnique(true, &ErrMsg)) {
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000759 errs() << "Error making unique filename: " << ErrMsg << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000760 exit(1);
761 }
Chris Lattnerb1aa85b2009-08-23 22:45:37 +0000762 OutputFile = uniqueFilename.str();
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000763
764 std::vector<const char*> GCCArgs;
765
766 GCCArgs.push_back(GCCPath.c_str());
Evan Chengf94bcf62009-03-12 00:53:34 +0000767
768 for (std::vector<std::string>::const_iterator
769 I = gccArgs.begin(), E = gccArgs.end(); I != E; ++I)
770 GCCArgs.push_back(I->c_str());
771
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000772 // Compile the C/asm file into a shared object
773 GCCArgs.push_back("-x");
774 GCCArgs.push_back(fileType == AsmFile ? "assembler" : "c");
775 GCCArgs.push_back("-fno-strict-aliasing");
776 GCCArgs.push_back(InputFile.c_str()); // Specify the input filename.
777 GCCArgs.push_back("-x");
778 GCCArgs.push_back("none");
Daniel Dunbard8590af2009-08-18 03:35:57 +0000779 if (TargetTriple.getArch() == Triple::sparc)
780 GCCArgs.push_back("-G"); // Compile a shared library, `-G' for Sparc
781 else if (TargetTriple.getOS() == Triple::Darwin) {
782 // link all source files into a single module in data segment, rather than
783 // generating blocks. dynamic_lookup requires that you set
784 // MACOSX_DEPLOYMENT_TARGET=10.3 in your env. FIXME: it would be better for
785 // bugpoint to just pass that in the environment of GCC.
786 GCCArgs.push_back("-single_module");
787 GCCArgs.push_back("-dynamiclib"); // `-dynamiclib' for MacOS X/PowerPC
788 GCCArgs.push_back("-undefined");
789 GCCArgs.push_back("dynamic_lookup");
790 } else
791 GCCArgs.push_back("-shared"); // `-shared' for Linux/X86, maybe others
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000792
Daniel Dunbard8590af2009-08-18 03:35:57 +0000793 if ((TargetTriple.getArch() == Triple::alpha) ||
794 (TargetTriple.getArch() == Triple::x86_64))
795 GCCArgs.push_back("-fPIC"); // Requires shared objs to contain PIC
796
797 if (TargetTriple.getArch() == Triple::sparc)
798 GCCArgs.push_back("-mcpu=v9");
799
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000800 GCCArgs.push_back("-o");
801 GCCArgs.push_back(OutputFile.c_str()); // Output to the right filename.
802 GCCArgs.push_back("-O2"); // Optimize the program a bit.
803
804
805
806 // Add any arguments intended for GCC. We locate them here because this is
807 // most likely -L and -l options that need to come before other libraries but
808 // after the source. Other options won't be sensitive to placement on the
809 // command line, so this should be safe.
810 for (unsigned i = 0, e = ArgsForGCC.size(); i != e; ++i)
811 GCCArgs.push_back(ArgsForGCC[i].c_str());
812 GCCArgs.push_back(0); // NULL terminator
813
814
815
Dan Gohmandd7ca8e2009-07-15 17:29:42 +0000816 outs() << "<gcc>"; outs().flush();
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000817 DEBUG(errs() << "\nAbout to run:\t";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000818 for (unsigned i=0, e = GCCArgs.size()-1; i != e; ++i)
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000819 errs() << " " << GCCArgs[i];
820 errs() << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000821 );
822 if (RunProgramWithTimeout(GCCPath, &GCCArgs[0], sys::Path(), sys::Path(),
823 sys::Path())) {
824 ProcessFailure(GCCPath, &GCCArgs[0]);
825 return 1;
826 }
827 return 0;
828}
829
830/// create - Try to find the `gcc' executable
831///
Dan Gohman8e0b7b9d2009-08-05 20:21:17 +0000832GCC *GCC::create(std::string &Message,
Bill Wendlingcd1389d2009-03-02 23:13:18 +0000833 const std::vector<std::string> *Args) {
Dan Gohman8e0b7b9d2009-08-05 20:21:17 +0000834 sys::Path GCCPath = sys::Program::FindProgramByName("gcc");
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000835 if (GCCPath.isEmpty()) {
836 Message = "Cannot find `gcc' in executable directory or PATH!\n";
837 return 0;
838 }
839
Evan Cheng0646fff2008-09-09 06:11:26 +0000840 sys::Path RemoteClientPath;
841 if (!RemoteClient.empty())
Dan Gohman8e0b7b9d2009-08-05 20:21:17 +0000842 RemoteClientPath = sys::Program::FindProgramByName(RemoteClient);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000843
Chris Lattnerb1aa85b2009-08-23 22:45:37 +0000844 Message = "Found gcc: " + GCCPath.str() + "\n";
Bill Wendlingcd1389d2009-03-02 23:13:18 +0000845 return new GCC(GCCPath, RemoteClientPath, Args);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000846}