]> Dogcows Code - chaz/tar/blob - src/buffer.c
(dump_file0): The conditional at line
[chaz/tar] / src / buffer.c
1 /* Buffer management for tar.
2
3 Copyright (C) 1988, 1992, 1993, 1994, 1996, 1997, 1999, 2000, 2001,
4 2003, 2004 Free Software Foundation, Inc.
5
6 Written by John Gilmore, on 1985-08-25.
7
8 This program is free software; you can redistribute it and/or modify it
9 under the terms of the GNU General Public License as published by the
10 Free Software Foundation; either version 2, or (at your option) any later
11 version.
12
13 This program is distributed in the hope that it will be useful, but
14 WITHOUT ANY WARRANTY; without even the implied warranty of
15 MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General
16 Public License for more details.
17
18 You should have received a copy of the GNU General Public License along
19 with this program; if not, write to the Free Software Foundation, Inc.,
20 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. */
21
22 #include "system.h"
23
24 #include <signal.h>
25
26 #include <fnmatch.h>
27 #include <human.h>
28 #include <quotearg.h>
29
30 #include "common.h"
31 #include "rmt.h"
32
33 /* Number of retries before giving up on read. */
34 #define READ_ERROR_MAX 10
35
36 /* Globbing pattern to append to volume label if initial match failed. */
37 #define VOLUME_LABEL_APPEND " Volume [1-9]*"
38 \f
39 /* Variables. */
40
41 static tarlong prev_written; /* bytes written on previous volumes */
42 static tarlong bytes_written; /* bytes written on this volume */
43
44 /* FIXME: The following variables should ideally be static to this
45 module. However, this cannot be done yet. The cleanup continues! */
46
47 union block *record_start; /* start of record of archive */
48 union block *record_end; /* last+1 block of archive record */
49 union block *current_block; /* current block of archive */
50 enum access_mode access_mode; /* how do we handle the archive */
51 off_t records_read; /* number of records read from this archive */
52 off_t records_written; /* likewise, for records written */
53
54 static off_t record_start_block; /* block ordinal at record_start */
55
56 /* Where we write list messages (not errors, not interactions) to. */
57 FILE *stdlis;
58
59 static void backspace_output (void);
60 static bool new_volume (enum access_mode);
61
62 /* PID of child program, if compress_option or remote archive access. */
63 static pid_t child_pid;
64
65 /* Error recovery stuff */
66 static int read_error_count;
67
68 /* Have we hit EOF yet? */
69 static int hit_eof;
70
71 /* Checkpointing counter */
72 static int checkpoint;
73
74 /* We're reading, but we just read the last block and it's time to update.
75 Declared in update.c
76
77 As least EXTERN like this one as possible. (?? --gray)
78 FIXME: Either eliminate it or move it to common.h.
79 */
80 extern bool time_to_start_writing;
81
82 static int volno = 1; /* which volume of a multi-volume tape we're
83 on */
84 static int global_volno = 1; /* volume number to print in external
85 messages */
86
87 /* The pointer save_name, which is set in function dump_file() of module
88 create.c, points to the original long filename instead of the new,
89 shorter mangled name that is set in start_header() of module create.c.
90 The pointer save_name is only used in multi-volume mode when the file
91 being processed is non-sparse; if a file is split between volumes, the
92 save_name is used in generating the LF_MULTIVOL record on the second
93 volume. (From Pierce Cantrell, 1991-08-13.) */
94
95 char *save_name; /* name of the file we are currently writing */
96 off_t save_totsize; /* total size of file we are writing, only
97 valid if save_name is nonzero */
98 off_t save_sizeleft; /* where we are in the file we are writing,
99 only valid if save_name is nonzero */
100
101 bool write_archive_to_stdout;
102
103 /* Used by flush_read and flush_write to store the real info about saved
104 names. */
105 static char *real_s_name;
106 static off_t real_s_totsize;
107 static off_t real_s_sizeleft;
108 \f
109 /* Functions. */
110
111 void
112 clear_read_error_count ()
113 {
114 read_error_count = 0;
115 }
116
117 void
118 print_total_written (void)
119 {
120 tarlong written = prev_written + bytes_written;
121 char bytes[sizeof (tarlong) * CHAR_BIT];
122 char abbr[LONGEST_HUMAN_READABLE + 1];
123 char rate[LONGEST_HUMAN_READABLE + 1];
124 double seconds;
125 int human_opts = human_autoscale | human_base_1024 | human_SI | human_B;
126
127 #if HAVE_CLOCK_GETTIME
128 struct timespec now;
129 if (clock_gettime (CLOCK_REALTIME, &now) == 0)
130 seconds = ((now.tv_sec - start_timespec.tv_sec)
131 + (now.tv_nsec - start_timespec.tv_nsec) / 1e9);
132 else
133 #endif
134 seconds = time (0) - start_time;
135
136 sprintf (bytes, TARLONG_FORMAT, written);
137
138 /* Amanda 2.4.1p1 looks for "Total bytes written: [0-9][0-9]*". */
139 fprintf (stderr, _("Total bytes written: %s (%s, %s/s)\n"), bytes,
140 human_readable (written, abbr, human_opts, 1, 1),
141 (0 < seconds && written / seconds < (uintmax_t) -1
142 ? human_readable (written / seconds, rate, human_opts, 1, 1)
143 : "?"));
144 }
145
146 /* Compute and return the block ordinal at current_block. */
147 off_t
148 current_block_ordinal (void)
149 {
150 return record_start_block + (current_block - record_start);
151 }
152
153 /* If the EOF flag is set, reset it, as well as current_block, etc. */
154 void
155 reset_eof (void)
156 {
157 if (hit_eof)
158 {
159 hit_eof = 0;
160 current_block = record_start;
161 record_end = record_start + blocking_factor;
162 access_mode = ACCESS_WRITE;
163 }
164 }
165
166 /* Return the location of the next available input or output block.
167 Return zero for EOF. Once we have returned zero, we just keep returning
168 it, to avoid accidentally going on to the next file on the tape. */
169 union block *
170 find_next_block (void)
171 {
172 if (current_block == record_end)
173 {
174 if (hit_eof)
175 return 0;
176 flush_archive ();
177 if (current_block == record_end)
178 {
179 hit_eof = 1;
180 return 0;
181 }
182 }
183 return current_block;
184 }
185
186 /* Indicate that we have used all blocks up thru BLOCK. */
187 void
188 set_next_block_after (union block *block)
189 {
190 while (block >= current_block)
191 current_block++;
192
193 /* Do *not* flush the archive here. If we do, the same argument to
194 set_next_block_after could mean the next block (if the input record
195 is exactly one block long), which is not what is intended. */
196
197 if (current_block > record_end)
198 abort ();
199 }
200
201 /* Return the number of bytes comprising the space between POINTER
202 through the end of the current buffer of blocks. This space is
203 available for filling with data, or taking data from. POINTER is
204 usually (but not always) the result of previous find_next_block call. */
205 size_t
206 available_space_after (union block *pointer)
207 {
208 return record_end->buffer - pointer->buffer;
209 }
210
211 /* Close file having descriptor FD, and abort if close unsuccessful. */
212 void
213 xclose (int fd)
214 {
215 if (close (fd) != 0)
216 close_error (_("(pipe)"));
217 }
218
219 /* Check the LABEL block against the volume label, seen as a globbing
220 pattern. Return true if the pattern matches. In case of failure,
221 retry matching a volume sequence number before giving up in
222 multi-volume mode. */
223 static bool
224 check_label_pattern (union block *label)
225 {
226 char *string;
227 bool result;
228
229 if (! memchr (label->header.name, '\0', sizeof label->header.name))
230 return false;
231
232 if (fnmatch (volume_label_option, label->header.name, 0) == 0)
233 return true;
234
235 if (!multi_volume_option)
236 return false;
237
238 string = xmalloc (strlen (volume_label_option)
239 + sizeof VOLUME_LABEL_APPEND + 1);
240 strcpy (string, volume_label_option);
241 strcat (string, VOLUME_LABEL_APPEND);
242 result = fnmatch (string, label->header.name, 0) == 0;
243 free (string);
244 return result;
245 }
246
247 /* Open an archive file. The argument specifies whether we are
248 reading or writing, or both. */
249 void
250 open_archive (enum access_mode wanted_access)
251 {
252 int backed_up_flag = 0;
253
254 if (index_file_name)
255 {
256 stdlis = fopen (index_file_name, "w");
257 if (! stdlis)
258 open_error (index_file_name);
259 }
260 else
261 stdlis = to_stdout_option ? stderr : stdout;
262
263 if (record_size == 0)
264 FATAL_ERROR ((0, 0, _("Invalid value for record_size")));
265
266 if (archive_names == 0)
267 FATAL_ERROR ((0, 0, _("No archive name given")));
268
269 tar_stat_destroy (&current_stat_info);
270 save_name = 0;
271 real_s_name = 0;
272
273 if (multi_volume_option)
274 {
275 record_start = valloc (record_size + (2 * BLOCKSIZE));
276 if (record_start)
277 record_start += 2;
278 }
279 else
280 record_start = valloc (record_size);
281 if (!record_start)
282 FATAL_ERROR ((0, 0, _("Cannot allocate memory for blocking factor %d"),
283 blocking_factor));
284
285 current_block = record_start;
286 record_end = record_start + blocking_factor;
287 /* When updating the archive, we start with reading. */
288 access_mode = wanted_access == ACCESS_UPDATE ? ACCESS_READ : wanted_access;
289
290 if (use_compress_program_option)
291 {
292 switch (wanted_access)
293 {
294 case ACCESS_READ:
295 child_pid = sys_child_open_for_uncompress ();
296 break;
297
298 case ACCESS_WRITE:
299 child_pid = sys_child_open_for_compress ();
300 break;
301
302 case ACCESS_UPDATE:
303 abort (); /* Should not happen */
304 break;
305 }
306
307 if (wanted_access == ACCESS_WRITE
308 && strcmp (archive_name_array[0], "-") == 0)
309 stdlis = stderr;
310 }
311 else if (strcmp (archive_name_array[0], "-") == 0)
312 {
313 read_full_records_option = 1; /* could be a pipe, be safe */
314 if (verify_option)
315 FATAL_ERROR ((0, 0, _("Cannot verify stdin/stdout archive")));
316
317 switch (wanted_access)
318 {
319 case ACCESS_READ:
320 archive = STDIN_FILENO;
321 break;
322
323 case ACCESS_WRITE:
324 archive = STDOUT_FILENO;
325 stdlis = stderr;
326 break;
327
328 case ACCESS_UPDATE:
329 archive = STDIN_FILENO;
330 stdlis = stderr;
331 write_archive_to_stdout = true;
332 break;
333 }
334 }
335 else if (verify_option)
336 archive = rmtopen (archive_name_array[0], O_RDWR | O_CREAT | O_BINARY,
337 MODE_RW, rsh_command_option);
338 else
339 switch (wanted_access)
340 {
341 case ACCESS_READ:
342 archive = rmtopen (archive_name_array[0], O_RDONLY | O_BINARY,
343 MODE_RW, rsh_command_option);
344 break;
345
346 case ACCESS_WRITE:
347 if (backup_option)
348 {
349 maybe_backup_file (archive_name_array[0], 1);
350 backed_up_flag = 1;
351 }
352 archive = rmtcreat (archive_name_array[0], MODE_RW,
353 rsh_command_option);
354 break;
355
356 case ACCESS_UPDATE:
357 archive = rmtopen (archive_name_array[0], O_RDWR | O_CREAT | O_BINARY,
358 MODE_RW, rsh_command_option);
359 break;
360 }
361
362 if (archive < 0
363 || (! _isrmt (archive) && !sys_get_archive_stat ()))
364 {
365 int saved_errno = errno;
366
367 if (backed_up_flag)
368 undo_last_backup ();
369 errno = saved_errno;
370 open_fatal (archive_name_array[0]);
371 }
372
373 sys_detect_dev_null_output ();
374 sys_save_archive_dev_ino ();
375 SET_BINARY_MODE (archive);
376
377 switch (wanted_access)
378 {
379 case ACCESS_UPDATE:
380 records_written = 0;
381 case ACCESS_READ:
382 records_read = 0;
383 record_end = record_start; /* set up for 1st record = # 0 */
384 find_next_block (); /* read it in, check for EOF */
385
386 if (volume_label_option)
387 {
388 union block *label = find_next_block ();
389
390 if (!label)
391 FATAL_ERROR ((0, 0, _("Archive not labeled to match %s"),
392 quote (volume_label_option)));
393 if (!check_label_pattern (label))
394 FATAL_ERROR ((0, 0, _("Volume %s does not match %s"),
395 quote_n (0, label->header.name),
396 quote_n (1, volume_label_option)));
397 }
398 break;
399
400 case ACCESS_WRITE:
401 records_written = 0;
402 if (volume_label_option)
403 {
404 memset (record_start, 0, BLOCKSIZE);
405 if (multi_volume_option)
406 sprintf (record_start->header.name, "%s Volume 1",
407 volume_label_option);
408 else
409 strcpy (record_start->header.name, volume_label_option);
410
411 assign_string (&current_stat_info.file_name, record_start->header.name);
412 current_stat_info.had_trailing_slash = strip_trailing_slashes (current_stat_info.file_name);
413
414 record_start->header.typeflag = GNUTYPE_VOLHDR;
415 TIME_TO_CHARS (start_time, record_start->header.mtime);
416 finish_header (&current_stat_info, record_start, -1);
417 }
418 break;
419 }
420 }
421
422 /* Perform a write to flush the buffer. */
423 void
424 flush_write (void)
425 {
426 int copy_back;
427 ssize_t status;
428
429 if (checkpoint_option && !(++checkpoint % 10))
430 WARN ((0, 0, _("Write checkpoint %d"), checkpoint));
431
432 if (tape_length_option && tape_length_option <= bytes_written)
433 {
434 errno = ENOSPC;
435 status = 0;
436 }
437 else if (dev_null_output)
438 status = record_size;
439 else
440 status = sys_write_archive_buffer ();
441 if (status != record_size && !multi_volume_option)
442 archive_write_error (status);
443
444 if (status > 0)
445 {
446 records_written++;
447 bytes_written += status;
448 }
449
450 if (status == record_size)
451 {
452 if (multi_volume_option)
453 {
454 if (save_name)
455 {
456 assign_string (&real_s_name, safer_name_suffix (save_name, false));
457 real_s_totsize = save_totsize;
458 real_s_sizeleft = save_sizeleft;
459 }
460 else
461 {
462 assign_string (&real_s_name, 0);
463 real_s_totsize = 0;
464 real_s_sizeleft = 0;
465 }
466 }
467 return;
468 }
469
470 /* We're multivol. Panic if we didn't get the right kind of response. */
471
472 /* ENXIO is for the UNIX PC. */
473 if (status < 0 && errno != ENOSPC && errno != EIO && errno != ENXIO)
474 archive_write_error (status);
475
476 /* If error indicates a short write, we just move to the next tape. */
477
478 if (!new_volume (ACCESS_WRITE))
479 return;
480
481 if (totals_option)
482 prev_written += bytes_written;
483 bytes_written = 0;
484
485 if (volume_label_option && real_s_name)
486 {
487 copy_back = 2;
488 record_start -= 2;
489 }
490 else if (volume_label_option || real_s_name)
491 {
492 copy_back = 1;
493 record_start--;
494 }
495 else
496 copy_back = 0;
497
498 if (volume_label_option)
499 {
500 memset (record_start, 0, BLOCKSIZE);
501 sprintf (record_start->header.name, "%s Volume %d",
502 volume_label_option, volno);
503 TIME_TO_CHARS (start_time, record_start->header.mtime);
504 record_start->header.typeflag = GNUTYPE_VOLHDR;
505 finish_header (&current_stat_info, record_start, -1);
506 }
507
508 if (real_s_name)
509 {
510 int tmp;
511
512 if (volume_label_option)
513 record_start++;
514
515 memset (record_start, 0, BLOCKSIZE);
516
517 /* FIXME: Michael P Urban writes: [a long name file] is being written
518 when a new volume rolls around [...] Looks like the wrong value is
519 being preserved in real_s_name, though. */
520
521 strcpy (record_start->header.name, real_s_name);
522 record_start->header.typeflag = GNUTYPE_MULTIVOL;
523 OFF_TO_CHARS (real_s_sizeleft, record_start->header.size);
524 OFF_TO_CHARS (real_s_totsize - real_s_sizeleft,
525 record_start->oldgnu_header.offset);
526 tmp = verbose_option;
527 verbose_option = 0;
528 finish_header (&current_stat_info, record_start, -1);
529 verbose_option = tmp;
530
531 if (volume_label_option)
532 record_start--;
533 }
534
535 status = sys_write_archive_buffer ();
536 if (status != record_size)
537 archive_write_error (status);
538
539 bytes_written += status;
540
541 if (copy_back)
542 {
543 record_start += copy_back;
544 memcpy (current_block,
545 record_start + blocking_factor - copy_back,
546 copy_back * BLOCKSIZE);
547 current_block += copy_back;
548
549 if (real_s_sizeleft >= copy_back * BLOCKSIZE)
550 real_s_sizeleft -= copy_back * BLOCKSIZE;
551 else if ((real_s_sizeleft + BLOCKSIZE - 1) / BLOCKSIZE <= copy_back)
552 assign_string (&real_s_name, 0);
553 else
554 {
555 assign_string (&real_s_name, safer_name_suffix (save_name, false));
556 real_s_sizeleft = save_sizeleft;
557 real_s_totsize = save_totsize;
558 }
559 copy_back = 0;
560 }
561 }
562
563 /* Handle write errors on the archive. Write errors are always fatal.
564 Hitting the end of a volume does not cause a write error unless the
565 write was the first record of the volume. */
566 void
567 archive_write_error (ssize_t status)
568 {
569 /* It might be useful to know how much was written before the error
570 occurred. */
571 if (totals_option)
572 {
573 int e = errno;
574 print_total_written ();
575 errno = e;
576 }
577
578 write_fatal_details (*archive_name_cursor, status, record_size);
579 }
580
581 /* Handle read errors on the archive. If the read should be retried,
582 return to the caller. */
583 void
584 archive_read_error (void)
585 {
586 read_error (*archive_name_cursor);
587
588 if (record_start_block == 0)
589 FATAL_ERROR ((0, 0, _("At beginning of tape, quitting now")));
590
591 /* Read error in mid archive. We retry up to READ_ERROR_MAX times and
592 then give up on reading the archive. */
593
594 if (read_error_count++ > READ_ERROR_MAX)
595 FATAL_ERROR ((0, 0, _("Too many errors, quitting")));
596 return;
597 }
598
599 static void
600 short_read (ssize_t status)
601 {
602 size_t left; /* bytes left */
603 char *more; /* pointer to next byte to read */
604
605 more = record_start->buffer + status;
606 left = record_size - status;
607
608 while (left % BLOCKSIZE != 0
609 || (left && status && read_full_records_option))
610 {
611 if (status)
612 while ((status = rmtread (archive, more, left)) < 0)
613 archive_read_error ();
614
615 if (status == 0)
616 {
617 char buf[UINTMAX_STRSIZE_BOUND];
618
619 WARN((0, 0, _("Read %s bytes from %s"),
620 STRINGIFY_BIGINT (record_size - left, buf),
621 *archive_name_cursor));
622 break;
623 }
624
625 if (! read_full_records_option)
626 {
627 unsigned long rest = record_size - left;
628
629 FATAL_ERROR ((0, 0,
630 ngettext ("Unaligned block (%lu byte) in archive",
631 "Unaligned block (%lu bytes) in archive",
632 rest),
633 rest));
634 }
635
636 /* User warned us about this. Fix up. */
637
638 left -= status;
639 more += status;
640 }
641
642 /* FIXME: for size=0, multi-volume support. On the first record, warn
643 about the problem. */
644
645 if (!read_full_records_option && verbose_option
646 && record_start_block == 0 && status > 0)
647 {
648 unsigned long rsize = (record_size - left) / BLOCKSIZE;
649 WARN ((0, 0,
650 ngettext ("Record size = %lu block",
651 "Record size = %lu blocks",
652 rsize),
653 rsize));
654 }
655
656 record_end = record_start + (record_size - left) / BLOCKSIZE;
657 records_read++;
658 }
659
660 /* Perform a read to flush the buffer. */
661 void
662 flush_read (void)
663 {
664 ssize_t status; /* result from system call */
665
666 if (checkpoint_option && !(++checkpoint % 10))
667 WARN ((0, 0, _("Read checkpoint %d"), checkpoint));
668
669 /* Clear the count of errors. This only applies to a single call to
670 flush_read. */
671
672 read_error_count = 0; /* clear error count */
673
674 if (write_archive_to_stdout && record_start_block != 0)
675 {
676 archive = STDOUT_FILENO;
677 status = sys_write_archive_buffer ();
678 archive = STDIN_FILENO;
679 if (status != record_size)
680 archive_write_error (status);
681 }
682 if (multi_volume_option)
683 {
684 if (save_name)
685 {
686 assign_string (&real_s_name, safer_name_suffix (save_name, false));
687 real_s_sizeleft = save_sizeleft;
688 real_s_totsize = save_totsize;
689 }
690 else
691 {
692 assign_string (&real_s_name, 0);
693 real_s_totsize = 0;
694 real_s_sizeleft = 0;
695 }
696 }
697
698 error_loop:
699 status = rmtread (archive, record_start->buffer, record_size);
700 if (status == record_size)
701 {
702 records_read++;
703 return;
704 }
705
706 if ((status == 0
707 || (status < 0 && errno == ENOSPC)
708 || (status > 0 && !read_full_records_option))
709 && multi_volume_option)
710 {
711 union block *cursor;
712
713 try_volume:
714 switch (subcommand_option)
715 {
716 case APPEND_SUBCOMMAND:
717 case CAT_SUBCOMMAND:
718 case UPDATE_SUBCOMMAND:
719 if (!new_volume (ACCESS_UPDATE))
720 return;
721 break;
722
723 default:
724 if (!new_volume (ACCESS_READ))
725 return;
726 break;
727 }
728
729 while ((status =
730 rmtread (archive, record_start->buffer, record_size)) < 0)
731 archive_read_error ();
732
733 if (status != record_size)
734 short_read (status);
735
736 cursor = record_start;
737
738 if (cursor->header.typeflag == GNUTYPE_VOLHDR)
739 {
740 if (volume_label_option)
741 {
742 if (!check_label_pattern (cursor))
743 {
744 WARN ((0, 0, _("Volume %s does not match %s"),
745 quote_n (0, cursor->header.name),
746 quote_n (1, volume_label_option)));
747 volno--;
748 global_volno--;
749 goto try_volume;
750 }
751 }
752 if (verbose_option)
753 fprintf (stdlis, _("Reading %s\n"), quote (cursor->header.name));
754 cursor++;
755 }
756 else if (volume_label_option)
757 WARN ((0, 0, _("WARNING: No volume header")));
758
759 if (real_s_name)
760 {
761 uintmax_t s1, s2;
762 if (cursor->header.typeflag != GNUTYPE_MULTIVOL
763 || strcmp (cursor->header.name, real_s_name))
764 {
765 WARN ((0, 0, _("%s is not continued on this volume"),
766 quote (real_s_name)));
767 volno--;
768 global_volno--;
769 goto try_volume;
770 }
771 s1 = UINTMAX_FROM_HEADER (cursor->header.size);
772 s2 = UINTMAX_FROM_HEADER (cursor->oldgnu_header.offset);
773 if (real_s_totsize != s1 + s2 || s1 + s2 < s2)
774 {
775 char totsizebuf[UINTMAX_STRSIZE_BOUND];
776 char s1buf[UINTMAX_STRSIZE_BOUND];
777 char s2buf[UINTMAX_STRSIZE_BOUND];
778
779 WARN ((0, 0, _("%s is the wrong size (%s != %s + %s)"),
780 quote (cursor->header.name),
781 STRINGIFY_BIGINT (save_totsize, totsizebuf),
782 STRINGIFY_BIGINT (s1, s1buf),
783 STRINGIFY_BIGINT (s2, s2buf)));
784 volno--;
785 global_volno--;
786 goto try_volume;
787 }
788 if (real_s_totsize - real_s_sizeleft
789 != OFF_FROM_HEADER (cursor->oldgnu_header.offset))
790 {
791 WARN ((0, 0, _("This volume is out of sequence")));
792 volno--;
793 global_volno--;
794 goto try_volume;
795 }
796 cursor++;
797 }
798 current_block = cursor;
799 records_read++;
800 return;
801 }
802 else if (status < 0)
803 {
804 archive_read_error ();
805 goto error_loop; /* try again */
806 }
807
808 short_read (status);
809 }
810
811 /* Flush the current buffer to/from the archive. */
812 void
813 flush_archive (void)
814 {
815 record_start_block += record_end - record_start;
816 current_block = record_start;
817 record_end = record_start + blocking_factor;
818
819 if (access_mode == ACCESS_READ && time_to_start_writing)
820 {
821 access_mode = ACCESS_WRITE;
822 time_to_start_writing = false;
823 backspace_output ();
824 }
825
826 switch (access_mode)
827 {
828 case ACCESS_READ:
829 flush_read ();
830 break;
831
832 case ACCESS_WRITE:
833 flush_write ();
834 break;
835
836 case ACCESS_UPDATE:
837 abort ();
838 }
839 }
840
841 /* Backspace the archive descriptor by one record worth. If it's a
842 tape, MTIOCTOP will work. If it's something else, try to seek on
843 it. If we can't seek, we lose! */
844 static void
845 backspace_output (void)
846 {
847 #ifdef MTIOCTOP
848 {
849 struct mtop operation;
850
851 operation.mt_op = MTBSR;
852 operation.mt_count = 1;
853 if (rmtioctl (archive, MTIOCTOP, (char *) &operation) >= 0)
854 return;
855 if (errno == EIO && rmtioctl (archive, MTIOCTOP, (char *) &operation) >= 0)
856 return;
857 }
858 #endif
859
860 {
861 off_t position = rmtlseek (archive, (off_t) 0, SEEK_CUR);
862
863 /* Seek back to the beginning of this record and start writing there. */
864
865 position -= record_size;
866 if (position < 0)
867 position = 0;
868 if (rmtlseek (archive, position, SEEK_SET) != position)
869 {
870 /* Lseek failed. Try a different method. */
871
872 WARN ((0, 0,
873 _("Cannot backspace archive file; it may be unreadable without -i")));
874
875 /* Replace the first part of the record with NULs. */
876
877 if (record_start->buffer != output_start)
878 memset (record_start->buffer, 0,
879 output_start - record_start->buffer);
880 }
881 }
882 }
883
884 /* Close the archive file. */
885 void
886 close_archive (void)
887 {
888 if (time_to_start_writing || access_mode == ACCESS_WRITE)
889 flush_archive ();
890
891 sys_drain_input_pipe ();
892
893 if (verify_option)
894 verify_volume ();
895
896 if (rmtclose (archive) != 0)
897 close_warn (*archive_name_cursor);
898
899 sys_wait_for_child (child_pid);
900
901 tar_stat_destroy (&current_stat_info);
902 if (save_name)
903 free (save_name);
904 if (real_s_name)
905 free (real_s_name);
906 free (multi_volume_option ? record_start - 2 : record_start);
907 }
908
909 /* Called to initialize the global volume number. */
910 void
911 init_volume_number (void)
912 {
913 FILE *file = fopen (volno_file_option, "r");
914
915 if (file)
916 {
917 if (fscanf (file, "%d", &global_volno) != 1
918 || global_volno < 0)
919 FATAL_ERROR ((0, 0, _("%s: contains invalid volume number"),
920 quotearg_colon (volno_file_option)));
921 if (ferror (file))
922 read_error (volno_file_option);
923 if (fclose (file) != 0)
924 close_error (volno_file_option);
925 }
926 else if (errno != ENOENT)
927 open_error (volno_file_option);
928 }
929
930 /* Called to write out the closing global volume number. */
931 void
932 closeout_volume_number (void)
933 {
934 FILE *file = fopen (volno_file_option, "w");
935
936 if (file)
937 {
938 fprintf (file, "%d\n", global_volno);
939 if (ferror (file))
940 write_error (volno_file_option);
941 if (fclose (file) != 0)
942 close_error (volno_file_option);
943 }
944 else
945 open_error (volno_file_option);
946 }
947
948 /* We've hit the end of the old volume. Close it and open the next one.
949 Return nonzero on success.
950 */
951 static bool
952 new_volume (enum access_mode access)
953 {
954 static FILE *read_file;
955 static int looped;
956
957 if (!read_file && !info_script_option)
958 /* FIXME: if fopen is used, it will never be closed. */
959 read_file = archive == STDIN_FILENO ? fopen (TTY_NAME, "r") : stdin;
960
961 if (now_verifying)
962 return false;
963 if (verify_option)
964 verify_volume ();
965
966 if (rmtclose (archive) != 0)
967 close_warn (*archive_name_cursor);
968
969 global_volno++;
970 if (global_volno < 0)
971 FATAL_ERROR ((0, 0, _("Volume number overflow")));
972 volno++;
973 archive_name_cursor++;
974 if (archive_name_cursor == archive_name_array + archive_names)
975 {
976 archive_name_cursor = archive_name_array;
977 looped = 1;
978 }
979
980 tryagain:
981 if (looped)
982 {
983 /* We have to prompt from now on. */
984
985 if (info_script_option)
986 {
987 if (volno_file_option)
988 closeout_volume_number ();
989 if (system (info_script_option) != 0)
990 FATAL_ERROR ((0, 0, _("`%s' command failed"), info_script_option));
991 }
992 else
993 while (1)
994 {
995 char input_buffer[80];
996
997 fputc ('\007', stderr);
998 fprintf (stderr,
999 _("Prepare volume #%d for %s and hit return: "),
1000 global_volno, quote (*archive_name_cursor));
1001 fflush (stderr);
1002
1003 if (fgets (input_buffer, sizeof input_buffer, read_file) == 0)
1004 {
1005 WARN ((0, 0, _("EOF where user reply was expected")));
1006
1007 if (subcommand_option != EXTRACT_SUBCOMMAND
1008 && subcommand_option != LIST_SUBCOMMAND
1009 && subcommand_option != DIFF_SUBCOMMAND)
1010 WARN ((0, 0, _("WARNING: Archive is incomplete")));
1011
1012 fatal_exit ();
1013 }
1014 if (input_buffer[0] == '\n'
1015 || input_buffer[0] == 'y'
1016 || input_buffer[0] == 'Y')
1017 break;
1018
1019 switch (input_buffer[0])
1020 {
1021 case '?':
1022 {
1023 /* FIXME: Might it be useful to disable the '!' command? */
1024 fprintf (stderr, _("\
1025 n [name] Give a new file name for the next (and subsequent) volume(s)\n\
1026 q Abort tar\n\
1027 ! Spawn a subshell\n\
1028 ? Print this list\n"));
1029 }
1030 break;
1031
1032 case 'q':
1033 /* Quit. */
1034
1035 WARN ((0, 0, _("No new volume; exiting.\n")));
1036
1037 if (subcommand_option != EXTRACT_SUBCOMMAND
1038 && subcommand_option != LIST_SUBCOMMAND
1039 && subcommand_option != DIFF_SUBCOMMAND)
1040 WARN ((0, 0, _("WARNING: Archive is incomplete")));
1041
1042 fatal_exit ();
1043
1044 case 'n':
1045 /* Get new file name. */
1046
1047 {
1048 char *name = &input_buffer[1];
1049 char *cursor;
1050
1051 for (name = input_buffer + 1;
1052 *name == ' ' || *name == '\t';
1053 name++)
1054 ;
1055
1056 for (cursor = name; *cursor && *cursor != '\n'; cursor++)
1057 ;
1058 *cursor = '\0';
1059
1060 /* FIXME: the following allocation is never reclaimed. */
1061 *archive_name_cursor = xstrdup (name);
1062 }
1063 break;
1064
1065 case '!':
1066 sys_spawn_shell ();
1067 break;
1068 }
1069 }
1070 }
1071
1072 if (strcmp (archive_name_cursor[0], "-") == 0)
1073 {
1074 read_full_records_option = true;
1075 archive = STDIN_FILENO;
1076 }
1077 else if (verify_option)
1078 archive = rmtopen (*archive_name_cursor, O_RDWR | O_CREAT, MODE_RW,
1079 rsh_command_option);
1080 else
1081 switch (access)
1082 {
1083 case ACCESS_READ:
1084 archive = rmtopen (*archive_name_cursor, O_RDONLY, MODE_RW,
1085 rsh_command_option);
1086 break;
1087
1088 case ACCESS_WRITE:
1089 if (backup_option)
1090 maybe_backup_file (*archive_name_cursor, 1);
1091 archive = rmtcreat (*archive_name_cursor, MODE_RW,
1092 rsh_command_option);
1093 break;
1094
1095 case ACCESS_UPDATE:
1096 archive = rmtopen (*archive_name_cursor, O_RDWR | O_CREAT, MODE_RW,
1097 rsh_command_option);
1098 break;
1099 }
1100
1101 if (archive < 0)
1102 {
1103 open_warn (*archive_name_cursor);
1104 if (!verify_option && access == ACCESS_WRITE && backup_option)
1105 undo_last_backup ();
1106 goto tryagain;
1107 }
1108
1109 SET_BINARY_MODE (archive);
1110
1111 return true;
1112 }
1113
This page took 0.090028 seconds and 4 git commands to generate.