xref: /freebsd/sbin/dump/main.c (revision f7c7b87fbd1c02016aa310f5d3866382cb38dc40)
18fae3551SRodney W. Grimes /*-
28fae3551SRodney W. Grimes  * Copyright (c) 1980, 1991, 1993, 1994
38fae3551SRodney W. Grimes  *	The Regents of the University of California.  All rights reserved.
48fae3551SRodney W. Grimes  *
58fae3551SRodney W. Grimes  * Redistribution and use in source and binary forms, with or without
68fae3551SRodney W. Grimes  * modification, are permitted provided that the following conditions
78fae3551SRodney W. Grimes  * are met:
88fae3551SRodney W. Grimes  * 1. Redistributions of source code must retain the above copyright
98fae3551SRodney W. Grimes  *    notice, this list of conditions and the following disclaimer.
108fae3551SRodney W. Grimes  * 2. Redistributions in binary form must reproduce the above copyright
118fae3551SRodney W. Grimes  *    notice, this list of conditions and the following disclaimer in the
128fae3551SRodney W. Grimes  *    documentation and/or other materials provided with the distribution.
138fae3551SRodney W. Grimes  * 4. Neither the name of the University nor the names of its contributors
148fae3551SRodney W. Grimes  *    may be used to endorse or promote products derived from this software
158fae3551SRodney W. Grimes  *    without specific prior written permission.
168fae3551SRodney W. Grimes  *
178fae3551SRodney W. Grimes  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
188fae3551SRodney W. Grimes  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
198fae3551SRodney W. Grimes  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
208fae3551SRodney W. Grimes  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
218fae3551SRodney W. Grimes  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
228fae3551SRodney W. Grimes  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
238fae3551SRodney W. Grimes  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
248fae3551SRodney W. Grimes  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
258fae3551SRodney W. Grimes  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
268fae3551SRodney W. Grimes  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
278fae3551SRodney W. Grimes  * SUCH DAMAGE.
288fae3551SRodney W. Grimes  */
298fae3551SRodney W. Grimes 
308fae3551SRodney W. Grimes #ifndef lint
317580ffbbSPhilippe Charnier static const char copyright[] =
328fae3551SRodney W. Grimes "@(#) Copyright (c) 1980, 1991, 1993, 1994\n\
338fae3551SRodney W. Grimes 	The Regents of the University of California.  All rights reserved.\n";
348fae3551SRodney W. Grimes #endif /* not lint */
358fae3551SRodney W. Grimes 
368fae3551SRodney W. Grimes #ifndef lint
377580ffbbSPhilippe Charnier #if 0
38a37c38b8SPeter Wemm static char sccsid[] = "@(#)main.c	8.6 (Berkeley) 5/1/95";
397580ffbbSPhilippe Charnier #endif
407580ffbbSPhilippe Charnier static const char rcsid[] =
417f3dea24SPeter Wemm   "$FreeBSD$";
428fae3551SRodney W. Grimes #endif /* not lint */
438fae3551SRodney W. Grimes 
448fae3551SRodney W. Grimes #include <sys/param.h>
4504285c06SJoerg Wunsch #include <sys/stat.h>
468fae3551SRodney W. Grimes #include <sys/time.h>
47194a6667SKirk McKusick #include <sys/mount.h>
4875766e17SPoul-Henning Kamp #include <sys/disklabel.h>
498fae3551SRodney W. Grimes 
508fae3551SRodney W. Grimes #include <ufs/ufs/dinode.h>
51194a6667SKirk McKusick #include <ufs/ufs/ufsmount.h>
52a37c38b8SPeter Wemm #include <ufs/ffs/fs.h>
538fae3551SRodney W. Grimes 
548fae3551SRodney W. Grimes #include <protocols/dumprestore.h>
558fae3551SRodney W. Grimes 
568fae3551SRodney W. Grimes #include <ctype.h>
57a37c38b8SPeter Wemm #include <err.h>
58194a6667SKirk McKusick #include <errno.h>
598fae3551SRodney W. Grimes #include <fcntl.h>
608fae3551SRodney W. Grimes #include <fstab.h>
6189fdc4e1SMike Barcroft #include <limits.h>
628fae3551SRodney W. Grimes #include <signal.h>
638a0453d6SJohan Karlsson #include <stdint.h>
648fae3551SRodney W. Grimes #include <stdio.h>
658fae3551SRodney W. Grimes #include <stdlib.h>
668fae3551SRodney W. Grimes #include <string.h>
67617dbd3cSIan Dowse #include <timeconv.h>
688fae3551SRodney W. Grimes #include <unistd.h>
698fae3551SRodney W. Grimes 
708fae3551SRodney W. Grimes #include "dump.h"
718fae3551SRodney W. Grimes #include "pathnames.h"
728fae3551SRodney W. Grimes 
738fae3551SRodney W. Grimes int	notify = 0;	/* notify operator flag */
74194a6667SKirk McKusick int	snapdump = 0;	/* dumping live filesystem, so use snapshot */
758fae3551SRodney W. Grimes int	blockswritten = 0;	/* number of blocks written on current tape */
768fae3551SRodney W. Grimes int	tapeno = 0;	/* current tape number */
77019420a5SJoerg Wunsch int	density = 0;	/* density in bytes/0.1" " <- this is for hilit19 */
788fae3551SRodney W. Grimes int	ntrec = NTREC;	/* # tape blocks in each tape record */
798fae3551SRodney W. Grimes int	cartridge = 0;	/* Assume non-cartridge tape */
805941e412SMatthew Dillon int	cachesize = 0;	/* block cache size (in bytes), defaults to 0 */
818fae3551SRodney W. Grimes long	dev_bsize = 1;	/* recalculated below */
828fae3551SRodney W. Grimes long	blocksperfile;	/* output blocks per file */
838fae3551SRodney W. Grimes char	*host = NULL;	/* remote host (if any) */
848fae3551SRodney W. Grimes 
851c85e6a3SKirk McKusick /*
861c85e6a3SKirk McKusick  * Possible superblock locations ordered from most to least likely.
871c85e6a3SKirk McKusick  */
881c85e6a3SKirk McKusick static int sblock_try[] = SBLOCKSEARCH;
891c85e6a3SKirk McKusick 
90194a6667SKirk McKusick static char *getmntpt(char *, int *);
912db673abSWarner Losh static long numarg(const char *, long, long);
922db673abSWarner Losh static void obsolete(int *, char **[]);
932db673abSWarner Losh static void usage(void) __dead2;
948fae3551SRodney W. Grimes 
958fae3551SRodney W. Grimes int
962db673abSWarner Losh main(int argc, char *argv[])
978fae3551SRodney W. Grimes {
98a50e99d1SIan Dowse 	struct stat sb;
99d2334e27SIan Dowse 	ino_t ino;
100d2334e27SIan Dowse 	int dirty;
1011c85e6a3SKirk McKusick 	union dinode *dp;
102d2334e27SIan Dowse 	struct fstab *dt;
103194a6667SKirk McKusick 	char *map, *mntpt;
104194a6667SKirk McKusick 	int ch, mode, mntflags;
1058fae3551SRodney W. Grimes 	int i, anydirskipped, bflag = 0, Tflag = 0, honorlevel = 1;
1069ac0768dSIan Dowse 	int just_estimate = 0;
1078fae3551SRodney W. Grimes 	ino_t maxino;
1085b3817c6SMatthew Dillon 	char *tmsg;
1098fae3551SRodney W. Grimes 
1101c85e6a3SKirk McKusick 	spcl.c_date = _time_to_time64(time(NULL));
1118fae3551SRodney W. Grimes 
1128fae3551SRodney W. Grimes 	tsize = 0;	/* Default later, based on 'c' option for cart tapes */
113f5dcc2f1SWarner Losh 	if ((tape = getenv("TAPE")) == NULL)
1148fae3551SRodney W. Grimes 		tape = _PATH_DEFTAPE;
1158fae3551SRodney W. Grimes 	dumpdates = _PATH_DUMPDATES;
1168fae3551SRodney W. Grimes 	temp = _PATH_DTMP;
1178fae3551SRodney W. Grimes 	if (TP_BSIZE / DEV_BSIZE == 0 || TP_BSIZE % DEV_BSIZE != 0)
1188fae3551SRodney W. Grimes 		quit("TP_BSIZE must be a multiple of DEV_BSIZE\n");
1198fae3551SRodney W. Grimes 	level = '0';
120a37c38b8SPeter Wemm 
121a37c38b8SPeter Wemm 	if (argc < 2)
122a37c38b8SPeter Wemm 		usage();
123a37c38b8SPeter Wemm 
124a37c38b8SPeter Wemm 	obsolete(&argc, &argv);
12544b81f0fSRuslan Ermilov 	while ((ch = getopt(argc, argv,
12644b81f0fSRuslan Ermilov 	    "0123456789aB:b:C:cD:d:f:h:LnSs:T:uWw")) != -1)
127a37c38b8SPeter Wemm 		switch (ch) {
128a37c38b8SPeter Wemm 		/* dump level */
129a37c38b8SPeter Wemm 		case '0': case '1': case '2': case '3': case '4':
130a37c38b8SPeter Wemm 		case '5': case '6': case '7': case '8': case '9':
131a37c38b8SPeter Wemm 			level = ch;
1328fae3551SRodney W. Grimes 			break;
1338fae3551SRodney W. Grimes 
134a37c38b8SPeter Wemm 		case 'a':		/* `auto-size', Write to EOM. */
135a37c38b8SPeter Wemm 			unlimited = 1;
1368fae3551SRodney W. Grimes 			break;
1378fae3551SRodney W. Grimes 
138a37c38b8SPeter Wemm 		case 'B':		/* blocks per output file */
139a37c38b8SPeter Wemm 			blocksperfile = numarg("number of blocks per file",
140a37c38b8SPeter Wemm 			    1L, 0L);
1418fae3551SRodney W. Grimes 			break;
1428fae3551SRodney W. Grimes 
1438fae3551SRodney W. Grimes 		case 'b':		/* blocks per tape write */
144a37c38b8SPeter Wemm 			ntrec = numarg("number of blocks per write",
145a37c38b8SPeter Wemm 			    1L, 1000L);
1468fae3551SRodney W. Grimes 			break;
1478fae3551SRodney W. Grimes 
1481f6a4631SRuslan Ermilov 		case 'C':
1491f6a4631SRuslan Ermilov 			cachesize = numarg("cachesize", 0, 0) * 1024 * 1024;
1501f6a4631SRuslan Ermilov 			break;
1511f6a4631SRuslan Ermilov 
1528fae3551SRodney W. Grimes 		case 'c':		/* Tape is cart. not 9-track */
1538fae3551SRodney W. Grimes 			cartridge = 1;
1548fae3551SRodney W. Grimes 			break;
1558fae3551SRodney W. Grimes 
1561f6a4631SRuslan Ermilov 		case 'D':
1571f6a4631SRuslan Ermilov 			dumpdates = optarg;
1581f6a4631SRuslan Ermilov 			break;
1591f6a4631SRuslan Ermilov 
160a37c38b8SPeter Wemm 		case 'd':		/* density, in bits per inch */
161a37c38b8SPeter Wemm 			density = numarg("density", 10L, 327670L) / 10;
162a37c38b8SPeter Wemm 			if (density >= 625 && !bflag)
163a37c38b8SPeter Wemm 				ntrec = HIGHDENSITYTREC;
164af00957eSJoerg Wunsch 			break;
165af00957eSJoerg Wunsch 
166a37c38b8SPeter Wemm 		case 'f':		/* output file */
167a37c38b8SPeter Wemm 			tape = optarg;
1688fae3551SRodney W. Grimes 			break;
1698fae3551SRodney W. Grimes 
170a37c38b8SPeter Wemm 		case 'h':
171a37c38b8SPeter Wemm 			honorlevel = numarg("honor level", 0L, 10L);
1728fae3551SRodney W. Grimes 			break;
1738fae3551SRodney W. Grimes 
174194a6667SKirk McKusick 		case 'L':
175194a6667SKirk McKusick 			snapdump = 1;
176194a6667SKirk McKusick 			break;
177194a6667SKirk McKusick 
1788fae3551SRodney W. Grimes 		case 'n':		/* notify operators */
1798fae3551SRodney W. Grimes 			notify = 1;
1808fae3551SRodney W. Grimes 			break;
1818fae3551SRodney W. Grimes 
1829ac0768dSIan Dowse 		case 'S':               /* exit after estimating # of tapes */
1839ac0768dSIan Dowse 			just_estimate = 1;
1849ac0768dSIan Dowse 			break;
1859ac0768dSIan Dowse 
1861f6a4631SRuslan Ermilov 		case 's':		/* tape size, feet */
1871f6a4631SRuslan Ermilov 			tsize = numarg("tape size", 1L, 0L) * 12 * 10;
1881f6a4631SRuslan Ermilov 			break;
1891f6a4631SRuslan Ermilov 
190a37c38b8SPeter Wemm 		case 'T':		/* time of last dump */
191a37c38b8SPeter Wemm 			spcl.c_ddate = unctime(optarg);
192a37c38b8SPeter Wemm 			if (spcl.c_ddate < 0) {
193a37c38b8SPeter Wemm 				(void)fprintf(stderr, "bad time \"%s\"\n",
194a37c38b8SPeter Wemm 				    optarg);
195f69e804dSJoseph Koshy 				exit(X_STARTUP);
1968fae3551SRodney W. Grimes 			}
197a37c38b8SPeter Wemm 			Tflag = 1;
198a37c38b8SPeter Wemm 			lastlevel = '?';
199a37c38b8SPeter Wemm 			break;
200a37c38b8SPeter Wemm 
201a37c38b8SPeter Wemm 		case 'u':		/* update /etc/dumpdates */
202a37c38b8SPeter Wemm 			uflag = 1;
203a37c38b8SPeter Wemm 			break;
204a37c38b8SPeter Wemm 
205a37c38b8SPeter Wemm 		case 'W':		/* what to do */
206a37c38b8SPeter Wemm 		case 'w':
207a37c38b8SPeter Wemm 			lastdump(ch);
208f69e804dSJoseph Koshy 			exit(X_FINOK);	/* do nothing else */
209a37c38b8SPeter Wemm 
210a37c38b8SPeter Wemm 		default:
211a37c38b8SPeter Wemm 			usage();
2128fae3551SRodney W. Grimes 		}
213a37c38b8SPeter Wemm 	argc -= optind;
214a37c38b8SPeter Wemm 	argv += optind;
215a37c38b8SPeter Wemm 
2168fae3551SRodney W. Grimes 	if (argc < 1) {
2178fae3551SRodney W. Grimes 		(void)fprintf(stderr, "Must specify disk or file system\n");
218f69e804dSJoseph Koshy 		exit(X_STARTUP);
2198fae3551SRodney W. Grimes 	}
2208fae3551SRodney W. Grimes 	disk = *argv++;
2218fae3551SRodney W. Grimes 	argc--;
2228fae3551SRodney W. Grimes 	if (argc >= 1) {
2238fae3551SRodney W. Grimes 		(void)fprintf(stderr, "Unknown arguments to dump:");
2248fae3551SRodney W. Grimes 		while (argc--)
2258fae3551SRodney W. Grimes 			(void)fprintf(stderr, " %s", *argv++);
2268fae3551SRodney W. Grimes 		(void)fprintf(stderr, "\n");
227f69e804dSJoseph Koshy 		exit(X_STARTUP);
2288fae3551SRodney W. Grimes 	}
2298fae3551SRodney W. Grimes 	if (Tflag && uflag) {
2308fae3551SRodney W. Grimes 	        (void)fprintf(stderr,
2318fae3551SRodney W. Grimes 		    "You cannot use the T and u flags together.\n");
232f69e804dSJoseph Koshy 		exit(X_STARTUP);
2338fae3551SRodney W. Grimes 	}
2348fae3551SRodney W. Grimes 	if (strcmp(tape, "-") == 0) {
2358fae3551SRodney W. Grimes 		pipeout++;
2368fae3551SRodney W. Grimes 		tape = "standard output";
2378fae3551SRodney W. Grimes 	}
2388fae3551SRodney W. Grimes 
2398fae3551SRodney W. Grimes 	if (blocksperfile)
2408fae3551SRodney W. Grimes 		blocksperfile = blocksperfile / ntrec * ntrec; /* round down */
241af00957eSJoerg Wunsch 	else if (!unlimited) {
2428fae3551SRodney W. Grimes 		/*
2438fae3551SRodney W. Grimes 		 * Determine how to default tape size and density
2448fae3551SRodney W. Grimes 		 *
2458fae3551SRodney W. Grimes 		 *         	density				tape size
2468fae3551SRodney W. Grimes 		 * 9-track	1600 bpi (160 bytes/.1")	2300 ft.
2478fae3551SRodney W. Grimes 		 * 9-track	6250 bpi (625 bytes/.1")	2300 ft.
2488fae3551SRodney W. Grimes 		 * cartridge	8000 bpi (100 bytes/.1")	1700 ft.
2498fae3551SRodney W. Grimes 		 *						(450*4 - slop)
250019420a5SJoerg Wunsch 		 * hilit19 hits again: "
2518fae3551SRodney W. Grimes 		 */
2528fae3551SRodney W. Grimes 		if (density == 0)
2538fae3551SRodney W. Grimes 			density = cartridge ? 100 : 160;
2548fae3551SRodney W. Grimes 		if (tsize == 0)
2558fae3551SRodney W. Grimes 			tsize = cartridge ? 1700L*120L : 2300L*120L;
2568fae3551SRodney W. Grimes 	}
2578fae3551SRodney W. Grimes 
258a37c38b8SPeter Wemm 	if (strchr(tape, ':')) {
2598fae3551SRodney W. Grimes 		host = tape;
260a37c38b8SPeter Wemm 		tape = strchr(host, ':');
2618fae3551SRodney W. Grimes 		*tape++ = '\0';
2628fae3551SRodney W. Grimes #ifdef RDUMP
2639e0dc595SPeter Wemm 		if (index(tape, '\n')) {
264f5dcc2f1SWarner Losh 		    (void)fprintf(stderr, "invalid characters in tape\n");
265f69e804dSJoseph Koshy 		    exit(X_STARTUP);
266f5dcc2f1SWarner Losh 		}
2678fae3551SRodney W. Grimes 		if (rmthost(host) == 0)
268f69e804dSJoseph Koshy 			exit(X_STARTUP);
2698fae3551SRodney W. Grimes #else
2708fae3551SRodney W. Grimes 		(void)fprintf(stderr, "remote dump not enabled\n");
271f69e804dSJoseph Koshy 		exit(X_STARTUP);
2728fae3551SRodney W. Grimes #endif
2738fae3551SRodney W. Grimes 	}
2748fae3551SRodney W. Grimes 	(void)setuid(getuid()); /* rmthost() is the only reason to be setuid */
2758fae3551SRodney W. Grimes 
2768fae3551SRodney W. Grimes 	if (signal(SIGHUP, SIG_IGN) != SIG_IGN)
2778fae3551SRodney W. Grimes 		signal(SIGHUP, sig);
2788fae3551SRodney W. Grimes 	if (signal(SIGTRAP, SIG_IGN) != SIG_IGN)
2798fae3551SRodney W. Grimes 		signal(SIGTRAP, sig);
2808fae3551SRodney W. Grimes 	if (signal(SIGFPE, SIG_IGN) != SIG_IGN)
2818fae3551SRodney W. Grimes 		signal(SIGFPE, sig);
2828fae3551SRodney W. Grimes 	if (signal(SIGBUS, SIG_IGN) != SIG_IGN)
2838fae3551SRodney W. Grimes 		signal(SIGBUS, sig);
2848fae3551SRodney W. Grimes 	if (signal(SIGSEGV, SIG_IGN) != SIG_IGN)
2858fae3551SRodney W. Grimes 		signal(SIGSEGV, sig);
2868fae3551SRodney W. Grimes 	if (signal(SIGTERM, SIG_IGN) != SIG_IGN)
2878fae3551SRodney W. Grimes 		signal(SIGTERM, sig);
2888fae3551SRodney W. Grimes 	if (signal(SIGINT, interrupt) == SIG_IGN)
2898fae3551SRodney W. Grimes 		signal(SIGINT, SIG_IGN);
2908fae3551SRodney W. Grimes 
291a3165d16SMatthew N. Dodd 	dump_getfstab();	/* /etc/fstab snarfed */
2928fae3551SRodney W. Grimes 	/*
2938fae3551SRodney W. Grimes 	 *	disk can be either the full special file name,
2948fae3551SRodney W. Grimes 	 *	the suffix of the special file name,
2958fae3551SRodney W. Grimes 	 *	the special name missing the leading '/',
2968fae3551SRodney W. Grimes 	 *	the file system name with or without the leading '/'.
2978fae3551SRodney W. Grimes 	 */
2988fae3551SRodney W. Grimes 	dt = fstabsearch(disk);
2998fae3551SRodney W. Grimes 	if (dt != NULL) {
3008fae3551SRodney W. Grimes 		disk = rawname(dt->fs_spec);
3018fae3551SRodney W. Grimes 		(void)strncpy(spcl.c_dev, dt->fs_spec, NAMELEN);
3028fae3551SRodney W. Grimes 		(void)strncpy(spcl.c_filesys, dt->fs_file, NAMELEN);
3038fae3551SRodney W. Grimes 	} else {
3048fae3551SRodney W. Grimes 		(void)strncpy(spcl.c_dev, disk, NAMELEN);
3058fae3551SRodney W. Grimes 		(void)strncpy(spcl.c_filesys, "an unlisted file system",
3068fae3551SRodney W. Grimes 		    NAMELEN);
3078fae3551SRodney W. Grimes 	}
308f5dcc2f1SWarner Losh 	spcl.c_dev[NAMELEN-1]='\0';
309f5dcc2f1SWarner Losh 	spcl.c_filesys[NAMELEN-1]='\0';
310194a6667SKirk McKusick 
311194a6667SKirk McKusick 	if ((mntpt = getmntpt(disk, &mntflags)) != 0) {
312f7c7b87fSBrian Feldman 		if (mntflags & MNT_RDONLY) {
313f7c7b87fSBrian Feldman 			if (snapdump != 0) {
314194a6667SKirk McKusick 				msg("WARNING: %s\n",
315f7c7b87fSBrian Feldman 				    "-L ignored for read-only filesystem.");
316f7c7b87fSBrian Feldman 				snapdump = 0;
317f7c7b87fSBrian Feldman 			}
318f7c7b87fSBrian Feldman 		} else if (snapdump == 0) {
319f7c7b87fSBrian Feldman 			msg("WARNING: %s%s\n",
320f7c7b87fSBrian Feldman 			    "should use -L when dumping live read-write ",
321f7c7b87fSBrian Feldman 			    "filesystems!");
322194a6667SKirk McKusick 		} else {
323d46b5285SKirk McKusick 			char snapname[BUFSIZ], snapcmd[BUFSIZ];
324194a6667SKirk McKusick 
325d46b5285SKirk McKusick 			snprintf(snapname, sizeof snapname,
326d46b5285SKirk McKusick 			    "%s/.snap/dump_snapshot", mntpt);
3276e3aaeb2SIan Dowse 			snprintf(snapcmd, sizeof snapcmd, "%s %s %s",
3286e3aaeb2SIan Dowse 			    _PATH_MKSNAP_FFS, mntpt, snapname);
329d46b5285SKirk McKusick 			unlink(snapname);
330d46b5285SKirk McKusick 			if (system(snapcmd) != 0)
331194a6667SKirk McKusick 				errx(X_STARTUP, "Cannot create %s: %s\n",
332194a6667SKirk McKusick 				    snapname, strerror(errno));
333194a6667SKirk McKusick 			if ((diskfd = open(snapname, O_RDONLY)) < 0) {
334194a6667SKirk McKusick 				unlink(snapname);
335194a6667SKirk McKusick 				errx(X_STARTUP, "Cannot open %s: %s\n",
336194a6667SKirk McKusick 				    snapname, strerror(errno));
337194a6667SKirk McKusick 			}
338194a6667SKirk McKusick 			unlink(snapname);
339194a6667SKirk McKusick 			if (fstat(diskfd, &sb) != 0)
340194a6667SKirk McKusick 				err(X_STARTUP, "%s: stat", snapname);
341194a6667SKirk McKusick 			spcl.c_date = _time_to_time64(sb.st_mtime);
342194a6667SKirk McKusick 		}
343194a6667SKirk McKusick 	} else if (snapdump != 0) {
344194a6667SKirk McKusick 		msg("WARNING: Cannot use -L on an unmounted filesystem.\n");
345194a6667SKirk McKusick 		snapdump = 0;
346194a6667SKirk McKusick 	}
347194a6667SKirk McKusick 	if (snapdump == 0) {
348194a6667SKirk McKusick 		if ((diskfd = open(disk, O_RDONLY)) < 0)
349194a6667SKirk McKusick 			err(X_STARTUP, "Cannot open %s", disk);
350194a6667SKirk McKusick 		if (fstat(diskfd, &sb) != 0)
351194a6667SKirk McKusick 			err(X_STARTUP, "%s: stat", disk);
352194a6667SKirk McKusick 		if (S_ISDIR(sb.st_mode))
353194a6667SKirk McKusick 			errx(X_STARTUP, "%s: unknown file system", disk);
354194a6667SKirk McKusick 	}
355194a6667SKirk McKusick 
3568fae3551SRodney W. Grimes 	(void)strcpy(spcl.c_label, "none");
3578fae3551SRodney W. Grimes 	(void)gethostname(spcl.c_host, NAMELEN);
3588fae3551SRodney W. Grimes 	spcl.c_level = level - '0';
3598fae3551SRodney W. Grimes 	spcl.c_type = TS_TAPE;
3608fae3551SRodney W. Grimes 
3615b3817c6SMatthew Dillon 	if (spcl.c_date == 0) {
3625b3817c6SMatthew Dillon 		tmsg = "the epoch\n";
3635b3817c6SMatthew Dillon 	} else {
3641c85e6a3SKirk McKusick 		time_t t = _time64_to_time(spcl.c_date);
3655b3817c6SMatthew Dillon 		tmsg = ctime(&t);
3665b3817c6SMatthew Dillon 	}
3675b3817c6SMatthew Dillon 	msg("Date of this level %c dump: %s", level, tmsg);
368194a6667SKirk McKusick 
369194a6667SKirk McKusick 	if (!Tflag)
370194a6667SKirk McKusick 	        getdumptime();		/* /etc/dumpdates snarfed */
3715b3817c6SMatthew Dillon 	if (spcl.c_ddate == 0) {
3725b3817c6SMatthew Dillon 		tmsg = "the epoch\n";
3735b3817c6SMatthew Dillon 	} else {
3741c85e6a3SKirk McKusick 		time_t t = _time64_to_time(spcl.c_ddate);
3755b3817c6SMatthew Dillon 		tmsg = ctime(&t);
3765b3817c6SMatthew Dillon 	}
3775b3817c6SMatthew Dillon  	msg("Date of last level %c dump: %s", lastlevel, tmsg);
378194a6667SKirk McKusick 
379194a6667SKirk McKusick 	msg("Dumping %s%s ", snapdump ? "snapshot of ": "", disk);
3808fae3551SRodney W. Grimes 	if (dt != NULL)
3818fae3551SRodney W. Grimes 		msgtail("(%s) ", dt->fs_file);
3828fae3551SRodney W. Grimes 	if (host)
3838fae3551SRodney W. Grimes 		msgtail("to %s on host %s\n", tape, host);
3848fae3551SRodney W. Grimes 	else
3858fae3551SRodney W. Grimes 		msgtail("to %s\n", tape);
3868fae3551SRodney W. Grimes 
3878fae3551SRodney W. Grimes 	sync();
3888fae3551SRodney W. Grimes 	sblock = (struct fs *)sblock_buf;
3891c85e6a3SKirk McKusick 	for (i = 0; sblock_try[i] != -1; i++) {
390924a7003SKirk McKusick 		sblock->fs_fsize = SBLOCKSIZE; /* needed in bread */
3911c85e6a3SKirk McKusick 		bread(sblock_try[i] >> dev_bshift, (char *) sblock, SBLOCKSIZE);
3921c85e6a3SKirk McKusick 		if ((sblock->fs_magic == FS_UFS1_MAGIC ||
3931c85e6a3SKirk McKusick 		     (sblock->fs_magic == FS_UFS2_MAGIC &&
394ada981b2SKirk McKusick 		      sblock->fs_sblockloc == sblock_try[i])) &&
3951c85e6a3SKirk McKusick 		    sblock->fs_bsize <= MAXBSIZE &&
3961c85e6a3SKirk McKusick 		    sblock->fs_bsize >= sizeof(struct fs))
3971c85e6a3SKirk McKusick 			break;
3981c85e6a3SKirk McKusick 	}
3991c85e6a3SKirk McKusick 	if (sblock_try[i] == -1)
4001c85e6a3SKirk McKusick 		quit("Cannot find file system superblock\n");
4018fae3551SRodney W. Grimes 	dev_bsize = sblock->fs_fsize / fsbtodb(sblock, 1);
4028fae3551SRodney W. Grimes 	dev_bshift = ffs(dev_bsize) - 1;
4038fae3551SRodney W. Grimes 	if (dev_bsize != (1 << dev_bshift))
404617dbd3cSIan Dowse 		quit("dev_bsize (%ld) is not a power of 2", dev_bsize);
4058fae3551SRodney W. Grimes 	tp_bshift = ffs(TP_BSIZE) - 1;
4068fae3551SRodney W. Grimes 	if (TP_BSIZE != (1 << tp_bshift))
4078fae3551SRodney W. Grimes 		quit("TP_BSIZE (%d) is not a power of 2", TP_BSIZE);
4088fae3551SRodney W. Grimes 	maxino = sblock->fs_ipg * sblock->fs_ncg;
40989fdc4e1SMike Barcroft 	mapsize = roundup(howmany(maxino, CHAR_BIT), TP_BSIZE);
4108fae3551SRodney W. Grimes 	usedinomap = (char *)calloc((unsigned) mapsize, sizeof(char));
4118fae3551SRodney W. Grimes 	dumpdirmap = (char *)calloc((unsigned) mapsize, sizeof(char));
4128fae3551SRodney W. Grimes 	dumpinomap = (char *)calloc((unsigned) mapsize, sizeof(char));
4138fae3551SRodney W. Grimes 	tapesize = 3 * (howmany(mapsize * sizeof(char), TP_BSIZE) + 1);
4148fae3551SRodney W. Grimes 
4158fae3551SRodney W. Grimes 	nonodump = spcl.c_level < honorlevel;
4168fae3551SRodney W. Grimes 
4172bb823d2SIan Dowse 	passno = 1;
4182bb823d2SIan Dowse 	setproctitle("%s: pass 1: regular files", disk);
4198fae3551SRodney W. Grimes 	msg("mapping (Pass I) [regular files]\n");
4208fae3551SRodney W. Grimes 	anydirskipped = mapfiles(maxino, &tapesize);
4218fae3551SRodney W. Grimes 
4222bb823d2SIan Dowse 	passno = 2;
4232bb823d2SIan Dowse 	setproctitle("%s: pass 2: directories", disk);
4248fae3551SRodney W. Grimes 	msg("mapping (Pass II) [directories]\n");
4258fae3551SRodney W. Grimes 	while (anydirskipped) {
4268fae3551SRodney W. Grimes 		anydirskipped = mapdirs(maxino, &tapesize);
4278fae3551SRodney W. Grimes 	}
4288fae3551SRodney W. Grimes 
429af00957eSJoerg Wunsch 	if (pipeout || unlimited) {
4308fae3551SRodney W. Grimes 		tapesize += 10;	/* 10 trailer blocks */
4318fae3551SRodney W. Grimes 		msg("estimated %ld tape blocks.\n", tapesize);
4328fae3551SRodney W. Grimes 	} else {
4338fae3551SRodney W. Grimes 		double fetapes;
4348fae3551SRodney W. Grimes 
4358fae3551SRodney W. Grimes 		if (blocksperfile)
4368fae3551SRodney W. Grimes 			fetapes = (double) tapesize / blocksperfile;
4378fae3551SRodney W. Grimes 		else if (cartridge) {
4388fae3551SRodney W. Grimes 			/* Estimate number of tapes, assuming streaming stops at
4398fae3551SRodney W. Grimes 			   the end of each block written, and not in mid-block.
4408fae3551SRodney W. Grimes 			   Assume no erroneous blocks; this can be compensated
4418fae3551SRodney W. Grimes 			   for with an artificially low tape size. */
4428fae3551SRodney W. Grimes 			fetapes =
44371fbc63cSPaul Traina 			(	  (double) tapesize	/* blocks */
4448fae3551SRodney W. Grimes 				* TP_BSIZE	/* bytes/block */
445019420a5SJoerg Wunsch 				* (1.0/density)	/* 0.1" / byte " */
4468fae3551SRodney W. Grimes 			  +
44771fbc63cSPaul Traina 				  (double) tapesize	/* blocks */
4488fae3551SRodney W. Grimes 				* (1.0/ntrec)	/* streaming-stops per block */
449019420a5SJoerg Wunsch 				* 15.48		/* 0.1" / streaming-stop " */
450019420a5SJoerg Wunsch 			) * (1.0 / tsize );	/* tape / 0.1" " */
4518fae3551SRodney W. Grimes 		} else {
4528fae3551SRodney W. Grimes 			/* Estimate number of tapes, for old fashioned 9-track
4538fae3551SRodney W. Grimes 			   tape */
4548fae3551SRodney W. Grimes 			int tenthsperirg = (density == 625) ? 3 : 7;
4558fae3551SRodney W. Grimes 			fetapes =
45671fbc63cSPaul Traina 			(	  (double) tapesize	/* blocks */
4578fae3551SRodney W. Grimes 				* TP_BSIZE	/* bytes / block */
458019420a5SJoerg Wunsch 				* (1.0/density)	/* 0.1" / byte " */
4598fae3551SRodney W. Grimes 			  +
46071fbc63cSPaul Traina 				  (double) tapesize	/* blocks */
4618fae3551SRodney W. Grimes 				* (1.0/ntrec)	/* IRG's / block */
462019420a5SJoerg Wunsch 				* tenthsperirg	/* 0.1" / IRG " */
463019420a5SJoerg Wunsch 			) * (1.0 / tsize );	/* tape / 0.1" " */
4648fae3551SRodney W. Grimes 		}
4658fae3551SRodney W. Grimes 		etapes = fetapes;		/* truncating assignment */
4668fae3551SRodney W. Grimes 		etapes++;
4678fae3551SRodney W. Grimes 		/* count the dumped inodes map on each additional tape */
4688fae3551SRodney W. Grimes 		tapesize += (etapes - 1) *
4698fae3551SRodney W. Grimes 			(howmany(mapsize * sizeof(char), TP_BSIZE) + 1);
4708fae3551SRodney W. Grimes 		tapesize += etapes + 10;	/* headers + 10 trailer blks */
4718fae3551SRodney W. Grimes 		msg("estimated %ld tape blocks on %3.2f tape(s).\n",
4728fae3551SRodney W. Grimes 		    tapesize, fetapes);
4738fae3551SRodney W. Grimes 	}
4748fae3551SRodney W. Grimes 
4758fae3551SRodney W. Grimes         /*
4769ac0768dSIan Dowse          * If the user only wants an estimate of the number of
4779ac0768dSIan Dowse          * tapes, exit now.
4789ac0768dSIan Dowse          */
4799ac0768dSIan Dowse         if (just_estimate)
4809ac0768dSIan Dowse                 exit(0);
4819ac0768dSIan Dowse 
4829ac0768dSIan Dowse 	/*
4838fae3551SRodney W. Grimes 	 * Allocate tape buffer.
4848fae3551SRodney W. Grimes 	 */
4858fae3551SRodney W. Grimes 	if (!alloctape())
4867580ffbbSPhilippe Charnier 		quit(
4877580ffbbSPhilippe Charnier 	"can't allocate tape buffers - try a smaller blocking factor.\n");
4888fae3551SRodney W. Grimes 
4898fae3551SRodney W. Grimes 	startnewtape(1);
4908fae3551SRodney W. Grimes 	(void)time((time_t *)&(tstart_writing));
4918fae3551SRodney W. Grimes 	dumpmap(usedinomap, TS_CLRI, maxino - 1);
4928fae3551SRodney W. Grimes 
4932bb823d2SIan Dowse 	passno = 3;
4942bb823d2SIan Dowse 	setproctitle("%s: pass 3: directories", disk);
4958fae3551SRodney W. Grimes 	msg("dumping (Pass III) [directories]\n");
4968fae3551SRodney W. Grimes 	dirty = 0;		/* XXX just to get gcc to shut up */
4978fae3551SRodney W. Grimes 	for (map = dumpdirmap, ino = 1; ino < maxino; ino++) {
49889fdc4e1SMike Barcroft 		if (((ino - 1) % CHAR_BIT) == 0)	/* map is offset by 1 */
4998fae3551SRodney W. Grimes 			dirty = *map++;
5008fae3551SRodney W. Grimes 		else
5018fae3551SRodney W. Grimes 			dirty >>= 1;
5028fae3551SRodney W. Grimes 		if ((dirty & 1) == 0)
5038fae3551SRodney W. Grimes 			continue;
5048fae3551SRodney W. Grimes 		/*
5058fae3551SRodney W. Grimes 		 * Skip directory inodes deleted and maybe reallocated
5068fae3551SRodney W. Grimes 		 */
5071c85e6a3SKirk McKusick 		dp = getino(ino, &mode);
5081c85e6a3SKirk McKusick 		if (mode != IFDIR)
5098fae3551SRodney W. Grimes 			continue;
5108fae3551SRodney W. Grimes 		(void)dumpino(dp, ino);
5118fae3551SRodney W. Grimes 	}
5128fae3551SRodney W. Grimes 
5132bb823d2SIan Dowse 	passno = 4;
5142bb823d2SIan Dowse 	setproctitle("%s: pass 4: regular files", disk);
5158fae3551SRodney W. Grimes 	msg("dumping (Pass IV) [regular files]\n");
5168fae3551SRodney W. Grimes 	for (map = dumpinomap, ino = 1; ino < maxino; ino++) {
51789fdc4e1SMike Barcroft 		if (((ino - 1) % CHAR_BIT) == 0)	/* map is offset by 1 */
5188fae3551SRodney W. Grimes 			dirty = *map++;
5198fae3551SRodney W. Grimes 		else
5208fae3551SRodney W. Grimes 			dirty >>= 1;
5218fae3551SRodney W. Grimes 		if ((dirty & 1) == 0)
5228fae3551SRodney W. Grimes 			continue;
5238fae3551SRodney W. Grimes 		/*
5248fae3551SRodney W. Grimes 		 * Skip inodes deleted and reallocated as directories.
5258fae3551SRodney W. Grimes 		 */
5261c85e6a3SKirk McKusick 		dp = getino(ino, &mode);
5278fae3551SRodney W. Grimes 		if (mode == IFDIR)
5288fae3551SRodney W. Grimes 			continue;
5298fae3551SRodney W. Grimes 		(void)dumpino(dp, ino);
5308fae3551SRodney W. Grimes 	}
5318fae3551SRodney W. Grimes 
532019420a5SJoerg Wunsch 	(void)time((time_t *)&(tend_writing));
5338fae3551SRodney W. Grimes 	spcl.c_type = TS_END;
5348fae3551SRodney W. Grimes 	for (i = 0; i < ntrec; i++)
5358fae3551SRodney W. Grimes 		writeheader(maxino - 1);
5368fae3551SRodney W. Grimes 	if (pipeout)
537617dbd3cSIan Dowse 		msg("DUMP: %jd tape blocks\n", (intmax_t)spcl.c_tapea);
5388fae3551SRodney W. Grimes 	else
539617dbd3cSIan Dowse 		msg("DUMP: %jd tape blocks on %d volume%s\n",
540617dbd3cSIan Dowse 		    (intmax_t)spcl.c_tapea, spcl.c_volume,
5418878be20SJean-Marc Zucconi 		    (spcl.c_volume == 1) ? "" : "s");
542019420a5SJoerg Wunsch 
543019420a5SJoerg Wunsch 	/* report dump performance, avoid division through zero */
544019420a5SJoerg Wunsch 	if (tend_writing - tstart_writing == 0)
545019420a5SJoerg Wunsch 		msg("finished in less than a second\n");
546019420a5SJoerg Wunsch 	else
5478a0453d6SJohan Karlsson 		msg("finished in %jd seconds, throughput %jd KBytes/sec\n",
5488a0453d6SJohan Karlsson 		    (intmax_t)tend_writing - tstart_writing,
5498a0453d6SJohan Karlsson 		    (intmax_t)(spcl.c_tapea /
550617dbd3cSIan Dowse 		    (tend_writing - tstart_writing)));
551019420a5SJoerg Wunsch 
5528fae3551SRodney W. Grimes 	putdumptime();
5538fae3551SRodney W. Grimes 	trewind();
554bfa0b298SPoul-Henning Kamp 	broadcast("DUMP IS DONE!\a\a\n");
5558fae3551SRodney W. Grimes 	msg("DUMP IS DONE\n");
5568fae3551SRodney W. Grimes 	Exit(X_FINOK);
5578fae3551SRodney W. Grimes 	/* NOTREACHED */
5588fae3551SRodney W. Grimes }
5598fae3551SRodney W. Grimes 
560a37c38b8SPeter Wemm static void
5612db673abSWarner Losh usage(void)
562a37c38b8SPeter Wemm {
563a1e5f53bSPhilippe Charnier 	fprintf(stderr,
56444b81f0fSRuslan Ermilov 		"usage: dump [-0123456789acLnSu] [-B records] [-b blocksize] [-C cachesize]\n"
5651f6a4631SRuslan Ermilov 		"            [-D dumpdates] [-d density] [-f file] [-h level] [-s feet]\n"
5661f6a4631SRuslan Ermilov 		"            [-T date] filesystem\n"
5674fb133adSBruce Evans 		"       dump -W | -w\n");
568f69e804dSJoseph Koshy 	exit(X_STARTUP);
569a37c38b8SPeter Wemm }
570a37c38b8SPeter Wemm 
5718fae3551SRodney W. Grimes /*
572194a6667SKirk McKusick  * Check to see if a disk is currently mounted.
573194a6667SKirk McKusick  */
574194a6667SKirk McKusick static char *
575194a6667SKirk McKusick getmntpt(char *name, int *mntflagsp)
576194a6667SKirk McKusick {
577194a6667SKirk McKusick 	long mntsize, i;
578194a6667SKirk McKusick 	struct statfs *mntbuf;
579194a6667SKirk McKusick 
580194a6667SKirk McKusick 	mntsize = getmntinfo(&mntbuf, MNT_NOWAIT);
581194a6667SKirk McKusick 	for (i = 0; i < mntsize; i++) {
582194a6667SKirk McKusick 		if (!strcmp(mntbuf[i].f_mntfromname, name)) {
583194a6667SKirk McKusick 			*mntflagsp = mntbuf[i].f_flags;
584194a6667SKirk McKusick 			return (mntbuf[i].f_mntonname);
585194a6667SKirk McKusick 		}
586194a6667SKirk McKusick 	}
587194a6667SKirk McKusick 	return (0);
588194a6667SKirk McKusick }
589194a6667SKirk McKusick 
590194a6667SKirk McKusick /*
5918fae3551SRodney W. Grimes  * Pick up a numeric argument.  It must be nonnegative and in the given
5928fae3551SRodney W. Grimes  * range (except that a vmax of 0 means unlimited).
5938fae3551SRodney W. Grimes  */
5948fae3551SRodney W. Grimes static long
5952db673abSWarner Losh numarg(const char *meaning, long vmin, long vmax)
5968fae3551SRodney W. Grimes {
597a37c38b8SPeter Wemm 	char *p;
5988fae3551SRodney W. Grimes 	long val;
5998fae3551SRodney W. Grimes 
600a37c38b8SPeter Wemm 	val = strtol(optarg, &p, 10);
601a37c38b8SPeter Wemm 	if (*p)
602a37c38b8SPeter Wemm 		errx(1, "illegal %s -- %s", meaning, optarg);
6038fae3551SRodney W. Grimes 	if (val < vmin || (vmax && val > vmax))
604a37c38b8SPeter Wemm 		errx(1, "%s must be between %ld and %ld", meaning, vmin, vmax);
6058fae3551SRodney W. Grimes 	return (val);
6068fae3551SRodney W. Grimes }
6078fae3551SRodney W. Grimes 
6088fae3551SRodney W. Grimes void
6092db673abSWarner Losh sig(int signo)
6108fae3551SRodney W. Grimes {
6118fae3551SRodney W. Grimes 	switch(signo) {
6128fae3551SRodney W. Grimes 	case SIGALRM:
6138fae3551SRodney W. Grimes 	case SIGBUS:
6148fae3551SRodney W. Grimes 	case SIGFPE:
6158fae3551SRodney W. Grimes 	case SIGHUP:
6168fae3551SRodney W. Grimes 	case SIGTERM:
6178fae3551SRodney W. Grimes 	case SIGTRAP:
6188fae3551SRodney W. Grimes 		if (pipeout)
6198fae3551SRodney W. Grimes 			quit("Signal on pipe: cannot recover\n");
6208fae3551SRodney W. Grimes 		msg("Rewriting attempted as response to unknown signal.\n");
6218fae3551SRodney W. Grimes 		(void)fflush(stderr);
6228fae3551SRodney W. Grimes 		(void)fflush(stdout);
6238fae3551SRodney W. Grimes 		close_rewind();
6248fae3551SRodney W. Grimes 		exit(X_REWRITE);
6258fae3551SRodney W. Grimes 		/* NOTREACHED */
6268fae3551SRodney W. Grimes 	case SIGSEGV:
6278fae3551SRodney W. Grimes 		msg("SIGSEGV: ABORTING!\n");
6288fae3551SRodney W. Grimes 		(void)signal(SIGSEGV, SIG_DFL);
6298fae3551SRodney W. Grimes 		(void)kill(0, SIGSEGV);
6308fae3551SRodney W. Grimes 		/* NOTREACHED */
6318fae3551SRodney W. Grimes 	}
6328fae3551SRodney W. Grimes }
6338fae3551SRodney W. Grimes 
6348fae3551SRodney W. Grimes char *
6352db673abSWarner Losh rawname(char *cp)
6368fae3551SRodney W. Grimes {
6378fae3551SRodney W. Grimes 	static char rawbuf[MAXPATHLEN];
63804285c06SJoerg Wunsch 	char *dp;
63904285c06SJoerg Wunsch 	struct stat sb;
64004285c06SJoerg Wunsch 
64104285c06SJoerg Wunsch 	if (stat(cp, &sb) == 0) {
64204285c06SJoerg Wunsch 		/*
64304285c06SJoerg Wunsch 		 * If the name already refers to a raw device, return
64404285c06SJoerg Wunsch 		 * it immediately without tampering.
64504285c06SJoerg Wunsch 		 */
64604285c06SJoerg Wunsch 		if ((sb.st_mode & S_IFMT) == S_IFCHR)
64704285c06SJoerg Wunsch 			return (cp);
64804285c06SJoerg Wunsch 	}
64904285c06SJoerg Wunsch 
65004285c06SJoerg Wunsch 	dp = strrchr(cp, '/');
6518fae3551SRodney W. Grimes 
6528fae3551SRodney W. Grimes 	if (dp == NULL)
6538fae3551SRodney W. Grimes 		return (NULL);
6548fae3551SRodney W. Grimes 	*dp = '\0';
6552db673abSWarner Losh 	(void)strlcpy(rawbuf, cp, MAXPATHLEN - 1);
6568fae3551SRodney W. Grimes 	*dp = '/';
6572db673abSWarner Losh 	(void)strlcat(rawbuf, "/r", MAXPATHLEN - 1 - strlen(rawbuf));
6582db673abSWarner Losh 	(void)strlcat(rawbuf, dp + 1, MAXPATHLEN - 1 - strlen(rawbuf));
6598fae3551SRodney W. Grimes 	return (rawbuf);
6608fae3551SRodney W. Grimes }
6618fae3551SRodney W. Grimes 
662a37c38b8SPeter Wemm /*
663a37c38b8SPeter Wemm  * obsolete --
664a37c38b8SPeter Wemm  *	Change set of key letters and ordered arguments into something
665a37c38b8SPeter Wemm  *	getopt(3) will like.
666a37c38b8SPeter Wemm  */
667a37c38b8SPeter Wemm static void
6682db673abSWarner Losh obsolete(int *argcp, char **argvp[])
6698fae3551SRodney W. Grimes {
670a37c38b8SPeter Wemm 	int argc, flags;
671a37c38b8SPeter Wemm 	char *ap, **argv, *flagsp, **nargv, *p;
6728fae3551SRodney W. Grimes 
673a37c38b8SPeter Wemm 	/* Setup. */
674a37c38b8SPeter Wemm 	argv = *argvp;
675a37c38b8SPeter Wemm 	argc = *argcp;
676a37c38b8SPeter Wemm 
677a37c38b8SPeter Wemm 	/* Return if no arguments or first argument has leading dash. */
678a37c38b8SPeter Wemm 	ap = argv[1];
679a37c38b8SPeter Wemm 	if (argc == 1 || *ap == '-')
680a37c38b8SPeter Wemm 		return;
681a37c38b8SPeter Wemm 
682a37c38b8SPeter Wemm 	/* Allocate space for new arguments. */
683a37c38b8SPeter Wemm 	if ((*argvp = nargv = malloc((argc + 1) * sizeof(char *))) == NULL ||
684a37c38b8SPeter Wemm 	    (p = flagsp = malloc(strlen(ap) + 2)) == NULL)
685a37c38b8SPeter Wemm 		err(1, NULL);
686a37c38b8SPeter Wemm 
687a37c38b8SPeter Wemm 	*nargv++ = *argv;
688a37c38b8SPeter Wemm 	argv += 2;
689a37c38b8SPeter Wemm 
690a37c38b8SPeter Wemm 	for (flags = 0; *ap; ++ap) {
691a37c38b8SPeter Wemm 		switch (*ap) {
692a37c38b8SPeter Wemm 		case 'B':
693a37c38b8SPeter Wemm 		case 'b':
694a37c38b8SPeter Wemm 		case 'd':
695a37c38b8SPeter Wemm 		case 'f':
696cedae1f4SMatthew Dillon 		case 'D':
697ea8123f9SMatthew Dillon 		case 'C':
698a37c38b8SPeter Wemm 		case 'h':
699a37c38b8SPeter Wemm 		case 's':
700a37c38b8SPeter Wemm 		case 'T':
701a37c38b8SPeter Wemm 			if (*argv == NULL) {
702a37c38b8SPeter Wemm 				warnx("option requires an argument -- %c", *ap);
703a37c38b8SPeter Wemm 				usage();
7048fae3551SRodney W. Grimes 			}
705a37c38b8SPeter Wemm 			if ((nargv[0] = malloc(strlen(*argv) + 2 + 1)) == NULL)
706a37c38b8SPeter Wemm 				err(1, NULL);
707a37c38b8SPeter Wemm 			nargv[0][0] = '-';
708a37c38b8SPeter Wemm 			nargv[0][1] = *ap;
709a37c38b8SPeter Wemm 			(void)strcpy(&nargv[0][2], *argv);
710a37c38b8SPeter Wemm 			++argv;
711a37c38b8SPeter Wemm 			++nargv;
712a37c38b8SPeter Wemm 			break;
713a37c38b8SPeter Wemm 		default:
714a37c38b8SPeter Wemm 			if (!flags) {
715a37c38b8SPeter Wemm 				*p++ = '-';
716a37c38b8SPeter Wemm 				flags = 1;
717a37c38b8SPeter Wemm 			}
718a37c38b8SPeter Wemm 			*p++ = *ap;
719a37c38b8SPeter Wemm 			break;
720a37c38b8SPeter Wemm 		}
721a37c38b8SPeter Wemm 	}
722a37c38b8SPeter Wemm 
723a37c38b8SPeter Wemm 	/* Terminate flags. */
724a37c38b8SPeter Wemm 	if (flags) {
725a37c38b8SPeter Wemm 		*p = '\0';
726a37c38b8SPeter Wemm 		*nargv++ = flagsp;
727a37c38b8SPeter Wemm 	}
728a37c38b8SPeter Wemm 
729a37c38b8SPeter Wemm 	/* Copy remaining arguments. */
7307580ffbbSPhilippe Charnier 	while ((*nargv++ = *argv++));
731a37c38b8SPeter Wemm 
732a37c38b8SPeter Wemm 	/* Update argument count. */
733a37c38b8SPeter Wemm 	*argcp = nargv - *argvp - 1;
734a37c38b8SPeter Wemm }
735