tcopy.c revision 330897 
1/*-
2 * SPDX-License-Identifier: BSD-3-Clause
3 *
4 * Copyright (c) 1985, 1987, 1993
5 *	The Regents of the University of California.  All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 *    notice, this list of conditions and the following disclaimer.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 *    notice, this list of conditions and the following disclaimer in the
14 *    documentation and/or other materials provided with the distribution.
15 * 4. Neither the name of the University nor the names of its contributors
16 *    may be used to endorse or promote products derived from this software
17 *    without specific prior written permission.
18 *
19 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
20 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
21 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
22 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
23 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
24 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
25 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
26 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
27 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
28 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
29 * SUCH DAMAGE.
30 */
31
32#include <sys/cdefs.h>
33
34__FBSDID("$FreeBSD: stable/11/usr.bin/tcopy/tcopy.c 330897 2018-03-14 03:19:51Z eadler $");
35
36#ifndef lint
37static const char copyright[] =
38"@(#) Copyright (c) 1985, 1987, 1993\n\
39	The Regents of the University of California.  All rights reserved.\n";
40#endif
41
42#ifndef lint
43static const char sccsid[] = "@(#)tcopy.c	8.2 (Berkeley) 4/17/94";
44#endif
45
46#include <sys/types.h>
47#include <sys/stat.h>
48#include <sys/ioctl.h>
49#include <sys/mtio.h>
50
51#include <err.h>
52#include <errno.h>
53#include <fcntl.h>
54#include <paths.h>
55#include <signal.h>
56#include <stdint.h>
57#include <stdio.h>
58#include <stdlib.h>
59#include <string.h>
60#include <unistd.h>
61
62#define	MAXREC	(64 * 1024)
63#define	NOCOUNT	(-2)
64
65static int	filen, guesslen, maxblk = MAXREC;
66static uint64_t	lastrec, record, size, tsize;
67static FILE	*msg;
68
69static void	*getspace(int);
70static void	 intr(int);
71static void	 usage(void) __dead2;
72static void	 verify(int, int, char *);
73static void	 writeop(int, int);
74static void	 rewind_tape(int);
75
76int
77main(int argc, char *argv[])
78{
79	int lastnread, nread, nw, inp, outp;
80	enum {READ, VERIFY, COPY, COPYVERIFY} op = READ;
81	sig_t oldsig;
82	int ch, needeof;
83	char *buff;
84	const char *inf;
85
86	msg = stdout;
87	guesslen = 1;
88	outp = -1;
89	while ((ch = getopt(argc, argv, "cs:vx")) != -1)
90		switch((char)ch) {
91		case 'c':
92			op = COPYVERIFY;
93			break;
94		case 's':
95			maxblk = atoi(optarg);
96			if (maxblk <= 0) {
97				warnx("illegal block size");
98				usage();
99			}
100			guesslen = 0;
101			break;
102		case 'v':
103			op = VERIFY;
104			break;
105		case 'x':
106			msg = stderr;
107			break;
108		case '?':
109		default:
110			usage();
111		}
112	argc -= optind;
113	argv += optind;
114
115	switch(argc) {
116	case 0:
117		if (op != READ)
118			usage();
119		inf = _PATH_DEFTAPE;
120		break;
121	case 1:
122		if (op != READ)
123			usage();
124		inf = argv[0];
125		break;
126	case 2:
127		if (op == READ)
128			op = COPY;
129		inf = argv[0];
130		if ((outp = open(argv[1], op == VERIFY ? O_RDONLY :
131		    op == COPY ? O_WRONLY : O_RDWR, DEFFILEMODE)) < 0)
132			err(3, "%s", argv[1]);
133		break;
134	default:
135		usage();
136	}
137
138	if ((inp = open(inf, O_RDONLY, 0)) < 0)
139		err(1, "%s", inf);
140
141	buff = getspace(maxblk);
142
143	if (op == VERIFY) {
144		verify(inp, outp, buff);
145		exit(0);
146	}
147
148	if ((oldsig = signal(SIGINT, SIG_IGN)) != SIG_IGN)
149		(void) signal(SIGINT, intr);
150
151	needeof = 0;
152	for (lastnread = NOCOUNT;;) {
153		if ((nread = read(inp, buff, maxblk)) == -1) {
154			while (errno == EINVAL && (maxblk -= 1024)) {
155				nread = read(inp, buff, maxblk);
156				if (nread >= 0)
157					goto r1;
158			}
159			err(1, "read error, file %d, record %ju", filen, (intmax_t)record);
160		} else if (nread != lastnread) {
161			if (lastnread != 0 && lastnread != NOCOUNT) {
162				if (lastrec == 0 && nread == 0)
163					fprintf(msg, "%ju records\n", (intmax_t)record);
164				else if (record - lastrec > 1)
165					fprintf(msg, "records %ju to %ju\n",
166					    (intmax_t)lastrec, (intmax_t)record);
167				else
168					fprintf(msg, "record %ju\n", (intmax_t)lastrec);
169			}
170			if (nread != 0)
171				fprintf(msg, "file %d: block size %d: ",
172				    filen, nread);
173			(void) fflush(stdout);
174			lastrec = record;
175		}
176r1:		guesslen = 0;
177		if (nread > 0) {
178			if (op == COPY || op == COPYVERIFY) {
179				if (needeof) {
180					writeop(outp, MTWEOF);
181					needeof = 0;
182				}
183				nw = write(outp, buff, nread);
184				if (nw != nread) {
185					if (nw == -1) {
186						warn("write error, file %d, record %ju", filen,
187						    (intmax_t)record);
188					} else {
189						warnx("write error, file %d, record %ju", filen,
190						    (intmax_t)record);
191						warnx("write (%d) != read (%d)", nw, nread);
192					}
193					errx(5, "copy aborted");
194				}
195			}
196			size += nread;
197			record++;
198		} else {
199			if (lastnread <= 0 && lastnread != NOCOUNT) {
200				fprintf(msg, "eot\n");
201				break;
202			}
203			fprintf(msg,
204			    "file %d: eof after %ju records: %ju bytes\n",
205			    filen, (intmax_t)record, (intmax_t)size);
206			needeof = 1;
207			filen++;
208			tsize += size;
209			size = record = lastrec = 0;
210			lastnread = 0;
211		}
212		lastnread = nread;
213	}
214	fprintf(msg, "total length: %ju bytes\n", (intmax_t)tsize);
215	(void)signal(SIGINT, oldsig);
216	if (op == COPY || op == COPYVERIFY) {
217		writeop(outp, MTWEOF);
218		writeop(outp, MTWEOF);
219		if (op == COPYVERIFY) {
220			rewind_tape(outp);
221			rewind_tape(inp);
222			verify(inp, outp, buff);
223		}
224	}
225	exit(0);
226}
227
228static void
229verify(int inp, int outp, char *outb)
230{
231	int eot, inmaxblk, inn, outmaxblk, outn;
232	char *inb;
233
234	inb = getspace(maxblk);
235	inmaxblk = outmaxblk = maxblk;
236	for (eot = 0;; guesslen = 0) {
237		if ((inn = read(inp, inb, inmaxblk)) == -1) {
238			if (guesslen)
239				while (errno == EINVAL && (inmaxblk -= 1024)) {
240					inn = read(inp, inb, inmaxblk);
241					if (inn >= 0)
242						goto r1;
243				}
244			warn("read error");
245			break;
246		}
247r1:		if ((outn = read(outp, outb, outmaxblk)) == -1) {
248			if (guesslen)
249				while (errno == EINVAL && (outmaxblk -= 1024)) {
250					outn = read(outp, outb, outmaxblk);
251					if (outn >= 0)
252						goto r2;
253				}
254			warn("read error");
255			break;
256		}
257r2:		if (inn != outn) {
258			fprintf(msg,
259			    "%s: tapes have different block sizes; %d != %d.\n",
260			    "tcopy", inn, outn);
261			break;
262		}
263		if (!inn) {
264			if (eot++) {
265				fprintf(msg, "tcopy: tapes are identical.\n");
266				return;
267			}
268		} else {
269			if (bcmp(inb, outb, inn)) {
270				fprintf(msg,
271				    "tcopy: tapes have different data.\n");
272				break;
273			}
274			eot = 0;
275		}
276	}
277	exit(1);
278}
279
280static void
281intr(int signo __unused)
282{
283	if (record) {
284		if (record - lastrec > 1)
285			fprintf(msg, "records %ju to %ju\n", (intmax_t)lastrec, (intmax_t)record);
286		else
287			fprintf(msg, "record %ju\n", (intmax_t)lastrec);
288	}
289	fprintf(msg, "interrupt at file %d: record %ju\n", filen, (intmax_t)record);
290	fprintf(msg, "total length: %ju bytes\n", (uintmax_t)(tsize + size));
291	exit(1);
292}
293
294static void *
295getspace(int blk)
296{
297	void *bp;
298
299	if ((bp = malloc((size_t)blk)) == NULL)
300		errx(11, "no memory");
301	return (bp);
302}
303
304static void
305writeop(int fd, int type)
306{
307	struct mtop op;
308
309	op.mt_op = type;
310	op.mt_count = (daddr_t)1;
311	if (ioctl(fd, MTIOCTOP, (char *)&op) < 0)
312		err(6, "tape op");
313}
314
315static void
316usage(void)
317{
318	fprintf(stderr, "usage: tcopy [-cvx] [-s maxblk] [src [dest]]\n");
319	exit(1);
320}
321
322static void
323rewind_tape(int fd)
324{
325	struct stat sp;
326
327	if(fstat(fd, &sp))
328		errx(12, "fstat in rewind");
329
330	/*
331	 * don't want to do tape ioctl on regular files:
332	 */
333	if( S_ISREG(sp.st_mode) ) {
334		if( lseek(fd, 0, SEEK_SET) == -1 )
335			errx(13, "lseek");
336	} else
337		/*  assume its a tape	*/
338		writeop(fd, MTREW);
339}
340