sbin/modstat/main.c

   1 /*      $NetBSD: main.c,v 1.7 2009/10/02 18:50:14 elad Exp $    */
   2
   3 /*-
   4  * Copyright (c) 2008 The NetBSD Foundation, Inc.
   5  * All rights reserved.
   6  *
   7  * Redistribution and use in source and binary forms, with or without
   8  * modification, are permitted provided that the following conditions
   9  * are met:
  10  * 1. Redistributions of source code must retain the above copyright
  11  *    notice, this list of conditions and the following disclaimer.
  12  * 2. Redistributions in binary form must reproduce the above copyright
  13  *    notice, this list of conditions and the following disclaimer in the
  14  *    documentation and/or other materials provided with the distribution.
  15  *
  16  * THIS SOFTWARE IS PROVIDED BY THE NETBSD FOUNDATION, INC. AND CONTRIBUTORS
  17  * ``AS IS'' AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED
  18  * TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR
  19  * PURPOSE ARE DISCLAIMED.  IN NO EVENT SHALL THE FOUNDATION OR CONTRIBUTORS
  20  * BE LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  21  * CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  22  * SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  23  * INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  24  * CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  25  * ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  26  * POSSIBILITY OF SUCH DAMAGE.
  27  */
  28
  29 #include <sys/cdefs.h>
  30 #ifndef lint
  31 __RCSID("$NetBSD: main.c,v 1.7 2009/10/02 18:50:14 elad Exp $");
  32 #endif /* !lint */
  33
  34 #include <sys/module.h>
  35
  36 #include <stdio.h>
  37 #include <stdlib.h>
  38 #include <string.h>
  39 #include <unistd.h>
  40 #include <err.h>
  41
  42 int     main(int, char **);
  43 static void     usage(void) __dead;
  44 static int      modstatcmp(const void *, const void *);
  45
  46 static const char *classes[] = {
  47         "any",
  48         "misc",
  49         "vfs",
  50         "driver",
  51         "exec",
  52         "secmodel",
  53 };
  54
  55 static const char *sources[] = {
  56         "builtin",
  57         "boot",
  58         "filesys",
  59 };
  60
  61 int
  62 main(int argc, char **argv)
  63 {
  64         struct iovec iov;
  65         modstat_t *ms;
  66         size_t len;
  67         const char *name;
  68         char sbuf[32];
  69         int ch;
  70
  71         name = NULL;
  72
  73         while ((ch = getopt(argc, argv, "n:")) != -1) {
  74                 switch (ch) {
  75                 case 'n':
  76                         name = optarg;
  77                         break;
  78                 default:
  79                         usage();
  80                         /* NOTREACHED */
  81                 }
  82         }
  83
  84         argc -= optind;
  85         argv += optind;
  86         if (argc != 0)
  87                 usage();
  88
  89         for (len = 4096;;) {
  90                 iov.iov_base = malloc(len);
  91                 iov.iov_len = len;
  92                 if (modctl(MODCTL_STAT, &iov)) {
  93                         err(EXIT_FAILURE, "modctl(MODCTL_STAT)");
  94                 }
  95                 if (len >= iov.iov_len) {
  96                         break;
  97                 }
  98                 free(iov.iov_base);
  99                 len = iov.iov_len;
 100         }
 101
 102         printf("%-16s %-10s %-10s %-5s %-8s %s\n",
 103             "NAME", "CLASS", "SOURCE", "REFS", "SIZE", "REQUIRES");
 104         len = iov.iov_len / sizeof(modstat_t);
 105         qsort(iov.iov_base, len, sizeof(modstat_t), modstatcmp);
 106         for (ms = iov.iov_base; len != 0; ms++, len--) {
 107                 if (name != NULL && strcmp(ms->ms_name, name) != 0) {
 108                         continue;
 109                 }
 110                 if (ms->ms_required[0] == '\0') {
 111                         ms->ms_required[0] = '-';
 112                         ms->ms_required[1] = '\0';
 113                 }
 114                 if (ms->ms_size == 0) {
 115                         sbuf[0] = '-';
 116                         sbuf[1] = '\0';
 117                 } else {
 118                         snprintf(sbuf, sizeof(sbuf), "%u", ms->ms_size);
 119                 }
 120                 printf("%-16s %-10s %-10s %-5d %-8s %s\n",
 121                     ms->ms_name, classes[ms->ms_class], sources[ms->ms_source],
 122                     ms->ms_refcnt, sbuf, ms->ms_required);
 123         }
 124
 125         exit(EXIT_SUCCESS);
 126 }
 127
 128 static void
 129 usage(void)
 130 {
 131
 132         (void)fprintf(stderr, "Usage: %s [-n name]\n", getprogname());
 133         exit(EXIT_FAILURE);
 134 }
 135
 136 static int
 137 modstatcmp(const void *a, const void *b)
 138 {
 139         const modstat_t *msa, *msb;
 140
 141         msa = a;
 142         msb = b;
 143
 144         return strcmp(msa->ms_name, msb->ms_name);
 145 }
 146