//===----------------------------------------------------------------------===//
#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;
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!");
return GV;
}
+// int isnan(double value);
+GenericValue lle_X_isnan(FunctionType *F, const vector<GenericValue> &Args) {
+ assert(Args.size() == 1);
+ GenericValue GV;
+ GV.IntVal = isnan(Args[0].DoubleVal);
+ return GV;
+}
+
// double floor(double)
GenericValue lle_X_floor(FunctionType *M, const vector<GenericValue> &Args) {
assert(Args.size() == 1);
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;
+ 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;
+}
+
+// int feof(FILE *stream);
+GenericValue lle_X_feof(FunctionType *M, const vector<GenericValue> &Args) {
+ assert(Args.size() == 1);
+ GenericValue GV;
+
+ GV.IntVal = feof(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;
+}
+
+// FILE *freopen(const char *path, const char *mode, FILE *stream);
+GenericValue lle_X_freopen(FunctionType *M, const vector<GenericValue> &Args) {
+ assert(Args.size() == 3);
+ GenericValue GV;
+ GV.PointerVal = (PointerTy)freopen((char*)Args[0].PointerVal,
+ (char*)Args[1].PointerVal,
+ getFILE(Args[2].PointerVal));
return GV;
}
GenericValue lle_X_fflush(FunctionType *M, const vector<GenericValue> &Args) {
assert(Args.size() == 1);
GenericValue GV;
+ 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;
+}
+
+// int fputc(int C, FILE *stream);
+GenericValue lle_X_fputc(FunctionType *M, const vector<GenericValue> &Args) {
+ assert(Args.size() == 2);
+ GenericValue GV;
+ GV.IntVal = fputc(Args[0].IntVal, getFILE(Args[1].PointerVal));
+ return GV;
+}
+
+// int ungetc(int C, FILE *stream);
+GenericValue lle_X_ungetc(FunctionType *M, const vector<GenericValue> &Args) {
+ assert(Args.size() == 2);
+ GenericValue GV;
+ GV.IntVal = ungetc(Args[0].IntVal, getFILE(Args[1].PointerVal));
+ return GV;
+}
+
+// int fprintf(FILE *,sbyte *, ...) - a very rough implementation to make output
+// useful.
+GenericValue lle_X_fprintf(FunctionType *M, const vector<GenericValue> &Args) {
+ assert(Args.size() > 2);
+ char Buffer[10000];
+ vector<GenericValue> NewArgs;
+ GenericValue GV; GV.PointerVal = (PointerTy)Buffer;
+ NewArgs.push_back(GV);
+ NewArgs.insert(NewArgs.end(), Args.begin()+1, Args.end());
+ GV = lle_X_sprintf(M, NewArgs);
- GV.IntVal = fflush((FILE*)Args[0].PointerVal);
+ fputs(Buffer, 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_pow"] = lle_X_pow;
FuncNames["lle_X_exp"] = lle_X_exp;
FuncNames["lle_X_log"] = lle_X_log;
+ FuncNames["lle_X_isnan"] = lle_X_isnan;
FuncNames["lle_X_floor"] = lle_X_floor;
FuncNames["lle_X_srand"] = lle_X_srand;
FuncNames["lle_X_drand48"] = lle_X_drand48;
FuncNames["lle_i_clock"] = lle_i_clock;
FuncNames["lle_X_fopen"] = lle_X_fopen;
FuncNames["lle_X_fclose"] = lle_X_fclose;
+ FuncNames["lle_X_feof"] = lle_X_feof;
FuncNames["lle_X_fread"] = lle_X_fread;
FuncNames["lle_X_fwrite"] = lle_X_fwrite;
FuncNames["lle_X_fgets"] = lle_X_fgets;
FuncNames["lle_X_fflush"] = lle_X_fflush;
+ FuncNames["lle_X_fgetc"] = lle_X_getc;
+ FuncNames["lle_X_getc"] = lle_X_getc;
+ FuncNames["lle_X_fputc"] = lle_X_fputc;
+ FuncNames["lle_X_ungetc"] = lle_X_ungetc;
+ FuncNames["lle_X_fprintf"] = lle_X_fprintf;
+ FuncNames["lle_X_freopen"] = lle_X_freopen;
}