Return to dirs.c CVS log

Up to [CSRG BSD Unix] / 43BSDTahoe / etc / restore

Annotation of 43BSDTahoe/etc/restore/dirs.c, revision 1.1

1.1     ! root        1: /*
        !             2:  * Copyright (c) 1983 Regents of the University of California.
        !             3:  * All rights reserved.  The Berkeley software License Agreement
        !             4:  * specifies the terms and conditions for redistribution.
        !             5:  */
        !             6: 
        !             7: #ifndef lint
        !             8: static char sccsid[] = "@(#)dirs.c     5.5 (Berkeley) 5/13/88";
        !             9: #endif not lint
        !            10: 
        !            11: #include "restore.h"
        !            12: #include <protocols/dumprestore.h>
        !            13: #include <sys/file.h>
        !            14: 
        !            15: /*
        !            16:  * Symbol table of directories read from tape.
        !            17:  */
        !            18: #define HASHSIZE       1000
        !            19: #define INOHASH(val) (val % HASHSIZE)
        !            20: struct inotab {
        !            21:        struct inotab *t_next;
        !            22:        ino_t   t_ino;
        !            23:        daddr_t t_seekpt;
        !            24:        long t_size;
        !            25: };
        !            26: static struct inotab *inotab[HASHSIZE];
        !            27: extern struct inotab *inotablookup();
        !            28: extern struct inotab *allocinotab();
        !            29: 
        !            30: /*
        !            31:  * Information retained about directories.
        !            32:  */
        !            33: struct modeinfo {
        !            34:        ino_t ino;
        !            35:        time_t timep[2];
        !            36:        short mode;
        !            37:        short uid;
        !            38:        short gid;
        !            39: };
        !            40: 
        !            41: /*
        !            42:  * Global variables for this file.
        !            43:  */
        !            44: static daddr_t seekpt;
        !            45: static FILE    *df, *mf;
        !            46: static DIR     *dirp;
        !            47: static char    dirfile[32] = "#";      /* No file */
        !            48: static char    modefile[32] = "#";     /* No file */
        !            49: extern ino_t   search();
        !            50: struct direct  *rst_readdir();
        !            51: extern void    rst_seekdir();
        !            52: 
        !            53: /*
        !            54:  * Format of old style directories.
        !            55:  */
        !            56: #define ODIRSIZ 14
        !            57: struct odirect {
        !            58:        u_short d_ino;
        !            59:        char    d_name[ODIRSIZ];
        !            60: };
        !            61: 
        !            62: /*
        !            63:  *     Extract directory contents, building up a directory structure
        !            64:  *     on disk for extraction by name.
        !            65:  *     If genmode is requested, save mode, owner, and times for all
        !            66:  *     directories on the tape.
        !            67:  */
        !            68: extractdirs(genmode)
        !            69:        int genmode;
        !            70: {
        !            71:        register int i;
        !            72:        register struct dinode *ip;
        !            73:        struct inotab *itp;
        !            74:        struct direct nulldir;
        !            75:        int putdir(), null();
        !            76: 
        !            77:        vprintf(stdout, "Extract directories from tape\n");
        !            78:        (void) sprintf(dirfile, "/tmp/rstdir%d", dumpdate);
        !            79:        df = fopen(dirfile, "w");
        !            80:        if (df == 0) {
        !            81:                fprintf(stderr,
        !            82:                    "restore: %s - cannot create directory temporary\n",
        !            83:                    dirfile);
        !            84:                perror("fopen");
        !            85:                done(1);
        !            86:        }
        !            87:        if (genmode != 0) {
        !            88:                (void) sprintf(modefile, "/tmp/rstmode%d", dumpdate);
        !            89:                mf = fopen(modefile, "w");
        !            90:                if (mf == 0) {
        !            91:                        fprintf(stderr,
        !            92:                            "restore: %s - cannot create modefile \n",
        !            93:                            modefile);
        !            94:                        perror("fopen");
        !            95:                        done(1);
        !            96:                }
        !            97:        }
        !            98:        nulldir.d_ino = 0;
        !            99:        nulldir.d_namlen = 1;
        !           100:        (void) strcpy(nulldir.d_name, "/");
        !           101:        nulldir.d_reclen = DIRSIZ(&nulldir);
        !           102:        for (;;) {
        !           103:                curfile.name = "<directory file - name unknown>";
        !           104:                curfile.action = USING;
        !           105:                ip = curfile.dip;
        !           106:                if (ip == NULL || (ip->di_mode & IFMT) != IFDIR) {
        !           107:                        (void) fclose(df);
        !           108:                        dirp = opendir(dirfile);
        !           109:                        if (dirp == NULL)
        !           110:                                perror("opendir");
        !           111:                        if (mf != NULL)
        !           112:                                (void) fclose(mf);
        !           113:                        i = dirlookup(".");
        !           114:                        if (i == 0)
        !           115:                                panic("Root directory is not on tape\n");
        !           116:                        return;
        !           117:                }
        !           118:                itp = allocinotab(curfile.ino, ip, seekpt);
        !           119:                getfile(putdir, null);
        !           120:                putent(&nulldir);
        !           121:                flushent();
        !           122:                itp->t_size = seekpt - itp->t_seekpt;
        !           123:        }
        !           124: }
        !           125: 
        !           126: /*
        !           127:  * skip over all the directories on the tape
        !           128:  */
        !           129: skipdirs()
        !           130: {
        !           131: 
        !           132:        while ((curfile.dip->di_mode & IFMT) == IFDIR) {
        !           133:                skipfile();
        !           134:        }
        !           135: }
        !           136: 
        !           137: /*
        !           138:  *     Recursively find names and inumbers of all files in subtree 
        !           139:  *     pname and pass them off to be processed.
        !           140:  */
        !           141: treescan(pname, ino, todo)
        !           142:        char *pname;
        !           143:        ino_t ino;
        !           144:        long (*todo)();
        !           145: {
        !           146:        register struct inotab *itp;
        !           147:        register struct direct *dp;
        !           148:        register struct entry *np;
        !           149:        int namelen;
        !           150:        daddr_t bpt;
        !           151:        char locname[MAXPATHLEN + 1];
        !           152: 
        !           153:        itp = inotablookup(ino);
        !           154:        if (itp == NULL) {
        !           155:                /*
        !           156:                 * Pname is name of a simple file or an unchanged directory.
        !           157:                 */
        !           158:                (void) (*todo)(pname, ino, LEAF);
        !           159:                return;
        !           160:        }
        !           161:        /*
        !           162:         * Pname is a dumped directory name.
        !           163:         */
        !           164:        if ((*todo)(pname, ino, NODE) == FAIL)
        !           165:                return;
        !           166:        /*
        !           167:         * begin search through the directory
        !           168:         * skipping over "." and ".."
        !           169:         */
        !           170:        (void) strncpy(locname, pname, MAXPATHLEN);
        !           171:        (void) strncat(locname, "/", MAXPATHLEN);
        !           172:        namelen = strlen(locname);
        !           173:        rst_seekdir(dirp, itp->t_seekpt, itp->t_seekpt);
        !           174:        dp = rst_readdir(dirp); /* "." */
        !           175:        if (dp != NULL && strcmp(dp->d_name, ".") == 0)
        !           176:                dp = rst_readdir(dirp); /* ".." */
        !           177:        else
        !           178:                fprintf(stderr, "Warning: `.' missing from directory %s\n",
        !           179:                        pname);
        !           180:        if (dp != NULL && strcmp(dp->d_name, "..") == 0)
        !           181:                dp = rst_readdir(dirp); /* first real entry */
        !           182:        else
        !           183:                fprintf(stderr, "Warning: `..' missing from directory %s\n",
        !           184:                        pname);
        !           185:        bpt = telldir(dirp);
        !           186:        /*
        !           187:         * a zero inode signals end of directory
        !           188:         */
        !           189:        while (dp != NULL && dp->d_ino != 0) {
        !           190:                locname[namelen] = '\0';
        !           191:                if (namelen + dp->d_namlen >= MAXPATHLEN) {
        !           192:                        fprintf(stderr, "%s%s: name exceeds %d char\n",
        !           193:                                locname, dp->d_name, MAXPATHLEN);
        !           194:                } else {
        !           195:                        (void) strncat(locname, dp->d_name, (int)dp->d_namlen);
        !           196:                        treescan(locname, dp->d_ino, todo);
        !           197:                        rst_seekdir(dirp, bpt, itp->t_seekpt);
        !           198:                }
        !           199:                dp = rst_readdir(dirp);
        !           200:                bpt = telldir(dirp);
        !           201:        }
        !           202:        if (dp == NULL)
        !           203:                fprintf(stderr, "corrupted directory: %s.\n", locname);
        !           204: }
        !           205: 
        !           206: /*
        !           207:  * Search the directory tree rooted at inode ROOTINO
        !           208:  * for the path pointed at by n
        !           209:  */
        !           210: ino_t
        !           211: psearch(n)
        !           212:        char    *n;
        !           213: {
        !           214:        register char *cp, *cp1;
        !           215:        ino_t ino;
        !           216:        char c;
        !           217: 
        !           218:        ino = ROOTINO;
        !           219:        if (*(cp = n) == '/')
        !           220:                cp++;
        !           221: next:
        !           222:        cp1 = cp + 1;
        !           223:        while (*cp1 != '/' && *cp1)
        !           224:                cp1++;
        !           225:        c = *cp1;
        !           226:        *cp1 = 0;
        !           227:        ino = search(ino, cp);
        !           228:        if (ino == 0) {
        !           229:                *cp1 = c;
        !           230:                return(0);
        !           231:        }
        !           232:        *cp1 = c;
        !           233:        if (c == '/') {
        !           234:                cp = cp1+1;
        !           235:                goto next;
        !           236:        }
        !           237:        return(ino);
        !           238: }
        !           239: 
        !           240: /*
        !           241:  * search the directory inode ino
        !           242:  * looking for entry cp
        !           243:  */
        !           244: ino_t
        !           245: search(inum, cp)
        !           246:        ino_t   inum;
        !           247:        char    *cp;
        !           248: {
        !           249:        register struct direct *dp;
        !           250:        register struct inotab *itp;
        !           251:        int len;
        !           252: 
        !           253:        itp = inotablookup(inum);
        !           254:        if (itp == NULL)
        !           255:                return(0);
        !           256:        rst_seekdir(dirp, itp->t_seekpt, itp->t_seekpt);
        !           257:        len = strlen(cp);
        !           258:        do {
        !           259:                dp = rst_readdir(dirp);
        !           260:                if (dp == NULL || dp->d_ino == 0)
        !           261:                        return (0);
        !           262:        } while (dp->d_namlen != len || strncmp(dp->d_name, cp, len) != 0);
        !           263:        return(dp->d_ino);
        !           264: }
        !           265: 
        !           266: /*
        !           267:  * Put the directory entries in the directory file
        !           268:  */
        !           269: putdir(buf, size)
        !           270:        char *buf;
        !           271:        int size;
        !           272: {
        !           273:        struct direct cvtbuf;
        !           274:        register struct odirect *odp;
        !           275:        struct odirect *eodp;
        !           276:        register struct direct *dp;
        !           277:        long loc, i;
        !           278:        extern int Bcvt;
        !           279: 
        !           280:        if (cvtflag) {
        !           281:                eodp = (struct odirect *)&buf[size];
        !           282:                for (odp = (struct odirect *)buf; odp < eodp; odp++)
        !           283:                        if (odp->d_ino != 0) {
        !           284:                                dcvt(odp, &cvtbuf);
        !           285:                                putent(&cvtbuf);
        !           286:                        }
        !           287:        } else {
        !           288:                for (loc = 0; loc < size; ) {
        !           289:                        dp = (struct direct *)(buf + loc);
        !           290:                        if (Bcvt) {
        !           291:                                swabst("l2s", (char *) dp);
        !           292:                        }
        !           293:                        i = DIRBLKSIZ - (loc & (DIRBLKSIZ - 1));
        !           294:                        if (dp->d_reclen == 0 || dp->d_reclen > i) {
        !           295:                                loc += i;
        !           296:                                continue;
        !           297:                        }
        !           298:                        loc += dp->d_reclen;
        !           299:                        if (dp->d_ino != 0) {
        !           300:                                putent(dp);
        !           301:                        }
        !           302:                }
        !           303:        }
        !           304: }
        !           305: 
        !           306: /*
        !           307:  * These variables are "local" to the following two functions.
        !           308:  */
        !           309: char dirbuf[DIRBLKSIZ];
        !           310: long dirloc = 0;
        !           311: long prev = 0;
        !           312: 
        !           313: /*
        !           314:  * add a new directory entry to a file.
        !           315:  */
        !           316: putent(dp)
        !           317:        struct direct *dp;
        !           318: {
        !           319:        dp->d_reclen = DIRSIZ(dp);
        !           320:        if (dirloc + dp->d_reclen > DIRBLKSIZ) {
        !           321:                ((struct direct *)(dirbuf + prev))->d_reclen =
        !           322:                    DIRBLKSIZ - prev;
        !           323:                (void) fwrite(dirbuf, 1, DIRBLKSIZ, df);
        !           324:                dirloc = 0;
        !           325:        }
        !           326:        bcopy((char *)dp, dirbuf + dirloc, (long)dp->d_reclen);
        !           327:        prev = dirloc;
        !           328:        dirloc += dp->d_reclen;
        !           329: }
        !           330: 
        !           331: /*
        !           332:  * flush out a directory that is finished.
        !           333:  */
        !           334: flushent()
        !           335: {
        !           336: 
        !           337:        ((struct direct *)(dirbuf + prev))->d_reclen = DIRBLKSIZ - prev;
        !           338:        (void) fwrite(dirbuf, (int)dirloc, 1, df);
        !           339:        seekpt = ftell(df);
        !           340:        dirloc = 0;
        !           341: }
        !           342: 
        !           343: dcvt(odp, ndp)
        !           344:        register struct odirect *odp;
        !           345:        register struct direct *ndp;
        !           346: {
        !           347: 
        !           348:        bzero((char *)ndp, (long)(sizeof *ndp));
        !           349:        ndp->d_ino =  odp->d_ino;
        !           350:        (void) strncpy(ndp->d_name, odp->d_name, ODIRSIZ);
        !           351:        ndp->d_namlen = strlen(ndp->d_name);
        !           352:        ndp->d_reclen = DIRSIZ(ndp);
        !           353: }
        !           354: 
        !           355: /*
        !           356:  * Seek to an entry in a directory.
        !           357:  * Only values returned by ``telldir'' should be passed to rst_seekdir.
        !           358:  * This routine handles many directories in a single file.
        !           359:  * It takes the base of the directory in the file, plus
        !           360:  * the desired seek offset into it.
        !           361:  */
        !           362: void
        !           363: rst_seekdir(dirp, loc, base)
        !           364:        register DIR *dirp;
        !           365:        daddr_t loc, base;
        !           366: {
        !           367: 
        !           368:        if (loc == telldir(dirp))
        !           369:                return;
        !           370:        loc -= base;
        !           371:        if (loc < 0)
        !           372:                fprintf(stderr, "bad seek pointer to rst_seekdir %d\n", loc);
        !           373:        (void) lseek(dirp->dd_fd, base + (loc & ~(DIRBLKSIZ - 1)), 0);
        !           374:        dirp->dd_loc = loc & (DIRBLKSIZ - 1);
        !           375:        if (dirp->dd_loc != 0)
        !           376:                dirp->dd_size = read(dirp->dd_fd, dirp->dd_buf, DIRBLKSIZ);
        !           377: }
        !           378: 
        !           379: /*
        !           380:  * get next entry in a directory.
        !           381:  */
        !           382: struct direct *
        !           383: rst_readdir(dirp)
        !           384:        register DIR *dirp;
        !           385: {
        !           386:        register struct direct *dp;
        !           387: 
        !           388:        for (;;) {
        !           389:                if (dirp->dd_loc == 0) {
        !           390:                        dirp->dd_size = read(dirp->dd_fd, dirp->dd_buf, 
        !           391:                            DIRBLKSIZ);
        !           392:                        if (dirp->dd_size <= 0) {
        !           393:                                dprintf(stderr, "error reading directory\n");
        !           394:                                return NULL;
        !           395:                        }
        !           396:                }
        !           397:                if (dirp->dd_loc >= dirp->dd_size) {
        !           398:                        dirp->dd_loc = 0;
        !           399:                        continue;
        !           400:                }
        !           401:                dp = (struct direct *)(dirp->dd_buf + dirp->dd_loc);
        !           402:                if (dp->d_reclen == 0 ||
        !           403:                    dp->d_reclen > DIRBLKSIZ + 1 - dirp->dd_loc) {
        !           404:                        dprintf(stderr, "corrupted directory: bad reclen %d\n",
        !           405:                                dp->d_reclen);
        !           406:                        return NULL;
        !           407:                }
        !           408:                dirp->dd_loc += dp->d_reclen;
        !           409:                if (dp->d_ino == 0 && strcmp(dp->d_name, "/") != 0)
        !           410:                        continue;
        !           411:                if (dp->d_ino >= maxino) {
        !           412:                        dprintf(stderr, "corrupted directory: bad inum %d\n",
        !           413:                                dp->d_ino);
        !           414:                        continue;
        !           415:                }
        !           416:                return (dp);
        !           417:        }
        !           418: }
        !           419: 
        !           420: /*
        !           421:  * Simulate the opening of a directory
        !           422:  */
        !           423: DIR *
        !           424: rst_opendir(name)
        !           425:        char *name;
        !           426: {
        !           427:        struct inotab *itp;
        !           428:        ino_t ino;
        !           429: 
        !           430:        if ((ino = dirlookup(name)) > 0 &&
        !           431:            (itp = inotablookup(ino)) != NULL) {
        !           432:                rst_seekdir(dirp, itp->t_seekpt, itp->t_seekpt);
        !           433:                return (dirp);
        !           434:        }
        !           435:        return (0);
        !           436: }
        !           437: 
        !           438: /*
        !           439:  * Set the mode, owner, and times for all new or changed directories
        !           440:  */
        !           441: setdirmodes()
        !           442: {
        !           443:        FILE *mf;
        !           444:        struct modeinfo node;
        !           445:        struct entry *ep;
        !           446:        char *cp;
        !           447:        
        !           448:        vprintf(stdout, "Set directory mode, owner, and times.\n");
        !           449:        (void) sprintf(modefile, "/tmp/rstmode%d", dumpdate);
        !           450:        mf = fopen(modefile, "r");
        !           451:        if (mf == NULL) {
        !           452:                perror("fopen");
        !           453:                fprintf(stderr, "cannot open mode file %s\n", modefile);
        !           454:                fprintf(stderr, "directory mode, owner, and times not set\n");
        !           455:                return;
        !           456:        }
        !           457:        clearerr(mf);
        !           458:        for (;;) {
        !           459:                (void) fread((char *)&node, 1, sizeof(struct modeinfo), mf);
        !           460:                if (feof(mf))
        !           461:                        break;
        !           462:                ep = lookupino(node.ino);
        !           463:                if (command == 'i' || command == 'x') {
        !           464:                        if (ep == NIL)
        !           465:                                continue;
        !           466:                        if (ep->e_flags & EXISTED) {
        !           467:                                ep->e_flags &= ~NEW;
        !           468:                                continue;
        !           469:                        }
        !           470:                        if (node.ino == ROOTINO &&
        !           471:                            reply("set owner/mode for '.'") == FAIL)
        !           472:                                continue;
        !           473:                }
        !           474:                if (ep == NIL)
        !           475:                        panic("cannot find directory inode %d\n", node.ino);
        !           476:                cp = myname(ep);
        !           477:                (void) chown(cp, node.uid, node.gid);
        !           478:                (void) chmod(cp, node.mode);
        !           479:                utime(cp, node.timep);
        !           480:                ep->e_flags &= ~NEW;
        !           481:        }
        !           482:        if (ferror(mf))
        !           483:                panic("error setting directory modes\n");
        !           484:        (void) fclose(mf);
        !           485: }
        !           486: 
        !           487: /*
        !           488:  * Generate a literal copy of a directory.
        !           489:  */
        !           490: genliteraldir(name, ino)
        !           491:        char *name;
        !           492:        ino_t ino;
        !           493: {
        !           494:        register struct inotab *itp;
        !           495:        int ofile, dp, i, size;
        !           496:        char buf[BUFSIZ];
        !           497: 
        !           498:        itp = inotablookup(ino);
        !           499:        if (itp == NULL)
        !           500:                panic("Cannot find directory inode %d named %s\n", ino, name);
        !           501:        if ((ofile = creat(name, 0666)) < 0) {
        !           502:                fprintf(stderr, "%s: ", name);
        !           503:                (void) fflush(stderr);
        !           504:                perror("cannot create file");
        !           505:                return (FAIL);
        !           506:        }
        !           507:        rst_seekdir(dirp, itp->t_seekpt, itp->t_seekpt);
        !           508:        dp = dup(dirp->dd_fd);
        !           509:        for (i = itp->t_size; i > 0; i -= BUFSIZ) {
        !           510:                size = i < BUFSIZ ? i : BUFSIZ;
        !           511:                if (read(dp, buf, (int) size) == -1) {
        !           512:                        fprintf(stderr,
        !           513:                                "write error extracting inode %d, name %s\n",
        !           514:                                curfile.ino, curfile.name);
        !           515:                        perror("read");
        !           516:                        done(1);
        !           517:                }
        !           518:                if (!Nflag && write(ofile, buf, (int) size) == -1) {
        !           519:                        fprintf(stderr,
        !           520:                                "write error extracting inode %d, name %s\n",
        !           521:                                curfile.ino, curfile.name);
        !           522:                        perror("write");
        !           523:                        done(1);
        !           524:                }
        !           525:        }
        !           526:        (void) close(dp);
        !           527:        (void) close(ofile);
        !           528:        return (GOOD);
        !           529: }
        !           530: 
        !           531: /*
        !           532:  * Determine the type of an inode
        !           533:  */
        !           534: inodetype(ino)
        !           535:        ino_t ino;
        !           536: {
        !           537:        struct inotab *itp;
        !           538: 
        !           539:        itp = inotablookup(ino);
        !           540:        if (itp == NULL)
        !           541:                return (LEAF);
        !           542:        return (NODE);
        !           543: }
        !           544: 
        !           545: /*
        !           546:  * Allocate and initialize a directory inode entry.
        !           547:  * If requested, save its pertinent mode, owner, and time info.
        !           548:  */
        !           549: struct inotab *
        !           550: allocinotab(ino, dip, seekpt)
        !           551:        ino_t ino;
        !           552:        struct dinode *dip;
        !           553:        daddr_t seekpt;
        !           554: {
        !           555:        register struct inotab  *itp;
        !           556:        struct modeinfo node;
        !           557: 
        !           558:        itp = (struct inotab *)calloc(1, sizeof(struct inotab));
        !           559:        if (itp == 0)
        !           560:                panic("no memory directory table\n");
        !           561:        itp->t_next = inotab[INOHASH(ino)];
        !           562:        inotab[INOHASH(ino)] = itp;
        !           563:        itp->t_ino = ino;
        !           564:        itp->t_seekpt = seekpt;
        !           565:        if (mf == NULL)
        !           566:                return(itp);
        !           567:        node.ino = ino;
        !           568:        node.timep[0] = dip->di_atime;
        !           569:        node.timep[1] = dip->di_mtime;
        !           570:        node.mode = dip->di_mode;
        !           571:        node.uid = dip->di_uid;
        !           572:        node.gid = dip->di_gid;
        !           573:        (void) fwrite((char *)&node, 1, sizeof(struct modeinfo), mf);
        !           574:        return(itp);
        !           575: }
        !           576: 
        !           577: /*
        !           578:  * Look up an inode in the table of directories
        !           579:  */
        !           580: struct inotab *
        !           581: inotablookup(ino)
        !           582:        ino_t   ino;
        !           583: {
        !           584:        register struct inotab *itp;
        !           585: 
        !           586:        for (itp = inotab[INOHASH(ino)]; itp != NULL; itp = itp->t_next)
        !           587:                if (itp->t_ino == ino)
        !           588:                        return(itp);
        !           589:        return ((struct inotab *)0);
        !           590: }
        !           591: 
        !           592: /*
        !           593:  * Clean up and exit
        !           594:  */
        !           595: done(exitcode)
        !           596:        int exitcode;
        !           597: {
        !           598: 
        !           599:        closemt();
        !           600:        if (modefile[0] != '#')
        !           601:                (void) unlink(modefile);
        !           602:        if (dirfile[0] != '#')
        !           603:                (void) unlink(dirfile);
        !           604:        exit(exitcode);
        !           605: }

This archive runs on limited infrastructure. Preserving old code on modern bandwidth. Automated agents are requested to crawl responsibly.