2 * Copyright (c) 2003-2007 Tim Kientzle
5 * Redistribution and use in source and binary forms, with or without
6 * modification, are permitted provided that the following conditions
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.
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.
26 __FBSDID("$FreeBSD$");
29 * Also see test_option_q for additional validation of -r support.
31 DEFINE_TEST(test_option_r)
35 size_t buff_size = 35000;
36 size_t s, buff_size_rounded;
43 /* Create an archive with one file. */
44 assertMakeFile("f1", 0644, "abc");
45 r = systemf("%s cf archive.tar --format=ustar f1 >step1.out 2>step1.err", testprog);
46 failure("Error invoking %s cf archive.tar f1", testprog);
48 assertEmptyFile("step1.out");
49 assertEmptyFile("step1.err");
51 /* Do some basic validation of the constructed archive. */
52 p0 = slurpfile(&s, "archive.tar");
53 if (!assert(p0 != NULL))
55 if (!assert(s >= 2048))
57 assertEqualMem(p0 + 0, "f1", 3);
58 assertEqualMem(p0 + 512, "abc", 3);
59 assertEqualMem(p0 + 1024, "\0\0\0\0\0\0\0\0", 8);
60 assertEqualMem(p0 + 1536, "\0\0\0\0\0\0\0\0", 8);
62 /* Edit that file with a lot more data and update the archive with a new copy. */
63 buff = malloc(buff_size);
68 for (i = 0; i < (int)buff_size; ++i)
69 buff[i] = "abcdefghijklmnopqrstuvwxyz"[rand() % 26];
70 buff[buff_size - 1] = '\0';
71 assertMakeFile("f1", 0644, buff);
72 r = systemf("%s rf archive.tar --format=ustar f1 >step2.out 2>step2.err", testprog);
73 failure("Error invoking %s rf archive.tar f1", testprog);
75 assertEmptyFile("step2.out");
76 assertEmptyFile("step2.err");
78 /* The constructed archive should just have the new entry appended. */
79 p1 = slurpfile(&s, "archive.tar");
80 if (!assert(p1 != NULL))
82 buff_size_rounded = ((buff_size + 511) / 512) * 512;
83 assert(s >= 2560 + buff_size_rounded);
84 /* Verify first entry is unchanged. */
85 assertEqualMem(p0, p1, 1024);
86 /* Verify that second entry is correct. */
87 assertEqualMem(p1 + 1024, "f1", 3);
88 assertEqualMem(p1 + 1536, buff, buff_size);
89 /* Verify end-of-archive marker. */
90 assertEqualMem(p1 + 1536 + buff_size_rounded, "\0\0\0\0\0\0\0\0", 8);
91 assertEqualMem(p1 + 2048 + buff_size_rounded, "\0\0\0\0\0\0\0\0", 8);
96 /* Update the archive by adding a different file. */
97 assertMakeFile("f2", 0644, "f2");
98 r = systemf("%s rf archive.tar --format=ustar f2 >step3.out 2>step3.err", testprog);
99 failure("Error invoking %s rf archive.tar f2", testprog);
100 assertEqualInt(r, 0);
101 assertEmptyFile("step3.out");
102 assertEmptyFile("step3.err");
104 /* Validate the constructed archive. */
105 p1 = slurpfile(&s, "archive.tar");
106 if (!assert(p1 != NULL))
108 assert(s >= 3584 + buff_size_rounded);
109 /* Verify first two entries are unchanged. */
110 assertEqualMem(p0, p1, 1536 + buff_size_rounded);
111 /* Verify that new entry is correct. */
112 assertEqualMem(p1 + 1536 + buff_size_rounded, "f2", 3);
113 assertEqualMem(p1 + 2048 + buff_size_rounded, "f2", 3);
114 /* Verify end-of-archive marker. */
115 assertEqualMem(p1 + 2560 + buff_size_rounded, "\0\0\0\0\0\0\0\0", 8);
116 assertEqualMem(p1 + 3072 + buff_size_rounded, "\0\0\0\0\0\0\0\0", 8);
119 /* Unpack everything */
120 assertMakeDir("extract", 0775);
121 assertChdir("extract");
122 r = systemf("%s xf ../archive.tar >extract.out 2>extract.err", testprog);
123 failure("Error invoking %s xf archive.tar", testprog);
124 assertEqualInt(r, 0);
125 assertEmptyFile("extract.out");
126 assertEmptyFile("extract.err");
128 /* Verify that the second copy of f1 overwrote the first. */
129 assertFileContents(buff, (int)strlen(buff), "f1");