1 files changed, 208 insertions, 3 deletions

diff --git a/libbb/appletlib.c b/libbb/appletlib.c
index 542211255..320cb5b13 100644
--- a/libbb/appletlib.c
+++ b/libbb/appletlib.c
@@ -53,6 +53,15 @@ static inline int *get_perrno(void) { return &errno; }
 # define IF_FEATURE_INDIVIDUAL(...) __VA_ARGS__
 #endif
 
+#if (ENABLE_FEATURE_INSTALLER && !ENABLE_PLATFORM_MINGW32) || \
+        (ENABLE_PLATFORM_MINGW32 && (ENABLE_FEATURE_PREFER_APPLETS \
+                || ENABLE_FEATURE_SH_STANDALONE \
+                || ENABLE_FEATURE_SH_NOFORK))
+# define IF_FULL_LIST_OPTION(...) __VA_ARGS__
+#else
+# define IF_FULL_LIST_OPTION(...)
+#endif
+
 #include "usage_compressed.h"
 
 #if ENABLE_FEATURE_SH_EMBEDDED_SCRIPTS
@@ -62,6 +71,7 @@ static inline int *get_perrno(void) { return &errno; }
 # define NUM_SCRIPTS 0
 #endif
 #if NUM_SCRIPTS > 0
+# define BB_ARCHIVE_PUBLIC
 # include "bb_archive.h"
 static const char packed_scripts[] ALIGN1 = { PACKED_SCRIPTS };
 #endif
@@ -111,6 +121,7 @@ static const char usage_messages[] ALIGN1 = UNPACKED_USAGE;
 #if ENABLE_FEATURE_COMPRESS_USAGE
 
 static const char packed_usage[] ALIGN1 = { PACKED_USAGE };
+# define BB_ARCHIVE_PUBLIC
 # include "bb_archive.h"
 # define unpack_usage_messages() \
         unpack_bz2_data(packed_usage, sizeof(packed_usage), sizeof(UNPACKED_USAGE))
@@ -154,7 +165,11 @@ void FAST_FUNC bb_show_usage(void)
                         ap--;
                 }
                 full_write2_str(bb_banner);
+#if ENABLE_PLATFORM_MINGW32
+                full_write2_str(" multi-call binary\n");
+#else
                 full_write2_str(" multi-call binary.\n"); /* common string */
+#endif
                 if (*p == '\b')
                         full_write2_str("\nNo help available\n");
                 else {
@@ -657,6 +672,7 @@ static const char *const install_dir[] = {
 #  endif
 };
 
+#  if !ENABLE_PLATFORM_MINGW32
 /* create (sym)links for each applet */
 static void install_links(const char *busybox, int use_symbolic_links,
                 char *custom_install_dir)
@@ -689,6 +705,42 @@ static void install_links(const char *busybox, int use_symbolic_links,
                         continue;
         }
 }
+#  else /* ENABLE_PLATFORM_MINGW32 */
+static void install_links(const char *busybox,
+                int use_symbolic_links UNUSED_PARAM, char *custom_install_dir)
+{
+        char *fpc;
+        const char *appname = applet_names;
+        const char *sd = NULL;
+        int i, rc;
+
+        if (custom_install_dir != NULL) {
+                bb_make_directory(custom_install_dir, 0755, FILEUTILS_RECUR);
+        }
+        else {
+                sd = get_system_drive();
+                for (i=1; i<ARRAY_SIZE(install_dir); ++i) {
+                        fpc = xasprintf("%s%s", sd ?: "", install_dir[i]);
+                        bb_make_directory(fpc, 0755, FILEUTILS_RECUR);
+                        free(fpc);
+                }
+        }
+
+        for (i = 0; i < ARRAY_SIZE(applet_main); i++) {
+                fpc = xasprintf("%s%s/%s.exe", sd ?: "",
+                                custom_install_dir ?: install_dir[APPLET_INSTALL_LOC(i)],
+                                appname);
+                rc = link(busybox, fpc);
+                if (rc != 0 && (errno != EEXIST ||
+                                strcmp("busybox.exe", bb_basename(fpc)) != 0)) {
+                        bb_simple_perror_msg(fpc);
+                }
+                free(fpc);
+                while (*appname++ != '\0')
+                        continue;
+        }
+}
+#  endif
 # elif ENABLE_BUSYBOX
 static void install_links(const char *busybox UNUSED_PARAM,
                 int use_symbolic_links UNUSED_PARAM,
@@ -775,20 +827,35 @@ int busybox_main(int argc UNUSED_PARAM, char **argv)
 
                 dup2(1, 2);
                 full_write2_str(bb_banner); /* reuse const string */
+#if ENABLE_PLATFORM_MINGW32
+                full_write2_str(" multi-call binary\n"); /* reuse */
+#else
                 full_write2_str(" multi-call binary.\n"); /* reuse */
+#endif
+#if defined(MINGW_VER)
+                if (sizeof(MINGW_VER) > 5) {
+                        full_write2_str(MINGW_VER "\n\n");
+                }
+#endif
                 full_write2_str(
-                        "BusyBox is copyrighted by many authors between 1998-2015.\n"
+                        "BusyBox is copyrighted by many authors between 1998-2021.\n"
                         "Licensed under GPLv2. See source distribution for detailed\n"
                         "copyright notices.\n"
                         "\n"
                         "Usage: busybox [function [arguments]...]\n"
-                        "   or: busybox --list"IF_FEATURE_INSTALLER("[-full]")"\n"
+                        "   or: busybox --list"IF_FULL_LIST_OPTION("[-full]")"\n"
 #  if ENABLE_FEATURE_SHOW_SCRIPT && NUM_SCRIPTS > 0
                         "   or: busybox --show SCRIPT\n"
 #  endif
                         IF_FEATURE_INSTALLER(
+                        IF_NOT_PLATFORM_MINGW32(
                         "   or: busybox --install [-s] [DIR]\n"
                         )
+                        IF_PLATFORM_MINGW32(
+                        "   or: busybox --install [-u|DIR]\n"
+                        "   or: busybox --uninstall [-n] file\n"
+                        )
+                        )
                         "   or: function [arguments]...\n"
                         "\n"
                         IF_NOT_FEATURE_SH_STANDALONE(
@@ -849,9 +916,28 @@ int busybox_main(int argc UNUSED_PARAM, char **argv)
                 const char *a = applet_names;
                 dup2(1, 2);
                 while (*a) {
-#  if ENABLE_FEATURE_INSTALLER
+#  if ENABLE_FEATURE_INSTALLER && !ENABLE_PLATFORM_MINGW32
                         if (argv[1][6]) /* --list-full? */
                                 full_write2_str(install_dir[APPLET_INSTALL_LOC(i)] + 1);
+#  elif ENABLE_PLATFORM_MINGW32 && (ENABLE_FEATURE_PREFER_APPLETS \
+                || ENABLE_FEATURE_SH_STANDALONE \
+                || ENABLE_FEATURE_SH_NOFORK)
+                        if (argv[1][6]) { /* --list-full? */
+                                const char *str;
+
+                                if (APPLET_IS_NOFORK(i))
+                                        str = "NOFORK  ";
+                                else if (APPLET_IS_NOEXEC(i))
+                                        str = "noexec  ";
+#   if NUM_SCRIPTS > 0
+                                else if (applet_main[i] == scripted_main)
+                                        str = "script  ";
+#   endif
+                                else
+                                        str = "        ";
+                                full_write2_str(str);
+                                full_write2_str(install_dir[APPLET_INSTALL_LOC(i)] + 1);
+                        }
 #  endif
                         full_write2_str(a);
                         full_write2_str("\n");
@@ -863,6 +949,7 @@ int busybox_main(int argc UNUSED_PARAM, char **argv)
         }
 
         if (ENABLE_FEATURE_INSTALLER && strcmp(argv[1], "--install") == 0) {
+#if !ENABLE_PLATFORM_MINGW32
                 int use_symbolic_links;
                 const char *busybox;
 
@@ -883,8 +970,50 @@ int busybox_main(int argc UNUSED_PARAM, char **argv)
                  */
                 use_symbolic_links = (argv[2] && strcmp(argv[2], "-s") == 0 && ++argv);
                 install_links(busybox, use_symbolic_links, argv[2]);
+#else
+                char *target = NULL;
+
+                /* busybox --install [-u|DIR]
+                 * -u: install to Unix-style directories in system drive
+                 * DIR: directory to install links to
+                 * If no argument is provided put the links in the same directory
+                 * as busybox.
+                 */
+                if (argv[2]) {
+                        if (strcmp(argv[2], "-u") != 0)
+                                target = argv[2];
+                }
+                else {
+                        target = dirname(xstrdup(bb_busybox_exec_path));
+                }
+                /* NULL target -> install to Unix-style dirs */
+                install_links(bb_busybox_exec_path, FALSE, target);
+#endif
+                return 0;
+        }
+
+#if ENABLE_PLATFORM_MINGW32 && ENABLE_FEATURE_INSTALLER
+        if (strcmp(argv[1], "--uninstall") == 0) {
+                char name[PATH_MAX];
+                int dry_run = (argv[2] && strcmp(argv[2], "-n") == 0 && ++argv);
+                const char *file = argv[2];
+
+                if (!argv[2])
+                        bb_error_msg_and_die(bb_msg_requires_arg, "--uninstall");
+
+                while (enumerate_links(file, name)) {
+                        if (dry_run) {
+                                full_write1_str(name);
+                                full_write1_str("\n");
+                        }
+                        else if (unlink(name) != 0) {
+                                bb_simple_perror_msg(name);
+                        }
+                        file = NULL;
+                }
                 return 0;
         }
+#endif
 
         if (strcmp(argv[1], "--help") == 0) {
                 /* "busybox --help [<applet>]" */
@@ -905,15 +1034,42 @@ int busybox_main(int argc UNUSED_PARAM, char **argv)
 # endif
 
 # if NUM_APPLETS > 0
+
+#  if ENABLE_PLATFORM_MINGW32
+static int interp = 0;
+char bb_comm[COMM_LEN];
+char bb_command_line[128];
+#  endif
+
 void FAST_FUNC run_applet_no_and_exit(int applet_no, const char *name, char **argv)
 {
         int argc = string_array_len(argv);
+#  if ENABLE_PLATFORM_MINGW32
+        int i;
+        const char *vmask;
+        unsigned int mask;
+#  endif
 
         /*
          * We do not use argv[0]: do not want to repeat massaging of
          * "-/sbin/halt" -> "halt", for example.
          */
         applet_name = name;
+#  if ENABLE_PLATFORM_MINGW32
+        safe_strncpy(bb_comm,
+                                        interp ? bb_basename(argv[interp]) : applet_name,
+                                        sizeof(bb_comm));
+
+        safe_strncpy(bb_command_line, applet_name, sizeof(bb_command_line));
+        for (i=1; i < argc && argv[i] &&
+                        strlen(bb_command_line) + strlen(argv[i]) + 2 < 128; ++i) {
+                strcat(strcat(bb_command_line, " "), argv[i]);
+        }
+
+        vmask = getenv("BB_UMASK");
+        if (vmask && sscanf(vmask, "%o", &mask) == 1)
+                umask((mode_t)(mask&0777));
+#  endif
 
         /* Special case. POSIX says "test --help"
          * should be no different from e.g. "test --foo".
@@ -930,6 +1086,9 @@ void FAST_FUNC run_applet_no_and_exit(int applet_no, const char *name, char **ar
 #  if defined APPLET_NO_false
          && applet_no != APPLET_NO_false
 #  endif
+#  if defined APPLET_NO_busybox
+         && applet_no != APPLET_NO_busybox
+#  endif
         ) {
                 if (argc == 2 && strcmp(argv[1], "--help") == 0) {
                         /* Make "foo --help" exit with 0: */
@@ -939,6 +1098,7 @@ void FAST_FUNC run_applet_no_and_exit(int applet_no, const char *name, char **ar
         }
         if (ENABLE_FEATURE_SUID)
                 check_suid(applet_no);
+
         xfunc_error_retval = applet_main[applet_no](argc, argv);
         /* Note: applet_main() may also not return (die on a xfunc or such) */
         xfunc_die();
@@ -1004,6 +1164,9 @@ int lbb_main(char **argv)
 int main(int argc UNUSED_PARAM, char **argv)
 #endif
 {
+#if ENABLE_PLATFORM_MINGW32
+        char *s;
+#endif
 #if 0
         /* TODO: find a use for a block of memory between end of .bss
          * and end of page. For example, I'm getting "_end:0x812e698 2408 bytes"
@@ -1047,6 +1210,38 @@ int main(int argc UNUSED_PARAM, char **argv)
                 argv[0][0] &= 0x7f;
         }
 #endif
+#if ENABLE_PLATFORM_MINGW32
+        /* detect if we're running an interpreted script */
+        if (argv[0][1] == ':' && argv[0][2] == '/') {
+                switch (argv[0][0]) {
+                case '2':
+                        ++interp;
+                        /* fall through */
+                case '1':
+                        ++interp;
+                        argv[0] += 3;
+                        break;
+                }
+        }
+# if ENABLE_FEATURE_EURO
+        init_codepage();
+# endif
+        /* Ignore critical errors, such as calling GetVolumeInformation() on
+         * a floppy or CDROM drive with no media. */
+        SetErrorMode(SEM_FAILCRITICALERRORS);
+#endif
+
+#if defined(__MINGW64_VERSION_MAJOR)
+        if ( stdin ) {
+                _setmode(fileno(stdin), _O_BINARY);
+        }
+        if ( stdout ) {
+                _setmode(fileno(stdout), _O_BINARY);
+        }
+        if ( stderr ) {
+                _setmode(fileno(stderr), _O_BINARY);
+        }
+#endif
 
 #if defined(SINGLE_APPLET_MAIN)
 
@@ -1071,6 +1266,10 @@ int main(int argc UNUSED_PARAM, char **argv)
 
 #else
 
+# if ENABLE_PLATFORM_MINGW32
+        if (argv[1] && argv[2] && strcmp(argv[1], "--busybox") == 0)
+                argv += 2;
+# endif
         lbb_prepare("busybox" IF_FEATURE_INDIVIDUAL(, argv));
 # if !ENABLE_BUSYBOX
         if (argv[1] && is_prefixed_with(bb_basename(argv[0]), "busybox"))
@@ -1079,6 +1278,12 @@ int main(int argc UNUSED_PARAM, char **argv)
         applet_name = argv[0];
         if (applet_name[0] == '-')
                 applet_name++;
+# if ENABLE_PLATFORM_MINGW32
+        str_tolower(argv[0]);
+        bs_to_slash(argv[0]);
+        if (has_exe_suffix_or_dot(argv[0]) && (s=strrchr(argv[0], '.')))
+                *s = '\0';
+# endif
         applet_name = bb_basename(applet_name);
 
         /* If we are a result of execv("/proc/self/exe"), fix ugly comm of "exe" */