/*- * Copyright (c) 2012 Michihiro NAKAJIMA * 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 * in this position and unchanged. * 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" __FBSDID("$FreeBSD$"); static char buff[4096]; static struct { const char *path; mode_t mode; int nlink; time_t mtime; uid_t uid; gid_t gid; } entries[] = { { ".", S_IFDIR | 0755, 3, 1231975636, 1001, 1001 }, { "./COPYING", S_IFREG | 0644, 1, 1231975636, 1001, 1001 }, { "./Makefile", S_IFREG | 0644, 1, 1233041050, 1001, 1001 }, { "./NEWS", S_IFREG | 0644, 1, 1231975636, 1001, 1001 }, { "./PROJECTS", S_IFREG | 0644, 1, 1231975636, 1001, 1001 }, { "./README", S_IFREG | 0644, 1, 1231975636, 1001, 1001 }, { "./subdir", S_IFDIR | 0755, 3, 1233504586, 1001, 1001 }, { "./subdir/README", S_IFREG | 0664, 1, 1231975636, 1002, 1001 }, { "./subdir/config", S_IFREG | 0664, 1, 1232266273, 1003, 1003 }, { "./subdir2", S_IFDIR | 0755, 3, 1233504586, 1001, 1001 }, { "./subdir3", S_IFDIR | 0755, 3, 1233504586, 1001, 1001 }, { "./subdir3/mtree", S_IFREG | 0664, 2, 1232266273, 1003, 1003 }, { NULL, 0, 0, 0, 0, 0 } }; static const char image [] = { "#mtree\n" "\n" "# .\n" "/set type=file uid=1001 gid=1001 mode=644\n" ". time=1231975636.0 mode=755 type=dir\n" " COPYING time=1231975636.0 size=8\n" " Makefile time=1233041050.0 size=8\n" " NEWS time=1231975636.0 size=8\n" " PROJECTS time=1231975636.0 size=8\n" " README time=1231975636.0 size=8\n" "\n" "# ./subdir\n" "/set mode=664\n" "subdir time=1233504586.0 mode=755 type=dir\n" " README time=1231975636.0 uid=1002 size=8\n" " config time=1232266273.0 gid=1003 uid=1003 size=8\n" "# ./subdir\n" "..\n" "\n" "\n" "# ./subdir2\n" "subdir2 time=1233504586.0 mode=755 type=dir\n" "# ./subdir2\n" "..\n" "\n" "\n" "# ./subdir3\n" "subdir3 time=1233504586.0 mode=755 type=dir\n" " mtree nlink=2 time=1232266273.0 gid=1003 uid=1003 size=8\n" "# ./subdir3\n" "..\n" "\n" "..\n\n" }; static const char image_dironly [] = { "#mtree\n" "# .\n" "/set type=dir uid=1001 gid=1001 mode=755\n" ". time=1231975636.0\n" "# ./subdir\n" "subdir time=1233504586.0\n" "# ./subdir\n" "..\n" "# ./subdir2\n" "subdir2 time=1233504586.0\n" "# ./subdir2\n" "..\n" "# ./subdir3\n" "subdir3 time=1233504586.0\n" "# ./subdir3\n" "..\n" "..\n" }; static void test_write_format_mtree_sub(int dironly) { struct archive_entry *ae; struct archive* a; size_t used; int i; /* Create a mtree format archive. */ assert((a = archive_write_new()) != NULL); assertEqualIntA(a, ARCHIVE_OK, archive_write_set_format_mtree_classic(a)); if (dironly) assertEqualIntA(a, ARCHIVE_OK, archive_write_set_format_option(a, NULL, "dironly", "1")); assertEqualIntA(a, ARCHIVE_OK, archive_write_open_memory(a, buff, sizeof(buff)-1, &used)); /* Write entries */ for (i = 0; entries[i].path != NULL; i++) { assert((ae = archive_entry_new()) != NULL); archive_entry_set_nlink(ae, entries[i].nlink); assertEqualInt(entries[i].nlink, archive_entry_nlink(ae)); archive_entry_set_mtime(ae, entries[i].mtime, 0); assertEqualInt(entries[i].mtime, archive_entry_mtime(ae)); archive_entry_set_mode(ae, entries[i].mode); assertEqualInt(entries[i].mode, archive_entry_mode(ae)); archive_entry_set_uid(ae, entries[i].uid); assertEqualInt(entries[i].uid, archive_entry_uid(ae)); archive_entry_set_gid(ae, entries[i].gid); assertEqualInt(entries[i].gid, archive_entry_gid(ae)); archive_entry_copy_pathname(ae, entries[i].path); if ((entries[i].mode & AE_IFMT) != S_IFDIR) archive_entry_set_size(ae, 8); assertEqualIntA(a, ARCHIVE_OK, archive_write_header(a, ae)); if ((entries[i].mode & AE_IFMT) != S_IFDIR) assertEqualIntA(a, 8, archive_write_data(a, "Hello012", 15)); archive_entry_free(ae); } assertEqualIntA(a, ARCHIVE_OK, archive_write_close(a)); assertEqualInt(ARCHIVE_OK, archive_write_free(a)); buff[used] = '\0'; if (dironly) assertEqualString(buff, image_dironly); else assertEqualString(buff, image); /* * Read the data and check it. */ assert((a = archive_read_new()) != NULL); assertEqualIntA(a, ARCHIVE_OK, archive_read_support_format_all(a)); assertEqualIntA(a, ARCHIVE_OK, archive_read_support_filter_all(a)); assertEqualIntA(a, ARCHIVE_OK, archive_read_open_memory(a, buff, used)); /* Read entries */ for (i = 0; entries[i].path != NULL; i++) { if (dironly && (entries[i].mode & AE_IFMT) != S_IFDIR) continue; assertEqualIntA(a, ARCHIVE_OK, archive_read_next_header(a, &ae)); assertEqualInt(entries[i].mtime, archive_entry_mtime(ae)); assertEqualInt(entries[i].mode, archive_entry_mode(ae)); assertEqualInt(entries[i].uid, archive_entry_uid(ae)); assertEqualInt(entries[i].gid, archive_entry_gid(ae)); if (i > 0) assertEqualString(entries[i].path + 2, archive_entry_pathname(ae)); else assertEqualString(entries[i].path, archive_entry_pathname(ae)); if ((entries[i].mode & AE_IFMT) != S_IFDIR) assertEqualInt(8, archive_entry_size(ae)); } assertEqualIntA(a, ARCHIVE_OK, archive_read_close(a)); assertEqualInt(ARCHIVE_OK, archive_read_free(a)); } DEFINE_TEST(test_write_format_mtree_classic) { /* Generate classic format. */ test_write_format_mtree_sub(0); /* Generate classic format and Write directory only. */ test_write_format_mtree_sub(1); }