/// calls to atexit(3), which we intercept and store in
/// AtExitHandlers.
///
-void JIT::runAtExitHandlers() {
+static void runAtExitHandlers() {
while (!AtExitHandlers.empty()) {
void (*Fn)() = AtExitHandlers.back();
AtExitHandlers.pop_back();
// jit_exit - Used to intercept the "exit" library call.
static void jit_exit(int Status) {
- JIT::runAtExitHandlers(); // Run atexit handlers...
+ runAtExitHandlers(); // Run atexit handlers...
exit(Status);
}
/// run - Start execution with the specified function and arguments.
///
-GenericValue JIT::run(Function *F, const std::vector<GenericValue> &ArgValues) {
+GenericValue JIT::runFunction(Function *F,
+ const std::vector<GenericValue> &ArgValues) {
assert (F && "Function *F was null at entry to run()");
+ GenericValue rv;
+
+ if (ArgValues.size() == 3) {
+ int (*PF)(int, char **, const char **) =
+ (int(*)(int, char **, const char **))getPointerToFunction(F);
+ assert(PF && "Pointer to fn's code was null after getPointerToFunction");
+
+ // Call the function.
+ int ExitCode = PF(ArgValues[0].IntVal, (char **) GVTOP (ArgValues[1]),
+ (const char **) GVTOP (ArgValues[2]));
+
+ rv.IntVal = ExitCode;
+ } else {
+ // FIXME: This code should handle a couple of common cases efficiently, but
+ // it should also implement the general case by code-gening a new anonymous
+ // nullary function to call.
+ assert(ArgValues.size() == 1);
+ void (*PF)(int) = (void(*)(int))getPointerToFunction(F);
+ assert(PF && "Pointer to fn's code was null after getPointerToFunction");
+ PF(ArgValues[0].IntVal);
+ }
- int (*PF)(int, char **, const char **) =
- (int(*)(int, char **, const char **))getPointerToFunction(F);
- assert(PF != 0 && "Pointer to fn's code was null after getPointerToFunction");
-
- // Call the function.
- int ExitCode = PF(ArgValues[0].IntVal, (char **) GVTOP (ArgValues[1]),
- (const char **) GVTOP (ArgValues[2]));
-
- // Run any atexit handlers now!
- runAtExitHandlers();
-
- GenericValue rv;
- rv.IntVal = ExitCode;
return rv;
}
/// run - Start execution with the specified function and arguments.
///
- virtual GenericValue run(Function *F,
- const std::vector<GenericValue> &ArgValues);
+ virtual GenericValue runFunction(Function *F,
+ const std::vector<GenericValue> &ArgValues);
/// getPointerToNamedFunction - This method returns the address of the
/// specified function by using the dlsym function call. As such it is only
//
static void CompilationCallback();
- /// runAtExitHandlers - Before exiting the program, at_exit functions must be
- /// called. This method calls them.
- ///
- static void runAtExitHandlers();
-
/// getPointerToFunction - This returns the address of the specified function,
/// compiling it if necessary.
///