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 static bool write_varint(pb_ostream_t *stream, const pb_field_t *field, void * const *arg)
15 return pb_encode_tag_for_field(stream, field) &&
16 pb_encode_varint(stream, (long)*arg);
19 static bool write_svarint(pb_ostream_t *stream, const pb_field_t *field, void * const *arg)
21 return pb_encode_tag_for_field(stream, field) &&
22 pb_encode_svarint(stream, (long)*arg);
25 static bool write_fixed32(pb_ostream_t *stream, const pb_field_t *field, void * const *arg)
27 return pb_encode_tag_for_field(stream, field) &&
28 pb_encode_fixed32(stream, *arg);
31 static bool write_fixed64(pb_ostream_t *stream, const pb_field_t *field, void * const *arg)
33 return pb_encode_tag_for_field(stream, field) &&
34 pb_encode_fixed64(stream, *arg);
37 static bool write_string(pb_ostream_t *stream, const pb_field_t *field, void * const *arg)
39 return pb_encode_tag_for_field(stream, field) &&
40 pb_encode_string(stream, *arg, strlen(*arg));
43 static bool write_submsg(pb_ostream_t *stream, const pb_field_t *field, void * const *arg)
46 return pb_encode_tag_for_field(stream, field) &&
47 pb_encode_submessage(stream, SubMessage_fields, *arg);
50 static bool write_emptymsg(pb_ostream_t *stream, const pb_field_t *field, void * const *arg)
52 EmptyMessage emptymsg = {0};
53 return pb_encode_tag_for_field(stream, field) &&
54 pb_encode_submessage(stream, EmptyMessage_fields, &emptymsg);
57 static bool write_repeated_varint(pb_ostream_t *stream, const pb_field_t *field, void * const *arg)
59 return pb_encode_tag_for_field(stream, field) &&
60 pb_encode_varint(stream, 0) &&
61 pb_encode_tag_for_field(stream, field) &&
62 pb_encode_varint(stream, 0) &&
63 pb_encode_tag_for_field(stream, field) &&
64 pb_encode_varint(stream, 0) &&
65 pb_encode_tag_for_field(stream, field) &&
66 pb_encode_varint(stream, 0) &&
67 pb_encode_tag_for_field(stream, field) &&
68 pb_encode_varint(stream, (long)*arg);
71 static bool write_repeated_svarint(pb_ostream_t *stream, const pb_field_t *field, void * const *arg)
73 return pb_encode_tag_for_field(stream, field) &&
74 pb_encode_svarint(stream, 0) &&
75 pb_encode_tag_for_field(stream, field) &&
76 pb_encode_svarint(stream, 0) &&
77 pb_encode_tag_for_field(stream, field) &&
78 pb_encode_svarint(stream, 0) &&
79 pb_encode_tag_for_field(stream, field) &&
80 pb_encode_svarint(stream, 0) &&
81 pb_encode_tag_for_field(stream, field) &&
82 pb_encode_svarint(stream, (long)*arg);
85 static bool write_repeated_fixed32(pb_ostream_t *stream, const pb_field_t *field, void * const *arg)
89 /* Make it a packed field */
90 return pb_encode_tag(stream, PB_WT_STRING, field->tag) &&
91 pb_encode_varint(stream, 5 * 4) && /* Number of bytes */
92 pb_encode_fixed32(stream, &dummy) &&
93 pb_encode_fixed32(stream, &dummy) &&
94 pb_encode_fixed32(stream, &dummy) &&
95 pb_encode_fixed32(stream, &dummy) &&
96 pb_encode_fixed32(stream, *arg);
99 static bool write_repeated_fixed64(pb_ostream_t *stream, const pb_field_t *field, void * const *arg)
103 /* Make it a packed field */
104 return pb_encode_tag(stream, PB_WT_STRING, field->tag) &&
105 pb_encode_varint(stream, 5 * 8) && /* Number of bytes */
106 pb_encode_fixed64(stream, &dummy) &&
107 pb_encode_fixed64(stream, &dummy) &&
108 pb_encode_fixed64(stream, &dummy) &&
109 pb_encode_fixed64(stream, &dummy) &&
110 pb_encode_fixed64(stream, *arg);
113 static bool write_repeated_string(pb_ostream_t *stream, const pb_field_t *field, void * const *arg)
115 return pb_encode_tag_for_field(stream, field) &&
116 pb_encode_string(stream, 0, 0) &&
117 pb_encode_tag_for_field(stream, field) &&
118 pb_encode_string(stream, 0, 0) &&
119 pb_encode_tag_for_field(stream, field) &&
120 pb_encode_string(stream, 0, 0) &&
121 pb_encode_tag_for_field(stream, field) &&
122 pb_encode_string(stream, 0, 0) &&
123 pb_encode_tag_for_field(stream, field) &&
124 pb_encode_string(stream, *arg, strlen(*arg));
127 static bool write_repeated_submsg(pb_ostream_t *stream, const pb_field_t *field, void * const *arg)
129 SubMessage dummy = {""};
131 return pb_encode_tag_for_field(stream, field) &&
132 pb_encode_submessage(stream, SubMessage_fields, &dummy) &&
133 pb_encode_tag_for_field(stream, field) &&
134 pb_encode_submessage(stream, SubMessage_fields, &dummy) &&
135 pb_encode_tag_for_field(stream, field) &&
136 pb_encode_submessage(stream, SubMessage_fields, &dummy) &&
137 pb_encode_tag_for_field(stream, field) &&
138 pb_encode_submessage(stream, SubMessage_fields, &dummy) &&
139 pb_encode_tag_for_field(stream, field) &&
140 pb_encode_submessage(stream, SubMessage_fields, *arg);
143 static bool write_limits(pb_ostream_t *stream, const pb_field_t *field, void * const *arg)
146 limits.int32_min = INT32_MIN;
147 limits.int32_max = INT32_MAX;
148 limits.uint32_min = 0;
149 limits.uint32_max = UINT32_MAX;
150 limits.int64_min = INT64_MIN;
151 limits.int64_max = INT64_MAX;
152 limits.uint64_min = 0;
153 limits.uint64_max = UINT64_MAX;
154 limits.enum_min = HugeEnum_Negative;
155 limits.enum_max = HugeEnum_Positive;
157 return pb_encode_tag_for_field(stream, field) &&
158 pb_encode_submessage(stream, Limits_fields, &limits);
161 static bool write_repeated_emptymsg(pb_ostream_t *stream, const pb_field_t *field, void * const *arg)
163 EmptyMessage emptymsg = {0};
164 return pb_encode_tag_for_field(stream, field) &&
165 pb_encode_submessage(stream, EmptyMessage_fields, &emptymsg) &&
166 pb_encode_tag_for_field(stream, field) &&
167 pb_encode_submessage(stream, EmptyMessage_fields, &emptymsg) &&
168 pb_encode_tag_for_field(stream, field) &&
169 pb_encode_submessage(stream, EmptyMessage_fields, &emptymsg) &&
170 pb_encode_tag_for_field(stream, field) &&
171 pb_encode_submessage(stream, EmptyMessage_fields, &emptymsg) &&
172 pb_encode_tag_for_field(stream, field) &&
173 pb_encode_submessage(stream, EmptyMessage_fields, &emptymsg);
176 int main(int argc, char **argv)
178 int mode = (argc > 1) ? atoi(argv[1]) : 0;
180 /* Values for use from callbacks through pointers. */
181 uint32_t req_fixed32 = 1008;
182 int32_t req_sfixed32 = -1009;
183 float req_float = 1010.0f;
184 uint64_t req_fixed64 = 1011;
185 int64_t req_sfixed64 = -1012;
186 double req_double = 1013.0;
187 SubMessage req_submsg = {"1016", 1016};
189 uint32_t rep_fixed32 = 2008;
190 int32_t rep_sfixed32 = -2009;
191 float rep_float = 2010.0f;
192 uint64_t rep_fixed64 = 2011;
193 int64_t rep_sfixed64 = -2012;
194 double rep_double = 2013.0;
195 SubMessage rep_submsg = {"2016", 2016, true, 2016};
197 uint32_t opt_fixed32 = 3048;
198 int32_t opt_sfixed32 = 3049;
199 float opt_float = 3050.0f;
200 uint64_t opt_fixed64 = 3051;
201 int64_t opt_sfixed64 = 3052;
202 double opt_double = 3053.0f;
203 SubMessage opt_submsg = {"3056", 3056};
205 SubMessage oneof_msg1 = {"4059", 4059};
207 /* Bind callbacks for required fields */
208 AllTypes alltypes = {{{0}}};
210 alltypes.req_int32.funcs.encode = &write_varint;
211 alltypes.req_int32.arg = (void*)-1001;
213 alltypes.req_int64.funcs.encode = &write_varint;
214 alltypes.req_int64.arg = (void*)-1002;
216 alltypes.req_uint32.funcs.encode = &write_varint;
217 alltypes.req_uint32.arg = (void*)1003;
219 alltypes.req_uint32.funcs.encode = &write_varint;
220 alltypes.req_uint32.arg = (void*)1003;
222 alltypes.req_uint64.funcs.encode = &write_varint;
223 alltypes.req_uint64.arg = (void*)1004;
225 alltypes.req_sint32.funcs.encode = &write_svarint;
226 alltypes.req_sint32.arg = (void*)-1005;
228 alltypes.req_sint64.funcs.encode = &write_svarint;
229 alltypes.req_sint64.arg = (void*)-1006;
231 alltypes.req_bool.funcs.encode = &write_varint;
232 alltypes.req_bool.arg = (void*)true;
234 alltypes.req_fixed32.funcs.encode = &write_fixed32;
235 alltypes.req_fixed32.arg = &req_fixed32;
237 alltypes.req_sfixed32.funcs.encode = &write_fixed32;
238 alltypes.req_sfixed32.arg = &req_sfixed32;
240 alltypes.req_float.funcs.encode = &write_fixed32;
241 alltypes.req_float.arg = &req_float;
243 alltypes.req_fixed64.funcs.encode = &write_fixed64;
244 alltypes.req_fixed64.arg = &req_fixed64;
246 alltypes.req_sfixed64.funcs.encode = &write_fixed64;
247 alltypes.req_sfixed64.arg = &req_sfixed64;
249 alltypes.req_double.funcs.encode = &write_fixed64;
250 alltypes.req_double.arg = &req_double;
252 alltypes.req_string.funcs.encode = &write_string;
253 alltypes.req_string.arg = "1014";
255 alltypes.req_bytes.funcs.encode = &write_string;
256 alltypes.req_bytes.arg = "1015";
258 alltypes.req_submsg.funcs.encode = &write_submsg;
259 alltypes.req_submsg.arg = &req_submsg;
261 alltypes.req_enum.funcs.encode = &write_varint;
262 alltypes.req_enum.arg = (void*)MyEnum_Truth;
264 alltypes.req_emptymsg.funcs.encode = &write_emptymsg;
266 alltypes.req_fbytes.funcs.encode = &write_string;
267 alltypes.req_fbytes.arg = "1019";
269 /* Bind callbacks for repeated fields */
270 alltypes.rep_int32.funcs.encode = &write_repeated_varint;
271 alltypes.rep_int32.arg = (void*)-2001;
273 alltypes.rep_int64.funcs.encode = &write_repeated_varint;
274 alltypes.rep_int64.arg = (void*)-2002;
276 alltypes.rep_uint32.funcs.encode = &write_repeated_varint;
277 alltypes.rep_uint32.arg = (void*)2003;
279 alltypes.rep_uint64.funcs.encode = &write_repeated_varint;
280 alltypes.rep_uint64.arg = (void*)2004;
282 alltypes.rep_sint32.funcs.encode = &write_repeated_svarint;
283 alltypes.rep_sint32.arg = (void*)-2005;
285 alltypes.rep_sint64.funcs.encode = &write_repeated_svarint;
286 alltypes.rep_sint64.arg = (void*)-2006;
288 alltypes.rep_bool.funcs.encode = &write_repeated_varint;
289 alltypes.rep_bool.arg = (void*)true;
291 alltypes.rep_fixed32.funcs.encode = &write_repeated_fixed32;
292 alltypes.rep_fixed32.arg = &rep_fixed32;
294 alltypes.rep_sfixed32.funcs.encode = &write_repeated_fixed32;
295 alltypes.rep_sfixed32.arg = &rep_sfixed32;
297 alltypes.rep_float.funcs.encode = &write_repeated_fixed32;
298 alltypes.rep_float.arg = &rep_float;
300 alltypes.rep_fixed64.funcs.encode = &write_repeated_fixed64;
301 alltypes.rep_fixed64.arg = &rep_fixed64;
303 alltypes.rep_sfixed64.funcs.encode = &write_repeated_fixed64;
304 alltypes.rep_sfixed64.arg = &rep_sfixed64;
306 alltypes.rep_double.funcs.encode = &write_repeated_fixed64;
307 alltypes.rep_double.arg = &rep_double;
309 alltypes.rep_string.funcs.encode = &write_repeated_string;
310 alltypes.rep_string.arg = "2014";
312 alltypes.rep_bytes.funcs.encode = &write_repeated_string;
313 alltypes.rep_bytes.arg = "2015";
315 alltypes.rep_submsg.funcs.encode = &write_repeated_submsg;
316 alltypes.rep_submsg.arg = &rep_submsg;
318 alltypes.rep_enum.funcs.encode = &write_repeated_varint;
319 alltypes.rep_enum.arg = (void*)MyEnum_Truth;
321 alltypes.rep_emptymsg.funcs.encode = &write_repeated_emptymsg;
323 alltypes.rep_fbytes.funcs.encode = &write_repeated_string;
324 alltypes.rep_fbytes.arg = "2019";
326 alltypes.req_limits.funcs.encode = &write_limits;
328 /* Bind callbacks for optional fields */
331 alltypes.opt_int32.funcs.encode = &write_varint;
332 alltypes.opt_int32.arg = (void*)3041;
334 alltypes.opt_int64.funcs.encode = &write_varint;
335 alltypes.opt_int64.arg = (void*)3042;
337 alltypes.opt_uint32.funcs.encode = &write_varint;
338 alltypes.opt_uint32.arg = (void*)3043;
340 alltypes.opt_uint64.funcs.encode = &write_varint;
341 alltypes.opt_uint64.arg = (void*)3044;
343 alltypes.opt_sint32.funcs.encode = &write_svarint;
344 alltypes.opt_sint32.arg = (void*)3045;
346 alltypes.opt_sint64.funcs.encode = &write_svarint;
347 alltypes.opt_sint64.arg = (void*)3046;
349 alltypes.opt_bool.funcs.encode = &write_varint;
350 alltypes.opt_bool.arg = (void*)true;
352 alltypes.opt_fixed32.funcs.encode = &write_fixed32;
353 alltypes.opt_fixed32.arg = &opt_fixed32;
355 alltypes.opt_sfixed32.funcs.encode = &write_fixed32;
356 alltypes.opt_sfixed32.arg = &opt_sfixed32;
358 alltypes.opt_float.funcs.encode = &write_fixed32;
359 alltypes.opt_float.arg = &opt_float;
361 alltypes.opt_fixed64.funcs.encode = &write_fixed64;
362 alltypes.opt_fixed64.arg = &opt_fixed64;
364 alltypes.opt_sfixed64.funcs.encode = &write_fixed64;
365 alltypes.opt_sfixed64.arg = &opt_sfixed64;
367 alltypes.opt_double.funcs.encode = &write_fixed64;
368 alltypes.opt_double.arg = &opt_double;
370 alltypes.opt_string.funcs.encode = &write_string;
371 alltypes.opt_string.arg = "3054";
373 alltypes.opt_bytes.funcs.encode = &write_string;
374 alltypes.opt_bytes.arg = "3055";
376 alltypes.opt_submsg.funcs.encode = &write_submsg;
377 alltypes.opt_submsg.arg = &opt_submsg;
379 alltypes.opt_enum.funcs.encode = &write_varint;
380 alltypes.opt_enum.arg = (void*)MyEnum_Truth;
382 alltypes.opt_emptymsg.funcs.encode = &write_emptymsg;
384 alltypes.opt_fbytes.funcs.encode = &write_string;
385 alltypes.opt_fbytes.arg = "3059";
387 alltypes.oneof_msg1.funcs.encode = &write_submsg;
388 alltypes.oneof_msg1.arg = &oneof_msg1;
391 alltypes.end.funcs.encode = &write_varint;
392 alltypes.end.arg = (void*)1099;
395 uint8_t buffer[2048];
396 pb_ostream_t stream = pb_ostream_from_buffer(buffer, sizeof(buffer));
398 /* Now encode it and check if we succeeded. */
399 if (pb_encode(&stream, AllTypes_fields, &alltypes))
401 SET_BINARY_MODE(stdout);
402 fwrite(buffer, 1, stream.bytes_written, stdout);
403 return 0; /* Success */
407 fprintf(stderr, "Encoding failed: %s\n", PB_GET_ERROR(&stream));
408 return 1; /* Failure */