blob: a5e1e8b8a42affc09dee156acf40b4e1015389fd [file] [log] [blame]
Chris Lattner7915a1e2003-10-14 21:34:11 +00001//===-- ToolRunner.cpp ----------------------------------------------------===//
Misha Brukmanf976c852005-04-21 22:55:34 +00002//
John Criswellb576c942003-10-20 19:43:21 +00003// The LLVM Compiler Infrastructure
4//
Chris Lattner21c62da2007-12-29 20:44:31 +00005// This file is distributed under the University of Illinois Open Source
6// License. See LICENSE.TXT for details.
Misha Brukmanf976c852005-04-21 22:55:34 +00007//
John Criswellb576c942003-10-20 19:43:21 +00008//===----------------------------------------------------------------------===//
Chris Lattner7915a1e2003-10-14 21:34:11 +00009//
10// This file implements the interfaces described in the ToolRunner.h file.
11//
12//===----------------------------------------------------------------------===//
13
Chris Lattner0b1fe842003-10-19 02:27:40 +000014#define DEBUG_TYPE "toolrunner"
Chris Lattnerf1b20d82006-06-06 22:30:59 +000015#include "ToolRunner.h"
Reid Spencer551ccae2004-09-01 22:55:40 +000016#include "llvm/Config/config.h" // for HAVE_LINK_R
Chris Lattner45495c52005-02-13 23:13:47 +000017#include "llvm/System/Program.h"
Evan Cheng34e400e2007-05-03 18:36:15 +000018#include "llvm/Support/CommandLine.h"
Reid Spencer551ccae2004-09-01 22:55:40 +000019#include "llvm/Support/Debug.h"
20#include "llvm/Support/FileUtilities.h"
Chris Lattner7915a1e2003-10-14 21:34:11 +000021#include <fstream>
Chris Lattner89bf9ea2004-02-18 20:38:00 +000022#include <sstream>
Chris Lattner2cdd21c2003-12-14 21:35:53 +000023using namespace llvm;
Brian Gaeked0fde302003-11-11 22:41:34 +000024
Evan Cheng34e400e2007-05-03 18:36:15 +000025namespace {
26 cl::opt<std::string>
Evan Cheng70f684f2008-09-09 06:11:26 +000027 RemoteClient("remote-client",
28 cl::desc("Remote execution client (rsh/ssh)"));
Evan Cheng34e400e2007-05-03 18:36:15 +000029
30 cl::opt<std::string>
Evan Cheng70f684f2008-09-09 06:11:26 +000031 RemoteHost("remote-host",
32 cl::desc("Remote execution (rsh/ssh) host"));
33
34 cl::opt<std::string>
David Goodwin80becf12009-07-10 21:39:28 +000035 RemotePort("remote-port",
36 cl::desc("Remote execution (rsh/ssh) port"));
37
38 cl::opt<std::string>
Evan Cheng70f684f2008-09-09 06:11:26 +000039 RemoteUser("remote-user",
40 cl::desc("Remote execution (rsh/ssh) user id"));
41
42 cl::opt<std::string>
43 RemoteExtra("remote-extra-options",
44 cl::desc("Remote execution (rsh/ssh) extra options"));
Evan Cheng34e400e2007-05-03 18:36:15 +000045}
46
Alkis Evlogimenos1d29a6d2004-02-19 07:39:26 +000047ToolExecutionError::~ToolExecutionError() throw() { }
48
Chris Lattner45495c52005-02-13 23:13:47 +000049/// RunProgramWithTimeout - This function provides an alternate interface to the
50/// sys::Program::ExecuteAndWait interface.
51/// @see sys:Program::ExecuteAndWait
52static int RunProgramWithTimeout(const sys::Path &ProgramPath,
53 const char **Args,
54 const sys::Path &StdInFile,
55 const sys::Path &StdOutFile,
56 const sys::Path &StdErrFile,
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +000057 unsigned NumSeconds = 0,
58 unsigned MemoryLimit = 0) {
Chris Lattner45495c52005-02-13 23:13:47 +000059 const sys::Path* redirects[3];
60 redirects[0] = &StdInFile;
61 redirects[1] = &StdOutFile;
62 redirects[2] = &StdErrFile;
Chris Lattnerc600f3c2006-09-15 21:29:15 +000063
Chris Lattnerd4223502006-09-15 23:01:10 +000064 if (0) {
Dan Gohman65f57c22009-07-15 16:35:29 +000065 errs() << "RUN:";
Chris Lattnerd4223502006-09-15 23:01:10 +000066 for (unsigned i = 0; Args[i]; ++i)
Dan Gohman65f57c22009-07-15 16:35:29 +000067 errs() << " " << Args[i];
68 errs() << "\n";
Chris Lattnerd4223502006-09-15 23:01:10 +000069 }
Misha Brukmanf976c852005-04-21 22:55:34 +000070
71 return
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +000072 sys::Program::ExecuteAndWait(ProgramPath, Args, 0, redirects,
73 NumSeconds, MemoryLimit);
Chris Lattner45495c52005-02-13 23:13:47 +000074}
75
76
77
Reid Spencerb31baa82004-12-19 18:00:21 +000078static void ProcessFailure(sys::Path ProgPath, const char** Args) {
Chris Lattner89bf9ea2004-02-18 20:38:00 +000079 std::ostringstream OS;
Chris Lattnera3de1172004-02-18 20:58:00 +000080 OS << "\nError running tool:\n ";
Chris Lattner89bf9ea2004-02-18 20:38:00 +000081 for (const char **Arg = Args; *Arg; ++Arg)
82 OS << " " << *Arg;
83 OS << "\n";
84
85 // Rerun the compiler, capturing any error messages to print them.
Dan Gohmana971b292009-02-12 20:53:27 +000086 sys::Path ErrorFilename("bugpoint.program_error_messages");
Reid Spencer51c5a282006-08-23 20:34:57 +000087 std::string ErrMsg;
88 if (ErrorFilename.makeUnique(true, &ErrMsg)) {
Dan Gohman65f57c22009-07-15 16:35:29 +000089 errs() << "Error making unique filename: " << ErrMsg << "\n";
Reid Spencer51c5a282006-08-23 20:34:57 +000090 exit(1);
91 }
Reid Spencerb31baa82004-12-19 18:00:21 +000092 RunProgramWithTimeout(ProgPath, Args, sys::Path(""), ErrorFilename,
Reid Spencer8ea5ecb2006-08-21 06:04:45 +000093 ErrorFilename); // FIXME: check return code ?
Chris Lattner89bf9ea2004-02-18 20:38:00 +000094
95 // Print out the error messages generated by GCC if possible...
96 std::ifstream ErrorFile(ErrorFilename.c_str());
97 if (ErrorFile) {
98 std::copy(std::istreambuf_iterator<char>(ErrorFile),
99 std::istreambuf_iterator<char>(),
100 std::ostreambuf_iterator<char>(OS));
101 ErrorFile.close();
102 }
103
Reid Spencera229c5c2005-07-08 03:08:58 +0000104 ErrorFilename.eraseFromDisk();
Chris Lattner89bf9ea2004-02-18 20:38:00 +0000105 throw ToolExecutionError(OS.str());
106}
107
Misha Brukman9558c6a2003-09-29 22:39:25 +0000108//===---------------------------------------------------------------------===//
109// LLI Implementation of AbstractIntepreter interface
110//
Chris Lattner2cdd21c2003-12-14 21:35:53 +0000111namespace {
112 class LLI : public AbstractInterpreter {
113 std::string LLIPath; // The path to the LLI executable
Brian Gaeked11577b2004-05-04 21:09:01 +0000114 std::vector<std::string> ToolArgs; // Args to pass to LLI
Chris Lattner2cdd21c2003-12-14 21:35:53 +0000115 public:
Brian Gaeked11577b2004-05-04 21:09:01 +0000116 LLI(const std::string &Path, const std::vector<std::string> *Args)
117 : LLIPath(Path) {
118 ToolArgs.clear ();
Brian Gaeke48b008d2004-05-04 22:02:41 +0000119 if (Args) { ToolArgs = *Args; }
Brian Gaeked11577b2004-05-04 21:09:01 +0000120 }
Misha Brukmanf976c852005-04-21 22:55:34 +0000121
Gabor Greif8ff70c22007-07-04 21:55:50 +0000122 virtual int ExecuteProgram(const std::string &Bitcode,
Chris Lattner2cdd21c2003-12-14 21:35:53 +0000123 const std::vector<std::string> &Args,
124 const std::string &InputFile,
125 const std::string &OutputFile,
Reid Spencer51ab5c82006-06-06 00:00:42 +0000126 const std::vector<std::string> &GCCArgs,
Misha Brukmanf976c852005-04-21 22:55:34 +0000127 const std::vector<std::string> &SharedLibs =
Chris Lattnere96b2ed2004-07-24 07:49:11 +0000128 std::vector<std::string>(),
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000129 unsigned Timeout = 0,
130 unsigned MemoryLimit = 0);
Chris Lattner2cdd21c2003-12-14 21:35:53 +0000131 };
132}
Misha Brukman9558c6a2003-09-29 22:39:25 +0000133
Gabor Greif8ff70c22007-07-04 21:55:50 +0000134int LLI::ExecuteProgram(const std::string &Bitcode,
Chris Lattner7915a1e2003-10-14 21:34:11 +0000135 const std::vector<std::string> &Args,
Misha Brukman9558c6a2003-09-29 22:39:25 +0000136 const std::string &InputFile,
137 const std::string &OutputFile,
Reid Spencer51ab5c82006-06-06 00:00:42 +0000138 const std::vector<std::string> &GCCArgs,
Chris Lattnere96b2ed2004-07-24 07:49:11 +0000139 const std::vector<std::string> &SharedLibs,
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000140 unsigned Timeout,
141 unsigned MemoryLimit) {
Chris Lattner8c56be52004-02-18 20:21:57 +0000142 if (!SharedLibs.empty())
143 throw ToolExecutionError("LLI currently does not support "
144 "loading shared libraries.");
Misha Brukman9558c6a2003-09-29 22:39:25 +0000145
146 std::vector<const char*> LLIArgs;
147 LLIArgs.push_back(LLIPath.c_str());
Misha Brukman9558c6a2003-09-29 22:39:25 +0000148 LLIArgs.push_back("-force-interpreter=true");
Brian Gaeked11577b2004-05-04 21:09:01 +0000149
150 // Add any extra LLI args.
151 for (unsigned i = 0, e = ToolArgs.size(); i != e; ++i)
152 LLIArgs.push_back(ToolArgs[i].c_str());
153
Gabor Greif8ff70c22007-07-04 21:55:50 +0000154 LLIArgs.push_back(Bitcode.c_str());
Misha Brukman9558c6a2003-09-29 22:39:25 +0000155 // Add optional parameters to the running program from Argv
156 for (unsigned i=0, e = Args.size(); i != e; ++i)
157 LLIArgs.push_back(Args[i].c_str());
158 LLIArgs.push_back(0);
159
Dan Gohmana1bdced2009-07-15 17:29:42 +0000160 outs() << "<lli>"; outs().flush();
Dan Gohman65f57c22009-07-15 16:35:29 +0000161 DEBUG(errs() << "\nAbout to run:\t";
Chris Lattner7b2ccff2003-10-19 02:14:58 +0000162 for (unsigned i=0, e = LLIArgs.size()-1; i != e; ++i)
Dan Gohman65f57c22009-07-15 16:35:29 +0000163 errs() << " " << LLIArgs[i];
164 errs() << "\n";
Misha Brukman9558c6a2003-09-29 22:39:25 +0000165 );
Reid Spencerb31baa82004-12-19 18:00:21 +0000166 return RunProgramWithTimeout(sys::Path(LLIPath), &LLIArgs[0],
Misha Brukmanf976c852005-04-21 22:55:34 +0000167 sys::Path(InputFile), sys::Path(OutputFile), sys::Path(OutputFile),
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000168 Timeout, MemoryLimit);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000169}
170
171// LLI create method - Try to find the LLI executable
Chris Lattner7915a1e2003-10-14 21:34:11 +0000172AbstractInterpreter *AbstractInterpreter::createLLI(const std::string &ProgPath,
Brian Gaeked11577b2004-05-04 21:09:01 +0000173 std::string &Message,
174 const std::vector<std::string> *ToolArgs) {
Reid Spencer51ab8ec2004-12-13 23:43:44 +0000175 std::string LLIPath = FindExecutable("lli", ProgPath).toString();
Misha Brukman9558c6a2003-09-29 22:39:25 +0000176 if (!LLIPath.empty()) {
177 Message = "Found lli: " + LLIPath + "\n";
Brian Gaeked11577b2004-05-04 21:09:01 +0000178 return new LLI(LLIPath, ToolArgs);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000179 }
180
181 Message = "Cannot find `lli' in executable directory or PATH!\n";
182 return 0;
183}
184
Anton Korobeynikov9ef74252008-04-28 20:53:48 +0000185//===---------------------------------------------------------------------===//
186// Custom execution command implementation of AbstractIntepreter interface
187//
188// Allows using a custom command for executing the bitcode, thus allows,
189// for example, to invoke a cross compiler for code generation followed by
190// a simulator that executes the generated binary.
191namespace {
192 class CustomExecutor : public AbstractInterpreter {
193 std::string ExecutionCommand;
194 std::vector<std::string> ExecutorArgs;
195 public:
196 CustomExecutor(
197 const std::string &ExecutionCmd, std::vector<std::string> ExecArgs) :
198 ExecutionCommand(ExecutionCmd), ExecutorArgs(ExecArgs) {}
199
200 virtual int ExecuteProgram(const std::string &Bitcode,
201 const std::vector<std::string> &Args,
202 const std::string &InputFile,
203 const std::string &OutputFile,
204 const std::vector<std::string> &GCCArgs,
205 const std::vector<std::string> &SharedLibs =
206 std::vector<std::string>(),
207 unsigned Timeout = 0,
208 unsigned MemoryLimit = 0);
209 };
210}
211
212int CustomExecutor::ExecuteProgram(const std::string &Bitcode,
213 const std::vector<std::string> &Args,
214 const std::string &InputFile,
215 const std::string &OutputFile,
216 const std::vector<std::string> &GCCArgs,
217 const std::vector<std::string> &SharedLibs,
218 unsigned Timeout,
219 unsigned MemoryLimit) {
220
221 std::vector<const char*> ProgramArgs;
222 ProgramArgs.push_back(ExecutionCommand.c_str());
223
224 for (std::size_t i = 0; i < ExecutorArgs.size(); ++i)
225 ProgramArgs.push_back(ExecutorArgs.at(i).c_str());
226 ProgramArgs.push_back(Bitcode.c_str());
227 ProgramArgs.push_back(0);
228
229 // Add optional parameters to the running program from Argv
230 for (unsigned i=0, e = Args.size(); i != e; ++i)
231 ProgramArgs.push_back(Args[i].c_str());
232
233 return RunProgramWithTimeout(
234 sys::Path(ExecutionCommand),
235 &ProgramArgs[0], sys::Path(InputFile), sys::Path(OutputFile),
236 sys::Path(OutputFile), Timeout, MemoryLimit);
237}
238
239// Custom execution environment create method, takes the execution command
240// as arguments
241AbstractInterpreter *AbstractInterpreter::createCustom(
242 const std::string &ProgramPath,
243 std::string &Message,
244 const std::string &ExecCommandLine) {
245
246 std::string Command = "";
247 std::vector<std::string> Args;
248 std::string delimiters = " ";
249
250 // Tokenize the ExecCommandLine to the command and the args to allow
251 // defining a full command line as the command instead of just the
252 // executed program. We cannot just pass the whole string after the command
253 // as a single argument because then program sees only a single
254 // command line argument (with spaces in it: "foo bar" instead
255 // of "foo" and "bar").
256
257 // code borrowed from:
258 // http://oopweb.com/CPP/Documents/CPPHOWTO/Volume/C++Programming-HOWTO-7.html
259 std::string::size_type lastPos =
260 ExecCommandLine.find_first_not_of(delimiters, 0);
261 std::string::size_type pos =
262 ExecCommandLine.find_first_of(delimiters, lastPos);
263
264 while (std::string::npos != pos || std::string::npos != lastPos) {
265 std::string token = ExecCommandLine.substr(lastPos, pos - lastPos);
266 if (Command == "")
267 Command = token;
268 else
269 Args.push_back(token);
270 // Skip delimiters. Note the "not_of"
271 lastPos = ExecCommandLine.find_first_not_of(delimiters, pos);
272 // Find next "non-delimiter"
273 pos = ExecCommandLine.find_first_of(delimiters, lastPos);
274 }
275
276 std::string CmdPath = FindExecutable(Command, ProgramPath).toString();
277 if (CmdPath.empty()) {
278 Message =
279 std::string("Cannot find '") + Command +
280 "' in executable directory or PATH!\n";
281 return 0;
282 }
283
284 Message = "Found command in: " + CmdPath + "\n";
285
286 return new CustomExecutor(CmdPath, Args);
287}
288
Misha Brukman9558c6a2003-09-29 22:39:25 +0000289//===----------------------------------------------------------------------===//
290// LLC Implementation of AbstractIntepreter interface
291//
Gabor Greif8ff70c22007-07-04 21:55:50 +0000292GCC::FileType LLC::OutputCode(const std::string &Bitcode,
Chris Lattnerc600f3c2006-09-15 21:29:15 +0000293 sys::Path &OutputAsmFile) {
Gabor Greif8ff70c22007-07-04 21:55:50 +0000294 sys::Path uniqueFile(Bitcode+".llc.s");
Reid Spencer51c5a282006-08-23 20:34:57 +0000295 std::string ErrMsg;
296 if (uniqueFile.makeUnique(true, &ErrMsg)) {
Dan Gohman65f57c22009-07-15 16:35:29 +0000297 errs() << "Error making unique filename: " << ErrMsg << "\n";
Reid Spencer51c5a282006-08-23 20:34:57 +0000298 exit(1);
299 }
Reid Spencer9ac14182004-12-16 23:01:34 +0000300 OutputAsmFile = uniqueFile;
Brian Gaeked11577b2004-05-04 21:09:01 +0000301 std::vector<const char *> LLCArgs;
302 LLCArgs.push_back (LLCPath.c_str());
303
304 // Add any extra LLC args.
305 for (unsigned i = 0, e = ToolArgs.size(); i != e; ++i)
306 LLCArgs.push_back(ToolArgs[i].c_str());
307
308 LLCArgs.push_back ("-o");
309 LLCArgs.push_back (OutputAsmFile.c_str()); // Output to the Asm file
310 LLCArgs.push_back ("-f"); // Overwrite as necessary...
Gabor Greif8ff70c22007-07-04 21:55:50 +0000311 LLCArgs.push_back (Bitcode.c_str()); // This is the input bitcode
Brian Gaeked11577b2004-05-04 21:09:01 +0000312 LLCArgs.push_back (0);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000313
Dan Gohmana1bdced2009-07-15 17:29:42 +0000314 outs() << "<llc>"; outs().flush();
Dan Gohman65f57c22009-07-15 16:35:29 +0000315 DEBUG(errs() << "\nAbout to run:\t";
Brian Gaeked11577b2004-05-04 21:09:01 +0000316 for (unsigned i=0, e = LLCArgs.size()-1; i != e; ++i)
Dan Gohman65f57c22009-07-15 16:35:29 +0000317 errs() << " " << LLCArgs[i];
318 errs() << "\n";
Brian Gaeked11577b2004-05-04 21:09:01 +0000319 );
Misha Brukmanf976c852005-04-21 22:55:34 +0000320 if (RunProgramWithTimeout(sys::Path(LLCPath), &LLCArgs[0],
Reid Spencerb31baa82004-12-19 18:00:21 +0000321 sys::Path(), sys::Path(), sys::Path()))
322 ProcessFailure(sys::Path(LLCPath), &LLCArgs[0]);
Chris Lattnerc600f3c2006-09-15 21:29:15 +0000323
324 return GCC::AsmFile;
Misha Brukman9558c6a2003-09-29 22:39:25 +0000325}
326
Gabor Greif8ff70c22007-07-04 21:55:50 +0000327void LLC::compileProgram(const std::string &Bitcode) {
Reid Spencer9ac14182004-12-16 23:01:34 +0000328 sys::Path OutputAsmFile;
Gabor Greif8ff70c22007-07-04 21:55:50 +0000329 OutputCode(Bitcode, OutputAsmFile);
Reid Spencera229c5c2005-07-08 03:08:58 +0000330 OutputAsmFile.eraseFromDisk();
Chris Lattner9cbbee32004-02-18 23:24:41 +0000331}
332
Gabor Greif8ff70c22007-07-04 21:55:50 +0000333int LLC::ExecuteProgram(const std::string &Bitcode,
Chris Lattner7915a1e2003-10-14 21:34:11 +0000334 const std::vector<std::string> &Args,
Misha Brukman9558c6a2003-09-29 22:39:25 +0000335 const std::string &InputFile,
336 const std::string &OutputFile,
Reid Spencer51ab5c82006-06-06 00:00:42 +0000337 const std::vector<std::string> &ArgsForGCC,
Chris Lattnere96b2ed2004-07-24 07:49:11 +0000338 const std::vector<std::string> &SharedLibs,
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000339 unsigned Timeout,
340 unsigned MemoryLimit) {
Chris Lattnereeed9832003-10-14 21:52:52 +0000341
Reid Spencer9ac14182004-12-16 23:01:34 +0000342 sys::Path OutputAsmFile;
Gabor Greif8ff70c22007-07-04 21:55:50 +0000343 OutputCode(Bitcode, OutputAsmFile);
Chris Lattner8c56be52004-02-18 20:21:57 +0000344 FileRemover OutFileRemover(OutputAsmFile);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000345
Reid Spencer51ab5c82006-06-06 00:00:42 +0000346 std::vector<std::string> GCCArgs(ArgsForGCC);
Bill Wendling38efa382009-03-02 23:13:18 +0000347 GCCArgs.insert(GCCArgs.end(), SharedLibs.begin(), SharedLibs.end());
348 GCCArgs.insert(GCCArgs.end(), gccArgs.begin(), gccArgs.end());
Reid Spencer51ab5c82006-06-06 00:00:42 +0000349
Misha Brukman9558c6a2003-09-29 22:39:25 +0000350 // Assuming LLC worked, compile the result with GCC and run it.
Reid Spencer9ac14182004-12-16 23:01:34 +0000351 return gcc->ExecuteProgram(OutputAsmFile.toString(), Args, GCC::AsmFile,
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000352 InputFile, OutputFile, GCCArgs,
353 Timeout, MemoryLimit);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000354}
355
Chris Lattner7915a1e2003-10-14 21:34:11 +0000356/// createLLC - Try to find the LLC executable
Misha Brukman9558c6a2003-09-29 22:39:25 +0000357///
Chris Lattner7915a1e2003-10-14 21:34:11 +0000358LLC *AbstractInterpreter::createLLC(const std::string &ProgramPath,
Brian Gaeked11577b2004-05-04 21:09:01 +0000359 std::string &Message,
Bill Wendling38efa382009-03-02 23:13:18 +0000360 const std::vector<std::string> *Args,
361 const std::vector<std::string> *GCCArgs) {
Reid Spencer51ab8ec2004-12-13 23:43:44 +0000362 std::string LLCPath = FindExecutable("llc", ProgramPath).toString();
Misha Brukman9558c6a2003-09-29 22:39:25 +0000363 if (LLCPath.empty()) {
364 Message = "Cannot find `llc' in executable directory or PATH!\n";
365 return 0;
366 }
367
368 Message = "Found llc: " + LLCPath + "\n";
Bill Wendling38efa382009-03-02 23:13:18 +0000369 GCC *gcc = GCC::create(ProgramPath, Message, GCCArgs);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000370 if (!gcc) {
Dan Gohman65f57c22009-07-15 16:35:29 +0000371 errs() << Message << "\n";
Misha Brukman9558c6a2003-09-29 22:39:25 +0000372 exit(1);
373 }
Bill Wendling38efa382009-03-02 23:13:18 +0000374 return new LLC(LLCPath, gcc, Args, GCCArgs);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000375}
376
377//===---------------------------------------------------------------------===//
378// JIT Implementation of AbstractIntepreter interface
379//
Chris Lattner2cdd21c2003-12-14 21:35:53 +0000380namespace {
381 class JIT : public AbstractInterpreter {
382 std::string LLIPath; // The path to the LLI executable
Brian Gaeked11577b2004-05-04 21:09:01 +0000383 std::vector<std::string> ToolArgs; // Args to pass to LLI
Chris Lattner2cdd21c2003-12-14 21:35:53 +0000384 public:
Brian Gaeked11577b2004-05-04 21:09:01 +0000385 JIT(const std::string &Path, const std::vector<std::string> *Args)
386 : LLIPath(Path) {
387 ToolArgs.clear ();
Brian Gaeke48b008d2004-05-04 22:02:41 +0000388 if (Args) { ToolArgs = *Args; }
Brian Gaeked11577b2004-05-04 21:09:01 +0000389 }
Misha Brukmanf976c852005-04-21 22:55:34 +0000390
Gabor Greif8ff70c22007-07-04 21:55:50 +0000391 virtual int ExecuteProgram(const std::string &Bitcode,
Chris Lattner2cdd21c2003-12-14 21:35:53 +0000392 const std::vector<std::string> &Args,
393 const std::string &InputFile,
394 const std::string &OutputFile,
Reid Spencer51ab5c82006-06-06 00:00:42 +0000395 const std::vector<std::string> &GCCArgs =
396 std::vector<std::string>(),
Misha Brukmanf976c852005-04-21 22:55:34 +0000397 const std::vector<std::string> &SharedLibs =
Reid Spencer51ab5c82006-06-06 00:00:42 +0000398 std::vector<std::string>(),
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000399 unsigned Timeout =0,
400 unsigned MemoryLimit =0);
Chris Lattner2cdd21c2003-12-14 21:35:53 +0000401 };
402}
Misha Brukman9558c6a2003-09-29 22:39:25 +0000403
Gabor Greif8ff70c22007-07-04 21:55:50 +0000404int JIT::ExecuteProgram(const std::string &Bitcode,
Chris Lattner7915a1e2003-10-14 21:34:11 +0000405 const std::vector<std::string> &Args,
Misha Brukman9558c6a2003-09-29 22:39:25 +0000406 const std::string &InputFile,
407 const std::string &OutputFile,
Reid Spencer51ab5c82006-06-06 00:00:42 +0000408 const std::vector<std::string> &GCCArgs,
Chris Lattnere96b2ed2004-07-24 07:49:11 +0000409 const std::vector<std::string> &SharedLibs,
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000410 unsigned Timeout,
411 unsigned MemoryLimit) {
Misha Brukman9558c6a2003-09-29 22:39:25 +0000412 // Construct a vector of parameters, incorporating those from the command-line
413 std::vector<const char*> JITArgs;
414 JITArgs.push_back(LLIPath.c_str());
Misha Brukman9558c6a2003-09-29 22:39:25 +0000415 JITArgs.push_back("-force-interpreter=false");
Chris Lattnereeed9832003-10-14 21:52:52 +0000416
Brian Gaeked11577b2004-05-04 21:09:01 +0000417 // Add any extra LLI args.
418 for (unsigned i = 0, e = ToolArgs.size(); i != e; ++i)
419 JITArgs.push_back(ToolArgs[i].c_str());
420
Chris Lattnereeed9832003-10-14 21:52:52 +0000421 for (unsigned i = 0, e = SharedLibs.size(); i != e; ++i) {
Misha Brukman9558c6a2003-09-29 22:39:25 +0000422 JITArgs.push_back("-load");
Chris Lattnereeed9832003-10-14 21:52:52 +0000423 JITArgs.push_back(SharedLibs[i].c_str());
Misha Brukman9558c6a2003-09-29 22:39:25 +0000424 }
Gabor Greif8ff70c22007-07-04 21:55:50 +0000425 JITArgs.push_back(Bitcode.c_str());
Misha Brukman9558c6a2003-09-29 22:39:25 +0000426 // Add optional parameters to the running program from Argv
427 for (unsigned i=0, e = Args.size(); i != e; ++i)
428 JITArgs.push_back(Args[i].c_str());
429 JITArgs.push_back(0);
430
Dan Gohmana1bdced2009-07-15 17:29:42 +0000431 outs() << "<jit>"; outs().flush();
Dan Gohman65f57c22009-07-15 16:35:29 +0000432 DEBUG(errs() << "\nAbout to run:\t";
Chris Lattner7b2ccff2003-10-19 02:14:58 +0000433 for (unsigned i=0, e = JITArgs.size()-1; i != e; ++i)
Dan Gohman65f57c22009-07-15 16:35:29 +0000434 errs() << " " << JITArgs[i];
435 errs() << "\n";
Misha Brukman9558c6a2003-09-29 22:39:25 +0000436 );
Dan Gohman65f57c22009-07-15 16:35:29 +0000437 DEBUG(errs() << "\nSending output to " << OutputFile << "\n");
Reid Spencerb31baa82004-12-19 18:00:21 +0000438 return RunProgramWithTimeout(sys::Path(LLIPath), &JITArgs[0],
Misha Brukmanf976c852005-04-21 22:55:34 +0000439 sys::Path(InputFile), sys::Path(OutputFile), sys::Path(OutputFile),
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000440 Timeout, MemoryLimit);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000441}
442
Chris Lattner7915a1e2003-10-14 21:34:11 +0000443/// createJIT - Try to find the LLI executable
Misha Brukman9558c6a2003-09-29 22:39:25 +0000444///
Chris Lattner7915a1e2003-10-14 21:34:11 +0000445AbstractInterpreter *AbstractInterpreter::createJIT(const std::string &ProgPath,
Brian Gaeked11577b2004-05-04 21:09:01 +0000446 std::string &Message, const std::vector<std::string> *Args) {
Reid Spencer51ab8ec2004-12-13 23:43:44 +0000447 std::string LLIPath = FindExecutable("lli", ProgPath).toString();
Misha Brukman9558c6a2003-09-29 22:39:25 +0000448 if (!LLIPath.empty()) {
449 Message = "Found lli: " + LLIPath + "\n";
Brian Gaeked11577b2004-05-04 21:09:01 +0000450 return new JIT(LLIPath, Args);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000451 }
452
453 Message = "Cannot find `lli' in executable directory or PATH!\n";
454 return 0;
455}
456
Gabor Greif8ff70c22007-07-04 21:55:50 +0000457GCC::FileType CBE::OutputCode(const std::string &Bitcode,
Chris Lattnerc600f3c2006-09-15 21:29:15 +0000458 sys::Path &OutputCFile) {
Gabor Greif8ff70c22007-07-04 21:55:50 +0000459 sys::Path uniqueFile(Bitcode+".cbe.c");
Reid Spencer51c5a282006-08-23 20:34:57 +0000460 std::string ErrMsg;
461 if (uniqueFile.makeUnique(true, &ErrMsg)) {
Dan Gohman65f57c22009-07-15 16:35:29 +0000462 errs() << "Error making unique filename: " << ErrMsg << "\n";
Reid Spencer51c5a282006-08-23 20:34:57 +0000463 exit(1);
464 }
Reid Spencer9ac14182004-12-16 23:01:34 +0000465 OutputCFile = uniqueFile;
Brian Gaeked11577b2004-05-04 21:09:01 +0000466 std::vector<const char *> LLCArgs;
467 LLCArgs.push_back (LLCPath.c_str());
468
469 // Add any extra LLC args.
470 for (unsigned i = 0, e = ToolArgs.size(); i != e; ++i)
471 LLCArgs.push_back(ToolArgs[i].c_str());
472
473 LLCArgs.push_back ("-o");
474 LLCArgs.push_back (OutputCFile.c_str()); // Output to the C file
475 LLCArgs.push_back ("-march=c"); // Output C language
476 LLCArgs.push_back ("-f"); // Overwrite as necessary...
Gabor Greif8ff70c22007-07-04 21:55:50 +0000477 LLCArgs.push_back (Bitcode.c_str()); // This is the input bitcode
Brian Gaeked11577b2004-05-04 21:09:01 +0000478 LLCArgs.push_back (0);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000479
Dan Gohmana1bdced2009-07-15 17:29:42 +0000480 outs() << "<cbe>"; outs().flush();
Dan Gohman65f57c22009-07-15 16:35:29 +0000481 DEBUG(errs() << "\nAbout to run:\t";
Brian Gaeked11577b2004-05-04 21:09:01 +0000482 for (unsigned i=0, e = LLCArgs.size()-1; i != e; ++i)
Dan Gohman65f57c22009-07-15 16:35:29 +0000483 errs() << " " << LLCArgs[i];
484 errs() << "\n";
Brian Gaeked11577b2004-05-04 21:09:01 +0000485 );
Misha Brukmanf976c852005-04-21 22:55:34 +0000486 if (RunProgramWithTimeout(LLCPath, &LLCArgs[0], sys::Path(), sys::Path(),
Reid Spencerb31baa82004-12-19 18:00:21 +0000487 sys::Path()))
Brian Gaeked11577b2004-05-04 21:09:01 +0000488 ProcessFailure(LLCPath, &LLCArgs[0]);
Chris Lattnerc600f3c2006-09-15 21:29:15 +0000489 return GCC::CFile;
Misha Brukman9558c6a2003-09-29 22:39:25 +0000490}
491
Gabor Greif8ff70c22007-07-04 21:55:50 +0000492void CBE::compileProgram(const std::string &Bitcode) {
Reid Spencer9ac14182004-12-16 23:01:34 +0000493 sys::Path OutputCFile;
Gabor Greif8ff70c22007-07-04 21:55:50 +0000494 OutputCode(Bitcode, OutputCFile);
Reid Spencera229c5c2005-07-08 03:08:58 +0000495 OutputCFile.eraseFromDisk();
Chris Lattner9cbbee32004-02-18 23:24:41 +0000496}
497
Gabor Greif8ff70c22007-07-04 21:55:50 +0000498int CBE::ExecuteProgram(const std::string &Bitcode,
Chris Lattner7915a1e2003-10-14 21:34:11 +0000499 const std::vector<std::string> &Args,
Misha Brukman9558c6a2003-09-29 22:39:25 +0000500 const std::string &InputFile,
501 const std::string &OutputFile,
Reid Spencer51ab5c82006-06-06 00:00:42 +0000502 const std::vector<std::string> &ArgsForGCC,
Chris Lattnere96b2ed2004-07-24 07:49:11 +0000503 const std::vector<std::string> &SharedLibs,
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000504 unsigned Timeout,
505 unsigned MemoryLimit) {
Reid Spencer9ac14182004-12-16 23:01:34 +0000506 sys::Path OutputCFile;
Gabor Greif8ff70c22007-07-04 21:55:50 +0000507 OutputCode(Bitcode, OutputCFile);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000508
Chris Lattner8c56be52004-02-18 20:21:57 +0000509 FileRemover CFileRemove(OutputCFile);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000510
Reid Spencer51ab5c82006-06-06 00:00:42 +0000511 std::vector<std::string> GCCArgs(ArgsForGCC);
Bill Wendling38efa382009-03-02 23:13:18 +0000512 GCCArgs.insert(GCCArgs.end(), SharedLibs.begin(), SharedLibs.end());
513
Misha Brukmanf976c852005-04-21 22:55:34 +0000514 return gcc->ExecuteProgram(OutputCFile.toString(), Args, GCC::CFile,
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000515 InputFile, OutputFile, GCCArgs,
516 Timeout, MemoryLimit);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000517}
518
Chris Lattner9915cd92004-02-17 06:40:06 +0000519/// createCBE - Try to find the 'llc' executable
Misha Brukman9558c6a2003-09-29 22:39:25 +0000520///
Chris Lattner7915a1e2003-10-14 21:34:11 +0000521CBE *AbstractInterpreter::createCBE(const std::string &ProgramPath,
Brian Gaeked11577b2004-05-04 21:09:01 +0000522 std::string &Message,
Bill Wendling38efa382009-03-02 23:13:18 +0000523 const std::vector<std::string> *Args,
524 const std::vector<std::string> *GCCArgs) {
Reid Spencerb31baa82004-12-19 18:00:21 +0000525 sys::Path LLCPath = FindExecutable("llc", ProgramPath);
526 if (LLCPath.isEmpty()) {
Misha Brukmanf976c852005-04-21 22:55:34 +0000527 Message =
Chris Lattner9915cd92004-02-17 06:40:06 +0000528 "Cannot find `llc' in executable directory or PATH!\n";
Misha Brukman9558c6a2003-09-29 22:39:25 +0000529 return 0;
530 }
531
Reid Spencerb31baa82004-12-19 18:00:21 +0000532 Message = "Found llc: " + LLCPath.toString() + "\n";
Bill Wendling38efa382009-03-02 23:13:18 +0000533 GCC *gcc = GCC::create(ProgramPath, Message, GCCArgs);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000534 if (!gcc) {
Dan Gohman65f57c22009-07-15 16:35:29 +0000535 errs() << Message << "\n";
Misha Brukman9558c6a2003-09-29 22:39:25 +0000536 exit(1);
537 }
Brian Gaeked11577b2004-05-04 21:09:01 +0000538 return new CBE(LLCPath, gcc, Args);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000539}
540
541//===---------------------------------------------------------------------===//
542// GCC abstraction
543//
David Goodwin80becf12009-07-10 21:39:28 +0000544
Daniel Dunbar84f9abe2009-07-11 21:24:52 +0000545#ifdef __APPLE__
David Goodwin80becf12009-07-10 21:39:28 +0000546static bool
547IsARMArchitecture(std::vector<std::string> Args)
548{
549 for (std::vector<std::string>::const_iterator
550 I = Args.begin(), E = Args.end(); I != E; ++I) {
551 if (!strcasecmp(I->c_str(), "-arch")) {
552 ++I;
553 if ((I != E) && !strncasecmp(I->c_str(), "arm", strlen("arm"))) {
554 return true;
555 }
556 }
557 }
558
559 return false;
560}
Daniel Dunbar84f9abe2009-07-11 21:24:52 +0000561#endif
David Goodwin80becf12009-07-10 21:39:28 +0000562
Misha Brukman9558c6a2003-09-29 22:39:25 +0000563int GCC::ExecuteProgram(const std::string &ProgramFile,
Chris Lattner7915a1e2003-10-14 21:34:11 +0000564 const std::vector<std::string> &Args,
Misha Brukman9558c6a2003-09-29 22:39:25 +0000565 FileType fileType,
566 const std::string &InputFile,
567 const std::string &OutputFile,
Reid Spencer51ab5c82006-06-06 00:00:42 +0000568 const std::vector<std::string> &ArgsForGCC,
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000569 unsigned Timeout,
570 unsigned MemoryLimit) {
Misha Brukman9558c6a2003-09-29 22:39:25 +0000571 std::vector<const char*> GCCArgs;
572
573 GCCArgs.push_back(GCCPath.c_str());
Chris Lattnereeed9832003-10-14 21:52:52 +0000574
Bill Wendling38efa382009-03-02 23:13:18 +0000575 for (std::vector<std::string>::const_iterator
576 I = gccArgs.begin(), E = gccArgs.end(); I != E; ++I)
577 GCCArgs.push_back(I->c_str());
578
Chris Lattnereeed9832003-10-14 21:52:52 +0000579 // Specify -x explicitly in case the extension is wonky
Misha Brukman9558c6a2003-09-29 22:39:25 +0000580 GCCArgs.push_back("-x");
581 if (fileType == CFile) {
582 GCCArgs.push_back("c");
583 GCCArgs.push_back("-fno-strict-aliasing");
584 } else {
585 GCCArgs.push_back("assembler");
Chris Lattnerd00b2882005-08-29 13:14:24 +0000586#ifdef __APPLE__
David Goodwin80becf12009-07-10 21:39:28 +0000587 // For ARM architectures we don't want this flag. bugpoint isn't
588 // explicitly told what architecture it is working on, so we get
589 // it from gcc flags
590 if (!IsARMArchitecture(ArgsForGCC))
591 GCCArgs.push_back("-force_cpusubtype_ALL");
Chris Lattnerd00b2882005-08-29 13:14:24 +0000592#endif
Misha Brukman9558c6a2003-09-29 22:39:25 +0000593 }
594 GCCArgs.push_back(ProgramFile.c_str()); // Specify the input filename...
Chris Lattner629e4872006-06-09 21:31:53 +0000595 GCCArgs.push_back("-x");
596 GCCArgs.push_back("none");
Misha Brukman9558c6a2003-09-29 22:39:25 +0000597 GCCArgs.push_back("-o");
Reid Spencercda985e2004-12-15 01:51:56 +0000598 sys::Path OutputBinary (ProgramFile+".gcc.exe");
Reid Spencer51c5a282006-08-23 20:34:57 +0000599 std::string ErrMsg;
600 if (OutputBinary.makeUnique(true, &ErrMsg)) {
Dan Gohman65f57c22009-07-15 16:35:29 +0000601 errs() << "Error making unique filename: " << ErrMsg << "\n";
Reid Spencer51c5a282006-08-23 20:34:57 +0000602 exit(1);
603 }
Misha Brukman9558c6a2003-09-29 22:39:25 +0000604 GCCArgs.push_back(OutputBinary.c_str()); // Output to the right file...
Reid Spencer51ab5c82006-06-06 00:00:42 +0000605
606 // Add any arguments intended for GCC. We locate them here because this is
607 // most likely -L and -l options that need to come before other libraries but
608 // after the source. Other options won't be sensitive to placement on the
609 // command line, so this should be safe.
610 for (unsigned i = 0, e = ArgsForGCC.size(); i != e; ++i)
611 GCCArgs.push_back(ArgsForGCC[i].c_str());
612
Misha Brukman9558c6a2003-09-29 22:39:25 +0000613 GCCArgs.push_back("-lm"); // Hard-code the math library...
614 GCCArgs.push_back("-O2"); // Optimize the program a bit...
Brian Gaekec8db76c2003-11-18 06:31:17 +0000615#if defined (HAVE_LINK_R)
Chris Lattner1f0f1622003-10-18 21:54:47 +0000616 GCCArgs.push_back("-Wl,-R."); // Search this dir for .so files
Brian Gaekec8db76c2003-11-18 06:31:17 +0000617#endif
Chris Lattnerfdcc71e2006-02-04 05:02:27 +0000618#ifdef __sparc__
619 GCCArgs.push_back("-mcpu=v9");
620#endif
Misha Brukman9558c6a2003-09-29 22:39:25 +0000621 GCCArgs.push_back(0); // NULL terminator
622
Dan Gohmana1bdced2009-07-15 17:29:42 +0000623 outs() << "<gcc>"; outs().flush();
Dan Gohman65f57c22009-07-15 16:35:29 +0000624 DEBUG(errs() << "\nAbout to run:\t";
Evan Cheng7f7fdcc2007-01-03 07:44:30 +0000625 for (unsigned i=0, e = GCCArgs.size()-1; i != e; ++i)
Dan Gohman65f57c22009-07-15 16:35:29 +0000626 errs() << " " << GCCArgs[i];
627 errs() << "\n";
Evan Cheng7f7fdcc2007-01-03 07:44:30 +0000628 );
Reid Spencerb31baa82004-12-19 18:00:21 +0000629 if (RunProgramWithTimeout(GCCPath, &GCCArgs[0], sys::Path(), sys::Path(),
630 sys::Path())) {
Chris Lattner89bf9ea2004-02-18 20:38:00 +0000631 ProcessFailure(GCCPath, &GCCArgs[0]);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000632 exit(1);
633 }
634
635 std::vector<const char*> ProgramArgs;
Chris Lattner45495c52005-02-13 23:13:47 +0000636
Viktor Kutuzov699220b2009-07-14 19:10:55 +0000637 std::string Exec;
Evan Cheng70f684f2008-09-09 06:11:26 +0000638 if (RemoteClientPath.isEmpty())
Evan Cheng34e400e2007-05-03 18:36:15 +0000639 ProgramArgs.push_back(OutputBinary.c_str());
640 else {
Evan Cheng70f684f2008-09-09 06:11:26 +0000641 ProgramArgs.push_back(RemoteClientPath.c_str());
642 ProgramArgs.push_back(RemoteHost.c_str());
Evan Cheng34e400e2007-05-03 18:36:15 +0000643 ProgramArgs.push_back("-l");
Evan Cheng70f684f2008-09-09 06:11:26 +0000644 ProgramArgs.push_back(RemoteUser.c_str());
David Goodwin80becf12009-07-10 21:39:28 +0000645 if (!RemotePort.empty()) {
646 ProgramArgs.push_back("-p");
647 ProgramArgs.push_back(RemotePort.c_str());
648 }
Evan Cheng70f684f2008-09-09 06:11:26 +0000649 if (!RemoteExtra.empty()) {
650 ProgramArgs.push_back(RemoteExtra.c_str());
651 }
Evan Cheng34e400e2007-05-03 18:36:15 +0000652
Viktor Kutuzov699220b2009-07-14 19:10:55 +0000653 // Full path to the binary
Evan Cheng34e400e2007-05-03 18:36:15 +0000654 char* env_pwd = getenv("PWD");
Evan Cheng34e400e2007-05-03 18:36:15 +0000655 Exec += env_pwd;
Viktor Kutuzov699220b2009-07-14 19:10:55 +0000656 Exec += "/";
Evan Cheng34e400e2007-05-03 18:36:15 +0000657 Exec += OutputBinary.c_str();
658 ProgramArgs.push_back(Exec.c_str());
659 }
660
Misha Brukman9558c6a2003-09-29 22:39:25 +0000661 // Add optional parameters to the running program from Argv
662 for (unsigned i=0, e = Args.size(); i != e; ++i)
663 ProgramArgs.push_back(Args[i].c_str());
664 ProgramArgs.push_back(0); // NULL terminator
665
666 // Now that we have a binary, run it!
Dan Gohmana1bdced2009-07-15 17:29:42 +0000667 outs() << "<program>"; outs().flush();
Dan Gohman65f57c22009-07-15 16:35:29 +0000668 DEBUG(errs() << "\nAbout to run:\t";
Chris Lattner7b2ccff2003-10-19 02:14:58 +0000669 for (unsigned i=0, e = ProgramArgs.size()-1; i != e; ++i)
Dan Gohman65f57c22009-07-15 16:35:29 +0000670 errs() << " " << ProgramArgs[i];
671 errs() << "\n";
Misha Brukman9558c6a2003-09-29 22:39:25 +0000672 );
Chris Lattner8c56be52004-02-18 20:21:57 +0000673
Reid Spencer9ac14182004-12-16 23:01:34 +0000674 FileRemover OutputBinaryRemover(OutputBinary);
Evan Cheng34e400e2007-05-03 18:36:15 +0000675
Viktor Kutuzov699220b2009-07-14 19:10:55 +0000676 if (RemoteClientPath.isEmpty()) {
Dan Gohman65f57c22009-07-15 16:35:29 +0000677 DEBUG(errs() << "<run locally>";);
Evan Cheng34e400e2007-05-03 18:36:15 +0000678 return RunProgramWithTimeout(OutputBinary, &ProgramArgs[0],
679 sys::Path(InputFile), sys::Path(OutputFile), sys::Path(OutputFile),
680 Timeout, MemoryLimit);
Viktor Kutuzov699220b2009-07-14 19:10:55 +0000681 } else {
Dan Gohmana1bdced2009-07-15 17:29:42 +0000682 outs() << "<run remotely>"; outs().flush();
Viktor Kutuzov699220b2009-07-14 19:10:55 +0000683 int RemoteClientStatus = RunProgramWithTimeout(sys::Path(RemoteClientPath),
684 &ProgramArgs[0], sys::Path(InputFile), sys::Path(OutputFile),
685 sys::Path(OutputFile), Timeout, MemoryLimit);
686 if (RemoteClientStatus != 0) {
Dan Gohman65f57c22009-07-15 16:35:29 +0000687 errs() << "Remote Client failed with an error: " <<
688 RemoteClientStatus << ".\n";
Viktor Kutuzov699220b2009-07-14 19:10:55 +0000689 }
690 }
Evan Chengf975e012009-07-14 23:55:32 +0000691
692 return 0;
Misha Brukman9558c6a2003-09-29 22:39:25 +0000693}
694
Chris Lattner1798e4a2003-10-14 21:07:25 +0000695int GCC::MakeSharedObject(const std::string &InputFile, FileType fileType,
Chris Lattner130e2a32006-06-27 20:35:36 +0000696 std::string &OutputFile,
697 const std::vector<std::string> &ArgsForGCC) {
Reid Spencercda985e2004-12-15 01:51:56 +0000698 sys::Path uniqueFilename(InputFile+LTDL_SHLIB_EXT);
Reid Spencer51c5a282006-08-23 20:34:57 +0000699 std::string ErrMsg;
700 if (uniqueFilename.makeUnique(true, &ErrMsg)) {
Dan Gohman65f57c22009-07-15 16:35:29 +0000701 errs() << "Error making unique filename: " << ErrMsg << "\n";
Reid Spencer51c5a282006-08-23 20:34:57 +0000702 exit(1);
703 }
Reid Spencercda985e2004-12-15 01:51:56 +0000704 OutputFile = uniqueFilename.toString();
705
Chris Lattner130e2a32006-06-27 20:35:36 +0000706 std::vector<const char*> GCCArgs;
707
708 GCCArgs.push_back(GCCPath.c_str());
Evan Chengfd829952009-03-12 00:53:34 +0000709
710 for (std::vector<std::string>::const_iterator
711 I = gccArgs.begin(), E = gccArgs.end(); I != E; ++I)
712 GCCArgs.push_back(I->c_str());
713
Misha Brukman9558c6a2003-09-29 22:39:25 +0000714 // Compile the C/asm file into a shared object
Chris Lattner130e2a32006-06-27 20:35:36 +0000715 GCCArgs.push_back("-x");
716 GCCArgs.push_back(fileType == AsmFile ? "assembler" : "c");
717 GCCArgs.push_back("-fno-strict-aliasing");
718 GCCArgs.push_back(InputFile.c_str()); // Specify the input filename.
Lauro Ramos Venancio497391a2007-06-06 23:10:56 +0000719 GCCArgs.push_back("-x");
720 GCCArgs.push_back("none");
Misha Brukman9558c6a2003-09-29 22:39:25 +0000721#if defined(sparc) || defined(__sparc__) || defined(__sparcv9)
Chris Lattner130e2a32006-06-27 20:35:36 +0000722 GCCArgs.push_back("-G"); // Compile a shared library, `-G' for Sparc
Nate Begeman72b286b2005-07-08 00:23:26 +0000723#elif defined(__APPLE__)
Chris Lattner130e2a32006-06-27 20:35:36 +0000724 // link all source files into a single module in data segment, rather than
725 // generating blocks. dynamic_lookup requires that you set
726 // MACOSX_DEPLOYMENT_TARGET=10.3 in your env. FIXME: it would be better for
727 // bugpoint to just pass that in the environment of GCC.
728 GCCArgs.push_back("-single_module");
729 GCCArgs.push_back("-dynamiclib"); // `-dynamiclib' for MacOS X/PowerPC
730 GCCArgs.push_back("-undefined");
731 GCCArgs.push_back("dynamic_lookup");
Misha Brukmanb3998ec2004-07-16 19:45:45 +0000732#else
Chris Lattner130e2a32006-06-27 20:35:36 +0000733 GCCArgs.push_back("-shared"); // `-shared' for Linux/X86, maybe others
Misha Brukman9558c6a2003-09-29 22:39:25 +0000734#endif
Chris Lattner1c81f132005-03-09 03:31:02 +0000735
Torok Edwinaf4fc282008-04-06 12:42:29 +0000736#if defined(__ia64__) || defined(__alpha__) || defined(__amd64__)
Chris Lattner130e2a32006-06-27 20:35:36 +0000737 GCCArgs.push_back("-fPIC"); // Requires shared objs to contain PIC
Chris Lattner1c81f132005-03-09 03:31:02 +0000738#endif
Chris Lattnerfdcc71e2006-02-04 05:02:27 +0000739#ifdef __sparc__
Chris Lattner130e2a32006-06-27 20:35:36 +0000740 GCCArgs.push_back("-mcpu=v9");
Chris Lattnerfdcc71e2006-02-04 05:02:27 +0000741#endif
Chris Lattner130e2a32006-06-27 20:35:36 +0000742 GCCArgs.push_back("-o");
743 GCCArgs.push_back(OutputFile.c_str()); // Output to the right filename.
744 GCCArgs.push_back("-O2"); // Optimize the program a bit.
745
746
747
748 // Add any arguments intended for GCC. We locate them here because this is
749 // most likely -L and -l options that need to come before other libraries but
750 // after the source. Other options won't be sensitive to placement on the
751 // command line, so this should be safe.
752 for (unsigned i = 0, e = ArgsForGCC.size(); i != e; ++i)
753 GCCArgs.push_back(ArgsForGCC[i].c_str());
754 GCCArgs.push_back(0); // NULL terminator
755
756
Misha Brukmanf976c852005-04-21 22:55:34 +0000757
Dan Gohmana1bdced2009-07-15 17:29:42 +0000758 outs() << "<gcc>"; outs().flush();
Dan Gohman65f57c22009-07-15 16:35:29 +0000759 DEBUG(errs() << "\nAbout to run:\t";
Evan Cheng7f7fdcc2007-01-03 07:44:30 +0000760 for (unsigned i=0, e = GCCArgs.size()-1; i != e; ++i)
Dan Gohman65f57c22009-07-15 16:35:29 +0000761 errs() << " " << GCCArgs[i];
762 errs() << "\n";
Evan Cheng7f7fdcc2007-01-03 07:44:30 +0000763 );
Chris Lattner130e2a32006-06-27 20:35:36 +0000764 if (RunProgramWithTimeout(GCCPath, &GCCArgs[0], sys::Path(), sys::Path(),
Reid Spencerb31baa82004-12-19 18:00:21 +0000765 sys::Path())) {
Chris Lattner130e2a32006-06-27 20:35:36 +0000766 ProcessFailure(GCCPath, &GCCArgs[0]);
Chris Lattner1798e4a2003-10-14 21:07:25 +0000767 return 1;
Misha Brukman9558c6a2003-09-29 22:39:25 +0000768 }
769 return 0;
770}
771
Chris Lattner7915a1e2003-10-14 21:34:11 +0000772/// create - Try to find the `gcc' executable
Misha Brukman9558c6a2003-09-29 22:39:25 +0000773///
Bill Wendling38efa382009-03-02 23:13:18 +0000774GCC *GCC::create(const std::string &ProgramPath, std::string &Message,
775 const std::vector<std::string> *Args) {
Reid Spencerb31baa82004-12-19 18:00:21 +0000776 sys::Path GCCPath = FindExecutable("gcc", ProgramPath);
777 if (GCCPath.isEmpty()) {
Misha Brukman9558c6a2003-09-29 22:39:25 +0000778 Message = "Cannot find `gcc' in executable directory or PATH!\n";
779 return 0;
780 }
781
Evan Cheng70f684f2008-09-09 06:11:26 +0000782 sys::Path RemoteClientPath;
783 if (!RemoteClient.empty())
784 RemoteClientPath = FindExecutable(RemoteClient.c_str(), ProgramPath);
Evan Cheng34e400e2007-05-03 18:36:15 +0000785
Reid Spencerb31baa82004-12-19 18:00:21 +0000786 Message = "Found gcc: " + GCCPath.toString() + "\n";
Bill Wendling38efa382009-03-02 23:13:18 +0000787 return new GCC(GCCPath, RemoteClientPath, Args);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000788}