X-Git-Url: http://demsky.eecs.uci.edu/git/?a=blobdiff_plain;f=lib%2FExecutionEngine%2FInterpreter%2FExternalFunctions.cpp;h=bde32bd269176980009282e28c9e6f6284ecfb26;hb=34008373748f334b380972c67938c41f2e98582a;hp=a16f31718e63c4d611ea3a842971110f65801739;hpb=1ee34a5a5ba52557697e7eb607e8bd5d4ce0d492;p=oota-llvm.git diff --git a/lib/ExecutionEngine/Interpreter/ExternalFunctions.cpp b/lib/ExecutionEngine/Interpreter/ExternalFunctions.cpp index a16f31718e6..bde32bd2691 100644 --- a/lib/ExecutionEngine/Interpreter/ExternalFunctions.cpp +++ b/lib/ExecutionEngine/Interpreter/ExternalFunctions.cpp @@ -11,16 +11,19 @@ //===----------------------------------------------------------------------===// #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; @@ -135,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); @@ -175,36 +179,36 @@ 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) +// void abort(void) GenericValue lle_X_abort(FunctionType *M, const vector &Args) { std::cerr << "***PROGRAM ABORTED***!\n"; GenericValue GV; @@ -350,9 +354,17 @@ GenericValue lle_X_sprintf(FunctionType *M, const vector &Args) { 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; @@ -409,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); @@ -424,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; } @@ -434,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; } @@ -444,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; } @@ -454,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; } @@ -463,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; } @@ -514,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; }