blob: 503a5dda136a62551532c815d71a84d54e0b0395 [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>
23#include <iostream>
24using namespace llvm;
25
26namespace {
27 cl::opt<std::string>
Evan Cheng0646fff2008-09-09 06:11:26 +000028 RemoteClient("remote-client",
29 cl::desc("Remote execution client (rsh/ssh)"));
Dan Gohmanf17a25c2007-07-18 16:29:46 +000030
31 cl::opt<std::string>
Evan Cheng0646fff2008-09-09 06:11:26 +000032 RemoteHost("remote-host",
33 cl::desc("Remote execution (rsh/ssh) host"));
34
35 cl::opt<std::string>
David Goodwinec1edba2009-07-10 21:39:28 +000036 RemotePort("remote-port",
37 cl::desc("Remote execution (rsh/ssh) port"));
38
39 cl::opt<std::string>
Evan Cheng0646fff2008-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"));
Dan Gohmanf17a25c2007-07-18 16:29:46 +000046}
47
48ToolExecutionError::~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,
58 unsigned NumSeconds = 0,
59 unsigned MemoryLimit = 0) {
60 const sys::Path* redirects[3];
61 redirects[0] = &StdInFile;
62 redirects[1] = &StdOutFile;
63 redirects[2] = &StdErrFile;
64
65 if (0) {
66 std::cerr << "RUN:";
67 for (unsigned i = 0; Args[i]; ++i)
68 std::cerr << " " << Args[i];
69 std::cerr << "\n";
70 }
71
72 return
73 sys::Program::ExecuteAndWait(ProgramPath, Args, 0, redirects,
74 NumSeconds, MemoryLimit);
75}
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 Gohman3c814b32009-02-12 20:53:27 +000087 sys::Path ErrorFilename("bugpoint.program_error_messages");
Dan Gohmanf17a25c2007-07-18 16:29:46 +000088 std::string ErrMsg;
89 if (ErrorFilename.makeUnique(true, &ErrMsg)) {
90 std::cerr << "Error making unique filename: " << ErrMsg << "\n";
91 exit(1);
92 }
93 RunProgramWithTimeout(ProgPath, Args, sys::Path(""), ErrorFilename,
94 ErrorFilename); // FIXME: check return code ?
95
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
123 virtual int ExecuteProgram(const std::string &Bitcode,
124 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>(),
130 unsigned Timeout = 0,
131 unsigned MemoryLimit = 0);
132 };
133}
134
135int LLI::ExecuteProgram(const std::string &Bitcode,
136 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,
141 unsigned Timeout,
142 unsigned MemoryLimit) {
143 if (!SharedLibs.empty())
144 throw ToolExecutionError("LLI currently does not support "
145 "loading shared libraries.");
146
Dan Gohmanf17a25c2007-07-18 16:29:46 +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
155 LLIArgs.push_back(Bitcode.c_str());
156 // 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),
169 Timeout, MemoryLimit);
170}
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 Korobeynikovfc0116f2008-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
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000290//===----------------------------------------------------------------------===//
291// LLC Implementation of AbstractIntepreter interface
292//
293GCC::FileType LLC::OutputCode(const std::string &Bitcode,
294 sys::Path &OutputAsmFile) {
295 sys::Path uniqueFile(Bitcode+".llc.s");
296 std::string ErrMsg;
297 if (uniqueFile.makeUnique(true, &ErrMsg)) {
298 std::cerr << "Error making unique filename: " << ErrMsg << "\n";
299 exit(1);
300 }
301 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...
312 LLCArgs.push_back (Bitcode.c_str()); // This is the input bitcode
313 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]);
324
325 return GCC::AsmFile;
326}
327
328void LLC::compileProgram(const std::string &Bitcode) {
329 sys::Path OutputAsmFile;
330 OutputCode(Bitcode, OutputAsmFile);
331 OutputAsmFile.eraseFromDisk();
332}
333
334int LLC::ExecuteProgram(const std::string &Bitcode,
335 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,
340 unsigned Timeout,
341 unsigned MemoryLimit) {
342
343 sys::Path OutputAsmFile;
344 OutputCode(Bitcode, OutputAsmFile);
345 FileRemover OutFileRemover(OutputAsmFile);
346
347 std::vector<std::string> GCCArgs(ArgsForGCC);
Bill Wendlingcd1389d2009-03-02 23:13:18 +0000348 GCCArgs.insert(GCCArgs.end(), SharedLibs.begin(), SharedLibs.end());
349 GCCArgs.insert(GCCArgs.end(), gccArgs.begin(), gccArgs.end());
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000350
351 // Assuming LLC worked, compile the result with GCC and run it.
352 return gcc->ExecuteProgram(OutputAsmFile.toString(), Args, GCC::AsmFile,
353 InputFile, OutputFile, GCCArgs,
354 Timeout, MemoryLimit);
355}
356
357/// createLLC - Try to find the LLC executable
358///
359LLC *AbstractInterpreter::createLLC(const std::string &ProgramPath,
360 std::string &Message,
Bill Wendlingcd1389d2009-03-02 23:13:18 +0000361 const std::vector<std::string> *Args,
362 const std::vector<std::string> *GCCArgs) {
Dan Gohmanf17a25c2007-07-18 16:29:46 +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 Wendlingcd1389d2009-03-02 23:13:18 +0000370 GCC *gcc = GCC::create(ProgramPath, Message, GCCArgs);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000371 if (!gcc) {
372 std::cerr << Message << "\n";
373 exit(1);
374 }
Bill Wendlingcd1389d2009-03-02 23:13:18 +0000375 return new LLC(LLCPath, gcc, Args, GCCArgs);
Dan Gohmanf17a25c2007-07-18 16:29:46 +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
392 virtual int 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> &GCCArgs =
397 std::vector<std::string>(),
398 const std::vector<std::string> &SharedLibs =
399 std::vector<std::string>(),
400 unsigned Timeout =0,
401 unsigned MemoryLimit =0);
402 };
403}
404
405int JIT::ExecuteProgram(const std::string &Bitcode,
406 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,
411 unsigned Timeout,
412 unsigned MemoryLimit) {
Dan Gohmanf17a25c2007-07-18 16:29:46 +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 }
426 JITArgs.push_back(Bitcode.c_str());
427 // 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),
441 Timeout, MemoryLimit);
442}
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
458GCC::FileType CBE::OutputCode(const std::string &Bitcode,
459 sys::Path &OutputCFile) {
460 sys::Path uniqueFile(Bitcode+".cbe.c");
461 std::string ErrMsg;
462 if (uniqueFile.makeUnique(true, &ErrMsg)) {
463 std::cerr << "Error making unique filename: " << ErrMsg << "\n";
464 exit(1);
465 }
466 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...
478 LLCArgs.push_back (Bitcode.c_str()); // This is the input bitcode
479 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]);
490 return GCC::CFile;
491}
492
493void CBE::compileProgram(const std::string &Bitcode) {
494 sys::Path OutputCFile;
495 OutputCode(Bitcode, OutputCFile);
496 OutputCFile.eraseFromDisk();
497}
498
499int CBE::ExecuteProgram(const std::string &Bitcode,
500 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,
505 unsigned Timeout,
506 unsigned MemoryLimit) {
507 sys::Path OutputCFile;
508 OutputCode(Bitcode, OutputCFile);
509
510 FileRemover CFileRemove(OutputCFile);
511
512 std::vector<std::string> GCCArgs(ArgsForGCC);
Bill Wendlingcd1389d2009-03-02 23:13:18 +0000513 GCCArgs.insert(GCCArgs.end(), SharedLibs.begin(), SharedLibs.end());
514
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000515 return gcc->ExecuteProgram(OutputCFile.toString(), Args, GCC::CFile,
516 InputFile, OutputFile, GCCArgs,
517 Timeout, MemoryLimit);
518}
519
520/// createCBE - Try to find the 'llc' executable
521///
522CBE *AbstractInterpreter::createCBE(const std::string &ProgramPath,
523 std::string &Message,
Bill Wendlingcd1389d2009-03-02 23:13:18 +0000524 const std::vector<std::string> *Args,
525 const std::vector<std::string> *GCCArgs) {
Dan Gohmanf17a25c2007-07-18 16:29:46 +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 Wendlingcd1389d2009-03-02 23:13:18 +0000534 GCC *gcc = GCC::create(ProgramPath, Message, GCCArgs);
Dan Gohmanf17a25c2007-07-18 16:29:46 +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 Goodwinec1edba2009-07-10 21:39:28 +0000545
Daniel Dunbar15a1bdf2009-07-11 21:24:52 +0000546#ifdef __APPLE__
David Goodwinec1edba2009-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 Dunbar15a1bdf2009-07-11 21:24:52 +0000562#endif
David Goodwinec1edba2009-07-10 21:39:28 +0000563
Dan Gohmanf17a25c2007-07-18 16:29:46 +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,
570 unsigned Timeout,
571 unsigned MemoryLimit) {
572 std::vector<const char*> GCCArgs;
573
574 GCCArgs.push_back(GCCPath.c_str());
575
Bill Wendlingcd1389d2009-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
Dan Gohmanf17a25c2007-07-18 16:29:46 +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 Goodwinec1edba2009-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");
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000593#endif
594 }
595 GCCArgs.push_back(ProgramFile.c_str()); // Specify the input filename...
596 GCCArgs.push_back("-x");
597 GCCArgs.push_back("none");
598 GCCArgs.push_back("-o");
599 sys::Path OutputBinary (ProgramFile+".gcc.exe");
600 std::string ErrMsg;
601 if (OutputBinary.makeUnique(true, &ErrMsg)) {
602 std::cerr << "Error making unique filename: " << ErrMsg << "\n";
603 exit(1);
604 }
605 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;
625 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 );
630 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 Kutuzov8c3e63c2009-07-14 19:10:55 +0000638 std::string Exec;
Evan Cheng0646fff2008-09-09 06:11:26 +0000639 if (RemoteClientPath.isEmpty())
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000640 ProgramArgs.push_back(OutputBinary.c_str());
641 else {
Evan Cheng0646fff2008-09-09 06:11:26 +0000642 ProgramArgs.push_back(RemoteClientPath.c_str());
643 ProgramArgs.push_back(RemoteHost.c_str());
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000644 ProgramArgs.push_back("-l");
Evan Cheng0646fff2008-09-09 06:11:26 +0000645 ProgramArgs.push_back(RemoteUser.c_str());
David Goodwinec1edba2009-07-10 21:39:28 +0000646 if (!RemotePort.empty()) {
647 ProgramArgs.push_back("-p");
648 ProgramArgs.push_back(RemotePort.c_str());
649 }
Evan Cheng0646fff2008-09-09 06:11:26 +0000650 if (!RemoteExtra.empty()) {
651 ProgramArgs.push_back(RemoteExtra.c_str());
652 }
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000653
Viktor Kutuzov8c3e63c2009-07-14 19:10:55 +0000654 // Full path to the binary
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000655 char* env_pwd = getenv("PWD");
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000656 Exec += env_pwd;
Viktor Kutuzov8c3e63c2009-07-14 19:10:55 +0000657 Exec += "/";
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000658 Exec += OutputBinary.c_str();
659 ProgramArgs.push_back(Exec.c_str());
660 }
661
662 // 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);
676
Viktor Kutuzov8c3e63c2009-07-14 19:10:55 +0000677 if (RemoteClientPath.isEmpty()) {
678 DEBUG(std::cerr << "<run locally>" << std::flush;);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000679 return RunProgramWithTimeout(OutputBinary, &ProgramArgs[0],
680 sys::Path(InputFile), sys::Path(OutputFile), sys::Path(OutputFile),
681 Timeout, MemoryLimit);
Viktor Kutuzov8c3e63c2009-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 }
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000692}
693
694int GCC::MakeSharedObject(const std::string &InputFile, FileType fileType,
695 std::string &OutputFile,
696 const std::vector<std::string> &ArgsForGCC) {
697 sys::Path uniqueFilename(InputFile+LTDL_SHLIB_EXT);
698 std::string ErrMsg;
699 if (uniqueFilename.makeUnique(true, &ErrMsg)) {
700 std::cerr << "Error making unique filename: " << ErrMsg << "\n";
701 exit(1);
702 }
703 OutputFile = uniqueFilename.toString();
704
705 std::vector<const char*> GCCArgs;
706
707 GCCArgs.push_back(GCCPath.c_str());
Evan Chengf94bcf62009-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
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000713 // Compile the C/asm file into a shared object
714 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.
718 GCCArgs.push_back("-x");
719 GCCArgs.push_back("none");
720#if defined(sparc) || defined(__sparc__) || defined(__sparcv9)
721 GCCArgs.push_back("-G"); // Compile a shared library, `-G' for Sparc
722#elif defined(__APPLE__)
723 // 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");
731#else
732 GCCArgs.push_back("-shared"); // `-shared' for Linux/X86, maybe others
733#endif
734
Edwin Török35594fa2008-04-06 12:42:29 +0000735#if defined(__ia64__) || defined(__alpha__) || defined(__amd64__)
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000736 GCCArgs.push_back("-fPIC"); // Requires shared objs to contain PIC
737#endif
738#ifdef __sparc__
739 GCCArgs.push_back("-mcpu=v9");
740#endif
741 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
756
757 std::cout << "<gcc>" << std::flush;
758 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 );
763 if (RunProgramWithTimeout(GCCPath, &GCCArgs[0], sys::Path(), sys::Path(),
764 sys::Path())) {
765 ProcessFailure(GCCPath, &GCCArgs[0]);
766 return 1;
767 }
768 return 0;
769}
770
771/// create - Try to find the `gcc' executable
772///
Bill Wendlingcd1389d2009-03-02 23:13:18 +0000773GCC *GCC::create(const std::string &ProgramPath, std::string &Message,
774 const std::vector<std::string> *Args) {
Dan Gohmanf17a25c2007-07-18 16:29:46 +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 Cheng0646fff2008-09-09 06:11:26 +0000781 sys::Path RemoteClientPath;
782 if (!RemoteClient.empty())
783 RemoteClientPath = FindExecutable(RemoteClient.c_str(), ProgramPath);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000784
785 Message = "Found gcc: " + GCCPath.toString() + "\n";
Bill Wendlingcd1389d2009-03-02 23:13:18 +0000786 return new GCC(GCCPath, RemoteClientPath, Args);
Dan Gohmanf17a25c2007-07-18 16:29:46 +0000787}