blob: 503a5dda136a62551532c815d71a84d54e0b0395 [file] [log] [blame]
Chris Lattnerffac2862006-06-06 22:30:59 +00001//===-- ToolRunner.cpp ----------------------------------------------------===//
2//
3// The LLVM Compiler Infrastructure
4//
Chris Lattner345353d2007-12-29 20:44:31 +00005// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
Chris Lattnerffac2862006-06-06 22:30:59 +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"
Evan Cheng779b52e2007-05-03 18:36:15 +000018#include "llvm/Support/CommandLine.h"
Chris Lattnerffac2862006-06-06 22:30:59 +000019#include "llvm/Support/Debug.h"
20#include "llvm/Support/FileUtilities.h"
21#include <fstream>
22#include <sstream>
23#include <iostream>
24using namespace llvm;
25
Evan Cheng779b52e2007-05-03 18:36:15 +000026namespace {
27 cl::opt<std::string>
Evan Cheng7aedcf12008-09-09 06:11:26 +000028 RemoteClient("remote-client",
29 cl::desc("Remote execution client (rsh/ssh)"));
Evan Cheng779b52e2007-05-03 18:36:15 +000030
31 cl::opt<std::string>
Evan Cheng7aedcf12008-09-09 06:11:26 +000032 RemoteHost("remote-host",
33 cl::desc("Remote execution (rsh/ssh) host"));
34
35 cl::opt<std::string>
David Goodwin73f4e3f2009-07-10 21:39:28 +000036 RemotePort("remote-port",
37 cl::desc("Remote execution (rsh/ssh) port"));
38
39 cl::opt<std::string>
Evan Cheng7aedcf12008-09-09 06:11:26 +000040 RemoteUser("remote-user",
41 cl::desc("Remote execution (rsh/ssh) user id"));
42
43 cl::opt<std::string>
44 RemoteExtra("remote-extra-options",
45 cl::desc("Remote execution (rsh/ssh) extra options"));
Evan Cheng779b52e2007-05-03 18:36:15 +000046}
47
Chris Lattnerffac2862006-06-06 22:30:59 +000048ToolExecutionError::~ToolExecutionError() throw() { }
49
50/// RunProgramWithTimeout - This function provides an alternate interface to the
51/// sys::Program::ExecuteAndWait interface.
52/// @see sys:Program::ExecuteAndWait
53static int RunProgramWithTimeout(const sys::Path &ProgramPath,
54 const char **Args,
55 const sys::Path &StdInFile,
56 const sys::Path &StdOutFile,
57 const sys::Path &StdErrFile,
Anton Korobeynikovd01defe2007-02-16 19:11:07 +000058 unsigned NumSeconds = 0,
59 unsigned MemoryLimit = 0) {
Chris Lattnerffac2862006-06-06 22:30:59 +000060 const sys::Path* redirects[3];
61 redirects[0] = &StdInFile;
62 redirects[1] = &StdOutFile;
63 redirects[2] = &StdErrFile;
Chris Lattner634bc042006-09-15 21:29:15 +000064
Chris Lattnerff8a6b82006-09-15 23:01:10 +000065 if (0) {
66 std::cerr << "RUN:";
67 for (unsigned i = 0; Args[i]; ++i)
68 std::cerr << " " << Args[i];
69 std::cerr << "\n";
70 }
Chris Lattnerffac2862006-06-06 22:30:59 +000071
72 return
Anton Korobeynikovd01defe2007-02-16 19:11:07 +000073 sys::Program::ExecuteAndWait(ProgramPath, Args, 0, redirects,
74 NumSeconds, MemoryLimit);
Chris Lattnerffac2862006-06-06 22:30:59 +000075}
76
77
78
79static void ProcessFailure(sys::Path ProgPath, const char** Args) {
80 std::ostringstream OS;
81 OS << "\nError running tool:\n ";
82 for (const char **Arg = Args; *Arg; ++Arg)
83 OS << " " << *Arg;
84 OS << "\n";
85
86 // Rerun the compiler, capturing any error messages to print them.
Dan Gohmanaf2037f2009-02-12 20:53:27 +000087 sys::Path ErrorFilename("bugpoint.program_error_messages");
Reid Spencere4ca7222006-08-23 20:34:57 +000088 std::string ErrMsg;
89 if (ErrorFilename.makeUnique(true, &ErrMsg)) {
90 std::cerr << "Error making unique filename: " << ErrMsg << "\n";
91 exit(1);
92 }
Chris Lattnerffac2862006-06-06 22:30:59 +000093 RunProgramWithTimeout(ProgPath, Args, sys::Path(""), ErrorFilename,
Reid Spencer944645a2006-08-21 06:04:45 +000094 ErrorFilename); // FIXME: check return code ?
Chris Lattnerffac2862006-06-06 22:30:59 +000095
96 // Print out the error messages generated by GCC if possible...
97 std::ifstream ErrorFile(ErrorFilename.c_str());
98 if (ErrorFile) {
99 std::copy(std::istreambuf_iterator<char>(ErrorFile),
100 std::istreambuf_iterator<char>(),
101 std::ostreambuf_iterator<char>(OS));
102 ErrorFile.close();
103 }
104
105 ErrorFilename.eraseFromDisk();
106 throw ToolExecutionError(OS.str());
107}
108
109//===---------------------------------------------------------------------===//
110// LLI Implementation of AbstractIntepreter interface
111//
112namespace {
113 class LLI : public AbstractInterpreter {
114 std::string LLIPath; // The path to the LLI executable
115 std::vector<std::string> ToolArgs; // Args to pass to LLI
116 public:
117 LLI(const std::string &Path, const std::vector<std::string> *Args)
118 : LLIPath(Path) {
119 ToolArgs.clear ();
120 if (Args) { ToolArgs = *Args; }
121 }
122
Gabor Greif0e535c3c2007-07-04 21:55:50 +0000123 virtual int ExecuteProgram(const std::string &Bitcode,
Chris Lattnerffac2862006-06-06 22:30:59 +0000124 const std::vector<std::string> &Args,
125 const std::string &InputFile,
126 const std::string &OutputFile,
127 const std::vector<std::string> &GCCArgs,
128 const std::vector<std::string> &SharedLibs =
129 std::vector<std::string>(),
Anton Korobeynikovd01defe2007-02-16 19:11:07 +0000130 unsigned Timeout = 0,
131 unsigned MemoryLimit = 0);
Chris Lattnerffac2862006-06-06 22:30:59 +0000132 };
133}
134
Gabor Greif0e535c3c2007-07-04 21:55:50 +0000135int LLI::ExecuteProgram(const std::string &Bitcode,
Chris Lattnerffac2862006-06-06 22:30:59 +0000136 const std::vector<std::string> &Args,
137 const std::string &InputFile,
138 const std::string &OutputFile,
139 const std::vector<std::string> &GCCArgs,
140 const std::vector<std::string> &SharedLibs,
Anton Korobeynikovd01defe2007-02-16 19:11:07 +0000141 unsigned Timeout,
142 unsigned MemoryLimit) {
Chris Lattnerffac2862006-06-06 22:30:59 +0000143 if (!SharedLibs.empty())
144 throw ToolExecutionError("LLI currently does not support "
145 "loading shared libraries.");
146
Chris Lattnerffac2862006-06-06 22:30:59 +0000147 std::vector<const char*> LLIArgs;
148 LLIArgs.push_back(LLIPath.c_str());
149 LLIArgs.push_back("-force-interpreter=true");
150
151 // Add any extra LLI args.
152 for (unsigned i = 0, e = ToolArgs.size(); i != e; ++i)
153 LLIArgs.push_back(ToolArgs[i].c_str());
154
Gabor Greif0e535c3c2007-07-04 21:55:50 +0000155 LLIArgs.push_back(Bitcode.c_str());
Chris Lattnerffac2862006-06-06 22:30:59 +0000156 // Add optional parameters to the running program from Argv
157 for (unsigned i=0, e = Args.size(); i != e; ++i)
158 LLIArgs.push_back(Args[i].c_str());
159 LLIArgs.push_back(0);
160
161 std::cout << "<lli>" << std::flush;
162 DEBUG(std::cerr << "\nAbout to run:\t";
163 for (unsigned i=0, e = LLIArgs.size()-1; i != e; ++i)
164 std::cerr << " " << LLIArgs[i];
165 std::cerr << "\n";
166 );
167 return RunProgramWithTimeout(sys::Path(LLIPath), &LLIArgs[0],
168 sys::Path(InputFile), sys::Path(OutputFile), sys::Path(OutputFile),
Anton Korobeynikovd01defe2007-02-16 19:11:07 +0000169 Timeout, MemoryLimit);
Chris Lattnerffac2862006-06-06 22:30:59 +0000170}
171
172// LLI create method - Try to find the LLI executable
173AbstractInterpreter *AbstractInterpreter::createLLI(const std::string &ProgPath,
174 std::string &Message,
175 const std::vector<std::string> *ToolArgs) {
176 std::string LLIPath = FindExecutable("lli", ProgPath).toString();
177 if (!LLIPath.empty()) {
178 Message = "Found lli: " + LLIPath + "\n";
179 return new LLI(LLIPath, ToolArgs);
180 }
181
182 Message = "Cannot find `lli' in executable directory or PATH!\n";
183 return 0;
184}
185
Anton Korobeynikovc53565c2008-04-28 20:53:48 +0000186//===---------------------------------------------------------------------===//
187// Custom execution command implementation of AbstractIntepreter interface
188//
189// Allows using a custom command for executing the bitcode, thus allows,
190// for example, to invoke a cross compiler for code generation followed by
191// a simulator that executes the generated binary.
192namespace {
193 class CustomExecutor : public AbstractInterpreter {
194 std::string ExecutionCommand;
195 std::vector<std::string> ExecutorArgs;
196 public:
197 CustomExecutor(
198 const std::string &ExecutionCmd, std::vector<std::string> ExecArgs) :
199 ExecutionCommand(ExecutionCmd), ExecutorArgs(ExecArgs) {}
200
201 virtual int ExecuteProgram(const std::string &Bitcode,
202 const std::vector<std::string> &Args,
203 const std::string &InputFile,
204 const std::string &OutputFile,
205 const std::vector<std::string> &GCCArgs,
206 const std::vector<std::string> &SharedLibs =
207 std::vector<std::string>(),
208 unsigned Timeout = 0,
209 unsigned MemoryLimit = 0);
210 };
211}
212
213int CustomExecutor::ExecuteProgram(const std::string &Bitcode,
214 const std::vector<std::string> &Args,
215 const std::string &InputFile,
216 const std::string &OutputFile,
217 const std::vector<std::string> &GCCArgs,
218 const std::vector<std::string> &SharedLibs,
219 unsigned Timeout,
220 unsigned MemoryLimit) {
221
222 std::vector<const char*> ProgramArgs;
223 ProgramArgs.push_back(ExecutionCommand.c_str());
224
225 for (std::size_t i = 0; i < ExecutorArgs.size(); ++i)
226 ProgramArgs.push_back(ExecutorArgs.at(i).c_str());
227 ProgramArgs.push_back(Bitcode.c_str());
228 ProgramArgs.push_back(0);
229
230 // Add optional parameters to the running program from Argv
231 for (unsigned i=0, e = Args.size(); i != e; ++i)
232 ProgramArgs.push_back(Args[i].c_str());
233
234 return RunProgramWithTimeout(
235 sys::Path(ExecutionCommand),
236 &ProgramArgs[0], sys::Path(InputFile), sys::Path(OutputFile),
237 sys::Path(OutputFile), Timeout, MemoryLimit);
238}
239
240// Custom execution environment create method, takes the execution command
241// as arguments
242AbstractInterpreter *AbstractInterpreter::createCustom(
243 const std::string &ProgramPath,
244 std::string &Message,
245 const std::string &ExecCommandLine) {
246
247 std::string Command = "";
248 std::vector<std::string> Args;
249 std::string delimiters = " ";
250
251 // Tokenize the ExecCommandLine to the command and the args to allow
252 // defining a full command line as the command instead of just the
253 // executed program. We cannot just pass the whole string after the command
254 // as a single argument because then program sees only a single
255 // command line argument (with spaces in it: "foo bar" instead
256 // of "foo" and "bar").
257
258 // code borrowed from:
259 // http://oopweb.com/CPP/Documents/CPPHOWTO/Volume/C++Programming-HOWTO-7.html
260 std::string::size_type lastPos =
261 ExecCommandLine.find_first_not_of(delimiters, 0);
262 std::string::size_type pos =
263 ExecCommandLine.find_first_of(delimiters, lastPos);
264
265 while (std::string::npos != pos || std::string::npos != lastPos) {
266 std::string token = ExecCommandLine.substr(lastPos, pos - lastPos);
267 if (Command == "")
268 Command = token;
269 else
270 Args.push_back(token);
271 // Skip delimiters. Note the "not_of"
272 lastPos = ExecCommandLine.find_first_not_of(delimiters, pos);
273 // Find next "non-delimiter"
274 pos = ExecCommandLine.find_first_of(delimiters, lastPos);
275 }
276
277 std::string CmdPath = FindExecutable(Command, ProgramPath).toString();
278 if (CmdPath.empty()) {
279 Message =
280 std::string("Cannot find '") + Command +
281 "' in executable directory or PATH!\n";
282 return 0;
283 }
284
285 Message = "Found command in: " + CmdPath + "\n";
286
287 return new CustomExecutor(CmdPath, Args);
288}
289
Chris Lattnerffac2862006-06-06 22:30:59 +0000290//===----------------------------------------------------------------------===//
291// LLC Implementation of AbstractIntepreter interface
292//
Gabor Greif0e535c3c2007-07-04 21:55:50 +0000293GCC::FileType LLC::OutputCode(const std::string &Bitcode,
Chris Lattner634bc042006-09-15 21:29:15 +0000294 sys::Path &OutputAsmFile) {
Gabor Greif0e535c3c2007-07-04 21:55:50 +0000295 sys::Path uniqueFile(Bitcode+".llc.s");
Reid Spencere4ca7222006-08-23 20:34:57 +0000296 std::string ErrMsg;
297 if (uniqueFile.makeUnique(true, &ErrMsg)) {
298 std::cerr << "Error making unique filename: " << ErrMsg << "\n";
299 exit(1);
300 }
Chris Lattnerffac2862006-06-06 22:30:59 +0000301 OutputAsmFile = uniqueFile;
302 std::vector<const char *> LLCArgs;
303 LLCArgs.push_back (LLCPath.c_str());
304
305 // Add any extra LLC args.
306 for (unsigned i = 0, e = ToolArgs.size(); i != e; ++i)
307 LLCArgs.push_back(ToolArgs[i].c_str());
308
309 LLCArgs.push_back ("-o");
310 LLCArgs.push_back (OutputAsmFile.c_str()); // Output to the Asm file
311 LLCArgs.push_back ("-f"); // Overwrite as necessary...
Gabor Greif0e535c3c2007-07-04 21:55:50 +0000312 LLCArgs.push_back (Bitcode.c_str()); // This is the input bitcode
Chris Lattnerffac2862006-06-06 22:30:59 +0000313 LLCArgs.push_back (0);
314
315 std::cout << "<llc>" << std::flush;
316 DEBUG(std::cerr << "\nAbout to run:\t";
317 for (unsigned i=0, e = LLCArgs.size()-1; i != e; ++i)
318 std::cerr << " " << LLCArgs[i];
319 std::cerr << "\n";
320 );
321 if (RunProgramWithTimeout(sys::Path(LLCPath), &LLCArgs[0],
322 sys::Path(), sys::Path(), sys::Path()))
323 ProcessFailure(sys::Path(LLCPath), &LLCArgs[0]);
Chris Lattner634bc042006-09-15 21:29:15 +0000324
325 return GCC::AsmFile;
Chris Lattnerffac2862006-06-06 22:30:59 +0000326}
327
Gabor Greif0e535c3c2007-07-04 21:55:50 +0000328void LLC::compileProgram(const std::string &Bitcode) {
Chris Lattnerffac2862006-06-06 22:30:59 +0000329 sys::Path OutputAsmFile;
Gabor Greif0e535c3c2007-07-04 21:55:50 +0000330 OutputCode(Bitcode, OutputAsmFile);
Chris Lattnerffac2862006-06-06 22:30:59 +0000331 OutputAsmFile.eraseFromDisk();
332}
333
Gabor Greif0e535c3c2007-07-04 21:55:50 +0000334int LLC::ExecuteProgram(const std::string &Bitcode,
Chris Lattnerffac2862006-06-06 22:30:59 +0000335 const std::vector<std::string> &Args,
336 const std::string &InputFile,
337 const std::string &OutputFile,
338 const std::vector<std::string> &ArgsForGCC,
339 const std::vector<std::string> &SharedLibs,
Anton Korobeynikovd01defe2007-02-16 19:11:07 +0000340 unsigned Timeout,
341 unsigned MemoryLimit) {
Chris Lattnerffac2862006-06-06 22:30:59 +0000342
343 sys::Path OutputAsmFile;
Gabor Greif0e535c3c2007-07-04 21:55:50 +0000344 OutputCode(Bitcode, OutputAsmFile);
Chris Lattnerffac2862006-06-06 22:30:59 +0000345 FileRemover OutFileRemover(OutputAsmFile);
346
347 std::vector<std::string> GCCArgs(ArgsForGCC);
Bill Wendlingbf5d8272009-03-02 23:13:18 +0000348 GCCArgs.insert(GCCArgs.end(), SharedLibs.begin(), SharedLibs.end());
349 GCCArgs.insert(GCCArgs.end(), gccArgs.begin(), gccArgs.end());
Chris Lattnerffac2862006-06-06 22:30:59 +0000350
351 // Assuming LLC worked, compile the result with GCC and run it.
352 return gcc->ExecuteProgram(OutputAsmFile.toString(), Args, GCC::AsmFile,
Anton Korobeynikovd01defe2007-02-16 19:11:07 +0000353 InputFile, OutputFile, GCCArgs,
354 Timeout, MemoryLimit);
Chris Lattnerffac2862006-06-06 22:30:59 +0000355}
356
357/// createLLC - Try to find the LLC executable
358///
359LLC *AbstractInterpreter::createLLC(const std::string &ProgramPath,
360 std::string &Message,
Bill Wendlingbf5d8272009-03-02 23:13:18 +0000361 const std::vector<std::string> *Args,
362 const std::vector<std::string> *GCCArgs) {
Chris Lattnerffac2862006-06-06 22:30:59 +0000363 std::string LLCPath = FindExecutable("llc", ProgramPath).toString();
364 if (LLCPath.empty()) {
365 Message = "Cannot find `llc' in executable directory or PATH!\n";
366 return 0;
367 }
368
369 Message = "Found llc: " + LLCPath + "\n";
Bill Wendlingbf5d8272009-03-02 23:13:18 +0000370 GCC *gcc = GCC::create(ProgramPath, Message, GCCArgs);
Chris Lattnerffac2862006-06-06 22:30:59 +0000371 if (!gcc) {
372 std::cerr << Message << "\n";
373 exit(1);
374 }
Bill Wendlingbf5d8272009-03-02 23:13:18 +0000375 return new LLC(LLCPath, gcc, Args, GCCArgs);
Chris Lattnerffac2862006-06-06 22:30:59 +0000376}
377
378//===---------------------------------------------------------------------===//
379// JIT Implementation of AbstractIntepreter interface
380//
381namespace {
382 class JIT : public AbstractInterpreter {
383 std::string LLIPath; // The path to the LLI executable
384 std::vector<std::string> ToolArgs; // Args to pass to LLI
385 public:
386 JIT(const std::string &Path, const std::vector<std::string> *Args)
387 : LLIPath(Path) {
388 ToolArgs.clear ();
389 if (Args) { ToolArgs = *Args; }
390 }
391
Gabor Greif0e535c3c2007-07-04 21:55:50 +0000392 virtual int ExecuteProgram(const std::string &Bitcode,
Chris Lattnerffac2862006-06-06 22:30:59 +0000393 const std::vector<std::string> &Args,
394 const std::string &InputFile,
395 const std::string &OutputFile,
396 const std::vector<std::string> &GCCArgs =
397 std::vector<std::string>(),
398 const std::vector<std::string> &SharedLibs =
399 std::vector<std::string>(),
Anton Korobeynikovd01defe2007-02-16 19:11:07 +0000400 unsigned Timeout =0,
401 unsigned MemoryLimit =0);
Chris Lattnerffac2862006-06-06 22:30:59 +0000402 };
403}
404
Gabor Greif0e535c3c2007-07-04 21:55:50 +0000405int JIT::ExecuteProgram(const std::string &Bitcode,
Chris Lattnerffac2862006-06-06 22:30:59 +0000406 const std::vector<std::string> &Args,
407 const std::string &InputFile,
408 const std::string &OutputFile,
409 const std::vector<std::string> &GCCArgs,
410 const std::vector<std::string> &SharedLibs,
Anton Korobeynikovd01defe2007-02-16 19:11:07 +0000411 unsigned Timeout,
412 unsigned MemoryLimit) {
Chris Lattnerffac2862006-06-06 22:30:59 +0000413 // Construct a vector of parameters, incorporating those from the command-line
414 std::vector<const char*> JITArgs;
415 JITArgs.push_back(LLIPath.c_str());
416 JITArgs.push_back("-force-interpreter=false");
417
418 // Add any extra LLI args.
419 for (unsigned i = 0, e = ToolArgs.size(); i != e; ++i)
420 JITArgs.push_back(ToolArgs[i].c_str());
421
422 for (unsigned i = 0, e = SharedLibs.size(); i != e; ++i) {
423 JITArgs.push_back("-load");
424 JITArgs.push_back(SharedLibs[i].c_str());
425 }
Gabor Greif0e535c3c2007-07-04 21:55:50 +0000426 JITArgs.push_back(Bitcode.c_str());
Chris Lattnerffac2862006-06-06 22:30:59 +0000427 // Add optional parameters to the running program from Argv
428 for (unsigned i=0, e = Args.size(); i != e; ++i)
429 JITArgs.push_back(Args[i].c_str());
430 JITArgs.push_back(0);
431
432 std::cout << "<jit>" << std::flush;
433 DEBUG(std::cerr << "\nAbout to run:\t";
434 for (unsigned i=0, e = JITArgs.size()-1; i != e; ++i)
435 std::cerr << " " << JITArgs[i];
436 std::cerr << "\n";
437 );
438 DEBUG(std::cerr << "\nSending output to " << OutputFile << "\n");
439 return RunProgramWithTimeout(sys::Path(LLIPath), &JITArgs[0],
440 sys::Path(InputFile), sys::Path(OutputFile), sys::Path(OutputFile),
Anton Korobeynikovd01defe2007-02-16 19:11:07 +0000441 Timeout, MemoryLimit);
Chris Lattnerffac2862006-06-06 22:30:59 +0000442}
443
444/// createJIT - Try to find the LLI executable
445///
446AbstractInterpreter *AbstractInterpreter::createJIT(const std::string &ProgPath,
447 std::string &Message, const std::vector<std::string> *Args) {
448 std::string LLIPath = FindExecutable("lli", ProgPath).toString();
449 if (!LLIPath.empty()) {
450 Message = "Found lli: " + LLIPath + "\n";
451 return new JIT(LLIPath, Args);
452 }
453
454 Message = "Cannot find `lli' in executable directory or PATH!\n";
455 return 0;
456}
457
Gabor Greif0e535c3c2007-07-04 21:55:50 +0000458GCC::FileType CBE::OutputCode(const std::string &Bitcode,
Chris Lattner634bc042006-09-15 21:29:15 +0000459 sys::Path &OutputCFile) {
Gabor Greif0e535c3c2007-07-04 21:55:50 +0000460 sys::Path uniqueFile(Bitcode+".cbe.c");
Reid Spencere4ca7222006-08-23 20:34:57 +0000461 std::string ErrMsg;
462 if (uniqueFile.makeUnique(true, &ErrMsg)) {
463 std::cerr << "Error making unique filename: " << ErrMsg << "\n";
464 exit(1);
465 }
Chris Lattnerffac2862006-06-06 22:30:59 +0000466 OutputCFile = uniqueFile;
467 std::vector<const char *> LLCArgs;
468 LLCArgs.push_back (LLCPath.c_str());
469
470 // Add any extra LLC args.
471 for (unsigned i = 0, e = ToolArgs.size(); i != e; ++i)
472 LLCArgs.push_back(ToolArgs[i].c_str());
473
474 LLCArgs.push_back ("-o");
475 LLCArgs.push_back (OutputCFile.c_str()); // Output to the C file
476 LLCArgs.push_back ("-march=c"); // Output C language
477 LLCArgs.push_back ("-f"); // Overwrite as necessary...
Gabor Greif0e535c3c2007-07-04 21:55:50 +0000478 LLCArgs.push_back (Bitcode.c_str()); // This is the input bitcode
Chris Lattnerffac2862006-06-06 22:30:59 +0000479 LLCArgs.push_back (0);
480
481 std::cout << "<cbe>" << std::flush;
482 DEBUG(std::cerr << "\nAbout to run:\t";
483 for (unsigned i=0, e = LLCArgs.size()-1; i != e; ++i)
484 std::cerr << " " << LLCArgs[i];
485 std::cerr << "\n";
486 );
487 if (RunProgramWithTimeout(LLCPath, &LLCArgs[0], sys::Path(), sys::Path(),
488 sys::Path()))
489 ProcessFailure(LLCPath, &LLCArgs[0]);
Chris Lattner634bc042006-09-15 21:29:15 +0000490 return GCC::CFile;
Chris Lattnerffac2862006-06-06 22:30:59 +0000491}
492
Gabor Greif0e535c3c2007-07-04 21:55:50 +0000493void CBE::compileProgram(const std::string &Bitcode) {
Chris Lattnerffac2862006-06-06 22:30:59 +0000494 sys::Path OutputCFile;
Gabor Greif0e535c3c2007-07-04 21:55:50 +0000495 OutputCode(Bitcode, OutputCFile);
Chris Lattnerffac2862006-06-06 22:30:59 +0000496 OutputCFile.eraseFromDisk();
497}
498
Gabor Greif0e535c3c2007-07-04 21:55:50 +0000499int CBE::ExecuteProgram(const std::string &Bitcode,
Chris Lattnerffac2862006-06-06 22:30:59 +0000500 const std::vector<std::string> &Args,
501 const std::string &InputFile,
502 const std::string &OutputFile,
503 const std::vector<std::string> &ArgsForGCC,
504 const std::vector<std::string> &SharedLibs,
Anton Korobeynikovd01defe2007-02-16 19:11:07 +0000505 unsigned Timeout,
506 unsigned MemoryLimit) {
Chris Lattnerffac2862006-06-06 22:30:59 +0000507 sys::Path OutputCFile;
Gabor Greif0e535c3c2007-07-04 21:55:50 +0000508 OutputCode(Bitcode, OutputCFile);
Chris Lattnerffac2862006-06-06 22:30:59 +0000509
510 FileRemover CFileRemove(OutputCFile);
511
512 std::vector<std::string> GCCArgs(ArgsForGCC);
Bill Wendlingbf5d8272009-03-02 23:13:18 +0000513 GCCArgs.insert(GCCArgs.end(), SharedLibs.begin(), SharedLibs.end());
514
Chris Lattnerffac2862006-06-06 22:30:59 +0000515 return gcc->ExecuteProgram(OutputCFile.toString(), Args, GCC::CFile,
Anton Korobeynikovd01defe2007-02-16 19:11:07 +0000516 InputFile, OutputFile, GCCArgs,
517 Timeout, MemoryLimit);
Chris Lattnerffac2862006-06-06 22:30:59 +0000518}
519
520/// createCBE - Try to find the 'llc' executable
521///
522CBE *AbstractInterpreter::createCBE(const std::string &ProgramPath,
523 std::string &Message,
Bill Wendlingbf5d8272009-03-02 23:13:18 +0000524 const std::vector<std::string> *Args,
525 const std::vector<std::string> *GCCArgs) {
Chris Lattnerffac2862006-06-06 22:30:59 +0000526 sys::Path LLCPath = FindExecutable("llc", ProgramPath);
527 if (LLCPath.isEmpty()) {
528 Message =
529 "Cannot find `llc' in executable directory or PATH!\n";
530 return 0;
531 }
532
533 Message = "Found llc: " + LLCPath.toString() + "\n";
Bill Wendlingbf5d8272009-03-02 23:13:18 +0000534 GCC *gcc = GCC::create(ProgramPath, Message, GCCArgs);
Chris Lattnerffac2862006-06-06 22:30:59 +0000535 if (!gcc) {
536 std::cerr << Message << "\n";
537 exit(1);
538 }
539 return new CBE(LLCPath, gcc, Args);
540}
541
542//===---------------------------------------------------------------------===//
543// GCC abstraction
544//
David Goodwin73f4e3f2009-07-10 21:39:28 +0000545
Daniel Dunbar9dc326f2009-07-11 21:24:52 +0000546#ifdef __APPLE__
David Goodwin73f4e3f2009-07-10 21:39:28 +0000547static bool
548IsARMArchitecture(std::vector<std::string> Args)
549{
550 for (std::vector<std::string>::const_iterator
551 I = Args.begin(), E = Args.end(); I != E; ++I) {
552 if (!strcasecmp(I->c_str(), "-arch")) {
553 ++I;
554 if ((I != E) && !strncasecmp(I->c_str(), "arm", strlen("arm"))) {
555 return true;
556 }
557 }
558 }
559
560 return false;
561}
Daniel Dunbar9dc326f2009-07-11 21:24:52 +0000562#endif
David Goodwin73f4e3f2009-07-10 21:39:28 +0000563
Chris Lattnerffac2862006-06-06 22:30:59 +0000564int GCC::ExecuteProgram(const std::string &ProgramFile,
565 const std::vector<std::string> &Args,
566 FileType fileType,
567 const std::string &InputFile,
568 const std::string &OutputFile,
569 const std::vector<std::string> &ArgsForGCC,
Anton Korobeynikovd01defe2007-02-16 19:11:07 +0000570 unsigned Timeout,
571 unsigned MemoryLimit) {
Chris Lattnerffac2862006-06-06 22:30:59 +0000572 std::vector<const char*> GCCArgs;
573
574 GCCArgs.push_back(GCCPath.c_str());
575
Bill Wendlingbf5d8272009-03-02 23:13:18 +0000576 for (std::vector<std::string>::const_iterator
577 I = gccArgs.begin(), E = gccArgs.end(); I != E; ++I)
578 GCCArgs.push_back(I->c_str());
579
Chris Lattnerffac2862006-06-06 22:30:59 +0000580 // Specify -x explicitly in case the extension is wonky
581 GCCArgs.push_back("-x");
582 if (fileType == CFile) {
583 GCCArgs.push_back("c");
584 GCCArgs.push_back("-fno-strict-aliasing");
585 } else {
586 GCCArgs.push_back("assembler");
587#ifdef __APPLE__
David Goodwin73f4e3f2009-07-10 21:39:28 +0000588 // For ARM architectures we don't want this flag. bugpoint isn't
589 // explicitly told what architecture it is working on, so we get
590 // it from gcc flags
591 if (!IsARMArchitecture(ArgsForGCC))
592 GCCArgs.push_back("-force_cpusubtype_ALL");
Chris Lattnerffac2862006-06-06 22:30:59 +0000593#endif
594 }
595 GCCArgs.push_back(ProgramFile.c_str()); // Specify the input filename...
Chris Lattner572e78c2006-06-09 21:31:53 +0000596 GCCArgs.push_back("-x");
597 GCCArgs.push_back("none");
Chris Lattnerffac2862006-06-06 22:30:59 +0000598 GCCArgs.push_back("-o");
599 sys::Path OutputBinary (ProgramFile+".gcc.exe");
Reid Spencere4ca7222006-08-23 20:34:57 +0000600 std::string ErrMsg;
601 if (OutputBinary.makeUnique(true, &ErrMsg)) {
602 std::cerr << "Error making unique filename: " << ErrMsg << "\n";
603 exit(1);
604 }
Chris Lattnerffac2862006-06-06 22:30:59 +0000605 GCCArgs.push_back(OutputBinary.c_str()); // Output to the right file...
606
607 // Add any arguments intended for GCC. We locate them here because this is
608 // most likely -L and -l options that need to come before other libraries but
609 // after the source. Other options won't be sensitive to placement on the
610 // command line, so this should be safe.
611 for (unsigned i = 0, e = ArgsForGCC.size(); i != e; ++i)
612 GCCArgs.push_back(ArgsForGCC[i].c_str());
613
614 GCCArgs.push_back("-lm"); // Hard-code the math library...
615 GCCArgs.push_back("-O2"); // Optimize the program a bit...
616#if defined (HAVE_LINK_R)
617 GCCArgs.push_back("-Wl,-R."); // Search this dir for .so files
618#endif
619#ifdef __sparc__
620 GCCArgs.push_back("-mcpu=v9");
621#endif
622 GCCArgs.push_back(0); // NULL terminator
623
624 std::cout << "<gcc>" << std::flush;
Evan Chenga50e3912007-01-03 07:44:30 +0000625 DEBUG(std::cerr << "\nAbout to run:\t";
626 for (unsigned i=0, e = GCCArgs.size()-1; i != e; ++i)
627 std::cerr << " " << GCCArgs[i];
628 std::cerr << "\n";
629 );
Chris Lattnerffac2862006-06-06 22:30:59 +0000630 if (RunProgramWithTimeout(GCCPath, &GCCArgs[0], sys::Path(), sys::Path(),
631 sys::Path())) {
632 ProcessFailure(GCCPath, &GCCArgs[0]);
633 exit(1);
634 }
635
636 std::vector<const char*> ProgramArgs;
637
Viktor Kutuzov1518de12009-07-14 19:10:55 +0000638 std::string Exec;
Evan Cheng7aedcf12008-09-09 06:11:26 +0000639 if (RemoteClientPath.isEmpty())
Evan Cheng779b52e2007-05-03 18:36:15 +0000640 ProgramArgs.push_back(OutputBinary.c_str());
641 else {
Evan Cheng7aedcf12008-09-09 06:11:26 +0000642 ProgramArgs.push_back(RemoteClientPath.c_str());
643 ProgramArgs.push_back(RemoteHost.c_str());
Evan Cheng779b52e2007-05-03 18:36:15 +0000644 ProgramArgs.push_back("-l");
Evan Cheng7aedcf12008-09-09 06:11:26 +0000645 ProgramArgs.push_back(RemoteUser.c_str());
David Goodwin73f4e3f2009-07-10 21:39:28 +0000646 if (!RemotePort.empty()) {
647 ProgramArgs.push_back("-p");
648 ProgramArgs.push_back(RemotePort.c_str());
649 }
Evan Cheng7aedcf12008-09-09 06:11:26 +0000650 if (!RemoteExtra.empty()) {
651 ProgramArgs.push_back(RemoteExtra.c_str());
652 }
Evan Cheng779b52e2007-05-03 18:36:15 +0000653
Viktor Kutuzov1518de12009-07-14 19:10:55 +0000654 // Full path to the binary
Evan Cheng779b52e2007-05-03 18:36:15 +0000655 char* env_pwd = getenv("PWD");
Evan Cheng779b52e2007-05-03 18:36:15 +0000656 Exec += env_pwd;
Viktor Kutuzov1518de12009-07-14 19:10:55 +0000657 Exec += "/";
Evan Cheng779b52e2007-05-03 18:36:15 +0000658 Exec += OutputBinary.c_str();
659 ProgramArgs.push_back(Exec.c_str());
660 }
661
Chris Lattnerffac2862006-06-06 22:30:59 +0000662 // Add optional parameters to the running program from Argv
663 for (unsigned i=0, e = Args.size(); i != e; ++i)
664 ProgramArgs.push_back(Args[i].c_str());
665 ProgramArgs.push_back(0); // NULL terminator
666
667 // Now that we have a binary, run it!
668 std::cout << "<program>" << std::flush;
669 DEBUG(std::cerr << "\nAbout to run:\t";
670 for (unsigned i=0, e = ProgramArgs.size()-1; i != e; ++i)
671 std::cerr << " " << ProgramArgs[i];
672 std::cerr << "\n";
673 );
674
675 FileRemover OutputBinaryRemover(OutputBinary);
Evan Cheng779b52e2007-05-03 18:36:15 +0000676
Viktor Kutuzov1518de12009-07-14 19:10:55 +0000677 if (RemoteClientPath.isEmpty()) {
678 DEBUG(std::cerr << "<run locally>" << std::flush;);
Evan Cheng779b52e2007-05-03 18:36:15 +0000679 return RunProgramWithTimeout(OutputBinary, &ProgramArgs[0],
680 sys::Path(InputFile), sys::Path(OutputFile), sys::Path(OutputFile),
681 Timeout, MemoryLimit);
Viktor Kutuzov1518de12009-07-14 19:10:55 +0000682 } else {
683 std::cout << "<run remotely>" << std::flush;
684 int RemoteClientStatus = RunProgramWithTimeout(sys::Path(RemoteClientPath),
685 &ProgramArgs[0], sys::Path(InputFile), sys::Path(OutputFile),
686 sys::Path(OutputFile), Timeout, MemoryLimit);
687 if (RemoteClientStatus != 0) {
688 std::cerr << "Remote Client failed with an error: " <<
689 RemoteClientStatus << ".\n" << std::flush;
690 }
691 }
Chris Lattnerffac2862006-06-06 22:30:59 +0000692}
693
694int GCC::MakeSharedObject(const std::string &InputFile, FileType fileType,
Chris Lattnercc21fa72006-06-27 20:35:36 +0000695 std::string &OutputFile,
696 const std::vector<std::string> &ArgsForGCC) {
Chris Lattnerffac2862006-06-06 22:30:59 +0000697 sys::Path uniqueFilename(InputFile+LTDL_SHLIB_EXT);
Reid Spencere4ca7222006-08-23 20:34:57 +0000698 std::string ErrMsg;
699 if (uniqueFilename.makeUnique(true, &ErrMsg)) {
700 std::cerr << "Error making unique filename: " << ErrMsg << "\n";
701 exit(1);
702 }
Chris Lattnerffac2862006-06-06 22:30:59 +0000703 OutputFile = uniqueFilename.toString();
704
Chris Lattnercc21fa72006-06-27 20:35:36 +0000705 std::vector<const char*> GCCArgs;
706
707 GCCArgs.push_back(GCCPath.c_str());
Evan Cheng690b6352009-03-12 00:53:34 +0000708
709 for (std::vector<std::string>::const_iterator
710 I = gccArgs.begin(), E = gccArgs.end(); I != E; ++I)
711 GCCArgs.push_back(I->c_str());
712
Chris Lattnerffac2862006-06-06 22:30:59 +0000713 // Compile the C/asm file into a shared object
Chris Lattnercc21fa72006-06-27 20:35:36 +0000714 GCCArgs.push_back("-x");
715 GCCArgs.push_back(fileType == AsmFile ? "assembler" : "c");
716 GCCArgs.push_back("-fno-strict-aliasing");
717 GCCArgs.push_back(InputFile.c_str()); // Specify the input filename.
Lauro Ramos Venanciof04823a2007-06-06 23:10:56 +0000718 GCCArgs.push_back("-x");
719 GCCArgs.push_back("none");
Chris Lattnerffac2862006-06-06 22:30:59 +0000720#if defined(sparc) || defined(__sparc__) || defined(__sparcv9)
Chris Lattnercc21fa72006-06-27 20:35:36 +0000721 GCCArgs.push_back("-G"); // Compile a shared library, `-G' for Sparc
Chris Lattnerffac2862006-06-06 22:30:59 +0000722#elif defined(__APPLE__)
Chris Lattnercc21fa72006-06-27 20:35:36 +0000723 // link all source files into a single module in data segment, rather than
724 // generating blocks. dynamic_lookup requires that you set
725 // MACOSX_DEPLOYMENT_TARGET=10.3 in your env. FIXME: it would be better for
726 // bugpoint to just pass that in the environment of GCC.
727 GCCArgs.push_back("-single_module");
728 GCCArgs.push_back("-dynamiclib"); // `-dynamiclib' for MacOS X/PowerPC
729 GCCArgs.push_back("-undefined");
730 GCCArgs.push_back("dynamic_lookup");
Chris Lattnerffac2862006-06-06 22:30:59 +0000731#else
Chris Lattnercc21fa72006-06-27 20:35:36 +0000732 GCCArgs.push_back("-shared"); // `-shared' for Linux/X86, maybe others
Chris Lattnerffac2862006-06-06 22:30:59 +0000733#endif
734
Torok Edwin538160b2008-04-06 12:42:29 +0000735#if defined(__ia64__) || defined(__alpha__) || defined(__amd64__)
Chris Lattnercc21fa72006-06-27 20:35:36 +0000736 GCCArgs.push_back("-fPIC"); // Requires shared objs to contain PIC
Chris Lattnerffac2862006-06-06 22:30:59 +0000737#endif
738#ifdef __sparc__
Chris Lattnercc21fa72006-06-27 20:35:36 +0000739 GCCArgs.push_back("-mcpu=v9");
Chris Lattnerffac2862006-06-06 22:30:59 +0000740#endif
Chris Lattnercc21fa72006-06-27 20:35:36 +0000741 GCCArgs.push_back("-o");
742 GCCArgs.push_back(OutputFile.c_str()); // Output to the right filename.
743 GCCArgs.push_back("-O2"); // Optimize the program a bit.
744
745
746
747 // Add any arguments intended for GCC. We locate them here because this is
748 // most likely -L and -l options that need to come before other libraries but
749 // after the source. Other options won't be sensitive to placement on the
750 // command line, so this should be safe.
751 for (unsigned i = 0, e = ArgsForGCC.size(); i != e; ++i)
752 GCCArgs.push_back(ArgsForGCC[i].c_str());
753 GCCArgs.push_back(0); // NULL terminator
754
755
Chris Lattnerffac2862006-06-06 22:30:59 +0000756
757 std::cout << "<gcc>" << std::flush;
Evan Chenga50e3912007-01-03 07:44:30 +0000758 DEBUG(std::cerr << "\nAbout to run:\t";
759 for (unsigned i=0, e = GCCArgs.size()-1; i != e; ++i)
760 std::cerr << " " << GCCArgs[i];
761 std::cerr << "\n";
762 );
Chris Lattnercc21fa72006-06-27 20:35:36 +0000763 if (RunProgramWithTimeout(GCCPath, &GCCArgs[0], sys::Path(), sys::Path(),
Chris Lattnerffac2862006-06-06 22:30:59 +0000764 sys::Path())) {
Chris Lattnercc21fa72006-06-27 20:35:36 +0000765 ProcessFailure(GCCPath, &GCCArgs[0]);
Chris Lattnerffac2862006-06-06 22:30:59 +0000766 return 1;
767 }
768 return 0;
769}
770
771/// create - Try to find the `gcc' executable
772///
Bill Wendlingbf5d8272009-03-02 23:13:18 +0000773GCC *GCC::create(const std::string &ProgramPath, std::string &Message,
774 const std::vector<std::string> *Args) {
Chris Lattnerffac2862006-06-06 22:30:59 +0000775 sys::Path GCCPath = FindExecutable("gcc", ProgramPath);
776 if (GCCPath.isEmpty()) {
777 Message = "Cannot find `gcc' in executable directory or PATH!\n";
778 return 0;
779 }
780
Evan Cheng7aedcf12008-09-09 06:11:26 +0000781 sys::Path RemoteClientPath;
782 if (!RemoteClient.empty())
783 RemoteClientPath = FindExecutable(RemoteClient.c_str(), ProgramPath);
Evan Cheng779b52e2007-05-03 18:36:15 +0000784
Chris Lattnerffac2862006-06-06 22:30:59 +0000785 Message = "Found gcc: " + GCCPath.toString() + "\n";
Bill Wendlingbf5d8272009-03-02 23:13:18 +0000786 return new GCC(GCCPath, RemoteClientPath, Args);
Chris Lattnerffac2862006-06-06 22:30:59 +0000787}