-//===-- ExternalMethods.cpp - Implement External Functions ----------------===//
+//===-- ExternalFunctions.cpp - Implement External Functions --------------===//
//
-// This file contains both code to deal with invoking "external" methods, but
-// also contains code that implements "exported" external methods.
+// This file contains both code to deal with invoking "external" functions, but
+// also contains code that implements "exported" external functions.
//
-// External methods in LLI are implemented by dlopen'ing the lli executable and
-// using dlsym to look op the methods that we want to invoke. If a method is
-// found, then the arguments are mangled and passed in to the function call.
+// External functions in LLI are implemented by dlopen'ing the lli executable
+// and using dlsym to look op the functions that we want to invoke. If a
+// function is found, then the arguments are mangled and passed in to the
+// function call.
//
//===----------------------------------------------------------------------===//
#include "Interpreter.h"
+#include "ExecutionAnnotations.h"
#include "llvm/DerivedTypes.h"
+#include "llvm/SymbolTable.h"
+#include "llvm/Target/TargetData.h"
#include <map>
#include <dlfcn.h>
-#include <iostream>
#include <link.h>
#include <math.h>
#include <stdio.h>
using std::vector;
using std::cout;
+extern TargetData TD;
+
typedef GenericValue (*ExFunc)(FunctionType *, const vector<GenericValue> &);
static std::map<const Function *, ExFunc> Functions;
static std::map<std::string, ExFunc> FuncNames;
const vector<GenericValue> &ArgVals) {
TheInterpreter = this;
- // Do a lookup to see if the method is in our cache... this should just be a
+ // Do a lookup to see if the function is in our cache... this should just be a
// defered annotation!
std::map<const Function *, ExFunc>::iterator FI = Functions.find(M);
ExFunc Fn = (FI == Functions.end()) ? lookupFunction(M) : FI->second;
if (Fn == 0) {
- cout << "Tried to execute an unknown external method: "
+ cout << "Tried to execute an unknown external function: "
<< M->getType()->getDescription() << " " << M->getName() << "\n";
return GenericValue();
}
// TODO: FIXME when types are not const!
- GenericValue Result = Fn(const_cast<FunctionType*>(M->getFunctionType()),ArgVals);
+ GenericValue Result = Fn(const_cast<FunctionType*>(M->getFunctionType()),
+ ArgVals);
return Result;
}
assert(ArgVal.size() == 1 && "generic print only takes one argument!");
// Specialize print([ubyte {x N} ] *) and print(sbyte *)
- if (PointerType *PTy = dyn_cast<PointerType>(M->getParamTypes()[0].get()))
+ if (const PointerType *PTy =
+ dyn_cast<PointerType>(M->getParamTypes()[0].get()))
if (PTy->getElementType() == Type::SByteTy ||
isa<ArrayType>(PTy->getElementType())) {
return lle_VP_printstr(M, ArgVal);
PRINT_TYPE_FUNC(Pointer, PointerTyID)
-// void "putchar"(sbyte)
+// void putchar(sbyte)
GenericValue lle_Vb_putchar(FunctionType *M, const vector<GenericValue> &Args) {
cout << Args[0].SByteVal;
return GenericValue();
}
-// int "putchar"(int)
+// int putchar(int)
GenericValue lle_ii_putchar(FunctionType *M, const vector<GenericValue> &Args) {
cout << ((char)Args[0].IntVal) << std::flush;
return Args[0];
}
-// void "putchar"(ubyte)
+// void putchar(ubyte)
GenericValue lle_VB_putchar(FunctionType *M, const vector<GenericValue> &Args) {
cout << Args[0].SByteVal << std::flush;
return Args[0];
}
-// void "__main"()
+// void __main()
GenericValue lle_V___main(FunctionType *M, const vector<GenericValue> &Args) {
return GenericValue();
}
-// void "exit"(int)
+// void exit(int)
GenericValue lle_X_exit(FunctionType *M, const vector<GenericValue> &Args) {
TheInterpreter->exitCalled(Args[0]);
return GenericValue();
}
+// void abort(void)
+GenericValue lle_X_abort(FunctionType *M, const vector<GenericValue> &Args) {
+ std::cerr << "***PROGRAM ABORTED***!\n";
+ GenericValue GV;
+ GV.IntVal = 1;
+ TheInterpreter->exitCalled(GV);
+ return GenericValue();
+}
+
// void *malloc(uint)
GenericValue lle_X_malloc(FunctionType *M, const vector<GenericValue> &Args) {
assert(Args.size() == 1 && "Malloc expects one argument!");
case '%':
sprintf(Buffer, FmtBuf); break;
case 'c':
- sprintf(Buffer, FmtBuf, Args[ArgNo++].SByteVal); break;
+ sprintf(Buffer, FmtBuf, Args[ArgNo++].IntVal); break;
case 'd': case 'i':
case 'u': case 'o':
case 'x': case 'X':
- if (HowLong == 2)
+ if (HowLong >= 1) {
+ if (HowLong == 1) {
+ // Make sure we use %lld with a 64 bit argument because we might be
+ // compiling LLI on a 32 bit compiler.
+ unsigned Size = strlen(FmtBuf);
+ FmtBuf[Size] = FmtBuf[Size-1];
+ FmtBuf[Size+1] = 0;
+ FmtBuf[Size-1] = 'l';
+ }
sprintf(Buffer, FmtBuf, Args[ArgNo++].ULongVal);
- else
+ } else
sprintf(Buffer, FmtBuf, Args[ArgNo++].IntVal); break;
case 'e': case 'E': case 'g': case 'G': case 'f':
sprintf(Buffer, FmtBuf, Args[ArgNo++].DoubleVal); break;
// IO Functions...
//===----------------------------------------------------------------------===//
+// getFILE - Turn a pointer in the host address space into a legit pointer in
+// the interpreter address space. For the most part, this is an identity
+// transformation, but if the program refers to stdio, stderr, stdin then they
+// have pointers that are relative to the __iob array. If this is the case,
+// change the FILE into the REAL stdio stream.
+//
+static FILE *getFILE(PointerTy Ptr) {
+ static Module *LastMod = 0;
+ static PointerTy IOBBase = 0;
+ static unsigned FILESize;
+
+ if (LastMod != TheInterpreter->getModule()) { // Module change or initialize?
+ Module *M = LastMod = TheInterpreter->getModule();
+
+ // Check to see if the currently loaded module contains an __iob symbol...
+ GlobalVariable *IOB = 0;
+ if (SymbolTable *ST = M->getSymbolTable()) {
+ for (SymbolTable::iterator I = ST->begin(), E = ST->end(); I != E; ++I) {
+ SymbolTable::VarMap &M = I->second;
+ for (SymbolTable::VarMap::iterator J = M.begin(), E = M.end();
+ J != E; ++J)
+ if (J->first == "__iob")
+ if ((IOB = dyn_cast<GlobalVariable>(J->second)))
+ break;
+ if (IOB) break;
+ }
+ }
+
+ // If we found an __iob symbol now, find out what the actual address it's
+ // held in is...
+ if (IOB) {
+ // Get the address the array lives in...
+ GlobalAddress *Address =
+ (GlobalAddress*)IOB->getOrCreateAnnotation(GlobalAddressAID);
+ IOBBase = (PointerTy)(GenericValue*)Address->Ptr;
+
+ // Figure out how big each element of the array is...
+ const ArrayType *AT =
+ dyn_cast<ArrayType>(IOB->getType()->getElementType());
+ if (AT)
+ FILESize = TD.getTypeSize(AT->getElementType());
+ else
+ FILESize = 16*8; // Default size
+ }
+ }
+
+ // Check to see if this is a reference to __iob...
+ if (IOBBase) {
+ unsigned FDNum = (Ptr-IOBBase)/FILESize;
+ if (FDNum == 0)
+ return stdin;
+ else if (FDNum == 1)
+ return stdout;
+ else if (FDNum == 2)
+ return stderr;
+ }
+
+ return (FILE*)Ptr;
+}
+
+
// FILE *fopen(const char *filename, const char *mode);
GenericValue lle_X_fopen(FunctionType *M, const vector<GenericValue> &Args) {
assert(Args.size() == 2);
assert(Args.size() == 1);
GenericValue GV;
- GV.IntVal = fclose((FILE *)Args[0].PointerVal);
+ GV.IntVal = fclose(getFILE(Args[0].PointerVal));
return GV;
}
GenericValue GV;
GV.UIntVal = fread((void*)Args[0].PointerVal, Args[1].UIntVal,
- Args[2].UIntVal, (FILE*)Args[3].PointerVal);
+ Args[2].UIntVal, getFILE(Args[3].PointerVal));
return GV;
}
GenericValue GV;
GV.UIntVal = fwrite((void*)Args[0].PointerVal, Args[1].UIntVal,
- Args[2].UIntVal, (FILE*)Args[3].PointerVal);
+ Args[2].UIntVal, getFILE(Args[3].PointerVal));
return GV;
}
GenericValue GV;
GV.PointerVal = (PointerTy)fgets((char*)Args[0].PointerVal, Args[1].IntVal,
- (FILE*)Args[2].PointerVal);
+ getFILE(Args[2].PointerVal));
return GV;
}
assert(Args.size() == 1);
GenericValue GV;
- GV.IntVal = fflush((FILE*)Args[0].PointerVal);
+ GV.IntVal = fflush(getFILE(Args[0].PointerVal));
+ return GV;
+}
+
+// int getc(FILE *stream);
+GenericValue lle_X_getc(FunctionType *M, const vector<GenericValue> &Args) {
+ assert(Args.size() == 1);
+ GenericValue GV;
+
+ GV.IntVal = getc(getFILE(Args[0].PointerVal));
return GV;
}
FuncNames["lle_VB_putchar"] = lle_VB_putchar;
FuncNames["lle_V___main"] = lle_V___main;
FuncNames["lle_X_exit"] = lle_X_exit;
+ FuncNames["lle_X_abort"] = lle_X_abort;
FuncNames["lle_X_malloc"] = lle_X_malloc;
FuncNames["lle_X_free"] = lle_X_free;
FuncNames["lle_X_atoi"] = lle_X_atoi;
FuncNames["lle_X_fwrite"] = lle_X_fwrite;
FuncNames["lle_X_fgets"] = lle_X_fgets;
FuncNames["lle_X_fflush"] = lle_X_fflush;
+ FuncNames["lle_X_getc"] = lle_X_getc;
}
projects / oota-llvm.git / blobdiff