/* -----------------------------------------------------------------------------
 *
 * (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 "RtsUtils.h"
#include "Prelude.h"
#include "Task.h"
#if defined(mingw32_HOST_OS)
#include "win32/seh_excn.h"
#endif

#ifdef DEBUG
# include "Printer.h"   /* for printing        */
#endif

#if defined(PARALLEL_RTS)
#include "Schedule.h"
#ifdef DEBUG
#include <unistd.h> // gethostname, getpid, sleep with "wait" flag
#endif
#endif

#ifdef HAVE_WINDOWS_H
# include <windows.h>
#endif

/* 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 StgClosure *progmain_closure;  /* This will be ZCMain_main_closure */
static RtsConfig rtsconfig;

/* Hack: we assume that we're building a batch-mode system unless
 * INTERPRETER is set
 */
#ifndef INTERPRETER /* Hack */
static void real_main(void) GNUC3_ATTRIBUTE(__noreturn__);
static void real_main(void)
{
    int exit_status;
    SchedulerStatus status;

    hs_init_ghc(&progargc, &progargv, rtsconfig);

    /* 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(PARALLEL_RTS)

# if defined(DEBUG)
    if (RtsFlags.ParFlags.wait != 0) {
    /* a wait loop to allow attachment of gdb to UNIX threads */
      char hostname[256];
      gethostname(hostname, sizeof(hostname));
      debugBelch("Process is ready for attaching debugger.\n"
		 "%s %d (on host %s) and set RtsFlags.ParFlags.wait=rtsFalse\n"
		 , progargv[0], getpid(), hostname);
      while (RtsFlags.ParFlags.wait != 0)
        sleep(1);
    }
# endif

    if (IAmMainThread == rtsTrue) {

    /* ToDo: want to start with a larger stack size */
      Capability *cap;
	IF_PAR_DEBUG(verbose,
		     debugBelch("==== [%x] Main Thread Started ...\n", thisPE));

	cap = rts_lock();
	// copied from below...
	rts_evalLazyIO(&cap,progmain_closure, NULL);
	status = rts_getSchedStatus(cap);
	taskTimeStamp(myTask());
	rts_unlock(cap);

	IF_PAR_DEBUG(verbose,
		     debugBelch("== [%x] Main PE stopping ...\n",
				thisPE));
    } else {
      Capability *cap = rts_lock();

      /* Just to show we're alive */
      IF_PAR_DEBUG(verbose,
		   debugBelch("== [%x] Non-Main PE enters scheduler via taskStart() without work ...\n",
			   thisPE));
      /* all non-main threads enter the scheduler without work */
      startEmptyScheduler(cap);

      rts_unlock(cap);
      IF_PAR_DEBUG(verbose,
		   debugBelch("== [%x] Non-Main PE stopping ...\n",
			   thisPE));
      status = Success;  // declare victory (see shutdownParallelSystem)
    }

#  else /* !PARALLEL_RTS */

    /* ToDo: want to start with a larger stack size */
    { 
	Capability *cap = rts_lock();
        rts_evalLazyIO(&cap,progmain_closure, NULL);
	status = rts_getSchedStatus(cap);
	taskTimeStamp(myTask());
	rts_unlock(cap);
    }

#endif /* PARALLEL_RTS */

    /* 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 HeapExhausted:
      exit_status = EXIT_HEAPOVERFLOW;
      break;
    case Success:
      exit_status = EXIT_SUCCESS;
      break;
    default:
      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 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.
 */
int hs_main (int argc, char *argv[],     // program args
             StgClosure *main_closure,   // closure for Main.main
             RtsConfig rts_config)    // RTS configuration
{
    /* 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_closure = main_closure;
    rtsconfig = rts_config;

#if defined(mingw32_HOST_OS)
    BEGIN_CATCH
#endif
    real_main();
#if defined(mingw32_HOST_OS)
    END_CATCH
#endif
}
# endif /* BATCH_MODE */