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