From af2db474c2bb80e29924430e3c730bc217e55189 Mon Sep 17 00:00:00 2001
From: Ian Lynagh <igloo@earth.li>
Date: Mon, 20 Aug 2007 22:27:01 +0000
Subject: [PATCH] Fix ghc-inplace.c on Windows; new code based on ghci.c

---
 compiler/Makefile      |    8 ++-
 compiler/ghc-inplace.c |  126 +++++++++++++++++++++++++++++++++++++++++++++++-
 2 files changed, 130 insertions(+), 4 deletions(-)

diff --git a/compiler/Makefile b/compiler/Makefile
index e881216..3f443ac 100644
--- a/compiler/Makefile
+++ b/compiler/Makefile
@@ -720,10 +720,14 @@ EXCLUDED_C_SRCS += ghc-inplace.c
 # will go wrong when we use it in a Haskell string below.
 TOP_ABS=$(subst \\,/,$(FPTOOLS_TOP_ABS_PLATFORM))
 
-GHC_PATH=$(TOP_ABS)/$(GHC_COMPILER_DIR_REL)/$(GHC_PROG)
+GHC_PATH=$(TOP_ABS)/$(GHC_COMPILER_DIR_REL)/$(GHC_PROG)$(exeext)
+
+ifeq "$(HOSTPLATFORM)" "i386-unknown-mingw32"
+INPLACE_EXTRA_FLAGS = -optc-DWINDOWS
+endif
 
 $(INPLACE_PROG): ghc-inplace.c
-	$(HC) -cpp -optc-DGHC_PATH=\"$(GHC_PATH)\" -optc-DTOP_ABS=\"$(TOP_ABS)\" $< -o $@
+	$(HC) -cpp -optc-DGHC_PATH=\"$(GHC_PATH)\" -optc-DTOP_ABS=\"$(TOP_ABS)\" $(INPLACE_EXTRA_FLAGS) $< -o $@
 
 all :: $(INPLACE_PROG)
 
diff --git a/compiler/ghc-inplace.c b/compiler/ghc-inplace.c
index 0129258..78cffba 100644
--- a/compiler/ghc-inplace.c
+++ b/compiler/ghc-inplace.c
@@ -1,9 +1,29 @@
 
 #include <stdio.h>
+#include <stdarg.h>
 #include <stdlib.h>
 #include <string.h>
 #include <unistd.h>
 
+#ifdef WINDOWS
+#include <windows.h>
+#include <process.h>
+#include <malloc.h>
+#include <signal.h>
+#include <io.h>
+#endif
+
+int run(char *this, char *program, int argc, char **argv);
+
+void error(const char *fmt, ...) {
+    va_list argp;
+    va_start(argp, fmt);
+    vfprintf(stderr, fmt, argp);
+    va_end(argp);
+    fprintf(stderr, "\n");
+    fflush(stderr);
+}
+
 int main(int argc, char **argv) {
     char **args;
     args = malloc(sizeof(char *) * (argc + 3));
@@ -19,6 +39,108 @@ int main(int argc, char **argv) {
     }
     memcpy(args + 3, argv + 1, sizeof(char *) * (argc - 1));
     args[argc+2] = NULL;
-    execv(GHC_PATH, args);
-    return 0;
+    return run(argv[0], GHC_PATH, argc + 2, args);
+}
+
+#ifndef WINDOWS
+int run(char *this, char *program, int argc, char** argv) {
+    execv(program, argv);
+    return 1; /* Not reached */
 }
+#else
+int run(char *this, char *program, int argc, char** argv) {
+    TCHAR  programShort[MAX_PATH+1];
+    DWORD  dwSize;
+    int    i;
+    char*  new_cmdline;
+    char   *ptr;
+    char   *src;
+    unsigned int cmdline_len;
+
+    STARTUPINFO si;
+    PROCESS_INFORMATION pi;
+  
+    ZeroMemory(&si, sizeof(STARTUPINFO));
+    ZeroMemory(&pi, sizeof(PROCESS_INFORMATION));
+
+    si.cb = sizeof(STARTUPINFO);
+
+    dwSize = MAX_PATH;
+    /* Turn the path into short form - LFN form causes problems
+       when passed in argv[0]. */
+    if ( !(GetShortPathName(program, programShort, dwSize)) ) {
+        error("%s: Unable to locate %s", this, program);
+        return 1;
+    }
+  
+    /* Compute length of the flattened 'argv', including spaces! */
+    cmdline_len = 0;
+    for(i = 1; i < argc; i++) {
+        /* Note: play it safe and quote all argv strings */
+        cmdline_len += 1 + strlen(argv[i]) + 2;
+    }
+    new_cmdline = (char*)malloc(sizeof(char) * (cmdline_len + 1));
+    if (!new_cmdline) {
+        error("%s: failed to start up ghc.exe; insufficient memory", this);
+        return 1;
+    }
+  
+    ptr = new_cmdline;
+    for(i = 1; i < argc; i++) {
+        *ptr++ = ' ';
+        *ptr++ = '"';
+        src = argv[i];
+        while(*src) {
+            *ptr++ = *src++;
+        }
+        *ptr++ = '"';
+    }
+    *ptr = '\0';
+  
+    /* Note: Used to use _spawnv(_P_WAIT, ...) here, but it suffered
+       from the parent intercepting console events such as Ctrl-C,
+       which it shouldn't. Installing an ignore-all console handler
+       didn't do the trick either.
+
+       Irrespective of this issue, using CreateProcess() is preferable,
+       as it makes this wrapper work on both mingw and cygwin.
+    */
+#if 0
+    fprintf(stderr, "Invoking ghc: %s %s\n", programShort, new_cmdline);
+    fflush(stderr);
+#endif
+    if (!CreateProcess(programShort,
+                       new_cmdline,
+                       NULL,
+                       NULL,
+                       TRUE,
+                       0, /* dwCreationFlags */
+                       NULL, /* lpEnvironment */
+                       NULL, /* lpCurrentDirectory */
+                       &si,  /* lpStartupInfo */
+                       &pi) ) {
+        error("%s: Unable to start ghc.exe (error code: %lu)",
+              this, GetLastError());
+        return 1;
+    }
+    /* Disable handling of console events in the parent by dropping its
+     * connection to the console. This has the (minor) downside of not being
+     * able to subsequently emit any error messages to the console.
+     */
+    FreeConsole();
+
+    switch (WaitForSingleObject(pi.hProcess, INFINITE) ) {
+        case WAIT_OBJECT_0:
+            return 0;
+        case WAIT_ABANDONED:
+        case WAIT_FAILED:
+            /* in the event we get any hard errors, bring the child
+               to a halt. */
+            TerminateProcess(pi.hProcess, 1);
+            return 1;
+        default:
+            return 1;
+    }
+}
+#endif
+
-- 
1.7.10.4