lib/ExecutionEngine/Interpreter/ExternalFunctions.cpp

   1 //===-- ExternalFunctions.cpp - Implement External Functions --------------===//
   2 //
   3 //  This file contains both code to deal with invoking "external" functions, but
   4 //  also contains code that implements "exported" external functions.
   5 //
   6 //  External functions in LLI are implemented by dlopen'ing the lli executable
   7 //  and using dlsym to look op the functions that we want to invoke.  If a
   8 //  function is found, then the arguments are mangled and passed in to the
   9 //  function call.
  10 //
  11 //===----------------------------------------------------------------------===//
  12
  13 #include "Interpreter.h"
  14 #include "ExecutionAnnotations.h"
  15 #include "llvm/Module.h"
  16 #include "llvm/DerivedTypes.h"
  17 #include "llvm/SymbolTable.h"
  18 #include "llvm/Target/TargetData.h"
  19 #include <map>
  20 #include <dlfcn.h>
  21 #include <link.h>
  22 #include <math.h>
  23 #include <stdio.h>
  24 using std::vector;
  25 using std::cout;
  26
  27 typedef GenericValue (*ExFunc)(FunctionType *, const vector<GenericValue> &);
  28 static std::map<const Function *, ExFunc> Functions;
  29 static std::map<std::string, ExFunc> FuncNames;
  30
  31 static Interpreter *TheInterpreter;
  32
  33 // getCurrentExecutablePath() - Return the directory that the lli executable
  34 // lives in.
  35 //
  36 std::string Interpreter::getCurrentExecutablePath() const {
  37   Dl_info Info;
  38   if (dladdr(&TheInterpreter, &Info) == 0) return "";
  39
  40   std::string LinkAddr(Info.dli_fname);
  41   unsigned SlashPos = LinkAddr.rfind('/');
  42   if (SlashPos != std::string::npos)
  43     LinkAddr.resize(SlashPos);    // Trim the executable name off...
  44
  45   return LinkAddr;
  46 }
  47
  48
  49 static char getTypeID(const Type *Ty) {
  50   switch (Ty->getPrimitiveID()) {
  51   case Type::VoidTyID:    return 'V';
  52   case Type::BoolTyID:    return 'o';
  53   case Type::UByteTyID:   return 'B';
  54   case Type::SByteTyID:   return 'b';
  55   case Type::UShortTyID:  return 'S';
  56   case Type::ShortTyID:   return 's';
  57   case Type::UIntTyID:    return 'I';
  58   case Type::IntTyID:     return 'i';
  59   case Type::ULongTyID:   return 'L';
  60   case Type::LongTyID:    return 'l';
  61   case Type::FloatTyID:   return 'F';
  62   case Type::DoubleTyID:  return 'D';
  63   case Type::PointerTyID: return 'P';
  64   case Type::FunctionTyID:  return 'M';
  65   case Type::StructTyID:  return 'T';
  66   case Type::ArrayTyID:   return 'A';
  67   case Type::OpaqueTyID:  return 'O';
  68   default: return 'U';
  69   }
  70 }
  71
  72 static ExFunc lookupFunction(const Function *M) {
  73   // Function not found, look it up... start by figuring out what the
  74   // composite function name should be.
  75   std::string ExtName = "lle_";
  76   const FunctionType *MT = M->getFunctionType();
  77   for (unsigned i = 0; const Type *Ty = MT->getContainedType(i); ++i)
  78     ExtName += getTypeID(Ty);
  79   ExtName += "_" + M->getName();
  80
  81   //cout << "Tried: '" << ExtName << "'\n";
  82   ExFunc FnPtr = FuncNames[ExtName];
  83   if (FnPtr == 0)
  84     FnPtr = (ExFunc)dlsym(RTLD_DEFAULT, ExtName.c_str());
  85   if (FnPtr == 0)
  86     FnPtr = FuncNames["lle_X_"+M->getName()];
  87   if (FnPtr == 0)  // Try calling a generic function... if it exists...
  88     FnPtr = (ExFunc)dlsym(RTLD_DEFAULT, ("lle_X_"+M->getName()).c_str());
  89   if (FnPtr != 0)
  90     Functions.insert(std::make_pair(M, FnPtr));  // Cache for later
  91   return FnPtr;
  92 }
  93
  94 GenericValue Interpreter::callExternalMethod(Function *M,
  95                                          const vector<GenericValue> &ArgVals) {
  96   TheInterpreter = this;
  97
  98   // Do a lookup to see if the function is in our cache... this should just be a
  99   // defered annotation!
 100   std::map<const Function *, ExFunc>::iterator FI = Functions.find(M);
 101   ExFunc Fn = (FI == Functions.end()) ? lookupFunction(M) : FI->second;
 102   if (Fn == 0) {
 103     cout << "Tried to execute an unknown external function: "
 104          << M->getType()->getDescription() << " " << M->getName() << "\n";
 105     return GenericValue();
 106   }
 107
 108   // TODO: FIXME when types are not const!
 109   GenericValue Result = Fn(const_cast<FunctionType*>(M->getFunctionType()),
 110                            ArgVals);
 111   return Result;
 112 }
 113
 114
 115 //===----------------------------------------------------------------------===//
 116 //  Functions "exported" to the running application...
 117 //
 118 extern "C" {  // Don't add C++ manglings to llvm mangling :)
 119
 120 // Implement void printstr([ubyte {x N}] *)
 121 GenericValue lle_VP_printstr(FunctionType *M,
 122                              const vector<GenericValue> &ArgVal){
 123   assert(ArgVal.size() == 1 && "printstr only takes one argument!");
 124   cout << (char*)GVTOP(ArgVal[0]);
 125   return GenericValue();
 126 }
 127
 128 // Implement 'void print(X)' for every type...
 129 GenericValue lle_X_print(FunctionType *M, const vector<GenericValue> &ArgVals) {
 130   assert(ArgVals.size() == 1 && "generic print only takes one argument!");
 131
 132   Interpreter::print(M->getParamTypes()[0], ArgVals[0]);
 133   return GenericValue();
 134 }
 135
 136 // Implement 'void printVal(X)' for every type...
 137 GenericValue lle_X_printVal(FunctionType *M,
 138                             const vector<GenericValue> &ArgVal) {
 139   assert(ArgVal.size() == 1 && "generic print only takes one argument!");
 140
 141   // Specialize print([ubyte {x N} ] *) and print(sbyte *)
 142   if (const PointerType *PTy =
 143       dyn_cast<PointerType>(M->getParamTypes()[0].get()))
 144     if (PTy->getElementType() == Type::SByteTy ||
 145         isa<ArrayType>(PTy->getElementType())) {
 146       return lle_VP_printstr(M, ArgVal);
 147     }
 148
 149   Interpreter::printValue(M->getParamTypes()[0], ArgVal[0]);
 150   return GenericValue();
 151 }
 152
 153 // Implement 'void printString(X)'
 154 // Argument must be [ubyte {x N} ] * or sbyte *
 155 GenericValue lle_X_printString(FunctionType *M,
 156                                const vector<GenericValue> &ArgVal) {
 157   assert(ArgVal.size() == 1 && "generic print only takes one argument!");
 158   return lle_VP_printstr(M, ArgVal);
 159 }
 160
 161 // Implement 'void print<TYPE>(X)' for each primitive type or pointer type
 162 #define PRINT_TYPE_FUNC(TYPENAME,TYPEID) \
 163   GenericValue lle_X_print##TYPENAME(FunctionType *M,\
 164                                      const vector<GenericValue> &ArgVal) {\
 165     assert(ArgVal.size() == 1 && "generic print only takes one argument!");\
 166     assert(M->getParamTypes()[0].get()->getPrimitiveID() == Type::TYPEID);\
 167     Interpreter::printValue(M->getParamTypes()[0], ArgVal[0]);\
 168     return GenericValue();\
 169   }
 170
 171 PRINT_TYPE_FUNC(SByte,   SByteTyID)
 172 PRINT_TYPE_FUNC(UByte,   UByteTyID)
 173 PRINT_TYPE_FUNC(Short,   ShortTyID)
 174 PRINT_TYPE_FUNC(UShort,  UShortTyID)
 175 PRINT_TYPE_FUNC(Int,     IntTyID)
 176 PRINT_TYPE_FUNC(UInt,    UIntTyID)
 177 PRINT_TYPE_FUNC(Long,    LongTyID)
 178 PRINT_TYPE_FUNC(ULong,   ULongTyID)
 179 PRINT_TYPE_FUNC(Float,   FloatTyID)
 180 PRINT_TYPE_FUNC(Double,  DoubleTyID)
 181 PRINT_TYPE_FUNC(Pointer, PointerTyID)
 182
 183
 184 // void putchar(sbyte)
 185 GenericValue lle_Vb_putchar(FunctionType *M, const vector<GenericValue> &Args) {
 186   cout << Args[0].SByteVal;
 187   return GenericValue();
 188 }
 189
 190 // int putchar(int)
 191 GenericValue lle_ii_putchar(FunctionType *M, const vector<GenericValue> &Args) {
 192   cout << ((char)Args[0].IntVal) << std::flush;
 193   return Args[0];
 194 }
 195
 196 // void putchar(ubyte)
 197 GenericValue lle_VB_putchar(FunctionType *M, const vector<GenericValue> &Args) {
 198   cout << Args[0].SByteVal << std::flush;
 199   return Args[0];
 200 }
 201
 202 // void __main()
 203 GenericValue lle_V___main(FunctionType *M, const vector<GenericValue> &Args) {
 204   return GenericValue();
 205 }
 206
 207 // void exit(int)
 208 GenericValue lle_X_exit(FunctionType *M, const vector<GenericValue> &Args) {
 209   TheInterpreter->exitCalled(Args[0]);
 210   return GenericValue();
 211 }
 212
 213 // void abort(void)
 214 GenericValue lle_X_abort(FunctionType *M, const vector<GenericValue> &Args) {
 215   std::cerr << "***PROGRAM ABORTED***!\n";
 216   GenericValue GV;
 217   GV.IntVal = 1;
 218   TheInterpreter->exitCalled(GV);
 219   return GenericValue();
 220 }
 221
 222 // void *malloc(uint)
 223 GenericValue lle_X_malloc(FunctionType *M, const vector<GenericValue> &Args) {
 224   assert(Args.size() == 1 && "Malloc expects one argument!");
 225   return PTOGV(malloc(Args[0].UIntVal));
 226 }
 227
 228 // void *calloc(uint, uint)
 229 GenericValue lle_X_calloc(FunctionType *M, const vector<GenericValue> &Args) {
 230   assert(Args.size() == 2 && "calloc expects two arguments!");
 231   return PTOGV(calloc(Args[0].UIntVal, Args[1].UIntVal));
 232 }
 233
 234 // void free(void *)
 235 GenericValue lle_X_free(FunctionType *M, const vector<GenericValue> &Args) {
 236   assert(Args.size() == 1);
 237   free(GVTOP(Args[0]));
 238   return GenericValue();
 239 }
 240
 241 // int atoi(char *)
 242 GenericValue lle_X_atoi(FunctionType *M, const vector<GenericValue> &Args) {
 243   assert(Args.size() == 1);
 244   GenericValue GV;
 245   GV.IntVal = atoi((char*)GVTOP(Args[0]));
 246   return GV;
 247 }
 248
 249 // double pow(double, double)
 250 GenericValue lle_X_pow(FunctionType *M, const vector<GenericValue> &Args) {
 251   assert(Args.size() == 2);
 252   GenericValue GV;
 253   GV.DoubleVal = pow(Args[0].DoubleVal, Args[1].DoubleVal);
 254   return GV;
 255 }
 256
 257 // double exp(double)
 258 GenericValue lle_X_exp(FunctionType *M, const vector<GenericValue> &Args) {
 259   assert(Args.size() == 1);
 260   GenericValue GV;
 261   GV.DoubleVal = exp(Args[0].DoubleVal);
 262   return GV;
 263 }
 264
 265 // double sqrt(double)
 266 GenericValue lle_X_sqrt(FunctionType *M, const vector<GenericValue> &Args) {
 267   assert(Args.size() == 1);
 268   GenericValue GV;
 269   GV.DoubleVal = sqrt(Args[0].DoubleVal);
 270   return GV;
 271 }
 272
 273 // double log(double)
 274 GenericValue lle_X_log(FunctionType *M, const vector<GenericValue> &Args) {
 275   assert(Args.size() == 1);
 276   GenericValue GV;
 277   GV.DoubleVal = log(Args[0].DoubleVal);
 278   return GV;
 279 }
 280
 281 // int isnan(double value);
 282 GenericValue lle_X_isnan(FunctionType *F, const vector<GenericValue> &Args) {
 283   assert(Args.size() == 1);
 284   GenericValue GV;
 285   GV.IntVal = isnan(Args[0].DoubleVal);
 286   return GV;
 287 }
 288
 289 // double floor(double)
 290 GenericValue lle_X_floor(FunctionType *M, const vector<GenericValue> &Args) {
 291   assert(Args.size() == 1);
 292   GenericValue GV;
 293   GV.DoubleVal = floor(Args[0].DoubleVal);
 294   return GV;
 295 }
 296
 297 // double drand48()
 298 GenericValue lle_X_drand48(FunctionType *M, const vector<GenericValue> &Args) {
 299   assert(Args.size() == 0);
 300   GenericValue GV;
 301   GV.DoubleVal = drand48();
 302   return GV;
 303 }
 304
 305 // long lrand48()
 306 GenericValue lle_X_lrand48(FunctionType *M, const vector<GenericValue> &Args) {
 307   assert(Args.size() == 0);
 308   GenericValue GV;
 309   GV.IntVal = lrand48();
 310   return GV;
 311 }
 312
 313 // void srand48(long)
 314 GenericValue lle_X_srand48(FunctionType *M, const vector<GenericValue> &Args) {
 315   assert(Args.size() == 1);
 316   srand48(Args[0].IntVal);
 317   return GenericValue();
 318 }
 319
 320 // void srand(uint)
 321 GenericValue lle_X_srand(FunctionType *M, const vector<GenericValue> &Args) {
 322   assert(Args.size() == 1);
 323   srand(Args[0].UIntVal);
 324   return GenericValue();
 325 }
 326
 327 // int puts(const char*)
 328 GenericValue lle_X_puts(FunctionType *M, const vector<GenericValue> &Args) {
 329   assert(Args.size() == 1);
 330   GenericValue GV;
 331   GV.IntVal = puts((char*)GVTOP(Args[0]));
 332   return GV;
 333 }
 334
 335 // int sprintf(sbyte *, sbyte *, ...) - a very rough implementation to make
 336 // output useful.
 337 GenericValue lle_X_sprintf(FunctionType *M, const vector<GenericValue> &Args) {
 338   char *OutputBuffer = (char *)GVTOP(Args[0]);
 339   const char *FmtStr = (const char *)GVTOP(Args[1]);
 340   unsigned ArgNo = 2;
 341
 342   // printf should return # chars printed.  This is completely incorrect, but
 343   // close enough for now.
 344   GenericValue GV; GV.IntVal = strlen(FmtStr);
 345   while (1) {
 346     switch (*FmtStr) {
 347     case 0: return GV;             // Null terminator...
 348     default:                       // Normal nonspecial character
 349       sprintf(OutputBuffer++, "%c", *FmtStr++);
 350       break;
 351     case '\\': {                   // Handle escape codes
 352       sprintf(OutputBuffer, "%c%c", *FmtStr, *(FmtStr+1));
 353       FmtStr += 2; OutputBuffer += 2;
 354       break;
 355     }
 356     case '%': {                    // Handle format specifiers
 357       char FmtBuf[100] = "", Buffer[1000] = "";
 358       char *FB = FmtBuf;
 359       *FB++ = *FmtStr++;
 360       char Last = *FB++ = *FmtStr++;
 361       unsigned HowLong = 0;
 362       while (Last != 'c' && Last != 'd' && Last != 'i' && Last != 'u' &&
 363              Last != 'o' && Last != 'x' && Last != 'X' && Last != 'e' &&
 364              Last != 'E' && Last != 'g' && Last != 'G' && Last != 'f' &&
 365              Last != 'p' && Last != 's' && Last != '%') {
 366         if (Last == 'l' || Last == 'L') HowLong++;  // Keep track of l's
 367         Last = *FB++ = *FmtStr++;
 368       }
 369       *FB = 0;
 370
 371       switch (Last) {
 372       case '%':
 373         sprintf(Buffer, FmtBuf); break;
 374       case 'c':
 375         sprintf(Buffer, FmtBuf, Args[ArgNo++].IntVal); break;
 376       case 'd': case 'i':
 377       case 'u': case 'o':
 378       case 'x': case 'X':
 379         if (HowLong >= 1) {
 380           if (HowLong == 1) {
 381             // Make sure we use %lld with a 64 bit argument because we might be
 382             // compiling LLI on a 32 bit compiler.
 383             unsigned Size = strlen(FmtBuf);
 384             FmtBuf[Size] = FmtBuf[Size-1];
 385             FmtBuf[Size+1] = 0;
 386             FmtBuf[Size-1] = 'l';
 387           }
 388           sprintf(Buffer, FmtBuf, Args[ArgNo++].ULongVal);
 389         } else
 390           sprintf(Buffer, FmtBuf, Args[ArgNo++].IntVal); break;
 391       case 'e': case 'E': case 'g': case 'G': case 'f':
 392         sprintf(Buffer, FmtBuf, Args[ArgNo++].DoubleVal); break;
 393       case 'p':
 394         sprintf(Buffer, FmtBuf, (void*)GVTOP(Args[ArgNo++])); break;
 395       case 's':
 396         sprintf(Buffer, FmtBuf, (char*)GVTOP(Args[ArgNo++])); break;
 397       default:  cout << "<unknown printf code '" << *FmtStr << "'!>";
 398         ArgNo++; break;
 399       }
 400       strcpy(OutputBuffer, Buffer);
 401       OutputBuffer += strlen(Buffer);
 402       }
 403       break;
 404     }
 405   }
 406 }
 407
 408 // int printf(sbyte *, ...) - a very rough implementation to make output useful.
 409 GenericValue lle_X_printf(FunctionType *M, const vector<GenericValue> &Args) {
 410   char Buffer[10000];
 411   vector<GenericValue> NewArgs;
 412   NewArgs.push_back(PTOGV(Buffer));
 413   NewArgs.insert(NewArgs.end(), Args.begin(), Args.end());
 414   GenericValue GV = lle_X_sprintf(M, NewArgs);
 415   cout << Buffer;
 416   return GV;
 417 }
 418
 419 static void ByteswapSCANFResults(const char *Fmt, void *Arg0, void *Arg1,
 420                                  void *Arg2, void *Arg3, void *Arg4, void *Arg5,
 421                                  void *Arg6, void *Arg7, void *Arg8) {
 422   void *Args[] = { Arg0, Arg1, Arg2, Arg3, Arg4, Arg5, Arg6, Arg7, Arg8, 0 };
 423
 424   // Loop over the format string, munging read values as appropriate (performs
 425   // byteswaps as neccesary).
 426   unsigned ArgNo = 0;
 427   while (*Fmt) {
 428     if (*Fmt++ == '%') {
 429       // Read any flag characters that may be present...
 430       bool Suppress = false;
 431       bool Half = false;
 432       bool Long = false;
 433       bool LongLong = false;  // long long or long double
 434
 435       while (1) {
 436         switch (*Fmt++) {
 437         case '*': Suppress = true; break;
 438         case 'a': /*Allocate = true;*/ break;  // We don't need to track this
 439         case 'h': Half = true; break;
 440         case 'l': Long = true; break;
 441         case 'q':
 442         case 'L': LongLong = true; break;
 443         default:
 444           if (Fmt[-1] > '9' || Fmt[-1] < '0')   // Ignore field width specs
 445             goto Out;
 446         }
 447       }
 448     Out:
 449
 450       // Read the conversion character
 451       if (!Suppress && Fmt[-1] != '%') { // Nothing to do?
 452         unsigned Size = 0;
 453         const Type *Ty = 0;
 454
 455         switch (Fmt[-1]) {
 456         case 'i': case 'o': case 'u': case 'x': case 'X': case 'n': case 'p':
 457         case 'd':
 458           if (Long || LongLong) {
 459             Size = 8; Ty = Type::ULongTy;
 460           } else if (Half) {
 461             Size = 4; Ty = Type::UShortTy;
 462           } else {
 463             Size = 4; Ty = Type::UIntTy;
 464           }
 465           break;
 466
 467         case 'e': case 'g': case 'E':
 468         case 'f':
 469           if (Long || LongLong) {
 470             Size = 8; Ty = Type::DoubleTy;
 471           } else {
 472             Size = 4; Ty = Type::FloatTy;
 473           }
 474           break;
 475
 476         case 's': case 'c': case '[':  // No byteswap needed
 477           Size = 1;
 478           Ty = Type::SByteTy;
 479           break;
 480
 481         default: break;
 482         }
 483
 484         if (Size) {
 485           GenericValue GV;
 486           void *Arg = Args[ArgNo++];
 487           memcpy(&GV, Arg, Size);
 488           TheInterpreter->StoreValueToMemory(GV, (GenericValue*)Arg, Ty);
 489         }
 490       }
 491     }
 492   }
 493 }
 494
 495 // int sscanf(const char *format, ...);
 496 GenericValue lle_X_sscanf(FunctionType *M, const vector<GenericValue> &args) {
 497   assert(args.size() < 10 && "Only handle up to 10 args to sscanf right now!");
 498
 499   char *Args[10];
 500   for (unsigned i = 0; i < args.size(); ++i)
 501     Args[i] = (char*)GVTOP(args[i]);
 502
 503   GenericValue GV;
 504   GV.IntVal = sscanf(Args[0], Args[1], Args[2], Args[3], Args[4],
 505                      Args[5], Args[6], Args[7], Args[8], Args[9]);
 506   ByteswapSCANFResults(Args[1], Args[2], Args[3], Args[4],
 507                        Args[5], Args[6], Args[7], Args[8], Args[9], 0);
 508   return GV;
 509 }
 510
 511 // int scanf(const char *format, ...);
 512 GenericValue lle_X_scanf(FunctionType *M, const vector<GenericValue> &args) {
 513   assert(args.size() < 10 && "Only handle up to 10 args to scanf right now!");
 514
 515   char *Args[10];
 516   for (unsigned i = 0; i < args.size(); ++i)
 517     Args[i] = (char*)GVTOP(args[i]);
 518
 519   GenericValue GV;
 520   GV.IntVal = scanf(Args[0], Args[1], Args[2], Args[3], Args[4],
 521                     Args[5], Args[6], Args[7], Args[8], Args[9]);
 522   ByteswapSCANFResults(Args[0], Args[1], Args[2], Args[3], Args[4],
 523                        Args[5], Args[6], Args[7], Args[8], Args[9]);
 524   return GV;
 525 }
 526
 527
 528 // int clock(void) - Profiling implementation
 529 GenericValue lle_i_clock(FunctionType *M, const vector<GenericValue> &Args) {
 530   extern int clock(void);
 531   GenericValue GV; GV.IntVal = clock();
 532   return GV;
 533 }
 534
 535
 536 //===----------------------------------------------------------------------===//
 537 // String Functions...
 538 //===----------------------------------------------------------------------===//
 539
 540 // int strcmp(const char *S1, const char *S2);
 541 GenericValue lle_X_strcmp(FunctionType *M, const vector<GenericValue> &Args) {
 542   assert(Args.size() == 2);
 543   GenericValue Ret;
 544   Ret.IntVal = strcmp((char*)GVTOP(Args[0]), (char*)GVTOP(Args[1]));
 545   return Ret;
 546 }
 547
 548 // char *strcat(char *Dest, const char *src);
 549 GenericValue lle_X_strcat(FunctionType *M, const vector<GenericValue> &Args) {
 550   assert(Args.size() == 2);
 551   return PTOGV(strcat((char*)GVTOP(Args[0]), (char*)GVTOP(Args[1])));
 552 }
 553
 554 // char *strcpy(char *Dest, const char *src);
 555 GenericValue lle_X_strcpy(FunctionType *M, const vector<GenericValue> &Args) {
 556   assert(Args.size() == 2);
 557   return PTOGV(strcpy((char*)GVTOP(Args[0]), (char*)GVTOP(Args[1])));
 558 }
 559
 560 // long strlen(const char *src);
 561 GenericValue lle_X_strlen(FunctionType *M, const vector<GenericValue> &Args) {
 562   assert(Args.size() == 1);
 563   GenericValue Ret;
 564   Ret.LongVal = strlen((char*)GVTOP(Args[0]));
 565   return Ret;
 566 }
 567
 568 // void *memset(void *S, int C, size_t N)
 569 GenericValue lle_X_memset(FunctionType *M, const vector<GenericValue> &Args) {
 570   assert(Args.size() == 3);
 571   return PTOGV(memset(GVTOP(Args[0]), Args[1].IntVal, Args[2].UIntVal));
 572 }
 573
 574 // void *memcpy(void *Dest, void *src, size_t Size);
 575 GenericValue lle_X_memcpy(FunctionType *M, const vector<GenericValue> &Args) {
 576   assert(Args.size() == 3);
 577   return PTOGV(memcpy((char*)GVTOP(Args[0]), (char*)GVTOP(Args[1]),
 578                       Args[2].UIntVal));
 579 }
 580
 581 //===----------------------------------------------------------------------===//
 582 // IO Functions...
 583 //===----------------------------------------------------------------------===//
 584
 585 // getFILE - Turn a pointer in the host address space into a legit pointer in
 586 // the interpreter address space.  For the most part, this is an identity
 587 // transformation, but if the program refers to stdio, stderr, stdin then they
 588 // have pointers that are relative to the __iob array.  If this is the case,
 589 // change the FILE into the REAL stdio stream.
 590 //
 591 static FILE *getFILE(void *Ptr) {
 592   static Module *LastMod = 0;
 593   static PointerTy IOBBase = 0;
 594   static unsigned FILESize;
 595
 596   if (LastMod != &TheInterpreter->getModule()) { // Module change or initialize?
 597     Module *M = LastMod = &TheInterpreter->getModule();
 598
 599     // Check to see if the currently loaded module contains an __iob symbol...
 600     GlobalVariable *IOB = 0;
 601     SymbolTable &ST = M->getSymbolTable();
 602     for (SymbolTable::iterator I = ST.begin(), E = ST.end(); I != E; ++I) {
 603       SymbolTable::VarMap &M = I->second;
 604       for (SymbolTable::VarMap::iterator J = M.begin(), E = M.end();
 605            J != E; ++J)
 606         if (J->first == "__iob")
 607           if ((IOB = dyn_cast<GlobalVariable>(J->second)))
 608             break;
 609       if (IOB) break;
 610     }
 611
 612 #if 0   /// FIXME!  __iob support for LLI
 613     // If we found an __iob symbol now, find out what the actual address it's
 614     // held in is...
 615     if (IOB) {
 616       // Get the address the array lives in...
 617       GlobalAddress *Address =
 618         (GlobalAddress*)IOB->getOrCreateAnnotation(GlobalAddressAID);
 619       IOBBase = (PointerTy)(GenericValue*)Address->Ptr;
 620
 621       // Figure out how big each element of the array is...
 622       const ArrayType *AT =
 623         dyn_cast<ArrayType>(IOB->getType()->getElementType());
 624       if (AT)
 625         FILESize = TD.getTypeSize(AT->getElementType());
 626       else
 627         FILESize = 16*8;  // Default size
 628     }
 629 #endif
 630   }
 631
 632   // Check to see if this is a reference to __iob...
 633   if (IOBBase) {
 634     unsigned FDNum = ((unsigned long)Ptr-IOBBase)/FILESize;
 635     if (FDNum == 0)
 636       return stdin;
 637     else if (FDNum == 1)
 638       return stdout;
 639     else if (FDNum == 2)
 640       return stderr;
 641   }
 642
 643   return (FILE*)Ptr;
 644 }
 645
 646
 647 // FILE *fopen(const char *filename, const char *mode);
 648 GenericValue lle_X_fopen(FunctionType *M, const vector<GenericValue> &Args) {
 649   assert(Args.size() == 2);
 650   return PTOGV(fopen((const char *)GVTOP(Args[0]),
 651                      (const char *)GVTOP(Args[1])));
 652 }
 653
 654 // int fclose(FILE *F);
 655 GenericValue lle_X_fclose(FunctionType *M, const vector<GenericValue> &Args) {
 656   assert(Args.size() == 1);
 657   GenericValue GV;
 658   GV.IntVal = fclose(getFILE(GVTOP(Args[0])));
 659   return GV;
 660 }
 661
 662 // int feof(FILE *stream);
 663 GenericValue lle_X_feof(FunctionType *M, const vector<GenericValue> &Args) {
 664   assert(Args.size() == 1);
 665   GenericValue GV;
 666
 667   GV.IntVal = feof(getFILE(GVTOP(Args[0])));
 668   return GV;
 669 }
 670
 671 // size_t fread(void *ptr, size_t size, size_t nitems, FILE *stream);
 672 GenericValue lle_X_fread(FunctionType *M, const vector<GenericValue> &Args) {
 673   assert(Args.size() == 4);
 674   GenericValue GV;
 675
 676   GV.UIntVal = fread((void*)GVTOP(Args[0]), Args[1].UIntVal,
 677                      Args[2].UIntVal, getFILE(GVTOP(Args[3])));
 678   return GV;
 679 }
 680
 681 // size_t fwrite(const void *ptr, size_t size, size_t nitems, FILE *stream);
 682 GenericValue lle_X_fwrite(FunctionType *M, const vector<GenericValue> &Args) {
 683   assert(Args.size() == 4);
 684   GenericValue GV;
 685
 686   GV.UIntVal = fwrite((void*)GVTOP(Args[0]), Args[1].UIntVal,
 687                       Args[2].UIntVal, getFILE(GVTOP(Args[3])));
 688   return GV;
 689 }
 690
 691 // char *fgets(char *s, int n, FILE *stream);
 692 GenericValue lle_X_fgets(FunctionType *M, const vector<GenericValue> &Args) {
 693   assert(Args.size() == 3);
 694   return GVTOP(fgets((char*)GVTOP(Args[0]), Args[1].IntVal,
 695                      getFILE(GVTOP(Args[2]))));
 696 }
 697
 698 // FILE *freopen(const char *path, const char *mode, FILE *stream);
 699 GenericValue lle_X_freopen(FunctionType *M, const vector<GenericValue> &Args) {
 700   assert(Args.size() == 3);
 701   return PTOGV(freopen((char*)GVTOP(Args[0]), (char*)GVTOP(Args[1]),
 702                        getFILE(GVTOP(Args[2]))));
 703 }
 704
 705 // int fflush(FILE *stream);
 706 GenericValue lle_X_fflush(FunctionType *M, const vector<GenericValue> &Args) {
 707   assert(Args.size() == 1);
 708   GenericValue GV;
 709   GV.IntVal = fflush(getFILE(GVTOP(Args[0])));
 710   return GV;
 711 }
 712
 713 // int getc(FILE *stream);
 714 GenericValue lle_X_getc(FunctionType *M, const vector<GenericValue> &Args) {
 715   assert(Args.size() == 1);
 716   GenericValue GV;
 717   GV.IntVal = getc(getFILE(GVTOP(Args[0])));
 718   return GV;
 719 }
 720
 721 // int _IO_getc(FILE *stream);
 722 GenericValue lle_X__IO_getc(FunctionType *F, const vector<GenericValue> &Args) {
 723   return lle_X_getc(F, Args);
 724 }
 725
 726 // int fputc(int C, FILE *stream);
 727 GenericValue lle_X_fputc(FunctionType *M, const vector<GenericValue> &Args) {
 728   assert(Args.size() == 2);
 729   GenericValue GV;
 730   GV.IntVal = fputc(Args[0].IntVal, getFILE(GVTOP(Args[1])));
 731   return GV;
 732 }
 733
 734 // int ungetc(int C, FILE *stream);
 735 GenericValue lle_X_ungetc(FunctionType *M, const vector<GenericValue> &Args) {
 736   assert(Args.size() == 2);
 737   GenericValue GV;
 738   GV.IntVal = ungetc(Args[0].IntVal, getFILE(GVTOP(Args[1])));
 739   return GV;
 740 }
 741
 742 // int fprintf(FILE *,sbyte *, ...) - a very rough implementation to make output
 743 // useful.
 744 GenericValue lle_X_fprintf(FunctionType *M, const vector<GenericValue> &Args) {
 745   assert(Args.size() >= 2);
 746   char Buffer[10000];
 747   vector<GenericValue> NewArgs;
 748   NewArgs.push_back(PTOGV(Buffer));
 749   NewArgs.insert(NewArgs.end(), Args.begin()+1, Args.end());
 750   GenericValue GV = lle_X_sprintf(M, NewArgs);
 751
 752   fputs(Buffer, getFILE(GVTOP(Args[0])));
 753   return GV;
 754 }
 755
 756 } // End extern "C"
 757
 758
 759 void Interpreter::initializeExternalMethods() {
 760   FuncNames["lle_VP_printstr"] = lle_VP_printstr;
 761   FuncNames["lle_X_print"] = lle_X_print;
 762   FuncNames["lle_X_printVal"] = lle_X_printVal;
 763   FuncNames["lle_X_printString"] = lle_X_printString;
 764   FuncNames["lle_X_printUByte"] = lle_X_printUByte;
 765   FuncNames["lle_X_printSByte"] = lle_X_printSByte;
 766   FuncNames["lle_X_printUShort"] = lle_X_printUShort;
 767   FuncNames["lle_X_printShort"] = lle_X_printShort;
 768   FuncNames["lle_X_printInt"] = lle_X_printInt;
 769   FuncNames["lle_X_printUInt"] = lle_X_printUInt;
 770   FuncNames["lle_X_printLong"] = lle_X_printLong;
 771   FuncNames["lle_X_printULong"] = lle_X_printULong;
 772   FuncNames["lle_X_printFloat"] = lle_X_printFloat;
 773   FuncNames["lle_X_printDouble"] = lle_X_printDouble;
 774   FuncNames["lle_X_printPointer"] = lle_X_printPointer;
 775   FuncNames["lle_Vb_putchar"]     = lle_Vb_putchar;
 776   FuncNames["lle_ii_putchar"]     = lle_ii_putchar;
 777   FuncNames["lle_VB_putchar"]     = lle_VB_putchar;
 778   FuncNames["lle_V___main"]       = lle_V___main;
 779   FuncNames["lle_X_exit"]         = lle_X_exit;
 780   FuncNames["lle_X_abort"]        = lle_X_abort;
 781   FuncNames["lle_X_malloc"]       = lle_X_malloc;
 782   FuncNames["lle_X_calloc"]       = lle_X_calloc;
 783   FuncNames["lle_X_free"]         = lle_X_free;
 784   FuncNames["lle_X_atoi"]         = lle_X_atoi;
 785   FuncNames["lle_X_pow"]          = lle_X_pow;
 786   FuncNames["lle_X_exp"]          = lle_X_exp;
 787   FuncNames["lle_X_log"]          = lle_X_log;
 788   FuncNames["lle_X_isnan"]        = lle_X_isnan;
 789   FuncNames["lle_X_floor"]        = lle_X_floor;
 790   FuncNames["lle_X_srand"]        = lle_X_srand;
 791   FuncNames["lle_X_drand48"]      = lle_X_drand48;
 792   FuncNames["lle_X_srand48"]      = lle_X_srand48;
 793   FuncNames["lle_X_lrand48"]      = lle_X_lrand48;
 794   FuncNames["lle_X_sqrt"]         = lle_X_sqrt;
 795   FuncNames["lle_X_puts"]         = lle_X_puts;
 796   FuncNames["lle_X_printf"]       = lle_X_printf;
 797   FuncNames["lle_X_sprintf"]      = lle_X_sprintf;
 798   FuncNames["lle_X_sscanf"]       = lle_X_sscanf;
 799   FuncNames["lle_X_scanf"]        = lle_X_scanf;
 800   FuncNames["lle_i_clock"]        = lle_i_clock;
 801
 802   FuncNames["lle_X_strcmp"]       = lle_X_strcmp;
 803   FuncNames["lle_X_strcat"]       = lle_X_strcat;
 804   FuncNames["lle_X_strcpy"]       = lle_X_strcpy;
 805   FuncNames["lle_X_strlen"]       = lle_X_strlen;
 806   FuncNames["lle_X_memset"]       = lle_X_memset;
 807   FuncNames["lle_X_memcpy"]       = lle_X_memcpy;
 808
 809   FuncNames["lle_X_fopen"]        = lle_X_fopen;
 810   FuncNames["lle_X_fclose"]       = lle_X_fclose;
 811   FuncNames["lle_X_feof"]         = lle_X_feof;
 812   FuncNames["lle_X_fread"]        = lle_X_fread;
 813   FuncNames["lle_X_fwrite"]       = lle_X_fwrite;
 814   FuncNames["lle_X_fgets"]        = lle_X_fgets;
 815   FuncNames["lle_X_fflush"]       = lle_X_fflush;
 816   FuncNames["lle_X_fgetc"]        = lle_X_getc;
 817   FuncNames["lle_X_getc"]         = lle_X_getc;
 818   FuncNames["lle_X__IO_getc"]     = lle_X__IO_getc;
 819   FuncNames["lle_X_fputc"]        = lle_X_fputc;
 820   FuncNames["lle_X_ungetc"]       = lle_X_ungetc;
 821   FuncNames["lle_X_fprintf"]      = lle_X_fprintf;
 822   FuncNames["lle_X_freopen"]      = lle_X_freopen;
 823 }