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