contrib/libarchive/libarchive/test/test_write_format_mtree_fflags.c

   1 /*-
   2  * Copyright (c) 2011 Michihiro NAKAJIMA
   3  * All rights reserved.
   4  *
   5  * Redistribution and use in source and binary forms, with or without
   6  * modification, are permitted provided that the following conditions
   7  * are met:
   8  * 1. Redistributions of source code must retain the above copyright
   9  *    notice, this list of conditions and the following disclaimer
  10  *    in this position and unchanged.
  11  * 2. Redistributions in binary form must reproduce the above copyright
  12  *    notice, this list of conditions and the following disclaimer in the
  13  *    documentation and/or other materials provided with the distribution.
  14  *
  15  * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS OR
  16  * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
  17  * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
  18  * IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY DIRECT, INDIRECT,
  19  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
  20  * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
  21  * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
  22  * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
  23  * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
  24  * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
  25  */
  26
  27 #include "test.h"
  28 __FBSDID("$FreeBSD$");
  29 #ifdef HAVE_SYS_STAT_H
  30 #include <sys/stat.h>
  31 #endif
  32
  33 /*
  34  * Test UFS file flags with/without use-set option.
  35  */
  36 #if defined(UF_IMMUTABLE) && defined(UF_NODUMP)
  37
  38 static char buff[4096];
  39 static struct {
  40         const char      *path;
  41         unsigned long    fflags;
  42 } entries[] = {
  43         { "./f1",       UF_IMMUTABLE | UF_NODUMP },
  44         { "./f11",      UF_IMMUTABLE | UF_NODUMP },
  45         { "./f2",       0 },
  46         { "./f3",       UF_NODUMP },
  47         { NULL, 0 }
  48 };
  49
  50 static void
  51 test_write_format_mtree_sub(int use_set)
  52 {
  53         struct archive_entry *ae;
  54         struct archive* a;
  55         size_t used;
  56         int i;
  57
  58         /* Create a mtree format archive. */
  59         assert((a = archive_write_new()) != NULL);
  60         assertEqualIntA(a, ARCHIVE_OK, archive_write_set_format_mtree(a));
  61         if (use_set)
  62                 assertEqualIntA(a, ARCHIVE_OK,
  63                     archive_write_set_options(a, "use-set,!all,flags,type"));
  64         else
  65                 assertEqualIntA(a, ARCHIVE_OK,
  66                     archive_write_set_options(a, "!all,flags,type"));
  67         assertEqualIntA(a, ARCHIVE_OK,
  68             archive_write_open_memory(a, buff, sizeof(buff)-1, &used));
  69
  70         /* Write entries */
  71         for (i = 0; entries[i].path != NULL; i++) {
  72                 assert((ae = archive_entry_new()) != NULL);
  73                 archive_entry_set_fflags(ae, entries[i].fflags, 0);
  74                 archive_entry_copy_pathname(ae, entries[i].path);
  75                 archive_entry_set_size(ae, 0);
  76                 assertEqualIntA(a, ARCHIVE_OK, archive_write_header(a, ae));
  77                 archive_entry_free(ae);
  78         }
  79         assertEqualIntA(a, ARCHIVE_OK, archive_write_close(a));
  80         assertEqualInt(ARCHIVE_OK, archive_write_free(a));
  81
  82         if (use_set) {
  83                 const char *p;
  84
  85                 buff[used] = '\0';
  86                 assert(NULL != (p = strstr(buff, "\n/set ")));
  87                 if (p != NULL) {
  88                         char *r;
  89                         const char *o;
  90                         p++;
  91                         r = strchr(p, '\n');
  92                         if (r != NULL)
  93                                 *r = '\0';
  94                         o = "/set type=file flags=uchg,nodump";
  95                         assertEqualString(o, p);
  96                         if (r != NULL)
  97                                 *r = '\n';
  98                 }
  99         }
 100
 101         /*
 102          * Read the data and check it.
 103          */
 104         assert((a = archive_read_new()) != NULL);
 105         assertEqualIntA(a, ARCHIVE_OK, archive_read_support_format_all(a));
 106         assertEqualIntA(a, ARCHIVE_OK, archive_read_support_filter_all(a));
 107         assertEqualIntA(a, ARCHIVE_OK, archive_read_open_memory(a, buff, used));
 108
 109         /* Read entries */
 110         for (i = 0; entries[i].path != NULL; i++) {
 111                 unsigned long fset, fclr;
 112
 113                 assertEqualIntA(a, ARCHIVE_OK, archive_read_next_header(a, &ae));
 114                 archive_entry_fflags(ae, &fset, &fclr);
 115                 assertEqualInt((int)entries[i].fflags, (int)fset);
 116                 assertEqualInt(0, (int)fclr);
 117                 assertEqualString(entries[i].path, archive_entry_pathname(ae));
 118         }
 119         assertEqualIntA(a, ARCHIVE_OK, archive_read_close(a));
 120         assertEqualInt(ARCHIVE_OK, archive_read_free(a));
 121 }
 122
 123 #endif
 124
 125 DEFINE_TEST(test_write_format_mtree_fflags)
 126 {
 127 #if defined(UF_IMMUTABLE) && defined(UF_NODUMP)
 128         /* Default setting */
 129         test_write_format_mtree_sub(0);
 130         /* Use /set keyword */
 131         test_write_format_mtree_sub(1);
 132 #else
 133         skipping("This platform does not support UFS file flags");
 134 #endif
 135 }