modutils/depmod.c

   1 /* vi: set sw=4 ts=4: */
   2 /*
   3  * depmod - generate modules.dep
   4  * Copyright (c) 2008 Bernhard Reutner-Fischer
   5  * Copyrihgt (c) 2008 Timo Teras <timo.teras@iki.fi>
   6  * Copyright (c) 2008 Vladimir Dronnikov
   7  *
   8  * Licensed under GPLv2 or later, see file LICENSE in this tarball for details.
   9  */
  10
  11 #undef _GNU_SOURCE
  12 #define _GNU_SOURCE
  13 #include "libbb.h"
  14 #include "modutils.h"
  15 #include <sys/utsname.h> /* uname() */
  16
  17 /*
  18  * Theory of operation:
  19  * - iterate over all modules and record their full path
  20  * - iterate over all modules looking for "depends=" entries
  21  *   for each depends, look through our list of full paths and emit if found
  22  */
  23
  24 typedef struct module_info {
  25         struct module_info *next;
  26         char *name, *modname;
  27         llist_t *dependencies;
  28         llist_t *aliases;
  29         llist_t *symbols;
  30         struct module_info *dnext, *dprev;
  31 } module_info;
  32
  33 enum {
  34         ARG_a = (1<<0), /* All modules, ignore mods in argv */
  35         ARG_A = (1<<1), /* Only emit .ko that are newer than modules.dep file */
  36         ARG_b = (1<<2), /* base directory when modules are in staging area */
  37         ARG_e = (1<<3), /* with -F, print unresolved symbols */
  38         ARG_F = (1<<4), /* System.map that contains the symbols */
  39         ARG_n = (1<<5), /* dry-run, print to stdout only */
  40         ARG_r = (1<<6)  /* Compat dummy. Linux Makefile uses it */
  41 };
  42
  43 static int FAST_FUNC parse_module(const char *fname, struct stat *sb UNUSED_PARAM,
  44                                   void *data, int depth UNUSED_PARAM)
  45 {
  46         char modname[MODULE_NAME_LEN];
  47         module_info **first = (module_info **) data;
  48         char *image, *ptr;
  49         module_info *info;
  50         /* Arbitrary. Was sb->st_size, but that breaks .gz etc */
  51         size_t len = (64*1024*1024 - 4096);
  52
  53         if (strrstr(fname, ".ko") == NULL)
  54                 return TRUE;
  55
  56         image = xmalloc_open_zipped_read_close(fname, &len);
  57         info = xzalloc(sizeof(*info));
  58
  59         info->next = *first;
  60         *first = info;
  61
  62         info->dnext = info->dprev = info;
  63         info->name = xasprintf("/%s", fname);
  64         info->modname = xstrdup(filename2modname(fname, modname));
  65         for (ptr = image; ptr < image + len - 10; ptr++) {
  66                 if (strncmp(ptr, "depends=", 8) == 0) {
  67                         char *u;
  68
  69                         ptr += 8;
  70                         for (u = ptr; *u; u++)
  71                                 if (*u == '-')
  72                                         *u = '_';
  73                         ptr += string_to_llist(ptr, &info->dependencies, ",");
  74                 } else if (ENABLE_FEATURE_MODUTILS_ALIAS
  75                  && strncmp(ptr, "alias=", 6) == 0
  76                 ) {
  77                         llist_add_to(&info->aliases, xstrdup(ptr + 6));
  78                         ptr += strlen(ptr);
  79                 } else if (ENABLE_FEATURE_MODUTILS_SYMBOLS
  80                  && strncmp(ptr, "__ksymtab_", 10) == 0
  81                 ) {
  82                         ptr += 10;
  83                         if (strncmp(ptr, "gpl", 3) == 0
  84                          || strcmp(ptr, "strings") == 0
  85                         ) {
  86                                 continue;
  87                         }
  88                         llist_add_to(&info->symbols, xstrdup(ptr));
  89                         ptr += strlen(ptr);
  90                 }
  91         }
  92         free(image);
  93
  94         return TRUE;
  95 }
  96
  97 static module_info *find_module(module_info *modules, const char *modname)
  98 {
  99         module_info *m;
 100
 101         for (m = modules; m != NULL; m = m->next)
 102                 if (strcmp(m->modname, modname) == 0)
 103                         return m;
 104         return NULL;
 105 }
 106
 107 static void order_dep_list(module_info *modules, module_info *start,
 108                            llist_t *add)
 109 {
 110         module_info *m;
 111         llist_t *n;
 112
 113         for (n = add; n != NULL; n = n->link) {
 114                 m = find_module(modules, n->data);
 115                 if (m == NULL)
 116                         continue;
 117
 118                 /* unlink current entry */
 119                 m->dnext->dprev = m->dprev;
 120                 m->dprev->dnext = m->dnext;
 121
 122                 /* and add it to tail */
 123                 m->dnext = start;
 124                 m->dprev = start->dprev;
 125                 start->dprev->dnext = m;
 126                 start->dprev = m;
 127
 128                 /* recurse */
 129                 order_dep_list(modules, start, m->dependencies);
 130         }
 131 }
 132
 133 static void xfreopen_write(const char *file, FILE *f)
 134 {
 135         if (freopen(file, "w", f) == NULL)
 136                 bb_perror_msg_and_die("can't open '%s'", file);
 137 }
 138
 139 int depmod_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
 140 int depmod_main(int argc UNUSED_PARAM, char **argv)
 141 {
 142         module_info *modules = NULL, *m, *dep;
 143         const char *moddir_base = "/";
 144         char *moddir, *version;
 145         struct utsname uts;
 146         int tmp;
 147
 148         getopt32(argv, "aAb:eF:nr", &moddir_base, NULL);
 149         argv += optind;
 150
 151         /* goto modules location */
 152         xchdir(moddir_base);
 153
 154         /* If a version is provided, then that kernel version's module directory
 155          * is used, rather than the current kernel version (as returned by
 156          * "uname -r").  */
 157         if (*argv && sscanf(*argv, "%d.%d.%d", &tmp, &tmp, &tmp) == 3) {
 158                 version = *argv++;
 159         } else {
 160                 uname(&uts);
 161                 version = uts.release;
 162         }
 163         moddir = concat_path_file(&CONFIG_DEFAULT_MODULES_DIR[1], version);
 164
 165         /* Scan modules */
 166         if (*argv) {
 167                 char *modfile;
 168                 struct stat sb;
 169                 do {
 170                         modfile = concat_path_file(moddir, *argv);
 171                         xstat(modfile, &sb);
 172                         parse_module(modfile, &sb, &modules, 0);
 173                         free(modfile);
 174                 } while (*(++argv));
 175         } else {
 176                 recursive_action(moddir, ACTION_RECURSE,
 177                                  parse_module, NULL, &modules, 0);
 178         }
 179
 180         /* Prepare for writing out the dep files */
 181         xchdir(moddir);
 182         if (ENABLE_FEATURE_CLEAN_UP)
 183                 free(moddir);
 184
 185         /* Generate dependency and alias files */
 186         if (!(option_mask32 & ARG_n))
 187                 xfreopen_write(CONFIG_DEFAULT_DEPMOD_FILE, stdout);
 188         for (m = modules; m != NULL; m = m->next) {
 189                 printf("%s:", m->name);
 190
 191                 order_dep_list(modules, m, m->dependencies);
 192                 while (m->dnext != m) {
 193                         dep = m->dnext;
 194                         printf(" %s", dep->name);
 195
 196                         /* unlink current entry */
 197                         dep->dnext->dprev = dep->dprev;
 198                         dep->dprev->dnext = dep->dnext;
 199                         dep->dnext = dep->dprev = dep;
 200                 }
 201                 bb_putchar('\n');
 202         }
 203
 204 #if ENABLE_FEATURE_MODUTILS_ALIAS
 205         if (!(option_mask32 & ARG_n))
 206                 xfreopen_write("modules.alias", stdout);
 207         for (m = modules; m != NULL; m = m->next) {
 208                 const char *fname = bb_basename(m->name);
 209                 int fnlen = strchrnul(fname, '.') - fname;
 210                 while (m->aliases) {
 211                         /* Last word can well be m->modname instead,
 212                          * but depmod from module-init-tools 3.4
 213                          * uses module basename, i.e., no s/-/_/g.
 214                          * (pathname and .ko.* are still stripped)
 215                          * Mimicking that... */
 216                         printf("alias %s %.*s\n",
 217                                 (char*)llist_pop(&m->aliases),
 218                                 fnlen, fname);
 219                 }
 220         }
 221 #endif
 222 #if ENABLE_FEATURE_MODUTILS_SYMBOLS
 223         if (!(option_mask32 & ARG_n))
 224                 xfreopen_write("modules.symbols", stdout);
 225         for (m = modules; m != NULL; m = m->next) {
 226                 const char *fname = bb_basename(m->name);
 227                 int fnlen = strchrnul(fname, '.') - fname;
 228                 while (m->symbols) {
 229                         printf("alias symbol:%s %.*s\n",
 230                                 (char*)llist_pop(&m->symbols),
 231                                 fnlen, fname);
 232                 }
 233         }
 234 #endif
 235
 236         if (ENABLE_FEATURE_CLEAN_UP) {
 237                 while (modules) {
 238                         module_info *old = modules;
 239                         modules = modules->next;
 240                         free(old->name);
 241                         free(old->modname);
 242                         free(old);
 243                 }
 244         }
 245
 246         return EXIT_SUCCESS;
 247 }