blob: 3e63d8050c4be5b3c905bf76234273c4c097be64 [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"
Chris Lattner45495c52005-02-13 23:13:47 +000016#include "llvm/System/Program.h"
Evan Cheng34e400e2007-05-03 18:36:15 +000017#include "llvm/Support/CommandLine.h"
Reid Spencer551ccae2004-09-01 22:55:40 +000018#include "llvm/Support/Debug.h"
19#include "llvm/Support/FileUtilities.h"
Chris Lattner74382b72009-08-23 22:45:37 +000020#include "llvm/Support/raw_ostream.h"
21#include "llvm/Config/config.h" // for HAVE_LINK_R
Chris Lattner7915a1e2003-10-14 21:34:11 +000022#include <fstream>
Chris Lattner89bf9ea2004-02-18 20:38:00 +000023#include <sstream>
Chris Lattner2cdd21c2003-12-14 21:35:53 +000024using namespace llvm;
Brian Gaeked0fde302003-11-11 22:41:34 +000025
Anton Korobeynikov86c006a2009-08-05 09:32:10 +000026namespace llvm {
27 cl::opt<bool>
28 SaveTemps("save-temps", cl::init(false), cl::desc("Save temporary files"));
29}
30
Evan Cheng34e400e2007-05-03 18:36:15 +000031namespace {
32 cl::opt<std::string>
Evan Cheng70f684f2008-09-09 06:11:26 +000033 RemoteClient("remote-client",
34 cl::desc("Remote execution client (rsh/ssh)"));
Evan Cheng34e400e2007-05-03 18:36:15 +000035
36 cl::opt<std::string>
Evan Cheng70f684f2008-09-09 06:11:26 +000037 RemoteHost("remote-host",
38 cl::desc("Remote execution (rsh/ssh) host"));
39
40 cl::opt<std::string>
David Goodwin80becf12009-07-10 21:39:28 +000041 RemotePort("remote-port",
42 cl::desc("Remote execution (rsh/ssh) port"));
43
44 cl::opt<std::string>
Evan Cheng70f684f2008-09-09 06:11:26 +000045 RemoteUser("remote-user",
46 cl::desc("Remote execution (rsh/ssh) user id"));
47
48 cl::opt<std::string>
49 RemoteExtra("remote-extra-options",
50 cl::desc("Remote execution (rsh/ssh) extra options"));
Evan Cheng34e400e2007-05-03 18:36:15 +000051}
52
Alkis Evlogimenos1d29a6d2004-02-19 07:39:26 +000053ToolExecutionError::~ToolExecutionError() throw() { }
54
Viktor Kutuzovfc2271f2009-07-18 18:39:24 +000055/// RunProgramWithTimeout - This function provides an alternate interface
56/// to the sys::Program::ExecuteAndWait interface.
Chris Lattner45495c52005-02-13 23:13:47 +000057/// @see sys:Program::ExecuteAndWait
58static int RunProgramWithTimeout(const sys::Path &ProgramPath,
59 const char **Args,
60 const sys::Path &StdInFile,
61 const sys::Path &StdOutFile,
62 const sys::Path &StdErrFile,
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +000063 unsigned NumSeconds = 0,
64 unsigned MemoryLimit = 0) {
Chris Lattner45495c52005-02-13 23:13:47 +000065 const sys::Path* redirects[3];
66 redirects[0] = &StdInFile;
67 redirects[1] = &StdOutFile;
68 redirects[2] = &StdErrFile;
Viktor Kutuzovfc2271f2009-07-18 18:39:24 +000069
70#if 0 // For debug purposes
71 {
Dan Gohman65f57c22009-07-15 16:35:29 +000072 errs() << "RUN:";
Chris Lattnerd4223502006-09-15 23:01:10 +000073 for (unsigned i = 0; Args[i]; ++i)
Dan Gohman65f57c22009-07-15 16:35:29 +000074 errs() << " " << Args[i];
75 errs() << "\n";
Chris Lattnerd4223502006-09-15 23:01:10 +000076 }
Viktor Kutuzovfc2271f2009-07-18 18:39:24 +000077#endif
Misha Brukmanf976c852005-04-21 22:55:34 +000078
79 return
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +000080 sys::Program::ExecuteAndWait(ProgramPath, Args, 0, redirects,
81 NumSeconds, MemoryLimit);
Chris Lattner45495c52005-02-13 23:13:47 +000082}
83
Viktor Kutuzovfc2271f2009-07-18 18:39:24 +000084/// RunProgramRemotelyWithTimeout - This function runs the given program
85/// remotely using the given remote client and the sys::Program::ExecuteAndWait.
86/// Returns the remote program exit code or reports a remote client error if it
87/// fails. Remote client is required to return 255 if it failed or program exit
88/// code otherwise.
89/// @see sys:Program::ExecuteAndWait
90static int RunProgramRemotelyWithTimeout(const sys::Path &RemoteClientPath,
91 const char **Args,
92 const sys::Path &StdInFile,
93 const sys::Path &StdOutFile,
94 const sys::Path &StdErrFile,
95 unsigned NumSeconds = 0,
96 unsigned MemoryLimit = 0) {
97 const sys::Path* redirects[3];
98 redirects[0] = &StdInFile;
99 redirects[1] = &StdOutFile;
100 redirects[2] = &StdErrFile;
Chris Lattner45495c52005-02-13 23:13:47 +0000101
Viktor Kutuzovfc2271f2009-07-18 18:39:24 +0000102#if 0 // For debug purposes
103 {
104 errs() << "RUN:";
105 for (unsigned i = 0; Args[i]; ++i)
106 errs() << " " << Args[i];
107 errs() << "\n";
108 }
109#endif
110
111 // Run the program remotely with the remote client
112 int ReturnCode = sys::Program::ExecuteAndWait(RemoteClientPath, Args,
113 0, redirects, NumSeconds, MemoryLimit);
114
115 // Has the remote client fail?
116 if (255 == ReturnCode) {
117 std::ostringstream OS;
118 OS << "\nError running remote client:\n ";
119 for (const char **Arg = Args; *Arg; ++Arg)
120 OS << " " << *Arg;
121 OS << "\n";
122
123 // The error message is in the output file, let's print it out from there.
124 std::ifstream ErrorFile(StdOutFile.c_str());
125 if (ErrorFile) {
126 std::copy(std::istreambuf_iterator<char>(ErrorFile),
127 std::istreambuf_iterator<char>(),
128 std::ostreambuf_iterator<char>(OS));
129 ErrorFile.close();
130 }
131
132 throw ToolExecutionError(OS.str());
133 }
134
135 return ReturnCode;
136}
Chris Lattner45495c52005-02-13 23:13:47 +0000137
Reid Spencerb31baa82004-12-19 18:00:21 +0000138static void ProcessFailure(sys::Path ProgPath, const char** Args) {
Chris Lattner89bf9ea2004-02-18 20:38:00 +0000139 std::ostringstream OS;
Chris Lattnera3de1172004-02-18 20:58:00 +0000140 OS << "\nError running tool:\n ";
Chris Lattner89bf9ea2004-02-18 20:38:00 +0000141 for (const char **Arg = Args; *Arg; ++Arg)
142 OS << " " << *Arg;
143 OS << "\n";
144
145 // Rerun the compiler, capturing any error messages to print them.
Dan Gohmana971b292009-02-12 20:53:27 +0000146 sys::Path ErrorFilename("bugpoint.program_error_messages");
Reid Spencer51c5a282006-08-23 20:34:57 +0000147 std::string ErrMsg;
148 if (ErrorFilename.makeUnique(true, &ErrMsg)) {
Dan Gohman65f57c22009-07-15 16:35:29 +0000149 errs() << "Error making unique filename: " << ErrMsg << "\n";
Reid Spencer51c5a282006-08-23 20:34:57 +0000150 exit(1);
151 }
Reid Spencerb31baa82004-12-19 18:00:21 +0000152 RunProgramWithTimeout(ProgPath, Args, sys::Path(""), ErrorFilename,
Reid Spencer8ea5ecb2006-08-21 06:04:45 +0000153 ErrorFilename); // FIXME: check return code ?
Chris Lattner89bf9ea2004-02-18 20:38:00 +0000154
155 // Print out the error messages generated by GCC if possible...
156 std::ifstream ErrorFile(ErrorFilename.c_str());
157 if (ErrorFile) {
158 std::copy(std::istreambuf_iterator<char>(ErrorFile),
159 std::istreambuf_iterator<char>(),
160 std::ostreambuf_iterator<char>(OS));
161 ErrorFile.close();
162 }
163
Reid Spencera229c5c2005-07-08 03:08:58 +0000164 ErrorFilename.eraseFromDisk();
Chris Lattner89bf9ea2004-02-18 20:38:00 +0000165 throw ToolExecutionError(OS.str());
166}
167
Misha Brukman9558c6a2003-09-29 22:39:25 +0000168//===---------------------------------------------------------------------===//
169// LLI Implementation of AbstractIntepreter interface
170//
Chris Lattner2cdd21c2003-12-14 21:35:53 +0000171namespace {
172 class LLI : public AbstractInterpreter {
173 std::string LLIPath; // The path to the LLI executable
Brian Gaeked11577b2004-05-04 21:09:01 +0000174 std::vector<std::string> ToolArgs; // Args to pass to LLI
Chris Lattner2cdd21c2003-12-14 21:35:53 +0000175 public:
Brian Gaeked11577b2004-05-04 21:09:01 +0000176 LLI(const std::string &Path, const std::vector<std::string> *Args)
177 : LLIPath(Path) {
178 ToolArgs.clear ();
Brian Gaeke48b008d2004-05-04 22:02:41 +0000179 if (Args) { ToolArgs = *Args; }
Brian Gaeked11577b2004-05-04 21:09:01 +0000180 }
Misha Brukmanf976c852005-04-21 22:55:34 +0000181
Gabor Greif8ff70c22007-07-04 21:55:50 +0000182 virtual int ExecuteProgram(const std::string &Bitcode,
Chris Lattner2cdd21c2003-12-14 21:35:53 +0000183 const std::vector<std::string> &Args,
184 const std::string &InputFile,
185 const std::string &OutputFile,
Reid Spencer51ab5c82006-06-06 00:00:42 +0000186 const std::vector<std::string> &GCCArgs,
Misha Brukmanf976c852005-04-21 22:55:34 +0000187 const std::vector<std::string> &SharedLibs =
Chris Lattnere96b2ed2004-07-24 07:49:11 +0000188 std::vector<std::string>(),
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000189 unsigned Timeout = 0,
190 unsigned MemoryLimit = 0);
Chris Lattner2cdd21c2003-12-14 21:35:53 +0000191 };
192}
Misha Brukman9558c6a2003-09-29 22:39:25 +0000193
Gabor Greif8ff70c22007-07-04 21:55:50 +0000194int LLI::ExecuteProgram(const std::string &Bitcode,
Chris Lattner7915a1e2003-10-14 21:34:11 +0000195 const std::vector<std::string> &Args,
Misha Brukman9558c6a2003-09-29 22:39:25 +0000196 const std::string &InputFile,
197 const std::string &OutputFile,
Reid Spencer51ab5c82006-06-06 00:00:42 +0000198 const std::vector<std::string> &GCCArgs,
Chris Lattnere96b2ed2004-07-24 07:49:11 +0000199 const std::vector<std::string> &SharedLibs,
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000200 unsigned Timeout,
201 unsigned MemoryLimit) {
Chris Lattner8c56be52004-02-18 20:21:57 +0000202 if (!SharedLibs.empty())
203 throw ToolExecutionError("LLI currently does not support "
204 "loading shared libraries.");
Misha Brukman9558c6a2003-09-29 22:39:25 +0000205
206 std::vector<const char*> LLIArgs;
207 LLIArgs.push_back(LLIPath.c_str());
Misha Brukman9558c6a2003-09-29 22:39:25 +0000208 LLIArgs.push_back("-force-interpreter=true");
Brian Gaeked11577b2004-05-04 21:09:01 +0000209
210 // Add any extra LLI args.
211 for (unsigned i = 0, e = ToolArgs.size(); i != e; ++i)
212 LLIArgs.push_back(ToolArgs[i].c_str());
213
Gabor Greif8ff70c22007-07-04 21:55:50 +0000214 LLIArgs.push_back(Bitcode.c_str());
Misha Brukman9558c6a2003-09-29 22:39:25 +0000215 // Add optional parameters to the running program from Argv
216 for (unsigned i=0, e = Args.size(); i != e; ++i)
217 LLIArgs.push_back(Args[i].c_str());
218 LLIArgs.push_back(0);
219
Dan Gohmana1bdced2009-07-15 17:29:42 +0000220 outs() << "<lli>"; outs().flush();
Dan Gohman65f57c22009-07-15 16:35:29 +0000221 DEBUG(errs() << "\nAbout to run:\t";
Chris Lattner7b2ccff2003-10-19 02:14:58 +0000222 for (unsigned i=0, e = LLIArgs.size()-1; i != e; ++i)
Dan Gohman65f57c22009-07-15 16:35:29 +0000223 errs() << " " << LLIArgs[i];
224 errs() << "\n";
Misha Brukman9558c6a2003-09-29 22:39:25 +0000225 );
Reid Spencerb31baa82004-12-19 18:00:21 +0000226 return RunProgramWithTimeout(sys::Path(LLIPath), &LLIArgs[0],
Misha Brukmanf976c852005-04-21 22:55:34 +0000227 sys::Path(InputFile), sys::Path(OutputFile), sys::Path(OutputFile),
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000228 Timeout, MemoryLimit);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000229}
230
231// LLI create method - Try to find the LLI executable
Dan Gohman197f7282009-08-05 20:21:17 +0000232AbstractInterpreter *AbstractInterpreter::createLLI(const char *Argv0,
Brian Gaeked11577b2004-05-04 21:09:01 +0000233 std::string &Message,
234 const std::vector<std::string> *ToolArgs) {
Dan Gohman197f7282009-08-05 20:21:17 +0000235 std::string LLIPath =
Chris Lattner74382b72009-08-23 22:45:37 +0000236 FindExecutable("lli", Argv0, (void *)(intptr_t)&createLLI).str();
Misha Brukman9558c6a2003-09-29 22:39:25 +0000237 if (!LLIPath.empty()) {
238 Message = "Found lli: " + LLIPath + "\n";
Brian Gaeked11577b2004-05-04 21:09:01 +0000239 return new LLI(LLIPath, ToolArgs);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000240 }
241
242 Message = "Cannot find `lli' in executable directory or PATH!\n";
243 return 0;
244}
245
Anton Korobeynikov9ef74252008-04-28 20:53:48 +0000246//===---------------------------------------------------------------------===//
247// Custom execution command implementation of AbstractIntepreter interface
248//
249// Allows using a custom command for executing the bitcode, thus allows,
250// for example, to invoke a cross compiler for code generation followed by
251// a simulator that executes the generated binary.
252namespace {
253 class CustomExecutor : public AbstractInterpreter {
254 std::string ExecutionCommand;
255 std::vector<std::string> ExecutorArgs;
256 public:
257 CustomExecutor(
258 const std::string &ExecutionCmd, std::vector<std::string> ExecArgs) :
259 ExecutionCommand(ExecutionCmd), ExecutorArgs(ExecArgs) {}
260
261 virtual int ExecuteProgram(const std::string &Bitcode,
262 const std::vector<std::string> &Args,
263 const std::string &InputFile,
264 const std::string &OutputFile,
265 const std::vector<std::string> &GCCArgs,
266 const std::vector<std::string> &SharedLibs =
267 std::vector<std::string>(),
268 unsigned Timeout = 0,
269 unsigned MemoryLimit = 0);
270 };
271}
272
273int CustomExecutor::ExecuteProgram(const std::string &Bitcode,
274 const std::vector<std::string> &Args,
275 const std::string &InputFile,
276 const std::string &OutputFile,
277 const std::vector<std::string> &GCCArgs,
278 const std::vector<std::string> &SharedLibs,
279 unsigned Timeout,
280 unsigned MemoryLimit) {
281
282 std::vector<const char*> ProgramArgs;
283 ProgramArgs.push_back(ExecutionCommand.c_str());
284
285 for (std::size_t i = 0; i < ExecutorArgs.size(); ++i)
286 ProgramArgs.push_back(ExecutorArgs.at(i).c_str());
287 ProgramArgs.push_back(Bitcode.c_str());
288 ProgramArgs.push_back(0);
289
290 // Add optional parameters to the running program from Argv
291 for (unsigned i=0, e = Args.size(); i != e; ++i)
292 ProgramArgs.push_back(Args[i].c_str());
293
294 return RunProgramWithTimeout(
295 sys::Path(ExecutionCommand),
296 &ProgramArgs[0], sys::Path(InputFile), sys::Path(OutputFile),
297 sys::Path(OutputFile), Timeout, MemoryLimit);
298}
299
300// Custom execution environment create method, takes the execution command
301// as arguments
302AbstractInterpreter *AbstractInterpreter::createCustom(
Anton Korobeynikov9ef74252008-04-28 20:53:48 +0000303 std::string &Message,
304 const std::string &ExecCommandLine) {
305
306 std::string Command = "";
307 std::vector<std::string> Args;
308 std::string delimiters = " ";
309
310 // Tokenize the ExecCommandLine to the command and the args to allow
311 // defining a full command line as the command instead of just the
312 // executed program. We cannot just pass the whole string after the command
313 // as a single argument because then program sees only a single
314 // command line argument (with spaces in it: "foo bar" instead
315 // of "foo" and "bar").
316
317 // code borrowed from:
318 // http://oopweb.com/CPP/Documents/CPPHOWTO/Volume/C++Programming-HOWTO-7.html
319 std::string::size_type lastPos =
320 ExecCommandLine.find_first_not_of(delimiters, 0);
321 std::string::size_type pos =
322 ExecCommandLine.find_first_of(delimiters, lastPos);
323
324 while (std::string::npos != pos || std::string::npos != lastPos) {
325 std::string token = ExecCommandLine.substr(lastPos, pos - lastPos);
326 if (Command == "")
327 Command = token;
328 else
329 Args.push_back(token);
330 // Skip delimiters. Note the "not_of"
331 lastPos = ExecCommandLine.find_first_not_of(delimiters, pos);
332 // Find next "non-delimiter"
333 pos = ExecCommandLine.find_first_of(delimiters, lastPos);
334 }
335
Chris Lattner74382b72009-08-23 22:45:37 +0000336 std::string CmdPath = sys::Program::FindProgramByName(Command).str();
Anton Korobeynikov9ef74252008-04-28 20:53:48 +0000337 if (CmdPath.empty()) {
338 Message =
339 std::string("Cannot find '") + Command +
340 "' in executable directory or PATH!\n";
341 return 0;
342 }
343
344 Message = "Found command in: " + CmdPath + "\n";
345
346 return new CustomExecutor(CmdPath, Args);
347}
348
Misha Brukman9558c6a2003-09-29 22:39:25 +0000349//===----------------------------------------------------------------------===//
350// LLC Implementation of AbstractIntepreter interface
351//
Gabor Greif8ff70c22007-07-04 21:55:50 +0000352GCC::FileType LLC::OutputCode(const std::string &Bitcode,
Chris Lattnerc600f3c2006-09-15 21:29:15 +0000353 sys::Path &OutputAsmFile) {
Gabor Greif8ff70c22007-07-04 21:55:50 +0000354 sys::Path uniqueFile(Bitcode+".llc.s");
Reid Spencer51c5a282006-08-23 20:34:57 +0000355 std::string ErrMsg;
356 if (uniqueFile.makeUnique(true, &ErrMsg)) {
Dan Gohman65f57c22009-07-15 16:35:29 +0000357 errs() << "Error making unique filename: " << ErrMsg << "\n";
Reid Spencer51c5a282006-08-23 20:34:57 +0000358 exit(1);
359 }
Reid Spencer9ac14182004-12-16 23:01:34 +0000360 OutputAsmFile = uniqueFile;
Brian Gaeked11577b2004-05-04 21:09:01 +0000361 std::vector<const char *> LLCArgs;
362 LLCArgs.push_back (LLCPath.c_str());
363
364 // Add any extra LLC args.
365 for (unsigned i = 0, e = ToolArgs.size(); i != e; ++i)
366 LLCArgs.push_back(ToolArgs[i].c_str());
367
368 LLCArgs.push_back ("-o");
369 LLCArgs.push_back (OutputAsmFile.c_str()); // Output to the Asm file
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.
Chris Lattner74382b72009-08-23 22:45:37 +0000410 return gcc->ExecuteProgram(OutputAsmFile.str(), 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///
Dan Gohman197f7282009-08-05 20:21:17 +0000417LLC *AbstractInterpreter::createLLC(const char *Argv0,
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) {
Dan Gohman197f7282009-08-05 20:21:17 +0000421 std::string LLCPath =
Chris Lattner74382b72009-08-23 22:45:37 +0000422 FindExecutable("llc", Argv0, (void *)(intptr_t)&createLLC).str();
Misha Brukman9558c6a2003-09-29 22:39:25 +0000423 if (LLCPath.empty()) {
424 Message = "Cannot find `llc' in executable directory or PATH!\n";
425 return 0;
426 }
427
428 Message = "Found llc: " + LLCPath + "\n";
Dan Gohman197f7282009-08-05 20:21:17 +0000429 GCC *gcc = GCC::create(Message, GCCArgs);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000430 if (!gcc) {
Dan Gohman65f57c22009-07-15 16:35:29 +0000431 errs() << Message << "\n";
Misha Brukman9558c6a2003-09-29 22:39:25 +0000432 exit(1);
433 }
Bill Wendling38efa382009-03-02 23:13:18 +0000434 return new LLC(LLCPath, gcc, Args, GCCArgs);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000435}
436
437//===---------------------------------------------------------------------===//
438// JIT Implementation of AbstractIntepreter interface
439//
Chris Lattner2cdd21c2003-12-14 21:35:53 +0000440namespace {
441 class JIT : public AbstractInterpreter {
442 std::string LLIPath; // The path to the LLI executable
Brian Gaeked11577b2004-05-04 21:09:01 +0000443 std::vector<std::string> ToolArgs; // Args to pass to LLI
Chris Lattner2cdd21c2003-12-14 21:35:53 +0000444 public:
Brian Gaeked11577b2004-05-04 21:09:01 +0000445 JIT(const std::string &Path, const std::vector<std::string> *Args)
446 : LLIPath(Path) {
447 ToolArgs.clear ();
Brian Gaeke48b008d2004-05-04 22:02:41 +0000448 if (Args) { ToolArgs = *Args; }
Brian Gaeked11577b2004-05-04 21:09:01 +0000449 }
Misha Brukmanf976c852005-04-21 22:55:34 +0000450
Gabor Greif8ff70c22007-07-04 21:55:50 +0000451 virtual int ExecuteProgram(const std::string &Bitcode,
Chris Lattner2cdd21c2003-12-14 21:35:53 +0000452 const std::vector<std::string> &Args,
453 const std::string &InputFile,
454 const std::string &OutputFile,
Reid Spencer51ab5c82006-06-06 00:00:42 +0000455 const std::vector<std::string> &GCCArgs =
456 std::vector<std::string>(),
Misha Brukmanf976c852005-04-21 22:55:34 +0000457 const std::vector<std::string> &SharedLibs =
Reid Spencer51ab5c82006-06-06 00:00:42 +0000458 std::vector<std::string>(),
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000459 unsigned Timeout =0,
460 unsigned MemoryLimit =0);
Chris Lattner2cdd21c2003-12-14 21:35:53 +0000461 };
462}
Misha Brukman9558c6a2003-09-29 22:39:25 +0000463
Gabor Greif8ff70c22007-07-04 21:55:50 +0000464int JIT::ExecuteProgram(const std::string &Bitcode,
Chris Lattner7915a1e2003-10-14 21:34:11 +0000465 const std::vector<std::string> &Args,
Misha Brukman9558c6a2003-09-29 22:39:25 +0000466 const std::string &InputFile,
467 const std::string &OutputFile,
Reid Spencer51ab5c82006-06-06 00:00:42 +0000468 const std::vector<std::string> &GCCArgs,
Chris Lattnere96b2ed2004-07-24 07:49:11 +0000469 const std::vector<std::string> &SharedLibs,
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000470 unsigned Timeout,
471 unsigned MemoryLimit) {
Misha Brukman9558c6a2003-09-29 22:39:25 +0000472 // Construct a vector of parameters, incorporating those from the command-line
473 std::vector<const char*> JITArgs;
474 JITArgs.push_back(LLIPath.c_str());
Misha Brukman9558c6a2003-09-29 22:39:25 +0000475 JITArgs.push_back("-force-interpreter=false");
Chris Lattnereeed9832003-10-14 21:52:52 +0000476
Brian Gaeked11577b2004-05-04 21:09:01 +0000477 // Add any extra LLI args.
478 for (unsigned i = 0, e = ToolArgs.size(); i != e; ++i)
479 JITArgs.push_back(ToolArgs[i].c_str());
480
Chris Lattnereeed9832003-10-14 21:52:52 +0000481 for (unsigned i = 0, e = SharedLibs.size(); i != e; ++i) {
Misha Brukman9558c6a2003-09-29 22:39:25 +0000482 JITArgs.push_back("-load");
Chris Lattnereeed9832003-10-14 21:52:52 +0000483 JITArgs.push_back(SharedLibs[i].c_str());
Misha Brukman9558c6a2003-09-29 22:39:25 +0000484 }
Gabor Greif8ff70c22007-07-04 21:55:50 +0000485 JITArgs.push_back(Bitcode.c_str());
Misha Brukman9558c6a2003-09-29 22:39:25 +0000486 // Add optional parameters to the running program from Argv
487 for (unsigned i=0, e = Args.size(); i != e; ++i)
488 JITArgs.push_back(Args[i].c_str());
489 JITArgs.push_back(0);
490
Dan Gohmana1bdced2009-07-15 17:29:42 +0000491 outs() << "<jit>"; outs().flush();
Dan Gohman65f57c22009-07-15 16:35:29 +0000492 DEBUG(errs() << "\nAbout to run:\t";
Chris Lattner7b2ccff2003-10-19 02:14:58 +0000493 for (unsigned i=0, e = JITArgs.size()-1; i != e; ++i)
Dan Gohman65f57c22009-07-15 16:35:29 +0000494 errs() << " " << JITArgs[i];
495 errs() << "\n";
Misha Brukman9558c6a2003-09-29 22:39:25 +0000496 );
Dan Gohman65f57c22009-07-15 16:35:29 +0000497 DEBUG(errs() << "\nSending output to " << OutputFile << "\n");
Reid Spencerb31baa82004-12-19 18:00:21 +0000498 return RunProgramWithTimeout(sys::Path(LLIPath), &JITArgs[0],
Misha Brukmanf976c852005-04-21 22:55:34 +0000499 sys::Path(InputFile), sys::Path(OutputFile), sys::Path(OutputFile),
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000500 Timeout, MemoryLimit);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000501}
502
Chris Lattner7915a1e2003-10-14 21:34:11 +0000503/// createJIT - Try to find the LLI executable
Misha Brukman9558c6a2003-09-29 22:39:25 +0000504///
Dan Gohman197f7282009-08-05 20:21:17 +0000505AbstractInterpreter *AbstractInterpreter::createJIT(const char *Argv0,
Brian Gaeked11577b2004-05-04 21:09:01 +0000506 std::string &Message, const std::vector<std::string> *Args) {
Dan Gohman197f7282009-08-05 20:21:17 +0000507 std::string LLIPath =
Chris Lattner74382b72009-08-23 22:45:37 +0000508 FindExecutable("lli", Argv0, (void *)(intptr_t)&createJIT).str();
Misha Brukman9558c6a2003-09-29 22:39:25 +0000509 if (!LLIPath.empty()) {
510 Message = "Found lli: " + LLIPath + "\n";
Brian Gaeked11577b2004-05-04 21:09:01 +0000511 return new JIT(LLIPath, Args);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000512 }
513
514 Message = "Cannot find `lli' in executable directory or PATH!\n";
515 return 0;
516}
517
Gabor Greif8ff70c22007-07-04 21:55:50 +0000518GCC::FileType CBE::OutputCode(const std::string &Bitcode,
Chris Lattnerc600f3c2006-09-15 21:29:15 +0000519 sys::Path &OutputCFile) {
Gabor Greif8ff70c22007-07-04 21:55:50 +0000520 sys::Path uniqueFile(Bitcode+".cbe.c");
Reid Spencer51c5a282006-08-23 20:34:57 +0000521 std::string ErrMsg;
522 if (uniqueFile.makeUnique(true, &ErrMsg)) {
Dan Gohman65f57c22009-07-15 16:35:29 +0000523 errs() << "Error making unique filename: " << ErrMsg << "\n";
Reid Spencer51c5a282006-08-23 20:34:57 +0000524 exit(1);
525 }
Reid Spencer9ac14182004-12-16 23:01:34 +0000526 OutputCFile = uniqueFile;
Brian Gaeked11577b2004-05-04 21:09:01 +0000527 std::vector<const char *> LLCArgs;
528 LLCArgs.push_back (LLCPath.c_str());
529
530 // Add any extra LLC args.
531 for (unsigned i = 0, e = ToolArgs.size(); i != e; ++i)
532 LLCArgs.push_back(ToolArgs[i].c_str());
533
534 LLCArgs.push_back ("-o");
535 LLCArgs.push_back (OutputCFile.c_str()); // Output to the C file
536 LLCArgs.push_back ("-march=c"); // Output C language
537 LLCArgs.push_back ("-f"); // Overwrite as necessary...
Gabor Greif8ff70c22007-07-04 21:55:50 +0000538 LLCArgs.push_back (Bitcode.c_str()); // This is the input bitcode
Brian Gaeked11577b2004-05-04 21:09:01 +0000539 LLCArgs.push_back (0);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000540
Dan Gohmana1bdced2009-07-15 17:29:42 +0000541 outs() << "<cbe>"; outs().flush();
Dan Gohman65f57c22009-07-15 16:35:29 +0000542 DEBUG(errs() << "\nAbout to run:\t";
Brian Gaeked11577b2004-05-04 21:09:01 +0000543 for (unsigned i=0, e = LLCArgs.size()-1; i != e; ++i)
Dan Gohman65f57c22009-07-15 16:35:29 +0000544 errs() << " " << LLCArgs[i];
545 errs() << "\n";
Brian Gaeked11577b2004-05-04 21:09:01 +0000546 );
Misha Brukmanf976c852005-04-21 22:55:34 +0000547 if (RunProgramWithTimeout(LLCPath, &LLCArgs[0], sys::Path(), sys::Path(),
Reid Spencerb31baa82004-12-19 18:00:21 +0000548 sys::Path()))
Brian Gaeked11577b2004-05-04 21:09:01 +0000549 ProcessFailure(LLCPath, &LLCArgs[0]);
Chris Lattnerc600f3c2006-09-15 21:29:15 +0000550 return GCC::CFile;
Misha Brukman9558c6a2003-09-29 22:39:25 +0000551}
552
Gabor Greif8ff70c22007-07-04 21:55:50 +0000553void CBE::compileProgram(const std::string &Bitcode) {
Reid Spencer9ac14182004-12-16 23:01:34 +0000554 sys::Path OutputCFile;
Gabor Greif8ff70c22007-07-04 21:55:50 +0000555 OutputCode(Bitcode, OutputCFile);
Reid Spencera229c5c2005-07-08 03:08:58 +0000556 OutputCFile.eraseFromDisk();
Chris Lattner9cbbee32004-02-18 23:24:41 +0000557}
558
Gabor Greif8ff70c22007-07-04 21:55:50 +0000559int CBE::ExecuteProgram(const std::string &Bitcode,
Chris Lattner7915a1e2003-10-14 21:34:11 +0000560 const std::vector<std::string> &Args,
Misha Brukman9558c6a2003-09-29 22:39:25 +0000561 const std::string &InputFile,
562 const std::string &OutputFile,
Reid Spencer51ab5c82006-06-06 00:00:42 +0000563 const std::vector<std::string> &ArgsForGCC,
Chris Lattnere96b2ed2004-07-24 07:49:11 +0000564 const std::vector<std::string> &SharedLibs,
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000565 unsigned Timeout,
566 unsigned MemoryLimit) {
Reid Spencer9ac14182004-12-16 23:01:34 +0000567 sys::Path OutputCFile;
Gabor Greif8ff70c22007-07-04 21:55:50 +0000568 OutputCode(Bitcode, OutputCFile);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000569
Anton Korobeynikov86c006a2009-08-05 09:32:10 +0000570 FileRemover CFileRemove(OutputCFile, !SaveTemps);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000571
Reid Spencer51ab5c82006-06-06 00:00:42 +0000572 std::vector<std::string> GCCArgs(ArgsForGCC);
Bill Wendling38efa382009-03-02 23:13:18 +0000573 GCCArgs.insert(GCCArgs.end(), SharedLibs.begin(), SharedLibs.end());
574
Chris Lattner74382b72009-08-23 22:45:37 +0000575 return gcc->ExecuteProgram(OutputCFile.str(), Args, GCC::CFile,
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000576 InputFile, OutputFile, GCCArgs,
577 Timeout, MemoryLimit);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000578}
579
Chris Lattner9915cd92004-02-17 06:40:06 +0000580/// createCBE - Try to find the 'llc' executable
Misha Brukman9558c6a2003-09-29 22:39:25 +0000581///
Dan Gohman197f7282009-08-05 20:21:17 +0000582CBE *AbstractInterpreter::createCBE(const char *Argv0,
Brian Gaeked11577b2004-05-04 21:09:01 +0000583 std::string &Message,
Bill Wendling38efa382009-03-02 23:13:18 +0000584 const std::vector<std::string> *Args,
585 const std::vector<std::string> *GCCArgs) {
Dan Gohman197f7282009-08-05 20:21:17 +0000586 sys::Path LLCPath =
Dan Gohman8608cf22009-08-05 21:03:39 +0000587 FindExecutable("llc", Argv0, (void *)(intptr_t)&createCBE);
Reid Spencerb31baa82004-12-19 18:00:21 +0000588 if (LLCPath.isEmpty()) {
Misha Brukmanf976c852005-04-21 22:55:34 +0000589 Message =
Chris Lattner9915cd92004-02-17 06:40:06 +0000590 "Cannot find `llc' in executable directory or PATH!\n";
Misha Brukman9558c6a2003-09-29 22:39:25 +0000591 return 0;
592 }
593
Chris Lattner74382b72009-08-23 22:45:37 +0000594 Message = "Found llc: " + LLCPath.str() + "\n";
Dan Gohman197f7282009-08-05 20:21:17 +0000595 GCC *gcc = GCC::create(Message, GCCArgs);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000596 if (!gcc) {
Dan Gohman65f57c22009-07-15 16:35:29 +0000597 errs() << Message << "\n";
Misha Brukman9558c6a2003-09-29 22:39:25 +0000598 exit(1);
599 }
Brian Gaeked11577b2004-05-04 21:09:01 +0000600 return new CBE(LLCPath, gcc, Args);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000601}
602
603//===---------------------------------------------------------------------===//
604// GCC abstraction
605//
David Goodwin80becf12009-07-10 21:39:28 +0000606
607static bool
608IsARMArchitecture(std::vector<std::string> Args)
609{
610 for (std::vector<std::string>::const_iterator
611 I = Args.begin(), E = Args.end(); I != E; ++I) {
Benjamin Kramer05872ea2009-11-12 20:36:59 +0000612 StringRef S(*I);
613 if (!S.equals_lower("-arch")) {
David Goodwin80becf12009-07-10 21:39:28 +0000614 ++I;
Benjamin Kramer05872ea2009-11-12 20:36:59 +0000615 if (I != E && !S.substr(0, strlen("arm")).equals_lower("arm")) {
David Goodwin80becf12009-07-10 21:39:28 +0000616 return true;
617 }
618 }
619 }
620
621 return false;
622}
623
Misha Brukman9558c6a2003-09-29 22:39:25 +0000624int GCC::ExecuteProgram(const std::string &ProgramFile,
Chris Lattner7915a1e2003-10-14 21:34:11 +0000625 const std::vector<std::string> &Args,
Misha Brukman9558c6a2003-09-29 22:39:25 +0000626 FileType fileType,
627 const std::string &InputFile,
628 const std::string &OutputFile,
Reid Spencer51ab5c82006-06-06 00:00:42 +0000629 const std::vector<std::string> &ArgsForGCC,
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000630 unsigned Timeout,
631 unsigned MemoryLimit) {
Misha Brukman9558c6a2003-09-29 22:39:25 +0000632 std::vector<const char*> GCCArgs;
633
634 GCCArgs.push_back(GCCPath.c_str());
Chris Lattnereeed9832003-10-14 21:52:52 +0000635
Bill Wendling38efa382009-03-02 23:13:18 +0000636 for (std::vector<std::string>::const_iterator
637 I = gccArgs.begin(), E = gccArgs.end(); I != E; ++I)
638 GCCArgs.push_back(I->c_str());
639
Chris Lattnereeed9832003-10-14 21:52:52 +0000640 // Specify -x explicitly in case the extension is wonky
Misha Brukman9558c6a2003-09-29 22:39:25 +0000641 GCCArgs.push_back("-x");
642 if (fileType == CFile) {
643 GCCArgs.push_back("c");
644 GCCArgs.push_back("-fno-strict-aliasing");
645 } else {
646 GCCArgs.push_back("assembler");
Daniel Dunbarca740962009-08-18 03:35:57 +0000647
David Goodwin80becf12009-07-10 21:39:28 +0000648 // For ARM architectures we don't want this flag. bugpoint isn't
649 // explicitly told what architecture it is working on, so we get
650 // it from gcc flags
Daniel Dunbarca740962009-08-18 03:35:57 +0000651 if ((TargetTriple.getOS() == Triple::Darwin) &&
652 !IsARMArchitecture(ArgsForGCC))
David Goodwin80becf12009-07-10 21:39:28 +0000653 GCCArgs.push_back("-force_cpusubtype_ALL");
Misha Brukman9558c6a2003-09-29 22:39:25 +0000654 }
655 GCCArgs.push_back(ProgramFile.c_str()); // Specify the input filename...
Chris Lattner629e4872006-06-09 21:31:53 +0000656 GCCArgs.push_back("-x");
657 GCCArgs.push_back("none");
Misha Brukman9558c6a2003-09-29 22:39:25 +0000658 GCCArgs.push_back("-o");
Reid Spencercda985e2004-12-15 01:51:56 +0000659 sys::Path OutputBinary (ProgramFile+".gcc.exe");
Reid Spencer51c5a282006-08-23 20:34:57 +0000660 std::string ErrMsg;
661 if (OutputBinary.makeUnique(true, &ErrMsg)) {
Dan Gohman65f57c22009-07-15 16:35:29 +0000662 errs() << "Error making unique filename: " << ErrMsg << "\n";
Reid Spencer51c5a282006-08-23 20:34:57 +0000663 exit(1);
664 }
Misha Brukman9558c6a2003-09-29 22:39:25 +0000665 GCCArgs.push_back(OutputBinary.c_str()); // Output to the right file...
Reid Spencer51ab5c82006-06-06 00:00:42 +0000666
667 // Add any arguments intended for GCC. We locate them here because this is
668 // most likely -L and -l options that need to come before other libraries but
669 // after the source. Other options won't be sensitive to placement on the
670 // command line, so this should be safe.
671 for (unsigned i = 0, e = ArgsForGCC.size(); i != e; ++i)
672 GCCArgs.push_back(ArgsForGCC[i].c_str());
673
Misha Brukman9558c6a2003-09-29 22:39:25 +0000674 GCCArgs.push_back("-lm"); // Hard-code the math library...
675 GCCArgs.push_back("-O2"); // Optimize the program a bit...
Brian Gaekec8db76c2003-11-18 06:31:17 +0000676#if defined (HAVE_LINK_R)
Chris Lattner1f0f1622003-10-18 21:54:47 +0000677 GCCArgs.push_back("-Wl,-R."); // Search this dir for .so files
Brian Gaekec8db76c2003-11-18 06:31:17 +0000678#endif
Daniel Dunbarca740962009-08-18 03:35:57 +0000679 if (TargetTriple.getArch() == Triple::sparc)
680 GCCArgs.push_back("-mcpu=v9");
Misha Brukman9558c6a2003-09-29 22:39:25 +0000681 GCCArgs.push_back(0); // NULL terminator
682
Dan Gohmana1bdced2009-07-15 17:29:42 +0000683 outs() << "<gcc>"; outs().flush();
Dan Gohman65f57c22009-07-15 16:35:29 +0000684 DEBUG(errs() << "\nAbout to run:\t";
Evan Cheng7f7fdcc2007-01-03 07:44:30 +0000685 for (unsigned i=0, e = GCCArgs.size()-1; i != e; ++i)
Dan Gohman65f57c22009-07-15 16:35:29 +0000686 errs() << " " << GCCArgs[i];
687 errs() << "\n";
Evan Cheng7f7fdcc2007-01-03 07:44:30 +0000688 );
Reid Spencerb31baa82004-12-19 18:00:21 +0000689 if (RunProgramWithTimeout(GCCPath, &GCCArgs[0], sys::Path(), sys::Path(),
690 sys::Path())) {
Chris Lattner89bf9ea2004-02-18 20:38:00 +0000691 ProcessFailure(GCCPath, &GCCArgs[0]);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000692 exit(1);
693 }
694
695 std::vector<const char*> ProgramArgs;
Chris Lattner45495c52005-02-13 23:13:47 +0000696
Evan Cheng70f684f2008-09-09 06:11:26 +0000697 if (RemoteClientPath.isEmpty())
Evan Cheng34e400e2007-05-03 18:36:15 +0000698 ProgramArgs.push_back(OutputBinary.c_str());
699 else {
Evan Cheng70f684f2008-09-09 06:11:26 +0000700 ProgramArgs.push_back(RemoteClientPath.c_str());
701 ProgramArgs.push_back(RemoteHost.c_str());
Anton Korobeynikov26964b62009-08-05 09:32:53 +0000702 if (!RemoteUser.empty()) {
703 ProgramArgs.push_back("-l");
704 ProgramArgs.push_back(RemoteUser.c_str());
705 }
David Goodwin80becf12009-07-10 21:39:28 +0000706 if (!RemotePort.empty()) {
707 ProgramArgs.push_back("-p");
708 ProgramArgs.push_back(RemotePort.c_str());
709 }
Evan Cheng70f684f2008-09-09 06:11:26 +0000710 if (!RemoteExtra.empty()) {
711 ProgramArgs.push_back(RemoteExtra.c_str());
712 }
Evan Cheng34e400e2007-05-03 18:36:15 +0000713
David Goodwin91cf3612009-07-20 17:15:03 +0000714 // Full path to the binary. We need to cd to the exec directory because
715 // there is a dylib there that the exec expects to find in the CWD
Evan Cheng34e400e2007-05-03 18:36:15 +0000716 char* env_pwd = getenv("PWD");
David Goodwin91cf3612009-07-20 17:15:03 +0000717 std::string Exec = "cd ";
Evan Cheng34e400e2007-05-03 18:36:15 +0000718 Exec += env_pwd;
David Goodwin91cf3612009-07-20 17:15:03 +0000719 Exec += "; ./";
Evan Cheng34e400e2007-05-03 18:36:15 +0000720 Exec += OutputBinary.c_str();
721 ProgramArgs.push_back(Exec.c_str());
722 }
723
Misha Brukman9558c6a2003-09-29 22:39:25 +0000724 // Add optional parameters to the running program from Argv
725 for (unsigned i=0, e = Args.size(); i != e; ++i)
726 ProgramArgs.push_back(Args[i].c_str());
727 ProgramArgs.push_back(0); // NULL terminator
728
729 // Now that we have a binary, run it!
Dan Gohmana1bdced2009-07-15 17:29:42 +0000730 outs() << "<program>"; outs().flush();
Dan Gohman65f57c22009-07-15 16:35:29 +0000731 DEBUG(errs() << "\nAbout to run:\t";
Chris Lattner7b2ccff2003-10-19 02:14:58 +0000732 for (unsigned i=0, e = ProgramArgs.size()-1; i != e; ++i)
Dan Gohman65f57c22009-07-15 16:35:29 +0000733 errs() << " " << ProgramArgs[i];
734 errs() << "\n";
Misha Brukman9558c6a2003-09-29 22:39:25 +0000735 );
Chris Lattner8c56be52004-02-18 20:21:57 +0000736
Anton Korobeynikov86c006a2009-08-05 09:32:10 +0000737 FileRemover OutputBinaryRemover(OutputBinary, !SaveTemps);
Evan Cheng34e400e2007-05-03 18:36:15 +0000738
Viktor Kutuzov699220b2009-07-14 19:10:55 +0000739 if (RemoteClientPath.isEmpty()) {
Dan Gohman65f57c22009-07-15 16:35:29 +0000740 DEBUG(errs() << "<run locally>";);
Evan Cheng34e400e2007-05-03 18:36:15 +0000741 return RunProgramWithTimeout(OutputBinary, &ProgramArgs[0],
742 sys::Path(InputFile), sys::Path(OutputFile), sys::Path(OutputFile),
743 Timeout, MemoryLimit);
Viktor Kutuzov699220b2009-07-14 19:10:55 +0000744 } else {
Dan Gohmana1bdced2009-07-15 17:29:42 +0000745 outs() << "<run remotely>"; outs().flush();
Viktor Kutuzovfc2271f2009-07-18 18:39:24 +0000746 return RunProgramRemotelyWithTimeout(sys::Path(RemoteClientPath),
Viktor Kutuzov699220b2009-07-14 19:10:55 +0000747 &ProgramArgs[0], sys::Path(InputFile), sys::Path(OutputFile),
748 sys::Path(OutputFile), Timeout, MemoryLimit);
Viktor Kutuzov699220b2009-07-14 19:10:55 +0000749 }
Misha Brukman9558c6a2003-09-29 22:39:25 +0000750}
751
Chris Lattner1798e4a2003-10-14 21:07:25 +0000752int GCC::MakeSharedObject(const std::string &InputFile, FileType fileType,
Chris Lattner130e2a32006-06-27 20:35:36 +0000753 std::string &OutputFile,
754 const std::vector<std::string> &ArgsForGCC) {
Reid Spencercda985e2004-12-15 01:51:56 +0000755 sys::Path uniqueFilename(InputFile+LTDL_SHLIB_EXT);
Reid Spencer51c5a282006-08-23 20:34:57 +0000756 std::string ErrMsg;
757 if (uniqueFilename.makeUnique(true, &ErrMsg)) {
Dan Gohman65f57c22009-07-15 16:35:29 +0000758 errs() << "Error making unique filename: " << ErrMsg << "\n";
Reid Spencer51c5a282006-08-23 20:34:57 +0000759 exit(1);
760 }
Chris Lattner74382b72009-08-23 22:45:37 +0000761 OutputFile = uniqueFilename.str();
Reid Spencercda985e2004-12-15 01:51:56 +0000762
Chris Lattner130e2a32006-06-27 20:35:36 +0000763 std::vector<const char*> GCCArgs;
764
765 GCCArgs.push_back(GCCPath.c_str());
Evan Chengfd829952009-03-12 00:53:34 +0000766
767 for (std::vector<std::string>::const_iterator
768 I = gccArgs.begin(), E = gccArgs.end(); I != E; ++I)
769 GCCArgs.push_back(I->c_str());
770
Misha Brukman9558c6a2003-09-29 22:39:25 +0000771 // Compile the C/asm file into a shared object
Chris Lattner130e2a32006-06-27 20:35:36 +0000772 GCCArgs.push_back("-x");
773 GCCArgs.push_back(fileType == AsmFile ? "assembler" : "c");
774 GCCArgs.push_back("-fno-strict-aliasing");
775 GCCArgs.push_back(InputFile.c_str()); // Specify the input filename.
Lauro Ramos Venancio497391a2007-06-06 23:10:56 +0000776 GCCArgs.push_back("-x");
777 GCCArgs.push_back("none");
Daniel Dunbarca740962009-08-18 03:35:57 +0000778 if (TargetTriple.getArch() == Triple::sparc)
779 GCCArgs.push_back("-G"); // Compile a shared library, `-G' for Sparc
780 else if (TargetTriple.getOS() == Triple::Darwin) {
781 // link all source files into a single module in data segment, rather than
782 // generating blocks. dynamic_lookup requires that you set
783 // MACOSX_DEPLOYMENT_TARGET=10.3 in your env. FIXME: it would be better for
784 // bugpoint to just pass that in the environment of GCC.
785 GCCArgs.push_back("-single_module");
786 GCCArgs.push_back("-dynamiclib"); // `-dynamiclib' for MacOS X/PowerPC
787 GCCArgs.push_back("-undefined");
788 GCCArgs.push_back("dynamic_lookup");
789 } else
790 GCCArgs.push_back("-shared"); // `-shared' for Linux/X86, maybe others
Chris Lattner1c81f132005-03-09 03:31:02 +0000791
Daniel Dunbarca740962009-08-18 03:35:57 +0000792 if ((TargetTriple.getArch() == Triple::alpha) ||
793 (TargetTriple.getArch() == Triple::x86_64))
794 GCCArgs.push_back("-fPIC"); // Requires shared objs to contain PIC
795
796 if (TargetTriple.getArch() == Triple::sparc)
797 GCCArgs.push_back("-mcpu=v9");
798
Chris Lattner130e2a32006-06-27 20:35:36 +0000799 GCCArgs.push_back("-o");
800 GCCArgs.push_back(OutputFile.c_str()); // Output to the right filename.
801 GCCArgs.push_back("-O2"); // Optimize the program a bit.
802
803
804
805 // Add any arguments intended for GCC. We locate them here because this is
806 // most likely -L and -l options that need to come before other libraries but
807 // after the source. Other options won't be sensitive to placement on the
808 // command line, so this should be safe.
809 for (unsigned i = 0, e = ArgsForGCC.size(); i != e; ++i)
810 GCCArgs.push_back(ArgsForGCC[i].c_str());
811 GCCArgs.push_back(0); // NULL terminator
812
813
Misha Brukmanf976c852005-04-21 22:55:34 +0000814
Dan Gohmana1bdced2009-07-15 17:29:42 +0000815 outs() << "<gcc>"; outs().flush();
Dan Gohman65f57c22009-07-15 16:35:29 +0000816 DEBUG(errs() << "\nAbout to run:\t";
Evan Cheng7f7fdcc2007-01-03 07:44:30 +0000817 for (unsigned i=0, e = GCCArgs.size()-1; i != e; ++i)
Dan Gohman65f57c22009-07-15 16:35:29 +0000818 errs() << " " << GCCArgs[i];
819 errs() << "\n";
Evan Cheng7f7fdcc2007-01-03 07:44:30 +0000820 );
Chris Lattner130e2a32006-06-27 20:35:36 +0000821 if (RunProgramWithTimeout(GCCPath, &GCCArgs[0], sys::Path(), sys::Path(),
Reid Spencerb31baa82004-12-19 18:00:21 +0000822 sys::Path())) {
Chris Lattner130e2a32006-06-27 20:35:36 +0000823 ProcessFailure(GCCPath, &GCCArgs[0]);
Chris Lattner1798e4a2003-10-14 21:07:25 +0000824 return 1;
Misha Brukman9558c6a2003-09-29 22:39:25 +0000825 }
826 return 0;
827}
828
Chris Lattner7915a1e2003-10-14 21:34:11 +0000829/// create - Try to find the `gcc' executable
Misha Brukman9558c6a2003-09-29 22:39:25 +0000830///
Dan Gohman197f7282009-08-05 20:21:17 +0000831GCC *GCC::create(std::string &Message,
Bill Wendling38efa382009-03-02 23:13:18 +0000832 const std::vector<std::string> *Args) {
Dan Gohman197f7282009-08-05 20:21:17 +0000833 sys::Path GCCPath = sys::Program::FindProgramByName("gcc");
Reid Spencerb31baa82004-12-19 18:00:21 +0000834 if (GCCPath.isEmpty()) {
Misha Brukman9558c6a2003-09-29 22:39:25 +0000835 Message = "Cannot find `gcc' in executable directory or PATH!\n";
836 return 0;
837 }
838
Evan Cheng70f684f2008-09-09 06:11:26 +0000839 sys::Path RemoteClientPath;
840 if (!RemoteClient.empty())
Dan Gohman197f7282009-08-05 20:21:17 +0000841 RemoteClientPath = sys::Program::FindProgramByName(RemoteClient);
Evan Cheng34e400e2007-05-03 18:36:15 +0000842
Chris Lattner74382b72009-08-23 22:45:37 +0000843 Message = "Found gcc: " + GCCPath.str() + "\n";
Bill Wendling38efa382009-03-02 23:13:18 +0000844 return new GCC(GCCPath, RemoteClientPath, Args);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000845}