text: /*
   * Copyright (c) 1985, 1993
   *	The Regents of the University of California.  All rights reserved.
   *
   * %sccs.include.redist.c%
   */
  
  #ifndef lint
  static char sccsid[] = "@(#)interactive.c	8.5 (Berkeley) 05/01/95";
  #endif /* not lint */
  
  #include <sys/param.h>
  #include <sys/time.h>
  #include <sys/stat.h>
  
  #include <ufs/ufs/dinode.h>
  #include <ufs/ufs/dir.h>
  #include <ufs/ffs/fs.h>
  #include <protocols/dumprestore.h>
  
  #include <setjmp.h>
  #include <glob.h>
  #include <stdio.h>
  #include <stdlib.h>
  #include <string.h>
  
  #include "restore.h"
  #include "extern.h"
  
  #define round(a, b) (((a) + (b) - 1) / (b) * (b))
  
  /*
   * Things to handle interruptions.
   */
  static int runshell;
  static jmp_buf reset;
  static char *nextarg = NULL;
  
  /*
   * Structure and routines associated with listing directories.
   */
  struct afile {
  	ino_t	fnum;		/* inode number of file */
  	char	*fname;		/* file name */
  	short	len;		/* name length */
  	char	prefix;		/* prefix character */
  	char	postfix;	/* postfix character */
  };
  struct arglist {
  	int	freeglob;	/* glob structure needs to be freed */
  	int	argcnt;		/* next globbed argument to return */
  	glob_t	glob;		/* globbing information */
  	char	*cmd;		/* the current command */
  };
  
  static char	*copynext __P((char *, char *));
  static int	 fcmp __P((const void *, const void *));
  static void	 formatf __P((struct afile *, int));
  static void	 getcmd __P((char *, char *, char *, struct arglist *));
  struct dirent	*glob_readdir __P((RST_DIR *dirp));
  static int	 glob_stat __P((const char *, struct stat *));
  static void	 mkentry __P((char *, struct direct *, struct afile *));
  static void	 printlist __P((char *, char *));
  
  /*
   * Read and execute commands from the terminal.
   */
  void
  runcmdshell()
  {
  	register struct entry *np;
  	ino_t ino;
  	struct arglist arglist;
  	char curdir[MAXPATHLEN];
  	char name[MAXPATHLEN];
  	char cmd[BUFSIZ];
  
  	arglist.freeglob = 0;
  	arglist.argcnt = 0;
  	arglist.glob.gl_flags = GLOB_ALTDIRFUNC;
  	arglist.glob.gl_opendir = (void *)rst_opendir;
  	arglist.glob.gl_readdir = (void *)glob_readdir;
  	arglist.glob.gl_closedir = (void *)rst_closedir;
  	arglist.glob.gl_lstat = glob_stat;
  	arglist.glob.gl_stat = glob_stat;
  	canon("/", curdir);
  loop:
  	if (setjmp(reset) != 0) {
  		if (arglist.freeglob != 0) {
  			arglist.freeglob = 0;
  			arglist.argcnt = 0;
  			globfree(&arglist.glob);
  		}
  		nextarg = NULL;
  		volno = 0;
  	}
  	runshell = 1;
  	getcmd(curdir, cmd, name, &arglist);
  	switch (cmd[0]) {
  	/*
  	 * Add elements to the extraction list.
  	 */
  	case 'a':
  		if (strncmp(cmd, "add", strlen(cmd)) != 0)
  			goto bad;
  		ino = dirlookup(name);
  		if (ino == 0)
  			break;
  		if (mflag)
  			pathcheck(name);
  		treescan(name, ino, addfile);
  		break;
  	/*
  	 * Change working directory.
  	 */
  	case 'c':
  		if (strncmp(cmd, "cd", strlen(cmd)) != 0)
  			goto bad;
  		ino = dirlookup(name);
  		if (ino == 0)
  			break;
  		if (inodetype(ino) == LEAF) {
  			fprintf(stderr, "%s: not a directory\n", name);
  			break;
  		}
  		(void) strcpy(curdir, name);
  		break;
  	/*
  	 * Delete elements from the extraction list.
  	 */
  	case 'd':
  		if (strncmp(cmd, "delete", strlen(cmd)) != 0)
  			goto bad;
  		np = lookupname(name);
  		if (np == NULL || (np->e_flags & NEW) == 0) {
  			fprintf(stderr, "%s: not on extraction list\n", name);
  			break;
  		}
  		treescan(name, np->e_ino, deletefile);
  		break;
  	/*
  	 * Extract the requested list.
  	 */
  	case 'e':
  		if (strncmp(cmd, "extract", strlen(cmd)) != 0)
  			goto bad;
  		createfiles();
  		createlinks();
  		setdirmodes(0);
  		if (dflag)
  			checkrestore();
  		volno = 0;
  		break;
  	/*
  	 * List available commands.
  	 */
  	case 'h':
  		if (strncmp(cmd, "help", strlen(cmd)) != 0)
  			goto bad;
  	case '?':
  		fprintf(stderr, "%s%s%s%s%s%s%s%s%s%s%s%s%s%s%s%s%s",
  			"Available commands are:\n",
  			"\tls [arg] - list directory\n",
  			"\tcd arg - change directory\n",
  			"\tpwd - print current directory\n",
  			"\tadd [arg] - add `arg' to list of",
  			" files to be extracted\n",
  			"\tdelete [arg] - delete `arg' from",
  			" list of files to be extracted\n",
  			"\textract - extract requested files\n",
  			"\tsetmodes - set modes of requested directories\n",
  			"\tquit - immediately exit program\n",
  			"\twhat - list dump header information\n",
  			"\tverbose - toggle verbose flag",
  			" (useful with ``ls'')\n",
  			"\thelp or `?' - print this list\n",
  			"If no `arg' is supplied, the current",
  			" directory is used\n");
  		break;
  	/*
  	 * List a directory.
  	 */
  	case 'l':
  		if (strncmp(cmd, "ls", strlen(cmd)) != 0)
  			goto bad;
  		printlist(name, curdir);
  		break;
  	/*
  	 * Print current directory.
  	 */
  	case 'p':
  		if (strncmp(cmd, "pwd", strlen(cmd)) != 0)
  			goto bad;
  		if (curdir[1] == '\0')
  			fprintf(stderr, "/\n");
  		else
  			fprintf(stderr, "%s\n", &curdir[1]);
  		break;
  	/*
  	 * Quit.
  	 */
  	case 'q':
  		if (strncmp(cmd, "quit", strlen(cmd)) != 0)
  			goto bad;
  		return;
  	case 'x':
  		if (strncmp(cmd, "xit", strlen(cmd)) != 0)
  			goto bad;
  		return;
  	/*
  	 * Toggle verbose mode.
  	 */
  	case 'v':
  		if (strncmp(cmd, "verbose", strlen(cmd)) != 0)
  			goto bad;
  		if (vflag) {
  			fprintf(stderr, "verbose mode off\n");
  			vflag = 0;
  			break;
  		}
  		fprintf(stderr, "verbose mode on\n");
  		vflag++;
  		break;
  	/*
  	 * Just restore requested directory modes.
  	 */
  	case 's':
  		if (strncmp(cmd, "setmodes", strlen(cmd)) != 0)
  			goto bad;
  		setdirmodes(FORCE);
  		break;
  	/*
  	 * Print out dump header information.
  	 */
  	case 'w':
  		if (strncmp(cmd, "what", strlen(cmd)) != 0)
  			goto bad;
  		printdumpinfo();
  		break;
  	/*
  	 * Turn on debugging.
  	 */
  	case 'D':
  		if (strncmp(cmd, "Debug", strlen(cmd)) != 0)
  			goto bad;
  		if (dflag) {
  			fprintf(stderr, "debugging mode off\n");
  			dflag = 0;
  			break;
  		}
  		fprintf(stderr, "debugging mode on\n");
  		dflag++;
  		break;
  	/*
  	 * Unknown command.
  	 */
  	default:
  	bad:
  		fprintf(stderr, "%s: unknown command; type ? for help\n", cmd);
  		break;
  	}
  	goto loop;
  }
  
  /*
   * Read and parse an interactive command.
   * The first word on the line is assigned to "cmd". If
   * there are no arguments on the command line, then "curdir"
   * is returned as the argument. If there are arguments
   * on the line they are returned one at a time on each
   * successive call to getcmd. Each argument is first assigned
   * to "name". If it does not start with "/" the pathname in
   * "curdir" is prepended to it. Finally "canon" is called to
   * eliminate any embedded ".." components.
   */
  static void
  getcmd(curdir, cmd, name, ap)
  	char *curdir, *cmd, *name;
  	struct arglist *ap;
  {
  	register char *cp;
  	static char input[BUFSIZ];
  	char output[BUFSIZ];
  #	define rawname input	/* save space by reusing input buffer */
  
  	/*
  	 * Check to see if still processing arguments.
  	 */
  	if (ap->argcnt > 0)
  		goto retnext;
  	if (nextarg != NULL)
  		goto getnext;
  	/*
  	 * Read a command line and trim off trailing white space.
  	 */
  	do	{
  		fprintf(stderr, "restore > ");
  		(void) fflush(stderr);
  		(void) fgets(input, BUFSIZ, terminal);
  	} while (!feof(terminal) && input[0] == '\n');
  	if (feof(terminal)) {
  		(void) strcpy(cmd, "quit");
  		return;
  	}
  	for (cp = &input[strlen(input) - 2]; *cp == ' ' || *cp == '\t'; cp--)
  		/* trim off trailing white space and newline */;
  	*++cp = '\0';
  	/*
  	 * Copy the command into "cmd".
  	 */
  	cp = copynext(input, cmd);
  	ap->cmd = cmd;
  	/*
  	 * If no argument, use curdir as the default.
  	 */
  	if (*cp == '\0') {
  		(void) strcpy(name, curdir);
  		return;
  	}
  	nextarg = cp;
  	/*
  	 * Find the next argument.
  	 */
  getnext:
  	cp = copynext(nextarg, rawname);
  	if (*cp == '\0')
  		nextarg = NULL;
  	else
  		nextarg = cp;
  	/*
  	 * If it is an absolute pathname, canonicalize it and return it.
  	 */
  	if (rawname[0] == '/') {
  		canon(rawname, name);
  	} else {
  		/*
  		 * For relative pathnames, prepend the current directory to
  		 * it then canonicalize and return it.
  		 */
  		(void) strcpy(output, curdir);
  		(void) strcat(output, "/");
  		(void) strcat(output, rawname);
  		canon(output, name);
  	}
  	if (glob(name, GLOB_ALTDIRFUNC, NULL, &ap->glob) < 0)
  		fprintf(stderr, "%s: out of memory\n", ap->cmd);
  	if (ap->glob.gl_pathc == 0)
  		return;
  	ap->freeglob = 1;
  	ap->argcnt = ap->glob.gl_pathc;
  
  retnext:
  	strcpy(name, ap->glob.gl_pathv[ap->glob.gl_pathc - ap->argcnt]);
  	if (--ap->argcnt == 0) {
  		ap->freeglob = 0;
  		globfree(&ap->glob);
  	}
  #	undef rawname
  }
  
  /*
   * Strip off the next token of the input.
   */
  static char *
  copynext(input, output)
  	char *input, *output;
  {
  	register char *cp, *bp;
  	char quote;
  
  	for (cp = input; *cp == ' ' || *cp == '\t'; cp++)
  		/* skip to argument */;
  	bp = output;
  	while (*cp != ' ' && *cp != '\t' && *cp != '\0') {
  		/*
  		 * Handle back slashes.
  		 */
  		if (*cp == '\\') {
  			if (*++cp == '\0') {
  				fprintf(stderr,
  					"command lines cannot be continued\n");
  				continue;
  			}
  			*bp++ = *cp++;
  			continue;
  		}
  		/*
  		 * The usual unquoted case.
  		 */
  		if (*cp != '\'' && *cp != '"') {
  			*bp++ = *cp++;
  			continue;
  		}
  		/*
  		 * Handle single and double quotes.
  		 */
  		quote = *cp++;
  		while (*cp != quote && *cp != '\0')
  			*bp++ = *cp++ | 0200;
  		if (*cp++ == '\0') {
  			fprintf(stderr, "missing %c\n", quote);
  			cp--;
  			continue;
  		}
  	}
  	*bp = '\0';
  	return (cp);
  }
  
  /*
   * Canonicalize file names to always start with ``./'' and
   * remove any imbedded "." and ".." components.
   */
  void
  canon(rawname, canonname)
  	char *rawname, *canonname;
  {
  	register char *cp, *np;
  
  	if (strcmp(rawname, ".") == 0 || strncmp(rawname, "./", 2) == 0)
  		(void) strcpy(canonname, "");
  	else if (rawname[0] == '/')
  		(void) strcpy(canonname, ".");
  	else
  		(void) strcpy(canonname, "./");
  	(void) strcat(canonname, rawname);
  	/*
  	 * Eliminate multiple and trailing '/'s
  	 */
  	for (cp = np = canonname; *np != '\0'; cp++) {
  		*cp = *np++;
  		while (*cp == '/' && *np == '/')
  			np++;
  	}
  	*cp = '\0';
  	if (*--cp == '/')
  		*cp = '\0';
  	/*
  	 * Eliminate extraneous "." and ".." from pathnames.
  	 */
  	for (np = canonname; *np != '\0'; ) {
  		np++;
  		cp = np;
  		while (*np != '/' && *np != '\0')
  			np++;
  		if (np - cp == 1 && *cp == '.') {
  			cp--;
  			(void) strcpy(cp, np);
  			np = cp;
  		}
  		if (np - cp == 2 && strncmp(cp, "..", 2) == 0) {
  			cp--;
  			while (cp > &canonname[1] && *--cp != '/')
  				/* find beginning of name */;
  			(void) strcpy(cp, np);
  			np = cp;
  		}
  	}
  }
  
  /*
   * Do an "ls" style listing of a directory
   */
  static void
  printlist(name, basename)
  	char *name;
  	char *basename;
  {
  	register struct afile *fp, *list, *listp;
  	register struct direct *dp;
  	struct afile single;
  	RST_DIR *dirp;
  	int entries, len, namelen;
  	char locname[MAXPATHLEN + 1];
  
  	dp = pathsearch(name);
  	if (dp == NULL || (!dflag && TSTINO(dp->d_ino, dumpmap) == 0) ||
  	    (!vflag && dp->d_ino == WINO))
  		return;
  	if ((dirp = rst_opendir(name)) == NULL) {
  		entries = 1;
  		list = &single;
  		mkentry(name, dp, list);
  		len = strlen(basename) + 1;
  		if (strlen(name) - len > single.len) {
  			freename(single.fname);
  			single.fname = savename(&name[len]);
  			single.len = strlen(single.fname);
  		}
  	} else {
  		entries = 0;
  		while (dp = rst_readdir(dirp))
  			entries++;
  		rst_closedir(dirp);
  		list = (struct afile *)malloc(entries * sizeof(struct afile));
  		if (list == NULL) {
  			fprintf(stderr, "ls: out of memory\n");
  			return;
  		}
  		if ((dirp = rst_opendir(name)) == NULL)
  			panic("directory reopen failed\n");
  		fprintf(stderr, "%s:\n", name);
  		entries = 0;
  		listp = list;
  		(void) strncpy(locname, name, MAXPATHLEN);
  		(void) strncat(locname, "/", MAXPATHLEN);
  		namelen = strlen(locname);
  		while (dp = rst_readdir(dirp)) {
  			if (dp == NULL)
  				break;
  			if (!dflag && TSTINO(dp->d_ino, dumpmap) == 0)
  				continue;
  			if (!vflag && (dp->d_ino == WINO ||
  			     strcmp(dp->d_name, ".") == 0 ||
  			     strcmp(dp->d_name, "..") == 0))
  				continue;
  			locname[namelen] = '\0';
  			if (namelen + dp->d_namlen >= MAXPATHLEN) {
  				fprintf(stderr, "%s%s: name exceeds %d char\n",
  					locname, dp->d_name, MAXPATHLEN);
  			} else {
  				(void) strncat(locname, dp->d_name,
  				    (int)dp->d_namlen);
  				mkentry(locname, dp, listp++);
  				entries++;
  			}
  		}
  		rst_closedir(dirp);
  		if (entries == 0) {
  			fprintf(stderr, "\n");
  			free(list);
  			return;
  		}
  		qsort((char *)list, entries, sizeof(struct afile), fcmp);
  	}
  	formatf(list, entries);
  	if (dirp != NULL) {
  		for (fp = listp - 1; fp >= list; fp--)
  			freename(fp->fname);
  		fprintf(stderr, "\n");
  		free(list);
  	}
  }
  
  /*
   * Read the contents of a directory.
   */
  static void
  mkentry(name, dp, fp)
  	char *name;
  	struct direct *dp;
  	register struct afile *fp;
  {
  	char *cp;
  	struct entry *np;
  
  	fp->fnum = dp->d_ino;
  	fp->fname = savename(dp->d_name);
  	for (cp = fp->fname; *cp; cp++)
  		if (!vflag && (*cp < ' ' || *cp >= 0177))
  			*cp = '?';
  	fp->len = cp - fp->fname;
  	if (dflag && TSTINO(fp->fnum, dumpmap) == 0)
  		fp->prefix = '^';
  	else if ((np = lookupname(name)) != NULL && (np->e_flags & NEW))
  		fp->prefix = '*';
  	else
  		fp->prefix = ' ';
  	switch(dp->d_type) {
  
  	default:
  		fprintf(stderr, "Warning: undefined file type %d\n",
  		    dp->d_type);
  		/* fall through */
  	case DT_REG:
  		fp->postfix = ' ';
  		break;
  
  	case DT_LNK:
  		fp->postfix = '@';
  		break;
  
  	case DT_FIFO:
  	case DT_SOCK:
  		fp->postfix = '=';
  		break;
  
  	case DT_CHR:
  	case DT_BLK:
  		fp->postfix = '#';
  		break;
  
  	case DT_WHT:
  		fp->postfix = '%';
  		break;
  
  	case DT_UNKNOWN:
  	case DT_DIR:
  		if (inodetype(dp->d_ino) == NODE)
  			fp->postfix = '/';
  		else
  			fp->postfix = ' ';
  		break;
  	}
  	return;
  }
  
  /*
   * Print out a pretty listing of a directory
   */
  static void
  formatf(list, nentry)
  	register struct afile *list;
  	int nentry;
  {
  	register struct afile *fp, *endlist;
  	int width, bigino, haveprefix, havepostfix;
  	int i, j, w, precision, columns, lines;
  
  	width = 0;
  	haveprefix = 0;
  	havepostfix = 0;
  	bigino = ROOTINO;
  	endlist = &list[nentry];
  	for (fp = &list[0]; fp < endlist; fp++) {
  		if (bigino < fp->fnum)
  			bigino = fp->fnum;
  		if (width < fp->len)
  			width = fp->len;
  		if (fp->prefix != ' ')
  			haveprefix = 1;
  		if (fp->postfix != ' ')
  			havepostfix = 1;
  	}
  	if (haveprefix)
  		width++;
  	if (havepostfix)
  		width++;
  	if (vflag) {
  		for (precision = 0, i = bigino; i > 0; i /= 10)
  			precision++;
  		width += precision + 1;
  	}
  	width++;
  	columns = 81 / width;
  	if (columns == 0)
  		columns = 1;
  	lines = (nentry + columns - 1) / columns;
  	for (i = 0; i < lines; i++) {
  		for (j = 0; j < columns; j++) {
  			fp = &list[j * lines + i];
  			if (vflag) {
  				fprintf(stderr, "%*d ", precision, fp->fnum);
  				fp->len += precision + 1;
  			}
  			if (haveprefix) {
  				putc(fp->prefix, stderr);
  				fp->len++;
  			}
  			fprintf(stderr, "%s", fp->fname);
  			if (havepostfix) {
  				putc(fp->postfix, stderr);
  				fp->len++;
  			}
  			if (fp + lines >= endlist) {
  				fprintf(stderr, "\n");
  				break;
  			}
  			for (w = fp->len; w < width; w++)
  				putc(' ', stderr);
  		}
  	}
  }
  
  /*
   * Skip over directory entries that are not on the tape
   *
   * First have to get definition of a dirent.
   */
  #undef DIRBLKSIZ
  #include <dirent.h>
  #undef d_ino
  
  struct dirent *
  glob_readdir(dirp)
  	RST_DIR *dirp;
  {
  	struct direct *dp;
  	static struct dirent adirent;
  
  	while ((dp = rst_readdir(dirp)) != NULL) {
  		if (!vflag && dp->d_ino == WINO)
  			continue;
  		if (dflag || TSTINO(dp->d_ino, dumpmap))
  			break;
  	}
  	if (dp == NULL)
  		return (NULL);
  	adirent.d_fileno = dp->d_ino;
  	adirent.d_namlen = dp->d_namlen;
  	memmove(adirent.d_name, dp->d_name, dp->d_namlen + 1);
  	return (&adirent);
  }
  
  /*
   * Return st_mode information in response to stat or lstat calls
   */
  static int
  glob_stat(name, stp)
  	const char *name;
  	struct stat *stp;
  {
  	register struct direct *dp;
  
  	dp = pathsearch(name);
  	if (dp == NULL || (!dflag && TSTINO(dp->d_ino, dumpmap) == 0) ||
  	    (!vflag && dp->d_ino == WINO))
  		return (-1);
  	if (inodetype(dp->d_ino) == NODE)
  		stp->st_mode = IFDIR;
  	else
  		stp->st_mode = IFREG;
  	return (0);
  }
  
  /*
   * Comparison routine for qsort.
   */
  static int
  fcmp(f1, f2)
  	register const void *f1, *f2;
  {
  	return (strcmp(((struct afile *)f1)->fname,
  	    ((struct afile *)f2)->fname));
  }
  
  /*
   * respond to interrupts
   */
  void
  onintr(signo)
  	int signo;
  {
  	if (command == 'i' && runshell)
  		longjmp(reset, 1);
  	if (reply("restore interrupted, continue") == FAIL)
  		done(1);
  }
  