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