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 * 2. Redistributions in binary form must reproduce the above copyright 11 * notice, this list of conditions and the following disclaimer in the 12 * documentation and/or other materials provided with the distribution. 13 * 14 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS OR 15 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 16 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 17 * IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY DIRECT, INDIRECT, 18 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 19 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 20 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 21 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 22 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 23 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 24 */ 25 26 27#include "test.h" 28__FBSDID("$FreeBSD$"); 29 30static void 31test_filter_by_name(const char *filter_name, int filter_code, 32 int (*can_filter_prog)(void)) 33{ 34 struct archive_entry *ae; 35 struct archive *a; 36 size_t used; 37 size_t buffsize = 1024 * 128; 38 char *buff; 39 int r; 40 41 assert((buff = malloc(buffsize)) != NULL); 42 if (buff == NULL) 43 return; 44 45 /* Create a new archive in memory. */ 46 assert((a = archive_write_new()) != NULL); 47 assertEqualIntA(a, ARCHIVE_OK, archive_write_set_format_ustar(a)); 48 r = archive_write_add_filter_by_name(a, filter_name); 49 if (r == ARCHIVE_WARN) { 50 if (!can_filter_prog()) { 51 skipping("%s filter not suported on this platform", 52 filter_name); 53 assertEqualInt(ARCHIVE_OK, archive_write_free(a)); 54 free(buff); 55 return; 56 } 57 } else if (r == ARCHIVE_FATAL && 58 (strcmp(archive_error_string(a), 59 "lzma compression not supported on this platform") == 0 || 60 strcmp(archive_error_string(a), 61 "xz compression not supported on this platform") == 0)) { 62 skipping("%s filter not suported 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 assertEqualIntA(a, ARCHIVE_OK, archive_write_set_bytes_per_block(a, 10)); 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 /* 101 * Now, read the data back. 102 */ 103 assert((a = archive_read_new()) != NULL); 104 assertEqualIntA(a, ARCHIVE_OK, archive_read_support_format_all(a)); 105 assertEqualIntA(a, ARCHIVE_OK, archive_read_support_filter_all(a)); 106 assertEqualIntA(a, ARCHIVE_OK, archive_read_open_memory(a, buff, used)); 107 108 /* 109 * Read and verify the file. 110 */ 111 assertEqualIntA(a, ARCHIVE_OK, archive_read_next_header(a, &ae)); 112 assertEqualInt(1, archive_entry_mtime(ae)); 113 assertEqualString("file", archive_entry_pathname(ae)); 114 assertEqualInt(AE_IFREG, archive_entry_filetype(ae)); 115 assertEqualInt(8, archive_entry_size(ae)); 116 117 /* Verify the end of the archive. */ 118 assertEqualIntA(a, ARCHIVE_EOF, archive_read_next_header(a, &ae)); 119 120 /* Verify archive format. */ 121 assertEqualIntA(a, filter_code, archive_filter_code(a, 0)); 122 assertEqualIntA(a, ARCHIVE_FORMAT_TAR_USTAR, archive_format(a)); 123 124 assertEqualInt(ARCHIVE_OK, archive_read_close(a)); 125 assertEqualInt(ARCHIVE_OK, archive_read_free(a)); 126 free(buff); 127} 128 129static int 130canAlways(void) 131{ 132 return 1; 133} 134 135static int 136cannot(void) 137{ 138 return 0; 139} 140 141DEFINE_TEST(test_archive_write_add_filter_by_name_b64encode) 142{ 143 test_filter_by_name("b64encode", ARCHIVE_FILTER_UU, canAlways); 144} 145 146DEFINE_TEST(test_archive_write_add_filter_by_name_bzip2) 147{ 148 test_filter_by_name("bzip2", ARCHIVE_FILTER_BZIP2, canBzip2); 149} 150 151DEFINE_TEST(test_archive_write_add_filter_by_name_compress) 152{ 153 test_filter_by_name("compress", ARCHIVE_FILTER_COMPRESS, canAlways); 154} 155 156DEFINE_TEST(test_archive_write_add_filter_by_name_grzip) 157{ 158 test_filter_by_name("grzip", ARCHIVE_FILTER_GRZIP, canGrzip); 159} 160 161DEFINE_TEST(test_archive_write_add_filter_by_name_gzip) 162{ 163 test_filter_by_name("gzip", ARCHIVE_FILTER_GZIP, canGzip); 164} 165 166DEFINE_TEST(test_archive_write_add_filter_by_name_lrzip) 167{ 168 test_filter_by_name("lrzip", ARCHIVE_FILTER_LRZIP, canLrzip); 169} 170 171DEFINE_TEST(test_archive_write_add_filter_by_name_lzip) 172{ 173 test_filter_by_name("lzip", ARCHIVE_FILTER_LZIP, cannot); 174} 175 176DEFINE_TEST(test_archive_write_add_filter_by_name_lzma) 177{ 178 test_filter_by_name("lzma", ARCHIVE_FILTER_LZMA, cannot); 179} 180 181DEFINE_TEST(test_archive_write_add_filter_by_name_lzop) 182{ 183 test_filter_by_name("lzop", ARCHIVE_FILTER_LZOP, canLzop); 184} 185 186DEFINE_TEST(test_archive_write_add_filter_by_name_uuencode) 187{ 188 test_filter_by_name("uuencode", ARCHIVE_FILTER_UU, canAlways); 189} 190 191DEFINE_TEST(test_archive_write_add_filter_by_name_xz) 192{ 193 test_filter_by_name("xz", ARCHIVE_FILTER_XZ, cannot); 194} 195