Home | History | Annotate | Line # | Download | only in savecore
savecore.c revision 1.86.28.2
      1 /*	$NetBSD: savecore.c,v 1.86.28.2 2020/04/08 14:07:20 martin Exp $	*/
      2 
      3 /*-
      4  * Copyright (c) 1986, 1992, 1993
      5  *	The Regents of the University of California.  All rights reserved.
      6  *
      7  * Redistribution and use in source and binary forms, with or without
      8  * modification, are permitted provided that the following conditions
      9  * are met:
     10  * 1. Redistributions of source code must retain the above copyright
     11  *    notice, this list of conditions and the following disclaimer.
     12  * 2. Redistributions in binary form must reproduce the above copyright
     13  *    notice, this list of conditions and the following disclaimer in the
     14  *    documentation and/or other materials provided with the distribution.
     15  * 3. Neither the name of the University nor the names of its contributors
     16  *    may be used to endorse or promote products derived from this software
     17  *    without specific prior written permission.
     18  *
     19  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
     20  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     21  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     22  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
     23  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     24  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
     25  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     26  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     27  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     28  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     29  * SUCH DAMAGE.
     30  */
     31 
     32 #include <sys/cdefs.h>
     33 #ifndef lint
     34 __COPYRIGHT("@(#) Copyright (c) 1986, 1992, 1993\
     35  The Regents of the University of California.  All rights reserved.");
     36 #endif /* not lint */
     37 
     38 #ifndef lint
     39 #if 0
     40 static char sccsid[] = "@(#)savecore.c	8.5 (Berkeley) 4/28/95";
     41 #else
     42 __RCSID("$NetBSD: savecore.c,v 1.86.28.2 2020/04/08 14:07:20 martin Exp $");
     43 #endif
     44 #endif /* not lint */
     45 
     46 #define _KSYMS_PRIVATE
     47 
     48 #include <stdbool.h>
     49 
     50 #include <sys/param.h>
     51 #include <sys/mount.h>
     52 #include <sys/msgbuf.h>
     53 #include <sys/syslog.h>
     54 #include <sys/time.h>
     55 #include <sys/ksyms.h>
     56 
     57 #include <dirent.h>
     58 #include <errno.h>
     59 #include <fcntl.h>
     60 #include <nlist.h>
     61 #include <paths.h>
     62 #include <stddef.h>
     63 #include <stdio.h>
     64 #include <stdlib.h>
     65 #include <string.h>
     66 #include <time.h>
     67 #include <tzfile.h>
     68 #include <unistd.h>
     69 #include <util.h>
     70 #include <limits.h>
     71 #include <stdarg.h>
     72 #include <kvm.h>
     73 
     74 extern FILE *zopen(const char *fname, const char *mode);
     75 
     76 /*
     77  * Note that KREAD_LOGWARN takes a variable name, not pointer to it, unlike
     78  * KREAD() itself.
     79  */
     80 #define	KREAD(kd, addr, p)\
     81 	(kvm_read(kd, addr, (char *)(p), sizeof(*(p))) != sizeof(*(p)))
     82 #define KREAD_LOGWARN(kd, addr, p, err)					\
     83 do {									\
     84 	if (KREAD(kd, addr, &(p)) != 0) {				\
     85 		syslog(LOG_WARNING, "%s:%d: kvm_read " #p ": %s",	\
     86 			__func__, __LINE__, kvm_geterr(kd));		\
     87 		err;							\
     88 	}								\
     89 } while (0)
     90 
     91 static struct nlist current_nl[] = {	/* Namelist for currently running system. */
     92 #define	X_DUMPDEV	0
     93 	{ .n_name = "_dumpdev" },
     94 #define	X_DUMPLO	1
     95 	{ .n_name = "_dumplo" },
     96 #define	X_TIME_SECOND	2
     97 	{ .n_name = "_time_second" },
     98 #define X_TIME		3
     99 	{ .n_name = "_time" },
    100 #define	X_DUMPSIZE	4
    101 	{ .n_name = "_dumpsize" },
    102 #define	X_VERSION	5
    103 	{ .n_name = "_version" },
    104 #define	X_DUMPMAG	6
    105 	{ .n_name = "_dumpmag" },
    106 #define	X_PANICSTR	7
    107 	{ .n_name = "_panicstr" },
    108 #define	X_PANICSTART	8
    109 	{ .n_name = "_panicstart" },
    110 #define	X_PANICEND	9
    111 	{ .n_name = "_panicend" },
    112 #define	X_MSGBUF	10
    113 	{ .n_name = "_msgbufp" },
    114 #define	X_DUMPCDEV	11
    115 	{ .n_name = "_dumpcdev" },
    116 #define X_SYMSZ		12
    117 	{ .n_name = "_ksyms_symsz" },
    118 #define X_STRSZ		13
    119 	{ .n_name = "_ksyms_strsz" },
    120 #define X_KHDR		14
    121 	{ .n_name = "_ksyms_hdr" },
    122 #define X_SYMTABS	15
    123 	{ .n_name = "_ksyms_symtabs" },
    124 	{ .n_name = NULL },
    125 };
    126 static int cursyms[] = { X_DUMPDEV, X_DUMPLO, X_VERSION, X_DUMPMAG, X_DUMPCDEV, -1 };
    127 static int dumpsyms[] = { X_TIME_SECOND, X_TIME, X_DUMPSIZE, X_VERSION, X_PANICSTR,
    128     X_DUMPMAG, X_SYMSZ, X_STRSZ, X_KHDR, X_SYMTABS, -1 };
    129 
    130 static struct nlist dump_nl[] = {	/* Name list for dumped system. */
    131 	{ .n_name = "_dumpdev" },	/* Entries MUST be the same as */
    132 	{ .n_name = "_dumplo" },	/*	those in current_nl[].  */
    133 	{ .n_name = "_time_second" },
    134 	{ .n_name = "_time" },
    135 	{ .n_name = "_dumpsize" },
    136 	{ .n_name = "_version" },
    137 	{ .n_name = "_dumpmag" },
    138 	{ .n_name = "_panicstr" },
    139 	{ .n_name = "_panicstart" },
    140 	{ .n_name = "_panicend" },
    141 	{ .n_name = "_msgbufp" },
    142 	{ .n_name = "_dumpcdev" },
    143 	{ .n_name = "_ksyms_symsz" },
    144 	{ .n_name = "_ksyms_strsz" },
    145 	{ .n_name = "_ksyms_hdr" },
    146 	{ .n_name = "_ksyms_symtabs" },
    147 	{ .n_name = NULL },
    148 };
    149 
    150 /* Types match kernel declarations. */
    151 static off_t	dumplo;				/* where dump starts on dumpdev */
    152 static u_int32_t dumpmag;			/* magic number in dump */
    153 static int	dumpsize;			/* amount of memory dumped */
    154 static off_t dumpbytes;			/* in bytes */
    155 
    156 static const char	*kernel;		/* name of used kernel */
    157 static const char	*dirname;		/* directory to save dumps in */
    158 static char	*ddname;			/* name of dump device */
    159 static dev_t	dumpdev;			/* dump device */
    160 static dev_t	dumpcdev = NODEV;		/* dump device (char equivalent) */
    161 static int	dumpfd;				/* read/write descriptor on dev */
    162 static kvm_t	*kd_dump;			/* kvm descriptor on dev	*/
    163 static time_t	now;				/* current date */
    164 static char	panic_mesg[1024];
    165 static long	panicstr;
    166 static char	vers[1024];
    167 static char	gzmode[3];
    168 
    169 static void	check_kmem(void);
    170 static int	check_space(void);
    171 static void	clear_dump(void);
    172 static int	Create(char *, int);
    173 static int	dump_exists(int);
    174 static char	*find_dev(dev_t, mode_t);
    175 static int	get_crashtime(void);
    176 static void	kmem_setup(int);
    177 static void	Lseek(int, off_t, int);
    178 static int	Open(const char *, int rw);
    179 static void	save_core(int);
    180 __dead static void	usage(const char *fmt, ...) __printflike(1, 2);
    181 
    182 int
    183 main(int argc, char *argv[])
    184 {
    185 	int ch, level, testonly, compress, force, clear, verbose;
    186 	char *ep;
    187 
    188 	kernel = NULL;
    189 	level = 1;		/* default to fastest gzip compression */
    190 	force = 0;
    191 	clear = 0;
    192 	testonly = 0;
    193 	verbose = 0;
    194 	compress = 0;
    195 	gzmode[0] = 'w';
    196 
    197 	openlog("savecore", LOG_PERROR, LOG_DAEMON);
    198 
    199 	while ((ch = getopt(argc, argv, "cdfnN:vzZ:")) != -1)
    200 		switch(ch) {
    201 		case 'c':
    202 			clear = 1;
    203 			break;
    204 		case 'd':		/* Not documented. */
    205 		case 'v':
    206 			verbose = 1;
    207 			break;
    208 		case 'f':
    209 			force = 1;
    210 			break;
    211 		case 'n':
    212 			testonly = 1;
    213 			break;
    214 		case 'N':
    215 			kernel = optarg;
    216 			break;
    217 		case 'z':
    218 			compress = 1;
    219 			break;
    220 		case 'Z':
    221 			level = (int)strtol(optarg, &ep, 10);
    222 			if (level < 0 || level > 9)
    223 				usage("Invalid compression `%s'", optarg);
    224 			break;
    225 		case '?':
    226 			usage("Missing argument for flag `%c'", optopt);
    227 		default:
    228 			usage("Unknown flag `%c'", ch);
    229 		}
    230 	argc -= optind;
    231 	argv += optind;
    232 
    233 	if (argc != 0)
    234 		dirname = argv[0];
    235 	else
    236 		dirname = "/var/crash";
    237 
    238 	gzmode[1] = level + '0';
    239 
    240 	(void)time(&now);
    241 	kmem_setup(verbose);
    242 
    243 	if (clear && !testonly) {
    244 		clear_dump();
    245 		exit(0);
    246 	}
    247 
    248 	if (!dump_exists(verbose) && !force)
    249 		exit(1);
    250 
    251 	if (testonly)
    252 		/* If -n was passed and there was a dump, exit at level 0 */
    253 		exit(0);
    254 
    255 	check_kmem();
    256 
    257 	if (panicstr)
    258 		syslog(LOG_ALERT, "reboot after panic: %s", panic_mesg);
    259 	else
    260 		syslog(LOG_ALERT, "reboot");
    261 
    262 	if ((!get_crashtime() || !check_space()) && !force)
    263 		exit(1);
    264 
    265 	save_core(compress);
    266 
    267 	clear_dump();
    268 	exit(0);
    269 }
    270 
    271 static void
    272 read_string(kvm_t *kd, u_long kva, char *buf, size_t size)
    273 {
    274 	size_t i;
    275 
    276 	for (i = 0; i < size - 1; i++) {
    277 		(void)kvm_read(kd, kva + i, buf + i, 1);
    278 		if (buf[i] == '\0')
    279 			return;
    280 	}
    281 
    282 	buf[size - 1] = '\0';
    283 }
    284 
    285 static void
    286 kmem_setup(int verbose)
    287 {
    288 	long l_dumplo;
    289 	kvm_t *kd_kern;
    290 	char errbuf[_POSIX2_LINE_MAX];
    291 	int i, hdrsz;
    292 
    293 	/*
    294 	 * Some names we need for the currently running system, others for
    295 	 * the system that was running when the dump was made.  The values
    296 	 * obtained from the current system are used to look for things in
    297 	 * /dev/kmem that cannot be found in the kernel namelist, but are
    298 	 * presumed to be the same (since the disk partitions are probably
    299 	 * the same!)
    300 	 */
    301 	kd_kern = kvm_openfiles(kernel, NULL, NULL, O_RDONLY, errbuf);
    302 	if (kd_kern == NULL) {
    303 		syslog(LOG_ERR, "%s: kvm_openfiles: %s", kernel, errbuf);
    304 		exit(1);
    305 	}
    306 	if (kvm_nlist(kd_kern, current_nl) == -1)
    307 		syslog(LOG_ERR, "%s: kvm_nlist: %s", kernel,
    308 		    kvm_geterr(kd_kern));
    309 
    310 	for (i = 0; cursyms[i] != -1; i++) {
    311 		if (current_nl[cursyms[i]].n_value != 0)
    312 			continue;
    313 		switch (cursyms[i]) {
    314 		case X_TIME_SECOND:
    315 		case X_TIME:
    316 		case X_DUMPCDEV:
    317 			break;
    318 		default:
    319 			syslog(LOG_ERR, "%s: %s not in namelist",
    320 			    kernel, current_nl[cursyms[i]].n_name);
    321 			exit(1);
    322 		}
    323 	}
    324 
    325 	KREAD_LOGWARN(kd_kern, current_nl[X_DUMPDEV].n_value, dumpdev, exit(1));
    326 	if (dumpdev == NODEV) {
    327 		syslog(LOG_WARNING, "no core dump (no dumpdev)");
    328 		exit(1);
    329 	}
    330 	KREAD_LOGWARN(kd_kern, current_nl[X_DUMPLO].n_value, l_dumplo, exit(1));
    331 	if (l_dumplo == -1) {
    332 		syslog(LOG_WARNING, "no core dump (invalid dumplo)");
    333 		exit(1);
    334 	}
    335 	dumplo = DEV_BSIZE * (off_t) l_dumplo;
    336 
    337 	if (verbose)
    338 		(void)printf("dumplo = %lld (%ld * %ld)\n",
    339 		    (long long)dumplo, (long)(dumplo / DEV_BSIZE), (long)DEV_BSIZE);
    340 	KREAD_LOGWARN(kd_kern, current_nl[X_DUMPMAG].n_value, dumpmag, exit(1));
    341 
    342 	read_string(kd_kern, current_nl[X_VERSION].n_value, vers,
    343 	    sizeof(vers));
    344 
    345 	if (current_nl[X_DUMPCDEV].n_value != 0) {
    346 		KREAD_LOGWARN(kd_kern, current_nl[X_DUMPCDEV].n_value, dumpcdev,
    347 		    exit(1));
    348 		ddname = find_dev(dumpcdev, S_IFCHR);
    349 	} else
    350 		ddname = find_dev(dumpdev, S_IFBLK);
    351 	if (strncmp(ddname, "/dev/cons", 8) == 0 ||
    352 	    strncmp(ddname, "/dev/tty", 7) == 0 ||
    353 	    strncmp(ddname, "/dev/pty", 7) == 0 ||
    354 	    strncmp(ddname, "/dev/pts", 7) == 0) {
    355 		syslog(LOG_ERR, "dumpdev %s is tty; override kernel", ddname);
    356 		exit(1);
    357 	}
    358 	dumpfd = Open(ddname, O_RDWR);
    359 
    360 	kd_dump = kvm_openfiles(kernel, ddname, NULL, O_RDWR, errbuf);
    361 	if (kd_dump == NULL) {
    362 		syslog(LOG_ERR, "%s: kvm_openfiles: %s", kernel, errbuf);
    363 		exit(1);
    364 	}
    365 
    366 	if (kvm_nlist(kd_dump, dump_nl) == -1)
    367 		syslog(LOG_ERR, "%s: kvm_nlist: %s", kernel,
    368 		    kvm_geterr(kd_dump));
    369 
    370 	for (i = 0; dumpsyms[i] != -1; i++)
    371 		if (dump_nl[dumpsyms[i]].n_value == 0 &&
    372 			dumpsyms[i] != X_TIME_SECOND &&
    373 			dumpsyms[i] != X_TIME) {
    374 			syslog(LOG_ERR, "%s: %s not in namelist",
    375 			    kernel, dump_nl[dumpsyms[i]].n_name);
    376 			exit(1);
    377 		}
    378 	hdrsz = kvm_dump_mkheader(kd_dump, dumplo);
    379 
    380 	/*
    381 	 * If 'hdrsz' == 0, kvm_dump_mkheader() failed on the magic-number
    382 	 * checks, ergo no dump is present...
    383 	 */
    384 	if (hdrsz == 0) {
    385 		syslog(LOG_WARNING, "no core dump");
    386 		exit(1);
    387 	}
    388 	if (hdrsz == -1) {
    389 		syslog(LOG_ERR, "%s: kvm_dump_mkheader: %s", kernel,
    390 		    kvm_geterr(kd_dump));
    391 		exit(1);
    392 	}
    393 	dumplo += hdrsz;
    394 	kvm_close(kd_kern);
    395 }
    396 
    397 static void
    398 check_kmem(void)
    399 {
    400 	char *cp, *bufdata;
    401 	struct kern_msgbuf msgbuf, *bufp;
    402 	long panicloc, panicstart, panicend;
    403 	char core_vers[1024];
    404 
    405 	read_string(kd_dump, dump_nl[X_VERSION].n_value, core_vers,
    406 	    sizeof(core_vers));
    407 
    408 	if (strcmp(vers, core_vers) != 0)
    409 		syslog(LOG_WARNING,
    410 		    "warning: %s version mismatch:\n\t%s\nand\t%s\n",
    411 		    kvm_getkernelname(kd_dump), vers, core_vers);
    412 
    413 	panicstart = panicend = 0;
    414 	KREAD_LOGWARN(kd_dump, dump_nl[X_PANICSTART].n_value, panicstart,
    415 	    goto nomsguf);
    416 	KREAD_LOGWARN(kd_dump, dump_nl[X_PANICEND].n_value, panicend,
    417 	    goto nomsguf);
    418 
    419 	if (panicstart != 0 && panicend != 0) {
    420 		KREAD_LOGWARN(kd_dump, dump_nl[X_MSGBUF].n_value, bufp,
    421 		    goto nomsguf);
    422 		/* Reads msg_bufs[1], but doesn't matter. */
    423 		KREAD_LOGWARN(kd_dump, (long)bufp, msgbuf,
    424 		    goto nomsguf);
    425 		if (msgbuf.msg_magic != MSG_MAGIC) {
    426 			syslog(LOG_WARNING, "msgbuf magic incorrect (%lx != %lx)",
    427 			    msgbuf.msg_magic, (long)MSG_MAGIC);
    428 			goto nomsguf;
    429 		}
    430 		bufdata = malloc(msgbuf.msg_bufs);
    431 		if (bufdata == NULL) {
    432 			syslog(LOG_WARNING, "couldn't allocate space for msgbuf data");
    433 			goto nomsguf;
    434 		}
    435 		if (kvm_read(kd_dump, (long)&bufp->msg_bufc, bufdata,
    436 		    msgbuf.msg_bufs) != msgbuf.msg_bufs) {
    437 			syslog(LOG_WARNING, "kvm_read dmesg buffer: %s",
    438 			    kvm_geterr(kd_dump));
    439 			free(bufdata);
    440 			goto nomsguf;
    441 		}
    442 		cp = panic_mesg;
    443 		while (panicstart != panicend && cp < &panic_mesg[sizeof(panic_mesg)-1]) {
    444 			*cp++ = bufdata[panicstart];
    445 			panicstart++;
    446 			if (panicstart >= msgbuf.msg_bufs)
    447 				panicstart = 0;
    448 		}
    449 		/* Don't end in a new-line */
    450 		cp = &panic_mesg[strlen(panic_mesg)] - 1;
    451 		if (*cp == '\n')
    452 			*cp = '\0';
    453 		panic_mesg[sizeof(panic_mesg) - 1] = '\0';
    454 		free(bufdata);
    455 
    456 		panicstr = 1;	/* anything not zero */
    457 		return;
    458 	}
    459 nomsguf:
    460 	KREAD_LOGWARN(kd_dump, dump_nl[X_PANICSTR].n_value, panicstr,
    461 	    return);
    462 	if (panicstr) {
    463 		cp = panic_mesg;
    464 		panicloc = panicstr;
    465 		do {
    466 			if (KREAD(kd_dump, panicloc, cp) != 0) {
    467 				syslog(LOG_WARNING, "kvm_read msgbuf: %s",
    468 				    kvm_geterr(kd_dump));
    469 				break;
    470 			}
    471 			panicloc++;
    472 		} while (*cp++ && cp < &panic_mesg[sizeof(panic_mesg)-1]);
    473 		panic_mesg[sizeof(panic_mesg) - 1] = '\0';
    474 	}
    475 }
    476 
    477 static int
    478 dump_exists(int verbose)
    479 {
    480 	u_int32_t newdumpmag;
    481 
    482 	/* Read the dump magic and size. */
    483 	KREAD_LOGWARN(kd_dump, dump_nl[X_DUMPMAG].n_value, newdumpmag, return 0);
    484 	KREAD_LOGWARN(kd_dump, dump_nl[X_DUMPSIZE].n_value, dumpsize, return 0);
    485 
    486 	dumpbytes = (off_t)dumpsize * getpagesize();
    487 
    488 	/*
    489 	 * Return zero if core dump doesn't seem to be there, and note
    490 	 * it for syslog.  This check and return happens after the dump size
    491 	 * is read, so dumpsize is whether or not the core is valid (for -f).
    492 	 */
    493 	if (newdumpmag != dumpmag) {
    494 		if (verbose)
    495 			syslog(LOG_WARNING, "magic number mismatch "
    496 			    "(%#x != %#x)", newdumpmag, dumpmag);
    497 		syslog(LOG_WARNING, "no core dump");
    498 		return (0);
    499 	}
    500 	return (1);
    501 }
    502 
    503 static void
    504 clear_dump(void)
    505 {
    506 	if (kvm_dump_inval(kd_dump) == -1)
    507 		syslog(LOG_ERR, "%s: kvm_dump_inval: %s", ddname,
    508 		    kvm_geterr(kd_dump));
    509 
    510 }
    511 
    512 static char buf[1024 * 1024];
    513 
    514 static void
    515 save_kernel(FILE *fp, char *path)
    516 {
    517 	int nw, nr, ifd;
    518 
    519 	ifd = Open(kernel, O_RDONLY);
    520 	while ((nr = read(ifd, buf, sizeof(buf))) > 0) {
    521 		nw = fwrite(buf, 1, nr, fp);
    522 		if (nw != nr) {
    523 			syslog(LOG_ERR, "%s: %s",
    524 			    path, strerror(nw == 0 ? EIO : errno));
    525 			syslog(LOG_WARNING,
    526 			    "WARNING: kernel may be incomplete");
    527 			exit(1);
    528 		}
    529 	}
    530 	if (nr < 0) {
    531 		syslog(LOG_ERR, "%s: %m", kernel);
    532 		syslog(LOG_WARNING, "WARNING: kernel may be incomplete");
    533 		exit(1);
    534 	}
    535 }
    536 
    537 static int
    538 ksymsget(u_long addr, void *ptr, size_t size)
    539 {
    540 
    541 	if ((size_t)kvm_read(kd_dump, addr, ptr, size) != size) {
    542 		syslog(LOG_WARNING, "kvm_read ksyms: %s", kvm_geterr(kd_dump));
    543 		return 1;
    544 	}
    545 	return 0;
    546 }
    547 
    548 static int
    549 save_ksyms(FILE *fp, char *path)
    550 {
    551 	struct ksyms_hdr khdr;
    552 	int nw, symsz, strsz;
    553 	TAILQ_HEAD(, ksyms_symtab) symtabs;
    554 	struct ksyms_symtab st, *stptr;
    555 	void *p;
    556 
    557 	/* Get basic info and ELF headers, check if ksyms was on. */
    558 	if (ksymsget(dump_nl[X_KHDR].n_value, &khdr, sizeof(khdr)))
    559 		return 1;
    560 	if (ksymsget(dump_nl[X_SYMSZ].n_value, &symsz, sizeof(symsz)))
    561 		return 1;
    562 	if (ksymsget(dump_nl[X_STRSZ].n_value, &strsz, sizeof(strsz)))
    563 		return 1;
    564 	if (symsz == 0 || strsz == 0)
    565 		return 1;
    566 
    567 	/* Update the ELF section headers for symbols/strings. */
    568 	khdr.kh_shdr[SYMTAB].sh_size = symsz;
    569 	khdr.kh_shdr[SYMTAB].sh_info = symsz / sizeof(Elf_Sym);
    570 	khdr.kh_shdr[STRTAB].sh_offset = symsz +
    571 	    khdr.kh_shdr[SYMTAB].sh_offset;
    572 	khdr.kh_shdr[STRTAB].sh_size = strsz;
    573 
    574 	/* Write out the ELF headers. */
    575 	nw = fwrite(&khdr, 1, sizeof(khdr), fp);
    576 	if (nw != sizeof(khdr)) {
    577 		syslog(LOG_ERR, "%s: %s",
    578 		    path, strerror(nw == 0 ? EIO : errno));
    579 		syslog(LOG_WARNING,
    580 		    "WARNING: kernel may be incomplete");
    581 		exit(1);
    582         }
    583 
    584         /* Dump symbol table. */
    585 	if (ksymsget(dump_nl[X_SYMTABS].n_value, &symtabs, sizeof(symtabs)))
    586 		return 1;
    587 	stptr = TAILQ_FIRST(&symtabs);
    588 	while (stptr != NULL) {
    589 		if (ksymsget((u_long)stptr, &st, sizeof(st)))
    590 			return 1;
    591 		stptr = TAILQ_NEXT(&st, sd_queue);
    592 		if ((p = malloc(st.sd_symsize)) == NULL)
    593 			return 1;
    594 		if (ksymsget((u_long)st.sd_symstart, p, st.sd_symsize)) {
    595 			free(p);
    596 			return 1;
    597 		}
    598 		nw = fwrite(p, 1, st.sd_symsize, fp);
    599 		free(p);
    600 		if (nw != st.sd_symsize) {
    601 			syslog(LOG_ERR, "%s: %s",
    602 			    path, strerror(nw == 0 ? EIO : errno));
    603 			syslog(LOG_WARNING,
    604 			    "WARNING: kernel may be incomplete");
    605 			exit(1);
    606 		}
    607 	}
    608 
    609 	/* Dump string table. */
    610 	if (ksymsget(dump_nl[X_SYMTABS].n_value, &symtabs, sizeof(symtabs)))
    611 		return 1;
    612 	stptr = TAILQ_FIRST(&symtabs);
    613 	while (stptr != NULL) {
    614 		if (ksymsget((u_long)stptr, &st, sizeof(st)))
    615 			return 1;
    616 		stptr = TAILQ_NEXT(&st, sd_queue);
    617 		if ((p = malloc(st.sd_symsize)) == NULL)
    618 			return 1;
    619 		if (ksymsget((u_long)st.sd_strstart, p, st.sd_strsize)) {
    620 			free(p);
    621 			return 1;
    622 		}
    623 		nw = fwrite(p, 1, st.sd_strsize, fp);
    624 		free(p);
    625 		if (nw != st.sd_strsize) {
    626 			syslog(LOG_ERR, "%s: %s",
    627 			    path, strerror(nw == 0 ? EIO : errno));
    628 			syslog(LOG_WARNING,
    629 			    "WARNING: kernel may be incomplete");
    630 			exit(1);
    631 		}
    632 	}
    633 
    634 	return 0;
    635 }
    636 
    637 static void
    638 save_core(int compress)
    639 {
    640 	FILE *fp;
    641 	int bounds, ifd, nr, nw, ofd, tryksyms;
    642 	char path[MAXPATHLEN], rbuf[MAXPATHLEN];
    643 	const char *rawp;
    644 
    645 	ofd = -1;
    646 	/*
    647 	 * Get the current number and update the bounds file.  Do the update
    648 	 * now, because may fail later and don't want to overwrite anything.
    649 	 */
    650 	umask(066);
    651 	(void)snprintf(path, sizeof(path), "%s/bounds", dirname);
    652 	if ((fp = fopen(path, "r")) == NULL)
    653 		goto err1;
    654 	if (fgets(buf, sizeof(buf), fp) == NULL) {
    655 		if (ferror(fp))
    656 err1:			syslog(LOG_WARNING, "%s: %m", path);
    657 		bounds = 0;
    658 	} else
    659 		bounds = atoi(buf);
    660 	if (fp != NULL)
    661 		(void)fclose(fp);
    662 	if ((fp = fopen(path, "w")) == NULL)
    663 		syslog(LOG_ERR, "%s: %m", path);
    664 	else {
    665 		(void)fprintf(fp, "%d\n", bounds + 1);
    666 		(void)fclose(fp);
    667 	}
    668 
    669 	/* Create the core file. */
    670 	(void)snprintf(path, sizeof(path), "%s/netbsd.%d.core%s",
    671 	    dirname, bounds, compress ? ".gz" : "");
    672 	if (compress) {
    673 		if ((fp = zopen(path, gzmode)) == NULL) {
    674 			syslog(LOG_ERR, "%s: %m", path);
    675 			exit(1);
    676 		}
    677 	} else {
    678 		ofd = Create(path, S_IRUSR | S_IWUSR | S_IRGRP | S_IROTH);
    679 		fp  = fdopen(ofd, "w");
    680 		if (fp == NULL) {
    681 			syslog(LOG_ERR, "%s: fdopen: %m", path);
    682 			exit(1);
    683 		}
    684 	}
    685 
    686 	if (dumpcdev == NODEV) {
    687 		/* Open the raw device. */
    688 		rawp = getdiskrawname(rbuf, sizeof(rbuf), ddname);
    689 		if (rawp == NULL) {
    690 			syslog(LOG_WARNING, "%s: %m; can't convert to raw",
    691 			    ddname);
    692 			rawp = ddname;
    693 		}
    694 		if ((ifd = open(rawp, O_RDONLY)) == -1) {
    695 			syslog(LOG_WARNING, "%s: %m; using block device",
    696 			    rawp);
    697 			ifd = dumpfd;
    698 		}
    699 	} else {
    700 		rawp = ddname;
    701 		ifd = dumpfd;
    702 	}
    703 
    704 	/* Seek to the start of the core. */
    705 	Lseek(ifd, dumplo, SEEK_SET);
    706 
    707 	if (kvm_dump_wrtheader(kd_dump, fp, (int32_t)dumpbytes) == -1) {
    708 		syslog(LOG_ERR, "kvm_dump_wrtheader: %s : %s", path,
    709 		    kvm_geterr(kd_dump));
    710 		exit(1);
    711 	}
    712 
    713 	/* Copy the core file. */
    714 	syslog(LOG_NOTICE, "writing %score to %s",
    715 	    compress ? "compressed " : "", path);
    716 	for (; dumpbytes > (off_t)0; dumpbytes -= (off_t)nr) {
    717 		char nbuf[7];
    718 		humanize_number(nbuf, 7, dumpbytes, "", HN_AUTOSCALE, 0);
    719 		(void)printf("%7s\r", nbuf);
    720 		(void)fflush(stdout);
    721 		nr = read(ifd, buf, MIN(dumpbytes, (off_t)sizeof(buf)));
    722 		if (nr <= 0) {
    723 			if (nr == 0)
    724 				syslog(LOG_WARNING,
    725 				    "WARNING: EOF on dump device");
    726 			else
    727 				syslog(LOG_ERR, "%s: %m", rawp);
    728 			goto err2;
    729 		}
    730 		nw = fwrite(buf, 1, nr, fp);
    731 		if (nw != nr) {
    732 			syslog(LOG_ERR, "%s: %s",
    733 			    path, strerror(nw == 0 ? EIO : errno));
    734 err2:			syslog(LOG_WARNING,
    735 			    "WARNING: core may be incomplete");
    736 			(void)printf("\n");
    737 			exit(1);
    738 		}
    739 	}
    740 	if (dumpcdev == NODEV)
    741 		(void)close(ifd);
    742 	(void)fclose(fp);
    743 
    744 	/* Create a kernel. */
    745 	(void)snprintf(path, sizeof(path), "%s/netbsd.%d%s",
    746 	    dirname, bounds, compress ? ".gz" : "");
    747 	syslog(LOG_NOTICE, "writing %skernel to %s",
    748 	    compress ? "compressed " : "", path);
    749 	for (tryksyms = 1;; tryksyms = 0) {
    750 		if (compress) {
    751 			if ((fp = zopen(path, gzmode)) == NULL) {
    752 				syslog(LOG_ERR, "%s: %m", path);
    753 				exit(1);
    754 			}
    755 		} else {
    756 			ofd = Create(path, S_IRUSR | S_IWUSR);
    757 			fp  = fdopen(ofd, "w");
    758 			if (fp == NULL) {
    759 				syslog(LOG_ERR, "fdopen: %m");
    760 				exit(1);
    761 			}
    762 		}
    763 		if (tryksyms) {
    764 			if (!save_ksyms(fp, path))
    765 				break;
    766 			(void)fclose(fp);
    767 			unlink(path);
    768 		} else {
    769 			save_kernel(fp, path);
    770 			break;
    771 		}
    772 	}
    773 	(void)fclose(fp);
    774 
    775 	/*
    776 	 * For development systems where the crash occurs during boot
    777 	 * to multiuser.
    778 	 */
    779 	sync();
    780 	sleep(1);
    781 	sync();
    782 	sleep(1);
    783 }
    784 
    785 static char *
    786 find_dev(dev_t dev, mode_t type)
    787 {
    788 	DIR *dfd;
    789 	struct dirent *dir;
    790 	struct stat sb;
    791 	char *dp, device[MAXPATHLEN + 1], *p;
    792 	size_t l;
    793 
    794 	if ((dfd = opendir(_PATH_DEV)) == NULL) {
    795 		syslog(LOG_ERR, "%s: %m", _PATH_DEV);
    796 		exit(1);
    797 	}
    798 	strlcpy(device, _PATH_DEV, sizeof(device));
    799 	p = &device[strlen(device)];
    800 	l = sizeof(device) - strlen(device);
    801 	while ((dir = readdir(dfd))) {
    802 		strlcpy(p, dir->d_name, l);
    803 		if (lstat(device, &sb)) {
    804 			syslog(LOG_ERR, "%s: %m", device);
    805 			continue;
    806 		}
    807 		if ((sb.st_mode & S_IFMT) != type)
    808 			continue;
    809 		if (dev == sb.st_rdev) {
    810 			closedir(dfd);
    811 			if ((dp = strdup(device)) == NULL) {
    812 				syslog(LOG_ERR, "%m");
    813 				exit(1);
    814 			}
    815 			return (dp);
    816 		}
    817 	}
    818 	closedir(dfd);
    819 	syslog(LOG_ERR, "can't find device %lld/%lld",
    820 	    (long long)major(dev), (long long)minor(dev));
    821 	exit(1);
    822 }
    823 
    824 static int
    825 get_crashtime(void)
    826 {
    827 	time_t dumptime;			/* Time the dump was taken. */
    828 	struct timeval dtime;
    829 
    830 	if (KREAD(kd_dump, dump_nl[X_TIME_SECOND].n_value, &dumptime) != 0) {
    831 		if (KREAD(kd_dump, dump_nl[X_TIME].n_value, &dtime) != 0) {
    832 			syslog(LOG_WARNING, "kvm_read dumptime: %s (and _time_second "
    833 			    "is not defined also)", kvm_geterr(kd_dump));
    834 			return (0);
    835 		}
    836 		dumptime = dtime.tv_sec;
    837 	}
    838 	if (dumptime == 0) {
    839 		syslog(LOG_WARNING, "dump time is zero");
    840 		return (0);
    841 	}
    842 	syslog(LOG_INFO, "system went down at %s", ctime(&dumptime));
    843 #define	LEEWAY	(60 * SECSPERDAY)
    844 	if (dumptime < now - LEEWAY || dumptime > now + LEEWAY) {
    845 		syslog(LOG_WARNING, "dump time is unreasonable");
    846 		return (0);
    847 	}
    848 	return (1);
    849 }
    850 
    851 static int
    852 check_space(void)
    853 {
    854 	FILE *fp;
    855 	off_t minfree, spacefree, kernelsize, needed;
    856 	struct stat st;
    857 	struct statvfs fsbuf;
    858 	char mbuf[100], path[MAXPATHLEN];
    859 
    860 	/* XXX assume a reasonable default, unless we find a kernel. */
    861 	kernelsize = 20 * 1024 * 1024;
    862 	if (!stat(kernel, &st)) kernelsize = st.st_blocks * S_BLKSIZE;
    863 	if (statvfs(dirname, &fsbuf) < 0) {
    864 		syslog(LOG_ERR, "%s: %m", dirname);
    865 		exit(1);
    866 	}
    867 	spacefree = fsbuf.f_bavail;
    868 	spacefree *= fsbuf.f_frsize;
    869 	spacefree /= 1024;
    870 
    871 	(void)snprintf(path, sizeof(path), "%s/minfree", dirname);
    872 	if ((fp = fopen(path, "r")) == NULL)
    873 		minfree = 0;
    874 	else {
    875 		if (fgets(mbuf, sizeof(mbuf), fp) == NULL)
    876 			minfree = 0;
    877 		else
    878 			minfree = atoi(mbuf);
    879 		(void)fclose(fp);
    880 	}
    881 
    882 	needed = (dumpbytes + kernelsize) / 1024;
    883  	if (minfree > 0 && spacefree - needed < minfree) {
    884 		syslog(LOG_WARNING,
    885 		    "no dump, not enough free space in %s", dirname);
    886 		return (0);
    887 	}
    888 	if (spacefree - needed < minfree)
    889 		syslog(LOG_WARNING,
    890 		    "dump performed, but free space threshold crossed");
    891 	return (1);
    892 }
    893 
    894 static int
    895 Open(const char *name, int rw)
    896 {
    897 	int fd;
    898 
    899 	if ((fd = open(name, rw, 0)) < 0) {
    900 		syslog(LOG_ERR, "%s: %m", name);
    901 		exit(1);
    902 	}
    903 	return (fd);
    904 }
    905 
    906 static void
    907 Lseek(int fd, off_t off, int flag)
    908 {
    909 	off_t ret;
    910 
    911 	ret = lseek(fd, off, flag);
    912 	if (ret == -1) {
    913 		syslog(LOG_ERR, "lseek: %m");
    914 		exit(1);
    915 	}
    916 }
    917 
    918 static int
    919 Create(char *file, int mode)
    920 {
    921 	int fd;
    922 
    923 	fd = open(file, O_WRONLY | O_CREAT | O_TRUNC, mode);
    924 	if (fd < 0) {
    925 		syslog(LOG_ERR, "%s: %m", file);
    926 		exit(1);
    927 	}
    928 	return (fd);
    929 }
    930 
    931 static void
    932 usage(const char *fmt, ...)
    933 {
    934 	va_list ap;
    935 	va_start(ap, fmt);
    936 	(void)vsyslog(LOG_ERR, fmt, ap);
    937 	va_end(ap);
    938 	(void)syslog(LOG_ERR,
    939 	    "Usage: %s [-cfnvz] [-N system] [-Z level] [directory]",
    940 	    getprogname());
    941 	exit(1);
    942 }
    943