1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
|
#include <stdio.h>
#include <string.h>
#include "pb_decode.h"
#include "unittests.h"
#define S(x) pb_istream_from_buffer((uint8_t*)x, sizeof(x))
bool stream_callback(pb_istream_t *stream, uint8_t *buf, size_t count)
{
if (stream->state != NULL)
return false; /* Simulate error */
if (buf != NULL)
memset(buf, 'x', count);
return true;
}
int main()
{
int status = 0;
{
uint8_t buffer1[] = "foobartest1234";
uint8_t buffer2[sizeof(buffer1)];
pb_istream_t stream = pb_istream_from_buffer(buffer1, sizeof(buffer1));
COMMENT("Test pb_read and pb_istream_t");
TEST(pb_read(&stream, buffer2, 6))
TEST(memcmp(buffer2, "foobar", 6) == 0)
TEST(stream.bytes_left == sizeof(buffer1) - 6)
TEST(pb_read(&stream, buffer2 + 6, stream.bytes_left))
TEST(memcmp(buffer1, buffer2, sizeof(buffer1)) == 0)
TEST(stream.bytes_left == 0)
TEST(!pb_read(&stream, buffer2, 1))
}
{
uint8_t buffer[20];
pb_istream_t stream = {&stream_callback, NULL, 20};
COMMENT("Test pb_read with custom callback");
TEST(pb_read(&stream, buffer, 5))
TEST(memcmp(buffer, "xxxxx", 5) == 0)
TEST(!pb_read(&stream, buffer, 50))
stream.state = (void*)1; /* Simulated error return from callback */
TEST(!pb_read(&stream, buffer, 5))
stream.state = NULL;
TEST(pb_read(&stream, buffer, 15))
}
{
pb_istream_t s;
uint64_t u;
int64_t i;
COMMENT("Test pb_decode_varint");
TEST((s = S("\x00"), pb_decode_varint(&s, &u) && u == 0));
TEST((s = S("\x01"), pb_decode_varint(&s, &u) && u == 1));
TEST((s = S("\xAC\x02"), pb_decode_varint(&s, &u) && u == 300));
TEST((s = S("\xFF\xFF\xFF\xFF\x0F"), pb_decode_varint(&s, &u) && u == UINT32_MAX));
TEST((s = S("\xFF\xFF\xFF\xFF\x0F"), pb_decode_varint(&s, (uint64_t*)&i) && i == UINT32_MAX));
TEST((s = S("\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF\x01"),
pb_decode_varint(&s, (uint64_t*)&i) && i == -1));
TEST((s = S("\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF\x01"),
pb_decode_varint(&s, &u) && u == UINT64_MAX));
}
{
pb_istream_t s;
COMMENT("Test pb_skip_varint");
TEST((s = S("\x00""foobar"), pb_skip_varint(&s) && s.bytes_left == 7))
TEST((s = S("\xAC\x02""foobar"), pb_skip_varint(&s) && s.bytes_left == 7))
TEST((s = S("\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF\x01""foobar"),
pb_skip_varint(&s) && s.bytes_left == 7))
}
{
pb_istream_t s;
COMMENT("Test pb_skip_string")
TEST((s = S("\x00""foobar"), pb_skip_string(&s) && s.bytes_left == 7))
TEST((s = S("\x04""testfoobar"), pb_skip_string(&s) && s.bytes_left == 7))
}
{
pb_istream_t s = S("\x01\xFF\xFF\x03");
pb_field_t f = {1, PB_LTYPE_VARINT, 0, 0, 4, 0, 0};
uint32_t d;
COMMENT("Test pb_dec_varint using uint32_t")
TEST(pb_dec_varint(&s, &f, &d) && d == 1)
/* Verify that no more than data_size is written. */
d = 0;
f.data_size = 1;
TEST(pb_dec_varint(&s, &f, &d) && (d == 0xFF || d == 0xFF000000))
}
{
pb_istream_t s;
pb_field_t f = {1, PB_LTYPE_SVARINT, 0, 0, 4, 0, 0};
int32_t d;
COMMENT("Test pb_dec_svarint using int32_t")
TEST((s = S("\x01"), pb_dec_svarint(&s, &f, &d) && d == -1))
TEST((s = S("\x02"), pb_dec_svarint(&s, &f, &d) && d == 1))
TEST((s = S("\xfe\xff\xff\xff\x0f"), pb_dec_svarint(&s, &f, &d) && d == INT32_MAX))
TEST((s = S("\xff\xff\xff\xff\x0f"), pb_dec_svarint(&s, &f, &d) && d == INT32_MIN))
}
{
pb_istream_t s;
pb_field_t f = {1, PB_LTYPE_SVARINT, 0, 0, 8, 0, 0};
uint64_t d;
COMMENT("Test pb_dec_svarint using uint64_t")
TEST((s = S("\x01"), pb_dec_svarint(&s, &f, &d) && d == -1))
TEST((s = S("\x02"), pb_dec_svarint(&s, &f, &d) && d == 1))
TEST((s = S("\xFE\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF\x01"), pb_dec_svarint(&s, &f, &d) && d == INT64_MAX))
TEST((s = S("\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF\x01"), pb_dec_svarint(&s, &f, &d) && d == INT64_MIN))
}
if (status != 0)
fprintf(stdout, "\n\nSome tests FAILED!\n");
return status;
}
|
