Changeset 2844 for trunk/src/kmk/kmkbuiltin

Timestamp:

Aug 29, 2016, 6:31:33 PM (9 years ago)

Author:

bird

Message:

kSubmit/kWorker: updates

File:

• trunk/src/kmk/kmkbuiltin/kSubmit.c (modified) (14 diffs)

trunk/src/kmk/kmkbuiltin/kSubmit.c

@@ -52 +52 @@
 #ifdef KBUILD_OS_WINDOWS
 # include "sub_proc.h"
+# include "quote_argv.h"
 #endif
 
@@ -366 +367 @@
                 if (pWorker->OverlappedRead.hEvent != NULL)
                 {
-                    char        szHandleArg[16];
+                    char        szHandleArg[32];
                     const char *apszArgs[4] = { szExecutable, "--pipe", szHandleArg, NULL };
                     _snprintf(szHandleArg, sizeof(szHandleArg), "%p", hWorkerPipe);
@@ -557 +558 @@
                                       const char *pszCwd, uint32_t *pcbMsg)
 {
-    size_t   i;
     size_t   cbTmp;
+    uint32_t i;
     uint32_t cbMsg;
+    uint32_t cArgs;
+    uint32_t cEnvVars;
     uint8_t *pbMsg;
     uint8_t *pbCursor;
+
+    /*
+     * Adjust input.
+     */
+    if (!pszExecutable)
+        pszExecutable = papszArgs[0];
 
     /*
@@ -569 +578 @@
     cbMsg += sizeof("JOB");
     cbMsg += strlen(pszExecutable) + 1;
-
+    cbMsg += strlen(pszCwd) + 1;
+
+    cbMsg += sizeof(cArgs);
     for (i = 0; papszArgs[i] != NULL; i++)
-        cbMsg += strlen(papszArgs[i]) + 1;
-    cbMsg += 1;
-
+        cbMsg += 1 + strlen(papszArgs[i]) + 1;
+    cArgs  = i;
+
+    cbMsg += sizeof(cArgs);
     for (i = 0; papszEnvVars[i] != NULL; i++)
         cbMsg += strlen(papszEnvVars[i]) + 1;
-    cbMsg += 1;
-
-    cbMsg += strlen(pszCwd) + 1;
+    cEnvVars = i;
+
 
     /*
@@ -594 +605 @@
     pbCursor += cbTmp;
 
+    cbTmp = strlen(pszCwd) + 1;
+    memcpy(pbCursor, pszCwd, cbTmp);
+    pbCursor += cbTmp;
+
+    memcpy(pbCursor, &cArgs, sizeof(cArgs));
+    pbCursor += sizeof(cArgs);
     for (i = 0; papszArgs[i] != NULL; i++)
     {
+        *pbCursor++ = 0; /* Argument expansion flags (MSC, EMX). */
         cbTmp = strlen(papszArgs[i]) + 1;
         memcpy(pbCursor, papszArgs[i], cbTmp);
         pbCursor += cbTmp;
     }
-    *pbCursor++ = '\0';
-
+    assert(i == cArgs);
+
+    memcpy(pbCursor, &cEnvVars, sizeof(cEnvVars));
+    pbCursor += sizeof(cEnvVars);
     for (i = 0; papszEnvVars[i] != NULL; i++)
     {
@@ -608 +628 @@
         pbCursor += cbTmp;
     }
-    *pbCursor++ = '\0';
-
-    cbTmp = strlen(pszCwd) + 1;
-    memcpy(pbCursor, pszCwd, cbTmp);
-    pbCursor += cbTmp;
+    assert(i == cEnvVars);
 
     assert(pbCursor - pbMsg == (size_t)cbMsg);
@@ -780 +796 @@
 /**
  * Used by
- * @returns 0 if we got the whole result, -1 if I/O is pending, windows last
+ * @returns 0 if we got the whole result, -1 if I/O is pending, and windows last
  *          error on ReadFile failure.
  * @param   pWorker             The worker instance.
@@ -806 +822 @@
             DWORD dwErr = GetLastError();
             if (dwErr == ERROR_IO_PENDING)
-                return 1;
-            return kSubmitWinReadFailed(pWorker, GetLastError());
+                return -1;
+            return kSubmitWinReadFailed(pWorker, dwErr);
         }
 
@@ -849 +865 @@
     if (rc == -1)
     {
-        if (process_kmk_register_submit(pWorker->OverlappedRead.hEvent, (intptr_t)pWorker) == 0)
+        if (process_kmk_register_submit(pWorker->OverlappedRead.hEvent, (intptr_t)pWorker, pPidSpawned) == 0)
         { /* likely */ }
         else
@@ -873 +889 @@
      */
     pWorker->pBusyWith = pChild;
+#ifndef KBUILD_OS_WINDOWS
     *pPidSpawned = pWorker->pid;
+#endif
 
     kSubmitListUnlink(&g_IdleList, pWorker);
@@ -977 +995 @@
         PWORKERINSTANCE apWorkers[MAXIMUM_WAIT_OBJECTS];
         HANDLE          ahHandles[MAXIMUM_WAIT_OBJECTS];
-        DWORD           cHandles;
+        DWORD           cHandles = 0;
 
         for (pWorker = g_IdleList.pHead; pWorker != NULL; pWorker = pWorker->pNext)
@@ -1329 +1347 @@
 
     papszEnv = pChild->environment;
-    if (papszEnv)
+    if (!papszEnv)
         pChild->environment = papszEnv = target_environment(pChild->file);
     cEnvVars = 0;
@@ -1495 +1513 @@
         if (pWorker)
         {
+#ifdef KBUILD_OS_WINDOWS
+            /* Quote the argv elements, but first we need unquoted pszExecute. */
+            char *pszFreeExec = NULL;
+            if (!pszExecutable)
+                pszExecutable = pszFreeExec = xstrdup(argv[0]);
+            quote_argv(argc, argv, fWatcomBrainDamage, 1 /*fFreeOrLeak*/);
+#endif
+
             rcExit = kSubmitSendJobMessage(pWorker, pvMsg, cbMsg, 0 /*fNoRespawning*/, cVerbosity);
             if (rcExit == 0)
@@ -1502 +1528 @@
                 if (atexit(kSubmitAtExitCallback) == 0)
                     g_fAtExitRegistered = 1;
+
+#ifdef KBUILD_OS_WINDOWS
+            free(pszFreeExec);
+#endif
         }
         else