blob: d610676d4dedecbd89fa94fca2d6a0e28c5ebbdb [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 Lattner86a54842006-01-22 22:53:01 +000023#include <iostream>
Chris Lattner2cdd21c2003-12-14 21:35:53 +000024using namespace llvm;
Brian Gaeked0fde302003-11-11 22:41:34 +000025
Evan Cheng34e400e2007-05-03 18:36:15 +000026namespace {
27 cl::opt<std::string>
Evan Cheng70f684f2008-09-09 06:11:26 +000028 RemoteClient("remote-client",
29 cl::desc("Remote execution client (rsh/ssh)"));
Evan Cheng34e400e2007-05-03 18:36:15 +000030
31 cl::opt<std::string>
Evan Cheng70f684f2008-09-09 06:11:26 +000032 RemoteHost("remote-host",
33 cl::desc("Remote execution (rsh/ssh) host"));
34
35 cl::opt<std::string>
David Goodwin80becf12009-07-10 21:39:28 +000036 RemotePort("remote-port",
37 cl::desc("Remote execution (rsh/ssh) port"));
38
39 cl::opt<std::string>
Evan Cheng70f684f2008-09-09 06:11:26 +000040 RemoteUser("remote-user",
41 cl::desc("Remote execution (rsh/ssh) user id"));
42
43 cl::opt<std::string>
44 RemoteExtra("remote-extra-options",
45 cl::desc("Remote execution (rsh/ssh) extra options"));
Evan Cheng34e400e2007-05-03 18:36:15 +000046}
47
Alkis Evlogimenos1d29a6d2004-02-19 07:39:26 +000048ToolExecutionError::~ToolExecutionError() throw() { }
49
Chris Lattner45495c52005-02-13 23:13:47 +000050/// RunProgramWithTimeout - This function provides an alternate interface to the
51/// sys::Program::ExecuteAndWait interface.
52/// @see sys:Program::ExecuteAndWait
53static int RunProgramWithTimeout(const sys::Path &ProgramPath,
54 const char **Args,
55 const sys::Path &StdInFile,
56 const sys::Path &StdOutFile,
57 const sys::Path &StdErrFile,
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +000058 unsigned NumSeconds = 0,
59 unsigned MemoryLimit = 0) {
Chris Lattner45495c52005-02-13 23:13:47 +000060 const sys::Path* redirects[3];
61 redirects[0] = &StdInFile;
62 redirects[1] = &StdOutFile;
63 redirects[2] = &StdErrFile;
Chris Lattnerc600f3c2006-09-15 21:29:15 +000064
Chris Lattnerd4223502006-09-15 23:01:10 +000065 if (0) {
Dan Gohman65f57c22009-07-15 16:35:29 +000066 errs() << "RUN:";
Chris Lattnerd4223502006-09-15 23:01:10 +000067 for (unsigned i = 0; Args[i]; ++i)
Dan Gohman65f57c22009-07-15 16:35:29 +000068 errs() << " " << Args[i];
69 errs() << "\n";
Chris Lattnerd4223502006-09-15 23:01:10 +000070 }
Misha Brukmanf976c852005-04-21 22:55:34 +000071
72 return
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +000073 sys::Program::ExecuteAndWait(ProgramPath, Args, 0, redirects,
74 NumSeconds, MemoryLimit);
Chris Lattner45495c52005-02-13 23:13:47 +000075}
76
77
78
Reid Spencerb31baa82004-12-19 18:00:21 +000079static void ProcessFailure(sys::Path ProgPath, const char** Args) {
Chris Lattner89bf9ea2004-02-18 20:38:00 +000080 std::ostringstream OS;
Chris Lattnera3de1172004-02-18 20:58:00 +000081 OS << "\nError running tool:\n ";
Chris Lattner89bf9ea2004-02-18 20:38:00 +000082 for (const char **Arg = Args; *Arg; ++Arg)
83 OS << " " << *Arg;
84 OS << "\n";
85
86 // Rerun the compiler, capturing any error messages to print them.
Dan Gohmana971b292009-02-12 20:53:27 +000087 sys::Path ErrorFilename("bugpoint.program_error_messages");
Reid Spencer51c5a282006-08-23 20:34:57 +000088 std::string ErrMsg;
89 if (ErrorFilename.makeUnique(true, &ErrMsg)) {
Dan Gohman65f57c22009-07-15 16:35:29 +000090 errs() << "Error making unique filename: " << ErrMsg << "\n";
Reid Spencer51c5a282006-08-23 20:34:57 +000091 exit(1);
92 }
Reid Spencerb31baa82004-12-19 18:00:21 +000093 RunProgramWithTimeout(ProgPath, Args, sys::Path(""), ErrorFilename,
Reid Spencer8ea5ecb2006-08-21 06:04:45 +000094 ErrorFilename); // FIXME: check return code ?
Chris Lattner89bf9ea2004-02-18 20:38:00 +000095
96 // Print out the error messages generated by GCC if possible...
97 std::ifstream ErrorFile(ErrorFilename.c_str());
98 if (ErrorFile) {
99 std::copy(std::istreambuf_iterator<char>(ErrorFile),
100 std::istreambuf_iterator<char>(),
101 std::ostreambuf_iterator<char>(OS));
102 ErrorFile.close();
103 }
104
Reid Spencera229c5c2005-07-08 03:08:58 +0000105 ErrorFilename.eraseFromDisk();
Chris Lattner89bf9ea2004-02-18 20:38:00 +0000106 throw ToolExecutionError(OS.str());
107}
108
Misha Brukman9558c6a2003-09-29 22:39:25 +0000109//===---------------------------------------------------------------------===//
110// LLI Implementation of AbstractIntepreter interface
111//
Chris Lattner2cdd21c2003-12-14 21:35:53 +0000112namespace {
113 class LLI : public AbstractInterpreter {
114 std::string LLIPath; // The path to the LLI executable
Brian Gaeked11577b2004-05-04 21:09:01 +0000115 std::vector<std::string> ToolArgs; // Args to pass to LLI
Chris Lattner2cdd21c2003-12-14 21:35:53 +0000116 public:
Brian Gaeked11577b2004-05-04 21:09:01 +0000117 LLI(const std::string &Path, const std::vector<std::string> *Args)
118 : LLIPath(Path) {
119 ToolArgs.clear ();
Brian Gaeke48b008d2004-05-04 22:02:41 +0000120 if (Args) { ToolArgs = *Args; }
Brian Gaeked11577b2004-05-04 21:09:01 +0000121 }
Misha Brukmanf976c852005-04-21 22:55:34 +0000122
Gabor Greif8ff70c22007-07-04 21:55:50 +0000123 virtual int ExecuteProgram(const std::string &Bitcode,
Chris Lattner2cdd21c2003-12-14 21:35:53 +0000124 const std::vector<std::string> &Args,
125 const std::string &InputFile,
126 const std::string &OutputFile,
Reid Spencer51ab5c82006-06-06 00:00:42 +0000127 const std::vector<std::string> &GCCArgs,
Misha Brukmanf976c852005-04-21 22:55:34 +0000128 const std::vector<std::string> &SharedLibs =
Chris Lattnere96b2ed2004-07-24 07:49:11 +0000129 std::vector<std::string>(),
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000130 unsigned Timeout = 0,
131 unsigned MemoryLimit = 0);
Chris Lattner2cdd21c2003-12-14 21:35:53 +0000132 };
133}
Misha Brukman9558c6a2003-09-29 22:39:25 +0000134
Gabor Greif8ff70c22007-07-04 21:55:50 +0000135int LLI::ExecuteProgram(const std::string &Bitcode,
Chris Lattner7915a1e2003-10-14 21:34:11 +0000136 const std::vector<std::string> &Args,
Misha Brukman9558c6a2003-09-29 22:39:25 +0000137 const std::string &InputFile,
138 const std::string &OutputFile,
Reid Spencer51ab5c82006-06-06 00:00:42 +0000139 const std::vector<std::string> &GCCArgs,
Chris Lattnere96b2ed2004-07-24 07:49:11 +0000140 const std::vector<std::string> &SharedLibs,
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000141 unsigned Timeout,
142 unsigned MemoryLimit) {
Chris Lattner8c56be52004-02-18 20:21:57 +0000143 if (!SharedLibs.empty())
144 throw ToolExecutionError("LLI currently does not support "
145 "loading shared libraries.");
Misha Brukman9558c6a2003-09-29 22:39:25 +0000146
147 std::vector<const char*> LLIArgs;
148 LLIArgs.push_back(LLIPath.c_str());
Misha Brukman9558c6a2003-09-29 22:39:25 +0000149 LLIArgs.push_back("-force-interpreter=true");
Brian Gaeked11577b2004-05-04 21:09:01 +0000150
151 // Add any extra LLI args.
152 for (unsigned i = 0, e = ToolArgs.size(); i != e; ++i)
153 LLIArgs.push_back(ToolArgs[i].c_str());
154
Gabor Greif8ff70c22007-07-04 21:55:50 +0000155 LLIArgs.push_back(Bitcode.c_str());
Misha Brukman9558c6a2003-09-29 22:39:25 +0000156 // Add optional parameters to the running program from Argv
157 for (unsigned i=0, e = Args.size(); i != e; ++i)
158 LLIArgs.push_back(Args[i].c_str());
159 LLIArgs.push_back(0);
160
161 std::cout << "<lli>" << std::flush;
Dan Gohman65f57c22009-07-15 16:35:29 +0000162 DEBUG(errs() << "\nAbout to run:\t";
Chris Lattner7b2ccff2003-10-19 02:14:58 +0000163 for (unsigned i=0, e = LLIArgs.size()-1; i != e; ++i)
Dan Gohman65f57c22009-07-15 16:35:29 +0000164 errs() << " " << LLIArgs[i];
165 errs() << "\n";
Misha Brukman9558c6a2003-09-29 22:39:25 +0000166 );
Reid Spencerb31baa82004-12-19 18:00:21 +0000167 return RunProgramWithTimeout(sys::Path(LLIPath), &LLIArgs[0],
Misha Brukmanf976c852005-04-21 22:55:34 +0000168 sys::Path(InputFile), sys::Path(OutputFile), sys::Path(OutputFile),
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000169 Timeout, MemoryLimit);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000170}
171
172// LLI create method - Try to find the LLI executable
Chris Lattner7915a1e2003-10-14 21:34:11 +0000173AbstractInterpreter *AbstractInterpreter::createLLI(const std::string &ProgPath,
Brian Gaeked11577b2004-05-04 21:09:01 +0000174 std::string &Message,
175 const std::vector<std::string> *ToolArgs) {
Reid Spencer51ab8ec2004-12-13 23:43:44 +0000176 std::string LLIPath = FindExecutable("lli", ProgPath).toString();
Misha Brukman9558c6a2003-09-29 22:39:25 +0000177 if (!LLIPath.empty()) {
178 Message = "Found lli: " + LLIPath + "\n";
Brian Gaeked11577b2004-05-04 21:09:01 +0000179 return new LLI(LLIPath, ToolArgs);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000180 }
181
182 Message = "Cannot find `lli' in executable directory or PATH!\n";
183 return 0;
184}
185
Anton Korobeynikov9ef74252008-04-28 20:53:48 +0000186//===---------------------------------------------------------------------===//
187// Custom execution command implementation of AbstractIntepreter interface
188//
189// Allows using a custom command for executing the bitcode, thus allows,
190// for example, to invoke a cross compiler for code generation followed by
191// a simulator that executes the generated binary.
192namespace {
193 class CustomExecutor : public AbstractInterpreter {
194 std::string ExecutionCommand;
195 std::vector<std::string> ExecutorArgs;
196 public:
197 CustomExecutor(
198 const std::string &ExecutionCmd, std::vector<std::string> ExecArgs) :
199 ExecutionCommand(ExecutionCmd), ExecutorArgs(ExecArgs) {}
200
201 virtual int ExecuteProgram(const std::string &Bitcode,
202 const std::vector<std::string> &Args,
203 const std::string &InputFile,
204 const std::string &OutputFile,
205 const std::vector<std::string> &GCCArgs,
206 const std::vector<std::string> &SharedLibs =
207 std::vector<std::string>(),
208 unsigned Timeout = 0,
209 unsigned MemoryLimit = 0);
210 };
211}
212
213int CustomExecutor::ExecuteProgram(const std::string &Bitcode,
214 const std::vector<std::string> &Args,
215 const std::string &InputFile,
216 const std::string &OutputFile,
217 const std::vector<std::string> &GCCArgs,
218 const std::vector<std::string> &SharedLibs,
219 unsigned Timeout,
220 unsigned MemoryLimit) {
221
222 std::vector<const char*> ProgramArgs;
223 ProgramArgs.push_back(ExecutionCommand.c_str());
224
225 for (std::size_t i = 0; i < ExecutorArgs.size(); ++i)
226 ProgramArgs.push_back(ExecutorArgs.at(i).c_str());
227 ProgramArgs.push_back(Bitcode.c_str());
228 ProgramArgs.push_back(0);
229
230 // Add optional parameters to the running program from Argv
231 for (unsigned i=0, e = Args.size(); i != e; ++i)
232 ProgramArgs.push_back(Args[i].c_str());
233
234 return RunProgramWithTimeout(
235 sys::Path(ExecutionCommand),
236 &ProgramArgs[0], sys::Path(InputFile), sys::Path(OutputFile),
237 sys::Path(OutputFile), Timeout, MemoryLimit);
238}
239
240// Custom execution environment create method, takes the execution command
241// as arguments
242AbstractInterpreter *AbstractInterpreter::createCustom(
243 const std::string &ProgramPath,
244 std::string &Message,
245 const std::string &ExecCommandLine) {
246
247 std::string Command = "";
248 std::vector<std::string> Args;
249 std::string delimiters = " ";
250
251 // Tokenize the ExecCommandLine to the command and the args to allow
252 // defining a full command line as the command instead of just the
253 // executed program. We cannot just pass the whole string after the command
254 // as a single argument because then program sees only a single
255 // command line argument (with spaces in it: "foo bar" instead
256 // of "foo" and "bar").
257
258 // code borrowed from:
259 // http://oopweb.com/CPP/Documents/CPPHOWTO/Volume/C++Programming-HOWTO-7.html
260 std::string::size_type lastPos =
261 ExecCommandLine.find_first_not_of(delimiters, 0);
262 std::string::size_type pos =
263 ExecCommandLine.find_first_of(delimiters, lastPos);
264
265 while (std::string::npos != pos || std::string::npos != lastPos) {
266 std::string token = ExecCommandLine.substr(lastPos, pos - lastPos);
267 if (Command == "")
268 Command = token;
269 else
270 Args.push_back(token);
271 // Skip delimiters. Note the "not_of"
272 lastPos = ExecCommandLine.find_first_not_of(delimiters, pos);
273 // Find next "non-delimiter"
274 pos = ExecCommandLine.find_first_of(delimiters, lastPos);
275 }
276
277 std::string CmdPath = FindExecutable(Command, ProgramPath).toString();
278 if (CmdPath.empty()) {
279 Message =
280 std::string("Cannot find '") + Command +
281 "' in executable directory or PATH!\n";
282 return 0;
283 }
284
285 Message = "Found command in: " + CmdPath + "\n";
286
287 return new CustomExecutor(CmdPath, Args);
288}
289
Misha Brukman9558c6a2003-09-29 22:39:25 +0000290//===----------------------------------------------------------------------===//
291// LLC Implementation of AbstractIntepreter interface
292//
Gabor Greif8ff70c22007-07-04 21:55:50 +0000293GCC::FileType LLC::OutputCode(const std::string &Bitcode,
Chris Lattnerc600f3c2006-09-15 21:29:15 +0000294 sys::Path &OutputAsmFile) {
Gabor Greif8ff70c22007-07-04 21:55:50 +0000295 sys::Path uniqueFile(Bitcode+".llc.s");
Reid Spencer51c5a282006-08-23 20:34:57 +0000296 std::string ErrMsg;
297 if (uniqueFile.makeUnique(true, &ErrMsg)) {
Dan Gohman65f57c22009-07-15 16:35:29 +0000298 errs() << "Error making unique filename: " << ErrMsg << "\n";
Reid Spencer51c5a282006-08-23 20:34:57 +0000299 exit(1);
300 }
Reid Spencer9ac14182004-12-16 23:01:34 +0000301 OutputAsmFile = uniqueFile;
Brian Gaeked11577b2004-05-04 21:09:01 +0000302 std::vector<const char *> LLCArgs;
303 LLCArgs.push_back (LLCPath.c_str());
304
305 // Add any extra LLC args.
306 for (unsigned i = 0, e = ToolArgs.size(); i != e; ++i)
307 LLCArgs.push_back(ToolArgs[i].c_str());
308
309 LLCArgs.push_back ("-o");
310 LLCArgs.push_back (OutputAsmFile.c_str()); // Output to the Asm file
311 LLCArgs.push_back ("-f"); // Overwrite as necessary...
Gabor Greif8ff70c22007-07-04 21:55:50 +0000312 LLCArgs.push_back (Bitcode.c_str()); // This is the input bitcode
Brian Gaeked11577b2004-05-04 21:09:01 +0000313 LLCArgs.push_back (0);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000314
315 std::cout << "<llc>" << std::flush;
Dan Gohman65f57c22009-07-15 16:35:29 +0000316 DEBUG(errs() << "\nAbout to run:\t";
Brian Gaeked11577b2004-05-04 21:09:01 +0000317 for (unsigned i=0, e = LLCArgs.size()-1; i != e; ++i)
Dan Gohman65f57c22009-07-15 16:35:29 +0000318 errs() << " " << LLCArgs[i];
319 errs() << "\n";
Brian Gaeked11577b2004-05-04 21:09:01 +0000320 );
Misha Brukmanf976c852005-04-21 22:55:34 +0000321 if (RunProgramWithTimeout(sys::Path(LLCPath), &LLCArgs[0],
Reid Spencerb31baa82004-12-19 18:00:21 +0000322 sys::Path(), sys::Path(), sys::Path()))
323 ProcessFailure(sys::Path(LLCPath), &LLCArgs[0]);
Chris Lattnerc600f3c2006-09-15 21:29:15 +0000324
325 return GCC::AsmFile;
Misha Brukman9558c6a2003-09-29 22:39:25 +0000326}
327
Gabor Greif8ff70c22007-07-04 21:55:50 +0000328void LLC::compileProgram(const std::string &Bitcode) {
Reid Spencer9ac14182004-12-16 23:01:34 +0000329 sys::Path OutputAsmFile;
Gabor Greif8ff70c22007-07-04 21:55:50 +0000330 OutputCode(Bitcode, OutputAsmFile);
Reid Spencera229c5c2005-07-08 03:08:58 +0000331 OutputAsmFile.eraseFromDisk();
Chris Lattner9cbbee32004-02-18 23:24:41 +0000332}
333
Gabor Greif8ff70c22007-07-04 21:55:50 +0000334int LLC::ExecuteProgram(const std::string &Bitcode,
Chris Lattner7915a1e2003-10-14 21:34:11 +0000335 const std::vector<std::string> &Args,
Misha Brukman9558c6a2003-09-29 22:39:25 +0000336 const std::string &InputFile,
337 const std::string &OutputFile,
Reid Spencer51ab5c82006-06-06 00:00:42 +0000338 const std::vector<std::string> &ArgsForGCC,
Chris Lattnere96b2ed2004-07-24 07:49:11 +0000339 const std::vector<std::string> &SharedLibs,
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000340 unsigned Timeout,
341 unsigned MemoryLimit) {
Chris Lattnereeed9832003-10-14 21:52:52 +0000342
Reid Spencer9ac14182004-12-16 23:01:34 +0000343 sys::Path OutputAsmFile;
Gabor Greif8ff70c22007-07-04 21:55:50 +0000344 OutputCode(Bitcode, OutputAsmFile);
Chris Lattner8c56be52004-02-18 20:21:57 +0000345 FileRemover OutFileRemover(OutputAsmFile);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000346
Reid Spencer51ab5c82006-06-06 00:00:42 +0000347 std::vector<std::string> GCCArgs(ArgsForGCC);
Bill Wendling38efa382009-03-02 23:13:18 +0000348 GCCArgs.insert(GCCArgs.end(), SharedLibs.begin(), SharedLibs.end());
349 GCCArgs.insert(GCCArgs.end(), gccArgs.begin(), gccArgs.end());
Reid Spencer51ab5c82006-06-06 00:00:42 +0000350
Misha Brukman9558c6a2003-09-29 22:39:25 +0000351 // Assuming LLC worked, compile the result with GCC and run it.
Reid Spencer9ac14182004-12-16 23:01:34 +0000352 return gcc->ExecuteProgram(OutputAsmFile.toString(), Args, GCC::AsmFile,
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000353 InputFile, OutputFile, GCCArgs,
354 Timeout, MemoryLimit);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000355}
356
Chris Lattner7915a1e2003-10-14 21:34:11 +0000357/// createLLC - Try to find the LLC executable
Misha Brukman9558c6a2003-09-29 22:39:25 +0000358///
Chris Lattner7915a1e2003-10-14 21:34:11 +0000359LLC *AbstractInterpreter::createLLC(const std::string &ProgramPath,
Brian Gaeked11577b2004-05-04 21:09:01 +0000360 std::string &Message,
Bill Wendling38efa382009-03-02 23:13:18 +0000361 const std::vector<std::string> *Args,
362 const std::vector<std::string> *GCCArgs) {
Reid Spencer51ab8ec2004-12-13 23:43:44 +0000363 std::string LLCPath = FindExecutable("llc", ProgramPath).toString();
Misha Brukman9558c6a2003-09-29 22:39:25 +0000364 if (LLCPath.empty()) {
365 Message = "Cannot find `llc' in executable directory or PATH!\n";
366 return 0;
367 }
368
369 Message = "Found llc: " + LLCPath + "\n";
Bill Wendling38efa382009-03-02 23:13:18 +0000370 GCC *gcc = GCC::create(ProgramPath, Message, GCCArgs);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000371 if (!gcc) {
Dan Gohman65f57c22009-07-15 16:35:29 +0000372 errs() << Message << "\n";
Misha Brukman9558c6a2003-09-29 22:39:25 +0000373 exit(1);
374 }
Bill Wendling38efa382009-03-02 23:13:18 +0000375 return new LLC(LLCPath, gcc, Args, GCCArgs);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000376}
377
378//===---------------------------------------------------------------------===//
379// JIT Implementation of AbstractIntepreter interface
380//
Chris Lattner2cdd21c2003-12-14 21:35:53 +0000381namespace {
382 class JIT : public AbstractInterpreter {
383 std::string LLIPath; // The path to the LLI executable
Brian Gaeked11577b2004-05-04 21:09:01 +0000384 std::vector<std::string> ToolArgs; // Args to pass to LLI
Chris Lattner2cdd21c2003-12-14 21:35:53 +0000385 public:
Brian Gaeked11577b2004-05-04 21:09:01 +0000386 JIT(const std::string &Path, const std::vector<std::string> *Args)
387 : LLIPath(Path) {
388 ToolArgs.clear ();
Brian Gaeke48b008d2004-05-04 22:02:41 +0000389 if (Args) { ToolArgs = *Args; }
Brian Gaeked11577b2004-05-04 21:09:01 +0000390 }
Misha Brukmanf976c852005-04-21 22:55:34 +0000391
Gabor Greif8ff70c22007-07-04 21:55:50 +0000392 virtual int ExecuteProgram(const std::string &Bitcode,
Chris Lattner2cdd21c2003-12-14 21:35:53 +0000393 const std::vector<std::string> &Args,
394 const std::string &InputFile,
395 const std::string &OutputFile,
Reid Spencer51ab5c82006-06-06 00:00:42 +0000396 const std::vector<std::string> &GCCArgs =
397 std::vector<std::string>(),
Misha Brukmanf976c852005-04-21 22:55:34 +0000398 const std::vector<std::string> &SharedLibs =
Reid Spencer51ab5c82006-06-06 00:00:42 +0000399 std::vector<std::string>(),
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000400 unsigned Timeout =0,
401 unsigned MemoryLimit =0);
Chris Lattner2cdd21c2003-12-14 21:35:53 +0000402 };
403}
Misha Brukman9558c6a2003-09-29 22:39:25 +0000404
Gabor Greif8ff70c22007-07-04 21:55:50 +0000405int JIT::ExecuteProgram(const std::string &Bitcode,
Chris Lattner7915a1e2003-10-14 21:34:11 +0000406 const std::vector<std::string> &Args,
Misha Brukman9558c6a2003-09-29 22:39:25 +0000407 const std::string &InputFile,
408 const std::string &OutputFile,
Reid Spencer51ab5c82006-06-06 00:00:42 +0000409 const std::vector<std::string> &GCCArgs,
Chris Lattnere96b2ed2004-07-24 07:49:11 +0000410 const std::vector<std::string> &SharedLibs,
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000411 unsigned Timeout,
412 unsigned MemoryLimit) {
Misha Brukman9558c6a2003-09-29 22:39:25 +0000413 // Construct a vector of parameters, incorporating those from the command-line
414 std::vector<const char*> JITArgs;
415 JITArgs.push_back(LLIPath.c_str());
Misha Brukman9558c6a2003-09-29 22:39:25 +0000416 JITArgs.push_back("-force-interpreter=false");
Chris Lattnereeed9832003-10-14 21:52:52 +0000417
Brian Gaeked11577b2004-05-04 21:09:01 +0000418 // Add any extra LLI args.
419 for (unsigned i = 0, e = ToolArgs.size(); i != e; ++i)
420 JITArgs.push_back(ToolArgs[i].c_str());
421
Chris Lattnereeed9832003-10-14 21:52:52 +0000422 for (unsigned i = 0, e = SharedLibs.size(); i != e; ++i) {
Misha Brukman9558c6a2003-09-29 22:39:25 +0000423 JITArgs.push_back("-load");
Chris Lattnereeed9832003-10-14 21:52:52 +0000424 JITArgs.push_back(SharedLibs[i].c_str());
Misha Brukman9558c6a2003-09-29 22:39:25 +0000425 }
Gabor Greif8ff70c22007-07-04 21:55:50 +0000426 JITArgs.push_back(Bitcode.c_str());
Misha Brukman9558c6a2003-09-29 22:39:25 +0000427 // Add optional parameters to the running program from Argv
428 for (unsigned i=0, e = Args.size(); i != e; ++i)
429 JITArgs.push_back(Args[i].c_str());
430 JITArgs.push_back(0);
431
432 std::cout << "<jit>" << std::flush;
Dan Gohman65f57c22009-07-15 16:35:29 +0000433 DEBUG(errs() << "\nAbout to run:\t";
Chris Lattner7b2ccff2003-10-19 02:14:58 +0000434 for (unsigned i=0, e = JITArgs.size()-1; i != e; ++i)
Dan Gohman65f57c22009-07-15 16:35:29 +0000435 errs() << " " << JITArgs[i];
436 errs() << "\n";
Misha Brukman9558c6a2003-09-29 22:39:25 +0000437 );
Dan Gohman65f57c22009-07-15 16:35:29 +0000438 DEBUG(errs() << "\nSending output to " << OutputFile << "\n");
Reid Spencerb31baa82004-12-19 18:00:21 +0000439 return RunProgramWithTimeout(sys::Path(LLIPath), &JITArgs[0],
Misha Brukmanf976c852005-04-21 22:55:34 +0000440 sys::Path(InputFile), sys::Path(OutputFile), sys::Path(OutputFile),
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000441 Timeout, MemoryLimit);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000442}
443
Chris Lattner7915a1e2003-10-14 21:34:11 +0000444/// createJIT - Try to find the LLI executable
Misha Brukman9558c6a2003-09-29 22:39:25 +0000445///
Chris Lattner7915a1e2003-10-14 21:34:11 +0000446AbstractInterpreter *AbstractInterpreter::createJIT(const std::string &ProgPath,
Brian Gaeked11577b2004-05-04 21:09:01 +0000447 std::string &Message, const std::vector<std::string> *Args) {
Reid Spencer51ab8ec2004-12-13 23:43:44 +0000448 std::string LLIPath = FindExecutable("lli", ProgPath).toString();
Misha Brukman9558c6a2003-09-29 22:39:25 +0000449 if (!LLIPath.empty()) {
450 Message = "Found lli: " + LLIPath + "\n";
Brian Gaeked11577b2004-05-04 21:09:01 +0000451 return new JIT(LLIPath, Args);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000452 }
453
454 Message = "Cannot find `lli' in executable directory or PATH!\n";
455 return 0;
456}
457
Gabor Greif8ff70c22007-07-04 21:55:50 +0000458GCC::FileType CBE::OutputCode(const std::string &Bitcode,
Chris Lattnerc600f3c2006-09-15 21:29:15 +0000459 sys::Path &OutputCFile) {
Gabor Greif8ff70c22007-07-04 21:55:50 +0000460 sys::Path uniqueFile(Bitcode+".cbe.c");
Reid Spencer51c5a282006-08-23 20:34:57 +0000461 std::string ErrMsg;
462 if (uniqueFile.makeUnique(true, &ErrMsg)) {
Dan Gohman65f57c22009-07-15 16:35:29 +0000463 errs() << "Error making unique filename: " << ErrMsg << "\n";
Reid Spencer51c5a282006-08-23 20:34:57 +0000464 exit(1);
465 }
Reid Spencer9ac14182004-12-16 23:01:34 +0000466 OutputCFile = uniqueFile;
Brian Gaeked11577b2004-05-04 21:09:01 +0000467 std::vector<const char *> LLCArgs;
468 LLCArgs.push_back (LLCPath.c_str());
469
470 // Add any extra LLC args.
471 for (unsigned i = 0, e = ToolArgs.size(); i != e; ++i)
472 LLCArgs.push_back(ToolArgs[i].c_str());
473
474 LLCArgs.push_back ("-o");
475 LLCArgs.push_back (OutputCFile.c_str()); // Output to the C file
476 LLCArgs.push_back ("-march=c"); // Output C language
477 LLCArgs.push_back ("-f"); // Overwrite as necessary...
Gabor Greif8ff70c22007-07-04 21:55:50 +0000478 LLCArgs.push_back (Bitcode.c_str()); // This is the input bitcode
Brian Gaeked11577b2004-05-04 21:09:01 +0000479 LLCArgs.push_back (0);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000480
481 std::cout << "<cbe>" << std::flush;
Dan Gohman65f57c22009-07-15 16:35:29 +0000482 DEBUG(errs() << "\nAbout to run:\t";
Brian Gaeked11577b2004-05-04 21:09:01 +0000483 for (unsigned i=0, e = LLCArgs.size()-1; i != e; ++i)
Dan Gohman65f57c22009-07-15 16:35:29 +0000484 errs() << " " << LLCArgs[i];
485 errs() << "\n";
Brian Gaeked11577b2004-05-04 21:09:01 +0000486 );
Misha Brukmanf976c852005-04-21 22:55:34 +0000487 if (RunProgramWithTimeout(LLCPath, &LLCArgs[0], sys::Path(), sys::Path(),
Reid Spencerb31baa82004-12-19 18:00:21 +0000488 sys::Path()))
Brian Gaeked11577b2004-05-04 21:09:01 +0000489 ProcessFailure(LLCPath, &LLCArgs[0]);
Chris Lattnerc600f3c2006-09-15 21:29:15 +0000490 return GCC::CFile;
Misha Brukman9558c6a2003-09-29 22:39:25 +0000491}
492
Gabor Greif8ff70c22007-07-04 21:55:50 +0000493void CBE::compileProgram(const std::string &Bitcode) {
Reid Spencer9ac14182004-12-16 23:01:34 +0000494 sys::Path OutputCFile;
Gabor Greif8ff70c22007-07-04 21:55:50 +0000495 OutputCode(Bitcode, OutputCFile);
Reid Spencera229c5c2005-07-08 03:08:58 +0000496 OutputCFile.eraseFromDisk();
Chris Lattner9cbbee32004-02-18 23:24:41 +0000497}
498
Gabor Greif8ff70c22007-07-04 21:55:50 +0000499int CBE::ExecuteProgram(const std::string &Bitcode,
Chris Lattner7915a1e2003-10-14 21:34:11 +0000500 const std::vector<std::string> &Args,
Misha Brukman9558c6a2003-09-29 22:39:25 +0000501 const std::string &InputFile,
502 const std::string &OutputFile,
Reid Spencer51ab5c82006-06-06 00:00:42 +0000503 const std::vector<std::string> &ArgsForGCC,
Chris Lattnere96b2ed2004-07-24 07:49:11 +0000504 const std::vector<std::string> &SharedLibs,
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000505 unsigned Timeout,
506 unsigned MemoryLimit) {
Reid Spencer9ac14182004-12-16 23:01:34 +0000507 sys::Path OutputCFile;
Gabor Greif8ff70c22007-07-04 21:55:50 +0000508 OutputCode(Bitcode, OutputCFile);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000509
Chris Lattner8c56be52004-02-18 20:21:57 +0000510 FileRemover CFileRemove(OutputCFile);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000511
Reid Spencer51ab5c82006-06-06 00:00:42 +0000512 std::vector<std::string> GCCArgs(ArgsForGCC);
Bill Wendling38efa382009-03-02 23:13:18 +0000513 GCCArgs.insert(GCCArgs.end(), SharedLibs.begin(), SharedLibs.end());
514
Misha Brukmanf976c852005-04-21 22:55:34 +0000515 return gcc->ExecuteProgram(OutputCFile.toString(), Args, GCC::CFile,
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000516 InputFile, OutputFile, GCCArgs,
517 Timeout, MemoryLimit);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000518}
519
Chris Lattner9915cd92004-02-17 06:40:06 +0000520/// createCBE - Try to find the 'llc' executable
Misha Brukman9558c6a2003-09-29 22:39:25 +0000521///
Chris Lattner7915a1e2003-10-14 21:34:11 +0000522CBE *AbstractInterpreter::createCBE(const std::string &ProgramPath,
Brian Gaeked11577b2004-05-04 21:09:01 +0000523 std::string &Message,
Bill Wendling38efa382009-03-02 23:13:18 +0000524 const std::vector<std::string> *Args,
525 const std::vector<std::string> *GCCArgs) {
Reid Spencerb31baa82004-12-19 18:00:21 +0000526 sys::Path LLCPath = FindExecutable("llc", ProgramPath);
527 if (LLCPath.isEmpty()) {
Misha Brukmanf976c852005-04-21 22:55:34 +0000528 Message =
Chris Lattner9915cd92004-02-17 06:40:06 +0000529 "Cannot find `llc' in executable directory or PATH!\n";
Misha Brukman9558c6a2003-09-29 22:39:25 +0000530 return 0;
531 }
532
Reid Spencerb31baa82004-12-19 18:00:21 +0000533 Message = "Found llc: " + LLCPath.toString() + "\n";
Bill Wendling38efa382009-03-02 23:13:18 +0000534 GCC *gcc = GCC::create(ProgramPath, Message, GCCArgs);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000535 if (!gcc) {
Dan Gohman65f57c22009-07-15 16:35:29 +0000536 errs() << Message << "\n";
Misha Brukman9558c6a2003-09-29 22:39:25 +0000537 exit(1);
538 }
Brian Gaeked11577b2004-05-04 21:09:01 +0000539 return new CBE(LLCPath, gcc, Args);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000540}
541
542//===---------------------------------------------------------------------===//
543// GCC abstraction
544//
David Goodwin80becf12009-07-10 21:39:28 +0000545
Daniel Dunbar84f9abe2009-07-11 21:24:52 +0000546#ifdef __APPLE__
David Goodwin80becf12009-07-10 21:39:28 +0000547static bool
548IsARMArchitecture(std::vector<std::string> Args)
549{
550 for (std::vector<std::string>::const_iterator
551 I = Args.begin(), E = Args.end(); I != E; ++I) {
552 if (!strcasecmp(I->c_str(), "-arch")) {
553 ++I;
554 if ((I != E) && !strncasecmp(I->c_str(), "arm", strlen("arm"))) {
555 return true;
556 }
557 }
558 }
559
560 return false;
561}
Daniel Dunbar84f9abe2009-07-11 21:24:52 +0000562#endif
David Goodwin80becf12009-07-10 21:39:28 +0000563
Misha Brukman9558c6a2003-09-29 22:39:25 +0000564int GCC::ExecuteProgram(const std::string &ProgramFile,
Chris Lattner7915a1e2003-10-14 21:34:11 +0000565 const std::vector<std::string> &Args,
Misha Brukman9558c6a2003-09-29 22:39:25 +0000566 FileType fileType,
567 const std::string &InputFile,
568 const std::string &OutputFile,
Reid Spencer51ab5c82006-06-06 00:00:42 +0000569 const std::vector<std::string> &ArgsForGCC,
Anton Korobeynikov9ba8a762007-02-16 19:11:07 +0000570 unsigned Timeout,
571 unsigned MemoryLimit) {
Misha Brukman9558c6a2003-09-29 22:39:25 +0000572 std::vector<const char*> GCCArgs;
573
574 GCCArgs.push_back(GCCPath.c_str());
Chris Lattnereeed9832003-10-14 21:52:52 +0000575
Bill Wendling38efa382009-03-02 23:13:18 +0000576 for (std::vector<std::string>::const_iterator
577 I = gccArgs.begin(), E = gccArgs.end(); I != E; ++I)
578 GCCArgs.push_back(I->c_str());
579
Chris Lattnereeed9832003-10-14 21:52:52 +0000580 // Specify -x explicitly in case the extension is wonky
Misha Brukman9558c6a2003-09-29 22:39:25 +0000581 GCCArgs.push_back("-x");
582 if (fileType == CFile) {
583 GCCArgs.push_back("c");
584 GCCArgs.push_back("-fno-strict-aliasing");
585 } else {
586 GCCArgs.push_back("assembler");
Chris Lattnerd00b2882005-08-29 13:14:24 +0000587#ifdef __APPLE__
David Goodwin80becf12009-07-10 21:39:28 +0000588 // For ARM architectures we don't want this flag. bugpoint isn't
589 // explicitly told what architecture it is working on, so we get
590 // it from gcc flags
591 if (!IsARMArchitecture(ArgsForGCC))
592 GCCArgs.push_back("-force_cpusubtype_ALL");
Chris Lattnerd00b2882005-08-29 13:14:24 +0000593#endif
Misha Brukman9558c6a2003-09-29 22:39:25 +0000594 }
595 GCCArgs.push_back(ProgramFile.c_str()); // Specify the input filename...
Chris Lattner629e4872006-06-09 21:31:53 +0000596 GCCArgs.push_back("-x");
597 GCCArgs.push_back("none");
Misha Brukman9558c6a2003-09-29 22:39:25 +0000598 GCCArgs.push_back("-o");
Reid Spencercda985e2004-12-15 01:51:56 +0000599 sys::Path OutputBinary (ProgramFile+".gcc.exe");
Reid Spencer51c5a282006-08-23 20:34:57 +0000600 std::string ErrMsg;
601 if (OutputBinary.makeUnique(true, &ErrMsg)) {
Dan Gohman65f57c22009-07-15 16:35:29 +0000602 errs() << "Error making unique filename: " << ErrMsg << "\n";
Reid Spencer51c5a282006-08-23 20:34:57 +0000603 exit(1);
604 }
Misha Brukman9558c6a2003-09-29 22:39:25 +0000605 GCCArgs.push_back(OutputBinary.c_str()); // Output to the right file...
Reid Spencer51ab5c82006-06-06 00:00:42 +0000606
607 // Add any arguments intended for GCC. We locate them here because this is
608 // most likely -L and -l options that need to come before other libraries but
609 // after the source. Other options won't be sensitive to placement on the
610 // command line, so this should be safe.
611 for (unsigned i = 0, e = ArgsForGCC.size(); i != e; ++i)
612 GCCArgs.push_back(ArgsForGCC[i].c_str());
613
Misha Brukman9558c6a2003-09-29 22:39:25 +0000614 GCCArgs.push_back("-lm"); // Hard-code the math library...
615 GCCArgs.push_back("-O2"); // Optimize the program a bit...
Brian Gaekec8db76c2003-11-18 06:31:17 +0000616#if defined (HAVE_LINK_R)
Chris Lattner1f0f1622003-10-18 21:54:47 +0000617 GCCArgs.push_back("-Wl,-R."); // Search this dir for .so files
Brian Gaekec8db76c2003-11-18 06:31:17 +0000618#endif
Chris Lattnerfdcc71e2006-02-04 05:02:27 +0000619#ifdef __sparc__
620 GCCArgs.push_back("-mcpu=v9");
621#endif
Misha Brukman9558c6a2003-09-29 22:39:25 +0000622 GCCArgs.push_back(0); // NULL terminator
623
624 std::cout << "<gcc>" << std::flush;
Dan Gohman65f57c22009-07-15 16:35:29 +0000625 DEBUG(errs() << "\nAbout to run:\t";
Evan Cheng7f7fdcc2007-01-03 07:44:30 +0000626 for (unsigned i=0, e = GCCArgs.size()-1; i != e; ++i)
Dan Gohman65f57c22009-07-15 16:35:29 +0000627 errs() << " " << GCCArgs[i];
628 errs() << "\n";
Evan Cheng7f7fdcc2007-01-03 07:44:30 +0000629 );
Reid Spencerb31baa82004-12-19 18:00:21 +0000630 if (RunProgramWithTimeout(GCCPath, &GCCArgs[0], sys::Path(), sys::Path(),
631 sys::Path())) {
Chris Lattner89bf9ea2004-02-18 20:38:00 +0000632 ProcessFailure(GCCPath, &GCCArgs[0]);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000633 exit(1);
634 }
635
636 std::vector<const char*> ProgramArgs;
Chris Lattner45495c52005-02-13 23:13:47 +0000637
Viktor Kutuzov699220b2009-07-14 19:10:55 +0000638 std::string Exec;
Evan Cheng70f684f2008-09-09 06:11:26 +0000639 if (RemoteClientPath.isEmpty())
Evan Cheng34e400e2007-05-03 18:36:15 +0000640 ProgramArgs.push_back(OutputBinary.c_str());
641 else {
Evan Cheng70f684f2008-09-09 06:11:26 +0000642 ProgramArgs.push_back(RemoteClientPath.c_str());
643 ProgramArgs.push_back(RemoteHost.c_str());
Evan Cheng34e400e2007-05-03 18:36:15 +0000644 ProgramArgs.push_back("-l");
Evan Cheng70f684f2008-09-09 06:11:26 +0000645 ProgramArgs.push_back(RemoteUser.c_str());
David Goodwin80becf12009-07-10 21:39:28 +0000646 if (!RemotePort.empty()) {
647 ProgramArgs.push_back("-p");
648 ProgramArgs.push_back(RemotePort.c_str());
649 }
Evan Cheng70f684f2008-09-09 06:11:26 +0000650 if (!RemoteExtra.empty()) {
651 ProgramArgs.push_back(RemoteExtra.c_str());
652 }
Evan Cheng34e400e2007-05-03 18:36:15 +0000653
Viktor Kutuzov699220b2009-07-14 19:10:55 +0000654 // Full path to the binary
Evan Cheng34e400e2007-05-03 18:36:15 +0000655 char* env_pwd = getenv("PWD");
Evan Cheng34e400e2007-05-03 18:36:15 +0000656 Exec += env_pwd;
Viktor Kutuzov699220b2009-07-14 19:10:55 +0000657 Exec += "/";
Evan Cheng34e400e2007-05-03 18:36:15 +0000658 Exec += OutputBinary.c_str();
659 ProgramArgs.push_back(Exec.c_str());
660 }
661
Misha Brukman9558c6a2003-09-29 22:39:25 +0000662 // Add optional parameters to the running program from Argv
663 for (unsigned i=0, e = Args.size(); i != e; ++i)
664 ProgramArgs.push_back(Args[i].c_str());
665 ProgramArgs.push_back(0); // NULL terminator
666
667 // Now that we have a binary, run it!
668 std::cout << "<program>" << std::flush;
Dan Gohman65f57c22009-07-15 16:35:29 +0000669 DEBUG(errs() << "\nAbout to run:\t";
Chris Lattner7b2ccff2003-10-19 02:14:58 +0000670 for (unsigned i=0, e = ProgramArgs.size()-1; i != e; ++i)
Dan Gohman65f57c22009-07-15 16:35:29 +0000671 errs() << " " << ProgramArgs[i];
672 errs() << "\n";
Misha Brukman9558c6a2003-09-29 22:39:25 +0000673 );
Chris Lattner8c56be52004-02-18 20:21:57 +0000674
Reid Spencer9ac14182004-12-16 23:01:34 +0000675 FileRemover OutputBinaryRemover(OutputBinary);
Evan Cheng34e400e2007-05-03 18:36:15 +0000676
Viktor Kutuzov699220b2009-07-14 19:10:55 +0000677 if (RemoteClientPath.isEmpty()) {
Dan Gohman65f57c22009-07-15 16:35:29 +0000678 DEBUG(errs() << "<run locally>";);
Evan Cheng34e400e2007-05-03 18:36:15 +0000679 return RunProgramWithTimeout(OutputBinary, &ProgramArgs[0],
680 sys::Path(InputFile), sys::Path(OutputFile), sys::Path(OutputFile),
681 Timeout, MemoryLimit);
Viktor Kutuzov699220b2009-07-14 19:10:55 +0000682 } else {
683 std::cout << "<run remotely>" << std::flush;
684 int RemoteClientStatus = RunProgramWithTimeout(sys::Path(RemoteClientPath),
685 &ProgramArgs[0], sys::Path(InputFile), sys::Path(OutputFile),
686 sys::Path(OutputFile), Timeout, MemoryLimit);
687 if (RemoteClientStatus != 0) {
Dan Gohman65f57c22009-07-15 16:35:29 +0000688 errs() << "Remote Client failed with an error: " <<
689 RemoteClientStatus << ".\n";
Viktor Kutuzov699220b2009-07-14 19:10:55 +0000690 }
691 }
Evan Chengf975e012009-07-14 23:55:32 +0000692
693 return 0;
Misha Brukman9558c6a2003-09-29 22:39:25 +0000694}
695
Chris Lattner1798e4a2003-10-14 21:07:25 +0000696int GCC::MakeSharedObject(const std::string &InputFile, FileType fileType,
Chris Lattner130e2a32006-06-27 20:35:36 +0000697 std::string &OutputFile,
698 const std::vector<std::string> &ArgsForGCC) {
Reid Spencercda985e2004-12-15 01:51:56 +0000699 sys::Path uniqueFilename(InputFile+LTDL_SHLIB_EXT);
Reid Spencer51c5a282006-08-23 20:34:57 +0000700 std::string ErrMsg;
701 if (uniqueFilename.makeUnique(true, &ErrMsg)) {
Dan Gohman65f57c22009-07-15 16:35:29 +0000702 errs() << "Error making unique filename: " << ErrMsg << "\n";
Reid Spencer51c5a282006-08-23 20:34:57 +0000703 exit(1);
704 }
Reid Spencercda985e2004-12-15 01:51:56 +0000705 OutputFile = uniqueFilename.toString();
706
Chris Lattner130e2a32006-06-27 20:35:36 +0000707 std::vector<const char*> GCCArgs;
708
709 GCCArgs.push_back(GCCPath.c_str());
Evan Chengfd829952009-03-12 00:53:34 +0000710
711 for (std::vector<std::string>::const_iterator
712 I = gccArgs.begin(), E = gccArgs.end(); I != E; ++I)
713 GCCArgs.push_back(I->c_str());
714
Misha Brukman9558c6a2003-09-29 22:39:25 +0000715 // Compile the C/asm file into a shared object
Chris Lattner130e2a32006-06-27 20:35:36 +0000716 GCCArgs.push_back("-x");
717 GCCArgs.push_back(fileType == AsmFile ? "assembler" : "c");
718 GCCArgs.push_back("-fno-strict-aliasing");
719 GCCArgs.push_back(InputFile.c_str()); // Specify the input filename.
Lauro Ramos Venancio497391a2007-06-06 23:10:56 +0000720 GCCArgs.push_back("-x");
721 GCCArgs.push_back("none");
Misha Brukman9558c6a2003-09-29 22:39:25 +0000722#if defined(sparc) || defined(__sparc__) || defined(__sparcv9)
Chris Lattner130e2a32006-06-27 20:35:36 +0000723 GCCArgs.push_back("-G"); // Compile a shared library, `-G' for Sparc
Nate Begeman72b286b2005-07-08 00:23:26 +0000724#elif defined(__APPLE__)
Chris Lattner130e2a32006-06-27 20:35:36 +0000725 // link all source files into a single module in data segment, rather than
726 // generating blocks. dynamic_lookup requires that you set
727 // MACOSX_DEPLOYMENT_TARGET=10.3 in your env. FIXME: it would be better for
728 // bugpoint to just pass that in the environment of GCC.
729 GCCArgs.push_back("-single_module");
730 GCCArgs.push_back("-dynamiclib"); // `-dynamiclib' for MacOS X/PowerPC
731 GCCArgs.push_back("-undefined");
732 GCCArgs.push_back("dynamic_lookup");
Misha Brukmanb3998ec2004-07-16 19:45:45 +0000733#else
Chris Lattner130e2a32006-06-27 20:35:36 +0000734 GCCArgs.push_back("-shared"); // `-shared' for Linux/X86, maybe others
Misha Brukman9558c6a2003-09-29 22:39:25 +0000735#endif
Chris Lattner1c81f132005-03-09 03:31:02 +0000736
Torok Edwinaf4fc282008-04-06 12:42:29 +0000737#if defined(__ia64__) || defined(__alpha__) || defined(__amd64__)
Chris Lattner130e2a32006-06-27 20:35:36 +0000738 GCCArgs.push_back("-fPIC"); // Requires shared objs to contain PIC
Chris Lattner1c81f132005-03-09 03:31:02 +0000739#endif
Chris Lattnerfdcc71e2006-02-04 05:02:27 +0000740#ifdef __sparc__
Chris Lattner130e2a32006-06-27 20:35:36 +0000741 GCCArgs.push_back("-mcpu=v9");
Chris Lattnerfdcc71e2006-02-04 05:02:27 +0000742#endif
Chris Lattner130e2a32006-06-27 20:35:36 +0000743 GCCArgs.push_back("-o");
744 GCCArgs.push_back(OutputFile.c_str()); // Output to the right filename.
745 GCCArgs.push_back("-O2"); // Optimize the program a bit.
746
747
748
749 // Add any arguments intended for GCC. We locate them here because this is
750 // most likely -L and -l options that need to come before other libraries but
751 // after the source. Other options won't be sensitive to placement on the
752 // command line, so this should be safe.
753 for (unsigned i = 0, e = ArgsForGCC.size(); i != e; ++i)
754 GCCArgs.push_back(ArgsForGCC[i].c_str());
755 GCCArgs.push_back(0); // NULL terminator
756
757
Misha Brukmanf976c852005-04-21 22:55:34 +0000758
Misha Brukman9558c6a2003-09-29 22:39:25 +0000759 std::cout << "<gcc>" << std::flush;
Dan Gohman65f57c22009-07-15 16:35:29 +0000760 DEBUG(errs() << "\nAbout to run:\t";
Evan Cheng7f7fdcc2007-01-03 07:44:30 +0000761 for (unsigned i=0, e = GCCArgs.size()-1; i != e; ++i)
Dan Gohman65f57c22009-07-15 16:35:29 +0000762 errs() << " " << GCCArgs[i];
763 errs() << "\n";
Evan Cheng7f7fdcc2007-01-03 07:44:30 +0000764 );
Chris Lattner130e2a32006-06-27 20:35:36 +0000765 if (RunProgramWithTimeout(GCCPath, &GCCArgs[0], sys::Path(), sys::Path(),
Reid Spencerb31baa82004-12-19 18:00:21 +0000766 sys::Path())) {
Chris Lattner130e2a32006-06-27 20:35:36 +0000767 ProcessFailure(GCCPath, &GCCArgs[0]);
Chris Lattner1798e4a2003-10-14 21:07:25 +0000768 return 1;
Misha Brukman9558c6a2003-09-29 22:39:25 +0000769 }
770 return 0;
771}
772
Chris Lattner7915a1e2003-10-14 21:34:11 +0000773/// create - Try to find the `gcc' executable
Misha Brukman9558c6a2003-09-29 22:39:25 +0000774///
Bill Wendling38efa382009-03-02 23:13:18 +0000775GCC *GCC::create(const std::string &ProgramPath, std::string &Message,
776 const std::vector<std::string> *Args) {
Reid Spencerb31baa82004-12-19 18:00:21 +0000777 sys::Path GCCPath = FindExecutable("gcc", ProgramPath);
778 if (GCCPath.isEmpty()) {
Misha Brukman9558c6a2003-09-29 22:39:25 +0000779 Message = "Cannot find `gcc' in executable directory or PATH!\n";
780 return 0;
781 }
782
Evan Cheng70f684f2008-09-09 06:11:26 +0000783 sys::Path RemoteClientPath;
784 if (!RemoteClient.empty())
785 RemoteClientPath = FindExecutable(RemoteClient.c_str(), ProgramPath);
Evan Cheng34e400e2007-05-03 18:36:15 +0000786
Reid Spencerb31baa82004-12-19 18:00:21 +0000787 Message = "Found gcc: " + GCCPath.toString() + "\n";
Bill Wendling38efa382009-03-02 23:13:18 +0000788 return new GCC(GCCPath, RemoteClientPath, Args);
Misha Brukman9558c6a2003-09-29 22:39:25 +0000789}