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