1/*-
2 * Copyright (c) 2003-2007 Tim Kientzle
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#include "test.h"
26
27static void
28test_open_filename_mbs(void)
29{
30	char buff[64];
31	struct archive_entry *ae;
32	struct archive *a;
33
34	/* Write an archive through this FILE *. */
35	assert((a = archive_write_new()) != NULL);
36	assertEqualIntA(a, ARCHIVE_OK, archive_write_set_format_ustar(a));
37	assertEqualIntA(a, ARCHIVE_OK, archive_write_add_filter_none(a));
38	assertEqualIntA(a, ARCHIVE_OK,
39	    archive_write_open_filename(a, "test.tar"));
40
41	/*
42	 * Write a file to it.
43	 */
44	assert((ae = archive_entry_new()) != NULL);
45	archive_entry_set_mtime(ae, 1, 0);
46	archive_entry_copy_pathname(ae, "file");
47	archive_entry_set_mode(ae, S_IFREG | 0755);
48	archive_entry_set_size(ae, 8);
49	assertEqualIntA(a, ARCHIVE_OK, archive_write_header(a, ae));
50	archive_entry_free(ae);
51	assertEqualIntA(a, 8, archive_write_data(a, "12345678", 9));
52
53	/*
54	 * Write a second file to it.
55	 */
56	assert((ae = archive_entry_new()) != NULL);
57	archive_entry_copy_pathname(ae, "file2");
58	archive_entry_set_mode(ae, S_IFREG | 0755);
59	archive_entry_set_size(ae, 819200);
60	assertEqualIntA(a, ARCHIVE_OK, archive_write_header(a, ae));
61	archive_entry_free(ae);
62
63	/* Close out the archive. */
64	assertEqualIntA(a, ARCHIVE_OK, archive_write_close(a));
65	assertEqualInt(ARCHIVE_OK, archive_write_free(a));
66
67	/*
68	 * Now, read the data back.
69	 */
70	assert((a = archive_read_new()) != NULL);
71	assertEqualIntA(a, ARCHIVE_OK, archive_read_support_format_all(a));
72	assertEqualIntA(a, ARCHIVE_OK, archive_read_support_filter_all(a));
73	assertEqualIntA(a, ARCHIVE_OK,
74	    archive_read_open_filename(a, "test.tar", 512));
75
76	assertEqualIntA(a, ARCHIVE_OK, archive_read_next_header(a, &ae));
77	assertEqualInt(1, archive_entry_mtime(ae));
78	assertEqualInt(0, archive_entry_mtime_nsec(ae));
79	assertEqualInt(0, archive_entry_atime(ae));
80	assertEqualInt(0, archive_entry_ctime(ae));
81	assertEqualString("file", archive_entry_pathname(ae));
82	assert((S_IFREG | 0755) == archive_entry_mode(ae));
83	assertEqualInt(8, archive_entry_size(ae));
84	assertEqualIntA(a, 8, archive_read_data(a, buff, 10));
85	assertEqualMem(buff, "12345678", 8);
86
87	assertEqualIntA(a, ARCHIVE_OK, archive_read_next_header(a, &ae));
88	assertEqualString("file2", archive_entry_pathname(ae));
89	assert((S_IFREG | 0755) == archive_entry_mode(ae));
90	assertEqualInt(819200, archive_entry_size(ae));
91	assertEqualIntA(a, ARCHIVE_OK, archive_read_data_skip(a));
92
93	/* Verify the end of the archive. */
94	assertEqualIntA(a, ARCHIVE_EOF, archive_read_next_header(a, &ae));
95	assertEqualIntA(a, ARCHIVE_OK, archive_read_close(a));
96	assertEqualInt(ARCHIVE_OK, archive_read_free(a));
97
98	/*
99	 * Verify some of the error handling.
100	 */
101	assert((a = archive_read_new()) != NULL);
102	assertEqualIntA(a, ARCHIVE_OK, archive_read_support_format_all(a));
103	assertEqualIntA(a, ARCHIVE_OK, archive_read_support_filter_all(a));
104	assertEqualIntA(a, ARCHIVE_FATAL,
105	    archive_read_open_filename(a, "nonexistent.tar", 512));
106	assertEqualIntA(a, ARCHIVE_OK, archive_read_close(a));
107	assertEqualInt(ARCHIVE_OK, archive_read_free(a));
108
109}
110
111static void
112test_open_filename_wcs(void)
113{
114	char buff[64];
115	struct archive_entry *ae;
116	struct archive *a;
117
118	/* Write an archive through this FILE *. */
119	assert((a = archive_write_new()) != NULL);
120	assertEqualIntA(a, ARCHIVE_OK, archive_write_set_format_ustar(a));
121	assertEqualIntA(a, ARCHIVE_OK, archive_write_add_filter_none(a));
122	assertEqualIntA(a, ARCHIVE_OK,
123	    archive_write_open_filename_w(a, L"test.tar"));
124
125	/*
126	 * Write a file to it.
127	 */
128	assert((ae = archive_entry_new()) != NULL);
129	archive_entry_set_mtime(ae, 1, 0);
130	archive_entry_copy_pathname_w(ae, L"file");
131	archive_entry_set_mode(ae, S_IFREG | 0755);
132	archive_entry_set_size(ae, 8);
133	assertEqualIntA(a, ARCHIVE_OK, archive_write_header(a, ae));
134	archive_entry_free(ae);
135	assertEqualIntA(a, 8, archive_write_data(a, "12345678", 9));
136
137	/*
138	 * Write a second file to it.
139	 */
140	assert((ae = archive_entry_new()) != NULL);
141	archive_entry_copy_pathname_w(ae, L"file2");
142	archive_entry_set_mode(ae, S_IFREG | 0755);
143	archive_entry_set_size(ae, 819200);
144	assertEqualIntA(a, ARCHIVE_OK, archive_write_header(a, ae));
145	archive_entry_free(ae);
146
147	/* Close out the archive. */
148	assertEqualIntA(a, ARCHIVE_OK, archive_write_close(a));
149	assertEqualInt(ARCHIVE_OK, archive_write_free(a));
150
151	/*
152	 * Now, read the data back.
153	 */
154	assert((a = archive_read_new()) != NULL);
155	assertEqualIntA(a, ARCHIVE_OK, archive_read_support_format_all(a));
156	assertEqualIntA(a, ARCHIVE_OK, archive_read_support_filter_all(a));
157	assertEqualIntA(a, ARCHIVE_OK,
158	    archive_read_open_filename_w(a, L"test.tar", 512));
159
160	assertEqualIntA(a, ARCHIVE_OK, archive_read_next_header(a, &ae));
161	assertEqualInt(1, archive_entry_mtime(ae));
162	assertEqualInt(0, archive_entry_mtime_nsec(ae));
163	assertEqualInt(0, archive_entry_atime(ae));
164	assertEqualInt(0, archive_entry_ctime(ae));
165	assertEqualWString(L"file", archive_entry_pathname_w(ae));
166	assert((S_IFREG | 0755) == archive_entry_mode(ae));
167	assertEqualInt(8, archive_entry_size(ae));
168	assertEqualIntA(a, 8, archive_read_data(a, buff, 10));
169	assertEqualMem(buff, "12345678", 8);
170
171	assertEqualIntA(a, ARCHIVE_OK, archive_read_next_header(a, &ae));
172	assertEqualWString(L"file2", archive_entry_pathname_w(ae));
173	assert((S_IFREG | 0755) == archive_entry_mode(ae));
174	assertEqualInt(819200, archive_entry_size(ae));
175	assertEqualIntA(a, ARCHIVE_OK, archive_read_data_skip(a));
176
177	/* Verify the end of the archive. */
178	assertEqualIntA(a, ARCHIVE_EOF, archive_read_next_header(a, &ae));
179	assertEqualIntA(a, ARCHIVE_OK, archive_read_close(a));
180	assertEqualInt(ARCHIVE_OK, archive_read_free(a));
181
182	/*
183	 * Verify some of the error handling.
184	 */
185	assert((a = archive_read_new()) != NULL);
186	assertEqualIntA(a, ARCHIVE_OK, archive_read_support_format_all(a));
187	assertEqualIntA(a, ARCHIVE_OK, archive_read_support_filter_all(a));
188	assertEqualIntA(a, ARCHIVE_FATAL,
189	    archive_read_open_filename_w(a, L"nonexistent.tar", 512));
190	assertEqualIntA(a, ARCHIVE_OK, archive_read_close(a));
191	assertEqualInt(ARCHIVE_OK, archive_read_free(a));
192
193}
194
195DEFINE_TEST(test_open_filename)
196{
197	test_open_filename_mbs();
198	test_open_filename_wcs();
199}
200