X-Git-Url: http://git.megacz.com/?a=blobdiff_plain;f=rts%2Fwin32%2FIOManager.c;h=c9a759fffbce2655f971f220272fb409ed7b4a06;hb=fff1f6194c3c39de53cd645bda9865fb131b1c68;hp=764be698722627b2b92c5cbfa20ae04b38e919ff;hpb=de6c8e5293c9ef68b597ab2e6d55c3f42a283489;p=ghc-hetmet.git

diff --git a/rts/win32/IOManager.c b/rts/win32/IOManager.c
index 764be69..c9a759f 100644
--- a/rts/win32/IOManager.c
+++ b/rts/win32/IOManager.c
@@ -16,6 +16,7 @@
 #include <io.h>
 #include <winsock.h>
 #include <process.h>
+#include <errno.h>
 
 /*
  * Internal state maintained by the IO manager.
@@ -181,7 +182,15 @@ IOWorkerProc(PVOID param)
 			len = write(work->workData.ioData.fd,
 				    work->workData.ioData.buf,
 				    work->workData.ioData.len);
-			if (len == -1) { errCode = errno; }
+			if (len == -1) {
+                            errCode = errno;
+                            // write() gets errno wrong for
+                            // ERROR_NO_DATA, we have to fix it here:
+                            if (errCode == EINVAL &&
+                                GetLastError() == ERROR_NO_DATA) {
+                                errCode = EPIPE;
+                            }
+                        }
 		    }
 		    complData = work->workData.ioData.buf;
 		    fd = work->workData.ioData.fd;
@@ -190,7 +199,7 @@ IOWorkerProc(PVOID param)
 		     * 
 		     * Note: Sleep() is in milliseconds, not micros.
 		     */
-		    Sleep(work->workData.delayData.msecs / 1000);
+		    Sleep((work->workData.delayData.msecs + 999) / 1000);
 		    len = work->workData.delayData.msecs;
 		    complData = NULL;
 		    fd = 0;
@@ -443,25 +452,29 @@ AddProcRequest ( void* proc,
     return depositWorkItem(reqID, wItem);
 }
 
-void ShutdownIOManager ( void )
+void ShutdownIOManager ( rtsBool wait_threads )
 {
     int num;
 
     SetEvent(ioMan->hExitEvent);
   
-    /* Wait for all worker threads to die. */
-    for (;;) {
-        EnterCriticalSection(&ioMan->manLock);
-	num = ioMan->numWorkers;
-	LeaveCriticalSection(&ioMan->manLock);
-	if (num == 0)
-	    break;
-	Sleep(10);
+    if (wait_threads) {
+        /* Wait for all worker threads to die. */
+        for (;;) {
+            EnterCriticalSection(&ioMan->manLock);
+            num = ioMan->numWorkers;
+            LeaveCriticalSection(&ioMan->manLock);
+            if (num == 0)
+                break;
+            Sleep(10);
+        }
+        FreeWorkQueue(ioMan->workQueue);
+        CloseHandle(ioMan->hExitEvent);
+        DeleteCriticalSection(&ioMan->active_work_lock);
+        DeleteCriticalSection(&ioMan->manLock);
+        free(ioMan);
+        ioMan = NULL;
     }
-    FreeWorkQueue(ioMan->workQueue);
-    CloseHandle(ioMan->hExitEvent);
-    free(ioMan);
-    ioMan = NULL;
 }
 
 /* Keep track of WorkItems currently being serviced. */