1 /* Attempts to test all the datatypes supported by ProtoBuf when used as callback fields.
2 * Note that normally there would be no reason to use callback fields for this,
3 * because each encoder defined here only gives a single field.
10 #include "alltypes.pb.h"
11 #include "test_helpers.h"
13 #define TEST(x) if (!(x)) { \
14 printf("Test " #x " failed (in field %d).\n", field->tag); \
18 static bool read_varint(pb_istream_t *stream, const pb_field_t *field, void **arg)
21 if (!pb_decode_varint(stream, &value))
24 TEST((int64_t)value == (long)*arg);
28 static bool read_svarint(pb_istream_t *stream, const pb_field_t *field, void **arg)
31 if (!pb_decode_svarint(stream, &value))
34 TEST(value == (long)*arg);
38 static bool read_fixed32(pb_istream_t *stream, const pb_field_t *field, void **arg)
41 if (!pb_decode_fixed32(stream, &value))
44 TEST(value == *(uint32_t*)*arg);
48 static bool read_fixed64(pb_istream_t *stream, const pb_field_t *field, void **arg)
51 if (!pb_decode_fixed64(stream, &value))
54 TEST(value == *(uint64_t*)*arg);
58 static bool read_string(pb_istream_t *stream, const pb_field_t *field, void **arg)
60 uint8_t buf[16] = {0};
61 size_t len = stream->bytes_left;
63 if (len > sizeof(buf) - 1 || !pb_read(stream, buf, len))
66 TEST(strcmp((char*)buf, *arg) == 0);
70 static bool read_submsg(pb_istream_t *stream, const pb_field_t *field, void **arg)
72 SubMessage submsg = {""};
74 if (!pb_decode(stream, SubMessage_fields, &submsg))
77 TEST(memcmp(&submsg, *arg, sizeof(submsg)));
81 static bool read_emptymsg(pb_istream_t *stream, const pb_field_t *field, void **arg)
83 EmptyMessage emptymsg = {0};
84 return pb_decode(stream, EmptyMessage_fields, &emptymsg);
87 static bool read_repeated_varint(pb_istream_t *stream, const pb_field_t *field, void **arg)
89 int32_t** expected = (int32_t**)arg;
91 if (!pb_decode_varint(stream, &value))
94 TEST(*(*expected)++ == value);
98 static bool read_repeated_svarint(pb_istream_t *stream, const pb_field_t *field, void **arg)
100 int32_t** expected = (int32_t**)arg;
102 if (!pb_decode_svarint(stream, &value))
105 TEST(*(*expected)++ == value);
109 static bool read_repeated_fixed32(pb_istream_t *stream, const pb_field_t *field, void **arg)
111 uint32_t** expected = (uint32_t**)arg;
113 if (!pb_decode_fixed32(stream, &value))
116 TEST(*(*expected)++ == value);
120 static bool read_repeated_fixed64(pb_istream_t *stream, const pb_field_t *field, void **arg)
122 uint64_t** expected = (uint64_t**)arg;
124 if (!pb_decode_fixed64(stream, &value))
127 TEST(*(*expected)++ == value);
131 static bool read_repeated_string(pb_istream_t *stream, const pb_field_t *field, void **arg)
133 uint8_t*** expected = (uint8_t***)arg;
134 uint8_t buf[16] = {0};
135 size_t len = stream->bytes_left;
137 if (len > sizeof(buf) - 1 || !pb_read(stream, buf, len))
140 TEST(strcmp((char*)*(*expected)++, (char*)buf) == 0);
144 static bool read_repeated_submsg(pb_istream_t *stream, const pb_field_t *field, void **arg)
146 SubMessage** expected = (SubMessage**)arg;
147 SubMessage decoded = {""};
148 if (!pb_decode(stream, SubMessage_fields, &decoded))
151 TEST(memcmp((*expected)++, &decoded, sizeof(decoded)) == 0);
155 static bool read_limits(pb_istream_t *stream, const pb_field_t *field, void **arg)
157 Limits decoded = {0};
158 if (!pb_decode(stream, Limits_fields, &decoded))
161 TEST(decoded.int32_min == INT32_MIN);
162 TEST(decoded.int32_max == INT32_MAX);
163 TEST(decoded.uint32_min == 0);
164 TEST(decoded.uint32_max == UINT32_MAX);
165 TEST(decoded.int64_min == INT64_MIN);
166 TEST(decoded.int64_max == INT64_MAX);
167 TEST(decoded.uint64_min == 0);
168 TEST(decoded.uint64_max == UINT64_MAX);
169 TEST(decoded.enum_min == HugeEnum_Negative);
170 TEST(decoded.enum_max == HugeEnum_Positive);
175 /* This function is called once from main(), it handles
176 the decoding and checks the fields. */
177 bool check_alltypes(pb_istream_t *stream, int mode)
179 /* Values for use from callbacks through pointers. */
180 uint32_t req_fixed32 = 1008;
181 int32_t req_sfixed32 = -1009;
182 float req_float = 1010.0f;
183 uint64_t req_fixed64 = 1011;
184 int64_t req_sfixed64 = -1012;
185 double req_double = 1013.0;
186 SubMessage req_submsg = {"1016", 1016};
188 int32_t rep_int32[5] = {0, 0, 0, 0, -2001};
189 int32_t rep_int64[5] = {0, 0, 0, 0, -2002};
190 int32_t rep_uint32[5] = {0, 0, 0, 0, 2003};
191 int32_t rep_uint64[5] = {0, 0, 0, 0, 2004};
192 int32_t rep_sint32[5] = {0, 0, 0, 0, -2005};
193 int32_t rep_sint64[5] = {0, 0, 0, 0, -2006};
194 int32_t rep_bool[5] = {false, false, false, false, true};
195 uint32_t rep_fixed32[5] = {0, 0, 0, 0, 2008};
196 int32_t rep_sfixed32[5] = {0, 0, 0, 0, -2009};
197 float rep_float[5] = {0, 0, 0, 0, 2010.0f};
198 uint64_t rep_fixed64[5] = {0, 0, 0, 0, 2011};
199 int64_t rep_sfixed64[5] = {0, 0, 0, 0, -2012};
200 double rep_double[5] = {0, 0, 0, 0, 2013.0};
201 char* rep_string[5] = {"", "", "", "", "2014"};
202 char* rep_bytes[5] = {"", "", "", "", "2015"};
203 SubMessage rep_submsg[5] = {{"", 0, 0, 3},
207 {"2016", 2016, true, 2016}};
208 int32_t rep_enum[5] = {0, 0, 0, 0, MyEnum_Truth};
210 uint32_t opt_fixed32 = 3048;
211 int32_t opt_sfixed32 = 3049;
212 float opt_float = 3050.0f;
213 uint64_t opt_fixed64 = 3051;
214 int64_t opt_sfixed64 = 3052;
215 double opt_double = 3053.0f;
216 SubMessage opt_submsg = {"3056", 3056};
218 SubMessage oneof_msg1 = {"4059", 4059};
220 /* Bind callbacks for required fields */
223 /* Fill with garbage to better detect initialization errors */
224 memset(&alltypes, 0xAA, sizeof(alltypes));
225 alltypes.extensions = 0;
227 alltypes.req_int32.funcs.decode = &read_varint;
228 alltypes.req_int32.arg = (void*)-1001;
230 alltypes.req_int64.funcs.decode = &read_varint;
231 alltypes.req_int64.arg = (void*)-1002;
233 alltypes.req_uint32.funcs.decode = &read_varint;
234 alltypes.req_uint32.arg = (void*)1003;
236 alltypes.req_uint32.funcs.decode = &read_varint;
237 alltypes.req_uint32.arg = (void*)1003;
239 alltypes.req_uint64.funcs.decode = &read_varint;
240 alltypes.req_uint64.arg = (void*)1004;
242 alltypes.req_sint32.funcs.decode = &read_svarint;
243 alltypes.req_sint32.arg = (void*)-1005;
245 alltypes.req_sint64.funcs.decode = &read_svarint;
246 alltypes.req_sint64.arg = (void*)-1006;
248 alltypes.req_bool.funcs.decode = &read_varint;
249 alltypes.req_bool.arg = (void*)true;
251 alltypes.req_fixed32.funcs.decode = &read_fixed32;
252 alltypes.req_fixed32.arg = &req_fixed32;
254 alltypes.req_sfixed32.funcs.decode = &read_fixed32;
255 alltypes.req_sfixed32.arg = &req_sfixed32;
257 alltypes.req_float.funcs.decode = &read_fixed32;
258 alltypes.req_float.arg = &req_float;
260 alltypes.req_fixed64.funcs.decode = &read_fixed64;
261 alltypes.req_fixed64.arg = &req_fixed64;
263 alltypes.req_sfixed64.funcs.decode = &read_fixed64;
264 alltypes.req_sfixed64.arg = &req_sfixed64;
266 alltypes.req_double.funcs.decode = &read_fixed64;
267 alltypes.req_double.arg = &req_double;
269 alltypes.req_string.funcs.decode = &read_string;
270 alltypes.req_string.arg = "1014";
272 alltypes.req_bytes.funcs.decode = &read_string;
273 alltypes.req_bytes.arg = "1015";
275 alltypes.req_submsg.funcs.decode = &read_submsg;
276 alltypes.req_submsg.arg = &req_submsg;
278 alltypes.req_enum.funcs.decode = &read_varint;
279 alltypes.req_enum.arg = (void*)MyEnum_Truth;
281 alltypes.req_emptymsg.funcs.decode = &read_emptymsg;
283 /* Bind callbacks for repeated fields */
284 alltypes.rep_int32.funcs.decode = &read_repeated_varint;
285 alltypes.rep_int32.arg = rep_int32;
287 alltypes.rep_int64.funcs.decode = &read_repeated_varint;
288 alltypes.rep_int64.arg = rep_int64;
290 alltypes.rep_uint32.funcs.decode = &read_repeated_varint;
291 alltypes.rep_uint32.arg = rep_uint32;
293 alltypes.rep_uint64.funcs.decode = &read_repeated_varint;
294 alltypes.rep_uint64.arg = rep_uint64;
296 alltypes.rep_sint32.funcs.decode = &read_repeated_svarint;
297 alltypes.rep_sint32.arg = rep_sint32;
299 alltypes.rep_sint64.funcs.decode = &read_repeated_svarint;
300 alltypes.rep_sint64.arg = rep_sint64;
302 alltypes.rep_bool.funcs.decode = &read_repeated_varint;
303 alltypes.rep_bool.arg = rep_bool;
305 alltypes.rep_fixed32.funcs.decode = &read_repeated_fixed32;
306 alltypes.rep_fixed32.arg = rep_fixed32;
308 alltypes.rep_sfixed32.funcs.decode = &read_repeated_fixed32;
309 alltypes.rep_sfixed32.arg = rep_sfixed32;
311 alltypes.rep_float.funcs.decode = &read_repeated_fixed32;
312 alltypes.rep_float.arg = rep_float;
314 alltypes.rep_fixed64.funcs.decode = &read_repeated_fixed64;
315 alltypes.rep_fixed64.arg = rep_fixed64;
317 alltypes.rep_sfixed64.funcs.decode = &read_repeated_fixed64;
318 alltypes.rep_sfixed64.arg = rep_sfixed64;
320 alltypes.rep_double.funcs.decode = &read_repeated_fixed64;
321 alltypes.rep_double.arg = rep_double;
323 alltypes.rep_string.funcs.decode = &read_repeated_string;
324 alltypes.rep_string.arg = rep_string;
326 alltypes.rep_bytes.funcs.decode = &read_repeated_string;
327 alltypes.rep_bytes.arg = rep_bytes;
329 alltypes.rep_submsg.funcs.decode = &read_repeated_submsg;
330 alltypes.rep_submsg.arg = rep_submsg;
332 alltypes.rep_enum.funcs.decode = &read_repeated_varint;
333 alltypes.rep_enum.arg = rep_enum;
335 alltypes.rep_emptymsg.funcs.decode = &read_emptymsg;
337 alltypes.req_limits.funcs.decode = &read_limits;
339 alltypes.end.funcs.decode = &read_varint;
340 alltypes.end.arg = (void*)1099;
342 /* Bind callbacks for optional fields */
345 alltypes.opt_int32.funcs.decode = &read_varint;
346 alltypes.opt_int32.arg = (void*)3041;
348 alltypes.opt_int64.funcs.decode = &read_varint;
349 alltypes.opt_int64.arg = (void*)3042;
351 alltypes.opt_uint32.funcs.decode = &read_varint;
352 alltypes.opt_uint32.arg = (void*)3043;
354 alltypes.opt_uint64.funcs.decode = &read_varint;
355 alltypes.opt_uint64.arg = (void*)3044;
357 alltypes.opt_sint32.funcs.decode = &read_svarint;
358 alltypes.opt_sint32.arg = (void*)3045;
360 alltypes.opt_sint64.funcs.decode = &read_svarint;
361 alltypes.opt_sint64.arg = (void*)3046;
363 alltypes.opt_bool.funcs.decode = &read_varint;
364 alltypes.opt_bool.arg = (void*)true;
366 alltypes.opt_fixed32.funcs.decode = &read_fixed32;
367 alltypes.opt_fixed32.arg = &opt_fixed32;
369 alltypes.opt_sfixed32.funcs.decode = &read_fixed32;
370 alltypes.opt_sfixed32.arg = &opt_sfixed32;
372 alltypes.opt_float.funcs.decode = &read_fixed32;
373 alltypes.opt_float.arg = &opt_float;
375 alltypes.opt_fixed64.funcs.decode = &read_fixed64;
376 alltypes.opt_fixed64.arg = &opt_fixed64;
378 alltypes.opt_sfixed64.funcs.decode = &read_fixed64;
379 alltypes.opt_sfixed64.arg = &opt_sfixed64;
381 alltypes.opt_double.funcs.decode = &read_fixed64;
382 alltypes.opt_double.arg = &opt_double;
384 alltypes.opt_string.funcs.decode = &read_string;
385 alltypes.opt_string.arg = "3054";
387 alltypes.opt_bytes.funcs.decode = &read_string;
388 alltypes.opt_bytes.arg = "3055";
390 alltypes.opt_submsg.funcs.decode = &read_submsg;
391 alltypes.opt_submsg.arg = &opt_submsg;
393 alltypes.opt_enum.funcs.decode = &read_varint;
394 alltypes.opt_enum.arg = (void*)MyEnum_Truth;
396 alltypes.opt_emptymsg.funcs.decode = &read_emptymsg;
398 alltypes.oneof_msg1.funcs.decode = &read_submsg;
399 alltypes.oneof_msg1.arg = &oneof_msg1;
402 return pb_decode(stream, AllTypes_fields, &alltypes);
405 int main(int argc, char **argv)
407 uint8_t buffer[1024];
411 /* Whether to expect the optional values or the default values. */
412 int mode = (argc > 1) ? atoi(argv[1]) : 0;
414 /* Read the data into buffer */
415 SET_BINARY_MODE(stdin);
416 count = fread(buffer, 1, sizeof(buffer), stdin);
418 /* Construct a pb_istream_t for reading from the buffer */
419 stream = pb_istream_from_buffer(buffer, count);
421 /* Decode and print out the stuff */
422 if (!check_alltypes(&stream, mode))
424 printf("Parsing failed: %s\n", PB_GET_ERROR(&stream));