1 // Copyright 2017 The Chromium OS Authors. All rights reserved.
2 // Use of this source code is governed by a BSD-style license that can be
3 // found in the LICENSE file.
4
5 #include <string>
6 #include <vector>
7
8 #include "gtest/gtest.h"
9
10 #include "puffin/src/include/puffin/common.h"
11 #include "puffin/src/include/puffin/puffdiff.h"
12 #include "puffin/src/include/puffin/puffpatch.h"
13 #include "puffin/src/include/puffin/utils.h"
14 #include "puffin/src/logging.h"
15 #include "puffin/src/memory_stream.h"
16 #include "puffin/src/puffin_stream.h"
17 #include "puffin/src/unittest_common.h"
18
19 #define PRINT_SAMPLE 0 // Set to 1 if you want to print the generated samples.
20
21 using std::string;
22 using std::vector;
23
24 namespace puffin {
25
26 namespace {
27
28 #if PRINT_SAMPLE
29 // Print an array into hex-format to the output. This can be used to create
30 // static arrays for unit testing of the puffer/huffer.
PrintArray(const string & name,const Buffer & array)31 void PrintArray(const string& name, const Buffer& array) {
32 std::cout << "const Buffer " << name << " = {" << std::endl << " ";
33 for (size_t idx = 0; idx < array.size(); idx++) {
34 std::cout << " 0x" << std::hex << std::uppercase << std::setfill('0')
35 << std::setw(2) << uint(array[idx]);
36 if (idx == array.size() - 1) {
37 std::cout << std::dec << "};" << std::endl;
38 return;
39 }
40 std::cout << ",";
41 if ((idx + 1) % 12 == 0) {
42 std::cout << std::endl << " ";
43 }
44 }
45 }
46 #endif
47
48 const Buffer kPatch1To2 = {
49 0x50, 0x55, 0x46, 0x31, 0x00, 0x00, 0x00, 0x51, 0x08, 0x01, 0x12, 0x27,
50 0x0A, 0x04, 0x08, 0x10, 0x10, 0x32, 0x0A, 0x04, 0x08, 0x50, 0x10, 0x0A,
51 0x0A, 0x04, 0x08, 0x60, 0x10, 0x12, 0x12, 0x04, 0x08, 0x10, 0x10, 0x58,
52 0x12, 0x04, 0x08, 0x78, 0x10, 0x28, 0x12, 0x05, 0x08, 0xA8, 0x01, 0x10,
53 0x38, 0x18, 0x1F, 0x1A, 0x24, 0x0A, 0x02, 0x10, 0x32, 0x0A, 0x04, 0x08,
54 0x48, 0x10, 0x50, 0x0A, 0x05, 0x08, 0x98, 0x01, 0x10, 0x12, 0x12, 0x02,
55 0x10, 0x58, 0x12, 0x04, 0x08, 0x70, 0x10, 0x58, 0x12, 0x05, 0x08, 0xC8,
56 0x01, 0x10, 0x38, 0x18, 0x21, 0x42, 0x53, 0x44, 0x46, 0x32, 0x01, 0x01,
57 0x01, 0x38, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x34, 0x00, 0x00,
58 0x00, 0x00, 0x00, 0x00, 0x00, 0x21, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
59 0x00, 0x42, 0x5A, 0x68, 0x39, 0x31, 0x41, 0x59, 0x26, 0x53, 0x59, 0xD1,
60 0x20, 0xBB, 0x7E, 0x00, 0x00, 0x03, 0x60, 0x40, 0x78, 0x0E, 0x08, 0x00,
61 0x40, 0x00, 0x20, 0x00, 0x31, 0x06, 0x4C, 0x40, 0x92, 0x8F, 0x46, 0xA7,
62 0xA8, 0xE0, 0xF3, 0xD6, 0x21, 0x12, 0xF4, 0xBC, 0x43, 0x32, 0x1F, 0x17,
63 0x72, 0x45, 0x38, 0x50, 0x90, 0xD1, 0x20, 0xBB, 0x7E, 0x42, 0x5A, 0x68,
64 0x39, 0x31, 0x41, 0x59, 0x26, 0x53, 0x59, 0xF1, 0x20, 0x5F, 0x0D, 0x00,
65 0x00, 0x02, 0x41, 0x15, 0x42, 0x08, 0x20, 0x00, 0x40, 0x00, 0x00, 0x02,
66 0x40, 0x00, 0x20, 0x00, 0x22, 0x3D, 0x23, 0x10, 0x86, 0x03, 0x96, 0x54,
67 0x11, 0x16, 0x5F, 0x17, 0x72, 0x45, 0x38, 0x50, 0x90, 0xF1, 0x20, 0x5F,
68 0x0D, 0x42, 0x5A, 0x68, 0x39, 0x31, 0x41, 0x59, 0x26, 0x53, 0x59, 0x07,
69 0xD4, 0xCB, 0x6E, 0x00, 0x00, 0x00, 0x01, 0x00, 0x01, 0x00, 0x20, 0x00,
70 0x21, 0x18, 0x46, 0x82, 0xEE, 0x48, 0xA7, 0x0A, 0x12, 0x00, 0xFA, 0x99,
71 0x6D, 0xC0};
72
73 const Buffer kPatch2To1 = {
74 0x50, 0x55, 0x46, 0x31, 0x00, 0x00, 0x00, 0x51, 0x08, 0x01, 0x12, 0x24,
75 0x0A, 0x02, 0x10, 0x32, 0x0A, 0x04, 0x08, 0x48, 0x10, 0x50, 0x0A, 0x05,
76 0x08, 0x98, 0x01, 0x10, 0x12, 0x12, 0x02, 0x10, 0x58, 0x12, 0x04, 0x08,
77 0x70, 0x10, 0x58, 0x12, 0x05, 0x08, 0xC8, 0x01, 0x10, 0x38, 0x18, 0x21,
78 0x1A, 0x27, 0x0A, 0x04, 0x08, 0x10, 0x10, 0x32, 0x0A, 0x04, 0x08, 0x50,
79 0x10, 0x0A, 0x0A, 0x04, 0x08, 0x60, 0x10, 0x12, 0x12, 0x04, 0x08, 0x10,
80 0x10, 0x58, 0x12, 0x04, 0x08, 0x78, 0x10, 0x28, 0x12, 0x05, 0x08, 0xA8,
81 0x01, 0x10, 0x38, 0x18, 0x1F, 0x42, 0x53, 0x44, 0x46, 0x32, 0x01, 0x01,
82 0x01, 0x33, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x25, 0x00, 0x00,
83 0x00, 0x00, 0x00, 0x00, 0x00, 0x1F, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
84 0x00, 0x42, 0x5A, 0x68, 0x39, 0x31, 0x41, 0x59, 0x26, 0x53, 0x59, 0x3D,
85 0xBD, 0x08, 0x91, 0x00, 0x00, 0x01, 0xE0, 0x40, 0x5C, 0x0A, 0x40, 0x00,
86 0x40, 0x00, 0x20, 0x00, 0x31, 0x0C, 0x08, 0x23, 0xD2, 0x34, 0xD1, 0xB1,
87 0x73, 0x60, 0x44, 0x54, 0xE4, 0xFC, 0x5D, 0xC9, 0x14, 0xE1, 0x42, 0x40,
88 0xF6, 0xF4, 0x22, 0x44, 0x42, 0x5A, 0x68, 0x39, 0x31, 0x41, 0x59, 0x26,
89 0x53, 0x59, 0x41, 0x62, 0x2E, 0xF0, 0x00, 0x00, 0x00, 0x40, 0x00, 0x40,
90 0x20, 0x20, 0x00, 0x21, 0x00, 0x82, 0x83, 0x17, 0x72, 0x45, 0x38, 0x50,
91 0x90, 0x41, 0x62, 0x2E, 0xF0, 0x42, 0x5A, 0x68, 0x39, 0x31, 0x41, 0x59,
92 0x26, 0x53, 0x59, 0xE0, 0x20, 0x04, 0x57, 0x00, 0x00, 0x04, 0x76, 0x50,
93 0xE0, 0x00, 0x20, 0x00, 0x10, 0x00, 0x04, 0x00, 0x02, 0x00, 0x20, 0x00,
94 0x40, 0x00, 0x00, 0x00, 0xA0, 0x00, 0x21, 0xA1, 0xA3, 0x10, 0x83, 0x26,
95 0x21, 0x5E, 0xB2, 0x69, 0xAC, 0x70, 0x60, 0x53, 0xC5, 0xDC, 0x91, 0x4E,
96 0x14, 0x24, 0x38, 0x08, 0x01, 0x15, 0xC0};
97
98 const Buffer kPatch1ToEmpty = {
99 0x50, 0x55, 0x46, 0x31, 0x00, 0x00, 0x00, 0x2D, 0x08, 0x01, 0x12, 0x27,
100 0x0A, 0x04, 0x08, 0x10, 0x10, 0x32, 0x0A, 0x04, 0x08, 0x50, 0x10, 0x0A,
101 0x0A, 0x04, 0x08, 0x60, 0x10, 0x12, 0x12, 0x04, 0x08, 0x10, 0x10, 0x58,
102 0x12, 0x04, 0x08, 0x78, 0x10, 0x28, 0x12, 0x05, 0x08, 0xA8, 0x01, 0x10,
103 0x38, 0x18, 0x1F, 0x1A, 0x00, 0x42, 0x53, 0x44, 0x46, 0x32, 0x01, 0x01,
104 0x01, 0x0E, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x0E, 0x00, 0x00,
105 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00,
106 0x00, 0x42, 0x5A, 0x68, 0x39, 0x17, 0x72, 0x45, 0x38, 0x50, 0x90, 0x00,
107 0x00, 0x00, 0x00, 0x42, 0x5A, 0x68, 0x39, 0x17, 0x72, 0x45, 0x38, 0x50,
108 0x90, 0x00, 0x00, 0x00, 0x00, 0x42, 0x5A, 0x68, 0x39, 0x17, 0x72, 0x45,
109 0x38, 0x50, 0x90, 0x00, 0x00, 0x00, 0x00};
110
111 const Buffer kPatch1ToNoDeflate = {
112 0x50, 0x55, 0x46, 0x31, 0x00, 0x00, 0x00, 0x2F, 0x08, 0x01, 0x12, 0x27,
113 0x0A, 0x04, 0x08, 0x10, 0x10, 0x32, 0x0A, 0x04, 0x08, 0x50, 0x10, 0x0A,
114 0x0A, 0x04, 0x08, 0x60, 0x10, 0x12, 0x12, 0x04, 0x08, 0x10, 0x10, 0x58,
115 0x12, 0x04, 0x08, 0x78, 0x10, 0x28, 0x12, 0x05, 0x08, 0xA8, 0x01, 0x10,
116 0x38, 0x18, 0x1F, 0x1A, 0x02, 0x18, 0x04, 0x42, 0x53, 0x44, 0x46, 0x32,
117 0x01, 0x01, 0x01, 0x28, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x0E,
118 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x00, 0x04, 0x00, 0x00, 0x00, 0x00,
119 0x00, 0x00, 0x00, 0x42, 0x5A, 0x68, 0x39, 0x31, 0x41, 0x59, 0x26, 0x53,
120 0x59, 0xBA, 0x8D, 0x7F, 0x2D, 0x00, 0x00, 0x00, 0x40, 0x00, 0x44, 0x08,
121 0x20, 0x00, 0x30, 0xCC, 0x09, 0x32, 0x54, 0x65, 0x38, 0xBB, 0x92, 0x29,
122 0xC2, 0x84, 0x85, 0xD4, 0x6B, 0xF9, 0x68, 0x42, 0x5A, 0x68, 0x39, 0x17,
123 0x72, 0x45, 0x38, 0x50, 0x90, 0x00, 0x00, 0x00, 0x00, 0x42, 0x5A, 0x68,
124 0x39, 0x31, 0x41, 0x59, 0x26, 0x53, 0x59, 0xE7, 0xAA, 0xF1, 0xFC, 0x00,
125 0x00, 0x00, 0x70, 0x00, 0x00, 0x08, 0x01, 0x00, 0x20, 0x04, 0x20, 0x00,
126 0x21, 0x9A, 0x68, 0x33, 0x4D, 0x13, 0x3C, 0x5D, 0xC9, 0x14, 0xE1, 0x42,
127 0x43, 0x9E, 0xAB, 0xC7, 0xF0};
128
129 } // namespace
130
TestPatching(const Buffer & src_buf,const Buffer & dst_buf,const vector<BitExtent> & src_deflates,const vector<BitExtent> & dst_deflates,const Buffer patch)131 void TestPatching(const Buffer& src_buf,
132 const Buffer& dst_buf,
133 const vector<BitExtent>& src_deflates,
134 const vector<BitExtent>& dst_deflates,
135 const Buffer patch) {
136 Buffer patch_out;
137 string patch_path;
138 ASSERT_TRUE(MakeTempFile(&patch_path, nullptr));
139 ScopedPathUnlinker scoped_unlinker(patch_path);
140 ASSERT_TRUE(PuffDiff(src_buf, dst_buf, src_deflates, dst_deflates,
141 {bsdiff::CompressorType::kBZ2}, patch_path, &patch_out));
142
143 #if PRINT_SAMPLE
144 PrintArray("kPatchXXXXX", patch_out);
145 #endif
146
147 EXPECT_EQ(patch_out, patch);
148
149 auto src_stream = MemoryStream::CreateForRead(src_buf);
150 Buffer dst_buf_out(dst_buf.size());
151 auto dst_stream = MemoryStream::CreateForWrite(&dst_buf_out);
152 ASSERT_TRUE(PuffPatch(std::move(src_stream), std::move(dst_stream),
153 patch.data(), patch.size()));
154 EXPECT_EQ(dst_buf_out, dst_buf);
155 }
156
TEST(PatchingTest,Patching1To2Test)157 TEST(PatchingTest, Patching1To2Test) {
158 TestPatching(kDeflatesSample1, kDeflatesSample2,
159 kSubblockDeflateExtentsSample1, kSubblockDeflateExtentsSample2,
160 kPatch1To2);
161 }
162
TEST(PatchingTest,Patching2To1Test)163 TEST(PatchingTest, Patching2To1Test) {
164 TestPatching(kDeflatesSample2, kDeflatesSample1,
165 kSubblockDeflateExtentsSample2, kSubblockDeflateExtentsSample1,
166 kPatch2To1);
167 }
168
TEST(PatchingTest,Patching1ToEmptyTest)169 TEST(PatchingTest, Patching1ToEmptyTest) {
170 TestPatching(kDeflatesSample1, {}, kSubblockDeflateExtentsSample1, {},
171 kPatch1ToEmpty);
172 }
173
TEST(PatchingTest,Patching1ToNoDeflateTest)174 TEST(PatchingTest, Patching1ToNoDeflateTest) {
175 TestPatching(kDeflatesSample1, {11, 22, 33, 44},
176 kSubblockDeflateExtentsSample1, {}, kPatch1ToNoDeflate);
177 }
178
179 // TODO(ahassani): add tests for:
180 // TestPatchingEmptyTo2
181 // TestPatchingNoDeflateTo2
182
183 // TODO(ahassani): Change tests data if you decided to compress the header of
184 // the patch.
185
186 } // namespace puffin
187