freebsd-dev/usr.sbin/asf/asf.c
Brian Feldman afdc68c42f Add a "-f" flag for asf(8) which performs a search to find the each module
no matter where in the directory structure it may be.  Use this and the "-k"
flag in the generated gdbinit files so that the "getsyms" function in gdb
requires no user intervention to run and will find every module if they're
in the kernel build's module directory.  This is still quite useful for
cases where gdb knows that the path for some modules is /boot/kernel and
others are in the object directory for /usr/src/sys/$ARCH/compile/kernel.

Approved by:	grog
2003-11-04 06:38:37 +00:00

330 lines
9.1 KiB
C

/*
* Copyright (c) 2002, 2003 Greg Lehey
* All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
* notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
* notice, this list of conditions and the following disclaimer in the
* documentation and/or other materials provided with the distribution.
*
* This software is provided by the author ``as is'' and any express
* or implied warranties, including, but not limited to, the implied
* warranties of merchantability and fitness for a particular purpose
* are disclaimed. In no event shall the author be liable for any
* direct, indirect, incidental, special, exemplary, or consequential
* damages (including, but not limited to, procurement of substitute
* goods or services; loss of use, data, or profits; or business
* interruption) however caused and on any theory of liability,
* whether in contract, strict liability, or tort (including
* negligence or otherwise) arising in any way out of the use of this
* software, even if advised of the possibility of such damage.
*/
/* $Id: asf.c,v 1.4 2003/05/04 02:55:20 grog Exp grog $ */
/* $FreeBSD$ */
#define MAXLINE 1024
#include <ctype.h>
#include <errno.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <sys/file.h>
#include <sys/param.h>
#include <sys/stat.h>
#include <sys/wait.h>
#include <sys/types.h>
#include <fts.h>
#include <unistd.h>
#define MAXTOKEN 10
const char *modules_path; /* path relative to kernel
* build directory */
const char *outfile; /* and where to write the output */
/*
* Take a blank separated list of tokens and turn it into a list of
* individual nul-delimited strings. Build a list of pointers at
* token, which must have enough space for the tokens. Return the
* number of tokens, or -1 on error (typically a missing string
* delimiter).
*/
static int
tokenize(char *cptr, char *token[], int maxtoken)
{
char delim; /* delimiter to search for */
int tokennr; /* index of this token */
for (tokennr = 0; tokennr < maxtoken;) {
while (isspace(*cptr))
cptr++; /* skip initial white space */
if ((*cptr == '\0') || (*cptr == '\n')
|| (*cptr == '#')) /* end of line */
return tokennr; /* return number of tokens found */
delim = *cptr;
token[tokennr] = cptr; /* point to it */
tokennr++; /* one more */
if (tokennr == maxtoken) /* run off the end? */
return tokennr;
if ((delim == '\'') || (delim == '"')) { /* delimitered */
for (;;) {
cptr++;
if ((*cptr == delim)
&& (cptr[-1] != '\\')) { /* found the partner */
cptr++; /* move on past */
if (!isspace(*cptr)) /* no space after closing quote */
return -1;
*cptr++ = '\0'; /* delimit */
} else if ((*cptr == '\0')
|| (*cptr == '\n')) /* end of line */
return -1;
}
} else { /* not quoted */
while ((*cptr != '\0') && (!isspace(*cptr)) && (*cptr != '\n'))
cptr++;
if (*cptr != '\0') /* not end of the line, */
*cptr++ = '\0'; /* delimit and move to the next */
}
}
return maxtoken; /* can't get here */
}
static char *
findmodule(char *modules_path, const char *module_name)
{
char *const path_argv[2] = { modules_path, NULL };
char *module_path = NULL;
int module_name_len = strlen(module_name);
FTS *fts;
FTSENT *ftsent;
if (modules_path == NULL) {
fprintf(stderr,
"Can't allocate memory to traverse a path: %s (%d)\n",
strerror(errno),
errno);
exit(1);
}
fts = fts_open(path_argv, FTS_PHYSICAL | FTS_NOCHDIR, NULL);
if (fts == NULL) {
fprintf(stderr,
"Can't begin traversing path %s: %s (%d)\n",
modules_path,
strerror(errno),
errno);
exit(1);
}
while ((ftsent = fts_read(fts)) != NULL) {
if (ftsent->fts_info == FTS_DNR ||
ftsent->fts_info == FTS_ERR ||
ftsent->fts_info == FTS_NS) {
fprintf(stderr,
"Error while traversing path %s: %s (%d)\n",
modules_path,
strerror(errno),
errno);
exit(1);
}
if (ftsent->fts_info != FTS_F ||
ftsent->fts_namelen != module_name_len ||
memcmp(module_name, ftsent->fts_name, module_name_len) != 0)
continue;
if (asprintf(&module_path,
"%.*s",
ftsent->fts_pathlen,
ftsent->fts_path) == -1) {
fprintf(stderr,
"Can't allocate memory traversing path %s: %s (%d)\n",
modules_path,
strerror(errno),
errno);
exit(1);
}
break;
}
if (ftsent == NULL && errno != 0) {
fprintf(stderr,
"Couldn't complete traversing path %s: %s (%d)\n",
modules_path,
strerror(errno),
errno);
exit(1);
}
fts_close(fts);
free(modules_path);
return (module_path);
}
static void
usage(const char *myname)
{
fprintf(stderr,
"Usage:\n"
"%s [-a] [-f] [-k] [-s] [-x] [modules-path [outfile]]\n\n"
"\t-a\tappend to outfile)\n"
"\t-f\tfind the module in any subdirectory of module-path\n"
"\t-k\ttake input from kldstat(8)\n"
"\t-s\tdon't prepend subdir for module path\n"
"\t-x\tdon't append \".debug\" to module name\n",
myname);
}
int
main(int argc, char *argv[])
{
char buf[MAXLINE];
FILE *kldstat;
FILE *objcopy;
FILE *out; /* output file */
char ocbuf[MAXLINE];
int tokens; /* number of tokens on line */
char basetoken[MAXLINE];
int i;
const char *filemode = "w"; /* mode for outfile */
char cwd[MAXPATHLEN]; /* current directory */
const char *debugname = ".debug"; /* some file names end in this */
char *token[MAXTOKEN];
int nosubdir = 0;
int dofind = 0;
getcwd(cwd, MAXPATHLEN); /* find where we are */
kldstat = stdin;
for (i = 1; i < argc; i++) {
if (argv[i][0] == '-') {
if (strcmp(argv[i], "-k") == 0) { /* get input from kldstat(8) */
if (!(kldstat = popen("kldstat", "r"))) {
perror("Can't start kldstat");
return 1;
}
} else if (strcmp(argv[i], "-a") == 0) /* append to outfile */
filemode = "a";
else if (strcmp(argv[i], "-x") == 0) /* no .debug extension */
debugname = ""; /* nothing */
else if (strcmp(argv[i], "-s") == 0) /* no subdir */
nosubdir = 1; /* nothing */
else if (strcmp(argv[i], "-f") == 0) /* find .ko (recursively) */
dofind = 1;
else {
fprintf(stderr,
"Invalid option: %s, aborting\n",
argv[i]);
usage(argv[0]);
return 1;
}
} else if (modules_path == NULL)
modules_path = argv[i];
else if (outfile == NULL)
outfile = argv[i];
else {
fprintf(stderr,
"Extraneous startup information: \"%s\", aborting\n",
argv[i]);
usage(argv[0]);
return 1;
}
}
if (modules_path == NULL)
modules_path = "modules";
if (outfile == NULL)
outfile = ".asf";
if ((out = fopen(outfile, filemode)) == NULL) {
fprintf(stderr,
"Can't open output file %s: %s (%d)\n",
outfile,
strerror(errno),
errno);
return 1;
}
while (fgets(buf, MAXLINE, kldstat)) {
if ((!(strstr(buf, "kernel")))
&& buf[0] != 'I') {
quad_t base;
quad_t textaddr;
quad_t dataaddr;
quad_t bssaddr;
tokens = tokenize(buf, token, MAXTOKEN);
base = strtoll(token[2], NULL, 16);
if (!dofind) {
strcpy(basetoken, token[4]);
basetoken[strlen(basetoken) - 3] = '/';
basetoken[strlen(basetoken) - 2] = '\0'; /* cut off the .ko */
snprintf(ocbuf,
MAXLINE,
"/usr/bin/objdump --section-headers %s/%s%s%s",
modules_path,
nosubdir ? "" : basetoken,
token[4],
debugname);
} else {
char *modpath;
modpath = findmodule(strdup(modules_path), token[4]);
if (modpath == NULL)
continue;
snprintf(ocbuf,
MAXLINE,
"/usr/bin/objdump --section-headers %s%s",
modpath,
debugname);
free(modpath);
}
if (!(objcopy = popen(ocbuf, "r"))) {
fprintf(stderr,
"Can't start %s: %s (%d)\n",
ocbuf,
strerror(errno),
errno);
return 1;
}
while (fgets(ocbuf, MAXLINE, objcopy)) {
int octokens;
char *octoken[MAXTOKEN];
octokens = tokenize(ocbuf, octoken, MAXTOKEN);
if (octokens > 1) {
if (!strcmp(octoken[1], ".text"))
textaddr = strtoll(octoken[3], NULL, 16) + base;
else if (!strcmp(octoken[1], ".data"))
dataaddr = strtoll(octoken[3], NULL, 16) + base;
else if (!strcmp(octoken[1], ".bss"))
bssaddr = strtoll(octoken[3], NULL, 16) + base;
}
}
if (textaddr) { /* we must have a text address */
if (!dofind) {
fprintf(out,
"add-symbol-file %s/%s/%s%s%s 0x%llx",
cwd,
modules_path,
nosubdir ? "" : basetoken,
token[4],
debugname,
textaddr);
} else {
char *modpath;
modpath = findmodule(strdup(modules_path), token[4]);
if (modpath == NULL)
continue;
fprintf(out,
"add-symbol-file %s%s 0x%llx",
modpath,
debugname,
textaddr);
free(modpath);
}
if (dataaddr)
fprintf(out, " -s .data 0x%llx", dataaddr);
if (bssaddr)
fprintf(out, " -s .bss 0x%llx", bssaddr);
fprintf(out, "\n");
}
}
}
return 0;
}