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