rts: Claim AP_STACK before adjusting Sp
[ghc.git] / rts / RtsMain.c
index aa2fe0f..d9f0557 100644 (file)
 #include "PosixSource.h"
 #include "Rts.h"
 #include "RtsAPI.h"
-#include "SchedAPI.h"
-#include "RtsFlags.h"
+
 #include "RtsUtils.h"
-#include "RtsMain.h"
 #include "Prelude.h"
 #include "Task.h"
-#if defined(mingw32_HOST_OS)
-#include "win32/seh_excn.h"
-#endif
-#include <stdlib.h>
+#include "Excn.h"
 
-#ifdef DEBUG
+#if defined(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);
-
-/* Annoying global vars for passing parameters to real_main() below
- * This is to get around problem with Windows SEH, see hs_main(). */
-static int progargc;
-static char **progargv;
-static void (*progmain_init)(void);   /* This will be __stginit_ZCMain */
-static StgClosure *progmain_closure;  /* This will be ZCMain_main_closure */
-
-/* Hack: we assume that we're building a batch-mode system unless 
- * INTERPRETER is set
- */
-#ifndef INTERPRETER /* Hack */
-static void real_main(void)
+// Hack: we assume that we're building a batch-mode system unless
+// INTERPRETER is set
+#if !defined(INTERPRETER) /* Hack */
+
+// The rts entry point from a compiled program using a Haskell main
+// function.  This gets called from a tiny main function generated by
+// GHC and linked into each compiled Haskell program that uses a
+// Haskell main function.
+// 
+// We expect the caller to pass ZCMain_main_closure for
+// main_closure. The reason we cannot refer to this symbol directly
+// is because we're inside the rts and we do not know for sure that
+// we'll be using a Haskell main function.
+// 
+// NOTE: This function is marked as _noreturn_ in Main.h
+
+int hs_main ( int argc, char *argv[],       // program args
+              StgClosure *main_closure,     // closure for Main.main
+              RtsConfig rts_config)         // RTS configuration
+              
 {
+    BEGIN_WINDOWS_VEH_HANDLER
+
     int exit_status;
     SchedulerStatus status;
-    /* all GranSim/GUM init is done in startupHaskell; sets IAmMainThread! */
-
-    startupHaskell(progargc,progargv,progmain_init);
-
-    /* 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(progmain_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)
+    hs_init_ghc(&argc, &argv, rts_config);
+
+    // 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;
+    // 
+    // 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)
+
+    // ToDo: want to start with a larger stack size
+    {
+        Capability *cap = rts_lock();
+        rts_evalLazyIO(&cap, main_closure, NULL);
+        status = rts_getSchedStatus(cap);
+        rts_unlock(cap);
     }
 
-#  elif defined(GRAN)
-
-    /* ToDo: Dump event for the main thread */
-    status = rts_mainLazyIO(progmain_closure, NULL);
-
-#  else /* !PAR && !GRAN */
-
-    /* ToDo: want to start with a larger stack size */
-    { 
-       Capability *cap = rts_lock();
-       cap = rts_evalLazyIO(cap,progmain_closure, NULL);
-       status = rts_getSchedStatus(cap);
-       taskTimeStamp(myTask());
-       rts_unlock(cap);
-    }
-
-#  endif /* !PAR && !GRAN */
-
-    /* check the status of the entire Haskell computation */
+    // check the status of the entire Haskell computation
     switch (status) {
     case Killed:
-      errorBelch("main thread exited (uncaught exception)");
-      exit_status = EXIT_KILLED;
-      break;
+        errorBelch("main thread exited (uncaught exception)");
+        exit_status = EXIT_KILLED;
+        break;
     case Interrupted:
-      errorBelch("interrupted");
-      exit_status = EXIT_INTERRUPTED;
-      break;
+        errorBelch("interrupted");
+        exit_status = EXIT_INTERRUPTED;
+        break;
     case HeapExhausted:
-      exit_status = EXIT_HEAPOVERFLOW;
-      break;
+        exit_status = EXIT_HEAPOVERFLOW;
+        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 
+        exit_status = EXIT_SUCCESS;
+        break;
     default:
-      barf("main thread completed with invalid status");
+        barf("main thread completed with invalid status");
     }
-    shutdownHaskellAndExit(exit_status);
-}
 
-/* The rts entry point from a compiled program using a Haskell main function.
- * This gets called from a tiny main function which gets linked into each
- * compiled Haskell program that uses a Haskell main function.
- *
- * We expect the caller to pass __stginit_ZCMain for main_init and
- * ZCMain_main_closure for main_closure. The reason we cannot refer to
- * these symbols directly is because we're inside the rts and we do not know
- * for sure that we'll be using a Haskell main function.
- */
-int hs_main(int argc, char *argv[], void (*main_init)(void), StgClosure *main_closure)
-{
-    /* We do this dance with argc and argv as otherwise the SEH exception
-       stuff (the BEGIN/END CATCH below) on Windows gets confused */
-    progargc = argc;
-    progargv = argv;
-    progmain_init    = main_init;
-    progmain_closure = main_closure;
+    END_WINDOWS_VEH_HANDLER
 
-#if defined(mingw32_HOST_OS)
-    BEGIN_CATCH
-#endif
-    real_main();
-#if defined(mingw32_HOST_OS)
-    END_CATCH
-#endif
-    return 0; /* not reached, but keeps gcc -Wall happy */
+    shutdownHaskellAndExit(exit_status, 0 /* !fastExit */);
+    // No code beyond this point. Dead code elimination will remove it 
 }
 # endif /* BATCH_MODE */