tcopy.c revision 73986 
1/*
2 * Copyright (c) 1985, 1987, 1993
3 *	The Regents of the University of California.  All rights reserved.
4 *
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
7 * are met:
8 * 1. Redistributions of source code must retain the above copyright
9 *    notice, this list of conditions and the following disclaimer.
10 * 2. Redistributions in binary form must reproduce the above copyright
11 *    notice, this list of conditions and the following disclaimer in the
12 *    documentation and/or other materials provided with the distribution.
13 * 3. All advertising materials mentioning features or use of this software
14 *    must display the following acknowledgement:
15 *	This product includes software developed by the University of
16 *	California, Berkeley and its contributors.
17 * 4. Neither the name of the University nor the names of its contributors
18 *    may be used to endorse or promote products derived from this software
19 *    without specific prior written permission.
20 *
21 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
22 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
23 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
24 * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
25 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
26 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
27 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
28 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
29 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
30 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
31 * SUCH DAMAGE.
32 */
33
34#ifndef lint
35static const char copyright[] =
36"@(#) Copyright (c) 1985, 1987, 1993\n\
37	The Regents of the University of California.  All rights reserved.\n";
38#endif /* not lint */
39
40#ifndef lint
41#if 0
42static char sccsid[] = "@(#)tcopy.c	8.2 (Berkeley) 4/17/94";
43#endif
44static const char rcsid[] =
45  "$FreeBSD: head/usr.bin/tcopy/tcopy.c 73986 2001-03-08 09:04:40Z obrien $";
46#endif /* not lint */
47
48#include <sys/types.h>
49#include <sys/stat.h>
50#include <sys/ioctl.h>
51#include <sys/mtio.h>
52
53#include <err.h>
54#include <errno.h>
55#include <fcntl.h>
56#include <paths.h>
57#include <signal.h>
58#include <stdio.h>
59#include <stdlib.h>
60#include <string.h>
61#include <unistd.h>
62
63#define	MAXREC	(64 * 1024)
64#define	NOCOUNT	(-2)
65
66int	filen, guesslen, maxblk = MAXREC;
67u_int64_t	lastrec, record, size, tsize;
68FILE	*msg = stdout;
69
70void	*getspace __P((int));
71void	 intr __P((int));
72static void	 usage __P((void));
73void	 verify __P((int, int, char *));
74void	 writeop __P((int, int));
75void	rewind_tape(int);
76
77int
78main(argc, argv)
79	int argc;
80	char *argv[];
81{
82	register int lastnread, nread, nw, inp, outp;
83	enum {READ, VERIFY, COPY, COPYVERIFY} op = READ;
84	sig_t oldsig;
85	int ch, needeof;
86	char *buff, *inf;
87
88	guesslen = 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 %qu", filen, record);
160		} else if (nread != lastnread) {
161			if (lastnread != 0 && lastnread != NOCOUNT) {
162				if (lastrec == 0 && nread == 0)
163					fprintf(msg, "%qu records\n", record);
164				else if (record - lastrec > 1)
165					fprintf(msg, "records %qu to %qu\n",
166					    lastrec, record);
167				else
168					fprintf(msg, "record %qu\n", 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 %qu", filen, record);
187					} else {
188					warnx("write error, file %d, record %qu", filen, record);
189					warnx("write (%d) != read (%d)", nw, nread);
190					}
191					errx(5, "copy aborted");
192				}
193			}
194			size += nread;
195			record++;
196		} else {
197			if (lastnread <= 0 && lastnread != NOCOUNT) {
198				fprintf(msg, "eot\n");
199				break;
200			}
201			fprintf(msg,
202			    "file %d: eof after %qu records: %qu bytes\n",
203			    filen, record, size);
204			needeof = 1;
205			filen++;
206			tsize += size;
207			size = record = lastrec = 0;
208			lastnread = 0;
209		}
210		lastnread = nread;
211	}
212	fprintf(msg, "total length: %qu bytes\n", tsize);
213	(void)signal(SIGINT, oldsig);
214	if (op == COPY || op == COPYVERIFY) {
215		writeop(outp, MTWEOF);
216		writeop(outp, MTWEOF);
217		if (op == COPYVERIFY) {
218			rewind_tape(outp);
219			rewind_tape(inp);
220			verify(inp, outp, buff);
221		}
222	}
223	exit(0);
224}
225
226void
227verify(inp, outp, outb)
228	register int inp, outp;
229	register char *outb;
230{
231	register int eot, inmaxblk, inn, outmaxblk, outn;
232	register 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
280void
281intr(signo)
282	int signo;
283{
284	if (record) {
285		if (record - lastrec > 1)
286			fprintf(msg, "records %qu to %qu\n", lastrec, record);
287		else
288			fprintf(msg, "record %qu\n", lastrec);
289	}
290	fprintf(msg, "interrupt at file %d: record %qu\n", filen, record);
291	fprintf(msg, "total length: %ld bytes\n", tsize + size);
292	exit(1);
293}
294
295void *
296getspace(blk)
297	int blk;
298{
299	void *bp;
300
301	if ((bp = malloc((size_t)blk)) == NULL)
302		errx(11, "no memory");
303	return (bp);
304}
305
306void
307writeop(fd, type)
308	int fd, type;
309{
310	struct mtop op;
311
312	op.mt_op = type;
313	op.mt_count = (daddr_t)1;
314	if (ioctl(fd, MTIOCTOP, (char *)&op) < 0)
315		err(6, "tape op");
316}
317
318static void
319usage()
320{
321	fprintf(stderr, "usage: tcopy [-cvx] [-s maxblk] [src [dest]]\n");
322	exit(1);
323}
324
325void
326rewind_tape(int fd)
327{
328	struct stat sp;
329
330	if(fstat(fd, &sp))
331		errx(12, "fstat in rewind");
332
333	/*
334	 * don't want to do tape ioctl on regular files:
335	 */
336	if( S_ISREG(sp.st_mode) ) {
337		if( lseek(fd, 0, SEEK_SET) == -1 )
338			errx(13, "lseek");
339	} else
340		/*  assume its a tape	*/
341		writeop(fd, MTREW);
342}
343