As we no longer use function pointers for read in common archiving code
archive_xread can be replaced with bb_full_read, and archive_copy_file with bb_copyfd* bb_copyfd is split into two functions bb_copyfd_size and bb_copyfd_eof, they share a common backend.
This commit is contained in:
parent
1a2d75fd72
commit
7ffe133864
@ -59,7 +59,7 @@ static void data_extract_regular_file(archive_handle_t *archive_handle)
|
|||||||
|
|
||||||
file_header = archive_handle->file_header;
|
file_header = archive_handle->file_header;
|
||||||
dst_fd = bb_xopen(file_header->name, O_WRONLY | O_CREAT);
|
dst_fd = bb_xopen(file_header->name, O_WRONLY | O_CREAT);
|
||||||
archive_copy_file(archive_handle, dst_fd);
|
bb_copyfd_eof(archive_handle->src_fd, dst_fd, file_header->size);
|
||||||
close(dst_fd);
|
close(dst_fd);
|
||||||
|
|
||||||
chmod(file_header->name, file_header->mode);
|
chmod(file_header->name, file_header->mode);
|
||||||
|
@ -37,14 +37,11 @@ LIBUNARCHIVE-y:= \
|
|||||||
header_list.o \
|
header_list.o \
|
||||||
header_verbose_list.o \
|
header_verbose_list.o \
|
||||||
\
|
\
|
||||||
archive_xread.o \
|
|
||||||
archive_xread_all.o \
|
archive_xread_all.o \
|
||||||
archive_xread_all_eof.o \
|
archive_xread_all_eof.o \
|
||||||
\
|
\
|
||||||
seek_by_char.o \
|
seek_by_char.o \
|
||||||
seek_by_jump.o \
|
seek_by_jump.o \
|
||||||
\
|
|
||||||
archive_copy_file.o \
|
|
||||||
\
|
\
|
||||||
data_align.o \
|
data_align.o \
|
||||||
find_list_entry.o \
|
find_list_entry.o \
|
||||||
|
@ -1,44 +0,0 @@
|
|||||||
/*
|
|
||||||
* 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 of the License, 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 <unistd.h>
|
|
||||||
|
|
||||||
#include "libbb.h"
|
|
||||||
#include "unarchive.h"
|
|
||||||
|
|
||||||
extern void archive_copy_file(const archive_handle_t *archive_handle, const int dst_fd)
|
|
||||||
{
|
|
||||||
char buffer[512];
|
|
||||||
off_t chunksize = archive_handle->file_header->size;
|
|
||||||
|
|
||||||
while (chunksize != 0) {
|
|
||||||
size_t size;
|
|
||||||
if (chunksize > 512) {
|
|
||||||
size = 512;
|
|
||||||
} else {
|
|
||||||
size = chunksize;
|
|
||||||
}
|
|
||||||
// archive_xread_all(archive_handle, buffer, size);
|
|
||||||
size = archive_xread(archive_handle, buffer, size);
|
|
||||||
|
|
||||||
if (write(dst_fd, buffer, size) != size) {
|
|
||||||
bb_error_msg_and_die ("Short write");
|
|
||||||
}
|
|
||||||
chunksize -= size;
|
|
||||||
}
|
|
||||||
|
|
||||||
return;
|
|
||||||
}
|
|
@ -1,33 +0,0 @@
|
|||||||
/*
|
|
||||||
* 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 of the License, 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 Library 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 <stdio.h>
|
|
||||||
#include <stdlib.h>
|
|
||||||
#include <string.h>
|
|
||||||
#include "unarchive.h"
|
|
||||||
#include "libbb.h"
|
|
||||||
|
|
||||||
extern ssize_t archive_xread(const archive_handle_t *archive_handle, unsigned char *buf, const size_t count)
|
|
||||||
{
|
|
||||||
ssize_t size;
|
|
||||||
|
|
||||||
size = bb_full_read(archive_handle->src_fd, buf, count);
|
|
||||||
if (size < 0) {
|
|
||||||
bb_perror_msg_and_die("Read error");
|
|
||||||
}
|
|
||||||
|
|
||||||
return(size);
|
|
||||||
}
|
|
@ -24,7 +24,7 @@ extern void archive_xread_all(const archive_handle_t *archive_handle, void *buf,
|
|||||||
{
|
{
|
||||||
ssize_t size;
|
ssize_t size;
|
||||||
|
|
||||||
size = archive_xread(archive_handle, buf, count);
|
size = bb_full_read(archive_handle->src_fd, buf, count);
|
||||||
if (size != count) {
|
if (size != count) {
|
||||||
bb_error_msg_and_die("Short read");
|
bb_error_msg_and_die("Short read");
|
||||||
}
|
}
|
||||||
|
@ -24,7 +24,7 @@ extern ssize_t archive_xread_all_eof(archive_handle_t *archive_handle, unsigned
|
|||||||
{
|
{
|
||||||
ssize_t size;
|
ssize_t size;
|
||||||
|
|
||||||
size = archive_xread(archive_handle, buf, count);
|
size = bb_full_read(archive_handle->src_fd, buf, count);
|
||||||
if ((size != 0) && (size != count)) {
|
if ((size != 0) && (size != count)) {
|
||||||
bb_perror_msg_and_die("Short read, read %d of %d", size, count);
|
bb_perror_msg_and_die("Short read, read %d of %d", size, count);
|
||||||
}
|
}
|
||||||
|
@ -79,7 +79,7 @@ extern void data_extract_all(archive_handle_t *archive_handle)
|
|||||||
case S_IFREG: {
|
case S_IFREG: {
|
||||||
/* Regular file */
|
/* Regular file */
|
||||||
dst_fd = bb_xopen(file_header->name, O_WRONLY | O_CREAT | O_EXCL);
|
dst_fd = bb_xopen(file_header->name, O_WRONLY | O_CREAT | O_EXCL);
|
||||||
archive_copy_file(archive_handle, dst_fd);
|
bb_copyfd_size(archive_handle->src_fd, dst_fd, file_header->size);
|
||||||
close(dst_fd);
|
close(dst_fd);
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
@ -18,5 +18,5 @@
|
|||||||
|
|
||||||
extern void data_extract_to_stdout(archive_handle_t *archive_handle)
|
extern void data_extract_to_stdout(archive_handle_t *archive_handle)
|
||||||
{
|
{
|
||||||
archive_copy_file(archive_handle, fileno(stdout));
|
bb_copyfd_eof(archive_handle->src_fd, fileno(stdout));
|
||||||
}
|
}
|
||||||
|
@ -57,7 +57,7 @@ extern char get_header_tar(archive_handle_t *archive_handle)
|
|||||||
/* Align header */
|
/* Align header */
|
||||||
data_align(archive_handle, 512);
|
data_align(archive_handle, 512);
|
||||||
|
|
||||||
if (archive_xread(archive_handle, tar.raw, 512) != 512) {
|
if (bb_full_read(archive_handle->src_fd, tar.raw, 512) != 512) {
|
||||||
/* Assume end of file */
|
/* Assume end of file */
|
||||||
return(EXIT_FAILURE);
|
return(EXIT_FAILURE);
|
||||||
}
|
}
|
||||||
|
@ -26,19 +26,7 @@
|
|||||||
*/
|
*/
|
||||||
extern void seek_by_char(const archive_handle_t *archive_handle, const unsigned int jump_size)
|
extern void seek_by_char(const archive_handle_t *archive_handle, const unsigned int jump_size)
|
||||||
{
|
{
|
||||||
unsigned int remaining = jump_size;
|
if (jump_size) {
|
||||||
unsigned int read_amount;
|
bb_full_fd_action(archive_handle->src_fd, -1, jump_size, NULL);
|
||||||
RESERVE_CONFIG_BUFFER(buf, BUFSIZ);
|
|
||||||
|
|
||||||
while (remaining > 0) {
|
|
||||||
if (remaining > BUFSIZ) {
|
|
||||||
read_amount = BUFSIZ;
|
|
||||||
} else {
|
|
||||||
read_amount = remaining;
|
|
||||||
}
|
}
|
||||||
read_amount = archive_xread(archive_handle, buf, read_amount);
|
|
||||||
remaining -= read_amount;
|
|
||||||
}
|
|
||||||
|
|
||||||
RELEASE_CONFIG_BUFFER(buf);
|
|
||||||
}
|
}
|
||||||
|
@ -414,8 +414,7 @@ static int writeFileToTarball(const char *fileName, struct stat *statbuf,
|
|||||||
if ((tbInfo->hlInfo == NULL)
|
if ((tbInfo->hlInfo == NULL)
|
||||||
&& (S_ISREG(statbuf->st_mode))) {
|
&& (S_ISREG(statbuf->st_mode))) {
|
||||||
int inputFileFd;
|
int inputFileFd;
|
||||||
char buffer[BUFSIZ];
|
ssize_t readSize = 0;
|
||||||
ssize_t size = 0, readSize = 0;
|
|
||||||
|
|
||||||
/* open the file we want to archive, and make sure all is well */
|
/* open the file we want to archive, and make sure all is well */
|
||||||
if ((inputFileFd = open(fileName, O_RDONLY)) < 0) {
|
if ((inputFileFd = open(fileName, O_RDONLY)) < 0) {
|
||||||
@ -424,18 +423,8 @@ static int writeFileToTarball(const char *fileName, struct stat *statbuf,
|
|||||||
}
|
}
|
||||||
|
|
||||||
/* write the file to the archive */
|
/* write the file to the archive */
|
||||||
while ((size = bb_full_read(inputFileFd, buffer, sizeof(buffer))) > 0) {
|
readSize = bb_copyfd_eof(inputFileFd, tbInfo->tarFd);
|
||||||
if (bb_full_write(tbInfo->tarFd, buffer, size) != size) {
|
|
||||||
/* Output file seems to have a problem */
|
|
||||||
bb_error_msg(bb_msg_io_error, fileName);
|
|
||||||
return (FALSE);
|
|
||||||
}
|
|
||||||
readSize += size;
|
|
||||||
}
|
|
||||||
if (size == -1) {
|
|
||||||
bb_error_msg(bb_msg_io_error, fileName);
|
|
||||||
return (FALSE);
|
|
||||||
}
|
|
||||||
/* Pad the file up to the tar block size */
|
/* Pad the file up to the tar block size */
|
||||||
for (; (readSize % TAR_BLOCK_SIZE) != 0; readSize++) {
|
for (; (readSize % TAR_BLOCK_SIZE) != 0; readSize++) {
|
||||||
write(tbInfo->tarFd, "\0", 1);
|
write(tbInfo->tarFd, "\0", 1);
|
||||||
|
@ -54,7 +54,7 @@ extern int cat_main(int argc, char **argv)
|
|||||||
|
|
||||||
do {
|
do {
|
||||||
if ((f = bb_wfopen_input(*argv)) != NULL) {
|
if ((f = bb_wfopen_input(*argv)) != NULL) {
|
||||||
int r = bb_copyfd(fileno(f), STDOUT_FILENO, 0);
|
int r = bb_copyfd_eof(fileno(f), STDOUT_FILENO);
|
||||||
bb_fclose_nonstdin(f);
|
bb_fclose_nonstdin(f);
|
||||||
if (r >= 0) {
|
if (r >= 0) {
|
||||||
continue;
|
continue;
|
||||||
|
@ -136,7 +136,8 @@ extern long *find_pid_by_name( const char* pidName);
|
|||||||
extern char *find_real_root_device_name(const char* name);
|
extern char *find_real_root_device_name(const char* name);
|
||||||
extern char *bb_get_line_from_file(FILE *file);
|
extern char *bb_get_line_from_file(FILE *file);
|
||||||
extern char *bb_get_chomped_line_from_file(FILE *file);
|
extern char *bb_get_chomped_line_from_file(FILE *file);
|
||||||
extern int bb_copyfd(int fd1, int fd2, const off_t chunksize);
|
extern int bb_copyfd_size(int fd1, int fd2, const off_t size);
|
||||||
|
extern int bb_copyfd_eof(int fd1, int fd2);
|
||||||
extern void bb_xprint_and_close_file(FILE *file);
|
extern void bb_xprint_and_close_file(FILE *file);
|
||||||
extern int bb_xprint_file_by_name(const char *filename);
|
extern int bb_xprint_file_by_name(const char *filename);
|
||||||
extern char bb_process_escape_sequence(const char **ptr);
|
extern char bb_process_escape_sequence(const char **ptr);
|
||||||
@ -480,6 +481,6 @@ extern void xregcomp(regex_t *preg, const char *regex, int cflags);
|
|||||||
#define HASH_SHA1 1
|
#define HASH_SHA1 1
|
||||||
#define HASH_MD5 2
|
#define HASH_MD5 2
|
||||||
extern int hash_fd(int fd, const size_t size, const uint8_t hash_algo, uint8_t *hashval);
|
extern int hash_fd(int fd, const size_t size, const uint8_t hash_algo, uint8_t *hashval);
|
||||||
|
extern size_t bb_full_fd_action(int src_fd, int dst_fd, const size_t size, ssize_t (*action)(int fd, const void *, size_t));
|
||||||
|
|
||||||
#endif /* __LIBCONFIG_H__ */
|
#endif /* __LIBCONFIG_H__ */
|
||||||
|
@ -90,12 +90,10 @@ extern char get_header_tar_gz(archive_handle_t *archive_handle);
|
|||||||
extern void seek_by_jump(const archive_handle_t *archive_handle, const unsigned int amount);
|
extern void seek_by_jump(const archive_handle_t *archive_handle, const unsigned int amount);
|
||||||
extern void seek_by_char(const archive_handle_t *archive_handle, const unsigned int amount);
|
extern void seek_by_char(const archive_handle_t *archive_handle, const unsigned int amount);
|
||||||
|
|
||||||
extern ssize_t archive_xread(const archive_handle_t *archive_handle, unsigned char *buf, const size_t count);
|
|
||||||
extern void archive_xread_all(const archive_handle_t *archive_handle, void *buf, const size_t count);
|
extern void archive_xread_all(const archive_handle_t *archive_handle, void *buf, const size_t count);
|
||||||
extern ssize_t archive_xread_all_eof(archive_handle_t *archive_handle, unsigned char *buf, size_t count);
|
extern ssize_t archive_xread_all_eof(archive_handle_t *archive_handle, unsigned char *buf, size_t count);
|
||||||
|
|
||||||
extern void data_align(archive_handle_t *archive_handle, const unsigned short boundary);
|
extern void data_align(archive_handle_t *archive_handle, const unsigned short boundary);
|
||||||
extern void archive_copy_file(const archive_handle_t *archive_handle, const int dst_fd);
|
|
||||||
extern const llist_t *find_list_entry(const llist_t *list, const char *filename);
|
extern const llist_t *find_list_entry(const llist_t *list, const char *filename);
|
||||||
|
|
||||||
extern int uncompressStream(int src_fd, int dst_fd);
|
extern int uncompressStream(int src_fd, int dst_fd);
|
||||||
|
@ -181,7 +181,7 @@ int copy_file(const char *source, const char *dest, int flags)
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
if (bb_copyfd(fileno(sfp), fileno(dfp), 0) == -1)
|
if (bb_copyfd_eof(fileno(sfp), fileno(dfp)) == -1)
|
||||||
status = -1;
|
status = -1;
|
||||||
|
|
||||||
if (fclose(dfp) < 0) {
|
if (fclose(dfp) < 0) {
|
||||||
|
@ -29,44 +29,55 @@
|
|||||||
#define BUFSIZ 4096
|
#define BUFSIZ 4096
|
||||||
#endif
|
#endif
|
||||||
|
|
||||||
/* If chunksize is 0 copy until EOF */
|
/* If size is 0 copy until EOF */
|
||||||
extern int bb_copyfd(int fd1, int fd2, const off_t chunksize)
|
extern size_t bb_full_fd_action(int src_fd, int dst_fd, const size_t size, ssize_t (*action)(int fd, const void *, size_t))
|
||||||
{
|
{
|
||||||
ssize_t nread;
|
size_t read_total = 0;
|
||||||
size_t size;
|
|
||||||
off_t remaining;
|
|
||||||
RESERVE_CONFIG_BUFFER(buffer,BUFSIZ);
|
RESERVE_CONFIG_BUFFER(buffer,BUFSIZ);
|
||||||
|
|
||||||
remaining = size = BUFSIZ;
|
while ((size == 0) || (read_total < size)) {
|
||||||
if (chunksize) {
|
size_t read_try;
|
||||||
remaining = chunksize;
|
ssize_t read_actual;
|
||||||
|
|
||||||
|
if ((size == 0) || (size - read_total > BUFSIZ)) {
|
||||||
|
read_try = BUFSIZ;
|
||||||
|
} else {
|
||||||
|
read_try = size - read_total;
|
||||||
}
|
}
|
||||||
|
|
||||||
do {
|
read_actual = safe_read(src_fd, buffer, read_try);
|
||||||
if (size > remaining) {
|
if (read_actual > 0) {
|
||||||
size = remaining;
|
if (action && (action(dst_fd, buffer, (size_t) read_actual) != read_actual)) {
|
||||||
}
|
|
||||||
|
|
||||||
if ((nread = safe_read(fd1, buffer, size)) > 0) {
|
|
||||||
if (bb_full_write(fd2, buffer, nread) < 0) {
|
|
||||||
bb_perror_msg(bb_msg_write_error); /* match Read error below */
|
bb_perror_msg(bb_msg_write_error); /* match Read error below */
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
if (chunksize && ((remaining -= nread) == 0)) {
|
|
||||||
return 0;
|
|
||||||
}
|
}
|
||||||
} else if (!nread) {
|
else if (read_actual == 0) {
|
||||||
if (chunksize) {
|
if (size) {
|
||||||
bb_error_msg("Unable to read all data");
|
bb_error_msg("Unable to read all data");
|
||||||
break;
|
|
||||||
}
|
}
|
||||||
return 0;
|
break;
|
||||||
} else { /* nread < 0 */
|
} else {
|
||||||
bb_perror_msg("Read error"); /* match bb_msg_write_error above */
|
/* read_actual < 0 */
|
||||||
|
bb_perror_msg("Read error");
|
||||||
break;
|
break;
|
||||||
}
|
}
|
||||||
|
|
||||||
} while (1);
|
read_total += read_actual;
|
||||||
|
}
|
||||||
|
|
||||||
return -1;
|
RELEASE_CONFIG_BUFFER(buffer);
|
||||||
|
|
||||||
|
return(read_total);
|
||||||
|
}
|
||||||
|
|
||||||
|
|
||||||
|
extern int bb_copyfd_size(int fd1, int fd2, const off_t size)
|
||||||
|
{
|
||||||
|
return(bb_full_fd_action(fd1, fd2, size, bb_full_write));
|
||||||
|
}
|
||||||
|
|
||||||
|
extern int bb_copyfd_eof(int fd1, int fd2)
|
||||||
|
{
|
||||||
|
return(bb_full_fd_action(fd1, fd2, 0, bb_full_write));
|
||||||
}
|
}
|
||||||
|
@ -28,7 +28,7 @@ extern void bb_xprint_and_close_file(FILE *file)
|
|||||||
bb_xfflush_stdout();
|
bb_xfflush_stdout();
|
||||||
/* Note: Do not use STDOUT_FILENO here, as this is a lib routine
|
/* Note: Do not use STDOUT_FILENO here, as this is a lib routine
|
||||||
* and the calling code may have reassigned stdout. */
|
* and the calling code may have reassigned stdout. */
|
||||||
if (bb_copyfd(fileno(file), fileno(stdout), 0) == -1) {
|
if (bb_copyfd_eof(fileno(file), fileno(stdout)) == -1) {
|
||||||
/* bb_copyfd outputs any needed messages, so just die. */
|
/* bb_copyfd outputs any needed messages, so just die. */
|
||||||
exit(bb_default_error_retval);
|
exit(bb_default_error_retval);
|
||||||
}
|
}
|
||||||
|
@ -210,7 +210,7 @@ static int ftp_recieve(ftp_host_info_t *server, FILE *control_stream,
|
|||||||
}
|
}
|
||||||
|
|
||||||
/* Copy the file */
|
/* Copy the file */
|
||||||
if (bb_copyfd(fd_data, fd_local, filesize) == -1) {
|
if (bb_copyfd_size(fd_data, fd_local, filesize) == -1) {
|
||||||
exit(EXIT_FAILURE);
|
exit(EXIT_FAILURE);
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -272,7 +272,7 @@ static int ftp_send(ftp_host_info_t *server, FILE *control_stream,
|
|||||||
}
|
}
|
||||||
|
|
||||||
/* transfer the file */
|
/* transfer the file */
|
||||||
if (bb_copyfd(fd_local, fd_data, 0) == -1) {
|
if (bb_copyfd_eof(fd_local, fd_data) == -1) {
|
||||||
exit(EXIT_FAILURE);
|
exit(EXIT_FAILURE);
|
||||||
}
|
}
|
||||||
|
|
||||||
|
Loading…
Reference in New Issue
Block a user