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