]> git.wh0rd.org - dump.git/blob - dump/tape.c
Added --with-oldstylefscript in configure.
[dump.git] / dump / tape.c
1 /*
2 * Ported to Linux's Second Extended File System as part of the
3 * dump and restore backup suit
4 * Remy Card <card@Linux.EU.Org>, 1994-1997
5 * Stelian Pop <pop@cybercable.fr>, 1999-2000
6 */
7
8 /*-
9 * Copyright (c) 1980, 1991, 1993
10 * The Regents of the University of California. All rights reserved.
11 *
12 * Redistribution and use in source and binary forms, with or without
13 * modification, are permitted provided that the following conditions
14 * are met:
15 * 1. Redistributions of source code must retain the above copyright
16 * notice, this list of conditions and the following disclaimer.
17 * 2. Redistributions in binary form must reproduce the above copyright
18 * notice, this list of conditions and the following disclaimer in the
19 * documentation and/or other materials provided with the distribution.
20 * 3. All advertising materials mentioning features or use of this software
21 * must display the following acknowledgement:
22 * This product includes software developed by the University of
23 * California, Berkeley and its contributors.
24 * 4. Neither the name of the University nor the names of its contributors
25 * may be used to endorse or promote products derived from this software
26 * without specific prior written permission.
27 *
28 * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
29 * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
30 * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
31 * ARE DISCLAIMED. IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
32 * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
33 * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
34 * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
35 * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
36 * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
37 * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
38 * SUCH DAMAGE.
39 */
40
41 #ifndef lint
42 static const char rcsid[] =
43 "$Id: tape.c,v 1.20 2000/06/01 18:30:08 stelian Exp $";
44 #endif /* not lint */
45
46 #ifdef __linux__
47 #include <sys/types.h>
48 #include <linux/types.h>
49 #endif
50 #include <sys/param.h>
51 #include <sys/socket.h>
52 #include <sys/time.h>
53 #include <sys/wait.h>
54 #ifdef __linux__
55 #include <linux/ext2_fs.h>
56 #include <bsdcompat.h>
57 #else /* __linux__ */
58 #ifdef sunos
59 #include <sys/vnode.h>
60
61 #include <ufs/fs.h>
62 #include <ufs/inode.h>
63 #else
64 #include <ufs/ufs/dinode.h>
65 #include <ufs/ffs/fs.h>
66 #endif
67 #endif /* __linux__ */
68
69 #include <protocols/dumprestore.h>
70
71 #include <errno.h>
72 #include <fcntl.h>
73 #include <setjmp.h>
74 #include <signal.h>
75 #include <stdio.h>
76 #include <compaterr.h>
77 #ifdef __STDC__
78 #include <stdlib.h>
79 #include <string.h>
80 #include <unistd.h>
81 #else
82 int write(), read();
83 #endif
84
85 #ifdef __linux__
86 #include <ext2fs/ext2fs.h>
87 #endif
88
89 #include "dump.h"
90
91 int writesize; /* size of malloc()ed buffer for tape */
92 long lastspclrec = -1; /* tape block number of last written header */
93 int trecno = 0; /* next record to write in current block */
94 extern long blocksperfile; /* number of blocks per output file */
95 long blocksthisvol; /* number of blocks on current output file */
96 extern int ntrec; /* blocking factor on tape */
97 extern int cartridge;
98 extern char *host;
99 char *nexttape;
100 extern pid_t rshpid;
101 int eot_code = 1;
102
103 static ssize_t atomic_read __P((int, void *, size_t));
104 static ssize_t atomic_write __P((int, const void *, size_t));
105 static void doslave __P((int, int));
106 static void enslave __P((void));
107 static void flushtape __P((void));
108 static void killall __P((void));
109 static void rollforward __P((void));
110 static int system_command __P((const char *, const char *, int));
111
112 /*
113 * Concurrent dump mods (Caltech) - disk block reading and tape writing
114 * are exported to several slave processes. While one slave writes the
115 * tape, the others read disk blocks; they pass control of the tape in
116 * a ring via signals. The parent process traverses the filesystem and
117 * sends writeheader()'s and lists of daddr's to the slaves via pipes.
118 * The following structure defines the instruction packets sent to slaves.
119 */
120 struct req {
121 daddr_t dblk;
122 int count;
123 };
124 int reqsiz;
125
126 #define SLAVES 3 /* 1 slave writing, 1 reading, 1 for slack */
127 struct slave {
128 int tapea; /* header number at start of this chunk */
129 int count; /* count to next header (used for TS_TAPE */
130 /* after EOT) */
131 int inode; /* inode that we are currently dealing with */
132 int fd; /* FD for this slave */
133 int pid; /* PID for this slave */
134 int sent; /* 1 == we've sent this slave requests */
135 int firstrec; /* record number of this block */
136 char (*tblock)[TP_BSIZE]; /* buffer for data blocks */
137 struct req *req; /* buffer for requests */
138 } slaves[SLAVES+1];
139 struct slave *slp;
140
141 char (*nextblock)[TP_BSIZE];
142
143 static time_t tstart_volume; /* time of volume start */
144 static int tapea_volume; /* value of spcl.c_tapea at volume start */
145
146 int master; /* pid of master, for sending error signals */
147 int tenths; /* length of tape used per block written */
148 static int caught; /* have we caught the signal to proceed? */
149 static int ready; /* have we reached the lock point without having */
150 /* received the SIGUSR2 signal from the prev slave? */
151 static sigjmp_buf jmpbuf; /* where to jump to if we are ready when the */
152 /* SIGUSR2 arrives from the previous slave */
153
154 int
155 alloctape(void)
156 {
157 int pgoff = getpagesize() - 1;
158 char *buf;
159 int i;
160
161 writesize = ntrec * TP_BSIZE;
162 reqsiz = (ntrec + 1) * sizeof(struct req);
163 /*
164 * CDC 92181's and 92185's make 0.8" gaps in 1600-bpi start/stop mode
165 * (see DEC TU80 User's Guide). The shorter gaps of 6250-bpi require
166 * repositioning after stopping, i.e, streaming mode, where the gap is
167 * variable, 0.30" to 0.45". The gap is maximal when the tape stops.
168 */
169 if (blocksperfile == 0 && !unlimited)
170 tenths = writesize / density +
171 (cartridge ? 16 : density == 625 ? 5 : 8);
172 /*
173 * Allocate tape buffer contiguous with the array of instruction
174 * packets, so flushtape() can write them together with one write().
175 * Align tape buffer on page boundary to speed up tape write().
176 */
177 for (i = 0; i <= SLAVES; i++) {
178 buf = (char *)
179 malloc((unsigned)(reqsiz + writesize + pgoff + TP_BSIZE));
180 if (buf == NULL)
181 return(0);
182 slaves[i].tblock = (char (*)[TP_BSIZE])
183 #ifdef __linux__
184 (((long)&buf[reqsiz] + pgoff) &~ pgoff);
185 #else
186 (((long)&buf[ntrec + 1] + pgoff) &~ pgoff);
187 #endif
188 slaves[i].req = (struct req *)slaves[i].tblock - ntrec - 1;
189 }
190 slp = &slaves[0];
191 slp->count = 1;
192 slp->tapea = 0;
193 slp->firstrec = 0;
194 nextblock = slp->tblock;
195 return(1);
196 }
197
198 void
199 writerec(const void *dp, int isspcl)
200 {
201
202 slp->req[trecno].dblk = (daddr_t)0;
203 slp->req[trecno].count = 1;
204 /* XXX post increment triggers an egcs-1.1.2-12 bug on alpha/sparc */
205 *(union u_spcl *)(*(nextblock)) = *(union u_spcl *)dp;
206 nextblock++;
207 if (isspcl)
208 lastspclrec = spcl.c_tapea;
209 trecno++;
210 spcl.c_tapea++;
211 if (trecno >= ntrec)
212 flushtape();
213 }
214
215 void
216 dumpblock(daddr_t blkno, int size)
217 {
218 int avail, tpblks, dblkno;
219
220 dblkno = fsbtodb(sblock, blkno);
221 tpblks = size >> tp_bshift;
222 while ((avail = MIN(tpblks, ntrec - trecno)) > 0) {
223 slp->req[trecno].dblk = dblkno;
224 slp->req[trecno].count = avail;
225 trecno += avail;
226 spcl.c_tapea += avail;
227 if (trecno >= ntrec)
228 flushtape();
229 dblkno += avail << (tp_bshift - dev_bshift);
230 tpblks -= avail;
231 }
232 }
233
234 int nogripe = 0;
235
236 static void
237 tperror(int signo)
238 {
239
240 if (pipeout) {
241 msg("write error on %s\n", tape);
242 quit("Cannot recover\n");
243 /* NOTREACHED */
244 }
245 msg("write error %d blocks into volume %d\n", blocksthisvol, tapeno);
246 broadcast("DUMP WRITE ERROR!\n");
247 if (!query("Do you want to restart?"))
248 dumpabort(0);
249 msg("Closing this volume. Prepare to restart with new media;\n");
250 msg("this dump volume will be rewritten.\n");
251 killall();
252 nogripe = 1;
253 close_rewind();
254 Exit(X_REWRITE);
255 }
256
257 static void
258 sigpipe(int signo)
259 {
260
261 quit("Broken pipe\n");
262 }
263
264 /*
265 * do_stats --
266 * Update xferrate stats
267 */
268 time_t
269 do_stats(void)
270 {
271 time_t tnow, ttaken;
272 int blocks;
273
274 #ifdef __linux__
275 (void)time4(&tnow);
276 #else
277 (void)time(&tnow);
278 #endif
279 ttaken = tnow - tstart_volume;
280 blocks = spcl.c_tapea - tapea_volume;
281 msg("Volume %d completed at: %s", tapeno,
282 #ifdef __linux__
283 ctime4(&tnow));
284 #else
285 ctime(&tnow));
286 #endif
287 if (ttaken > 0) {
288 msg("Volume %d took %d:%02d:%02d\n", tapeno,
289 ttaken / 3600, (ttaken % 3600) / 60, ttaken % 60);
290 msg("Volume %d transfer rate: %ld KB/s\n", tapeno,
291 blocks / ttaken);
292 xferrate += blocks / ttaken;
293 }
294 return(tnow);
295 }
296
297 #if defined(SIGINFO)
298 /*
299 * statussig --
300 * information message upon receipt of SIGINFO
301 * (derived from optr.c::timeest())
302 */
303 void
304 statussig(int notused)
305 {
306 time_t tnow, deltat;
307 char msgbuf[128];
308 int save_errno = errno;
309
310 if (blockswritten < 500)
311 return;
312 #ifdef __linux__
313 (void) time4(&tnow);
314 #else
315 (void) time((time_t *) &tnow);
316 #endif
317 if (blockswritten > tapesize)
318 tapesize = blockswritten;
319 deltat = tstart_writing - tnow + (1.0 * (tnow - tstart_writing))
320 / blockswritten * tapesize;
321 (void)snprintf(msgbuf, sizeof(msgbuf),
322 "%3.2f%% done at %ld KB/s, finished in %d:%02d\n",
323 (blockswritten * 100.0) / tapesize,
324 (spcl.c_tapea - tapea_volume) / (tnow - tstart_volume),
325 (int)(deltat / 3600), (int)((deltat % 3600) / 60));
326 write(STDERR_FILENO, msgbuf, strlen(msgbuf));
327 errno = save_errno;
328 }
329 #endif
330
331 static void
332 flushtape(void)
333 {
334 int i, blks, got;
335 long lastfirstrec;
336
337 int siz = (char *)nextblock - (char *)slp->req;
338
339 slp->req[trecno].count = 0; /* Sentinel */
340
341 if (atomic_write( slp->fd, (char *)slp->req, siz) != siz)
342 quit("error writing command pipe: %s\n", strerror(errno));
343 slp->sent = 1; /* we sent a request, read the response later */
344
345 lastfirstrec = slp->firstrec;
346
347 if (++slp >= &slaves[SLAVES])
348 slp = &slaves[0];
349
350 /* Read results back from next slave */
351 if (slp->sent) {
352 if (atomic_read( slp->fd, (char *)&got, sizeof got)
353 != sizeof got) {
354 perror(" DUMP: error reading command pipe in master");
355 dumpabort(0);
356 }
357 slp->sent = 0;
358
359 /* Check for end of tape */
360 if (got < writesize) {
361 msg("End of tape detected\n");
362
363 /*
364 * Drain the results, don't care what the values were.
365 * If we read them here then trewind won't...
366 */
367 for (i = 0; i < SLAVES; i++) {
368 if (slaves[i].sent) {
369 if (atomic_read( slaves[i].fd,
370 (char *)&got, sizeof got)
371 != sizeof got) {
372 perror(" DUMP: error reading command pipe in master");
373 dumpabort(0);
374 }
375 slaves[i].sent = 0;
376 }
377 }
378
379 close_rewind();
380 rollforward();
381 return;
382 }
383 }
384
385 blks = 0;
386 if (spcl.c_type != TS_END) {
387 for (i = 0; i < spcl.c_count; i++)
388 if (spcl.c_addr[i] != 0)
389 blks++;
390 }
391 slp->count = lastspclrec + blks + 1 - spcl.c_tapea;
392 slp->tapea = spcl.c_tapea;
393 slp->firstrec = lastfirstrec + ntrec;
394 slp->inode = curino;
395 nextblock = slp->tblock;
396 trecno = 0;
397 asize += tenths;
398 blockswritten += ntrec;
399 blocksthisvol += ntrec;
400 if (!pipeout && !unlimited && (blocksperfile ?
401 (blocksthisvol >= blocksperfile) : (asize > tsize))) {
402 close_rewind();
403 startnewtape(0);
404 }
405 timeest();
406 }
407
408 /*
409 * Executes the command in a shell.
410 * Returns -1 if an error occured, the exit status of
411 * the command on success.
412 */
413 int system_command(const char *command, const char *device, int volnum) {
414 int pid, status;
415 char commandstr[4096];
416
417 pid = fork();
418 if (pid == -1) {
419 perror(" DUMP: unable to fork");
420 return -1;
421 }
422 if (pid == 0) {
423 setuid(getuid());
424 setgid(getgid());
425 #if OLD_STYLE_FSCRIPT
426 snprintf(commandstr, sizeof(commandstr), "%s", command);
427 #else
428 snprintf(commandstr, sizeof(commandstr), "%s %s %d", command, device, volnum);
429 #endif
430 commandstr[sizeof(commandstr) - 1] = '\0';
431 execl("/bin/sh", "sh", "-c", commandstr, NULL);
432 perror(" DUMP: unable to execute shell");
433 exit(-1);
434 }
435 do {
436 if (waitpid(pid, &status, 0) == -1) {
437 if (errno != EINTR) {
438 perror(" DUMP: waitpid error");
439 return -1;
440 }
441 } else {
442 if (WIFEXITED(status))
443 return WEXITSTATUS(status);
444 else
445 return -1;
446 }
447 } while(1);
448 }
449
450 time_t
451 trewind(void)
452 {
453 int f;
454 int got;
455
456 for (f = 0; f < SLAVES; f++) {
457 /*
458 * Drain the results, but unlike EOT we DO (or should) care
459 * what the return values were, since if we detect EOT after
460 * we think we've written the last blocks to the tape anyway,
461 * we have to replay those blocks with rollforward.
462 *
463 * fixme: punt for now.
464 */
465 if (slaves[f].sent) {
466 if (atomic_read( slaves[f].fd, (char *)&got, sizeof got)
467 != sizeof got) {
468 perror(" DUMP: error reading command pipe in master");
469 dumpabort(0);
470 }
471 slaves[f].sent = 0;
472 if (got != writesize) {
473 msg("EOT detected in last 2 tape records!\n");
474 msg("Use a longer tape, decrease the size estimate\n");
475 quit("or use no size estimate at all.\n");
476 }
477 }
478 (void) close(slaves[f].fd);
479 }
480 while (wait((int *)NULL) >= 0) /* wait for any signals from slaves */
481 /* void */;
482
483 if (!pipeout) {
484
485 msg("Closing %s\n", tape);
486
487 #ifdef RDUMP
488 if (host) {
489 rmtclose();
490 while (rmtopen(tape, 0) < 0)
491 sleep(10);
492 rmtclose();
493 }
494 else
495 #endif
496 {
497 (void) close(tapefd);
498 while ((f = open(tape, 0)) < 0)
499 sleep (10);
500 (void) close(f);
501 }
502 eot_code = 1;
503 if (eot_script && spcl.c_type != TS_END) {
504 msg("Launching %s\n", eot_script);
505 eot_code = system_command(eot_script, tape, tapeno);
506 }
507 if (eot_code != 0 && eot_code != 1) {
508 msg("Dump aborted by the end of tape script\n");
509 dumpabort(0);
510 }
511 }
512 return do_stats();
513 }
514
515
516 void
517 close_rewind(void)
518 {
519 (void)trewind();
520 if (nexttape || Mflag || (eot_code == 0) )
521 return;
522 if (!nogripe) {
523 msg("Change Volumes: Mount volume #%d\n", tapeno+1);
524 broadcast("CHANGE DUMP VOLUMES!\7\7\n");
525 }
526 while (!query("Is the new volume mounted and ready to go?"))
527 if (query("Do you want to abort?")) {
528 dumpabort(0);
529 /*NOTREACHED*/
530 }
531 }
532
533 void
534 rollforward(void)
535 {
536 register struct req *p, *q, *prev;
537 register struct slave *tslp;
538 int i, size, savedtapea, got;
539 union u_spcl *ntb, *otb;
540 #ifdef __linux__
541 int blks;
542 long lastfirstrec;
543 #endif
544 tslp = &slaves[SLAVES];
545 ntb = (union u_spcl *)tslp->tblock[1];
546
547 /*
548 * Each of the N slaves should have requests that need to
549 * be replayed on the next tape. Use the extra slave buffers
550 * (slaves[SLAVES]) to construct request lists to be sent to
551 * each slave in turn.
552 */
553 for (i = 0; i < SLAVES; i++) {
554 q = &tslp->req[1];
555 otb = (union u_spcl *)slp->tblock;
556
557 /*
558 * For each request in the current slave, copy it to tslp.
559 */
560
561 prev = NULL;
562 for (p = slp->req; p->count > 0; p += p->count) {
563 *q = *p;
564 if (p->dblk == 0)
565 *ntb++ = *otb++; /* copy the datablock also */
566 prev = q;
567 q += q->count;
568 }
569 if (prev == NULL)
570 quit("rollforward: protocol botch");
571 if (prev->dblk != 0)
572 prev->count -= 1;
573 else
574 ntb--;
575 q -= 1;
576 q->count = 0;
577 q = &tslp->req[0];
578 if (i == 0) {
579 q->dblk = 0;
580 q->count = 1;
581 trecno = 0;
582 nextblock = tslp->tblock;
583 savedtapea = spcl.c_tapea;
584 spcl.c_tapea = slp->tapea;
585 startnewtape(0);
586 spcl.c_tapea = savedtapea;
587 lastspclrec = savedtapea - 1;
588 }
589 size = (char *)ntb - (char *)q;
590 if (atomic_write( slp->fd, (char *)q, size) != size) {
591 perror(" DUMP: error writing command pipe");
592 dumpabort(0);
593 }
594 slp->sent = 1;
595 #ifdef __linux__
596 lastfirstrec = slp->firstrec;
597 #endif
598 if (++slp >= &slaves[SLAVES])
599 slp = &slaves[0];
600
601 q->count = 1;
602
603 if (prev->dblk != 0) {
604 /*
605 * If the last one was a disk block, make the
606 * first of this one be the last bit of that disk
607 * block...
608 */
609 q->dblk = prev->dblk +
610 prev->count * (TP_BSIZE / DEV_BSIZE);
611 ntb = (union u_spcl *)tslp->tblock;
612 } else {
613 /*
614 * It wasn't a disk block. Copy the data to its
615 * new location in the buffer.
616 */
617 q->dblk = 0;
618 *((union u_spcl *)tslp->tblock) = *ntb;
619 ntb = (union u_spcl *)tslp->tblock[1];
620 }
621 }
622 slp->req[0] = *q;
623 nextblock = slp->tblock;
624 if (q->dblk == 0) {
625 #ifdef __linux__
626 /* XXX post increment triggers an egcs-1.1.2-12 bug on alpha/sparc */
627 *(union u_spcl *)(*nextblock) = *(union u_spcl *)tslp->tblock;
628 #endif
629 nextblock++;
630 }
631 trecno = 1;
632
633 /*
634 * Clear the first slaves' response. One hopes that it
635 * worked ok, otherwise the tape is much too short!
636 */
637 if (slp->sent) {
638 if (atomic_read( slp->fd, (char *)&got, sizeof got)
639 != sizeof got) {
640 perror(" DUMP: error reading command pipe in master");
641 dumpabort(0);
642 }
643 slp->sent = 0;
644
645 if (got != writesize) {
646 quit("EOT detected at start of the tape!\n");
647 }
648 }
649
650 #ifdef __linux__
651 blks = 0;
652 if (spcl.c_type != TS_END) {
653 for (i = 0; i < spcl.c_count; i++)
654 if (spcl.c_addr[i] != 0)
655 blks++;
656 }
657
658 slp->firstrec = lastfirstrec + ntrec;
659 slp->count = lastspclrec + blks + 1 - spcl.c_tapea;
660 slp->inode = curino;
661 asize += tenths;
662 blockswritten += ntrec;
663 blocksthisvol += ntrec;
664 #endif
665 }
666
667 /*
668 * We implement taking and restoring checkpoints on the tape level.
669 * When each tape is opened, a new process is created by forking; this
670 * saves all of the necessary context in the parent. The child
671 * continues the dump; the parent waits around, saving the context.
672 * If the child returns X_REWRITE, then it had problems writing that tape;
673 * this causes the parent to fork again, duplicating the context, and
674 * everything continues as if nothing had happened.
675 */
676 void
677 startnewtape(int top)
678 {
679 int parentpid;
680 int childpid;
681 int status;
682 int waitpid;
683 char *p;
684
685 #ifdef __linux__
686 sigset_t sigs;
687 sigemptyset(&sigs);
688 sigaddset(&sigs, SIGINT);
689 sigprocmask(SIG_BLOCK, &sigs, NULL);
690 #else /* __linux__ */
691 #ifdef sunos
692 void (*interrupt_save)();
693 #else
694 sig_t interrupt_save;
695 #endif
696 interrupt_save = signal(SIGINT, SIG_IGN);
697 #endif /* __linux__ */
698
699 parentpid = getpid();
700 tapea_volume = spcl.c_tapea;
701 #ifdef __linux__
702 (void)time4(&tstart_volume);
703 #else
704 (void)time((&tstart_volume);
705 #endif
706
707 restore_check_point:
708 #ifdef __linux__
709 sigprocmask(SIG_UNBLOCK, &sigs, NULL);
710 #else
711 (void)signal(SIGINT, interrupt_save);
712 #endif
713 /*
714 * All signals are inherited...
715 */
716 childpid = fork();
717 if (childpid < 0) {
718 msg("Context save fork fails in parent %d\n", parentpid);
719 Exit(X_ABORT);
720 }
721 if (childpid != 0) {
722 /*
723 * PARENT:
724 * save the context by waiting
725 * until the child doing all of the work returns.
726 * don't catch the interrupt
727 */
728 #ifdef __linux__
729 sigprocmask(SIG_BLOCK, &sigs, NULL);
730 #else
731 signal(SIGINT, SIG_IGN);
732 #endif
733 #ifdef TDEBUG
734 msg("Tape: %d; parent process: %d child process %d\n",
735 tapeno+1, parentpid, childpid);
736 #endif /* TDEBUG */
737 while ((waitpid = wait(&status)) != childpid)
738 if (waitpid != rshpid)
739 msg("Parent %d waiting for child %d has another child %d return\n",
740 parentpid, childpid, waitpid);
741 if (status & 0xFF) {
742 msg("Child %d returns LOB status %o\n",
743 childpid, status&0xFF);
744 }
745 status = (status >> 8) & 0xFF;
746 #ifdef TDEBUG
747 switch(status) {
748 case X_FINOK:
749 msg("Child %d finishes X_FINOK\n", childpid);
750 break;
751 case X_ABORT:
752 msg("Child %d finishes X_ABORT\n", childpid);
753 break;
754 case X_REWRITE:
755 msg("Child %d finishes X_REWRITE\n", childpid);
756 break;
757 default:
758 msg("Child %d finishes unknown %d\n",
759 childpid, status);
760 break;
761 }
762 #endif /* TDEBUG */
763 switch(status) {
764 case X_FINOK:
765 Exit(X_FINOK);
766 case X_ABORT:
767 Exit(X_ABORT);
768 case X_REWRITE:
769 goto restore_check_point;
770 default:
771 msg("Bad return code from dump: %d\n", status);
772 Exit(X_ABORT);
773 }
774 /*NOTREACHED*/
775 } else { /* we are the child; just continue */
776 #ifdef TDEBUG
777 sleep(4); /* allow time for parent's message to get out */
778 msg("Child on Tape %d has parent %d, my pid = %d\n",
779 tapeno+1, parentpid, getpid());
780 #endif /* TDEBUG */
781 /*
782 * If we have a name like "/dev/rmt0,/dev/rmt1",
783 * use the name before the comma first, and save
784 * the remaining names for subsequent volumes.
785 */
786 tapeno++; /* current tape sequence */
787 if (Mflag) {
788 snprintf(tape, MAXPATHLEN, "%s%03d", tapeprefix, tapeno);
789 tape[MAXPATHLEN - 1] = '\0';
790 msg("Dumping volume %d on %s\n", tapeno, tape);
791 }
792 else if (nexttape || strchr(tapeprefix, ',')) {
793 if (nexttape && *nexttape)
794 tapeprefix = nexttape;
795 if ((p = strchr(tapeprefix, ',')) != NULL) {
796 *p = '\0';
797 nexttape = p + 1;
798 } else
799 nexttape = NULL;
800 strncpy(tape, tapeprefix, MAXPATHLEN);
801 tape[MAXPATHLEN - 1] = '\0';
802 msg("Dumping volume %d on %s\n", tapeno, tape);
803 }
804 #ifdef RDUMP
805 while ((tapefd = (host ? rmtopen(tape, 2) : pipeout ?
806 fileno(stdout) :
807 open(tape, O_WRONLY|O_CREAT, 0666))) < 0)
808 #else
809 while ((tapefd = (pipeout ? fileno(stdout) :
810 open(tape, O_WRONLY|O_CREAT, 0666))) < 0)
811 #endif
812 {
813 msg("Cannot open output \"%s\".\n", tape);
814 if (!query("Do you want to retry the open?"))
815 dumpabort(0);
816 }
817
818 enslave(); /* Share open tape file descriptor with slaves */
819
820 asize = 0;
821 blocksthisvol = 0;
822 if (top)
823 newtape++; /* new tape signal */
824 spcl.c_count = slp->count;
825 /*
826 * measure firstrec in TP_BSIZE units since restore doesn't
827 * know the correct ntrec value...
828 */
829 spcl.c_firstrec = slp->firstrec;
830 spcl.c_volume++;
831 spcl.c_type = TS_TAPE;
832 spcl.c_flags |= DR_NEWHEADER;
833 writeheader((ino_t)slp->inode);
834 spcl.c_flags &=~ DR_NEWHEADER;
835 msg("Volume %d started at: %s", tapeno,
836 #ifdef __linux__
837 ctime4(&tstart_volume));
838 #else
839 ctime(&tstart_volume));
840 #endif
841 if (tapeno > 1)
842 msg("Volume %d begins with blocks from inode %d\n",
843 tapeno, slp->inode);
844 }
845 }
846
847 void
848 dumpabort(int signo)
849 {
850
851 if (master != 0 && master != getpid())
852 /* Signals master to call dumpabort */
853 (void) kill(master, SIGTERM);
854 else {
855 killall();
856 msg("The ENTIRE dump is aborted.\n");
857 }
858 #ifdef RDUMP
859 rmtclose();
860 #endif
861 Exit(X_ABORT);
862 }
863
864 void
865 Exit(int status)
866 {
867
868 #ifdef TDEBUG
869 msg("pid = %d exits with status %d\n", getpid(), status);
870 #endif /* TDEBUG */
871 exit(status);
872 }
873
874 /*
875 * proceed - handler for SIGUSR2, used to synchronize IO between the slaves.
876 */
877 static void
878 proceed(int signo)
879 {
880 caught++;
881 if (ready)
882 siglongjmp(jmpbuf, 1);
883 }
884
885 void
886 enslave(void)
887 {
888 int cmd[2];
889 #ifdef LINUX_FORK_BUG
890 int i, j;
891 #else
892 register int i, j;
893 #endif
894
895 master = getpid();
896
897 { struct sigaction sa;
898 sa.sa_handler = dumpabort;
899 sigaction(SIGTERM, &sa, NULL); /* Slave sends SIGTERM on dumpabort() */
900 sa.sa_handler = sigpipe;
901 sigaction(SIGPIPE, &sa, NULL);
902 sa.sa_handler = tperror;
903 sigaction(SIGUSR1, &sa, NULL); /* Slave sends SIGUSR1 on tape errors */
904 sa.sa_handler = proceed;
905 sa.sa_flags = SA_RESTART;
906 sigaction(SIGUSR2, &sa, NULL); /* Slave sends SIGUSR2 to next slave */
907 }
908
909 for (i = 0; i < SLAVES; i++) {
910 if (i == slp - &slaves[0]) {
911 caught = 1;
912 } else {
913 caught = 0;
914 }
915
916 if (socketpair(AF_UNIX, SOCK_STREAM, 0, cmd) < 0 ||
917 (slaves[i].pid = fork()) < 0)
918 quit("too many slaves, %d (recompile smaller): %s\n",
919 i, strerror(errno));
920
921 slaves[i].fd = cmd[1];
922 slaves[i].sent = 0;
923 if (slaves[i].pid == 0) { /* Slave starts up here */
924 sigset_t sigs;
925 for (j = 0; j <= i; j++)
926 (void) close(slaves[j].fd);
927 sigemptyset(&sigs);
928 sigaddset(&sigs, SIGINT); /* Master handles this */
929 #if defined(SIGINFO)
930 sigaddset(&sigs, SIGINFO);
931 #endif
932 sigprocmask(SIG_BLOCK, &sigs, NULL);
933
934 #ifdef LINUX_FORK_BUG
935 if (atomic_write( cmd[0], (char *) &i, sizeof i)
936 != sizeof i)
937 quit("master/slave protocol botched 3\n");
938 #endif
939 doslave(cmd[0], i);
940 Exit(X_FINOK);
941 }
942 }
943
944 #ifdef LINUX_FORK_BUG
945 /*
946 * Wait for all slaves to _actually_ start to circumvent a bug in
947 * Linux kernels >= 2.1.3 where a signal sent to a child that hasn't
948 * returned from fork() causes a SEGV in the child process
949 */
950 for (i = 0; i < SLAVES; i++)
951 if (atomic_read( slaves[i].fd, (char *) &j, sizeof j) != sizeof j)
952 quit("master/slave protocol botched 4\n");
953 #endif
954
955 for (i = 0; i < SLAVES; i++)
956 (void) atomic_write( slaves[i].fd,
957 (char *) &slaves[(i + 1) % SLAVES].pid,
958 sizeof slaves[0].pid);
959
960 master = 0;
961 }
962
963 void
964 killall(void)
965 {
966 register int i;
967
968 for (i = 0; i < SLAVES; i++)
969 if (slaves[i].pid > 0) {
970 (void) kill(slaves[i].pid, SIGKILL);
971 slaves[i].sent = 0;
972 }
973 }
974
975 /*
976 * Synchronization - each process has a lockfile, and shares file
977 * descriptors to the following process's lockfile. When our write
978 * completes, we release our lock on the following process's lock-
979 * file, allowing the following process to lock it and proceed. We
980 * get the lock back for the next cycle by swapping descriptors.
981 */
982 static void
983 doslave(int cmd, int slave_number)
984 {
985 register int nread;
986 int nextslave, size, eot_count;
987 volatile int wrote = 0;
988 sigset_t sigset;
989 #ifdef __linux__
990 errcode_t retval;
991 #endif
992
993 /*
994 * Need our own seek pointer.
995 */
996 (void) close(diskfd);
997 if ((diskfd = open(disk, O_RDONLY)) < 0)
998 quit("slave couldn't reopen disk: %s\n", strerror(errno));
999 #ifdef __linux__
1000 ext2fs_close(fs);
1001 retval = dump_fs_open(disk, &fs);
1002 if (retval)
1003 quit("slave couldn't reopen disk: %s\n", error_message(retval));
1004 #endif /* __linux__ */
1005
1006 /*
1007 * Need the pid of the next slave in the loop...
1008 */
1009 if ((nread = atomic_read( cmd, (char *)&nextslave, sizeof nextslave))
1010 != sizeof nextslave) {
1011 quit("master/slave protocol botched - didn't get pid of next slave.\n");
1012 }
1013
1014 /*
1015 * Get list of blocks to dump, read the blocks into tape buffer
1016 */
1017 while ((nread = atomic_read( cmd, (char *)slp->req, reqsiz)) == reqsiz) {
1018 register struct req *p = slp->req;
1019
1020 for (trecno = 0; trecno < ntrec;
1021 trecno += p->count, p += p->count) {
1022 if (p->dblk) {
1023 bread(p->dblk, slp->tblock[trecno],
1024 p->count * TP_BSIZE);
1025 } else {
1026 if (p->count != 1 || atomic_read( cmd,
1027 (char *)slp->tblock[trecno],
1028 TP_BSIZE) != TP_BSIZE)
1029 quit("master/slave protocol botched.\n");
1030 }
1031 }
1032 if (sigsetjmp(jmpbuf, 1) == 0) {
1033 ready = 1;
1034 if (!caught)
1035 (void) pause();
1036 }
1037 ready = 0;
1038 caught = 0;
1039
1040 /* Try to write the data... */
1041 wrote = 0;
1042 eot_count = 0;
1043 size = 0;
1044
1045 while (eot_count < 10 && size < writesize) {
1046 #ifdef RDUMP
1047 if (host)
1048 wrote = rmtwrite(slp->tblock[0]+size,
1049 writesize-size);
1050 else
1051 #endif
1052 wrote = write(tapefd, slp->tblock[0]+size,
1053 writesize-size);
1054 #ifdef WRITEDEBUG
1055 printf("slave %d wrote %d\n", slave_number, wrote);
1056 #endif
1057 if (wrote < 0)
1058 break;
1059 if (wrote == 0)
1060 eot_count++;
1061 size += wrote;
1062 }
1063
1064 #ifdef WRITEDEBUG
1065 if (size != writesize)
1066 printf("slave %d only wrote %d out of %d bytes and gave up.\n",
1067 slave_number, size, writesize);
1068 #endif
1069
1070 /*
1071 * Handle ENOSPC as an EOT condition.
1072 */
1073 if (wrote < 0 && errno == ENOSPC) {
1074 wrote = 0;
1075 eot_count++;
1076 }
1077
1078 if (eot_count > 0)
1079 size = 0;
1080
1081 if (wrote < 0) {
1082 (void) kill(master, SIGUSR1);
1083 sigemptyset(&sigset);
1084 for (;;)
1085 sigsuspend(&sigset);
1086 } else {
1087 /*
1088 * pass size of write back to master
1089 * (for EOT handling)
1090 */
1091 (void) atomic_write( cmd, (char *)&size, sizeof size);
1092 }
1093
1094 /*
1095 * If partial write, don't want next slave to go.
1096 * Also jolts him awake.
1097 */
1098 (void) kill(nextslave, SIGUSR2);
1099 }
1100 if (nread != 0)
1101 quit("error reading command pipe: %s\n", strerror(errno));
1102 }
1103
1104 /*
1105 * Since a read from a pipe may not return all we asked for,
1106 * or a write may not write all we ask if we get a signal,
1107 * loop until the count is satisfied (or error).
1108 */
1109 static ssize_t
1110 atomic_read(int fd, void *buf, size_t count)
1111 {
1112 int got, need = count;
1113
1114 do {
1115 while ((got = read(fd, buf, need)) > 0 && (need -= got) > 0)
1116 (char *)buf += got;
1117 } while (got == -1 && errno == EINTR);
1118 return (got < 0 ? got : count - need);
1119 }
1120
1121 /*
1122 * Since a read from a pipe may not return all we asked for,
1123 * or a write may not write all we ask if we get a signal,
1124 * loop until the count is satisfied (or error).
1125 */
1126 static ssize_t
1127 atomic_write(int fd, const void *buf, size_t count)
1128 {
1129 int got, need = count;
1130
1131 do {
1132 while ((got = write(fd, buf, need)) > 0 && (need -= got) > 0)
1133 (char *)buf += got;
1134 } while (got == -1 && errno == EINTR);
1135 return (got < 0 ? got : count - need);
1136 }