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
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
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
{
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
to the archive */
struct tar_stat_info *stat_info; /* Information about the file */
@@ -54,6 +57,39 @@ struct tar_sparse_file
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
tar_sparse_member_p (struct tar_sparse_file *file)
{
@@ -130,9 +166,16 @@ tar_sparse_fixup_header (struct tar_sparse_file *file)
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);
return false;
@@ -185,7 +228,7 @@ sparse_scan_file (struct tar_sparse_file *file)
size_t offset = 0;
struct sp_array sp = {0, 0};
if (!lseek_or_error (file, 0, SEEK_SET))
if (!lseek_or_error (file, 0))
return false;
clear_block (buffer);
@@ -269,8 +312,7 @@ sparse_dump_region (struct tar_sparse_file *file, size_t i)
union block *blk;
off_t bytes_left = file->stat_info->sparse_map[i].numbytes;
if (!lseek_or_error (file, file->stat_info->sparse_map[i].offset,
SEEK_SET))
if (!lseek_or_error (file, file->stat_info->sparse_map[i].offset))
return false;
while (bytes_left > 0)
@@ -304,8 +346,7 @@ sparse_extract_region (struct tar_sparse_file *file, size_t i)
{
size_t write_size;
if (!lseek_or_error (file, file->stat_info->sparse_map[i].offset,
SEEK_SET))
if (!lseek_or_error (file, file->stat_info->sparse_map[i].offset))
return false;
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)
{
/* 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);
}
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);
write_size -= count;
file->dumped_size += count;
file->offset += count;
if (count != wrbytes)
{
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.fd = fd;
file.seekable = true; /* File *must* be seekable for dump to work */
file.offset = 0;
if (!sparse_select_optab (&file)
|| !tar_sparse_init (&file))
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.fd = fd;
file.seekable = lseek (fd, 0, SEEK_SET) == 0;
file.offset = 0;
if (!sparse_select_optab (&file)
|| !tar_sparse_init (&file))
return dump_status_not_implemented;
@@ -450,7 +496,7 @@ static char diff_buffer[BLOCKSIZE];
static bool
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;
while (beg < end)
@@ -486,8 +532,7 @@ check_data_region (struct tar_sparse_file *file, size_t i)
{
size_t size_left;
if (!lseek_or_error (file, file->stat_info->sparse_map[i].offset,
SEEK_SET))
if (!lseek_or_error (file, file->stat_info->sparse_map[i].offset))
return false;
size_left = file->stat_info->sparse_map[i].numbytes;
while (size_left > 0)