| /*- |
| * Copyright (c) 2009 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: head/lib/libarchive/test/test_write_format_mtree.c 191183 2009-04-17 01:06:31Z kientzle $"); |
| |
| static char buff[4096]; |
| static struct { |
| const char *path; |
| mode_t mode; |
| time_t mtime; |
| uid_t uid; |
| gid_t gid; |
| } entries[] = { |
| { "./COPYING", S_IFREG | 0644, 1231975636, 1001, 1001 }, |
| { "./Makefile", S_IFREG | 0644, 1233041050, 1001, 1001 }, |
| { "./NEWS", S_IFREG | 0644, 1231975636, 1001, 1001 }, |
| { "./PROJECTS", S_IFREG | 0644, 1231975636, 1001, 1001 }, |
| { "./README", S_IFREG | 0644, 1231975636, 1001, 1001 }, |
| { "./subdir", S_IFDIR | 0755, 1233504586, 1001, 1001 }, |
| { "./subdir/README", S_IFREG | 0664, 1231975636, 1002, 1001 }, |
| { "./subdir/config", S_IFREG | 0664, 1232266273, 1003, 1003 }, |
| { "./subdir2", S_IFDIR | 0755, 1233504586, 1001, 1001 }, |
| { "./subdir3", S_IFDIR | 0755, 1233504586, 1001, 1001 }, |
| { "./subdir3/mtree", S_IFREG | 0664, 1232266273, 1003, 1003 }, |
| { NULL, 0, 0, 0, 0 } |
| }; |
| static struct { |
| const char *path; |
| mode_t mode; |
| time_t mtime; |
| uid_t uid; |
| gid_t gid; |
| } entries2[] = { |
| { "COPYING", S_IFREG | 0644, 1231975636, 1001, 1001 }, |
| { "Makefile", S_IFREG | 0644, 1233041050, 1001, 1001 }, |
| { "NEWS", S_IFREG | 0644, 1231975636, 1001, 1001 }, |
| { "PROJECTS", S_IFREG | 0644, 1231975636, 1001, 1001 }, |
| { "README", S_IFREG | 0644, 1231975636, 1001, 1001 }, |
| { "subdir", S_IFDIR | 0755, 1233504586, 1001, 1001 }, |
| { "subdir/README", S_IFREG | 0664, 1231975636, 1002, 1001 }, |
| { "subdir/config", S_IFREG | 0664, 1232266273, 1003, 1003 }, |
| { "subdir2", S_IFDIR | 0755, 1233504586, 1001, 1001 }, |
| { "subdir3", S_IFDIR | 0755, 1233504586, 1001, 1001 }, |
| { "subdir3/mtree", S_IFREG | 0664, 1232266273, 1003, 1003 }, |
| { NULL, 0, 0, 0, 0 } |
| }; |
| |
| static void |
| test_write_format_mtree_sub(int use_set, 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(a)); |
| if (use_set) |
| assertEqualIntA(a, ARCHIVE_OK, archive_write_set_format_option(a, NULL, "use-set", "1")); |
| 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_mtime(ae, entries[i].mtime, 0); |
| assert(entries[i].mtime == archive_entry_mtime(ae)); |
| archive_entry_set_mode(ae, entries[i].mode); |
| assert(entries[i].mode == archive_entry_mode(ae)); |
| archive_entry_set_uid(ae, entries[i].uid); |
| assert(entries[i].uid == archive_entry_uid(ae)); |
| archive_entry_set_gid(ae, entries[i].gid); |
| assert(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)); |
| |
| if (use_set) { |
| const char *p; |
| |
| buff[used] = '\0'; |
| assert(NULL != (p = strstr(buff, "\n/set "))); |
| if (p != NULL) { |
| char *r; |
| const char *o; |
| p++; |
| r = strchr(p, '\n'); |
| if (r != NULL) |
| *r = '\0'; |
| if (dironly) |
| o = "/set type=dir uid=1001 gid=1001 mode=755"; |
| else |
| o = "/set type=file uid=1001 gid=1001 mode=644"; |
| assertEqualString(o, p); |
| if (r != NULL) |
| *r = '\n'; |
| } |
| } |
| |
| /* |
| * 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)); |
| 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)); |
| } |
| |
| static void |
| test_write_format_mtree_sub2(int use_set, int dironly) |
| { |
| struct archive_entry *ae; |
| struct archive* a; |
| size_t used; |
| int i; |
| char str[32]; |
| |
| /* Create a mtree format archive. */ |
| assert((a = archive_write_new()) != NULL); |
| assertEqualIntA(a, ARCHIVE_OK, archive_write_set_format_mtree(a)); |
| if (use_set) |
| assertEqualIntA(a, ARCHIVE_OK, archive_write_set_format_option(a, NULL, "use-set", "1")); |
| 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 entries2 */ |
| for (i = 0; entries2[i].path != NULL; i++) { |
| assert((ae = archive_entry_new()) != NULL); |
| archive_entry_set_mtime(ae, entries2[i].mtime, 0); |
| assert(entries2[i].mtime == archive_entry_mtime(ae)); |
| archive_entry_set_mode(ae, entries2[i].mode); |
| assert(entries2[i].mode == archive_entry_mode(ae)); |
| archive_entry_set_uid(ae, entries2[i].uid); |
| assert(entries2[i].uid == archive_entry_uid(ae)); |
| archive_entry_set_gid(ae, entries2[i].gid); |
| assert(entries2[i].gid == archive_entry_gid(ae)); |
| archive_entry_copy_pathname(ae, entries2[i].path); |
| if ((entries2[i].mode & AE_IFMT) != S_IFDIR) |
| archive_entry_set_size(ae, 8); |
| assertEqualIntA(a, ARCHIVE_OK, archive_write_header(a, ae)); |
| if ((entries2[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)); |
| |
| if (use_set) { |
| const char *p; |
| |
| buff[used] = '\0'; |
| assert(NULL != (p = strstr(buff, "\n/set "))); |
| if (p != NULL) { |
| char *r; |
| const char *o; |
| p++; |
| r = strchr(p, '\n'); |
| if (r != NULL) |
| *r = '\0'; |
| if (dironly) |
| o = "/set type=dir uid=1001 gid=1001 mode=755"; |
| else |
| o = "/set type=file uid=1001 gid=1001 mode=644"; |
| assertEqualString(o, p); |
| if (r != NULL) |
| *r = '\n'; |
| } |
| } |
| |
| /* |
| * 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 entries2 */ |
| memset(str, 0, sizeof(str)); |
| strcpy(str, "./"); |
| for (i = 0; entries2[i].path != NULL; i++) { |
| if (dironly && (entries2[i].mode & AE_IFMT) != S_IFDIR) |
| continue; |
| assertEqualIntA(a, ARCHIVE_OK, archive_read_next_header(a, &ae)); |
| assertEqualInt(entries2[i].mtime, archive_entry_mtime(ae)); |
| assertEqualInt(entries2[i].mode, archive_entry_mode(ae)); |
| assertEqualInt(entries2[i].uid, archive_entry_uid(ae)); |
| assertEqualInt(entries2[i].gid, archive_entry_gid(ae)); |
| strcpy(str + 2, entries2[i].path); |
| assertEqualString(str, archive_entry_pathname(ae)); |
| if ((entries2[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) |
| { |
| /* Default setting */ |
| test_write_format_mtree_sub(0, 0); |
| /* Directory only */ |
| test_write_format_mtree_sub(0, 1); |
| /* Use /set keyword */ |
| test_write_format_mtree_sub(1, 0); |
| /* Use /set keyword with directory only */ |
| test_write_format_mtree_sub(1, 1); |
| } |
| |
| DEFINE_TEST(test_write_format_mtree_no_leading_dotslash) |
| { |
| /* Default setting */ |
| test_write_format_mtree_sub2(0, 0); |
| /* Directory only */ |
| test_write_format_mtree_sub2(0, 1); |
| /* Use /set keyword */ |
| test_write_format_mtree_sub2(1, 0); |
| /* Use /set keyword with directory only */ |
| test_write_format_mtree_sub2(1, 1); |
| } |