]> git.wh0rd.org Git - dump.git/blob - restore/tape.c
Some corrections from Uwe Gohlke.
[dump.git] / restore / tape.c
1 /*
2  *      Ported to Linux's Second Extended File System as part of the
3  *      dump and restore backup suit
4  *      Remy Card <card@Linux.EU.Org>, 1994-1997
5  *      Stelian Pop <stelian@popies.net>, 1999-2000
6  *      Stelian Pop <stelian@popies.net> - AlcĂ´ve <www.alcove.com>, 2000-2002
7  */
8
9 /*
10  * Copyright (c) 1983, 1993
11  *      The Regents of the University of California.  All rights reserved.
12  * (c) UNIX System Laboratories, Inc.
13  * All or some portions of this file are derived from material licensed
14  * to the University of California by American Telephone and Telegraph
15  * Co. or Unix System Laboratories, Inc. and are reproduced herein with
16  * the permission of UNIX System Laboratories, Inc.
17  *
18  * Redistribution and use in source and binary forms, with or without
19  * modification, are permitted provided that the following conditions
20  * are met:
21  * 1. Redistributions of source code must retain the above copyright
22  *    notice, this list of conditions and the following disclaimer.
23  * 2. Redistributions in binary form must reproduce the above copyright
24  *    notice, this list of conditions and the following disclaimer in the
25  *    documentation and/or other materials provided with the distribution.
26  * 3. Neither the name of the University nor the names of its contributors
27  *    may be used to endorse or promote products derived from this software
28  *    without specific prior written permission.
29  *
30  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
31  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
32  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
33  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
34  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
35  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
36  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
37  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
38  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
39  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
40  * SUCH DAMAGE.
41  */
42
43 #ifndef lint
44 static const char rcsid[] =
45         "$Id: tape.c,v 1.76 2003/11/22 16:52:16 stelian Exp $";
46 #endif /* not lint */
47
48 #include <config.h>
49 #include <compatlfs.h>
50 #include <sys/types.h>
51 #include <errno.h>
52 #include <compaterr.h>
53 #include <system.h>
54 #include <setjmp.h>
55 #include <stdio.h>
56 #include <stdlib.h>
57 #include <string.h>
58 #include <unistd.h>
59
60 #include <sys/param.h>
61 #include <sys/file.h>
62 #include <sys/mtio.h>
63 #include <sys/stat.h>
64
65 #ifdef  __linux__
66 #include <sys/time.h>
67 #include <time.h>
68 #ifdef HAVE_EXT2FS_EXT2_FS_H
69 #include <ext2fs/ext2_fs.h>
70 #else
71 #include <linux/ext2_fs.h>
72 #endif
73 #include <ext2fs/ext2fs.h>
74 #include <bsdcompat.h>
75 #else   /* __linux__ */
76 #ifdef sunos
77 #define quad_t int64_t
78 #include <sys/time.h>
79 #include <sys/fcntl.h>
80 #include <bsdcompat.h>
81 #else
82 #include <ufs/ufs/dinode.h>
83 #endif
84 #endif  /* __linux__ */
85 #ifdef DUMP_MACOSX
86 #include "darwin.h"
87 #endif
88 #include <protocols/dumprestore.h>
89
90 #ifdef HAVE_ZLIB
91 #include <zlib.h>
92 #endif /* HAVE_ZLIB */
93
94 #ifdef HAVE_BZLIB
95 #include <bzlib.h>
96 #endif /* HAVE_BZLIB */
97
98 #ifdef HAVE_LZO
99 #include <minilzo.h>
100 #endif /* HAVE_LZO */
101
102 #include "restore.h"
103 #include "extern.h"
104 #include "pathnames.h"
105
106 #ifdef USE_QFA
107 int             noresyncmesg = 0;
108 #endif /* USE_QFA */
109 static long     fssize = MAXBSIZE;
110 static int      mt = -1;
111 int             pipein = 0;
112 static int      magtapein = 0;          /* input is from magtape */
113 static char     magtape[MAXPATHLEN];
114 static char     magtapeprefix[MAXPATHLEN];
115 static int      blkcnt;
116 static int      numtrec;
117 static char     *tapebuf;               /* input buffer for read */
118 static int      bufsize;                /* buffer size without prefix */
119 static char     *tbufptr = NULL;        /* active tape buffer */
120 #if defined(HAVE_ZLIB) || defined(HAVE_BZLIB) || defined(HAVE_LZO)
121 static char     *comprbuf;              /* uncompress work buf */
122 static size_t   comprlen;               /* size including prefix */
123 #endif
124 static union    u_spcl endoftapemark;
125 static long     blksread;               /* blocks read since last header */
126 static long     tpblksread = 0;         /* TP_BSIZE blocks read */
127 static long     tapesread;
128 static sigjmp_buf       restart;
129 static int      gettingfile = 0;        /* restart has a valid frame */
130 char            *host = NULL;
131
132 static int      ofile;
133 static char     *map;
134 static char     lnkbuf[MAXPATHLEN + 1];
135 static int      pathlen;
136
137 int             oldinofmt;      /* old inode format conversion required */
138 int             Bcvt;           /* Swap Bytes (for CCI or sun) */
139 static int      Qcvt;           /* Swap quads (for sun) */
140
141 #define FLUSHTAPEBUF()  blkcnt = ntrec + 1
142
143 static void      accthdr __P((struct s_spcl *));
144 static int       checksum __P((int *));
145 static void      findinode __P((struct s_spcl *));
146 static void      findtapeblksize __P((void));
147 static int       gethead __P((struct s_spcl *));
148 static int       converthead __P((struct s_spcl *));
149 static void      converttapebuf __P((struct tapebuf *));
150 static void      readtape __P((char *));
151 static void      setdumpnum __P((void));
152 #ifdef DUMP_MACOSX
153 static void      xtrfilefinderinfo __P((char *, size_t));
154 #endif
155
156 static u_int     swabi __P((u_int));
157 #if 0
158 static u_long    swabl __P((u_long));
159 #endif
160 static u_char   *swab64 __P((u_char *, int));
161 static u_char   *swab32 __P((u_char *, int));
162 static u_char   *swab16 __P((u_char *, int));
163 static void      terminateinput __P((void));
164 static void      xtrfile __P((char *, size_t));
165 static void      xtrlnkfile __P((char *, size_t));
166 static void      xtrlnkskip __P((char *, size_t));
167 static void      xtrmap __P((char *, size_t));
168 static void      xtrmapskip __P((char *, size_t));
169 static void      xtrskip __P((char *, size_t));
170 static void      setmagtapein __P((void));
171
172 #if defined(HAVE_ZLIB) || defined(HAVE_BZLIB) || defined(HAVE_LZO)
173 static void     newcomprbuf __P((int));
174 static void     (*readtape_func) __P((char *));
175 static void     readtape_set __P((char *));
176 static void     readtape_uncompr __P((char *));
177 static void     readtape_comprfile __P((char *));
178 static void     readtape_comprtape __P((char *));
179 static char     *decompress_tapebuf __P((struct tapebuf *, int));
180 static void     msg_read_error __P((char *));
181 #endif
182 static int      read_a_block __P((int, char *, size_t, long *));
183 #define PREFIXSIZE      sizeof(struct tapebuf)
184
185 #define COMPARE_ONTHEFLY 1
186
187 #if COMPARE_ONTHEFLY
188 static int      ifile;          /* input file for compare */
189 static int      cmperror;       /* compare error */
190 static void     xtrcmpfile __P((char *, size_t));
191 static void     xtrcmpskip __P((char *, size_t));
192 #endif
193
194 static int readmapflag;
195 static int readingmaps;         /* set to 1 while reading the maps */
196
197 #ifdef DUMP_MACOSX
198 static DumpFinderInfo   gFndrInfo;
199 #endif
200
201 /*
202  * Set up an input source. This is called from main.c before setup() is.
203  */
204 void
205 setinput(char *source)
206 {
207         int i;
208         char *n;
209
210         FLUSHTAPEBUF();
211         if (bflag)
212                 newtapebuf(ntrec);
213         else
214                 newtapebuf(NTREC > HIGHDENSITYTREC ? NTREC : HIGHDENSITYTREC);
215         terminal = stdin;
216
217 #ifdef RRESTORE
218         if ((n = strchr(source, ':'))) {
219                 for (i = 0; i < (n - source); i++) {
220                         if (source[i] == '/')
221                                 break;
222                 }
223                 if (source[i] != '/') {
224                         host = source;
225                         source = strchr(host, ':');
226                         *source++ = '\0';
227                         if (rmthost(host) == 0)
228                                 exit(1);
229                 }
230         } else
231 #endif
232         if (strcmp(source, "-") == 0) {
233                 /*
234                  * Since input is coming from a pipe we must establish
235                  * our own connection to the terminal.
236                  */
237                 terminal = fopen(_PATH_TTY, "r");
238                 if (terminal == NULL) {
239                         warn("cannot open %s", _PATH_TTY);
240                         terminal = fopen(_PATH_DEVNULL, "r");
241                         if (terminal == NULL)
242                                 err(1, "cannot open %s", _PATH_DEVNULL);
243                 }
244                 pipein++;
245         }
246         setuid(getuid());       /* no longer need or want root privileges */
247         if (Mflag) {
248                 strncpy(magtapeprefix, source, MAXPATHLEN);
249                 magtapeprefix[MAXPATHLEN-1] = '\0';
250                 snprintf(magtape, MAXPATHLEN, "%s%03d", source, 1);
251         }
252         else
253                 strncpy(magtape, source, MAXPATHLEN);
254         magtape[MAXPATHLEN - 1] = '\0';
255 }
256
257 void
258 newtapebuf(long size)
259 {
260         static int tapebufsize = -1;
261
262         ntrec = size;
263         bufsize = ntrec * TP_BSIZE;
264         if (size <= tapebufsize)
265                 return;
266         if (tapebuf != NULL)
267                 free(tapebuf);
268         tapebuf = malloc(size * TP_BSIZE + sizeof(struct tapebuf));
269         if (tapebuf == NULL)
270                 errx(1, "Cannot allocate space for tape buffer");
271         tapebufsize = size;
272 }
273
274 #if defined(HAVE_ZLIB) || defined(HAVE_BZLIB) || defined(HAVE_LZO)
275 static void
276 newcomprbuf(int size)
277 {
278         size_t buf_size = (size+1) * TP_BSIZE + sizeof(struct tapebuf);
279         if (buf_size <= comprlen)
280                 return;
281         comprlen = buf_size;
282         if (comprbuf != NULL)
283                 free(comprbuf);
284         comprbuf = malloc(comprlen);
285         if (comprbuf == NULL)
286                 errx(1, "Cannot allocate space for decompress buffer");
287 }
288 #endif /* HAVE_ZLIB || HAVE_BZLIB || HAVE_LZO */
289
290 /*
291  * Verify that the tape drive can be accessed and
292  * that it actually is a dump tape.
293  */
294 void
295 setup(void)
296 {
297         int i, j, *ip, bot_code;
298         struct STAT stbuf;
299         char *temptape;
300
301         Vprintf(stdout, "Verify tape and initialize maps\n");
302         if (Afile == NULL && bot_script) {
303                 msg("Launching %s\n", bot_script);
304                 bot_code = system_command(bot_script, magtape, 1);
305                 if (bot_code != 0 && bot_code != 1) {
306                         msg("Restore aborted by the beginning of tape script\n");
307                         exit(1);
308                 }
309         }
310
311         if (Afile)
312                 temptape = Afile;
313         else
314                 temptape = magtape;
315
316 #ifdef RRESTORE
317         if (!Afile && host)
318                 mt = rmtopen(temptape, O_RDONLY);
319         else
320 #endif
321         if (pipein)
322                 mt = 0;
323         else
324                 mt = OPEN(temptape, O_RDONLY, 0);
325         if (mt < 0)
326                 err(1, "%s", temptape);
327         if (!Afile) {
328                 volno = 1;
329                 setmagtapein();
330                 setdumpnum();
331         }
332 #if defined(HAVE_ZLIB) || defined(HAVE_BZLIB) || defined(HAVE_LZO)
333         readtape_func = readtape_set;
334 #if defined(HAVE_LZO)
335         if (lzo_init() != LZO_E_OK) {
336           msg("internal error - lzo_init failed \n");
337           exit(1);
338         }
339 #endif
340 #endif
341         FLUSHTAPEBUF();
342         findtapeblksize();
343         if (gethead(&spcl) == FAIL) {
344                 blkcnt--; /* push back this block */
345                 blksread--;
346                 tpblksread--;
347                 cvtflag++;
348                 if (gethead(&spcl) == FAIL)
349                         errx(1, "Tape is not a dump tape");
350                 fprintf(stderr, "Converting to new file system format.\n");
351         }
352
353         if (zflag) {
354                 fprintf(stderr, "Dump tape is compressed.\n");
355 #if !defined(HAVE_ZLIB) && !defined(HAVE_BZLIB) && !defined(HAVE_LZO)
356                 errx(1,"This restore version doesn't support decompression");
357 #endif /* !HAVE_ZLIB && !HAVE_BZLIB */
358         }
359         if (pipein) {
360                 endoftapemark.s_spcl.c_magic = cvtflag ? OFS_MAGIC : NFS_MAGIC;
361                 endoftapemark.s_spcl.c_type = TS_END;
362                 ip = (int *)&endoftapemark;
363                 j = sizeof(union u_spcl) / sizeof(int);
364                 i = 0;
365                 do
366                         i += *ip++;
367                 while (--j);
368                 endoftapemark.s_spcl.c_checksum = CHECKSUM - i;
369         }
370         if (vflag || command == 't' || command == 'C')
371                 printdumpinfo();
372 #ifdef USE_QFA
373         if (tapeposflag && (unsigned long)spcl.c_date != qfadumpdate)
374                 errx(1, "different QFA/dumpdates detected\n");
375 #endif
376         if (filesys[0] == '\0') {
377                 char *dirptr;
378                 strncpy(filesys, spcl.c_filesys, NAMELEN);
379                 filesys[NAMELEN - 1] = '\0';
380                 dirptr = strstr(filesys, " (dir");
381                 if (dirptr != NULL)
382                         *dirptr = '\0';
383         }
384         dumptime = spcl.c_ddate;
385         dumpdate = spcl.c_date;
386         if (STAT(".", &stbuf) < 0)
387                 err(1, "cannot stat .");
388         if (stbuf.st_blksize > 0 && stbuf.st_blksize < TP_BSIZE )
389                 fssize = TP_BSIZE;
390         if (stbuf.st_blksize >= TP_BSIZE && stbuf.st_blksize <= MAXBSIZE)
391                 fssize = stbuf.st_blksize;
392         if (((fssize - 1) & fssize) != 0)
393                 errx(1, "bad block size %ld", fssize);
394         if (spcl.c_volume != 1)
395                 errx(1, "Tape is not volume 1 of the dump");
396         if (gethead(&spcl) == FAIL) {
397                 Dprintf(stdout, "header read failed at %ld blocks\n", (long)blksread);
398                 panic("no header after volume mark!\n");
399         }
400         readingmaps = 1;
401         findinode(&spcl);
402         if (spcl.c_type != TS_CLRI)
403                 errx(1, "Cannot find file removal list");
404         maxino = (spcl.c_count * TP_BSIZE * NBBY) + 1;
405         map = calloc((unsigned)1, (unsigned)howmany(maxino, NBBY));
406         if (map == NULL)
407                 errx(1, "no memory for active inode map");
408         usedinomap = map;
409         curfile.action = USING;
410         getfile(xtrmap, xtrmapskip);
411         while (spcl.c_type == TS_ADDR) {
412                 /* Recompute maxino and the map */
413                 dump_ino_t oldmaxino = maxino;
414                 maxino += (spcl.c_count * TP_BSIZE * NBBY) + 1;
415                 resizemaps(oldmaxino, maxino);
416
417                 spcl.c_dinode.di_size = spcl.c_count * TP_BSIZE;
418                 getfile(xtrmap, xtrmapskip);
419         }
420         Dprintf(stdout, "maxino = %lu\n", (unsigned long)maxino);
421         if (spcl.c_type != TS_BITS)
422                 errx(1, "Cannot find file dump list");
423         map = calloc((unsigned)1, (unsigned)howmany(maxino, NBBY));
424         if (map == (char *)NULL)
425                 errx(1, "no memory for file dump list");
426         dumpmap = map;
427         curfile.action = USING;
428         getfile(xtrmap, xtrmapskip);
429         while (spcl.c_type == TS_ADDR) {
430                 spcl.c_dinode.di_size = spcl.c_count * TP_BSIZE;
431                 getfile(xtrmap, xtrmapskip);
432         }
433         /*
434          * If there may be whiteout entries on the tape, pretend that the
435          * whiteout inode exists, so that the whiteout entries can be
436          * extracted.
437          */
438         if (oldinofmt == 0)
439                 SETINO(WINO, dumpmap);
440         readingmaps = 0;
441         findinode(&spcl);
442 }
443
444 /*
445  * Prompt user to load a new dump volume.
446  * "Nextvol" is the next suggested volume to use.
447  * This suggested volume is enforced when doing full
448  * or incremental restores, but can be overridden by
449  * the user when only extracting a subset of the files.
450  */
451 void
452 getvol(long nextvol)
453 {
454         long newvol = 0, wantnext = 0, i;
455         long saved_blksread = 0, saved_tpblksread = 0;
456         union u_spcl tmpspcl;
457 #       define tmpbuf tmpspcl.s_spcl
458         char buf[TP_BSIZE];
459         int haderror = 0, bot_code = 1;
460
461         if (nextvol == 1) {
462                 tapesread = 0;
463                 gettingfile = 0;
464                 tpblksread = 0;
465                 blksread = 0;
466         }
467         if (pipein) {
468                 if (nextvol != 1)
469                         panic("Changing volumes on pipe input?\n");
470                 if (volno == 1)
471                         return;
472                 goto gethdr;
473         }
474         saved_blksread = blksread;
475         saved_tpblksread = tpblksread;
476 #if defined(USE_QFA) && defined(sunos)
477         if (createtapeposflag || tapeposflag) 
478                 close(fdsmtc);
479 #endif
480 again:
481         if (pipein)
482                 exit(1); /* pipes do not get a second chance */
483         if (aflag || curfile.action != SKIP) {
484                 newvol = nextvol;
485                 wantnext = 1;
486         } else {
487                 newvol = 0;
488                 wantnext = 0;
489         }
490         while (newvol <= 0) {
491                 if (tapesread == 0) {
492                         fprintf(stderr, "%s%s%s%s%s",
493                             "You have not read any volumes yet.\n",
494                             "Unless you know which volume your",
495                             " file(s) are on you should start\n",
496                             "with the last volume and work",
497                             " towards the first.\n");
498                 } else {
499                         fprintf(stderr, "You have read volumes");
500                         strcpy(buf, ": ");
501                         for (i = 1; i < 32; i++)
502                                 if (tapesread & (1 << i)) {
503                                         fprintf(stderr, "%s%ld", buf, (long)i);
504                                         strcpy(buf, ", ");
505                                 }
506                         fprintf(stderr, "\n");
507                 }
508                 do      {
509                         fprintf(stderr, "Specify next volume # (none if no more volumes): ");
510                         (void) fflush(stderr);
511                         (void) fgets(buf, TP_BSIZE, terminal);
512                 } while (!feof(terminal) && buf[0] == '\n');
513                 if (feof(terminal))
514                         exit(1);
515                 if (!strcmp(buf, "none\n")) {
516                         terminateinput();
517                         return;
518                 }
519                 newvol = atoi(buf);
520                 if (newvol <= 0) {
521                         fprintf(stderr,
522                             "Volume numbers are positive numerics\n");
523                 }
524         }
525         if (newvol == volno) {
526                 tapesread |= 1 << volno;
527 #if defined(USE_QFA) && defined(sunos)
528                 if (createtapeposflag || tapeposflag) {
529                         if (OpenSMTCmt(magtape) < 0) {
530                                 volno = -1;
531                                 haderror = 1;
532                                 goto again;
533                         }
534                 }
535 #endif  /* USE_QFA */
536                 return;
537         }
538         closemt();
539
540         /* 
541          * if using an archive file, reset its name so readtape()
542          * could properly use remote access.
543          */
544         Afile = NULL;
545
546         if (Mflag) {
547                 snprintf(magtape, MAXPATHLEN, "%s%03ld", magtapeprefix, newvol);
548                 magtape[MAXPATHLEN - 1] = '\0';
549         }
550         if (bot_script && !haderror) {
551                 msg("Launching %s\n", bot_script);
552                 bot_code = system_command(bot_script, magtape, newvol);
553                 if (bot_code != 0 && bot_code != 1) {
554                         msg("Restore aborted by the beginning of tape script\n");
555                         exit(1);
556                 }
557         }
558         if (haderror || (bot_code && !Mflag)) {
559                 haderror = 0;
560 #ifdef sunos
561                 fprintf(stderr, "Mount volume %ld\n", (long)newvol);
562 #else
563                 fprintf(stderr, "Mount tape volume %ld\n", (long)newvol);
564 #endif
565                 fprintf(stderr, "Enter ``none'' if there are no more tapes\n");
566 #ifdef sunos
567                 fprintf(stderr, "then enter volume name (default: %s) ", magtape);
568 #else
569                 fprintf(stderr, "otherwise enter tape name (default: %s) ", magtape);
570 #endif
571                 (void) fflush(stderr);
572                 (void) fgets(buf, TP_BSIZE, terminal);
573                 if (feof(terminal))
574                         exit(1);
575                 if (!strcmp(buf, "none\n")) {
576                         terminateinput();
577                         return;
578                 }
579                 if (buf[0] != '\n') {
580                         char *pos;
581                         (void) strncpy(magtape, buf, sizeof(magtape));
582                         magtape[sizeof(magtape) - 1] = '\0';
583                         if ((pos = strchr(magtape, '\n'))) 
584                                 magtape[pos - magtape] = '\0';
585                 }
586         }
587 #if defined(USE_QFA) && defined(sunos)
588         if (createtapeposflag || tapeposflag) {
589                 if (OpenSMTCmt(magtape) < 0) {
590                         volno = -1;
591                         haderror = 1;
592                         goto again;
593                 }
594         }
595 #endif  /* USE_QFA */
596 #ifdef RRESTORE
597         if (host)
598                 mt = rmtopen(magtape, O_RDONLY);
599         else
600 #endif
601                 mt = OPEN(magtape, O_RDONLY, 0);
602
603         if (mt == -1) {
604                 fprintf(stderr, "Cannot open %s\n", magtape);
605                 volno = -1;
606                 haderror = 1;
607                 goto again;
608         }
609 gethdr:
610         setmagtapein();
611 #if defined(HAVE_ZLIB) || defined(HAVE_BZLIB) || defined(HAVE_LZO)
612         readtape_func = readtape_set;
613 #endif
614         volno = newvol;
615         setdumpnum();
616         FLUSHTAPEBUF();
617         findtapeblksize();
618         if (gethead(&tmpbuf) == FAIL) {
619                 Dprintf(stdout, "header read failed at %ld blocks\n", (long)blksread);
620                 fprintf(stderr, "tape is not dump tape\n");
621                 volno = 0;
622                 haderror = 1;
623                 blksread = saved_blksread;
624                 tpblksread = saved_tpblksread;
625                 goto again;
626         }
627         if (tmpbuf.c_volume != volno) {
628                 fprintf(stderr, "Wrong volume (%d)\n", tmpbuf.c_volume);
629                 volno = 0;
630                 haderror = 1;
631                 blksread = saved_blksread;
632                 tpblksread = saved_tpblksread;
633                 goto again;
634         }
635         if (tmpbuf.c_date != dumpdate || tmpbuf.c_ddate != dumptime) {
636                 fprintf(stderr, "Wrong dump date\n\tgot: %s",
637 #ifdef sunos
638                         ctime(&tmpbuf.c_date));
639 #else
640                         ctime4(&tmpbuf.c_date));
641 #endif
642                 fprintf(stderr, "\twanted: %s", ctime(&dumpdate));
643                 volno = 0;
644                 haderror = 1;
645                 blksread = saved_blksread;
646                 tpblksread = saved_tpblksread;
647                 goto again;
648         }
649         tapesread |= 1 << volno;
650         /*
651          * If continuing from the previous volume, skip over any
652          * blocks read already at the end of the previous volume.
653          *
654          * If coming to this volume at random, skip to the beginning
655          * of the next record.
656          */
657         if (zflag) {
658                 fprintf(stderr, "Dump tape is compressed.\n");
659 #if !defined(HAVE_ZLIB) && !defined(HAVE_BZLIB) && !defined(HAVE_LZO)
660                 errx(1,"This restore version doesn't support decompression");
661 #endif /* !HAVE_ZLIB && !HAVE_BZLIB */
662         }
663         Dprintf(stdout, "read %ld recs, tape starts with %ld\n",
664                 tpblksread - 1, (long)tmpbuf.c_firstrec);
665         if (tmpbuf.c_type == TS_TAPE && (tmpbuf.c_flags & DR_NEWHEADER)) {
666                 if (!wantnext) {
667                         tpblksread = tmpbuf.c_firstrec + 1;
668                         for (i = tmpbuf.c_count; i > 0; i--)
669                                 readtape(buf);
670                 } else if (tmpbuf.c_firstrec > 0 &&
671                            tmpbuf.c_firstrec < tpblksread - 1) {
672                         /*
673                          * -1 since we've read the volume header
674                          */
675                         i = tpblksread - tmpbuf.c_firstrec - 1;
676                         Dprintf(stderr, "Skipping %ld duplicate record%s.\n",
677                                 (long)i, i > 1 ? "s" : "");
678                         while (--i >= 0)
679                                 readtape(buf);
680                 }
681         }
682         if (curfile.action == USING) {
683                 if (volno == 1)
684                         panic("active file into volume 1\n");
685                 return;
686         }
687         /*
688          * Skip up to the beginning of the next record
689          */
690         if (tmpbuf.c_type == TS_TAPE && (tmpbuf.c_flags & DR_NEWHEADER))
691                 for (i = tmpbuf.c_count; i > 0; i--)
692                         readtape(buf);
693         (void) gethead(&spcl);
694         findinode(&spcl);
695         if (gettingfile) {
696                 gettingfile = 0;
697                 siglongjmp(restart, 1);
698         }
699 }
700
701 /*
702  * Handle unexpected EOF.
703  */
704 static void
705 terminateinput(void)
706 {
707
708         if (gettingfile && curfile.action == USING) {
709                 printf("Warning: %s %s\n",
710                     "End-of-input encountered while extracting", curfile.name);
711         }
712         curfile.name = "<name unknown>";
713         curfile.action = UNKNOWN;
714         curfile.dip = NULL;
715         curfile.ino = maxino;
716         if (gettingfile) {
717                 gettingfile = 0;
718                 siglongjmp(restart, 1);
719         }
720 }
721
722 /*
723  * handle multiple dumps per tape by skipping forward to the
724  * appropriate one.
725  */
726 static void
727 setdumpnum(void)
728 {
729         struct mtop tcom;
730
731         if (dumpnum == 1 || volno != 1)
732                 return;
733         if (pipein)
734                 errx(1, "Cannot have multiple dumps on pipe input");
735         tcom.mt_op = MTFSF;
736         tcom.mt_count = dumpnum - 1;
737 #ifdef RRESTORE
738         if (host) {
739                 if (rmtioctl(MTFSF, dumpnum - 1) < 0) {
740                         fprintf(stderr, "rmtioctl MTFSF: %s\n", strerror(errno));
741             exit(1);
742                 }
743         } else
744 #endif
745                 if (ioctl(mt, (int)MTIOCTOP, (char *)&tcom) < 0) {
746                         fprintf(stderr, "rmtioctl MTFSF: %s\n", strerror(errno));
747                         exit(1);
748                         /* warn("ioctl MTFSF"); */
749                 }
750 }
751
752 void
753 printdumpinfo(void)
754 {
755 #ifdef sunos
756         Vprintf(stdout, "Dump   date: %s", ctime(&spcl.c_date));
757         Vprintf(stdout, "Dumped from: %s",
758             (spcl.c_ddate == 0) ? "the epoch\n" : ctime(&spcl.c_ddate));
759         if (spcl.c_host[0] == '\0')
760                 return;
761         Vprintf(stdout, "Level %d dump of %s on %s:%s\n",
762                 spcl.c_level, spcl.c_filesys, spcl.c_host, spcl.c_dev);
763         Vprintf(stdout, "Label: %s\n", spcl.c_label);
764 #else
765         fprintf(stdout, "Dump   date: %s", ctime4(&spcl.c_date));
766         fprintf(stdout, "Dumped from: %s",
767             (spcl.c_ddate == 0) ? "the epoch\n" : ctime4(&spcl.c_ddate));
768         if (spcl.c_host[0] == '\0')
769                 return;
770         fprintf(stdout, "Level %d dump of %s on %s:%s\n",
771                 spcl.c_level, spcl.c_filesys, spcl.c_host, spcl.c_dev);
772         fprintf(stdout, "Label: %s\n", spcl.c_label);
773 #endif
774 }
775
776 void 
777 printvolinfo(void)
778 {
779         int i;
780
781         if (volinfo[1] == ROOTINO) {
782                 printf("Starting inode numbers by volume:\n");
783                 for (i = 1; i < (int)TP_NINOS && volinfo[i] != 0; ++i)
784                         printf("\tVolume %d: %lu\n", i, (unsigned long)volinfo[i]);
785         }
786 }
787
788
789 int
790 extractfile(struct entry *ep, int doremove)
791 {
792         unsigned int flags;
793         mode_t mode;
794         struct timeval timep[2];
795         char *name = myname(ep);
796
797         /* If removal is requested (-r mode) do remove it unless
798          * we are extracting a metadata only inode */
799         if (spcl.c_flags & DR_METAONLY) {
800                 Vprintf(stdout, "file %s is metadata only\n", name);
801         }
802         else {
803                 if (doremove) {
804                         removeleaf(ep);
805                         ep->e_flags &= ~REMOVED;
806                 }
807         }
808
809         curfile.name = name;
810         curfile.action = USING;
811 #if defined(__linux__) || defined(sunos)
812         timep[0].tv_sec = curfile.dip->di_atime.tv_sec;
813         timep[0].tv_usec = curfile.dip->di_atime.tv_usec;
814         timep[1].tv_sec = curfile.dip->di_mtime.tv_sec;
815         timep[1].tv_usec = curfile.dip->di_mtime.tv_usec;
816 #else   /* __linux__ || sunos */
817         timep[0].tv_sec = curfile.dip->di_atime;
818         timep[0].tv_usec = curfile.dip->di_atimensec / 1000;
819         timep[1].tv_sec = curfile.dip->di_mtime;
820         timep[1].tv_usec = curfile.dip->di_mtimensec / 1000;
821 #endif  /* __linux__ */
822         mode = curfile.dip->di_mode;
823         flags = curfile.dip->di_flags;
824         switch (mode & IFMT) {
825
826         default:
827                 fprintf(stderr, "%s: unknown file mode 0%o\n", name, mode);
828                 skipfile();
829                 return (FAIL);
830
831         case IFSOCK:
832                 Vprintf(stdout, "skipped socket %s\n", name);
833                 skipfile();
834                 return (GOOD);
835
836         case IFDIR:
837                 if (mflag) {
838                         if (ep == NULL || ep->e_flags & EXTRACT)
839                                 panic("unextracted directory %s\n", name);
840                         skipfile();
841                         return (GOOD);
842                 }
843                 Vprintf(stdout, "extract file %s\n", name);
844                 return (genliteraldir(name, curfile.ino));
845
846         case IFLNK:
847         {
848 #ifdef HAVE_LCHOWN
849                 uid_t luid = curfile.dip->di_uid;
850                 gid_t lgid = curfile.dip->di_gid;
851 #endif
852                 if (! (spcl.c_flags & DR_METAONLY)) {
853                         lnkbuf[0] = '\0';
854                         pathlen = 0;
855                         getfile(xtrlnkfile, xtrlnkskip);
856                         if (pathlen == 0) {
857                                 Vprintf(stdout,
858                                     "%s: zero length symbolic link (ignored)\n", name);
859                                 return (GOOD);
860                         }
861                         if (linkit(lnkbuf, name, SYMLINK) == FAIL)
862                                 return (FAIL);
863                 }
864                 else
865                         skipfile();
866
867 #ifdef HAVE_LCHOWN
868                 (void) lchown(name, luid, lgid);
869 #endif
870                 return (GOOD);
871         }
872
873         case IFIFO:
874                 Vprintf(stdout, "extract fifo %s\n", name);
875                 if (Nflag) {
876                         skipfile();
877                         return (GOOD);
878                 }
879                 if (! (spcl.c_flags & DR_METAONLY)) {
880                         if (uflag && !Nflag)
881                                 (void)unlink(name);
882                         if (mkfifo(name, mode) < 0) {
883                                 warn("%s: cannot create fifo", name);
884                                 skipfile();
885                                 return (FAIL);
886                         }
887                 }
888                 (void) chown(name, curfile.dip->di_uid, curfile.dip->di_gid);
889                 (void) chmod(name, mode);
890                 if (flags)
891 #ifdef  __linux__
892                         (void) fsetflags(name, flags);
893 #else
894 #ifdef sunos
895                         {
896                         warn("%s: cannot call chflags", name);
897                         /* (void) chflags(name, flags); */
898                         }
899 #else
900                         (void) chflags(name, flags);
901 #endif
902 #endif
903                 skipfile();
904                 utimes(name, timep);
905                 return (GOOD);
906
907         case IFCHR:
908         case IFBLK:
909                 Vprintf(stdout, "extract special file %s\n", name);
910                 if (Nflag) {
911                         skipfile();
912                         return (GOOD);
913                 }
914                 if (! (spcl.c_flags & DR_METAONLY)) {
915                         if (uflag)
916                                 (void)unlink(name);
917                         if (mknod(name, mode, (int)curfile.dip->di_rdev) < 0) {
918                                 warn("%s: cannot create special file", name);
919                                 skipfile();
920                                 return (FAIL);
921                         }
922                 }
923                 (void) chown(name, curfile.dip->di_uid, curfile.dip->di_gid);
924                 (void) chmod(name, mode);
925                 if (flags)
926 #ifdef  __linux__
927                         {
928                         warn("%s: fsetflags called on a special file", name);
929                         (void) fsetflags(name, flags);
930                         }
931 #else
932 #ifdef sunos
933                         {
934                         warn("%s: cannot call chflags on a special file", name);
935                         /* (void) chflags(name, flags); */
936                         }
937 #else
938                         {
939                         warn("%s: chflags called on a special file", name);
940                         (void) chflags(name, flags);
941                         }
942 #endif
943 #endif
944                 skipfile();
945                 utimes(name, timep);
946                 return (GOOD);
947
948         case IFREG:
949         {
950                 uid_t luid = curfile.dip->di_uid;
951                 gid_t lgid = curfile.dip->di_gid;
952
953                 Vprintf(stdout, "extract file %s\n", name);
954                 if (Nflag) {
955                         skipfile();
956                         return (GOOD);
957                 }
958                 if (! (spcl.c_flags & DR_METAONLY)) {
959                         if (uflag)
960                                 (void)unlink(name);
961                         if ((ofile = OPEN(name, O_WRONLY | O_CREAT | O_TRUNC,
962                             0666)) < 0) {
963                                 warn("%s: cannot create file", name);
964                                 skipfile();
965                                 return (FAIL);
966                         }
967                         getfile(xtrfile, xtrskip);
968                         (void) close(ofile);
969                 }
970                 else
971                         skipfile();
972                 (void) chown(name, luid, lgid);
973                 (void) chmod(name, mode);
974                 if (flags)
975 #ifdef  __linux__
976                         (void) fsetflags(name, flags);
977 #else
978 #ifdef sunos
979                         {
980                         warn("%s: cannot call chflags", name);
981                         /* (void) chflags(name, flags); */
982                         }
983 #else
984                         (void) chflags(name, flags);
985 #endif
986 #endif
987                 utimes(name, timep);
988                 return (GOOD);
989         }
990         }
991         /* NOTREACHED */
992 }
993
994 #ifdef DUMP_MACOSX
995 int
996 extractfinderinfoufs(char *name)
997 {
998         int err;
999         char                    oFileRsrc[MAXPATHLEN];
1000         int flags;
1001         mode_t mode;
1002         struct timeval timep[2];
1003         u_int32_t       uid;
1004         u_int32_t       gid;
1005         char    path[MAXPATHLEN], fname[MAXPATHLEN];
1006         int toto;
1007
1008         curfile.name = name;
1009         curfile.action = USING;
1010         timep[0].tv_sec = curfile.dip->di_atime.tv_sec;
1011         timep[0].tv_usec = curfile.dip->di_atime.tv_usec;
1012         timep[1].tv_sec = curfile.dip->di_mtime.tv_sec;
1013         timep[1].tv_usec = curfile.dip->di_mtime.tv_usec;
1014         mode = curfile.dip->di_mode;
1015         flags = curfile.dip->di_flags;
1016         uid = curfile.dip->di_uid;
1017         gid =  curfile.dip->di_gid;
1018
1019         switch (mode & IFMT) {
1020
1021         default:
1022                 fprintf(stderr, "%s: (extr. finfoufs) unknown file mode 0%o\n", name, mode);
1023                 skipfile();
1024                 return (FAIL);
1025
1026         case IFDIR:
1027                 fprintf(stderr, "%s: (extr. finfoufs[IFDIR]) unknown file mode 0%o\n", name, mode);
1028                 skipfile();
1029                 return (FAIL);
1030
1031         case IFLNK:
1032                 skipfile();
1033                 return (GOOD);
1034
1035         case IFREG:
1036                 Vprintf(stdout, "extract finderinfo file %s\n", name);
1037                 if (Nflag) {
1038                         skipfile();
1039                         return (GOOD);
1040                 }
1041                 getfile(xtrfilefinderinfo, xtrskip);
1042
1043                 GetPathFile(name, path, fname);
1044                 strcpy(oFileRsrc, path);
1045                 strcat(oFileRsrc, "._");
1046                 strcat(oFileRsrc, fname);
1047
1048                 if ((err = CreateAppleDoubleFileRes(oFileRsrc, &gFndrInfo.fndrinfo,
1049                                 mode, flags, timep, uid, gid)) != 0) {
1050                         fprintf(stderr, "%s: cannot create finderinfo: %s\n",
1051                         name, strerror(errno));
1052                         skipfile();
1053                         return (FAIL);
1054                 }
1055                 return (GOOD);
1056         }
1057         /* NOTREACHED */
1058 }
1059
1060
1061 int
1062 extractresourceufs(char *name)
1063 {
1064         char                    oFileRsrc[MAXPATHLEN];
1065         int flags;
1066         mode_t mode;
1067         struct timeval timep[2];
1068         char    path[MAXPATHLEN], fname[MAXPATHLEN];
1069         ASDHeaderPtr    hp;
1070         ASDEntryPtr     ep;
1071         u_long  loff;
1072          u_int32_t      uid;
1073         u_int32_t       gid;
1074         u_int64_t       di_size;
1075         char            *p;
1076         char            buf[1024];
1077
1078         curfile.name = name;
1079         curfile.action = USING;
1080         timep[0].tv_sec = curfile.dip->di_atime.tv_sec;
1081         timep[0].tv_usec = curfile.dip->di_atime.tv_usec;
1082         timep[1].tv_sec = curfile.dip->di_mtime.tv_sec;
1083         timep[1].tv_usec = curfile.dip->di_mtime.tv_usec;
1084         mode = curfile.dip->di_mode;
1085         flags = curfile.dip->di_flags;
1086         uid = curfile.dip->di_uid;
1087         gid =  curfile.dip->di_gid;
1088         di_size = curfile.dip->di_size;
1089
1090         switch (mode & IFMT) {
1091
1092         default:
1093                 fprintf(stderr, "%s: (extr. resufs) unknown file mode 0%o\n", name, mode);
1094                 skipfile();
1095                 return (FAIL);
1096
1097         case IFDIR:
1098                 fprintf(stderr, "%s: (extr. resufs [IFDIR]) unknown file mode 0%o\n", name, mode);
1099                 skipfile();
1100                 return (FAIL);
1101
1102         case IFLNK:
1103                 skipfile();
1104                 return (GOOD);
1105
1106         case IFREG:
1107                 Vprintf(stdout, "extract resource file %s\n", name);
1108                 if (Nflag) {
1109                         skipfile();
1110                         return (GOOD);
1111                 }
1112
1113                 GetPathFile(name, path, fname);
1114                 strcpy(oFileRsrc, path);
1115                 strcat(oFileRsrc, "._");
1116                 strcat(oFileRsrc, fname);
1117
1118                 if ((ofile = open(oFileRsrc, O_RDONLY, 0)) < 0) {
1119                         fprintf(stderr, "%s: cannot read finderinfo: %s\n",
1120                             name, strerror(errno));
1121                         skipfile();
1122                         return (FAIL);
1123                 }
1124                 read(ofile, buf, 70);
1125                 (void) close(ofile);
1126                 p = buf;
1127                 hp = (ASDHeaderPtr)p;
1128                 /* the header */
1129                 hp->entries++;
1130                 p += sizeof(ASDHeader) - CORRECT;
1131                 ep = (ASDEntryPtr)p;
1132                 /* the finderinfo entry */
1133                 ep->offset += sizeof(ASDEntry);
1134                 loff = ep->offset;
1135
1136                 p += sizeof(ASDEntry);
1137                 /* the finderinfo data */
1138                 bcopy(p, p + sizeof(ASDEntry), INFOLEN);
1139                 ep = (ASDEntryPtr)p;
1140                 /* the new resourcefork entry */
1141                 ep->entryID = EntryRSRCFork;
1142                 ep->offset = loff + INFOLEN;
1143                 ep->len = di_size;
1144                 /* write the new appledouble entries to the file */
1145                 if ((ofile = open(oFileRsrc, O_WRONLY | O_CREAT | O_TRUNC, 0666)) < 0) {
1146                         fprintf(stderr, "%s: cannot create resource file: %s\n",
1147                             name, strerror(errno));
1148                         skipfile();
1149                         return (FAIL);
1150                 }
1151                 write(ofile, buf, 70 + sizeof(ASDEntry));
1152                 /* and add the resource data from tape */
1153                 getfile(xtrfile, xtrskip);
1154
1155                 (void) fchown(ofile, uid, gid);
1156                 (void) fchmod(ofile, mode);
1157                 (void) close(ofile);
1158                 (void) fsetflags(oFileRsrc, flags);
1159                 utimes(oFileRsrc, timep);
1160                 return (GOOD);
1161         }
1162         /* NOTREACHED */
1163 }
1164 #endif /* DUMP_MACOSX */
1165
1166 /*
1167  * skip over bit maps on the tape
1168  */
1169 void
1170 skipmaps(void)
1171 {
1172
1173         while (spcl.c_type == TS_BITS || spcl.c_type == TS_CLRI)
1174                 skipfile();
1175 }
1176
1177 /*
1178  * skip over a file on the tape
1179  */
1180 void
1181 skipfile(void)
1182 {
1183
1184         curfile.action = SKIP;
1185         getfile(xtrnull, xtrnull);
1186 }
1187
1188 /*
1189  * Extract a file from the tape.
1190  * When an allocated block is found it is passed to the fill function;
1191  * when an unallocated block (hole) is found, a zeroed buffer is passed
1192  * to the skip function.
1193  */
1194 void
1195 getfile(void (*fill) __P((char *, size_t)), void (*skip) __P((char *, size_t)))
1196 {
1197         int i;
1198         volatile int curblk = 0;
1199         volatile quad_t size = spcl.c_dinode.di_size;
1200         volatile int last_write_was_hole = 0;
1201         quad_t origsize = size;
1202         static char clearedbuf[MAXBSIZE];
1203         char buf[MAXBSIZE / TP_BSIZE][TP_BSIZE];
1204         char junk[TP_BSIZE];
1205
1206         if (spcl.c_type == TS_END)
1207                 panic("ran off end of tape\n");
1208         if (spcl.c_magic != NFS_MAGIC)
1209                 panic("not at beginning of a file\n");
1210         if (!gettingfile && setjmp(restart) != 0)
1211                 return;
1212         gettingfile++;
1213 loop:
1214         for (i = 0; i < spcl.c_count; i++) {
1215                 if (readmapflag || spcl.c_addr[i]) {
1216                         readtape(&buf[curblk++][0]);
1217                         if (curblk == fssize / TP_BSIZE) {
1218                                 (*fill)((char *)buf, (size_t)(size > TP_BSIZE ?
1219                                      fssize : (curblk - 1) * TP_BSIZE + size));
1220                                 curblk = 0;
1221                                 last_write_was_hole = 0;
1222                         }
1223                 } else {
1224                         if (curblk > 0) {
1225                                 (*fill)((char *)buf, (size_t)(size > TP_BSIZE ?
1226                                      curblk * TP_BSIZE :
1227                                      (curblk - 1) * TP_BSIZE + size));
1228                                 curblk = 0;
1229                         }
1230                         (*skip)(clearedbuf, (long)(size > TP_BSIZE ?
1231                                 TP_BSIZE : size));
1232                         last_write_was_hole = 1;
1233                 }
1234                 if ((size -= TP_BSIZE) <= 0) {
1235                         for (i++; i < spcl.c_count; i++)
1236                                 if (readmapflag || spcl.c_addr[i])
1237                                         readtape(junk);
1238                         break;
1239                 }
1240         }
1241         if (gethead(&spcl) == GOOD && size > 0) {
1242                 if (spcl.c_type == TS_ADDR)
1243                         goto loop;
1244                 Dprintf(stdout,
1245                         "Missing address (header) block for %s at %ld blocks\n",
1246                         curfile.name, (long)blksread);
1247         }
1248         if (curblk > 0) {
1249                 (*fill)((char *)buf, (size_t)((curblk * TP_BSIZE) + size));
1250                 last_write_was_hole = 0;
1251         }
1252         if (size > 0) {
1253                 fprintf(stderr, "Missing blocks at the end of %s, assuming hole\n", curfile.name);
1254                 while (size > 0) {
1255                         size_t skp = size > TP_BSIZE ? TP_BSIZE : size;
1256                         (*skip)(clearedbuf, skp);
1257                         size -= skp;
1258                 }
1259                 last_write_was_hole = 1;
1260         }
1261         if (last_write_was_hole) {
1262                 FTRUNCATE(ofile, origsize);
1263         }
1264         if (!readingmaps) 
1265                 findinode(&spcl);
1266         gettingfile = 0;
1267 }
1268
1269 /*
1270  * Write out the next block of a file.
1271  */
1272 static void
1273 xtrfile(char *buf, size_t size)
1274 {
1275
1276         if (Nflag)
1277                 return;
1278         if (write(ofile, buf, (int) size) == -1)
1279                 err(1, "write error extracting inode %lu, name %s\nwrite",
1280                         (unsigned long)curfile.ino, curfile.name);
1281 }
1282
1283 #ifdef DUMP_MACOSX
1284 static void
1285 xtrfilefinderinfo(char *buf, size_t size)
1286 {
1287         if (Nflag)
1288                 return;
1289         bcopy(buf, &gFndrInfo, size);
1290 }
1291 #endif /* DUMP_MACOSX */
1292
1293 /*
1294  * Skip over a hole in a file.
1295  */
1296 /* ARGSUSED */
1297 static void
1298 xtrskip(UNUSED(char *buf), size_t size)
1299 {
1300
1301         if (LSEEK(ofile, (OFF_T)size, SEEK_CUR) == -1)
1302                 err(1, "seek error extracting inode %lu, name %s\nlseek",
1303                         (unsigned long)curfile.ino, curfile.name);
1304 }
1305
1306 /*
1307  * Collect the next block of a symbolic link.
1308  */
1309 static void
1310 xtrlnkfile(char *buf, size_t size)
1311 {
1312
1313         pathlen += size;
1314         if (pathlen > MAXPATHLEN)
1315                 errx(1, "symbolic link name: %s->%s%s; too long %d",
1316                     curfile.name, lnkbuf, buf, pathlen);
1317         (void) strcat(lnkbuf, buf);
1318 }
1319
1320 /*
1321  * Skip over a hole in a symbolic link (should never happen).
1322  */
1323 /* ARGSUSED */
1324 static void
1325 xtrlnkskip(UNUSED(char *buf), UNUSED(size_t size))
1326 {
1327
1328         errx(1, "unallocated block in symbolic link %s", curfile.name);
1329 }
1330
1331 /*
1332  * Collect the next block of a bit map.
1333  */
1334 static void
1335 xtrmap(char *buf, size_t size)
1336 {
1337
1338         memmove(map, buf, size);
1339         map += size;
1340 }
1341
1342 /*
1343  * Skip over a hole in a bit map (should never happen).
1344  */
1345 /* ARGSUSED */
1346 static void
1347 xtrmapskip(UNUSED(char *buf), size_t size)
1348 {
1349
1350         panic("hole in map\n");
1351         map += size;
1352 }
1353
1354 /*
1355  * Noop, when an extraction function is not needed.
1356  */
1357 /* ARGSUSED */
1358 void
1359 xtrnull(UNUSED(char *buf), UNUSED(size_t size))
1360 {
1361
1362         return;
1363 }
1364
1365 #if COMPARE_ONTHEFLY
1366 /*
1367  * Compare the next block of a file.
1368  */
1369 static void
1370 xtrcmpfile(char *buf, size_t size)
1371 {
1372         static char cmpbuf[MAXBSIZE];
1373
1374         if (cmperror)
1375                 return;
1376         
1377         if (read(ifile, cmpbuf, size) != (ssize_t)size) {
1378                 fprintf(stderr, "%s: size has changed.\n", 
1379                         curfile.name);
1380                 cmperror = 1;
1381                 return;
1382         }
1383         
1384         if (memcmp(buf, cmpbuf, size) != 0) {
1385                 fprintf(stderr, "%s: tape and disk copies are different\n",
1386                         curfile.name);
1387                 cmperror = 1;
1388                 return;
1389         }
1390 }
1391
1392 /*
1393  * Skip over a hole in a file.
1394  */
1395 static void
1396 xtrcmpskip(UNUSED(char *buf), size_t size)
1397 {
1398         static char cmpbuf[MAXBSIZE];
1399         int i;
1400
1401         if (cmperror)
1402                 return;
1403         
1404         if (read(ifile, cmpbuf, size) != (ssize_t)size) {
1405                 fprintf(stderr, "%s: size has changed.\n", 
1406                         curfile.name);
1407                 cmperror = 1;
1408                 return;
1409         }
1410
1411         for (i = 0; i < (int)size; ++i)
1412                 if (cmpbuf[i] != '\0') {
1413                         fprintf(stderr, "%s: tape and disk copies are different\n",
1414                                 curfile.name);
1415                         cmperror = 1;
1416                         return;
1417                 }
1418 }
1419 #endif /* COMPARE_ONTHEFLY */
1420
1421 #if !COMPARE_ONTHEFLY
1422 static int
1423 do_cmpfiles(int fd_tape, int fd_disk, long size)
1424 {
1425         static char buf_tape[BUFSIZ];
1426         static char buf_disk[BUFSIZ];
1427         ssize_t n_tape;
1428         ssize_t n_disk;
1429
1430         while (size > 0) {
1431                 if ((n_tape = read(fd_tape, buf_tape, sizeof(buf_tape))) < 1) {
1432                         close(fd_tape), close(fd_disk);
1433                         panic("do_cmpfiles: unexpected EOF[1]");
1434                 }
1435                 if ((n_disk = read(fd_disk, buf_disk, sizeof(buf_tape))) < 1) {
1436                         close(fd_tape), close(fd_disk);
1437                         panic("do_cmpfiles: unexpected EOF[2]");
1438                 }
1439                 if (n_tape != n_disk) {
1440                         close(fd_tape), close(fd_disk);
1441                         panic("do_cmpfiles: sizes different!");
1442                 }
1443                 if (memcmp(buf_tape, buf_disk, (size_t)n_tape) != 0) return (1);
1444                 size -= n_tape;
1445         }
1446         return (0);
1447 }
1448
1449 /* for debugging compare problems */
1450 #undef COMPARE_FAIL_KEEP_FILE
1451
1452 static
1453 #ifdef COMPARE_FAIL_KEEP_FILE
1454 /* return true if tapefile should be unlinked after compare */
1455 int
1456 #else
1457 void
1458 #endif
1459 cmpfiles(char *tapefile, char *diskfile, struct STAT *sbuf_disk)
1460 {
1461         struct STAT sbuf_tape;
1462         int fd_tape, fd_disk;
1463
1464         if (STAT(tapefile, &sbuf_tape) != 0) {
1465                 panic("Can't lstat tmp file %s: %s\n", tapefile,
1466                       strerror(errno));
1467                 do_compare_error;
1468         }
1469
1470         if (sbuf_disk->st_size != sbuf_tape.st_size) {
1471                 fprintf(stderr,
1472                         "%s: size changed from %ld to %ld.\n",
1473                         diskfile, (long)sbuf_tape.st_size, (long)sbuf_disk->st_size);
1474                 do_compare_error;
1475 #ifdef COMPARE_FAIL_KEEP_FILE
1476                 return (0);
1477 #else
1478                 return;
1479 #endif
1480         }
1481
1482         if ((fd_tape = OPEN(tapefile, O_RDONLY)) < 0) {
1483                 panic("Can't open %s: %s\n", tapefile, strerror(errno));
1484                 do_compare_error;
1485         }
1486         if ((fd_disk = OPEN(diskfile, O_RDONLY)) < 0) {
1487                 close(fd_tape);
1488                 panic("Can't open %s: %s\n", diskfile, strerror(errno));
1489                 do_compare_error;
1490         }
1491
1492         if (do_cmpfiles(fd_tape, fd_disk, sbuf_tape.st_size)) {
1493                 fprintf(stderr, "%s: tape and disk copies are different\n",
1494                         diskfile);
1495                 close(fd_tape);
1496                 close(fd_disk);
1497                 do_compare_error;
1498 #ifdef COMPARE_FAIL_KEEP_FILE
1499                 /* rename the file to live in /tmp */
1500                 /* rename `tapefile' to /tmp/<basename of diskfile> */
1501                 {
1502                         char *p = strrchr(diskfile, '/');
1503                         char newname[MAXPATHLEN];
1504                         if (!p) {
1505                                 panic("can't find / in %s\n", diskfile);
1506                         }
1507                         snprintf(newname, sizeof(newname), "%s/debug/%s", tmpdir, p + 1);
1508                         if (rename(tapefile, newname)) {
1509                                 panic("rename from %s to %s failed: %s\n",
1510                                       tapefile, newname,
1511                                       strerror(errno));
1512                         } else {
1513                                 fprintf(stderr, "*** %s saved to %s\n",
1514                                         tapefile, newname);
1515                         }
1516                 }
1517                 
1518                 /* don't unlink the file (it's not there anymore */
1519                 /* anyway) */
1520                 return (0);
1521 #else
1522                 return;
1523 #endif
1524         }
1525         close(fd_tape);
1526         close(fd_disk);
1527 #ifdef COMPARE_FAIL_KEEP_FILE
1528         return (1);
1529 #endif
1530 }
1531 #endif /* !COMPARE_ONTHEFLY */
1532
1533 #if !COMPARE_ONTHEFLY
1534 static char tmpfilename[MAXPATHLEN];
1535 #endif
1536
1537 void
1538 comparefile(char *name)
1539 {
1540         unsigned int mode;
1541         struct STAT sb;
1542         int r;
1543 #if !COMPARE_ONTHEFLY
1544         static char *tmpfile = NULL;
1545         struct STAT stemp;
1546 #endif
1547
1548         curfile.name = name;
1549         curfile.action = USING;
1550         mode = curfile.dip->di_mode;
1551
1552         if ((mode & IFMT) == IFSOCK) {
1553                 Vprintf(stdout, "skipped socket %s\n", name);
1554                 skipfile();
1555                 return;
1556         }
1557
1558         if ((r = LSTAT(name, &sb)) != 0) {
1559                 warn("%s: does not exist (%d)", name, r);
1560                 do_compare_error;
1561                 skipfile();
1562                 return;
1563         }
1564
1565         Vprintf(stdout, "comparing %s (size: %ld, mode: 0%o)\n", name,
1566                 (long)sb.st_size, mode);
1567
1568         if (sb.st_mode != mode) {
1569                 fprintf(stderr, "%s: mode changed from 0%o to 0%o.\n",
1570                         name, mode & 07777, sb.st_mode & 07777);
1571                 do_compare_error;
1572         }
1573         if (spcl.c_flags & DR_METAONLY) {
1574                 skipfile();
1575                 return;
1576         }
1577         switch (mode & IFMT) {
1578         default:
1579                 skipfile();
1580                 return;
1581
1582         case IFSOCK:
1583                 skipfile();
1584                 return;
1585
1586         case IFDIR:
1587                 skipfile();
1588                 return;
1589
1590         case IFLNK: {
1591                 char lbuf[MAXPATHLEN + 1];
1592                 int lsize;
1593
1594                 if (!(sb.st_mode & S_IFLNK)) {
1595                         fprintf(stderr, "%s: is no longer a symbolic link\n",
1596                                 name);
1597                         do_compare_error;
1598                         return;
1599                 }
1600                 lnkbuf[0] = '\0';
1601                 pathlen = 0;
1602                 getfile(xtrlnkfile, xtrlnkskip);
1603                 if (pathlen == 0) {
1604                         fprintf(stderr,
1605                                 "%s: zero length symbolic link (ignored)\n",
1606                                 name);
1607                         do_compare_error;
1608                         return;
1609                 }
1610                 if ((lsize = readlink(name, lbuf, MAXPATHLEN)) < 0) {
1611                         panic("readlink of %s failed: %s", name,
1612                               strerror(errno));
1613                         do_compare_error;
1614                 }
1615                 lbuf[lsize] = 0;
1616                 if (strcmp(lbuf, lnkbuf) != 0) {
1617                         fprintf(stderr,
1618                                 "%s: symbolic link changed from %s to %s.\n",
1619                                 name, lnkbuf, lbuf);
1620                         do_compare_error;
1621                         return;
1622                 }
1623                 return;
1624         }
1625
1626         case IFCHR:
1627         case IFBLK:
1628                 if (!(sb.st_mode & (S_IFCHR|S_IFBLK))) {
1629                         fprintf(stderr, "%s: no longer a special file\n",
1630                                 name);
1631                         do_compare_error;
1632                         skipfile();
1633                         return;
1634                 }
1635
1636                 if (sb.st_rdev != (dev_t)curfile.dip->di_rdev) {
1637                         fprintf(stderr,
1638                                 "%s: device changed from %d,%d to %d,%d.\n",
1639                                 name,
1640                                 ((int)curfile.dip->di_rdev >> 8) & 0xff,
1641                                 (int)curfile.dip->di_rdev & 0xff,
1642                                 ((int)sb.st_rdev >> 8) & 0xff,
1643                                 (int)sb.st_rdev & 0xff);
1644                         do_compare_error;
1645                 }
1646                 skipfile();
1647                 return;
1648
1649         case IFREG:
1650 #if COMPARE_ONTHEFLY
1651                 if ((ifile = OPEN(name, O_RDONLY)) < 0) {
1652                         panic("Can't open %s: %s\n", name, strerror(errno));
1653                         skipfile();
1654                         do_compare_error;
1655                 }
1656                 else {
1657                         cmperror = 0;
1658                         getfile(xtrcmpfile, xtrcmpskip);
1659                         if (!cmperror) {
1660                                 char c;
1661                                 if (read(ifile, &c, 1) != 0) {
1662                                         fprintf(stderr, "%s: size has changed.\n", 
1663                                                 name);
1664                                         cmperror = 1;
1665                                 }
1666                         }
1667                         if (cmperror)
1668                                 do_compare_error;
1669                         close(ifile);
1670                 }
1671 #else
1672                 if (tmpfile == NULL) {
1673                         /* argument to mktemp() must not be in RO space: */
1674                         snprintf(tmpfilename, sizeof(tmpfilename), "%s/restoreCXXXXXX", tmpdir);
1675                         tmpfile = mktemp(&tmpfilename[0]);
1676                 }
1677                 if ((STAT(tmpfile, &stemp) == 0) && (unlink(tmpfile) != 0)) {
1678                         panic("cannot delete tmp file %s: %s\n",
1679                               tmpfile, strerror(errno));
1680                 }
1681                 if ((ofile = OPEN(tmpfile, O_WRONLY | O_CREAT | O_TRUNC, 0600)) < 0) {
1682                         panic("cannot create file temp file %s: %s\n",
1683                               name, strerror(errno));
1684                 }
1685                 getfile(xtrfile, xtrskip);
1686                 (void) close(ofile);
1687 #ifdef COMPARE_FAIL_KEEP_FILE
1688                 if (cmpfiles(tmpfile, name, &sb))
1689                         unlink(tmpfile);
1690 #else
1691                 cmpfiles(tmpfile, name, &sb);
1692                 unlink(tmpfile);
1693 #endif
1694 #endif /* COMPARE_ONTHEFLY */
1695                 return;
1696         }
1697         /* NOTREACHED */
1698 }
1699
1700 #if defined(HAVE_ZLIB) || defined(HAVE_BZLIB) || defined(HAVE_LZO)
1701 static void (*readtape_func)(char *) = readtape_set;
1702
1703 /*
1704  * Read TP_BSIZE blocks from the input.
1705  * Handle read errors, and end of media.
1706  * Decompress compressed blocks.
1707  */
1708 static void
1709 readtape(char *buf)
1710 {
1711         (*readtape_func)(buf);  /* call the actual processing routine */
1712 }
1713
1714 /*
1715  * Set function pointer for readtape() routine. zflag and magtapein must
1716  * be correctly set before the first call to readtape().
1717  */
1718 static void
1719 readtape_set(char *buf)
1720 {
1721         if (!zflag) 
1722                 readtape_func = readtape_uncompr;
1723         else {
1724                 newcomprbuf(ntrec);
1725                 if (magtapein)
1726                         readtape_func = readtape_comprtape;
1727                 else
1728                         readtape_func = readtape_comprfile;
1729         }
1730         readtape(buf);
1731 }
1732
1733 #endif /* HAVE_ZLIB || HAVE_BZLIB || HAVE_LZO */
1734
1735 /*
1736  * This is the original readtape(), it's used for reading uncompressed input.
1737  * Read TP_BSIZE blocks from the input.
1738  * Handle read errors, and end of media.
1739  */
1740 static void
1741 #if defined(HAVE_ZLIB) || defined(HAVE_BZLIB) || defined(HAVE_LZO)
1742 readtape_uncompr(char *buf)
1743 #else
1744 readtape(char *buf)
1745 #endif
1746 {
1747         ssize_t rd, newvol, i;
1748         int cnt, seek_failed;
1749
1750         if (blkcnt < numtrec) {
1751                 memmove(buf, &tbufptr[(blkcnt++ * TP_BSIZE)], TP_BSIZE);
1752                 blksread++;
1753                 tpblksread++;
1754                 return;
1755         }
1756         tbufptr = tapebuf;
1757         for (i = 0; i < ntrec; i++)
1758                 ((struct s_spcl *)&tapebuf[i * TP_BSIZE])->c_magic = 0;
1759         if (numtrec == 0)
1760                 numtrec = ntrec;
1761         cnt = ntrec * TP_BSIZE;
1762         rd = 0;
1763 #ifdef USE_QFA
1764         if (createtapeposflag)
1765                 (void)GetTapePos(&curtapepos);
1766 #endif
1767 getmore:
1768 #ifdef RRESTORE
1769         if (!Afile && host)
1770                 i = rmtread(&tapebuf[rd], cnt);
1771         else
1772 #endif
1773                 i = read(mt, &tapebuf[rd], cnt);
1774
1775         /*
1776          * Check for mid-tape short read error.
1777          * If found, skip rest of buffer and start with the next.
1778          */
1779         if (!pipein && numtrec < ntrec && i > 0) {
1780                 Dprintf(stdout, "mid-media short read error.\n");
1781                 numtrec = ntrec;
1782         }
1783         /*
1784          * Handle partial block read.
1785          */
1786         if (pipein && i == 0 && rd > 0)
1787                 i = rd;
1788         else if (i > 0 && i != ntrec * TP_BSIZE) {
1789                 if (pipein) {
1790                         rd += i;
1791                         cnt -= i;
1792                         if (cnt > 0)
1793                                 goto getmore;
1794                         i = rd;
1795                 } else {
1796                         /*
1797                          * Short read. Process the blocks read.
1798                          */
1799                         if (i % TP_BSIZE != 0)
1800                                 Vprintf(stdout,
1801                                     "partial block read: %ld should be %ld\n",
1802                                     (long)i, ntrec * TP_BSIZE);
1803                         numtrec = i / TP_BSIZE;
1804                 }
1805         }
1806         /*
1807          * Handle read error.
1808          */
1809         if (i < 0) {
1810                 fprintf(stderr, "Tape read error while ");
1811                 switch (curfile.action) {
1812                 default:
1813                         fprintf(stderr, "trying to set up tape\n");
1814                         break;
1815                 case UNKNOWN:
1816                         fprintf(stderr, "trying to resynchronize\n");
1817                         break;
1818                 case USING:
1819                         fprintf(stderr, "restoring %s\n", curfile.name);
1820                         break;
1821                 case SKIP:
1822                         fprintf(stderr, "skipping over inode %lu\n",
1823                                 (unsigned long)curfile.ino);
1824                         break;
1825                 }
1826                 if (!yflag && !reply("continue"))
1827                         exit(1);
1828                 i = ntrec * TP_BSIZE;
1829                 memset(tapebuf, 0, (size_t)i);
1830 #ifdef RRESTORE
1831                 if (!Afile && host)
1832                         seek_failed = (rmtseek(i, 1) < 0);
1833                 else
1834 #endif
1835                         seek_failed = (LSEEK(mt, i, SEEK_CUR) == (OFF_T)-1);
1836
1837                 if (seek_failed) {
1838                         warn("continuation failed");
1839                         if (!yflag && !reply("assume end-of-tape and continue"))
1840                                 exit(1);
1841                         i = 0;
1842                 }
1843         }
1844         /*
1845          * Handle end of tape.
1846          */
1847         if (i == 0) {
1848                 Vprintf(stdout, "End-of-tape encountered\n");
1849                 if (!pipein) {
1850                         newvol = volno + 1;
1851                         volno = 0;
1852                         numtrec = 0;
1853                         getvol(newvol);
1854                         readtape(buf);
1855                         return;
1856                 }
1857                 if (rd % TP_BSIZE != 0)
1858                         panic("partial block read: %d should be %d\n",
1859                                 rd, ntrec * TP_BSIZE);
1860                 terminateinput();
1861                 memmove(&tapebuf[rd], &endoftapemark, TP_BSIZE);
1862         }
1863         blkcnt = 0;
1864         memmove(buf, &tbufptr[(blkcnt++ * TP_BSIZE)], TP_BSIZE);
1865         blksread++;
1866         tpblksread++;
1867 }
1868
1869 #if defined(HAVE_ZLIB) || defined(HAVE_BZLIB) || defined(HAVE_LZO)
1870
1871 /*
1872  * Read a compressed format block from a file or pipe and uncompress it.
1873  * Attempt to handle read errors, and end of file. 
1874  */
1875 static void
1876 readtape_comprfile(char *buf)
1877 {
1878         long rl, size, i, ret;
1879         int newvol; 
1880         struct tapebuf *tpb;
1881
1882         if (blkcnt < numtrec) {
1883                 memmove(buf, &tbufptr[(blkcnt++ * TP_BSIZE)], TP_BSIZE);
1884                 blksread++;
1885                 tpblksread++;
1886                 return;
1887         }
1888         /* need to read the next block */
1889         tbufptr = tapebuf;
1890         for (i = 0; i < ntrec; i++)
1891                 ((struct s_spcl *)&tapebuf[i * TP_BSIZE])->c_magic = 0;
1892         numtrec = ntrec;
1893         tpb = (struct tapebuf *) tapebuf;
1894
1895         /* read the block prefix */
1896         ret = read_a_block(mt, tapebuf, PREFIXSIZE, &rl);
1897         converttapebuf(tpb);
1898
1899         if (Vflag && (ret == 0 || rl < (int)PREFIXSIZE  ||  tpb->length == 0))
1900                 ret = 0;
1901         if (ret <= 0)
1902                 goto readerr;
1903
1904         /* read the data */
1905         size = tpb->length;
1906         if (size > bufsize)  {
1907                 /* something's wrong */
1908                 Vprintf(stdout, "Prefix size error, max size %d, got %ld\n",
1909                         bufsize, size);
1910                 size = bufsize;
1911                 tpb->length = bufsize;
1912         }
1913         ret = read_a_block(mt, tpb->buf, size, &rl);
1914         if (ret <= 0)
1915                 goto readerr;
1916
1917         tbufptr = decompress_tapebuf(tpb, rl + PREFIXSIZE);
1918         if (tbufptr == NULL) {
1919                 msg_read_error("File decompression error while");
1920                 if (!yflag && !reply("continue"))
1921                         exit(1);
1922                 memset(tapebuf, 0, bufsize);
1923                 tbufptr = tapebuf;
1924         }
1925
1926         blkcnt = 0;
1927         memmove(buf, &tbufptr[(blkcnt++ * TP_BSIZE)], TP_BSIZE);
1928         blksread++;
1929         tpblksread++;
1930         return;
1931
1932 readerr:
1933         /* Errors while reading from a file or pipe are catastrophic. Since
1934          * there are no block boundaries, it's impossible to bypass the
1935          * block in error and find the start of the next block.
1936          */
1937         if (ret == 0) {
1938                 /* It's possible to have multiple input files using -M
1939                  * and -f file1,file2...
1940                  */
1941                 Vprintf(stdout, "End-of-File encountered\n");
1942                 if (!pipein) {
1943                         newvol = volno + 1;
1944                         volno = 0;
1945                         numtrec = 0;
1946                         getvol(newvol);
1947                         readtape(buf);
1948                         return;
1949                 }
1950         }
1951         msg_read_error("Read error while");
1952         /* if (!yflag && !reply("continue")) */
1953                 exit(1);
1954 }
1955
1956 /*
1957  * Read compressed data from a tape and uncompress it.
1958  * Handle read errors, and end of media.
1959  * Since a tape consists of separate physical blocks, we try
1960  * to recover from errors by repositioning the tape to the next
1961  * block.
1962  */
1963 static void
1964 readtape_comprtape(char *buf)
1965 {
1966         long rl, size, i;
1967         int ret, newvol;
1968         struct tapebuf *tpb;
1969         struct mtop tcom;
1970
1971         if (blkcnt < numtrec) {
1972                 memmove(buf, &tbufptr[(blkcnt++ * TP_BSIZE)], TP_BSIZE);
1973                 blksread++;
1974                 tpblksread++;
1975                 return;
1976         }
1977         /* need to read the next block */
1978         tbufptr = tapebuf;
1979         for (i = 0; i < ntrec; i++)
1980                 ((struct s_spcl *)&tapebuf[i * TP_BSIZE])->c_magic = 0;
1981         numtrec = ntrec;
1982         tpb = (struct tapebuf *) tapebuf;
1983
1984         /* read the block */
1985         size = bufsize + PREFIXSIZE;
1986         ret = read_a_block(mt, tapebuf, size, &rl);
1987         if (ret <= 0)
1988                 goto readerr;
1989
1990         converttapebuf(tpb);
1991         tbufptr = decompress_tapebuf(tpb, rl);
1992         if (tbufptr == NULL) {
1993                 msg_read_error("Tape decompression error while");
1994                 if (!yflag && !reply("continue"))
1995                         exit(1);
1996                 memset(tapebuf, 0, PREFIXSIZE + bufsize);
1997                 tbufptr = tapebuf;
1998         }
1999         goto moverecord;
2000
2001 readerr:
2002         /* Handle errors: EOT switches to the next volume, other errors
2003          * attempt to position the tape to the next block.
2004          */
2005         if (ret == 0) {
2006                 Vprintf(stdout, "End-of-tape encountered\n");
2007                 newvol = volno + 1;
2008                 volno = 0;
2009                 numtrec = 0;
2010                 getvol(newvol);
2011                 readtape(buf);
2012                 return;
2013         }
2014
2015         msg_read_error("Tape read error while");
2016         if (!yflag && !reply("continue"))
2017                 exit(1);
2018         memset(tapebuf, 0, PREFIXSIZE + bufsize);
2019         tbufptr = tapebuf;
2020
2021 #ifdef RRESTORE
2022         if (host)
2023                 rl = rmtioctl(MTFSR, 1);
2024         else
2025 #endif
2026         {
2027                 tcom.mt_op = MTFSR;
2028                 tcom.mt_count = 1;
2029                 rl = ioctl(mt, MTIOCTOP, &tcom);
2030         }
2031
2032         if (rl < 0) {
2033                 warn("continuation failed");
2034                 if (!yflag && !reply("assume end-of-tape and continue"))
2035                         exit(1);
2036                 ret = 0;         /* end of tape */
2037                 goto readerr;
2038         }
2039
2040 moverecord:
2041         blkcnt = 0;
2042         memmove(buf, &tbufptr[(blkcnt++ * TP_BSIZE)], TP_BSIZE);
2043         blksread++;
2044         tpblksread++;
2045 }
2046
2047 /*
2048  *  Decompress a struct tapebuf into a buffer. readsize is the size read
2049  *  from the tape/file and is used for error messages. Returns a pointer
2050  *  to the location of the uncompressed buffer or NULL on errors.
2051  *  Adjust numtrec and complain for a short block.
2052  */
2053 static char *
2054 decompress_tapebuf(struct tapebuf *tpbin, int readsize)
2055 {
2056         /* If zflag is on, all blocks have a struct tapebuf prefix */
2057         /* zflag gets set in setup() from the dump header          */
2058         int cresult, blocklen;        
2059         unsigned long worklen;
2060 #ifdef HAVE_BZLIB
2061         unsigned int worklen2;
2062 #endif
2063         char *output = NULL,*reason = NULL, *lengtherr = NULL;              
2064        
2065         /* build a length error message */
2066         blocklen = tpbin->length;
2067         if (readsize < blocklen + (int)PREFIXSIZE)
2068                 lengtherr = "short";
2069         else
2070                 if (readsize > blocklen + (int)PREFIXSIZE)
2071                         lengtherr = "long";
2072
2073         worklen = comprlen;
2074         cresult = 1;
2075         if (tpbin->compressed) {
2076                 /* uncompress whatever we read, if it fails, complain later */
2077                 if (tpbin->flags == COMPRESS_ZLIB) {
2078 #ifndef HAVE_ZLIB
2079                         errx(1,"This restore version doesn't support zlib decompression");
2080 #else
2081                         cresult = uncompress(comprbuf, &worklen, 
2082                                              tpbin->buf, blocklen);
2083                         output = comprbuf;
2084                         switch (cresult) {
2085                                 case Z_OK:
2086                                         break;
2087                                 case Z_MEM_ERROR:
2088                                         reason = "not enough memory";
2089                                         break;
2090                                 case Z_BUF_ERROR:
2091                                         reason = "buffer too small";
2092                                         break;
2093                                 case Z_DATA_ERROR:
2094                                         reason = "data error";
2095                                         break;
2096                                 default:
2097                                         reason = "unknown";
2098                         }
2099                         if (cresult == Z_OK)
2100                                 cresult = 1;
2101                         else
2102                                 cresult = 0;
2103 #endif /* HAVE_ZLIB */
2104                 }
2105                 if (tpbin->flags == COMPRESS_BZLIB) {
2106 #ifndef HAVE_BZLIB
2107                         errx(1,"This restore version doesn't support bzlib decompression");
2108 #else
2109                         worklen2 = worklen;
2110                         cresult = BZ2_bzBuffToBuffDecompress(
2111                                         comprbuf, &worklen2, 
2112                                         tpbin->buf, blocklen, 0, 0);
2113                         worklen = worklen2;
2114                         output = comprbuf;
2115                         switch (cresult) {
2116                                 case BZ_OK:
2117                                         break;
2118                                 case BZ_MEM_ERROR:
2119                                         reason = "not enough memory";
2120                                         break;
2121                                 case BZ_OUTBUFF_FULL:
2122                                         reason = "buffer too small";
2123                                         break;
2124                                 case BZ_DATA_ERROR:
2125                                 case BZ_DATA_ERROR_MAGIC:
2126                                 case BZ_UNEXPECTED_EOF:
2127                                         reason = "data error";
2128                                         break;
2129                                 default:
2130                                         reason = "unknown";
2131                         }
2132                         if (cresult == BZ_OK)
2133                                 cresult = 1;
2134                         else
2135                                 cresult = 0;
2136 #endif /* HAVE_BZLIB */
2137                 }
2138                 if (tpbin->flags == COMPRESS_LZO) {
2139 #ifndef HAVE_LZO
2140                         errx(1,"This restore version doesn't support lzo decompression");
2141 #else
2142                         cresult = lzo1x_decompress(tpbin->buf, blocklen,
2143                                                    comprbuf, (lzo_uintp) &worklen,NULL);
2144                         output = comprbuf;
2145                         switch (cresult) {
2146                                 case LZO_E_OK:
2147                                         break;
2148                                 case LZO_E_ERROR:
2149                                 case LZO_E_EOF_NOT_FOUND:
2150                                         reason = "data error";
2151                                         break;
2152                                 default:
2153                                         reason = "unknown";
2154                         }
2155                         if (cresult == LZO_E_OK)
2156                                 cresult = 1;
2157                         else
2158                                 cresult = 0;
2159 #endif /* HAVE_LZO */
2160                 }
2161         }
2162         else {
2163                 output = tpbin->buf;
2164                 worklen = blocklen;
2165         }
2166         if (cresult) {
2167                 numtrec = worklen / TP_BSIZE;
2168                 if (worklen % TP_BSIZE != 0)
2169                         reason = "length mismatch";
2170         }
2171         if (reason) {
2172                 if (lengtherr)
2173                         fprintf(stderr, "%s compressed block: %d expected: %d\n",
2174                                 lengtherr, readsize, tpbin->length + PREFIXSIZE);
2175                 fprintf(stderr, "decompression error, block %ld: %s\n",
2176                         tpblksread+1, reason);
2177                 if (!cresult)
2178                         output = NULL;
2179         }
2180         return output;
2181 }
2182
2183 /*
2184  * Print an error message for a read error.
2185  * This was exteracted from the original readtape().
2186  */
2187 static void
2188 msg_read_error(char *m)
2189 {
2190         switch (curfile.action) {
2191                 default:
2192                         fprintf(stderr, "%s trying to set up tape\n", m);
2193                         break;
2194                 case UNKNOWN:
2195                         fprintf(stderr, "%s trying to resynchronize\n", m);
2196                         break;
2197                 case USING:
2198                         fprintf(stderr, "%s restoring %s\n", m, curfile.name);
2199                         break;
2200                 case SKIP:
2201                         fprintf(stderr, "%s skipping over inode %lu\n", m,
2202                                 (unsigned long)curfile.ino);
2203                         break;
2204         }
2205 }
2206 #endif /* HAVE_ZLIB || HAVE_BZLIB || HAVE_LZO */
2207
2208 /*
2209  * Read the first block and get the blocksize from it. Test
2210  * for a compressed dump tape/file. setup() will make the final
2211  * determination by checking the compressed flag if gethead()
2212  * finds a valid header. The test here is necessary to offset the buffer
2213  * by the size of the compressed prefix. zflag is set here so that
2214  * readtape_set can set the correct function pointer for readtape().
2215  * Note that the first block of each tape/file is not compressed
2216  * and does not have a prefix.
2217  */ 
2218 static void
2219 findtapeblksize(void)
2220 {
2221         long i;
2222         size_t len;
2223         struct tapebuf *tpb = (struct tapebuf *) tapebuf;
2224         struct s_spcl spclpt;
2225
2226         for (i = 0; i < ntrec; i++)
2227                 ((struct s_spcl *)&tapebuf[i * TP_BSIZE])->c_magic = 0;
2228         blkcnt = 0;
2229         tbufptr = tapebuf;
2230         /*
2231          * For a pipe or file, read in the first record. For a tape, read
2232          * the first block.
2233          */
2234         len = magtapein ? ntrec * TP_BSIZE : TP_BSIZE;
2235
2236         if (read_a_block(mt, tapebuf, len, &i) <= 0)
2237                 errx(1, "Tape read error on first record");
2238
2239         memcpy(&spclpt, tapebuf, TP_BSIZE);
2240         if (converthead(&spclpt) == FAIL) {
2241                 cvtflag++;
2242                 if (converthead(&spclpt) == FAIL) {
2243                         /* Special case for old compressed tapes with prefix */
2244                         if (magtapein && (i % TP_BSIZE != 0)) 
2245                                 goto oldformat;
2246                         errx(1, "Tape is not a dump tape");
2247                 }
2248                 fprintf(stderr, "Converting to new file system format.\n");
2249         }
2250         /*
2251          * If the input is from a file or a pipe, we read TP_BSIZE
2252          * bytes looking for a dump header. If the dump is compressed
2253          * we need to read in the rest of the block, as determined
2254          * by c_ntrec in the dump header. The first block of the
2255          * dump is not compressed and does not have a prefix.
2256          */
2257         if (!magtapein) {
2258                 if (spclpt.c_type == TS_TAPE
2259                     && spclpt.c_flags & DR_COMPRESSED) {
2260                         /* It's a compressed dump file, read in the */
2261                         /* rest of the block based on spclpt.c_ntrec. */
2262                         if (spclpt.c_ntrec > ntrec)
2263                                 errx(1, "Tape blocksize is too large, use "
2264                                      "\'-b %d\' ", spclpt.c_ntrec);
2265                         ntrec = spclpt.c_ntrec;
2266                         len = (ntrec - 1) * TP_BSIZE;
2267                         zflag = 1;   
2268                 }
2269                 else {
2270                         /* read in the rest of the block based on bufsize */
2271                         len = bufsize - TP_BSIZE;
2272                 }
2273                 if (read_a_block(mt, tapebuf+TP_BSIZE, len, &i) < 0
2274                     || (i != (long)len && i % TP_BSIZE != 0))
2275                         errx(1,"Error reading dump file header");
2276                 tbufptr = tapebuf;
2277                 numtrec = ntrec;
2278                 Vprintf(stdout, "Input block size is %ld\n", ntrec);
2279                 return;
2280         } /* if (!magtapein) */
2281
2282         /*
2283          * If the input is a tape, we tried to read ntrec * TP_BSIZE bytes.
2284          * If the value of ntrec is too large, we read less than
2285          * what we asked for; adjust the value of ntrec and test for 
2286          * a compressed dump tape.
2287          */
2288         if (i % TP_BSIZE != 0) {
2289 oldformat:
2290                 /* may be old format compressed dump tape with a prefix */
2291                 memcpy(&spclpt, tpb->buf, TP_BSIZE);
2292                 cvtflag = 0;
2293                 if (converthead(&spclpt) == FAIL) {
2294                         cvtflag++;
2295                         if (converthead(&spclpt) == FAIL)
2296                                 errx(1, "Tape is not a dump tape");
2297                         fprintf(stderr, "Converting to new file system format.\n");
2298                 }
2299                 if (i % TP_BSIZE == PREFIXSIZE
2300                     && tpb->compressed == 0
2301                     && spclpt.c_type == TS_TAPE
2302                     && spclpt.c_flags & DR_COMPRESSED) {
2303                         zflag = 1;
2304                         tbufptr = tpb->buf;
2305                         if (tpb->length > bufsize)
2306                                 errx(1, "Tape blocksize is too large, use "
2307                                         "\'-b %d\' ", tpb->length / TP_BSIZE);
2308                 }
2309                 else
2310                         errx(1, "Tape block size (%ld) is not a multiple of dump block size (%d)",
2311                                 i, TP_BSIZE);
2312         }
2313         ntrec = i / TP_BSIZE;
2314         if (spclpt.c_type == TS_TAPE) {
2315                 if (spclpt.c_flags & DR_COMPRESSED)
2316                         zflag = 1;
2317                 if (spclpt.c_ntrec > ntrec)
2318                         errx(1, "Tape blocksize is too large, use "
2319                                 "\'-b %d\' ", spclpt.c_ntrec);
2320         }
2321         numtrec = ntrec;
2322         Vprintf(stdout, "Tape block size is %ld\n", ntrec);
2323 }
2324
2325 /*
2326  * Read a block of data handling all of the messy details.
2327  */
2328 static int read_a_block(int fd, char *buf, size_t len, long *lengthread)
2329 {
2330         long i = 1, size;
2331
2332         size = len;
2333         while (size > 0) {
2334 #ifdef RRESTORE
2335                 if (!Afile && host)
2336                         i = rmtread(buf, size);
2337                 else
2338 #endif
2339                         i = read(fd, buf, size);                 
2340
2341                 if (i <= 0)
2342                         break; /* EOD or error */
2343                 size -= i;
2344                 if (magtapein)
2345                         break; /* block at a time for mt */
2346                 buf += i;
2347         }
2348         *lengthread = len - size;
2349         return i;
2350 }
2351
2352 void
2353 closemt(void)
2354 {
2355
2356         if (mt < 0)
2357                 return;
2358 #ifdef RRESTORE
2359         if (!Afile && host)
2360                 rmtclose();
2361         else
2362 #endif
2363                 (void) close(mt);
2364 }
2365
2366 static void
2367 setmagtapein(void) {
2368         struct mtget mt_stat;
2369         static int done = 0;
2370         if (done)
2371                 return;
2372         done = 1;
2373         if (!pipein) {
2374                 /* need to know if input is really from a tape */
2375 #ifdef RRESTORE
2376                 if (host)
2377                         magtapein = !lflag;
2378                 else
2379 #endif
2380                         magtapein = ioctl(mt, MTIOCGET, (char *)&mt_stat) == 0;
2381         }
2382
2383         Vprintf(stdout,"Input is from a %s %s\n",
2384                         host ? "remote" : "local",
2385                         magtapein ? "tape" :
2386                         Vflag ? "multi-volume (no tape)" : "file/pipe");
2387 }
2388
2389 /*
2390  * Read the next block from the tape.
2391  * Check to see if it is one of several vintage headers.
2392  * If it is an old style header, convert it to a new style header.
2393  * If it is not any valid header, return an error.
2394  */
2395 static int
2396 gethead(struct s_spcl *buf)
2397 {
2398         readtape((char *)buf);
2399         return converthead(buf);
2400 }
2401
2402 static int
2403 converthead(struct s_spcl *buf)
2404 {
2405         int32_t i;
2406         union {
2407                 quad_t  qval;
2408                 int32_t val[2];
2409         } qcvt;
2410         union u_ospcl {
2411                 char dummy[TP_BSIZE];
2412                 struct  s_ospcl {
2413                         int32_t c_type;
2414                         int32_t c_date;
2415                         int32_t c_ddate;
2416                         int32_t c_volume;
2417                         int32_t c_tapea;
2418                         u_int16_t c_inumber;
2419                         int32_t c_magic;
2420                         int32_t c_checksum;
2421                         struct odinode {
2422                                 u_int16_t odi_mode;
2423                                 u_int16_t odi_nlink;
2424                                 u_int16_t odi_uid;
2425                                 u_int16_t odi_gid;
2426                                 int32_t odi_size;
2427                                 int32_t odi_rdev;
2428                                 char    odi_addr[36];
2429                                 int32_t odi_atime;
2430                                 int32_t odi_mtime;
2431                                 int32_t odi_ctime;
2432                         } c_dinode;
2433                         int32_t c_count;
2434                         char    c_fill[256];
2435                 } s_ospcl;
2436         } u_ospcl;
2437
2438         if (!cvtflag) {
2439                 if (buf->c_magic != NFS_MAGIC) {
2440                         if (swabi(buf->c_magic) != NFS_MAGIC)
2441                                 return (FAIL);
2442                         if (!Bcvt) {
2443                                 Vprintf(stdout, "Note: Doing Byte swapping\n");
2444                                 Bcvt = 1;
2445                         }
2446                 }
2447                 if (checksum((int *)buf) == FAIL)
2448                         return (FAIL);
2449                 if (Bcvt)
2450                         swabst((u_char *)"8i4s31i528bi192b3i", (u_char *)buf);
2451                 goto good;
2452         }
2453         memcpy(&u_ospcl.s_ospcl, buf, TP_BSIZE);
2454         memset((char *)buf, 0, (long)TP_BSIZE);
2455         buf->c_type = u_ospcl.s_ospcl.c_type;
2456         buf->c_date = u_ospcl.s_ospcl.c_date;
2457         buf->c_ddate = u_ospcl.s_ospcl.c_ddate;
2458         buf->c_volume = u_ospcl.s_ospcl.c_volume;
2459         buf->c_tapea = u_ospcl.s_ospcl.c_tapea;
2460         buf->c_inumber = u_ospcl.s_ospcl.c_inumber;
2461         buf->c_checksum = u_ospcl.s_ospcl.c_checksum;
2462         buf->c_magic = u_ospcl.s_ospcl.c_magic;
2463         buf->c_dinode.di_mode = u_ospcl.s_ospcl.c_dinode.odi_mode;
2464         buf->c_dinode.di_nlink = u_ospcl.s_ospcl.c_dinode.odi_nlink;
2465         buf->c_dinode.di_uid = u_ospcl.s_ospcl.c_dinode.odi_uid;
2466         buf->c_dinode.di_gid = u_ospcl.s_ospcl.c_dinode.odi_gid;
2467         buf->c_dinode.di_size = u_ospcl.s_ospcl.c_dinode.odi_size;
2468         buf->c_dinode.di_rdev = u_ospcl.s_ospcl.c_dinode.odi_rdev;
2469 #if defined(__linux__) || defined(sunos)
2470         buf->c_dinode.di_atime.tv_sec = u_ospcl.s_ospcl.c_dinode.odi_atime;
2471         buf->c_dinode.di_mtime.tv_sec = u_ospcl.s_ospcl.c_dinode.odi_mtime;
2472         buf->c_dinode.di_ctime.tv_sec = u_ospcl.s_ospcl.c_dinode.odi_ctime;
2473 #else   /* __linux__ || sunos */
2474         buf->c_dinode.di_atime = u_ospcl.s_ospcl.c_dinode.odi_atime;
2475         buf->c_dinode.di_mtime = u_ospcl.s_ospcl.c_dinode.odi_mtime;
2476         buf->c_dinode.di_ctime = u_ospcl.s_ospcl.c_dinode.odi_ctime;
2477 #endif  /* __linux__ || sunos */
2478         buf->c_count = u_ospcl.s_ospcl.c_count;
2479         memmove(buf->c_addr, u_ospcl.s_ospcl.c_fill, (long)256);
2480         if (u_ospcl.s_ospcl.c_magic != OFS_MAGIC ||
2481             checksum((int *)(&u_ospcl.s_ospcl)) == FAIL)
2482                 return(FAIL);
2483         buf->c_magic = NFS_MAGIC;
2484
2485 good:
2486         if ((buf->c_dinode.di_size == 0 || buf->c_dinode.di_size > 0xfffffff) &&
2487             (buf->c_dinode.di_mode & IFMT) == IFDIR && Qcvt == 0) {
2488                 qcvt.qval = buf->c_dinode.di_size;
2489                 if (qcvt.val[0] || qcvt.val[1]) {
2490                         Vprintf(stdout, "Note: Doing Quad swapping\n");
2491                         Qcvt = 1;
2492                 }
2493         }
2494         if (Qcvt) {
2495                 qcvt.qval = buf->c_dinode.di_size;
2496                 i = qcvt.val[1];
2497                 qcvt.val[1] = qcvt.val[0];
2498                 qcvt.val[0] = i;
2499                 buf->c_dinode.di_size = qcvt.qval;
2500         }
2501         readmapflag = 0;
2502
2503         switch (buf->c_type) {
2504
2505         case TS_CLRI:
2506         case TS_BITS:
2507                 /*
2508                  * Have to patch up missing information in bit map headers
2509                  */
2510                 buf->c_inumber = 0;
2511                 buf->c_dinode.di_size = buf->c_count * TP_BSIZE;
2512                 if (buf->c_count > TP_NINDIR)
2513                         readmapflag = 1;
2514                 else 
2515                         for (i = 0; i < buf->c_count; i++)
2516                                 buf->c_addr[i]++;
2517                 break;
2518
2519         case TS_TAPE:
2520                 if ((buf->c_flags & DR_NEWINODEFMT) == 0)
2521                         oldinofmt = 1;
2522                 /* fall through */
2523         case TS_END:
2524                 buf->c_inumber = 0;
2525                 if (buf->c_flags & DR_INODEINFO) {
2526                         memcpy(volinfo, buf->c_inos, TP_NINOS * sizeof(dump_ino_t));
2527                         if (Bcvt)
2528                                 swabst((u_char *)"128i", (u_char *)volinfo);
2529                 }
2530                 break;
2531
2532         case TS_INODE:
2533         case TS_ADDR:
2534                 break;
2535
2536         default:
2537                 panic("gethead: unknown inode type %d\n", buf->c_type);
2538                 break;
2539         }
2540         /*
2541          * If we are restoring a filesystem with old format inodes,
2542          * copy the uid/gid to the new location.
2543          */
2544         if (oldinofmt) {
2545                 buf->c_dinode.di_uid = buf->c_dinode.di_ouid;
2546                 buf->c_dinode.di_gid = buf->c_dinode.di_ogid;
2547         }
2548         if (dflag)
2549                 accthdr(buf);
2550         return(GOOD);
2551 }
2552
2553 static void
2554 converttapebuf(struct tapebuf *tpb)
2555 {
2556         if (Bcvt) {
2557                 struct tb {
2558                         unsigned int    length:28;
2559                         unsigned int    flags:3;
2560                         unsigned int    compressed:1;
2561                 } tb;
2562                 swabst((u_char *)"i", (u_char *)tpb);
2563                 memcpy(&tb, tpb, 4);    
2564                 tpb->length = tb.length;
2565                 tpb->flags = tb.flags;
2566                 tpb->compressed = tb.compressed;
2567         }
2568 }
2569
2570 /*
2571  * Check that a header is where it belongs and predict the next header
2572  */
2573 static void
2574 accthdr(struct s_spcl *header)
2575 {
2576         static dump_ino_t previno = 0x7fffffff;
2577         static int prevtype;
2578         static long predict;
2579         long blks, i;
2580
2581         if (header->c_type == TS_TAPE) {
2582                 fprintf(stderr, "Volume header (%s inode format) ",
2583                     oldinofmt ? "old" : "new");
2584                 if (header->c_firstrec)
2585                         fprintf(stderr, "begins with record %d",
2586                                 header->c_firstrec);
2587                 fprintf(stderr, "\n");
2588                 previno = 0x7fffffff;
2589                 return;
2590         }
2591         if (previno == 0x7fffffff)
2592                 goto newcalc;
2593         switch (prevtype) {
2594         case TS_BITS:
2595                 fprintf(stderr, "Dumped inodes map header");
2596                 break;
2597         case TS_CLRI:
2598                 fprintf(stderr, "Used inodes map header");
2599                 break;
2600         case TS_INODE:
2601                 fprintf(stderr, "File header, ino %lu", (unsigned long)previno);
2602                 break;
2603         case TS_ADDR:
2604                 fprintf(stderr, "File continuation header, ino %ld", (long)previno);
2605                 break;
2606         case TS_END:
2607                 fprintf(stderr, "End of tape header");
2608                 break;
2609         }
2610         if (predict != blksread - 1)
2611                 fprintf(stderr, "; predicted %ld blocks, got %ld blocks",
2612                         predict, blksread - 1);
2613         fprintf(stderr, "\n");
2614 newcalc:
2615         blks = 0;
2616         if (header->c_type != TS_END)
2617                 for (i = 0; i < header->c_count; i++)
2618                         if (readmapflag || header->c_addr[i] != 0)
2619                                 blks++;
2620         predict = blks;
2621         blksread = 0;
2622         prevtype = header->c_type;
2623         previno = header->c_inumber;
2624 }
2625
2626 /*
2627  * Find an inode header.
2628  * Complain if had to skip, and complain is set.
2629  */
2630 static void
2631 findinode(struct s_spcl *header)
2632 {
2633         static long skipcnt = 0;
2634         long i;
2635         char buf[TP_BSIZE];
2636
2637         curfile.name = "<name unknown>";
2638         curfile.action = UNKNOWN;
2639         curfile.dip = NULL;
2640         curfile.ino = 0;
2641         do {
2642                 if (header->c_magic != NFS_MAGIC) {
2643                         skipcnt++;
2644                         while (gethead(header) == FAIL ||
2645                             header->c_date != dumpdate)
2646                                 skipcnt++;
2647                 }
2648                 switch (header->c_type) {
2649
2650                 case TS_ADDR:
2651                         /*
2652                          * Skip up to the beginning of the next record
2653                          */
2654                         for (i = 0; i < header->c_count; i++)
2655                                 if (header->c_addr[i])
2656                                         readtape(buf);
2657                         while (gethead(header) == FAIL ||
2658                             header->c_date != dumpdate)
2659                                 skipcnt++;
2660                         break;
2661
2662                 case TS_INODE:
2663                         curfile.dip = &header->c_dinode;
2664                         curfile.ino = header->c_inumber;
2665                         break;
2666
2667                 case TS_END:
2668                         curfile.ino = maxino;
2669                         break;
2670
2671                 case TS_CLRI:
2672                         curfile.name = "<file removal list>";
2673                         break;
2674
2675                 case TS_BITS:
2676                         curfile.name = "<file dump list>";
2677                         break;
2678
2679                 case TS_TAPE:
2680                         panic("unexpected tape header\n");
2681                         /* NOTREACHED */
2682
2683                 default:
2684                         panic("unknown tape header type %d\n", spcl.c_type);
2685                         /* NOTREACHED */
2686
2687                 }
2688         } while (header->c_type == TS_ADDR);
2689         if (skipcnt > 0)
2690 #ifdef USE_QFA
2691                 if (!noresyncmesg)
2692 #endif
2693                         fprintf(stderr, "resync restore, skipped %ld blocks\n",
2694                                 skipcnt);
2695         skipcnt = 0;
2696 }
2697
2698 static int
2699 checksum(int *buf)
2700 {
2701         int i, j;
2702
2703         j = sizeof(union u_spcl) / sizeof(int);
2704         i = 0;
2705         if(!Bcvt) {
2706                 do
2707                         i += *buf++;
2708                 while (--j);
2709         } else {
2710                 /* What happens if we want to read restore tapes
2711                         for a 16bit int machine??? */
2712                 do
2713                         i += swabi(*buf++);
2714                 while (--j);
2715         }
2716
2717         if (i != CHECKSUM) {
2718                 fprintf(stderr, "Checksum error %o, inode %lu file %s\n", i,
2719                         (unsigned long)curfile.ino, curfile.name);
2720                 return(FAIL);
2721         }
2722         return(GOOD);
2723 }
2724
2725 #ifdef RRESTORE
2726 #ifdef __STDC__
2727 #include <stdarg.h>
2728 #else
2729 #include <varargs.h>
2730 #endif
2731
2732 void
2733 #ifdef __STDC__
2734 msg(const char *fmt, ...)
2735 #else
2736 msg(fmt, va_alist)
2737         char *fmt;
2738         va_dcl
2739 #endif
2740 {
2741         va_list ap;
2742 #ifdef __STDC__
2743         va_start(ap, fmt);
2744 #else
2745         va_start(ap);
2746 #endif
2747         (void)vfprintf(stderr, fmt, ap);
2748         va_end(ap);
2749 }
2750 #endif /* RRESTORE */
2751
2752 static u_char *
2753 swab16(u_char *sp, int n)
2754 {
2755         char c;
2756
2757         while (--n >= 0) {
2758                 c = sp[0]; sp[0] = sp[1]; sp[1] = c;
2759                 sp += 2;
2760         }
2761         return (sp);
2762 }
2763
2764 static u_char *
2765 swab32(u_char *sp, int n)
2766 {
2767         char c;
2768
2769         while (--n >= 0) {
2770                 c = sp[0]; sp[0] = sp[3]; sp[3] = c;
2771                 c = sp[1]; sp[1] = sp[2]; sp[2] = c;
2772                 sp += 4;
2773         }
2774         return (sp);
2775 }
2776
2777 static u_char *
2778 swab64(u_char *sp, int n)
2779 {
2780         char c;
2781
2782         while (--n >= 0) {
2783                 c = sp[0]; sp[0] = sp[7]; sp[7] = c;
2784                 c = sp[1]; sp[1] = sp[6]; sp[6] = c;
2785                 c = sp[2]; sp[2] = sp[5]; sp[5] = c;
2786                 c = sp[3]; sp[3] = sp[4]; sp[4] = c;
2787                 sp += 8;
2788         }
2789         return (sp);
2790 }
2791
2792 void
2793 swabst(u_char *cp, u_char *sp)
2794 {
2795         int n = 0;
2796
2797         while (*cp) {
2798                 switch (*cp) {
2799                 case '0': case '1': case '2': case '3': case '4':
2800                 case '5': case '6': case '7': case '8': case '9':
2801                         n = (n * 10) + (*cp++ - '0');
2802                         continue;
2803
2804                 case 's': case 'w': case 'h':
2805                         if (n == 0)
2806                                 n = 1;
2807                         sp = swab16(sp, n);
2808                         break;
2809
2810                 case 'i':
2811                         if (n == 0)
2812                                 n = 1;
2813                         sp = swab32(sp, n);
2814                         break;
2815
2816                 case 'l':
2817                         if (n == 0)
2818                                 n = 1;
2819                         sp = swab64(sp, n);
2820                         break;
2821
2822                 default: /* Any other character, like 'b' counts as byte. */
2823                         if (n == 0)
2824                                 n = 1;
2825                         sp += n;
2826                         break;
2827                 }
2828                 cp++;
2829                 n = 0;
2830         }
2831 }
2832
2833 static u_int
2834 swabi(u_int x)
2835 {
2836         swabst((u_char *)"i", (u_char *)&x);
2837         return (x);
2838 }
2839
2840 #if 0
2841 static u_long
2842 swabl(u_long x)
2843 {
2844         swabst((u_char *)"l", (u_char *)&x);
2845         return (x);
2846 }
2847 #endif
2848
2849 void
2850 RequestVol(long tnum)
2851 {
2852         FLUSHTAPEBUF();
2853         getvol(tnum);
2854 }
2855
2856 #ifdef USE_QFA
2857 #ifdef sunos
2858 extern int fdsmtc;
2859
2860 struct uscsi_cmd {
2861         int     uscsi_flags;            /* read, write, etc. see below */
2862         short   uscsi_status;           /* resulting status  */
2863         short   uscsi_timeout;          /* Command Timeout */
2864         caddr_t uscsi_cdb;              /* cdb to send to target */
2865         caddr_t uscsi_bufaddr;          /* i/o source/destination */
2866         u_int   uscsi_buflen;           /* size of i/o to take place */
2867         u_int   uscsi_resid;            /* resid from i/o operation */
2868         u_char  uscsi_cdblen;           /* # of valid cdb bytes */
2869         u_char  uscsi_rqlen;            /* size of uscsi_rqbuf */
2870         u_char  uscsi_rqstatus;         /* status of request sense cmd */
2871         u_char  uscsi_rqresid;          /* resid of request sense cmd */
2872         caddr_t uscsi_rqbuf;            /* request sense buffer */
2873         void   *uscsi_reserved_5;       /* Reserved for Future Use */
2874 };
2875
2876 #define CDB_GROUP0      6       /*  6-byte cdb's */
2877 #define CDB_GROUP1      10      /* 10-byte cdb's */
2878 #define CDB_GROUP2      10      /* 10-byte cdb's */
2879 #define CDB_GROUP3      0       /* reserved */
2880 #define CDB_GROUP4      16      /* 16-byte cdb's */
2881 #define CDB_GROUP5      12      /* 12-byte cdb's */
2882 #define CDB_GROUP6      0       /* reserved */
2883 #define CDB_GROUP7      0       /* reserved */
2884
2885 #define USCSI_WRITE     0x00000 /* send data to device */
2886 #define USCSI_SILENT    0x00001 /* no error messages */
2887 #define USCSI_DIAGNOSE  0x00002 /* fail if any error occurs */
2888 #define USCSI_ISOLATE   0x00004 /* isolate from normal commands */
2889 #define USCSI_READ      0x00008 /* get data from device */
2890 #define USCSI_RESET     0x04000 /* Reset target */
2891 #define USCSI_RESET_ALL 0x08000 /* Reset all targets */
2892 #define USCSI_RQENABLE  0x10000 /* Enable Request Sense extensions */
2893
2894 #define USCSIIOC        (0x04 << 8)
2895 #define USCSICMD        (USCSIIOC|201)  /* user scsi command */
2896
2897 #define USCSI_TIMEOUT   30
2898 #define USCSI_SHORT_TIMEOUT     900
2899 #define USCSI_LONG_TIMEOUT      14000
2900
2901 #define B(s,i) ((unsigned char)((s) >> i))
2902 #define B1(s)                           ((unsigned char)(s))
2903
2904 #define MSB4(s,v) *(s)=B(v,24),(s)[1]=B(v,16), (s)[2]=B(v,8), (s)[3]=B1(v)
2905
2906
2907 int
2908 GetTapePos(long long *pos)
2909 {
2910         int                                     err = 0;
2911         struct uscsi_cmd        scmd;
2912         char                            buf[512];
2913         char                            parm[512 * 8];
2914         long                            lpos;
2915
2916         (void)memset((void *)buf, 0, sizeof(buf));
2917         (void)memset((void *)&scmd, 0, sizeof(scmd));
2918         scmd.uscsi_flags = USCSI_READ|USCSI_SILENT;
2919         scmd.uscsi_timeout = USCSI_TIMEOUT;
2920         scmd.uscsi_cdb = buf;
2921         scmd.uscsi_cdblen = CDB_GROUP1;
2922         buf[0] = 0x34;  /* read position */
2923         buf[1] = 0;
2924         (void)memset((void *)parm, 0, 512);
2925         scmd.uscsi_bufaddr = parm;
2926         scmd.uscsi_buflen = 56;
2927         if (ioctl(fdsmtc, USCSICMD, &scmd) == -1) {
2928                 err = errno;
2929                 return err;
2930         }
2931         (void)memcpy(&lpos, &parm[4], sizeof(long));
2932         *pos = lpos;
2933         return err;
2934 }
2935
2936 int
2937 GotoTapePos(long long pos)
2938 {
2939         int                                     err = 0;
2940         struct uscsi_cmd        scmd;
2941         char                            buf[512];
2942         char                            parm[512 * 8];
2943         long                            lpos = (long)pos;
2944
2945         (void)memset((void *)buf, 0, sizeof(buf));
2946         (void)memset((void *)&scmd, 0, sizeof(scmd));
2947         scmd.uscsi_flags = USCSI_WRITE|USCSI_SILENT;
2948         scmd.uscsi_timeout = 360;       /* 5 Minutes */
2949         scmd.uscsi_cdb = buf;
2950         scmd.uscsi_cdblen = CDB_GROUP1;
2951         buf[0] = 0x2b;  /* locate */
2952         buf[1] = 0;
2953         MSB4(&buf[3], lpos);
2954         (void)memset((void *)parm, 0, 512);
2955         scmd.uscsi_bufaddr = NULL;
2956         scmd.uscsi_buflen = 0;
2957         if (ioctl(fdsmtc, USCSICMD, &scmd) == -1) {
2958                 err = errno;
2959                 return err;
2960         }
2961         return err;
2962 }
2963 #endif
2964
2965 #define LSEEK_GET_TAPEPOS       10
2966 #define LSEEK_GO2_TAPEPOS       11
2967
2968 #ifdef  __linux__
2969 typedef struct mt_pos {
2970         short    mt_op;
2971         int      mt_count;
2972 } MTPosRec, *MTPosPtr;
2973
2974
2975 /*
2976  * get the current position of the tape
2977  */
2978 int
2979 GetTapePos(long long *pos)
2980 {
2981         int err = 0;
2982
2983 #ifdef RDUMP
2984         if (host) {
2985                 *pos = (long long) rmtseek((OFF_T)0, (int)LSEEK_GET_TAPEPOS);
2986                 err = *pos < 0;
2987         }
2988         else
2989 #endif
2990         {
2991         if (magtapein) {
2992                 long mtpos;
2993                 *pos = 0;
2994                 err = (ioctl(mt, MTIOCPOS, &mtpos) < 0);
2995                 *pos = (long long)mtpos;
2996         }
2997         else {
2998                 *pos = LSEEK(mt, 0, SEEK_CUR);
2999                 err = (*pos < 0);
3000         }
3001         }
3002         if (err) {
3003                 err = errno;
3004                 fprintf(stdout, "[%ld] error: %d (getting tapepos: %lld)\n", 
3005                         (unsigned long)getpid(), err, *pos);
3006                 return err;
3007         }
3008         return err;
3009 }
3010
3011 /*
3012  * go to specified position on tape
3013  */
3014 int
3015 GotoTapePos(long long pos)
3016 {
3017         int err = 0;
3018
3019 #ifdef RDUMP
3020         if (host)
3021                 err = (rmtseek((OFF_T)pos, (int)LSEEK_GO2_TAPEPOS) < 0);
3022         else
3023 #endif
3024         {
3025         if (magtapein) {
3026                 struct mt_pos buf;
3027                 buf.mt_op = MTSEEK;
3028                 buf.mt_count = (int) pos;
3029                 err = (ioctl(mt, MTIOCTOP, &buf) < 0);
3030         }
3031         else {
3032                 pos = LSEEK(mt, pos, SEEK_SET);
3033                 err = (pos < 0);
3034         }
3035         }
3036         if (err) {
3037                 err = errno;
3038                 fprintf(stdout, "[%ld] error: %d (setting tapepos: %lld)\n", 
3039                         (unsigned long)getpid(), err, pos);
3040                 return err;
3041         }
3042         return err;
3043 }
3044 #endif /* __linux__ */
3045
3046 /*
3047  * read next data from tape to re-sync
3048  */
3049 void
3050 ReReadFromTape(void)
3051 {
3052         FLUSHTAPEBUF();
3053         noresyncmesg = 1;
3054         if (gethead(&spcl) == FAIL) {
3055 #ifdef DEBUG_QFA
3056                 fprintf(stdout, "DEBUG 1 gethead failed\n");
3057 #endif
3058         }
3059         findinode(&spcl);
3060         noresyncmesg = 0;
3061 }
3062
3063 void
3064 ReReadInodeFromTape(dump_ino_t theino)
3065 {
3066         long    cntloop = 0;
3067
3068         FLUSHTAPEBUF();
3069         noresyncmesg = 1;
3070         do {
3071                 cntloop++;
3072                 gethead(&spcl);
3073         } while (!(spcl.c_inumber == theino && spcl.c_type == TS_INODE && spcl.c_date == dumpdate));
3074 #ifdef DEBUG_QFA
3075         fprintf(stderr, "DEBUG: %ld reads\n", cntloop);
3076         fprintf(stderr, "DEBUG: bufsize %ld\n", bufsize);
3077         fprintf(stderr, "DEBUG: ntrec %ld\n", ntrec);
3078         fprintf(stderr, "DEBUG: %ld reads\n", cntloop);
3079 #endif
3080         findinode(&spcl);
3081         noresyncmesg = 0;
3082 }
3083
3084 #ifdef sunos
3085 int
3086 OpenSMTCmt(char *themagtape)
3087 {
3088         if (GetSCSIIDFromPath(themagtape, &scsiid)) {
3089                 fprintf(stderr, "can't get SCSI-ID for %s\n", themagtape);
3090                 return -1;
3091         }
3092         if (scsiid < 0) {
3093                 fprintf(stderr, "can't get SCSI-ID for %s\n", themagtape);
3094                 return -1;
3095         }
3096         sprintf(smtcpath, "/dev/rsmtc%ld,0", scsiid);
3097         if ((fdsmtc = open(smtcpath, O_RDWR)) == -1) {
3098                 fprintf(stderr, "can't open smtc device: %s, %d\n", smtcpath, errno);
3099                 return -1;
3100         }
3101         return 0;
3102 }
3103 #endif /* sunos */
3104 #endif /* USE_QFA */