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 14static 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 23static 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 31static 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 41static 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 71static 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 85static 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 123static 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 160DEF_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 171DEF_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 182DEF_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 193DEF_TEST(Writer32_misc, reporter) { 194 test_reserve(reporter); 195 test_string_null(reporter); 196 test_rewind(reporter); 197} 198 199DEF_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