xref: /dragonfly/usr.sbin/asf/asf.c (revision 3641b7ca) 
1 /*
2  * Copyright (c) 2002, 2003 Greg Lehey
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 ``as is'' and any express
15  * or implied warranties, including, but not limited to, the implied
16  * warranties of merchantability and fitness for a particular purpose
17  * are disclaimed.  In no event shall the author be liable for any
18  * direct, indirect, incidental, special, exemplary, or consequential
19  * damages (including, but not limited to, procurement of substitute
20  * goods or services; loss of use, data, or profits; or business
21  * interruption) however caused and on any theory of liability,
22  * whether in contract, strict liability, or tort (including
23  * negligence or otherwise) arising in any way out of the use of this
24  * software, even if advised of the possibility of such damage.
25  */
26 /* $Id: asf.c,v 1.6 2003/11/04 06:38:37 green Exp $ */
27 /* $FreeBSD: src/usr.sbin/asf/asf.c,v 1.6 2003/11/04 06:38:37 green Exp $ */
28 /* $DragonFly: src/usr.sbin/asf/asf.c,v 1.5 2008/05/19 10:13:08 corecode Exp $ */
29 
30 #define MAXLINE 1024
31 #include <ctype.h>
32 #include <errno.h>
33 #include <stdio.h>
34 #include <stdlib.h>
35 #include <string.h>
36 #include <sys/file.h>
37 #include <sys/param.h>
38 #include <sys/stat.h>
39 #include <sys/wait.h>
40 #include <sys/types.h>
41 #include <fts.h>
42 #include <unistd.h>
43 
44 #define MAXTOKEN 10
45 const char *modules_path;		/* path relative to kernel
46 					 * build directory */
47 const char *outfile;			/* and where to write the output */
48 
49 /*
50  * Take a blank separated list of tokens and turn it into a list of
51  * individual nul-delimited strings.  Build a list of pointers at
52  * token, which must have enough space for the tokens.  Return the
53  * number of tokens, or -1 on error (typically a missing string
54  * delimiter).
55  */
56 static int
57 tokenize(char *cptr, char *token[], int maxtoken)
58 {
59     char delim;				/* delimiter to search for */
60     int tokennr;			/* index of this token */
61 
62     for (tokennr = 0; tokennr < maxtoken;) {
63 	while (isspace(*cptr))
64 	    cptr++;			/* skip initial white space */
65 	if ((*cptr == '\0') || (*cptr == '\n')
66 	    || (*cptr == '#'))		/* end of line */
67 	    return tokennr;		/* return number of tokens found */
68 	delim = *cptr;
69 	token[tokennr] = cptr;		/* point to it */
70 	tokennr++;			/* one more */
71 	if (tokennr == maxtoken)	/* run off the end? */
72 	    return tokennr;
73 	if ((delim == '\'') || (delim == '"')) { /* delimitered */
74 	    for (;;) {
75 		cptr++;
76 		if ((*cptr == delim)
77 		    && (cptr[-1] != '\\')) { /* found the partner */
78 		    cptr++;		/* move on past */
79 		    if (!isspace(*cptr)) /* no space after closing quote */
80 			return -1;
81 		    *cptr++ = '\0';	/* delimit */
82 		} else if ((*cptr == '\0')
83 		    || (*cptr == '\n'))	/* end of line */
84 		    return -1;
85 	    }
86 	} else {			/* not quoted */
87 	    while ((*cptr != '\0') && (!isspace(*cptr)) && (*cptr != '\n'))
88 		cptr++;
89 	    if (*cptr != '\0')		/* not end of the line, */
90 		*cptr++ = '\0';		/* delimit and move to the next */
91 	}
92     }
93     return maxtoken;			/* can't get here */
94 }
95 
96 static char *
97 findmodule(char *modules_path, const char *module_name)
98 {
99     char *const path_argv[2] = { modules_path, NULL };
100     char *module_path = NULL;
101     int module_name_len = strlen(module_name);
102     FTS *fts;
103     FTSENT *ftsent;
104 
105     if (modules_path == NULL) {
106 	fprintf(stderr,
107 	    "Can't allocate memory to traverse a path: %s (%d)\n",
108 	    strerror(errno),
109 	    errno);
110 	exit(1);
111     }
112     fts = fts_open(path_argv, FTS_PHYSICAL | FTS_NOCHDIR, NULL);
113     if (fts == NULL) {
114 	fprintf(stderr,
115 	    "Can't begin traversing path %s: %s (%d)\n",
116 	    modules_path,
117 	    strerror(errno),
118 	    errno);
119 	exit(1);
120     }
121     while ((ftsent = fts_read(fts)) != NULL) {
122 	if (ftsent->fts_info == FTS_DNR ||
123 	    ftsent->fts_info == FTS_ERR ||
124 	    ftsent->fts_info == FTS_NS) {
125 	    fprintf(stderr,
126 		"Error while traversing path %s: %s (%d)\n",
127 		modules_path,
128 		strerror(errno),
129 		errno);
130 	    exit(1);
131 	}
132 	if (ftsent->fts_info != FTS_F ||
133 	    ftsent->fts_namelen != module_name_len ||
134 	    memcmp(module_name, ftsent->fts_name, module_name_len) != 0)
135 		continue;
136 	if (asprintf(&module_path,
137 	    "%.*s",
138 	    ftsent->fts_pathlen,
139 	    ftsent->fts_path) == -1) {
140 	    fprintf(stderr,
141 		"Can't allocate memory traversing path %s: %s (%d)\n",
142 		modules_path,
143 		strerror(errno),
144 		errno);
145 	    exit(1);
146 	}
147 	break;
148     }
149     if (ftsent == NULL && errno != 0) {
150 	fprintf(stderr,
151 	    "Couldn't complete traversing path %s: %s (%d)\n",
152 	    modules_path,
153 	    strerror(errno),
154 	    errno);
155 	exit(1);
156     }
157     fts_close(fts);
158     free(modules_path);
159     return (module_path);
160 }
161 
162 static void
163 usage(const char *myname)
164 {
165     fprintf(stderr,
166 	"Usage:\n"
167 	"%s [-a] [-f] [-k] [modules-path [outfile]]\n\n"
168 	"\t-a\tappend to outfile)\n"
169 	"\t-f\tfind the module in any subdirectory of module-path\n"
170 	"\t-k\ttake input from kldstat(8)\n",
171 	myname);
172 }
173 
174 int
175 main(int argc, char *argv[])
176 {
177     char buf[MAXLINE];
178     FILE *kldstat;
179     FILE *objcopy;
180     FILE *out;				/* output file */
181     char ocbuf[MAXLINE];
182     int tokens;				/* number of tokens on line */
183     int i, ch;
184     const char *filemode = "w";		/* mode for outfile */
185     char cwd[MAXPATHLEN];		/* current directory */
186     char *token[MAXTOKEN];
187     int dofind = 0;
188 
189     getcwd(cwd, MAXPATHLEN);		/* find where we are */
190     kldstat = stdin;
191     while ((ch = getopt(argc, argv, "afks")) != -1) {
192 	switch (ch) {
193 	case 'k': /* get input from kldstat(8) */
194 	    if (!(kldstat = popen("kldstat", "r"))) {
195 		perror("Can't start kldstat");
196 		return 1;
197 	    }
198 	    break;
199 	case 'a': /* append to outfile */
200 	    filemode = "a";
201 	    break;
202 	case 'f': /* find .ko (recursively) */
203 	    dofind = 1;
204 	    break;
205 	default:
206 	    fprintf(stderr,
207 		    "Invalid option: %s, aborting\n",
208 		    argv[i]);
209 	    usage(argv[0]);
210 	    return 1;
211 	}
212     }
213 
214     argv += optind;
215     argc -= optind;
216 
217     if (argc >= 1) {
218 	modules_path = argv[0];
219 	argc--;
220 	argv++;
221     }
222 
223     if (argc >= 1) {
224 	outfile = argv[0];
225 	argc--;
226 	argv++;
227     }
228 
229     if (argc > 0) {
230 	fprintf(stderr,
231 	    "Extraneous startup information: \"%s\", aborting\n",
232 	    argv[0]);
233 	usage(getprogname());
234 	return 1;
235     }
236     if (modules_path == NULL)
237 	modules_path = "/modules";
238     if (outfile == NULL)
239 	outfile = ".asf";
240     if ((out = fopen(outfile, filemode)) == NULL) {
241 	fprintf(stderr,
242 	    "Can't open output file %s: %s (%d)\n",
243 	    outfile,
244 	    strerror(errno),
245 	    errno);
246 	return 1;
247     }
248     while (fgets(buf, MAXLINE, kldstat)) {
249 	if ((!(strstr(buf, "kernel")))
250 	    && buf[0] != 'I') {
251 	    quad_t base;
252 	    quad_t textaddr;
253 	    quad_t dataaddr;
254 	    quad_t bssaddr;
255 
256 	    tokens = tokenize(buf, token, MAXTOKEN);
257 	    base = strtoll(token[2], NULL, 16);
258 	    if (!dofind) {
259 		snprintf(ocbuf,
260 		    MAXLINE,
261 		    "/usr/bin/objdump --section-headers %s/%s",
262 		    modules_path,
263 		    token[4]);
264 	    } else {
265 		char *modpath;
266 
267 		modpath = findmodule(strdup(modules_path), token[4]);
268 		if (modpath == NULL)
269 		    continue;
270 		snprintf(ocbuf,
271 		    MAXLINE,
272 		    "/usr/bin/objdump --section-headers %s",
273 		    modpath);
274 		free(modpath);
275 	    }
276 	    if (!(objcopy = popen(ocbuf, "r"))) {
277 		fprintf(stderr,
278 		    "Can't start %s: %s (%d)\n",
279 		    ocbuf,
280 		    strerror(errno),
281 		    errno);
282 		return 1;
283 	    }
284 	    while (fgets(ocbuf, MAXLINE, objcopy)) {
285 		int octokens;
286 		char *octoken[MAXTOKEN];
287 
288 		octokens = tokenize(ocbuf, octoken, MAXTOKEN);
289 		if (octokens > 1) {
290 		    if (!strcmp(octoken[1], ".text"))
291 			textaddr = strtoll(octoken[3], NULL, 16) + base;
292 		    else if (!strcmp(octoken[1], ".data"))
293 			dataaddr = strtoll(octoken[3], NULL, 16) + base;
294 		    else if (!strcmp(octoken[1], ".bss"))
295 			bssaddr = strtoll(octoken[3], NULL, 16) + base;
296 		}
297 	    }
298 	    if (textaddr) {		/* we must have a text address */
299 		if (!dofind) {
300 		    fprintf(out,
301 			"add-symbol-file %s%s%s/%s 0x%llx",
302 			modules_path[0] != '/' ? cwd : "",
303 			modules_path[0] != '/' ? "/" : "",
304 			modules_path,
305 			token[4],
306 			textaddr);
307 		} else {
308 		    char *modpath;
309 
310 		    modpath = findmodule(strdup(modules_path), token[4]);
311 		    if (modpath == NULL)
312 			continue;
313 		    fprintf(out,
314 			"add-symbol-file %s 0x%llx",
315 			modpath,
316 			textaddr);
317 		    free(modpath);
318 		}
319 		if (dataaddr)
320 		    fprintf(out, " -s .data 0x%llx", dataaddr);
321 		if (bssaddr)
322 		    fprintf(out, " -s .bss 0x%llx", bssaddr);
323 		fprintf(out, "\n");
324 	    }
325 	}
326     }
327     return 0;
328 }
329