xref: /openbsd-src/sbin/dump/tape.c (revision a7b9eedcb4c1ab985f28fc923822c24bc6417908)
1*a7b9eedcSflorian /*	$OpenBSD: tape.c,v 1.49 2024/05/09 08:35:40 florian Exp $	*/
22904a520Smillert /*	$NetBSD: tape.c,v 1.11 1997/06/05 11:13:26 lukem Exp $	*/
3df930be7Sderaadt 
4df930be7Sderaadt /*-
5df930be7Sderaadt  * Copyright (c) 1980, 1991, 1993
6df930be7Sderaadt  *	The Regents of the University of California.  All rights reserved.
7df930be7Sderaadt  *
8df930be7Sderaadt  * Redistribution and use in source and binary forms, with or without
9df930be7Sderaadt  * modification, are permitted provided that the following conditions
10df930be7Sderaadt  * are met:
11df930be7Sderaadt  * 1. Redistributions of source code must retain the above copyright
12df930be7Sderaadt  *    notice, this list of conditions and the following disclaimer.
13df930be7Sderaadt  * 2. Redistributions in binary form must reproduce the above copyright
14df930be7Sderaadt  *    notice, this list of conditions and the following disclaimer in the
15df930be7Sderaadt  *    documentation and/or other materials provided with the distribution.
161ef0d710Smillert  * 3. Neither the name of the University nor the names of its contributors
17df930be7Sderaadt  *    may be used to endorse or promote products derived from this software
18df930be7Sderaadt  *    without specific prior written permission.
19df930be7Sderaadt  *
20df930be7Sderaadt  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
21df930be7Sderaadt  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
22df930be7Sderaadt  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
23df930be7Sderaadt  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
24df930be7Sderaadt  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
25df930be7Sderaadt  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
26df930be7Sderaadt  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
27df930be7Sderaadt  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
28df930be7Sderaadt  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
29df930be7Sderaadt  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
30df930be7Sderaadt  * SUCH DAMAGE.
31df930be7Sderaadt  */
32df930be7Sderaadt 
3378eb0b7eSderaadt #include <sys/param.h>	/* MAXBSIZE DEV_BSIZE */
34df930be7Sderaadt #include <sys/socket.h>
35df930be7Sderaadt #include <sys/time.h>
36df930be7Sderaadt #include <sys/wait.h>
372509e7acSkrw #include <sys/stat.h>
38df930be7Sderaadt #include <ufs/ffs/fs.h>
39df930be7Sderaadt #include <ufs/ufs/dinode.h>
40df930be7Sderaadt 
41df930be7Sderaadt #include <protocols/dumprestore.h>
42df930be7Sderaadt 
43df930be7Sderaadt #include <errno.h>
44df930be7Sderaadt #include <fcntl.h>
45df930be7Sderaadt #include <signal.h>
46df930be7Sderaadt #include <stdio.h>
47df930be7Sderaadt #include <stdlib.h>
48df930be7Sderaadt #include <string.h>
492904a520Smillert #include <time.h>
50df930be7Sderaadt #include <unistd.h>
51b9fc9a72Sderaadt #include <limits.h>
52df930be7Sderaadt 
53df930be7Sderaadt #include "dump.h"
54df930be7Sderaadt #include "pathnames.h"
55df930be7Sderaadt 
56b9fc9a72Sderaadt #define MINIMUM(a, b)	(((a) < (b)) ? (a) : (b))
57b9fc9a72Sderaadt 
589be7043fSmortimer ino_t   curino; 		/* current inumber; used globally */
599be7043fSmortimer int     newtape;		/* new tape flag */
609be7043fSmortimer union u_spcl u_spcl;		/* mapping of variables in a control block */
619be7043fSmortimer 
629be7043fSmortimer static int tapefd;		/* tape file descriptor */
639be7043fSmortimer static int64_t asize;		/* number of 0.1" units written on cur tape */
649be7043fSmortimer static int writesize;		/* size of malloc()ed buffer for tape */
659be7043fSmortimer static int64_t	lastspclrec = -1; /* tape block number of last written header */
669be7043fSmortimer static int trecno = 0;		/* next record to write in current block */
679be7043fSmortimer static int64_t blocksthisvol;	/* number of blocks on current output file */
689be7043fSmortimer static char *nexttape;
69df930be7Sderaadt 
7053fedb49Sderaadt static	ssize_t atomic(ssize_t (*)(int, void *, size_t), int, char *, int);
71c72b5b24Smillert static	void doslave(int, int);
72c72b5b24Smillert static	void enslave(void);
73c72b5b24Smillert static	void flushtape(void);
74c72b5b24Smillert static	void killall(void);
75c72b5b24Smillert static	void rollforward(void);
76df930be7Sderaadt 
77e07939ebSderaadt void	tperror(int signo);
78e07939ebSderaadt void	sigpipe(int signo);
79e07939ebSderaadt void	proceed(int signo);
80e07939ebSderaadt 
81df930be7Sderaadt /*
82df930be7Sderaadt  * Concurrent dump mods (Caltech) - disk block reading and tape writing
83df930be7Sderaadt  * are exported to several slave processes.  While one slave writes the
84df930be7Sderaadt  * tape, the others read disk blocks; they pass control of the tape in
85df930be7Sderaadt  * a ring via signals. The parent process traverses the filesystem and
86df930be7Sderaadt  * sends writeheader()'s and lists of daddr's to the slaves via pipes.
87df930be7Sderaadt  * The following structure defines the instruction packets sent to slaves.
88df930be7Sderaadt  */
89df930be7Sderaadt struct req {
901abdbfdeSderaadt 	daddr_t dblk;
91df930be7Sderaadt 	int count;
92df930be7Sderaadt };
939be7043fSmortimer static int reqsiz;
94df930be7Sderaadt 
95df930be7Sderaadt #define SLAVES 3		/* 1 slave writing, 1 reading, 1 for slack */
969be7043fSmortimer static struct slave {
970155e653Smillert 	int64_t tapea;		/* header number at start of this chunk */
980155e653Smillert 	int64_t firstrec;	/* record number of this block */
99df930be7Sderaadt 	int count;		/* count to next header (used for TS_TAPE */
100df930be7Sderaadt 				/* after EOT) */
101df930be7Sderaadt 	int inode;		/* inode that we are currently dealing with */
102df930be7Sderaadt 	int fd;			/* FD for this slave */
1030a3b99ccScloder 	pid_t pid;		/* PID for this slave */
104df930be7Sderaadt 	int sent;		/* 1 == we've sent this slave requests */
105df930be7Sderaadt 	char (*tblock)[TP_BSIZE]; /* buffer for data blocks */
106df930be7Sderaadt 	struct req *req;	/* buffer for requests */
107df930be7Sderaadt } slaves[SLAVES+1];
1089be7043fSmortimer static struct slave *slp;
109df930be7Sderaadt 
1109be7043fSmortimer static char	(*nextblock)[TP_BSIZE];
111df930be7Sderaadt 
1122904a520Smillert static time_t tstart_volume;	/* time of volume start */
1130155e653Smillert static int64_t tapea_volume;	/* value of spcl.c_tapea at volume start */
1142904a520Smillert 
1159be7043fSmortimer static pid_t master;		/* pid of master, for sending error signals */
1169be7043fSmortimer static int tenths;		/* length of tape used per block written */
117bc7f19ddSmillert static volatile sig_atomic_t caught;	/* have we caught the signal to proceed? */
118df930be7Sderaadt 
119df930be7Sderaadt int
alloctape(void)12053fedb49Sderaadt alloctape(void)
121df930be7Sderaadt {
122df930be7Sderaadt 	int pgoff = getpagesize() - 1;
123df930be7Sderaadt 	char *buf;
124df930be7Sderaadt 	int i;
125df930be7Sderaadt 
126df930be7Sderaadt 	writesize = ntrec * TP_BSIZE;
127df930be7Sderaadt 	reqsiz = (ntrec + 1) * sizeof(struct req);
128df930be7Sderaadt 	/*
129df930be7Sderaadt 	 * CDC 92181's and 92185's make 0.8" gaps in 1600-bpi start/stop mode
130df930be7Sderaadt 	 * (see DEC TU80 User's Guide).  The shorter gaps of 6250-bpi require
131df930be7Sderaadt 	 * repositioning after stopping, i.e, streaming mode, where the gap is
132df930be7Sderaadt 	 * variable, 0.30" to 0.45".  The gap is maximal when the tape stops.
133df930be7Sderaadt 	 */
13431ebca96Sderaadt 	if (blocksperfile == 0 && !unlimited)
135df930be7Sderaadt 		tenths = writesize / density +
136df930be7Sderaadt 		    (cartridge ? 16 : density == 625 ? 5 : 8);
137df930be7Sderaadt 	/*
138df930be7Sderaadt 	 * Allocate tape buffer contiguous with the array of instruction
139df930be7Sderaadt 	 * packets, so flushtape() can write them together with one write().
140df930be7Sderaadt 	 * Align tape buffer on page boundary to speed up tape write().
141df930be7Sderaadt 	 */
142df930be7Sderaadt 	for (i = 0; i <= SLAVES; i++) {
1435ae94ef8Sderaadt 		buf = malloc((unsigned)(reqsiz + writesize + pgoff + TP_BSIZE));
144df930be7Sderaadt 		if (buf == NULL)
145df930be7Sderaadt 			return(0);
146df930be7Sderaadt 		slaves[i].tblock = (char (*)[TP_BSIZE])
147df930be7Sderaadt 		    (((long)&buf[ntrec + 1] + pgoff) &~ pgoff);
148df930be7Sderaadt 		slaves[i].req = (struct req *)slaves[i].tblock - ntrec - 1;
149df930be7Sderaadt 	}
150df930be7Sderaadt 	slp = &slaves[0];
151df930be7Sderaadt 	slp->count = 1;
152df930be7Sderaadt 	slp->tapea = 0;
153df930be7Sderaadt 	slp->firstrec = 0;
154df930be7Sderaadt 	nextblock = slp->tblock;
155df930be7Sderaadt 	return(1);
156df930be7Sderaadt }
157df930be7Sderaadt 
158df930be7Sderaadt void
writerec(char * dp,int isspcl)15953fedb49Sderaadt writerec(char *dp, int isspcl)
160df930be7Sderaadt {
161df930be7Sderaadt 
162fd08467eSkrw 	slp->req[trecno].dblk = 0;
163df930be7Sderaadt 	slp->req[trecno].count = 1;
164df930be7Sderaadt 	*(union u_spcl *)(*(nextblock)++) = *(union u_spcl *)dp;
165df930be7Sderaadt 	if (isspcl)
166df930be7Sderaadt 		lastspclrec = spcl.c_tapea;
167df930be7Sderaadt 	trecno++;
168df930be7Sderaadt 	spcl.c_tapea++;
169df930be7Sderaadt 	if (trecno >= ntrec)
170df930be7Sderaadt 		flushtape();
171df930be7Sderaadt }
172df930be7Sderaadt 
173df930be7Sderaadt void
dumpblock(daddr_t blkno,int size)1741abdbfdeSderaadt dumpblock(daddr_t blkno, int size)
175df930be7Sderaadt {
1760155e653Smillert 	int avail, tpblks;
1771abdbfdeSderaadt 	daddr_t dblkno;
178df930be7Sderaadt 
179df930be7Sderaadt 	dblkno = fsbtodb(sblock, blkno);
180df930be7Sderaadt 	tpblks = size >> tp_bshift;
181b9fc9a72Sderaadt 	while ((avail = MINIMUM(tpblks, ntrec - trecno)) > 0) {
182df930be7Sderaadt 		slp->req[trecno].dblk = dblkno;
183df930be7Sderaadt 		slp->req[trecno].count = avail;
184df930be7Sderaadt 		trecno += avail;
185df930be7Sderaadt 		spcl.c_tapea += avail;
186df930be7Sderaadt 		if (trecno >= ntrec)
187df930be7Sderaadt 			flushtape();
188171a3685Skrw 		dblkno += avail << (tp_bshift - (ffs(DEV_BSIZE) - 1));
189df930be7Sderaadt 		tpblks -= avail;
190df930be7Sderaadt 	}
191df930be7Sderaadt }
192df930be7Sderaadt 
193df930be7Sderaadt int	nogripe = 0;
194df930be7Sderaadt 
195df930be7Sderaadt void
tperror(int signo)19653fedb49Sderaadt tperror(int signo)
197df930be7Sderaadt {
198bc7f19ddSmillert 	/* XXX - signal races */
199df930be7Sderaadt 
200df930be7Sderaadt 	if (pipeout) {
201df930be7Sderaadt 		msg("write error on %s\n", tape);
202df930be7Sderaadt 		quit("Cannot recover\n");
203df930be7Sderaadt 		/* NOTREACHED */
204df930be7Sderaadt 	}
2050603cbefSnaddy 	msg("write error %lld blocks into volume %d\n",
2060603cbefSnaddy 	    (long long)blocksthisvol, tapeno);
207df930be7Sderaadt 	broadcast("DUMP WRITE ERROR!\n");
208df930be7Sderaadt 	if (!query("Do you want to restart?"))
209df930be7Sderaadt 		dumpabort(0);
210df930be7Sderaadt 	msg("Closing this volume.  Prepare to restart with new media;\n");
211df930be7Sderaadt 	msg("this dump volume will be rewritten.\n");
212df930be7Sderaadt 	killall();
213df930be7Sderaadt 	nogripe = 1;
214df930be7Sderaadt 	close_rewind();
215df930be7Sderaadt 	Exit(X_REWRITE);
216df930be7Sderaadt }
217df930be7Sderaadt 
218df930be7Sderaadt void
sigpipe(int signo)21953fedb49Sderaadt sigpipe(int signo)
220df930be7Sderaadt {
221df930be7Sderaadt 
222df930be7Sderaadt 	quit("Broken pipe\n");
223df930be7Sderaadt }
224df930be7Sderaadt 
2252904a520Smillert /*
2262904a520Smillert  * do_stats --
2272904a520Smillert  *	Update xferrate stats
2282904a520Smillert  */
2292904a520Smillert time_t
do_stats(void)23053fedb49Sderaadt do_stats(void)
2312904a520Smillert {
2322904a520Smillert 	time_t tnow, ttaken;
233a59aca37Sotto 	int64_t blocks;
234*a7b9eedcSflorian 	char *ct;
2352904a520Smillert 
2362904a520Smillert 	(void)time(&tnow);
2372904a520Smillert 	ttaken = tnow - tstart_volume;
2382904a520Smillert 	blocks = spcl.c_tapea - tapea_volume;
239*a7b9eedcSflorian 	ct = ctime(&tnow);
240*a7b9eedcSflorian 	msg("Volume %d completed at: %s", tapeno, ct ? ct : "?\n");
2412904a520Smillert 	if (ttaken > 0) {
242520eb379Sotto 		msg("Volume %d took %lld:%02lld:%02lld\n", tapeno,
243520eb379Sotto 		    (long long)ttaken / 3600, ((long long)ttaken % 3600) / 60,
244520eb379Sotto 		    (long long)ttaken % 60);
245a59aca37Sotto 		blocks /= ttaken;
246a59aca37Sotto 		msg("Volume %d transfer rate: %lld KB/s\n", tapeno, blocks);
247a59aca37Sotto 		xferrate += blocks;
2482904a520Smillert 	}
2492904a520Smillert 	return(tnow);
2502904a520Smillert }
2512904a520Smillert 
2522904a520Smillert /*
2532904a520Smillert  * statussig --
2542904a520Smillert  *	information message upon receipt of SIGINFO
2552904a520Smillert  *	(derived from optr.c::timeest())
256f72e66a0Sderaadt  * XXX not safe
2572904a520Smillert  */
2582904a520Smillert void
statussig(int signo)2590e419460Sderaadt statussig(int signo)
2602904a520Smillert {
2612904a520Smillert 	time_t	tnow, deltat;
262e792503aSangelos 	int save_errno = errno;
2632904a520Smillert 
2642904a520Smillert 	if (blockswritten < 500)
2652904a520Smillert 		return;
26691edd796Sderaadt 	(void) time(&tnow);
2672904a520Smillert 	deltat = tstart_writing - tnow + (1.0 * (tnow - tstart_writing))
2682904a520Smillert 		/ blockswritten * tapesize;
269753a1adaSderaadt 	/* XXX not safe due to floating point printf */
270753a1adaSderaadt 	dprintf(STDERR_FILENO,
2710155e653Smillert 	    "dump: %s %3.2f%% done at %lld KB/s, finished in %d:%02d\n",
272e50609ceSderaadt 	    tape, (blockswritten * 100.0) / tapesize,
2732904a520Smillert 	    (spcl.c_tapea - tapea_volume) / (tnow - tstart_volume),
2742904a520Smillert 	    (int)(deltat / 3600), (int)((deltat % 3600) / 60));
275e792503aSangelos 	errno = save_errno;
2762904a520Smillert }
2772904a520Smillert 
278df930be7Sderaadt static void
flushtape(void)27953fedb49Sderaadt flushtape(void)
280df930be7Sderaadt {
281df930be7Sderaadt 	int i, blks, got;
2820155e653Smillert 	int64_t lastfirstrec;
283df930be7Sderaadt 
284df930be7Sderaadt 	int siz = (char *)nextblock - (char *)slp->req;
285df930be7Sderaadt 
286df930be7Sderaadt 	slp->req[trecno].count = 0;			/* Sentinel */
287df930be7Sderaadt 
28853fedb49Sderaadt 	if (atomic((ssize_t (*)(int, void *, size_t))write, slp->fd,
28953fedb49Sderaadt 	    (char *)slp->req, siz) != siz)
290df930be7Sderaadt 		quit("error writing command pipe: %s\n", strerror(errno));
291df930be7Sderaadt 	slp->sent = 1; /* we sent a request, read the response later */
292df930be7Sderaadt 
293df930be7Sderaadt 	lastfirstrec = slp->firstrec;
294df930be7Sderaadt 
295df930be7Sderaadt 	if (++slp >= &slaves[SLAVES])
296df930be7Sderaadt 		slp = &slaves[0];
297df930be7Sderaadt 
298df930be7Sderaadt 	/* Read results back from next slave */
299df930be7Sderaadt 	if (slp->sent) {
3002904a520Smillert 		if (atomic(read, slp->fd, (char *)&got, sizeof(got))
3012904a520Smillert 		    != sizeof(got)) {
302df930be7Sderaadt 			perror("  DUMP: error reading command pipe in master");
303df930be7Sderaadt 			dumpabort(0);
304df930be7Sderaadt 		}
305df930be7Sderaadt 		slp->sent = 0;
306df930be7Sderaadt 
307df930be7Sderaadt 		/* Check for end of tape */
308df930be7Sderaadt 		if (got < writesize) {
309df930be7Sderaadt 			msg("End of tape detected\n");
310df930be7Sderaadt 
311df930be7Sderaadt 			/*
312df930be7Sderaadt 			 * Drain the results, don't care what the values were.
313df930be7Sderaadt 			 * If we read them here then trewind won't...
314df930be7Sderaadt 			 */
315df930be7Sderaadt 			for (i = 0; i < SLAVES; i++) {
316df930be7Sderaadt 				if (slaves[i].sent) {
317df930be7Sderaadt 					if (atomic(read, slaves[i].fd,
3182904a520Smillert 					    (char *)&got, sizeof(got))
3192904a520Smillert 					    != sizeof(got)) {
320df930be7Sderaadt 						perror("  DUMP: error reading command pipe in master");
321df930be7Sderaadt 						dumpabort(0);
322df930be7Sderaadt 					}
323df930be7Sderaadt 					slaves[i].sent = 0;
324df930be7Sderaadt 				}
325df930be7Sderaadt 			}
326df930be7Sderaadt 
327df930be7Sderaadt 			close_rewind();
328df930be7Sderaadt 			rollforward();
329df930be7Sderaadt 			return;
330df930be7Sderaadt 		}
331df930be7Sderaadt 	}
332df930be7Sderaadt 
333df930be7Sderaadt 	blks = 0;
334a08df88eSotto 	if (spcl.c_type != TS_END && spcl.c_type != TS_CLRI &&
335a08df88eSotto 	    spcl.c_type != TS_BITS) {
336a08df88eSotto 		if (spcl.c_count > TP_NINDIR)
337a08df88eSotto 			quit("c_count too large\n");
338df930be7Sderaadt 		for (i = 0; i < spcl.c_count; i++)
339df930be7Sderaadt 			if (spcl.c_addr[i] != 0)
340df930be7Sderaadt 				blks++;
341df930be7Sderaadt 	}
342df930be7Sderaadt 	slp->count = lastspclrec + blks + 1 - spcl.c_tapea;
343df930be7Sderaadt 	slp->tapea = spcl.c_tapea;
344df930be7Sderaadt 	slp->firstrec = lastfirstrec + ntrec;
345df930be7Sderaadt 	slp->inode = curino;
346df930be7Sderaadt 	nextblock = slp->tblock;
347df930be7Sderaadt 	trecno = 0;
348df930be7Sderaadt 	asize += tenths;
349df930be7Sderaadt 	blockswritten += ntrec;
350df930be7Sderaadt 	blocksthisvol += ntrec;
35131ebca96Sderaadt 	if (!pipeout && !unlimited && (blocksperfile ?
352df930be7Sderaadt 	    (blocksthisvol >= blocksperfile) : (asize > tsize))) {
353df930be7Sderaadt 		close_rewind();
354df930be7Sderaadt 		startnewtape(0);
355df930be7Sderaadt 	}
356df930be7Sderaadt 	timeest();
357df930be7Sderaadt }
358df930be7Sderaadt 
359df930be7Sderaadt void
trewind(void)36053fedb49Sderaadt trewind(void)
361df930be7Sderaadt {
3622509e7acSkrw 	struct stat sb;
3632509e7acSkrw 	int f, got;
364df930be7Sderaadt 
365df930be7Sderaadt 	for (f = 0; f < SLAVES; f++) {
366df930be7Sderaadt 		/*
367df930be7Sderaadt 		 * Drain the results, but unlike EOT we DO (or should) care
368df930be7Sderaadt 		 * what the return values were, since if we detect EOT after
369df930be7Sderaadt 		 * we think we've written the last blocks to the tape anyway,
370df930be7Sderaadt 		 * we have to replay those blocks with rollforward.
371df930be7Sderaadt 		 *
372df930be7Sderaadt 		 * fixme: punt for now.
373df930be7Sderaadt 		 */
374df930be7Sderaadt 		if (slaves[f].sent) {
3752904a520Smillert 			if (atomic(read, slaves[f].fd, (char *)&got, sizeof(got))
3762904a520Smillert 			    != sizeof(got)) {
377df930be7Sderaadt 				perror("  DUMP: error reading command pipe in master");
378df930be7Sderaadt 				dumpabort(0);
379df930be7Sderaadt 			}
380df930be7Sderaadt 			slaves[f].sent = 0;
381df930be7Sderaadt 			if (got != writesize) {
382df930be7Sderaadt 				msg("EOT detected in last 2 tape records!\n");
383df930be7Sderaadt 				msg("Use a longer tape, decrease the size estimate\n");
384df930be7Sderaadt 				quit("or use no size estimate at all.\n");
385df930be7Sderaadt 			}
386df930be7Sderaadt 		}
387df930be7Sderaadt 		(void) close(slaves[f].fd);
388df930be7Sderaadt 	}
389df930be7Sderaadt 	while (wait((int *)NULL) >= 0)	/* wait for any signals from slaves */
390df930be7Sderaadt 		/* void */;
391df930be7Sderaadt 
392df930be7Sderaadt 	if (pipeout)
393df930be7Sderaadt 		return;
394df930be7Sderaadt 
395df930be7Sderaadt 	msg("Closing %s\n", tape);
396df930be7Sderaadt 
397df930be7Sderaadt #ifdef RDUMP
398df930be7Sderaadt 	if (host) {
399df930be7Sderaadt 		rmtclose();
40040d1eb0cShalex 		while (rmtopen(tape, O_RDONLY) < 0)
401df930be7Sderaadt 			sleep(10);
402df930be7Sderaadt 		rmtclose();
403df930be7Sderaadt 		return;
404df930be7Sderaadt 	}
405df930be7Sderaadt #endif
4062509e7acSkrw 	/*
4076d5c2f5fSkrw 	 * st(4) says: "Bit 1 of the minor number specifies whether an eject is
4086d5c2f5fSkrw 	 * attempted when the device is closed.  When it is set, the device
4096d5c2f5fSkrw 	 * will attempt to eject its media on close ...".
4102509e7acSkrw 	 *
4112509e7acSkrw 	 * If the tape has been ejected, looping on open() will generate 'Media
4122509e7acSkrw 	 * not present' errors until a tape is loaded. Once loaded the tape
4132509e7acSkrw 	 * will be immediately ejected as a result of the second close().
4142509e7acSkrw 	 *
4152509e7acSkrw 	 * So if the tape will be ejected, just close and return.
4162509e7acSkrw 	 */
4172509e7acSkrw 	if ((fstat(tapefd, &sb) == 0) && (minor(sb.st_rdev) & 0x02)) {
4182509e7acSkrw 		(void) close(tapefd);
4192509e7acSkrw 		return;
4202509e7acSkrw 	}
4212509e7acSkrw 
422df930be7Sderaadt 	(void) close(tapefd);
423df69c215Sderaadt 	while ((f = open(tape, O_RDONLY)) == -1)
424df930be7Sderaadt 		sleep (10);
425df930be7Sderaadt 	(void) close(f);
426df930be7Sderaadt }
427df930be7Sderaadt 
428df930be7Sderaadt void
close_rewind(void)42953fedb49Sderaadt close_rewind(void)
430df930be7Sderaadt {
431df930be7Sderaadt 	trewind();
4322904a520Smillert 	(void)do_stats();
433df930be7Sderaadt 	if (nexttape)
434df930be7Sderaadt 		return;
435df930be7Sderaadt 	if (!nogripe) {
436df930be7Sderaadt 		msg("Change Volumes: Mount volume #%d\n", tapeno+1);
437df930be7Sderaadt 		broadcast("CHANGE DUMP VOLUMES!\7\7\n");
438df930be7Sderaadt 	}
439df930be7Sderaadt 	while (!query("Is the new volume mounted and ready to go?"))
440df930be7Sderaadt 		if (query("Do you want to abort?")) {
441df930be7Sderaadt 			dumpabort(0);
442df930be7Sderaadt 			/*NOTREACHED*/
443df930be7Sderaadt 		}
444df930be7Sderaadt }
445df930be7Sderaadt 
446df930be7Sderaadt void
rollforward(void)44753fedb49Sderaadt rollforward(void)
448df930be7Sderaadt {
449e073c79dSmpech 	struct req *p, *q, *prev;
450e073c79dSmpech 	struct slave *tslp;
4510155e653Smillert 	int i, size, got;
4520155e653Smillert 	int64_t savedtapea;
453df930be7Sderaadt 	union u_spcl *ntb, *otb;
454df930be7Sderaadt 	tslp = &slaves[SLAVES];
455df930be7Sderaadt 	ntb = (union u_spcl *)tslp->tblock[1];
456df930be7Sderaadt 
457df930be7Sderaadt 	/*
458df930be7Sderaadt 	 * Each of the N slaves should have requests that need to
459df930be7Sderaadt 	 * be replayed on the next tape.  Use the extra slave buffers
460df930be7Sderaadt 	 * (slaves[SLAVES]) to construct request lists to be sent to
461df930be7Sderaadt 	 * each slave in turn.
462df930be7Sderaadt 	 */
463df930be7Sderaadt 	for (i = 0; i < SLAVES; i++) {
464df930be7Sderaadt 		q = &tslp->req[1];
465df930be7Sderaadt 		otb = (union u_spcl *)slp->tblock;
466df930be7Sderaadt 
467df930be7Sderaadt 		/*
468df930be7Sderaadt 		 * For each request in the current slave, copy it to tslp.
469df930be7Sderaadt 		 */
470df930be7Sderaadt 
471df930be7Sderaadt 		prev = NULL;
472df930be7Sderaadt 		for (p = slp->req; p->count > 0; p += p->count) {
473df930be7Sderaadt 			*q = *p;
474df930be7Sderaadt 			if (p->dblk == 0)
475df930be7Sderaadt 				*ntb++ = *otb++; /* copy the datablock also */
476df930be7Sderaadt 			prev = q;
477df930be7Sderaadt 			q += q->count;
478df930be7Sderaadt 		}
479df930be7Sderaadt 		if (prev == NULL)
48099ea2379Sderaadt 			quit("rollforward: protocol botch\n");
481df930be7Sderaadt 		if (prev->dblk != 0)
482df930be7Sderaadt 			prev->count -= 1;
483df930be7Sderaadt 		else
484df930be7Sderaadt 			ntb--;
485df930be7Sderaadt 		q -= 1;
486df930be7Sderaadt 		q->count = 0;
487df930be7Sderaadt 		q = &tslp->req[0];
488df930be7Sderaadt 		if (i == 0) {
489df930be7Sderaadt 			q->dblk = 0;
490df930be7Sderaadt 			q->count = 1;
491df930be7Sderaadt 			trecno = 0;
492df930be7Sderaadt 			nextblock = tslp->tblock;
493df930be7Sderaadt 			savedtapea = spcl.c_tapea;
494df930be7Sderaadt 			spcl.c_tapea = slp->tapea;
495df930be7Sderaadt 			startnewtape(0);
496df930be7Sderaadt 			spcl.c_tapea = savedtapea;
497df930be7Sderaadt 			lastspclrec = savedtapea - 1;
498df930be7Sderaadt 		}
499df930be7Sderaadt 		size = (char *)ntb - (char *)q;
50053fedb49Sderaadt 		if (atomic((ssize_t (*)(int, void *, size_t))write,
50153fedb49Sderaadt 		    slp->fd, (char *)q, size) != size) {
502df930be7Sderaadt 			perror("  DUMP: error writing command pipe");
503df930be7Sderaadt 			dumpabort(0);
504df930be7Sderaadt 		}
505df930be7Sderaadt 		slp->sent = 1;
506df930be7Sderaadt 		if (++slp >= &slaves[SLAVES])
507df930be7Sderaadt 			slp = &slaves[0];
508df930be7Sderaadt 
509df930be7Sderaadt 		q->count = 1;
510df930be7Sderaadt 
511df930be7Sderaadt 		if (prev->dblk != 0) {
512df930be7Sderaadt 			/*
513df930be7Sderaadt 			 * If the last one was a disk block, make the
514df930be7Sderaadt 			 * first of this one be the last bit of that disk
515df930be7Sderaadt 			 * block...
516df930be7Sderaadt 			 */
517df930be7Sderaadt 			q->dblk = prev->dblk +
518df930be7Sderaadt 				prev->count * (TP_BSIZE / DEV_BSIZE);
519df930be7Sderaadt 			ntb = (union u_spcl *)tslp->tblock;
520df930be7Sderaadt 		} else {
521df930be7Sderaadt 			/*
522df930be7Sderaadt 			 * It wasn't a disk block.  Copy the data to its
523df930be7Sderaadt 			 * new location in the buffer.
524df930be7Sderaadt 			 */
525df930be7Sderaadt 			q->dblk = 0;
526df930be7Sderaadt 			*((union u_spcl *)tslp->tblock) = *ntb;
527df930be7Sderaadt 			ntb = (union u_spcl *)tslp->tblock[1];
528df930be7Sderaadt 		}
529df930be7Sderaadt 	}
530df930be7Sderaadt 	slp->req[0] = *q;
531df930be7Sderaadt 	nextblock = slp->tblock;
532df930be7Sderaadt 	if (q->dblk == 0)
533df930be7Sderaadt 		nextblock++;
534df930be7Sderaadt 	trecno = 1;
535df930be7Sderaadt 
536df930be7Sderaadt 	/*
537df930be7Sderaadt 	 * Clear the first slaves' response.  One hopes that it
538df930be7Sderaadt 	 * worked ok, otherwise the tape is much too short!
539df930be7Sderaadt 	 */
540df930be7Sderaadt 	if (slp->sent) {
5412904a520Smillert 		if (atomic(read, slp->fd, (char *)&got, sizeof(got))
5422904a520Smillert 		    != sizeof(got)) {
543df930be7Sderaadt 			perror("  DUMP: error reading command pipe in master");
544df930be7Sderaadt 			dumpabort(0);
545df930be7Sderaadt 		}
546df930be7Sderaadt 		slp->sent = 0;
547df930be7Sderaadt 
548df930be7Sderaadt 		if (got != writesize) {
549df930be7Sderaadt 			quit("EOT detected at start of the tape!\n");
550df930be7Sderaadt 		}
551df930be7Sderaadt 	}
552df930be7Sderaadt }
553df930be7Sderaadt 
554df930be7Sderaadt /*
555df930be7Sderaadt  * We implement taking and restoring checkpoints on the tape level.
556df930be7Sderaadt  * When each tape is opened, a new process is created by forking; this
557df930be7Sderaadt  * saves all of the necessary context in the parent.  The child
558df930be7Sderaadt  * continues the dump; the parent waits around, saving the context.
559df930be7Sderaadt  * If the child returns X_REWRITE, then it had problems writing that tape;
560df930be7Sderaadt  * this causes the parent to fork again, duplicating the context, and
561df930be7Sderaadt  * everything continues as if nothing had happened.
562df930be7Sderaadt  */
563df930be7Sderaadt void
startnewtape(int top)56453fedb49Sderaadt startnewtape(int top)
565df930be7Sderaadt {
5660a3b99ccScloder 	pid_t	parentpid;
5670a3b99ccScloder 	pid_t	childpid;
568df930be7Sderaadt 	int	status;
5690a3b99ccScloder 	pid_t	waitingpid;
570*a7b9eedcSflorian 	char	*p, *ct;
571df930be7Sderaadt 	sig_t	interrupt_save;
572df930be7Sderaadt 
573df930be7Sderaadt 	interrupt_save = signal(SIGINT, SIG_IGN);
574df930be7Sderaadt 	parentpid = getpid();
5752904a520Smillert 	tapea_volume = spcl.c_tapea;
5762904a520Smillert 	(void)time(&tstart_volume);
577df930be7Sderaadt 
578df930be7Sderaadt restore_check_point:
579df930be7Sderaadt 	(void)signal(SIGINT, interrupt_save);
580df930be7Sderaadt 	/*
581df930be7Sderaadt 	 *	All signals are inherited...
582df930be7Sderaadt 	 */
583df930be7Sderaadt 	childpid = fork();
584df69c215Sderaadt 	if (childpid == -1) {
585df930be7Sderaadt 		msg("Context save fork fails in parent %d\n", parentpid);
586df930be7Sderaadt 		Exit(X_ABORT);
587df930be7Sderaadt 	}
588df930be7Sderaadt 	if (childpid != 0) {
589df930be7Sderaadt 		/*
590df930be7Sderaadt 		 *	PARENT:
591df930be7Sderaadt 		 *	save the context by waiting
592df930be7Sderaadt 		 *	until the child doing all of the work returns.
593df930be7Sderaadt 		 *	don't catch the interrupt
594df930be7Sderaadt 		 */
595df930be7Sderaadt 		signal(SIGINT, SIG_IGN);
596df930be7Sderaadt #ifdef TDEBUG
597df930be7Sderaadt 		msg("Tape: %d; parent process: %d child process %d\n",
598df930be7Sderaadt 			tapeno+1, parentpid, childpid);
599df930be7Sderaadt #endif /* TDEBUG */
6000e419460Sderaadt 		while ((waitingpid = wait(&status)) != childpid)
601df930be7Sderaadt 			msg("Parent %d waiting for child %d has another child %d return\n",
6020e419460Sderaadt 				parentpid, childpid, waitingpid);
603df930be7Sderaadt 		if (status & 0xFF) {
604df930be7Sderaadt 			msg("Child %d returns LOB status %o\n",
605df930be7Sderaadt 				childpid, status&0xFF);
606df930be7Sderaadt 		}
607df930be7Sderaadt 		status = (status >> 8) & 0xFF;
608df930be7Sderaadt #ifdef TDEBUG
609df930be7Sderaadt 		switch(status) {
610df930be7Sderaadt 			case X_FINOK:
611df930be7Sderaadt 				msg("Child %d finishes X_FINOK\n", childpid);
612df930be7Sderaadt 				break;
613df930be7Sderaadt 			case X_ABORT:
614df930be7Sderaadt 				msg("Child %d finishes X_ABORT\n", childpid);
615df930be7Sderaadt 				break;
616df930be7Sderaadt 			case X_REWRITE:
617df930be7Sderaadt 				msg("Child %d finishes X_REWRITE\n", childpid);
618df930be7Sderaadt 				break;
619df930be7Sderaadt 			default:
620df930be7Sderaadt 				msg("Child %d finishes unknown %d\n",
621df930be7Sderaadt 					childpid, status);
622df930be7Sderaadt 				break;
623df930be7Sderaadt 		}
624df930be7Sderaadt #endif /* TDEBUG */
625df930be7Sderaadt 		switch(status) {
626df930be7Sderaadt 			case X_FINOK:
627df930be7Sderaadt 				Exit(X_FINOK);
6280e419460Sderaadt 				break;
629df930be7Sderaadt 			case X_ABORT:
630df930be7Sderaadt 				Exit(X_ABORT);
6310e419460Sderaadt 				break;
632df930be7Sderaadt 			case X_REWRITE:
633df930be7Sderaadt 				goto restore_check_point;
634df930be7Sderaadt 			default:
635df930be7Sderaadt 				msg("Bad return code from dump: %d\n", status);
636df930be7Sderaadt 				Exit(X_ABORT);
637df930be7Sderaadt 		}
638df930be7Sderaadt 		/*NOTREACHED*/
639df930be7Sderaadt 	} else {	/* we are the child; just continue */
640df930be7Sderaadt #ifdef TDEBUG
641df930be7Sderaadt 		sleep(4);	/* allow time for parent's message to get out */
642df930be7Sderaadt 		msg("Child on Tape %d has parent %d, my pid = %d\n",
643df930be7Sderaadt 			tapeno+1, parentpid, getpid());
644df930be7Sderaadt #endif /* TDEBUG */
645df930be7Sderaadt 		/*
6462904a520Smillert 		 * If we have a name like "/dev/rst0,/dev/rst1",
647df930be7Sderaadt 		 * use the name before the comma first, and save
648df930be7Sderaadt 		 * the remaining names for subsequent volumes.
649df930be7Sderaadt 		 */
650df930be7Sderaadt 		tapeno++;               /* current tape sequence */
651df930be7Sderaadt 		if (nexttape || strchr(tape, ',')) {
652df930be7Sderaadt 			if (nexttape && *nexttape)
653df930be7Sderaadt 				tape = nexttape;
654df930be7Sderaadt 			if ((p = strchr(tape, ',')) != NULL) {
655df930be7Sderaadt 				*p = '\0';
656df930be7Sderaadt 				nexttape = p + 1;
657df930be7Sderaadt 			} else
658df930be7Sderaadt 				nexttape = NULL;
659df930be7Sderaadt 			msg("Dumping volume %d on %s\n", tapeno, tape);
660df930be7Sderaadt 		}
661df930be7Sderaadt #ifdef RDUMP
66240d1eb0cShalex 		while ((tapefd = (host ? rmtopen(tape, O_WRONLY|O_CREAT) :
663df69c215Sderaadt 			pipeout ? 1 : open(tape, O_WRONLY|O_CREAT, 0666))) == -1)
664df930be7Sderaadt #else
665df930be7Sderaadt 		while ((tapefd = (pipeout ? 1 :
666df69c215Sderaadt 				  open(tape, O_WRONLY|O_CREAT, 0666))) == -1)
667df930be7Sderaadt #endif
668df930be7Sderaadt 		    {
669df930be7Sderaadt 			msg("Cannot open output \"%s\".\n", tape);
670df930be7Sderaadt 			if (!query("Do you want to retry the open?"))
671df930be7Sderaadt 				dumpabort(0);
672df930be7Sderaadt 		}
673df930be7Sderaadt 
674df930be7Sderaadt 		enslave();  /* Share open tape file descriptor with slaves */
675df930be7Sderaadt 
676df930be7Sderaadt 		asize = 0;
677df930be7Sderaadt 		blocksthisvol = 0;
678df930be7Sderaadt 		if (top)
679df930be7Sderaadt 			newtape++;		/* new tape signal */
680df930be7Sderaadt 		spcl.c_count = slp->count;
681df930be7Sderaadt 		/*
682df930be7Sderaadt 		 * measure firstrec in TP_BSIZE units since restore doesn't
683df930be7Sderaadt 		 * know the correct ntrec value...
684df930be7Sderaadt 		 */
685df930be7Sderaadt 		spcl.c_firstrec = slp->firstrec;
686df930be7Sderaadt 		spcl.c_volume++;
687df930be7Sderaadt 		spcl.c_type = TS_TAPE;
6880155e653Smillert 		if (sblock->fs_magic != FS_UFS2_MAGIC)
689df930be7Sderaadt 			spcl.c_flags |= DR_NEWHEADER;
690df930be7Sderaadt 		writeheader((ino_t)slp->inode);
6910155e653Smillert 		if (sblock->fs_magic != FS_UFS2_MAGIC)
692df930be7Sderaadt 			spcl.c_flags &=~ DR_NEWHEADER;
693*a7b9eedcSflorian 		ct = ctime(&tstart_volume);
694*a7b9eedcSflorian 		msg("Volume %d started at: %s", tapeno, ct ? ct : "?\n");
695df930be7Sderaadt 		if (tapeno > 1)
696d0f688a9Sderaadt 			msg("Volume %d begins with blocks from inode %llu\n",
697d0f688a9Sderaadt 			    tapeno, (unsigned long long)slp->inode);
698df930be7Sderaadt 	}
699df930be7Sderaadt }
700df930be7Sderaadt 
701df930be7Sderaadt void
dumpabort(int signo)70253fedb49Sderaadt dumpabort(int signo)
703df930be7Sderaadt {
704df930be7Sderaadt 
705df930be7Sderaadt 	if (master != 0 && master != getpid())
706df930be7Sderaadt 		/* Signals master to call dumpabort */
707df930be7Sderaadt 		(void) kill(master, SIGTERM);
708df930be7Sderaadt 	else {
709df930be7Sderaadt 		killall();
710df930be7Sderaadt 		msg("The ENTIRE dump is aborted.\n");
711df930be7Sderaadt 	}
712df930be7Sderaadt #ifdef RDUMP
713df930be7Sderaadt 	rmtclose();
714df930be7Sderaadt #endif
715df930be7Sderaadt 	Exit(X_ABORT);
716df930be7Sderaadt }
717df930be7Sderaadt 
718df930be7Sderaadt __dead void
Exit(int status)71953fedb49Sderaadt Exit(int status)
720df930be7Sderaadt {
721df930be7Sderaadt 
722df930be7Sderaadt #ifdef TDEBUG
723df930be7Sderaadt 	msg("pid = %d exits with status %d\n", getpid(), status);
724df930be7Sderaadt #endif /* TDEBUG */
725df930be7Sderaadt 	exit(status);
726df930be7Sderaadt }
727df930be7Sderaadt 
728df930be7Sderaadt /*
729df930be7Sderaadt  * proceed - handler for SIGUSR2, used to synchronize IO between the slaves.
730df930be7Sderaadt  */
731df930be7Sderaadt void
proceed(int signo)73253fedb49Sderaadt proceed(int signo)
733df930be7Sderaadt {
734df930be7Sderaadt 	caught++;
735df930be7Sderaadt }
736df930be7Sderaadt 
737df930be7Sderaadt void
enslave(void)73853fedb49Sderaadt enslave(void)
739df930be7Sderaadt {
740df930be7Sderaadt 	int cmd[2];
741e073c79dSmpech 	int i, j;
742df930be7Sderaadt 
743df930be7Sderaadt 	master = getpid();
744df930be7Sderaadt 
745df930be7Sderaadt 	signal(SIGTERM, dumpabort);  /* Slave sends SIGTERM on dumpabort() */
746df930be7Sderaadt 	signal(SIGPIPE, sigpipe);
747df930be7Sderaadt 	signal(SIGUSR1, tperror);    /* Slave sends SIGUSR1 on tape errors */
748df930be7Sderaadt 	signal(SIGUSR2, proceed);    /* Slave sends SIGUSR2 to next slave */
749df930be7Sderaadt 
750df930be7Sderaadt 	for (i = 0; i < SLAVES; i++) {
751df930be7Sderaadt 		if (i == slp - &slaves[0]) {
752df930be7Sderaadt 			caught = 1;
753df930be7Sderaadt 		} else {
754df930be7Sderaadt 			caught = 0;
755df930be7Sderaadt 		}
756df930be7Sderaadt 
757df69c215Sderaadt 		if (socketpair(AF_UNIX, SOCK_STREAM, 0, cmd) == -1 ||
758df69c215Sderaadt 		    (slaves[i].pid = fork()) == -1)
759df930be7Sderaadt 			quit("too many slaves, %d (recompile smaller): %s\n",
760df930be7Sderaadt 			    i, strerror(errno));
761df930be7Sderaadt 
762df930be7Sderaadt 		slaves[i].fd = cmd[1];
763df930be7Sderaadt 		slaves[i].sent = 0;
764df930be7Sderaadt 		if (slaves[i].pid == 0) { 	    /* Slave starts up here */
765df930be7Sderaadt 			for (j = 0; j <= i; j++)
766df930be7Sderaadt 			        (void) close(slaves[j].fd);
767df930be7Sderaadt 			signal(SIGINT, SIG_IGN);    /* Master handles this */
7682904a520Smillert 			signal(SIGINFO, SIG_IGN);
769df930be7Sderaadt 			doslave(cmd[0], i);
770df930be7Sderaadt 			Exit(X_FINOK);
771df930be7Sderaadt 		}
772df930be7Sderaadt 	}
773df930be7Sderaadt 
774df930be7Sderaadt 	for (i = 0; i < SLAVES; i++)
77553fedb49Sderaadt 		(void) atomic((ssize_t (*)(int, void *, size_t))write,
77653fedb49Sderaadt 		    slaves[i].fd, (char *) &slaves[(i + 1) % SLAVES].pid,
7772904a520Smillert 		    sizeof(slaves[0].pid));
778df930be7Sderaadt 	master = 0;
779df930be7Sderaadt }
780df930be7Sderaadt 
781df930be7Sderaadt void
killall(void)78253fedb49Sderaadt killall(void)
783df930be7Sderaadt {
784e073c79dSmpech 	int i;
785df930be7Sderaadt 
786df930be7Sderaadt 	for (i = 0; i < SLAVES; i++)
7876cbcad2bSderaadt 		if (slaves[i].pid > 0) {
788df930be7Sderaadt 			(void) kill(slaves[i].pid, SIGKILL);
7896cbcad2bSderaadt 			slaves[i].pid = 0;
7906cbcad2bSderaadt 		}
791df930be7Sderaadt }
792df930be7Sderaadt 
793df930be7Sderaadt /*
794df930be7Sderaadt  * Synchronization - each process has a lockfile, and shares file
795df930be7Sderaadt  * descriptors to the following process's lockfile.  When our write
796df930be7Sderaadt  * completes, we release our lock on the following process's lock-
797df930be7Sderaadt  * file, allowing the following process to lock it and proceed. We
798df930be7Sderaadt  * get the lock back for the next cycle by swapping descriptors.
799df930be7Sderaadt  */
800df930be7Sderaadt static void
doslave(int cmd,int slave_number)80153fedb49Sderaadt doslave(int cmd, int slave_number)
802df930be7Sderaadt {
8030155e653Smillert 	int nread, nextslave, size, wrote = 0, eot_count;
8045e769311Smillert 	sigset_t nsigset, osigset;
805df930be7Sderaadt 
806df930be7Sderaadt 	/*
807df930be7Sderaadt 	 * Need our own seek pointer.
808df930be7Sderaadt 	 */
809df930be7Sderaadt 	(void) close(diskfd);
810df69c215Sderaadt 	if ((diskfd = open(disk, O_RDONLY)) == -1)
811df930be7Sderaadt 		quit("slave couldn't reopen disk: %s\n", strerror(errno));
812df930be7Sderaadt 
813df930be7Sderaadt 	/*
814df930be7Sderaadt 	 * Need the pid of the next slave in the loop...
815df930be7Sderaadt 	 */
8162904a520Smillert 	if ((nread = atomic(read, cmd, (char *)&nextslave, sizeof(nextslave)))
8172904a520Smillert 	    != sizeof(nextslave)) {
818df930be7Sderaadt 		quit("master/slave protocol botched - didn't get pid of next slave.\n");
819df930be7Sderaadt 	}
820df930be7Sderaadt 
821df930be7Sderaadt 	/*
822df930be7Sderaadt 	 * Get list of blocks to dump, read the blocks into tape buffer
823df930be7Sderaadt 	 */
824df930be7Sderaadt 	while ((nread = atomic(read, cmd, (char *)slp->req, reqsiz)) == reqsiz) {
825e073c79dSmpech 		struct req *p = slp->req;
826df930be7Sderaadt 
827df930be7Sderaadt 		for (trecno = 0; trecno < ntrec;
828df930be7Sderaadt 		     trecno += p->count, p += p->count) {
829df930be7Sderaadt 			if (p->dblk) {
830df930be7Sderaadt 				bread(p->dblk, slp->tblock[trecno],
831df930be7Sderaadt 					p->count * TP_BSIZE);
832df930be7Sderaadt 			} else {
833df930be7Sderaadt 				if (p->count != 1 || atomic(read, cmd,
834df930be7Sderaadt 				    (char *)slp->tblock[trecno],
835df930be7Sderaadt 				    TP_BSIZE) != TP_BSIZE)
836df930be7Sderaadt 				       quit("master/slave protocol botched.\n");
837df930be7Sderaadt 			}
838df930be7Sderaadt 		}
8395e769311Smillert 
8405e769311Smillert 		sigemptyset(&nsigset);
8415e769311Smillert 		sigaddset(&nsigset, SIGUSR2);
8425e769311Smillert 		sigprocmask(SIG_BLOCK, &nsigset, &osigset);
8435e769311Smillert 		while (!caught)
8445e769311Smillert 			sigsuspend(&osigset);
845df930be7Sderaadt 		caught = 0;
8465e769311Smillert 		sigprocmask(SIG_SETMASK, &osigset, NULL);
847df930be7Sderaadt 
848df930be7Sderaadt 		/* Try to write the data... */
849df930be7Sderaadt 		eot_count = 0;
850df930be7Sderaadt 		size = 0;
851df930be7Sderaadt 
852df930be7Sderaadt 		while (eot_count < 10 && size < writesize) {
853df930be7Sderaadt #ifdef RDUMP
854df930be7Sderaadt 			if (host)
855df930be7Sderaadt 				wrote = rmtwrite(slp->tblock[0]+size,
856df930be7Sderaadt 				    writesize-size);
857df930be7Sderaadt 			else
858df930be7Sderaadt #endif
859df930be7Sderaadt 				wrote = write(tapefd, slp->tblock[0]+size,
860df930be7Sderaadt 				    writesize-size);
861df930be7Sderaadt #ifdef WRITEDEBUG
862df930be7Sderaadt 			printf("slave %d wrote %d\n", slave_number, wrote);
863df930be7Sderaadt #endif
864df930be7Sderaadt 			if (wrote < 0)
865df930be7Sderaadt 				break;
866df930be7Sderaadt 			if (wrote == 0)
867df930be7Sderaadt 				eot_count++;
868df930be7Sderaadt 			size += wrote;
869df930be7Sderaadt 		}
870df930be7Sderaadt 
871df930be7Sderaadt #ifdef WRITEDEBUG
872df930be7Sderaadt 		if (size != writesize)
873df930be7Sderaadt 		 printf("slave %d only wrote %d out of %d bytes and gave up.\n",
874df930be7Sderaadt 		     slave_number, size, writesize);
875df930be7Sderaadt #endif
876df930be7Sderaadt 
877df930be7Sderaadt 		if (eot_count > 0)
878df930be7Sderaadt 			size = 0;
879df930be7Sderaadt 
880df930be7Sderaadt 		/*
8812df6c92dSderaadt 		 * Handle ENOSPC as an EOT condition
882df930be7Sderaadt 		 */
8832df6c92dSderaadt 		if (wrote < 0 && errno == ENOSPC) {
8842df6c92dSderaadt 			wrote = 0;
8852df6c92dSderaadt 			eot_count++;
8862df6c92dSderaadt 		}
8872df6c92dSderaadt 
888df930be7Sderaadt 		if (size < 0) {
889df930be7Sderaadt 			(void) kill(master, SIGUSR1);
8905e769311Smillert 			sigemptyset(&nsigset);
891df930be7Sderaadt 			for (;;)
8925e769311Smillert 				sigsuspend(&nsigset);
893df930be7Sderaadt 		} else {
894df930be7Sderaadt 			/*
895df930be7Sderaadt 			 * pass size of write back to master
896df930be7Sderaadt 			 * (for EOT handling)
897df930be7Sderaadt 			 */
89853fedb49Sderaadt 			(void) atomic((ssize_t (*)(int, void *, size_t))write,
89953fedb49Sderaadt 			    cmd, (char *)&size, sizeof(size));
900df930be7Sderaadt 		}
901df930be7Sderaadt 
902df930be7Sderaadt 		/*
903df930be7Sderaadt 		 * If partial write, don't want next slave to go.
904df930be7Sderaadt 		 * Also jolts him awake.
905df930be7Sderaadt 		 */
906df930be7Sderaadt 		(void) kill(nextslave, SIGUSR2);
907df930be7Sderaadt 	}
908df930be7Sderaadt 	if (nread != 0)
909df930be7Sderaadt 		quit("error reading command pipe: %s\n", strerror(errno));
910df930be7Sderaadt }
911df930be7Sderaadt 
912df930be7Sderaadt /*
913df930be7Sderaadt  * Since a read from a pipe may not return all we asked for,
914df930be7Sderaadt  * or a write may not write all we ask if we get a signal,
915df930be7Sderaadt  * loop until the count is satisfied (or error).
916df930be7Sderaadt  */
917df930be7Sderaadt static ssize_t
atomic(ssize_t (* func)(int,void *,size_t),int fd,char * buf,int count)91853fedb49Sderaadt atomic(ssize_t (*func)(int, void *, size_t), int fd, char *buf, int count)
919df930be7Sderaadt {
920df930be7Sderaadt 	ssize_t got, need = count;
921df930be7Sderaadt 
922df930be7Sderaadt 	while ((got = (*func)(fd, buf, need)) > 0 && (need -= got) > 0)
923df930be7Sderaadt 		buf += got;
924df930be7Sderaadt 	return (got < 0 ? got : count - need);
925df930be7Sderaadt }
926