1 /*-
2 * Copyright (c) 2003-2007 Tim Kientzle
3 * Copyright (c) 2012 Andres Mejia
4 * Copyright (c) 2011-2012 Michihiro NAKAJIMA
5 * All rights reserved.
6 *
7 * Redistribution and use in source and binary forms, with or without
8 * modification, are permitted provided that the following conditions
9 * are met:
10 * 1. Redistributions of source code must retain the above copyright
11 * notice, this list of conditions and the following disclaimer.
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.
15 *
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.
26 */
27 #include "test.h"
28
DEFINE_TEST(test_read_set_format)29 DEFINE_TEST(test_read_set_format)
30 {
31 char buff[64];
32 const char reffile[] = "test_read_format_rar.rar";
33 const char test_txt[] = "test text document\r\n";
34 int size = sizeof(test_txt)-1;
35 struct archive_entry *ae;
36 struct archive *a;
37
38 extract_reference_file(reffile);
39 assert((a = archive_read_new()) != NULL);
40 assertA(0 == archive_read_set_format(a, ARCHIVE_FORMAT_RAR));
41 assertA(0 == archive_read_append_filter(a, ARCHIVE_FILTER_NONE));
42 assertA(0 == archive_read_open_filename(a, reffile, 10240));
43
44 /* First header. */
45 assertA(0 == archive_read_next_header(a, &ae));
46 assertEqualString("test.txt", archive_entry_pathname(ae));
47 assertA((int)archive_entry_mtime(ae));
48 assertA((int)archive_entry_ctime(ae));
49 assertA((int)archive_entry_atime(ae));
50 assertEqualInt(20, archive_entry_size(ae));
51 assertEqualInt(33188, archive_entry_mode(ae));
52 assertA(size == archive_read_data(a, buff, size));
53 assertEqualMem(buff, test_txt, size);
54
55 /* Second header. */
56 assertA(0 == archive_read_next_header(a, &ae));
57 assertEqualString("testlink", archive_entry_pathname(ae));
58 assertA((int)archive_entry_mtime(ae));
59 assertA((int)archive_entry_ctime(ae));
60 assertA((int)archive_entry_atime(ae));
61 assertEqualInt(0, archive_entry_size(ae));
62 assertEqualInt(41471, archive_entry_mode(ae));
63 assertEqualString("test.txt", archive_entry_symlink(ae));
64 assertEqualIntA(a, 0, archive_read_data(a, buff, sizeof(buff)));
65
66 /* Third header. */
67 assertA(0 == archive_read_next_header(a, &ae));
68 assertEqualString("testdir/test.txt", archive_entry_pathname(ae));
69 assertA((int)archive_entry_mtime(ae));
70 assertA((int)archive_entry_ctime(ae));
71 assertA((int)archive_entry_atime(ae));
72 assertEqualInt(20, archive_entry_size(ae));
73 assertEqualInt(33188, archive_entry_mode(ae));
74 assertA(size == archive_read_data(a, buff, size));
75 assertEqualMem(buff, test_txt, size);
76
77 /* Fourth header. */
78 assertA(0 == archive_read_next_header(a, &ae));
79 assertEqualString("testdir", archive_entry_pathname(ae));
80 assertA((int)archive_entry_mtime(ae));
81 assertA((int)archive_entry_ctime(ae));
82 assertA((int)archive_entry_atime(ae));
83 assertEqualInt(0, archive_entry_size(ae));
84 assertEqualInt(16877, archive_entry_mode(ae));
85
86 /* Fifth header. */
87 assertA(0 == archive_read_next_header(a, &ae));
88 assertEqualString("testemptydir", archive_entry_pathname(ae));
89 assertA((int)archive_entry_mtime(ae));
90 assertA((int)archive_entry_ctime(ae));
91 assertA((int)archive_entry_atime(ae));
92 assertEqualInt(0, archive_entry_size(ae));
93 assertEqualInt(16877, archive_entry_mode(ae));
94
95 /* Test EOF */
96 assertA(1 == archive_read_next_header(a, &ae));
97 assertEqualInt(5, archive_file_count(a));
98 assertEqualIntA(a, ARCHIVE_OK, archive_read_close(a));
99 assertEqualInt(ARCHIVE_OK, archive_read_free(a));
100 }
101
DEFINE_TEST(test_read_set_wrong_format)102 DEFINE_TEST(test_read_set_wrong_format)
103 {
104 const char reffile[] = "test_read_format_zip.zip";
105 struct archive_entry *ae;
106 struct archive *a;
107
108 extract_reference_file(reffile);
109 assert((a = archive_read_new()) != NULL);
110 assertA(0 == archive_read_set_format(a, ARCHIVE_FORMAT_RAR));
111 assertA(0 == archive_read_append_filter(a, ARCHIVE_FILTER_NONE));
112 assertA(0 == archive_read_open_filename(a, reffile, 10240));
113
114 /* Check that this actually fails, then close the archive. */
115 assertA(archive_read_next_header(a, &ae) < (ARCHIVE_WARN));
116 assertEqualIntA(a, ARCHIVE_OK, archive_read_close(a));
117 assertEqualInt(ARCHIVE_OK, archive_read_free(a));
118 }
119
120 static unsigned char archive[] = {
121 31,139,8,0,222,'C','p','C',0,3,211,'c',160,'=','0','0','0','0','7','5','U',
122 0,210,134,230,166,6,200,'4',28,'(',24,26,24,27,155,24,152,24,154,27,155,')',
123 24,24,26,152,154,25,'2','(',152,210,193,'m',12,165,197,'%',137,'E','@',167,
124 148,'d',230,226,'U','G','H',30,234,15,'8','=',10,'F',193,'(',24,5,131,28,
125 0,0,29,172,5,240,0,6,0,0};
126
DEFINE_TEST(test_read_append_filter)127 DEFINE_TEST(test_read_append_filter)
128 {
129 struct archive_entry *ae;
130 struct archive *a;
131 int r;
132
133 assert((a = archive_read_new()) != NULL);
134 assertA(0 == archive_read_set_format(a, ARCHIVE_FORMAT_TAR));
135 r = archive_read_append_filter(a, ARCHIVE_FILTER_GZIP);
136 if (r != ARCHIVE_OK && archive_zlib_version() == NULL && !canGzip()) {
137 skipping("gzip tests require zlib or working gzip command");
138 assertEqualInt(ARCHIVE_OK, archive_read_free(a));
139 return;
140 }
141 if (r == ARCHIVE_WARN && canGzip())
142 assertEqualString(archive_error_string(a), "Using external gzip program");
143 else
144 assertEqualIntA(a, ARCHIVE_OK, r);
145 assertEqualInt(ARCHIVE_OK,
146 archive_read_open_memory(a, archive, sizeof(archive)));
147 assertEqualInt(ARCHIVE_OK, archive_read_next_header(a, &ae));
148 assertEqualInt(1, archive_file_count(a));
149 assertEqualInt(archive_filter_code(a, 0), ARCHIVE_COMPRESSION_GZIP);
150 assertEqualInt(archive_format(a), ARCHIVE_FORMAT_TAR_USTAR);
151 assertEqualInt(ARCHIVE_OK, archive_read_close(a));
152 assertEqualInt(ARCHIVE_OK,archive_read_free(a));
153 }
154
DEFINE_TEST(test_read_append_wrong_filter)155 DEFINE_TEST(test_read_append_wrong_filter)
156 {
157 struct archive_entry *ae;
158 struct archive *a;
159 int r;
160
161 assert((a = archive_read_new()) != NULL);
162 assertA(0 == archive_read_set_format(a, ARCHIVE_FORMAT_TAR));
163 r = archive_read_append_filter(a, ARCHIVE_FILTER_XZ);
164 if (r == ARCHIVE_WARN && !canXz()) {
165 skipping("xz reading not fully supported on this platform");
166 assertEqualInt(ARCHIVE_OK, archive_read_free(a));
167 return;
168 }
169 assertEqualInt(ARCHIVE_OK,
170 archive_read_open_memory(a, archive, sizeof(archive)));
171 assertA(archive_read_next_header(a, &ae) < (ARCHIVE_WARN));
172 if (r == ARCHIVE_WARN && canXz()) {
173 assertEqualIntA(a, ARCHIVE_WARN, archive_read_close(a));
174 } else {
175 assertEqualIntA(a, ARCHIVE_OK, archive_read_close(a));
176 }
177 assertEqualInt(ARCHIVE_OK,archive_read_free(a));
178 }
179
DEFINE_TEST(test_read_append_lzop_filter)180 DEFINE_TEST(test_read_append_lzop_filter)
181 {
182 struct archive *a;
183 int r;
184
185 assert((a = archive_read_new()) != NULL);
186 assertA(0 == archive_read_set_format(a, ARCHIVE_FORMAT_TAR));
187 r = archive_read_append_filter(a, ARCHIVE_FILTER_LZOP);
188 if (archive_liblzo2_version() != NULL) {
189 assertEqualIntA(a, ARCHIVE_OK, r);
190 } else if (canLzop()) {
191 // We're using an external program
192 assertEqualIntA(a, ARCHIVE_WARN, r);
193 }
194
195 archive_read_free(a);
196 }
197
DEFINE_TEST(test_read_append_grzip_filter)198 DEFINE_TEST(test_read_append_grzip_filter)
199 {
200 struct archive *a;
201 int r;
202
203 assert((a = archive_read_new()) != NULL);
204 assertA(0 == archive_read_set_format(a, ARCHIVE_FORMAT_TAR));
205 r = archive_read_append_filter(a, ARCHIVE_FILTER_GRZIP);
206 // Grzip currently always uses an external program.
207 if (canGrzip()) {
208 assertEqualIntA(a, ARCHIVE_WARN, r);
209 }
210
211 archive_read_free(a);
212 }
213
DEFINE_TEST(test_read_append_filter_program)214 DEFINE_TEST(test_read_append_filter_program)
215 {
216 struct archive_entry *ae;
217 struct archive *a;
218
219 if (!canGzip()) {
220 skipping("Can't run gzip program on this platform");
221 return;
222 }
223 assert((a = archive_read_new()) != NULL);
224 assertA(0 == archive_read_set_format(a, ARCHIVE_FORMAT_TAR));
225 assertEqualIntA(a, ARCHIVE_OK,
226 archive_read_append_filter_program(a, "gzip -d"));
227 assertEqualIntA(a, ARCHIVE_OK,
228 archive_read_open_memory(a, archive, sizeof(archive)));
229 assertEqualIntA(a, ARCHIVE_OK,
230 archive_read_next_header(a, &ae));
231 assertEqualInt(archive_filter_code(a, 0), ARCHIVE_FILTER_PROGRAM);
232 assertEqualInt(archive_format(a), ARCHIVE_FORMAT_TAR_USTAR);
233 assertEqualIntA(a, ARCHIVE_OK, archive_read_close(a));
234 assertEqualInt(ARCHIVE_OK, archive_read_free(a));
235 }
236
DEFINE_TEST(test_read_append_filter_wrong_program)237 DEFINE_TEST(test_read_append_filter_wrong_program)
238 {
239 struct archive_entry *ae;
240 struct archive *a;
241 #if !defined(_WIN32) || defined(__CYGWIN__)
242 FILE * fp;
243 int fd;
244 fpos_t pos;
245 #endif
246
247 /*
248 * If we have "bunzip2 -q", try using that.
249 */
250 if (!canRunCommand("bunzip2 -h", NULL)) {
251 skipping("Can't run bunzip2 program on this platform");
252 return;
253 }
254
255 #if !defined(_WIN32) || defined(__CYGWIN__)
256 /* bunzip2 will write to stderr, redirect it to a file */
257 fflush(stderr);
258 fgetpos(stderr, &pos);
259 assert((fd = dup(fileno(stderr))) != -1);
260 fp = freopen("stderr1", "w", stderr);
261 #endif
262
263 assert((a = archive_read_new()) != NULL);
264 assertA(0 == archive_read_set_format(a, ARCHIVE_FORMAT_TAR));
265 assertEqualIntA(a, ARCHIVE_OK,
266 archive_read_append_filter_program(a, "bunzip2 -q"));
267 assertEqualIntA(a, ARCHIVE_OK,
268 archive_read_open_memory(a, archive, sizeof(archive)));
269 assertA(archive_read_next_header(a, &ae) < (ARCHIVE_WARN));
270 assertEqualIntA(a, ARCHIVE_WARN, archive_read_close(a));
271 assertEqualInt(ARCHIVE_OK, archive_read_free(a));
272
273 #if !defined(_WIN32) || defined(__CYGWIN__)
274 /* restore stderr */
275 if (fp != NULL) {
276 fflush(stderr);
277 dup2(fd, fileno(stderr));
278 clearerr(stderr);
279 (void)fsetpos(stderr, &pos);
280 }
281 close(fd);
282 #endif
283 }
284