From 4a5bb95729a5335bfdddd401454003ac05ba4316 Mon Sep 17 00:00:00 2001 From: Chris Lattner Date: Wed, 14 May 2003 14:21:30 +0000 Subject: Add support for atexit function, remove support for __main function llvm-svn: 6194 --- llvm/lib/ExecutionEngine/Interpreter/Interpreter.cpp | 19 +++++++++++++++---- 1 file changed, 15 insertions(+), 4 deletions(-) (limited to 'llvm/lib/ExecutionEngine/Interpreter/Interpreter.cpp') diff --git a/llvm/lib/ExecutionEngine/Interpreter/Interpreter.cpp b/llvm/lib/ExecutionEngine/Interpreter/Interpreter.cpp index 4582a6f26d4..3453d831957 100644 --- a/llvm/lib/ExecutionEngine/Interpreter/Interpreter.cpp +++ b/llvm/lib/ExecutionEngine/Interpreter/Interpreter.cpp @@ -48,10 +48,21 @@ int Interpreter::run(const std::string &MainFunction, run(); } - // If debug mode, allow the user to interact... also, if the user pressed - // ctrl-c or execution hit an error, enter the event loop... - if (Debug || isStopped()) - handleUserInput(); + do { + // If debug mode, allow the user to interact... also, if the user pressed + // ctrl-c or execution hit an error, enter the event loop... + if (Debug || isStopped()) + handleUserInput(); + + // If the program has exited, run atexit handlers... + if (ECStack.empty() && !AtExitHandlers.empty()) { + callFunction(AtExitHandlers.back(), std::vector()); + AtExitHandlers.pop_back(); + run(); + } + } while (!ECStack.empty()); + + PerformExitStuff(); return ExitCode; } -- cgit v1.2.3