xref: /netbsd-src/usr.bin/tcopy/tcopy.c (revision ce0bb6e8d2e560ecacbe865a848624f94498063b)
1 /*	$NetBSD: tcopy.c,v 1.3 1994/12/09 02:14:39 jtc Exp $	*/
2 
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  * 3. All advertising materials mentioning features or use of this software
16  *    must display the following acknowledgement:
17  *	This product includes software developed by the University of
18  *	California, Berkeley and its contributors.
19  * 4. Neither the name of the University nor the names of its contributors
20  *    may be used to endorse or promote products derived from this software
21  *    without specific prior written permission.
22  *
23  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
24  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
25  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
26  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
27  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
28  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
29  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
30  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
31  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
32  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
33  * SUCH DAMAGE.
34  */
35 
36 #ifndef lint
37 static char copyright[] =
38 "@(#) Copyright (c) 1985, 1987, 1993\n\
39 	The Regents of the University of California.  All rights reserved.\n";
40 #endif /* not lint */
41 
42 #ifndef lint
43 #if 0
44 static char sccsid[] = "@(#)tcopy.c	8.2 (Berkeley) 4/17/94";
45 #endif
46 static char rcsid[] = "$NetBSD: tcopy.c,v 1.3 1994/12/09 02:14:39 jtc Exp $";
47 #endif /* not lint */
48 
49 #include <sys/types.h>
50 #include <sys/stat.h>
51 #include <sys/ioctl.h>
52 #include <sys/mtio.h>
53 
54 #include <errno.h>
55 #include <fcntl.h>
56 #include <signal.h>
57 #include <stdio.h>
58 #include <stdlib.h>
59 #include <string.h>
60 #include <unistd.h>
61 
62 #include "pathnames.h"
63 
64 #define	MAXREC	(64 * 1024)
65 #define	NOCOUNT	(-2)
66 
67 int	filen, guesslen, maxblk = MAXREC;
68 long	lastrec, record, size, tsize;
69 FILE	*msg = stdout;
70 
71 void	*getspace __P((int));
72 void	 intr __P((int));
73 void	 usage __P((void));
74 void	 verify __P((int, int, char *));
75 void	 writeop __P((int, int));
76 
77 int
78 main(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")) != EOF)
90 		switch((char)ch) {
91 		case 'c':
92 			op = COPYVERIFY;
93 			break;
94 		case 's':
95 			maxblk = atoi(optarg);
96 			if (maxblk <= 0) {
97 				fprintf(stderr, "tcopy: illegal block size\n");
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 			perror(argv[1]);
133 			exit(3);
134 		}
135 		break;
136 	default:
137 		usage();
138 	}
139 
140 	if ((inp = open(inf, O_RDONLY, 0)) < 0) {
141 		perror(inf);
142 		exit(1);
143 	}
144 
145 	buff = getspace(maxblk);
146 
147 	if (op == VERIFY) {
148 		verify(inp, outp, buff);
149 		exit(0);
150 	}
151 
152 	if ((oldsig = signal(SIGINT, SIG_IGN)) != SIG_IGN)
153 		(void) signal(SIGINT, intr);
154 
155 	needeof = 0;
156 	for (lastnread = NOCOUNT;;) {
157 		if ((nread = read(inp, buff, maxblk)) == -1) {
158 			while (errno == EINVAL && (maxblk -= 1024)) {
159 				nread = read(inp, buff, maxblk);
160 				if (nread >= 0)
161 					goto r1;
162 			}
163 			fprintf(stderr, "read error, file %d, record %ld: ",
164 			    filen, record);
165 			perror("");
166 			exit(1);
167 		} else if (nread != lastnread) {
168 			if (lastnread != 0 && lastnread != NOCOUNT) {
169 				if (lastrec == 0 && nread == 0)
170 					fprintf(msg, "%ld records\n", record);
171 				else if (record - lastrec > 1)
172 					fprintf(msg, "records %ld to %ld\n",
173 					    lastrec, record);
174 				else
175 					fprintf(msg, "record %ld\n", lastrec);
176 			}
177 			if (nread != 0)
178 				fprintf(msg, "file %d: block size %d: ",
179 				    filen, nread);
180 			(void) fflush(stdout);
181 			lastrec = record;
182 		}
183 r1:		guesslen = 0;
184 		if (nread > 0) {
185 			if (op == COPY || op == COPYVERIFY) {
186 				if (needeof) {
187 					writeop(outp, MTWEOF);
188 					needeof = 0;
189 				}
190 				nw = write(outp, buff, nread);
191 				if (nw != nread) {
192 				    fprintf(stderr,
193 					"write error, file %d, record %ld: ",
194 					filen, record);
195 				    if (nw == -1)
196 					perror("");
197 				    else
198 					fprintf(stderr,
199 					    "write (%d) != read (%d)\n",
200 					    nw, nread);
201 				    fprintf(stderr, "copy aborted\n");
202 				    exit(5);
203 				}
204 			}
205 			size += nread;
206 			record++;
207 		} else {
208 			if (lastnread <= 0 && lastnread != NOCOUNT) {
209 				fprintf(msg, "eot\n");
210 				break;
211 			}
212 			fprintf(msg,
213 			    "file %d: eof after %ld records: %ld bytes\n",
214 			    filen, record, size);
215 			needeof = 1;
216 			filen++;
217 			tsize += size;
218 			size = record = lastrec = 0;
219 			lastnread = 0;
220 		}
221 		lastnread = nread;
222 	}
223 	fprintf(msg, "total length: %ld bytes\n", tsize);
224 	(void)signal(SIGINT, oldsig);
225 	if (op == COPY || op == COPYVERIFY) {
226 		writeop(outp, MTWEOF);
227 		writeop(outp, MTWEOF);
228 		if (op == COPYVERIFY) {
229 			writeop(outp, MTREW);
230 			writeop(inp, MTREW);
231 			verify(inp, outp, buff);
232 		}
233 	}
234 	exit(0);
235 }
236 
237 void
238 verify(inp, outp, outb)
239 	register int inp, outp;
240 	register char *outb;
241 {
242 	register int eot, inmaxblk, inn, outmaxblk, outn;
243 	register char *inb;
244 
245 	inb = getspace(maxblk);
246 	inmaxblk = outmaxblk = maxblk;
247 	for (eot = 0;; guesslen = 0) {
248 		if ((inn = read(inp, inb, inmaxblk)) == -1) {
249 			if (guesslen)
250 				while (errno == EINVAL && (inmaxblk -= 1024)) {
251 					inn = read(inp, inb, inmaxblk);
252 					if (inn >= 0)
253 						goto r1;
254 				}
255 			perror("tcopy: read error");
256 			break;
257 		}
258 r1:		if ((outn = read(outp, outb, outmaxblk)) == -1) {
259 			if (guesslen)
260 				while (errno == EINVAL && (outmaxblk -= 1024)) {
261 					outn = read(outp, outb, outmaxblk);
262 					if (outn >= 0)
263 						goto r2;
264 				}
265 			perror("tcopy: read error");
266 			break;
267 		}
268 r2:		if (inn != outn) {
269 			fprintf(msg,
270 			    "%s: tapes have different block sizes; %d != %d.\n",
271 			    "tcopy", inn, outn);
272 			break;
273 		}
274 		if (!inn) {
275 			if (eot++) {
276 				fprintf(msg, "tcopy: tapes are identical.\n");
277 				return;
278 			}
279 		} else {
280 			if (bcmp(inb, outb, inn)) {
281 				fprintf(msg,
282 				    "tcopy: tapes have different data.\n");
283 				break;
284 			}
285 			eot = 0;
286 		}
287 	}
288 	exit(1);
289 }
290 
291 void
292 intr(signo)
293 	int signo;
294 {
295 	if (record)
296 		if (record - lastrec > 1)
297 			fprintf(msg, "records %ld to %ld\n", lastrec, record);
298 		else
299 			fprintf(msg, "record %ld\n", lastrec);
300 	fprintf(msg, "interrupt at file %d: record %ld\n", filen, record);
301 	fprintf(msg, "total length: %ld bytes\n", tsize + size);
302 	exit(1);
303 }
304 
305 void *
306 getspace(blk)
307 	int blk;
308 {
309 	void *bp;
310 
311 	if ((bp = malloc((size_t)blk)) == NULL) {
312 		fprintf(stderr, "tcopy: no memory\n");
313 		exit(11);
314 	}
315 	return (bp);
316 }
317 
318 void
319 writeop(fd, type)
320 	int fd, type;
321 {
322 	struct mtop op;
323 
324 	op.mt_op = type;
325 	op.mt_count = (daddr_t)1;
326 	if (ioctl(fd, MTIOCTOP, (char *)&op) < 0) {
327 		perror("tcopy: tape op");
328 		exit(6);
329 	}
330 }
331 
332 void
333 usage()
334 {
335 	fprintf(stderr, "usage: tcopy [-cvx] [-s maxblk] src [dest]\n");
336 	exit(1);
337 }
338