blob: 3149a7a494a99be36bfa6e8aa5730a75b156c4ff [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
Viktor Kutuzovfc2271f2009-07-18 18:39:24 +000053/// RunProgramWithTimeout - This function provides an alternate interface
54/// to the sys::Program::ExecuteAndWait interface.
Nick Lewycky22ff7482010-04-12 05:08:25 +000055/// @see sys::Program::ExecuteAndWait
Chris Lattner45495c52005-02-13 23:13:47 +000056static int RunProgramWithTimeout(const sys::Path &ProgramPath,
57 const char **Args,
58 const sys::Path &StdInFile,
59 const sys::Path &StdOutFile,
60 const sys::Path &StdErrFile,
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +000061 unsigned NumSeconds = 0,
62 unsigned MemoryLimit = 0) {
Chris Lattner45495c52005-02-13 23:13:47 +000063 const sys::Path* redirects[3];
64 redirects[0] = &StdInFile;
65 redirects[1] = &StdOutFile;
66 redirects[2] = &StdErrFile;
Viktor Kutuzovfc2271f2009-07-18 18:39:24 +000067
68#if 0 // For debug purposes
69 {
Dan Gohman65f57c22009-07-15 16:35:29 +000070 errs() << "RUN:";
Chris Lattnerd4223502006-09-15 23:01:10 +000071 for (unsigned i = 0; Args[i]; ++i)
Dan Gohman65f57c22009-07-15 16:35:29 +000072 errs() << " " << Args[i];
73 errs() << "\n";
Chris Lattnerd4223502006-09-15 23:01:10 +000074 }
Viktor Kutuzovfc2271f2009-07-18 18:39:24 +000075#endif
Misha Brukmanf976c852005-04-21 22:55:34 +000076
77 return
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +000078 sys::Program::ExecuteAndWait(ProgramPath, Args, 0, redirects,
79 NumSeconds, MemoryLimit);
Chris Lattner45495c52005-02-13 23:13:47 +000080}
81
Viktor Kutuzovfc2271f2009-07-18 18:39:24 +000082/// RunProgramRemotelyWithTimeout - This function runs the given program
83/// remotely using the given remote client and the sys::Program::ExecuteAndWait.
84/// Returns the remote program exit code or reports a remote client error if it
85/// fails. Remote client is required to return 255 if it failed or program exit
86/// code otherwise.
Nick Lewycky22ff7482010-04-12 05:08:25 +000087/// @see sys::Program::ExecuteAndWait
Viktor Kutuzovfc2271f2009-07-18 18:39:24 +000088static int RunProgramRemotelyWithTimeout(const sys::Path &RemoteClientPath,
89 const char **Args,
90 const sys::Path &StdInFile,
91 const sys::Path &StdOutFile,
92 const sys::Path &StdErrFile,
93 unsigned NumSeconds = 0,
94 unsigned MemoryLimit = 0) {
95 const sys::Path* redirects[3];
96 redirects[0] = &StdInFile;
97 redirects[1] = &StdOutFile;
98 redirects[2] = &StdErrFile;
Chris Lattner45495c52005-02-13 23:13:47 +000099
Viktor Kutuzovfc2271f2009-07-18 18:39:24 +0000100#if 0 // For debug purposes
101 {
102 errs() << "RUN:";
103 for (unsigned i = 0; Args[i]; ++i)
104 errs() << " " << Args[i];
105 errs() << "\n";
106 }
107#endif
108
109 // Run the program remotely with the remote client
110 int ReturnCode = sys::Program::ExecuteAndWait(RemoteClientPath, Args,
111 0, redirects, NumSeconds, MemoryLimit);
112
113 // Has the remote client fail?
114 if (255 == ReturnCode) {
115 std::ostringstream OS;
116 OS << "\nError running remote client:\n ";
117 for (const char **Arg = Args; *Arg; ++Arg)
118 OS << " " << *Arg;
119 OS << "\n";
120
121 // The error message is in the output file, let's print it out from there.
122 std::ifstream ErrorFile(StdOutFile.c_str());
123 if (ErrorFile) {
124 std::copy(std::istreambuf_iterator<char>(ErrorFile),
125 std::istreambuf_iterator<char>(),
126 std::ostreambuf_iterator<char>(OS));
127 ErrorFile.close();
128 }
129
Nick Lewycky22ff7482010-04-12 05:08:25 +0000130 errs() << OS;
Viktor Kutuzovfc2271f2009-07-18 18:39:24 +0000131 }
132
133 return ReturnCode;
134}
Chris Lattner45495c52005-02-13 23:13:47 +0000135
Duncan Sands41396302010-05-24 07:49:55 +0000136static std::string ProcessFailure(sys::Path ProgPath, const char** Args,
137 unsigned Timeout = 0,
138 unsigned MemoryLimit = 0) {
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";
Chris Lattner50010422010-03-16 06:41:47 +0000144
Chris Lattner89bf9ea2004-02-18 20:38:00 +0000145 // 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,
Duncan Sands41396302010-05-24 07:49:55 +0000153 ErrorFilename, Timeout, MemoryLimit);
154 // FIXME: check return code ?
Chris Lattner89bf9ea2004-02-18 20:38:00 +0000155
156 // Print out the error messages generated by GCC if possible...
157 std::ifstream ErrorFile(ErrorFilename.c_str());
158 if (ErrorFile) {
159 std::copy(std::istreambuf_iterator<char>(ErrorFile),
160 std::istreambuf_iterator<char>(),
161 std::ostreambuf_iterator<char>(OS));
162 ErrorFile.close();
163 }
164
Reid Spencera229c5c2005-07-08 03:08:58 +0000165 ErrorFilename.eraseFromDisk();
Nick Lewycky22ff7482010-04-12 05:08:25 +0000166 return OS.str();
Chris Lattner89bf9ea2004-02-18 20:38:00 +0000167}
168
Misha Brukman9558c6a2003-09-29 22:39:25 +0000169//===---------------------------------------------------------------------===//
170// LLI Implementation of AbstractIntepreter interface
171//
Chris Lattner2cdd21c2003-12-14 21:35:53 +0000172namespace {
173 class LLI : public AbstractInterpreter {
174 std::string LLIPath; // The path to the LLI executable
Brian Gaeked11577b2004-05-04 21:09:01 +0000175 std::vector<std::string> ToolArgs; // Args to pass to LLI
Chris Lattner2cdd21c2003-12-14 21:35:53 +0000176 public:
Brian Gaeked11577b2004-05-04 21:09:01 +0000177 LLI(const std::string &Path, const std::vector<std::string> *Args)
178 : LLIPath(Path) {
179 ToolArgs.clear ();
Brian Gaeke48b008d2004-05-04 22:02:41 +0000180 if (Args) { ToolArgs = *Args; }
Brian Gaeked11577b2004-05-04 21:09:01 +0000181 }
Misha Brukmanf976c852005-04-21 22:55:34 +0000182
Gabor Greif8ff70c22007-07-04 21:55:50 +0000183 virtual int ExecuteProgram(const std::string &Bitcode,
Chris Lattner2cdd21c2003-12-14 21:35:53 +0000184 const std::vector<std::string> &Args,
185 const std::string &InputFile,
186 const std::string &OutputFile,
Nick Lewycky22ff7482010-04-12 05:08:25 +0000187 std::string *Error,
Reid Spencer51ab5c82006-06-06 00:00:42 +0000188 const std::vector<std::string> &GCCArgs,
Misha Brukmanf976c852005-04-21 22:55:34 +0000189 const std::vector<std::string> &SharedLibs =
Chris Lattnere96b2ed2004-07-24 07:49:11 +0000190 std::vector<std::string>(),
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000191 unsigned Timeout = 0,
192 unsigned MemoryLimit = 0);
Chris Lattner2cdd21c2003-12-14 21:35:53 +0000193 };
194}
Misha Brukman9558c6a2003-09-29 22:39:25 +0000195
Gabor Greif8ff70c22007-07-04 21:55:50 +0000196int LLI::ExecuteProgram(const std::string &Bitcode,
Chris Lattner7915a1e2003-10-14 21:34:11 +0000197 const std::vector<std::string> &Args,
Misha Brukman9558c6a2003-09-29 22:39:25 +0000198 const std::string &InputFile,
199 const std::string &OutputFile,
Nick Lewycky22ff7482010-04-12 05:08:25 +0000200 std::string *Error,
Reid Spencer51ab5c82006-06-06 00:00:42 +0000201 const std::vector<std::string> &GCCArgs,
Chris Lattnere96b2ed2004-07-24 07:49:11 +0000202 const std::vector<std::string> &SharedLibs,
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000203 unsigned Timeout,
204 unsigned MemoryLimit) {
Misha Brukman9558c6a2003-09-29 22:39:25 +0000205 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
Duncan Sands3ea11cf2009-11-17 10:20:22 +0000209 for (std::vector<std::string>::const_iterator i = SharedLibs.begin(), e = SharedLibs.end(); i != e; ++i) {
210 LLIArgs.push_back("-load");
211 LLIArgs.push_back((*i).c_str());
212 }
213
Brian Gaeked11577b2004-05-04 21:09:01 +0000214 // Add any extra LLI args.
215 for (unsigned i = 0, e = ToolArgs.size(); i != e; ++i)
216 LLIArgs.push_back(ToolArgs[i].c_str());
217
Gabor Greif8ff70c22007-07-04 21:55:50 +0000218 LLIArgs.push_back(Bitcode.c_str());
Misha Brukman9558c6a2003-09-29 22:39:25 +0000219 // Add optional parameters to the running program from Argv
220 for (unsigned i=0, e = Args.size(); i != e; ++i)
221 LLIArgs.push_back(Args[i].c_str());
222 LLIArgs.push_back(0);
223
Dan Gohmana1bdced2009-07-15 17:29:42 +0000224 outs() << "<lli>"; outs().flush();
Dan Gohman65f57c22009-07-15 16:35:29 +0000225 DEBUG(errs() << "\nAbout to run:\t";
Chris Lattner7b2ccff2003-10-19 02:14:58 +0000226 for (unsigned i=0, e = LLIArgs.size()-1; i != e; ++i)
Dan Gohman65f57c22009-07-15 16:35:29 +0000227 errs() << " " << LLIArgs[i];
228 errs() << "\n";
Misha Brukman9558c6a2003-09-29 22:39:25 +0000229 );
Reid Spencerb31baa82004-12-19 18:00:21 +0000230 return RunProgramWithTimeout(sys::Path(LLIPath), &LLIArgs[0],
Misha Brukmanf976c852005-04-21 22:55:34 +0000231 sys::Path(InputFile), sys::Path(OutputFile), sys::Path(OutputFile),
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000232 Timeout, MemoryLimit);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000233}
234
235// LLI create method - Try to find the LLI executable
Dan Gohman197f7282009-08-05 20:21:17 +0000236AbstractInterpreter *AbstractInterpreter::createLLI(const char *Argv0,
Brian Gaeked11577b2004-05-04 21:09:01 +0000237 std::string &Message,
238 const std::vector<std::string> *ToolArgs) {
Dan Gohman197f7282009-08-05 20:21:17 +0000239 std::string LLIPath =
Chris Lattner74382b72009-08-23 22:45:37 +0000240 FindExecutable("lli", Argv0, (void *)(intptr_t)&createLLI).str();
Misha Brukman9558c6a2003-09-29 22:39:25 +0000241 if (!LLIPath.empty()) {
242 Message = "Found lli: " + LLIPath + "\n";
Brian Gaeked11577b2004-05-04 21:09:01 +0000243 return new LLI(LLIPath, ToolArgs);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000244 }
245
246 Message = "Cannot find `lli' in executable directory or PATH!\n";
247 return 0;
248}
249
Anton Korobeynikov9ef74252008-04-28 20:53:48 +0000250//===---------------------------------------------------------------------===//
251// Custom execution command implementation of AbstractIntepreter interface
252//
253// Allows using a custom command for executing the bitcode, thus allows,
254// for example, to invoke a cross compiler for code generation followed by
255// a simulator that executes the generated binary.
256namespace {
257 class CustomExecutor : public AbstractInterpreter {
258 std::string ExecutionCommand;
259 std::vector<std::string> ExecutorArgs;
260 public:
261 CustomExecutor(
262 const std::string &ExecutionCmd, std::vector<std::string> ExecArgs) :
263 ExecutionCommand(ExecutionCmd), ExecutorArgs(ExecArgs) {}
264
265 virtual int ExecuteProgram(const std::string &Bitcode,
266 const std::vector<std::string> &Args,
267 const std::string &InputFile,
268 const std::string &OutputFile,
Nick Lewycky22ff7482010-04-12 05:08:25 +0000269 std::string *Error,
Anton Korobeynikov9ef74252008-04-28 20:53:48 +0000270 const std::vector<std::string> &GCCArgs,
271 const std::vector<std::string> &SharedLibs =
Nick Lewycky22ff7482010-04-12 05:08:25 +0000272 std::vector<std::string>(),
Anton Korobeynikov9ef74252008-04-28 20:53:48 +0000273 unsigned Timeout = 0,
274 unsigned MemoryLimit = 0);
275 };
276}
277
278int CustomExecutor::ExecuteProgram(const std::string &Bitcode,
279 const std::vector<std::string> &Args,
280 const std::string &InputFile,
281 const std::string &OutputFile,
Nick Lewycky22ff7482010-04-12 05:08:25 +0000282 std::string *Error,
Anton Korobeynikov9ef74252008-04-28 20:53:48 +0000283 const std::vector<std::string> &GCCArgs,
284 const std::vector<std::string> &SharedLibs,
285 unsigned Timeout,
286 unsigned MemoryLimit) {
287
288 std::vector<const char*> ProgramArgs;
289 ProgramArgs.push_back(ExecutionCommand.c_str());
290
291 for (std::size_t i = 0; i < ExecutorArgs.size(); ++i)
292 ProgramArgs.push_back(ExecutorArgs.at(i).c_str());
293 ProgramArgs.push_back(Bitcode.c_str());
294 ProgramArgs.push_back(0);
295
296 // Add optional parameters to the running program from Argv
Nick Lewycky22ff7482010-04-12 05:08:25 +0000297 for (unsigned i = 0, e = Args.size(); i != e; ++i)
Anton Korobeynikov9ef74252008-04-28 20:53:48 +0000298 ProgramArgs.push_back(Args[i].c_str());
299
300 return RunProgramWithTimeout(
301 sys::Path(ExecutionCommand),
302 &ProgramArgs[0], sys::Path(InputFile), sys::Path(OutputFile),
303 sys::Path(OutputFile), Timeout, MemoryLimit);
304}
305
306// Custom execution environment create method, takes the execution command
307// as arguments
308AbstractInterpreter *AbstractInterpreter::createCustom(
Anton Korobeynikov9ef74252008-04-28 20:53:48 +0000309 std::string &Message,
310 const std::string &ExecCommandLine) {
311
312 std::string Command = "";
313 std::vector<std::string> Args;
314 std::string delimiters = " ";
315
316 // Tokenize the ExecCommandLine to the command and the args to allow
317 // defining a full command line as the command instead of just the
318 // executed program. We cannot just pass the whole string after the command
319 // as a single argument because then program sees only a single
320 // command line argument (with spaces in it: "foo bar" instead
321 // of "foo" and "bar").
322
323 // code borrowed from:
324 // http://oopweb.com/CPP/Documents/CPPHOWTO/Volume/C++Programming-HOWTO-7.html
325 std::string::size_type lastPos =
326 ExecCommandLine.find_first_not_of(delimiters, 0);
327 std::string::size_type pos =
328 ExecCommandLine.find_first_of(delimiters, lastPos);
329
330 while (std::string::npos != pos || std::string::npos != lastPos) {
331 std::string token = ExecCommandLine.substr(lastPos, pos - lastPos);
332 if (Command == "")
333 Command = token;
334 else
335 Args.push_back(token);
336 // Skip delimiters. Note the "not_of"
337 lastPos = ExecCommandLine.find_first_not_of(delimiters, pos);
338 // Find next "non-delimiter"
339 pos = ExecCommandLine.find_first_of(delimiters, lastPos);
340 }
341
Chris Lattner74382b72009-08-23 22:45:37 +0000342 std::string CmdPath = sys::Program::FindProgramByName(Command).str();
Anton Korobeynikov9ef74252008-04-28 20:53:48 +0000343 if (CmdPath.empty()) {
344 Message =
345 std::string("Cannot find '") + Command +
346 "' in executable directory or PATH!\n";
347 return 0;
348 }
349
350 Message = "Found command in: " + CmdPath + "\n";
351
352 return new CustomExecutor(CmdPath, Args);
353}
354
Misha Brukman9558c6a2003-09-29 22:39:25 +0000355//===----------------------------------------------------------------------===//
356// LLC Implementation of AbstractIntepreter interface
357//
Gabor Greif8ff70c22007-07-04 21:55:50 +0000358GCC::FileType LLC::OutputCode(const std::string &Bitcode,
Duncan Sands41396302010-05-24 07:49:55 +0000359 sys::Path &OutputAsmFile, std::string &Error,
360 unsigned Timeout, unsigned MemoryLimit) {
Chris Lattner50010422010-03-16 06:41:47 +0000361 const char *Suffix = (UseIntegratedAssembler ? ".llc.o" : ".llc.s");
362 sys::Path uniqueFile(Bitcode + Suffix);
Reid Spencer51c5a282006-08-23 20:34:57 +0000363 std::string ErrMsg;
364 if (uniqueFile.makeUnique(true, &ErrMsg)) {
Dan Gohman65f57c22009-07-15 16:35:29 +0000365 errs() << "Error making unique filename: " << ErrMsg << "\n";
Reid Spencer51c5a282006-08-23 20:34:57 +0000366 exit(1);
367 }
Reid Spencer9ac14182004-12-16 23:01:34 +0000368 OutputAsmFile = uniqueFile;
Brian Gaeked11577b2004-05-04 21:09:01 +0000369 std::vector<const char *> LLCArgs;
Chris Lattner50010422010-03-16 06:41:47 +0000370 LLCArgs.push_back(LLCPath.c_str());
Brian Gaeked11577b2004-05-04 21:09:01 +0000371
372 // Add any extra LLC args.
373 for (unsigned i = 0, e = ToolArgs.size(); i != e; ++i)
374 LLCArgs.push_back(ToolArgs[i].c_str());
375
Chris Lattner50010422010-03-16 06:41:47 +0000376 LLCArgs.push_back("-o");
377 LLCArgs.push_back(OutputAsmFile.c_str()); // Output to the Asm file
378 LLCArgs.push_back(Bitcode.c_str()); // This is the input bitcode
379
380 if (UseIntegratedAssembler)
381 LLCArgs.push_back("-filetype=obj");
382
Brian Gaeked11577b2004-05-04 21:09:01 +0000383 LLCArgs.push_back (0);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000384
Chris Lattner50010422010-03-16 06:41:47 +0000385 outs() << (UseIntegratedAssembler ? "<llc-ia>" : "<llc>");
386 outs().flush();
Dan Gohman65f57c22009-07-15 16:35:29 +0000387 DEBUG(errs() << "\nAbout to run:\t";
Nick Lewycky22ff7482010-04-12 05:08:25 +0000388 for (unsigned i = 0, e = LLCArgs.size()-1; i != e; ++i)
Dan Gohman65f57c22009-07-15 16:35:29 +0000389 errs() << " " << LLCArgs[i];
390 errs() << "\n";
Brian Gaeked11577b2004-05-04 21:09:01 +0000391 );
Misha Brukmanf976c852005-04-21 22:55:34 +0000392 if (RunProgramWithTimeout(sys::Path(LLCPath), &LLCArgs[0],
Duncan Sands41396302010-05-24 07:49:55 +0000393 sys::Path(), sys::Path(), sys::Path(),
394 Timeout, MemoryLimit))
395 Error = ProcessFailure(sys::Path(LLCPath), &LLCArgs[0],
396 Timeout, MemoryLimit);
Nick Lewycky22ff7482010-04-12 05:08:25 +0000397 return UseIntegratedAssembler ? GCC::ObjectFile : GCC::AsmFile;
Misha Brukman9558c6a2003-09-29 22:39:25 +0000398}
399
Duncan Sands41396302010-05-24 07:49:55 +0000400void LLC::compileProgram(const std::string &Bitcode, std::string *Error,
401 unsigned Timeout, unsigned MemoryLimit) {
Reid Spencer9ac14182004-12-16 23:01:34 +0000402 sys::Path OutputAsmFile;
Duncan Sands41396302010-05-24 07:49:55 +0000403 OutputCode(Bitcode, OutputAsmFile, *Error, Timeout, MemoryLimit);
Reid Spencera229c5c2005-07-08 03:08:58 +0000404 OutputAsmFile.eraseFromDisk();
Chris Lattner9cbbee32004-02-18 23:24:41 +0000405}
406
Gabor Greif8ff70c22007-07-04 21:55:50 +0000407int LLC::ExecuteProgram(const std::string &Bitcode,
Chris Lattner7915a1e2003-10-14 21:34:11 +0000408 const std::vector<std::string> &Args,
Misha Brukman9558c6a2003-09-29 22:39:25 +0000409 const std::string &InputFile,
410 const std::string &OutputFile,
Nick Lewycky22ff7482010-04-12 05:08:25 +0000411 std::string *Error,
Reid Spencer51ab5c82006-06-06 00:00:42 +0000412 const std::vector<std::string> &ArgsForGCC,
Chris Lattnere96b2ed2004-07-24 07:49:11 +0000413 const std::vector<std::string> &SharedLibs,
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000414 unsigned Timeout,
415 unsigned MemoryLimit) {
Chris Lattnereeed9832003-10-14 21:52:52 +0000416
Reid Spencer9ac14182004-12-16 23:01:34 +0000417 sys::Path OutputAsmFile;
Duncan Sands41396302010-05-24 07:49:55 +0000418 GCC::FileType FileKind = OutputCode(Bitcode, OutputAsmFile, *Error, Timeout,
419 MemoryLimit);
Anton Korobeynikov86c006a2009-08-05 09:32:10 +0000420 FileRemover OutFileRemover(OutputAsmFile, !SaveTemps);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000421
Reid Spencer51ab5c82006-06-06 00:00:42 +0000422 std::vector<std::string> GCCArgs(ArgsForGCC);
Bill Wendling38efa382009-03-02 23:13:18 +0000423 GCCArgs.insert(GCCArgs.end(), SharedLibs.begin(), SharedLibs.end());
Reid Spencer51ab5c82006-06-06 00:00:42 +0000424
Misha Brukman9558c6a2003-09-29 22:39:25 +0000425 // Assuming LLC worked, compile the result with GCC and run it.
Chris Lattner50010422010-03-16 06:41:47 +0000426 return gcc->ExecuteProgram(OutputAsmFile.str(), Args, FileKind,
Nick Lewycky22ff7482010-04-12 05:08:25 +0000427 InputFile, OutputFile, Error, GCCArgs,
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000428 Timeout, MemoryLimit);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000429}
430
Chris Lattner7915a1e2003-10-14 21:34:11 +0000431/// createLLC - Try to find the LLC executable
Misha Brukman9558c6a2003-09-29 22:39:25 +0000432///
Dan Gohman197f7282009-08-05 20:21:17 +0000433LLC *AbstractInterpreter::createLLC(const char *Argv0,
Brian Gaeked11577b2004-05-04 21:09:01 +0000434 std::string &Message,
Kalle Raiskilafaa95762010-05-10 07:38:37 +0000435 const std::string &GCCBinary,
Bill Wendling38efa382009-03-02 23:13:18 +0000436 const std::vector<std::string> *Args,
Chris Lattner50010422010-03-16 06:41:47 +0000437 const std::vector<std::string> *GCCArgs,
438 bool UseIntegratedAssembler) {
Dan Gohman197f7282009-08-05 20:21:17 +0000439 std::string LLCPath =
Chris Lattner74382b72009-08-23 22:45:37 +0000440 FindExecutable("llc", Argv0, (void *)(intptr_t)&createLLC).str();
Misha Brukman9558c6a2003-09-29 22:39:25 +0000441 if (LLCPath.empty()) {
442 Message = "Cannot find `llc' in executable directory or PATH!\n";
443 return 0;
444 }
445
446 Message = "Found llc: " + LLCPath + "\n";
Kalle Raiskilafaa95762010-05-10 07:38:37 +0000447 GCC *gcc = GCC::create(Message, GCCBinary, GCCArgs);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000448 if (!gcc) {
Dan Gohman65f57c22009-07-15 16:35:29 +0000449 errs() << Message << "\n";
Misha Brukman9558c6a2003-09-29 22:39:25 +0000450 exit(1);
451 }
Kalle Raiskilafaa95762010-05-10 07:38:37 +0000452 return new LLC(LLCPath, gcc, Args, UseIntegratedAssembler);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000453}
454
455//===---------------------------------------------------------------------===//
456// JIT Implementation of AbstractIntepreter interface
457//
Chris Lattner2cdd21c2003-12-14 21:35:53 +0000458namespace {
459 class JIT : public AbstractInterpreter {
460 std::string LLIPath; // The path to the LLI executable
Brian Gaeked11577b2004-05-04 21:09:01 +0000461 std::vector<std::string> ToolArgs; // Args to pass to LLI
Chris Lattner2cdd21c2003-12-14 21:35:53 +0000462 public:
Brian Gaeked11577b2004-05-04 21:09:01 +0000463 JIT(const std::string &Path, const std::vector<std::string> *Args)
464 : LLIPath(Path) {
465 ToolArgs.clear ();
Brian Gaeke48b008d2004-05-04 22:02:41 +0000466 if (Args) { ToolArgs = *Args; }
Brian Gaeked11577b2004-05-04 21:09:01 +0000467 }
Misha Brukmanf976c852005-04-21 22:55:34 +0000468
Gabor Greif8ff70c22007-07-04 21:55:50 +0000469 virtual int ExecuteProgram(const std::string &Bitcode,
Chris Lattner2cdd21c2003-12-14 21:35:53 +0000470 const std::vector<std::string> &Args,
471 const std::string &InputFile,
472 const std::string &OutputFile,
Nick Lewycky22ff7482010-04-12 05:08:25 +0000473 std::string *Error,
Reid Spencer51ab5c82006-06-06 00:00:42 +0000474 const std::vector<std::string> &GCCArgs =
475 std::vector<std::string>(),
Misha Brukmanf976c852005-04-21 22:55:34 +0000476 const std::vector<std::string> &SharedLibs =
Reid Spencer51ab5c82006-06-06 00:00:42 +0000477 std::vector<std::string>(),
Nick Lewycky22ff7482010-04-12 05:08:25 +0000478 unsigned Timeout = 0,
479 unsigned MemoryLimit = 0);
Chris Lattner2cdd21c2003-12-14 21:35:53 +0000480 };
481}
Misha Brukman9558c6a2003-09-29 22:39:25 +0000482
Gabor Greif8ff70c22007-07-04 21:55:50 +0000483int JIT::ExecuteProgram(const std::string &Bitcode,
Chris Lattner7915a1e2003-10-14 21:34:11 +0000484 const std::vector<std::string> &Args,
Misha Brukman9558c6a2003-09-29 22:39:25 +0000485 const std::string &InputFile,
486 const std::string &OutputFile,
Nick Lewycky22ff7482010-04-12 05:08:25 +0000487 std::string *Error,
Reid Spencer51ab5c82006-06-06 00:00:42 +0000488 const std::vector<std::string> &GCCArgs,
Chris Lattnere96b2ed2004-07-24 07:49:11 +0000489 const std::vector<std::string> &SharedLibs,
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000490 unsigned Timeout,
491 unsigned MemoryLimit) {
Misha Brukman9558c6a2003-09-29 22:39:25 +0000492 // Construct a vector of parameters, incorporating those from the command-line
493 std::vector<const char*> JITArgs;
494 JITArgs.push_back(LLIPath.c_str());
Misha Brukman9558c6a2003-09-29 22:39:25 +0000495 JITArgs.push_back("-force-interpreter=false");
Chris Lattnereeed9832003-10-14 21:52:52 +0000496
Brian Gaeked11577b2004-05-04 21:09:01 +0000497 // Add any extra LLI args.
498 for (unsigned i = 0, e = ToolArgs.size(); i != e; ++i)
499 JITArgs.push_back(ToolArgs[i].c_str());
500
Chris Lattnereeed9832003-10-14 21:52:52 +0000501 for (unsigned i = 0, e = SharedLibs.size(); i != e; ++i) {
Misha Brukman9558c6a2003-09-29 22:39:25 +0000502 JITArgs.push_back("-load");
Chris Lattnereeed9832003-10-14 21:52:52 +0000503 JITArgs.push_back(SharedLibs[i].c_str());
Misha Brukman9558c6a2003-09-29 22:39:25 +0000504 }
Gabor Greif8ff70c22007-07-04 21:55:50 +0000505 JITArgs.push_back(Bitcode.c_str());
Misha Brukman9558c6a2003-09-29 22:39:25 +0000506 // Add optional parameters to the running program from Argv
507 for (unsigned i=0, e = Args.size(); i != e; ++i)
508 JITArgs.push_back(Args[i].c_str());
509 JITArgs.push_back(0);
510
Dan Gohmana1bdced2009-07-15 17:29:42 +0000511 outs() << "<jit>"; outs().flush();
Dan Gohman65f57c22009-07-15 16:35:29 +0000512 DEBUG(errs() << "\nAbout to run:\t";
Chris Lattner7b2ccff2003-10-19 02:14:58 +0000513 for (unsigned i=0, e = JITArgs.size()-1; i != e; ++i)
Dan Gohman65f57c22009-07-15 16:35:29 +0000514 errs() << " " << JITArgs[i];
515 errs() << "\n";
Misha Brukman9558c6a2003-09-29 22:39:25 +0000516 );
Dan Gohman65f57c22009-07-15 16:35:29 +0000517 DEBUG(errs() << "\nSending output to " << OutputFile << "\n");
Reid Spencerb31baa82004-12-19 18:00:21 +0000518 return RunProgramWithTimeout(sys::Path(LLIPath), &JITArgs[0],
Misha Brukmanf976c852005-04-21 22:55:34 +0000519 sys::Path(InputFile), sys::Path(OutputFile), sys::Path(OutputFile),
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000520 Timeout, MemoryLimit);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000521}
522
Chris Lattner7915a1e2003-10-14 21:34:11 +0000523/// createJIT - Try to find the LLI executable
Misha Brukman9558c6a2003-09-29 22:39:25 +0000524///
Dan Gohman197f7282009-08-05 20:21:17 +0000525AbstractInterpreter *AbstractInterpreter::createJIT(const char *Argv0,
Brian Gaeked11577b2004-05-04 21:09:01 +0000526 std::string &Message, const std::vector<std::string> *Args) {
Dan Gohman197f7282009-08-05 20:21:17 +0000527 std::string LLIPath =
Chris Lattner74382b72009-08-23 22:45:37 +0000528 FindExecutable("lli", Argv0, (void *)(intptr_t)&createJIT).str();
Misha Brukman9558c6a2003-09-29 22:39:25 +0000529 if (!LLIPath.empty()) {
530 Message = "Found lli: " + LLIPath + "\n";
Brian Gaeked11577b2004-05-04 21:09:01 +0000531 return new JIT(LLIPath, Args);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000532 }
533
534 Message = "Cannot find `lli' in executable directory or PATH!\n";
535 return 0;
536}
537
Gabor Greif8ff70c22007-07-04 21:55:50 +0000538GCC::FileType CBE::OutputCode(const std::string &Bitcode,
Duncan Sands41396302010-05-24 07:49:55 +0000539 sys::Path &OutputCFile, std::string &Error,
540 unsigned Timeout, unsigned MemoryLimit) {
Gabor Greif8ff70c22007-07-04 21:55:50 +0000541 sys::Path uniqueFile(Bitcode+".cbe.c");
Reid Spencer51c5a282006-08-23 20:34:57 +0000542 std::string ErrMsg;
543 if (uniqueFile.makeUnique(true, &ErrMsg)) {
Dan Gohman65f57c22009-07-15 16:35:29 +0000544 errs() << "Error making unique filename: " << ErrMsg << "\n";
Reid Spencer51c5a282006-08-23 20:34:57 +0000545 exit(1);
546 }
Reid Spencer9ac14182004-12-16 23:01:34 +0000547 OutputCFile = uniqueFile;
Brian Gaeked11577b2004-05-04 21:09:01 +0000548 std::vector<const char *> LLCArgs;
Nick Lewycky22ff7482010-04-12 05:08:25 +0000549 LLCArgs.push_back(LLCPath.c_str());
Brian Gaeked11577b2004-05-04 21:09:01 +0000550
551 // Add any extra LLC args.
552 for (unsigned i = 0, e = ToolArgs.size(); i != e; ++i)
553 LLCArgs.push_back(ToolArgs[i].c_str());
554
Nick Lewycky22ff7482010-04-12 05:08:25 +0000555 LLCArgs.push_back("-o");
556 LLCArgs.push_back(OutputCFile.c_str()); // Output to the C file
557 LLCArgs.push_back("-march=c"); // Output C language
Nick Lewycky22ff7482010-04-12 05:08:25 +0000558 LLCArgs.push_back(Bitcode.c_str()); // This is the input bitcode
559 LLCArgs.push_back(0);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000560
Dan Gohmana1bdced2009-07-15 17:29:42 +0000561 outs() << "<cbe>"; outs().flush();
Dan Gohman65f57c22009-07-15 16:35:29 +0000562 DEBUG(errs() << "\nAbout to run:\t";
Nick Lewycky22ff7482010-04-12 05:08:25 +0000563 for (unsigned i = 0, e = LLCArgs.size()-1; i != e; ++i)
Dan Gohman65f57c22009-07-15 16:35:29 +0000564 errs() << " " << LLCArgs[i];
565 errs() << "\n";
Brian Gaeked11577b2004-05-04 21:09:01 +0000566 );
Misha Brukmanf976c852005-04-21 22:55:34 +0000567 if (RunProgramWithTimeout(LLCPath, &LLCArgs[0], sys::Path(), sys::Path(),
Duncan Sands41396302010-05-24 07:49:55 +0000568 sys::Path(), Timeout, MemoryLimit))
569 Error = ProcessFailure(LLCPath, &LLCArgs[0], Timeout, MemoryLimit);
Chris Lattnerc600f3c2006-09-15 21:29:15 +0000570 return GCC::CFile;
Misha Brukman9558c6a2003-09-29 22:39:25 +0000571}
572
Duncan Sands41396302010-05-24 07:49:55 +0000573void CBE::compileProgram(const std::string &Bitcode, std::string *Error,
574 unsigned Timeout, unsigned MemoryLimit) {
Reid Spencer9ac14182004-12-16 23:01:34 +0000575 sys::Path OutputCFile;
Duncan Sands41396302010-05-24 07:49:55 +0000576 OutputCode(Bitcode, OutputCFile, *Error, Timeout, MemoryLimit);
Reid Spencera229c5c2005-07-08 03:08:58 +0000577 OutputCFile.eraseFromDisk();
Chris Lattner9cbbee32004-02-18 23:24:41 +0000578}
579
Gabor Greif8ff70c22007-07-04 21:55:50 +0000580int CBE::ExecuteProgram(const std::string &Bitcode,
Chris Lattner7915a1e2003-10-14 21:34:11 +0000581 const std::vector<std::string> &Args,
Misha Brukman9558c6a2003-09-29 22:39:25 +0000582 const std::string &InputFile,
583 const std::string &OutputFile,
Nick Lewycky22ff7482010-04-12 05:08:25 +0000584 std::string *Error,
Reid Spencer51ab5c82006-06-06 00:00:42 +0000585 const std::vector<std::string> &ArgsForGCC,
Chris Lattnere96b2ed2004-07-24 07:49:11 +0000586 const std::vector<std::string> &SharedLibs,
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000587 unsigned Timeout,
588 unsigned MemoryLimit) {
Reid Spencer9ac14182004-12-16 23:01:34 +0000589 sys::Path OutputCFile;
Duncan Sands41396302010-05-24 07:49:55 +0000590 OutputCode(Bitcode, OutputCFile, *Error, Timeout, MemoryLimit);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000591
Anton Korobeynikov86c006a2009-08-05 09:32:10 +0000592 FileRemover CFileRemove(OutputCFile, !SaveTemps);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000593
Reid Spencer51ab5c82006-06-06 00:00:42 +0000594 std::vector<std::string> GCCArgs(ArgsForGCC);
Bill Wendling38efa382009-03-02 23:13:18 +0000595 GCCArgs.insert(GCCArgs.end(), SharedLibs.begin(), SharedLibs.end());
596
Chris Lattner74382b72009-08-23 22:45:37 +0000597 return gcc->ExecuteProgram(OutputCFile.str(), Args, GCC::CFile,
Nick Lewycky22ff7482010-04-12 05:08:25 +0000598 InputFile, OutputFile, Error, GCCArgs,
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000599 Timeout, MemoryLimit);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000600}
601
Chris Lattner9915cd92004-02-17 06:40:06 +0000602/// createCBE - Try to find the 'llc' executable
Misha Brukman9558c6a2003-09-29 22:39:25 +0000603///
Dan Gohman197f7282009-08-05 20:21:17 +0000604CBE *AbstractInterpreter::createCBE(const char *Argv0,
Brian Gaeked11577b2004-05-04 21:09:01 +0000605 std::string &Message,
Kalle Raiskilafaa95762010-05-10 07:38:37 +0000606 const std::string &GCCBinary,
Bill Wendling38efa382009-03-02 23:13:18 +0000607 const std::vector<std::string> *Args,
608 const std::vector<std::string> *GCCArgs) {
Dan Gohman197f7282009-08-05 20:21:17 +0000609 sys::Path LLCPath =
Dan Gohman8608cf22009-08-05 21:03:39 +0000610 FindExecutable("llc", Argv0, (void *)(intptr_t)&createCBE);
Reid Spencerb31baa82004-12-19 18:00:21 +0000611 if (LLCPath.isEmpty()) {
Misha Brukmanf976c852005-04-21 22:55:34 +0000612 Message =
Chris Lattner9915cd92004-02-17 06:40:06 +0000613 "Cannot find `llc' in executable directory or PATH!\n";
Misha Brukman9558c6a2003-09-29 22:39:25 +0000614 return 0;
615 }
616
Chris Lattner74382b72009-08-23 22:45:37 +0000617 Message = "Found llc: " + LLCPath.str() + "\n";
Kalle Raiskilafaa95762010-05-10 07:38:37 +0000618 GCC *gcc = GCC::create(Message, GCCBinary, GCCArgs);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000619 if (!gcc) {
Dan Gohman65f57c22009-07-15 16:35:29 +0000620 errs() << Message << "\n";
Misha Brukman9558c6a2003-09-29 22:39:25 +0000621 exit(1);
622 }
Brian Gaeked11577b2004-05-04 21:09:01 +0000623 return new CBE(LLCPath, gcc, Args);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000624}
625
626//===---------------------------------------------------------------------===//
627// GCC abstraction
628//
David Goodwin80becf12009-07-10 21:39:28 +0000629
Chris Lattner50010422010-03-16 06:41:47 +0000630static bool IsARMArchitecture(std::vector<std::string> Args) {
David Goodwin80becf12009-07-10 21:39:28 +0000631 for (std::vector<std::string>::const_iterator
632 I = Args.begin(), E = Args.end(); I != E; ++I) {
Jakob Stoklund Olesen18e05b42010-05-13 17:58:15 +0000633 if (StringRef(*I).equals_lower("-arch")) {
David Goodwin80becf12009-07-10 21:39:28 +0000634 ++I;
Jakob Stoklund Olesen18e05b42010-05-13 17:58:15 +0000635 if (I != E && StringRef(*I).substr(0, strlen("arm")).equals_lower("arm"))
David Goodwin80becf12009-07-10 21:39:28 +0000636 return true;
David Goodwin80becf12009-07-10 21:39:28 +0000637 }
638 }
639
640 return false;
641}
642
Misha Brukman9558c6a2003-09-29 22:39:25 +0000643int GCC::ExecuteProgram(const std::string &ProgramFile,
Chris Lattner7915a1e2003-10-14 21:34:11 +0000644 const std::vector<std::string> &Args,
Misha Brukman9558c6a2003-09-29 22:39:25 +0000645 FileType fileType,
646 const std::string &InputFile,
647 const std::string &OutputFile,
Nick Lewycky22ff7482010-04-12 05:08:25 +0000648 std::string *Error,
Reid Spencer51ab5c82006-06-06 00:00:42 +0000649 const std::vector<std::string> &ArgsForGCC,
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000650 unsigned Timeout,
651 unsigned MemoryLimit) {
Misha Brukman9558c6a2003-09-29 22:39:25 +0000652 std::vector<const char*> GCCArgs;
653
654 GCCArgs.push_back(GCCPath.c_str());
Chris Lattnereeed9832003-10-14 21:52:52 +0000655
Chris Lattner50010422010-03-16 06:41:47 +0000656 if (TargetTriple.getArch() == Triple::x86)
657 GCCArgs.push_back("-m32");
658
Bill Wendling38efa382009-03-02 23:13:18 +0000659 for (std::vector<std::string>::const_iterator
660 I = gccArgs.begin(), E = gccArgs.end(); I != E; ++I)
661 GCCArgs.push_back(I->c_str());
662
Chris Lattnereeed9832003-10-14 21:52:52 +0000663 // Specify -x explicitly in case the extension is wonky
Chris Lattner50010422010-03-16 06:41:47 +0000664 if (fileType != ObjectFile) {
665 GCCArgs.push_back("-x");
666 if (fileType == CFile) {
667 GCCArgs.push_back("c");
668 GCCArgs.push_back("-fno-strict-aliasing");
669 } else {
670 GCCArgs.push_back("assembler");
Daniel Dunbarca740962009-08-18 03:35:57 +0000671
Chris Lattner50010422010-03-16 06:41:47 +0000672 // For ARM architectures we don't want this flag. bugpoint isn't
673 // explicitly told what architecture it is working on, so we get
674 // it from gcc flags
675 if ((TargetTriple.getOS() == Triple::Darwin) &&
676 !IsARMArchitecture(ArgsForGCC))
677 GCCArgs.push_back("-force_cpusubtype_ALL");
678 }
Misha Brukman9558c6a2003-09-29 22:39:25 +0000679 }
Chris Lattner50010422010-03-16 06:41:47 +0000680
681 GCCArgs.push_back(ProgramFile.c_str()); // Specify the input filename.
682
Chris Lattner629e4872006-06-09 21:31:53 +0000683 GCCArgs.push_back("-x");
684 GCCArgs.push_back("none");
Misha Brukman9558c6a2003-09-29 22:39:25 +0000685 GCCArgs.push_back("-o");
Reid Spencercda985e2004-12-15 01:51:56 +0000686 sys::Path OutputBinary (ProgramFile+".gcc.exe");
Reid Spencer51c5a282006-08-23 20:34:57 +0000687 std::string ErrMsg;
688 if (OutputBinary.makeUnique(true, &ErrMsg)) {
Dan Gohman65f57c22009-07-15 16:35:29 +0000689 errs() << "Error making unique filename: " << ErrMsg << "\n";
Reid Spencer51c5a282006-08-23 20:34:57 +0000690 exit(1);
691 }
Misha Brukman9558c6a2003-09-29 22:39:25 +0000692 GCCArgs.push_back(OutputBinary.c_str()); // Output to the right file...
Reid Spencer51ab5c82006-06-06 00:00:42 +0000693
694 // Add any arguments intended for GCC. We locate them here because this is
695 // most likely -L and -l options that need to come before other libraries but
696 // after the source. Other options won't be sensitive to placement on the
697 // command line, so this should be safe.
698 for (unsigned i = 0, e = ArgsForGCC.size(); i != e; ++i)
699 GCCArgs.push_back(ArgsForGCC[i].c_str());
700
Misha Brukman9558c6a2003-09-29 22:39:25 +0000701 GCCArgs.push_back("-lm"); // Hard-code the math library...
702 GCCArgs.push_back("-O2"); // Optimize the program a bit...
Brian Gaekec8db76c2003-11-18 06:31:17 +0000703#if defined (HAVE_LINK_R)
Chris Lattner1f0f1622003-10-18 21:54:47 +0000704 GCCArgs.push_back("-Wl,-R."); // Search this dir for .so files
Brian Gaekec8db76c2003-11-18 06:31:17 +0000705#endif
Daniel Dunbarca740962009-08-18 03:35:57 +0000706 if (TargetTriple.getArch() == Triple::sparc)
707 GCCArgs.push_back("-mcpu=v9");
Misha Brukman9558c6a2003-09-29 22:39:25 +0000708 GCCArgs.push_back(0); // NULL terminator
709
Dan Gohmana1bdced2009-07-15 17:29:42 +0000710 outs() << "<gcc>"; outs().flush();
Dan Gohman65f57c22009-07-15 16:35:29 +0000711 DEBUG(errs() << "\nAbout to run:\t";
Nick Lewycky22ff7482010-04-12 05:08:25 +0000712 for (unsigned i = 0, e = GCCArgs.size()-1; i != e; ++i)
Dan Gohman65f57c22009-07-15 16:35:29 +0000713 errs() << " " << GCCArgs[i];
714 errs() << "\n";
Evan Cheng7f7fdcc2007-01-03 07:44:30 +0000715 );
Reid Spencerb31baa82004-12-19 18:00:21 +0000716 if (RunProgramWithTimeout(GCCPath, &GCCArgs[0], sys::Path(), sys::Path(),
717 sys::Path())) {
Nick Lewycky22ff7482010-04-12 05:08:25 +0000718 *Error = ProcessFailure(GCCPath, &GCCArgs[0]);
719 return -1;
Misha Brukman9558c6a2003-09-29 22:39:25 +0000720 }
721
722 std::vector<const char*> ProgramArgs;
Chris Lattner45495c52005-02-13 23:13:47 +0000723
Evan Cheng70f684f2008-09-09 06:11:26 +0000724 if (RemoteClientPath.isEmpty())
Evan Cheng34e400e2007-05-03 18:36:15 +0000725 ProgramArgs.push_back(OutputBinary.c_str());
726 else {
Evan Cheng70f684f2008-09-09 06:11:26 +0000727 ProgramArgs.push_back(RemoteClientPath.c_str());
728 ProgramArgs.push_back(RemoteHost.c_str());
Anton Korobeynikov26964b62009-08-05 09:32:53 +0000729 if (!RemoteUser.empty()) {
730 ProgramArgs.push_back("-l");
731 ProgramArgs.push_back(RemoteUser.c_str());
732 }
David Goodwin80becf12009-07-10 21:39:28 +0000733 if (!RemotePort.empty()) {
734 ProgramArgs.push_back("-p");
735 ProgramArgs.push_back(RemotePort.c_str());
736 }
Evan Cheng70f684f2008-09-09 06:11:26 +0000737 if (!RemoteExtra.empty()) {
738 ProgramArgs.push_back(RemoteExtra.c_str());
739 }
Evan Cheng34e400e2007-05-03 18:36:15 +0000740
David Goodwin91cf3612009-07-20 17:15:03 +0000741 // Full path to the binary. We need to cd to the exec directory because
742 // there is a dylib there that the exec expects to find in the CWD
Evan Cheng34e400e2007-05-03 18:36:15 +0000743 char* env_pwd = getenv("PWD");
David Goodwin91cf3612009-07-20 17:15:03 +0000744 std::string Exec = "cd ";
Evan Cheng34e400e2007-05-03 18:36:15 +0000745 Exec += env_pwd;
David Goodwin91cf3612009-07-20 17:15:03 +0000746 Exec += "; ./";
Evan Cheng34e400e2007-05-03 18:36:15 +0000747 Exec += OutputBinary.c_str();
748 ProgramArgs.push_back(Exec.c_str());
749 }
750
Misha Brukman9558c6a2003-09-29 22:39:25 +0000751 // Add optional parameters to the running program from Argv
Nick Lewycky22ff7482010-04-12 05:08:25 +0000752 for (unsigned i = 0, e = Args.size(); i != e; ++i)
Misha Brukman9558c6a2003-09-29 22:39:25 +0000753 ProgramArgs.push_back(Args[i].c_str());
754 ProgramArgs.push_back(0); // NULL terminator
755
756 // Now that we have a binary, run it!
Dan Gohmana1bdced2009-07-15 17:29:42 +0000757 outs() << "<program>"; outs().flush();
Dan Gohman65f57c22009-07-15 16:35:29 +0000758 DEBUG(errs() << "\nAbout to run:\t";
Nick Lewycky22ff7482010-04-12 05:08:25 +0000759 for (unsigned i = 0, e = ProgramArgs.size()-1; i != e; ++i)
Dan Gohman65f57c22009-07-15 16:35:29 +0000760 errs() << " " << ProgramArgs[i];
761 errs() << "\n";
Misha Brukman9558c6a2003-09-29 22:39:25 +0000762 );
Chris Lattner8c56be52004-02-18 20:21:57 +0000763
Anton Korobeynikov86c006a2009-08-05 09:32:10 +0000764 FileRemover OutputBinaryRemover(OutputBinary, !SaveTemps);
Evan Cheng34e400e2007-05-03 18:36:15 +0000765
Viktor Kutuzov699220b2009-07-14 19:10:55 +0000766 if (RemoteClientPath.isEmpty()) {
Nick Lewycky22ff7482010-04-12 05:08:25 +0000767 DEBUG(errs() << "<run locally>");
Evan Cheng34e400e2007-05-03 18:36:15 +0000768 return RunProgramWithTimeout(OutputBinary, &ProgramArgs[0],
769 sys::Path(InputFile), sys::Path(OutputFile), sys::Path(OutputFile),
770 Timeout, MemoryLimit);
Viktor Kutuzov699220b2009-07-14 19:10:55 +0000771 } else {
Dan Gohmana1bdced2009-07-15 17:29:42 +0000772 outs() << "<run remotely>"; outs().flush();
Viktor Kutuzovfc2271f2009-07-18 18:39:24 +0000773 return RunProgramRemotelyWithTimeout(sys::Path(RemoteClientPath),
Viktor Kutuzov699220b2009-07-14 19:10:55 +0000774 &ProgramArgs[0], sys::Path(InputFile), sys::Path(OutputFile),
775 sys::Path(OutputFile), Timeout, MemoryLimit);
Viktor Kutuzov699220b2009-07-14 19:10:55 +0000776 }
Misha Brukman9558c6a2003-09-29 22:39:25 +0000777}
778
Chris Lattner1798e4a2003-10-14 21:07:25 +0000779int GCC::MakeSharedObject(const std::string &InputFile, FileType fileType,
Chris Lattner130e2a32006-06-27 20:35:36 +0000780 std::string &OutputFile,
Nick Lewycky22ff7482010-04-12 05:08:25 +0000781 const std::vector<std::string> &ArgsForGCC,
782 std::string &Error) {
Reid Spencercda985e2004-12-15 01:51:56 +0000783 sys::Path uniqueFilename(InputFile+LTDL_SHLIB_EXT);
Reid Spencer51c5a282006-08-23 20:34:57 +0000784 std::string ErrMsg;
785 if (uniqueFilename.makeUnique(true, &ErrMsg)) {
Dan Gohman65f57c22009-07-15 16:35:29 +0000786 errs() << "Error making unique filename: " << ErrMsg << "\n";
Reid Spencer51c5a282006-08-23 20:34:57 +0000787 exit(1);
788 }
Chris Lattner74382b72009-08-23 22:45:37 +0000789 OutputFile = uniqueFilename.str();
Reid Spencercda985e2004-12-15 01:51:56 +0000790
Chris Lattner130e2a32006-06-27 20:35:36 +0000791 std::vector<const char*> GCCArgs;
792
793 GCCArgs.push_back(GCCPath.c_str());
Evan Chengfd829952009-03-12 00:53:34 +0000794
Chris Lattner50010422010-03-16 06:41:47 +0000795 if (TargetTriple.getArch() == Triple::x86)
796 GCCArgs.push_back("-m32");
797
Evan Chengfd829952009-03-12 00:53:34 +0000798 for (std::vector<std::string>::const_iterator
799 I = gccArgs.begin(), E = gccArgs.end(); I != E; ++I)
800 GCCArgs.push_back(I->c_str());
801
Misha Brukman9558c6a2003-09-29 22:39:25 +0000802 // Compile the C/asm file into a shared object
Chris Lattner50010422010-03-16 06:41:47 +0000803 if (fileType != ObjectFile) {
804 GCCArgs.push_back("-x");
805 GCCArgs.push_back(fileType == AsmFile ? "assembler" : "c");
806 }
Chris Lattner130e2a32006-06-27 20:35:36 +0000807 GCCArgs.push_back("-fno-strict-aliasing");
808 GCCArgs.push_back(InputFile.c_str()); // Specify the input filename.
Lauro Ramos Venancio497391a2007-06-06 23:10:56 +0000809 GCCArgs.push_back("-x");
810 GCCArgs.push_back("none");
Daniel Dunbarca740962009-08-18 03:35:57 +0000811 if (TargetTriple.getArch() == Triple::sparc)
812 GCCArgs.push_back("-G"); // Compile a shared library, `-G' for Sparc
813 else if (TargetTriple.getOS() == Triple::Darwin) {
814 // link all source files into a single module in data segment, rather than
815 // generating blocks. dynamic_lookup requires that you set
816 // MACOSX_DEPLOYMENT_TARGET=10.3 in your env. FIXME: it would be better for
817 // bugpoint to just pass that in the environment of GCC.
818 GCCArgs.push_back("-single_module");
819 GCCArgs.push_back("-dynamiclib"); // `-dynamiclib' for MacOS X/PowerPC
820 GCCArgs.push_back("-undefined");
821 GCCArgs.push_back("dynamic_lookup");
822 } else
823 GCCArgs.push_back("-shared"); // `-shared' for Linux/X86, maybe others
Chris Lattner1c81f132005-03-09 03:31:02 +0000824
Daniel Dunbarca740962009-08-18 03:35:57 +0000825 if ((TargetTriple.getArch() == Triple::alpha) ||
826 (TargetTriple.getArch() == Triple::x86_64))
827 GCCArgs.push_back("-fPIC"); // Requires shared objs to contain PIC
828
829 if (TargetTriple.getArch() == Triple::sparc)
830 GCCArgs.push_back("-mcpu=v9");
831
Chris Lattner130e2a32006-06-27 20:35:36 +0000832 GCCArgs.push_back("-o");
833 GCCArgs.push_back(OutputFile.c_str()); // Output to the right filename.
834 GCCArgs.push_back("-O2"); // Optimize the program a bit.
835
836
837
838 // Add any arguments intended for GCC. We locate them here because this is
839 // most likely -L and -l options that need to come before other libraries but
840 // after the source. Other options won't be sensitive to placement on the
841 // command line, so this should be safe.
842 for (unsigned i = 0, e = ArgsForGCC.size(); i != e; ++i)
843 GCCArgs.push_back(ArgsForGCC[i].c_str());
844 GCCArgs.push_back(0); // NULL terminator
845
846
Misha Brukmanf976c852005-04-21 22:55:34 +0000847
Dan Gohmana1bdced2009-07-15 17:29:42 +0000848 outs() << "<gcc>"; outs().flush();
Dan Gohman65f57c22009-07-15 16:35:29 +0000849 DEBUG(errs() << "\nAbout to run:\t";
Nick Lewycky22ff7482010-04-12 05:08:25 +0000850 for (unsigned i = 0, e = GCCArgs.size()-1; i != e; ++i)
Dan Gohman65f57c22009-07-15 16:35:29 +0000851 errs() << " " << GCCArgs[i];
852 errs() << "\n";
Evan Cheng7f7fdcc2007-01-03 07:44:30 +0000853 );
Chris Lattner130e2a32006-06-27 20:35:36 +0000854 if (RunProgramWithTimeout(GCCPath, &GCCArgs[0], sys::Path(), sys::Path(),
Reid Spencerb31baa82004-12-19 18:00:21 +0000855 sys::Path())) {
Nick Lewycky22ff7482010-04-12 05:08:25 +0000856 Error = ProcessFailure(GCCPath, &GCCArgs[0]);
Chris Lattner1798e4a2003-10-14 21:07:25 +0000857 return 1;
Misha Brukman9558c6a2003-09-29 22:39:25 +0000858 }
859 return 0;
860}
861
Chris Lattner7915a1e2003-10-14 21:34:11 +0000862/// create - Try to find the `gcc' executable
Misha Brukman9558c6a2003-09-29 22:39:25 +0000863///
Dan Gohman197f7282009-08-05 20:21:17 +0000864GCC *GCC::create(std::string &Message,
Kalle Raiskilafaa95762010-05-10 07:38:37 +0000865 const std::string &GCCBinary,
Bill Wendling38efa382009-03-02 23:13:18 +0000866 const std::vector<std::string> *Args) {
Kalle Raiskilafaa95762010-05-10 07:38:37 +0000867 sys::Path GCCPath = sys::Program::FindProgramByName(GCCBinary);
Reid Spencerb31baa82004-12-19 18:00:21 +0000868 if (GCCPath.isEmpty()) {
Kalle Raiskilafaa95762010-05-10 07:38:37 +0000869 Message = "Cannot find `"+ GCCBinary +"' in executable directory or PATH!\n";
Misha Brukman9558c6a2003-09-29 22:39:25 +0000870 return 0;
871 }
872
Evan Cheng70f684f2008-09-09 06:11:26 +0000873 sys::Path RemoteClientPath;
874 if (!RemoteClient.empty())
Dan Gohman197f7282009-08-05 20:21:17 +0000875 RemoteClientPath = sys::Program::FindProgramByName(RemoteClient);
Evan Cheng34e400e2007-05-03 18:36:15 +0000876
Chris Lattner74382b72009-08-23 22:45:37 +0000877 Message = "Found gcc: " + GCCPath.str() + "\n";
Bill Wendling38efa382009-03-02 23:13:18 +0000878 return new GCC(GCCPath, RemoteClientPath, Args);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000879}