| /* Attempts to test all the datatypes supported by ProtoBuf when used as callback fields. |
| * Note that normally there would be no reason to use callback fields for this, |
| * because each encoder defined here only gives a single field. |
| */ |
| |
| #include <stdio.h> |
| #include <stdlib.h> |
| #include <string.h> |
| #include <pb_encode.h> |
| #include "alltypes.pb.h" |
| #include "test_helpers.h" |
| |
| static bool write_varint(pb_ostream_t *stream, const pb_field_t *field, void * const *arg) |
| { |
| return pb_encode_tag_for_field(stream, field) && |
| pb_encode_varint(stream, (intptr_t)*arg); |
| } |
| |
| static bool write_svarint(pb_ostream_t *stream, const pb_field_t *field, void * const *arg) |
| { |
| return pb_encode_tag_for_field(stream, field) && |
| pb_encode_svarint(stream, (intptr_t)*arg); |
| } |
| |
| static bool write_fixed32(pb_ostream_t *stream, const pb_field_t *field, void * const *arg) |
| { |
| return pb_encode_tag_for_field(stream, field) && |
| pb_encode_fixed32(stream, *arg); |
| } |
| |
| static bool write_fixed64(pb_ostream_t *stream, const pb_field_t *field, void * const *arg) |
| { |
| return pb_encode_tag_for_field(stream, field) && |
| pb_encode_fixed64(stream, *arg); |
| } |
| |
| static bool write_double(pb_ostream_t *stream, const pb_field_t *field, void * const *arg) |
| { |
| #ifdef PB_CONVERT_DOUBLE_FLOAT |
| if (sizeof(double) == sizeof(float)) |
| return pb_encode_tag_for_field(stream, field) && |
| pb_encode_float_as_double(stream, *(float*)*arg); |
| #endif |
| |
| return pb_encode_tag_for_field(stream, field) && |
| pb_encode_fixed64(stream, *arg); |
| } |
| |
| static bool write_string(pb_ostream_t *stream, const pb_field_t *field, void * const *arg) |
| { |
| return pb_encode_tag_for_field(stream, field) && |
| pb_encode_string(stream, *arg, strlen(*arg)); |
| } |
| |
| static bool write_submsg(pb_ostream_t *stream, const pb_field_t *field, void * const *arg) |
| { |
| |
| return pb_encode_tag_for_field(stream, field) && |
| pb_encode_submessage(stream, SubMessage_fields, *arg); |
| } |
| |
| static bool write_emptymsg(pb_ostream_t *stream, const pb_field_t *field, void * const *arg) |
| { |
| EmptyMessage emptymsg = {0}; |
| return pb_encode_tag_for_field(stream, field) && |
| pb_encode_submessage(stream, EmptyMessage_fields, &emptymsg); |
| } |
| |
| static bool write_repeated_varint(pb_ostream_t *stream, const pb_field_t *field, void * const *arg) |
| { |
| return pb_encode_tag_for_field(stream, field) && |
| pb_encode_varint(stream, 0) && |
| pb_encode_tag_for_field(stream, field) && |
| pb_encode_varint(stream, 0) && |
| pb_encode_tag_for_field(stream, field) && |
| pb_encode_varint(stream, 0) && |
| pb_encode_tag_for_field(stream, field) && |
| pb_encode_varint(stream, 0) && |
| pb_encode_tag_for_field(stream, field) && |
| pb_encode_varint(stream, (intptr_t)*arg); |
| } |
| |
| static bool write_repeated_svarint(pb_ostream_t *stream, const pb_field_t *field, void * const *arg) |
| { |
| return pb_encode_tag_for_field(stream, field) && |
| pb_encode_svarint(stream, 0) && |
| pb_encode_tag_for_field(stream, field) && |
| pb_encode_svarint(stream, 0) && |
| pb_encode_tag_for_field(stream, field) && |
| pb_encode_svarint(stream, 0) && |
| pb_encode_tag_for_field(stream, field) && |
| pb_encode_svarint(stream, 0) && |
| pb_encode_tag_for_field(stream, field) && |
| pb_encode_svarint(stream, (intptr_t)*arg); |
| } |
| |
| static bool write_repeated_fixed32(pb_ostream_t *stream, const pb_field_t *field, void * const *arg) |
| { |
| uint32_t dummy = 0; |
| |
| /* Make it a packed field */ |
| return pb_encode_tag(stream, PB_WT_STRING, field->tag) && |
| pb_encode_varint(stream, 5 * 4) && /* Number of bytes */ |
| pb_encode_fixed32(stream, &dummy) && |
| pb_encode_fixed32(stream, &dummy) && |
| pb_encode_fixed32(stream, &dummy) && |
| pb_encode_fixed32(stream, &dummy) && |
| pb_encode_fixed32(stream, *arg); |
| } |
| |
| static bool write_repeated_fixed64(pb_ostream_t *stream, const pb_field_t *field, void * const *arg) |
| { |
| uint64_t dummy = 0; |
| |
| /* Make it a packed field */ |
| return pb_encode_tag(stream, PB_WT_STRING, field->tag) && |
| pb_encode_varint(stream, 5 * 8) && /* Number of bytes */ |
| pb_encode_fixed64(stream, &dummy) && |
| pb_encode_fixed64(stream, &dummy) && |
| pb_encode_fixed64(stream, &dummy) && |
| pb_encode_fixed64(stream, &dummy) && |
| pb_encode_fixed64(stream, *arg); |
| } |
| |
| static bool write_repeated_double(pb_ostream_t *stream, const pb_field_t *field, void * const *arg) |
| { |
| uint64_t dummy = 0; |
| |
| #ifdef PB_CONVERT_DOUBLE_FLOAT |
| if (sizeof(double) == sizeof(float)) |
| return pb_encode_tag(stream, PB_WT_STRING, field->tag) && |
| pb_encode_varint(stream, 5 * 8) && /* Number of bytes */ |
| pb_encode_float_as_double(stream, 0.0f) && |
| pb_encode_float_as_double(stream, 0.0f) && |
| pb_encode_float_as_double(stream, 0.0f) && |
| pb_encode_float_as_double(stream, 0.0f) && |
| pb_encode_float_as_double(stream, *(float*)*arg); |
| #endif |
| |
| /* Make it a packed field */ |
| return pb_encode_tag(stream, PB_WT_STRING, field->tag) && |
| pb_encode_varint(stream, 5 * 8) && /* Number of bytes */ |
| pb_encode_fixed64(stream, &dummy) && |
| pb_encode_fixed64(stream, &dummy) && |
| pb_encode_fixed64(stream, &dummy) && |
| pb_encode_fixed64(stream, &dummy) && |
| pb_encode_fixed64(stream, *arg); |
| } |
| |
| static bool write_repeated_string(pb_ostream_t *stream, const pb_field_t *field, void * const *arg) |
| { |
| return pb_encode_tag_for_field(stream, field) && |
| pb_encode_string(stream, 0, 0) && |
| pb_encode_tag_for_field(stream, field) && |
| pb_encode_string(stream, 0, 0) && |
| pb_encode_tag_for_field(stream, field) && |
| pb_encode_string(stream, 0, 0) && |
| pb_encode_tag_for_field(stream, field) && |
| pb_encode_string(stream, 0, 0) && |
| pb_encode_tag_for_field(stream, field) && |
| pb_encode_string(stream, *arg, strlen(*arg)); |
| } |
| |
| static bool write_repeated_submsg(pb_ostream_t *stream, const pb_field_t *field, void * const *arg) |
| { |
| SubMessage dummy = {""}; |
| |
| return pb_encode_tag_for_field(stream, field) && |
| pb_encode_submessage(stream, SubMessage_fields, &dummy) && |
| pb_encode_tag_for_field(stream, field) && |
| pb_encode_submessage(stream, SubMessage_fields, &dummy) && |
| pb_encode_tag_for_field(stream, field) && |
| pb_encode_submessage(stream, SubMessage_fields, &dummy) && |
| pb_encode_tag_for_field(stream, field) && |
| pb_encode_submessage(stream, SubMessage_fields, &dummy) && |
| pb_encode_tag_for_field(stream, field) && |
| pb_encode_submessage(stream, SubMessage_fields, *arg); |
| } |
| |
| static bool write_limits(pb_ostream_t *stream, const pb_field_t *field, void * const *arg) |
| { |
| Limits limits = {0}; |
| limits.int32_min = INT32_MIN; |
| limits.int32_max = INT32_MAX; |
| limits.uint32_min = 0; |
| limits.uint32_max = UINT32_MAX; |
| limits.int64_min = INT64_MIN; |
| limits.int64_max = INT64_MAX; |
| limits.uint64_min = 0; |
| limits.uint64_max = UINT64_MAX; |
| limits.enum_min = HugeEnum_Negative; |
| limits.enum_max = HugeEnum_Positive; |
| limits.largetag = 1001; |
| |
| return pb_encode_tag_for_field(stream, field) && |
| pb_encode_submessage(stream, Limits_fields, &limits); |
| } |
| |
| static bool write_ds8(pb_ostream_t *stream, const pb_field_t *field, void * const *arg) |
| { |
| DescriptorSize8 ds8 = {9991,9992}; |
| |
| return pb_encode_tag_for_field(stream, field) && |
| pb_encode_submessage(stream, DescriptorSize8_fields, &ds8); |
| } |
| |
| static bool write_intsizes(pb_ostream_t *stream, const pb_field_t *field, void * const *arg) |
| { |
| IntSizes intsizes = {-128, 255, -128, -32768, 65535, -32768}; |
| |
| return pb_encode_tag_for_field(stream, field) && |
| pb_encode_submessage(stream, IntSizes_fields, &intsizes); |
| } |
| |
| static bool write_repeated_emptymsg(pb_ostream_t *stream, const pb_field_t *field, void * const *arg) |
| { |
| EmptyMessage emptymsg = {0}; |
| return pb_encode_tag_for_field(stream, field) && |
| pb_encode_submessage(stream, EmptyMessage_fields, &emptymsg) && |
| pb_encode_tag_for_field(stream, field) && |
| pb_encode_submessage(stream, EmptyMessage_fields, &emptymsg) && |
| pb_encode_tag_for_field(stream, field) && |
| pb_encode_submessage(stream, EmptyMessage_fields, &emptymsg) && |
| pb_encode_tag_for_field(stream, field) && |
| pb_encode_submessage(stream, EmptyMessage_fields, &emptymsg) && |
| pb_encode_tag_for_field(stream, field) && |
| pb_encode_submessage(stream, EmptyMessage_fields, &emptymsg); |
| } |
| |
| static bool write_farray2(pb_ostream_t *stream, const pb_field_t *field, void * const *arg) |
| { |
| uint32_t dummy = 0; |
| uint32_t value = (uint32_t)(intptr_t)*arg; |
| |
| /* Make it a packed field */ |
| return pb_encode_tag(stream, PB_WT_STRING, field->tag) && |
| pb_encode_varint(stream, 3 * 4) && /* Number of bytes */ |
| pb_encode_fixed32(stream, &dummy) && |
| pb_encode_fixed32(stream, &dummy) && |
| pb_encode_fixed32(stream, &value); |
| } |
| |
| int main(int argc, char **argv) |
| { |
| int mode = (argc > 1) ? atoi(argv[1]) : 0; |
| |
| /* Values for use from callbacks through pointers. */ |
| uint32_t req_fixed32 = 1008; |
| int32_t req_sfixed32 = -1009; |
| float req_float = 1010.0f; |
| uint64_t req_fixed64 = 1011; |
| int64_t req_sfixed64 = -1012; |
| double req_double = 1013.0; |
| SubMessage req_submsg = {"1016", 1016}; |
| |
| uint32_t rep_fixed32 = 2008; |
| int32_t rep_sfixed32 = -2009; |
| float rep_float = 2010.0f; |
| uint64_t rep_fixed64 = 2011; |
| int64_t rep_sfixed64 = -2012; |
| double rep_double = 2013.0; |
| SubMessage rep_submsg = {"2016", 2016, true, 2016}; |
| |
| uint32_t opt_fixed32 = 3048; |
| int32_t opt_sfixed32 = 3049; |
| float opt_float = 3050.0f; |
| uint64_t opt_fixed64 = 3051; |
| int64_t opt_sfixed64 = 3052; |
| double opt_double = 3053.0f; |
| SubMessage opt_submsg = {"3056", 3056}; |
| |
| SubMessage oneof_msg1 = {"4059", 4059}; |
| |
| /* Bind callbacks for required fields */ |
| AllTypes alltypes = {{{0}}}; |
| |
| alltypes.req_int32.funcs.encode = &write_varint; |
| alltypes.req_int32.arg = (void*)-1001; |
| |
| alltypes.req_int64.funcs.encode = &write_varint; |
| alltypes.req_int64.arg = (void*)-1002; |
| |
| alltypes.req_uint32.funcs.encode = &write_varint; |
| alltypes.req_uint32.arg = (void*)1003; |
| |
| alltypes.req_uint32.funcs.encode = &write_varint; |
| alltypes.req_uint32.arg = (void*)1003; |
| |
| alltypes.req_uint64.funcs.encode = &write_varint; |
| alltypes.req_uint64.arg = (void*)1004; |
| |
| alltypes.req_sint32.funcs.encode = &write_svarint; |
| alltypes.req_sint32.arg = (void*)-1005; |
| |
| alltypes.req_sint64.funcs.encode = &write_svarint; |
| alltypes.req_sint64.arg = (void*)-1006; |
| |
| alltypes.req_bool.funcs.encode = &write_varint; |
| alltypes.req_bool.arg = (void*)true; |
| |
| alltypes.req_fixed32.funcs.encode = &write_fixed32; |
| alltypes.req_fixed32.arg = &req_fixed32; |
| |
| alltypes.req_sfixed32.funcs.encode = &write_fixed32; |
| alltypes.req_sfixed32.arg = &req_sfixed32; |
| |
| alltypes.req_float.funcs.encode = &write_fixed32; |
| alltypes.req_float.arg = &req_float; |
| |
| alltypes.req_fixed64.funcs.encode = &write_fixed64; |
| alltypes.req_fixed64.arg = &req_fixed64; |
| |
| alltypes.req_sfixed64.funcs.encode = &write_fixed64; |
| alltypes.req_sfixed64.arg = &req_sfixed64; |
| |
| alltypes.req_double.funcs.encode = &write_double; |
| alltypes.req_double.arg = &req_double; |
| |
| alltypes.req_string.funcs.encode = &write_string; |
| alltypes.req_string.arg = "1014"; |
| |
| alltypes.req_bytes.funcs.encode = &write_string; |
| alltypes.req_bytes.arg = "1015"; |
| |
| alltypes.req_submsg.funcs.encode = &write_submsg; |
| alltypes.req_submsg.arg = &req_submsg; |
| |
| alltypes.req_enum.funcs.encode = &write_varint; |
| alltypes.req_enum.arg = (void*)MyEnum_Truth; |
| |
| alltypes.req_emptymsg.funcs.encode = &write_emptymsg; |
| |
| alltypes.req_fbytes.funcs.encode = &write_string; |
| alltypes.req_fbytes.arg = "1019"; |
| |
| /* Bind callbacks for repeated fields */ |
| alltypes.rep_int32.funcs.encode = &write_repeated_varint; |
| alltypes.rep_int32.arg = (void*)-2001; |
| |
| alltypes.rep_int64.funcs.encode = &write_repeated_varint; |
| alltypes.rep_int64.arg = (void*)-2002; |
| |
| alltypes.rep_uint32.funcs.encode = &write_repeated_varint; |
| alltypes.rep_uint32.arg = (void*)2003; |
| |
| alltypes.rep_uint64.funcs.encode = &write_repeated_varint; |
| alltypes.rep_uint64.arg = (void*)2004; |
| |
| alltypes.rep_sint32.funcs.encode = &write_repeated_svarint; |
| alltypes.rep_sint32.arg = (void*)-2005; |
| |
| alltypes.rep_sint64.funcs.encode = &write_repeated_svarint; |
| alltypes.rep_sint64.arg = (void*)-2006; |
| |
| alltypes.rep_bool.funcs.encode = &write_repeated_varint; |
| alltypes.rep_bool.arg = (void*)true; |
| |
| alltypes.rep_fixed32.funcs.encode = &write_repeated_fixed32; |
| alltypes.rep_fixed32.arg = &rep_fixed32; |
| |
| alltypes.rep_sfixed32.funcs.encode = &write_repeated_fixed32; |
| alltypes.rep_sfixed32.arg = &rep_sfixed32; |
| |
| alltypes.rep_float.funcs.encode = &write_repeated_fixed32; |
| alltypes.rep_float.arg = &rep_float; |
| |
| alltypes.rep_fixed64.funcs.encode = &write_repeated_fixed64; |
| alltypes.rep_fixed64.arg = &rep_fixed64; |
| |
| alltypes.rep_sfixed64.funcs.encode = &write_repeated_fixed64; |
| alltypes.rep_sfixed64.arg = &rep_sfixed64; |
| |
| alltypes.rep_double.funcs.encode = &write_repeated_double; |
| alltypes.rep_double.arg = &rep_double; |
| |
| alltypes.rep_string.funcs.encode = &write_repeated_string; |
| alltypes.rep_string.arg = "2014"; |
| |
| alltypes.rep_bytes.funcs.encode = &write_repeated_string; |
| alltypes.rep_bytes.arg = "2015"; |
| |
| alltypes.rep_submsg.funcs.encode = &write_repeated_submsg; |
| alltypes.rep_submsg.arg = &rep_submsg; |
| |
| alltypes.rep_enum.funcs.encode = &write_repeated_varint; |
| alltypes.rep_enum.arg = (void*)MyEnum_Truth; |
| |
| alltypes.rep_emptymsg.funcs.encode = &write_repeated_emptymsg; |
| |
| alltypes.rep_fbytes.funcs.encode = &write_repeated_string; |
| alltypes.rep_fbytes.arg = "2019"; |
| |
| alltypes.rep_farray.funcs.encode = &write_repeated_varint; |
| alltypes.rep_farray.arg = (void*)2040; |
| |
| alltypes.rep_farray2.funcs.encode = &write_farray2; |
| alltypes.rep_farray2.arg = (void*)2095; |
| |
| alltypes.req_limits.funcs.encode = &write_limits; |
| |
| alltypes.req_ds8.funcs.encode = &write_ds8; |
| |
| alltypes.req_intsizes.funcs.encode = &write_intsizes; |
| |
| /* Bind callbacks for optional fields */ |
| if (mode != 0) |
| { |
| alltypes.opt_int32.funcs.encode = &write_varint; |
| alltypes.opt_int32.arg = (void*)3041; |
| |
| alltypes.opt_int64.funcs.encode = &write_varint; |
| alltypes.opt_int64.arg = (void*)3042; |
| |
| alltypes.opt_uint32.funcs.encode = &write_varint; |
| alltypes.opt_uint32.arg = (void*)3043; |
| |
| alltypes.opt_uint64.funcs.encode = &write_varint; |
| alltypes.opt_uint64.arg = (void*)3044; |
| |
| alltypes.opt_sint32.funcs.encode = &write_svarint; |
| alltypes.opt_sint32.arg = (void*)3045; |
| |
| alltypes.opt_sint64.funcs.encode = &write_svarint; |
| alltypes.opt_sint64.arg = (void*)3046; |
| |
| alltypes.opt_bool.funcs.encode = &write_varint; |
| alltypes.opt_bool.arg = (void*)true; |
| |
| alltypes.opt_fixed32.funcs.encode = &write_fixed32; |
| alltypes.opt_fixed32.arg = &opt_fixed32; |
| |
| alltypes.opt_sfixed32.funcs.encode = &write_fixed32; |
| alltypes.opt_sfixed32.arg = &opt_sfixed32; |
| |
| alltypes.opt_float.funcs.encode = &write_fixed32; |
| alltypes.opt_float.arg = &opt_float; |
| |
| alltypes.opt_fixed64.funcs.encode = &write_fixed64; |
| alltypes.opt_fixed64.arg = &opt_fixed64; |
| |
| alltypes.opt_sfixed64.funcs.encode = &write_fixed64; |
| alltypes.opt_sfixed64.arg = &opt_sfixed64; |
| |
| alltypes.opt_double.funcs.encode = &write_double; |
| alltypes.opt_double.arg = &opt_double; |
| |
| alltypes.opt_string.funcs.encode = &write_string; |
| alltypes.opt_string.arg = "3054"; |
| |
| alltypes.opt_bytes.funcs.encode = &write_string; |
| alltypes.opt_bytes.arg = "3055"; |
| |
| alltypes.opt_submsg.funcs.encode = &write_submsg; |
| alltypes.opt_submsg.arg = &opt_submsg; |
| |
| alltypes.opt_enum.funcs.encode = &write_varint; |
| alltypes.opt_enum.arg = (void*)MyEnum_Truth; |
| |
| alltypes.opt_emptymsg.funcs.encode = &write_emptymsg; |
| |
| alltypes.opt_fbytes.funcs.encode = &write_string; |
| alltypes.opt_fbytes.arg = "3059"; |
| |
| alltypes.oneof_msg1.funcs.encode = &write_submsg; |
| alltypes.oneof_msg1.arg = &oneof_msg1; |
| |
| alltypes.opt_non_zero_based_enum.funcs.encode = &write_varint; |
| alltypes.opt_non_zero_based_enum.arg = (void *)NonZeroBasedEnum_Three; |
| } |
| |
| alltypes.end.funcs.encode = &write_varint; |
| alltypes.end.arg = (void*)1099; |
| |
| { |
| uint8_t buffer[2048]; |
| pb_ostream_t stream = pb_ostream_from_buffer(buffer, sizeof(buffer)); |
| |
| /* Now encode it and check if we succeeded. */ |
| if (pb_encode(&stream, AllTypes_fields, &alltypes)) |
| { |
| SET_BINARY_MODE(stdout); |
| fwrite(buffer, 1, stream.bytes_written, stdout); |
| return 0; /* Success */ |
| } |
| else |
| { |
| fprintf(stderr, "Encoding failed: %s\n", PB_GET_ERROR(&stream)); |
| return 1; /* Failure */ |
| } |
| } |
| } |
