/* -----------------------------------------------------------------------------
- * $Id: RtsStartup.c,v 1.2 1998/12/02 13:28:41 simonm Exp $
+ * $Id: RtsStartup.c,v 1.25 1999/12/20 10:34:37 simonpj Exp $
+ *
+ * (c) The GHC Team, 1998-1999
*
* Main function for a standalone Haskell program.
*
#include "RtsUtils.h"
#include "RtsFlags.h"
#include "Storage.h" /* initStorage, exitStorage */
-#include "StablePtr.h" /* initStablePtrTable */
+#include "StablePriv.h" /* initStablePtrTable */
#include "Schedule.h" /* initScheduler */
#include "Stats.h" /* initStats */
+#include "Signals.h"
+#include "Itimer.h"
#include "Weak.h"
+#include "Ticky.h"
#if defined(PROFILING)
-# include "ProfRTS.h"
-#elif defined(DEBUG)
-# include "DebugProf.h"
+# include "ProfRts.h"
+# include "ProfHeap.h"
#endif
#ifdef PAR
*/
struct RTS_FLAGS RtsFlags;
-extern void startupHaskell(int argc, char *argv[])
+static int rts_has_started_up = 0;
+
+void
+startupHaskell(int argc, char *argv[])
{
+#ifdef ENABLE_WIN32_DLL_SUPPORT
+ int i;
+#endif
+
+ /* To avoid repeated initialisations of the RTS */
+ if (rts_has_started_up)
+ return;
+ else
+ rts_has_started_up=1;
+
#if defined(PAR)
int nPEs = 0; /* Number of PEs */
#endif
prog_argc = argc;
prog_argv = argv;
-#if defined(PAR)
+#ifdef PAR
/* Initialise the parallel system -- before initHeap! */
initParallelSystem();
/* And start GranSim profiling if required: omitted for now
*/
#endif /* PAR */
+ /* initialise scheduler data structures (needs to be done before
+ * initStorage()).
+ */
+ initScheduler();
+
/* initialize the storage manager */
initStorage();
initProfiling();
#endif
- /* Initialise the scheduler */
- initScheduler();
+ /* start the ticker */
+ install_vtalrm_handler();
+#if 0 /* tmp--SDM */
+ initialize_virtual_timer(TICK_MILLISECS);
+#endif
+
+ /* start our haskell execution tasks */
+#ifdef SMP
+ startTasks();
+#endif
/* Initialise the stats department */
initStats();
-#if 0
+#if !defined(mingw32_TARGET_OS) && !defined(PAR)
+ /* Initialise the user signal handler set */
initUserSignals();
+ /* Set up handler to run on SIGINT */
+ init_shutdown_handler();
#endif
+
+ /* When the RTS and Prelude live in separate DLLs,
+ we need to patch up the char- and int-like tables
+ that the RTS keep after both DLLs have been loaded,
+ filling in the tables with references to where the
+ static info tables have been loaded inside the running
+ process.
+ */
+#ifdef ENABLE_WIN32_DLL_SUPPORT
+ for(i=0;i<=255;i++)
+ (CHARLIKE_closure[i]).header.info = (const StgInfoTable*)&Czh_static_info;
+ for(i=0;i<=32;i++)
+ (INTLIKE_closure[i]).header.info = (const StgInfoTable*)&Izh_static_info;
+
+#endif
/* Record initialization times */
end_init();
}
+/*
+ * Shutting down the RTS - two ways of doing this, one which
+ * calls exit(), one that doesn't.
+ *
+ * (shutdownHaskellAndExit() is called by System.exitWith).
+ */
+void
+shutdownHaskellAndExit(int n)
+{
+ OnExitHook();
+ shutdownHaskell();
+ stg_exit(n);
+}
+
void
shutdownHaskell(void)
{
- /* Finalise any remaining weak pointers */
- finaliseWeakPointersNow();
+ if (!rts_has_started_up)
+ return;
+
+ /* start timing the shutdown */
+ stat_startExit();
+
+ /* Finalize any remaining weak pointers */
+ finalizeWeakPointersNow();
#if defined(GRAN)
#error FixMe.
end_gr_simulation();
#endif
- /* clean up things from the storage manager's point of view */
+ /* stop all running tasks */
+ exitScheduler();
+
+ /* stop the ticker */
+ initialize_virtual_timer(0);
+
+ /* reset the standard file descriptors to blocking mode */
+ resetNonBlockingFd(0);
+ resetNonBlockingFd(1);
+ resetNonBlockingFd(2);
+
+ /* stop timing the shutdown, we're about to print stats */
+ stat_endExit();
+
+ /* clean up things from the storage manager's point of view.
+ * also outputs the stats (+RTS -s) info.
+ */
exitStorage();
#if defined(PROFILING) || defined(DEBUG)
#endif
#if defined(TICKY_TICKY)
- #error FixMe.
- if (RTSflags.TickyFlags.showTickyStats) PrintTickyInfo();
+ if (RtsFlags.TickyFlags.showTickyStats) PrintTickyInfo();
#endif
- /*
- This fflush is important, because: if "main" just returns,
- then we will end up in pre-supplied exit code that will close
- streams and flush buffers. In particular we have seen: it
- will close fd 0 (stdin), then flush fd 1 (stdout), then <who
- cares>...
-
- But if you're playing with sockets, that "close fd 0" might
- suggest to the daemon that all is over, only to be presented
- with more stuff on "fd 1" at the flush.
-
- The fflush avoids this sad possibility.
- */
- fflush(stdout);
+ rts_has_started_up=0;
}
/*
- * called from STG-land to exit the program cleanly
+ * called from STG-land to exit the program
*/
void
#ifdef PAR
par_exit(n);
#else
- OnExitHook();
exit(n);
#endif
}
+