xref: /dragonfly/sbin/kldstat/kldstat.c (revision 956939d5)
1 /*-
2  * Copyright (c) 1997 Doug Rabson
3  * All rights reserved.
4  *
5  * Redistribution and use in source and binary forms, with or without
6  * modification, are permitted provided that the following conditions
7  * are met:
8  * 1. Redistributions of source code must retain the above copyright
9  *    notice, this list of conditions and the following disclaimer.
10  * 2. Redistributions in binary form must reproduce the above copyright
11  *    notice, this list of conditions and the following disclaimer in the
12  *    documentation and/or other materials provided with the distribution.
13  *
14  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR AND CONTRIBUTORS ``AS IS'' AND
15  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
16  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
17  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
18  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
19  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
20  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
21  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
22  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
23  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
24  * SUCH DAMAGE.
25  *
26  * $FreeBSD: src/sbin/kldstat/kldstat.c,v 1.20 2007/10/22 04:12:57 jb Exp $
27  * $DragonFly: src/sbin/kldstat/kldstat.c,v 1.5 2008/01/08 15:27:57 matthias Exp $
28  */
29 
30 #include <sys/cdefs.h>
31 
32 #include <err.h>
33 #include <stdint.h>
34 #include <stdio.h>
35 #include <stdlib.h>
36 #include <unistd.h>
37 #include <sys/types.h>
38 #include <sys/param.h>
39 #include <sys/module.h>
40 #include <sys/linker.h>
41 
42 #define	POINTER_WIDTH	((int)(sizeof(void *) * 2 + 2))
43 
44 static void
45 printmod(int modid)
46 {
47     struct module_stat stat;
48 
49     stat.version = sizeof(struct module_stat);
50     if (modstat(modid, &stat) < 0)
51 	warn("can't stat module id %d", modid);
52     else
53 	printf("\t\t%2d %s\n", stat.id, stat.name);
54 }
55 
56 static void
57 printfile(int fileid, int verbose)
58 {
59     struct kld_file_stat stat;
60     int modid;
61 
62     stat.version = sizeof(struct kld_file_stat);
63     if (kldstat(fileid, &stat) < 0)
64 	warn("can't stat file id %d", fileid);
65     else
66 	printf("%2d %4d %p %-8jx %s\n",
67 	       stat.id, stat.refs, stat.address, (uintmax_t)stat.size,
68 	       stat.name);
69 
70     if (verbose) {
71 	printf("\tContains modules:\n");
72 	printf("\t\tId Name\n");
73 	for (modid = kldfirstmod(fileid); modid > 0;
74 	     modid = modfnext(modid))
75 	    printmod(modid);
76     }
77 }
78 
79 static void
80 usage(void)
81 {
82     fprintf(stderr, "usage: kldstat [-v] [-i id] [-n filename]\n");
83     fprintf(stderr, "       kldstat [-q] [-m modname]\n");
84     exit(1);
85 }
86 
87 int
88 main(int argc, char** argv)
89 {
90     int c;
91     int verbose = 0;
92     int fileid = 0;
93     int quiet = 0;
94     char* filename = NULL;
95     char* modname = NULL;
96     char* p;
97 
98     while ((c = getopt(argc, argv, "i:m:n:qv")) != -1)
99 	switch (c) {
100 	case 'i':
101 	    fileid = (int)strtoul(optarg, &p, 10);
102 	    if (*p != '\0')
103 		usage();
104 	    break;
105 	case 'm':
106 	    modname = optarg;
107 	    break;
108 	case 'n':
109 	    filename = optarg;
110 	    break;
111 	case 'q':
112 	    quiet = 1;
113 	    break;
114 	case 'v':
115 	    verbose = 1;
116 	    break;
117 	default:
118 	    usage();
119 	}
120     argc -= optind;
121     argv += optind;
122 
123     if (argc != 0)
124 	usage();
125 
126     if (modname != NULL) {
127 	int modid;
128 	struct module_stat stat;
129 
130 	if ((modid = modfind(modname)) < 0) {
131 	    if (!quiet)
132 		warn("can't find module %s", modname);
133 	    return 1;
134 	} else if (quiet) {
135 	    return 0;
136 	}
137 
138 	stat.version = sizeof(struct module_stat);
139 	if (modstat(modid, &stat) < 0)
140 	    warn("can't stat module id %d", modid);
141 	else {
142 	    printf("Id  Refs Name\n");
143 	    printf("%3d %4d %s\n", stat.id, stat.refs, stat.name);
144 	}
145 
146 	return 0;
147     }
148 
149     if (filename != NULL) {
150 	if ((fileid = kldfind(filename)) < 0)
151 	    err(1, "can't find file %s", filename);
152     }
153 
154     printf("Id Refs Address%*c Size     Name\n", POINTER_WIDTH - 7, ' ');
155     if (fileid != 0)
156 	printfile(fileid, verbose);
157     else
158 	for (fileid = kldnext(0); fileid > 0; fileid = kldnext(fileid))
159 	    printfile(fileid, verbose);
160 
161     return 0;
162 }
163