1 files changed, 250 insertions, 3 deletions

diff --git a/libbb/appletlib.c b/libbb/appletlib.c
index d8ab2a450..99f0cb89c 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,10 +71,18 @@ 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
 
+#if defined(find_applet_by_name)
+# undef find_applet_by_name
+#endif
+#if defined(is_applet_preferred)
+# undef is_applet_preferred
+#endif
+
 /* "Do not compress usage text if uncompressed text is small
  *  and we don't include bunzip2 code for other reasons"
  *
@@ -111,6 +128,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 +172,11 @@ void FAST_FUNC bb_show_usage(void)
                         ap--;
                 }
                 full_write2_str(bb_banner);
+#if ENABLE_PLATFORM_MINGW32
+                full_write2_str("\n");
+#else
                 full_write2_str(" multi-call binary.\n"); /* common string */
+#endif
                 if (*p == '\b')
                         full_write2_str("\nNo help available\n");
                 else {
@@ -239,6 +261,45 @@ int FAST_FUNC find_applet_by_name(const char *name)
         return -1;
 }
 
+#if ENABLE_PLATFORM_MINGW32 && NUM_APPLETS > 1 && \
+                (ENABLE_FEATURE_PREFER_APPLETS || ENABLE_FEATURE_SH_STANDALONE)
+int FAST_FUNC is_applet_preferred(const char *name)
+{
+        const char *var, *s;
+        size_t len;
+
+        var = getenv(BB_OVERRIDE_APPLETS);
+        if (var && *var) {
+                /* '-' overrides all applets */
+                if (var[0] == '-' && var[1] == '\0')
+                        return FALSE;
+
+                /* Override applets from a space-separated list */
+                len = strlen(name);
+                s = var - 1;
+                while (1) {
+                        s = strstr(s + 1, name);
+                        if (!s)
+                                break;
+                        /* neither "name.." nor "xxx,name.."? */
+                        if (s != var && s[-1] != ' ')
+                                continue;
+                        /* neither "..name" nor "..name,xxx"? */
+                        if (s[len] != '\0' && s[len] != ' ')
+                                continue;
+                        return FALSE;
+                }
+        }
+        return TRUE;
+}
+
+int FAST_FUNC find_preferred_applet_by_name(const char *name)
+{
+        int applet_no = find_applet_by_name(name);
+        return applet_no >= 0 && is_applet_preferred(name) ? applet_no : -1;
+}
+#endif
+
 
 void lbb_prepare(const char *applet
                 IF_FEATURE_INDIVIDUAL(, char **argv))
@@ -288,6 +349,11 @@ const char *applet_name;
 #if !BB_MMU
 bool re_execed;
 #endif
+#if ENABLE_PLATFORM_MINGW32
+static int interp = 0;
+char bb_comm[COMM_LEN];
+char bb_command_line[128];
+#endif
 
 
 /* If not built as a single-applet executable... */
@@ -673,15 +739,36 @@ static void install_links(const char *busybox, int use_symbolic_links,
         const char *appname = applet_names;
         unsigned i;
         int rc;
+#  if ENABLE_PLATFORM_MINGW32
+        const char *sd = NULL;
+
+        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);
+                }
+        }
+#  endif
 
         lf = link;
         if (use_symbolic_links)
                 lf = symlink;
 
         for (i = 0; i < ARRAY_SIZE(applet_main); i++) {
+#  if ENABLE_PLATFORM_MINGW32
+                fpc = xasprintf("%s%s/%s.exe", sd ?: "",
+                                custom_install_dir ?: install_dir[APPLET_INSTALL_LOC(i)],
+                                appname);
+#  else
                 fpc = concat_path_file(
                                 custom_install_dir ? custom_install_dir : install_dir[APPLET_INSTALL_LOC(i)],
                                 appname);
+#  endif
                 // debug: bb_error_msg("%slinking %s to busybox",
                 //              use_symbolic_links ? "sym" : "", fpc);
                 rc = lf(busybox, fpc);
@@ -779,20 +866,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("\n");
+#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-2022.\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 [-s] [-u|DIR]\n"
+                        "   or: busybox --uninstall [-n] file\n"
+                        )
+                        )
                         "   or: function [arguments]...\n"
                         "\n"
                         IF_NOT_FEATURE_SH_STANDALONE(
@@ -853,9 +955,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");
@@ -868,6 +989,7 @@ int busybox_main(int argc UNUSED_PARAM, char **argv)
 
         if (ENABLE_FEATURE_INSTALLER && strcmp(argv[1], "--install") == 0) {
                 int use_symbolic_links;
+#if !ENABLE_PLATFORM_MINGW32
                 const char *busybox;
 
                 busybox = xmalloc_readlink(bb_busybox_exec_path);
@@ -887,9 +1009,63 @@ 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;
+                uint32_t opt;
+                enum { OPT_s = (1 << 0), OPT_u = (1 << 1) };
+
+                /* busybox --install [-s] [-u|DIR]
+                 * -s: make symlinks
+                 * -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.
+                 */
+                argv += 1;
+                opt = getopt32(argv, "!su");
+                argv += optind;
+
+                if (opt == (uint32_t)-1 ||
+                                (*argv != NULL && (opt & OPT_u || *(argv + 1) != NULL)))
+                        bb_simple_error_msg_and_die("busybox --install [-s] [-u|DIR]");
+
+                if (opt & OPT_u)
+                        target = NULL;
+                else if (*argv != NULL)
+                        target = *argv;
+                else
+                        target = dirname(xstrdup(bb_busybox_exec_path));
+
+                use_symbolic_links = opt & OPT_s;
+                /* NULL target -> install to Unix-style dirs */
+                install_links(bb_busybox_exec_path, use_symbolic_links, 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>]" */
                 if (!argv[2]
@@ -939,6 +1115,9 @@ void FAST_FUNC show_usage_if_dash_dash_help(int applet_no, char **argv)
 #  if defined APPLET_NO_echo
          && applet_no != APPLET_NO_echo
 #  endif
+#  if ENABLE_PLATFORM_MINGW32 && defined APPLET_NO_busybox
+         && applet_no != APPLET_NO_busybox
+#  endif
         ) {
                 if (argv[1] && !argv[2] && strcmp(argv[1], "--help") == 0) {
                         /* Make "foo --help" exit with 0: */
@@ -950,7 +1129,14 @@ void FAST_FUNC show_usage_if_dash_dash_help(int applet_no, char **argv)
 
 void FAST_FUNC run_applet_no_and_exit(int applet_no, const char *name, char **argv)
 {
+#  if ENABLE_PLATFORM_MINGW32
+        int argc = string_array_len(argv);
+        int i;
+        const char *vmask;
+        unsigned int mask;
+#  else
         int argc;
+#  endif
 
         /*
          * We do not use argv[0]: do not want to repeat massaging of
@@ -963,7 +1149,23 @@ void FAST_FUNC run_applet_no_and_exit(int applet_no, const char *name, char **ar
         if (ENABLE_FEATURE_SUID)
                 check_suid(applet_no);
 
+#  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);
+#  else
         argc = string_array_len(argv);
+#  endif
         xfunc_error_retval = applet_main[applet_no](argc, argv);
 
         /* Note: applet_main() may also not return (die on a xfunc or such) */
@@ -1073,6 +1275,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)
 
@@ -1097,6 +1331,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"))
@@ -1105,6 +1343,15 @@ 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])) {
+                char *s = strrchr(argv[0], '.');
+                if (s)
+                        *s = '\0';
+        }
+# endif
         applet_name = bb_basename(applet_name);
 
         /* If we are a result of execv("/proc/self/exe"), fix ugly comm of "exe" */