X-Git-Url: https://git.librecmc.org/?a=blobdiff_plain;f=ar.c;h=4c3d0bbd5110e6c5974fcb6b9f65a883c5e1b266;hb=9f0fedb8f693829e7869439a2ad43cc1c21a3786;hp=27dfadd1db62cb3a6bf426957c0e41297556d86f;hpb=57ebebfb01a9a29378b2f0179724661bfc5402e9;p=oweals%2Fbusybox.git diff --git a/ar.c b/ar.c index 27dfadd1d..4c3d0bbd5 100644 --- a/ar.c +++ b/ar.c @@ -3,8 +3,8 @@ * Mini ar implementation for busybox * * Copyright (C) 2000 by Glenn McGrath - * Written by Glenn McGrath 1 June 2000 - * + * Written by Glenn McGrath 1 June 2000 + * * Based in part on BusyBox tar, Debian dpkg-deb and GNU ar. * * This program is free software; you can redistribute it and/or modify @@ -21,312 +21,107 @@ * along with this program; if not, write to the Free Software * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA * - * Last modified 10 June 2000 */ - - -#include #include -#include -#include -#include -#include -#include -#include "internal.h" - -#define AR_BLOCK_SIZE 60 -#define AR_PRESERVE_DATE 1 /* preserve original dates */ -#define AR_VERBOSE 2 /* be verbose */ -#define AR_DISPLAY 4 /* display contents */ -#define AR_EXT_TO_FILE 8 /* extract contents of archive */ -#define AR_EXT_TO_STDOUT 16 /* extract to stdout */ - -#define BB_DECLARE_EXTERN -#define bb_need_io_error -#include "messages.c" - -struct ArHeader { /* Byte Offset */ - char ar_name[16]; /* 0-15 */ - char ar_date[12]; /* 16-27 */ - char ar_uid[6], ar_gid[6]; /* 28-39 */ - char ar_mode[8]; /* 40-47 */ - char ar_size[10]; /* 48-57 */ - char ar_fmag[2]; /* 58-59 */ -}; -typedef struct ArHeader ArHeader; - -struct ArInfo { - char name[17]; /* File name */ - time_t date; /* long int, No of seconds since epoch */ - uid_t uid; /* unsigned int, Numeric UID */ - gid_t gid; /* unsigned int, Numeric GID */ - mode_t mode; /* unsigned int, Unix mode */ - size_t size; /* int, Size of the file */ -}; -typedef struct ArInfo ArInfo; - -static const char ar_usage[] = "ar [optxvV] archive [filenames] \n" -#ifndef BB_FEATURE_TRIVIAL_HELP - "\nExtract or list files from an ar archive.\n\n" - "Options:\n" - "\to\t\tpreserve original dates\n" - "\tp\t\textract to stdout\n" - "\tt\t\tlist\n" - "\tx\t\textract\n" - "\tv\t\tverbosely list files processed\n" -#endif - ; - -/* - * Display details of a file, verbosly if funct=2 - */ -static void displayEntry(struct ArInfo *entry, int funct) -{ - /* TODO convert mode to string */ - if ((funct & AR_VERBOSE) == AR_VERBOSE) - printf("%i %i/%i %8i %s ", entry->mode, entry->uid, entry->gid, - entry->size, timeString(entry->date)); - printf("%s\n", entry->name); -} - -/* this is from tar.c remove later*/ -static long getOctal(const char *cp, int size) -{ - long val = 0; - - for(;(size > 0) && (*cp == ' '); cp++, size--); - if ((size == 0) || !isOctal(*cp)) - return -1; - for(; (size > 0) && isOctal(*cp); size--) { - val = val * 8 + *cp++ - '0'; - } - for (;(size > 0) && (*cp == ' '); cp++, size--); - if ((size > 0) && *cp) - return -1; - return val; -} - -/* - * Converts from the char based struct to a new struct with stricter types - */ -static int processArHeader(struct ArHeader *rawHeader, struct ArInfo *header) -{ - int count2; - int count; - - /* check end of header marker is valid */ - if ((rawHeader->ar_fmag[0]!='`') || (rawHeader->ar_fmag[1]!='\n')) - return(FALSE); - - /* convert filename */ - for (count = 0; count < 16; count++) { - /* allow spaces in filename except at the end */ - if (rawHeader->ar_name[count] == ' ') { - for (count2 = count; count2 < 16; count2++) - if (!isspace(rawHeader->ar_name[count2])) - break; - if (count2 >= 16) - break; - } - /* GNU ar uses '/' as an end of filename marker */ - if (rawHeader->ar_name[count] == '/') - break; - header->name[count] = rawHeader->ar_name[count]; - } - header->name[count] = '\0'; - header->date = atoi(rawHeader->ar_date); - header->uid = atoi(rawHeader->ar_uid); - header->gid = atoi(rawHeader->ar_gid); - header->mode = getOctal(rawHeader->ar_mode, sizeof(rawHeader->ar_mode)); - header->size = atoi(rawHeader->ar_size); - return (TRUE); -} - -/* - * Copy size bytes from current position if srcFd to current position in dstFd - * taken from tarExtractRegularFile in tar.c, remove later - */ -static int copySubFile(int srcFd, int dstFd, int copySize) -{ - int readSize, writeSize, doneSize; - char buffer[BUFSIZ]; - - while (copySize > 0) { - if (copySize > BUFSIZ) - readSize = BUFSIZ; - else - readSize = copySize; - writeSize = fullRead(srcFd, buffer, readSize); - if (writeSize <= 0) { - errorMsg(io_error, "copySubFile :", strerror(errno)); - return (FALSE); - } - doneSize = fullWrite(dstFd, buffer, writeSize); - if (doneSize <= 0) { - errorMsg(io_error, "copySubFile :", strerror(errno)); - return (FALSE); - } - copySize -= doneSize; - } - return (TRUE); -} - -/* - * Extract the file described in ArInfo to the specified path - * set the new files uid, gid and mode - */ -static int extractToFile(struct ArInfo *file, int funct, int srcFd, const char *path) -{ - int dstFd, temp; - struct stat tmpStat; - char *pathname = NULL; - struct utimbuf newtime; - - if ((temp = isDirectory(path, TRUE, &tmpStat)) != TRUE) { - if (!createPath(path, 0777)) { - fatalError("Cannot extract to specified path"); - return (FALSE); - } - } - temp = (strlen(path) + 16); - pathname = (char *) xmalloc(temp); - pathname = strcpy(pathname, path); - pathname = strcat(pathname, file->name); - dstFd = device_open(pathname, O_WRONLY | O_CREAT); - temp = copySubFile(srcFd, dstFd, file->size); - fchown(dstFd, file->uid, file->gid); - fchmod(dstFd, file->mode); - close(dstFd); - if ((funct&AR_PRESERVE_DATE)==AR_PRESERVE_DATE) - newtime.modtime=file->date; - else - newtime.modtime=time(0); - newtime.actime=time(0); - temp = utime(pathname, &newtime); - return (TRUE); -} - -/* - * Return a file descriptor for the specified file and do error checks - */ -static int getArFd(char *filename) -{ - int arFd; - char arVersion[8]; - - arFd = open(filename, O_RDONLY); - if (arFd < 0) { - errorMsg("Error opening '%s': %s\n", filename, strerror(errno)); - return (FALSE); - } - if (fullRead(arFd, arVersion, 8) <= 0) { - errorMsg( "ar: Unexpected EOF in archive\n"); - return (FALSE); - } - if (strncmp(arVersion,"!",7) != 0) { - errorMsg("ar header fails check "); - return(FALSE); - } - return arFd; -} +#include +#include +#include +#include "busybox.h" -/* - * Step through the ar file and process it one entry at a time - * fileList[0] is the name of the ar archive - * fileList[1] and up are filenames to extract from the archive - * funct contains flags to specify the actions to be performed - */ -static int readArFile(char *fileList[16], int fileListSize, int funct) +extern int ar_main(int argc, char **argv) { - int arFd, status, extFileFlag, i, lastOffset=0; - ArHeader rawArHeader; - ArInfo arEntry; - - /* open the ar archive */ - arFd=getArFd(fileList[0]); - - /* read the first header, then loop until ono more headers */ - while ((status = fullRead(arFd, (char *) &rawArHeader, AR_BLOCK_SIZE)) - == AR_BLOCK_SIZE) { + const int preserve_date = 1; /* preserve original dates */ + const int verbose = 2; /* be verbose */ + const int display = 4; /* display contents */ + const int extract_to_file = 8; /* extract contents of archive */ + const int extract_to_stdout = 16; /* extract to stdout */ - /* check the header is valid, if not try reading the header - agian with an offset of 1, needed as some ar archive end - with a '\n' which isnt counted in specified file size */ - if ((status=processArHeader(&rawArHeader, &arEntry))==FALSE ) { - if ((i=lseek(arFd, 0, SEEK_CUR))==(lastOffset+60)) - lseek(arFd, lastOffset+1, SEEK_SET); - else - return(FALSE); - } - else { - extFileFlag=0; - - if (funct&AR_DISPLAY) - displayEntry(&arEntry, funct); + FILE *src_file = NULL, *dst_file = NULL; + int funct = 0, opt=0; - /* check file was specified to be extracted only if - some file were specified */ - if ((funct&AR_EXT_TO_FILE) || (funct&AR_EXT_TO_STDOUT)){ - if (fileListSize==1) - extFileFlag=1; - else { - for( i=1; i<=fileListSize; i++) - if ((status=(strcmp(fileList[i],arEntry.name)))==0) - extFileFlag=1; - } - } - if (extFileFlag==1) { - if (funct&AR_EXT_TO_FILE) - extractToFile(&arEntry, funct, arFd, "./"); - else - copySubFile(arFd,fileno(stdout),arEntry.size); - } - else - lseek(arFd, arEntry.size, SEEK_CUR); - lastOffset=lseek(arFd, 0, SEEK_CUR); - } /* if processArHeader */ - } /* while */ - return (TRUE); -} - -extern int ar_main(int argc, char **argv) -{ - int funct = 0, ret=0, i=0; - char *fileList[16], c, *opt_ptr; + ar_headers_t *head, *ar_extract_list=NULL; - if (argc < 2) - usage(ar_usage); + ar_extract_list = (ar_headers_t *) xcalloc(1, sizeof(ar_headers_t)); + head = (ar_headers_t *) xcalloc(1, sizeof(ar_headers_t)); - opt_ptr = argv[1]; - if (*opt_ptr == '-') - ++opt_ptr; - while ((c = *opt_ptr++) != '\0') { - switch (c) { + while ((opt = getopt(argc, argv, "ovtpx")) != -1) { + switch (opt) { case 'o': - funct = funct | AR_PRESERVE_DATE; + funct |= preserve_date; break; case 'v': - funct = funct | AR_VERBOSE; + funct |= verbose; break; case 't': - funct = funct | AR_DISPLAY; - break; - case 'x': - funct = funct | AR_EXT_TO_FILE; + funct |= display; break; case 'p': - funct = funct | AR_EXT_TO_STDOUT; + funct |= extract_to_stdout; + break; + case 'x': + funct |= extract_to_file; break; default: - usage(ar_usage); + show_usage(); } } + + /* check the src filename was specified */ + if (optind == argc) { + show_usage(); + } - for(i=0; i<(argc-2); i++) - fileList[i]=argv[i+2]; - - if (funct > 3) - ret = readArFile(fileList, (argc-2), funct); - - return (ret); + if ( (src_file = wfopen(argv[optind], "r")) < 0) { + error_msg_and_die("Cannot read %s", argv[optind]); + } + + optind++; + head = get_ar_headers(src_file); + /* find files to extract or display */ + /* search through argv and build extract list */ + for (;optind < argc; optind++) { + ar_headers_t *ar_entry; + ar_entry = (ar_headers_t *) xcalloc(1, sizeof(ar_headers_t)); + ar_entry = head; + while (ar_entry->next != NULL) { + if (strcmp(argv[optind], ar_entry->name) == 0) { + ar_headers_t *tmp; + tmp = (ar_headers_t *) xmalloc(sizeof(ar_headers_t)); + *tmp = *ar_extract_list; + *ar_extract_list = *ar_entry; + ar_extract_list->next = tmp; + break; + } + ar_entry=ar_entry->next; + } + } + + /* if individual files not found extract all files */ + if (ar_extract_list->next==NULL) { + ar_extract_list = head; + } + + /* find files to extract or display */ + while (ar_extract_list->next != NULL) { + if (funct & extract_to_file) { + dst_file = wfopen(ar_extract_list->name, "w"); + } + else if (funct & extract_to_stdout) { + dst_file = stdout; + } + if ((funct & extract_to_file) || (funct & extract_to_stdout)) { + fseek(src_file, ar_extract_list->offset, SEEK_SET); + copy_file_chunk(src_file, dst_file, ar_extract_list->size); + } + if (funct & verbose) { + printf("%s %d/%d %8d %s ", mode_string(ar_extract_list->mode), + ar_extract_list->uid, ar_extract_list->gid, + (int) ar_extract_list->size, time_string(ar_extract_list->mtime)); + } + if ((funct & display) || (funct & verbose)){ + printf("%s\n", ar_extract_list->name); + } + ar_extract_list = ar_extract_list->next; + } + return EXIT_SUCCESS; }