On Friday 12 March 2010 13:49, Alexander Shishkin wrote:
> This implements archive creation in busybox ar. Long file names
> support is to be implemented yet.
> 
>  int ar_main(int argc UNUSED_PARAM, char **argv)
>  {
> -     static const char msg_unsupported_err[] ALIGN1 =
> -             "archive %s is not supported";
> -
>       archive_handle_t *archive_handle;
>       unsigned opt;
> +     bool wr = 0;

redundant init

>       archive_handle = init_handle();
>  
>       /* Prepend '-' to the first argument if required */
> -     opt_complementary = "--:p:t:x:-1:p--tx:t--px:x--pt";
> +     opt_complementary = "--:p:t:x:-1:p--tx:t--px:x--pt:c:r";

Why the options are not optional?
Can c really be specified without r?
manpage says -c it isn't a command, it's a modifier
to r command. I'm confused. There should be a comment about it.

> --- /dev/null
> +++ b/archival/ar.h
> @@ -0,0 +1,27 @@
> +/*
> + * busybox ar archive data structures
> + * Licensed under the GPL v2 or later, see the file LICENSE in this source 
> tree.
> + */
> +
> +#ifndef AR_H
> +#define AR_H

missing PUSH_AND_SET_FUNCTION_VISIBILITY_TO_HIDDEN

> +struct ar_header {
> +     char name[16];
> +     char date[12];
> +     char uid[6];
> +     char gid[6];
> +     char mode[8];
> +     char size[10];
> +     char magic[2];
> +};
> +
> +#define AR_HEADER_LEN sizeof(struct ar_header)
> +#define AR_MAGIC "!<arch>"
> +#define AR_MAGIC_LEN 7
> +
> +#if ENABLE_FEATURE_AR_CREATE
> +int write_ar_archive(archive_handle_t *handle, int append);

missing FAST_FUNC

> +     dprintf(handle->src_fd, "%-16s%-12lu%-6u%-6u%-8o%-10llu`\n",
> +              fh->name, fh->mtime, fh->uid, fh->gid, fh->mode, fh->size);
> +     handle->offset += AR_HEADER_LEN;

Why do you think fh->size can always be printf'ed with %llu?
What if some field (e.g. uid = 999999999) overflowed -
will handle->offset be wrong?


> +        * create a new handle for output, which is a new file created
> +        * in place of the old one
> +        */
> +       if (append && st.st_size) {
> +               out_handle = init_handle();
> +               unlink(handle->ar_name);
> +               out_handle->src_fd = xopen(handle->ar_name, O_WRONLY | 
> O_CREAT);

Forgotten O_TRUNC?


> +#if ENABLE_FEATURE_AR_LONG_FILENAMES
> +#error nope, not yet
> +#else
> +     snprintf(fn_h, 17, "%.15s/", bb_basename(fn));

looks like sprintf will be safe too

> --- a/archival/libunarchive/get_header_ar.c
> +++ b/archival/libunarchive/get_header_ar.c
> @@ -6,6 +6,7 @@
>  
>  #include "libbb.h"
>  #include "unarchive.h"
> +#include "../ar.h"

No other .h file requires such ugly #include.

> --- a/include/unarchive.h
> +++ b/include/unarchive.h
> @@ -73,6 +73,10 @@ typedef struct archive_handle_t {
>       /* Contains the handle to a sub archive */
>       struct archive_handle_t *dpkg__sub_archive;
>  #endif
> +#if ENABLE_FEATURE_AR_CREATE
> +     const char *ar_name;
> +     struct archive_handle_t *ar_out;

The other members use <applet>__name, follow the example.


+testing "ar creates archives" \
+       "ar c test.a README && ar p test.a README | md5sum | cut -f1 -d' '" \
+       "$(md5sum README | cut -f1 -d' ')\n" \
+       "" \
+       ""

Just for fun, rm test.a before tests too.
"ar c ARCHIVE FILE" is not a valid syntax.

Please see the patch below where some of those are fixed.
Please fix or comment about field overflow and send
the updated patch back.
-- 
vda











diff -ad -urpN busybox.5/archival/ar.c busybox.6/archival/ar.c
--- busybox.5/archival/ar.c     2010-03-06 20:30:44.000000000 +0100
+++ busybox.6/archival/ar.c     2010-03-15 04:38:23.000000000 +0100
@@ -15,6 +15,7 @@
 
 #include "libbb.h"
 #include "unarchive.h"
+#include "ar.h"
 
 static void FAST_FUNC header_verbose_list_ar(const file_header_t *file_header)
 {
@@ -25,34 +26,41 @@ static void FAST_FUNC header_verbose_lis
        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);
+       printf("%s %u/%u%7"OFF_FMT"u %s %s\n", &mode[1],
+                       (int)file_header->uid, (int)file_header->gid,
+                       file_header->size,
+                       &mtime[4], file_header->name
+       );
 }
 
-#define AR_CTX_PRINT           0x01
-#define AR_CTX_LIST            0x02
-#define AR_CTX_EXTRACT         0x04
-#define AR_OPT_PRESERVE_DATE   0x08
-#define AR_OPT_VERBOSE         0x10
-#define AR_OPT_CREATE          0x20
-#define AR_OPT_INSERT          0x40
+#define AR_OPT_VERBOSE         (1 << 0)
+#define AR_OPT_PRESERVE_DATE   (1 << 1)
+#define AR_CTX_PRINT           (1 << 2)
+#define AR_CTX_LIST            (1 << 3)
+#define AR_CTX_EXTRACT         (1 << 4)
+#define AR_OPT_INSERT          (1 << 5)
+/* insert implies create, but warns about it. -c suppresses warning */
+#define AR_OPT_CREATE          (1 << 6)
 
 int ar_main(int argc, char **argv) MAIN_EXTERNALLY_VISIBLE;
 int ar_main(int argc UNUSED_PARAM, char **argv)
 {
-       static const char msg_unsupported_err[] ALIGN1 =
-               "archive %s is not supported";
-
        archive_handle_t *archive_handle;
-       unsigned opt;
+       unsigned opt, t;
 
        archive_handle = init_handle();
 
-       /* Prepend '-' to the first argument if required */
-       opt_complementary = "--:p:t:x:-1:p--tx:t--px:x--pt";
-       opt = getopt32(argv, "ptxovcr");
+       /* --: prepend '-' to the first argument if required */
+       /* -1: at least one param is reqd */
+       /* one of p,t,x[,r] is required */
+       opt_complementary = "--:-1:p:t:x"IF_FEATURE_AR_CREATE(":r");
+       opt = getopt32(argv, "voptx"IF_FEATURE_AR_CREATE("rc"));
        argv += optind;
 
+       t = (opt / AR_CTX_PRINT) & 0xf;
+       if (t & (t-1)) /* more than one of p,t,x[,r] are specified */
+               bb_show_usage();
+
        if (opt & AR_CTX_PRINT) {
                archive_handle->action_data = data_extract_to_stdout;
        }
@@ -68,20 +76,25 @@ int ar_main(int argc UNUSED_PARAM, char 
        if (opt & AR_OPT_VERBOSE) {
                archive_handle->action_header = header_verbose_list_ar;
        }
-       if (opt & AR_OPT_CREATE) {
-               bb_error_msg_and_die(msg_unsupported_err, "creation");
-       }
-       if (opt & AR_OPT_INSERT) {
-               bb_error_msg_and_die(msg_unsupported_err, "insertion");
-       }
-
-       archive_handle->src_fd = xopen(*argv++, O_RDONLY);
+#if ENABLE_FEATURE_AR_CREATE
+       archive_handle->ar__name = *argv;
+#endif
+       archive_handle->src_fd = xopen(*argv++,
+                       (opt & (AR_OPT_CREATE | AR_OPT_INSERT))
+                               ? O_RDWR | O_CREAT
+                               : O_RDONLY
+       );
 
        while (*argv) {
                archive_handle->filter = filter_accept_list;
                llist_add_to(&archive_handle->accept, *argv++);
        }
 
+#if ENABLE_FEATURE_AR_CREATE
+       if (opt & (AR_OPT_CREATE | AR_OPT_INSERT))
+               return write_ar_archive(archive_handle, opt & AR_OPT_INSERT);
+#endif
+
        unpack_ar_archive(archive_handle);
 
        return EXIT_SUCCESS;
diff -ad -urpN busybox.5/archival/ar_create.c busybox.6/archival/ar_create.c
--- busybox.5/archival/ar_create.c      1970-01-01 01:00:00.000000000 +0100
+++ busybox.6/archival/ar_create.c      2010-03-15 04:58:42.000000000 +0100
@@ -0,0 +1,135 @@
+/*
+ * ar archive creation support
+ *
+ * Copyright (C) 2010 Nokia Corporation. All rights reserved.
+ * Written by Alexander Shishkin.
+ *
+ * Licensed under GPLv2 or later, see file LICENSE in the source tree for 
details.
+ */
+
+#include "libbb.h"
+#include "unarchive.h"
+#include "ar.h"
+
+/* filter out entries with same names as specified on the command line */
+static char FAST_FUNC filter_replaceable(archive_handle_t *handle)
+{
+       if (find_list_entry(handle->accept, handle->file_header->name))
+               return EXIT_FAILURE;
+
+       return EXIT_SUCCESS;
+}
+
+static void FAST_FUNC output_ar_header(archive_handle_t *handle)
+{
+       struct file_header_t *fh = handle->file_header;
+
+       if (handle->offset & 1) {
+               xwrite(handle->src_fd, "\n", 1);
+               handle->offset++;
+       }
+
+       fdprintf(handle->src_fd, "%-16s%-12lu%-6u%-6u%-8o%-10"OFF_FMT"u`\n",
+                               fh->name, (long)fh->mtime,
+                               (int)fh->uid, (int)fh->gid,
+                               (int)fh->mode, fh->size
+       );
+//TODO: What if some field (e.g. uid = 999999999) overflowed - will 
handle->offset be wrong?
+       handle->offset += AR_HEADER_LEN;
+}
+
+/*
+ * when replacing files in an existing archive, copy from the the
+ * original archive those files that are to be left intact
+ */
+static void FAST_FUNC copy_data(archive_handle_t *handle)
+{
+       archive_handle_t *out_handle = handle->ar__out;
+       struct file_header_t *fh = handle->file_header;
+
+       out_handle->file_header = fh;
+       output_ar_header(out_handle);
+
+       bb_copyfd_exact_size(handle->src_fd, out_handle->src_fd, fh->size);
+       out_handle->offset += fh->size;
+}
+
+static int write_ar_header(archive_handle_t *handle)
+{
+       char *fn;
+       char fn_h[FILENAME_MAX];
+       struct stat st;
+       int fd;
+
+       fn = llist_pop(&handle->accept);
+       if (!fn)
+               return -1;
+
+       xstat(fn, &st);
+
+       handle->file_header->mtime = st.st_mtime;
+       handle->file_header->uid = st.st_uid;
+       handle->file_header->gid = st.st_gid;
+       handle->file_header->mode = st.st_mode;
+       handle->file_header->size = st.st_size;
+       handle->file_header->name = fn_h;
+
+#if ENABLE_FEATURE_AR_LONG_FILENAMES
+#error nope, not yet
+#else
+       sprintf(fn_h, "%.15s/", bb_basename(fn));
+#endif
+       output_ar_header(handle);
+
+       fd = xopen(fn, O_RDONLY);
+       bb_copyfd_exact_size(fd, handle->src_fd, st.st_size);
+       close(fd);
+       handle->offset += st.st_size;
+
+       return 0;
+}
+
+int FAST_FUNC write_ar_archive(archive_handle_t *handle, int append)
+{
+       struct stat st;
+       archive_handle_t *out_handle;
+
+       if (fstat(handle->src_fd, &st) == -1)
+               return EXIT_FAILURE;
+
+       /*
+        * if we're appending/replacing stuff in an existing archive,
+        * create a new handle for output, which is a new file created
+        * in place of the old one
+        */
+       if (append && st.st_size != 0) {
+               out_handle = init_handle();
+               unlink(handle->ar__name);
+               out_handle->src_fd = xopen(handle->ar__name, O_WRONLY | O_CREAT 
| O_TRUNC);
+               out_handle->accept = handle->accept;
+       } else {
+               out_handle = handle;
+       }
+
+       handle->ar__out = out_handle;
+
+       xwrite(out_handle->src_fd, AR_MAGIC "\n", AR_MAGIC_LEN + 1);
+       out_handle->offset += AR_MAGIC_LEN + 1;
+
+       /* skip to the end of the archive if we have to append stuff */
+       if (append && st.st_size != 0) {
+               handle->filter = filter_replaceable;
+               handle->action_data = copy_data;
+               unpack_ar_archive(handle);
+       }
+
+       while (write_ar_header(out_handle) == 0)
+               continue;
+
+       close(handle->src_fd);
+
+       if (out_handle->src_fd != handle->src_fd)
+               close(out_handle->src_fd);
+
+       return EXIT_SUCCESS;
+}
diff -ad -urpN busybox.5/archival/Config.in busybox.6/archival/Config.in
--- busybox.5/archival/Config.in        2010-03-06 20:30:44.000000000 +0100
+++ busybox.6/archival/Config.in        2010-03-15 03:34:14.000000000 +0100
@@ -62,6 +62,13 @@ config FEATURE_AR_LONG_FILENAMES
          It supports the GNU ar long filename method which moves multiple long
          filenames into a the data section of a new ar entry.
 
+config FEATURE_AR_CREATE
+       bool "Support archive creation"
+       default n
+       depends on AR
+       help
+         This enables archive creation (-c and -r) with busybox ar.
+
 config BUNZIP2
        bool "bunzip2"
        default n
diff -ad -urpN busybox.5/archival/Kbuild busybox.6/archival/Kbuild
--- busybox.5/archival/Kbuild   2010-03-06 20:30:44.000000000 +0100
+++ busybox.6/archival/Kbuild   2010-03-15 03:34:14.000000000 +0100
@@ -8,6 +8,7 @@ libs-y                          += libunarchive/
 
 lib-y:=
 lib-$(CONFIG_AR)               += ar.o
+lib-$(CONFIG_FEATURE_AR_CREATE)        += ar_create.o
 lib-$(CONFIG_BUNZIP2)          += bbunzip.o
 lib-$(CONFIG_BZIP2)            += bzip2.o bbunzip.o
 lib-$(CONFIG_UNLZMA)           += bbunzip.o
diff -ad -urpN busybox.5/archival/libunarchive/get_header_ar.c 
busybox.6/archival/libunarchive/get_header_ar.c
--- busybox.5/archival/libunarchive/get_header_ar.c     2010-03-06 
20:30:44.000000000 +0100
+++ busybox.6/archival/libunarchive/get_header_ar.c     2010-03-15 
03:49:50.000000000 +0100
@@ -6,6 +6,7 @@
 
 #include "libbb.h"
 #include "unarchive.h"
+#include "ar.h"
 
 static unsigned read_num(const char *str, int base)
 {
@@ -23,15 +24,7 @@ char FAST_FUNC get_header_ar(archive_han
        unsigned size;
        union {
                char raw[60];
-               struct {
-                       char name[16];
-                       char date[12];
-                       char uid[6];
-                       char gid[6];
-                       char mode[8];
-                       char size[10];
-                       char magic[2];
-               } formatted;
+               struct ar_header formatted;
        } ar;
 #if ENABLE_FEATURE_AR_LONG_FILENAMES
        static char *ar_long_names;
diff -ad -urpN busybox.5/archival/libunarchive/unpack_ar_archive.c 
busybox.6/archival/libunarchive/unpack_ar_archive.c
--- busybox.5/archival/libunarchive/unpack_ar_archive.c 2010-03-06 
20:30:44.000000000 +0100
+++ busybox.6/archival/libunarchive/unpack_ar_archive.c 2010-03-15 
03:49:13.000000000 +0100
@@ -5,16 +5,17 @@
 
 #include "libbb.h"
 #include "unarchive.h"
+#include "ar.h"
 
 void FAST_FUNC unpack_ar_archive(archive_handle_t *ar_archive)
 {
        char magic[7];
 
-       xread(ar_archive->src_fd, magic, 7);
-       if (strncmp(magic, "!<arch>", 7) != 0) {
+       xread(ar_archive->src_fd, magic, AR_MAGIC_LEN);
+       if (strncmp(magic, AR_MAGIC, AR_MAGIC_LEN) != 0) {
                bb_error_msg_and_die("invalid ar magic");
        }
-       ar_archive->offset += 7;
+       ar_archive->offset += AR_MAGIC_LEN;
 
        while (get_header_ar(ar_archive) == EXIT_SUCCESS)
                continue;
diff -ad -urpN busybox.5/include/ar.h busybox.6/include/ar.h
--- busybox.5/include/ar.h      1970-01-01 01:00:00.000000000 +0100
+++ busybox.6/include/ar.h      2010-03-15 03:47:41.000000000 +0100
@@ -0,0 +1,30 @@
+/*
+ * busybox ar archive data structures
+ * Licensed under the GPL v2 or later, see the file LICENSE in this source 
tree.
+ */
+#ifndef AR_H
+#define AR_H
+
+PUSH_AND_SET_FUNCTION_VISIBILITY_TO_HIDDEN
+
+struct ar_header {
+       char name[16];
+       char date[12];
+       char uid[6];
+       char gid[6];
+       char mode[8];
+       char size[10];
+       char magic[2];
+};
+
+#define AR_HEADER_LEN sizeof(struct ar_header)
+#define AR_MAGIC      "!<arch>"
+#define AR_MAGIC_LEN  7
+
+#if ENABLE_FEATURE_AR_CREATE
+int FAST_FUNC write_ar_archive(archive_handle_t *handle, int append);
+#endif
+
+POP_SAVED_FUNCTION_VISIBILITY
+
+#endif
diff -ad -urpN busybox.5/include/unarchive.h busybox.6/include/unarchive.h
--- busybox.5/include/unarchive.h       2010-03-06 20:30:44.000000000 +0100
+++ busybox.6/include/unarchive.h       2010-03-15 04:02:37.000000000 +0100
@@ -73,6 +73,10 @@ typedef struct archive_handle_t {
        /* Contains the handle to a sub archive */
        struct archive_handle_t *dpkg__sub_archive;
 #endif
+#if ENABLE_FEATURE_AR_CREATE
+       const char *ar__name;
+       struct archive_handle_t *ar__out;
+#endif
 } archive_handle_t;
 /* bits in ah_flags */
 #define ARCHIVE_RESTORE_DATE        (1 << 0)
diff -ad -urpN busybox.5/testsuite/ar.tests busybox.6/testsuite/ar.tests
--- busybox.5/testsuite/ar.tests        1970-01-01 01:00:00.000000000 +0100
+++ busybox.6/testsuite/ar.tests        2010-03-15 04:58:42.000000000 +0100
@@ -0,0 +1,27 @@
+#!/bin/sh
+# Copyright 2010 Nokia Corporation
+# written by Alexander Shishkin
+# Licensed under GPL v2 or later, see file LICENSE for details.
+
+. testing.sh
+
+# testing "test name" "options" "expected result" "file input" "stdin"
+
+optional FEATURE_AR_CREATE
+
+testing "ar creates archives" \
+       "ar rc test.a README && ar p test.a README | md5sum | cut -f1 -d' '" \
+       "$(md5sum README | cut -f1 -d' ')\n" \
+       "" \
+       ""
+rm test.a
+
+testing "ar replaces things in archives" \
+       "echo 'blah!' > file1 && echo 'blast!' > file2 && ar cr test.a README 
file1 file2 && mv file2 file1 && ar cr test.a file1 && ar p test.a file1" \
+       "blast!\n" \
+       "" \
+       ""
+
+rm test.a
+
+exit $FAILCOUNT
_______________________________________________
busybox mailing list
[email protected]
http://lists.busybox.net/mailman/listinfo/busybox

Reply via email to