Changeset 2733 in kBuild

Timestamp:

Oct 16, 2014 6:29:41 PM (11 years ago)

Author:

bird

Message:

mscfakes.c: Make sure STDOUT and STDERR are blocking before we use them. Fixes #120.

File:

• trunk/src/kmk/kmkbuiltin/mscfakes.c (modified) (2 diffs)

trunk/src/kmk/kmkbuiltin/mscfakes.c

@@ -110 +110 @@
 
 int
-birdSetErrno(DWORD dwErr)
+birdSetErrno(unsigned dwErr)
 {
     switch (dwErr)
@@ -535 +535 @@
 
 
+/**
+ * This is a kludge to make pipe handles blocking.
+ *
+ * @returns TRUE if it's now blocking, FALSE if not a pipe or we failed to fix
+ *          the blocking mode.
+ * @param   fd                  The libc file descriptor number.
+ */
+static BOOL makePipeBlocking(int fd)
+{
+    /* Is pipe? */
+    HANDLE hFile = (HANDLE)_get_osfhandle(fd);
+    if (hFile != INVALID_HANDLE_VALUE)
+    {
+        DWORD fType = GetFileType(hFile);
+        fType &= ~FILE_TYPE_REMOTE;
+        if (fType == FILE_TYPE_PIPE)
+        {
+            /* Try fix it. */
+            DWORD fState = 0;
+            if (GetNamedPipeHandleState(hFile, &fState, NULL, NULL, NULL, NULL,  0))
+            {
+                fState &= ~PIPE_NOWAIT;
+                fState |= PIPE_WAIT;
+                if (SetNamedPipeHandleState(hFile, &fState, NULL, NULL))
+                    return TRUE;
+            }
+        }
+    }
+    return FALSE;
+}
+
+
+/**
+ * Initializes the msc fake stuff.
+ * @returns 0 on success (non-zero would indicate failure, see rterr.h).
+ */
+int mscfake_init(void)
+{
+    /*
+     * Kludge against _write returning ENOSPC on non-blocking pipes.
+     */
+    makePipeBlocking(STDOUT_FILENO);
+    makePipeBlocking(STDERR_FILENO);
+
+    return 0;
+}
+
+/*
+ * Do this before main is called.
+ */
+#pragma section(".CRT$XIA", read)
+#pragma section(".CRT$XIU", read)
+#pragma section(".CRT$XIZ", read)
+typedef int (__cdecl *PFNCRTINIT)(void);
+static __declspec(allocate(".CRT$XIU")) PFNCRTINIT g_MscFakeInitVectorEntry = mscfake_init;
+