1 /*-
2 * SPDX-License-Identifier: BSD-3-Clause
3 *
4 * Copyright (c) 1983, 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 #ifndef lint
33 static const char copyright[] =
34 "@(#) Copyright (c) 1983, 1993\n\
35 The Regents of the University of California. All rights reserved.\n";
36 #endif /* not lint */
37
38 #ifndef lint
39 #if 0
40 static char sccsid[] = "@(#)main.c 8.6 (Berkeley) 5/4/95";
41 #endif
42 #endif /* not lint */
43
44 #include <sys/cdefs.h>
45 __FBSDID("$FreeBSD$");
46
47 #include <sys/param.h>
48 #include <sys/stat.h>
49
50 #include <ufs/ufs/dinode.h>
51 #include <protocols/dumprestore.h>
52
53 #include <err.h>
54 #include <limits.h>
55 #include <locale.h>
56 #include <paths.h>
57 #include <stdio.h>
58 #include <stdlib.h>
59 #include <string.h>
60 #include <unistd.h>
61
62 #include "restore.h"
63 #include "extern.h"
64
65 int bflag = 0, cvtflag = 0, dflag = 0, Dflag = 0, vflag = 0, yflag = 0;
66 int hflag = 1, mflag = 1, Nflag = 0;
67 int uflag = 0;
68 int pipecmd = 0;
69 char command = '\0';
70 long dumpnum = 1;
71 long volno = 0;
72 long ntrec;
73 char *dumpmap;
74 char *usedinomap;
75 ino_t maxino;
76 time_t dumptime;
77 time_t dumpdate;
78 FILE *terminal;
79
80 static void obsolete(int *, char **[]);
81 static void usage(void) __dead2;
82
83 int
main(int argc,char * argv[])84 main(int argc, char *argv[])
85 {
86 int ch;
87 ino_t ino;
88 char *inputdev;
89 char *symtbl = "./restoresymtable";
90 char *p, name[MAXPATHLEN];
91
92 /* Temp files should *not* be readable. We set permissions later. */
93 (void) umask(077);
94
95 if (argc < 2)
96 usage();
97
98 (void)setlocale(LC_ALL, "");
99
100 inputdev = NULL;
101 obsolete(&argc, &argv);
102 while ((ch = getopt(argc, argv, "b:dDf:himNP:Rrs:tuvxy")) != -1)
103 switch(ch) {
104 case 'b':
105 /* Change default tape blocksize. */
106 bflag = 1;
107 ntrec = strtol(optarg, &p, 10);
108 if (*p)
109 errx(1, "illegal blocksize -- %s", optarg);
110 if (ntrec <= 0)
111 errx(1, "block size must be greater than 0");
112 break;
113 case 'd':
114 dflag = 1;
115 break;
116 case 'D':
117 Dflag = 1;
118 break;
119 case 'f':
120 if (pipecmd)
121 errx(1,
122 "-P and -f options are mutually exclusive");
123 inputdev = optarg;
124 break;
125 case 'P':
126 if (!pipecmd && inputdev)
127 errx(1,
128 "-P and -f options are mutually exclusive");
129 inputdev = optarg;
130 pipecmd = 1;
131 break;
132 case 'h':
133 hflag = 0;
134 break;
135 case 'i':
136 case 'R':
137 case 'r':
138 case 't':
139 case 'x':
140 if (command != '\0')
141 errx(1,
142 "%c and %c options are mutually exclusive",
143 ch, command);
144 command = ch;
145 break;
146 case 'm':
147 mflag = 0;
148 break;
149 case 'N':
150 Nflag = 1;
151 break;
152 case 's':
153 /* Dumpnum (skip to) for multifile dump tapes. */
154 dumpnum = strtol(optarg, &p, 10);
155 if (*p)
156 errx(1, "illegal dump number -- %s", optarg);
157 if (dumpnum <= 0)
158 errx(1, "dump number must be greater than 0");
159 break;
160 case 'u':
161 uflag = 1;
162 break;
163 case 'v':
164 vflag = 1;
165 break;
166 case 'y':
167 yflag = 1;
168 break;
169 default:
170 usage();
171 }
172 argc -= optind;
173 argv += optind;
174
175 if (command == '\0')
176 errx(1, "none of i, R, r, t or x options specified");
177
178 if (signal(SIGINT, onintr) == SIG_IGN)
179 (void) signal(SIGINT, SIG_IGN);
180 if (signal(SIGTERM, onintr) == SIG_IGN)
181 (void) signal(SIGTERM, SIG_IGN);
182 setlinebuf(stderr);
183
184 if (inputdev == NULL && (inputdev = getenv("TAPE")) == NULL)
185 inputdev = _PATH_DEFTAPE;
186 setinput(inputdev, pipecmd);
187
188 if (argc == 0) {
189 argc = 1;
190 *--argv = ".";
191 }
192
193 switch (command) {
194 /*
195 * Interactive mode.
196 */
197 case 'i':
198 setup();
199 extractdirs(1);
200 initsymtable(NULL);
201 runcmdshell();
202 break;
203 /*
204 * Incremental restoration of a file system.
205 */
206 case 'r':
207 setup();
208 if (dumptime > 0) {
209 /*
210 * This is an incremental dump tape.
211 */
212 vprintf(stdout, "Begin incremental restore\n");
213 initsymtable(symtbl);
214 extractdirs(1);
215 removeoldleaves();
216 vprintf(stdout, "Calculate node updates.\n");
217 treescan(".", UFS_ROOTINO, nodeupdates);
218 findunreflinks();
219 removeoldnodes();
220 } else {
221 /*
222 * This is a level zero dump tape.
223 */
224 vprintf(stdout, "Begin level 0 restore\n");
225 initsymtable((char *)0);
226 extractdirs(1);
227 vprintf(stdout, "Calculate extraction list.\n");
228 treescan(".", UFS_ROOTINO, nodeupdates);
229 }
230 createleaves(symtbl);
231 createlinks();
232 setdirmodes(FORCE);
233 checkrestore();
234 if (dflag) {
235 vprintf(stdout, "Verify the directory structure\n");
236 treescan(".", UFS_ROOTINO, verifyfile);
237 }
238 dumpsymtable(symtbl, (long)1);
239 break;
240 /*
241 * Resume an incremental file system restoration.
242 */
243 case 'R':
244 initsymtable(symtbl);
245 skipmaps();
246 skipdirs();
247 createleaves(symtbl);
248 createlinks();
249 setdirmodes(FORCE);
250 checkrestore();
251 dumpsymtable(symtbl, (long)1);
252 break;
253 /*
254 * List contents of tape.
255 */
256 case 't':
257 setup();
258 extractdirs(0);
259 initsymtable((char *)0);
260 while (argc--) {
261 canon(*argv++, name, sizeof(name));
262 ino = dirlookup(name);
263 if (ino == 0)
264 continue;
265 treescan(name, ino, listfile);
266 }
267 break;
268 /*
269 * Batch extraction of tape contents.
270 */
271 case 'x':
272 setup();
273 extractdirs(1);
274 initsymtable((char *)0);
275 while (argc--) {
276 canon(*argv++, name, sizeof(name));
277 ino = dirlookup(name);
278 if (ino == 0)
279 continue;
280 if (mflag)
281 pathcheck(name);
282 treescan(name, ino, addfile);
283 }
284 createfiles();
285 createlinks();
286 setdirmodes(0);
287 if (dflag)
288 checkrestore();
289 break;
290 }
291 done(0);
292 /* NOTREACHED */
293 }
294
295 static void
usage()296 usage()
297 {
298 const char *const common =
299 "[-b blocksize] [-f file | -P pipecommand] [-s fileno]";
300 const char *const fileell = "[file ...]";
301
302 (void)fprintf(stderr, "usage:\t%s %s\n\t%s %s\n\t%s %s\n"
303 "\t%s %s %s\n\t%s %s %s\n",
304 "restore -i [-dhmNuvy]", common,
305 "restore -R [-dNuvy]", common,
306 "restore -r [-dNuvy]", common,
307 "restore -t [-dhNuvy]", common, fileell,
308 "restore -x [-dhmNuvy]", common, fileell);
309 done(1);
310 }
311
312 /*
313 * obsolete --
314 * Change set of key letters and ordered arguments into something
315 * getopt(3) will like.
316 */
317 static void
obsolete(int * argcp,char ** argvp[])318 obsolete(int *argcp, char **argvp[])
319 {
320 int argc, flags;
321 char *ap, **argv, *flagsp, **nargv, *p;
322
323 /* Setup. */
324 argv = *argvp;
325 argc = *argcp;
326
327 /* Return if no arguments or first argument has leading dash. */
328 ap = argv[1];
329 if (argc == 1 || *ap == '-')
330 return;
331
332 /* Allocate space for new arguments. */
333 if ((*argvp = nargv = malloc((argc + 1) * sizeof(char *))) == NULL ||
334 (p = flagsp = malloc(strlen(ap) + 2)) == NULL)
335 err(1, NULL);
336
337 *nargv++ = *argv;
338 argv += 2, argc -= 2;
339
340 for (flags = 0; *ap; ++ap) {
341 switch (*ap) {
342 case 'b':
343 case 'f':
344 case 's':
345 if (*argv == NULL) {
346 warnx("option requires an argument -- %c", *ap);
347 usage();
348 }
349 if ((nargv[0] = malloc(strlen(*argv) + 2 + 1)) == NULL)
350 err(1, NULL);
351 nargv[0][0] = '-';
352 nargv[0][1] = *ap;
353 (void)strcpy(&nargv[0][2], *argv);
354 ++argv;
355 ++nargv;
356 break;
357 default:
358 if (!flags) {
359 *p++ = '-';
360 flags = 1;
361 }
362 *p++ = *ap;
363 break;
364 }
365 }
366
367 /* Terminate flags. */
368 if (flags) {
369 *p = '\0';
370 *nargv++ = flagsp;
371 } else
372 free(flagsp);
373
374 /* Copy remaining arguments. */
375 while ((*nargv++ = *argv++));
376
377 /* Update argument count. */
378 *argcp = nargv - *argvp - 1;
379 }
380