X-Git-Url: http://demsky.eecs.uci.edu/git/?a=blobdiff_plain;f=lib%2FExecutionEngine%2FInterpreter%2FExternalFunctions.cpp;h=bde32bd269176980009282e28c9e6f6284ecfb26;hb=34008373748f334b380972c67938c41f2e98582a;hp=c8aa46a186e384f9cc69519fe2f8041426cb494b;hpb=b408b1255646f31895dcbf099184e087bfa1ca39;p=oota-llvm.git diff --git a/lib/ExecutionEngine/Interpreter/ExternalFunctions.cpp b/lib/ExecutionEngine/Interpreter/ExternalFunctions.cpp index c8aa46a186e..bde32bd2691 100644 --- a/lib/ExecutionEngine/Interpreter/ExternalFunctions.cpp +++ b/lib/ExecutionEngine/Interpreter/ExternalFunctions.cpp @@ -1,25 +1,30 @@ -//===-- 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 #include -#include #include #include #include using std::vector; using std::cout; +extern TargetData TD; + typedef GenericValue (*ExFunc)(FunctionType *, const vector &); static std::map Functions; static std::map FuncNames; @@ -91,18 +96,19 @@ GenericValue Interpreter::callExternalMethod(Function *M, const vector &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::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(M->getFunctionType()),ArgVals); + GenericValue Result = Fn(const_cast(M->getFunctionType()), + ArgVals); return Result; } @@ -132,7 +138,8 @@ GenericValue lle_X_printVal(FunctionType *M, const vector &ArgVal) assert(ArgVal.size() == 1 && "generic print only takes one argument!"); // Specialize print([ubyte {x N} ] *) and print(sbyte *) - if (PointerType *PTy = dyn_cast(M->getParamTypes()[0].get())) + if (const PointerType *PTy = + dyn_cast(M->getParamTypes()[0].get())) if (PTy->getElementType() == Type::SByteTy || isa(PTy->getElementType())) { return lle_VP_printstr(M, ArgVal); @@ -172,35 +179,44 @@ PRINT_TYPE_FUNC(Double, DoubleTyID) PRINT_TYPE_FUNC(Pointer, PointerTyID) -// void "putchar"(sbyte) +// void putchar(sbyte) GenericValue lle_Vb_putchar(FunctionType *M, const vector &Args) { cout << Args[0].SByteVal; return GenericValue(); } -// int "putchar"(int) +// int putchar(int) GenericValue lle_ii_putchar(FunctionType *M, const vector &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 &Args) { cout << Args[0].SByteVal << std::flush; return Args[0]; } -// void "__main"() +// void __main() GenericValue lle_V___main(FunctionType *M, const vector &Args) { return GenericValue(); } -// void "exit"(int) +// void exit(int) GenericValue lle_X_exit(FunctionType *M, const vector &Args) { TheInterpreter->exitCalled(Args[0]); return GenericValue(); } +// void abort(void) +GenericValue lle_X_abort(FunctionType *M, const vector &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 &Args) { assert(Args.size() == 1 && "Malloc expects one argument!"); @@ -334,13 +350,21 @@ GenericValue lle_X_sprintf(FunctionType *M, const vector &Args) { 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; @@ -397,6 +421,67 @@ GenericValue lle_i_clock(FunctionType *M, const vector &Args) { // 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(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(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 &Args) { assert(Args.size() == 2); @@ -412,7 +497,7 @@ GenericValue lle_X_fclose(FunctionType *M, const vector &Args) { assert(Args.size() == 1); GenericValue GV; - GV.IntVal = fclose((FILE *)Args[0].PointerVal); + GV.IntVal = fclose(getFILE(Args[0].PointerVal)); return GV; } @@ -422,7 +507,7 @@ GenericValue lle_X_fread(FunctionType *M, const vector &Args) { 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; } @@ -432,7 +517,7 @@ GenericValue lle_X_fwrite(FunctionType *M, const vector &Args) { 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; } @@ -442,7 +527,7 @@ GenericValue lle_X_fgets(FunctionType *M, const vector &Args) { GenericValue GV; GV.PointerVal = (PointerTy)fgets((char*)Args[0].PointerVal, Args[1].IntVal, - (FILE*)Args[2].PointerVal); + getFILE(Args[2].PointerVal)); return GV; } @@ -451,7 +536,16 @@ GenericValue lle_X_fflush(FunctionType *M, const vector &Args) { 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 &Args) { + assert(Args.size() == 1); + GenericValue GV; + + GV.IntVal = getc(getFILE(Args[0].PointerVal)); return GV; } @@ -479,6 +573,7 @@ void Interpreter::initializeExternalMethods() { 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; @@ -501,4 +596,5 @@ void Interpreter::initializeExternalMethods() { 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; }