Vladimir N. Oleynik writes:
[oweals/busybox.git] / archival / ar.c
index 8ab38682139112fc3c89d06eece9987744e53593..57ec927191c229ec8d937a3c67dcb342cb7ba70a 100644 (file)
  * along with this program; if not, write to the Free Software
  * Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
  *
+ * There is no signle standard to adhere to so ar may not portable
+ * between different systems
+ * http://www.unix-systems.org/single_unix_specification_v2/xcu/ar.html
  */
+#include <sys/types.h>
 #include <fcntl.h>
+#include <fnmatch.h>
+#include <stdio.h>
 #include <stdlib.h>
-#include <getopt.h>
+#include <string.h>
+#include <time.h>
+#include <utime.h>
 #include <unistd.h>
+#include <getopt.h>
+#include "unarchive.h"
 #include "busybox.h"
 
-extern int ar_main(int argc, char **argv)
+static void header_verbose_list_ar(const file_header_t *file_header)
+{
+       const char *mode = bb_mode_string(file_header->mode);
+       char *mtime;
+
+       mtime = ctime(&file_header->mtime);
+       mtime[16] = ' ';
+       memmove(&mtime[17], &mtime[20], 4);
+       mtime[21] = '\0';
+       printf("%s %d/%d%7d %s %s\n", &mode[1], file_header->uid, file_header->gid, (int) file_header->size, &mtime[4], file_header->name);
+}
+
+#if !defined CONFIG_TAR && !defined CONFIG_DPKG_DEB && !defined CONFIG_CPIO
+/* This is simpler than data_extract_all */
+static void data_extract_regular_file(archive_handle_t *archive_handle)
 {
-       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_header_t *file_header;
+       int dst_fd;
+
+       file_header = archive_handle->file_header;
+       dst_fd = bb_xopen(file_header->name, O_WRONLY | O_CREAT);
+       archive_copy_file(archive_handle, dst_fd);      
+       close(dst_fd);
 
-       FILE *src_file = NULL, *dst_file = NULL;
-       int funct = 0, opt=0;
+       chmod(file_header->name, file_header->mode);
+       chown(file_header->name, file_header->uid, file_header->gid);
 
-       ar_headers_t *head, *extract_list=NULL;
+       if (archive_handle->flags & ARCHIVE_PRESERVE_DATE) {
+               struct utimbuf t;
+               t.actime = t.modtime = file_header->mtime;
+               utime(file_header->name, &t);
+       }
 
-       extract_list = (ar_headers_t *) xcalloc(1, sizeof(ar_headers_t));
-       head = (ar_headers_t *) xcalloc(1, sizeof(ar_headers_t));
+       return;
+}
+#endif
 
-       while ((opt = getopt(argc, argv, "ovtpx")) != -1) {
+extern int ar_main(int argc, char **argv)
+{
+       archive_handle_t *archive_handle;
+       int opt;
+
+#if !defined CONFIG_DPKG_DEB && !defined CONFIG_DPKG
+       char magic[8];
+#endif
+       archive_handle = init_handle();
+
+       while ((opt = getopt(argc, argv, "covtpxX")) != -1) {
                switch (opt) {
-               case 'o':
-                       funct |= preserve_date;
+               case 'p':       /* print */
+                       archive_handle->action_data = data_extract_to_stdout;
                        break;
-               case 'v':
-                       funct |= verbose;
+               case 't':       /* list contents */
+                       archive_handle->action_header = header_list;
                        break;
-               case 't':
-                       funct |= display;
+               case 'X':
+                       archive_handle->action_header = header_verbose_list_ar;
+               case 'x':       /* extract */
+#if defined CONFIG_TAR || defined CONFIG_DPKG_DEB || defined CONFIG_CPIO
+                       archive_handle->action_data = data_extract_all;
+#else
+                       archive_handle->action_data = data_extract_regular_file;
+#endif
                        break;
-               case 'p':
-                       funct |= extract_to_stdout;
+               /* Modifiers */
+               case 'o':       /* preserve original dates */
+                       archive_handle->flags |= ARCHIVE_PRESERVE_DATE;
                        break;
-               case 'x':
-                       funct |= extract_to_file;
+               case 'v':       /* verbose */
+                       archive_handle->action_header = header_verbose_list_ar;
                        break;
                default:
-                       show_usage();
+                       bb_show_usage();
                }
        }
  
        /* check the src filename was specified */
        if (optind == argc) {
-               show_usage();
+               bb_show_usage();
        }
 
-       if ( (src_file = wfopen(argv[optind], "r")) < 0) {
-               error_msg_and_die("Cannot read %s", argv[optind]);
-       }
+       archive_handle->src_fd = bb_xopen(argv[optind++], O_RDONLY);
 
-       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 = *extract_list;
-                               *extract_list = *ar_entry;
-                               extract_list->next = tmp;
-                               break;                                  
-                       }
-                       ar_entry=ar_entry->next;
-               }
+       /* TODO: This is the same as in tar, seperate function ? */
+       while (optind < argc) {
+               archive_handle->filter = filter_accept_list;
+               archive_handle->accept = llist_add_to(archive_handle->accept, argv[optind]);
+               optind++;
        }
 
-       /* if individual files not found extract all files */   
-       if (extract_list->next==NULL) {
-               extract_list = head;
+#if defined CONFIG_DPKG_DEB || defined CONFIG_DPKG
+       unpack_ar_archive(archive_handle);
+#else
+       archive_xread_all(archive_handle, magic, 7);
+       if (strncmp(magic, "!<arch>", 7) != 0) {
+               bb_error_msg_and_die("Invalid ar magic");
        }
+       archive_handle->offset += 7;
+
+       while (get_header_ar(archive_handle) == EXIT_SUCCESS);
+#endif
 
-       /* find files to extract or display */  
-       while (extract_list->next != NULL) {
-               if (funct & extract_to_file) {
-                       dst_file = wfopen(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, extract_list->offset, SEEK_SET);
-                       copy_file_chunk(src_file, dst_file, (off_t) extract_list->size);                        
-               }
-               if (funct & verbose) {
-                       printf("%s %d/%d %8d %s ", mode_string(extract_list->mode), 
-                               extract_list->uid, extract_list->gid,
-                               (int) extract_list->size, time_string(extract_list->mtime));
-               }
-               if ((funct & display) || (funct & verbose)){
-                       printf("%s\n", extract_list->name);
-               }
-               extract_list=extract_list->next;
-       }
        return EXIT_SUCCESS;
 }