1 #include "Test.h"
2 #include "SkRandom.h"
3 #include "SkStream.h"
4
5 #define MAX_SIZE (256 * 1024)
6
random_fill(SkRandom & rand,void * buffer,size_t size)7 static void random_fill(SkRandom& rand, void* buffer, size_t size) {
8 char* p = (char*)buffer;
9 char* stop = p + size;
10 while (p < stop) {
11 *p++ = (char)(rand.nextU() >> 8);
12 }
13 }
14
test_buffer(skiatest::Reporter * reporter)15 static void test_buffer(skiatest::Reporter* reporter) {
16 SkRandom rand;
17 SkAutoMalloc am(MAX_SIZE * 2);
18 char* storage = (char*)am.get();
19 char* storage2 = storage + MAX_SIZE;
20
21 random_fill(rand, storage, MAX_SIZE);
22
23 for (int sizeTimes = 0; sizeTimes < 100; sizeTimes++) {
24 int size = rand.nextU() % MAX_SIZE;
25 if (size == 0) {
26 size = MAX_SIZE;
27 }
28 for (int times = 0; times < 100; times++) {
29 int bufferSize = 1 + (rand.nextU() & 0xFFFF);
30 SkMemoryStream mstream(storage, size);
31 SkBufferStream bstream(&mstream, bufferSize);
32
33 int bytesRead = 0;
34 while (bytesRead < size) {
35 int s = 17 + (rand.nextU() & 0xFFFF);
36 int ss = bstream.read(storage2, s);
37 REPORTER_ASSERT(reporter, ss > 0 && ss <= s);
38 REPORTER_ASSERT(reporter, bytesRead + ss <= size);
39 REPORTER_ASSERT(reporter,
40 memcmp(storage + bytesRead, storage2, ss) == 0);
41 bytesRead += ss;
42 }
43 REPORTER_ASSERT(reporter, bytesRead == size);
44 }
45 }
46 }
47
TestRStream(skiatest::Reporter * reporter)48 static void TestRStream(skiatest::Reporter* reporter) {
49 static const char s[] =
50 "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz";
51 char copy[sizeof(s)];
52 SkRandom rand;
53
54 for (int i = 0; i < 65; i++) {
55 char* copyPtr = copy;
56 SkMemoryStream mem(s, sizeof(s));
57 SkBufferStream buff(&mem, i);
58
59 do {
60 copyPtr += buff.read(copyPtr, rand.nextU() & 15);
61 } while (copyPtr < copy + sizeof(s));
62 REPORTER_ASSERT(reporter, copyPtr == copy + sizeof(s));
63 REPORTER_ASSERT(reporter, memcmp(s, copy, sizeof(s)) == 0);
64 }
65 test_buffer(reporter);
66 }
67
TestWStream(skiatest::Reporter * reporter)68 static void TestWStream(skiatest::Reporter* reporter) {
69 SkDynamicMemoryWStream ds;
70 const char s[] = "abcdefghijklmnopqrstuvwxyz";
71 int i;
72 for (i = 0; i < 100; i++) {
73 REPORTER_ASSERT(reporter, ds.write(s, 26));
74 }
75 REPORTER_ASSERT(reporter, ds.getOffset() == 100 * 26);
76 char* dst = new char[100 * 26 + 1];
77 dst[100*26] = '*';
78 ds.copyTo(dst);
79 REPORTER_ASSERT(reporter, dst[100*26] == '*');
80 // char* p = dst;
81 for (i = 0; i < 100; i++) {
82 REPORTER_ASSERT(reporter, memcmp(&dst[i * 26], s, 26) == 0);
83 }
84 REPORTER_ASSERT(reporter, memcmp(dst, ds.getStream(), 100*26) == 0);
85 delete[] dst;
86 }
87
TestStreams(skiatest::Reporter * reporter)88 static void TestStreams(skiatest::Reporter* reporter) {
89 TestRStream(reporter);
90 TestWStream(reporter);
91 }
92
93 #include "TestClassDef.h"
94 DEFINE_TESTCLASS("Stream", StreamTestClass, TestStreams)
95