test_write_format_mtree_classic.c revision 256281
1/*-
2 * Copyright (c) 2012 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
30static char buff[4096];
31static struct {
32	const char	*path;
33	mode_t		 mode;
34	int		 nlink;
35	time_t		 mtime;
36	uid_t	 	 uid;
37	gid_t		 gid;
38} entries[] = {
39	{ ".",			S_IFDIR | 0755, 3, 1231975636, 1001, 1001 },
40	{ "./COPYING",		S_IFREG | 0644, 1, 1231975636, 1001, 1001 },
41	{ "./Makefile", 	S_IFREG | 0644, 1, 1233041050, 1001, 1001 },
42	{ "./NEWS", 		S_IFREG | 0644, 1, 1231975636, 1001, 1001 },
43	{ "./PROJECTS", 	S_IFREG | 0644, 1, 1231975636, 1001, 1001 },
44	{ "./README",		S_IFREG | 0644, 1, 1231975636, 1001, 1001 },
45	{ "./subdir",		S_IFDIR | 0755, 3, 1233504586, 1001, 1001 },
46	{ "./subdir/README",	S_IFREG | 0664, 1, 1231975636, 1002, 1001 },
47	{ "./subdir/config",	S_IFREG | 0664, 1, 1232266273, 1003, 1003 },
48	{ "./subdir2",		S_IFDIR | 0755, 3, 1233504586, 1001, 1001 },
49	{ "./subdir3",		S_IFDIR | 0755, 3, 1233504586, 1001, 1001 },
50	{ "./subdir3/mtree",	S_IFREG | 0664, 2, 1232266273, 1003, 1003 },
51	{ NULL, 0, 0, 0, 0, 0 }
52};
53
54static const char image [] = {
55"#mtree\n"
56"\n"
57"# .\n"
58"/set type=file uid=1001 gid=1001 mode=644\n"
59".               time=1231975636.0 mode=755 type=dir\n"
60"    COPYING     time=1231975636.0 size=8\n"
61"    Makefile    time=1233041050.0 size=8\n"
62"    NEWS        time=1231975636.0 size=8\n"
63"    PROJECTS    time=1231975636.0 size=8\n"
64"    README      time=1231975636.0 size=8\n"
65"\n"
66"# ./subdir\n"
67"/set mode=664\n"
68"subdir          time=1233504586.0 mode=755 type=dir\n"
69"    README      time=1231975636.0 uid=1002 size=8\n"
70"    config      time=1232266273.0 gid=1003 uid=1003 size=8\n"
71"# ./subdir\n"
72"..\n"
73"\n"
74"\n"
75"# ./subdir2\n"
76"subdir2         time=1233504586.0 mode=755 type=dir\n"
77"# ./subdir2\n"
78"..\n"
79"\n"
80"\n"
81"# ./subdir3\n"
82"subdir3         time=1233504586.0 mode=755 type=dir\n"
83"    mtree       nlink=2 time=1232266273.0 gid=1003 uid=1003 size=8\n"
84"# ./subdir3\n"
85"..\n"
86"\n"
87"..\n\n"
88};
89
90static const char image_dironly [] = {
91"#mtree\n"
92"# .\n"
93"/set type=dir uid=1001 gid=1001 mode=755\n"
94".               time=1231975636.0\n"
95"# ./subdir\n"
96"subdir          time=1233504586.0\n"
97"# ./subdir\n"
98"..\n"
99"# ./subdir2\n"
100"subdir2         time=1233504586.0\n"
101"# ./subdir2\n"
102"..\n"
103"# ./subdir3\n"
104"subdir3         time=1233504586.0\n"
105"# ./subdir3\n"
106"..\n"
107"..\n"
108};
109
110static void
111test_write_format_mtree_sub(int dironly)
112{
113	struct archive_entry *ae;
114	struct archive* a;
115	size_t used;
116	int i;
117
118	/* Create a mtree format archive. */
119	assert((a = archive_write_new()) != NULL);
120	assertEqualIntA(a, ARCHIVE_OK, archive_write_set_format_mtree_classic(a));
121	if (dironly)
122		assertEqualIntA(a, ARCHIVE_OK,
123			archive_write_set_format_option(a, NULL, "dironly", "1"));
124	assertEqualIntA(a, ARCHIVE_OK,
125		archive_write_open_memory(a, buff, sizeof(buff)-1, &used));
126
127	/* Write entries */
128	for (i = 0; entries[i].path != NULL; i++) {
129		assert((ae = archive_entry_new()) != NULL);
130		archive_entry_set_nlink(ae, entries[i].nlink);
131		assertEqualInt(entries[i].nlink, archive_entry_nlink(ae));
132		archive_entry_set_mtime(ae, entries[i].mtime, 0);
133		assertEqualInt(entries[i].mtime, archive_entry_mtime(ae));
134		archive_entry_set_mode(ae, entries[i].mode);
135		assertEqualInt(entries[i].mode, archive_entry_mode(ae));
136		archive_entry_set_uid(ae, entries[i].uid);
137		assertEqualInt(entries[i].uid, archive_entry_uid(ae));
138		archive_entry_set_gid(ae, entries[i].gid);
139		assertEqualInt(entries[i].gid, archive_entry_gid(ae));
140		archive_entry_copy_pathname(ae, entries[i].path);
141		if ((entries[i].mode & AE_IFMT) != S_IFDIR)
142			archive_entry_set_size(ae, 8);
143		assertEqualIntA(a, ARCHIVE_OK, archive_write_header(a, ae));
144		if ((entries[i].mode & AE_IFMT) != S_IFDIR)
145			assertEqualIntA(a, 8,
146			    archive_write_data(a, "Hello012", 15));
147		archive_entry_free(ae);
148	}
149	assertEqualIntA(a, ARCHIVE_OK, archive_write_close(a));
150        assertEqualInt(ARCHIVE_OK, archive_write_free(a));
151
152	buff[used] = '\0';
153	if (dironly)
154		assertEqualString(buff, image_dironly);
155	else
156		assertEqualString(buff, image);
157
158	/*
159	 * Read the data and check it.
160	 */
161	assert((a = archive_read_new()) != NULL);
162	assertEqualIntA(a, ARCHIVE_OK, archive_read_support_format_all(a));
163	assertEqualIntA(a, ARCHIVE_OK, archive_read_support_filter_all(a));
164	assertEqualIntA(a, ARCHIVE_OK, archive_read_open_memory(a, buff, used));
165
166	/* Read entries */
167	for (i = 0; entries[i].path != NULL; i++) {
168		if (dironly && (entries[i].mode & AE_IFMT) != S_IFDIR)
169			continue;
170		assertEqualIntA(a, ARCHIVE_OK, archive_read_next_header(a, &ae));
171		assertEqualInt(entries[i].mtime, archive_entry_mtime(ae));
172		assertEqualInt(entries[i].mode, archive_entry_mode(ae));
173		assertEqualInt(entries[i].uid, archive_entry_uid(ae));
174		assertEqualInt(entries[i].gid, archive_entry_gid(ae));
175		if (i > 0)
176			assertEqualString(entries[i].path + 2,
177				archive_entry_pathname(ae));
178		else
179			assertEqualString(entries[i].path,
180				archive_entry_pathname(ae));
181		if ((entries[i].mode & AE_IFMT) != S_IFDIR)
182			assertEqualInt(8, archive_entry_size(ae));
183	}
184	assertEqualIntA(a, ARCHIVE_OK, archive_read_close(a));
185	assertEqualInt(ARCHIVE_OK, archive_read_free(a));
186}
187
188DEFINE_TEST(test_write_format_mtree_classic)
189{
190	/* Generate classic format. */
191	test_write_format_mtree_sub(0);
192	/* Generate classic format and Write directory only. */
193	test_write_format_mtree_sub(1);
194}
195