cleanup
[oweals/busybox.git] / modutils / lsmod.c
index c97d199a69e40d262cd86011ac05cc9b52998dc8..d51da2d1699a4f26de7e739c38b4d740b52f8321 100644 (file)
@@ -2,8 +2,8 @@
 /*
  * Mini lsmod implementation for busybox
  *
- * Copyright (C) 1999,2000,2001 by Lineo, inc.
- * Written by Erik Andersen <andersen@lineo.com>, <andersee@debian.org>
+ * Copyright (C) 1999,2000 by Lineo, inc. and Erik Andersen
+ * Copyright (C) 1999,2000,2001 by Erik Andersen <andersee@debian.org>
  *
  * Modified by Alcove, Julien Gaulmin <julien.gaulmin@alcove.fr> and
  * Nicolas Ferre <nicolas.ferre@alcove.fr> to support pre 2.1 kernels
@@ -27,6 +27,7 @@
 
 #include <stdlib.h>
 #include <stdio.h>
+#include <string.h>
 #include <stddef.h>
 #include <errno.h>
 #include <unistd.h>
 #include "busybox.h"
 
 
+#ifndef CONFIG_FEATURE_CHECK_TAINTED_MODULE
+static inline void check_tainted(void) { printf("\n"); }
+#else
+#define TAINT_FILENAME                  "/proc/sys/kernel/tainted"
+#define TAINT_PROPRIETORY_MODULE        (1<<0)
+#define TAINT_FORCED_MODULE             (1<<1)
+#define TAINT_UNSAFE_SMP                (1<<2)
 
-#ifdef BB_FEATURE_NEW_MODULE_INTERFACE
+static void check_tainted(void)
+{
+       int tainted;
+       FILE *f;
+
+       tainted = 0;
+       if ((f = fopen(TAINT_FILENAME, "r"))) {
+               fscanf(f, "%d", &tainted);
+               fclose(f);
+       }
+       if (f && tainted) {
+               printf("    Tainted: %c%c%c\n",
+                               tainted & TAINT_PROPRIETORY_MODULE      ? 'P' : 'G',
+                               tainted & TAINT_FORCED_MODULE           ? 'F' : ' ',
+                               tainted & TAINT_UNSAFE_SMP              ? 'S' : ' ');
+       }
+       else {
+               printf("    Not tainted\n");
+       }
+}
+#endif
+
+#ifdef CONFIG_FEATURE_QUERY_MODULE_INTERFACE
 
 struct module_info
 {
@@ -51,8 +81,7 @@ struct module_info
 };
 
 
-int query_module(const char *name, int which, void *buf, size_t bufsize,
-                size_t *ret);
+int query_module(const char *name, int which, void *buf, size_t bufsize, size_t *ret);
 
 /* Values for query_module's which.  */
 static const int QM_MODULES = 1;
@@ -69,20 +98,39 @@ static const int NEW_MOD_VISITED = 8;
 static const int NEW_MOD_USED_ONCE = 16;
 static const int NEW_MOD_INITIALIZING = 64;
 
+static int my_query_module(const char *name, int which, void **buf,
+               size_t *bufsize, size_t *ret)
+{
+       int my_ret;
+
+       my_ret = query_module(name, which, *buf, *bufsize, ret);
+
+       if (my_ret == -1 && errno == ENOSPC) {
+               *buf = xrealloc(*buf, *ret);
+               *bufsize = *ret;
+
+               my_ret = query_module(name, which, *buf, *bufsize, ret);
+       }
+
+       return my_ret;
+}
 
 extern int lsmod_main(int argc, char **argv)
 {
        struct module_info info;
        char *module_names, *mn, *deps, *dn;
-       size_t bufsize, nmod, count, i, j;
+       size_t bufsize, depsize, nmod, count, i, j;
 
        module_names = xmalloc(bufsize = 256);
-       deps = xmalloc(bufsize);
-       if (query_module(NULL, QM_MODULES, module_names, bufsize, &nmod)) {
+       if (my_query_module(NULL, QM_MODULES, (void **)&module_names, &bufsize,
+                               &nmod)) {
                perror_msg_and_die("QM_MODULES");
        }
 
-       printf("Module                  Size  Used by\n");
+       deps = xmalloc(depsize = 256);
+       printf("Module                  Size  Used by");
+       check_tainted();
+
        for (i = 0, mn = module_names; i < nmod; mn += strlen(mn) + 1, i++) {
                if (query_module(mn, QM_INFO, &info, sizeof(info), &count)) {
                        if (errno == ENOENT) {
@@ -92,61 +140,52 @@ extern int lsmod_main(int argc, char **argv)
                        /* else choke */
                        perror_msg_and_die("module %s: QM_INFO", mn);
                }
-               while (query_module(mn, QM_REFS, deps, bufsize, &count)) {
+               if (my_query_module(mn, QM_REFS, (void **)&deps, &depsize, &count)) {
                        if (errno == ENOENT) {
                                /* The module was removed out from underneath us. */
                                continue;
                        }
-                       if (errno != ENOSPC) {
-                               error_msg_and_die("module %s: QM_REFS", mn);
-                       }
-                       deps = xrealloc(deps, bufsize = count);
+                       perror_msg_and_die("module %s: QM_REFS", mn);
                }
-               printf("%-20s%8lu%4ld ", mn, info.size, info.usecount);
-               if (count) printf("[");
-               for (j = 0, dn = deps; j < count; dn += strlen(dn) + 1, j++) {
-                       printf("%s%s", dn, (j==count-1)? "":" ");
-               }
-               if (count) printf("] ");
-
+               printf("%-20s%8lu%4ld", mn, info.size, info.usecount);
                if (info.flags & NEW_MOD_DELETED)
-                       printf("(deleted)");
+                       printf(" (deleted)");
                else if (info.flags & NEW_MOD_INITIALIZING)
-                       printf("(initializing)");
+                       printf(" (initializing)");
                else if (!(info.flags & NEW_MOD_RUNNING))
-                       printf("(uninitialized)");
+                       printf(" (uninitialized)");
                else {
                        if (info.flags & NEW_MOD_AUTOCLEAN)
-                               printf("(autoclean) ");
+                               printf(" (autoclean) ");
                        if (!(info.flags & NEW_MOD_USED_ONCE))
-                               printf("(unused)");
+                               printf(" (unused)");
                }
+               if (count) printf(" [");
+               for (j = 0, dn = deps; j < count; dn += strlen(dn) + 1, j++) {
+                       printf("%s%s", dn, (j==count-1)? "":" ");
+               }
+               if (count) printf("]");
+
                printf("\n");
        }
 
+#ifdef CONFIG_FEATURE_CLEAN_UP
+       free(module_names);
+#endif
 
        return( 0);
 }
 
-#else /*BB_FEATURE_OLD_MODULE_INTERFACE*/
+#else /* CONFIG_FEATURE_QUERY_MODULE_INTERFACE */
 
 extern int lsmod_main(int argc, char **argv)
 {
-       int fd, i;
-       char line[128];
+       printf("Module                  Size  Used by");
+       check_tainted();
 
-       puts("Module                  Size  Used by");
-       fflush(stdout);
-
-       if ((fd = open("/proc/modules", O_RDONLY)) >= 0 ) {
-               while ((i = read(fd, line, sizeof(line))) > 0) {
-                       write(fileno(stdout), line, i);
-               }
-               close(fd);
-               return 0;
-       }
-       perror_msg_and_die("/proc/modules");
+       if(print_file_by_name("/proc/modules") == FALSE)
        return 1;
+       return 0;
 }
 
-#endif /*BB_FEATURE_OLD_MODULE_INTERFACE*/
+#endif /* CONFIG_FEATURE_QUERY_MODULE_INTERFACE */