1 /*
2 * Copyright 2011 Google Inc.
3 *
4 * Use of this source code is governed by a BSD-style license that can be
5 * found in the LICENSE file.
6 */
7
8 #include "include/utils/SkRandom.h"
9 #include "src/core/SkAutoMalloc.h"
10 #include "src/core/SkReadBuffer.h"
11 #include "src/core/SkWriter32.h"
12 #include "tests/Test.h"
13
check_contents(skiatest::Reporter * reporter,const SkWriter32 & writer,const void * expected,size_t size)14 static void check_contents(skiatest::Reporter* reporter, const SkWriter32& writer,
15 const void* expected, size_t size) {
16 SkAutoSMalloc<256> storage(size);
17 REPORTER_ASSERT(reporter, writer.bytesWritten() == size);
18 writer.flatten(storage.get());
19 REPORTER_ASSERT(reporter, !memcmp(storage.get(), expected, size));
20 }
21
22
test_reserve(skiatest::Reporter * reporter)23 static void test_reserve(skiatest::Reporter* reporter) {
24 // There used to be a bug where we'd assert your first reservation had to
25 // fit in external storage if you used it. This would crash in debug mode.
26 uint8_t storage[4];
27 SkWriter32 writer(storage, sizeof(storage));
28 writer.reserve(40);
29 }
30
test_string_null(skiatest::Reporter * reporter)31 static void test_string_null(skiatest::Reporter* reporter) {
32 uint8_t storage[8];
33 SkWriter32 writer(storage, sizeof(storage));
34
35 // Can we write nullptr?
36 writer.writeString(nullptr);
37 const int32_t expected[] = { 0x0, 0x0 };
38 check_contents(reporter, writer, expected, sizeof(expected));
39 }
40
test_rewind(skiatest::Reporter * reporter)41 static void test_rewind(skiatest::Reporter* reporter) {
42 SkSWriter32<32> swriter;
43 int32_t array[3] = { 1, 2, 4 };
44
45 REPORTER_ASSERT(reporter, 0 == swriter.bytesWritten());
46 for (size_t i = 0; i < SK_ARRAY_COUNT(array); ++i) {
47 swriter.writeInt(array[i]);
48 }
49 check_contents(reporter, swriter, array, sizeof(array));
50
51 swriter.rewindToOffset(2*sizeof(int32_t));
52 REPORTER_ASSERT(reporter, sizeof(array) - 4 == swriter.bytesWritten());
53 swriter.writeInt(3);
54 REPORTER_ASSERT(reporter, sizeof(array) == swriter.bytesWritten());
55 array[2] = 3;
56 check_contents(reporter, swriter, array, sizeof(array));
57
58 // test rewinding past allocated chunks. This used to crash because we
59 // didn't truncate our link-list after freeing trailing blocks
60 SkWriter32 writer;
61 for (int i = 0; i < 100; ++i) {
62 writer.writeInt(i);
63 }
64 REPORTER_ASSERT(reporter, 100*4 == writer.bytesWritten());
65 for (int j = 100*4; j >= 0; j -= 16) {
66 writer.rewindToOffset(j);
67 }
68 REPORTER_ASSERT(reporter, writer.bytesWritten() < 16);
69 }
70
test1(skiatest::Reporter * reporter,SkWriter32 * writer)71 static void test1(skiatest::Reporter* reporter, SkWriter32* writer) {
72 const uint32_t data[] = { 0, 1, 2, 3, 4, 5, 6, 7, 8, 9 };
73 for (size_t i = 0; i < SK_ARRAY_COUNT(data); ++i) {
74 REPORTER_ASSERT(reporter, i*4 == writer->bytesWritten());
75 writer->write32(data[i]);
76 REPORTER_ASSERT(reporter, data[i] == writer->readTAt<uint32_t>(i * 4));
77 }
78
79 char buffer[sizeof(data)];
80 REPORTER_ASSERT(reporter, sizeof(buffer) == writer->bytesWritten());
81 writer->flatten(buffer);
82 REPORTER_ASSERT(reporter, !memcmp(data, buffer, sizeof(buffer)));
83 }
84
testWritePad(skiatest::Reporter * reporter,SkWriter32 * writer)85 static void testWritePad(skiatest::Reporter* reporter, SkWriter32* writer) {
86 // Create some random data to write.
87 const size_t dataSize = 10;
88
89 SkAutoTMalloc<uint32_t> originalData(dataSize);
90 {
91 SkRandom rand(0);
92 for (size_t i = 0; i < dataSize; i++) {
93 originalData[(int) i] = rand.nextU();
94 }
95
96 // Write the random data to the writer at different lengths for
97 // different alignments.
98 for (size_t len = 0; len < dataSize; len++) {
99 writer->writePad(originalData.get(), len);
100 }
101 }
102
103 size_t totalBytes = writer->bytesWritten();
104
105 SkAutoMalloc readStorage(totalBytes);
106 writer->flatten(readStorage.get());
107
108 SkReadBuffer reader(readStorage.get(), totalBytes);
109
110 for (size_t len = 0; len < dataSize; len++) {
111 const char* readPtr = static_cast<const char*>(reader.skip(len));
112 // Ensure that the data read is the same as what was written.
113 REPORTER_ASSERT(reporter, memcmp(readPtr, originalData.get(), len) == 0);
114 // Ensure that the rest is padded with zeroes.
115 const char* stop = readPtr + SkAlign4(len);
116 readPtr += len;
117 while (readPtr < stop) {
118 REPORTER_ASSERT(reporter, *readPtr++ == 0);
119 }
120 }
121 }
122
testOverwriteT(skiatest::Reporter * reporter,SkWriter32 * writer)123 static void testOverwriteT(skiatest::Reporter* reporter, SkWriter32* writer) {
124 const size_t padding = 64;
125
126 const uint32_t uint1 = 0x12345678;
127 const uint32_t uint2 = 0x98765432;
128 const SkScalar scalar1 = 1234.5678f;
129 const SkScalar scalar2 = 9876.5432f;
130 const SkRect rect1 = SkRect::MakeXYWH(1, 2, 3, 4);
131 const SkRect rect2 = SkRect::MakeXYWH(5, 6, 7, 8);
132
133 for (size_t i = 0; i < (padding / 4); ++i) {
134 writer->write32(0);
135 }
136
137 writer->write32(uint1);
138 writer->writeRect(rect1);
139 writer->writeScalar(scalar1);
140
141 for (size_t i = 0; i < (padding / 4); ++i) {
142 writer->write32(0);
143 }
144
145 REPORTER_ASSERT(reporter, writer->readTAt<uint32_t>(padding) == uint1);
146 REPORTER_ASSERT(reporter, writer->readTAt<SkRect>(padding + sizeof(uint32_t)) == rect1);
147 REPORTER_ASSERT(reporter, writer->readTAt<SkScalar>(
148 padding + sizeof(uint32_t) + sizeof(SkRect)) == scalar1);
149
150 writer->overwriteTAt(padding, uint2);
151 writer->overwriteTAt(padding + sizeof(uint32_t), rect2);
152 writer->overwriteTAt(padding + sizeof(uint32_t) + sizeof(SkRect), scalar2);
153
154 REPORTER_ASSERT(reporter, writer->readTAt<uint32_t>(padding) == uint2);
155 REPORTER_ASSERT(reporter, writer->readTAt<SkRect>(padding + sizeof(uint32_t)) == rect2);
156 REPORTER_ASSERT(reporter, writer->readTAt<SkScalar>(
157 padding + sizeof(uint32_t) + sizeof(SkRect)) == scalar2);
158 }
159
DEF_TEST(Writer32_dynamic,reporter)160 DEF_TEST(Writer32_dynamic, reporter) {
161 SkWriter32 writer;
162 test1(reporter, &writer);
163
164 writer.reset();
165 testWritePad(reporter, &writer);
166
167 writer.reset();
168 testOverwriteT(reporter, &writer);
169 }
170
DEF_TEST(Writer32_small,reporter)171 DEF_TEST(Writer32_small, reporter) {
172 SkSWriter32<8 * sizeof(intptr_t)> writer;
173 test1(reporter, &writer);
174
175 writer.reset(); // should just rewind our storage
176 testWritePad(reporter, &writer);
177
178 writer.reset();
179 testOverwriteT(reporter, &writer);
180 }
181
DEF_TEST(Writer32_large,reporter)182 DEF_TEST(Writer32_large, reporter) {
183 SkSWriter32<1024 * sizeof(intptr_t)> writer;
184 test1(reporter, &writer);
185
186 writer.reset(); // should just rewind our storage
187 testWritePad(reporter, &writer);
188
189 writer.reset();
190 testOverwriteT(reporter, &writer);
191 }
192
DEF_TEST(Writer32_misc,reporter)193 DEF_TEST(Writer32_misc, reporter) {
194 test_reserve(reporter);
195 test_string_null(reporter);
196 test_rewind(reporter);
197 }
198
DEF_TEST(Writer32_data,reporter)199 DEF_TEST(Writer32_data, reporter) {
200 const char* str = "0123456789";
201 sk_sp<SkData> data0(SkData::MakeWithCString(str));
202 sk_sp<SkData> data1(SkData::MakeEmpty());
203
204 const size_t sizes[] = {
205 SkWriter32::WriteDataSize(nullptr),
206 SkWriter32::WriteDataSize(data0.get()),
207 SkWriter32::WriteDataSize(data1.get()),
208 };
209
210 SkSWriter32<1000> writer;
211 size_t sizeWritten = 0;
212
213 writer.writeData(nullptr);
214 sizeWritten += sizes[0];
215 REPORTER_ASSERT(reporter, sizeWritten == writer.bytesWritten());
216
217 writer.writeData(data0.get());
218 sizeWritten += sizes[1];
219 REPORTER_ASSERT(reporter, sizeWritten == writer.bytesWritten());
220
221 writer.writeData(data1.get());
222 sizeWritten += sizes[2];
223 REPORTER_ASSERT(reporter, sizeWritten == writer.bytesWritten());
224
225 auto result(writer.snapshotAsData());
226
227 SkReadBuffer reader(result->data(), result->size());
228 auto d0(reader.readByteArrayAsData()),
229 d1(reader.readByteArrayAsData()),
230 d2(reader.readByteArrayAsData());
231
232 REPORTER_ASSERT(reporter, 0 == d0->size());
233 REPORTER_ASSERT(reporter, strlen(str)+1 == d1->size());
234 REPORTER_ASSERT(reporter, !memcmp(str, d1->data(), strlen(str)+1));
235 REPORTER_ASSERT(reporter, 0 == d2->size());
236
237 REPORTER_ASSERT(reporter, reader.offset() == sizeWritten);
238 REPORTER_ASSERT(reporter, reader.eof());
239 }
240