/*      $NetBSD: tcopy.c,v 1.17 2011/09/06 18:32:26 joerg Exp $ */

/*
* Copyright (c) 1985, 1987, 1993, 1995
*      The Regents of the University of California.  All rights reserved.
*
* Redistribution and use in source and binary forms, with or without
* modification, are permitted provided that the following conditions
* are met:
* 1. Redistributions of source code must retain the above copyright
*    notice, this list of conditions and the following disclaimer.
* 2. Redistributions in binary form must reproduce the above copyright
*    notice, this list of conditions and the following disclaimer in the
*    documentation and/or other materials provided with the distribution.
* 3. Neither the name of the University nor the names of its contributors
*    may be used to endorse or promote products derived from this software
*    without specific prior written permission.
*
* THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
* ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
* IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
* ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
* FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
* DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
* OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
* HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
* LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
* OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
* SUCH DAMAGE.
*/

#include <sys/cdefs.h>
#ifndef lint
__COPYRIGHT("@(#) Copyright (c) 1985, 1987, 1993\
The Regents of the University of California.  All rights reserved.");
#endif /* not lint */

#ifndef lint
#if 0
static char sccsid[] = "@(#)tcopy.c     8.3 (Berkeley) 1/23/95";
#endif
__RCSID("$NetBSD: tcopy.c,v 1.17 2011/09/06 18:32:26 joerg Exp $");
#endif /* not lint */

#include <sys/types.h>
#include <sys/stat.h>
#include <sys/ioctl.h>
#include <sys/mtio.h>

#include <err.h>
#include <errno.h>
#include <paths.h>
#include <fcntl.h>
#include <signal.h>
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <unistd.h>
#include <util.h>

#define MAXREC  (64 * 1024)
#define NOCOUNT (-2)

static int      filen, guesslen, maxblk = MAXREC;
static long     lastrec, record;
static off_t    size, tsize;
static FILE     *msg = stdout;

static void     *getspace(int);
__dead static void       intr(int);
__dead static void       usage(void);
static void      verify(int, int, char *);
static void      writeop(int, int);

int
main(int argc, char *argv[])
{
       int ch, needeof, nw, inp, outp;
       ssize_t lastnread, nread;
       enum {READ, VERIFY, COPY, COPYVERIFY} op = READ;
       sig_t oldsig;
       char *buff;
       const char *inf;

       outp = 0;
       inf = NULL;
       guesslen = 1;
       while ((ch = getopt(argc, argv, "cs:vx")) != -1)
               switch((char)ch) {
               case 'c':
                       op = COPYVERIFY;
                       break;
               case 's':
                       maxblk = atoi(optarg);
                       if (maxblk <= 0) {
                               warnx("illegal block size");
                               usage();
                       }
                       guesslen = 0;
                       break;
               case 'v':
                       op = VERIFY;
                       break;
               case 'x':
                       msg = stderr;
                       break;
               case '?':
               default:
                       usage();
               }
       argc -= optind;
       argv += optind;

       switch(argc) {
       case 0:
               if (op != READ)
                       usage();
               inf = _PATH_DEFTAPE;
               break;
       case 1:
               if (op != READ)
                       usage();
               inf = argv[0];
               break;
       case 2:
               if (op == READ)
                       op = COPY;
               inf = argv[0];
               if ((outp = open(argv[1], op == VERIFY ? O_RDONLY :
                   op == COPY ? O_WRONLY : O_RDWR, DEFFILEMODE)) < 0) {
                       err(3, "%s", argv[1]);
               }
               break;
       default:
               usage();
       }

       if ((inp = open(inf, O_RDONLY, 0)) < 0)
               err(1, "%s", inf);

       buff = getspace(maxblk);

       if (op == VERIFY) {
               verify(inp, outp, buff);
               exit(0);
       }

       if ((oldsig = signal(SIGINT, SIG_IGN)) != SIG_IGN)
               (void) signal(SIGINT, intr);

       needeof = 0;
       for (lastnread = NOCOUNT;;) {
               if ((nread = read(inp, buff, maxblk)) == -1) {
                       while (errno == EINVAL && (maxblk -= 1024)) {
                               nread = read(inp, buff, maxblk);
                               if (nread >= 0)
                                       goto r1;
                       }
                       err(1, "read error, file %d, record %ld",
                           filen, record);
               } else if (nread != lastnread) {
                       if (lastnread != 0 && lastnread != NOCOUNT) {
                               if (lastrec == 0 && nread == 0)
                                       fprintf(msg, "%ld records\n", record);
                               else if (record - lastrec > 1)
                                       fprintf(msg, "records %ld to %ld\n",
                                           lastrec, record);
                               else
                                       fprintf(msg, "record %ld\n", lastrec);
                       }
                       if (nread != 0)
                               fprintf(msg, "file %d: block size %ld: ",
                                   filen, (long)nread);
                       (void) fflush(stdout);
                       lastrec = record;
               }
r1:             guesslen = 0;
               if (nread > 0) {
                       if (op == COPY || op == COPYVERIFY) {
                               if (needeof) {
                                       writeop(outp, MTWEOF);
                                       needeof = 0;
                               }
                               nw = write(outp, buff, nread);
                               if (nw != nread) {
                                   int error = errno;
                                   fprintf(stderr,
                                       "write error, file %d, record %ld: ",
                                       filen, record);
                                   if (nw == -1)
                                       fprintf(stderr,
                                               ": %s", strerror(error));
                                   else
                                       fprintf(stderr,
                                           "write (%d) != read (%ld)\n",
                                           nw, (long)nread);
                                   fprintf(stderr, "copy aborted\n");
                                   exit(5);
                               }
                       }
                       size += nread;
                       record++;
               } else {
                       if (lastnread <= 0 && lastnread != NOCOUNT) {
                               fprintf(msg, "eot\n");
                               break;
                       }
                       fprintf(msg,
                           "file %d: eof after %ld records: %lld bytes\n",
                           filen, record, (long long)size);
                       needeof = 1;
                       filen++;
                       tsize += size;
                       size = record = lastrec = 0;
                       lastnread = 0;
               }
               lastnread = nread;
       }
       fprintf(msg, "total length: %lld bytes\n", (long long)tsize);
       (void)signal(SIGINT, oldsig);
       if (op == COPY || op == COPYVERIFY) {
               writeop(outp, MTWEOF);
               writeop(outp, MTWEOF);
               if (op == COPYVERIFY) {
                       writeop(outp, MTREW);
                       writeop(inp, MTREW);
                       verify(inp, outp, buff);
               }
       }
       exit(0);
}

static void
verify(int inp, int outp, char *outb)
{
       int eot, inmaxblk, inn, outmaxblk, outn;
       char *inb;

       inb = getspace(maxblk);
       inmaxblk = outmaxblk = maxblk;
       for (eot = 0;; guesslen = 0) {
               if ((inn = read(inp, inb, inmaxblk)) == -1) {
                       if (guesslen)
                               while (errno == EINVAL && (inmaxblk -= 1024)) {
                                       inn = read(inp, inb, inmaxblk);
                                       if (inn >= 0)
                                               goto r1;
                               }
                       warn("read error");
                       break;
               }
r1:             if ((outn = read(outp, outb, outmaxblk)) == -1) {
                       if (guesslen)
                               while (errno == EINVAL && (outmaxblk -= 1024)) {
                                       outn = read(outp, outb, outmaxblk);
                                       if (outn >= 0)
                                               goto r2;
                               }
                       warn("read error");
                       break;
               }
r2:             if (inn != outn) {
                       fprintf(msg,
                           "%s: tapes have different block sizes; %d != %d.\n",
                           "tcopy", inn, outn);
                       break;
               }
               if (!inn) {
                       if (eot++) {
                               fprintf(msg, "%s: tapes are identical.\n",
                                       "tcopy");
                               free(inb);
                               return;
                       }
               } else {
                       if (memcmp(inb, outb, inn)) {
                               fprintf(msg,
                                   "%s: tapes have different data.\n",
                                       "tcopy");
                               break;
                       }
                       eot = 0;
               }
       }
       free(inb);
       exit(1);
}

static void
intr(int signo)
{
       if (record) {
               if (record - lastrec > 1)
                       fprintf(msg, "records %ld to %ld\n", lastrec, record);
               else
                       fprintf(msg, "record %ld\n", lastrec);
       }
       fprintf(msg, "interrupt at file %d: record %ld\n", filen, record);
       fprintf(msg, "total length: %lld bytes\n", (long long)(tsize + size));
       (void)raise_default_signal(signo);
       exit(1);
}

static void *
getspace(int blk)
{
       void *bp;

       if ((bp = malloc((size_t)blk)) == NULL)
               errx(11, "no memory");

       return (bp);
}

static void
writeop(int fd, int type)
{
       struct mtop op;

       op.mt_op = type;
       op.mt_count = (daddr_t)1;
       if (ioctl(fd, MTIOCTOP, (char *)&op) < 0)
               err(6, "tape op");
}

static void
usage(void)
{

       fprintf(stderr, "usage: tcopy [-cvx] [-s maxblk] src [dest]\n");
       exit(1);
}