usr/src/sbin/restore/utilities.c

/* Copyright (c) 1983 Regents of the University of California */

#ifndef lint
static char sccsid[] = "@(#)utilities.c 3.13    (Berkeley)      83/05/19";
#endif

#include "restore.h"

/*
 * Insure that all the components of a pathname exist.
 */
pathcheck(name)
        char *name;
{
        register char *cp;
        struct entry *ep;
        char *start;

        start = index(name, '/');
        if (start == 0)
                return;
        for (cp = start; *cp != '\0'; cp++) {
                if (*cp != '/')
                        continue;
                *cp = '\0';
                ep = lookupname(name);
                if (ep == NIL) {
                        ep = addentry(name, psearch(name), NODE);
                        newnode(ep);
                        ep->e_flags |= NEW|KEEP;
                }
                *cp = '/';
        }
}

/*
 * Change a name to a unique temporary name.
 */
mktempname(ep)
        register struct entry *ep;
{
        char oldname[MAXPATHLEN];

        if (ep->e_flags & TMPNAME)
                badentry(ep, "mktempname: called with TMPNAME");
        ep->e_flags |= TMPNAME;
        (void) strcpy(oldname, myname(ep));
        freename(ep->e_name);
        ep->e_name = savename(gentempname(ep));
        ep->e_namlen = strlen(ep->e_name);
        renameit(oldname, myname(ep));
}

/*
 * Generate a temporary name for an entry.
 */
char *
gentempname(ep)
        struct entry *ep;
{
        static char name[MAXPATHLEN];
        struct entry *np;
        long i = 0;

        for (np = lookupino(ep->e_ino); np != NIL && np != ep; np = np->e_links)
                i++;
        if (np == NIL)
                badentry(ep, "not on ino list");
        (void) sprintf(name, "%s%d%d", TMPHDR, i, ep->e_ino);
        return (name);
}

/*
 * Rename a file or directory.
 */
renameit(from, to)
        char *from, *to;
{
        if (rename(from, to) < 0) {
                fprintf(stderr, "Warning: cannot rename %s to %s", from, to);
                (void) fflush(stderr);
                perror("");
                return;
        }
        vprintf(stdout, "rename %s to %s\n", from, to);
}

/*
 * Create a new node (directory).
 */
newnode(np)
        struct entry *np;
{
        char *cp;

        if (np->e_type != NODE)
                badentry(np, "newnode: not a node");
        cp = myname(np);
        if (mkdir(cp, 0777) < 0) {
                fprintf(stderr, "Warning: ");
                (void) fflush(stderr);
                perror(cp);
                return;
        }
        vprintf(stdout, "Make node %s\n", cp);
}

/*
 * Remove an old node (directory).
 */
removenode(ep)
        register struct entry *ep;
{
        char *cp;

        if (ep->e_type != NODE)
                badentry(ep, "removenode: not a node");
        if (ep->e_entries != NIL)
                badentry(ep, "removenode: non-empty directory");
        ep->e_flags |= REMOVED;
        ep->e_flags &= ~TMPNAME;
        cp = myname(ep);
        if (rmdir(cp) < 0) {
                fprintf(stderr, "Warning: ");
                (void) fflush(stderr);
                perror(cp);
                return;
        }
        vprintf(stdout, "Remove node %s\n", cp);
}

/*
 * Remove a leaf.
 */
removeleaf(ep)
        register struct entry *ep;
{
        char *cp;

        if (ep->e_type != LEAF)
                badentry(ep, "removeleaf: not a leaf");
        ep->e_flags |= REMOVED;
        ep->e_flags &= ~TMPNAME;
        cp = myname(ep);
        if (unlink(cp) < 0) {
                fprintf(stderr, "Warning: ");
                (void) fflush(stderr);
                perror(cp);
                return;
        }
        vprintf(stdout, "Remove leaf %s\n", cp);
}

/*
 * Create a link.
 */
linkit(existing, new, type)
        char *existing, *new;
        int type;
{

        if (type == SYMLINK) {
                if (symlink(existing, new) < 0) {
                        fprintf(stderr,
                                "Warning: cannot create symbolic link %s->%s",
                                new, existing);
                        (void) fflush(stderr);
                        perror("");
                        return;
                }
        } else if (type == HARDLINK) {
                if (link(existing, new) < 0) {
                        fprintf(stderr,
                                "Warning: cannot create hard link %s->%s",
                                new, existing);
                        (void) fflush(stderr);
                        perror("");
                        return;
                }
        } else {
                panic("linkit: unknown type %d\n", type);
        }
        vprintf(stdout, "Create %s link %s->%s\n",
                type == SYMLINK ? "symbolic" : "hard", new, existing);
}

/*
 * find lowest number file (above "start") that needs to be extracted
 */
ino_t
lowerbnd(start)
        ino_t start;
{
        register struct entry *ep;

        for ( ; start < maxino; start++) {
                ep = lookupino(start);
                if (ep == NIL || ep->e_type == NODE)
                        continue;
                if (ep->e_flags & (NEW|EXTRACT))
                        return (start);
        }
        return (start);
}

/*
 * find highest number file (below "start") that needs to be extracted
 */
ino_t
upperbnd(start)
        ino_t start;
{
        register struct entry *ep;

        for ( ; start > ROOTINO; start--) {
                ep = lookupino(start);
                if (ep == NIL || ep->e_type == NODE)
                        continue;
                if (ep->e_flags & (NEW|EXTRACT))
                        return (start);
        }
        return (start);
}

/*
 * report on a badly formed entry
 */
badentry(ep, msg)
        register struct entry *ep;
        char *msg;
{

        fprintf(stderr, "bad entry: %s\n", msg);
        fprintf(stderr, "name: %s\n", myname(ep));
        fprintf(stderr, "parent name %s\n", myname(ep->e_parent));
        if (ep->e_sibling != NIL)
                fprintf(stderr, "sibling name: %s\n", myname(ep->e_sibling));
        if (ep->e_entries != NIL)
                fprintf(stderr, "next entry name: %s\n", myname(ep->e_entries));
        if (ep->e_links != NIL)
                fprintf(stderr, "next link name: %s\n", myname(ep->e_links));
        if (ep->e_next != NIL)
                fprintf(stderr, "next hashchain name: %s\n", myname(ep->e_next));
        fprintf(stderr, "entry type: %s\n",
                ep->e_type == NODE ? "NODE" : "LEAF");
        fprintf(stderr, "inode number: %ld\n", ep->e_ino);
        panic("flags: %s\n", flagvalues(ep));
}

/*
 * Construct a string indicating the active flag bits of an entry.
 */
char *
flagvalues(ep)
        register struct entry *ep;
{
        static char flagbuf[BUFSIZ];

        (void) strcpy(flagbuf, "|NIL");
        flagbuf[0] = '\0';
        if (ep->e_flags & REMOVED)
                (void) strcat(flagbuf, "|REMOVED");
        if (ep->e_flags & TMPNAME)
                (void) strcat(flagbuf, "|TMPNAME");
        if (ep->e_flags & EXTRACT)
                (void) strcat(flagbuf, "|EXTRACT");
        if (ep->e_flags & NEW)
                (void) strcat(flagbuf, "|NEW");
        if (ep->e_flags & KEEP)
                (void) strcat(flagbuf, "|KEEP");
        return (&flagbuf[1]);
}

/*
 * Check to see if a name is on a dump tape.
 */
ino_t
dirlookup(name)
        char *name;
{
        ino_t ino;

        ino = psearch(name);
        if (ino == 0 || BIT(ino, dumpmap) == 0)
                fprintf(stderr, "%s is not on tape\n", name);
        return (ino);
}

/*
 * Canonicalize file names to always start with ``./'' and
 * remove any imbedded ".." components.
 */
canon(rawname, canonname)
        char *rawname, *canonname;
{
        register char *cp, *np;
        int len;

        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);
        len = strlen(canonname) - 1;
        if (canonname[len] == '/')
                canonname[len] = '\0';
        /*
         * Eliminate extraneous ".." from pathnames.
         */
        for (np = canonname; *np != '\0'; ) {
                np++;
                cp = np;
                while (*np != '/' && *np != '\0')
                        np++;
                if (np - cp == 2 && strncmp(cp, "..", 2) == 0) {
                        cp--;
                        while (cp > &canonname[1] && *--cp != '/')
                                /* find beginning of name */;
                        (void) strcpy(cp, np);
                        np = cp;
                }
        }
}

/*
 * Elicit a reply.
 */
reply(question)
        char *question;
{
        char c;

        fprintf(stderr, "%s? ", question);
        do      {
                fprintf(stderr, "[yn] ");
                (void) fflush(stderr);
                c = getc(terminal);
                while (c != '\n' && getc(terminal) != '\n')
                        /* void */;
        } while (c != 'y' && c != 'n');
        if (c == 'y')
                return (GOOD);
        return (FAIL);
}

/*
 * 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.
 */
getcmd(curdir, cmd, name)
        char *curdir, *cmd, *name;
{
        register char *cp, *bp;
        char output[BUFSIZ];
        static char input[BUFSIZ];
        static char *nextarg = NULL;

        /*
         * Check to see if still processing arguments.
         */
        if (nextarg != NULL)
                goto getnext;
        nextarg = NULL;
        /*
         * 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".
         */
        for (cp = input; *cp == ' ' || *cp == '\t'; cp++)
                /* skip to command */;
        for (bp = cmd; *cp != ' ' && *cp != '\t' && *cp != '\0'; )
                *bp++ = *cp++;
        *bp = '\0';
        /*
         * If no argument, use curdir as the default.
         */
        if (*cp == '\0') {
                (void) strcpy(name, curdir);
                return;
        }
        nextarg = cp;
        /*
         * Find the next argument.
         */
getnext:
        for (cp = nextarg + 1; *cp == ' ' || *cp == '\t'; cp++)
                /* skip to argument */;
        for (bp = cp; *cp != ' ' && *cp != '\t' && *cp != '\0'; cp++)
                /* skip to end of argument */;
        if (*cp == '\0')
                nextarg = NULL;
        else
                nextarg = cp;
        *cp = '\0';
        /*
         * If it an absolute pathname, canonicalize it and return it.
         */
        if (*bp == '/') {
                canon(bp, name);
                return;
        }
        /*
         * For relative pathnames, prepend the current directory to
         * it then canonicalize and return it.
         */
        (void) strcpy(output, curdir);
        (void) strcat(output, "/");
        (void) strcat(output, bp);
        canon(output, name);
}

/*
 * respond to interrupts
 */
onintr()
{
        if (reply("restore interrupted, continue") == FAIL)
                done(1);
        if (signal(SIGINT, onintr) == SIG_IGN)
                (void) signal(SIGINT, SIG_IGN);
        if (signal(SIGTERM, onintr) == SIG_IGN)
                (void) signal(SIGTERM, SIG_IGN);
}

/*
 * handle unexpected inconsistencies
 */
/* VARARGS1 */
panic(msg, d1, d2)
        char *msg;
        long d1, d2;
{

        fprintf(stderr, msg, d1, d2);
        if (reply("abort") == GOOD) {
                if (reply("dump core") == GOOD)
                        abort();
                done(1);
        }
}