blob: 91975547e5f0cba5b94302eaf2f6f2018d30970b [file] [log] [blame]
Chris Lattner2fbfdcf2002-04-07 20:49:59 +00001//===-- ExternalFunctions.cpp - Implement External Functions --------------===//
Chris Lattner7720c8e2001-09-10 04:50:17 +00002//
John Criswellb576c942003-10-20 19:43:21 +00003// The LLVM Compiler Infrastructure
4//
5// This file was developed by the LLVM research group and is distributed under
6// the University of Illinois Open Source License. See LICENSE.TXT for details.
7//
8//===----------------------------------------------------------------------===//
9//
Chris Lattner2fbfdcf2002-04-07 20:49:59 +000010// This file contains both code to deal with invoking "external" functions, but
11// also contains code that implements "exported" external functions.
Chris Lattner7720c8e2001-09-10 04:50:17 +000012//
Brian Gaeke58a6faa2003-10-10 17:03:10 +000013// External functions in the interpreter are implemented by
14// using the system's dynamic loader to look up the address of the function
15// we want to invoke. If a function is found, then one of the
16// many lle_* wrapper functions in this file will translate its arguments from
17// GenericValues to the types the function is actually expecting, before the
18// function is called.
Chris Lattner7720c8e2001-09-10 04:50:17 +000019//
20//===----------------------------------------------------------------------===//
21
22#include "Interpreter.h"
23#include "llvm/DerivedTypes.h"
Misha Brukmanb8d15b22003-10-14 21:42:11 +000024#include "llvm/Module.h"
Reid Spencerdf5a37e2004-11-29 14:11:29 +000025#include "llvm/System/DynamicLibrary.h"
Chris Lattner005cbce2002-10-02 21:12:13 +000026#include "llvm/Target/TargetData.h"
Brian Gaeke70337982003-06-23 19:41:55 +000027#include <cmath>
Brian Gaekeb56a6bc2003-11-05 01:18:49 +000028#include <csignal>
Misha Brukmanb8d15b22003-10-14 21:42:11 +000029#include <map>
Chris Lattner697954c2002-01-20 22:54:45 +000030using std::vector;
Chris Lattner7720c8e2001-09-10 04:50:17 +000031
Chris Lattnerf7a743d2003-12-14 23:25:48 +000032using namespace llvm;
Brian Gaeked0fde302003-11-11 22:41:34 +000033
Chris Lattnerb408b122002-03-29 03:57:15 +000034typedef GenericValue (*ExFunc)(FunctionType *, const vector<GenericValue> &);
35static std::map<const Function *, ExFunc> Functions;
Chris Lattner697954c2002-01-20 22:54:45 +000036static std::map<std::string, ExFunc> FuncNames;
Chris Lattner7720c8e2001-09-10 04:50:17 +000037
Chris Lattnere43db882001-10-27 04:15:57 +000038static Interpreter *TheInterpreter;
39
Chris Lattner7720c8e2001-09-10 04:50:17 +000040static char getTypeID(const Type *Ty) {
Chris Lattnerf70c22b2004-06-17 18:19:28 +000041 switch (Ty->getTypeID()) {
Chris Lattner7720c8e2001-09-10 04:50:17 +000042 case Type::VoidTyID: return 'V';
43 case Type::BoolTyID: return 'o';
44 case Type::UByteTyID: return 'B';
45 case Type::SByteTyID: return 'b';
46 case Type::UShortTyID: return 'S';
47 case Type::ShortTyID: return 's';
48 case Type::UIntTyID: return 'I';
49 case Type::IntTyID: return 'i';
50 case Type::ULongTyID: return 'L';
51 case Type::LongTyID: return 'l';
52 case Type::FloatTyID: return 'F';
53 case Type::DoubleTyID: return 'D';
54 case Type::PointerTyID: return 'P';
Chris Lattnerb408b122002-03-29 03:57:15 +000055 case Type::FunctionTyID: return 'M';
Chris Lattner7720c8e2001-09-10 04:50:17 +000056 case Type::StructTyID: return 'T';
57 case Type::ArrayTyID: return 'A';
58 case Type::OpaqueTyID: return 'O';
59 default: return 'U';
60 }
61}
62
Brian Gaeke58a6faa2003-10-10 17:03:10 +000063static ExFunc lookupFunction(const Function *F) {
Chris Lattner7720c8e2001-09-10 04:50:17 +000064 // Function not found, look it up... start by figuring out what the
65 // composite function name should be.
Chris Lattner697954c2002-01-20 22:54:45 +000066 std::string ExtName = "lle_";
Brian Gaeke58a6faa2003-10-10 17:03:10 +000067 const FunctionType *FT = F->getFunctionType();
68 for (unsigned i = 0, e = FT->getNumContainedTypes(); i != e; ++i)
69 ExtName += getTypeID(FT->getContainedType(i));
70 ExtName += "_" + F->getName();
Chris Lattner7720c8e2001-09-10 04:50:17 +000071
Chris Lattner4721f132001-10-30 20:28:00 +000072 ExFunc FnPtr = FuncNames[ExtName];
73 if (FnPtr == 0)
Reid Spencerdf5a37e2004-11-29 14:11:29 +000074 FnPtr = (ExFunc)sys::DynamicLibrary::SearchForAddressOfSymbol(ExtName);
Chris Lattner4721f132001-10-30 20:28:00 +000075 if (FnPtr == 0)
Brian Gaeke58a6faa2003-10-10 17:03:10 +000076 FnPtr = FuncNames["lle_X_"+F->getName()];
Chris Lattner7720c8e2001-09-10 04:50:17 +000077 if (FnPtr == 0) // Try calling a generic function... if it exists...
Reid Spencerdf5a37e2004-11-29 14:11:29 +000078 FnPtr = (ExFunc)sys::DynamicLibrary::SearchForAddressOfSymbol(
79 ("lle_X_"+F->getName()).c_str());
Chris Lattner7720c8e2001-09-10 04:50:17 +000080 if (FnPtr != 0)
Brian Gaeke58a6faa2003-10-10 17:03:10 +000081 Functions.insert(std::make_pair(F, FnPtr)); // Cache for later
Chris Lattner7720c8e2001-09-10 04:50:17 +000082 return FnPtr;
83}
84
Chris Lattnerda82ed52003-05-08 16:18:31 +000085GenericValue Interpreter::callExternalFunction(Function *M,
Chris Lattner44edb6b2003-05-14 14:21:30 +000086 const std::vector<GenericValue> &ArgVals) {
Chris Lattnere43db882001-10-27 04:15:57 +000087 TheInterpreter = this;
88
Chris Lattner2fbfdcf2002-04-07 20:49:59 +000089 // Do a lookup to see if the function is in our cache... this should just be a
Misha Brukmand5d96b92003-10-10 17:42:19 +000090 // deferred annotation!
Chris Lattnerb408b122002-03-29 03:57:15 +000091 std::map<const Function *, ExFunc>::iterator FI = Functions.find(M);
92 ExFunc Fn = (FI == Functions.end()) ? lookupFunction(M) : FI->second;
Chris Lattner7720c8e2001-09-10 04:50:17 +000093 if (Fn == 0) {
Chris Lattnerda82ed52003-05-08 16:18:31 +000094 std::cout << "Tried to execute an unknown external function: "
95 << M->getType()->getDescription() << " " << M->getName() << "\n";
Chris Lattner4721f132001-10-30 20:28:00 +000096 return GenericValue();
Chris Lattner7720c8e2001-09-10 04:50:17 +000097 }
98
99 // TODO: FIXME when types are not const!
Chris Lattner2fbfdcf2002-04-07 20:49:59 +0000100 GenericValue Result = Fn(const_cast<FunctionType*>(M->getFunctionType()),
101 ArgVals);
Chris Lattner4721f132001-10-30 20:28:00 +0000102 return Result;
Chris Lattner7720c8e2001-09-10 04:50:17 +0000103}
104
105
106//===----------------------------------------------------------------------===//
Chris Lattnerb408b122002-03-29 03:57:15 +0000107// Functions "exported" to the running application...
Chris Lattner7720c8e2001-09-10 04:50:17 +0000108//
109extern "C" { // Don't add C++ manglings to llvm mangling :)
110
Chris Lattner005cbce2002-10-02 21:12:13 +0000111// void putchar(sbyte)
Chris Lattnerb408b122002-03-29 03:57:15 +0000112GenericValue lle_Vb_putchar(FunctionType *M, const vector<GenericValue> &Args) {
Chris Lattnerda82ed52003-05-08 16:18:31 +0000113 std::cout << Args[0].SByteVal;
Chris Lattner7720c8e2001-09-10 04:50:17 +0000114 return GenericValue();
115}
116
Chris Lattner005cbce2002-10-02 21:12:13 +0000117// int putchar(int)
Chris Lattnerb408b122002-03-29 03:57:15 +0000118GenericValue lle_ii_putchar(FunctionType *M, const vector<GenericValue> &Args) {
Chris Lattnerda82ed52003-05-08 16:18:31 +0000119 std::cout << ((char)Args[0].IntVal) << std::flush;
Chris Lattnere43db882001-10-27 04:15:57 +0000120 return Args[0];
121}
122
Chris Lattner005cbce2002-10-02 21:12:13 +0000123// void putchar(ubyte)
Chris Lattnerb408b122002-03-29 03:57:15 +0000124GenericValue lle_VB_putchar(FunctionType *M, const vector<GenericValue> &Args) {
Chris Lattnerda82ed52003-05-08 16:18:31 +0000125 std::cout << Args[0].SByteVal << std::flush;
Chris Lattnere43db882001-10-27 04:15:57 +0000126 return Args[0];
Chris Lattner2e42d3a2001-10-15 05:51:48 +0000127}
128
Chris Lattner44edb6b2003-05-14 14:21:30 +0000129// void atexit(Function*)
130GenericValue lle_X_atexit(FunctionType *M, const vector<GenericValue> &Args) {
131 assert(Args.size() == 1);
132 TheInterpreter->addAtExitHandler((Function*)GVTOP(Args[0]));
133 GenericValue GV;
134 GV.IntVal = 0;
135 return GV;
Chris Lattnerf8f2afb2001-10-18 21:55:32 +0000136}
137
Chris Lattner005cbce2002-10-02 21:12:13 +0000138// void exit(int)
Chris Lattnerb408b122002-03-29 03:57:15 +0000139GenericValue lle_X_exit(FunctionType *M, const vector<GenericValue> &Args) {
Chris Lattnere43db882001-10-27 04:15:57 +0000140 TheInterpreter->exitCalled(Args[0]);
141 return GenericValue();
142}
143
Chris Lattner005cbce2002-10-02 21:12:13 +0000144// void abort(void)
Chris Lattner1ee34a52002-05-20 21:17:16 +0000145GenericValue lle_X_abort(FunctionType *M, const vector<GenericValue> &Args) {
Brian Gaekeb56a6bc2003-11-05 01:18:49 +0000146 raise (SIGABRT);
Chris Lattner1ee34a52002-05-20 21:17:16 +0000147 return GenericValue();
148}
149
Chris Lattnerc2593162001-10-27 08:28:11 +0000150// void *malloc(uint)
Chris Lattnerb408b122002-03-29 03:57:15 +0000151GenericValue lle_X_malloc(FunctionType *M, const vector<GenericValue> &Args) {
Chris Lattner4721f132001-10-30 20:28:00 +0000152 assert(Args.size() == 1 && "Malloc expects one argument!");
Chris Lattnerb1118742003-01-13 00:59:47 +0000153 return PTOGV(malloc(Args[0].UIntVal));
Chris Lattnerc2593162001-10-27 08:28:11 +0000154}
155
Chris Lattner957d62a2003-04-23 19:55:24 +0000156// void *calloc(uint, uint)
157GenericValue lle_X_calloc(FunctionType *M, const vector<GenericValue> &Args) {
158 assert(Args.size() == 2 && "calloc expects two arguments!");
159 return PTOGV(calloc(Args[0].UIntVal, Args[1].UIntVal));
160}
161
Chris Lattnerc2593162001-10-27 08:28:11 +0000162// void free(void *)
Chris Lattnerb408b122002-03-29 03:57:15 +0000163GenericValue lle_X_free(FunctionType *M, const vector<GenericValue> &Args) {
Chris Lattnerc063d382001-11-06 21:52:18 +0000164 assert(Args.size() == 1);
Chris Lattnerb1118742003-01-13 00:59:47 +0000165 free(GVTOP(Args[0]));
Chris Lattnerc2593162001-10-27 08:28:11 +0000166 return GenericValue();
167}
168
Chris Lattner782b9392001-11-26 18:18:18 +0000169// int atoi(char *)
Chris Lattnerb408b122002-03-29 03:57:15 +0000170GenericValue lle_X_atoi(FunctionType *M, const vector<GenericValue> &Args) {
Chris Lattner782b9392001-11-26 18:18:18 +0000171 assert(Args.size() == 1);
172 GenericValue GV;
Chris Lattnerb1118742003-01-13 00:59:47 +0000173 GV.IntVal = atoi((char*)GVTOP(Args[0]));
Chris Lattner782b9392001-11-26 18:18:18 +0000174 return GV;
175}
176
Chris Lattnerc2593162001-10-27 08:28:11 +0000177// double pow(double, double)
Chris Lattnerb408b122002-03-29 03:57:15 +0000178GenericValue lle_X_pow(FunctionType *M, const vector<GenericValue> &Args) {
Chris Lattnerc063d382001-11-06 21:52:18 +0000179 assert(Args.size() == 2);
Chris Lattnerc2593162001-10-27 08:28:11 +0000180 GenericValue GV;
Chris Lattner08845a22001-10-29 20:27:45 +0000181 GV.DoubleVal = pow(Args[0].DoubleVal, Args[1].DoubleVal);
Chris Lattnerc2593162001-10-27 08:28:11 +0000182 return GV;
183}
184
Chris Lattner34dd24b2002-02-18 19:06:25 +0000185// double exp(double)
Chris Lattnerb408b122002-03-29 03:57:15 +0000186GenericValue lle_X_exp(FunctionType *M, const vector<GenericValue> &Args) {
Chris Lattner34dd24b2002-02-18 19:06:25 +0000187 assert(Args.size() == 1);
188 GenericValue GV;
189 GV.DoubleVal = exp(Args[0].DoubleVal);
190 return GV;
191}
192
Chris Lattnerc063d382001-11-06 21:52:18 +0000193// double sqrt(double)
Chris Lattnerb408b122002-03-29 03:57:15 +0000194GenericValue lle_X_sqrt(FunctionType *M, const vector<GenericValue> &Args) {
Chris Lattnerc063d382001-11-06 21:52:18 +0000195 assert(Args.size() == 1);
196 GenericValue GV;
197 GV.DoubleVal = sqrt(Args[0].DoubleVal);
198 return GV;
199}
Chris Lattnerc2593162001-10-27 08:28:11 +0000200
Chris Lattner86790052001-11-06 22:53:25 +0000201// double log(double)
Chris Lattnerb408b122002-03-29 03:57:15 +0000202GenericValue lle_X_log(FunctionType *M, const vector<GenericValue> &Args) {
Chris Lattner86790052001-11-06 22:53:25 +0000203 assert(Args.size() == 1);
204 GenericValue GV;
205 GV.DoubleVal = log(Args[0].DoubleVal);
206 return GV;
207}
208
Chris Lattner782b9392001-11-26 18:18:18 +0000209// double floor(double)
Chris Lattnerb408b122002-03-29 03:57:15 +0000210GenericValue lle_X_floor(FunctionType *M, const vector<GenericValue> &Args) {
Chris Lattner782b9392001-11-26 18:18:18 +0000211 assert(Args.size() == 1);
212 GenericValue GV;
213 GV.DoubleVal = floor(Args[0].DoubleVal);
214 return GV;
215}
216
Reid Spencerabec8f92004-10-27 23:03:44 +0000217#ifdef HAVE_RAND48
218
Chris Lattner86790052001-11-06 22:53:25 +0000219// double drand48()
Chris Lattnerb408b122002-03-29 03:57:15 +0000220GenericValue lle_X_drand48(FunctionType *M, const vector<GenericValue> &Args) {
Chris Lattner86790052001-11-06 22:53:25 +0000221 assert(Args.size() == 0);
222 GenericValue GV;
223 GV.DoubleVal = drand48();
224 return GV;
225}
226
Chris Lattner1b600142001-11-13 05:46:08 +0000227// long lrand48()
Chris Lattnerb408b122002-03-29 03:57:15 +0000228GenericValue lle_X_lrand48(FunctionType *M, const vector<GenericValue> &Args) {
Chris Lattner1b600142001-11-13 05:46:08 +0000229 assert(Args.size() == 0);
230 GenericValue GV;
231 GV.IntVal = lrand48();
232 return GV;
233}
234
235// void srand48(long)
Chris Lattnerb408b122002-03-29 03:57:15 +0000236GenericValue lle_X_srand48(FunctionType *M, const vector<GenericValue> &Args) {
Chris Lattner1b600142001-11-13 05:46:08 +0000237 assert(Args.size() == 1);
238 srand48(Args[0].IntVal);
239 return GenericValue();
240}
241
Reid Spencerabec8f92004-10-27 23:03:44 +0000242#endif
243
244// int rand()
245GenericValue lle_X_rand(FunctionType *M, const vector<GenericValue> &Args) {
246 assert(Args.size() == 0);
247 GenericValue GV;
248 GV.IntVal = rand();
249 return GV;
250}
251
Chris Lattner782b9392001-11-26 18:18:18 +0000252// void srand(uint)
Chris Lattnerb408b122002-03-29 03:57:15 +0000253GenericValue lle_X_srand(FunctionType *M, const vector<GenericValue> &Args) {
Chris Lattner782b9392001-11-26 18:18:18 +0000254 assert(Args.size() == 1);
255 srand(Args[0].UIntVal);
256 return GenericValue();
257}
Chris Lattner86790052001-11-06 22:53:25 +0000258
Chris Lattnerb1118742003-01-13 00:59:47 +0000259// int puts(const char*)
260GenericValue lle_X_puts(FunctionType *M, const vector<GenericValue> &Args) {
261 assert(Args.size() == 1);
262 GenericValue GV;
263 GV.IntVal = puts((char*)GVTOP(Args[0]));
264 return GV;
265}
266
Chris Lattnere7c6f722001-12-13 00:43:47 +0000267// int sprintf(sbyte *, sbyte *, ...) - a very rough implementation to make
268// output useful.
Chris Lattnerb408b122002-03-29 03:57:15 +0000269GenericValue lle_X_sprintf(FunctionType *M, const vector<GenericValue> &Args) {
Chris Lattnerb1118742003-01-13 00:59:47 +0000270 char *OutputBuffer = (char *)GVTOP(Args[0]);
271 const char *FmtStr = (const char *)GVTOP(Args[1]);
Chris Lattnere7c6f722001-12-13 00:43:47 +0000272 unsigned ArgNo = 2;
Chris Lattner08845a22001-10-29 20:27:45 +0000273
274 // printf should return # chars printed. This is completely incorrect, but
275 // close enough for now.
276 GenericValue GV; GV.IntVal = strlen(FmtStr);
277 while (1) {
278 switch (*FmtStr) {
279 case 0: return GV; // Null terminator...
280 default: // Normal nonspecial character
Chris Lattnere7c6f722001-12-13 00:43:47 +0000281 sprintf(OutputBuffer++, "%c", *FmtStr++);
Chris Lattner08845a22001-10-29 20:27:45 +0000282 break;
283 case '\\': { // Handle escape codes
Chris Lattnere7c6f722001-12-13 00:43:47 +0000284 sprintf(OutputBuffer, "%c%c", *FmtStr, *(FmtStr+1));
285 FmtStr += 2; OutputBuffer += 2;
Chris Lattner08845a22001-10-29 20:27:45 +0000286 break;
287 }
288 case '%': { // Handle format specifiers
Chris Lattnerea38c0e2001-11-07 19:46:27 +0000289 char FmtBuf[100] = "", Buffer[1000] = "";
290 char *FB = FmtBuf;
291 *FB++ = *FmtStr++;
292 char Last = *FB++ = *FmtStr++;
293 unsigned HowLong = 0;
294 while (Last != 'c' && Last != 'd' && Last != 'i' && Last != 'u' &&
295 Last != 'o' && Last != 'x' && Last != 'X' && Last != 'e' &&
296 Last != 'E' && Last != 'g' && Last != 'G' && Last != 'f' &&
297 Last != 'p' && Last != 's' && Last != '%') {
298 if (Last == 'l' || Last == 'L') HowLong++; // Keep track of l's
299 Last = *FB++ = *FmtStr++;
Chris Lattner08845a22001-10-29 20:27:45 +0000300 }
Chris Lattnerea38c0e2001-11-07 19:46:27 +0000301 *FB = 0;
302
303 switch (Last) {
304 case '%':
305 sprintf(Buffer, FmtBuf); break;
306 case 'c':
Chris Lattner2012d5e2002-04-17 17:43:01 +0000307 sprintf(Buffer, FmtBuf, Args[ArgNo++].IntVal); break;
Chris Lattnerea38c0e2001-11-07 19:46:27 +0000308 case 'd': case 'i':
309 case 'u': case 'o':
310 case 'x': case 'X':
Chris Lattner69ab7a82002-08-02 23:08:32 +0000311 if (HowLong >= 1) {
Chris Lattner1543e402003-08-24 14:02:47 +0000312 if (HowLong == 1 &&
313 TheInterpreter->getModule().getPointerSize()==Module::Pointer64 &&
Chris Lattner7471c482003-04-25 18:28:44 +0000314 sizeof(long) < sizeof(long long)) {
Chris Lattner69ab7a82002-08-02 23:08:32 +0000315 // Make sure we use %lld with a 64 bit argument because we might be
316 // compiling LLI on a 32 bit compiler.
317 unsigned Size = strlen(FmtBuf);
318 FmtBuf[Size] = FmtBuf[Size-1];
319 FmtBuf[Size+1] = 0;
320 FmtBuf[Size-1] = 'l';
321 }
Chris Lattnerea38c0e2001-11-07 19:46:27 +0000322 sprintf(Buffer, FmtBuf, Args[ArgNo++].ULongVal);
Chris Lattner69ab7a82002-08-02 23:08:32 +0000323 } else
Chris Lattnerea38c0e2001-11-07 19:46:27 +0000324 sprintf(Buffer, FmtBuf, Args[ArgNo++].IntVal); break;
325 case 'e': case 'E': case 'g': case 'G': case 'f':
326 sprintf(Buffer, FmtBuf, Args[ArgNo++].DoubleVal); break;
327 case 'p':
Chris Lattnerb1118742003-01-13 00:59:47 +0000328 sprintf(Buffer, FmtBuf, (void*)GVTOP(Args[ArgNo++])); break;
Chris Lattnerea38c0e2001-11-07 19:46:27 +0000329 case 's':
Chris Lattnerb1118742003-01-13 00:59:47 +0000330 sprintf(Buffer, FmtBuf, (char*)GVTOP(Args[ArgNo++])); break;
Chris Lattnerda82ed52003-05-08 16:18:31 +0000331 default: std::cout << "<unknown printf code '" << *FmtStr << "'!>";
Chris Lattnerea38c0e2001-11-07 19:46:27 +0000332 ArgNo++; break;
Chris Lattner08845a22001-10-29 20:27:45 +0000333 }
Chris Lattnere7c6f722001-12-13 00:43:47 +0000334 strcpy(OutputBuffer, Buffer);
335 OutputBuffer += strlen(Buffer);
Chris Lattnerea38c0e2001-11-07 19:46:27 +0000336 }
Chris Lattner08845a22001-10-29 20:27:45 +0000337 break;
338 }
Chris Lattner08845a22001-10-29 20:27:45 +0000339 }
340}
341
Chris Lattnere7c6f722001-12-13 00:43:47 +0000342// int printf(sbyte *, ...) - a very rough implementation to make output useful.
Chris Lattnerb408b122002-03-29 03:57:15 +0000343GenericValue lle_X_printf(FunctionType *M, const vector<GenericValue> &Args) {
Chris Lattnere7c6f722001-12-13 00:43:47 +0000344 char Buffer[10000];
345 vector<GenericValue> NewArgs;
Chris Lattnerb1118742003-01-13 00:59:47 +0000346 NewArgs.push_back(PTOGV(Buffer));
Chris Lattnere7c6f722001-12-13 00:43:47 +0000347 NewArgs.insert(NewArgs.end(), Args.begin(), Args.end());
Chris Lattnerb1118742003-01-13 00:59:47 +0000348 GenericValue GV = lle_X_sprintf(M, NewArgs);
Chris Lattnerda82ed52003-05-08 16:18:31 +0000349 std::cout << Buffer;
Chris Lattnere7c6f722001-12-13 00:43:47 +0000350 return GV;
351}
352
Chris Lattnerf9a88b62003-03-31 22:12:37 +0000353static void ByteswapSCANFResults(const char *Fmt, void *Arg0, void *Arg1,
354 void *Arg2, void *Arg3, void *Arg4, void *Arg5,
355 void *Arg6, void *Arg7, void *Arg8) {
356 void *Args[] = { Arg0, Arg1, Arg2, Arg3, Arg4, Arg5, Arg6, Arg7, Arg8, 0 };
357
358 // Loop over the format string, munging read values as appropriate (performs
Misha Brukman5560c9d2003-08-18 14:43:39 +0000359 // byteswaps as necessary).
Chris Lattnerf9a88b62003-03-31 22:12:37 +0000360 unsigned ArgNo = 0;
361 while (*Fmt) {
362 if (*Fmt++ == '%') {
363 // Read any flag characters that may be present...
364 bool Suppress = false;
365 bool Half = false;
366 bool Long = false;
367 bool LongLong = false; // long long or long double
368
369 while (1) {
370 switch (*Fmt++) {
371 case '*': Suppress = true; break;
372 case 'a': /*Allocate = true;*/ break; // We don't need to track this
373 case 'h': Half = true; break;
374 case 'l': Long = true; break;
375 case 'q':
376 case 'L': LongLong = true; break;
377 default:
378 if (Fmt[-1] > '9' || Fmt[-1] < '0') // Ignore field width specs
379 goto Out;
380 }
381 }
382 Out:
383
384 // Read the conversion character
385 if (!Suppress && Fmt[-1] != '%') { // Nothing to do?
386 unsigned Size = 0;
387 const Type *Ty = 0;
388
389 switch (Fmt[-1]) {
390 case 'i': case 'o': case 'u': case 'x': case 'X': case 'n': case 'p':
391 case 'd':
392 if (Long || LongLong) {
393 Size = 8; Ty = Type::ULongTy;
394 } else if (Half) {
395 Size = 4; Ty = Type::UShortTy;
396 } else {
397 Size = 4; Ty = Type::UIntTy;
398 }
399 break;
400
401 case 'e': case 'g': case 'E':
402 case 'f':
403 if (Long || LongLong) {
404 Size = 8; Ty = Type::DoubleTy;
405 } else {
406 Size = 4; Ty = Type::FloatTy;
407 }
408 break;
409
410 case 's': case 'c': case '[': // No byteswap needed
411 Size = 1;
412 Ty = Type::SByteTy;
413 break;
414
415 default: break;
416 }
417
418 if (Size) {
419 GenericValue GV;
420 void *Arg = Args[ArgNo++];
421 memcpy(&GV, Arg, Size);
422 TheInterpreter->StoreValueToMemory(GV, (GenericValue*)Arg, Ty);
423 }
424 }
425 }
426 }
427}
428
Chris Lattner665ee882002-03-08 22:51:07 +0000429// int sscanf(const char *format, ...);
Chris Lattnerb408b122002-03-29 03:57:15 +0000430GenericValue lle_X_sscanf(FunctionType *M, const vector<GenericValue> &args) {
Chris Lattner665ee882002-03-08 22:51:07 +0000431 assert(args.size() < 10 && "Only handle up to 10 args to sscanf right now!");
432
Chris Lattnerf9a88b62003-03-31 22:12:37 +0000433 char *Args[10];
Chris Lattner665ee882002-03-08 22:51:07 +0000434 for (unsigned i = 0; i < args.size(); ++i)
Chris Lattnerf9a88b62003-03-31 22:12:37 +0000435 Args[i] = (char*)GVTOP(args[i]);
Chris Lattner665ee882002-03-08 22:51:07 +0000436
437 GenericValue GV;
438 GV.IntVal = sscanf(Args[0], Args[1], Args[2], Args[3], Args[4],
439 Args[5], Args[6], Args[7], Args[8], Args[9]);
Chris Lattnerf9a88b62003-03-31 22:12:37 +0000440 ByteswapSCANFResults(Args[1], Args[2], Args[3], Args[4],
441 Args[5], Args[6], Args[7], Args[8], Args[9], 0);
442 return GV;
443}
444
445// int scanf(const char *format, ...);
446GenericValue lle_X_scanf(FunctionType *M, const vector<GenericValue> &args) {
447 assert(args.size() < 10 && "Only handle up to 10 args to scanf right now!");
448
449 char *Args[10];
450 for (unsigned i = 0; i < args.size(); ++i)
451 Args[i] = (char*)GVTOP(args[i]);
452
453 GenericValue GV;
454 GV.IntVal = scanf(Args[0], Args[1], Args[2], Args[3], Args[4],
455 Args[5], Args[6], Args[7], Args[8], Args[9]);
456 ByteswapSCANFResults(Args[0], Args[1], Args[2], Args[3], Args[4],
457 Args[5], Args[6], Args[7], Args[8], Args[9]);
Chris Lattner665ee882002-03-08 22:51:07 +0000458 return GV;
459}
460
461
Chris Lattner295fe672002-01-23 21:38:07 +0000462// int clock(void) - Profiling implementation
Chris Lattnerb408b122002-03-29 03:57:15 +0000463GenericValue lle_i_clock(FunctionType *M, const vector<GenericValue> &Args) {
Chris Lattner295fe672002-01-23 21:38:07 +0000464 extern int clock(void);
465 GenericValue GV; GV.IntVal = clock();
466 return GV;
467}
Chris Lattnere7c6f722001-12-13 00:43:47 +0000468
Chris Lattner957d62a2003-04-23 19:55:24 +0000469
470//===----------------------------------------------------------------------===//
471// String Functions...
472//===----------------------------------------------------------------------===//
473
474// int strcmp(const char *S1, const char *S2);
475GenericValue lle_X_strcmp(FunctionType *M, const vector<GenericValue> &Args) {
476 assert(Args.size() == 2);
477 GenericValue Ret;
478 Ret.IntVal = strcmp((char*)GVTOP(Args[0]), (char*)GVTOP(Args[1]));
479 return Ret;
480}
481
482// char *strcat(char *Dest, const char *src);
483GenericValue lle_X_strcat(FunctionType *M, const vector<GenericValue> &Args) {
484 assert(Args.size() == 2);
485 return PTOGV(strcat((char*)GVTOP(Args[0]), (char*)GVTOP(Args[1])));
486}
487
488// char *strcpy(char *Dest, const char *src);
489GenericValue lle_X_strcpy(FunctionType *M, const vector<GenericValue> &Args) {
490 assert(Args.size() == 2);
491 return PTOGV(strcpy((char*)GVTOP(Args[0]), (char*)GVTOP(Args[1])));
492}
493
Brian Gaeke3c66c352004-05-01 06:42:15 +0000494static GenericValue size_t_to_GV (size_t n) {
495 GenericValue Ret;
496 if (sizeof (size_t) == sizeof (uint64_t)) {
497 Ret.ULongVal = n;
498 } else {
499 assert (sizeof (size_t) == sizeof (unsigned int));
500 Ret.UIntVal = n;
501 }
502 return Ret;
503}
504
505static size_t GV_to_size_t (GenericValue GV) {
506 size_t count;
507 if (sizeof (size_t) == sizeof (uint64_t)) {
508 count = GV.ULongVal;
509 } else {
510 assert (sizeof (size_t) == sizeof (unsigned int));
511 count = GV.UIntVal;
512 }
513 return count;
514}
515
Brian Gaeke6c9d5822003-12-12 15:38:06 +0000516// size_t strlen(const char *src);
Chris Lattner957d62a2003-04-23 19:55:24 +0000517GenericValue lle_X_strlen(FunctionType *M, const vector<GenericValue> &Args) {
518 assert(Args.size() == 1);
Brian Gaeke6c9d5822003-12-12 15:38:06 +0000519 size_t strlenResult = strlen ((char *) GVTOP (Args[0]));
Brian Gaeke3c66c352004-05-01 06:42:15 +0000520 return size_t_to_GV (strlenResult);
Chris Lattner957d62a2003-04-23 19:55:24 +0000521}
522
Brian Gaeke70975ee2003-09-05 18:42:01 +0000523// char *strdup(const char *src);
524GenericValue lle_X_strdup(FunctionType *M, const vector<GenericValue> &Args) {
525 assert(Args.size() == 1);
526 return PTOGV(strdup((char*)GVTOP(Args[0])));
527}
528
Chris Lattnerc8cff9e2003-04-25 18:23:38 +0000529// char *__strdup(const char *src);
530GenericValue lle_X___strdup(FunctionType *M, const vector<GenericValue> &Args) {
531 assert(Args.size() == 1);
532 return PTOGV(strdup((char*)GVTOP(Args[0])));
533}
534
Chris Lattner957d62a2003-04-23 19:55:24 +0000535// void *memset(void *S, int C, size_t N)
536GenericValue lle_X_memset(FunctionType *M, const vector<GenericValue> &Args) {
537 assert(Args.size() == 3);
Brian Gaeke3c66c352004-05-01 06:42:15 +0000538 size_t count = GV_to_size_t (Args[2]);
539 return PTOGV(memset(GVTOP(Args[0]), Args[1].IntVal, count));
Chris Lattner957d62a2003-04-23 19:55:24 +0000540}
541
Chris Lattner5f311a72003-04-23 20:23:16 +0000542// void *memcpy(void *Dest, void *src, size_t Size);
543GenericValue lle_X_memcpy(FunctionType *M, const vector<GenericValue> &Args) {
544 assert(Args.size() == 3);
Brian Gaeke3c66c352004-05-01 06:42:15 +0000545 size_t count = GV_to_size_t (Args[2]);
546 return PTOGV(memcpy((char*)GVTOP(Args[0]), (char*)GVTOP(Args[1]), count));
Chris Lattner5f311a72003-04-23 20:23:16 +0000547}
Chris Lattner957d62a2003-04-23 19:55:24 +0000548
Chris Lattner665ee882002-03-08 22:51:07 +0000549//===----------------------------------------------------------------------===//
550// IO Functions...
551//===----------------------------------------------------------------------===//
552
Chris Lattner005cbce2002-10-02 21:12:13 +0000553// getFILE - Turn a pointer in the host address space into a legit pointer in
Reid Spencer0098bdf2004-05-25 08:46:15 +0000554// the interpreter address space. This is an identity transformation.
555#define getFILE(ptr) ((FILE*)ptr)
Chris Lattner005cbce2002-10-02 21:12:13 +0000556
Chris Lattner665ee882002-03-08 22:51:07 +0000557// FILE *fopen(const char *filename, const char *mode);
Chris Lattnerb408b122002-03-29 03:57:15 +0000558GenericValue lle_X_fopen(FunctionType *M, const vector<GenericValue> &Args) {
Chris Lattner665ee882002-03-08 22:51:07 +0000559 assert(Args.size() == 2);
Chris Lattnerb1118742003-01-13 00:59:47 +0000560 return PTOGV(fopen((const char *)GVTOP(Args[0]),
561 (const char *)GVTOP(Args[1])));
Chris Lattner665ee882002-03-08 22:51:07 +0000562}
563
564// int fclose(FILE *F);
Chris Lattnerb408b122002-03-29 03:57:15 +0000565GenericValue lle_X_fclose(FunctionType *M, const vector<GenericValue> &Args) {
Chris Lattner665ee882002-03-08 22:51:07 +0000566 assert(Args.size() == 1);
567 GenericValue GV;
Chris Lattnerb1118742003-01-13 00:59:47 +0000568 GV.IntVal = fclose(getFILE(GVTOP(Args[0])));
Chris Lattner665ee882002-03-08 22:51:07 +0000569 return GV;
570}
571
Chris Lattner25f6f372002-11-08 19:10:26 +0000572// int feof(FILE *stream);
573GenericValue lle_X_feof(FunctionType *M, const vector<GenericValue> &Args) {
574 assert(Args.size() == 1);
575 GenericValue GV;
576
Chris Lattnerb1118742003-01-13 00:59:47 +0000577 GV.IntVal = feof(getFILE(GVTOP(Args[0])));
Chris Lattner25f6f372002-11-08 19:10:26 +0000578 return GV;
579}
580
Chris Lattner665ee882002-03-08 22:51:07 +0000581// size_t fread(void *ptr, size_t size, size_t nitems, FILE *stream);
Chris Lattnerb408b122002-03-29 03:57:15 +0000582GenericValue lle_X_fread(FunctionType *M, const vector<GenericValue> &Args) {
Chris Lattner665ee882002-03-08 22:51:07 +0000583 assert(Args.size() == 4);
Brian Gaeke3c66c352004-05-01 06:42:15 +0000584 size_t result;
Chris Lattner665ee882002-03-08 22:51:07 +0000585
Brian Gaeke3c66c352004-05-01 06:42:15 +0000586 result = fread((void*)GVTOP(Args[0]), GV_to_size_t (Args[1]),
587 GV_to_size_t (Args[2]), getFILE(GVTOP(Args[3])));
588 return size_t_to_GV (result);
Chris Lattner665ee882002-03-08 22:51:07 +0000589}
590
591// size_t fwrite(const void *ptr, size_t size, size_t nitems, FILE *stream);
Chris Lattnerb408b122002-03-29 03:57:15 +0000592GenericValue lle_X_fwrite(FunctionType *M, const vector<GenericValue> &Args) {
Chris Lattner665ee882002-03-08 22:51:07 +0000593 assert(Args.size() == 4);
Brian Gaeke3c66c352004-05-01 06:42:15 +0000594 size_t result;
Chris Lattner665ee882002-03-08 22:51:07 +0000595
Brian Gaeke3c66c352004-05-01 06:42:15 +0000596 result = fwrite((void*)GVTOP(Args[0]), GV_to_size_t (Args[1]),
597 GV_to_size_t (Args[2]), getFILE(GVTOP(Args[3])));
598 return size_t_to_GV (result);
Chris Lattner665ee882002-03-08 22:51:07 +0000599}
600
601// char *fgets(char *s, int n, FILE *stream);
Chris Lattnerb408b122002-03-29 03:57:15 +0000602GenericValue lle_X_fgets(FunctionType *M, const vector<GenericValue> &Args) {
Chris Lattner665ee882002-03-08 22:51:07 +0000603 assert(Args.size() == 3);
Chris Lattnerb1118742003-01-13 00:59:47 +0000604 return GVTOP(fgets((char*)GVTOP(Args[0]), Args[1].IntVal,
605 getFILE(GVTOP(Args[2]))));
Chris Lattner665ee882002-03-08 22:51:07 +0000606}
607
Chris Lattnera4479cd2002-11-07 19:33:50 +0000608// FILE *freopen(const char *path, const char *mode, FILE *stream);
609GenericValue lle_X_freopen(FunctionType *M, const vector<GenericValue> &Args) {
610 assert(Args.size() == 3);
Chris Lattnerb1118742003-01-13 00:59:47 +0000611 return PTOGV(freopen((char*)GVTOP(Args[0]), (char*)GVTOP(Args[1]),
612 getFILE(GVTOP(Args[2]))));
Chris Lattnera4479cd2002-11-07 19:33:50 +0000613}
614
Chris Lattner665ee882002-03-08 22:51:07 +0000615// int fflush(FILE *stream);
Chris Lattnerb408b122002-03-29 03:57:15 +0000616GenericValue lle_X_fflush(FunctionType *M, const vector<GenericValue> &Args) {
Chris Lattner665ee882002-03-08 22:51:07 +0000617 assert(Args.size() == 1);
618 GenericValue GV;
Chris Lattnerb1118742003-01-13 00:59:47 +0000619 GV.IntVal = fflush(getFILE(GVTOP(Args[0])));
Chris Lattner005cbce2002-10-02 21:12:13 +0000620 return GV;
621}
622
623// int getc(FILE *stream);
624GenericValue lle_X_getc(FunctionType *M, const vector<GenericValue> &Args) {
625 assert(Args.size() == 1);
626 GenericValue GV;
Chris Lattnerb1118742003-01-13 00:59:47 +0000627 GV.IntVal = getc(getFILE(GVTOP(Args[0])));
Chris Lattner665ee882002-03-08 22:51:07 +0000628 return GV;
629}
630
Chris Lattnerf87a1982003-04-23 19:20:50 +0000631// int _IO_getc(FILE *stream);
632GenericValue lle_X__IO_getc(FunctionType *F, const vector<GenericValue> &Args) {
633 return lle_X_getc(F, Args);
634}
635
Chris Lattnera5c0bfe2002-11-06 22:59:28 +0000636// int fputc(int C, FILE *stream);
637GenericValue lle_X_fputc(FunctionType *M, const vector<GenericValue> &Args) {
638 assert(Args.size() == 2);
639 GenericValue GV;
Chris Lattnerb1118742003-01-13 00:59:47 +0000640 GV.IntVal = fputc(Args[0].IntVal, getFILE(GVTOP(Args[1])));
Chris Lattnera5c0bfe2002-11-06 22:59:28 +0000641 return GV;
642}
643
644// int ungetc(int C, FILE *stream);
645GenericValue lle_X_ungetc(FunctionType *M, const vector<GenericValue> &Args) {
646 assert(Args.size() == 2);
647 GenericValue GV;
Chris Lattnerb1118742003-01-13 00:59:47 +0000648 GV.IntVal = ungetc(Args[0].IntVal, getFILE(GVTOP(Args[1])));
Chris Lattnera5c0bfe2002-11-06 22:59:28 +0000649 return GV;
650}
651
Brian Gaeke59108d32004-06-16 02:56:40 +0000652// int ferror (FILE *stream);
653GenericValue lle_X_ferror(FunctionType *M, const vector<GenericValue> &Args) {
654 assert(Args.size() == 1);
655 GenericValue GV;
656 GV.IntVal = ferror (getFILE(GVTOP(Args[0])));
657 return GV;
658}
659
Chris Lattnercf9b4f02002-11-06 23:05:03 +0000660// int fprintf(FILE *,sbyte *, ...) - a very rough implementation to make output
661// useful.
662GenericValue lle_X_fprintf(FunctionType *M, const vector<GenericValue> &Args) {
Chris Lattner9dbf6dd2003-04-21 22:43:20 +0000663 assert(Args.size() >= 2);
Chris Lattnercf9b4f02002-11-06 23:05:03 +0000664 char Buffer[10000];
665 vector<GenericValue> NewArgs;
Chris Lattnerb1118742003-01-13 00:59:47 +0000666 NewArgs.push_back(PTOGV(Buffer));
Chris Lattnercf9b4f02002-11-06 23:05:03 +0000667 NewArgs.insert(NewArgs.end(), Args.begin()+1, Args.end());
Chris Lattnerb1118742003-01-13 00:59:47 +0000668 GenericValue GV = lle_X_sprintf(M, NewArgs);
Chris Lattnercf9b4f02002-11-06 23:05:03 +0000669
Chris Lattnerb1118742003-01-13 00:59:47 +0000670 fputs(Buffer, getFILE(GVTOP(Args[0])));
Chris Lattnercf9b4f02002-11-06 23:05:03 +0000671 return GV;
672}
673
Chris Lattner7720c8e2001-09-10 04:50:17 +0000674} // End extern "C"
Chris Lattner4721f132001-10-30 20:28:00 +0000675
676
Chris Lattnerda82ed52003-05-08 16:18:31 +0000677void Interpreter::initializeExternalFunctions() {
Chris Lattner0f279b22001-11-03 10:15:32 +0000678 FuncNames["lle_Vb_putchar"] = lle_Vb_putchar;
679 FuncNames["lle_ii_putchar"] = lle_ii_putchar;
680 FuncNames["lle_VB_putchar"] = lle_VB_putchar;
Chris Lattner0f279b22001-11-03 10:15:32 +0000681 FuncNames["lle_X_exit"] = lle_X_exit;
Chris Lattner1ee34a52002-05-20 21:17:16 +0000682 FuncNames["lle_X_abort"] = lle_X_abort;
Chris Lattner0f279b22001-11-03 10:15:32 +0000683 FuncNames["lle_X_malloc"] = lle_X_malloc;
Chris Lattner957d62a2003-04-23 19:55:24 +0000684 FuncNames["lle_X_calloc"] = lle_X_calloc;
Chris Lattner0f279b22001-11-03 10:15:32 +0000685 FuncNames["lle_X_free"] = lle_X_free;
Chris Lattner782b9392001-11-26 18:18:18 +0000686 FuncNames["lle_X_atoi"] = lle_X_atoi;
Chris Lattner0f279b22001-11-03 10:15:32 +0000687 FuncNames["lle_X_pow"] = lle_X_pow;
Chris Lattner34dd24b2002-02-18 19:06:25 +0000688 FuncNames["lle_X_exp"] = lle_X_exp;
Chris Lattner1b600142001-11-13 05:46:08 +0000689 FuncNames["lle_X_log"] = lle_X_log;
Chris Lattner782b9392001-11-26 18:18:18 +0000690 FuncNames["lle_X_floor"] = lle_X_floor;
691 FuncNames["lle_X_srand"] = lle_X_srand;
Reid Spencerabec8f92004-10-27 23:03:44 +0000692 FuncNames["lle_X_rand"] = lle_X_rand;
693#ifdef HAVE_RAND48
Chris Lattner1b600142001-11-13 05:46:08 +0000694 FuncNames["lle_X_drand48"] = lle_X_drand48;
695 FuncNames["lle_X_srand48"] = lle_X_srand48;
696 FuncNames["lle_X_lrand48"] = lle_X_lrand48;
Reid Spencerabec8f92004-10-27 23:03:44 +0000697#endif
Chris Lattnerc063d382001-11-06 21:52:18 +0000698 FuncNames["lle_X_sqrt"] = lle_X_sqrt;
Chris Lattnerb1118742003-01-13 00:59:47 +0000699 FuncNames["lle_X_puts"] = lle_X_puts;
Chris Lattner0f279b22001-11-03 10:15:32 +0000700 FuncNames["lle_X_printf"] = lle_X_printf;
Chris Lattnere7c6f722001-12-13 00:43:47 +0000701 FuncNames["lle_X_sprintf"] = lle_X_sprintf;
Chris Lattner665ee882002-03-08 22:51:07 +0000702 FuncNames["lle_X_sscanf"] = lle_X_sscanf;
Chris Lattnerf9a88b62003-03-31 22:12:37 +0000703 FuncNames["lle_X_scanf"] = lle_X_scanf;
Chris Lattner295fe672002-01-23 21:38:07 +0000704 FuncNames["lle_i_clock"] = lle_i_clock;
Chris Lattner957d62a2003-04-23 19:55:24 +0000705
706 FuncNames["lle_X_strcmp"] = lle_X_strcmp;
707 FuncNames["lle_X_strcat"] = lle_X_strcat;
708 FuncNames["lle_X_strcpy"] = lle_X_strcpy;
709 FuncNames["lle_X_strlen"] = lle_X_strlen;
Chris Lattnerda82ed52003-05-08 16:18:31 +0000710 FuncNames["lle_X___strdup"] = lle_X___strdup;
Chris Lattner957d62a2003-04-23 19:55:24 +0000711 FuncNames["lle_X_memset"] = lle_X_memset;
Chris Lattner5f311a72003-04-23 20:23:16 +0000712 FuncNames["lle_X_memcpy"] = lle_X_memcpy;
Chris Lattner957d62a2003-04-23 19:55:24 +0000713
Chris Lattner665ee882002-03-08 22:51:07 +0000714 FuncNames["lle_X_fopen"] = lle_X_fopen;
715 FuncNames["lle_X_fclose"] = lle_X_fclose;
Chris Lattner25f6f372002-11-08 19:10:26 +0000716 FuncNames["lle_X_feof"] = lle_X_feof;
Chris Lattner665ee882002-03-08 22:51:07 +0000717 FuncNames["lle_X_fread"] = lle_X_fread;
718 FuncNames["lle_X_fwrite"] = lle_X_fwrite;
719 FuncNames["lle_X_fgets"] = lle_X_fgets;
720 FuncNames["lle_X_fflush"] = lle_X_fflush;
Chris Lattnera5c0bfe2002-11-06 22:59:28 +0000721 FuncNames["lle_X_fgetc"] = lle_X_getc;
Chris Lattner005cbce2002-10-02 21:12:13 +0000722 FuncNames["lle_X_getc"] = lle_X_getc;
Chris Lattnerf87a1982003-04-23 19:20:50 +0000723 FuncNames["lle_X__IO_getc"] = lle_X__IO_getc;
Chris Lattnera5c0bfe2002-11-06 22:59:28 +0000724 FuncNames["lle_X_fputc"] = lle_X_fputc;
725 FuncNames["lle_X_ungetc"] = lle_X_ungetc;
Chris Lattnercf9b4f02002-11-06 23:05:03 +0000726 FuncNames["lle_X_fprintf"] = lle_X_fprintf;
Chris Lattnera4479cd2002-11-07 19:33:50 +0000727 FuncNames["lle_X_freopen"] = lle_X_freopen;
Chris Lattner4721f132001-10-30 20:28:00 +0000728}
Brian Gaeked0fde302003-11-11 22:41:34 +0000729