blob: dfb65cb0e724cf2004649895766d8f3f7d8f4101 [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"
16#include "llvm/Config/config.h" // for HAVE_LINK_R
17#include "llvm/System/Program.h"
18#include "llvm/Support/CommandLine.h"
19#include "llvm/Support/Debug.h"
20#include "llvm/Support/FileUtilities.h"
21#include <fstream>
22#include <sstream>
Dan Gohmanf17a25c2007-07-18 16:29:46 +000023using namespace llvm;
24
Anton Korobeynikov43cf4f12009-08-05 09:32:10 +000025namespace llvm {
26 cl::opt<bool>
27 SaveTemps("save-temps", cl::init(false), cl::desc("Save temporary files"));
28}
29
Dan Gohmanf17a25c2007-07-18 16:29:46 +000030namespace {
31 cl::opt<std::string>
Evan Cheng0646fff2008-09-09 06:11:26 +000032 RemoteClient("remote-client",
33 cl::desc("Remote execution client (rsh/ssh)"));
Dan Gohmanf17a25c2007-07-18 16:29:46 +000034
35 cl::opt<std::string>
Evan Cheng0646fff2008-09-09 06:11:26 +000036 RemoteHost("remote-host",
37 cl::desc("Remote execution (rsh/ssh) host"));
38
39 cl::opt<std::string>
David Goodwinec1edba2009-07-10 21:39:28 +000040 RemotePort("remote-port",
41 cl::desc("Remote execution (rsh/ssh) port"));
42
43 cl::opt<std::string>
Evan Cheng0646fff2008-09-09 06:11:26 +000044 RemoteUser("remote-user",
45 cl::desc("Remote execution (rsh/ssh) user id"));
46
47 cl::opt<std::string>
48 RemoteExtra("remote-extra-options",
49 cl::desc("Remote execution (rsh/ssh) extra options"));
Dan Gohmanf17a25c2007-07-18 16:29:46 +000050}
51
52ToolExecutionError::~ToolExecutionError() throw() { }
53
Viktor Kutuzov2743a2e2009-07-18 18:39:24 +000054/// RunProgramWithTimeout - This function provides an alternate interface
55/// to the sys::Program::ExecuteAndWait interface.
Dan Gohmanf17a25c2007-07-18 16:29:46 +000056/// @see sys:Program::ExecuteAndWait
57static int RunProgramWithTimeout(const sys::Path &ProgramPath,
58 const char **Args,
59 const sys::Path &StdInFile,
60 const sys::Path &StdOutFile,
61 const sys::Path &StdErrFile,
62 unsigned NumSeconds = 0,
63 unsigned MemoryLimit = 0) {
64 const sys::Path* redirects[3];
65 redirects[0] = &StdInFile;
66 redirects[1] = &StdOutFile;
67 redirects[2] = &StdErrFile;
Viktor Kutuzov2743a2e2009-07-18 18:39:24 +000068
69#if 0 // For debug purposes
70 {
Dan Gohmanf8b81bf2009-07-15 16:35:29 +000071 errs() << "RUN:";
Dan Gohmanf17a25c2007-07-18 16:29:46 +000072 for (unsigned i = 0; Args[i]; ++i)
Dan Gohmanf8b81bf2009-07-15 16:35:29 +000073 errs() << " " << Args[i];
74 errs() << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +000075 }
Viktor Kutuzov2743a2e2009-07-18 18:39:24 +000076#endif
Dan Gohmanf17a25c2007-07-18 16:29:46 +000077
78 return
79 sys::Program::ExecuteAndWait(ProgramPath, Args, 0, redirects,
80 NumSeconds, MemoryLimit);
81}
82
Viktor Kutuzov2743a2e2009-07-18 18:39:24 +000083/// RunProgramRemotelyWithTimeout - This function runs the given program
84/// remotely using the given remote client and the sys::Program::ExecuteAndWait.
85/// Returns the remote program exit code or reports a remote client error if it
86/// fails. Remote client is required to return 255 if it failed or program exit
87/// code otherwise.
88/// @see sys:Program::ExecuteAndWait
89static int RunProgramRemotelyWithTimeout(const sys::Path &RemoteClientPath,
90 const char **Args,
91 const sys::Path &StdInFile,
92 const sys::Path &StdOutFile,
93 const sys::Path &StdErrFile,
94 unsigned NumSeconds = 0,
95 unsigned MemoryLimit = 0) {
96 const sys::Path* redirects[3];
97 redirects[0] = &StdInFile;
98 redirects[1] = &StdOutFile;
99 redirects[2] = &StdErrFile;
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000100
Viktor Kutuzov2743a2e2009-07-18 18:39:24 +0000101#if 0 // For debug purposes
102 {
103 errs() << "RUN:";
104 for (unsigned i = 0; Args[i]; ++i)
105 errs() << " " << Args[i];
106 errs() << "\n";
107 }
108#endif
109
110 // Run the program remotely with the remote client
111 int ReturnCode = sys::Program::ExecuteAndWait(RemoteClientPath, Args,
112 0, redirects, NumSeconds, MemoryLimit);
113
114 // Has the remote client fail?
115 if (255 == ReturnCode) {
116 std::ostringstream OS;
117 OS << "\nError running remote client:\n ";
118 for (const char **Arg = Args; *Arg; ++Arg)
119 OS << " " << *Arg;
120 OS << "\n";
121
122 // The error message is in the output file, let's print it out from there.
123 std::ifstream ErrorFile(StdOutFile.c_str());
124 if (ErrorFile) {
125 std::copy(std::istreambuf_iterator<char>(ErrorFile),
126 std::istreambuf_iterator<char>(),
127 std::ostreambuf_iterator<char>(OS));
128 ErrorFile.close();
129 }
130
131 throw ToolExecutionError(OS.str());
132 }
133
134 return ReturnCode;
135}
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000136
137static void ProcessFailure(sys::Path ProgPath, const char** Args) {
138 std::ostringstream OS;
139 OS << "\nError running tool:\n ";
140 for (const char **Arg = Args; *Arg; ++Arg)
141 OS << " " << *Arg;
142 OS << "\n";
143
144 // Rerun the compiler, capturing any error messages to print them.
Dan Gohman3c814b32009-02-12 20:53:27 +0000145 sys::Path ErrorFilename("bugpoint.program_error_messages");
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000146 std::string ErrMsg;
147 if (ErrorFilename.makeUnique(true, &ErrMsg)) {
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000148 errs() << "Error making unique filename: " << ErrMsg << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000149 exit(1);
150 }
151 RunProgramWithTimeout(ProgPath, Args, sys::Path(""), ErrorFilename,
152 ErrorFilename); // FIXME: check return code ?
153
154 // Print out the error messages generated by GCC if possible...
155 std::ifstream ErrorFile(ErrorFilename.c_str());
156 if (ErrorFile) {
157 std::copy(std::istreambuf_iterator<char>(ErrorFile),
158 std::istreambuf_iterator<char>(),
159 std::ostreambuf_iterator<char>(OS));
160 ErrorFile.close();
161 }
162
163 ErrorFilename.eraseFromDisk();
164 throw ToolExecutionError(OS.str());
165}
166
167//===---------------------------------------------------------------------===//
168// LLI Implementation of AbstractIntepreter interface
169//
170namespace {
171 class LLI : public AbstractInterpreter {
172 std::string LLIPath; // The path to the LLI executable
173 std::vector<std::string> ToolArgs; // Args to pass to LLI
174 public:
175 LLI(const std::string &Path, const std::vector<std::string> *Args)
176 : LLIPath(Path) {
177 ToolArgs.clear ();
178 if (Args) { ToolArgs = *Args; }
179 }
180
181 virtual int ExecuteProgram(const std::string &Bitcode,
182 const std::vector<std::string> &Args,
183 const std::string &InputFile,
184 const std::string &OutputFile,
185 const std::vector<std::string> &GCCArgs,
186 const std::vector<std::string> &SharedLibs =
187 std::vector<std::string>(),
188 unsigned Timeout = 0,
189 unsigned MemoryLimit = 0);
190 };
191}
192
193int LLI::ExecuteProgram(const std::string &Bitcode,
194 const std::vector<std::string> &Args,
195 const std::string &InputFile,
196 const std::string &OutputFile,
197 const std::vector<std::string> &GCCArgs,
198 const std::vector<std::string> &SharedLibs,
199 unsigned Timeout,
200 unsigned MemoryLimit) {
201 if (!SharedLibs.empty())
202 throw ToolExecutionError("LLI currently does not support "
203 "loading shared libraries.");
204
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
209 // Add any extra LLI args.
210 for (unsigned i = 0, e = ToolArgs.size(); i != e; ++i)
211 LLIArgs.push_back(ToolArgs[i].c_str());
212
213 LLIArgs.push_back(Bitcode.c_str());
214 // Add optional parameters to the running program from Argv
215 for (unsigned i=0, e = Args.size(); i != e; ++i)
216 LLIArgs.push_back(Args[i].c_str());
217 LLIArgs.push_back(0);
218
Dan Gohmandd7ca8e2009-07-15 17:29:42 +0000219 outs() << "<lli>"; outs().flush();
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000220 DEBUG(errs() << "\nAbout to run:\t";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000221 for (unsigned i=0, e = LLIArgs.size()-1; i != e; ++i)
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000222 errs() << " " << LLIArgs[i];
223 errs() << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000224 );
225 return RunProgramWithTimeout(sys::Path(LLIPath), &LLIArgs[0],
226 sys::Path(InputFile), sys::Path(OutputFile), sys::Path(OutputFile),
227 Timeout, MemoryLimit);
228}
229
230// LLI create method - Try to find the LLI executable
Dan Gohman8e0b7b9d2009-08-05 20:21:17 +0000231AbstractInterpreter *AbstractInterpreter::createLLI(const char *Argv0,
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000232 std::string &Message,
233 const std::vector<std::string> *ToolArgs) {
Dan Gohman8e0b7b9d2009-08-05 20:21:17 +0000234 std::string LLIPath =
Dan Gohman108e0ed2009-08-05 21:03:39 +0000235 FindExecutable("lli", Argv0, (void *)(intptr_t)&createLLI).toString();
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000236 if (!LLIPath.empty()) {
237 Message = "Found lli: " + LLIPath + "\n";
238 return new LLI(LLIPath, ToolArgs);
239 }
240
241 Message = "Cannot find `lli' in executable directory or PATH!\n";
242 return 0;
243}
244
Anton Korobeynikovfc0116f2008-04-28 20:53:48 +0000245//===---------------------------------------------------------------------===//
246// Custom execution command implementation of AbstractIntepreter interface
247//
248// Allows using a custom command for executing the bitcode, thus allows,
249// for example, to invoke a cross compiler for code generation followed by
250// a simulator that executes the generated binary.
251namespace {
252 class CustomExecutor : public AbstractInterpreter {
253 std::string ExecutionCommand;
254 std::vector<std::string> ExecutorArgs;
255 public:
256 CustomExecutor(
257 const std::string &ExecutionCmd, std::vector<std::string> ExecArgs) :
258 ExecutionCommand(ExecutionCmd), ExecutorArgs(ExecArgs) {}
259
260 virtual int ExecuteProgram(const std::string &Bitcode,
261 const std::vector<std::string> &Args,
262 const std::string &InputFile,
263 const std::string &OutputFile,
264 const std::vector<std::string> &GCCArgs,
265 const std::vector<std::string> &SharedLibs =
266 std::vector<std::string>(),
267 unsigned Timeout = 0,
268 unsigned MemoryLimit = 0);
269 };
270}
271
272int CustomExecutor::ExecuteProgram(const std::string &Bitcode,
273 const std::vector<std::string> &Args,
274 const std::string &InputFile,
275 const std::string &OutputFile,
276 const std::vector<std::string> &GCCArgs,
277 const std::vector<std::string> &SharedLibs,
278 unsigned Timeout,
279 unsigned MemoryLimit) {
280
281 std::vector<const char*> ProgramArgs;
282 ProgramArgs.push_back(ExecutionCommand.c_str());
283
284 for (std::size_t i = 0; i < ExecutorArgs.size(); ++i)
285 ProgramArgs.push_back(ExecutorArgs.at(i).c_str());
286 ProgramArgs.push_back(Bitcode.c_str());
287 ProgramArgs.push_back(0);
288
289 // Add optional parameters to the running program from Argv
290 for (unsigned i=0, e = Args.size(); i != e; ++i)
291 ProgramArgs.push_back(Args[i].c_str());
292
293 return RunProgramWithTimeout(
294 sys::Path(ExecutionCommand),
295 &ProgramArgs[0], sys::Path(InputFile), sys::Path(OutputFile),
296 sys::Path(OutputFile), Timeout, MemoryLimit);
297}
298
299// Custom execution environment create method, takes the execution command
300// as arguments
301AbstractInterpreter *AbstractInterpreter::createCustom(
Anton Korobeynikovfc0116f2008-04-28 20:53:48 +0000302 std::string &Message,
303 const std::string &ExecCommandLine) {
304
305 std::string Command = "";
306 std::vector<std::string> Args;
307 std::string delimiters = " ";
308
309 // Tokenize the ExecCommandLine to the command and the args to allow
310 // defining a full command line as the command instead of just the
311 // executed program. We cannot just pass the whole string after the command
312 // as a single argument because then program sees only a single
313 // command line argument (with spaces in it: "foo bar" instead
314 // of "foo" and "bar").
315
316 // code borrowed from:
317 // http://oopweb.com/CPP/Documents/CPPHOWTO/Volume/C++Programming-HOWTO-7.html
318 std::string::size_type lastPos =
319 ExecCommandLine.find_first_not_of(delimiters, 0);
320 std::string::size_type pos =
321 ExecCommandLine.find_first_of(delimiters, lastPos);
322
323 while (std::string::npos != pos || std::string::npos != lastPos) {
324 std::string token = ExecCommandLine.substr(lastPos, pos - lastPos);
325 if (Command == "")
326 Command = token;
327 else
328 Args.push_back(token);
329 // Skip delimiters. Note the "not_of"
330 lastPos = ExecCommandLine.find_first_not_of(delimiters, pos);
331 // Find next "non-delimiter"
332 pos = ExecCommandLine.find_first_of(delimiters, lastPos);
333 }
334
Dan Gohman8e0b7b9d2009-08-05 20:21:17 +0000335 std::string CmdPath = sys::Program::FindProgramByName(Command).toString();
Anton Korobeynikovfc0116f2008-04-28 20:53:48 +0000336 if (CmdPath.empty()) {
337 Message =
338 std::string("Cannot find '") + Command +
339 "' in executable directory or PATH!\n";
340 return 0;
341 }
342
343 Message = "Found command in: " + CmdPath + "\n";
344
345 return new CustomExecutor(CmdPath, Args);
346}
347
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000348//===----------------------------------------------------------------------===//
349// LLC Implementation of AbstractIntepreter interface
350//
351GCC::FileType LLC::OutputCode(const std::string &Bitcode,
352 sys::Path &OutputAsmFile) {
353 sys::Path uniqueFile(Bitcode+".llc.s");
354 std::string ErrMsg;
355 if (uniqueFile.makeUnique(true, &ErrMsg)) {
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000356 errs() << "Error making unique filename: " << ErrMsg << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000357 exit(1);
358 }
359 OutputAsmFile = uniqueFile;
360 std::vector<const char *> LLCArgs;
361 LLCArgs.push_back (LLCPath.c_str());
362
363 // Add any extra LLC args.
364 for (unsigned i = 0, e = ToolArgs.size(); i != e; ++i)
365 LLCArgs.push_back(ToolArgs[i].c_str());
366
367 LLCArgs.push_back ("-o");
368 LLCArgs.push_back (OutputAsmFile.c_str()); // Output to the Asm file
369 LLCArgs.push_back ("-f"); // Overwrite as necessary...
370 LLCArgs.push_back (Bitcode.c_str()); // This is the input bitcode
371 LLCArgs.push_back (0);
372
Dan Gohmandd7ca8e2009-07-15 17:29:42 +0000373 outs() << "<llc>"; outs().flush();
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000374 DEBUG(errs() << "\nAbout to run:\t";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000375 for (unsigned i=0, e = LLCArgs.size()-1; i != e; ++i)
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000376 errs() << " " << LLCArgs[i];
377 errs() << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000378 );
379 if (RunProgramWithTimeout(sys::Path(LLCPath), &LLCArgs[0],
380 sys::Path(), sys::Path(), sys::Path()))
381 ProcessFailure(sys::Path(LLCPath), &LLCArgs[0]);
382
383 return GCC::AsmFile;
384}
385
386void LLC::compileProgram(const std::string &Bitcode) {
387 sys::Path OutputAsmFile;
388 OutputCode(Bitcode, OutputAsmFile);
389 OutputAsmFile.eraseFromDisk();
390}
391
392int LLC::ExecuteProgram(const std::string &Bitcode,
393 const std::vector<std::string> &Args,
394 const std::string &InputFile,
395 const std::string &OutputFile,
396 const std::vector<std::string> &ArgsForGCC,
397 const std::vector<std::string> &SharedLibs,
398 unsigned Timeout,
399 unsigned MemoryLimit) {
400
401 sys::Path OutputAsmFile;
402 OutputCode(Bitcode, OutputAsmFile);
Anton Korobeynikov43cf4f12009-08-05 09:32:10 +0000403 FileRemover OutFileRemover(OutputAsmFile, !SaveTemps);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000404
405 std::vector<std::string> GCCArgs(ArgsForGCC);
Bill Wendlingcd1389d2009-03-02 23:13:18 +0000406 GCCArgs.insert(GCCArgs.end(), SharedLibs.begin(), SharedLibs.end());
407 GCCArgs.insert(GCCArgs.end(), gccArgs.begin(), gccArgs.end());
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000408
409 // Assuming LLC worked, compile the result with GCC and run it.
410 return gcc->ExecuteProgram(OutputAsmFile.toString(), Args, GCC::AsmFile,
411 InputFile, OutputFile, GCCArgs,
412 Timeout, MemoryLimit);
413}
414
415/// createLLC - Try to find the LLC executable
416///
Dan Gohman8e0b7b9d2009-08-05 20:21:17 +0000417LLC *AbstractInterpreter::createLLC(const char *Argv0,
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000418 std::string &Message,
Bill Wendlingcd1389d2009-03-02 23:13:18 +0000419 const std::vector<std::string> *Args,
420 const std::vector<std::string> *GCCArgs) {
Dan Gohman8e0b7b9d2009-08-05 20:21:17 +0000421 std::string LLCPath =
Dan Gohman108e0ed2009-08-05 21:03:39 +0000422 FindExecutable("llc", Argv0, (void *)(intptr_t)&createLLC).toString();
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000423 if (LLCPath.empty()) {
424 Message = "Cannot find `llc' in executable directory or PATH!\n";
425 return 0;
426 }
427
428 Message = "Found llc: " + LLCPath + "\n";
Dan Gohman8e0b7b9d2009-08-05 20:21:17 +0000429 GCC *gcc = GCC::create(Message, GCCArgs);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000430 if (!gcc) {
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000431 errs() << Message << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000432 exit(1);
433 }
Bill Wendlingcd1389d2009-03-02 23:13:18 +0000434 return new LLC(LLCPath, gcc, Args, GCCArgs);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000435}
436
437//===---------------------------------------------------------------------===//
438// JIT Implementation of AbstractIntepreter interface
439//
440namespace {
441 class JIT : public AbstractInterpreter {
442 std::string LLIPath; // The path to the LLI executable
443 std::vector<std::string> ToolArgs; // Args to pass to LLI
444 public:
445 JIT(const std::string &Path, const std::vector<std::string> *Args)
446 : LLIPath(Path) {
447 ToolArgs.clear ();
448 if (Args) { ToolArgs = *Args; }
449 }
450
451 virtual int ExecuteProgram(const std::string &Bitcode,
452 const std::vector<std::string> &Args,
453 const std::string &InputFile,
454 const std::string &OutputFile,
455 const std::vector<std::string> &GCCArgs =
456 std::vector<std::string>(),
457 const std::vector<std::string> &SharedLibs =
458 std::vector<std::string>(),
459 unsigned Timeout =0,
460 unsigned MemoryLimit =0);
461 };
462}
463
464int JIT::ExecuteProgram(const std::string &Bitcode,
465 const std::vector<std::string> &Args,
466 const std::string &InputFile,
467 const std::string &OutputFile,
468 const std::vector<std::string> &GCCArgs,
469 const std::vector<std::string> &SharedLibs,
470 unsigned Timeout,
471 unsigned MemoryLimit) {
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000472 // Construct a vector of parameters, incorporating those from the command-line
473 std::vector<const char*> JITArgs;
474 JITArgs.push_back(LLIPath.c_str());
475 JITArgs.push_back("-force-interpreter=false");
476
477 // Add any extra LLI args.
478 for (unsigned i = 0, e = ToolArgs.size(); i != e; ++i)
479 JITArgs.push_back(ToolArgs[i].c_str());
480
481 for (unsigned i = 0, e = SharedLibs.size(); i != e; ++i) {
482 JITArgs.push_back("-load");
483 JITArgs.push_back(SharedLibs[i].c_str());
484 }
485 JITArgs.push_back(Bitcode.c_str());
486 // Add optional parameters to the running program from Argv
487 for (unsigned i=0, e = Args.size(); i != e; ++i)
488 JITArgs.push_back(Args[i].c_str());
489 JITArgs.push_back(0);
490
Dan Gohmandd7ca8e2009-07-15 17:29:42 +0000491 outs() << "<jit>"; outs().flush();
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000492 DEBUG(errs() << "\nAbout to run:\t";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000493 for (unsigned i=0, e = JITArgs.size()-1; i != e; ++i)
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000494 errs() << " " << JITArgs[i];
495 errs() << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000496 );
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000497 DEBUG(errs() << "\nSending output to " << OutputFile << "\n");
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000498 return RunProgramWithTimeout(sys::Path(LLIPath), &JITArgs[0],
499 sys::Path(InputFile), sys::Path(OutputFile), sys::Path(OutputFile),
500 Timeout, MemoryLimit);
501}
502
503/// createJIT - Try to find the LLI executable
504///
Dan Gohman8e0b7b9d2009-08-05 20:21:17 +0000505AbstractInterpreter *AbstractInterpreter::createJIT(const char *Argv0,
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000506 std::string &Message, const std::vector<std::string> *Args) {
Dan Gohman8e0b7b9d2009-08-05 20:21:17 +0000507 std::string LLIPath =
Dan Gohman108e0ed2009-08-05 21:03:39 +0000508 FindExecutable("lli", Argv0, (void *)(intptr_t)&createJIT).toString();
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000509 if (!LLIPath.empty()) {
510 Message = "Found lli: " + LLIPath + "\n";
511 return new JIT(LLIPath, Args);
512 }
513
514 Message = "Cannot find `lli' in executable directory or PATH!\n";
515 return 0;
516}
517
518GCC::FileType CBE::OutputCode(const std::string &Bitcode,
519 sys::Path &OutputCFile) {
520 sys::Path uniqueFile(Bitcode+".cbe.c");
521 std::string ErrMsg;
522 if (uniqueFile.makeUnique(true, &ErrMsg)) {
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000523 errs() << "Error making unique filename: " << ErrMsg << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000524 exit(1);
525 }
526 OutputCFile = uniqueFile;
527 std::vector<const char *> LLCArgs;
528 LLCArgs.push_back (LLCPath.c_str());
529
530 // Add any extra LLC args.
531 for (unsigned i = 0, e = ToolArgs.size(); i != e; ++i)
532 LLCArgs.push_back(ToolArgs[i].c_str());
533
534 LLCArgs.push_back ("-o");
535 LLCArgs.push_back (OutputCFile.c_str()); // Output to the C file
536 LLCArgs.push_back ("-march=c"); // Output C language
537 LLCArgs.push_back ("-f"); // Overwrite as necessary...
538 LLCArgs.push_back (Bitcode.c_str()); // This is the input bitcode
539 LLCArgs.push_back (0);
540
Dan Gohmandd7ca8e2009-07-15 17:29:42 +0000541 outs() << "<cbe>"; outs().flush();
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000542 DEBUG(errs() << "\nAbout to run:\t";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000543 for (unsigned i=0, e = LLCArgs.size()-1; i != e; ++i)
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000544 errs() << " " << LLCArgs[i];
545 errs() << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000546 );
547 if (RunProgramWithTimeout(LLCPath, &LLCArgs[0], sys::Path(), sys::Path(),
548 sys::Path()))
549 ProcessFailure(LLCPath, &LLCArgs[0]);
550 return GCC::CFile;
551}
552
553void CBE::compileProgram(const std::string &Bitcode) {
554 sys::Path OutputCFile;
555 OutputCode(Bitcode, OutputCFile);
556 OutputCFile.eraseFromDisk();
557}
558
559int CBE::ExecuteProgram(const std::string &Bitcode,
560 const std::vector<std::string> &Args,
561 const std::string &InputFile,
562 const std::string &OutputFile,
563 const std::vector<std::string> &ArgsForGCC,
564 const std::vector<std::string> &SharedLibs,
565 unsigned Timeout,
566 unsigned MemoryLimit) {
567 sys::Path OutputCFile;
568 OutputCode(Bitcode, OutputCFile);
569
Anton Korobeynikov43cf4f12009-08-05 09:32:10 +0000570 FileRemover CFileRemove(OutputCFile, !SaveTemps);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000571
572 std::vector<std::string> GCCArgs(ArgsForGCC);
Bill Wendlingcd1389d2009-03-02 23:13:18 +0000573 GCCArgs.insert(GCCArgs.end(), SharedLibs.begin(), SharedLibs.end());
574
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000575 return gcc->ExecuteProgram(OutputCFile.toString(), Args, GCC::CFile,
576 InputFile, OutputFile, GCCArgs,
577 Timeout, MemoryLimit);
578}
579
580/// createCBE - Try to find the 'llc' executable
581///
Dan Gohman8e0b7b9d2009-08-05 20:21:17 +0000582CBE *AbstractInterpreter::createCBE(const char *Argv0,
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000583 std::string &Message,
Bill Wendlingcd1389d2009-03-02 23:13:18 +0000584 const std::vector<std::string> *Args,
585 const std::vector<std::string> *GCCArgs) {
Dan Gohman8e0b7b9d2009-08-05 20:21:17 +0000586 sys::Path LLCPath =
Dan Gohman108e0ed2009-08-05 21:03:39 +0000587 FindExecutable("llc", Argv0, (void *)(intptr_t)&createCBE);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000588 if (LLCPath.isEmpty()) {
589 Message =
590 "Cannot find `llc' in executable directory or PATH!\n";
591 return 0;
592 }
593
594 Message = "Found llc: " + LLCPath.toString() + "\n";
Dan Gohman8e0b7b9d2009-08-05 20:21:17 +0000595 GCC *gcc = GCC::create(Message, GCCArgs);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000596 if (!gcc) {
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000597 errs() << Message << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000598 exit(1);
599 }
600 return new CBE(LLCPath, gcc, Args);
601}
602
603//===---------------------------------------------------------------------===//
604// GCC abstraction
605//
David Goodwinec1edba2009-07-10 21:39:28 +0000606
607static bool
608IsARMArchitecture(std::vector<std::string> Args)
609{
610 for (std::vector<std::string>::const_iterator
611 I = Args.begin(), E = Args.end(); I != E; ++I) {
612 if (!strcasecmp(I->c_str(), "-arch")) {
613 ++I;
614 if ((I != E) && !strncasecmp(I->c_str(), "arm", strlen("arm"))) {
615 return true;
616 }
617 }
618 }
619
620 return false;
621}
622
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000623int GCC::ExecuteProgram(const std::string &ProgramFile,
624 const std::vector<std::string> &Args,
625 FileType fileType,
626 const std::string &InputFile,
627 const std::string &OutputFile,
628 const std::vector<std::string> &ArgsForGCC,
629 unsigned Timeout,
630 unsigned MemoryLimit) {
631 std::vector<const char*> GCCArgs;
632
633 GCCArgs.push_back(GCCPath.c_str());
634
Bill Wendlingcd1389d2009-03-02 23:13:18 +0000635 for (std::vector<std::string>::const_iterator
636 I = gccArgs.begin(), E = gccArgs.end(); I != E; ++I)
637 GCCArgs.push_back(I->c_str());
638
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000639 // Specify -x explicitly in case the extension is wonky
640 GCCArgs.push_back("-x");
641 if (fileType == CFile) {
642 GCCArgs.push_back("c");
643 GCCArgs.push_back("-fno-strict-aliasing");
644 } else {
645 GCCArgs.push_back("assembler");
Daniel Dunbard8590af2009-08-18 03:35:57 +0000646
David Goodwinec1edba2009-07-10 21:39:28 +0000647 // For ARM architectures we don't want this flag. bugpoint isn't
648 // explicitly told what architecture it is working on, so we get
649 // it from gcc flags
Daniel Dunbard8590af2009-08-18 03:35:57 +0000650 if ((TargetTriple.getOS() == Triple::Darwin) &&
651 !IsARMArchitecture(ArgsForGCC))
David Goodwinec1edba2009-07-10 21:39:28 +0000652 GCCArgs.push_back("-force_cpusubtype_ALL");
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000653 }
654 GCCArgs.push_back(ProgramFile.c_str()); // Specify the input filename...
655 GCCArgs.push_back("-x");
656 GCCArgs.push_back("none");
657 GCCArgs.push_back("-o");
658 sys::Path OutputBinary (ProgramFile+".gcc.exe");
659 std::string ErrMsg;
660 if (OutputBinary.makeUnique(true, &ErrMsg)) {
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000661 errs() << "Error making unique filename: " << ErrMsg << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000662 exit(1);
663 }
664 GCCArgs.push_back(OutputBinary.c_str()); // Output to the right file...
665
666 // Add any arguments intended for GCC. We locate them here because this is
667 // most likely -L and -l options that need to come before other libraries but
668 // after the source. Other options won't be sensitive to placement on the
669 // command line, so this should be safe.
670 for (unsigned i = 0, e = ArgsForGCC.size(); i != e; ++i)
671 GCCArgs.push_back(ArgsForGCC[i].c_str());
672
673 GCCArgs.push_back("-lm"); // Hard-code the math library...
674 GCCArgs.push_back("-O2"); // Optimize the program a bit...
675#if defined (HAVE_LINK_R)
676 GCCArgs.push_back("-Wl,-R."); // Search this dir for .so files
677#endif
Daniel Dunbard8590af2009-08-18 03:35:57 +0000678 if (TargetTriple.getArch() == Triple::sparc)
679 GCCArgs.push_back("-mcpu=v9");
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000680 GCCArgs.push_back(0); // NULL terminator
681
Dan Gohmandd7ca8e2009-07-15 17:29:42 +0000682 outs() << "<gcc>"; outs().flush();
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000683 DEBUG(errs() << "\nAbout to run:\t";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000684 for (unsigned i=0, e = GCCArgs.size()-1; i != e; ++i)
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000685 errs() << " " << GCCArgs[i];
686 errs() << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000687 );
688 if (RunProgramWithTimeout(GCCPath, &GCCArgs[0], sys::Path(), sys::Path(),
689 sys::Path())) {
690 ProcessFailure(GCCPath, &GCCArgs[0]);
691 exit(1);
692 }
693
694 std::vector<const char*> ProgramArgs;
695
Evan Cheng0646fff2008-09-09 06:11:26 +0000696 if (RemoteClientPath.isEmpty())
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000697 ProgramArgs.push_back(OutputBinary.c_str());
698 else {
Evan Cheng0646fff2008-09-09 06:11:26 +0000699 ProgramArgs.push_back(RemoteClientPath.c_str());
700 ProgramArgs.push_back(RemoteHost.c_str());
Anton Korobeynikov33f97a62009-08-05 09:32:53 +0000701 if (!RemoteUser.empty()) {
702 ProgramArgs.push_back("-l");
703 ProgramArgs.push_back(RemoteUser.c_str());
704 }
David Goodwinec1edba2009-07-10 21:39:28 +0000705 if (!RemotePort.empty()) {
706 ProgramArgs.push_back("-p");
707 ProgramArgs.push_back(RemotePort.c_str());
708 }
Evan Cheng0646fff2008-09-09 06:11:26 +0000709 if (!RemoteExtra.empty()) {
710 ProgramArgs.push_back(RemoteExtra.c_str());
711 }
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000712
David Goodwinfbb0faf2009-07-20 17:15:03 +0000713 // Full path to the binary. We need to cd to the exec directory because
714 // there is a dylib there that the exec expects to find in the CWD
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000715 char* env_pwd = getenv("PWD");
David Goodwinfbb0faf2009-07-20 17:15:03 +0000716 std::string Exec = "cd ";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000717 Exec += env_pwd;
David Goodwinfbb0faf2009-07-20 17:15:03 +0000718 Exec += "; ./";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000719 Exec += OutputBinary.c_str();
720 ProgramArgs.push_back(Exec.c_str());
721 }
722
723 // Add optional parameters to the running program from Argv
724 for (unsigned i=0, e = Args.size(); i != e; ++i)
725 ProgramArgs.push_back(Args[i].c_str());
726 ProgramArgs.push_back(0); // NULL terminator
727
728 // Now that we have a binary, run it!
Dan Gohmandd7ca8e2009-07-15 17:29:42 +0000729 outs() << "<program>"; outs().flush();
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000730 DEBUG(errs() << "\nAbout to run:\t";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000731 for (unsigned i=0, e = ProgramArgs.size()-1; i != e; ++i)
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000732 errs() << " " << ProgramArgs[i];
733 errs() << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000734 );
735
Anton Korobeynikov43cf4f12009-08-05 09:32:10 +0000736 FileRemover OutputBinaryRemover(OutputBinary, !SaveTemps);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000737
Viktor Kutuzov8c3e63c2009-07-14 19:10:55 +0000738 if (RemoteClientPath.isEmpty()) {
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000739 DEBUG(errs() << "<run locally>";);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000740 return RunProgramWithTimeout(OutputBinary, &ProgramArgs[0],
741 sys::Path(InputFile), sys::Path(OutputFile), sys::Path(OutputFile),
742 Timeout, MemoryLimit);
Viktor Kutuzov8c3e63c2009-07-14 19:10:55 +0000743 } else {
Dan Gohmandd7ca8e2009-07-15 17:29:42 +0000744 outs() << "<run remotely>"; outs().flush();
Viktor Kutuzov2743a2e2009-07-18 18:39:24 +0000745 return RunProgramRemotelyWithTimeout(sys::Path(RemoteClientPath),
Viktor Kutuzov8c3e63c2009-07-14 19:10:55 +0000746 &ProgramArgs[0], sys::Path(InputFile), sys::Path(OutputFile),
747 sys::Path(OutputFile), Timeout, MemoryLimit);
Viktor Kutuzov8c3e63c2009-07-14 19:10:55 +0000748 }
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000749}
750
751int GCC::MakeSharedObject(const std::string &InputFile, FileType fileType,
752 std::string &OutputFile,
753 const std::vector<std::string> &ArgsForGCC) {
754 sys::Path uniqueFilename(InputFile+LTDL_SHLIB_EXT);
755 std::string ErrMsg;
756 if (uniqueFilename.makeUnique(true, &ErrMsg)) {
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000757 errs() << "Error making unique filename: " << ErrMsg << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000758 exit(1);
759 }
760 OutputFile = uniqueFilename.toString();
761
762 std::vector<const char*> GCCArgs;
763
764 GCCArgs.push_back(GCCPath.c_str());
Evan Chengf94bcf62009-03-12 00:53:34 +0000765
766 for (std::vector<std::string>::const_iterator
767 I = gccArgs.begin(), E = gccArgs.end(); I != E; ++I)
768 GCCArgs.push_back(I->c_str());
769
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000770 // Compile the C/asm file into a shared object
771 GCCArgs.push_back("-x");
772 GCCArgs.push_back(fileType == AsmFile ? "assembler" : "c");
773 GCCArgs.push_back("-fno-strict-aliasing");
774 GCCArgs.push_back(InputFile.c_str()); // Specify the input filename.
775 GCCArgs.push_back("-x");
776 GCCArgs.push_back("none");
Daniel Dunbard8590af2009-08-18 03:35:57 +0000777 if (TargetTriple.getArch() == Triple::sparc)
778 GCCArgs.push_back("-G"); // Compile a shared library, `-G' for Sparc
779 else if (TargetTriple.getOS() == Triple::Darwin) {
780 // link all source files into a single module in data segment, rather than
781 // generating blocks. dynamic_lookup requires that you set
782 // MACOSX_DEPLOYMENT_TARGET=10.3 in your env. FIXME: it would be better for
783 // bugpoint to just pass that in the environment of GCC.
784 GCCArgs.push_back("-single_module");
785 GCCArgs.push_back("-dynamiclib"); // `-dynamiclib' for MacOS X/PowerPC
786 GCCArgs.push_back("-undefined");
787 GCCArgs.push_back("dynamic_lookup");
788 } else
789 GCCArgs.push_back("-shared"); // `-shared' for Linux/X86, maybe others
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000790
Daniel Dunbard8590af2009-08-18 03:35:57 +0000791 if ((TargetTriple.getArch() == Triple::alpha) ||
792 (TargetTriple.getArch() == Triple::x86_64))
793 GCCArgs.push_back("-fPIC"); // Requires shared objs to contain PIC
794
795 if (TargetTriple.getArch() == Triple::sparc)
796 GCCArgs.push_back("-mcpu=v9");
797
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000798 GCCArgs.push_back("-o");
799 GCCArgs.push_back(OutputFile.c_str()); // Output to the right filename.
800 GCCArgs.push_back("-O2"); // Optimize the program a bit.
801
802
803
804 // Add any arguments intended for GCC. We locate them here because this is
805 // most likely -L and -l options that need to come before other libraries but
806 // after the source. Other options won't be sensitive to placement on the
807 // command line, so this should be safe.
808 for (unsigned i = 0, e = ArgsForGCC.size(); i != e; ++i)
809 GCCArgs.push_back(ArgsForGCC[i].c_str());
810 GCCArgs.push_back(0); // NULL terminator
811
812
813
Dan Gohmandd7ca8e2009-07-15 17:29:42 +0000814 outs() << "<gcc>"; outs().flush();
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000815 DEBUG(errs() << "\nAbout to run:\t";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000816 for (unsigned i=0, e = GCCArgs.size()-1; i != e; ++i)
Dan Gohmanf8b81bf2009-07-15 16:35:29 +0000817 errs() << " " << GCCArgs[i];
818 errs() << "\n";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000819 );
820 if (RunProgramWithTimeout(GCCPath, &GCCArgs[0], sys::Path(), sys::Path(),
821 sys::Path())) {
822 ProcessFailure(GCCPath, &GCCArgs[0]);
823 return 1;
824 }
825 return 0;
826}
827
828/// create - Try to find the `gcc' executable
829///
Dan Gohman8e0b7b9d2009-08-05 20:21:17 +0000830GCC *GCC::create(std::string &Message,
Bill Wendlingcd1389d2009-03-02 23:13:18 +0000831 const std::vector<std::string> *Args) {
Dan Gohman8e0b7b9d2009-08-05 20:21:17 +0000832 sys::Path GCCPath = sys::Program::FindProgramByName("gcc");
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000833 if (GCCPath.isEmpty()) {
834 Message = "Cannot find `gcc' in executable directory or PATH!\n";
835 return 0;
836 }
837
Evan Cheng0646fff2008-09-09 06:11:26 +0000838 sys::Path RemoteClientPath;
839 if (!RemoteClient.empty())
Dan Gohman8e0b7b9d2009-08-05 20:21:17 +0000840 RemoteClientPath = sys::Program::FindProgramByName(RemoteClient);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000841
842 Message = "Found gcc: " + GCCPath.toString() + "\n";
Bill Wendlingcd1389d2009-03-02 23:13:18 +0000843 return new GCC(GCCPath, RemoteClientPath, Args);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000844}