1248590Smm/*- 2248590Smm * Copyright (c) 2012 Michihiro NAKAJIMA 3248590Smm * All rights reserved. 4248590Smm * 5248590Smm * Redistribution and use in source and binary forms, with or without 6248590Smm * modification, are permitted provided that the following conditions 7248590Smm * are met: 8248590Smm * 1. Redistributions of source code must retain the above copyright 9248590Smm * notice, this list of conditions and the following disclaimer. 10248590Smm * 2. Redistributions in binary form must reproduce the above copyright 11248590Smm * notice, this list of conditions and the following disclaimer in the 12248590Smm * documentation and/or other materials provided with the distribution. 13248590Smm * 14248590Smm * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS OR 15248590Smm * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES 16248590Smm * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED. 17248590Smm * IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY DIRECT, INDIRECT, 18248590Smm * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT 19248590Smm * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, 20248590Smm * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY 21248590Smm * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT 22248590Smm * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF 23248590Smm * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE. 24248590Smm */ 25248590Smm 26248590Smm 27248590Smm#include "test.h" 28248590Smm__FBSDID("$FreeBSD$"); 29248590Smm 30248590Smmstatic void 31248590Smmtest_filter_by_name(const char *filter_name, int filter_code, 32248590Smm int (*can_filter_prog)(void)) 33248590Smm{ 34248590Smm struct archive_entry *ae; 35248590Smm struct archive *a; 36248590Smm size_t used; 37248590Smm size_t buffsize = 1024 * 128; 38248590Smm char *buff; 39248590Smm int r; 40248590Smm 41248590Smm assert((buff = malloc(buffsize)) != NULL); 42248590Smm if (buff == NULL) 43248590Smm return; 44248590Smm 45248590Smm /* Create a new archive in memory. */ 46248590Smm assert((a = archive_write_new()) != NULL); 47248590Smm assertEqualIntA(a, ARCHIVE_OK, archive_write_set_format_ustar(a)); 48248590Smm r = archive_write_add_filter_by_name(a, filter_name); 49248590Smm if (r == ARCHIVE_WARN) { 50248590Smm if (!can_filter_prog()) { 51248590Smm skipping("%s filter not suported on this platform", 52248590Smm filter_name); 53248590Smm assertEqualInt(ARCHIVE_OK, archive_write_free(a)); 54248590Smm free(buff); 55248590Smm return; 56248590Smm } 57248590Smm } else if (r == ARCHIVE_FATAL && 58248590Smm (strcmp(archive_error_string(a), 59248590Smm "lzma compression not supported on this platform") == 0 || 60248590Smm strcmp(archive_error_string(a), 61248590Smm "xz compression not supported on this platform") == 0)) { 62248590Smm skipping("%s filter not suported on this platform", filter_name); 63248590Smm assertEqualInt(ARCHIVE_OK, archive_write_free(a)); 64248590Smm free(buff); 65248590Smm return; 66248590Smm } else { 67248590Smm if (!assertEqualIntA(a, ARCHIVE_OK, r)) { 68248590Smm assertEqualInt(ARCHIVE_OK, archive_write_free(a)); 69248590Smm free(buff); 70248590Smm return; 71248590Smm } 72248590Smm } 73248590Smm assertEqualIntA(a, ARCHIVE_OK, archive_write_set_bytes_per_block(a, 10)); 74248590Smm assertEqualIntA(a, ARCHIVE_OK, 75248590Smm archive_write_open_memory(a, buff, buffsize, &used)); 76248590Smm 77248590Smm /* 78248590Smm * Write a file to it. 79248590Smm */ 80248590Smm assert((ae = archive_entry_new()) != NULL); 81248590Smm archive_entry_set_mtime(ae, 1, 0); 82248590Smm assertEqualInt(1, archive_entry_mtime(ae)); 83248590Smm archive_entry_set_ctime(ae, 1, 0); 84248590Smm assertEqualInt(1, archive_entry_ctime(ae)); 85248590Smm archive_entry_set_atime(ae, 1, 0); 86248590Smm assertEqualInt(1, archive_entry_atime(ae)); 87248590Smm archive_entry_copy_pathname(ae, "file"); 88248590Smm assertEqualString("file", archive_entry_pathname(ae)); 89248590Smm archive_entry_set_mode(ae, AE_IFREG | 0755); 90248590Smm assertEqualInt((AE_IFREG | 0755), archive_entry_mode(ae)); 91248590Smm archive_entry_set_size(ae, 8); 92248590Smm assertEqualInt(0, archive_write_header(a, ae)); 93248590Smm archive_entry_free(ae); 94248590Smm assertEqualInt(8, archive_write_data(a, "12345678", 8)); 95248590Smm 96248590Smm /* Close out the archive. */ 97248590Smm assertEqualIntA(a, ARCHIVE_OK, archive_write_close(a)); 98248590Smm assertEqualInt(ARCHIVE_OK, archive_write_free(a)); 99248590Smm 100248590Smm /* 101248590Smm * Now, read the data back. 102248590Smm */ 103248590Smm assert((a = archive_read_new()) != NULL); 104248590Smm assertEqualIntA(a, ARCHIVE_OK, archive_read_support_format_all(a)); 105248590Smm assertEqualIntA(a, ARCHIVE_OK, archive_read_support_filter_all(a)); 106248590Smm assertEqualIntA(a, ARCHIVE_OK, archive_read_open_memory(a, buff, used)); 107248590Smm 108248590Smm /* 109248590Smm * Read and verify the file. 110248590Smm */ 111248590Smm assertEqualIntA(a, ARCHIVE_OK, archive_read_next_header(a, &ae)); 112248590Smm assertEqualInt(1, archive_entry_mtime(ae)); 113248590Smm assertEqualString("file", archive_entry_pathname(ae)); 114248590Smm assertEqualInt(AE_IFREG, archive_entry_filetype(ae)); 115248590Smm assertEqualInt(8, archive_entry_size(ae)); 116248590Smm 117248590Smm /* Verify the end of the archive. */ 118248590Smm assertEqualIntA(a, ARCHIVE_EOF, archive_read_next_header(a, &ae)); 119248590Smm 120248590Smm /* Verify archive format. */ 121248590Smm assertEqualIntA(a, filter_code, archive_filter_code(a, 0)); 122248590Smm assertEqualIntA(a, ARCHIVE_FORMAT_TAR_USTAR, archive_format(a)); 123248590Smm 124248590Smm assertEqualInt(ARCHIVE_OK, archive_read_close(a)); 125248590Smm assertEqualInt(ARCHIVE_OK, archive_read_free(a)); 126248590Smm free(buff); 127248590Smm} 128248590Smm 129248590Smmstatic int 130248590SmmcanAlways(void) 131248590Smm{ 132248590Smm return 1; 133248590Smm} 134248590Smm 135248590Smmstatic int 136248590Smmcannot(void) 137248590Smm{ 138248590Smm return 0; 139248590Smm} 140248590Smm 141248590SmmDEFINE_TEST(test_archive_write_add_filter_by_name_b64encode) 142248590Smm{ 143248590Smm test_filter_by_name("b64encode", ARCHIVE_FILTER_UU, canAlways); 144248590Smm} 145248590Smm 146248590SmmDEFINE_TEST(test_archive_write_add_filter_by_name_bzip2) 147248590Smm{ 148248590Smm test_filter_by_name("bzip2", ARCHIVE_FILTER_BZIP2, canBzip2); 149248590Smm} 150248590Smm 151248590SmmDEFINE_TEST(test_archive_write_add_filter_by_name_compress) 152248590Smm{ 153248590Smm test_filter_by_name("compress", ARCHIVE_FILTER_COMPRESS, canAlways); 154248590Smm} 155248590Smm 156248590SmmDEFINE_TEST(test_archive_write_add_filter_by_name_grzip) 157248590Smm{ 158248590Smm test_filter_by_name("grzip", ARCHIVE_FILTER_GRZIP, canGrzip); 159248590Smm} 160248590Smm 161248590SmmDEFINE_TEST(test_archive_write_add_filter_by_name_gzip) 162248590Smm{ 163248590Smm test_filter_by_name("gzip", ARCHIVE_FILTER_GZIP, canGzip); 164248590Smm} 165248590Smm 166248590SmmDEFINE_TEST(test_archive_write_add_filter_by_name_lrzip) 167248590Smm{ 168248590Smm test_filter_by_name("lrzip", ARCHIVE_FILTER_LRZIP, canLrzip); 169248590Smm} 170248590Smm 171248590SmmDEFINE_TEST(test_archive_write_add_filter_by_name_lzip) 172248590Smm{ 173248590Smm test_filter_by_name("lzip", ARCHIVE_FILTER_LZIP, cannot); 174248590Smm} 175248590Smm 176248590SmmDEFINE_TEST(test_archive_write_add_filter_by_name_lzma) 177248590Smm{ 178248590Smm test_filter_by_name("lzma", ARCHIVE_FILTER_LZMA, cannot); 179248590Smm} 180248590Smm 181248590SmmDEFINE_TEST(test_archive_write_add_filter_by_name_lzop) 182248590Smm{ 183248590Smm test_filter_by_name("lzop", ARCHIVE_FILTER_LZOP, canLzop); 184248590Smm} 185248590Smm 186248590SmmDEFINE_TEST(test_archive_write_add_filter_by_name_uuencode) 187248590Smm{ 188248590Smm test_filter_by_name("uuencode", ARCHIVE_FILTER_UU, canAlways); 189248590Smm} 190248590Smm 191248590SmmDEFINE_TEST(test_archive_write_add_filter_by_name_xz) 192248590Smm{ 193248590Smm test_filter_by_name("xz", ARCHIVE_FILTER_XZ, cannot); 194248590Smm} 195