tftp: do not risk invoking Sorcerer's Apprentice syndrome
[oweals/busybox.git] / modutils / depmod.c
index 9131dc1abd2abe93c37c27b76cd4b0196c057769..405ba9e2a09431a17ae89d521aae09473c352850 100644 (file)
@@ -1,7 +1,9 @@
 /* vi: set sw=4 ts=4: */
 /*
  * depmod - generate modules.dep
- * Copyright (c) 2008 Bernhard Fischer
+ * Copyright (c) 2008 Bernhard Reutner-Fischer
+ * Copyrihgt (c) 2008 Timo Teras <timo.teras@iki.fi>
+ * Copyright (c) 2008 Vladimir Dronnikov
  *
  * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
  */
 #define _GNU_SOURCE
 #include <libbb.h>
 #include <sys/utsname.h> /* uname() */
+#include "modutils.h"
 
-struct globals {
-       llist_t *lst;
+/*
+ * Theory of operation:
+ * - iterate over all modules and record their full path
+ * - iterate over all modules looking for "depends=" entries
+ *   for each depends, look through our list of full paths and emit if found
+ */
+
+typedef struct module_info {
+       struct module_info *next;
+       char *name, *modname;
+       llist_t *dependencies;
+       llist_t *aliases;
+       llist_t *symbols;
+       struct module_info *dnext, *dprev;
+} module_info;
+
+enum {
+       ARG_a = (1<<0), /* All modules, ignore mods in argv */
+       ARG_A = (1<<1), /* Only emit .ko that are newer than modules.dep file */
+       ARG_b = (1<<2), /* base directory when modules are in staging area */
+       ARG_e = (1<<3), /* with -F, print unresolved symbols */
+       ARG_F = (1<<4), /* System.map that contains the symbols */
+       ARG_n = (1<<5), /* dry-run, print to stdout only */
+       ARG_r = (1<<6)  /* Compat dummy. Linux Makefile uses it */
 };
-#define G (*(struct globals*)&bb_common_bufsiz1)
-/* We have to zero it out because of NOEXEC */
-#define INIT_G() memset(&G, 0, sizeof(G))
 
-static int fill_lst(const char *modulename, struct stat ATTRIBUTE_UNUSED *sb,
-                                       void ATTRIBUTE_UNUSED *data, int ATTRIBUTE_UNUSED depth)
+static int FAST_FUNC parse_module(const char *fname, struct stat *sb,
+                                 void *data, int UNUSED_PARAM depth)
 {
-       llist_add_to_end(&G.lst, strdup(modulename));
+       module_info **first = (module_info **) data;
+       char *image, *ptr;
+       module_info *info;
+       size_t len = sb->st_size;
+
+       if (strrstr(fname, ".ko") == NULL)
+               return TRUE;
+
+       image = xmalloc_open_zipped_read_close(fname, &len);
+       info = xzalloc(sizeof(module_info));
+
+       info->next = *first;
+       *first = info;
+
+       info->dnext = info->dprev = info;
+       info->name = xasprintf("/%s", fname);
+       info->modname = filename2modname(fname, NULL);
+       for (ptr = image; ptr < image + len - 10; ptr++) {
+               if (strncmp(ptr, "depends=", 8) == 0) {
+                       char *u;
+
+                       ptr += 8;
+                       for (u = ptr; *u; u++)
+                               if (*u == '-')
+                                       *u = '_';
+                       ptr += string_to_llist(ptr, &info->dependencies, ",");
+               } else if (ENABLE_FEATURE_MODUTILS_ALIAS &&
+                          strncmp(ptr, "alias=", 6) == 0) {
+                       llist_add_to(&info->aliases, xstrdup(ptr + 6));
+                       ptr += strlen(ptr);
+               } else if (ENABLE_FEATURE_MODUTILS_SYMBOLS &&
+                          strncmp(ptr, "__ksymtab_", 10) == 0) {
+                       ptr += 10;
+                       if (strncmp(ptr, "gpl", 3) == 0 ||
+                           strcmp(ptr, "strings") == 0)
+                               continue;
+                       llist_add_to(&info->symbols, xstrdup(ptr));
+                       ptr += strlen(ptr);
+               }
+       }
+       free(image);
+
        return TRUE;
 }
 
-static int fileAction(const char *fname, struct stat ATTRIBUTE_UNUSED *sb,
-                                       void *data, int ATTRIBUTE_UNUSED depth)
+static module_info *find_module(module_info *modules, const char *modname)
 {
-       size_t seen = 0;
-       size_t len = MAXINT(ssize_t);
-       void *the_module = xmalloc_open_read_close(fname, &len), *ptr = the_module;
-       const char *deps;
-       RESERVE_CONFIG_BUFFER(depends, 512);
-       RESERVE_CONFIG_BUFFER(buf1, 512);
-
-       memset(buf1, 0, sizeof(buf1));
-       memset(depends, 0, sizeof(depends));
-       sprintf(buf1, "\n%s:", fname);
-
-       if (last_char_is(fname, 'o') == NULL) /* not a module */
-               goto done;
-       write((int)data, buf1, strlen(buf1));
-//bb_info_msg("[%d] fname='%s'", (int)data, fname);
-       do {
-               /* search for a 'd' */
-               ptr = memchr(ptr, 'd', len - seen);
-               if (ptr == NULL) /* no d left, done */
-                       break;
-               if (sscanf(ptr, "depends=%s", depends) == 1)
-                       break;
-               seen = ++ptr - the_module;
-       } while (1);
-//bb_info_msg(" depends='%s'", depends);
-       deps = depends;
-       while (*deps) {
-               llist_t * _lst = G.lst;
-               ptr = memchr(deps, ',', strlen(deps));
-               if (ptr != NULL)
-                       *(char*)ptr = '\0';
-               /* remember the length of the current dependency plus eventual 0 byte */
-               len = strlen(deps) + (ptr != NULL);
-               sprintf(buf1, "/%s.", deps); /* make sure we match the correct file */
-               while (_lst) {
-                       ptr = strstr(_lst->data, buf1);
-                       if (ptr != NULL)
-                               break; /* found it */
-                       _lst = _lst->link;
-               }
-               if (_lst && _lst->data) {
-                       const char separator = ' ';
-//bb_info_msg("[%s] -> '%s'", deps, _lst->data);
-                       write((int)data, &separator, 1);
-                       write((int)data, _lst->data, strlen(_lst->data));
+       module_info *m;
 
-                       deps += len;
-               }
+       for (m = modules; m != NULL; m = m->next)
+               if (strcmp(m->modname, modname) == 0)
+                       return m;
+       return NULL;
+}
+
+static void order_dep_list(module_info *modules, module_info *start,
+                          llist_t *add)
+{
+       module_info *m;
+       llist_t *n;
+
+       for (n = add; n != NULL; n = n->link) {
+               m = find_module(modules, n->data);
+               if (m == NULL)
+                       continue;
+
+               /* unlink current entry */
+               m->dnext->dprev = m->dprev;
+               m->dprev->dnext = m->dnext;
+
+               /* and add it to tail */
+               m->dnext = start;
+               m->dprev = start->dprev;
+               start->dprev->dnext = m;
+               start->dprev = m;
+
+               /* recurse */
+               order_dep_list(modules, start, m->dependencies);
        }
-done:
-       RELEASE_CONFIG_BUFFER(depends);
-       RELEASE_CONFIG_BUFFER(buf1);
-       free(the_module);
-       return TRUE;
+}
+
+static void xfreopen_write(const char *file, FILE *f)
+{
+       if (freopen(file, "w", f) == NULL)
+               bb_perror_msg_and_die("can't open '%s'", file);
 }
 
 int depmod_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
-int depmod_main(int ATTRIBUTE_UNUSED argc, char **argv)
+int depmod_main(int argc UNUSED_PARAM, char **argv)
 {
-       int retval = EXIT_SUCCESS;
-//     static const char moddir_base[] ALIGN1 = "/lib/modules/%s";
-
-       int fd = xopen3("/tmp/modules.dep", O_CREAT|O_WRONLY|O_TRUNC,
-                                       S_IWUSR|S_IRUSR|S_IRGRP|S_IROTH);
-       argv++;
-       do {
-               if (!recursive_action(*argv,
-                               ACTION_RECURSE, /* flags */
-                               fill_lst, /* file action */
-                               NULL, /* dir action */
-                               NULL, /* user data */
-                               0) || /* depth */
-                       !recursive_action(*argv,
-                               ACTION_RECURSE, /* flags */
-                               fileAction, /* file action */
-                               NULL, /* dir action */
-                               (void*)fd, /* user data */
-                               0)) { /* depth */
-                       retval = EXIT_FAILURE;
-               }
-       } while (*++argv);
+       module_info *modules = NULL, *m, *dep;
+       const char *moddir_base = "/";
+       char *moddir, *version;
+       struct utsname uts;
+       int tmp;
+
+       getopt32(argv, "aAb:eF:nr", &moddir_base, NULL);
+       argv += optind;
+
+       /* goto modules location */
+       xchdir(moddir_base);
+
+       /* If a version is provided, then that kernel version's module directory
+        * is used, rather than the current kernel version (as returned by
+        * "uname -r").  */
+       if (*argv && sscanf(*argv, "%d.%d.%d", &tmp, &tmp, &tmp) == 3) {
+               version = *argv++;
+       } else {
+               uname(&uts);
+               version = uts.release;
+       }
+       moddir = concat_path_file(&CONFIG_DEFAULT_MODULES_DIR[1], version);
+
+       /* Scan modules */
+       if (*argv) {
+               char *modfile;
+               struct stat sb;
+               do {
+                       modfile = concat_path_file(moddir, *argv);
+                       xstat(modfile, &sb);
+                       parse_module(modfile, &sb, &modules, 0);
+                       free(modfile);
+               } while (*(++argv));
+       } else {
+               recursive_action(moddir, ACTION_RECURSE,
+                                parse_module, NULL, &modules, 0);
+       }
 
+       /* Prepare for writing out the dep files */
+       xchdir(moddir);
        if (ENABLE_FEATURE_CLEAN_UP)
-               close(fd);
-       return retval;
+               free(moddir);
+
+       /* Generate dependency and alias files */
+       if (!(option_mask32 & ARG_n))
+               xfreopen_write(CONFIG_DEFAULT_DEPMOD_FILE, stdout);
+       for (m = modules; m != NULL; m = m->next) {
+               printf("%s:", m->name);
+
+               order_dep_list(modules, m, m->dependencies);
+               while (m->dnext != m) {
+                       dep = m->dnext;
+                       printf(" %s", dep->name);
+
+                       /* unlink current entry */
+                       dep->dnext->dprev = dep->dprev;
+                       dep->dprev->dnext = dep->dnext;
+                       dep->dnext = dep->dprev = dep;
+               }
+               bb_putchar('\n');
+       }
+
+#if ENABLE_FEATURE_MODUTILS_ALIAS
+       if (!(option_mask32 & ARG_n))
+               xfreopen_write("modules.alias", stdout);
+       for (m = modules; m != NULL; m = m->next) {
+               while (m->aliases) {
+                       printf("alias %s %s\n",
+                               (char*)llist_pop(&m->aliases),
+                               m->modname);
+               }
+       }
+#endif
+#if ENABLE_FEATURE_MODUTILS_SYMBOLS
+       if (!(option_mask32 & ARG_n))
+               xfreopen_write("modules.symbols", stdout);
+       for (m = modules; m != NULL; m = m->next) {
+               while (m->symbols) {
+                       printf("alias symbol:%s %s\n",
+                               (char*)llist_pop(&m->symbols),
+                               m->modname);
+               }
+       }
+#endif
+
+       if (ENABLE_FEATURE_CLEAN_UP) {
+               while (modules) {
+                       module_info *old = modules;
+                       modules = modules->next;
+                       free(old->name);
+                       free(old->modname);
+                       free(old);
+               }
+       }
+
+       return EXIT_SUCCESS;
 }