File: [local] / src / usr.sbin / kvm_mkdb / kvm_mkdb.c (download)
Revision 1.15, Fri Nov 21 09:11:25 2003 UTC (20 years, 6 months ago) by djm
Branch: MAIN
CVS Tags: OPENBSD_3_7_BASE, OPENBSD_3_7, OPENBSD_3_6_BASE, OPENBSD_3_6, OPENBSD_3_5_BASE, OPENBSD_3_5 Changes since 1.14: +35 -13 lines
add -o option to generate kvm database in alternate directory;
manpage nits jmc@ ok tedu@
|
/* $OpenBSD: kvm_mkdb.c,v 1.15 2003/11/21 09:11:25 djm Exp $ */
/*-
* Copyright (c) 1990, 1993
* The Regents of the University of California. 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.
* 3. Neither the name of the University nor the names of its contributors
* may be used to endorse or promote products derived from this software
* without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``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 REGENTS OR CONTRIBUTORS 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.
*/
#ifndef lint
static const char copyright[] =
"@(#) Copyright (c) 1990, 1993\n\
The Regents of the University of California. All rights reserved.\n";
#endif /* not lint */
#ifndef lint
#if 0
static char sccsid[] = "from: @(#)kvm_mkdb.c 8.3 (Berkeley) 5/4/95";
#else
static const char rcsid[] = "$OpenBSD: kvm_mkdb.c,v 1.15 2003/11/21 09:11:25 djm Exp $";
#endif
#endif /* not lint */
#include <sys/param.h>
#include <sys/stat.h>
#include <db.h>
#include <err.h>
#include <errno.h>
#include <fcntl.h>
#include <libgen.h>
#include <paths.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <sys/types.h>
#include <sys/time.h>
#include <sys/resource.h>
#include "extern.h"
void usage(void);
int kvm_mkdb(int, const char *, char *, char *, int);
HASHINFO openinfo = {
4096, /* bsize */
128, /* ffactor */
1024, /* nelem */
2048 * 1024, /* cachesize */
NULL, /* hash() */
0 /* lorder */
};
int
main(int argc, char *argv[])
{
struct rlimit rl;
int fd, rval, ch, verbose = 0;
char *nlistpath, *nlistname;
char dbdir[MAXPATHLEN];
/* Increase our data size to the max if we can. */
if (getrlimit(RLIMIT_DATA, &rl) == 0) {
rl.rlim_cur = rl.rlim_max;
if (setrlimit(RLIMIT_DATA, &rl) < 0)
warn("can't set rlimit data size");
}
strlcpy(dbdir, _PATH_VARDB, sizeof(dbdir));
while ((ch = getopt(argc, argv, "vo:")) != -1)
switch (ch) {
case 'v':
verbose = 1;
break;
case 'o':
rval = strlcpy(dbdir, optarg, sizeof(dbdir));
if (rval == 0 || rval + 1 >= sizeof(dbdir))
errx(1, "Invalid directory");
/* Make sure there is a '/' at the end of the path */
if (dbdir[strlen(dbdir) - 1] != '/')
strlcat(dbdir, "/", sizeof(dbdir));
break;
default:
usage();
}
argc -= optind;
argv += optind;
if (argc > 1)
usage();
/* If no kernel specified use _PATH_KSYMS and fall back to _PATH_UNIX */
if (argc > 0) {
nlistpath = argv[0];
nlistname = basename(nlistpath);
if ((fd = open(nlistpath, O_RDONLY, 0)) == -1)
err(1, "can't open %s", nlistpath);
rval = kvm_mkdb(fd, dbdir, nlistpath, nlistname, verbose);
} else {
nlistname = basename(_PATH_UNIX);
if ((fd = open((nlistpath = _PATH_KSYMS), O_RDONLY, 0)) == -1 ||
(rval = kvm_mkdb(fd, dbdir, nlistpath, nlistname,
verbose)) != 0) {
if (fd == -1)
warnx("can't open %s", _PATH_KSYMS);
else
warnx("will try again using %s instead", _PATH_UNIX);
if ((fd = open((nlistpath = _PATH_UNIX), O_RDONLY, 0)) == -1)
err(1, "can't open %s", nlistpath);
rval = kvm_mkdb(fd, dbdir, nlistpath, nlistname,
verbose);
}
}
exit(rval);
}
int
kvm_mkdb(int fd, const char *dbdir, char *nlistpath, char *nlistname,
int verbose)
{
DB *db;
char dbtemp[MAXPATHLEN], dbname[MAXPATHLEN];
int r;
r = snprintf(dbtemp, sizeof(dbtemp), "%skvm_%s.tmp",
dbdir, nlistname);
if (r < 0 || r > sizeof(dbtemp)) {
warnx("Directory name too long");
return (1);
}
r = snprintf(dbname, sizeof(dbname), "%skvm_%s.db",
dbdir, nlistname);
if (r < 0 || r > sizeof(dbtemp)) {
warnx("Directory name too long");
return (1);
}
/* If the existing db file matches the currently running kernel, exit */
if (testdb(dbname)) {
if (verbose)
warnx("%s already up to date", dbname);
return(0);
} else if (verbose)
warnx("rebuilding %s", dbname);
(void)umask(0);
db = dbopen(dbtemp, O_CREAT | O_EXLOCK | O_TRUNC | O_RDWR,
S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH, DB_HASH, &openinfo);
if (db == NULL) {
warn("can't dbopen %s", dbtemp);
return(1);
}
if (create_knlist(nlistpath, fd, db) != 0) {
(void)unlink(dbtemp);
warn("cannot determine executable type of %s", nlistpath);
return(1);
}
if (db->close(db)) {
warn("can't dbclose %s", dbtemp);
(void)unlink(dbtemp);
return(1);
}
if (rename(dbtemp, dbname)) {
warn("rename %s to %s", dbtemp, dbname);
(void)unlink(dbtemp);
return(1);
}
return(0);
}
void
usage(void)
{
(void)fprintf(stderr, "usage: kvm_mkdb [-v] [-o directory] [file]\n");
exit(1);
}