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