#include <dlfcn.h> // dlsym access
#include <iostream>
+// AtExitList - List of functions registered with the at_exit function
+static std::vector<void (*)()> AtExitList;
+
+void VM::runAtExitHandlers() {
+ while (!AtExitList.empty()) {
+ void (*Fn)() = AtExitList.back();
+ AtExitList.pop_back();
+ Fn();
+ }
+}
+
//===----------------------------------------------------------------------===//
// Function stubs that are invoked instead of raw system calls
//===----------------------------------------------------------------------===//
// jit_exit - Used to intercept the "exit" system call.
static void jit_exit(int Status) {
- exit(Status); // Do nothing for now.
+ VM::runAtExitHandlers(); // Run at_exit handlers...
+ exit(Status);
}
// jit_atexit - Used to intercept the "at_exit" system call.
static int jit_atexit(void (*Fn)(void)) {
- return atexit(Fn); // Do nothing for now.
+ AtExitList.push_back(Fn); // Take note of at_exit handler...
+ return 0; // Always successful
}
//===----------------------------------------------------------------------===//
void *VM::getPointerToNamedFunction(const std::string &Name) {
// Check to see if this is one of the functions we want to intercept...
if (Name == "exit") return (void*)&jit_exit;
- if (Name == "at_exit") return (void*)&jit_atexit;
+ if (Name == "atexit") return (void*)&jit_atexit;
// If it's an external function, look it up in the process image...
void *Ptr = dlsym(0, Name.c_str());
char **Argv = (char**)CreateArgv(Args);
// Call the main function...
- return PF(Args.size(), Argv);
+ int Result = PF(Args.size(), Argv);
+
+ // Run any atexit handlers now!
+ runAtExitHandlers();
+ return Result;
}
// handler to lazily patch up references...
//
std::map<void*, Function*> FunctionRefs;
+
public:
VM(Module *M, TargetMachine *tm);
~VM();
// which causes lazy compilation of the target function.
//
static void CompilationCallback();
+
+ /// runAtExitHandlers - Before exiting the program, at_exit functions must be
+ /// called. This method calls them.
+ ///
+ static void runAtExitHandlers();
+
private:
static MachineCodeEmitter *createEmitter(VM &V);
void setupPassManager();
// handler to lazily patch up references...
//
std::map<void*, Function*> FunctionRefs;
+
public:
VM(Module *M, TargetMachine *tm);
~VM();
// which causes lazy compilation of the target function.
//
static void CompilationCallback();
+
+ /// runAtExitHandlers - Before exiting the program, at_exit functions must be
+ /// called. This method calls them.
+ ///
+ static void runAtExitHandlers();
+
private:
static MachineCodeEmitter *createEmitter(VM &V);
void setupPassManager();