blob: 8d4935b558f7793be243d1d668ea628ba91d1d5b [file] [log] [blame]
Chris Lattnerffac2862006-06-06 22:30:59 +00001//===-- ToolRunner.cpp ----------------------------------------------------===//
2//
3// The LLVM Compiler Infrastructure
4//
Chris Lattner345353d2007-12-29 20:44:31 +00005// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
Chris Lattnerffac2862006-06-06 22:30:59 +00007//
8//===----------------------------------------------------------------------===//
9//
10// This file implements the interfaces described in the ToolRunner.h file.
11//
12//===----------------------------------------------------------------------===//
13
14#define DEBUG_TYPE "toolrunner"
15#include "ToolRunner.h"
Chris Lattnerffac2862006-06-06 22:30:59 +000016#include "llvm/System/Program.h"
Evan Cheng779b52e2007-05-03 18:36:15 +000017#include "llvm/Support/CommandLine.h"
Chris Lattnerffac2862006-06-06 22:30:59 +000018#include "llvm/Support/Debug.h"
19#include "llvm/Support/FileUtilities.h"
Chris Lattnerc521f542009-08-23 22:45:37 +000020#include "llvm/Support/raw_ostream.h"
21#include "llvm/Config/config.h" // for HAVE_LINK_R
Chris Lattnerffac2862006-06-06 22:30:59 +000022#include <fstream>
23#include <sstream>
Chris Lattnerffac2862006-06-06 22:30:59 +000024using namespace llvm;
25
Anton Korobeynikov7cbff912009-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 Cheng779b52e2007-05-03 18:36:15 +000031namespace {
32 cl::opt<std::string>
Evan Cheng7aedcf12008-09-09 06:11:26 +000033 RemoteClient("remote-client",
34 cl::desc("Remote execution client (rsh/ssh)"));
Evan Cheng779b52e2007-05-03 18:36:15 +000035
36 cl::opt<std::string>
Evan Cheng7aedcf12008-09-09 06:11:26 +000037 RemoteHost("remote-host",
38 cl::desc("Remote execution (rsh/ssh) host"));
39
40 cl::opt<std::string>
David Goodwin73f4e3f2009-07-10 21:39:28 +000041 RemotePort("remote-port",
42 cl::desc("Remote execution (rsh/ssh) port"));
43
44 cl::opt<std::string>
Evan Cheng7aedcf12008-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 Cheng779b52e2007-05-03 18:36:15 +000051}
52
Viktor Kutuzov85870252009-07-18 18:39:24 +000053/// RunProgramWithTimeout - This function provides an alternate interface
54/// to the sys::Program::ExecuteAndWait interface.
Nick Lewycky6ba630b2010-04-12 05:08:25 +000055/// @see sys::Program::ExecuteAndWait
Chris Lattnerffac2862006-06-06 22:30:59 +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 Korobeynikovd01defe2007-02-16 19:11:07 +000061 unsigned NumSeconds = 0,
62 unsigned MemoryLimit = 0) {
Chris Lattnerffac2862006-06-06 22:30:59 +000063 const sys::Path* redirects[3];
64 redirects[0] = &StdInFile;
65 redirects[1] = &StdOutFile;
66 redirects[2] = &StdErrFile;
Viktor Kutuzov85870252009-07-18 18:39:24 +000067
68#if 0 // For debug purposes
69 {
Dan Gohmand8db3762009-07-15 16:35:29 +000070 errs() << "RUN:";
Chris Lattnerff8a6b82006-09-15 23:01:10 +000071 for (unsigned i = 0; Args[i]; ++i)
Dan Gohmand8db3762009-07-15 16:35:29 +000072 errs() << " " << Args[i];
73 errs() << "\n";
Chris Lattnerff8a6b82006-09-15 23:01:10 +000074 }
Viktor Kutuzov85870252009-07-18 18:39:24 +000075#endif
Chris Lattnerffac2862006-06-06 22:30:59 +000076
77 return
Anton Korobeynikovd01defe2007-02-16 19:11:07 +000078 sys::Program::ExecuteAndWait(ProgramPath, Args, 0, redirects,
79 NumSeconds, MemoryLimit);
Chris Lattnerffac2862006-06-06 22:30:59 +000080}
81
Viktor Kutuzov85870252009-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 Lewycky6ba630b2010-04-12 05:08:25 +000087/// @see sys::Program::ExecuteAndWait
Viktor Kutuzov85870252009-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 Lattnerffac2862006-06-06 22:30:59 +000099
Viktor Kutuzov85870252009-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 Lewycky6ba630b2010-04-12 05:08:25 +0000130 errs() << OS;
Viktor Kutuzov85870252009-07-18 18:39:24 +0000131 }
132
133 return ReturnCode;
134}
Chris Lattnerffac2862006-06-06 22:30:59 +0000135
Duncan Sandse9cd6d02010-05-24 07:49:55 +0000136static std::string ProcessFailure(sys::Path ProgPath, const char** Args,
137 unsigned Timeout = 0,
138 unsigned MemoryLimit = 0) {
Chris Lattnerffac2862006-06-06 22:30:59 +0000139 std::ostringstream OS;
140 OS << "\nError running tool:\n ";
141 for (const char **Arg = Args; *Arg; ++Arg)
142 OS << " " << *Arg;
143 OS << "\n";
Chris Lattnerfd381322010-03-16 06:41:47 +0000144
Chris Lattnerffac2862006-06-06 22:30:59 +0000145 // Rerun the compiler, capturing any error messages to print them.
Dan Gohmanaf2037f2009-02-12 20:53:27 +0000146 sys::Path ErrorFilename("bugpoint.program_error_messages");
Reid Spencere4ca7222006-08-23 20:34:57 +0000147 std::string ErrMsg;
148 if (ErrorFilename.makeUnique(true, &ErrMsg)) {
Dan Gohmand8db3762009-07-15 16:35:29 +0000149 errs() << "Error making unique filename: " << ErrMsg << "\n";
Reid Spencere4ca7222006-08-23 20:34:57 +0000150 exit(1);
151 }
Chris Lattnerffac2862006-06-06 22:30:59 +0000152 RunProgramWithTimeout(ProgPath, Args, sys::Path(""), ErrorFilename,
Duncan Sandse9cd6d02010-05-24 07:49:55 +0000153 ErrorFilename, Timeout, MemoryLimit);
154 // FIXME: check return code ?
Chris Lattnerffac2862006-06-06 22:30:59 +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
165 ErrorFilename.eraseFromDisk();
Nick Lewycky6ba630b2010-04-12 05:08:25 +0000166 return OS.str();
Chris Lattnerffac2862006-06-06 22:30:59 +0000167}
168
169//===---------------------------------------------------------------------===//
170// LLI Implementation of AbstractIntepreter interface
171//
172namespace {
173 class LLI : public AbstractInterpreter {
174 std::string LLIPath; // The path to the LLI executable
175 std::vector<std::string> ToolArgs; // Args to pass to LLI
176 public:
177 LLI(const std::string &Path, const std::vector<std::string> *Args)
178 : LLIPath(Path) {
179 ToolArgs.clear ();
180 if (Args) { ToolArgs = *Args; }
181 }
182
Gabor Greif0e535c3c2007-07-04 21:55:50 +0000183 virtual int ExecuteProgram(const std::string &Bitcode,
Chris Lattnerffac2862006-06-06 22:30:59 +0000184 const std::vector<std::string> &Args,
185 const std::string &InputFile,
186 const std::string &OutputFile,
Nick Lewycky6ba630b2010-04-12 05:08:25 +0000187 std::string *Error,
Chris Lattnerffac2862006-06-06 22:30:59 +0000188 const std::vector<std::string> &GCCArgs,
189 const std::vector<std::string> &SharedLibs =
190 std::vector<std::string>(),
Anton Korobeynikovd01defe2007-02-16 19:11:07 +0000191 unsigned Timeout = 0,
192 unsigned MemoryLimit = 0);
Chris Lattnerffac2862006-06-06 22:30:59 +0000193 };
194}
195
Gabor Greif0e535c3c2007-07-04 21:55:50 +0000196int LLI::ExecuteProgram(const std::string &Bitcode,
Chris Lattnerffac2862006-06-06 22:30:59 +0000197 const std::vector<std::string> &Args,
198 const std::string &InputFile,
199 const std::string &OutputFile,
Nick Lewycky6ba630b2010-04-12 05:08:25 +0000200 std::string *Error,
Chris Lattnerffac2862006-06-06 22:30:59 +0000201 const std::vector<std::string> &GCCArgs,
202 const std::vector<std::string> &SharedLibs,
Anton Korobeynikovd01defe2007-02-16 19:11:07 +0000203 unsigned Timeout,
204 unsigned MemoryLimit) {
Chris Lattnerffac2862006-06-06 22:30:59 +0000205 std::vector<const char*> LLIArgs;
206 LLIArgs.push_back(LLIPath.c_str());
207 LLIArgs.push_back("-force-interpreter=true");
208
Duncan Sands3246fe02009-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
Chris Lattnerffac2862006-06-06 22:30:59 +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 Greif0e535c3c2007-07-04 21:55:50 +0000218 LLIArgs.push_back(Bitcode.c_str());
Chris Lattnerffac2862006-06-06 22:30:59 +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 Gohman607818a2009-07-15 17:29:42 +0000224 outs() << "<lli>"; outs().flush();
Dan Gohmand8db3762009-07-15 16:35:29 +0000225 DEBUG(errs() << "\nAbout to run:\t";
Chris Lattnerffac2862006-06-06 22:30:59 +0000226 for (unsigned i=0, e = LLIArgs.size()-1; i != e; ++i)
Dan Gohmand8db3762009-07-15 16:35:29 +0000227 errs() << " " << LLIArgs[i];
228 errs() << "\n";
Chris Lattnerffac2862006-06-06 22:30:59 +0000229 );
230 return RunProgramWithTimeout(sys::Path(LLIPath), &LLIArgs[0],
231 sys::Path(InputFile), sys::Path(OutputFile), sys::Path(OutputFile),
Anton Korobeynikovd01defe2007-02-16 19:11:07 +0000232 Timeout, MemoryLimit);
Chris Lattnerffac2862006-06-06 22:30:59 +0000233}
234
235// LLI create method - Try to find the LLI executable
Dan Gohman46ffffa2009-08-05 20:21:17 +0000236AbstractInterpreter *AbstractInterpreter::createLLI(const char *Argv0,
Chris Lattnerffac2862006-06-06 22:30:59 +0000237 std::string &Message,
238 const std::vector<std::string> *ToolArgs) {
Dan Gohman46ffffa2009-08-05 20:21:17 +0000239 std::string LLIPath =
Chris Lattnerc521f542009-08-23 22:45:37 +0000240 FindExecutable("lli", Argv0, (void *)(intptr_t)&createLLI).str();
Chris Lattnerffac2862006-06-06 22:30:59 +0000241 if (!LLIPath.empty()) {
242 Message = "Found lli: " + LLIPath + "\n";
243 return new LLI(LLIPath, ToolArgs);
244 }
245
Dan Gohman9e082462010-10-29 16:15:23 +0000246 Message = "Cannot find `lli' in executable directory!\n";
Chris Lattnerffac2862006-06-06 22:30:59 +0000247 return 0;
248}
249
Anton Korobeynikovc53565c2008-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 Lewycky6ba630b2010-04-12 05:08:25 +0000269 std::string *Error,
Anton Korobeynikovc53565c2008-04-28 20:53:48 +0000270 const std::vector<std::string> &GCCArgs,
271 const std::vector<std::string> &SharedLibs =
Nick Lewycky6ba630b2010-04-12 05:08:25 +0000272 std::vector<std::string>(),
Anton Korobeynikovc53565c2008-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 Lewycky6ba630b2010-04-12 05:08:25 +0000282 std::string *Error,
Anton Korobeynikovc53565c2008-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 Lewycky6ba630b2010-04-12 05:08:25 +0000297 for (unsigned i = 0, e = Args.size(); i != e; ++i)
Anton Korobeynikovc53565c2008-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 Korobeynikovc53565c2008-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 Lattnerc521f542009-08-23 22:45:37 +0000342 std::string CmdPath = sys::Program::FindProgramByName(Command).str();
Anton Korobeynikovc53565c2008-04-28 20:53:48 +0000343 if (CmdPath.empty()) {
344 Message =
345 std::string("Cannot find '") + Command +
Dan Gohman9e082462010-10-29 16:15:23 +0000346 "' in PATH!\n";
Anton Korobeynikovc53565c2008-04-28 20:53:48 +0000347 return 0;
348 }
349
350 Message = "Found command in: " + CmdPath + "\n";
351
352 return new CustomExecutor(CmdPath, Args);
353}
354
Chris Lattnerffac2862006-06-06 22:30:59 +0000355//===----------------------------------------------------------------------===//
356// LLC Implementation of AbstractIntepreter interface
357//
Gabor Greif0e535c3c2007-07-04 21:55:50 +0000358GCC::FileType LLC::OutputCode(const std::string &Bitcode,
Duncan Sandse9cd6d02010-05-24 07:49:55 +0000359 sys::Path &OutputAsmFile, std::string &Error,
360 unsigned Timeout, unsigned MemoryLimit) {
Chris Lattnerfd381322010-03-16 06:41:47 +0000361 const char *Suffix = (UseIntegratedAssembler ? ".llc.o" : ".llc.s");
362 sys::Path uniqueFile(Bitcode + Suffix);
Reid Spencere4ca7222006-08-23 20:34:57 +0000363 std::string ErrMsg;
364 if (uniqueFile.makeUnique(true, &ErrMsg)) {
Dan Gohmand8db3762009-07-15 16:35:29 +0000365 errs() << "Error making unique filename: " << ErrMsg << "\n";
Reid Spencere4ca7222006-08-23 20:34:57 +0000366 exit(1);
367 }
Chris Lattnerffac2862006-06-06 22:30:59 +0000368 OutputAsmFile = uniqueFile;
369 std::vector<const char *> LLCArgs;
Chris Lattnerfd381322010-03-16 06:41:47 +0000370 LLCArgs.push_back(LLCPath.c_str());
Chris Lattnerffac2862006-06-06 22:30:59 +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 Lattnerfd381322010-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
Chris Lattnerffac2862006-06-06 22:30:59 +0000383 LLCArgs.push_back (0);
384
Chris Lattnerfd381322010-03-16 06:41:47 +0000385 outs() << (UseIntegratedAssembler ? "<llc-ia>" : "<llc>");
386 outs().flush();
Dan Gohmand8db3762009-07-15 16:35:29 +0000387 DEBUG(errs() << "\nAbout to run:\t";
Nick Lewycky6ba630b2010-04-12 05:08:25 +0000388 for (unsigned i = 0, e = LLCArgs.size()-1; i != e; ++i)
Dan Gohmand8db3762009-07-15 16:35:29 +0000389 errs() << " " << LLCArgs[i];
390 errs() << "\n";
Chris Lattnerffac2862006-06-06 22:30:59 +0000391 );
392 if (RunProgramWithTimeout(sys::Path(LLCPath), &LLCArgs[0],
Duncan Sandse9cd6d02010-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 Lewycky6ba630b2010-04-12 05:08:25 +0000397 return UseIntegratedAssembler ? GCC::ObjectFile : GCC::AsmFile;
Chris Lattnerffac2862006-06-06 22:30:59 +0000398}
399
Duncan Sandse9cd6d02010-05-24 07:49:55 +0000400void LLC::compileProgram(const std::string &Bitcode, std::string *Error,
401 unsigned Timeout, unsigned MemoryLimit) {
Chris Lattnerffac2862006-06-06 22:30:59 +0000402 sys::Path OutputAsmFile;
Duncan Sandse9cd6d02010-05-24 07:49:55 +0000403 OutputCode(Bitcode, OutputAsmFile, *Error, Timeout, MemoryLimit);
Chris Lattnerffac2862006-06-06 22:30:59 +0000404 OutputAsmFile.eraseFromDisk();
405}
406
Gabor Greif0e535c3c2007-07-04 21:55:50 +0000407int LLC::ExecuteProgram(const std::string &Bitcode,
Chris Lattnerffac2862006-06-06 22:30:59 +0000408 const std::vector<std::string> &Args,
409 const std::string &InputFile,
410 const std::string &OutputFile,
Nick Lewycky6ba630b2010-04-12 05:08:25 +0000411 std::string *Error,
Chris Lattnerffac2862006-06-06 22:30:59 +0000412 const std::vector<std::string> &ArgsForGCC,
413 const std::vector<std::string> &SharedLibs,
Anton Korobeynikovd01defe2007-02-16 19:11:07 +0000414 unsigned Timeout,
415 unsigned MemoryLimit) {
Chris Lattnerffac2862006-06-06 22:30:59 +0000416
417 sys::Path OutputAsmFile;
Duncan Sandse9cd6d02010-05-24 07:49:55 +0000418 GCC::FileType FileKind = OutputCode(Bitcode, OutputAsmFile, *Error, Timeout,
419 MemoryLimit);
Anton Korobeynikov7cbff912009-08-05 09:32:10 +0000420 FileRemover OutFileRemover(OutputAsmFile, !SaveTemps);
Chris Lattnerffac2862006-06-06 22:30:59 +0000421
422 std::vector<std::string> GCCArgs(ArgsForGCC);
Bill Wendlingbf5d8272009-03-02 23:13:18 +0000423 GCCArgs.insert(GCCArgs.end(), SharedLibs.begin(), SharedLibs.end());
Chris Lattnerffac2862006-06-06 22:30:59 +0000424
425 // Assuming LLC worked, compile the result with GCC and run it.
Chris Lattnerfd381322010-03-16 06:41:47 +0000426 return gcc->ExecuteProgram(OutputAsmFile.str(), Args, FileKind,
Nick Lewycky6ba630b2010-04-12 05:08:25 +0000427 InputFile, OutputFile, Error, GCCArgs,
Anton Korobeynikovd01defe2007-02-16 19:11:07 +0000428 Timeout, MemoryLimit);
Chris Lattnerffac2862006-06-06 22:30:59 +0000429}
430
431/// createLLC - Try to find the LLC executable
432///
Dan Gohman46ffffa2009-08-05 20:21:17 +0000433LLC *AbstractInterpreter::createLLC(const char *Argv0,
Chris Lattnerffac2862006-06-06 22:30:59 +0000434 std::string &Message,
Kalle Raiskila6be58292010-05-10 07:38:37 +0000435 const std::string &GCCBinary,
Bill Wendlingbf5d8272009-03-02 23:13:18 +0000436 const std::vector<std::string> *Args,
Chris Lattnerfd381322010-03-16 06:41:47 +0000437 const std::vector<std::string> *GCCArgs,
438 bool UseIntegratedAssembler) {
Dan Gohman46ffffa2009-08-05 20:21:17 +0000439 std::string LLCPath =
Chris Lattnerc521f542009-08-23 22:45:37 +0000440 FindExecutable("llc", Argv0, (void *)(intptr_t)&createLLC).str();
Chris Lattnerffac2862006-06-06 22:30:59 +0000441 if (LLCPath.empty()) {
Dan Gohman9e082462010-10-29 16:15:23 +0000442 Message = "Cannot find `llc' in executable directory!\n";
Chris Lattnerffac2862006-06-06 22:30:59 +0000443 return 0;
444 }
445
446 Message = "Found llc: " + LLCPath + "\n";
Kalle Raiskila6be58292010-05-10 07:38:37 +0000447 GCC *gcc = GCC::create(Message, GCCBinary, GCCArgs);
Chris Lattnerffac2862006-06-06 22:30:59 +0000448 if (!gcc) {
Dan Gohmand8db3762009-07-15 16:35:29 +0000449 errs() << Message << "\n";
Chris Lattnerffac2862006-06-06 22:30:59 +0000450 exit(1);
451 }
Kalle Raiskila6be58292010-05-10 07:38:37 +0000452 return new LLC(LLCPath, gcc, Args, UseIntegratedAssembler);
Chris Lattnerffac2862006-06-06 22:30:59 +0000453}
454
455//===---------------------------------------------------------------------===//
456// JIT Implementation of AbstractIntepreter interface
457//
458namespace {
459 class JIT : public AbstractInterpreter {
460 std::string LLIPath; // The path to the LLI executable
461 std::vector<std::string> ToolArgs; // Args to pass to LLI
462 public:
463 JIT(const std::string &Path, const std::vector<std::string> *Args)
464 : LLIPath(Path) {
465 ToolArgs.clear ();
466 if (Args) { ToolArgs = *Args; }
467 }
468
Gabor Greif0e535c3c2007-07-04 21:55:50 +0000469 virtual int ExecuteProgram(const std::string &Bitcode,
Chris Lattnerffac2862006-06-06 22:30:59 +0000470 const std::vector<std::string> &Args,
471 const std::string &InputFile,
472 const std::string &OutputFile,
Nick Lewycky6ba630b2010-04-12 05:08:25 +0000473 std::string *Error,
Chris Lattnerffac2862006-06-06 22:30:59 +0000474 const std::vector<std::string> &GCCArgs =
475 std::vector<std::string>(),
476 const std::vector<std::string> &SharedLibs =
477 std::vector<std::string>(),
Nick Lewycky6ba630b2010-04-12 05:08:25 +0000478 unsigned Timeout = 0,
479 unsigned MemoryLimit = 0);
Chris Lattnerffac2862006-06-06 22:30:59 +0000480 };
481}
482
Gabor Greif0e535c3c2007-07-04 21:55:50 +0000483int JIT::ExecuteProgram(const std::string &Bitcode,
Chris Lattnerffac2862006-06-06 22:30:59 +0000484 const std::vector<std::string> &Args,
485 const std::string &InputFile,
486 const std::string &OutputFile,
Nick Lewycky6ba630b2010-04-12 05:08:25 +0000487 std::string *Error,
Chris Lattnerffac2862006-06-06 22:30:59 +0000488 const std::vector<std::string> &GCCArgs,
489 const std::vector<std::string> &SharedLibs,
Anton Korobeynikovd01defe2007-02-16 19:11:07 +0000490 unsigned Timeout,
491 unsigned MemoryLimit) {
Chris Lattnerffac2862006-06-06 22:30:59 +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());
495 JITArgs.push_back("-force-interpreter=false");
496
497 // 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
501 for (unsigned i = 0, e = SharedLibs.size(); i != e; ++i) {
502 JITArgs.push_back("-load");
503 JITArgs.push_back(SharedLibs[i].c_str());
504 }
Gabor Greif0e535c3c2007-07-04 21:55:50 +0000505 JITArgs.push_back(Bitcode.c_str());
Chris Lattnerffac2862006-06-06 22:30:59 +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 Gohman607818a2009-07-15 17:29:42 +0000511 outs() << "<jit>"; outs().flush();
Dan Gohmand8db3762009-07-15 16:35:29 +0000512 DEBUG(errs() << "\nAbout to run:\t";
Chris Lattnerffac2862006-06-06 22:30:59 +0000513 for (unsigned i=0, e = JITArgs.size()-1; i != e; ++i)
Dan Gohmand8db3762009-07-15 16:35:29 +0000514 errs() << " " << JITArgs[i];
515 errs() << "\n";
Chris Lattnerffac2862006-06-06 22:30:59 +0000516 );
Dan Gohmand8db3762009-07-15 16:35:29 +0000517 DEBUG(errs() << "\nSending output to " << OutputFile << "\n");
Chris Lattnerffac2862006-06-06 22:30:59 +0000518 return RunProgramWithTimeout(sys::Path(LLIPath), &JITArgs[0],
519 sys::Path(InputFile), sys::Path(OutputFile), sys::Path(OutputFile),
Anton Korobeynikovd01defe2007-02-16 19:11:07 +0000520 Timeout, MemoryLimit);
Chris Lattnerffac2862006-06-06 22:30:59 +0000521}
522
523/// createJIT - Try to find the LLI executable
524///
Dan Gohman46ffffa2009-08-05 20:21:17 +0000525AbstractInterpreter *AbstractInterpreter::createJIT(const char *Argv0,
Chris Lattnerffac2862006-06-06 22:30:59 +0000526 std::string &Message, const std::vector<std::string> *Args) {
Dan Gohman46ffffa2009-08-05 20:21:17 +0000527 std::string LLIPath =
Chris Lattnerc521f542009-08-23 22:45:37 +0000528 FindExecutable("lli", Argv0, (void *)(intptr_t)&createJIT).str();
Chris Lattnerffac2862006-06-06 22:30:59 +0000529 if (!LLIPath.empty()) {
530 Message = "Found lli: " + LLIPath + "\n";
531 return new JIT(LLIPath, Args);
532 }
533
Dan Gohman9e082462010-10-29 16:15:23 +0000534 Message = "Cannot find `lli' in executable directory!\n";
Chris Lattnerffac2862006-06-06 22:30:59 +0000535 return 0;
536}
537
Gabor Greif0e535c3c2007-07-04 21:55:50 +0000538GCC::FileType CBE::OutputCode(const std::string &Bitcode,
Duncan Sandse9cd6d02010-05-24 07:49:55 +0000539 sys::Path &OutputCFile, std::string &Error,
540 unsigned Timeout, unsigned MemoryLimit) {
Gabor Greif0e535c3c2007-07-04 21:55:50 +0000541 sys::Path uniqueFile(Bitcode+".cbe.c");
Reid Spencere4ca7222006-08-23 20:34:57 +0000542 std::string ErrMsg;
543 if (uniqueFile.makeUnique(true, &ErrMsg)) {
Dan Gohmand8db3762009-07-15 16:35:29 +0000544 errs() << "Error making unique filename: " << ErrMsg << "\n";
Reid Spencere4ca7222006-08-23 20:34:57 +0000545 exit(1);
546 }
Chris Lattnerffac2862006-06-06 22:30:59 +0000547 OutputCFile = uniqueFile;
548 std::vector<const char *> LLCArgs;
Nick Lewycky6ba630b2010-04-12 05:08:25 +0000549 LLCArgs.push_back(LLCPath.c_str());
Chris Lattnerffac2862006-06-06 22:30:59 +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 Lewycky6ba630b2010-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 Lewycky6ba630b2010-04-12 05:08:25 +0000558 LLCArgs.push_back(Bitcode.c_str()); // This is the input bitcode
559 LLCArgs.push_back(0);
Chris Lattnerffac2862006-06-06 22:30:59 +0000560
Dan Gohman607818a2009-07-15 17:29:42 +0000561 outs() << "<cbe>"; outs().flush();
Dan Gohmand8db3762009-07-15 16:35:29 +0000562 DEBUG(errs() << "\nAbout to run:\t";
Nick Lewycky6ba630b2010-04-12 05:08:25 +0000563 for (unsigned i = 0, e = LLCArgs.size()-1; i != e; ++i)
Dan Gohmand8db3762009-07-15 16:35:29 +0000564 errs() << " " << LLCArgs[i];
565 errs() << "\n";
Chris Lattnerffac2862006-06-06 22:30:59 +0000566 );
567 if (RunProgramWithTimeout(LLCPath, &LLCArgs[0], sys::Path(), sys::Path(),
Duncan Sandse9cd6d02010-05-24 07:49:55 +0000568 sys::Path(), Timeout, MemoryLimit))
569 Error = ProcessFailure(LLCPath, &LLCArgs[0], Timeout, MemoryLimit);
Chris Lattner634bc042006-09-15 21:29:15 +0000570 return GCC::CFile;
Chris Lattnerffac2862006-06-06 22:30:59 +0000571}
572
Duncan Sandse9cd6d02010-05-24 07:49:55 +0000573void CBE::compileProgram(const std::string &Bitcode, std::string *Error,
574 unsigned Timeout, unsigned MemoryLimit) {
Chris Lattnerffac2862006-06-06 22:30:59 +0000575 sys::Path OutputCFile;
Duncan Sandse9cd6d02010-05-24 07:49:55 +0000576 OutputCode(Bitcode, OutputCFile, *Error, Timeout, MemoryLimit);
Chris Lattnerffac2862006-06-06 22:30:59 +0000577 OutputCFile.eraseFromDisk();
578}
579
Gabor Greif0e535c3c2007-07-04 21:55:50 +0000580int CBE::ExecuteProgram(const std::string &Bitcode,
Chris Lattnerffac2862006-06-06 22:30:59 +0000581 const std::vector<std::string> &Args,
582 const std::string &InputFile,
583 const std::string &OutputFile,
Nick Lewycky6ba630b2010-04-12 05:08:25 +0000584 std::string *Error,
Chris Lattnerffac2862006-06-06 22:30:59 +0000585 const std::vector<std::string> &ArgsForGCC,
586 const std::vector<std::string> &SharedLibs,
Anton Korobeynikovd01defe2007-02-16 19:11:07 +0000587 unsigned Timeout,
588 unsigned MemoryLimit) {
Chris Lattnerffac2862006-06-06 22:30:59 +0000589 sys::Path OutputCFile;
Duncan Sandse9cd6d02010-05-24 07:49:55 +0000590 OutputCode(Bitcode, OutputCFile, *Error, Timeout, MemoryLimit);
Chris Lattnerffac2862006-06-06 22:30:59 +0000591
Anton Korobeynikov7cbff912009-08-05 09:32:10 +0000592 FileRemover CFileRemove(OutputCFile, !SaveTemps);
Chris Lattnerffac2862006-06-06 22:30:59 +0000593
594 std::vector<std::string> GCCArgs(ArgsForGCC);
Bill Wendlingbf5d8272009-03-02 23:13:18 +0000595 GCCArgs.insert(GCCArgs.end(), SharedLibs.begin(), SharedLibs.end());
596
Chris Lattnerc521f542009-08-23 22:45:37 +0000597 return gcc->ExecuteProgram(OutputCFile.str(), Args, GCC::CFile,
Nick Lewycky6ba630b2010-04-12 05:08:25 +0000598 InputFile, OutputFile, Error, GCCArgs,
Anton Korobeynikovd01defe2007-02-16 19:11:07 +0000599 Timeout, MemoryLimit);
Chris Lattnerffac2862006-06-06 22:30:59 +0000600}
601
602/// createCBE - Try to find the 'llc' executable
603///
Dan Gohman46ffffa2009-08-05 20:21:17 +0000604CBE *AbstractInterpreter::createCBE(const char *Argv0,
Chris Lattnerffac2862006-06-06 22:30:59 +0000605 std::string &Message,
Kalle Raiskila6be58292010-05-10 07:38:37 +0000606 const std::string &GCCBinary,
Bill Wendlingbf5d8272009-03-02 23:13:18 +0000607 const std::vector<std::string> *Args,
608 const std::vector<std::string> *GCCArgs) {
Dan Gohman46ffffa2009-08-05 20:21:17 +0000609 sys::Path LLCPath =
Dan Gohman198f5e82009-08-05 21:03:39 +0000610 FindExecutable("llc", Argv0, (void *)(intptr_t)&createCBE);
Chris Lattnerffac2862006-06-06 22:30:59 +0000611 if (LLCPath.isEmpty()) {
612 Message =
Dan Gohman9e082462010-10-29 16:15:23 +0000613 "Cannot find `llc' in executable directory!\n";
Chris Lattnerffac2862006-06-06 22:30:59 +0000614 return 0;
615 }
616
Chris Lattnerc521f542009-08-23 22:45:37 +0000617 Message = "Found llc: " + LLCPath.str() + "\n";
Kalle Raiskila6be58292010-05-10 07:38:37 +0000618 GCC *gcc = GCC::create(Message, GCCBinary, GCCArgs);
Chris Lattnerffac2862006-06-06 22:30:59 +0000619 if (!gcc) {
Dan Gohmand8db3762009-07-15 16:35:29 +0000620 errs() << Message << "\n";
Chris Lattnerffac2862006-06-06 22:30:59 +0000621 exit(1);
622 }
623 return new CBE(LLCPath, gcc, Args);
624}
625
626//===---------------------------------------------------------------------===//
627// GCC abstraction
628//
David Goodwin73f4e3f2009-07-10 21:39:28 +0000629
Jakob Stoklund Olesen12222492010-07-29 00:52:16 +0000630static bool IsARMArchitecture(std::vector<const char*> Args) {
631 for (std::vector<const char*>::const_iterator
David Goodwin73f4e3f2009-07-10 21:39:28 +0000632 I = Args.begin(), E = Args.end(); I != E; ++I) {
Jakob Stoklund Olesen3e0ddc02010-05-13 17:58:15 +0000633 if (StringRef(*I).equals_lower("-arch")) {
David Goodwin73f4e3f2009-07-10 21:39:28 +0000634 ++I;
Jakob Stoklund Olesen3e0ddc02010-05-13 17:58:15 +0000635 if (I != E && StringRef(*I).substr(0, strlen("arm")).equals_lower("arm"))
David Goodwin73f4e3f2009-07-10 21:39:28 +0000636 return true;
David Goodwin73f4e3f2009-07-10 21:39:28 +0000637 }
638 }
639
640 return false;
641}
642
Chris Lattnerffac2862006-06-06 22:30:59 +0000643int GCC::ExecuteProgram(const std::string &ProgramFile,
644 const std::vector<std::string> &Args,
645 FileType fileType,
646 const std::string &InputFile,
647 const std::string &OutputFile,
Nick Lewycky6ba630b2010-04-12 05:08:25 +0000648 std::string *Error,
Chris Lattnerffac2862006-06-06 22:30:59 +0000649 const std::vector<std::string> &ArgsForGCC,
Anton Korobeynikovd01defe2007-02-16 19:11:07 +0000650 unsigned Timeout,
651 unsigned MemoryLimit) {
Chris Lattnerffac2862006-06-06 22:30:59 +0000652 std::vector<const char*> GCCArgs;
653
654 GCCArgs.push_back(GCCPath.c_str());
655
Chris Lattnerfd381322010-03-16 06:41:47 +0000656 if (TargetTriple.getArch() == Triple::x86)
657 GCCArgs.push_back("-m32");
658
Bill Wendlingbf5d8272009-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 Lattnerffac2862006-06-06 22:30:59 +0000663 // Specify -x explicitly in case the extension is wonky
Chris Lattnerfd381322010-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 Dunbar8575a602009-08-18 03:35:57 +0000671
Chris Lattnerfd381322010-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) &&
Jakob Stoklund Olesen12222492010-07-29 00:52:16 +0000676 !IsARMArchitecture(GCCArgs))
Chris Lattnerfd381322010-03-16 06:41:47 +0000677 GCCArgs.push_back("-force_cpusubtype_ALL");
678 }
Chris Lattnerffac2862006-06-06 22:30:59 +0000679 }
Chris Lattnerfd381322010-03-16 06:41:47 +0000680
681 GCCArgs.push_back(ProgramFile.c_str()); // Specify the input filename.
682
Chris Lattner572e78c2006-06-09 21:31:53 +0000683 GCCArgs.push_back("-x");
684 GCCArgs.push_back("none");
Chris Lattnerffac2862006-06-06 22:30:59 +0000685 GCCArgs.push_back("-o");
686 sys::Path OutputBinary (ProgramFile+".gcc.exe");
Reid Spencere4ca7222006-08-23 20:34:57 +0000687 std::string ErrMsg;
688 if (OutputBinary.makeUnique(true, &ErrMsg)) {
Dan Gohmand8db3762009-07-15 16:35:29 +0000689 errs() << "Error making unique filename: " << ErrMsg << "\n";
Reid Spencere4ca7222006-08-23 20:34:57 +0000690 exit(1);
691 }
Chris Lattnerffac2862006-06-06 22:30:59 +0000692 GCCArgs.push_back(OutputBinary.c_str()); // Output to the right file...
693
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
701 GCCArgs.push_back("-lm"); // Hard-code the math library...
702 GCCArgs.push_back("-O2"); // Optimize the program a bit...
703#if defined (HAVE_LINK_R)
704 GCCArgs.push_back("-Wl,-R."); // Search this dir for .so files
705#endif
Daniel Dunbar8575a602009-08-18 03:35:57 +0000706 if (TargetTriple.getArch() == Triple::sparc)
707 GCCArgs.push_back("-mcpu=v9");
Chris Lattnerffac2862006-06-06 22:30:59 +0000708 GCCArgs.push_back(0); // NULL terminator
709
Dan Gohman607818a2009-07-15 17:29:42 +0000710 outs() << "<gcc>"; outs().flush();
Dan Gohmand8db3762009-07-15 16:35:29 +0000711 DEBUG(errs() << "\nAbout to run:\t";
Nick Lewycky6ba630b2010-04-12 05:08:25 +0000712 for (unsigned i = 0, e = GCCArgs.size()-1; i != e; ++i)
Dan Gohmand8db3762009-07-15 16:35:29 +0000713 errs() << " " << GCCArgs[i];
714 errs() << "\n";
Evan Chenga50e3912007-01-03 07:44:30 +0000715 );
Chris Lattnerffac2862006-06-06 22:30:59 +0000716 if (RunProgramWithTimeout(GCCPath, &GCCArgs[0], sys::Path(), sys::Path(),
717 sys::Path())) {
Nick Lewycky6ba630b2010-04-12 05:08:25 +0000718 *Error = ProcessFailure(GCCPath, &GCCArgs[0]);
719 return -1;
Chris Lattnerffac2862006-06-06 22:30:59 +0000720 }
721
Rafael Espindola74cc8b22010-07-24 23:05:45 +0000722 std::vector<const char*> ProgramArgs;
Rafael Espindolaf8203a32010-07-24 23:02:11 +0000723
724 // Declared here so that the destructor only runs after
725 // ProgramArgs is used.
726 std::string Exec;
Chris Lattnerffac2862006-06-06 22:30:59 +0000727
Evan Cheng7aedcf12008-09-09 06:11:26 +0000728 if (RemoteClientPath.isEmpty())
Evan Cheng779b52e2007-05-03 18:36:15 +0000729 ProgramArgs.push_back(OutputBinary.c_str());
730 else {
Evan Cheng7aedcf12008-09-09 06:11:26 +0000731 ProgramArgs.push_back(RemoteClientPath.c_str());
732 ProgramArgs.push_back(RemoteHost.c_str());
Anton Korobeynikov0c3f8d52009-08-05 09:32:53 +0000733 if (!RemoteUser.empty()) {
734 ProgramArgs.push_back("-l");
735 ProgramArgs.push_back(RemoteUser.c_str());
736 }
David Goodwin73f4e3f2009-07-10 21:39:28 +0000737 if (!RemotePort.empty()) {
738 ProgramArgs.push_back("-p");
739 ProgramArgs.push_back(RemotePort.c_str());
740 }
Evan Cheng7aedcf12008-09-09 06:11:26 +0000741 if (!RemoteExtra.empty()) {
742 ProgramArgs.push_back(RemoteExtra.c_str());
743 }
Evan Cheng779b52e2007-05-03 18:36:15 +0000744
David Goodwin47ca4612009-07-20 17:15:03 +0000745 // Full path to the binary. We need to cd to the exec directory because
746 // there is a dylib there that the exec expects to find in the CWD
Evan Cheng779b52e2007-05-03 18:36:15 +0000747 char* env_pwd = getenv("PWD");
Rafael Espindolaf8203a32010-07-24 23:02:11 +0000748 Exec = "cd ";
Evan Cheng779b52e2007-05-03 18:36:15 +0000749 Exec += env_pwd;
David Goodwin47ca4612009-07-20 17:15:03 +0000750 Exec += "; ./";
Evan Cheng779b52e2007-05-03 18:36:15 +0000751 Exec += OutputBinary.c_str();
752 ProgramArgs.push_back(Exec.c_str());
753 }
754
Chris Lattnerffac2862006-06-06 22:30:59 +0000755 // Add optional parameters to the running program from Argv
Nick Lewycky6ba630b2010-04-12 05:08:25 +0000756 for (unsigned i = 0, e = Args.size(); i != e; ++i)
Chris Lattnerffac2862006-06-06 22:30:59 +0000757 ProgramArgs.push_back(Args[i].c_str());
758 ProgramArgs.push_back(0); // NULL terminator
759
760 // Now that we have a binary, run it!
Dan Gohman607818a2009-07-15 17:29:42 +0000761 outs() << "<program>"; outs().flush();
Dan Gohmand8db3762009-07-15 16:35:29 +0000762 DEBUG(errs() << "\nAbout to run:\t";
Nick Lewycky6ba630b2010-04-12 05:08:25 +0000763 for (unsigned i = 0, e = ProgramArgs.size()-1; i != e; ++i)
Dan Gohmand8db3762009-07-15 16:35:29 +0000764 errs() << " " << ProgramArgs[i];
765 errs() << "\n";
Chris Lattnerffac2862006-06-06 22:30:59 +0000766 );
767
Anton Korobeynikov7cbff912009-08-05 09:32:10 +0000768 FileRemover OutputBinaryRemover(OutputBinary, !SaveTemps);
Evan Cheng779b52e2007-05-03 18:36:15 +0000769
Viktor Kutuzov1518de12009-07-14 19:10:55 +0000770 if (RemoteClientPath.isEmpty()) {
Nick Lewycky6ba630b2010-04-12 05:08:25 +0000771 DEBUG(errs() << "<run locally>");
Evan Cheng779b52e2007-05-03 18:36:15 +0000772 return RunProgramWithTimeout(OutputBinary, &ProgramArgs[0],
773 sys::Path(InputFile), sys::Path(OutputFile), sys::Path(OutputFile),
774 Timeout, MemoryLimit);
Viktor Kutuzov1518de12009-07-14 19:10:55 +0000775 } else {
Dan Gohman607818a2009-07-15 17:29:42 +0000776 outs() << "<run remotely>"; outs().flush();
Viktor Kutuzov85870252009-07-18 18:39:24 +0000777 return RunProgramRemotelyWithTimeout(sys::Path(RemoteClientPath),
Viktor Kutuzov1518de12009-07-14 19:10:55 +0000778 &ProgramArgs[0], sys::Path(InputFile), sys::Path(OutputFile),
779 sys::Path(OutputFile), Timeout, MemoryLimit);
Viktor Kutuzov1518de12009-07-14 19:10:55 +0000780 }
Chris Lattnerffac2862006-06-06 22:30:59 +0000781}
782
783int GCC::MakeSharedObject(const std::string &InputFile, FileType fileType,
Chris Lattnercc21fa72006-06-27 20:35:36 +0000784 std::string &OutputFile,
Nick Lewycky6ba630b2010-04-12 05:08:25 +0000785 const std::vector<std::string> &ArgsForGCC,
786 std::string &Error) {
Chris Lattnerffac2862006-06-06 22:30:59 +0000787 sys::Path uniqueFilename(InputFile+LTDL_SHLIB_EXT);
Reid Spencere4ca7222006-08-23 20:34:57 +0000788 std::string ErrMsg;
789 if (uniqueFilename.makeUnique(true, &ErrMsg)) {
Dan Gohmand8db3762009-07-15 16:35:29 +0000790 errs() << "Error making unique filename: " << ErrMsg << "\n";
Reid Spencere4ca7222006-08-23 20:34:57 +0000791 exit(1);
792 }
Chris Lattnerc521f542009-08-23 22:45:37 +0000793 OutputFile = uniqueFilename.str();
Chris Lattnerffac2862006-06-06 22:30:59 +0000794
Chris Lattnercc21fa72006-06-27 20:35:36 +0000795 std::vector<const char*> GCCArgs;
796
797 GCCArgs.push_back(GCCPath.c_str());
Evan Cheng690b6352009-03-12 00:53:34 +0000798
Chris Lattnerfd381322010-03-16 06:41:47 +0000799 if (TargetTriple.getArch() == Triple::x86)
800 GCCArgs.push_back("-m32");
801
Evan Cheng690b6352009-03-12 00:53:34 +0000802 for (std::vector<std::string>::const_iterator
803 I = gccArgs.begin(), E = gccArgs.end(); I != E; ++I)
804 GCCArgs.push_back(I->c_str());
805
Chris Lattnerffac2862006-06-06 22:30:59 +0000806 // Compile the C/asm file into a shared object
Chris Lattnerfd381322010-03-16 06:41:47 +0000807 if (fileType != ObjectFile) {
808 GCCArgs.push_back("-x");
809 GCCArgs.push_back(fileType == AsmFile ? "assembler" : "c");
810 }
Chris Lattnercc21fa72006-06-27 20:35:36 +0000811 GCCArgs.push_back("-fno-strict-aliasing");
812 GCCArgs.push_back(InputFile.c_str()); // Specify the input filename.
Lauro Ramos Venanciof04823a2007-06-06 23:10:56 +0000813 GCCArgs.push_back("-x");
814 GCCArgs.push_back("none");
Daniel Dunbar8575a602009-08-18 03:35:57 +0000815 if (TargetTriple.getArch() == Triple::sparc)
816 GCCArgs.push_back("-G"); // Compile a shared library, `-G' for Sparc
817 else if (TargetTriple.getOS() == Triple::Darwin) {
818 // link all source files into a single module in data segment, rather than
819 // generating blocks. dynamic_lookup requires that you set
820 // MACOSX_DEPLOYMENT_TARGET=10.3 in your env. FIXME: it would be better for
821 // bugpoint to just pass that in the environment of GCC.
822 GCCArgs.push_back("-single_module");
823 GCCArgs.push_back("-dynamiclib"); // `-dynamiclib' for MacOS X/PowerPC
824 GCCArgs.push_back("-undefined");
825 GCCArgs.push_back("dynamic_lookup");
826 } else
827 GCCArgs.push_back("-shared"); // `-shared' for Linux/X86, maybe others
Chris Lattnerffac2862006-06-06 22:30:59 +0000828
Daniel Dunbar8575a602009-08-18 03:35:57 +0000829 if ((TargetTriple.getArch() == Triple::alpha) ||
830 (TargetTriple.getArch() == Triple::x86_64))
831 GCCArgs.push_back("-fPIC"); // Requires shared objs to contain PIC
832
833 if (TargetTriple.getArch() == Triple::sparc)
834 GCCArgs.push_back("-mcpu=v9");
835
Chris Lattnercc21fa72006-06-27 20:35:36 +0000836 GCCArgs.push_back("-o");
837 GCCArgs.push_back(OutputFile.c_str()); // Output to the right filename.
838 GCCArgs.push_back("-O2"); // Optimize the program a bit.
839
840
841
842 // Add any arguments intended for GCC. We locate them here because this is
843 // most likely -L and -l options that need to come before other libraries but
844 // after the source. Other options won't be sensitive to placement on the
845 // command line, so this should be safe.
846 for (unsigned i = 0, e = ArgsForGCC.size(); i != e; ++i)
847 GCCArgs.push_back(ArgsForGCC[i].c_str());
848 GCCArgs.push_back(0); // NULL terminator
849
850
Chris Lattnerffac2862006-06-06 22:30:59 +0000851
Dan Gohman607818a2009-07-15 17:29:42 +0000852 outs() << "<gcc>"; outs().flush();
Dan Gohmand8db3762009-07-15 16:35:29 +0000853 DEBUG(errs() << "\nAbout to run:\t";
Nick Lewycky6ba630b2010-04-12 05:08:25 +0000854 for (unsigned i = 0, e = GCCArgs.size()-1; i != e; ++i)
Dan Gohmand8db3762009-07-15 16:35:29 +0000855 errs() << " " << GCCArgs[i];
856 errs() << "\n";
Evan Chenga50e3912007-01-03 07:44:30 +0000857 );
Chris Lattnercc21fa72006-06-27 20:35:36 +0000858 if (RunProgramWithTimeout(GCCPath, &GCCArgs[0], sys::Path(), sys::Path(),
Chris Lattnerffac2862006-06-06 22:30:59 +0000859 sys::Path())) {
Nick Lewycky6ba630b2010-04-12 05:08:25 +0000860 Error = ProcessFailure(GCCPath, &GCCArgs[0]);
Chris Lattnerffac2862006-06-06 22:30:59 +0000861 return 1;
862 }
863 return 0;
864}
865
866/// create - Try to find the `gcc' executable
867///
Dan Gohman46ffffa2009-08-05 20:21:17 +0000868GCC *GCC::create(std::string &Message,
Kalle Raiskila6be58292010-05-10 07:38:37 +0000869 const std::string &GCCBinary,
Bill Wendlingbf5d8272009-03-02 23:13:18 +0000870 const std::vector<std::string> *Args) {
Kalle Raiskila6be58292010-05-10 07:38:37 +0000871 sys::Path GCCPath = sys::Program::FindProgramByName(GCCBinary);
Chris Lattnerffac2862006-06-06 22:30:59 +0000872 if (GCCPath.isEmpty()) {
Dan Gohman9e082462010-10-29 16:15:23 +0000873 Message = "Cannot find `"+ GCCBinary +"' in PATH!\n";
Chris Lattnerffac2862006-06-06 22:30:59 +0000874 return 0;
875 }
876
Evan Cheng7aedcf12008-09-09 06:11:26 +0000877 sys::Path RemoteClientPath;
878 if (!RemoteClient.empty())
Dan Gohman46ffffa2009-08-05 20:21:17 +0000879 RemoteClientPath = sys::Program::FindProgramByName(RemoteClient);
Evan Cheng779b52e2007-05-03 18:36:15 +0000880
Chris Lattnerc521f542009-08-23 22:45:37 +0000881 Message = "Found gcc: " + GCCPath.str() + "\n";
Bill Wendlingbf5d8272009-03-02 23:13:18 +0000882 return new GCC(GCCPath, RemoteClientPath, Args);
Chris Lattnerffac2862006-06-06 22:30:59 +0000883}