Home | History | Annotate | Line # | Download | only in restore
main.c revision 1.35.30.1
      1  1.35.30.1    martin /*	$NetBSD: main.c,v 1.35.30.1 2020/04/08 14:07:20 martin Exp $	*/
      2        1.9       cgd 
      3        1.1       cgd /*
      4        1.4   mycroft  * Copyright (c) 1983, 1993
      5        1.4   mycroft  *	The Regents of the University of California.  All rights reserved.
      6        1.1       cgd  *
      7        1.1       cgd  * Redistribution and use in source and binary forms, with or without
      8        1.1       cgd  * modification, are permitted provided that the following conditions
      9        1.1       cgd  * are met:
     10        1.1       cgd  * 1. Redistributions of source code must retain the above copyright
     11        1.1       cgd  *    notice, this list of conditions and the following disclaimer.
     12        1.1       cgd  * 2. Redistributions in binary form must reproduce the above copyright
     13        1.1       cgd  *    notice, this list of conditions and the following disclaimer in the
     14        1.1       cgd  *    documentation and/or other materials provided with the distribution.
     15       1.23       agc  * 3. Neither the name of the University nor the names of its contributors
     16        1.1       cgd  *    may be used to endorse or promote products derived from this software
     17        1.1       cgd  *    without specific prior written permission.
     18        1.1       cgd  *
     19        1.1       cgd  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
     20        1.1       cgd  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     21        1.1       cgd  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     22        1.1       cgd  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
     23        1.1       cgd  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     24        1.1       cgd  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
     25        1.1       cgd  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     26        1.1       cgd  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     27        1.1       cgd  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     28        1.1       cgd  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     29        1.1       cgd  * SUCH DAMAGE.
     30        1.1       cgd  */
     31        1.1       cgd 
     32       1.15     lukem #include <sys/cdefs.h>
     33        1.1       cgd #ifndef lint
     34       1.33     lukem __COPYRIGHT("@(#) Copyright (c) 1983, 1993\
     35       1.33     lukem  The Regents of the University of California.  All rights reserved.");
     36        1.1       cgd #endif /* not lint */
     37        1.1       cgd 
     38        1.1       cgd #ifndef lint
     39        1.9       cgd #if 0
     40       1.16     lukem static char sccsid[] = "@(#)main.c	8.6 (Berkeley) 5/4/95";
     41        1.9       cgd #else
     42  1.35.30.1    martin __RCSID("$NetBSD: main.c,v 1.35.30.1 2020/04/08 14:07:20 martin Exp $");
     43        1.9       cgd #endif
     44        1.1       cgd #endif /* not lint */
     45        1.1       cgd 
     46        1.3       cgd #include <sys/param.h>
     47        1.3       cgd #include <sys/time.h>
     48        1.1       cgd 
     49       1.16     lukem #include <ufs/ufs/dinode.h>
     50        1.4   mycroft #include <ufs/ffs/fs.h>
     51        1.1       cgd #include <protocols/dumprestore.h>
     52        1.3       cgd 
     53        1.4   mycroft #include <err.h>
     54        1.3       cgd #include <errno.h>
     55       1.12     lukem #include <paths.h>
     56        1.3       cgd #include <signal.h>
     57        1.3       cgd #include <stdio.h>
     58        1.3       cgd #include <stdlib.h>
     59        1.3       cgd #include <string.h>
     60       1.16     lukem #include <unistd.h>
     61        1.3       cgd 
     62        1.3       cgd #include "restore.h"
     63        1.3       cgd #include "extern.h"
     64        1.1       cgd 
     65  1.35.30.1    martin struct context curfile;
     66  1.35.30.1    martin union u_spcl u_spcl;
     67  1.35.30.1    martin int	uflag;
     68        1.1       cgd int	bflag = 0, cvtflag = 0, dflag = 0, vflag = 0, yflag = 0;
     69       1.24     enami int	hflag = 1, mflag = 1, Dflag = 0, Nflag = 0;
     70        1.1       cgd char	command = '\0';
     71       1.15     lukem int32_t	dumpnum = 1;
     72       1.15     lukem int32_t	volno = 0;
     73       1.15     lukem int32_t	ntrec;
     74        1.1       cgd char	*dumpmap;
     75        1.5   mycroft char	*usedinomap;
     76        1.1       cgd ino_t	maxino;
     77        1.1       cgd time_t	dumptime;
     78        1.1       cgd time_t	dumpdate;
     79       1.22     enami size_t	pagesize;
     80       1.25     enami FILE	*terminal;
     81       1.31  christos const char	*tmpdir;
     82       1.29     fredb int	dotflag = 0;
     83        1.1       cgd 
     84       1.27    bouyer FILE *Mtreefile = NULL;
     85       1.27    bouyer 
     86       1.30   xtraeme static	void obsolete(int *, char **[]);
     87       1.34     joerg __dead static	void usage(void);
     88        1.3       cgd 
     89        1.3       cgd int
     90       1.30   xtraeme main(int argc, char *argv[])
     91        1.1       cgd {
     92        1.3       cgd 	int ch;
     93        1.1       cgd 	ino_t ino;
     94       1.31  christos 	const char *inputdev;
     95       1.31  christos 	const char *symtbl = "./restoresymtable";
     96        1.3       cgd 	char *p, name[MAXPATHLEN];
     97       1.31  christos 	static char dot[] = ".";
     98       1.11    scottr 
     99        1.3       cgd 	if (argc < 2)
    100        1.3       cgd 		usage();
    101        1.1       cgd 
    102       1.10       mrg 	if ((inputdev = getenv("TAPE")) == NULL)
    103       1.10       mrg 		inputdev = _PATH_DEFTAPE;
    104       1.13     lukem 	if ((tmpdir = getenv("TMPDIR")) == NULL)
    105       1.13     lukem 		tmpdir = _PATH_TMP;
    106        1.3       cgd 	obsolete(&argc, &argv);
    107       1.27    bouyer 	while ((ch = getopt(argc, argv, "b:cD:df:himM:NRrs:tuvxy")) != -1)
    108       1.25     enami 		switch (ch) {
    109        1.3       cgd 		case 'b':
    110        1.3       cgd 			/* Change default tape blocksize. */
    111        1.3       cgd 			bflag = 1;
    112        1.3       cgd 			ntrec = strtol(optarg, &p, 10);
    113        1.3       cgd 			if (*p)
    114        1.4   mycroft 				errx(1, "illegal blocksize -- %s", optarg);
    115        1.3       cgd 			if (ntrec <= 0)
    116        1.4   mycroft 				errx(1, "block size must be greater than 0");
    117        1.1       cgd 			break;
    118        1.1       cgd 		case 'c':
    119        1.3       cgd 			cvtflag = 1;
    120        1.1       cgd 			break;
    121       1.24     enami 		case 'D':
    122       1.24     enami 			ddesc = digest_lookup(optarg);
    123       1.24     enami 			if (ddesc == NULL)
    124       1.24     enami 				err(1, "unknown digest algorithm: %s", optarg);
    125       1.24     enami 			Dflag = 1;
    126       1.24     enami 			break;
    127        1.1       cgd 		case 'd':
    128        1.3       cgd 			dflag = 1;
    129        1.3       cgd 			break;
    130        1.3       cgd 		case 'f':
    131        1.3       cgd 			inputdev = optarg;
    132        1.1       cgd 			break;
    133        1.1       cgd 		case 'h':
    134        1.1       cgd 			hflag = 0;
    135        1.1       cgd 			break;
    136        1.3       cgd 		case 'i':
    137        1.3       cgd 		case 'R':
    138        1.3       cgd 		case 'r':
    139        1.3       cgd 		case 't':
    140        1.3       cgd 		case 'x':
    141        1.3       cgd 			if (command != '\0')
    142        1.4   mycroft 				errx(1,
    143        1.4   mycroft 				    "%c and %c options are mutually exclusive",
    144        1.3       cgd 				    ch, command);
    145        1.3       cgd 			command = ch;
    146        1.3       cgd 			break;
    147        1.1       cgd 		case 'm':
    148        1.1       cgd 			mflag = 0;
    149        1.1       cgd 			break;
    150        1.1       cgd 		case 'N':
    151        1.3       cgd 			Nflag = 1;
    152        1.3       cgd 			break;
    153       1.27    bouyer 		case 'M':
    154       1.27    bouyer 			Mtreefile = fopen(optarg, "a");
    155       1.27    bouyer 			if (Mtreefile == NULL)
    156       1.27    bouyer 				err(1, "can't open %s", optarg);
    157       1.27    bouyer 			break;
    158        1.3       cgd 		case 's':
    159        1.3       cgd 			/* Dumpnum (skip to) for multifile dump tapes. */
    160        1.3       cgd 			dumpnum = strtol(optarg, &p, 10);
    161        1.3       cgd 			if (*p)
    162        1.4   mycroft 				errx(1, "illegal dump number -- %s", optarg);
    163        1.3       cgd 			if (dumpnum <= 0)
    164        1.4   mycroft 				errx(1, "dump number must be greater than 0");
    165       1.18  christos 			break;
    166       1.18  christos 		case 'u':
    167       1.18  christos 			uflag = 1;
    168        1.1       cgd 			break;
    169        1.1       cgd 		case 'v':
    170        1.3       cgd 			vflag = 1;
    171        1.1       cgd 			break;
    172        1.1       cgd 		case 'y':
    173        1.3       cgd 			yflag = 1;
    174        1.1       cgd 			break;
    175        1.1       cgd 		default:
    176        1.3       cgd 			usage();
    177        1.1       cgd 		}
    178        1.3       cgd 	argc -= optind;
    179        1.3       cgd 	argv += optind;
    180        1.3       cgd 
    181        1.3       cgd 	if (command == '\0')
    182        1.4   mycroft 		errx(1, "none of i, R, r, t or x options specified");
    183       1.20     enami 
    184       1.20     enami 	if (Nflag || command == 't')
    185       1.20     enami 		uflag = 0;
    186        1.3       cgd 
    187        1.3       cgd 	if (signal(SIGINT, onintr) == SIG_IGN)
    188        1.3       cgd 		(void) signal(SIGINT, SIG_IGN);
    189        1.3       cgd 	if (signal(SIGTERM, onintr) == SIG_IGN)
    190        1.3       cgd 		(void) signal(SIGTERM, SIG_IGN);
    191        1.3       cgd 	setlinebuf(stderr);
    192       1.22     enami 	pagesize = sysconf(_SC_PAGESIZE);
    193        1.3       cgd 
    194        1.7   mycroft 	atexit(cleanup);
    195        1.7   mycroft 
    196        1.1       cgd 	setinput(inputdev);
    197        1.3       cgd 
    198        1.1       cgd 	if (argc == 0) {
    199        1.1       cgd 		argc = 1;
    200       1.31  christos 		*--argv = dot;
    201        1.1       cgd 	}
    202        1.3       cgd 
    203        1.1       cgd 	switch (command) {
    204        1.1       cgd 	/*
    205        1.1       cgd 	 * Interactive mode.
    206        1.1       cgd 	 */
    207        1.1       cgd 	case 'i':
    208        1.1       cgd 		setup();
    209        1.1       cgd 		extractdirs(1);
    210        1.3       cgd 		initsymtable(NULL);
    211        1.1       cgd 		runcmdshell();
    212        1.3       cgd 		break;
    213        1.1       cgd 	/*
    214        1.1       cgd 	 * Incremental restoration of a file system.
    215        1.1       cgd 	 */
    216        1.1       cgd 	case 'r':
    217        1.1       cgd 		setup();
    218        1.1       cgd 		if (dumptime > 0) {
    219        1.1       cgd 			/*
    220        1.1       cgd 			 * This is an incremental dump tape.
    221        1.1       cgd 			 */
    222        1.1       cgd 			vprintf(stdout, "Begin incremental restore\n");
    223        1.1       cgd 			initsymtable(symtbl);
    224        1.1       cgd 			extractdirs(1);
    225        1.1       cgd 			removeoldleaves();
    226        1.1       cgd 			vprintf(stdout, "Calculate node updates.\n");
    227       1.35  dholland 			treescan(".", UFS_ROOTINO, nodeupdates);
    228        1.1       cgd 			findunreflinks();
    229        1.1       cgd 			removeoldnodes();
    230        1.1       cgd 		} else {
    231        1.1       cgd 			/*
    232        1.1       cgd 			 * This is a level zero dump tape.
    233        1.1       cgd 			 */
    234        1.1       cgd 			vprintf(stdout, "Begin level 0 restore\n");
    235        1.1       cgd 			initsymtable((char *)0);
    236        1.1       cgd 			extractdirs(1);
    237        1.1       cgd 			vprintf(stdout, "Calculate extraction list.\n");
    238       1.35  dholland 			treescan(".", UFS_ROOTINO, nodeupdates);
    239        1.1       cgd 		}
    240        1.1       cgd 		createleaves(symtbl);
    241        1.1       cgd 		createlinks();
    242        1.3       cgd 		setdirmodes(FORCE);
    243        1.1       cgd 		checkrestore();
    244        1.1       cgd 		if (dflag) {
    245        1.1       cgd 			vprintf(stdout, "Verify the directory structure\n");
    246       1.35  dholland 			treescan(".", UFS_ROOTINO, verifyfile);
    247        1.1       cgd 		}
    248        1.1       cgd 		dumpsymtable(symtbl, (long)1);
    249        1.3       cgd 		break;
    250        1.1       cgd 	/*
    251        1.1       cgd 	 * Resume an incremental file system restoration.
    252        1.1       cgd 	 */
    253        1.1       cgd 	case 'R':
    254        1.1       cgd 		initsymtable(symtbl);
    255        1.1       cgd 		skipmaps();
    256        1.1       cgd 		skipdirs();
    257        1.1       cgd 		createleaves(symtbl);
    258        1.1       cgd 		createlinks();
    259        1.3       cgd 		setdirmodes(FORCE);
    260        1.1       cgd 		checkrestore();
    261        1.1       cgd 		dumpsymtable(symtbl, (long)1);
    262        1.3       cgd 		break;
    263        1.1       cgd 	/*
    264        1.1       cgd 	 * List contents of tape.
    265        1.1       cgd 	 */
    266        1.1       cgd 	case 't':
    267        1.1       cgd 		setup();
    268        1.1       cgd 		extractdirs(0);
    269        1.1       cgd 		initsymtable((char *)0);
    270        1.1       cgd 		while (argc--) {
    271        1.1       cgd 			canon(*argv++, name);
    272        1.1       cgd 			ino = dirlookup(name);
    273        1.1       cgd 			if (ino == 0)
    274        1.1       cgd 				continue;
    275        1.1       cgd 			treescan(name, ino, listfile);
    276        1.1       cgd 		}
    277        1.3       cgd 		break;
    278        1.1       cgd 	/*
    279        1.1       cgd 	 * Batch extraction of tape contents.
    280        1.1       cgd 	 */
    281        1.1       cgd 	case 'x':
    282        1.1       cgd 		setup();
    283        1.1       cgd 		extractdirs(1);
    284        1.1       cgd 		initsymtable((char *)0);
    285        1.1       cgd 		while (argc--) {
    286        1.1       cgd 			canon(*argv++, name);
    287        1.1       cgd 			ino = dirlookup(name);
    288        1.1       cgd 			if (ino == 0)
    289        1.1       cgd 				continue;
    290       1.35  dholland 			if (ino == UFS_ROOTINO)
    291       1.29     fredb 				dotflag = 1;
    292        1.1       cgd 			if (mflag)
    293        1.1       cgd 				pathcheck(name);
    294        1.1       cgd 			treescan(name, ino, addfile);
    295        1.1       cgd 		}
    296        1.1       cgd 		createfiles();
    297        1.1       cgd 		createlinks();
    298        1.3       cgd 		setdirmodes(0);
    299        1.1       cgd 		if (dflag)
    300        1.1       cgd 			checkrestore();
    301        1.3       cgd 		break;
    302        1.1       cgd 	}
    303        1.7   mycroft 	exit(0);
    304        1.3       cgd 	/* NOTREACHED */
    305        1.1       cgd }
    306        1.3       cgd 
    307        1.4   mycroft static void
    308       1.30   xtraeme usage(void)
    309        1.3       cgd {
    310       1.21       cgd 	const char *progname = getprogname();
    311        1.8   mycroft 
    312       1.14     lukem 	(void)fprintf(stderr,
    313       1.26       wiz 	    "usage: %s -i [-cdhmvyN] [-b bsize] [-D algorithm] "
    314       1.28       wiz 	    "[-f file] [-M mtreefile] [-s fileno]\n", progname);
    315       1.14     lukem 	(void)fprintf(stderr,
    316       1.26       wiz 	    "       %s -R [-cdvyN] [-b bsize] [-D algorithm] [-f file] "
    317       1.28       wiz 	    "[-M mtreefile] [-s fileno]\n", progname);
    318       1.14     lukem 	(void)fprintf(stderr,
    319       1.26       wiz 	    "       %s -r [-cdvyN] [-b bsize] [-D algorithm] [-f file] "
    320       1.28       wiz 	    "[-M mtreefile] [-s fileno]\n", progname);
    321       1.14     lukem 	(void)fprintf(stderr,
    322       1.26       wiz 	    "       %s -t [-cdhvy] [-b bsize] [-D algorithm] [-f file]\n"
    323       1.26       wiz 	    "           [-s fileno] [file ...]\n", progname);
    324       1.14     lukem 	(void)fprintf(stderr,
    325       1.26       wiz 	    "       %s -x [-cdhmvyN] [-b bsize] [-D algorithm] [-f file]\n"
    326       1.28       wiz 	    "           [-M mtreefile] [-s fileno] [file ...]\n", progname);
    327        1.7   mycroft 	exit(1);
    328        1.3       cgd }
    329        1.3       cgd 
    330        1.3       cgd /*
    331        1.3       cgd  * obsolete --
    332        1.3       cgd  *	Change set of key letters and ordered arguments into something
    333        1.3       cgd  *	getopt(3) will like.
    334        1.3       cgd  */
    335        1.3       cgd static void
    336       1.30   xtraeme obsolete(int *argcp, char **argvp[])
    337        1.3       cgd {
    338        1.3       cgd 	int argc, flags;
    339        1.3       cgd 	char *ap, **argv, *flagsp, **nargv, *p;
    340        1.3       cgd 
    341        1.3       cgd 	/* Setup. */
    342        1.3       cgd 	argv = *argvp;
    343        1.3       cgd 	argc = *argcp;
    344        1.3       cgd 
    345        1.3       cgd 	/* Return if no arguments or first argument has leading dash. */
    346        1.3       cgd 	ap = argv[1];
    347        1.3       cgd 	if (argc == 1 || *ap == '-')
    348        1.3       cgd 		return;
    349        1.3       cgd 
    350        1.3       cgd 	/* Allocate space for new arguments. */
    351        1.3       cgd 	if ((*argvp = nargv = malloc((argc + 1) * sizeof(char *))) == NULL ||
    352        1.3       cgd 	    (p = flagsp = malloc(strlen(ap) + 2)) == NULL)
    353       1.19  drochner 		err(1, NULL);
    354        1.3       cgd 
    355        1.3       cgd 	*nargv++ = *argv;
    356        1.3       cgd 	argv += 2;
    357        1.3       cgd 
    358        1.3       cgd 	for (flags = 0; *ap; ++ap) {
    359        1.6   mycroft 		switch (*ap) {
    360        1.3       cgd 		case 'b':
    361        1.3       cgd 		case 'f':
    362        1.3       cgd 		case 's':
    363        1.6   mycroft 			if (*argv == NULL) {
    364        1.6   mycroft 				warnx("option requires an argument -- %c", *ap);
    365        1.6   mycroft 				usage();
    366        1.6   mycroft 			}
    367        1.3       cgd 			if ((nargv[0] = malloc(strlen(*argv) + 2 + 1)) == NULL)
    368       1.19  drochner 				err(1, NULL);
    369        1.3       cgd 			nargv[0][0] = '-';
    370        1.3       cgd 			nargv[0][1] = *ap;
    371        1.3       cgd 			(void)strcpy(&nargv[0][2], *argv);
    372        1.6   mycroft 			++argv;
    373        1.3       cgd 			++nargv;
    374        1.3       cgd 			break;
    375        1.3       cgd 		default:
    376        1.3       cgd 			if (!flags) {
    377        1.3       cgd 				*p++ = '-';
    378        1.3       cgd 				flags = 1;
    379        1.3       cgd 			}
    380        1.3       cgd 			*p++ = *ap;
    381        1.3       cgd 			break;
    382        1.3       cgd 		}
    383        1.3       cgd 	}
    384        1.3       cgd 
    385        1.3       cgd 	/* Terminate flags. */
    386        1.3       cgd 	if (flags) {
    387        1.3       cgd 		*p = '\0';
    388        1.3       cgd 		*nargv++ = flagsp;
    389       1.32  christos 	} else {
    390       1.32  christos 		free(flagsp);
    391        1.3       cgd 	}
    392        1.3       cgd 
    393        1.3       cgd 	/* Copy remaining arguments. */
    394       1.14     lukem 	while ((*nargv++ = *argv++) != NULL)
    395       1.14     lukem 		;
    396        1.6   mycroft 
    397        1.6   mycroft 	/* Update argument count. */
    398        1.6   mycroft 	*argcp = nargv - *argvp - 1;
    399        1.3       cgd }
    400