udhcpc: fix a problem with binary-encoded options #2
[oweals/busybox.git] / modutils / modutils.c
index 5a4ba84e79c330c729bbc224d211facda6061a0f..6187ca72fd3835fa28a18158205a5856913cfd75 100644 (file)
@@ -3,9 +3,8 @@
  *
  * Copyright (C) 2008 by Timo Teras <timo.teras@iki.fi>
  *
- * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
+ * Licensed under GPLv2 or later, see file LICENSE in this source tree.
  */
-
 #include "modutils.h"
 
 #ifdef __UCLIBC__
@@ -17,19 +16,6 @@ extern int delete_module(const char *module, unsigned int flags);
 # define delete_module(mod, flags) syscall(__NR_delete_module, mod, flags)
 #endif
 
-/*
- a libbb candidate from ice age!
-*/
-llist_t FAST_FUNC *llist_find(llist_t *first, const char *str)
-{
-       while (first != NULL) {
-               if (strcmp(first->data, str) == 0)
-                       return first;
-               first = first->link;
-       }
-       return NULL;
-}
-
 void FAST_FUNC replace(char *s, char what, char with)
 {
        while (*s) {
@@ -39,7 +25,7 @@ void FAST_FUNC replace(char *s, char what, char with)
        }
 }
 
-char * FAST_FUNC replace_underscores(char *s)
+char* FAST_FUNC replace_underscores(char *s)
 {
        replace(s, '-', '_');
        return s;
@@ -59,7 +45,7 @@ int FAST_FUNC string_to_llist(char *string, llist_t **llist, const char *delim)
        return len;
 }
 
-char * FAST_FUNC filename2modname(const char *filename, char *modname)
+char* FAST_FUNC filename2modname(const char *filename, char *modname)
 {
        int i;
        char *from;
@@ -71,30 +57,12 @@ char * FAST_FUNC filename2modname(const char *filename, char *modname)
        from = bb_get_last_path_component_nostrip(filename);
        for (i = 0; i < (MODULE_NAME_LEN-1) && from[i] != '\0' && from[i] != '.'; i++)
                modname[i] = (from[i] == '-') ? '_' : from[i];
-       modname[i] = 0;
+       modname[i] = '\0';
 
        return modname;
 }
 
-const char * FAST_FUNC moderror(int err)
-{
-       switch (err) {
-       case -1:
-               return "no such module";
-       case ENOEXEC:
-               return "invalid module format";
-       case ENOENT:
-               return "unknown symbol in module, or unknown parameter";
-       case ESRCH:
-               return "module has wrong symbol version";
-       case ENOSYS:
-               return "kernel does not support requested operation";
-       default:
-               return strerror(err);
-       }
-}
-
-char * FAST_FUNC parse_cmdline_module_options(char **argv)
+char* FAST_FUNC parse_cmdline_module_options(char **argv, int quote_spaces)
 {
        char *options;
        int optlen;
@@ -102,38 +70,133 @@ char * FAST_FUNC parse_cmdline_module_options(char **argv)
        options = xzalloc(1);
        optlen = 0;
        while (*++argv) {
-               options = xrealloc(options, optlen + 2 + strlen(*argv) + 2);
-               /* Spaces handled by "" pairs, but no way of escaping quotes */
-               optlen += sprintf(options + optlen, (strchr(*argv, ' ') ? "\"%s\" " : "%s "), *argv);
+               const char *fmt;
+               const char *var;
+               const char *val;
+
+               var = *argv;
+               options = xrealloc(options, optlen + 2 + strlen(var) + 2);
+               fmt = "%.*s%s ";
+               val = strchrnul(var, '=');
+               if (quote_spaces) {
+                       /*
+                        * modprobe (module-init-tools version 3.11.1) compat:
+                        * quote only value:
+                        * var="val with spaces", not "var=val with spaces"
+                        * (note: var *name* is not checked for spaces!)
+                        */
+                       if (*val) { /* has var=val format. skip '=' */
+                               val++;
+                               if (strchr(val, ' '))
+                                       fmt = "%.*s\"%s\" ";
+                       }
+               }
+               optlen += sprintf(options + optlen, fmt, (int)(val - var), var, val);
        }
+       /* Remove trailing space. Disabled */
+       /* if (optlen != 0) options[optlen-1] = '\0'; */
        return options;
 }
 
+#if ENABLE_FEATURE_INSMOD_TRY_MMAP
+void* FAST_FUNC try_to_mmap_module(const char *filename, size_t *image_size_p)
+{
+       /* We have user reports of failure to load 3MB module
+        * on a 16MB RAM machine. Apparently even a transient
+        * memory spike to 6MB during module load
+        * is too big for that system. */
+       void *image;
+       struct stat st;
+       int fd;
+
+       fd = xopen(filename, O_RDONLY);
+       fstat(fd, &st);
+       image = NULL;
+       /* st.st_size is off_t, we can't just pass it to mmap */
+       if (st.st_size <= *image_size_p) {
+               size_t image_size = st.st_size;
+               image = mmap(NULL, image_size, PROT_READ, MAP_PRIVATE, fd, 0);
+               if (image == MAP_FAILED) {
+                       image = NULL;
+               } else if (*(uint32_t*)image != SWAP_BE32(0x7f454C46)) {
+                       /* No ELF signature. Compressed module? */
+                       munmap(image, image_size);
+                       image = NULL;
+               } else {
+                       /* Success. Report the size */
+                       *image_size_p = image_size;
+               }
+       }
+       close(fd);
+       return image;
+}
+#endif
+
+/* Return:
+ * 0 on success,
+ * -errno on open/read error,
+ * errno on init_module() error
+ */
 int FAST_FUNC bb_init_module(const char *filename, const char *options)
 {
-       size_t len = MAXINT(ssize_t);
+       size_t image_size;
        char *image;
-       int rc = ENOENT;
+       int rc;
+       bool mmaped;
+
+       if (!options)
+               options = "";
 
+//TODO: audit bb_init_module_24 to match error code convention
 #if ENABLE_FEATURE_2_4_MODULES
        if (get_linux_version_code() < KERNEL_VERSION(2,6,0))
                return bb_init_module_24(filename, options);
 #endif
 
-       /* Use the 2.6 way */
-       image = xmalloc_open_zipped_read_close(filename, &len);
+       image_size = INT_MAX - 4095;
+       mmaped = 0;
+       image = try_to_mmap_module(filename, &image_size);
        if (image) {
-               if (init_module(image, len, options) != 0)
-                       rc = errno;
-               else
-                       rc = 0;
-               free(image);
+               mmaped = 1;
+       } else {
+               errno = ENOMEM; /* may be changed by e.g. open errors below */
+               image = xmalloc_open_zipped_read_close(filename, &image_size);
+               if (!image)
+                       return -errno;
        }
 
+       errno = 0;
+       init_module(image, image_size, options);
+       rc = errno;
+       if (mmaped)
+               munmap(image, image_size);
+       else
+               free(image);
        return rc;
 }
 
 int FAST_FUNC bb_delete_module(const char *module, unsigned int flags)
 {
-       return delete_module(module, flags);
+       errno = 0;
+       delete_module(module, flags);
+       return errno;
+}
+
+const char* FAST_FUNC moderror(int err)
+{
+       switch (err) {
+       case -1: /* btw: it's -EPERM */
+               return "no such module";
+       case ENOEXEC:
+               return "invalid module format";
+       case ENOENT:
+               return "unknown symbol in module, or unknown parameter";
+       case ESRCH:
+               return "module has wrong symbol version";
+       case ENOSYS:
+               return "kernel does not support requested operation";
+       }
+       if (err < 0) /* should always be */
+               err = -err;
+       return strerror(err);
 }