Don't delete source file when decompressing to stdout

[oweals/busybox.git] / archival / ar.c

diff --git a/archival/ar.c b/archival/ar.c
index 08cd5c50125cdd6649c8f338f61e48a0a6c812c8..ac8c41f966ca61ee2164ce3c7cf725c1d1a0b47c 100644 (file)
--- a/archival/ar.c
+++ b/archival/ar.c
@@ -23,43 +23,39 @@
  *
  */
 #include <fcntl.h>
+#include <stdio.h>
 #include <stdlib.h>
+#include <string.h>
 #include <getopt.h>
-#include <unistd.h>
+#include "unarchive.h"
 #include "busybox.h"
 
 extern int ar_main(int argc, char **argv)
 {
-       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 */
-
-       FILE *src_file = NULL, *dst_file = NULL;
-       int funct = 0, opt=0;
-
-       ar_headers_t *head, *ar_extract_list=NULL;
-
-       ar_extract_list = (ar_headers_t *) xcalloc(1, sizeof(ar_headers_t));
-       head = (ar_headers_t *) xcalloc(1, sizeof(ar_headers_t));
+       FILE *src_stream = NULL;
+       char **extract_names = NULL;
+       char ar_magic[8];
+       int extract_function =  extract_unconditional;
+       int opt;
+       int num_of_entries = 0;
+       extern off_t archive_offset;
 
        while ((opt = getopt(argc, argv, "ovtpx")) != -1) {
                switch (opt) {
                case 'o':
-                       funct |= preserve_date;
+                       extract_function |= extract_preserve_date;
                        break;
                case 'v':
-                       funct |= verbose;
+                       extract_function |= extract_verbose_list;
                        break;
                case 't':
-                       funct |= display;
+                       extract_function |= extract_list;
                        break;
                case 'p':
-                       funct |= extract_to_stdout;
+                       extract_function |= extract_to_stdout;
                        break;
                case 'x':
-                       funct |= extract_to_file;
+                       extract_function |= extract_all_to_fs;
                        break;
                default:
                        show_usage();
@@ -71,57 +67,24 @@ extern int ar_main(int argc, char **argv)
                show_usage();
        }
 
-       if ( (src_file = wfopen(argv[optind], "r")) < 0) {
-               error_msg_and_die("Cannot read %s", argv[optind]);
-       }
+       src_stream = xfopen(argv[optind++], "r");
 
-       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;
-               }
+       /* check ar magic */
+       fread(ar_magic, 1, 8, src_stream);
+       archive_offset = 8;
+       if (strncmp(ar_magic,"!<arch>",7) != 0) {
+               error_msg_and_die("invalid magic");
        }
 
-       /* if individual files not found extract all files */   
-       if (ar_extract_list->next==NULL) {
-               ar_extract_list = head;
+       /* Create a list of files to extract */
+       while (optind < argc) {
+               extract_names = xrealloc(extract_names, sizeof(char *) * (num_of_entries + 2));
+               extract_names[num_of_entries] = xstrdup(argv[optind]);
+               num_of_entries++;
+               extract_names[num_of_entries] = NULL;
+               optind++;
        }
 
-       /* 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)){
-                       puts(ar_extract_list->name);
-               }
-               ar_extract_list = ar_extract_list->next;
-       }
+       unarchive(src_stream, stdout, &get_header_ar, extract_function, "./", extract_names, NULL);
        return EXIT_SUCCESS;
 }