1 /*-
2 * SPDX-License-Identifier: BSD-3-Clause
3 *
4 * Copyright (c) 1985, 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 #if 0
34 static char sccsid[] = "@(#)interactive.c 8.5 (Berkeley) 5/1/95";
35 #endif
36 #endif /* not lint */
37
38 #include <sys/cdefs.h>
39 #include <sys/param.h>
40 #include <sys/stat.h>
41
42 #include <ufs/ufs/dinode.h>
43 #include <ufs/ufs/dir.h>
44 #include <protocols/dumprestore.h>
45
46 #include <ctype.h>
47 #include <glob.h>
48 #include <limits.h>
49 #include <setjmp.h>
50 #include <stdint.h>
51 #include <stdio.h>
52 #include <stdlib.h>
53 #include <string.h>
54
55 #include "restore.h"
56 #include "extern.h"
57
58 /*
59 * Things to handle interruptions.
60 */
61 static int runshell;
62 static jmp_buf reset;
63 static char *nextarg = NULL;
64
65 /*
66 * Structure and routines associated with listing directories.
67 */
68 struct afile {
69 ino_t fnum; /* inode number of file */
70 char *fname; /* file name */
71 short len; /* name length */
72 char prefix; /* prefix character */
73 char postfix; /* postfix character */
74 };
75 struct arglist {
76 int freeglob; /* glob structure needs to be freed */
77 int argcnt; /* next globbed argument to return */
78 glob_t glob; /* globbing information */
79 char *cmd; /* the current command */
80 };
81
82 static char *copynext(char *, char *);
83 static int fcmp(const void *, const void *);
84 static void formatf(struct afile *, int);
85 static void getcmd(char *, char *, char *, size_t, struct arglist *);
86 struct dirent *glob_readdir(void *);
87 static int glob_stat(const char *, struct stat *);
88 static void mkentry(char *, struct direct *, struct afile *);
89 static void printlist(char *, char *);
90
91 /*
92 * Read and execute commands from the terminal.
93 */
94 void
runcmdshell(void)95 runcmdshell(void)
96 {
97 struct entry *np;
98 ino_t ino;
99 struct arglist arglist;
100 char curdir[MAXPATHLEN];
101 char name[MAXPATHLEN];
102 char cmd[BUFSIZ];
103
104 arglist.freeglob = 0;
105 arglist.argcnt = 0;
106 arglist.glob.gl_flags = GLOB_ALTDIRFUNC;
107 arglist.glob.gl_opendir = rst_opendir;
108 arglist.glob.gl_readdir = glob_readdir;
109 arglist.glob.gl_closedir = rst_closedir;
110 arglist.glob.gl_lstat = glob_stat;
111 arglist.glob.gl_stat = glob_stat;
112 canon("/", curdir, sizeof(curdir));
113 loop:
114 if (setjmp(reset) != 0) {
115 if (arglist.freeglob != 0) {
116 arglist.freeglob = 0;
117 arglist.argcnt = 0;
118 globfree(&arglist.glob);
119 }
120 nextarg = NULL;
121 volno = 0;
122 }
123 runshell = 1;
124 getcmd(curdir, cmd, name, sizeof(name), &arglist);
125 switch (cmd[0]) {
126 /*
127 * Add elements to the extraction list.
128 */
129 case 'a':
130 if (strncmp(cmd, "add", strlen(cmd)) != 0)
131 goto bad;
132 ino = dirlookup(name);
133 if (ino == 0)
134 break;
135 if (mflag)
136 pathcheck(name);
137 treescan(name, ino, addfile);
138 break;
139 /*
140 * Change working directory.
141 */
142 case 'c':
143 if (strncmp(cmd, "cd", strlen(cmd)) != 0)
144 goto bad;
145 ino = dirlookup(name);
146 if (ino == 0)
147 break;
148 if (inodetype(ino) == LEAF) {
149 fprintf(stderr, "%s: not a directory\n", name);
150 break;
151 }
152 (void) strcpy(curdir, name);
153 break;
154 /*
155 * Delete elements from the extraction list.
156 */
157 case 'd':
158 if (strncmp(cmd, "delete", strlen(cmd)) != 0)
159 goto bad;
160 np = lookupname(name);
161 if (np == NULL || (np->e_flags & NEW) == 0) {
162 fprintf(stderr, "%s: not on extraction list\n", name);
163 break;
164 }
165 treescan(name, np->e_ino, deletefile);
166 break;
167 /*
168 * Extract the requested list.
169 */
170 case 'e':
171 if (strncmp(cmd, "extract", strlen(cmd)) != 0)
172 goto bad;
173 createfiles();
174 createlinks();
175 setdirmodes(0);
176 if (dflag)
177 checkrestore();
178 volno = 0;
179 break;
180 /*
181 * List available commands.
182 */
183 case 'h':
184 if (strncmp(cmd, "help", strlen(cmd)) != 0)
185 goto bad;
186 case '?':
187 fprintf(stderr, "%s%s%s%s%s%s%s%s%s%s%s%s%s%s%s%s%s",
188 "Available commands are:\n",
189 "\tls [arg] - list directory\n",
190 "\tcd arg - change directory\n",
191 "\tpwd - print current directory\n",
192 "\tadd [arg] - add `arg' to list of",
193 " files to be extracted\n",
194 "\tdelete [arg] - delete `arg' from",
195 " list of files to be extracted\n",
196 "\textract - extract requested files\n",
197 "\tsetmodes - set modes of requested directories\n",
198 "\tquit - immediately exit program\n",
199 "\twhat - list dump header information\n",
200 "\tverbose - toggle verbose flag",
201 " (useful with ``ls'')\n",
202 "\thelp or `?' - print this list\n",
203 "If no `arg' is supplied, the current",
204 " directory is used\n");
205 break;
206 /*
207 * List a directory.
208 */
209 case 'l':
210 if (strncmp(cmd, "ls", strlen(cmd)) != 0)
211 goto bad;
212 printlist(name, curdir);
213 break;
214 /*
215 * Print current directory.
216 */
217 case 'p':
218 if (strncmp(cmd, "pwd", strlen(cmd)) != 0)
219 goto bad;
220 if (curdir[1] == '\0')
221 fprintf(stderr, "/\n");
222 else
223 fprintf(stderr, "%s\n", &curdir[1]);
224 break;
225 /*
226 * Quit.
227 */
228 case 'q':
229 if (strncmp(cmd, "quit", strlen(cmd)) != 0)
230 goto bad;
231 return;
232 case 'x':
233 if (strncmp(cmd, "xit", strlen(cmd)) != 0)
234 goto bad;
235 return;
236 /*
237 * Toggle verbose mode.
238 */
239 case 'v':
240 if (strncmp(cmd, "verbose", strlen(cmd)) != 0)
241 goto bad;
242 if (vflag) {
243 fprintf(stderr, "verbose mode off\n");
244 vflag = 0;
245 break;
246 }
247 fprintf(stderr, "verbose mode on\n");
248 vflag++;
249 break;
250 /*
251 * Just restore requested directory modes.
252 */
253 case 's':
254 if (strncmp(cmd, "setmodes", strlen(cmd)) != 0)
255 goto bad;
256 setdirmodes(FORCE);
257 break;
258 /*
259 * Print out dump header information.
260 */
261 case 'w':
262 if (strncmp(cmd, "what", strlen(cmd)) != 0)
263 goto bad;
264 printdumpinfo();
265 break;
266 /*
267 * Turn on debugging.
268 */
269 case 'D':
270 if (strncmp(cmd, "Debug", strlen(cmd)) != 0)
271 goto bad;
272 if (dflag) {
273 fprintf(stderr, "debugging mode off\n");
274 dflag = 0;
275 break;
276 }
277 fprintf(stderr, "debugging mode on\n");
278 dflag++;
279 break;
280 /*
281 * Unknown command.
282 */
283 default:
284 bad:
285 fprintf(stderr, "%s: unknown command; type ? for help\n", cmd);
286 break;
287 }
288 goto loop;
289 }
290
291 /*
292 * Read and parse an interactive command.
293 * The first word on the line is assigned to "cmd". If
294 * there are no arguments on the command line, then "curdir"
295 * is returned as the argument. If there are arguments
296 * on the line they are returned one at a time on each
297 * successive call to getcmd. Each argument is first assigned
298 * to "name". If it does not start with "/" the pathname in
299 * "curdir" is prepended to it. Finally "canon" is called to
300 * eliminate any embedded ".." components.
301 */
302 static void
getcmd(char * curdir,char * cmd,char * name,size_t size,struct arglist * ap)303 getcmd(char *curdir, char *cmd, char *name, size_t size, struct arglist *ap)
304 {
305 char *cp;
306 static char input[BUFSIZ];
307 char output[BUFSIZ];
308 # define rawname input /* save space by reusing input buffer */
309
310 /*
311 * Check to see if still processing arguments.
312 */
313 if (ap->argcnt > 0)
314 goto retnext;
315 if (nextarg != NULL)
316 goto getnext;
317 /*
318 * Read a command line and trim off trailing white space.
319 */
320 do {
321 fprintf(stderr, "restore > ");
322 (void) fflush(stderr);
323 if (fgets(input, BUFSIZ, terminal) == NULL) {
324 strcpy(cmd, "quit");
325 return;
326 }
327 } while (input[0] == '\n');
328 for (cp = &input[strlen(input) - 2]; *cp == ' ' || *cp == '\t'; cp--)
329 /* trim off trailing white space and newline */;
330 *++cp = '\0';
331 /*
332 * Copy the command into "cmd".
333 */
334 cp = copynext(input, cmd);
335 ap->cmd = cmd;
336 /*
337 * If no argument, use curdir as the default.
338 */
339 if (*cp == '\0') {
340 (void) strncpy(name, curdir, size);
341 name[size - 1] = '\0';
342 return;
343 }
344 nextarg = cp;
345 /*
346 * Find the next argument.
347 */
348 getnext:
349 cp = copynext(nextarg, rawname);
350 if (*cp == '\0')
351 nextarg = NULL;
352 else
353 nextarg = cp;
354 /*
355 * If it is an absolute pathname, canonicalize it and return it.
356 */
357 if (rawname[0] == '/') {
358 canon(rawname, name, size);
359 } else {
360 /*
361 * For relative pathnames, prepend the current directory to
362 * it then canonicalize and return it.
363 */
364 snprintf(output, sizeof(output), "%s/%s", curdir, rawname);
365 canon(output, name, size);
366 }
367 switch (glob(name, GLOB_ALTDIRFUNC, NULL, &ap->glob)) {
368 case GLOB_NOSPACE:
369 fprintf(stderr, "%s: out of memory\n", ap->cmd);
370 break;
371 case GLOB_NOMATCH:
372 fprintf(stderr, "%s %s: no such file or directory\n", ap->cmd, name);
373 break;
374 }
375 if (ap->glob.gl_pathc == 0)
376 return;
377 ap->freeglob = 1;
378 ap->argcnt = ap->glob.gl_pathc;
379
380 retnext:
381 strncpy(name, ap->glob.gl_pathv[ap->glob.gl_pathc - ap->argcnt], size);
382 name[size - 1] = '\0';
383 if (--ap->argcnt == 0) {
384 ap->freeglob = 0;
385 globfree(&ap->glob);
386 }
387 # undef rawname
388 }
389
390 /*
391 * Strip off the next token of the input.
392 */
393 static char *
copynext(char * input,char * output)394 copynext(char *input, char *output)
395 {
396 char *cp, *bp;
397 char quote;
398
399 for (cp = input; *cp == ' ' || *cp == '\t'; cp++)
400 /* skip to argument */;
401 bp = output;
402 while (*cp != ' ' && *cp != '\t' && *cp != '\0') {
403 /*
404 * Handle back slashes.
405 */
406 if (*cp == '\\') {
407 if (*++cp == '\0') {
408 fprintf(stderr,
409 "command lines cannot be continued\n");
410 continue;
411 }
412 *bp++ = *cp++;
413 continue;
414 }
415 /*
416 * The usual unquoted case.
417 */
418 if (*cp != '\'' && *cp != '"') {
419 *bp++ = *cp++;
420 continue;
421 }
422 /*
423 * Handle single and double quotes.
424 */
425 quote = *cp++;
426 while (*cp != quote && *cp != '\0')
427 *bp++ = *cp++;
428 if (*cp++ == '\0') {
429 fprintf(stderr, "missing %c\n", quote);
430 cp--;
431 continue;
432 }
433 }
434 *bp = '\0';
435 return (cp);
436 }
437
438 /*
439 * Canonicalize file names to always start with ``./'' and
440 * remove any embedded "." and ".." components.
441 */
442 void
canon(char * rawname,char * canonname,size_t len)443 canon(char *rawname, char *canonname, size_t len)
444 {
445 char *cp, *np;
446
447 if (strcmp(rawname, ".") == 0 || strncmp(rawname, "./", 2) == 0)
448 (void) strcpy(canonname, "");
449 else if (rawname[0] == '/')
450 (void) strcpy(canonname, ".");
451 else
452 (void) strcpy(canonname, "./");
453 if (strlen(canonname) + strlen(rawname) >= len) {
454 fprintf(stderr, "canonname: not enough buffer space\n");
455 done(1);
456 }
457
458 (void) strcat(canonname, rawname);
459 /*
460 * Eliminate multiple and trailing '/'s
461 */
462 for (cp = np = canonname; *np != '\0'; cp++) {
463 *cp = *np++;
464 while (*cp == '/' && *np == '/')
465 np++;
466 }
467 *cp = '\0';
468 if (*--cp == '/')
469 *cp = '\0';
470 /*
471 * Eliminate extraneous "." and ".." from pathnames.
472 */
473 for (np = canonname; *np != '\0'; ) {
474 np++;
475 cp = np;
476 while (*np != '/' && *np != '\0')
477 np++;
478 if (np - cp == 1 && *cp == '.') {
479 cp--;
480 (void) strcpy(cp, np);
481 np = cp;
482 }
483 if (np - cp == 2 && strncmp(cp, "..", 2) == 0) {
484 cp--;
485 while (cp > &canonname[1] && *--cp != '/')
486 /* find beginning of name */;
487 (void) strcpy(cp, np);
488 np = cp;
489 }
490 }
491 }
492
493 /*
494 * Do an "ls" style listing of a directory
495 */
496 static void
printlist(char * name,char * basename)497 printlist(char *name, char *basename)
498 {
499 struct afile *fp, *list, *listp;
500 struct direct *dp;
501 struct afile single;
502 RST_DIR *dirp;
503 int entries, len, namelen;
504 char locname[MAXPATHLEN];
505
506 dp = pathsearch(name);
507 if (dp == NULL || (!dflag && TSTINO(dp->d_ino, dumpmap) == 0) ||
508 (!vflag && dp->d_ino == UFS_WINO))
509 return;
510 if ((dirp = rst_opendir(name)) == NULL) {
511 entries = 1;
512 list = &single;
513 mkentry(name, dp, list);
514 len = strlen(basename) + 1;
515 if (strlen(name) - len > single.len) {
516 freename(single.fname);
517 single.fname = savename(&name[len]);
518 single.len = strlen(single.fname);
519 }
520 } else {
521 entries = 0;
522 while ((dp = rst_readdir(dirp)))
523 entries++;
524 rst_closedir(dirp);
525 list = (struct afile *)malloc(entries * sizeof(struct afile));
526 if (list == NULL) {
527 fprintf(stderr, "ls: out of memory\n");
528 return;
529 }
530 if ((dirp = rst_opendir(name)) == NULL)
531 panic("directory reopen failed\n");
532 fprintf(stderr, "%s:\n", name);
533 entries = 0;
534 listp = list;
535 (void)strlcpy(locname, name, MAXPATHLEN);
536 (void)strlcat(locname, "/", MAXPATHLEN);
537 namelen = strlen(locname);
538 while ((dp = rst_readdir(dirp))) {
539 if (dp == NULL)
540 break;
541 if (!dflag && TSTINO(dp->d_ino, dumpmap) == 0)
542 continue;
543 if (!vflag && (dp->d_ino == UFS_WINO ||
544 strcmp(dp->d_name, ".") == 0 ||
545 strcmp(dp->d_name, "..") == 0))
546 continue;
547 locname[namelen] = '\0';
548 if (namelen + dp->d_namlen >= MAXPATHLEN) {
549 fprintf(stderr, "%s%s: name exceeds %d char\n",
550 locname, dp->d_name, MAXPATHLEN);
551 } else {
552 (void)strlcat(locname, dp->d_name, MAXPATHLEN);
553 mkentry(locname, dp, listp++);
554 entries++;
555 }
556 }
557 rst_closedir(dirp);
558 if (entries == 0) {
559 fprintf(stderr, "\n");
560 free(list);
561 return;
562 }
563 qsort((char *)list, entries, sizeof(struct afile), fcmp);
564 }
565 formatf(list, entries);
566 if (dirp != NULL) {
567 for (fp = listp - 1; fp >= list; fp--)
568 freename(fp->fname);
569 fprintf(stderr, "\n");
570 free(list);
571 }
572 }
573
574 /*
575 * Read the contents of a directory.
576 */
577 static void
mkentry(char * name,struct direct * dp,struct afile * fp)578 mkentry(char *name, struct direct *dp, struct afile *fp)
579 {
580 char *cp;
581 struct entry *np;
582
583 fp->fnum = dp->d_ino;
584 fp->fname = savename(dp->d_name);
585 for (cp = fp->fname; *cp; cp++)
586 if (!vflag && !isprint((unsigned char)*cp))
587 *cp = '?';
588 fp->len = cp - fp->fname;
589 if (dflag && TSTINO(fp->fnum, dumpmap) == 0)
590 fp->prefix = '^';
591 else if ((np = lookupname(name)) != NULL && (np->e_flags & NEW))
592 fp->prefix = '*';
593 else
594 fp->prefix = ' ';
595 switch(dp->d_type) {
596
597 default:
598 fprintf(stderr, "Warning: undefined file type %d\n",
599 dp->d_type);
600 /* FALLTHROUGH */
601 case DT_REG:
602 fp->postfix = ' ';
603 break;
604
605 case DT_LNK:
606 fp->postfix = '@';
607 break;
608
609 case DT_FIFO:
610 case DT_SOCK:
611 fp->postfix = '=';
612 break;
613
614 case DT_CHR:
615 case DT_BLK:
616 fp->postfix = '#';
617 break;
618
619 case DT_WHT:
620 fp->postfix = '%';
621 break;
622
623 case DT_UNKNOWN:
624 case DT_DIR:
625 if (inodetype(dp->d_ino) == NODE)
626 fp->postfix = '/';
627 else
628 fp->postfix = ' ';
629 break;
630 }
631 return;
632 }
633
634 /*
635 * Print out a pretty listing of a directory
636 */
637 static void
formatf(struct afile * list,int nentry)638 formatf(struct afile *list, int nentry)
639 {
640 struct afile *fp, *endlist;
641 int width, bigino, haveprefix, havepostfix;
642 int i, j, w, precision, columns, lines;
643
644 width = 0;
645 haveprefix = 0;
646 havepostfix = 0;
647 bigino = UFS_ROOTINO;
648 endlist = &list[nentry];
649 for (fp = &list[0]; fp < endlist; fp++) {
650 if (bigino < fp->fnum)
651 bigino = fp->fnum;
652 if (width < fp->len)
653 width = fp->len;
654 if (fp->prefix != ' ')
655 haveprefix = 1;
656 if (fp->postfix != ' ')
657 havepostfix = 1;
658 }
659 if (haveprefix)
660 width++;
661 if (havepostfix)
662 width++;
663 if (vflag) {
664 for (precision = 0, i = bigino; i > 0; i /= 10)
665 precision++;
666 width += precision + 1;
667 }
668 width++;
669 columns = 81 / width;
670 if (columns == 0)
671 columns = 1;
672 lines = howmany(nentry, columns);
673 for (i = 0; i < lines; i++) {
674 for (j = 0; j < columns; j++) {
675 fp = &list[j * lines + i];
676 if (vflag) {
677 fprintf(stderr, "%*ju ",
678 precision, (uintmax_t)fp->fnum);
679 fp->len += precision + 1;
680 }
681 if (haveprefix) {
682 putc(fp->prefix, stderr);
683 fp->len++;
684 }
685 fprintf(stderr, "%s", fp->fname);
686 if (havepostfix) {
687 putc(fp->postfix, stderr);
688 fp->len++;
689 }
690 if (fp + lines >= endlist) {
691 fprintf(stderr, "\n");
692 break;
693 }
694 for (w = fp->len; w < width; w++)
695 putc(' ', stderr);
696 }
697 }
698 }
699
700 /*
701 * Skip over directory entries that are not on the tape
702 *
703 * First have to get definition of a dirent.
704 */
705 #undef DIRBLKSIZ
706 #include <dirent.h>
707 #undef d_ino
708
709 struct dirent *
glob_readdir(void * dirp)710 glob_readdir(void *dirp)
711 {
712 struct direct *dp;
713 static struct dirent adirent;
714
715 while ((dp = rst_readdir(dirp)) != NULL) {
716 if (!vflag && dp->d_ino == UFS_WINO)
717 continue;
718 if (dflag || TSTINO(dp->d_ino, dumpmap))
719 break;
720 }
721 if (dp == NULL)
722 return (NULL);
723 adirent.d_fileno = dp->d_ino;
724 adirent.d_namlen = dp->d_namlen;
725 memmove(adirent.d_name, dp->d_name, dp->d_namlen + 1);
726 return (&adirent);
727 }
728
729 /*
730 * Return st_mode information in response to stat or lstat calls
731 */
732 static int
glob_stat(const char * name,struct stat * stp)733 glob_stat(const char *name, struct stat *stp)
734 {
735 struct direct *dp;
736
737 dp = pathsearch(name);
738 if (dp == NULL || (!dflag && TSTINO(dp->d_ino, dumpmap) == 0) ||
739 (!vflag && dp->d_ino == UFS_WINO))
740 return (-1);
741 if (inodetype(dp->d_ino) == NODE)
742 stp->st_mode = IFDIR;
743 else
744 stp->st_mode = IFREG;
745 return (0);
746 }
747
748 /*
749 * Comparison routine for qsort.
750 */
751 static int
fcmp(const void * f1,const void * f2)752 fcmp(const void *f1, const void *f2)
753 {
754 return (strcoll(((struct afile *)f1)->fname,
755 ((struct afile *)f2)->fname));
756 }
757
758 /*
759 * respond to interrupts
760 */
761 void
onintr(int signo __unused)762 onintr(int signo __unused)
763 {
764 if (command == 'i' && runshell)
765 longjmp(reset, 1);
766 if (reply("restore interrupted, continue") == FAIL)
767 done(1);
768 }
769