blob: 81314a85c389d7c85fd21f1ce69ae867a02b8b0d [file] [log] [blame]
Chris Lattner7915a1e2003-10-14 21:34:11 +00001//===-- ToolRunner.cpp ----------------------------------------------------===//
Misha Brukmanf976c852005-04-21 22:55:34 +00002//
John Criswellb576c942003-10-20 19:43:21 +00003// The LLVM Compiler Infrastructure
4//
Chris Lattner21c62da2007-12-29 20:44:31 +00005// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
Misha Brukmanf976c852005-04-21 22:55:34 +00007//
John Criswellb576c942003-10-20 19:43:21 +00008//===----------------------------------------------------------------------===//
Chris Lattner7915a1e2003-10-14 21:34:11 +00009//
10// This file implements the interfaces described in the ToolRunner.h file.
11//
12//===----------------------------------------------------------------------===//
13
Chris Lattner0b1fe842003-10-19 02:27:40 +000014#define DEBUG_TYPE "toolrunner"
Chris Lattnerf1b20d82006-06-06 22:30:59 +000015#include "ToolRunner.h"
Reid Spencer551ccae2004-09-01 22:55:40 +000016#include "llvm/Config/config.h" // for HAVE_LINK_R
Chris Lattner45495c52005-02-13 23:13:47 +000017#include "llvm/System/Program.h"
Evan Cheng34e400e2007-05-03 18:36:15 +000018#include "llvm/Support/CommandLine.h"
Reid Spencer551ccae2004-09-01 22:55:40 +000019#include "llvm/Support/Debug.h"
20#include "llvm/Support/FileUtilities.h"
Chris Lattner7915a1e2003-10-14 21:34:11 +000021#include <fstream>
Chris Lattner89bf9ea2004-02-18 20:38:00 +000022#include <sstream>
Chris Lattner2cdd21c2003-12-14 21:35:53 +000023using namespace llvm;
Brian Gaeked0fde302003-11-11 22:41:34 +000024
Anton Korobeynikov86c006a2009-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
Evan Cheng34e400e2007-05-03 18:36:15 +000030namespace {
31 cl::opt<std::string>
Evan Cheng70f684f2008-09-09 06:11:26 +000032 RemoteClient("remote-client",
33 cl::desc("Remote execution client (rsh/ssh)"));
Evan Cheng34e400e2007-05-03 18:36:15 +000034
35 cl::opt<std::string>
Evan Cheng70f684f2008-09-09 06:11:26 +000036 RemoteHost("remote-host",
37 cl::desc("Remote execution (rsh/ssh) host"));
38
39 cl::opt<std::string>
David Goodwin80becf12009-07-10 21:39:28 +000040 RemotePort("remote-port",
41 cl::desc("Remote execution (rsh/ssh) port"));
42
43 cl::opt<std::string>
Evan Cheng70f684f2008-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"));
Evan Cheng34e400e2007-05-03 18:36:15 +000050}
51
Alkis Evlogimenos1d29a6d2004-02-19 07:39:26 +000052ToolExecutionError::~ToolExecutionError() throw() { }
53
Viktor Kutuzovfc2271f2009-07-18 18:39:24 +000054/// RunProgramWithTimeout - This function provides an alternate interface
55/// to the sys::Program::ExecuteAndWait interface.
Chris Lattner45495c52005-02-13 23:13:47 +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,
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +000062 unsigned NumSeconds = 0,
63 unsigned MemoryLimit = 0) {
Chris Lattner45495c52005-02-13 23:13:47 +000064 const sys::Path* redirects[3];
65 redirects[0] = &StdInFile;
66 redirects[1] = &StdOutFile;
67 redirects[2] = &StdErrFile;
Viktor Kutuzovfc2271f2009-07-18 18:39:24 +000068
69#if 0 // For debug purposes
70 {
Dan Gohman65f57c22009-07-15 16:35:29 +000071 errs() << "RUN:";
Chris Lattnerd4223502006-09-15 23:01:10 +000072 for (unsigned i = 0; Args[i]; ++i)
Dan Gohman65f57c22009-07-15 16:35:29 +000073 errs() << " " << Args[i];
74 errs() << "\n";
Chris Lattnerd4223502006-09-15 23:01:10 +000075 }
Viktor Kutuzovfc2271f2009-07-18 18:39:24 +000076#endif
Misha Brukmanf976c852005-04-21 22:55:34 +000077
78 return
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +000079 sys::Program::ExecuteAndWait(ProgramPath, Args, 0, redirects,
80 NumSeconds, MemoryLimit);
Chris Lattner45495c52005-02-13 23:13:47 +000081}
82
Viktor Kutuzovfc2271f2009-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;
Chris Lattner45495c52005-02-13 23:13:47 +0000100
Viktor Kutuzovfc2271f2009-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}
Chris Lattner45495c52005-02-13 23:13:47 +0000136
Reid Spencerb31baa82004-12-19 18:00:21 +0000137static void ProcessFailure(sys::Path ProgPath, const char** Args) {
Chris Lattner89bf9ea2004-02-18 20:38:00 +0000138 std::ostringstream OS;
Chris Lattnera3de1172004-02-18 20:58:00 +0000139 OS << "\nError running tool:\n ";
Chris Lattner89bf9ea2004-02-18 20:38:00 +0000140 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 Gohmana971b292009-02-12 20:53:27 +0000145 sys::Path ErrorFilename("bugpoint.program_error_messages");
Reid Spencer51c5a282006-08-23 20:34:57 +0000146 std::string ErrMsg;
147 if (ErrorFilename.makeUnique(true, &ErrMsg)) {
Dan Gohman65f57c22009-07-15 16:35:29 +0000148 errs() << "Error making unique filename: " << ErrMsg << "\n";
Reid Spencer51c5a282006-08-23 20:34:57 +0000149 exit(1);
150 }
Reid Spencerb31baa82004-12-19 18:00:21 +0000151 RunProgramWithTimeout(ProgPath, Args, sys::Path(""), ErrorFilename,
Reid Spencer8ea5ecb2006-08-21 06:04:45 +0000152 ErrorFilename); // FIXME: check return code ?
Chris Lattner89bf9ea2004-02-18 20:38:00 +0000153
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
Reid Spencera229c5c2005-07-08 03:08:58 +0000163 ErrorFilename.eraseFromDisk();
Chris Lattner89bf9ea2004-02-18 20:38:00 +0000164 throw ToolExecutionError(OS.str());
165}
166
Misha Brukman9558c6a2003-09-29 22:39:25 +0000167//===---------------------------------------------------------------------===//
168// LLI Implementation of AbstractIntepreter interface
169//
Chris Lattner2cdd21c2003-12-14 21:35:53 +0000170namespace {
171 class LLI : public AbstractInterpreter {
172 std::string LLIPath; // The path to the LLI executable
Brian Gaeked11577b2004-05-04 21:09:01 +0000173 std::vector<std::string> ToolArgs; // Args to pass to LLI
Chris Lattner2cdd21c2003-12-14 21:35:53 +0000174 public:
Brian Gaeked11577b2004-05-04 21:09:01 +0000175 LLI(const std::string &Path, const std::vector<std::string> *Args)
176 : LLIPath(Path) {
177 ToolArgs.clear ();
Brian Gaeke48b008d2004-05-04 22:02:41 +0000178 if (Args) { ToolArgs = *Args; }
Brian Gaeked11577b2004-05-04 21:09:01 +0000179 }
Misha Brukmanf976c852005-04-21 22:55:34 +0000180
Gabor Greif8ff70c22007-07-04 21:55:50 +0000181 virtual int ExecuteProgram(const std::string &Bitcode,
Chris Lattner2cdd21c2003-12-14 21:35:53 +0000182 const std::vector<std::string> &Args,
183 const std::string &InputFile,
184 const std::string &OutputFile,
Reid Spencer51ab5c82006-06-06 00:00:42 +0000185 const std::vector<std::string> &GCCArgs,
Misha Brukmanf976c852005-04-21 22:55:34 +0000186 const std::vector<std::string> &SharedLibs =
Chris Lattnere96b2ed2004-07-24 07:49:11 +0000187 std::vector<std::string>(),
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000188 unsigned Timeout = 0,
189 unsigned MemoryLimit = 0);
Chris Lattner2cdd21c2003-12-14 21:35:53 +0000190 };
191}
Misha Brukman9558c6a2003-09-29 22:39:25 +0000192
Gabor Greif8ff70c22007-07-04 21:55:50 +0000193int LLI::ExecuteProgram(const std::string &Bitcode,
Chris Lattner7915a1e2003-10-14 21:34:11 +0000194 const std::vector<std::string> &Args,
Misha Brukman9558c6a2003-09-29 22:39:25 +0000195 const std::string &InputFile,
196 const std::string &OutputFile,
Reid Spencer51ab5c82006-06-06 00:00:42 +0000197 const std::vector<std::string> &GCCArgs,
Chris Lattnere96b2ed2004-07-24 07:49:11 +0000198 const std::vector<std::string> &SharedLibs,
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000199 unsigned Timeout,
200 unsigned MemoryLimit) {
Chris Lattner8c56be52004-02-18 20:21:57 +0000201 if (!SharedLibs.empty())
202 throw ToolExecutionError("LLI currently does not support "
203 "loading shared libraries.");
Misha Brukman9558c6a2003-09-29 22:39:25 +0000204
205 std::vector<const char*> LLIArgs;
206 LLIArgs.push_back(LLIPath.c_str());
Misha Brukman9558c6a2003-09-29 22:39:25 +0000207 LLIArgs.push_back("-force-interpreter=true");
Brian Gaeked11577b2004-05-04 21:09:01 +0000208
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
Gabor Greif8ff70c22007-07-04 21:55:50 +0000213 LLIArgs.push_back(Bitcode.c_str());
Misha Brukman9558c6a2003-09-29 22:39:25 +0000214 // 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 Gohmana1bdced2009-07-15 17:29:42 +0000219 outs() << "<lli>"; outs().flush();
Dan Gohman65f57c22009-07-15 16:35:29 +0000220 DEBUG(errs() << "\nAbout to run:\t";
Chris Lattner7b2ccff2003-10-19 02:14:58 +0000221 for (unsigned i=0, e = LLIArgs.size()-1; i != e; ++i)
Dan Gohman65f57c22009-07-15 16:35:29 +0000222 errs() << " " << LLIArgs[i];
223 errs() << "\n";
Misha Brukman9558c6a2003-09-29 22:39:25 +0000224 );
Reid Spencerb31baa82004-12-19 18:00:21 +0000225 return RunProgramWithTimeout(sys::Path(LLIPath), &LLIArgs[0],
Misha Brukmanf976c852005-04-21 22:55:34 +0000226 sys::Path(InputFile), sys::Path(OutputFile), sys::Path(OutputFile),
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000227 Timeout, MemoryLimit);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000228}
229
230// LLI create method - Try to find the LLI executable
Chris Lattner7915a1e2003-10-14 21:34:11 +0000231AbstractInterpreter *AbstractInterpreter::createLLI(const std::string &ProgPath,
Brian Gaeked11577b2004-05-04 21:09:01 +0000232 std::string &Message,
233 const std::vector<std::string> *ToolArgs) {
Reid Spencer51ab8ec2004-12-13 23:43:44 +0000234 std::string LLIPath = FindExecutable("lli", ProgPath).toString();
Misha Brukman9558c6a2003-09-29 22:39:25 +0000235 if (!LLIPath.empty()) {
236 Message = "Found lli: " + LLIPath + "\n";
Brian Gaeked11577b2004-05-04 21:09:01 +0000237 return new LLI(LLIPath, ToolArgs);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000238 }
239
240 Message = "Cannot find `lli' in executable directory or PATH!\n";
241 return 0;
242}
243
Anton Korobeynikov9ef74252008-04-28 20:53:48 +0000244//===---------------------------------------------------------------------===//
245// Custom execution command implementation of AbstractIntepreter interface
246//
247// Allows using a custom command for executing the bitcode, thus allows,
248// for example, to invoke a cross compiler for code generation followed by
249// a simulator that executes the generated binary.
250namespace {
251 class CustomExecutor : public AbstractInterpreter {
252 std::string ExecutionCommand;
253 std::vector<std::string> ExecutorArgs;
254 public:
255 CustomExecutor(
256 const std::string &ExecutionCmd, std::vector<std::string> ExecArgs) :
257 ExecutionCommand(ExecutionCmd), ExecutorArgs(ExecArgs) {}
258
259 virtual int ExecuteProgram(const std::string &Bitcode,
260 const std::vector<std::string> &Args,
261 const std::string &InputFile,
262 const std::string &OutputFile,
263 const std::vector<std::string> &GCCArgs,
264 const std::vector<std::string> &SharedLibs =
265 std::vector<std::string>(),
266 unsigned Timeout = 0,
267 unsigned MemoryLimit = 0);
268 };
269}
270
271int CustomExecutor::ExecuteProgram(const std::string &Bitcode,
272 const std::vector<std::string> &Args,
273 const std::string &InputFile,
274 const std::string &OutputFile,
275 const std::vector<std::string> &GCCArgs,
276 const std::vector<std::string> &SharedLibs,
277 unsigned Timeout,
278 unsigned MemoryLimit) {
279
280 std::vector<const char*> ProgramArgs;
281 ProgramArgs.push_back(ExecutionCommand.c_str());
282
283 for (std::size_t i = 0; i < ExecutorArgs.size(); ++i)
284 ProgramArgs.push_back(ExecutorArgs.at(i).c_str());
285 ProgramArgs.push_back(Bitcode.c_str());
286 ProgramArgs.push_back(0);
287
288 // Add optional parameters to the running program from Argv
289 for (unsigned i=0, e = Args.size(); i != e; ++i)
290 ProgramArgs.push_back(Args[i].c_str());
291
292 return RunProgramWithTimeout(
293 sys::Path(ExecutionCommand),
294 &ProgramArgs[0], sys::Path(InputFile), sys::Path(OutputFile),
295 sys::Path(OutputFile), Timeout, MemoryLimit);
296}
297
298// Custom execution environment create method, takes the execution command
299// as arguments
300AbstractInterpreter *AbstractInterpreter::createCustom(
301 const std::string &ProgramPath,
302 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
335 std::string CmdPath = FindExecutable(Command, ProgramPath).toString();
336 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
Misha Brukman9558c6a2003-09-29 22:39:25 +0000348//===----------------------------------------------------------------------===//
349// LLC Implementation of AbstractIntepreter interface
350//
Gabor Greif8ff70c22007-07-04 21:55:50 +0000351GCC::FileType LLC::OutputCode(const std::string &Bitcode,
Chris Lattnerc600f3c2006-09-15 21:29:15 +0000352 sys::Path &OutputAsmFile) {
Gabor Greif8ff70c22007-07-04 21:55:50 +0000353 sys::Path uniqueFile(Bitcode+".llc.s");
Reid Spencer51c5a282006-08-23 20:34:57 +0000354 std::string ErrMsg;
355 if (uniqueFile.makeUnique(true, &ErrMsg)) {
Dan Gohman65f57c22009-07-15 16:35:29 +0000356 errs() << "Error making unique filename: " << ErrMsg << "\n";
Reid Spencer51c5a282006-08-23 20:34:57 +0000357 exit(1);
358 }
Reid Spencer9ac14182004-12-16 23:01:34 +0000359 OutputAsmFile = uniqueFile;
Brian Gaeked11577b2004-05-04 21:09:01 +0000360 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...
Gabor Greif8ff70c22007-07-04 21:55:50 +0000370 LLCArgs.push_back (Bitcode.c_str()); // This is the input bitcode
Brian Gaeked11577b2004-05-04 21:09:01 +0000371 LLCArgs.push_back (0);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000372
Dan Gohmana1bdced2009-07-15 17:29:42 +0000373 outs() << "<llc>"; outs().flush();
Dan Gohman65f57c22009-07-15 16:35:29 +0000374 DEBUG(errs() << "\nAbout to run:\t";
Brian Gaeked11577b2004-05-04 21:09:01 +0000375 for (unsigned i=0, e = LLCArgs.size()-1; i != e; ++i)
Dan Gohman65f57c22009-07-15 16:35:29 +0000376 errs() << " " << LLCArgs[i];
377 errs() << "\n";
Brian Gaeked11577b2004-05-04 21:09:01 +0000378 );
Misha Brukmanf976c852005-04-21 22:55:34 +0000379 if (RunProgramWithTimeout(sys::Path(LLCPath), &LLCArgs[0],
Reid Spencerb31baa82004-12-19 18:00:21 +0000380 sys::Path(), sys::Path(), sys::Path()))
381 ProcessFailure(sys::Path(LLCPath), &LLCArgs[0]);
Chris Lattnerc600f3c2006-09-15 21:29:15 +0000382
383 return GCC::AsmFile;
Misha Brukman9558c6a2003-09-29 22:39:25 +0000384}
385
Gabor Greif8ff70c22007-07-04 21:55:50 +0000386void LLC::compileProgram(const std::string &Bitcode) {
Reid Spencer9ac14182004-12-16 23:01:34 +0000387 sys::Path OutputAsmFile;
Gabor Greif8ff70c22007-07-04 21:55:50 +0000388 OutputCode(Bitcode, OutputAsmFile);
Reid Spencera229c5c2005-07-08 03:08:58 +0000389 OutputAsmFile.eraseFromDisk();
Chris Lattner9cbbee32004-02-18 23:24:41 +0000390}
391
Gabor Greif8ff70c22007-07-04 21:55:50 +0000392int LLC::ExecuteProgram(const std::string &Bitcode,
Chris Lattner7915a1e2003-10-14 21:34:11 +0000393 const std::vector<std::string> &Args,
Misha Brukman9558c6a2003-09-29 22:39:25 +0000394 const std::string &InputFile,
395 const std::string &OutputFile,
Reid Spencer51ab5c82006-06-06 00:00:42 +0000396 const std::vector<std::string> &ArgsForGCC,
Chris Lattnere96b2ed2004-07-24 07:49:11 +0000397 const std::vector<std::string> &SharedLibs,
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000398 unsigned Timeout,
399 unsigned MemoryLimit) {
Chris Lattnereeed9832003-10-14 21:52:52 +0000400
Reid Spencer9ac14182004-12-16 23:01:34 +0000401 sys::Path OutputAsmFile;
Gabor Greif8ff70c22007-07-04 21:55:50 +0000402 OutputCode(Bitcode, OutputAsmFile);
Anton Korobeynikov86c006a2009-08-05 09:32:10 +0000403 FileRemover OutFileRemover(OutputAsmFile, !SaveTemps);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000404
Reid Spencer51ab5c82006-06-06 00:00:42 +0000405 std::vector<std::string> GCCArgs(ArgsForGCC);
Bill Wendling38efa382009-03-02 23:13:18 +0000406 GCCArgs.insert(GCCArgs.end(), SharedLibs.begin(), SharedLibs.end());
407 GCCArgs.insert(GCCArgs.end(), gccArgs.begin(), gccArgs.end());
Reid Spencer51ab5c82006-06-06 00:00:42 +0000408
Misha Brukman9558c6a2003-09-29 22:39:25 +0000409 // Assuming LLC worked, compile the result with GCC and run it.
Reid Spencer9ac14182004-12-16 23:01:34 +0000410 return gcc->ExecuteProgram(OutputAsmFile.toString(), Args, GCC::AsmFile,
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000411 InputFile, OutputFile, GCCArgs,
412 Timeout, MemoryLimit);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000413}
414
Chris Lattner7915a1e2003-10-14 21:34:11 +0000415/// createLLC - Try to find the LLC executable
Misha Brukman9558c6a2003-09-29 22:39:25 +0000416///
Chris Lattner7915a1e2003-10-14 21:34:11 +0000417LLC *AbstractInterpreter::createLLC(const std::string &ProgramPath,
Brian Gaeked11577b2004-05-04 21:09:01 +0000418 std::string &Message,
Bill Wendling38efa382009-03-02 23:13:18 +0000419 const std::vector<std::string> *Args,
420 const std::vector<std::string> *GCCArgs) {
Reid Spencer51ab8ec2004-12-13 23:43:44 +0000421 std::string LLCPath = FindExecutable("llc", ProgramPath).toString();
Misha Brukman9558c6a2003-09-29 22:39:25 +0000422 if (LLCPath.empty()) {
423 Message = "Cannot find `llc' in executable directory or PATH!\n";
424 return 0;
425 }
426
427 Message = "Found llc: " + LLCPath + "\n";
Bill Wendling38efa382009-03-02 23:13:18 +0000428 GCC *gcc = GCC::create(ProgramPath, Message, GCCArgs);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000429 if (!gcc) {
Dan Gohman65f57c22009-07-15 16:35:29 +0000430 errs() << Message << "\n";
Misha Brukman9558c6a2003-09-29 22:39:25 +0000431 exit(1);
432 }
Bill Wendling38efa382009-03-02 23:13:18 +0000433 return new LLC(LLCPath, gcc, Args, GCCArgs);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000434}
435
436//===---------------------------------------------------------------------===//
437// JIT Implementation of AbstractIntepreter interface
438//
Chris Lattner2cdd21c2003-12-14 21:35:53 +0000439namespace {
440 class JIT : public AbstractInterpreter {
441 std::string LLIPath; // The path to the LLI executable
Brian Gaeked11577b2004-05-04 21:09:01 +0000442 std::vector<std::string> ToolArgs; // Args to pass to LLI
Chris Lattner2cdd21c2003-12-14 21:35:53 +0000443 public:
Brian Gaeked11577b2004-05-04 21:09:01 +0000444 JIT(const std::string &Path, const std::vector<std::string> *Args)
445 : LLIPath(Path) {
446 ToolArgs.clear ();
Brian Gaeke48b008d2004-05-04 22:02:41 +0000447 if (Args) { ToolArgs = *Args; }
Brian Gaeked11577b2004-05-04 21:09:01 +0000448 }
Misha Brukmanf976c852005-04-21 22:55:34 +0000449
Gabor Greif8ff70c22007-07-04 21:55:50 +0000450 virtual int ExecuteProgram(const std::string &Bitcode,
Chris Lattner2cdd21c2003-12-14 21:35:53 +0000451 const std::vector<std::string> &Args,
452 const std::string &InputFile,
453 const std::string &OutputFile,
Reid Spencer51ab5c82006-06-06 00:00:42 +0000454 const std::vector<std::string> &GCCArgs =
455 std::vector<std::string>(),
Misha Brukmanf976c852005-04-21 22:55:34 +0000456 const std::vector<std::string> &SharedLibs =
Reid Spencer51ab5c82006-06-06 00:00:42 +0000457 std::vector<std::string>(),
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000458 unsigned Timeout =0,
459 unsigned MemoryLimit =0);
Chris Lattner2cdd21c2003-12-14 21:35:53 +0000460 };
461}
Misha Brukman9558c6a2003-09-29 22:39:25 +0000462
Gabor Greif8ff70c22007-07-04 21:55:50 +0000463int JIT::ExecuteProgram(const std::string &Bitcode,
Chris Lattner7915a1e2003-10-14 21:34:11 +0000464 const std::vector<std::string> &Args,
Misha Brukman9558c6a2003-09-29 22:39:25 +0000465 const std::string &InputFile,
466 const std::string &OutputFile,
Reid Spencer51ab5c82006-06-06 00:00:42 +0000467 const std::vector<std::string> &GCCArgs,
Chris Lattnere96b2ed2004-07-24 07:49:11 +0000468 const std::vector<std::string> &SharedLibs,
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000469 unsigned Timeout,
470 unsigned MemoryLimit) {
Misha Brukman9558c6a2003-09-29 22:39:25 +0000471 // Construct a vector of parameters, incorporating those from the command-line
472 std::vector<const char*> JITArgs;
473 JITArgs.push_back(LLIPath.c_str());
Misha Brukman9558c6a2003-09-29 22:39:25 +0000474 JITArgs.push_back("-force-interpreter=false");
Chris Lattnereeed9832003-10-14 21:52:52 +0000475
Brian Gaeked11577b2004-05-04 21:09:01 +0000476 // Add any extra LLI args.
477 for (unsigned i = 0, e = ToolArgs.size(); i != e; ++i)
478 JITArgs.push_back(ToolArgs[i].c_str());
479
Chris Lattnereeed9832003-10-14 21:52:52 +0000480 for (unsigned i = 0, e = SharedLibs.size(); i != e; ++i) {
Misha Brukman9558c6a2003-09-29 22:39:25 +0000481 JITArgs.push_back("-load");
Chris Lattnereeed9832003-10-14 21:52:52 +0000482 JITArgs.push_back(SharedLibs[i].c_str());
Misha Brukman9558c6a2003-09-29 22:39:25 +0000483 }
Gabor Greif8ff70c22007-07-04 21:55:50 +0000484 JITArgs.push_back(Bitcode.c_str());
Misha Brukman9558c6a2003-09-29 22:39:25 +0000485 // Add optional parameters to the running program from Argv
486 for (unsigned i=0, e = Args.size(); i != e; ++i)
487 JITArgs.push_back(Args[i].c_str());
488 JITArgs.push_back(0);
489
Dan Gohmana1bdced2009-07-15 17:29:42 +0000490 outs() << "<jit>"; outs().flush();
Dan Gohman65f57c22009-07-15 16:35:29 +0000491 DEBUG(errs() << "\nAbout to run:\t";
Chris Lattner7b2ccff2003-10-19 02:14:58 +0000492 for (unsigned i=0, e = JITArgs.size()-1; i != e; ++i)
Dan Gohman65f57c22009-07-15 16:35:29 +0000493 errs() << " " << JITArgs[i];
494 errs() << "\n";
Misha Brukman9558c6a2003-09-29 22:39:25 +0000495 );
Dan Gohman65f57c22009-07-15 16:35:29 +0000496 DEBUG(errs() << "\nSending output to " << OutputFile << "\n");
Reid Spencerb31baa82004-12-19 18:00:21 +0000497 return RunProgramWithTimeout(sys::Path(LLIPath), &JITArgs[0],
Misha Brukmanf976c852005-04-21 22:55:34 +0000498 sys::Path(InputFile), sys::Path(OutputFile), sys::Path(OutputFile),
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000499 Timeout, MemoryLimit);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000500}
501
Chris Lattner7915a1e2003-10-14 21:34:11 +0000502/// createJIT - Try to find the LLI executable
Misha Brukman9558c6a2003-09-29 22:39:25 +0000503///
Chris Lattner7915a1e2003-10-14 21:34:11 +0000504AbstractInterpreter *AbstractInterpreter::createJIT(const std::string &ProgPath,
Brian Gaeked11577b2004-05-04 21:09:01 +0000505 std::string &Message, const std::vector<std::string> *Args) {
Reid Spencer51ab8ec2004-12-13 23:43:44 +0000506 std::string LLIPath = FindExecutable("lli", ProgPath).toString();
Misha Brukman9558c6a2003-09-29 22:39:25 +0000507 if (!LLIPath.empty()) {
508 Message = "Found lli: " + LLIPath + "\n";
Brian Gaeked11577b2004-05-04 21:09:01 +0000509 return new JIT(LLIPath, Args);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000510 }
511
512 Message = "Cannot find `lli' in executable directory or PATH!\n";
513 return 0;
514}
515
Gabor Greif8ff70c22007-07-04 21:55:50 +0000516GCC::FileType CBE::OutputCode(const std::string &Bitcode,
Chris Lattnerc600f3c2006-09-15 21:29:15 +0000517 sys::Path &OutputCFile) {
Gabor Greif8ff70c22007-07-04 21:55:50 +0000518 sys::Path uniqueFile(Bitcode+".cbe.c");
Reid Spencer51c5a282006-08-23 20:34:57 +0000519 std::string ErrMsg;
520 if (uniqueFile.makeUnique(true, &ErrMsg)) {
Dan Gohman65f57c22009-07-15 16:35:29 +0000521 errs() << "Error making unique filename: " << ErrMsg << "\n";
Reid Spencer51c5a282006-08-23 20:34:57 +0000522 exit(1);
523 }
Reid Spencer9ac14182004-12-16 23:01:34 +0000524 OutputCFile = uniqueFile;
Brian Gaeked11577b2004-05-04 21:09:01 +0000525 std::vector<const char *> LLCArgs;
526 LLCArgs.push_back (LLCPath.c_str());
527
528 // Add any extra LLC args.
529 for (unsigned i = 0, e = ToolArgs.size(); i != e; ++i)
530 LLCArgs.push_back(ToolArgs[i].c_str());
531
532 LLCArgs.push_back ("-o");
533 LLCArgs.push_back (OutputCFile.c_str()); // Output to the C file
534 LLCArgs.push_back ("-march=c"); // Output C language
535 LLCArgs.push_back ("-f"); // Overwrite as necessary...
Gabor Greif8ff70c22007-07-04 21:55:50 +0000536 LLCArgs.push_back (Bitcode.c_str()); // This is the input bitcode
Brian Gaeked11577b2004-05-04 21:09:01 +0000537 LLCArgs.push_back (0);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000538
Dan Gohmana1bdced2009-07-15 17:29:42 +0000539 outs() << "<cbe>"; outs().flush();
Dan Gohman65f57c22009-07-15 16:35:29 +0000540 DEBUG(errs() << "\nAbout to run:\t";
Brian Gaeked11577b2004-05-04 21:09:01 +0000541 for (unsigned i=0, e = LLCArgs.size()-1; i != e; ++i)
Dan Gohman65f57c22009-07-15 16:35:29 +0000542 errs() << " " << LLCArgs[i];
543 errs() << "\n";
Brian Gaeked11577b2004-05-04 21:09:01 +0000544 );
Misha Brukmanf976c852005-04-21 22:55:34 +0000545 if (RunProgramWithTimeout(LLCPath, &LLCArgs[0], sys::Path(), sys::Path(),
Reid Spencerb31baa82004-12-19 18:00:21 +0000546 sys::Path()))
Brian Gaeked11577b2004-05-04 21:09:01 +0000547 ProcessFailure(LLCPath, &LLCArgs[0]);
Chris Lattnerc600f3c2006-09-15 21:29:15 +0000548 return GCC::CFile;
Misha Brukman9558c6a2003-09-29 22:39:25 +0000549}
550
Gabor Greif8ff70c22007-07-04 21:55:50 +0000551void CBE::compileProgram(const std::string &Bitcode) {
Reid Spencer9ac14182004-12-16 23:01:34 +0000552 sys::Path OutputCFile;
Gabor Greif8ff70c22007-07-04 21:55:50 +0000553 OutputCode(Bitcode, OutputCFile);
Reid Spencera229c5c2005-07-08 03:08:58 +0000554 OutputCFile.eraseFromDisk();
Chris Lattner9cbbee32004-02-18 23:24:41 +0000555}
556
Gabor Greif8ff70c22007-07-04 21:55:50 +0000557int CBE::ExecuteProgram(const std::string &Bitcode,
Chris Lattner7915a1e2003-10-14 21:34:11 +0000558 const std::vector<std::string> &Args,
Misha Brukman9558c6a2003-09-29 22:39:25 +0000559 const std::string &InputFile,
560 const std::string &OutputFile,
Reid Spencer51ab5c82006-06-06 00:00:42 +0000561 const std::vector<std::string> &ArgsForGCC,
Chris Lattnere96b2ed2004-07-24 07:49:11 +0000562 const std::vector<std::string> &SharedLibs,
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000563 unsigned Timeout,
564 unsigned MemoryLimit) {
Reid Spencer9ac14182004-12-16 23:01:34 +0000565 sys::Path OutputCFile;
Gabor Greif8ff70c22007-07-04 21:55:50 +0000566 OutputCode(Bitcode, OutputCFile);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000567
Anton Korobeynikov86c006a2009-08-05 09:32:10 +0000568 FileRemover CFileRemove(OutputCFile, !SaveTemps);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000569
Reid Spencer51ab5c82006-06-06 00:00:42 +0000570 std::vector<std::string> GCCArgs(ArgsForGCC);
Bill Wendling38efa382009-03-02 23:13:18 +0000571 GCCArgs.insert(GCCArgs.end(), SharedLibs.begin(), SharedLibs.end());
572
Misha Brukmanf976c852005-04-21 22:55:34 +0000573 return gcc->ExecuteProgram(OutputCFile.toString(), Args, GCC::CFile,
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000574 InputFile, OutputFile, GCCArgs,
575 Timeout, MemoryLimit);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000576}
577
Chris Lattner9915cd92004-02-17 06:40:06 +0000578/// createCBE - Try to find the 'llc' executable
Misha Brukman9558c6a2003-09-29 22:39:25 +0000579///
Chris Lattner7915a1e2003-10-14 21:34:11 +0000580CBE *AbstractInterpreter::createCBE(const std::string &ProgramPath,
Brian Gaeked11577b2004-05-04 21:09:01 +0000581 std::string &Message,
Bill Wendling38efa382009-03-02 23:13:18 +0000582 const std::vector<std::string> *Args,
583 const std::vector<std::string> *GCCArgs) {
Reid Spencerb31baa82004-12-19 18:00:21 +0000584 sys::Path LLCPath = FindExecutable("llc", ProgramPath);
585 if (LLCPath.isEmpty()) {
Misha Brukmanf976c852005-04-21 22:55:34 +0000586 Message =
Chris Lattner9915cd92004-02-17 06:40:06 +0000587 "Cannot find `llc' in executable directory or PATH!\n";
Misha Brukman9558c6a2003-09-29 22:39:25 +0000588 return 0;
589 }
590
Reid Spencerb31baa82004-12-19 18:00:21 +0000591 Message = "Found llc: " + LLCPath.toString() + "\n";
Bill Wendling38efa382009-03-02 23:13:18 +0000592 GCC *gcc = GCC::create(ProgramPath, Message, GCCArgs);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000593 if (!gcc) {
Dan Gohman65f57c22009-07-15 16:35:29 +0000594 errs() << Message << "\n";
Misha Brukman9558c6a2003-09-29 22:39:25 +0000595 exit(1);
596 }
Brian Gaeked11577b2004-05-04 21:09:01 +0000597 return new CBE(LLCPath, gcc, Args);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000598}
599
600//===---------------------------------------------------------------------===//
601// GCC abstraction
602//
David Goodwin80becf12009-07-10 21:39:28 +0000603
Daniel Dunbar84f9abe2009-07-11 21:24:52 +0000604#ifdef __APPLE__
David Goodwin80becf12009-07-10 21:39:28 +0000605static bool
606IsARMArchitecture(std::vector<std::string> Args)
607{
608 for (std::vector<std::string>::const_iterator
609 I = Args.begin(), E = Args.end(); I != E; ++I) {
610 if (!strcasecmp(I->c_str(), "-arch")) {
611 ++I;
612 if ((I != E) && !strncasecmp(I->c_str(), "arm", strlen("arm"))) {
613 return true;
614 }
615 }
616 }
617
618 return false;
619}
Daniel Dunbar84f9abe2009-07-11 21:24:52 +0000620#endif
David Goodwin80becf12009-07-10 21:39:28 +0000621
Misha Brukman9558c6a2003-09-29 22:39:25 +0000622int GCC::ExecuteProgram(const std::string &ProgramFile,
Chris Lattner7915a1e2003-10-14 21:34:11 +0000623 const std::vector<std::string> &Args,
Misha Brukman9558c6a2003-09-29 22:39:25 +0000624 FileType fileType,
625 const std::string &InputFile,
626 const std::string &OutputFile,
Reid Spencer51ab5c82006-06-06 00:00:42 +0000627 const std::vector<std::string> &ArgsForGCC,
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000628 unsigned Timeout,
629 unsigned MemoryLimit) {
Misha Brukman9558c6a2003-09-29 22:39:25 +0000630 std::vector<const char*> GCCArgs;
631
632 GCCArgs.push_back(GCCPath.c_str());
Chris Lattnereeed9832003-10-14 21:52:52 +0000633
Bill Wendling38efa382009-03-02 23:13:18 +0000634 for (std::vector<std::string>::const_iterator
635 I = gccArgs.begin(), E = gccArgs.end(); I != E; ++I)
636 GCCArgs.push_back(I->c_str());
637
Chris Lattnereeed9832003-10-14 21:52:52 +0000638 // Specify -x explicitly in case the extension is wonky
Misha Brukman9558c6a2003-09-29 22:39:25 +0000639 GCCArgs.push_back("-x");
640 if (fileType == CFile) {
641 GCCArgs.push_back("c");
642 GCCArgs.push_back("-fno-strict-aliasing");
643 } else {
644 GCCArgs.push_back("assembler");
Chris Lattnerd00b2882005-08-29 13:14:24 +0000645#ifdef __APPLE__
David Goodwin80becf12009-07-10 21:39:28 +0000646 // For ARM architectures we don't want this flag. bugpoint isn't
647 // explicitly told what architecture it is working on, so we get
648 // it from gcc flags
649 if (!IsARMArchitecture(ArgsForGCC))
650 GCCArgs.push_back("-force_cpusubtype_ALL");
Chris Lattnerd00b2882005-08-29 13:14:24 +0000651#endif
Misha Brukman9558c6a2003-09-29 22:39:25 +0000652 }
653 GCCArgs.push_back(ProgramFile.c_str()); // Specify the input filename...
Chris Lattner629e4872006-06-09 21:31:53 +0000654 GCCArgs.push_back("-x");
655 GCCArgs.push_back("none");
Misha Brukman9558c6a2003-09-29 22:39:25 +0000656 GCCArgs.push_back("-o");
Reid Spencercda985e2004-12-15 01:51:56 +0000657 sys::Path OutputBinary (ProgramFile+".gcc.exe");
Reid Spencer51c5a282006-08-23 20:34:57 +0000658 std::string ErrMsg;
659 if (OutputBinary.makeUnique(true, &ErrMsg)) {
Dan Gohman65f57c22009-07-15 16:35:29 +0000660 errs() << "Error making unique filename: " << ErrMsg << "\n";
Reid Spencer51c5a282006-08-23 20:34:57 +0000661 exit(1);
662 }
Misha Brukman9558c6a2003-09-29 22:39:25 +0000663 GCCArgs.push_back(OutputBinary.c_str()); // Output to the right file...
Reid Spencer51ab5c82006-06-06 00:00:42 +0000664
665 // Add any arguments intended for GCC. We locate them here because this is
666 // most likely -L and -l options that need to come before other libraries but
667 // after the source. Other options won't be sensitive to placement on the
668 // command line, so this should be safe.
669 for (unsigned i = 0, e = ArgsForGCC.size(); i != e; ++i)
670 GCCArgs.push_back(ArgsForGCC[i].c_str());
671
Misha Brukman9558c6a2003-09-29 22:39:25 +0000672 GCCArgs.push_back("-lm"); // Hard-code the math library...
673 GCCArgs.push_back("-O2"); // Optimize the program a bit...
Brian Gaekec8db76c2003-11-18 06:31:17 +0000674#if defined (HAVE_LINK_R)
Chris Lattner1f0f1622003-10-18 21:54:47 +0000675 GCCArgs.push_back("-Wl,-R."); // Search this dir for .so files
Brian Gaekec8db76c2003-11-18 06:31:17 +0000676#endif
Chris Lattnerfdcc71e2006-02-04 05:02:27 +0000677#ifdef __sparc__
678 GCCArgs.push_back("-mcpu=v9");
679#endif
Misha Brukman9558c6a2003-09-29 22:39:25 +0000680 GCCArgs.push_back(0); // NULL terminator
681
Dan Gohmana1bdced2009-07-15 17:29:42 +0000682 outs() << "<gcc>"; outs().flush();
Dan Gohman65f57c22009-07-15 16:35:29 +0000683 DEBUG(errs() << "\nAbout to run:\t";
Evan Cheng7f7fdcc2007-01-03 07:44:30 +0000684 for (unsigned i=0, e = GCCArgs.size()-1; i != e; ++i)
Dan Gohman65f57c22009-07-15 16:35:29 +0000685 errs() << " " << GCCArgs[i];
686 errs() << "\n";
Evan Cheng7f7fdcc2007-01-03 07:44:30 +0000687 );
Reid Spencerb31baa82004-12-19 18:00:21 +0000688 if (RunProgramWithTimeout(GCCPath, &GCCArgs[0], sys::Path(), sys::Path(),
689 sys::Path())) {
Chris Lattner89bf9ea2004-02-18 20:38:00 +0000690 ProcessFailure(GCCPath, &GCCArgs[0]);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000691 exit(1);
692 }
693
694 std::vector<const char*> ProgramArgs;
Chris Lattner45495c52005-02-13 23:13:47 +0000695
Evan Cheng70f684f2008-09-09 06:11:26 +0000696 if (RemoteClientPath.isEmpty())
Evan Cheng34e400e2007-05-03 18:36:15 +0000697 ProgramArgs.push_back(OutputBinary.c_str());
698 else {
Evan Cheng70f684f2008-09-09 06:11:26 +0000699 ProgramArgs.push_back(RemoteClientPath.c_str());
700 ProgramArgs.push_back(RemoteHost.c_str());
Evan Cheng34e400e2007-05-03 18:36:15 +0000701 ProgramArgs.push_back("-l");
Evan Cheng70f684f2008-09-09 06:11:26 +0000702 ProgramArgs.push_back(RemoteUser.c_str());
David Goodwin80becf12009-07-10 21:39:28 +0000703 if (!RemotePort.empty()) {
704 ProgramArgs.push_back("-p");
705 ProgramArgs.push_back(RemotePort.c_str());
706 }
Evan Cheng70f684f2008-09-09 06:11:26 +0000707 if (!RemoteExtra.empty()) {
708 ProgramArgs.push_back(RemoteExtra.c_str());
709 }
Evan Cheng34e400e2007-05-03 18:36:15 +0000710
David Goodwin91cf3612009-07-20 17:15:03 +0000711 // Full path to the binary. We need to cd to the exec directory because
712 // there is a dylib there that the exec expects to find in the CWD
Evan Cheng34e400e2007-05-03 18:36:15 +0000713 char* env_pwd = getenv("PWD");
David Goodwin91cf3612009-07-20 17:15:03 +0000714 std::string Exec = "cd ";
Evan Cheng34e400e2007-05-03 18:36:15 +0000715 Exec += env_pwd;
David Goodwin91cf3612009-07-20 17:15:03 +0000716 Exec += "; ./";
Evan Cheng34e400e2007-05-03 18:36:15 +0000717 Exec += OutputBinary.c_str();
718 ProgramArgs.push_back(Exec.c_str());
719 }
720
Misha Brukman9558c6a2003-09-29 22:39:25 +0000721 // Add optional parameters to the running program from Argv
722 for (unsigned i=0, e = Args.size(); i != e; ++i)
723 ProgramArgs.push_back(Args[i].c_str());
724 ProgramArgs.push_back(0); // NULL terminator
725
726 // Now that we have a binary, run it!
Dan Gohmana1bdced2009-07-15 17:29:42 +0000727 outs() << "<program>"; outs().flush();
Dan Gohman65f57c22009-07-15 16:35:29 +0000728 DEBUG(errs() << "\nAbout to run:\t";
Chris Lattner7b2ccff2003-10-19 02:14:58 +0000729 for (unsigned i=0, e = ProgramArgs.size()-1; i != e; ++i)
Dan Gohman65f57c22009-07-15 16:35:29 +0000730 errs() << " " << ProgramArgs[i];
731 errs() << "\n";
Misha Brukman9558c6a2003-09-29 22:39:25 +0000732 );
Chris Lattner8c56be52004-02-18 20:21:57 +0000733
Anton Korobeynikov86c006a2009-08-05 09:32:10 +0000734 FileRemover OutputBinaryRemover(OutputBinary, !SaveTemps);
Evan Cheng34e400e2007-05-03 18:36:15 +0000735
Viktor Kutuzov699220b2009-07-14 19:10:55 +0000736 if (RemoteClientPath.isEmpty()) {
Dan Gohman65f57c22009-07-15 16:35:29 +0000737 DEBUG(errs() << "<run locally>";);
Evan Cheng34e400e2007-05-03 18:36:15 +0000738 return RunProgramWithTimeout(OutputBinary, &ProgramArgs[0],
739 sys::Path(InputFile), sys::Path(OutputFile), sys::Path(OutputFile),
740 Timeout, MemoryLimit);
Viktor Kutuzov699220b2009-07-14 19:10:55 +0000741 } else {
Dan Gohmana1bdced2009-07-15 17:29:42 +0000742 outs() << "<run remotely>"; outs().flush();
Viktor Kutuzovfc2271f2009-07-18 18:39:24 +0000743 return RunProgramRemotelyWithTimeout(sys::Path(RemoteClientPath),
Viktor Kutuzov699220b2009-07-14 19:10:55 +0000744 &ProgramArgs[0], sys::Path(InputFile), sys::Path(OutputFile),
745 sys::Path(OutputFile), Timeout, MemoryLimit);
Viktor Kutuzov699220b2009-07-14 19:10:55 +0000746 }
Misha Brukman9558c6a2003-09-29 22:39:25 +0000747}
748
Chris Lattner1798e4a2003-10-14 21:07:25 +0000749int GCC::MakeSharedObject(const std::string &InputFile, FileType fileType,
Chris Lattner130e2a32006-06-27 20:35:36 +0000750 std::string &OutputFile,
751 const std::vector<std::string> &ArgsForGCC) {
Reid Spencercda985e2004-12-15 01:51:56 +0000752 sys::Path uniqueFilename(InputFile+LTDL_SHLIB_EXT);
Reid Spencer51c5a282006-08-23 20:34:57 +0000753 std::string ErrMsg;
754 if (uniqueFilename.makeUnique(true, &ErrMsg)) {
Dan Gohman65f57c22009-07-15 16:35:29 +0000755 errs() << "Error making unique filename: " << ErrMsg << "\n";
Reid Spencer51c5a282006-08-23 20:34:57 +0000756 exit(1);
757 }
Reid Spencercda985e2004-12-15 01:51:56 +0000758 OutputFile = uniqueFilename.toString();
759
Chris Lattner130e2a32006-06-27 20:35:36 +0000760 std::vector<const char*> GCCArgs;
761
762 GCCArgs.push_back(GCCPath.c_str());
Evan Chengfd829952009-03-12 00:53:34 +0000763
764 for (std::vector<std::string>::const_iterator
765 I = gccArgs.begin(), E = gccArgs.end(); I != E; ++I)
766 GCCArgs.push_back(I->c_str());
767
Misha Brukman9558c6a2003-09-29 22:39:25 +0000768 // Compile the C/asm file into a shared object
Chris Lattner130e2a32006-06-27 20:35:36 +0000769 GCCArgs.push_back("-x");
770 GCCArgs.push_back(fileType == AsmFile ? "assembler" : "c");
771 GCCArgs.push_back("-fno-strict-aliasing");
772 GCCArgs.push_back(InputFile.c_str()); // Specify the input filename.
Lauro Ramos Venancio497391a2007-06-06 23:10:56 +0000773 GCCArgs.push_back("-x");
774 GCCArgs.push_back("none");
Misha Brukman9558c6a2003-09-29 22:39:25 +0000775#if defined(sparc) || defined(__sparc__) || defined(__sparcv9)
Chris Lattner130e2a32006-06-27 20:35:36 +0000776 GCCArgs.push_back("-G"); // Compile a shared library, `-G' for Sparc
Nate Begeman72b286b2005-07-08 00:23:26 +0000777#elif defined(__APPLE__)
Chris Lattner130e2a32006-06-27 20:35:36 +0000778 // link all source files into a single module in data segment, rather than
779 // generating blocks. dynamic_lookup requires that you set
780 // MACOSX_DEPLOYMENT_TARGET=10.3 in your env. FIXME: it would be better for
781 // bugpoint to just pass that in the environment of GCC.
782 GCCArgs.push_back("-single_module");
783 GCCArgs.push_back("-dynamiclib"); // `-dynamiclib' for MacOS X/PowerPC
784 GCCArgs.push_back("-undefined");
785 GCCArgs.push_back("dynamic_lookup");
Misha Brukmanb3998ec2004-07-16 19:45:45 +0000786#else
Chris Lattner130e2a32006-06-27 20:35:36 +0000787 GCCArgs.push_back("-shared"); // `-shared' for Linux/X86, maybe others
Misha Brukman9558c6a2003-09-29 22:39:25 +0000788#endif
Chris Lattner1c81f132005-03-09 03:31:02 +0000789
Torok Edwinaf4fc282008-04-06 12:42:29 +0000790#if defined(__ia64__) || defined(__alpha__) || defined(__amd64__)
Chris Lattner130e2a32006-06-27 20:35:36 +0000791 GCCArgs.push_back("-fPIC"); // Requires shared objs to contain PIC
Chris Lattner1c81f132005-03-09 03:31:02 +0000792#endif
Chris Lattnerfdcc71e2006-02-04 05:02:27 +0000793#ifdef __sparc__
Chris Lattner130e2a32006-06-27 20:35:36 +0000794 GCCArgs.push_back("-mcpu=v9");
Chris Lattnerfdcc71e2006-02-04 05:02:27 +0000795#endif
Chris Lattner130e2a32006-06-27 20:35:36 +0000796 GCCArgs.push_back("-o");
797 GCCArgs.push_back(OutputFile.c_str()); // Output to the right filename.
798 GCCArgs.push_back("-O2"); // Optimize the program a bit.
799
800
801
802 // Add any arguments intended for GCC. We locate them here because this is
803 // most likely -L and -l options that need to come before other libraries but
804 // after the source. Other options won't be sensitive to placement on the
805 // command line, so this should be safe.
806 for (unsigned i = 0, e = ArgsForGCC.size(); i != e; ++i)
807 GCCArgs.push_back(ArgsForGCC[i].c_str());
808 GCCArgs.push_back(0); // NULL terminator
809
810
Misha Brukmanf976c852005-04-21 22:55:34 +0000811
Dan Gohmana1bdced2009-07-15 17:29:42 +0000812 outs() << "<gcc>"; outs().flush();
Dan Gohman65f57c22009-07-15 16:35:29 +0000813 DEBUG(errs() << "\nAbout to run:\t";
Evan Cheng7f7fdcc2007-01-03 07:44:30 +0000814 for (unsigned i=0, e = GCCArgs.size()-1; i != e; ++i)
Dan Gohman65f57c22009-07-15 16:35:29 +0000815 errs() << " " << GCCArgs[i];
816 errs() << "\n";
Evan Cheng7f7fdcc2007-01-03 07:44:30 +0000817 );
Chris Lattner130e2a32006-06-27 20:35:36 +0000818 if (RunProgramWithTimeout(GCCPath, &GCCArgs[0], sys::Path(), sys::Path(),
Reid Spencerb31baa82004-12-19 18:00:21 +0000819 sys::Path())) {
Chris Lattner130e2a32006-06-27 20:35:36 +0000820 ProcessFailure(GCCPath, &GCCArgs[0]);
Chris Lattner1798e4a2003-10-14 21:07:25 +0000821 return 1;
Misha Brukman9558c6a2003-09-29 22:39:25 +0000822 }
823 return 0;
824}
825
Chris Lattner7915a1e2003-10-14 21:34:11 +0000826/// create - Try to find the `gcc' executable
Misha Brukman9558c6a2003-09-29 22:39:25 +0000827///
Bill Wendling38efa382009-03-02 23:13:18 +0000828GCC *GCC::create(const std::string &ProgramPath, std::string &Message,
829 const std::vector<std::string> *Args) {
Reid Spencerb31baa82004-12-19 18:00:21 +0000830 sys::Path GCCPath = FindExecutable("gcc", ProgramPath);
831 if (GCCPath.isEmpty()) {
Misha Brukman9558c6a2003-09-29 22:39:25 +0000832 Message = "Cannot find `gcc' in executable directory or PATH!\n";
833 return 0;
834 }
835
Evan Cheng70f684f2008-09-09 06:11:26 +0000836 sys::Path RemoteClientPath;
837 if (!RemoteClient.empty())
838 RemoteClientPath = FindExecutable(RemoteClient.c_str(), ProgramPath);
Evan Cheng34e400e2007-05-03 18:36:15 +0000839
Reid Spencerb31baa82004-12-19 18:00:21 +0000840 Message = "Found gcc: " + GCCPath.toString() + "\n";
Bill Wendling38efa382009-03-02 23:13:18 +0000841 return new GCC(GCCPath, RemoteClientPath, Args);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000842}