/* Buffer management for tar. Copyright (C) 1988, 1992, 1993, 1994, 1996, 1997, 1999, 2000, 2001, 2003, 2004 Free Software Foundation, Inc. Written by John Gilmore, on 1985-08-25. This program is free software; you can redistribute it and/or modify it under the terms of the GNU General Public License as published by the Free Software Foundation; either version 2, or (at your option) any later version. This program is distributed in the hope that it will be useful, but WITHOUT ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License for more details. You should have received a copy of the GNU General Public License along with this program; if not, write to the Free Software Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111-1307, USA. */ #include #include #include #include #include #include "common.h" #include /* Number of retries before giving up on read. */ #define READ_ERROR_MAX 10 /* Globbing pattern to append to volume label if initial match failed. */ #define VOLUME_LABEL_APPEND " Volume [1-9]*" /* Variables. */ static tarlong prev_written; /* bytes written on previous volumes */ static tarlong bytes_written; /* bytes written on this volume */ static void *record_buffer; /* allocated memory */ /* FIXME: The following variables should ideally be static to this module. However, this cannot be done yet. The cleanup continues! */ union block *record_start; /* start of record of archive */ union block *record_end; /* last+1 block of archive record */ union block *current_block; /* current block of archive */ enum access_mode access_mode; /* how do we handle the archive */ off_t records_read; /* number of records read from this archive */ off_t records_written; /* likewise, for records written */ static off_t record_start_block; /* block ordinal at record_start */ /* Where we write list messages (not errors, not interactions) to. */ FILE *stdlis; static void backspace_output (void); static bool new_volume (enum access_mode); /* PID of child program, if compress_option or remote archive access. */ static pid_t child_pid; /* Error recovery stuff */ static int read_error_count; /* Have we hit EOF yet? */ static bool hit_eof; /* Checkpointing counter */ static int checkpoint; static bool read_full_records = false; static bool reading_from_pipe = false; /* We're reading, but we just read the last block and it's time to update. Declared in update.c As least EXTERN like this one as possible. (?? --gray) FIXME: Either eliminate it or move it to common.h. */ extern bool time_to_start_writing; static int volno = 1; /* which volume of a multi-volume tape we're on */ static int global_volno = 1; /* volume number to print in external messages */ /* The pointer save_name, which is set in function dump_file() of module create.c, points to the original long filename instead of the new, shorter mangled name that is set in start_header() of module create.c. The pointer save_name is only used in multi-volume mode when the file being processed is non-sparse; if a file is split between volumes, the save_name is used in generating the LF_MULTIVOL record on the second volume. (From Pierce Cantrell, 1991-08-13.) */ char *save_name; /* name of the file we are currently writing */ off_t save_totsize; /* total size of file we are writing, only valid if save_name is nonzero */ off_t save_sizeleft; /* where we are in the file we are writing, only valid if save_name is nonzero */ bool write_archive_to_stdout; /* Used by flush_read and flush_write to store the real info about saved names. */ static char *real_s_name; static off_t real_s_totsize; static off_t real_s_sizeleft; /* Functions. */ void clear_read_error_count (void) { read_error_count = 0; } /* Time-related functions */ double duration; void set_start_time () { #if HAVE_CLOCK_GETTIME if (clock_gettime (CLOCK_REALTIME, &start_timespec) != 0) #endif start_time = time (0); } void compute_duration () { #if HAVE_CLOCK_GETTIME struct timespec now; if (clock_gettime (CLOCK_REALTIME, &now) == 0) duration += ((now.tv_sec - start_timespec.tv_sec) + (now.tv_nsec - start_timespec.tv_nsec) / 1e9); else #endif duration += time (NULL) - start_time; set_start_time (); } /* Compression detection */ enum compress_type { ct_none, ct_compress, ct_gzip, ct_bzip2 }; struct zip_magic { enum compress_type type; unsigned char *magic; size_t length; char *program; char *option; }; static struct zip_magic magic[] = { { ct_none, }, { ct_compress, "\037\235", 2, "compress", "-Z" }, { ct_gzip, "\037\213", 2, "gzip", "-z" }, { ct_bzip2, "BZh", 3, "bzip2", "-j" }, }; #define NMAGIC (sizeof(magic)/sizeof(magic[0])) #define compress_option(t) magic[t].option #define compress_program(t) magic[t].program /* Check if the file ARCHIVE is a compressed archive. */ enum compress_type check_compressed_archive () { struct zip_magic *p; size_t status; bool sfr, srp; /* Prepare global data needed for find_next_block: */ record_end = record_start; /* set up for 1st record = # 0 */ sfr = read_full_records; read_full_records = true; /* Suppress fatal error on reading a partial record */ srp = reading_from_pipe; reading_from_pipe = true; /* Suppress warning message on reading a partial record */ find_next_block (); /* Restore global values */ read_full_records = sfr; reading_from_pipe = srp; if (tar_checksum (record_start, true) == HEADER_SUCCESS) /* Probably a valid header */ return ct_none; for (p = magic + 1; p < magic + NMAGIC; p++) if (memcmp (record_start->buffer, p->magic, p->length) == 0) { hit_eof = false; /* It might have been set by find_next_block */ return p->type; } return ct_none; } /* Open an archive named archive_name_array[0]. Detect if it is a compressed archive of known type and use corresponding decompression program if so */ int open_compressed_archive () { enum compress_type type; archive = rmtopen (archive_name_array[0], O_RDONLY | O_BINARY, MODE_RW, rsh_command_option); if (archive == -1) return archive; type = check_compressed_archive (); if (type == ct_none) return archive; /* FD is not needed any more */ rmtclose (archive); /* Open compressed archive */ use_compress_program_option = compress_program (type); child_pid = sys_child_open_for_uncompress (); read_full_records = reading_from_pipe = true; records_read = 0; record_end = record_start; /* set up for 1st record = # 0 */ return archive; } void print_total_written (void) { tarlong written = prev_written + bytes_written; char bytes[sizeof (tarlong) * CHAR_BIT]; char abbr[LONGEST_HUMAN_READABLE + 1]; char rate[LONGEST_HUMAN_READABLE + 1]; int human_opts = human_autoscale | human_base_1024 | human_SI | human_B; sprintf (bytes, TARLONG_FORMAT, written); /* Amanda 2.4.1p1 looks for "Total bytes written: [0-9][0-9]*". */ fprintf (stderr, _("Total bytes written: %s (%s, %s/s)\n"), bytes, human_readable (written, abbr, human_opts, 1, 1), (0 < duration && written / duration < (uintmax_t) -1 ? human_readable (written / duration, rate, human_opts, 1, 1) : "?")); } /* Compute and return the block ordinal at current_block. */ off_t current_block_ordinal (void) { return record_start_block + (current_block - record_start); } /* If the EOF flag is set, reset it, as well as current_block, etc. */ void reset_eof (void) { if (hit_eof) { hit_eof = false; current_block = record_start; record_end = record_start + blocking_factor; access_mode = ACCESS_WRITE; } } /* Return the location of the next available input or output block. Return zero for EOF. Once we have returned zero, we just keep returning it, to avoid accidentally going on to the next file on the tape. */ union block * find_next_block (void) { if (current_block == record_end) { if (hit_eof) return 0; flush_archive (); if (current_block == record_end) { hit_eof = true; return 0; } } return current_block; } /* Indicate that we have used all blocks up thru BLOCK. */ void set_next_block_after (union block *block) { while (block >= current_block) current_block++; /* Do *not* flush the archive here. If we do, the same argument to set_next_block_after could mean the next block (if the input record is exactly one block long), which is not what is intended. */ if (current_block > record_end) abort (); } /* Return the number of bytes comprising the space between POINTER through the end of the current buffer of blocks. This space is available for filling with data, or taking data from. POINTER is usually (but not always) the result of previous find_next_block call. */ size_t available_space_after (union block *pointer) { return record_end->buffer - pointer->buffer; } /* Close file having descriptor FD, and abort if close unsuccessful. */ void xclose (int fd) { if (close (fd) != 0) close_error (_("(pipe)")); } /* Check the LABEL block against the volume label, seen as a globbing pattern. Return true if the pattern matches. In case of failure, retry matching a volume sequence number before giving up in multi-volume mode. */ static bool check_label_pattern (union block *label) { char *string; bool result; if (! memchr (label->header.name, '\0', sizeof label->header.name)) return false; if (fnmatch (volume_label_option, label->header.name, 0) == 0) return true; if (!multi_volume_option) return false; string = xmalloc (strlen (volume_label_option) + sizeof VOLUME_LABEL_APPEND + 1); strcpy (string, volume_label_option); strcat (string, VOLUME_LABEL_APPEND); result = fnmatch (string, label->header.name, 0) == 0; free (string); return result; } /* Open an archive file. The argument specifies whether we are reading or writing, or both. */ void open_archive (enum access_mode wanted_access) { int backed_up_flag = 0; if (index_file_name) { stdlis = fopen (index_file_name, "w"); if (! stdlis) open_error (index_file_name); } else stdlis = to_stdout_option ? stderr : stdout; if (record_size == 0) FATAL_ERROR ((0, 0, _("Invalid value for record_size"))); if (archive_names == 0) FATAL_ERROR ((0, 0, _("No archive name given"))); tar_stat_destroy (¤t_stat_info); save_name = 0; real_s_name = 0; record_start = page_aligned_alloc (&record_buffer, (record_size + (multi_volume_option ? 2 * BLOCKSIZE : 0))); if (multi_volume_option) record_start += 2; current_block = record_start; record_end = record_start + blocking_factor; /* When updating the archive, we start with reading. */ access_mode = wanted_access == ACCESS_UPDATE ? ACCESS_READ : wanted_access; read_full_records = read_full_records_option; reading_from_pipe = false; records_read = 0; if (use_compress_program_option) { switch (wanted_access) { case ACCESS_READ: child_pid = sys_child_open_for_uncompress (); read_full_records = reading_from_pipe = true; record_end = record_start; /* set up for 1st record = # 0 */ break; case ACCESS_WRITE: child_pid = sys_child_open_for_compress (); break; case ACCESS_UPDATE: abort (); /* Should not happen */ break; } if (wanted_access == ACCESS_WRITE && strcmp (archive_name_array[0], "-") == 0) stdlis = stderr; } else if (strcmp (archive_name_array[0], "-") == 0) { read_full_records = true; /* could be a pipe, be safe */ if (verify_option) FATAL_ERROR ((0, 0, _("Cannot verify stdin/stdout archive"))); switch (wanted_access) { case ACCESS_READ: { enum compress_type type; archive = STDIN_FILENO; type = check_compressed_archive (archive); if (type != ct_none) FATAL_ERROR ((0, 0, _("Archive is compressed. Use %s option"), compress_option (type))); } break; case ACCESS_WRITE: archive = STDOUT_FILENO; stdlis = stderr; break; case ACCESS_UPDATE: archive = STDIN_FILENO; stdlis = stderr; write_archive_to_stdout = true; break; } } else if (verify_option) archive = rmtopen (archive_name_array[0], O_RDWR | O_CREAT | O_BINARY, MODE_RW, rsh_command_option); else switch (wanted_access) { case ACCESS_READ: archive = open_compressed_archive (); break; case ACCESS_WRITE: if (backup_option) { maybe_backup_file (archive_name_array[0], 1); backed_up_flag = 1; } archive = rmtcreat (archive_name_array[0], MODE_RW, rsh_command_option); break; case ACCESS_UPDATE: archive = rmtopen (archive_name_array[0], O_RDWR | O_CREAT | O_BINARY, MODE_RW, rsh_command_option); break; } if (archive < 0 || (! _isrmt (archive) && !sys_get_archive_stat ())) { int saved_errno = errno; if (backed_up_flag) undo_last_backup (); errno = saved_errno; open_fatal (archive_name_array[0]); } sys_detect_dev_null_output (); sys_save_archive_dev_ino (); SET_BINARY_MODE (archive); switch (wanted_access) { case ACCESS_UPDATE: records_written = 0; record_end = record_start; /* set up for 1st record = # 0 */ case ACCESS_READ: find_next_block (); /* read it in, check for EOF */ if (volume_label_option) { union block *label = find_next_block (); if (!label) FATAL_ERROR ((0, 0, _("Archive not labeled to match %s"), quote (volume_label_option))); if (!check_label_pattern (label)) FATAL_ERROR ((0, 0, _("Volume %s does not match %s"), quote_n (0, label->header.name), quote_n (1, volume_label_option))); } break; case ACCESS_WRITE: records_written = 0; if (volume_label_option) { memset (record_start, 0, BLOCKSIZE); if (multi_volume_option) sprintf (record_start->header.name, "%s Volume 1", volume_label_option); else strcpy (record_start->header.name, volume_label_option); assign_string (¤t_stat_info.file_name, record_start->header.name); current_stat_info.had_trailing_slash = strip_trailing_slashes (current_stat_info.file_name); record_start->header.typeflag = GNUTYPE_VOLHDR; TIME_TO_CHARS (start_time, record_start->header.mtime); finish_header (¤t_stat_info, record_start, -1); } break; } } /* Perform a write to flush the buffer. */ void flush_write (void) { int copy_back; ssize_t status; if (checkpoint_option && !(++checkpoint % 10)) WARN ((0, 0, _("Write checkpoint %d"), checkpoint)); if (tape_length_option && tape_length_option <= bytes_written) { errno = ENOSPC; status = 0; } else if (dev_null_output) status = record_size; else status = sys_write_archive_buffer (); if (status != record_size && !multi_volume_option) archive_write_error (status); if (status > 0) { records_written++; bytes_written += status; } if (status == record_size) { if (multi_volume_option) { if (save_name) { assign_string (&real_s_name, safer_name_suffix (save_name, false)); real_s_totsize = save_totsize; real_s_sizeleft = save_sizeleft; } else { assign_string (&real_s_name, 0); real_s_totsize = 0; real_s_sizeleft = 0; } } return; } /* We're multivol. Panic if we didn't get the right kind of response. */ /* ENXIO is for the UNIX PC. */ if (status < 0 && errno != ENOSPC && errno != EIO && errno != ENXIO) archive_write_error (status); /* If error indicates a short write, we just move to the next tape. */ if (!new_volume (ACCESS_WRITE)) return; if (totals_option) prev_written += bytes_written; bytes_written = 0; if (volume_label_option && real_s_name) { copy_back = 2; record_start -= 2; } else if (volume_label_option || real_s_name) { copy_back = 1; record_start--; } else copy_back = 0; if (volume_label_option) { memset (record_start, 0, BLOCKSIZE); sprintf (record_start->header.name, "%s Volume %d", volume_label_option, volno); TIME_TO_CHARS (start_time, record_start->header.mtime); record_start->header.typeflag = GNUTYPE_VOLHDR; finish_header (¤t_stat_info, record_start, -1); } if (real_s_name) { int tmp; if (volume_label_option) record_start++; if (strlen (real_s_name) > NAME_FIELD_SIZE) FATAL_ERROR ((0, 0, _("%s: file name too long to be stored in a GNU multivolume header"), quotearg_colon (real_s_name))); memset (record_start, 0, BLOCKSIZE); /* FIXME: Michael P Urban writes: [a long name file] is being written when a new volume rolls around [...] Looks like the wrong value is being preserved in real_s_name, though. */ strncpy (record_start->header.name, real_s_name, NAME_FIELD_SIZE); record_start->header.typeflag = GNUTYPE_MULTIVOL; OFF_TO_CHARS (real_s_sizeleft, record_start->header.size); OFF_TO_CHARS (real_s_totsize - real_s_sizeleft, record_start->oldgnu_header.offset); tmp = verbose_option; verbose_option = 0; finish_header (¤t_stat_info, record_start, -1); verbose_option = tmp; if (volume_label_option) record_start--; } status = sys_write_archive_buffer (); if (status != record_size) archive_write_error (status); bytes_written += status; if (copy_back) { record_start += copy_back; memcpy (current_block, record_start + blocking_factor - copy_back, copy_back * BLOCKSIZE); current_block += copy_back; if (real_s_sizeleft >= copy_back * BLOCKSIZE) real_s_sizeleft -= copy_back * BLOCKSIZE; else if ((real_s_sizeleft + BLOCKSIZE - 1) / BLOCKSIZE <= copy_back) assign_string (&real_s_name, 0); else { assign_string (&real_s_name, safer_name_suffix (save_name, false)); real_s_sizeleft = save_sizeleft; real_s_totsize = save_totsize; } copy_back = 0; } } /* Handle write errors on the archive. Write errors are always fatal. Hitting the end of a volume does not cause a write error unless the write was the first record of the volume. */ void archive_write_error (ssize_t status) { /* It might be useful to know how much was written before the error occurred. */ if (totals_option) { int e = errno; print_total_written (); errno = e; } write_fatal_details (*archive_name_cursor, status, record_size); } /* Handle read errors on the archive. If the read should be retried, return to the caller. */ void archive_read_error (void) { read_error (*archive_name_cursor); if (record_start_block == 0) FATAL_ERROR ((0, 0, _("At beginning of tape, quitting now"))); /* Read error in mid archive. We retry up to READ_ERROR_MAX times and then give up on reading the archive. */ if (read_error_count++ > READ_ERROR_MAX) FATAL_ERROR ((0, 0, _("Too many errors, quitting"))); return; } static void short_read (size_t status) { size_t left; /* bytes left */ char *more; /* pointer to next byte to read */ more = record_start->buffer + status; left = record_size - status; while (left % BLOCKSIZE != 0 || (left && status && read_full_records)) { if (status) while ((status = rmtread (archive, more, left)) == SAFE_READ_ERROR) archive_read_error (); if (status == 0) { if (!reading_from_pipe) { char buf[UINTMAX_STRSIZE_BOUND]; WARN((0, 0, _("Read %s bytes from %s"), STRINGIFY_BIGINT (record_size - left, buf), *archive_name_cursor)); } break; } if (! read_full_records) { unsigned long rest = record_size - left; FATAL_ERROR ((0, 0, ngettext ("Unaligned block (%lu byte) in archive", "Unaligned block (%lu bytes) in archive", rest), rest)); } /* User warned us about this. Fix up. */ left -= status; more += status; } /* FIXME: for size=0, multi-volume support. On the first record, warn about the problem. */ if (!read_full_records && verbose_option > 1 && record_start_block == 0 && status != 0) { unsigned long rsize = (record_size - left) / BLOCKSIZE; WARN ((0, 0, ngettext ("Record size = %lu block", "Record size = %lu blocks", rsize), rsize)); } record_end = record_start + (record_size - left) / BLOCKSIZE; records_read++; } /* Perform a read to flush the buffer. */ void flush_read (void) { size_t status; /* result from system call */ if (checkpoint_option && !(++checkpoint % 10)) WARN ((0, 0, _("Read checkpoint %d"), checkpoint)); /* Clear the count of errors. This only applies to a single call to flush_read. */ read_error_count = 0; /* clear error count */ if (write_archive_to_stdout && record_start_block != 0) { archive = STDOUT_FILENO; status = sys_write_archive_buffer (); archive = STDIN_FILENO; if (status != record_size) archive_write_error (status); } if (multi_volume_option) { if (save_name) { assign_string (&real_s_name, safer_name_suffix (save_name, false)); real_s_sizeleft = save_sizeleft; real_s_totsize = save_totsize; } else { assign_string (&real_s_name, 0); real_s_totsize = 0; real_s_sizeleft = 0; } } error_loop: status = rmtread (archive, record_start->buffer, record_size); if (status == record_size) { records_read++; return; } /* The condition below used to include || (status > 0 && !read_full_records) This is incorrect since even if new_volume() succeeds, the subsequent call to rmtread will overwrite the chunk of data already read in the buffer, so the processing will fail */ if ((status == 0 || (status == SAFE_READ_ERROR && errno == ENOSPC)) && multi_volume_option) { union block *cursor; try_volume: switch (subcommand_option) { case APPEND_SUBCOMMAND: case CAT_SUBCOMMAND: case UPDATE_SUBCOMMAND: if (!new_volume (ACCESS_UPDATE)) return; break; default: if (!new_volume (ACCESS_READ)) return; break; } while ((status = rmtread (archive, record_start->buffer, record_size)) == SAFE_READ_ERROR) archive_read_error (); if (status != record_size) short_read (status); cursor = record_start; if (cursor->header.typeflag == GNUTYPE_VOLHDR) { if (volume_label_option) { if (!check_label_pattern (cursor)) { WARN ((0, 0, _("Volume %s does not match %s"), quote_n (0, cursor->header.name), quote_n (1, volume_label_option))); volno--; global_volno--; goto try_volume; } } if (verbose_option) fprintf (stdlis, _("Reading %s\n"), quote (cursor->header.name)); cursor++; } else if (volume_label_option) WARN ((0, 0, _("WARNING: No volume header"))); if (real_s_name) { uintmax_t s1, s2; if (cursor->header.typeflag != GNUTYPE_MULTIVOL || strncmp (cursor->header.name, real_s_name, NAME_FIELD_SIZE)) { WARN ((0, 0, _("%s is not continued on this volume"), quote (real_s_name))); volno--; global_volno--; goto try_volume; } s1 = UINTMAX_FROM_HEADER (cursor->header.size); s2 = UINTMAX_FROM_HEADER (cursor->oldgnu_header.offset); if (real_s_totsize != s1 + s2 || s1 + s2 < s2) { char totsizebuf[UINTMAX_STRSIZE_BOUND]; char s1buf[UINTMAX_STRSIZE_BOUND]; char s2buf[UINTMAX_STRSIZE_BOUND]; WARN ((0, 0, _("%s is the wrong size (%s != %s + %s)"), quote (cursor->header.name), STRINGIFY_BIGINT (save_totsize, totsizebuf), STRINGIFY_BIGINT (s1, s1buf), STRINGIFY_BIGINT (s2, s2buf))); volno--; global_volno--; goto try_volume; } if (real_s_totsize - real_s_sizeleft != OFF_FROM_HEADER (cursor->oldgnu_header.offset)) { WARN ((0, 0, _("This volume is out of sequence"))); volno--; global_volno--; goto try_volume; } cursor++; } current_block = cursor; records_read++; return; } else if (status == SAFE_READ_ERROR) { archive_read_error (); goto error_loop; /* try again */ } short_read (status); } /* Flush the current buffer to/from the archive. */ void flush_archive (void) { record_start_block += record_end - record_start; current_block = record_start; record_end = record_start + blocking_factor; if (access_mode == ACCESS_READ && time_to_start_writing) { access_mode = ACCESS_WRITE; time_to_start_writing = false; backspace_output (); } switch (access_mode) { case ACCESS_READ: flush_read (); break; case ACCESS_WRITE: flush_write (); break; case ACCESS_UPDATE: abort (); } } /* Backspace the archive descriptor by one record worth. If it's a tape, MTIOCTOP will work. If it's something else, try to seek on it. If we can't seek, we lose! */ static void backspace_output (void) { #ifdef MTIOCTOP { struct mtop operation; operation.mt_op = MTBSR; operation.mt_count = 1; if (rmtioctl (archive, MTIOCTOP, (char *) &operation) >= 0) return; if (errno == EIO && rmtioctl (archive, MTIOCTOP, (char *) &operation) >= 0) return; } #endif { off_t position = rmtlseek (archive, (off_t) 0, SEEK_CUR); /* Seek back to the beginning of this record and start writing there. */ position -= record_size; if (position < 0) position = 0; if (rmtlseek (archive, position, SEEK_SET) != position) { /* Lseek failed. Try a different method. */ WARN ((0, 0, _("Cannot backspace archive file; it may be unreadable without -i"))); /* Replace the first part of the record with NULs. */ if (record_start->buffer != output_start) memset (record_start->buffer, 0, output_start - record_start->buffer); } } } off_t seek_archive (off_t size) { off_t start = current_block_ordinal (); off_t offset; off_t nrec, nblk; off_t skipped = (blocking_factor - (current_block - record_start)); size -= skipped * BLOCKSIZE; if (size < record_size) return 0; /* FIXME: flush? */ /* Compute number of records to skip */ nrec = size / record_size; offset = rmtlseek (archive, nrec * record_size, SEEK_CUR); if (offset < 0) return offset; if (offset % record_size) FATAL_ERROR ((0, 0, _("rmtlseek not stopped at a record boundary"))); /* Convert to number of records */ offset /= BLOCKSIZE; /* Compute number of skipped blocks */ nblk = offset - start; /* Update buffering info */ records_read += nblk / blocking_factor; record_start_block = offset - blocking_factor; current_block = record_end; return nblk; } /* Close the archive file. */ void close_archive (void) { if (time_to_start_writing || access_mode == ACCESS_WRITE) flush_archive (); sys_drain_input_pipe (); compute_duration (); if (verify_option) verify_volume (); if (rmtclose (archive) != 0) close_warn (*archive_name_cursor); sys_wait_for_child (child_pid); tar_stat_destroy (¤t_stat_info); if (save_name) free (save_name); if (real_s_name) free (real_s_name); free (record_buffer); } /* Called to initialize the global volume number. */ void init_volume_number (void) { FILE *file = fopen (volno_file_option, "r"); if (file) { if (fscanf (file, "%d", &global_volno) != 1 || global_volno < 0) FATAL_ERROR ((0, 0, _("%s: contains invalid volume number"), quotearg_colon (volno_file_option))); if (ferror (file)) read_error (volno_file_option); if (fclose (file) != 0) close_error (volno_file_option); } else if (errno != ENOENT) open_error (volno_file_option); } /* Called to write out the closing global volume number. */ void closeout_volume_number (void) { FILE *file = fopen (volno_file_option, "w"); if (file) { fprintf (file, "%d\n", global_volno); if (ferror (file)) write_error (volno_file_option); if (fclose (file) != 0) close_error (volno_file_option); } else open_error (volno_file_option); } /* We've hit the end of the old volume. Close it and open the next one. Return nonzero on success. */ static bool new_volume (enum access_mode mode) { static FILE *read_file; static int looped; if (!read_file && !info_script_option) /* FIXME: if fopen is used, it will never be closed. */ read_file = archive == STDIN_FILENO ? fopen (TTY_NAME, "r") : stdin; if (now_verifying) return false; if (verify_option) verify_volume (); if (rmtclose (archive) != 0) close_warn (*archive_name_cursor); global_volno++; if (global_volno < 0) FATAL_ERROR ((0, 0, _("Volume number overflow"))); volno++; archive_name_cursor++; if (archive_name_cursor == archive_name_array + archive_names) { archive_name_cursor = archive_name_array; looped = 1; } tryagain: if (looped) { /* We have to prompt from now on. */ if (info_script_option) { if (volno_file_option) closeout_volume_number (); if (system (info_script_option) != 0) FATAL_ERROR ((0, 0, _("`%s' command failed"), info_script_option)); } else while (1) { char input_buffer[80]; fputc ('\007', stderr); fprintf (stderr, _("Prepare volume #%d for %s and hit return: "), global_volno, quote (*archive_name_cursor)); fflush (stderr); if (fgets (input_buffer, sizeof input_buffer, read_file) == 0) { WARN ((0, 0, _("EOF where user reply was expected"))); if (subcommand_option != EXTRACT_SUBCOMMAND && subcommand_option != LIST_SUBCOMMAND && subcommand_option != DIFF_SUBCOMMAND) WARN ((0, 0, _("WARNING: Archive is incomplete"))); fatal_exit (); } if (input_buffer[0] == '\n' || input_buffer[0] == 'y' || input_buffer[0] == 'Y') break; switch (input_buffer[0]) { case '?': { /* FIXME: Might it be useful to disable the '!' command? */ fprintf (stderr, _("\ n [name] Give a new file name for the next (and subsequent) volume(s)\n\ q Abort tar\n\ ! Spawn a subshell\n\ ? Print this list\n")); } break; case 'q': /* Quit. */ WARN ((0, 0, _("No new volume; exiting.\n"))); if (subcommand_option != EXTRACT_SUBCOMMAND && subcommand_option != LIST_SUBCOMMAND && subcommand_option != DIFF_SUBCOMMAND) WARN ((0, 0, _("WARNING: Archive is incomplete"))); fatal_exit (); case 'n': /* Get new file name. */ { char *name = &input_buffer[1]; char *cursor; for (name = input_buffer + 1; *name == ' ' || *name == '\t'; name++) ; for (cursor = name; *cursor && *cursor != '\n'; cursor++) ; *cursor = '\0'; /* FIXME: the following allocation is never reclaimed. */ *archive_name_cursor = xstrdup (name); } break; case '!': sys_spawn_shell (); break; } } } if (strcmp (archive_name_cursor[0], "-") == 0) { read_full_records = true; archive = STDIN_FILENO; } else if (verify_option) archive = rmtopen (*archive_name_cursor, O_RDWR | O_CREAT, MODE_RW, rsh_command_option); else switch (mode) { case ACCESS_READ: archive = rmtopen (*archive_name_cursor, O_RDONLY, MODE_RW, rsh_command_option); break; case ACCESS_WRITE: if (backup_option) maybe_backup_file (*archive_name_cursor, 1); archive = rmtcreat (*archive_name_cursor, MODE_RW, rsh_command_option); break; case ACCESS_UPDATE: archive = rmtopen (*archive_name_cursor, O_RDWR | O_CREAT, MODE_RW, rsh_command_option); break; } if (archive < 0) { open_warn (*archive_name_cursor); if (!verify_option && mode == ACCESS_WRITE && backup_option) undo_last_backup (); goto tryagain; } SET_BINARY_MODE (archive); return true; }