1 /*-
2 * Copyright (c) 2012 Michihiro NAKAJIMA
3 * Copyright (c) 2015 Okhotnikov Kirill
4 * All rights reserved.
5 *
6 * Redistribution and use in source and binary forms, with or without
7 * modification, are permitted provided that the following conditions
8 * are met:
9 * 1. Redistributions of source code must retain the above copyright
10 * notice, this list of conditions and the following disclaimer.
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
28 #include "test.h"
29 __FBSDID("$FreeBSD$");
30
31 static void
test_format_filter_by_ext(const char * output_file,int format_id,int filter_id,int dot_stored,const char * def_ext)32 test_format_filter_by_ext(const char *output_file,
33 int format_id, int filter_id, int dot_stored, const char * def_ext)
34 {
35 struct archive_entry *ae;
36 struct archive *a;
37 size_t used;
38 size_t buffsize = 1024 * 1024;
39 char *buff;
40 int r;
41
42 assert((buff = malloc(buffsize)) != NULL);
43 if (buff == NULL)
44 return;
45
46 /* Create a new archive in memory. */
47 assert((a = archive_write_new()) != NULL);
48 if( def_ext == NULL)
49 r = archive_write_set_format_filter_by_ext(a, output_file);
50 else
51 r = archive_write_set_format_filter_by_ext_def(a, output_file, def_ext);
52 if (r == ARCHIVE_WARN) {
53 skipping("%s format not fully supported on this platform",
54 archive_format_name(a));
55 assertEqualInt(ARCHIVE_OK, archive_write_free(a));
56 free(buff);
57 return;
58 } else if (r == ARCHIVE_FATAL &&
59 (strcmp(archive_error_string(a),
60 "lzma compression not supported on this platform") == 0 ||
61 strcmp(archive_error_string(a),
62 "xz compression not supported on this platform") == 0)) {
63 const char *filter_name = archive_filter_name(a, 0);
64 skipping("%s filter not supported on this platform", filter_name);
65 assertEqualInt(ARCHIVE_OK, archive_write_free(a));
66 free(buff);
67 return;
68 } else {
69 if (!assertEqualIntA(a, ARCHIVE_OK, r)) {
70 assertEqualInt(ARCHIVE_OK, archive_write_free(a));
71 free(buff);
72 return;
73 }
74 }
75
76 assertEqualIntA(a, ARCHIVE_OK,
77 archive_write_open_memory(a, buff, buffsize, &used));
78
79 /*
80 * Write a file to it.
81 */
82 assert((ae = archive_entry_new()) != NULL);
83 archive_entry_set_mtime(ae, 1, 0);
84 assertEqualInt(1, archive_entry_mtime(ae));
85 archive_entry_set_ctime(ae, 1, 0);
86 assertEqualInt(1, archive_entry_ctime(ae));
87 archive_entry_set_atime(ae, 1, 0);
88 assertEqualInt(1, archive_entry_atime(ae));
89 archive_entry_copy_pathname(ae, "file");
90 assertEqualString("file", archive_entry_pathname(ae));
91 archive_entry_set_mode(ae, AE_IFREG | 0755);
92 assertEqualInt((AE_IFREG | 0755), archive_entry_mode(ae));
93 archive_entry_set_size(ae, 8);
94 assertEqualInt(0, archive_write_header(a, ae));
95 archive_entry_free(ae);
96 assertEqualInt(8, archive_write_data(a, "12345678", 8));
97
98 /* Close out the archive. */
99 assertEqualIntA(a, ARCHIVE_OK, archive_write_close(a));
100 assertEqualInt(ARCHIVE_OK, archive_write_free(a));
101
102 if (format_id > 0) {
103 /*
104 * Now, read the data back.
105 */
106 /* With the test memory reader -- seeking mode. */
107 assert((a = archive_read_new()) != NULL);
108 assertEqualIntA(a, ARCHIVE_OK,
109 archive_read_support_format_all(a));
110 assertEqualIntA(a, ARCHIVE_OK,
111 archive_read_support_filter_all(a));
112 assertEqualIntA(a, ARCHIVE_OK,
113 read_open_memory_seek(a, buff, used, 7));
114
115 if (dot_stored & 1) {
116 assertEqualIntA(a, ARCHIVE_OK,
117 archive_read_next_header(a, &ae));
118 assertEqualString(".", archive_entry_pathname(ae));
119 assertEqualInt(AE_IFDIR, archive_entry_filetype(ae));
120 }
121 /*
122 * Read and verify the file.
123 */
124 assertEqualIntA(a, ARCHIVE_OK, archive_read_next_header(a, &ae));
125 assertEqualInt(1, archive_entry_mtime(ae));
126 if (dot_stored & 2) {
127 assertEqualString("./file", archive_entry_pathname(ae));
128 } else {
129 assertEqualString("file", archive_entry_pathname(ae));
130 }
131 assertEqualInt(AE_IFREG, archive_entry_filetype(ae));
132 assertEqualInt(8, archive_entry_size(ae));
133
134 /* Verify the end of the archive. */
135 assertEqualIntA(a, ARCHIVE_EOF, archive_read_next_header(a, &ae));
136
137 /* Verify archive format. */
138 assertEqualIntA(a, filter_id,
139 archive_filter_code(a, 0));
140 assertEqualIntA(a, format_id, archive_format(a) & ARCHIVE_FORMAT_BASE_MASK );
141
142 assertEqualInt(ARCHIVE_OK, archive_read_close(a));
143 assertEqualInt(ARCHIVE_OK, archive_read_free(a));
144 }
145 free(buff);
146 }
147
DEFINE_TEST(test_archive_write_set_format_filter_by_ext_7zip)148 DEFINE_TEST(test_archive_write_set_format_filter_by_ext_7zip)
149 {
150 test_format_filter_by_ext("./data/test.7z", ARCHIVE_FORMAT_7ZIP, ARCHIVE_FILTER_NONE, 0, NULL);
151 }
152
DEFINE_TEST(test_archive_write_set_format_filter_by_ext_zip)153 DEFINE_TEST(test_archive_write_set_format_filter_by_ext_zip)
154 {
155 test_format_filter_by_ext("./data/test.zip", ARCHIVE_FORMAT_ZIP, ARCHIVE_FILTER_NONE, 0, NULL);
156 }
157
DEFINE_TEST(test_archive_write_set_format_filter_by_ext_jar)158 DEFINE_TEST(test_archive_write_set_format_filter_by_ext_jar)
159 {
160 test_format_filter_by_ext("./data/test.jar", ARCHIVE_FORMAT_ZIP, ARCHIVE_FILTER_NONE, 0, NULL);
161 }
162
DEFINE_TEST(test_archive_write_set_format_filter_by_ext_a)163 DEFINE_TEST(test_archive_write_set_format_filter_by_ext_a)
164 {
165 test_format_filter_by_ext("./data/test.a", ARCHIVE_FORMAT_AR, ARCHIVE_FILTER_NONE, 0, NULL);
166 }
167
DEFINE_TEST(test_archive_write_set_format_filter_by_ext_ar)168 DEFINE_TEST(test_archive_write_set_format_filter_by_ext_ar)
169 {
170 test_format_filter_by_ext("./data/test.ar", ARCHIVE_FORMAT_AR, ARCHIVE_FILTER_NONE, 0, NULL);
171 }
172
DEFINE_TEST(test_archive_write_set_format_filter_by_ext_cpio)173 DEFINE_TEST(test_archive_write_set_format_filter_by_ext_cpio)
174 {
175 test_format_filter_by_ext("./data/test.cpio", ARCHIVE_FORMAT_CPIO, ARCHIVE_FILTER_NONE, 0, NULL);
176 }
177
DEFINE_TEST(test_archive_write_set_format_filter_by_ext_iso)178 DEFINE_TEST(test_archive_write_set_format_filter_by_ext_iso)
179 {
180 test_format_filter_by_ext("./data/test.iso", ARCHIVE_FORMAT_ISO9660, ARCHIVE_FILTER_NONE, 1, NULL);
181 }
182
DEFINE_TEST(test_archive_write_set_format_filter_by_ext_tar)183 DEFINE_TEST(test_archive_write_set_format_filter_by_ext_tar)
184 {
185 test_format_filter_by_ext("./data/test.tar", ARCHIVE_FORMAT_TAR, ARCHIVE_FILTER_NONE, 0, NULL);
186 }
187
DEFINE_TEST(test_archive_write_set_format_filter_by_ext_tar_gz)188 DEFINE_TEST(test_archive_write_set_format_filter_by_ext_tar_gz)
189 {
190 test_format_filter_by_ext("./data/test.tar.gz", ARCHIVE_FORMAT_TAR, ARCHIVE_FILTER_GZIP, 20, NULL);
191 }
192
DEFINE_TEST(test_archive_write_set_format_filter_by_ext_tar_bz2)193 DEFINE_TEST(test_archive_write_set_format_filter_by_ext_tar_bz2)
194 {
195 test_format_filter_by_ext("./data/test.tar.bz2", ARCHIVE_FORMAT_TAR, ARCHIVE_FILTER_BZIP2, 0, NULL);
196 }
197
DEFINE_TEST(test_archive_write_set_format_filter_by_ext_tar_xz)198 DEFINE_TEST(test_archive_write_set_format_filter_by_ext_tar_xz)
199 {
200 test_format_filter_by_ext("./data/test.tar.xz", ARCHIVE_FORMAT_TAR, ARCHIVE_FILTER_XZ, 0, NULL);
201 }
202
DEFINE_TEST(test_archive_write_set_format_filter_by_no_ext_def_zip)203 DEFINE_TEST(test_archive_write_set_format_filter_by_no_ext_def_zip)
204 {
205 test_format_filter_by_ext("./data/test", ARCHIVE_FORMAT_ZIP, ARCHIVE_FILTER_NONE, 0, ".zip");
206 }
207
DEFINE_TEST(test_archive_write_set_format_filter_by_ext_tar_bz2_def_zip)208 DEFINE_TEST(test_archive_write_set_format_filter_by_ext_tar_bz2_def_zip)
209 {
210 test_format_filter_by_ext("./data/test.tar.bz2", ARCHIVE_FORMAT_TAR, ARCHIVE_FILTER_BZIP2, 0, ".zip");
211 }
212