Fix compress support and prevent a segfault

This commit is contained in:
Glenn L McGrath 2002-09-27 06:46:02 +00:00
parent a47a3eada6
commit 2e41d0cb77
12 changed files with 71 additions and 66 deletions

View File

@ -163,11 +163,25 @@ extern int gunzip_main(int argc, char **argv)
} }
/* do the decompression, and cleanup */ /* do the decompression, and cleanup */
if (xread_char(src_fd) == 0x1f) {
unsigned char magic2;
magic2 = xread_char(src_fd);
#ifdef CONFIG_FEATURE_UNCOMPRESS
if (magic2 == 0x9d) {
return(uncompress(src_fd, dst_fd));
} else
#endif
if (magic2 == 0x8b) {
check_header_gzip(src_fd); check_header_gzip(src_fd);
if (inflate(src_fd, dst_fd) != 0) { if (inflate(src_fd, dst_fd) != 0) {
error_msg("Error inflating"); error_msg("Error inflating");
} }
check_trailer_gzip(src_fd); check_trailer_gzip(src_fd);
} else {
error_msg_and_die("Invalid magic\n");
}
}
if ((status != EXIT_SUCCESS) && (new_path)) { if ((status != EXIT_SUCCESS) && (new_path)) {
/* Unzip failed, remove new path instead of old path */ /* Unzip failed, remove new path instead of old path */

View File

@ -5,9 +5,8 @@
extern void check_header_gzip(int src_fd) extern void check_header_gzip(int src_fd)
{ {
union { union {
unsigned char raw[10]; unsigned char raw[8];
struct { struct {
unsigned char magic[2];
unsigned char method; unsigned char method;
unsigned char flags; unsigned char flags;
unsigned int mtime; unsigned int mtime;
@ -16,13 +15,7 @@ extern void check_header_gzip(int src_fd)
} formated; } formated;
} header; } header;
xread_all(src_fd, header.raw, 10); xread_all(src_fd, header.raw, 8);
/* Magic header for gzip files, 1F 8B = \037\213 */
if ((header.formated.magic[0] != 0x1F)
|| (header.formated.magic[1] != 0x8b)) {
error_msg_and_die("Invalid gzip magic");
}
/* Check the compression method */ /* Check the compression method */
if (header.formated.method != 8) { if (header.formated.method != 8) {

View File

@ -18,7 +18,6 @@ extern void data_extract_all(archive_handle_t *archive_handle)
if (archive_handle->flags & ARCHIVE_CREATE_LEADING_DIRS) { if (archive_handle->flags & ARCHIVE_CREATE_LEADING_DIRS) {
char *dir = dirname(strdup(file_header->name)); char *dir = dirname(strdup(file_header->name));
make_directory (dir, 0777, FILEUTILS_RECUR); make_directory (dir, 0777, FILEUTILS_RECUR);
free(dir);
} }
/* Create the file */ /* Create the file */

View File

@ -29,7 +29,8 @@
* *
*/ */
#include <stdio.h> #include <stdio.h>
#include <string.h>
#include <unistd.h>
#define IBUFSIZ 2048 /* Defailt input buffer size */ #define IBUFSIZ 2048 /* Defailt input buffer size */
#define OBUFSIZ 2048 /* Default output buffer size */ #define OBUFSIZ 2048 /* Default output buffer size */
@ -95,9 +96,6 @@ unsigned short codetab[HSIZE];
#define clear_tab_prefixof() memset(codetab, 0, 256); #define clear_tab_prefixof() memset(codetab, 0, 256);
extern int uncompress ( FILE *, FILE * );
/* /*
* Decompress stdin to stdout. This routine adapts to the codes in the * Decompress stdin to stdout. This routine adapts to the codes in the
* file building the "string" table on-the-fly; requiring no table to * file building the "string" table on-the-fly; requiring no table to
@ -105,7 +103,7 @@ extern int uncompress ( FILE *, FILE * );
* with those of the compress() routine. See the definitions above. * with those of the compress() routine. See the definitions above.
*/ */
int uncompress ( FILE * fdin, FILE * fdout ) extern int uncompress(int fd_in, int fd_out)
{ {
char_type *stackp; char_type *stackp;
code_int code; code_int code;
@ -125,7 +123,7 @@ int uncompress ( FILE * fdin, FILE * fdout )
insize = 0; insize = 0;
inbuf [0] = fgetc(fdin); inbuf [0] = xread_char(fd_in);
maxbits = inbuf[0] & BIT_MASK; maxbits = inbuf[0] & BIT_MASK;
block_mode = inbuf[0] & BLOCK_MODE; block_mode = inbuf[0] & BLOCK_MODE;
@ -173,11 +171,7 @@ resetbuf: ;
if (insize < (int) sizeof(inbuf)-IBUFSIZ) if (insize < (int) sizeof(inbuf)-IBUFSIZ)
{ {
rsize = fread(inbuf+insize, 1,IBUFSIZ,fdin); xread_all(fd_in, inbuf+insize, IBUFSIZ);
if ( !rsize && ferror(fdin))
return -1;
insize += rsize; insize += rsize;
} }
@ -275,8 +269,7 @@ resetbuf: ;
if (outpos >= OBUFSIZ) if (outpos >= OBUFSIZ)
{ {
fwrite(outbuf, 1,outpos,fdout); write(fd_out, outbuf, outpos);
outpos = 0; outpos = 0;
} }
stackp+= i; stackp+= i;
@ -303,8 +296,9 @@ resetbuf: ;
} }
while (rsize > 0); while (rsize > 0);
if (outpos > 0) if (outpos > 0) {
fwrite(outbuf, outpos,1, fdout); write(fd_out, outbuf, outpos);
}
return 0; return 0;
} }

View File

@ -6,5 +6,9 @@
*/ */
extern char filter_accept_all(const llist_t *accept_list, const llist_t *reject_list, const char *key) extern char filter_accept_all(const llist_t *accept_list, const llist_t *reject_list, const char *key)
{ {
if (key) {
return(EXIT_SUCCESS); return(EXIT_SUCCESS);
} else {
return(EXIT_FAILURE);
}
} }

View File

@ -155,7 +155,6 @@ extern char get_header_tar(archive_handle_t *archive_handle)
# endif # endif
} }
#endif #endif
if (archive_handle->filter(archive_handle->accept, archive_handle->reject, archive_handle->file_header->name) == EXIT_SUCCESS) { if (archive_handle->filter(archive_handle->accept, archive_handle->reject, archive_handle->file_header->name) == EXIT_SUCCESS) {
archive_handle->action_header(archive_handle->file_header); archive_handle->action_header(archive_handle->file_header);
archive_handle->flags |= ARCHIVE_EXTRACT_QUIET; archive_handle->flags |= ARCHIVE_EXTRACT_QUIET;

View File

@ -27,11 +27,17 @@ extern char get_header_tar_gz(archive_handle_t *archive_handle)
{ {
int fd_pipe[2]; int fd_pipe[2];
int pid; int pid;
unsigned char magic[2];
xread_all(archive_handle->src_fd, &magic, 2);
if ((magic[0] != 0x1f) || (magic[1] != 0x8b)) {
error_msg_and_die("Invalid gzip magic");
}
check_header_gzip(archive_handle->src_fd); check_header_gzip(archive_handle->src_fd);
if (pipe(fd_pipe) != 0) { if (pipe(fd_pipe) != 0) {
error_msg_and_die("Can't create pipe\n"); error_msg_and_die("Can't create pipe");
} }
pid = fork(); pid = fork();
@ -54,6 +60,7 @@ extern char get_header_tar_gz(archive_handle_t *archive_handle)
archive_handle->src_fd = fd_pipe[0]; archive_handle->src_fd = fd_pipe[0];
archive_handle->offset = 0;
while (get_header_tar(archive_handle) == EXIT_SUCCESS); while (get_header_tar(archive_handle) == EXIT_SUCCESS);
if (kill(pid, SIGTERM) == -1) { if (kill(pid, SIGTERM) == -1) {

View File

@ -29,7 +29,8 @@
* *
*/ */
#include <stdio.h> #include <stdio.h>
#include <string.h>
#include <unistd.h>
#define IBUFSIZ 2048 /* Defailt input buffer size */ #define IBUFSIZ 2048 /* Defailt input buffer size */
#define OBUFSIZ 2048 /* Default output buffer size */ #define OBUFSIZ 2048 /* Default output buffer size */
@ -95,9 +96,6 @@ unsigned short codetab[HSIZE];
#define clear_tab_prefixof() memset(codetab, 0, 256); #define clear_tab_prefixof() memset(codetab, 0, 256);
extern int uncompress ( FILE *, FILE * );
/* /*
* Decompress stdin to stdout. This routine adapts to the codes in the * Decompress stdin to stdout. This routine adapts to the codes in the
* file building the "string" table on-the-fly; requiring no table to * file building the "string" table on-the-fly; requiring no table to
@ -105,7 +103,7 @@ extern int uncompress ( FILE *, FILE * );
* with those of the compress() routine. See the definitions above. * with those of the compress() routine. See the definitions above.
*/ */
int uncompress ( FILE * fdin, FILE * fdout ) extern int uncompress(int fd_in, int fd_out)
{ {
char_type *stackp; char_type *stackp;
code_int code; code_int code;
@ -125,7 +123,7 @@ int uncompress ( FILE * fdin, FILE * fdout )
insize = 0; insize = 0;
inbuf [0] = fgetc(fdin); inbuf [0] = xread_char(fd_in);
maxbits = inbuf[0] & BIT_MASK; maxbits = inbuf[0] & BIT_MASK;
block_mode = inbuf[0] & BLOCK_MODE; block_mode = inbuf[0] & BLOCK_MODE;
@ -173,11 +171,7 @@ resetbuf: ;
if (insize < (int) sizeof(inbuf)-IBUFSIZ) if (insize < (int) sizeof(inbuf)-IBUFSIZ)
{ {
rsize = fread(inbuf+insize, 1,IBUFSIZ,fdin); xread_all(fd_in, inbuf+insize, IBUFSIZ);
if ( !rsize && ferror(fdin))
return -1;
insize += rsize; insize += rsize;
} }
@ -275,8 +269,7 @@ resetbuf: ;
if (outpos >= OBUFSIZ) if (outpos >= OBUFSIZ)
{ {
fwrite(outbuf, 1,outpos,fdout); write(fd_out, outbuf, outpos);
outpos = 0; outpos = 0;
} }
stackp+= i; stackp+= i;
@ -303,8 +296,9 @@ resetbuf: ;
} }
while (rsize > 0); while (rsize > 0);
if (outpos > 0) if (outpos > 0) {
fwrite(outbuf, outpos,1, fdout); write(fd_out, outbuf, outpos);
}
return 0; return 0;
} }

View File

@ -70,6 +70,7 @@ extern int rpm2cpio_main(int argc, char **argv)
{ {
struct rpm_lead lead; struct rpm_lead lead;
int rpm_fd; int rpm_fd;
unsigned char magic[2];
if (argc == 1) { if (argc == 1) {
rpm_fd = fileno(stdin); rpm_fd = fileno(stdin);
@ -89,6 +90,11 @@ extern int rpm2cpio_main(int argc, char **argv)
/* Skip the main header */ /* Skip the main header */
skip_header(rpm_fd); skip_header(rpm_fd);
xread_all(rpm_fd, &magic, 2);
if ((magic[0] != 0x1f) || (magic[1] != 0x8b))
error_msg_and_die("Invalid gzip magic");
}
check_header_gzip(rpm_fd); check_header_gzip(rpm_fd);
if (inflate(rpm_fd, fileno(stdout)) != 0) { if (inflate(rpm_fd, fileno(stdout)) != 0) {
error_msg("Error inflating"); error_msg("Error inflating");

View File

@ -226,6 +226,7 @@ extern long arith (const char *startbuf, int *errcode);
int read_package_field(const char *package_buffer, char **field_name, char **field_value); int read_package_field(const char *package_buffer, char **field_name, char **field_value);
char *fgets_str(FILE *file, const char *terminating_string); char *fgets_str(FILE *file, const char *terminating_string);
extern int uncompress(int fd_in, int fd_out);
extern int inflate(int in, int out); extern int inflate(int in, int out);
extern struct hostent *xgethostbyname(const char *name); extern struct hostent *xgethostbyname(const char *name);

View File

@ -85,7 +85,7 @@ extern char get_header_ar(archive_handle_t *archive_handle);
extern char get_header_tar(archive_handle_t *archive_handle); extern char get_header_tar(archive_handle_t *archive_handle);
extern char get_header_tar_gz(archive_handle_t *archive_handle); extern char get_header_tar_gz(archive_handle_t *archive_handle);
//extern void seek_sub_file(int src_fd, unsigned int amount); extern void seek_sub_file(int src_fd, unsigned int amount);
extern const unsigned short data_align(const int src_fd, const unsigned int offset, const unsigned short align_to); extern const unsigned short data_align(const int src_fd, const unsigned int offset, const unsigned short align_to);
extern const llist_t *add_to_list(const llist_t *old_head, const char *new_item); extern const llist_t *add_to_list(const llist_t *old_head, const char *new_item);
extern int copy_file_chunk_fd(int src_fd, int dst_fd, unsigned long long chunksize); extern int copy_file_chunk_fd(int src_fd, int dst_fd, unsigned long long chunksize);

View File

@ -29,7 +29,8 @@
* *
*/ */
#include <stdio.h> #include <stdio.h>
#include <string.h>
#include <unistd.h>
#define IBUFSIZ 2048 /* Defailt input buffer size */ #define IBUFSIZ 2048 /* Defailt input buffer size */
#define OBUFSIZ 2048 /* Default output buffer size */ #define OBUFSIZ 2048 /* Default output buffer size */
@ -95,9 +96,6 @@ unsigned short codetab[HSIZE];
#define clear_tab_prefixof() memset(codetab, 0, 256); #define clear_tab_prefixof() memset(codetab, 0, 256);
extern int uncompress ( FILE *, FILE * );
/* /*
* Decompress stdin to stdout. This routine adapts to the codes in the * Decompress stdin to stdout. This routine adapts to the codes in the
* file building the "string" table on-the-fly; requiring no table to * file building the "string" table on-the-fly; requiring no table to
@ -105,7 +103,7 @@ extern int uncompress ( FILE *, FILE * );
* with those of the compress() routine. See the definitions above. * with those of the compress() routine. See the definitions above.
*/ */
int uncompress ( FILE * fdin, FILE * fdout ) extern int uncompress(int fd_in, int fd_out)
{ {
char_type *stackp; char_type *stackp;
code_int code; code_int code;
@ -125,7 +123,7 @@ int uncompress ( FILE * fdin, FILE * fdout )
insize = 0; insize = 0;
inbuf [0] = fgetc(fdin); inbuf [0] = xread_char(fd_in);
maxbits = inbuf[0] & BIT_MASK; maxbits = inbuf[0] & BIT_MASK;
block_mode = inbuf[0] & BLOCK_MODE; block_mode = inbuf[0] & BLOCK_MODE;
@ -173,11 +171,7 @@ resetbuf: ;
if (insize < (int) sizeof(inbuf)-IBUFSIZ) if (insize < (int) sizeof(inbuf)-IBUFSIZ)
{ {
rsize = fread(inbuf+insize, 1,IBUFSIZ,fdin); xread_all(fd_in, inbuf+insize, IBUFSIZ);
if ( !rsize && ferror(fdin))
return -1;
insize += rsize; insize += rsize;
} }
@ -275,8 +269,7 @@ resetbuf: ;
if (outpos >= OBUFSIZ) if (outpos >= OBUFSIZ)
{ {
fwrite(outbuf, 1,outpos,fdout); write(fd_out, outbuf, outpos);
outpos = 0; outpos = 0;
} }
stackp+= i; stackp+= i;
@ -303,8 +296,9 @@ resetbuf: ;
} }
while (rsize > 0); while (rsize > 0);
if (outpos > 0) if (outpos > 0) {
fwrite(outbuf, outpos,1, fdout); write(fd_out, outbuf, outpos);
}
return 0; return 0;
} }