]> git.ipfire.org Git - thirdparty/libarchive.git/commitdiff
write_raw support from Marek Kubica
authorTim Kientzle <kientzle@acm.org>
Sat, 23 Feb 2013 21:19:59 +0000 (13:19 -0800)
committerTim Kientzle <kientzle@acm.org>
Sat, 23 Feb 2013 21:19:59 +0000 (13:19 -0800)
Merged branch 'feature/write-format-raw' of https://github.com/Leonidas-from-XIV/libarchive

Makefile.am
libarchive/CMakeLists.txt
libarchive/archive.h
libarchive/archive_write_format.3
libarchive/archive_write_set_format.c
libarchive/archive_write_set_format_by_name.c
libarchive/archive_write_set_format_raw.c [new file with mode: 0644]
libarchive/test/CMakeLists.txt
libarchive/test/test_write_format_raw.c [new file with mode: 0644]
libarchive/test/test_write_format_raw_b64.c [new file with mode: 0644]

index 57efe7916d0aa5c1ca00155e4b9edbb8a7c5d64f..6c156807f48f750881ac8677aacd2a6a1fbf7a56 100644 (file)
@@ -199,6 +199,7 @@ libarchive_la_SOURCES=                                              \
        libarchive/archive_write_set_format_iso9660.c           \
        libarchive/archive_write_set_format_mtree.c             \
        libarchive/archive_write_set_format_pax.c               \
+       libarchive/archive_write_set_format_raw.c               \
        libarchive/archive_write_set_format_shar.c              \
        libarchive/archive_write_set_format_ustar.c             \
        libarchive/archive_write_set_format_v7tar.c             \
@@ -486,6 +487,8 @@ libarchive_test_SOURCES=                                    \
        libarchive/test/test_write_format_mtree_no_separator.c  \
        libarchive/test/test_write_format_mtree_quoted_filename.c\
        libarchive/test/test_write_format_pax.c                 \
+       libarchive/test/test_write_format_raw.c                 \
+       libarchive/test/test_write_format_raw_b64.c             \
        libarchive/test/test_write_format_shar_empty.c          \
        libarchive/test/test_write_format_tar.c                 \
        libarchive/test/test_write_format_tar_empty.c           \
index ecb0409bd9d881b98608098f9f72a6f8bc287809..23d36325fe33b2b90bdd4628ca0d9ea68be85dc5 100644 (file)
@@ -125,6 +125,7 @@ SET(libarchive_SOURCES
   archive_write_set_format_iso9660.c
   archive_write_set_format_mtree.c
   archive_write_set_format_pax.c
+  archive_write_set_format_raw.c
   archive_write_set_format_shar.c
   archive_write_set_format_ustar.c
   archive_write_set_format_v7tar.c
index 5f5c38159928ef63908aa69f0e6e95e9734ca285..1e467a1bab0e2049d37138dd81db8fd559da4a3a 100644 (file)
@@ -670,6 +670,7 @@ __LA_DECL int archive_write_set_format_mtree_classic(struct archive *);
 /* TODO: int archive_write_set_format_old_tar(struct archive *); */
 __LA_DECL int archive_write_set_format_pax(struct archive *);
 __LA_DECL int archive_write_set_format_pax_restricted(struct archive *);
+__LA_DECL int archive_write_set_format_raw(struct archive *);
 __LA_DECL int archive_write_set_format_shar(struct archive *);
 __LA_DECL int archive_write_set_format_shar_dump(struct archive *);
 __LA_DECL int archive_write_set_format_ustar(struct archive *);
index dad2f7d7eb31a0f97d6c4af571ae77bed9aac5f7..39d300646e0909dd365841901a7aab571614a81c 100644 (file)
 .\"
 .\" $FreeBSD$
 .\"
-.Dd February 2, 2012
+.Dd February 14, 2013
 .Dt ARCHIVE_WRITE_FORMAT 3
 .Os
 .Sh NAME
 .Nm archive_write_set_format_cpio ,
 .Nm archive_write_set_format_pax ,
 .Nm archive_write_set_format_pax_restricted ,
+.Nm archive_write_set_format_raw ,
 .Nm archive_write_set_format_shar ,
 .Nm archive_write_set_format_shar_dump ,
 .Nm archive_write_set_format_ustar
@@ -46,6 +47,8 @@ Streaming Archive Library (libarchive, -larchive)
 .Ft int
 .Fn archive_write_set_format_pax_restricted "struct archive *"
 .Ft int
+.Fn archive_write_set_format_raw "struct archive *"
+.Ft int
 .Fn archive_write_set_format_shar "struct archive *"
 .Ft int
 .Fn archive_write_set_format_shar_dump "struct archive *"
index 641d56f6c27eeb3651a98a89af5281a20b2517dc..9055753b293a0d40ad3c1e300cd72e0183f1e527 100644 (file)
@@ -47,6 +47,7 @@ struct { int code; int (*setter)(struct archive *); } codes[] =
        { ARCHIVE_FORMAT_CPIO_SVR4_NOCRC,       archive_write_set_format_cpio_newc },
        { ARCHIVE_FORMAT_ISO9660,       archive_write_set_format_iso9660 },
        { ARCHIVE_FORMAT_MTREE,         archive_write_set_format_mtree },
+       { ARCHIVE_FORMAT_RAW,           archive_write_set_format_raw },
        { ARCHIVE_FORMAT_SHAR,          archive_write_set_format_shar },
        { ARCHIVE_FORMAT_SHAR_BASE,     archive_write_set_format_shar },
        { ARCHIVE_FORMAT_SHAR_DUMP,     archive_write_set_format_shar_dump },
@@ -57,7 +58,7 @@ struct { int code; int (*setter)(struct archive *); } codes[] =
                                archive_write_set_format_pax_restricted },
        { ARCHIVE_FORMAT_TAR_USTAR,     archive_write_set_format_ustar },
        { ARCHIVE_FORMAT_XAR,           archive_write_set_format_xar },
-       { ARCHIVE_FORMAT_ZIP,   archive_write_set_format_zip },
+       { ARCHIVE_FORMAT_ZIP,           archive_write_set_format_zip },
        { 0,            NULL }
 };
 
index af3105e482d646af290fc51122b115a33b3a33a1..4f3ce7d3518d7258d59a47371006a577429e2f28 100644 (file)
@@ -63,6 +63,7 @@ struct { const char *name; int (*setter)(struct archive *); } names[] =
        { "pax",        archive_write_set_format_pax },
        { "paxr",       archive_write_set_format_pax_restricted },
        { "posix",      archive_write_set_format_pax },
+       { "raw",        archive_write_set_format_raw },
        { "rpax",       archive_write_set_format_pax_restricted },
        { "shar",       archive_write_set_format_shar },
        { "shardump",   archive_write_set_format_shar_dump },
diff --git a/libarchive/archive_write_set_format_raw.c b/libarchive/archive_write_set_format_raw.c
new file mode 100644 (file)
index 0000000..feff936
--- /dev/null
@@ -0,0 +1,125 @@
+/*-
+ * Copyright (c) 2013 Marek Kubica
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following disclaimer in the
+ *    documentation and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS OR
+ * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
+ * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
+ * IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY DIRECT, INDIRECT,
+ * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
+ * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+ * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+ * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
+ * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+
+#include "archive_platform.h"
+
+#ifdef HAVE_ERRNO_H
+#include <errno.h>
+#endif
+
+#include "archive_entry.h"
+#include "archive_write_private.h"
+
+static ssize_t archive_write_raw_data(struct archive_write *,
+                   const void *buff, size_t s);
+static int     archive_write_raw_free(struct archive_write *);
+static int     archive_write_raw_header(struct archive_write *,
+                   struct archive_entry *);
+
+struct raw {
+        int entries_written;
+};
+
+/*
+ * Set output format to 'raw' format.
+ */
+int
+archive_write_set_format_raw(struct archive *_a)
+{
+       struct archive_write *a = (struct archive_write *)_a;
+       struct raw *raw;
+
+       archive_check_magic(_a, ARCHIVE_WRITE_MAGIC,
+           ARCHIVE_STATE_NEW, "archive_write_set_format_raw");
+
+       /* If someone else was already registered, unregister them. */
+       if (a->format_free != NULL)
+               (a->format_free)(a);
+
+       raw = (struct raw *)calloc(1, sizeof(*raw));
+       if (raw == NULL) {
+               archive_set_error(&a->archive, ENOMEM, "Can't allocate raw data");
+               return (ARCHIVE_FATAL);
+       }
+       raw->entries_written = 0;
+       a->format_data = raw;
+       a->format_name = "raw";
+        /* no options exist for this format */
+       a->format_options = NULL;
+       a->format_write_header = archive_write_raw_header;
+       a->format_write_data = archive_write_raw_data;
+       a->format_finish_entry = NULL;
+        /* nothing needs to be done on closing */
+       a->format_close = NULL;
+       a->format_free = archive_write_raw_free;
+       a->archive.archive_format = ARCHIVE_FORMAT_RAW;
+       a->archive.archive_format_name = "RAW";
+       return (ARCHIVE_OK);
+}
+
+static int
+archive_write_raw_header(struct archive_write *a, struct archive_entry *entry)
+{
+       struct raw *raw = (struct raw *)a->format_data;
+
+       if (archive_entry_filetype(entry) != AE_IFREG) {
+               archive_set_error(&a->archive, ERANGE,
+                   "Raw format only supports filetype AE_IFREG");
+               return (ARCHIVE_FATAL);
+       }
+
+
+       if (raw->entries_written > 0) {
+               archive_set_error(&a->archive, ERANGE,
+                   "Raw format only supports one entry per archive");
+               return (ARCHIVE_FATAL);
+       }
+       raw->entries_written++;
+
+       return (ARCHIVE_OK);
+}
+
+static ssize_t
+archive_write_raw_data(struct archive_write *a, const void *buff, size_t s)
+{
+       int ret;
+
+       ret = __archive_write_output(a, buff, s);
+       if (ret >= 0)
+               return (s);
+       else
+               return (ret);
+}
+
+static int
+archive_write_raw_free(struct archive_write *a)
+{
+       struct raw *raw;
+
+       raw = (struct raw *)a->format_data;
+       free(raw);
+       a->format_data = NULL;
+       return (ARCHIVE_OK);
+}
index d2eb2c2f39ef4d8a58cfbda642d1bff65365f92a..d74206d8933076fb88798f9f8b4779fbf9e868b0 100644 (file)
@@ -201,6 +201,8 @@ IF(ENABLE_TEST)
     test_write_format_mtree_no_separator.c
     test_write_format_mtree_quoted_filename.c
     test_write_format_pax.c
+    test_write_format_raw.c
+    test_write_format_raw_b64.c
     test_write_format_shar_empty.c
     test_write_format_tar.c
     test_write_format_tar_empty.c
diff --git a/libarchive/test/test_write_format_raw.c b/libarchive/test/test_write_format_raw.c
new file mode 100644 (file)
index 0000000..bd887ff
--- /dev/null
@@ -0,0 +1,123 @@
+/*-
+ * Copyright (c) 2013 Marek Kubica
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following disclaimer in the
+ *    documentation and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS OR
+ * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
+ * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
+ * IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY DIRECT, INDIRECT,
+ * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
+ * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+ * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+ * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
+ * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+#include "test.h"
+
+static void
+test_format(int        (*set_format)(struct archive *))
+{
+       char filedata[64];
+       struct archive_entry *ae;
+       struct archive *a;
+       size_t used;
+       size_t buffsize = 1000000;
+       char *buff;
+       const char *err;
+
+       buff = malloc(buffsize);
+
+       /* Create a new archive in memory. */
+       assert((a = archive_write_new()) != NULL);
+       assertEqualIntA(a, ARCHIVE_OK, (*set_format)(a));
+       assertEqualIntA(a, ARCHIVE_OK, archive_write_add_filter_none(a));
+       assertEqualIntA(a, ARCHIVE_OK, archive_write_open_memory(a, buff, buffsize, &used));
+
+       /*
+        * Write a file to it.
+        */
+       assert((ae = archive_entry_new()) != NULL);
+       archive_entry_set_pathname(ae, "test");
+       archive_entry_set_filetype(ae, AE_IFREG);
+       assertEqualIntA(a, ARCHIVE_OK, archive_write_header(a, ae));
+       archive_entry_free(ae);
+       assertEqualIntA(a, 9, archive_write_data(a, "12345678", 9));
+       assertEqualIntA(a, ARCHIVE_OK, archive_write_close(a));
+       assertEqualInt(ARCHIVE_OK, archive_write_free(a));
+
+       /*
+        * Read from it.
+        */
+       assert((a = archive_read_new()) != NULL);
+       assertEqualIntA(a, ARCHIVE_OK, archive_read_support_format_raw(a));
+       assertEqualIntA(a, ARCHIVE_OK, archive_read_support_filter_none(a));
+       assertEqualIntA(a, ARCHIVE_OK, archive_read_open_memory(a, buff, used));
+
+       assertEqualIntA(a, ARCHIVE_OK, archive_read_next_header(a, &ae));
+       assertEqualIntA(a, 9, archive_read_data(a, filedata, 10));
+       assertEqualMem(filedata, "12345678", 9);
+       assertEqualIntA(a, ARCHIVE_OK, archive_read_close(a));
+       assertEqualInt(ARCHIVE_OK, archive_read_free(a));
+
+       /* Create a new archive */
+       assert((a = archive_write_new()) != NULL);
+       assertEqualIntA(a, ARCHIVE_OK, (*set_format)(a));
+       assertEqualIntA(a, ARCHIVE_OK, archive_write_add_filter_none(a));
+       assertEqualIntA(a, ARCHIVE_OK, archive_write_open_memory(a, buff, buffsize, &used));
+
+       /* write first file: that should succeed */
+       assert((ae = archive_entry_new()) != NULL);
+       archive_entry_set_pathname(ae, "test");
+       archive_entry_set_filetype(ae, AE_IFREG);
+       assertEqualIntA(a, ARCHIVE_OK, archive_write_header(a, ae));
+       archive_entry_free(ae);
+       assertEqualIntA(a, 9, archive_write_data(a, "12345678", 9));
+
+       /* write second file: this should fail */
+       assert((ae = archive_entry_new()) != NULL);
+       archive_entry_set_pathname(ae, "test2");
+       archive_entry_set_filetype(ae, AE_IFREG);
+       assertEqualIntA(a, ARCHIVE_FATAL, archive_write_header(a, ae));
+       err = archive_error_string(a);
+       assertEqualMem(err, "Raw format only supports one entry per archive", 47);
+       archive_entry_free(ae);
+
+       assertEqualIntA(a, ARCHIVE_OK, archive_write_close(a));
+       assertEqualInt(ARCHIVE_OK, archive_write_free(a));
+
+       /* Create a new archive */
+       assert((a = archive_write_new()) != NULL);
+       assertEqualIntA(a, ARCHIVE_OK, (*set_format)(a));
+       assertEqualIntA(a, ARCHIVE_OK, archive_write_add_filter_none(a));
+       assertEqualIntA(a, ARCHIVE_OK, archive_write_open_memory(a, buff, buffsize, &used));
+
+       /* write a directory: this should fail */
+       assert((ae = archive_entry_new()) != NULL);
+       archive_entry_copy_pathname(ae, "dir");
+       archive_entry_set_filetype(ae, AE_IFDIR);
+       archive_entry_set_size(ae, 512);
+       assertEqualIntA(a, ARCHIVE_FATAL, archive_write_header(a, ae));
+       err = archive_error_string(a);
+       assertEqualMem(err, "Raw format only supports filetype AE_IFREG", 43);
+       archive_entry_free(ae);
+
+       assertEqualIntA(a, ARCHIVE_OK, archive_write_close(a));
+       assertEqualInt(ARCHIVE_OK, archive_write_free(a));
+
+       free(buff);
+}
+
+DEFINE_TEST(test_write_format_raw)
+{
+       test_format(archive_write_set_format_raw);
+}
diff --git a/libarchive/test/test_write_format_raw_b64.c b/libarchive/test/test_write_format_raw_b64.c
new file mode 100644 (file)
index 0000000..f8f6728
--- /dev/null
@@ -0,0 +1,77 @@
+/*-
+ * Copyright (c) 2013 Marek Kubica
+ * All rights reserved.
+ *
+ * Redistribution and use in source and binary forms, with or without
+ * modification, are permitted provided that the following conditions
+ * are met:
+ * 1. Redistributions of source code must retain the above copyright
+ *    notice, this list of conditions and the following disclaimer.
+ * 2. Redistributions in binary form must reproduce the above copyright
+ *    notice, this list of conditions and the following disclaimer in the
+ *    documentation and/or other materials provided with the distribution.
+ *
+ * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS OR
+ * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
+ * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
+ * IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY DIRECT, INDIRECT,
+ * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
+ * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
+ * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
+ * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
+ * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
+ * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
+ */
+#include "test.h"
+
+static void
+test_format(int        (*set_format)(struct archive *))
+{
+       char filedata[64];
+       struct archive_entry *ae;
+       struct archive *a;
+       size_t used;
+       size_t buffsize = 1000000;
+       char *buff;
+
+       buff = malloc(buffsize);
+
+       /* Create a new archive in memory. */
+       assert((a = archive_write_new()) != NULL);
+       assertEqualIntA(a, ARCHIVE_OK, (*set_format)(a));
+       assertEqualIntA(a, ARCHIVE_OK, archive_write_add_filter_b64encode(a));
+       assertEqualIntA(a, ARCHIVE_OK, archive_write_open_memory(a, buff, buffsize, &used));
+
+       /*
+        * Write a file to it.
+        */
+       assert((ae = archive_entry_new()) != NULL);
+       archive_entry_set_pathname(ae, "test");
+       archive_entry_set_filetype(ae, AE_IFREG);
+       assertEqualIntA(a, ARCHIVE_OK, archive_write_header(a, ae));
+       archive_entry_free(ae);
+       assertEqualIntA(a, 9, archive_write_data(a, "12345678", 9));
+       assertEqualIntA(a, ARCHIVE_OK, archive_write_close(a));
+       assertEqualInt(ARCHIVE_OK, archive_write_free(a));
+
+       /*
+        * Read from it.
+        */
+       assert((a = archive_read_new()) != NULL);
+       assertEqualIntA(a, ARCHIVE_OK, archive_read_support_format_raw(a));
+       assertEqualIntA(a, ARCHIVE_OK, archive_read_support_filter_none(a));
+       assertEqualIntA(a, ARCHIVE_OK, archive_read_open_memory(a, buff, used));
+
+       assertEqualIntA(a, ARCHIVE_OK, archive_read_next_header(a, &ae));
+       assertEqualIntA(a, 37, archive_read_data(a, filedata, 64));
+       assertEqualMem(filedata, "begin-base64 644 -\nMTIzNDU2NzgA\n====\n", 37);
+       assertEqualIntA(a, ARCHIVE_OK, archive_read_close(a));
+       assertEqualInt(ARCHIVE_OK, archive_read_free(a));
+
+       free(buff);
+}
+
+DEFINE_TEST(test_write_format_raw_b64)
+{
+       test_format(archive_write_set_format_raw);
+}