sbin/restore/tape.c

   1 /*
   2  * Copyright (c) 1983, 1993
   3  *      The Regents of the University of California.  All rights reserved.
   4  * (c) UNIX System Laboratories, Inc.
   5  * All or some portions of this file are derived from material licensed
   6  * to the University of California by American Telephone and Telegraph
   7  * Co. or Unix System Laboratories, Inc. and are reproduced herein with
   8  * the permission of UNIX System Laboratories, Inc.
   9  *
  10  * Redistribution and use in source and binary forms, with or without
  11  * modification, are permitted provided that the following conditions
  12  * are met:
  13  * 1. Redistributions of source code must retain the above copyright
  14  *    notice, this list of conditions and the following disclaimer.
  15  * 2. Redistributions in binary form must reproduce the above copyright
  16  *    notice, this list of conditions and the following disclaimer in the
  17  *    documentation and/or other materials provided with the distribution.
  18  * 4. Neither the name of the University nor the names of its contributors
  19  *    may be used to endorse or promote products derived from this software
  20  *    without specific prior written permission.
  21  *
  22  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  23  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  24  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  25  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  26  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  27  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  28  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  29  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  30  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  31  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  32  * SUCH DAMAGE.
  33  */
  34
  35 #ifndef lint
  36 #if 0
  37 static char sccsid[] = "@(#)tape.c      8.9 (Berkeley) 5/1/95";
  38 #endif
  39 #endif /* not lint */
  40
  41 #include <sys/cdefs.h>
  42 __FBSDID("$FreeBSD$");
  43
  44 #include <sys/param.h>
  45 #include <sys/file.h>
  46 #include <sys/mtio.h>
  47 #include <sys/stat.h>
  48 #include <sys/time.h>
  49 #include <sys/extattr.h>
  50 #include <sys/acl.h>
  51
  52 #include <ufs/ufs/extattr.h>
  53 #include <ufs/ufs/dinode.h>
  54 #include <protocols/dumprestore.h>
  55
  56 #include <errno.h>
  57 #include <limits.h>
  58 #include <paths.h>
  59 #include <setjmp.h>
  60 #include <stdint.h>
  61 #include <stdio.h>
  62 #include <stdlib.h>
  63 #include <string.h>
  64 #include <time.h>
  65 #include <timeconv.h>
  66 #include <unistd.h>
  67
  68 #include "restore.h"
  69 #include "extern.h"
  70
  71 static long     fssize = MAXBSIZE;
  72 static int      mt = -1;
  73 static int      pipein = 0;
  74 static int      pipecmdin = 0;
  75 static FILE     *popenfp = NULL;
  76 static char     *magtape;
  77 static int      blkcnt;
  78 static int      numtrec;
  79 static char     *tapebuf;
  80 static union    u_spcl endoftapemark;
  81 static long     byteslide = 0;
  82 static long     blksread;               /* blocks read since last header */
  83 static int64_t  tapeaddr = 0;           /* current TP_BSIZE tape record */
  84 static long     tapesread;
  85 static jmp_buf  restart;
  86 static int      gettingfile = 0;        /* restart has a valid frame */
  87 static char     *host = NULL;
  88 static int      readmapflag;
  89
  90 static int      ofile;
  91 static char     *map;
  92 static char     lnkbuf[MAXPATHLEN + 1];
  93 static int      pathlen;
  94
  95 int             Bcvt;           /* Swap Bytes */
  96 int             oldinofmt;      /* FreeBSD 1 inode format needs cvt */
  97
  98 #define FLUSHTAPEBUF()  blkcnt = ntrec + 1
  99
 100 char *namespace_names[] = EXTATTR_NAMESPACE_NAMES;
 101
 102 static void      accthdr(struct s_spcl *);
 103 static int       checksum(int *);
 104 static void      findinode(struct s_spcl *);
 105 static void      findtapeblksize(void);
 106 static char     *setupextattr(int);
 107 static void      xtrattr(char *, long);
 108 static void      set_extattr_link(char *, void *, int);
 109 static void      set_extattr_fd(int, char *, void *, int);
 110 static int       gethead(struct s_spcl *);
 111 static void      readtape(char *);
 112 static void      setdumpnum(void);
 113 static u_long    swabl(u_long);
 114 static u_char   *swablong(u_char *, int);
 115 static u_char   *swabshort(u_char *, int);
 116 static void      terminateinput(void);
 117 static void      xtrfile(char *, long);
 118 static void      xtrlnkfile(char *, long);
 119 static void      xtrlnkskip(char *, long);
 120 static void      xtrmap(char *, long);
 121 static void      xtrmapskip(char *, long);
 122 static void      xtrskip(char *, long);
 123
 124 /*
 125  * Set up an input source
 126  */
 127 void
 128 setinput(char *source, int ispipecommand)
 129 {
 130         FLUSHTAPEBUF();
 131         if (bflag)
 132                 newtapebuf(ntrec);
 133         else
 134                 newtapebuf(NTREC > HIGHDENSITYTREC ? NTREC : HIGHDENSITYTREC);
 135         terminal = stdin;
 136
 137         if (ispipecommand)
 138                 pipecmdin++;
 139         else
 140 #ifdef RRESTORE
 141         if (strchr(source, ':')) {
 142                 host = source;
 143                 source = strchr(host, ':');
 144                 *source++ = '\0';
 145                 if (rmthost(host) == 0)
 146                         done(1);
 147         } else
 148 #endif
 149         if (strcmp(source, "-") == 0) {
 150                 /*
 151                  * Since input is coming from a pipe we must establish
 152                  * our own connection to the terminal.
 153                  */
 154                 terminal = fopen(_PATH_TTY, "r");
 155                 if (terminal == NULL) {
 156                         (void)fprintf(stderr, "cannot open %s: %s\n",
 157                             _PATH_TTY, strerror(errno));
 158                         terminal = fopen(_PATH_DEVNULL, "r");
 159                         if (terminal == NULL) {
 160                                 (void)fprintf(stderr, "cannot open %s: %s\n",
 161                                     _PATH_DEVNULL, strerror(errno));
 162                                 done(1);
 163                         }
 164                 }
 165                 pipein++;
 166         }
 167         setuid(getuid());       /* no longer need or want root privileges */
 168         magtape = strdup(source);
 169         if (magtape == NULL) {
 170                 fprintf(stderr, "Cannot allocate space for magtape buffer\n");
 171                 done(1);
 172         }
 173 }
 174
 175 void
 176 newtapebuf(long size)
 177 {
 178         static int tapebufsize = -1;
 179
 180         ntrec = size;
 181         if (size <= tapebufsize)
 182                 return;
 183         if (tapebuf != NULL)
 184                 free(tapebuf - TP_BSIZE);
 185         tapebuf = malloc((size+1) * TP_BSIZE);
 186         if (tapebuf == NULL) {
 187                 fprintf(stderr, "Cannot allocate space for tape buffer\n");
 188                 done(1);
 189         }
 190         tapebuf += TP_BSIZE;
 191         tapebufsize = size;
 192 }
 193
 194 /*
 195  * Verify that the tape drive can be accessed and
 196  * that it actually is a dump tape.
 197  */
 198 void
 199 setup(void)
 200 {
 201         int i, j, *ip;
 202         struct stat stbuf;
 203
 204         vprintf(stdout, "Verify tape and initialize maps\n");
 205         if (pipecmdin) {
 206                 if (setenv("RESTORE_VOLUME", "1", 1) == -1) {
 207                         fprintf(stderr, "Cannot set $RESTORE_VOLUME: %s\n",
 208                             strerror(errno));
 209                         done(1);
 210                 }
 211                 popenfp = popen(magtape, "r");
 212                 mt = popenfp ? fileno(popenfp) : -1;
 213         } else
 214 #ifdef RRESTORE
 215         if (host)
 216                 mt = rmtopen(magtape, 0);
 217         else
 218 #endif
 219         if (pipein)
 220                 mt = 0;
 221         else
 222                 mt = open(magtape, O_RDONLY, 0);
 223         if (mt < 0) {
 224                 fprintf(stderr, "%s: %s\n", magtape, strerror(errno));
 225                 done(1);
 226         }
 227         volno = 1;
 228         setdumpnum();
 229         FLUSHTAPEBUF();
 230         if (!pipein && !pipecmdin && !bflag)
 231                 findtapeblksize();
 232         if (gethead(&spcl) == FAIL) {
 233                 fprintf(stderr, "Tape is not a dump tape\n");
 234                 done(1);
 235         }
 236         if (pipein) {
 237                 endoftapemark.s_spcl.c_magic = FS_UFS2_MAGIC;
 238                 endoftapemark.s_spcl.c_type = TS_END;
 239                 ip = (int *)&endoftapemark;
 240                 j = sizeof(union u_spcl) / sizeof(int);
 241                 i = 0;
 242                 do
 243                         i += *ip++;
 244                 while (--j);
 245                 endoftapemark.s_spcl.c_checksum = CHECKSUM - i;
 246         }
 247         if (vflag || command == 't')
 248                 printdumpinfo();
 249         dumptime = _time64_to_time(spcl.c_ddate);
 250         dumpdate = _time64_to_time(spcl.c_date);
 251         if (stat(".", &stbuf) < 0) {
 252                 fprintf(stderr, "cannot stat .: %s\n", strerror(errno));
 253                 done(1);
 254         }
 255         if (stbuf.st_blksize > 0 && stbuf.st_blksize < TP_BSIZE )
 256                 fssize = TP_BSIZE;
 257         if (stbuf.st_blksize >= TP_BSIZE && stbuf.st_blksize <= MAXBSIZE)
 258                 fssize = stbuf.st_blksize;
 259         if (((fssize - 1) & fssize) != 0) {
 260                 fprintf(stderr, "bad block size %ld\n", fssize);
 261                 done(1);
 262         }
 263         if (spcl.c_volume != 1) {
 264                 fprintf(stderr, "Tape is not volume 1 of the dump\n");
 265                 done(1);
 266         }
 267         if (gethead(&spcl) == FAIL) {
 268                 dprintf(stdout, "header read failed at %ld blocks\n", blksread);
 269                 panic("no header after volume mark!\n");
 270         }
 271         findinode(&spcl);
 272         if (spcl.c_type != TS_CLRI) {
 273                 fprintf(stderr, "Cannot find file removal list\n");
 274                 done(1);
 275         }
 276         maxino = (spcl.c_count * TP_BSIZE * NBBY) + 1;
 277         dprintf(stdout, "maxino = %d\n", maxino);
 278         map = calloc((unsigned)1, (unsigned)howmany(maxino, NBBY));
 279         if (map == NULL)
 280                 panic("no memory for active inode map\n");
 281         usedinomap = map;
 282         curfile.action = USING;
 283         getfile(xtrmap, xtrmapskip, xtrmapskip);
 284         if (spcl.c_type != TS_BITS) {
 285                 fprintf(stderr, "Cannot find file dump list\n");
 286                 done(1);
 287         }
 288         map = calloc((unsigned)1, (unsigned)howmany(maxino, NBBY));
 289         if (map == (char *)NULL)
 290                 panic("no memory for file dump list\n");
 291         dumpmap = map;
 292         curfile.action = USING;
 293         getfile(xtrmap, xtrmapskip, xtrmapskip);
 294         /*
 295          * If there may be whiteout entries on the tape, pretend that the
 296          * whiteout inode exists, so that the whiteout entries can be
 297          * extracted.
 298          */
 299         SETINO(WINO, dumpmap);
 300         /* 'r' restores don't call getvol() for tape 1, so mark it as read. */
 301         if (command == 'r')
 302                 tapesread = 1;
 303 }
 304
 305 /*
 306  * Prompt user to load a new dump volume.
 307  * "Nextvol" is the next suggested volume to use.
 308  * This suggested volume is enforced when doing full
 309  * or incremental restores, but can be overridden by
 310  * the user when only extracting a subset of the files.
 311  */
 312 void
 313 getvol(long nextvol)
 314 {
 315         int64_t prevtapea;
 316         long i, newvol, savecnt;
 317         union u_spcl tmpspcl;
 318 #       define tmpbuf tmpspcl.s_spcl
 319         char buf[TP_BSIZE];
 320
 321         if (nextvol == 1) {
 322                 tapesread = 0;
 323                 gettingfile = 0;
 324         }
 325         prevtapea = tapeaddr;
 326         savecnt = blksread;
 327         if (pipein) {
 328                 if (nextvol != 1) {
 329                         panic("Changing volumes on pipe input?\n");
 330                         /* Avoid looping if we couldn't ask the user. */
 331                         if (yflag || ferror(terminal) || feof(terminal))
 332                                 done(1);
 333                 }
 334                 if (volno == 1)
 335                         return;
 336                 goto gethdr;
 337         }
 338 again:
 339         if (pipein)
 340                 done(1); /* pipes do not get a second chance */
 341         if (command == 'R' || command == 'r' || curfile.action != SKIP)
 342                 newvol = nextvol;
 343         else
 344                 newvol = 0;
 345         while (newvol <= 0) {
 346                 if (tapesread == 0) {
 347                         fprintf(stderr, "%s%s%s%s%s%s%s",
 348                             "You have not read any tapes yet.\n",
 349                             "If you are extracting just a few files,",
 350                             " start with the last volume\n",
 351                             "and work towards the first; restore",
 352                             " can quickly skip tapes that\n",
 353                             "have no further files to extract.",
 354                             " Otherwise, begin with volume 1.\n");
 355                 } else {
 356                         fprintf(stderr, "You have read volumes");
 357                         strcpy(buf, ": ");
 358                         for (i = 0; i < 32; i++)
 359                                 if (tapesread & (1 << i)) {
 360                                         fprintf(stderr, "%s%ld", buf, i + 1);
 361                                         strcpy(buf, ", ");
 362                                 }
 363                         fprintf(stderr, "\n");
 364                 }
 365                 do      {
 366                         fprintf(stderr, "Specify next volume #: ");
 367                         (void) fflush(stderr);
 368                         if (fgets(buf, BUFSIZ, terminal) == NULL)
 369                                 done(1);
 370                 } while (buf[0] == '\n');
 371                 newvol = atoi(buf);
 372                 if (newvol <= 0) {
 373                         fprintf(stderr,
 374                             "Volume numbers are positive numerics\n");
 375                 }
 376         }
 377         if (newvol == volno) {
 378                 tapesread |= 1 << (volno - 1);
 379                 return;
 380         }
 381         closemt();
 382         fprintf(stderr, "Mount tape volume %ld\n", newvol);
 383         fprintf(stderr, "Enter ``none'' if there are no more tapes\n");
 384         fprintf(stderr, "otherwise enter tape name (default: %s) ", magtape);
 385         (void) fflush(stderr);
 386         if (fgets(buf, BUFSIZ, terminal) == NULL)
 387                 done(1);
 388         if (!strcmp(buf, "none\n")) {
 389                 terminateinput();
 390                 return;
 391         }
 392         if (buf[0] != '\n') {
 393                 (void) strcpy(magtape, buf);
 394                 magtape[strlen(magtape) - 1] = '\0';
 395         }
 396         if (pipecmdin) {
 397                 char volno[sizeof("2147483647")];
 398
 399                 (void)sprintf(volno, "%ld", newvol);
 400                 if (setenv("RESTORE_VOLUME", volno, 1) == -1) {
 401                         fprintf(stderr, "Cannot set $RESTORE_VOLUME: %s\n",
 402                             strerror(errno));
 403                         done(1);
 404                 }
 405                 popenfp = popen(magtape, "r");
 406                 mt = popenfp ? fileno(popenfp) : -1;
 407         } else
 408 #ifdef RRESTORE
 409         if (host)
 410                 mt = rmtopen(magtape, 0);
 411         else
 412 #endif
 413                 mt = open(magtape, O_RDONLY, 0);
 414
 415         if (mt == -1) {
 416                 fprintf(stderr, "Cannot open %s\n", magtape);
 417                 volno = -1;
 418                 goto again;
 419         }
 420 gethdr:
 421         volno = newvol;
 422         setdumpnum();
 423         FLUSHTAPEBUF();
 424         if (gethead(&tmpbuf) == FAIL) {
 425                 dprintf(stdout, "header read failed at %ld blocks\n", blksread);
 426                 fprintf(stderr, "tape is not dump tape\n");
 427                 volno = 0;
 428                 goto again;
 429         }
 430         if (tmpbuf.c_volume != volno) {
 431                 fprintf(stderr, "Wrong volume (%jd)\n",
 432                     (intmax_t)tmpbuf.c_volume);
 433                 volno = 0;
 434                 goto again;
 435         }
 436         if (_time64_to_time(tmpbuf.c_date) != dumpdate ||
 437             _time64_to_time(tmpbuf.c_ddate) != dumptime) {
 438                 time_t t = _time64_to_time(tmpbuf.c_date);
 439                 fprintf(stderr, "Wrong dump date\n\tgot: %s", ctime(&t));
 440                 fprintf(stderr, "\twanted: %s", ctime(&dumpdate));
 441                 volno = 0;
 442                 goto again;
 443         }
 444         tapesread |= 1 << (volno - 1);
 445         blksread = savecnt;
 446         /*
 447          * If continuing from the previous volume, skip over any
 448          * blocks read already at the end of the previous volume.
 449          *
 450          * If coming to this volume at random, skip to the beginning
 451          * of the next record.
 452          */
 453         dprintf(stdout, "last rec %jd, tape starts with %jd\n",
 454             (intmax_t)prevtapea, (intmax_t)tmpbuf.c_tapea);
 455         if (tmpbuf.c_type == TS_TAPE) {
 456                 if (curfile.action != USING) {
 457                         /*
 458                          * XXX Dump incorrectly sets c_count to 1 in the
 459                          * volume header of the first tape, so ignore
 460                          * c_count when volno == 1.
 461                          */
 462                         if (volno != 1)
 463                                 for (i = tmpbuf.c_count; i > 0; i--)
 464                                         readtape(buf);
 465                 } else if (tmpbuf.c_tapea <= prevtapea) {
 466                         /*
 467                          * Normally the value of c_tapea in the volume
 468                          * header is the record number of the header itself.
 469                          * However in the volume header following an EOT-
 470                          * terminated tape, it is the record number of the
 471                          * first continuation data block (dump bug?).
 472                          *
 473                          * The next record we want is `prevtapea + 1'.
 474                          */
 475                         i = prevtapea + 1 - tmpbuf.c_tapea;
 476                         dprintf(stderr, "Skipping %ld duplicate record%s.\n",
 477                                 i, i > 1 ? "s" : "");
 478                         while (--i >= 0)
 479                                 readtape(buf);
 480                 }
 481         }
 482         if (curfile.action == USING) {
 483                 if (volno == 1)
 484                         panic("active file into volume 1\n");
 485                 return;
 486         }
 487         (void) gethead(&spcl);
 488         findinode(&spcl);
 489         if (gettingfile) {
 490                 gettingfile = 0;
 491                 longjmp(restart, 1);
 492         }
 493 }
 494
 495 /*
 496  * Handle unexpected EOF.
 497  */
 498 static void
 499 terminateinput(void)
 500 {
 501
 502         if (gettingfile && curfile.action == USING) {
 503                 printf("Warning: %s %s\n",
 504                     "End-of-input encountered while extracting", curfile.name);
 505         }
 506         curfile.name = "<name unknown>";
 507         curfile.action = UNKNOWN;
 508         curfile.mode = 0;
 509         curfile.ino = maxino;
 510         if (gettingfile) {
 511                 gettingfile = 0;
 512                 longjmp(restart, 1);
 513         }
 514 }
 515
 516 /*
 517  * handle multiple dumps per tape by skipping forward to the
 518  * appropriate one.
 519  */
 520 static void
 521 setdumpnum(void)
 522 {
 523         struct mtop tcom;
 524
 525         if (dumpnum == 1 || volno != 1)
 526                 return;
 527         if (pipein) {
 528                 fprintf(stderr, "Cannot have multiple dumps on pipe input\n");
 529                 done(1);
 530         }
 531         tcom.mt_op = MTFSF;
 532         tcom.mt_count = dumpnum - 1;
 533 #ifdef RRESTORE
 534         if (host)
 535                 rmtioctl(MTFSF, dumpnum - 1);
 536         else
 537 #endif
 538                 if (!pipecmdin && ioctl(mt, MTIOCTOP, (char *)&tcom) < 0)
 539                         fprintf(stderr, "ioctl MTFSF: %s\n", strerror(errno));
 540 }
 541
 542 void
 543 printdumpinfo(void)
 544 {
 545         time_t t;
 546         t = _time64_to_time(spcl.c_date);
 547         fprintf(stdout, "Dump   date: %s", ctime(&t));
 548         t = _time64_to_time(spcl.c_ddate);
 549         fprintf(stdout, "Dumped from: %s",
 550             (spcl.c_ddate == 0) ? "the epoch\n" : ctime(&t));
 551         if (spcl.c_host[0] == '\0')
 552                 return;
 553         fprintf(stderr, "Level %jd dump of %s on %s:%s\n",
 554             (intmax_t)spcl.c_level, spcl.c_filesys, spcl.c_host, spcl.c_dev);
 555         fprintf(stderr, "Label: %s\n", spcl.c_label);
 556 }
 557
 558 int
 559 extractfile(char *name)
 560 {
 561         int flags;
 562         uid_t uid;
 563         gid_t gid;
 564         mode_t mode;
 565         int extsize;
 566         struct timeval mtimep[2], ctimep[2];
 567         struct entry *ep;
 568         char *buf;
 569
 570         curfile.name = name;
 571         curfile.action = USING;
 572         mtimep[0].tv_sec = curfile.atime_sec;
 573         mtimep[0].tv_usec = curfile.atime_nsec / 1000;
 574         mtimep[1].tv_sec = curfile.mtime_sec;
 575         mtimep[1].tv_usec = curfile.mtime_nsec / 1000;
 576         ctimep[0].tv_sec = curfile.atime_sec;
 577         ctimep[0].tv_usec = curfile.atime_nsec / 1000;
 578         ctimep[1].tv_sec = curfile.birthtime_sec;
 579         ctimep[1].tv_usec = curfile.birthtime_nsec / 1000;
 580         extsize = curfile.extsize;
 581         uid = getuid();
 582         if (uid == 0)
 583                 uid = curfile.uid;
 584         gid = curfile.gid;
 585         mode = curfile.mode;
 586         flags = curfile.file_flags;
 587         switch (mode & IFMT) {
 588
 589         default:
 590                 fprintf(stderr, "%s: unknown file mode 0%o\n", name, mode);
 591                 skipfile();
 592                 return (FAIL);
 593
 594         case IFSOCK:
 595                 vprintf(stdout, "skipped socket %s\n", name);
 596                 skipfile();
 597                 return (GOOD);
 598
 599         case IFDIR:
 600                 if (mflag) {
 601                         ep = lookupname(name);
 602                         if (ep == NULL || ep->e_flags & EXTRACT)
 603                                 panic("unextracted directory %s\n", name);
 604                         skipfile();
 605                         return (GOOD);
 606                 }
 607                 vprintf(stdout, "extract file %s\n", name);
 608                 return (genliteraldir(name, curfile.ino));
 609
 610         case IFLNK:
 611                 lnkbuf[0] = '\0';
 612                 pathlen = 0;
 613                 buf = setupextattr(extsize);
 614                 getfile(xtrlnkfile, xtrattr, xtrlnkskip);
 615                 if (pathlen == 0) {
 616                         vprintf(stdout,
 617                             "%s: zero length symbolic link (ignored)\n", name);
 618                         return (GOOD);
 619                 }
 620                 if (linkit(lnkbuf, name, SYMLINK) == GOOD) {
 621                         if (extsize > 0)
 622                                 set_extattr_link(name, buf, extsize);
 623                         (void) lchown(name, uid, gid);
 624                         (void) lchmod(name, mode);
 625                         (void) lutimes(name, ctimep);
 626                         (void) lutimes(name, mtimep);
 627                         (void) lchflags(name, flags);
 628                         return (GOOD);
 629                 }
 630                 return (FAIL);
 631
 632         case IFIFO:
 633                 vprintf(stdout, "extract fifo %s\n", name);
 634                 if (Nflag) {
 635                         skipfile();
 636                         return (GOOD);
 637                 }
 638                 if (uflag)
 639                         (void) unlink(name);
 640                 if (mkfifo(name, 0600) < 0) {
 641                         fprintf(stderr, "%s: cannot create fifo: %s\n",
 642                             name, strerror(errno));
 643                         skipfile();
 644                         return (FAIL);
 645                 }
 646                 if (extsize == 0) {
 647                         skipfile();
 648                 } else {
 649                         buf = setupextattr(extsize);
 650                         getfile(xtrnull, xtrattr, xtrnull);
 651                         set_extattr_file(name, buf, extsize);
 652                 }
 653                 (void) chown(name, uid, gid);
 654                 (void) chmod(name, mode);
 655                 (void) utimes(name, ctimep);
 656                 (void) utimes(name, mtimep);
 657                 (void) chflags(name, flags);
 658                 return (GOOD);
 659
 660         case IFCHR:
 661         case IFBLK:
 662                 vprintf(stdout, "extract special file %s\n", name);
 663                 if (Nflag) {
 664                         skipfile();
 665                         return (GOOD);
 666                 }
 667                 if (uflag)
 668                         (void) unlink(name);
 669                 if (mknod(name, (mode & (IFCHR | IFBLK)) | 0600,
 670                     (int)curfile.rdev) < 0) {
 671                         fprintf(stderr, "%s: cannot create special file: %s\n",
 672                             name, strerror(errno));
 673                         skipfile();
 674                         return (FAIL);
 675                 }
 676                 if (extsize == 0) {
 677                         skipfile();
 678                 } else {
 679                         buf = setupextattr(extsize);
 680                         getfile(xtrnull, xtrattr, xtrnull);
 681                         set_extattr_file(name, buf, extsize);
 682                 }
 683                 (void) chown(name, uid, gid);
 684                 (void) chmod(name, mode);
 685                 (void) utimes(name, ctimep);
 686                 (void) utimes(name, mtimep);
 687                 (void) chflags(name, flags);
 688                 return (GOOD);
 689
 690         case IFREG:
 691                 vprintf(stdout, "extract file %s\n", name);
 692                 if (Nflag) {
 693                         skipfile();
 694                         return (GOOD);
 695                 }
 696                 if (uflag)
 697                         (void) unlink(name);
 698                 if ((ofile = open(name, O_WRONLY | O_CREAT | O_TRUNC,
 699                     0600)) < 0) {
 700                         fprintf(stderr, "%s: cannot create file: %s\n",
 701                             name, strerror(errno));
 702                         skipfile();
 703                         return (FAIL);
 704                 }
 705                 buf = setupextattr(extsize);
 706                 getfile(xtrfile, xtrattr, xtrskip);
 707                 if (extsize > 0)
 708                         set_extattr_fd(ofile, name, buf, extsize);
 709                 (void) fchown(ofile, uid, gid);
 710                 (void) fchmod(ofile, mode);
 711                 (void) futimes(ofile, ctimep);
 712                 (void) futimes(ofile, mtimep);
 713                 (void) fchflags(ofile, flags);
 714                 (void) close(ofile);
 715                 return (GOOD);
 716         }
 717         /* NOTREACHED */
 718 }
 719
 720 /*
 721  * Set attributes for a file.
 722  */
 723 void
 724 set_extattr_file(char *name, void *buf, int size)
 725 {
 726         struct extattr *eap, *eaend;
 727
 728         vprintf(stdout, "Set attributes for %s:", name);
 729         eaend = buf + size;
 730         for (eap = buf; eap < eaend; eap = EXTATTR_NEXT(eap)) {
 731                 /*
 732                  * Make sure this entry is complete.
 733                  */
 734                 if (EXTATTR_NEXT(eap) > eaend || eap->ea_length <= 0) {
 735                         dprintf(stdout, "\n\t%scorrupted",
 736                                 eap == buf ? "" : "remainder ");
 737                         break;
 738                 }
 739                 if (eap->ea_namespace == EXTATTR_NAMESPACE_EMPTY)
 740                         continue;
 741                 vprintf(stdout, "\n\t%s, (%d bytes), %*s",
 742                         namespace_names[eap->ea_namespace], eap->ea_length,
 743                         eap->ea_namelength, eap->ea_name);
 744                 /*
 745                  * First we try the general attribute setting interface.
 746                  * However, some attributes can only be set by root or
 747                  * by using special interfaces (for example, ACLs).
 748                  */
 749                 if (extattr_set_file(name, eap->ea_namespace, eap->ea_name,
 750                     EXTATTR_CONTENT(eap), EXTATTR_CONTENT_SIZE(eap)) != -1) {
 751                         dprintf(stdout, " (set using extattr_set_file)");
 752                         continue;
 753                 }
 754                 /*
 755                  * If the general interface refuses to set the attribute,
 756                  * then we try all the specialized interfaces that we
 757                  * know about.
 758                  */
 759                 if (eap->ea_namespace == EXTATTR_NAMESPACE_SYSTEM &&
 760                     !strcmp(eap->ea_name, POSIX1E_ACL_ACCESS_EXTATTR_NAME)) {
 761                         if (acl_set_file(name, ACL_TYPE_ACCESS,
 762                             EXTATTR_CONTENT(eap)) != -1) {
 763                                 dprintf(stdout, " (set using acl_set_file)");
 764                                 continue;
 765                         }
 766                 }
 767                 if (eap->ea_namespace == EXTATTR_NAMESPACE_SYSTEM &&
 768                     !strcmp(eap->ea_name, POSIX1E_ACL_DEFAULT_EXTATTR_NAME)) {
 769                         if (acl_set_file(name, ACL_TYPE_DEFAULT,
 770                             EXTATTR_CONTENT(eap)) != -1) {
 771                                 dprintf(stdout, " (set using acl_set_file)");
 772                                 continue;
 773                         }
 774                 }
 775                 vprintf(stdout, " (unable to set)");
 776         }
 777         vprintf(stdout, "\n");
 778 }
 779
 780 /*
 781  * Set attributes for a symbolic link.
 782  */
 783 static void
 784 set_extattr_link(char *name, void *buf, int size)
 785 {
 786         struct extattr *eap, *eaend;
 787
 788         vprintf(stdout, "Set attributes for %s:", name);
 789         eaend = buf + size;
 790         for (eap = buf; eap < eaend; eap = EXTATTR_NEXT(eap)) {
 791                 /*
 792                  * Make sure this entry is complete.
 793                  */
 794                 if (EXTATTR_NEXT(eap) > eaend || eap->ea_length <= 0) {
 795                         dprintf(stdout, "\n\t%scorrupted",
 796                                 eap == buf ? "" : "remainder ");
 797                         break;
 798                 }
 799                 if (eap->ea_namespace == EXTATTR_NAMESPACE_EMPTY)
 800                         continue;
 801                 vprintf(stdout, "\n\t%s, (%d bytes), %*s",
 802                         namespace_names[eap->ea_namespace], eap->ea_length,
 803                         eap->ea_namelength, eap->ea_name);
 804                 /*
 805                  * First we try the general attribute setting interface.
 806                  * However, some attributes can only be set by root or
 807                  * by using special interfaces (for example, ACLs).
 808                  */
 809                 if (extattr_set_link(name, eap->ea_namespace, eap->ea_name,
 810                     EXTATTR_CONTENT(eap), EXTATTR_CONTENT_SIZE(eap)) != -1) {
 811                         dprintf(stdout, " (set using extattr_set_link)");
 812                         continue;
 813                 }
 814                 /*
 815                  * If the general interface refuses to set the attribute,
 816                  * then we try all the specialized interfaces that we
 817                  * know about.
 818                  */
 819                 if (eap->ea_namespace == EXTATTR_NAMESPACE_SYSTEM &&
 820                     !strcmp(eap->ea_name, POSIX1E_ACL_ACCESS_EXTATTR_NAME)) {
 821                         if (acl_set_link_np(name, ACL_TYPE_ACCESS,
 822                             EXTATTR_CONTENT(eap)) != -1) {
 823                                 dprintf(stdout, " (set using acl_set_link_np)");
 824                                 continue;
 825                         }
 826                 }
 827                 if (eap->ea_namespace == EXTATTR_NAMESPACE_SYSTEM &&
 828                     !strcmp(eap->ea_name, POSIX1E_ACL_DEFAULT_EXTATTR_NAME)) {
 829                         if (acl_set_link_np(name, ACL_TYPE_DEFAULT,
 830                             EXTATTR_CONTENT(eap)) != -1) {
 831                                 dprintf(stdout, " (set using acl_set_link_np)");
 832                                 continue;
 833                         }
 834                 }
 835                 vprintf(stdout, " (unable to set)");
 836         }
 837         vprintf(stdout, "\n");
 838 }
 839
 840 /*
 841  * Set attributes on a file descriptor.
 842  */
 843 static void
 844 set_extattr_fd(int fd, char *name, void *buf, int size)
 845 {
 846         struct extattr *eap, *eaend;
 847
 848         vprintf(stdout, "Set attributes for %s:", name);
 849         eaend = buf + size;
 850         for (eap = buf; eap < eaend; eap = EXTATTR_NEXT(eap)) {
 851                 /*
 852                  * Make sure this entry is complete.
 853                  */
 854                 if (EXTATTR_NEXT(eap) > eaend || eap->ea_length <= 0) {
 855                         dprintf(stdout, "\n\t%scorrupted",
 856                                 eap == buf ? "" : "remainder ");
 857                         break;
 858                 }
 859                 if (eap->ea_namespace == EXTATTR_NAMESPACE_EMPTY)
 860                         continue;
 861                 vprintf(stdout, "\n\t%s, (%d bytes), %*s",
 862                         namespace_names[eap->ea_namespace], eap->ea_length,
 863                         eap->ea_namelength, eap->ea_name);
 864                 /*
 865                  * First we try the general attribute setting interface.
 866                  * However, some attributes can only be set by root or
 867                  * by using special interfaces (for example, ACLs).
 868                  */
 869                 if (extattr_set_fd(fd, eap->ea_namespace, eap->ea_name,
 870                     EXTATTR_CONTENT(eap), EXTATTR_CONTENT_SIZE(eap)) != -1) {
 871                         dprintf(stdout, " (set using extattr_set_fd)");
 872                         continue;
 873                 }
 874                 /*
 875                  * If the general interface refuses to set the attribute,
 876                  * then we try all the specialized interfaces that we
 877                  * know about.
 878                  */
 879                 if (eap->ea_namespace == EXTATTR_NAMESPACE_SYSTEM &&
 880                     !strcmp(eap->ea_name, POSIX1E_ACL_ACCESS_EXTATTR_NAME)) {
 881                         if (acl_set_fd(fd, EXTATTR_CONTENT(eap)) != -1) {
 882                                 dprintf(stdout, " (set using acl_set_fd)");
 883                                 continue;
 884                         }
 885                 }
 886                 if (eap->ea_namespace == EXTATTR_NAMESPACE_SYSTEM &&
 887                     !strcmp(eap->ea_name, POSIX1E_ACL_DEFAULT_EXTATTR_NAME)) {
 888                         if (acl_set_file(name, ACL_TYPE_DEFAULT,
 889                             EXTATTR_CONTENT(eap)) != -1) {
 890                                 dprintf(stdout, " (set using acl_set_file)");
 891                                 continue;
 892                         }
 893                 }
 894                 vprintf(stdout, " (unable to set)");
 895         }
 896         vprintf(stdout, "\n");
 897 }
 898
 899 /*
 900  * skip over bit maps on the tape
 901  */
 902 void
 903 skipmaps(void)
 904 {
 905
 906         while (spcl.c_type == TS_BITS || spcl.c_type == TS_CLRI)
 907                 skipfile();
 908 }
 909
 910 /*
 911  * skip over a file on the tape
 912  */
 913 void
 914 skipfile(void)
 915 {
 916
 917         curfile.action = SKIP;
 918         getfile(xtrnull, xtrnull, xtrnull);
 919 }
 920
 921 /*
 922  * Extract a file from the tape.
 923  * When an allocated block is found it is passed to the fill function;
 924  * when an unallocated block (hole) is found, a zeroed buffer is passed
 925  * to the skip function.
 926  */
 927 void
 928 getfile(void (*datafill)(char *, long), void (*attrfill)(char *, long),
 929         void (*skip)(char *, long))
 930 {
 931         int i;
 932         off_t size;
 933         int curblk, attrsize;
 934         void (*fillit)(char *, long);
 935         static char clearedbuf[MAXBSIZE];
 936         char buf[MAXBSIZE / TP_BSIZE][TP_BSIZE];
 937         char junk[TP_BSIZE];
 938
 939         curblk = 0;
 940         size = spcl.c_size;
 941         attrsize = spcl.c_extsize;
 942         if (spcl.c_type == TS_END)
 943                 panic("ran off end of tape\n");
 944         if (spcl.c_magic != FS_UFS2_MAGIC)
 945                 panic("not at beginning of a file\n");
 946         if (!gettingfile && setjmp(restart) != 0)
 947                 return;
 948         gettingfile++;
 949         fillit = datafill;
 950         if (size == 0 && attrsize > 0) {
 951                 fillit = attrfill;
 952                 size = attrsize;
 953                 attrsize = 0;
 954         }
 955 loop:
 956         for (i = 0; i < spcl.c_count; i++) {
 957                 if (!readmapflag && i > TP_NINDIR) {
 958                         if (Dflag) {
 959                                 fprintf(stderr, "spcl.c_count = %jd\n",
 960                                     (intmax_t)spcl.c_count);
 961                                 break;
 962                         } else
 963                                 panic("spcl.c_count = %jd\n",
 964                                     (intmax_t)spcl.c_count);
 965                 }
 966                 if (readmapflag || spcl.c_addr[i]) {
 967                         readtape(&buf[curblk++][0]);
 968                         if (curblk == fssize / TP_BSIZE) {
 969                                 (*fillit)((char *)buf, (long)(size > TP_BSIZE ?
 970                                      fssize : (curblk - 1) * TP_BSIZE + size));
 971                                 curblk = 0;
 972                         }
 973                 } else {
 974                         if (curblk > 0) {
 975                                 (*fillit)((char *)buf, (long)(size > TP_BSIZE ?
 976                                      curblk * TP_BSIZE :
 977                                      (curblk - 1) * TP_BSIZE + size));
 978                                 curblk = 0;
 979                         }
 980                         (*skip)(clearedbuf, (long)(size > TP_BSIZE ?
 981                                 TP_BSIZE : size));
 982                 }
 983                 if ((size -= TP_BSIZE) <= 0) {
 984                         if (size > -TP_BSIZE && curblk > 0) {
 985                                 (*fillit)((char *)buf,
 986                                         (long)((curblk * TP_BSIZE) + size));
 987                                 curblk = 0;
 988                         }
 989                         if (attrsize > 0) {
 990                                 fillit = attrfill;
 991                                 size = attrsize;
 992                                 attrsize = 0;
 993                                 continue;
 994                         }
 995                         if (spcl.c_count - i > 1)
 996                                 dprintf(stdout, "skipping %d junk block(s)\n",
 997                                         spcl.c_count - i - 1);
 998                         for (i++; i < spcl.c_count; i++) {
 999                                 if (!readmapflag && i > TP_NINDIR) {
1000                                         if (Dflag) {
1001                                                 fprintf(stderr,
1002                                                     "spcl.c_count = %jd\n",
1003                                                     (intmax_t)spcl.c_count);
1004                                                 break;
1005                                         } else
1006                                                 panic("spcl.c_count = %jd\n",
1007                                                     (intmax_t)spcl.c_count);
1008                                 }
1009                                 if (readmapflag || spcl.c_addr[i])
1010                                         readtape(junk);
1011                         }
1012                         break;
1013                 }
1014         }
1015         if (gethead(&spcl) == GOOD && size > 0) {
1016                 if (spcl.c_type == TS_ADDR)
1017                         goto loop;
1018                 dprintf(stdout,
1019                         "Missing address (header) block for %s at %ld blocks\n",
1020                         curfile.name, blksread);
1021         }
1022         if (curblk > 0)
1023                 panic("getfile: lost data\n");
1024         findinode(&spcl);
1025         gettingfile = 0;
1026 }
1027
1028 /*
1029  * These variables are shared between the next two functions.
1030  */
1031 static int extbufsize = 0;
1032 static char *extbuf;
1033 static int extloc;
1034
1035 /*
1036  * Allocate a buffer into which to extract extended attributes.
1037  */
1038 static char *
1039 setupextattr(int extsize)
1040 {
1041
1042         extloc = 0;
1043         if (extsize <= extbufsize)
1044                 return (extbuf);
1045         if (extbufsize > 0)
1046                 free(extbuf);
1047         if ((extbuf = malloc(extsize)) != NULL) {
1048                 extbufsize = extsize;
1049                 return (extbuf);
1050         }
1051         extbufsize = 0;
1052         extbuf = NULL;
1053         fprintf(stderr, "Cannot extract %d bytes %s for inode %d, name %s\n",
1054             extsize, "of extended attributes", curfile.ino, curfile.name);
1055         return (NULL);
1056 }
1057
1058 /*
1059  * Extract the next block of extended attributes.
1060  */
1061 static void
1062 xtrattr(char *buf, long size)
1063 {
1064
1065         if (extloc + size > extbufsize)
1066                 panic("overrun attribute buffer\n");
1067         memmove(&extbuf[extloc], buf, size);
1068         extloc += size;
1069 }
1070
1071 /*
1072  * Write out the next block of a file.
1073  */
1074 static void
1075 xtrfile(char *buf, long size)
1076 {
1077
1078         if (Nflag)
1079                 return;
1080         if (write(ofile, buf, (int) size) == -1) {
1081                 fprintf(stderr,
1082                     "write error extracting inode %d, name %s\nwrite: %s\n",
1083                         curfile.ino, curfile.name, strerror(errno));
1084         }
1085 }
1086
1087 /*
1088  * Skip over a hole in a file.
1089  */
1090 /* ARGSUSED */
1091 static void
1092 xtrskip(char *buf, long size)
1093 {
1094
1095         if (lseek(ofile, size, SEEK_CUR) == -1) {
1096                 fprintf(stderr,
1097                     "seek error extracting inode %d, name %s\nlseek: %s\n",
1098                         curfile.ino, curfile.name, strerror(errno));
1099                 done(1);
1100         }
1101 }
1102
1103 /*
1104  * Collect the next block of a symbolic link.
1105  */
1106 static void
1107 xtrlnkfile(char *buf, long size)
1108 {
1109
1110         pathlen += size;
1111         if (pathlen > MAXPATHLEN) {
1112                 fprintf(stderr, "symbolic link name: %s->%s%s; too long %d\n",
1113                     curfile.name, lnkbuf, buf, pathlen);
1114                 done(1);
1115         }
1116         (void) strcat(lnkbuf, buf);
1117 }
1118
1119 /*
1120  * Skip over a hole in a symbolic link (should never happen).
1121  */
1122 /* ARGSUSED */
1123 static void
1124 xtrlnkskip(char *buf, long size)
1125 {
1126
1127         fprintf(stderr, "unallocated block in symbolic link %s\n",
1128                 curfile.name);
1129         done(1);
1130 }
1131
1132 /*
1133  * Collect the next block of a bit map.
1134  */
1135 static void
1136 xtrmap(char *buf, long size)
1137 {
1138
1139         memmove(map, buf, size);
1140         map += size;
1141 }
1142
1143 /*
1144  * Skip over a hole in a bit map (should never happen).
1145  */
1146 /* ARGSUSED */
1147 static void
1148 xtrmapskip(char *buf, long size)
1149 {
1150
1151         panic("hole in map\n");
1152         map += size;
1153 }
1154
1155 /*
1156  * Noop, when an extraction function is not needed.
1157  */
1158 /* ARGSUSED */
1159 void
1160 xtrnull(char *buf, long size)
1161 {
1162
1163         return;
1164 }
1165
1166 /*
1167  * Read TP_BSIZE blocks from the input.
1168  * Handle read errors, and end of media.
1169  */
1170 static void
1171 readtape(char *buf)
1172 {
1173         long rd, newvol, i, oldnumtrec;
1174         int cnt, seek_failed;
1175
1176         if (blkcnt + (byteslide > 0) < numtrec) {
1177                 memmove(buf, &tapebuf[(blkcnt++ * TP_BSIZE) + byteslide], (long)TP_BSIZE);
1178                 blksread++;
1179                 tapeaddr++;
1180                 return;
1181         }
1182         if (numtrec > 0)
1183                 memmove(&tapebuf[-TP_BSIZE],
1184                     &tapebuf[(numtrec-1) * TP_BSIZE], (long)TP_BSIZE);
1185         oldnumtrec = numtrec;
1186         for (i = 0; i < ntrec; i++)
1187                 ((struct s_spcl *)&tapebuf[i * TP_BSIZE])->c_magic = 0;
1188         if (numtrec == 0)
1189                 numtrec = ntrec;
1190         cnt = ntrec * TP_BSIZE;
1191         rd = 0;
1192 getmore:
1193 #ifdef RRESTORE
1194         if (host)
1195                 i = rmtread(&tapebuf[rd], cnt);
1196         else
1197 #endif
1198                 i = read(mt, &tapebuf[rd], cnt);
1199         /*
1200          * Check for mid-tape short read error.
1201          * If found, skip rest of buffer and start with the next.
1202          */
1203         if (!pipein && !pipecmdin && numtrec < ntrec && i > 0) {
1204                 dprintf(stdout, "mid-media short read error.\n");
1205                 numtrec = ntrec;
1206         }
1207         /*
1208          * Handle partial block read.
1209          */
1210         if ((pipein || pipecmdin) && i == 0 && rd > 0)
1211                 i = rd;
1212         else if (i > 0 && i != ntrec * TP_BSIZE) {
1213                 if (pipein || pipecmdin) {
1214                         rd += i;
1215                         cnt -= i;
1216                         if (cnt > 0)
1217                                 goto getmore;
1218                         i = rd;
1219                 } else {
1220                         /*
1221                          * Short read. Process the blocks read.
1222                          */
1223                         if (i % TP_BSIZE != 0)
1224                                 vprintf(stdout,
1225                                     "partial block read: %ld should be %ld\n",
1226                                     i, ntrec * TP_BSIZE);
1227                         numtrec = i / TP_BSIZE;
1228                 }
1229         }
1230         /*
1231          * Handle read error.
1232          */
1233         if (i < 0) {
1234                 fprintf(stderr, "Tape read error while ");
1235                 switch (curfile.action) {
1236                 default:
1237                         fprintf(stderr, "trying to set up tape\n");
1238                         break;
1239                 case UNKNOWN:
1240                         fprintf(stderr, "trying to resynchronize\n");
1241                         break;
1242                 case USING:
1243                         fprintf(stderr, "restoring %s\n", curfile.name);
1244                         break;
1245                 case SKIP:
1246                         fprintf(stderr, "skipping over inode %d\n",
1247                                 curfile.ino);
1248                         break;
1249                 }
1250                 if (!yflag && !reply("continue"))
1251                         done(1);
1252                 i = ntrec * TP_BSIZE;
1253                 memset(tapebuf, 0, i);
1254 #ifdef RRESTORE
1255                 if (host)
1256                         seek_failed = (rmtseek(i, 1) < 0);
1257                 else
1258 #endif
1259                         seek_failed = (lseek(mt, i, SEEK_CUR) == (off_t)-1);
1260
1261                 if (seek_failed) {
1262                         fprintf(stderr,
1263                             "continuation failed: %s\n", strerror(errno));
1264                         done(1);
1265                 }
1266         }
1267         /*
1268          * Handle end of tape.
1269          */
1270         if (i == 0) {
1271                 vprintf(stdout, "End-of-tape encountered\n");
1272                 if (!pipein) {
1273                         newvol = volno + 1;
1274                         volno = 0;
1275                         numtrec = 0;
1276                         getvol(newvol);
1277                         readtape(buf);
1278                         return;
1279                 }
1280                 if (rd % TP_BSIZE != 0)
1281                         panic("partial block read: %ld should be %ld\n",
1282                                 rd, ntrec * TP_BSIZE);
1283                 terminateinput();
1284                 memmove(&tapebuf[rd], &endoftapemark, (long)TP_BSIZE);
1285         }
1286         if (oldnumtrec == 0)
1287                 blkcnt = 0;
1288         else
1289                 blkcnt -= oldnumtrec;
1290         memmove(buf,
1291             &tapebuf[(blkcnt++ * TP_BSIZE) + byteslide], (long)TP_BSIZE);
1292         blksread++;
1293         tapeaddr++;
1294 }
1295
1296 static void
1297 findtapeblksize(void)
1298 {
1299         long i;
1300
1301         for (i = 0; i < ntrec; i++)
1302                 ((struct s_spcl *)&tapebuf[i * TP_BSIZE])->c_magic = 0;
1303         blkcnt = 0;
1304 #ifdef RRESTORE
1305         if (host)
1306                 i = rmtread(tapebuf, ntrec * TP_BSIZE);
1307         else
1308 #endif
1309                 i = read(mt, tapebuf, ntrec * TP_BSIZE);
1310
1311         if (i <= 0) {
1312                 fprintf(stderr, "tape read error: %s\n", strerror(errno));
1313                 done(1);
1314         }
1315         if (i % TP_BSIZE != 0) {
1316                 fprintf(stderr, "Tape block size (%ld) %s (%d)\n",
1317                         i, "is not a multiple of dump block size", TP_BSIZE);
1318                 done(1);
1319         }
1320         ntrec = i / TP_BSIZE;
1321         numtrec = ntrec;
1322         vprintf(stdout, "Tape block size is %ld\n", ntrec);
1323 }
1324
1325 void
1326 closemt(void)
1327 {
1328
1329         if (mt < 0)
1330                 return;
1331         if (pipecmdin) {
1332                 pclose(popenfp);
1333                 popenfp = NULL;
1334         } else
1335 #ifdef RRESTORE
1336         if (host)
1337                 rmtclose();
1338         else
1339 #endif
1340                 (void) close(mt);
1341 }
1342
1343 /*
1344  * Read the next block from the tape.
1345  * If it is not any valid header, return an error.
1346  */
1347 static int
1348 gethead(struct s_spcl *buf)
1349 {
1350         long i;
1351
1352         readtape((char *)buf);
1353         if (buf->c_magic != FS_UFS2_MAGIC && buf->c_magic != NFS_MAGIC) {
1354                 if (buf->c_magic == OFS_MAGIC) {
1355                         fprintf(stderr,
1356                             "Format of dump tape is too old. Must use\n");
1357                         fprintf(stderr,
1358                             "a version of restore from before 2002.\n");
1359                         return (FAIL);
1360                 }
1361                 if (swabl(buf->c_magic) != FS_UFS2_MAGIC &&
1362                     buf->c_magic != NFS_MAGIC) {
1363                         if (buf->c_magic == OFS_MAGIC) {
1364                                 fprintf(stderr,
1365                                   "Format of dump tape is too old. Must use\n");
1366                                 fprintf(stderr,
1367                                   "a version of restore from before 2002.\n");
1368                         }
1369                         return (FAIL);
1370                 }
1371                 if (!Bcvt) {
1372                         vprintf(stdout, "Note: Doing Byte swapping\n");
1373                         Bcvt = 1;
1374                 }
1375         }
1376         if (checksum((int *)buf) == FAIL)
1377                 return (FAIL);
1378         if (Bcvt) {
1379                 swabst((u_char *)"8l4s1q8l2q17l", (u_char *)buf);
1380                 swabst((u_char *)"l",(u_char *) &buf->c_level);
1381                 swabst((u_char *)"2l4q",(u_char *) &buf->c_flags);
1382         }
1383         readmapflag = 0;
1384
1385         switch (buf->c_type) {
1386
1387         case TS_CLRI:
1388         case TS_BITS:
1389                 /*
1390                  * Have to patch up missing information in bit map headers
1391                  */
1392                 buf->c_size = buf->c_count * TP_BSIZE;
1393                 if (buf->c_count > TP_NINDIR)
1394                         readmapflag = 1;
1395                 else
1396                         for (i = 0; i < buf->c_count; i++)
1397                                 buf->c_addr[i]++;
1398                 /* FALL THROUGH */
1399
1400         case TS_TAPE:
1401                 if (buf->c_magic == NFS_MAGIC &&
1402                     (buf->c_flags & NFS_DR_NEWINODEFMT) == 0)
1403                         oldinofmt = 1;
1404                 /* FALL THROUGH */
1405
1406         case TS_END:
1407                 buf->c_inumber = 0;
1408                 /* FALL THROUGH */
1409
1410         case TS_ADDR:
1411         case TS_INODE:
1412                 /*
1413                  * For old dump tapes, have to copy up old fields to
1414                  * new locations.
1415                  */
1416                 if (buf->c_magic == NFS_MAGIC) {
1417                         buf->c_tapea = buf->c_old_tapea;
1418                         buf->c_firstrec = buf->c_old_firstrec;
1419                         buf->c_date = _time32_to_time(buf->c_old_date);
1420                         buf->c_ddate = _time32_to_time(buf->c_old_ddate);
1421                         buf->c_atime = _time32_to_time(buf->c_old_atime);
1422                         buf->c_mtime = _time32_to_time(buf->c_old_mtime);
1423                         buf->c_birthtime = 0;
1424                         buf->c_birthtimensec = 0;
1425                         buf->c_extsize = 0;
1426                 }
1427                 break;
1428
1429         default:
1430                 panic("gethead: unknown inode type %d\n", buf->c_type);
1431                 break;
1432         }
1433         if (dumpdate != 0 && _time64_to_time(buf->c_date) != dumpdate)
1434                 fprintf(stderr, "Header with wrong dumpdate.\n");
1435         /*
1436          * If we're restoring a filesystem with the old (FreeBSD 1)
1437          * format inodes, copy the uid/gid to the new location
1438          */
1439         if (oldinofmt) {
1440                 buf->c_uid = buf->c_spare1[1];
1441                 buf->c_gid = buf->c_spare1[2];
1442         }
1443         buf->c_magic = FS_UFS2_MAGIC;
1444         tapeaddr = buf->c_tapea;
1445         if (dflag)
1446                 accthdr(buf);
1447         return(GOOD);
1448 }
1449
1450 /*
1451  * Check that a header is where it belongs and predict the next header
1452  */
1453 static void
1454 accthdr(struct s_spcl *header)
1455 {
1456         static ino_t previno = 0x7fffffff;
1457         static int prevtype;
1458         static long predict;
1459         long blks, i;
1460
1461         if (header->c_type == TS_TAPE) {
1462                 fprintf(stderr, "Volume header ");
1463                 if (header->c_firstrec)
1464                         fprintf(stderr, "begins with record %jd",
1465                             (intmax_t)header->c_firstrec);
1466                 fprintf(stderr, "\n");
1467                 previno = 0x7fffffff;
1468                 return;
1469         }
1470         if (previno == 0x7fffffff)
1471                 goto newcalc;
1472         switch (prevtype) {
1473         case TS_BITS:
1474                 fprintf(stderr, "Dumped inodes map header");
1475                 break;
1476         case TS_CLRI:
1477                 fprintf(stderr, "Used inodes map header");
1478                 break;
1479         case TS_INODE:
1480                 fprintf(stderr, "File header, ino %d", previno);
1481                 break;
1482         case TS_ADDR:
1483                 fprintf(stderr, "File continuation header, ino %d", previno);
1484                 break;
1485         case TS_END:
1486                 fprintf(stderr, "End of tape header");
1487                 break;
1488         }
1489         if (predict != blksread - 1)
1490                 fprintf(stderr, "; predicted %ld blocks, got %ld blocks",
1491                         predict, blksread - 1);
1492         fprintf(stderr, "\n");
1493 newcalc:
1494         blks = 0;
1495         if (header->c_type != TS_END)
1496                 for (i = 0; i < header->c_count; i++)
1497                         if (readmapflag || header->c_addr[i] != 0)
1498                                 blks++;
1499         predict = blks;
1500         blksread = 0;
1501         prevtype = header->c_type;
1502         previno = header->c_inumber;
1503 }
1504
1505 /*
1506  * Find an inode header.
1507  * Complain if had to skip.
1508  */
1509 static void
1510 findinode(struct s_spcl *header)
1511 {
1512         static long skipcnt = 0;
1513         long i;
1514         char buf[TP_BSIZE];
1515         int htype;
1516
1517         curfile.name = "<name unknown>";
1518         curfile.action = UNKNOWN;
1519         curfile.mode = 0;
1520         curfile.ino = 0;
1521         do {
1522                 htype = header->c_type;
1523                 switch (htype) {
1524
1525                 case TS_ADDR:
1526                         /*
1527                          * Skip up to the beginning of the next record
1528                          */
1529                         for (i = 0; i < header->c_count; i++)
1530                                 if (header->c_addr[i])
1531                                         readtape(buf);
1532                         while (gethead(header) == FAIL ||
1533                             _time64_to_time(header->c_date) != dumpdate) {
1534                                 skipcnt++;
1535                                 if (Dflag) {
1536                                         byteslide++;
1537                                         if (byteslide < TP_BSIZE) {
1538                                                 blkcnt--;
1539                                                 blksread--;
1540                                         } else
1541                                                 byteslide = 0;
1542                                 }
1543                         }
1544                         break;
1545
1546                 case TS_INODE:
1547                         curfile.mode = header->c_mode;
1548                         curfile.uid = header->c_uid;
1549                         curfile.gid = header->c_gid;
1550                         curfile.file_flags = header->c_file_flags;
1551                         curfile.rdev = header->c_rdev;
1552                         curfile.atime_sec = header->c_atime;
1553                         curfile.atime_nsec = header->c_atimensec;
1554                         curfile.mtime_sec = header->c_mtime;
1555                         curfile.mtime_nsec = header->c_mtimensec;
1556                         curfile.birthtime_sec = header->c_birthtime;
1557                         curfile.birthtime_nsec = header->c_birthtimensec;
1558                         curfile.extsize = header->c_extsize;
1559                         curfile.size = header->c_size;
1560                         curfile.ino = header->c_inumber;
1561                         break;
1562
1563                 case TS_END:
1564                         /* If we missed some tapes, get another volume. */
1565                         if (tapesread & (tapesread + 1)) {
1566                                 getvol(0);
1567                                 continue;
1568                         }
1569                         curfile.ino = maxino;
1570                         break;
1571
1572                 case TS_CLRI:
1573                         curfile.name = "<file removal list>";
1574                         break;
1575
1576                 case TS_BITS:
1577                         curfile.name = "<file dump list>";
1578                         break;
1579
1580                 case TS_TAPE:
1581                         if (Dflag)
1582                                 fprintf(stderr, "unexpected tape header\n");
1583                         else
1584                                 panic("unexpected tape header\n");
1585
1586                 default:
1587                         if (Dflag)
1588                                 fprintf(stderr, "unknown tape header type %d\n",
1589                                     spcl.c_type);
1590                         else
1591                                 panic("unknown tape header type %d\n",
1592                                     spcl.c_type);
1593                         while (gethead(header) == FAIL ||
1594                             _time64_to_time(header->c_date) != dumpdate) {
1595                                 skipcnt++;
1596                                 if (Dflag) {
1597                                         byteslide++;
1598                                         if (byteslide < TP_BSIZE) {
1599                                                 blkcnt--;
1600                                                 blksread--;
1601                                         } else
1602                                                 byteslide = 0;
1603                                 }
1604                         }
1605
1606                 }
1607         } while (htype == TS_ADDR);
1608         if (skipcnt > 0)
1609                 fprintf(stderr, "resync restore, skipped %ld %s\n",
1610                     skipcnt, Dflag ? "bytes" : "blocks");
1611         skipcnt = 0;
1612 }
1613
1614 static int
1615 checksum(int *buf)
1616 {
1617         int i, j;
1618
1619         j = sizeof(union u_spcl) / sizeof(int);
1620         i = 0;
1621         if (!Bcvt) {
1622                 do
1623                         i += *buf++;
1624                 while (--j);
1625         } else {
1626                 /* What happens if we want to read restore tapes
1627                         for a 16bit int machine??? */
1628                 do
1629                         i += swabl(*buf++);
1630                 while (--j);
1631         }
1632
1633         if (i != CHECKSUM) {
1634                 fprintf(stderr, "Checksum error %o, inode %d file %s\n", i,
1635                         curfile.ino, curfile.name);
1636                 return(FAIL);
1637         }
1638         return(GOOD);
1639 }
1640
1641 #ifdef RRESTORE
1642 #include <stdarg.h>
1643
1644 void
1645 msg(const char *fmt, ...)
1646 {
1647         va_list ap;
1648         va_start(ap, fmt);
1649         (void)vfprintf(stderr, fmt, ap);
1650         va_end(ap);
1651 }
1652 #endif /* RRESTORE */
1653
1654 static u_char *
1655 swabshort(u_char *sp, int n)
1656 {
1657         char c;
1658
1659         while (--n >= 0) {
1660                 c = sp[0]; sp[0] = sp[1]; sp[1] = c;
1661                 sp += 2;
1662         }
1663         return (sp);
1664 }
1665
1666 static u_char *
1667 swablong(u_char *sp, int n)
1668 {
1669         char c;
1670
1671         while (--n >= 0) {
1672                 c = sp[0]; sp[0] = sp[3]; sp[3] = c;
1673                 c = sp[2]; sp[2] = sp[1]; sp[1] = c;
1674                 sp += 4;
1675         }
1676         return (sp);
1677 }
1678
1679 static u_char *
1680 swabquad(u_char *sp, int n)
1681 {
1682         char c;
1683
1684         while (--n >= 0) {
1685                 c = sp[0]; sp[0] = sp[7]; sp[7] = c;
1686                 c = sp[1]; sp[1] = sp[6]; sp[6] = c;
1687                 c = sp[2]; sp[2] = sp[5]; sp[5] = c;
1688                 c = sp[3]; sp[3] = sp[4]; sp[4] = c;
1689                 sp += 8;
1690         }
1691         return (sp);
1692 }
1693
1694 void
1695 swabst(u_char *cp, u_char *sp)
1696 {
1697         int n = 0;
1698
1699         while (*cp) {
1700                 switch (*cp) {
1701                 case '0': case '1': case '2': case '3': case '4':
1702                 case '5': case '6': case '7': case '8': case '9':
1703                         n = (n * 10) + (*cp++ - '0');
1704                         continue;
1705
1706                 case 's': case 'w': case 'h':
1707                         if (n == 0)
1708                                 n = 1;
1709                         sp = swabshort(sp, n);
1710                         break;
1711
1712                 case 'l':
1713                         if (n == 0)
1714                                 n = 1;
1715                         sp = swablong(sp, n);
1716                         break;
1717
1718                 case 'q':
1719                         if (n == 0)
1720                                 n = 1;
1721                         sp = swabquad(sp, n);
1722                         break;
1723
1724                 case 'b':
1725                         if (n == 0)
1726                                 n = 1;
1727                         sp += n;
1728                         break;
1729
1730                 default:
1731                         fprintf(stderr, "Unknown conversion character: %c\n",
1732                             *cp);
1733                         done(0);
1734                         break;
1735                 }
1736                 cp++;
1737                 n = 0;
1738         }
1739 }
1740
1741 static u_long
1742 swabl(u_long x)
1743 {
1744         swabst((u_char *)"l", (u_char *)&x);
1745         return (x);
1746 }