1 files changed, 164 insertions, 3 deletions

diff --git a/findutils/xargs.c b/findutils/xargs.c
index 890c37534..37064be9d 100644
--- a/findutils/xargs.c
+++ b/findutils/xargs.c
@@ -74,6 +74,11 @@
 
 //kbuild:lib-$(CONFIG_XARGS) += xargs.o
 
+#if ENABLE_PLATFORM_MINGW32
+#include <conio.h>
+#include "busybox.h"
+#include "NUM_APPLETS.h"
+#endif
 #include "libbb.h"
 #include "common_bufsiz.h"
 
@@ -111,11 +116,19 @@ struct globals {
 #endif
         const char *eof_str;
         int idx;
+#if !ENABLE_PLATFORM_MINGW32
         int fd_tty;
         int fd_stdin;
+#endif
 #if ENABLE_FEATURE_XARGS_SUPPORT_PARALLEL
         int running_procs;
         int max_procs;
+# if ENABLE_PLATFORM_MINGW32
+        HANDLE *procs;
+# endif
+#endif
+#if ENABLE_PLATFORM_MINGW32
+        pid_t pid;
 #endif
         smalluint xargs_exitcode;
 #if ENABLE_FEATURE_XARGS_SUPPORT_QUOTES
@@ -137,6 +150,7 @@ struct globals {
         G.idx = 0; \
         IF_FEATURE_XARGS_SUPPORT_PARALLEL(G.running_procs = 0;) \
         IF_FEATURE_XARGS_SUPPORT_PARALLEL(G.max_procs = 1;) \
+        IF_FEATURE_XARGS_SUPPORT_PARALLEL(IF_PLATFORM_MINGW32(G.procs = NULL;)) \
         G.xargs_exitcode = 0; \
         IF_FEATURE_XARGS_SUPPORT_QUOTES(G.process_stdin__state = NORM;) \
         IF_FEATURE_XARGS_SUPPORT_QUOTES(G.process_stdin__q = '\0';) \
@@ -150,7 +164,7 @@ enum {
         OPTBIT_UPTO_SIZE,
         OPTBIT_EOF_STRING,
         OPTBIT_EOF_STRING1,
-        OPTBIT_STDIN_TTY,
+        IF_NOT_PLATFORM_MINGW32(              OPTBIT_STDIN_TTY  ,)
         IF_FEATURE_XARGS_SUPPORT_CONFIRMATION(OPTBIT_INTERACTIVE,)
         IF_FEATURE_XARGS_SUPPORT_TERMOPT(     OPTBIT_TERMINATE  ,)
         IF_FEATURE_XARGS_SUPPORT_ZERO_TERM(   OPTBIT_ZEROTERM   ,)
@@ -163,14 +177,15 @@ enum {
         OPT_UPTO_SIZE   = 1 << OPTBIT_UPTO_SIZE  ,
         OPT_EOF_STRING  = 1 << OPTBIT_EOF_STRING , /* GNU: -e[<param>] */
         OPT_EOF_STRING1 = 1 << OPTBIT_EOF_STRING1, /* SUS: -E<param> */
-        OPT_STDIN_TTY   = 1 << OPTBIT_STDIN_TTY,
+        OPT_STDIN_TTY   = IF_NOT_PLATFORM_MINGW32(              (1 << OPTBIT_STDIN_TTY  )) + 0,
         OPT_INTERACTIVE = IF_FEATURE_XARGS_SUPPORT_CONFIRMATION((1 << OPTBIT_INTERACTIVE)) + 0,
         OPT_TERMINATE   = IF_FEATURE_XARGS_SUPPORT_TERMOPT(     (1 << OPTBIT_TERMINATE  )) + 0,
         OPT_ZEROTERM    = IF_FEATURE_XARGS_SUPPORT_ZERO_TERM(   (1 << OPTBIT_ZEROTERM   )) + 0,
         OPT_REPLSTR     = IF_FEATURE_XARGS_SUPPORT_REPL_STR(    (1 << OPTBIT_REPLSTR    )) + 0,
         OPT_REPLSTR1    = IF_FEATURE_XARGS_SUPPORT_REPL_STR(    (1 << OPTBIT_REPLSTR1   )) + 0,
 };
-#define OPTION_STR "+trn:s:e::E:o" \
+#define OPTION_STR "+trn:s:e::E:" \
+        IF_NOT_PLATFORM_MINGW32(              "o") \
         IF_FEATURE_XARGS_SUPPORT_CONFIRMATION("p") \
         IF_FEATURE_XARGS_SUPPORT_TERMOPT(     "x") \
         IF_FEATURE_XARGS_SUPPORT_ZERO_TERM(   "0") \
@@ -179,6 +194,83 @@ enum {
         IF_FEATURE_XARGS_SUPPORT_ARGS_FILE(   "a:")
 
 
+#if ENABLE_PLATFORM_MINGW32
+static BOOL WINAPI ctrl_handler(DWORD dwCtrlType)
+{
+        if (dwCtrlType == CTRL_C_EVENT || dwCtrlType == CTRL_BREAK_EVENT) {
+# if ENABLE_FEATURE_XARGS_SUPPORT_PARALLEL
+                if (G.max_procs == 1)
+# endif
+                {
+                        if (G.pid > 0)
+                                kill(-G.pid, SIGTERM);
+                }
+# if ENABLE_FEATURE_XARGS_SUPPORT_PARALLEL
+                else {
+                        int i;
+
+                        for (i = 0; i < G.running_procs; ++i) {
+                                pid_t pid = GetProcessId(G.procs[i]);
+                                if (pid > 0)
+                                        kill(-pid, SIGTERM);
+                        }
+                }
+# endif
+                exit(SIGINT << 24);
+                return TRUE;
+        }
+        return FALSE;
+}
+#endif
+
+#if ENABLE_FEATURE_XARGS_SUPPORT_PARALLEL && ENABLE_PLATFORM_MINGW32
+static int wait_for_slot(int *idx)
+{
+        int i;
+
+        /* if less than max_procs running, set status to 0, return next free slot */
+        if (G.running_procs < G.max_procs) {
+                *idx = G.running_procs++;
+                return 0;
+        }
+
+check_exit_codes:
+        for (i = G.running_procs - 1; i >= 0; i--) {
+                DWORD status = 0;
+                if (!GetExitCodeProcess(G.procs[i], &status) ||
+                                status != STILL_ACTIVE) {
+                        CloseHandle(G.procs[i]);
+                        if (i + 1 < G.running_procs)
+                                G.procs[i] = G.procs[G.running_procs - 1];
+                        *idx = G.running_procs - 1;
+                        if (!G.max_procs)
+                                G.running_procs--;
+                        return status;
+                }
+        }
+
+        if (G.running_procs < MAXIMUM_WAIT_OBJECTS)
+                WaitForMultipleObjects((DWORD)G.running_procs, G.procs, FALSE,
+                                INFINITE);
+        else {
+                /* Fall back to polling */
+                for (;;) {
+                        DWORD nr = i + MAXIMUM_WAIT_OBJECTS > G.running_procs ?
+                                MAXIMUM_WAIT_OBJECTS : (DWORD)(G.running_procs - i);
+                        DWORD ret = WaitForMultipleObjects(nr, G.procs + i, FALSE, 100);
+
+                        if (ret != WAIT_TIMEOUT)
+                                break;
+                        i += MAXIMUM_WAIT_OBJECTS;
+                        if (i > G.running_procs)
+                                i = 0;
+                }
+        }
+
+        goto check_exit_codes;
+}
+#endif /* SUPPORT_PARALLEL && PLATFORM_MINGW32 */
+
 /*
  * Returns 0 if xargs should continue (but may set G.xargs_exitcode to 123).
  * Else sets G.xargs_exitcode to error code and returns nonzero.
@@ -189,6 +281,7 @@ static int xargs_exec(void)
 {
         int status;
 
+#if !ENABLE_PLATFORM_MINGW32
         if (option_mask32 & OPT_STDIN_TTY)
                 xdup2(G.fd_tty, STDIN_FILENO);
 
@@ -240,6 +333,43 @@ static int xargs_exec(void)
                 }
         }
 #endif
+#endif
+
+#if ENABLE_PLATFORM_MINGW32
+# if ENABLE_FEATURE_XARGS_SUPPORT_PARALLEL
+        if (G.max_procs == 1) {
+# endif
+# if ENABLE_FEATURE_PREFER_APPLETS && (NUM_APPLETS > 1)
+                int applet = find_applet_by_name(G.args[0]);
+                if (applet >= 0 && APPLET_IS_NOFORK(applet)) {
+                        status = run_nofork_applet(applet, G.args);
+                } else
+# endif
+                {
+                        G.pid = spawn(G.args);
+                        status = G.pid < 0 ? -1 : wait4pid(G.pid);
+                }
+# if ENABLE_FEATURE_XARGS_SUPPORT_PARALLEL
+        }
+        else {
+                int idx;
+                status = !G.running_procs && !G.max_procs ? 0 : wait_for_slot(&idx);
+                if (G.max_procs) {
+                        HANDLE p = (HANDLE)mingw_spawn_proc((const char **)G.args);
+                        if (p < 0)
+                                status = -1;
+                        else
+                                G.procs[idx] = p;
+                } else {
+                        while (G.running_procs) {
+                                int status2 = wait_for_slot(&idx);
+                                if (status2 && !status)
+                                        status = status2;
+                        }
+                }
+        }
+# endif
+#endif
         /* Manpage:
          * """xargs exits with the following status:
          * 0 if it succeeds
@@ -278,8 +408,10 @@ static int xargs_exec(void)
  ret:
         if (status != 0)
                 G.xargs_exitcode = status;
+#if !ENABLE_PLATFORM_MINGW32
         if (option_mask32 & OPT_STDIN_TTY)
                 xdup2(G.fd_stdin, STDIN_FILENO);
+#endif
         return status;
 }
 
@@ -559,6 +691,7 @@ static char* FAST_FUNC process_stdin_with_replace(int n_max_chars, int n_max_arg
  */
 static int xargs_ask_confirmation(void)
 {
+#if !ENABLE_PLATFORM_MINGW32
         FILE *tty_stream;
         int r;
 
@@ -568,6 +701,18 @@ static int xargs_ask_confirmation(void)
         r = bb_ask_y_confirmation_FILE(tty_stream);
 
         fclose(tty_stream);
+#else
+        int r, c, savec;
+
+        fputs(" ?...", stderr);
+        fflush_all();
+        c = savec = getche();
+        while (c != EOF && c != '\r')
+                c = getche();
+        fputs("\n", stderr);
+        fflush_all();
+        r = (savec == 'y' || savec == 'Y');
+#endif
 
         return r;
 }
@@ -585,7 +730,9 @@ static int xargs_ask_confirmation(void)
 //usage:        IF_FEATURE_XARGS_SUPPORT_ARGS_FILE(
 //usage:     "\n        -a FILE Read from FILE instead of stdin"
 //usage:        )
+//usage:        IF_NOT_PLATFORM_MINGW32(
 //usage:     "\n        -o      Reopen stdin as /dev/tty"
+//usage:        )
 //usage:     "\n        -r      Don't run command if input is empty"
 //usage:     "\n        -t      Print the command on stderr before execution"
 //usage:        IF_FEATURE_XARGS_SUPPORT_CONFIRMATION(
@@ -628,6 +775,9 @@ int xargs_main(int argc UNUSED_PARAM, char **argv)
 
         INIT_G();
 
+#if ENABLE_PLATFORM_MINGW32
+        SetConsoleCtrlHandler(ctrl_handler, TRUE);
+#endif
         opt = getopt32long(argv, OPTION_STR,
                 "no-run-if-empty\0" No_argument "r",
                 &max_args, &max_chars, &G.eof_str, &G.eof_str
@@ -638,7 +788,12 @@ int xargs_main(int argc UNUSED_PARAM, char **argv)
 
 #if ENABLE_FEATURE_XARGS_SUPPORT_PARALLEL
         if (G.max_procs <= 0) /* -P0 means "run lots of them" */
+#if !ENABLE_PLATFORM_MINGW32
                 G.max_procs = 100; /* let's not go crazy high */
+#else
+                G.max_procs = MAXIMUM_WAIT_OBJECTS;
+        G.procs = xmalloc(sizeof(G.procs[0]) * G.max_procs);
+#endif
 #endif
 
 #if ENABLE_FEATURE_XARGS_SUPPORT_ARGS_FILE
@@ -736,12 +891,14 @@ int xargs_main(int argc UNUSED_PARAM, char **argv)
                         store_param(argv[i]);
         }
 
+#if !ENABLE_PLATFORM_MINGW32
         if (opt & OPT_STDIN_TTY) {
                 G.fd_tty = xopen(CURRENT_TTY, O_RDONLY);
                 close_on_exec_on(G.fd_tty);
                 G.fd_stdin = dup(STDIN_FILENO);
                 close_on_exec_on(G.fd_stdin);
         }
+#endif
 
         initial_idx = G.idx;
         while (1) {
@@ -767,7 +924,11 @@ int xargs_main(int argc UNUSED_PARAM, char **argv)
                                 fmt = " %s";
                         }
                         if (!(opt & OPT_INTERACTIVE))
+#if !ENABLE_PLATFORM_MINGW32
                                 bb_putchar_stderr('\n');
+#else
+                                fprintf(stderr, "\n");
+#endif
                 }
 
                 if (!(opt & OPT_INTERACTIVE) || xargs_ask_confirmation()) {