1 /* This includes the whole .c file to get access to static functions. */
2 #include "pb_decode.c"
3
4 #include <stdio.h>
5 #include <string.h>
6 #include "unittests.h"
7 #include "unittestproto.pb.h"
8
9 #define S(x) pb_istream_from_buffer((uint8_t*)x, sizeof(x) - 1)
10
stream_callback(pb_istream_t * stream,uint8_t * buf,size_t count)11 bool stream_callback(pb_istream_t *stream, uint8_t *buf, size_t count)
12 {
13 if (stream->state != NULL)
14 return false; /* Simulate error */
15
16 if (buf != NULL)
17 memset(buf, 'x', count);
18 return true;
19 }
20
21 /* Verifies that the stream passed to callback matches the byte array pointed to by arg. */
callback_check(pb_istream_t * stream,const pb_field_t * field,void ** arg)22 bool callback_check(pb_istream_t *stream, const pb_field_t *field, void **arg)
23 {
24 int i;
25 uint8_t byte;
26 pb_bytes_array_t *ref = (pb_bytes_array_t*) *arg;
27
28 for (i = 0; i < ref->size; i++)
29 {
30 if (!pb_read(stream, &byte, 1))
31 return false;
32
33 if (byte != ref->bytes[i])
34 return false;
35 }
36
37 return true;
38 }
39
main()40 int main()
41 {
42 int status = 0;
43
44 {
45 uint8_t buffer1[] = "foobartest1234";
46 uint8_t buffer2[sizeof(buffer1)];
47 pb_istream_t stream = pb_istream_from_buffer(buffer1, sizeof(buffer1));
48
49 COMMENT("Test pb_read and pb_istream_t");
50 TEST(pb_read(&stream, buffer2, 6))
51 TEST(memcmp(buffer2, "foobar", 6) == 0)
52 TEST(stream.bytes_left == sizeof(buffer1) - 6)
53 TEST(pb_read(&stream, buffer2 + 6, stream.bytes_left))
54 TEST(memcmp(buffer1, buffer2, sizeof(buffer1)) == 0)
55 TEST(stream.bytes_left == 0)
56 TEST(!pb_read(&stream, buffer2, 1))
57 }
58
59 {
60 uint8_t buffer[20];
61 pb_istream_t stream = {&stream_callback, NULL, 20};
62
63 COMMENT("Test pb_read with custom callback");
64 TEST(pb_read(&stream, buffer, 5))
65 TEST(memcmp(buffer, "xxxxx", 5) == 0)
66 TEST(!pb_read(&stream, buffer, 50))
67 stream.state = (void*)1; /* Simulated error return from callback */
68 TEST(!pb_read(&stream, buffer, 5))
69 stream.state = NULL;
70 TEST(pb_read(&stream, buffer, 15))
71 }
72
73 {
74 pb_istream_t s;
75 uint64_t u;
76 int64_t i;
77
78 COMMENT("Test pb_decode_varint");
79 TEST((s = S("\x00"), pb_decode_varint(&s, &u) && u == 0));
80 TEST((s = S("\x01"), pb_decode_varint(&s, &u) && u == 1));
81 TEST((s = S("\xAC\x02"), pb_decode_varint(&s, &u) && u == 300));
82 TEST((s = S("\xFF\xFF\xFF\xFF\x0F"), pb_decode_varint(&s, &u) && u == UINT32_MAX));
83 TEST((s = S("\xFF\xFF\xFF\xFF\x0F"), pb_decode_varint(&s, (uint64_t*)&i) && i == UINT32_MAX));
84 TEST((s = S("\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF\x01"),
85 pb_decode_varint(&s, (uint64_t*)&i) && i == -1));
86 TEST((s = S("\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF\x01"),
87 pb_decode_varint(&s, &u) && u == UINT64_MAX));
88 }
89
90 {
91 pb_istream_t s;
92 COMMENT("Test pb_skip_varint");
93 TEST((s = S("\x00""foobar"), pb_skip_varint(&s) && s.bytes_left == 6))
94 TEST((s = S("\xAC\x02""foobar"), pb_skip_varint(&s) && s.bytes_left == 6))
95 TEST((s = S("\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF\x01""foobar"),
96 pb_skip_varint(&s) && s.bytes_left == 6))
97 TEST((s = S("\xFF"), !pb_skip_varint(&s)))
98 }
99
100 {
101 pb_istream_t s;
102 COMMENT("Test pb_skip_string")
103 TEST((s = S("\x00""foobar"), pb_skip_string(&s) && s.bytes_left == 6))
104 TEST((s = S("\x04""testfoobar"), pb_skip_string(&s) && s.bytes_left == 6))
105 TEST((s = S("\x04"), !pb_skip_string(&s)))
106 TEST((s = S("\xFF"), !pb_skip_string(&s)))
107 }
108
109 {
110 pb_istream_t s = S("\x01\xFF\xFF\x03");
111 pb_field_t f = {1, PB_LTYPE_VARINT, 0, 0, 4, 0, 0};
112 uint32_t d;
113 COMMENT("Test pb_dec_varint using uint32_t")
114 TEST(pb_dec_varint(&s, &f, &d) && d == 1)
115
116 /* Verify that no more than data_size is written. */
117 d = 0;
118 f.data_size = 1;
119 TEST(pb_dec_varint(&s, &f, &d) && (d == 0xFF || d == 0xFF000000))
120 }
121
122 {
123 pb_istream_t s;
124 pb_field_t f = {1, PB_LTYPE_SVARINT, 0, 0, 4, 0, 0};
125 int32_t d;
126
127 COMMENT("Test pb_dec_svarint using int32_t")
128 TEST((s = S("\x01"), pb_dec_svarint(&s, &f, &d) && d == -1))
129 TEST((s = S("\x02"), pb_dec_svarint(&s, &f, &d) && d == 1))
130 TEST((s = S("\xfe\xff\xff\xff\x0f"), pb_dec_svarint(&s, &f, &d) && d == INT32_MAX))
131 TEST((s = S("\xff\xff\xff\xff\x0f"), pb_dec_svarint(&s, &f, &d) && d == INT32_MIN))
132 }
133
134 {
135 pb_istream_t s;
136 pb_field_t f = {1, PB_LTYPE_SVARINT, 0, 0, 8, 0, 0};
137 uint64_t d;
138
139 COMMENT("Test pb_dec_svarint using uint64_t")
140 TEST((s = S("\x01"), pb_dec_svarint(&s, &f, &d) && d == -1))
141 TEST((s = S("\x02"), pb_dec_svarint(&s, &f, &d) && d == 1))
142 TEST((s = S("\xFE\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF\x01"), pb_dec_svarint(&s, &f, &d) && d == INT64_MAX))
143 TEST((s = S("\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF\x01"), pb_dec_svarint(&s, &f, &d) && d == INT64_MIN))
144 }
145
146 {
147 pb_istream_t s;
148 pb_field_t f = {1, PB_LTYPE_FIXED32, 0, 0, 4, 0, 0};
149 float d;
150
151 COMMENT("Test pb_dec_fixed32 using float (failures here may be caused by imperfect rounding)")
152 TEST((s = S("\x00\x00\x00\x00"), pb_dec_fixed32(&s, &f, &d) && d == 0.0f))
153 TEST((s = S("\x00\x00\xc6\x42"), pb_dec_fixed32(&s, &f, &d) && d == 99.0f))
154 TEST((s = S("\x4e\x61\x3c\xcb"), pb_dec_fixed32(&s, &f, &d) && d == -12345678.0f))
155 TEST((s = S("\x00"), !pb_dec_fixed32(&s, &f, &d) && d == -12345678.0f))
156 }
157
158 {
159 pb_istream_t s;
160 pb_field_t f = {1, PB_LTYPE_FIXED64, 0, 0, 8, 0, 0};
161 double d;
162
163 COMMENT("Test pb_dec_fixed64 using double (failures here may be caused by imperfect rounding)")
164 TEST((s = S("\x00\x00\x00\x00\x00\x00\x00\x00"), pb_dec_fixed64(&s, &f, &d) && d == 0.0))
165 TEST((s = S("\x00\x00\x00\x00\x00\xc0\x58\x40"), pb_dec_fixed64(&s, &f, &d) && d == 99.0))
166 TEST((s = S("\x00\x00\x00\xc0\x29\x8c\x67\xc1"), pb_dec_fixed64(&s, &f, &d) && d == -12345678.0f))
167 }
168
169 {
170 pb_istream_t s;
171 struct { size_t size; uint8_t bytes[5]; } d;
172 pb_field_t f = {1, PB_LTYPE_BYTES, 0, 0, sizeof(d), 0, 0};
173
174 COMMENT("Test pb_dec_bytes")
175 TEST((s = S("\x00"), pb_dec_bytes(&s, &f, &d) && d.size == 0))
176 TEST((s = S("\x01\xFF"), pb_dec_bytes(&s, &f, &d) && d.size == 1 && d.bytes[0] == 0xFF))
177 TEST((s = S("\x05xxxxx"), pb_dec_bytes(&s, &f, &d) && d.size == 5))
178 TEST((s = S("\x05xxxx"), !pb_dec_bytes(&s, &f, &d)))
179
180 /* Note: the size limit on bytes-fields is not strictly obeyed, as
181 * the compiler may add some padding to the struct. Using this padding
182 * is not a very good thing to do, but it is difficult to avoid when
183 * we use only a single uint8_t to store the size of the field.
184 * Therefore this tests against a 10-byte string, while otherwise even
185 * 6 bytes should error out.
186 */
187 TEST((s = S("\x10xxxxxxxxxx"), !pb_dec_bytes(&s, &f, &d)))
188 }
189
190 {
191 pb_istream_t s;
192 pb_field_t f = {1, PB_LTYPE_STRING, 0, 0, 5, 0, 0};
193 char d[5];
194
195 COMMENT("Test pb_dec_string")
196 TEST((s = S("\x00"), pb_dec_string(&s, &f, &d) && d[0] == '\0'))
197 TEST((s = S("\x04xyzz"), pb_dec_string(&s, &f, &d) && strcmp(d, "xyzz") == 0))
198 TEST((s = S("\x05xyzzy"), !pb_dec_string(&s, &f, &d)))
199 }
200
201 {
202 pb_istream_t s;
203 IntegerArray dest;
204
205 COMMENT("Testing pb_decode with repeated int32 field")
206 TEST((s = S(""), pb_decode(&s, IntegerArray_fields, &dest) && dest.data_count == 0))
207 TEST((s = S("\x08\x01\x08\x02"), pb_decode(&s, IntegerArray_fields, &dest)
208 && dest.data_count == 2 && dest.data[0] == 1 && dest.data[1] == 2))
209 s = S("\x08\x01\x08\x02\x08\x03\x08\x04\x08\x05\x08\x06\x08\x07\x08\x08\x08\x09\x08\x0A");
210 TEST(pb_decode(&s, IntegerArray_fields, &dest) && dest.data_count == 10 && dest.data[9] == 10)
211 s = S("\x08\x01\x08\x02\x08\x03\x08\x04\x08\x05\x08\x06\x08\x07\x08\x08\x08\x09\x08\x0A\x08\x0B");
212 TEST(!pb_decode(&s, IntegerArray_fields, &dest))
213 }
214
215 {
216 pb_istream_t s;
217 IntegerArray dest;
218
219 COMMENT("Testing pb_decode with packed int32 field")
220 TEST((s = S("\x0A\x00"), pb_decode(&s, IntegerArray_fields, &dest)
221 && dest.data_count == 0))
222 TEST((s = S("\x0A\x01\x01"), pb_decode(&s, IntegerArray_fields, &dest)
223 && dest.data_count == 1 && dest.data[0] == 1))
224 TEST((s = S("\x0A\x0A\x01\x02\x03\x04\x05\x06\x07\x08\x09\x0A"), pb_decode(&s, IntegerArray_fields, &dest)
225 && dest.data_count == 10 && dest.data[0] == 1 && dest.data[9] == 10))
226 TEST((s = S("\x0A\x0B\x01\x02\x03\x04\x05\x06\x07\x08\x09\x0A\x0B"), !pb_decode(&s, IntegerArray_fields, &dest)))
227
228 /* Test invalid wire data */
229 TEST((s = S("\x0A\xFF"), !pb_decode(&s, IntegerArray_fields, &dest)))
230 TEST((s = S("\x0A\x01"), !pb_decode(&s, IntegerArray_fields, &dest)))
231 }
232
233 {
234 pb_istream_t s;
235 IntegerArray dest;
236
237 COMMENT("Testing pb_decode with unknown fields")
238 TEST((s = S("\x18\x0F\x08\x01"), pb_decode(&s, IntegerArray_fields, &dest)
239 && dest.data_count == 1 && dest.data[0] == 1))
240 TEST((s = S("\x19\x00\x00\x00\x00\x00\x00\x00\x00\x08\x01"), pb_decode(&s, IntegerArray_fields, &dest)
241 && dest.data_count == 1 && dest.data[0] == 1))
242 TEST((s = S("\x1A\x00\x08\x01"), pb_decode(&s, IntegerArray_fields, &dest)
243 && dest.data_count == 1 && dest.data[0] == 1))
244 TEST((s = S("\x1B\x08\x01"), !pb_decode(&s, IntegerArray_fields, &dest)))
245 TEST((s = S("\x1D\x00\x00\x00\x00\x08\x01"), pb_decode(&s, IntegerArray_fields, &dest)
246 && dest.data_count == 1 && dest.data[0] == 1))
247 }
248
249 {
250 pb_istream_t s;
251 CallbackArray dest;
252 struct { size_t size; uint8_t bytes[10]; } ref;
253 dest.data.funcs.decode = &callback_check;
254 dest.data.arg = &ref;
255
256 COMMENT("Testing pb_decode with callbacks")
257 /* Single varint */
258 ref.size = 1; ref.bytes[0] = 0x55;
259 TEST((s = S("\x08\x55"), pb_decode(&s, CallbackArray_fields, &dest)))
260 /* Packed varint */
261 ref.size = 3; ref.bytes[0] = ref.bytes[1] = ref.bytes[2] = 0x55;
262 TEST((s = S("\x0A\x03\x55\x55\x55"), pb_decode(&s, CallbackArray_fields, &dest)))
263 /* Packed varint with loop */
264 ref.size = 1; ref.bytes[0] = 0x55;
265 TEST((s = S("\x0A\x03\x55\x55\x55"), pb_decode(&s, CallbackArray_fields, &dest)))
266 /* Single fixed32 */
267 ref.size = 4; ref.bytes[0] = ref.bytes[1] = ref.bytes[2] = ref.bytes[3] = 0xAA;
268 TEST((s = S("\x0D\xAA\xAA\xAA\xAA"), pb_decode(&s, CallbackArray_fields, &dest)))
269 /* Single fixed64 */
270 ref.size = 8; memset(ref.bytes, 0xAA, 8);
271 TEST((s = S("\x09\xAA\xAA\xAA\xAA\xAA\xAA\xAA\xAA"), pb_decode(&s, CallbackArray_fields, &dest)))
272 /* Unsupported field type */
273 TEST((s = S("\x0B\x00"), !pb_decode(&s, CallbackArray_fields, &dest)))
274
275 /* Just make sure that our test function works */
276 ref.size = 1; ref.bytes[0] = 0x56;
277 TEST((s = S("\x08\x55"), !pb_decode(&s, CallbackArray_fields, &dest)))
278 }
279
280 {
281 pb_istream_t s;
282 IntegerArray dest;
283
284 COMMENT("Testing pb_decode message termination")
285 TEST((s = S(""), pb_decode(&s, IntegerArray_fields, &dest)))
286 TEST((s = S("\x00"), pb_decode(&s, IntegerArray_fields, &dest)))
287 TEST((s = S("\x08\x01"), pb_decode(&s, IntegerArray_fields, &dest)))
288 TEST((s = S("\x08\x01\x00"), pb_decode(&s, IntegerArray_fields, &dest)))
289 TEST((s = S("\x08"), !pb_decode(&s, IntegerArray_fields, &dest)))
290 }
291
292 {
293 pb_istream_t s;
294 IntegerContainer dest = {{0}};
295
296 COMMENT("Testing pb_decode_delimited")
297 TEST((s = S("\x09\x0A\x07\x0A\x05\x01\x02\x03\x04\x05"),
298 pb_decode_delimited(&s, IntegerContainer_fields, &dest)) &&
299 dest.submsg.data_count == 5)
300 }
301
302 if (status != 0)
303 fprintf(stdout, "\n\nSome tests FAILED!\n");
304
305 return status;
306 }
307