#include <system.h>
#include <system-ioctl.h>
#include <signal.h>
#include <closeout.h>
#include <fnmatch.h>
#include <getline.h>
#include <human.h>
#include <quotearg.h>
#include "common.h"
#include <rmt.h>
#define READ_ERROR_MAX 10
#define VOLUME_LABEL_APPEND " Volume [1-9]*"
static tarlong prev_written;
static tarlong bytes_written;
static void *record_buffer[2];
union block *record_buffer_aligned[2];
static int record_index;
union block *record_start;
union block *record_end;
union block *current_block;
enum access_mode access_mode;
off_t records_read;
off_t records_written;
extern off_t records_skipped;
static off_t record_start_block;
FILE *stdlis;
static void backspace_output (void);
static pid_t child_pid;
static int read_error_count;
static bool hit_eof;
static unsigned checkpoint;
static bool read_full_records = false;
extern bool time_to_start_writing;
bool write_archive_to_stdout;
void (*flush_write_ptr) (size_t);
void (*flush_read_ptr) (void);
char *volume_label;
char *continued_file_name;
uintmax_t continued_file_size;
uintmax_t continued_file_offset;
static int volno = 1;
static int global_volno = 1;
bool write_archive_to_stdout;
static char *real_s_name;
static off_t real_s_totsize;
static off_t real_s_sizeleft;
static char *save_name;
static off_t save_totsize;
static off_t save_sizeleft;
static struct tar_stat_info dummy;
void
buffer_write_global_xheader ()
{
xheader_write_global (&dummy.xhdr);
}
void
mv_begin (struct tar_stat_info *st)
{
if (multi_volume_option)
{
assign_string (&save_name, st->orig_file_name);
save_totsize = save_sizeleft = st->stat.st_size;
}
}
void
mv_end ()
{
if (multi_volume_option)
assign_string (&save_name, 0);
}
void
mv_total_size (off_t size)
{
save_totsize = size;
}
void
mv_size_left (off_t size)
{
save_sizeleft = size;
}
void
clear_read_error_count (void)
{
read_error_count = 0;
}
double duration;
void
set_start_time ()
{
gettime (&start_time);
volume_start_time = start_time;
last_stat_time = start_time;
}
void
set_volume_start_time ()
{
gettime (&volume_start_time);
last_stat_time = volume_start_time;
}
void
compute_duration ()
{
struct timespec now;
gettime (&now);
duration += ((now.tv_sec - last_stat_time.tv_sec)
+ (now.tv_nsec - last_stat_time.tv_nsec) / 1e9);
gettime (&last_stat_time);
}
enum compress_type {
ct_none,
ct_compress,
ct_gzip,
ct_bzip2
};
struct zip_magic
{
enum compress_type type;
size_t length;
char *magic;
char *program;
char *option;
};
static struct zip_magic const magic[] = {
{ ct_none, },
{ ct_compress, 2, "\037\235", "compress", "-Z" },
{ ct_gzip, 2, "\037\213", "gzip", "-z" },
{ ct_bzip2, 3, "BZh", "bzip2", "-j" },
};
#define NMAGIC (sizeof(magic)/sizeof(magic[0]))
#define compress_option(t) magic[t].option
#define compress_program(t) magic[t].program
enum compress_type
check_compressed_archive ()
{
struct zip_magic const *p;
bool sfr;
record_end = record_start;
sfr = read_full_records;
read_full_records = true;
find_next_block ();
read_full_records = sfr;
if (tar_checksum (record_start, true) == HEADER_SUCCESS)
return ct_none;
for (p = magic + 1; p < magic + NMAGIC; p++)
if (memcmp (record_start->buffer, p->magic, p->length) == 0)
return p->type;
return ct_none;
}
int
open_compressed_archive ()
{
archive = rmtopen (archive_name_array[0], O_RDONLY | O_BINARY,
MODE_RW, rsh_command_option);
if (archive == -1)
return archive;
if (!multi_volume_option)
{
enum compress_type type = check_compressed_archive ();
if (type == ct_none)
return archive;
rmtclose (archive);
hit_eof = false;
use_compress_program_option = compress_program (type);
child_pid = sys_child_open_for_uncompress ();
read_full_records = true;
}
records_read = 0;
record_end = record_start;
return archive;
}
static void
print_stats (FILE *fp, const char *text, tarlong numbytes)
{
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, numbytes);
fprintf (fp, "%s: %s (%s, %s/s)\n",
text, bytes,
human_readable (numbytes, abbr, human_opts, 1, 1),
(0 < duration && numbytes / duration < (uintmax_t) -1
? human_readable (numbytes / duration, rate, human_opts, 1, 1)
: "?"));
}
void
print_total_stats ()
{
switch (subcommand_option)
{
case CREATE_SUBCOMMAND:
case CAT_SUBCOMMAND:
case UPDATE_SUBCOMMAND:
case APPEND_SUBCOMMAND:
print_stats (stderr, _("Total bytes written"),
prev_written + bytes_written);
break;
case DELETE_SUBCOMMAND:
{
char buf[UINTMAX_STRSIZE_BOUND];
print_stats (stderr, _("Total bytes read"),
records_read * record_size);
print_stats (stderr, _("Total bytes written"),
prev_written + bytes_written);
fprintf (stderr, _("Total bytes deleted: %s\n"),
STRINGIFY_BIGINT ((records_read - records_skipped)
* record_size
- (prev_written + bytes_written), buf));
}
break;
case EXTRACT_SUBCOMMAND:
case LIST_SUBCOMMAND:
case DIFF_SUBCOMMAND:
print_stats (stderr, _("Total bytes read"),
records_read * record_size);
break;
default:
abort ();
}
}
off_t
current_block_ordinal (void)
{
return record_start_block + (current_block - record_start);
}
void
reset_eof (void)
{
if (hit_eof)
{
hit_eof = false;
current_block = record_start;
record_end = record_start + blocking_factor;
access_mode = ACCESS_WRITE;
}
}
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;
}
void
set_next_block_after (union block *block)
{
while (block >= current_block)
current_block++;
if (current_block > record_end)
abort ();
}
size_t
available_space_after (union block *pointer)
{
return record_end->buffer - pointer->buffer;
}
void
xclose (int fd)
{
if (close (fd) != 0)
close_error (_("(pipe)"));
}
static void
init_buffer ()
{
if (! record_buffer_aligned[record_index])
record_buffer_aligned[record_index] =
page_aligned_alloc (&record_buffer[record_index], record_size);
record_start = record_buffer_aligned[record_index];
current_block = record_start;
record_end = record_start + blocking_factor;
}
#if HAVE_QUARANTINE
void
init_qtn(const char *file_name)
{
int fd;
if (strcmp(file_name, "-") == 0)
return;
if ((fd = open(file_name, O_RDONLY)) < 0)
return;
if ((archive_qtn_file = qtn_file_alloc()) != NULL) {
if (qtn_file_init_with_fd(archive_qtn_file, fd) != 0) {
qtn_file_free(archive_qtn_file);
archive_qtn_file = NULL;
}
}
close(fd);
}
void
finish_qtn(void)
{
if (archive_qtn_file != NULL) {
qtn_file_free(archive_qtn_file);
archive_qtn_file = NULL;
}
}
#else
void init_qtn(const char *file_name) {}
void finish_qtn(void) {}
#endif
static void
_open_archive (enum access_mode wanted_access)
{
int backed_up_flag = 0;
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_index = 0;
init_buffer ();
access_mode = wanted_access == ACCESS_UPDATE ? ACCESS_READ : wanted_access;
read_full_records = read_full_records_option;
records_read = 0;
if (use_compress_program_option)
{
switch (wanted_access)
{
case ACCESS_READ:
child_pid = sys_child_open_for_uncompress ();
read_full_records = true;
record_end = record_start;
break;
case ACCESS_WRITE:
child_pid = sys_child_open_for_compress ();
break;
case ACCESS_UPDATE:
abort ();
break;
}
if (!index_file_name
&& wanted_access == ACCESS_WRITE
&& strcmp (archive_name_array[0], "-") == 0)
stdlis = stderr;
}
else if (strcmp (archive_name_array[0], "-") == 0)
{
read_full_records = true;
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 ();
if (type != ct_none)
FATAL_ERROR ((0, 0,
_("Archive is compressed. Use %s option"),
compress_option (type)));
}
break;
case ACCESS_WRITE:
archive = STDOUT_FILENO;
if (!index_file_name)
stdlis = stderr;
break;
case ACCESS_UPDATE:
archive = STDIN_FILENO;
write_archive_to_stdout = true;
record_end = record_start;
if (!index_file_name)
stdlis = stderr;
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);
if (check_compressed_archive () != ct_none)
FATAL_ERROR ((0, 0,
_("Cannot update compressed archives")));
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);
init_qtn(archive_name_array[0]);
switch (wanted_access)
{
case ACCESS_READ:
find_next_block ();
break;
case ACCESS_UPDATE:
case ACCESS_WRITE:
records_written = 0;
break;
}
}
static void
do_checkpoint (bool write)
{
if (checkpoint_option && !(++checkpoint % checkpoint_option))
{
switch (checkpoint_style)
{
case checkpoint_dot:
fputc ('.', stdlis);
fflush (stdlis);
break;
case checkpoint_text:
if (write)
WARN ((0, 0, _("Write checkpoint %u"), checkpoint));
else
WARN ((0, 0, _("Read checkpoint %u"), checkpoint));
break;
}
}
}
ssize_t
_flush_write (void)
{
ssize_t status;
do_checkpoint (true);
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 ();
return status;
}
void
archive_write_error (ssize_t status)
{
if (totals_option)
{
int e = errno;
print_total_stats ();
errno = e;
}
write_fatal_details (*archive_name_cursor, status, record_size);
}
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")));
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;
char *more;
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)
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));
}
left -= status;
more += status;
}
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++;
}
void
flush_archive (void)
{
size_t buffer_level = current_block->buffer - record_start->buffer;
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_ptr (buffer_level);
break;
case ACCESS_UPDATE:
abort ();
}
}
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);
position -= record_size;
if (position < 0)
position = 0;
if (rmtlseek (archive, position, SEEK_SET) != position)
{
WARN ((0, 0,
_("Cannot backspace archive file; it may be unreadable without -i")));
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;
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")));
offset /= BLOCKSIZE;
nblk = offset - start;
records_read += nblk / blocking_factor;
record_start_block = offset - blocking_factor;
current_block = record_end;
return nblk;
}
void
close_archive (void)
{
if (time_to_start_writing || access_mode == ACCESS_WRITE)
{
flush_archive ();
if (current_block > record_start)
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[0]);
free (record_buffer[1]);
finish_qtn();
}
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);
}
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);
}
static void
increase_volume_number ()
{
global_volno++;
if (global_volno < 0)
FATAL_ERROR ((0, 0, _("Volume number overflow")));
volno++;
}
void
change_tape_menu (FILE *read_file)
{
char *input_buffer = NULL;
size_t size = 0;
bool stop = false;
while (!stop)
{
fputc ('\007', stderr);
fprintf (stderr,
_("Prepare volume #%d for %s and hit return: "),
global_volno + 1, quote (*archive_name_cursor));
fflush (stderr);
if (getline (&input_buffer, &size, 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 '?':
{
fprintf (stderr, _("\
n name Give a new file name for the next (and subsequent) volume(s)\n\
q Abort tar\n\
y or newline Continue operation\n"));
if (!restrict_option)
fprintf (stderr, _(" ! Spawn a subshell\n"));
fprintf (stderr, _(" ? Print this list\n"));
}
break;
case 'q':
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':
{
char *name;
char *cursor;
for (name = input_buffer + 1;
*name == ' ' || *name == '\t';
name++)
;
for (cursor = name; *cursor && *cursor != '\n'; cursor++)
;
*cursor = '\0';
if (name[0])
{
*archive_name_cursor = xstrdup (name);
stop = true;
}
else
fprintf (stderr, "%s",
_("File name not specified. Try again.\n"));
}
break;
case '!':
if (!restrict_option)
{
sys_spawn_shell ();
break;
}
default:
fprintf (stderr, _("Invalid input. Type ? for help.\n"));
}
}
free (input_buffer);
}
static bool
new_volume (enum access_mode mode)
{
static FILE *read_file;
static int looped;
int prompt;
if (!read_file && !info_script_option)
read_file = archive == STDIN_FILENO ? fopen (TTY_NAME, "r") : stdin;
if (now_verifying)
return false;
if (verify_option)
verify_volume ();
assign_string (&volume_label, NULL);
assign_string (&continued_file_name, NULL);
continued_file_size = continued_file_offset = 0;
current_block = record_start;
if (rmtclose (archive) != 0)
close_warn (*archive_name_cursor);
archive_name_cursor++;
if (archive_name_cursor == archive_name_array + archive_names)
{
archive_name_cursor = archive_name_array;
looped = 1;
}
prompt = looped;
tryagain:
if (prompt)
{
if (info_script_option)
{
if (volno_file_option)
closeout_volume_number ();
if (sys_exec_info_script (archive_name_cursor, global_volno+1))
FATAL_ERROR ((0, 0, _("%s command failed"),
quote (info_script_option)));
}
else
change_tape_menu (read_file);
}
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 ();
prompt = 1;
goto tryagain;
}
SET_BINARY_MODE (archive);
return true;
}
static bool
read_header0 (struct tar_stat_info *info)
{
enum read_header rc;
tar_stat_init (info);
rc = read_header_primitive (false, info);
if (rc == HEADER_SUCCESS)
{
set_next_block_after (current_header);
return true;
}
ERROR ((0, 0, _("This does not look like a tar archive")));
return false;
}
bool
try_new_volume ()
{
size_t status;
union block *header;
int access;
switch (subcommand_option)
{
case APPEND_SUBCOMMAND:
case CAT_SUBCOMMAND:
case UPDATE_SUBCOMMAND:
access = ACCESS_UPDATE;
break;
default:
access = ACCESS_READ;
break;
}
if (!new_volume (access))
return true;
while ((status = rmtread (archive, record_start->buffer, record_size))
== SAFE_READ_ERROR)
archive_read_error ();
if (status != record_size)
short_read (status);
header = find_next_block ();
if (!header)
return false;
switch (header->header.typeflag)
{
case XGLTYPE:
{
if (!read_header0 (&dummy))
return false;
xheader_decode (&dummy);
tar_stat_destroy (&dummy);
if (!real_s_name)
{
current_block = record_start;
}
break;
}
case GNUTYPE_VOLHDR:
if (!read_header0 (&dummy))
return false;
tar_stat_destroy (&dummy);
assign_string (&volume_label, current_header->header.name);
set_next_block_after (header);
header = find_next_block ();
if (header->header.typeflag != GNUTYPE_MULTIVOL)
break;
case GNUTYPE_MULTIVOL:
if (!read_header0 (&dummy))
return false;
tar_stat_destroy (&dummy);
assign_string (&continued_file_name, current_header->header.name);
continued_file_size =
UINTMAX_FROM_HEADER (current_header->header.size);
continued_file_offset =
UINTMAX_FROM_HEADER (current_header->oldgnu_header.offset);
break;
default:
break;
}
if (real_s_name)
{
uintmax_t s;
if (!continued_file_name
|| strcmp (continued_file_name, real_s_name))
{
if ((archive_format == GNU_FORMAT || archive_format == OLDGNU_FORMAT)
&& strlen (real_s_name) >= NAME_FIELD_SIZE
&& strncmp (continued_file_name, real_s_name,
NAME_FIELD_SIZE) == 0)
WARN ((0, 0,
_("%s is possibly continued on this volume: header contains truncated name"),
quote (real_s_name)));
else
{
WARN ((0, 0, _("%s is not continued on this volume"),
quote (real_s_name)));
return false;
}
}
s = continued_file_size + continued_file_offset;
if (real_s_totsize != s || s < continued_file_offset)
{
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 (continued_file_name),
STRINGIFY_BIGINT (save_totsize, totsizebuf),
STRINGIFY_BIGINT (continued_file_size, s1buf),
STRINGIFY_BIGINT (continued_file_offset, s2buf)));
return false;
}
if (real_s_totsize - real_s_sizeleft != continued_file_offset)
{
WARN ((0, 0, _("This volume is out of sequence")));
return false;
}
}
increase_volume_number ();
return true;
}
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;
}
static void
match_volume_label (void)
{
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)));
}
static void
_write_volume_label (const char *str)
{
if (archive_format == POSIX_FORMAT)
xheader_store ("GNU.volume.label", &dummy, str);
else
{
union block *label = find_next_block ();
memset (label, 0, BLOCKSIZE);
strcpy (label->header.name, volume_label_option);
assign_string (¤t_stat_info.file_name,
label->header.name);
current_stat_info.had_trailing_slash =
strip_trailing_slashes (current_stat_info.file_name);
label->header.typeflag = GNUTYPE_VOLHDR;
TIME_TO_CHARS (start_time.tv_sec, label->header.mtime);
finish_header (¤t_stat_info, label, -1);
set_next_block_after (label);
}
}
#define VOL_SUFFIX "Volume"
static void
add_volume_label (void)
{
char buf[UINTMAX_STRSIZE_BOUND];
char *p = STRINGIFY_BIGINT (volno, buf);
char *s = xmalloc (strlen (volume_label_option) + sizeof VOL_SUFFIX
+ strlen (p) + 2);
sprintf (s, "%s %s %s", volume_label_option, VOL_SUFFIX, p);
_write_volume_label (s);
free (s);
}
static void
add_chunk_header ()
{
if (archive_format == POSIX_FORMAT)
{
off_t block_ordinal;
union block *blk;
struct tar_stat_info st;
static size_t real_s_part_no;
real_s_part_no++;
memset (&st, 0, sizeof st);
st.orig_file_name = st.file_name = real_s_name;
st.stat.st_mode = S_IFREG|S_IRUSR|S_IWUSR|S_IRGRP|S_IROTH;
st.stat.st_uid = getuid ();
st.stat.st_gid = getgid ();
st.orig_file_name = xheader_format_name (&st,
"%d/GNUFileParts.%p/%f.%n",
real_s_part_no);
st.file_name = st.orig_file_name;
st.archive_file_size = st.stat.st_size = real_s_sizeleft;
block_ordinal = current_block_ordinal ();
blk = start_header (&st);
if (!blk)
abort ();
finish_header (&st, blk, block_ordinal);
free (st.orig_file_name);
}
}
static void
write_volume_label (void)
{
if (multi_volume_option)
add_volume_label ();
else
_write_volume_label (volume_label_option);
}
static void
gnu_add_multi_volume_header (void)
{
int tmp;
union block *block = find_next_block ();
if (strlen (real_s_name) > NAME_FIELD_SIZE)
WARN ((0, 0,
_("%s: file name too long to be stored in a GNU multivolume header, truncated"),
quotearg_colon (real_s_name)));
memset (block, 0, BLOCKSIZE);
strncpy (block->header.name, real_s_name, NAME_FIELD_SIZE);
block->header.typeflag = GNUTYPE_MULTIVOL;
OFF_TO_CHARS (real_s_sizeleft, block->header.size);
OFF_TO_CHARS (real_s_totsize - real_s_sizeleft,
block->oldgnu_header.offset);
tmp = verbose_option;
verbose_option = 0;
finish_header (¤t_stat_info, block, -1);
verbose_option = tmp;
set_next_block_after (block);
}
static void
add_multi_volume_header (void)
{
if (archive_format == POSIX_FORMAT)
{
off_t d = real_s_totsize - real_s_sizeleft;
xheader_store ("GNU.volume.filename", &dummy, real_s_name);
xheader_store ("GNU.volume.size", &dummy, &real_s_sizeleft);
xheader_store ("GNU.volume.offset", &dummy, &d);
}
else
gnu_add_multi_volume_header ();
}
static void
multi_volume_sync ()
{
if (multi_volume_option)
{
if (save_name)
{
assign_string (&real_s_name,
safer_name_suffix (save_name, false,
absolute_names_option));
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;
}
}
}
static void
simple_flush_read (void)
{
size_t status;
do_checkpoint (false);
read_error_count = 0;
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);
}
for (;;)
{
status = rmtread (archive, record_start->buffer, record_size);
if (status == record_size)
{
records_read++;
return;
}
if (status == SAFE_READ_ERROR)
{
archive_read_error ();
continue;
}
break;
}
short_read (status);
}
static void
simple_flush_write (size_t level __attribute__((unused)))
{
ssize_t status;
status = _flush_write ();
if (status != record_size)
archive_write_error (status);
else
{
records_written++;
bytes_written += status;
}
}
static void
_gnu_flush_read (void)
{
size_t status;
do_checkpoint (false);
read_error_count = 0;
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);
}
multi_volume_sync ();
for (;;)
{
status = rmtread (archive, record_start->buffer, record_size);
if (status == record_size)
{
records_read++;
return;
}
if ((status == 0
|| (status == SAFE_READ_ERROR && errno == ENOSPC))
&& multi_volume_option)
{
while (!try_new_volume ())
;
return;
}
else if (status == SAFE_READ_ERROR)
{
archive_read_error ();
continue;
}
break;
}
short_read (status);
}
static void
gnu_flush_read (void)
{
flush_read_ptr = simple_flush_read;
_gnu_flush_read ();
flush_read_ptr = gnu_flush_read;
}
static void
_gnu_flush_write (size_t buffer_level)
{
ssize_t status;
union block *header;
char *copy_ptr;
size_t copy_size;
size_t bufsize;
status = _flush_write ();
if (status != record_size && !multi_volume_option)
archive_write_error (status);
else
{
records_written++;
bytes_written += status;
}
if (status == record_size)
{
multi_volume_sync ();
return;
}
if (status < 0 && errno != ENOSPC && errno != EIO && errno != ENXIO)
archive_write_error (status);
if (!new_volume (ACCESS_WRITE))
return;
tar_stat_destroy (&dummy);
increase_volume_number ();
prev_written += bytes_written;
bytes_written = 0;
copy_ptr = record_start->buffer + status;
copy_size = buffer_level - status;
record_index = !record_index;
init_buffer ();
if (volume_label_option)
add_volume_label ();
if (real_s_name)
add_multi_volume_header ();
write_extended (true, &dummy, find_next_block ());
tar_stat_destroy (&dummy);
if (real_s_name)
add_chunk_header ();
header = find_next_block ();
bufsize = available_space_after (header);
while (bufsize < copy_size)
{
memcpy (header->buffer, copy_ptr, bufsize);
copy_ptr += bufsize;
copy_size -= bufsize;
set_next_block_after (header + (bufsize - 1) / BLOCKSIZE);
header = find_next_block ();
bufsize = available_space_after (header);
}
memcpy (header->buffer, copy_ptr, copy_size);
memset (header->buffer + copy_size, 0, bufsize - copy_size);
set_next_block_after (header + (copy_size - 1) / BLOCKSIZE);
find_next_block ();
}
static void
gnu_flush_write (size_t buffer_level)
{
flush_write_ptr = simple_flush_write;
_gnu_flush_write (buffer_level);
flush_write_ptr = gnu_flush_write;
}
void
flush_read ()
{
flush_read_ptr ();
}
void
flush_write ()
{
flush_write_ptr (record_size);
}
void
open_archive (enum access_mode wanted_access)
{
flush_read_ptr = gnu_flush_read;
flush_write_ptr = gnu_flush_write;
_open_archive (wanted_access);
switch (wanted_access)
{
case ACCESS_READ:
if (volume_label_option)
match_volume_label ();
break;
case ACCESS_WRITE:
records_written = 0;
if (volume_label_option)
write_volume_label ();
break;
default:
break;
}
set_volume_start_time ();
}