1248590Smm/*-
2248590Smm * Copyright (c) 2007 Tim Kientzle
3248590Smm * Copyright (c) 2012 Michihiro NAKAJIMA
4248590Smm * All rights reserved.
5248590Smm *
6248590Smm * Redistribution and use in source and binary forms, with or without
7248590Smm * modification, are permitted provided that the following conditions
8248590Smm * are met:
9248590Smm * 1. Redistributions of source code must retain the above copyright
10248590Smm *    notice, this list of conditions and the following disclaimer
11248590Smm *    in this position and unchanged.
12248590Smm * 2. Redistributions in binary form must reproduce the above copyright
13248590Smm *    notice, this list of conditions and the following disclaimer in the
14248590Smm *    documentation and/or other materials provided with the distribution.
15248590Smm *
16248590Smm * THIS SOFTWARE IS PROVIDED BY THE AUTHOR(S) ``AS IS'' AND ANY EXPRESS OR
17248590Smm * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
18248590Smm * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
19248590Smm * IN NO EVENT SHALL THE AUTHOR(S) BE LIABLE FOR ANY DIRECT, INDIRECT,
20248590Smm * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
21248590Smm * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
22248590Smm * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
23248590Smm * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
24248590Smm * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
25248590Smm * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
26248590Smm */
27248590Smm
28248590Smm#include "test.h"
29248590Smm__FBSDID("$FreeBSD$");
30248590Smm
31248590SmmDEFINE_TEST(test_write_filter_gzip_timestamp)
32248590Smm{
33248590Smm	struct archive_entry *ae;
34248590Smm	struct archive* a;
35248590Smm	char *buff, *data;
36248590Smm	size_t buffsize, datasize;
37248590Smm	size_t used1;
38248590Smm	int r, use_prog = 0;
39248590Smm
40248590Smm	buffsize = 10000;
41248590Smm	assert(NULL != (buff = (char *)malloc(buffsize)));
42248590Smm	if (buff == NULL)
43248590Smm		return;
44248590Smm
45248590Smm	datasize = 10000;
46248590Smm	assert(NULL != (data = (char *)malloc(datasize)));
47248590Smm	if (data == NULL) {
48248590Smm		free(buff);
49248590Smm		return;
50248590Smm	}
51248590Smm	memset(data, 0, datasize);
52248590Smm
53248590Smm	/* Test1: set "gzip:timestamp" option. */
54248590Smm	assert((a = archive_write_new()) != NULL);
55248590Smm	assertEqualIntA(a, ARCHIVE_OK, archive_write_set_format_ustar(a));
56248590Smm	r = archive_write_add_filter_gzip(a);
57248590Smm	if (r != ARCHIVE_OK) {
58248590Smm		if (canGzip() && r == ARCHIVE_WARN)
59248590Smm			use_prog = 1;
60248590Smm		else {
61248590Smm			skipping("gzip writing not supported on this platform");
62248590Smm			assertEqualInt(ARCHIVE_OK, archive_write_free(a));
63248590Smm			free(buff);
64248590Smm			free(data);
65248590Smm			return;
66248590Smm		}
67248590Smm	}
68248590Smm	assertEqualIntA(a, ARCHIVE_OK,
69248590Smm	    archive_write_set_options(a, "gzip:timestamp"));
70248590Smm	assertEqualIntA(a, ARCHIVE_OK,
71248590Smm	    archive_write_set_bytes_per_block(a, 10));
72248590Smm	assertEqualInt(ARCHIVE_FILTER_GZIP, archive_filter_code(a, 0));
73248590Smm	assertEqualString("gzip", archive_filter_name(a, 0));
74248590Smm	assertEqualIntA(a, ARCHIVE_OK,
75248590Smm	    archive_write_open_memory(a, buff, buffsize, &used1));
76248590Smm	assert((ae = archive_entry_new()) != NULL);
77248590Smm	archive_entry_set_filetype(ae, AE_IFREG);
78248590Smm	archive_entry_set_size(ae, datasize);
79248590Smm	archive_entry_copy_pathname(ae, "file");
80248590Smm	assertEqualIntA(a, ARCHIVE_OK, archive_write_header(a, ae));
81248590Smm	assertEqualIntA(a, datasize, archive_write_data(a, data, datasize));
82248590Smm	archive_entry_free(ae);
83248590Smm	assertEqualIntA(a, ARCHIVE_OK, archive_write_close(a));
84248590Smm	assertEqualInt(ARCHIVE_OK, archive_write_free(a));
85248590Smm	failure("Timestamp should be recorded");
86248590Smm	assert(memcmp(buff + 4, "\x00\x00\x00\x00", 4) != 0);
87248590Smm
88248590Smm	/* Test2: set "gzip:!timestamp" option. */
89248590Smm	assert((a = archive_write_new()) != NULL);
90248590Smm	assertEqualIntA(a, ARCHIVE_OK, archive_write_set_format_ustar(a));
91248590Smm	assertEqualIntA(a, (use_prog)?ARCHIVE_WARN:ARCHIVE_OK,
92248590Smm	    archive_write_add_filter_gzip(a));
93248590Smm	assertEqualIntA(a, ARCHIVE_OK,
94248590Smm	    archive_write_set_options(a, "gzip:!timestamp"));
95248590Smm	assertEqualIntA(a, ARCHIVE_OK,
96248590Smm	    archive_write_set_bytes_per_block(a, 10));
97248590Smm	assertEqualInt(ARCHIVE_FILTER_GZIP, archive_filter_code(a, 0));
98248590Smm	assertEqualString("gzip", archive_filter_name(a, 0));
99248590Smm	assertEqualIntA(a, ARCHIVE_OK,
100248590Smm	    archive_write_open_memory(a, buff, buffsize, &used1));
101248590Smm	assert((ae = archive_entry_new()) != NULL);
102248590Smm	archive_entry_set_filetype(ae, AE_IFREG);
103248590Smm	archive_entry_set_size(ae, datasize);
104248590Smm	archive_entry_copy_pathname(ae, "file");
105248590Smm	assertEqualIntA(a, ARCHIVE_OK, archive_write_header(a, ae));
106248590Smm	assertEqualIntA(a, datasize, archive_write_data(a, data, datasize));
107248590Smm	archive_entry_free(ae);
108248590Smm	assertEqualIntA(a, ARCHIVE_OK, archive_write_close(a));
109248590Smm	assertEqualInt(ARCHIVE_OK, archive_write_free(a));
110248590Smm	failure("Timestamp should not be recorded");
111248590Smm	assertEqualMem(buff + 4, "\x00\x00\x00\x00", 4);
112248590Smm
113248590Smm	/*
114248590Smm	 * Clean up.
115248590Smm	 */
116248590Smm	free(data);
117248590Smm	free(buff);
118248590Smm}
119