2 * Copyright (c) 2002, 2003 Greg Lehey
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
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.
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.
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.8 2008/09/19 17:49:31 swildner Exp $ */
37 #include <sys/param.h>
40 #include <sys/types.h>
45 const char *modules_path; /* path relative to kernel
47 const char *outfile; /* and where to write the output */
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
57 tokenize(char *cptr, char *token[], int maxtoken)
59 char delim; /* delimiter to search for */
60 int tokennr; /* index of this token */
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 */
69 token[tokennr] = cptr; /* point to it */
70 tokennr++; /* one more */
71 if (tokennr == maxtoken) /* run off the end? */
73 if ((delim == '\'') || (delim == '"')) { /* delimitered */
77 && (cptr[-1] != '\\')) { /* found the partner */
78 cptr++; /* move on past */
79 if (!isspace(*cptr)) /* no space after closing quote */
81 *cptr++ = '\0'; /* delimit */
82 } else if ((*cptr == '\0')
83 || (*cptr == '\n')) /* end of line */
86 } else { /* not quoted */
87 while ((*cptr != '\0') && (!isspace(*cptr)) && (*cptr != '\n'))
89 if (*cptr != '\0') /* not end of the line, */
90 *cptr++ = '\0'; /* delimit and move to the next */
93 return maxtoken; /* can't get here */
97 findmodule(char *mod_path, const char *module_name)
99 char *const path_argv[2] = { mod_path, NULL };
100 char *module_path = NULL;
101 size_t module_name_len = strlen(module_name);
105 if (mod_path == NULL) {
107 "Can't allocate memory to traverse a path: %s (%d)\n",
112 fts = fts_open(path_argv, FTS_PHYSICAL | FTS_NOCHDIR, NULL);
115 "Can't begin traversing path %s: %s (%d)\n",
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) {
126 "Error while traversing path %s: %s (%d)\n",
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)
136 if (asprintf(&module_path,
138 (int)ftsent->fts_pathlen,
139 ftsent->fts_path) == -1) {
141 "Can't allocate memory traversing path %s: %s (%d)\n",
149 if (ftsent == NULL && errno != 0) {
151 "Couldn't complete traversing path %s: %s (%d)\n",
159 return (module_path);
163 usage(const char *myname)
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",
175 main(int argc, char *argv[])
180 FILE *out; /* output file */
182 int tokens; /* number of tokens on line */
184 const char *filemode = "w"; /* mode for outfile */
185 char cwd[MAXPATHLEN]; /* current directory */
186 char *token[MAXTOKEN];
189 getcwd(cwd, MAXPATHLEN); /* find where we are */
191 while ((ch = getopt(argc, argv, "afk")) != -1) {
193 case 'k': /* get input from kldstat(8) */
194 if (!(kldstat = popen("kldstat", "r"))) {
195 perror("Can't start kldstat");
199 case 'a': /* append to outfile */
202 case 'f': /* find .ko (recursively) */
215 modules_path = argv[0];
228 "Extraneous startup information: \"%s\", aborting\n",
230 usage(getprogname());
233 if (modules_path == NULL)
234 modules_path = "/boot/modules";
237 if ((out = fopen(outfile, filemode)) == NULL) {
239 "Can't open output file %s: %s (%d)\n",
245 while (fgets(buf, MAXLINE, kldstat)) {
246 if ((!(strstr(buf, "kernel")))
249 long long textaddr = 0;
250 long long dataaddr = 0;
251 long long bssaddr = 0;
253 tokens = tokenize(buf, token, MAXTOKEN);
256 base = strtoll(token[2], NULL, 16);
260 "/usr/bin/objdump --section-headers %s/%s",
266 modpath = findmodule(strdup(modules_path), token[4]);
271 "/usr/bin/objdump --section-headers %s",
275 if (!(objcopy = popen(ocbuf, "r"))) {
277 "Can't start %s: %s (%d)\n",
283 while (fgets(ocbuf, MAXLINE, objcopy)) {
285 char *octoken[MAXTOKEN];
287 octokens = tokenize(ocbuf, octoken, MAXTOKEN);
289 if (!strcmp(octoken[1], ".text"))
290 textaddr = strtoll(octoken[3], NULL, 16) + base;
291 else if (!strcmp(octoken[1], ".data"))
292 dataaddr = strtoll(octoken[3], NULL, 16) + base;
293 else if (!strcmp(octoken[1], ".bss"))
294 bssaddr = strtoll(octoken[3], NULL, 16) + base;
297 if (textaddr) { /* we must have a text address */
300 "add-symbol-file %s%s%s/%s 0x%llx",
301 modules_path[0] != '/' ? cwd : "",
302 modules_path[0] != '/' ? "/" : "",
309 modpath = findmodule(strdup(modules_path), token[4]);
313 "add-symbol-file %s 0x%llx",
319 fprintf(out, " -s .data 0x%llx", dataaddr);
321 fprintf(out, " -s .bss 0x%llx", bssaddr);