diff options
Diffstat (limited to 'rts/Main.c')
-rw-r--r-- | rts/Main.c | 138 |
1 files changed, 138 insertions, 0 deletions
diff --git a/rts/Main.c b/rts/Main.c new file mode 100644 index 0000000000..6aef280e25 --- /dev/null +++ b/rts/Main.c @@ -0,0 +1,138 @@ +/* ----------------------------------------------------------------------------- + * + * (c) The GHC Team 1998-2000 + * + * Main function for a standalone Haskell program. + * + * ---------------------------------------------------------------------------*/ + +#define COMPILING_RTS_MAIN + +#include "PosixSource.h" +#include "Rts.h" +#include "RtsAPI.h" +#include "SchedAPI.h" +#include "Schedule.h" +#include "RtsFlags.h" +#include "RtsUtils.h" +#include "Prelude.h" +#include "Task.h" +#include <stdlib.h> + +#ifdef DEBUG +# include "Printer.h" /* for printing */ +#endif + +#ifdef PAR +# include "Parallel.h" +# include "ParallelRts.h" +# include "LLC.h" +#endif + +#if defined(GRAN) || defined(PAR) +# include "GranSimRts.h" +#endif + +#ifdef HAVE_WINDOWS_H +# include <windows.h> +#endif + +extern void __stginit_ZCMain(void); + +/* Hack: we assume that we're building a batch-mode system unless + * INTERPRETER is set + */ +#ifndef INTERPRETER /* Hack */ +int main(int argc, char *argv[]) +{ + int exit_status; + SchedulerStatus status; + /* all GranSim/GUM init is done in startupHaskell; sets IAmMainThread! */ + + startupHaskell(argc,argv,__stginit_ZCMain); + + /* kick off the computation by creating the main thread with a pointer + to mainIO_closure representing the computation of the overall program; + then enter the scheduler with this thread and off we go; + + the same for GranSim (we have only one instance of this code) + + in a parallel setup, where we have many instances of this code + running on different PEs, we should do this only for the main PE + (IAmMainThread is set in startupHaskell) + */ + +# if defined(PAR) + +# if defined(DEBUG) + { /* a wait loop to allow attachment of gdb to UNIX threads */ + nat i, j, s; + + for (i=0, s=0; i<(nat)RtsFlags.ParFlags.wait; i++) + for (j=0; j<1000000; j++) + s += j % 65536; + } + IF_PAR_DEBUG(verbose, + belch("Passed wait loop")); +# endif + + if (IAmMainThread == rtsTrue) { + IF_PAR_DEBUG(verbose, + debugBelch("==== [%x] Main Thread Started ...\n", mytid)); + + /* ToDo: Dump event for the main thread */ + status = rts_mainLazyIO((HaskellObj)mainIO_closure, NULL); + } else { + /* Just to show we're alive */ + IF_PAR_DEBUG(verbose, + debugBelch("== [%x] Non-Main PE enters scheduler via taskStart() without work ...\n", + mytid)); + + /* all non-main threads enter the scheduler without work */ + taskStart(); + status = Success; // declare victory (see shutdownParallelSystem) + } + +# elif defined(GRAN) + + /* ToDo: Dump event for the main thread */ + status = rts_mainLazyIO(mainIO_closure, NULL); + +# else /* !PAR && !GRAN */ + + /* ToDo: want to start with a larger stack size */ + { + void *cap = rts_lock(); + cap = rts_evalLazyIO(cap,(HaskellObj)(void *)mainIO_closure, NULL); + status = rts_getSchedStatus(cap); + rts_unlock(cap); + } + +# endif /* !PAR && !GRAN */ + + /* check the status of the entire Haskell computation */ + switch (status) { + case Killed: + errorBelch("main thread exited (uncaught exception)"); + exit_status = EXIT_KILLED; + break; + case Interrupted: + errorBelch("interrupted"); + exit_status = EXIT_INTERRUPTED; + break; + case Success: + exit_status = EXIT_SUCCESS; + break; +#if defined(PAR) + case NoStatus: + errorBelch("main thread PE killed; probably due to failure of another PE; check /tmp/pvml..."); + exit_status = EXIT_KILLED; + break; +#endif + default: + barf("main thread completed with invalid status"); + } + shutdownHaskellAndExit(exit_status); + return 0; /* never reached, keep gcc -Wall happy */ +} +# endif /* BATCH_MODE */ |