Main.c 2.44 KB
Newer Older
1
/* -----------------------------------------------------------------------------
sof's avatar
sof committed
2
 * $Id: Main.c,v 1.8 1999/05/10 10:06:24 sof Exp $
3 4
 *
 * (c) The GHC Team 1998-1999
5 6 7 8 9
 *
 * Main function for a standalone Haskell program.
 *
 * ---------------------------------------------------------------------------*/

sof's avatar
sof committed
10 11
#define COMPILING_RTS_MAIN

12 13
#include "Rts.h"
#include "RtsAPI.h"
14
#include "RtsFlags.h"
15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31
#include "Schedule.h"  /* for MainTSO */
#include "RtsUtils.h"

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

#ifdef INTERPRETER
#include "Assembler.h"
#endif

#ifdef PAR
#include "ParInit.h"
#include "Parallel.h"
#include "LLC.h"
#endif

sof's avatar
sof committed
32 33 34 35 36 37 38
#ifdef HAVE_WINDOWS_H
#include <windows.h>
#endif


#ifndef ENABLE_WIN32_DLL_SUPPORT

39 40 41
/* Hack: we assume that we're building a batch-mode system unless 
 * INTERPRETER is set
 */
sof's avatar
sof committed
42
# ifndef INTERPRETER /* Hack */
43 44 45 46 47
int main(int argc, char *argv[])
{
    SchedulerStatus status;
    startupHaskell(argc,argv);

sof's avatar
sof committed
48
#  ifndef PAR
49 50 51
    MainTSO = createIOThread(stg_max(BLOCK_SIZE_W,
				     RtsFlags.GcFlags.initialStkSize),
			     (StgClosure *)&mainIO_closure);
52
    status = schedule(MainTSO,NULL);
sof's avatar
sof committed
53
#  else
54 55 56 57
    if (IAmMainThread == rtsTrue) {
    /*Just to show we're alive */
      fprintf(stderr, "Main Thread Started ...\n");
     
58 59 60
      MainTSO = createIOThread(stg_max(BLOCK_SIZE_W,
				       RtsFlags.GcFlags.initialStkSize),
			       (StgClosure *)&mainIO_closure);
61 62 63 64 65
      status = schedule(MainTSO,NULL);
    } else {
      WaitForPEOp(PP_FINISH,SysManTask);
      exit(EXIT_SUCCESS);
    }
sof's avatar
sof committed
66
#  endif /* PAR */
67 68 69 70 71 72 73 74 75 76 77 78 79 80 81
    switch (status) {
    case AllBlocked:
      barf("Scheduler stopped, all threads blocked");
    case Deadlock:
      shutdownHaskell();
      barf("No threads to run!  Deadlock?");
    case Killed:
      belch("%s: warning: main thread killed", prog_argv[0]);
    case Success:
    case Interrupted:
      /* carry on */
    }
    shutdownHaskell();
    stg_exit(EXIT_SUCCESS);
}
sof's avatar
sof committed
82 83 84 85 86 87 88 89 90 91 92 93 94 95 96 97 98 99 100
# endif /* BATCH_MODE */

#else   /* !ENABLE_WIN32_DLL_SUPPORT */

static char* args[] = { "ghcRts" };

BOOL
WINAPI
DllMain ( HINSTANCE hInstance
        , DWORD reason
	, LPVOID reserved
	)
{
  /*
    ToDo: let the user configure RTS options to use
          via the registry.
   */
  switch (reason) {
  case DLL_PROCESS_ATTACH:
sof's avatar
sof committed
101
    startupHaskell(1,args);
sof's avatar
sof committed
102 103 104 105 106 107 108 109 110
    /* ToDo: gracefully handle startupHaskell() failures.. */
    return TRUE;
  case DLL_PROCESS_DETACH:
    shutdownHaskell();
  }
  return TRUE;
}

#endif /* !ENABLE_WIN32_DLL_SUPPORT */