2 * Copyright (c) 2010 Michihiro NAKAJIMA
3 * Copyright (c) 2007-2009 Tim Kientzle
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
9 * 1. Redistributions of source code must retain the above copyright
10 * notice, this list of conditions and the following disclaimer
11 * in this position and unchanged.
12 * 2. Redistributions in binary form must reproduce the above copyright
13 * notice, this list of conditions and the following disclaimer in the
14 * documentation and/or other materials provided with the distribution.
16 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS OR
17 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
18 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
19 * IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY DIRECT, INDIRECT,
20 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
21 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
22 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
23 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
24 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
25 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
29 __FBSDID("$FreeBSD$");
32 * A basic exercise of lzip reading and writing.
36 DEFINE_TEST(test_write_filter_lzip)
38 struct archive_entry *ae;
41 size_t buffsize, datasize;
47 assert(NULL != (buff = (char *)malloc(buffsize)));
52 assert(NULL != (data = (char *)malloc(datasize)));
57 memset(data, 0, datasize);
60 * Write a 100 files and read them all back.
62 assert((a = archive_write_new()) != NULL);
63 assertEqualIntA(a, ARCHIVE_OK, archive_write_set_format_ustar(a));
64 r = archive_write_add_filter_lzip(a);
65 if (r == ARCHIVE_FATAL) {
66 skipping("lzip writing not supported on this platform");
67 assertEqualInt(ARCHIVE_OK, archive_write_free(a));
72 assertEqualIntA(a, ARCHIVE_OK,
73 archive_write_set_bytes_per_block(a, 10));
74 assertEqualInt(ARCHIVE_FILTER_LZIP, archive_filter_code(a, 0));
75 assertEqualString("lzip", archive_filter_name(a, 0));
76 assertEqualIntA(a, ARCHIVE_OK,
77 archive_write_open_memory(a, buff, buffsize, &used1));
78 assertEqualInt(ARCHIVE_FILTER_LZIP, archive_filter_code(a, 0));
79 assertEqualString("lzip", archive_filter_name(a, 0));
80 assert((ae = archive_entry_new()) != NULL);
81 archive_entry_set_filetype(ae, AE_IFREG);
82 archive_entry_set_size(ae, datasize);
83 for (i = 0; i < 100; i++) {
84 sprintf(path, "file%03d", i);
85 archive_entry_copy_pathname(ae, path);
86 assertEqualIntA(a, ARCHIVE_OK, archive_write_header(a, ae));
88 == (size_t)archive_write_data(a, data, datasize));
90 archive_entry_free(ae);
91 assertEqualIntA(a, ARCHIVE_OK, archive_write_close(a));
92 assertEqualInt(ARCHIVE_OK, archive_write_free(a));
94 assert((a = archive_read_new()) != NULL);
95 assertEqualIntA(a, ARCHIVE_OK, archive_read_support_format_all(a));
96 r = archive_read_support_filter_lzip(a);
97 if (r == ARCHIVE_WARN) {
98 skipping("Can't verify lzip writing by reading back;"
99 " lzip reading not fully supported on this platform");
101 assertEqualIntA(a, ARCHIVE_OK,
102 archive_read_support_filter_all(a));
103 assertEqualIntA(a, ARCHIVE_OK,
104 archive_read_open_memory(a, buff, used1));
105 for (i = 0; i < 100; i++) {
106 sprintf(path, "file%03d", i);
107 if (!assertEqualInt(ARCHIVE_OK,
108 archive_read_next_header(a, &ae)))
110 assertEqualString(path, archive_entry_pathname(ae));
111 assertEqualInt((int)datasize, archive_entry_size(ae));
113 assertEqualIntA(a, ARCHIVE_OK, archive_read_close(a));
115 assertEqualInt(ARCHIVE_OK, archive_read_free(a));
118 * Repeat the cycle again, this time setting some compression
121 assert((a = archive_write_new()) != NULL);
122 assertEqualIntA(a, ARCHIVE_OK, archive_write_set_format_ustar(a));
123 assertEqualIntA(a, ARCHIVE_OK,
124 archive_write_set_bytes_per_block(a, 10));
125 assertEqualIntA(a, ARCHIVE_OK, archive_write_add_filter_lzip(a));
126 assertEqualIntA(a, ARCHIVE_FAILED,
127 archive_write_set_filter_option(a, NULL, "nonexistent-option", "0"));
128 assertEqualIntA(a, ARCHIVE_FAILED,
129 archive_write_set_filter_option(a, NULL, "compression-level", "abc"));
130 assertEqualIntA(a, ARCHIVE_FAILED,
131 archive_write_set_filter_option(a, NULL, "compression-level", "99"));
132 assertEqualIntA(a, ARCHIVE_OK,
133 archive_write_set_filter_option(a, NULL, "compression-level", "9"));
134 assertEqualIntA(a, ARCHIVE_OK, archive_write_open_memory(a, buff, buffsize, &used2));
135 for (i = 0; i < 100; i++) {
136 sprintf(path, "file%03d", i);
137 assert((ae = archive_entry_new()) != NULL);
138 archive_entry_copy_pathname(ae, path);
139 archive_entry_set_size(ae, datasize);
140 archive_entry_set_filetype(ae, AE_IFREG);
141 assertEqualIntA(a, ARCHIVE_OK, archive_write_header(a, ae));
142 assertA(datasize == (size_t)archive_write_data(a, data, datasize));
143 archive_entry_free(ae);
145 assertEqualIntA(a, ARCHIVE_OK, archive_write_close(a));
146 assertEqualInt(ARCHIVE_OK, archive_write_free(a));
149 assert((a = archive_read_new()) != NULL);
150 assertEqualIntA(a, ARCHIVE_OK, archive_read_support_format_all(a));
151 r = archive_read_support_filter_lzip(a);
152 if (r == ARCHIVE_WARN) {
153 skipping("lzip reading not fully supported on this platform");
155 assertEqualIntA(a, ARCHIVE_OK,
156 archive_read_support_filter_all(a));
157 assertEqualIntA(a, ARCHIVE_OK,
158 archive_read_open_memory(a, buff, used2));
159 for (i = 0; i < 100; i++) {
160 sprintf(path, "file%03d", i);
161 failure("Trying to read %s", path);
162 if (!assertEqualIntA(a, ARCHIVE_OK,
163 archive_read_next_header(a, &ae)))
165 assertEqualString(path, archive_entry_pathname(ae));
166 assertEqualInt((int)datasize, archive_entry_size(ae));
168 assertEqualIntA(a, ARCHIVE_OK, archive_read_close(a));
170 assertEqualInt(ARCHIVE_OK, archive_read_free(a));
173 * Repeat again, with much lower compression.
175 assert((a = archive_write_new()) != NULL);
176 assertEqualIntA(a, ARCHIVE_OK, archive_write_set_format_ustar(a));
177 assertEqualIntA(a, ARCHIVE_OK,
178 archive_write_set_bytes_per_block(a, 10));
179 assertEqualIntA(a, ARCHIVE_OK, archive_write_add_filter_lzip(a));
180 assertEqualIntA(a, ARCHIVE_OK,
181 archive_write_set_filter_option(a, NULL, "compression-level", "0"));
182 assertEqualIntA(a, ARCHIVE_OK, archive_write_open_memory(a, buff, buffsize, &used2));
183 for (i = 0; i < 100; i++) {
184 sprintf(path, "file%03d", i);
185 assert((ae = archive_entry_new()) != NULL);
186 archive_entry_copy_pathname(ae, path);
187 archive_entry_set_size(ae, datasize);
188 archive_entry_set_filetype(ae, AE_IFREG);
189 assertEqualIntA(a, ARCHIVE_OK, archive_write_header(a, ae));
190 failure("Writing file %s", path);
191 assertEqualIntA(a, datasize,
192 (size_t)archive_write_data(a, data, datasize));
193 archive_entry_free(ae);
195 assertEqualIntA(a, ARCHIVE_OK, archive_write_close(a));
196 assertEqualInt(ARCHIVE_OK, archive_write_free(a));
198 /* Level 0 really does result in larger data. */
199 failure("Compression-level=0 wrote %d bytes; default wrote %d bytes",
200 (int)used2, (int)used1);
201 assert(used2 > used1);
203 assert((a = archive_read_new()) != NULL);
204 assertEqualIntA(a, ARCHIVE_OK, archive_read_support_format_all(a));
205 assertEqualIntA(a, ARCHIVE_OK, archive_read_support_filter_all(a));
206 r = archive_read_support_filter_lzip(a);
207 if (r == ARCHIVE_WARN) {
208 skipping("lzip reading not fully supported on this platform");
210 assertEqualIntA(a, ARCHIVE_OK,
211 archive_read_open_memory(a, buff, used2));
212 for (i = 0; i < 100; i++) {
213 sprintf(path, "file%03d", i);
214 if (!assertEqualInt(ARCHIVE_OK,
215 archive_read_next_header(a, &ae)))
217 assertEqualString(path, archive_entry_pathname(ae));
218 assertEqualInt((int)datasize, archive_entry_size(ae));
220 assertEqualIntA(a, ARCHIVE_OK, archive_read_close(a));
222 assertEqualInt(ARCHIVE_OK, archive_read_free(a));
225 * Test various premature shutdown scenarios to make sure we
226 * don't crash or leak memory.
228 assert((a = archive_write_new()) != NULL);
229 assertEqualIntA(a, ARCHIVE_OK, archive_write_add_filter_lzip(a));
230 assertEqualInt(ARCHIVE_OK, archive_write_free(a));
232 assert((a = archive_write_new()) != NULL);
233 assertEqualIntA(a, ARCHIVE_OK, archive_write_add_filter_lzip(a));
234 assertEqualInt(ARCHIVE_OK, archive_write_close(a));
235 assertEqualInt(ARCHIVE_OK, archive_write_free(a));
237 assert((a = archive_write_new()) != NULL);
238 assertEqualIntA(a, ARCHIVE_OK, archive_write_set_format_ustar(a));
239 assertEqualIntA(a, ARCHIVE_OK, archive_write_add_filter_lzip(a));
240 assertEqualInt(ARCHIVE_OK, archive_write_close(a));
241 assertEqualInt(ARCHIVE_OK, archive_write_free(a));
243 assert((a = archive_write_new()) != NULL);
244 assertEqualIntA(a, ARCHIVE_OK, archive_write_set_format_ustar(a));
245 assertEqualIntA(a, ARCHIVE_OK, archive_write_add_filter_lzip(a));
246 assertEqualIntA(a, ARCHIVE_OK, archive_write_open_memory(a, buff, buffsize, &used2));
247 assertEqualInt(ARCHIVE_OK, archive_write_close(a));
248 assertEqualInt(ARCHIVE_OK, archive_write_free(a));