Extract sparse files even if the output fd is not seekable.

This commit is contained in:
Sergey Poznyakoff
2005-02-02 11:01:49 +00:00
parent ccac216e67
commit 198596f161

View File

@@ -1,6 +1,6 @@
/* Functions for dealing with sparse files /* Functions for dealing with sparse files
Copyright (C) 2003, 2004 Free Software Foundation, Inc. Copyright (C) 2003, 2004, 2005 Free Software Foundation, Inc.
This program is free software; you can redistribute it and/or modify it 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 under the terms of the GNU General Public License as published by the
@@ -46,6 +46,9 @@ struct tar_sparse_optab
struct tar_sparse_file struct tar_sparse_file
{ {
int fd; /* File descriptor */ int fd; /* File descriptor */
bool seekable; /* Is fd seekable? */
size_t offset; /* Current offset in fd if seekable==false.
Otherwise unused */
size_t dumped_size; /* Number of bytes actually written size_t dumped_size; /* Number of bytes actually written
to the archive */ to the archive */
struct tar_stat_info *stat_info; /* Information about the file */ struct tar_stat_info *stat_info; /* Information about the file */
@@ -54,6 +57,39 @@ struct tar_sparse_file
reqiure */ reqiure */
}; };
/* Dump zeros to file->fd until offset is reached. It is used instead of
lseek if the output file is not seekable */
static long
dump_zeros (struct tar_sparse_file *file, off_t offset)
{
char buf[BLOCKSIZE];
if (offset - file->offset < 0)
{
errno = EINVAL;
return -1;
}
memset (buf, 0, sizeof buf);
while (file->offset < offset)
{
size_t size = offset - file->offset;
size_t wrbytes;
if (size > sizeof buf)
size = sizeof buf;
wrbytes = write (file->fd, buf, size);
if (wrbytes <= 0)
{
if (wrbytes == 0)
errno = EINVAL;
return -1;
}
file->offset += wrbytes;
}
return file->offset;
}
static bool static bool
tar_sparse_member_p (struct tar_sparse_file *file) tar_sparse_member_p (struct tar_sparse_file *file)
{ {
@@ -130,9 +166,16 @@ tar_sparse_fixup_header (struct tar_sparse_file *file)
static bool static bool
lseek_or_error (struct tar_sparse_file *file, off_t offset, int whence) lseek_or_error (struct tar_sparse_file *file, off_t offset)
{ {
if (lseek (file->fd, offset, whence) < 0) off_t off;
if (file->seekable)
off = lseek (file->fd, offset, SEEK_SET);
else
off = dump_zeros (file, offset);
if (off < 0)
{ {
seek_diag_details (file->stat_info->orig_file_name, offset); seek_diag_details (file->stat_info->orig_file_name, offset);
return false; return false;
@@ -185,7 +228,7 @@ sparse_scan_file (struct tar_sparse_file *file)
size_t offset = 0; size_t offset = 0;
struct sp_array sp = {0, 0}; struct sp_array sp = {0, 0};
if (!lseek_or_error (file, 0, SEEK_SET)) if (!lseek_or_error (file, 0))
return false; return false;
clear_block (buffer); clear_block (buffer);
@@ -269,8 +312,7 @@ sparse_dump_region (struct tar_sparse_file *file, size_t i)
union block *blk; union block *blk;
off_t bytes_left = file->stat_info->sparse_map[i].numbytes; off_t bytes_left = file->stat_info->sparse_map[i].numbytes;
if (!lseek_or_error (file, file->stat_info->sparse_map[i].offset, if (!lseek_or_error (file, file->stat_info->sparse_map[i].offset))
SEEK_SET))
return false; return false;
while (bytes_left > 0) while (bytes_left > 0)
@@ -304,8 +346,7 @@ sparse_extract_region (struct tar_sparse_file *file, size_t i)
{ {
size_t write_size; size_t write_size;
if (!lseek_or_error (file, file->stat_info->sparse_map[i].offset, if (!lseek_or_error (file, file->stat_info->sparse_map[i].offset))
SEEK_SET))
return false; return false;
write_size = file->stat_info->sparse_map[i].numbytes; write_size = file->stat_info->sparse_map[i].numbytes;
@@ -313,7 +354,7 @@ sparse_extract_region (struct tar_sparse_file *file, size_t i)
if (write_size == 0) if (write_size == 0)
{ {
/* Last block of the file is a hole */ /* Last block of the file is a hole */
if (sys_truncate (file->fd)) if (file->seekable && sys_truncate (file->fd))
truncate_warn (file->stat_info->orig_file_name); truncate_warn (file->stat_info->orig_file_name);
} }
else while (write_size > 0) else while (write_size > 0)
@@ -330,6 +371,7 @@ sparse_extract_region (struct tar_sparse_file *file, size_t i)
count = full_write (file->fd, blk->buffer, wrbytes); count = full_write (file->fd, blk->buffer, wrbytes);
write_size -= count; write_size -= count;
file->dumped_size += count; file->dumped_size += count;
file->offset += count;
if (count != wrbytes) if (count != wrbytes)
{ {
write_error_details (file->stat_info->orig_file_name, write_error_details (file->stat_info->orig_file_name,
@@ -351,7 +393,9 @@ sparse_dump_file (int fd, struct tar_stat_info *st)
file.stat_info = st; file.stat_info = st;
file.fd = fd; file.fd = fd;
file.seekable = true; /* File *must* be seekable for dump to work */
file.offset = 0;
if (!sparse_select_optab (&file) if (!sparse_select_optab (&file)
|| !tar_sparse_init (&file)) || !tar_sparse_init (&file))
return dump_status_not_implemented; return dump_status_not_implemented;
@@ -414,7 +458,9 @@ sparse_extract_file (int fd, struct tar_stat_info *st, off_t *size)
file.stat_info = st; file.stat_info = st;
file.fd = fd; file.fd = fd;
file.seekable = lseek (fd, 0, SEEK_SET) == 0;
file.offset = 0;
if (!sparse_select_optab (&file) if (!sparse_select_optab (&file)
|| !tar_sparse_init (&file)) || !tar_sparse_init (&file))
return dump_status_not_implemented; return dump_status_not_implemented;
@@ -450,7 +496,7 @@ static char diff_buffer[BLOCKSIZE];
static bool static bool
check_sparse_region (struct tar_sparse_file *file, off_t beg, off_t end) check_sparse_region (struct tar_sparse_file *file, off_t beg, off_t end)
{ {
if (!lseek_or_error (file, beg, SEEK_SET)) if (!lseek_or_error (file, beg))
return false; return false;
while (beg < end) while (beg < end)
@@ -486,8 +532,7 @@ check_data_region (struct tar_sparse_file *file, size_t i)
{ {
size_t size_left; size_t size_left;
if (!lseek_or_error (file, file->stat_info->sparse_map[i].offset, if (!lseek_or_error (file, file->stat_info->sparse_map[i].offset))
SEEK_SET))
return false; return false;
size_left = file->stat_info->sparse_map[i].numbytes; size_left = file->stat_info->sparse_map[i].numbytes;
while (size_left > 0) while (size_left > 0)