17ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen/* Attempts to test all the datatypes supported by ProtoBuf. 27ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen */ 37ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen 47ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen#include <stdio.h> 57ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen#include <stdlib.h> 67ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen#include <string.h> 77ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen#include <pb_encode.h> 87ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen#include "alltypes.pb.h" 97ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen#include "test_helpers.h" 107ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen 117ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohenint main(int argc, char **argv) 127ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen{ 137ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen int mode = (argc > 1) ? atoi(argv[1]) : 0; 147ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen 157ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen /* Values for required fields */ 167ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen int32_t req_int32 = -1001; 177ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen int64_t req_int64 = -1002; 187ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen uint32_t req_uint32 = 1003; 197ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen uint64_t req_uint64 = 1004; 207ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen int32_t req_sint32 = -1005; 217ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen int64_t req_sint64 = -1006; 227ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen bool req_bool = true; 237ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen uint32_t req_fixed32 = 1008; 247ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen int32_t req_sfixed32 = -1009; 257ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen float req_float = 1010.0f; 267ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen uint64_t req_fixed64 = 1011; 277ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen int64_t req_sfixed64 = -1012; 287ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen double req_double = 1013.0; 297ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen char* req_string = "1014"; 307ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen PB_BYTES_ARRAY_T(4) req_bytes = {4, {'1', '0', '1', '5'}}; 317ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen static int32_t req_substuff = 1016; 327ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen SubMessage req_submsg = {"1016", &req_substuff}; 337ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen MyEnum req_enum = MyEnum_Truth; 347ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen EmptyMessage req_emptymsg = {0}; 357ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen 367ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen int32_t end = 1099; 377ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen 387ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen /* Values for repeated fields */ 397ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen int32_t rep_int32[5] = {0, 0, 0, 0, -2001}; 407ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen int64_t rep_int64[5] = {0, 0, 0, 0, -2002}; 417ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen uint32_t rep_uint32[5] = {0, 0, 0, 0, 2003}; 427ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen uint64_t rep_uint64[5] = {0, 0, 0, 0, 2004}; 437ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen int32_t rep_sint32[5] = {0, 0, 0, 0, -2005}; 447ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen int64_t rep_sint64[5] = {0, 0, 0, 0, -2006}; 457ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen bool rep_bool[5] = {false, false, false, false, true}; 467ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen uint32_t rep_fixed32[5] = {0, 0, 0, 0, 2008}; 477ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen int32_t rep_sfixed32[5] = {0, 0, 0, 0, -2009}; 487ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen float rep_float[5] = {0, 0, 0, 0, 2010.0f}; 497ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen uint64_t rep_fixed64[5] = {0, 0, 0, 0, 2011}; 507ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen int64_t rep_sfixed64[5] = {0, 0, 0, 0, -2012}; 517ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen double rep_double[5] = {0, 0, 0, 0, 2013.0f}; 527ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen char* rep_string[5] = {"", "", "", "", "2014"}; 537ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen static PB_BYTES_ARRAY_T(4) rep_bytes_4 = {4, {'2', '0', '1', '5'}}; 547ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen pb_bytes_array_t *rep_bytes[5]= {NULL, NULL, NULL, NULL, (pb_bytes_array_t*)&rep_bytes_4}; 557ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen static int32_t rep_sub2zero = 0; 567ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen static int32_t rep_substuff2 = 2016; 577ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen static uint32_t rep_substuff3 = 2016; 587ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen SubMessage rep_submsg[5] = {{"", &rep_sub2zero}, 597ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen {"", &rep_sub2zero}, 607ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen {"", &rep_sub2zero}, 617ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen {"", &rep_sub2zero}, 627ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen {"2016", &rep_substuff2, &rep_substuff3}}; 637ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen MyEnum rep_enum[5] = {0, 0, 0, 0, MyEnum_Truth}; 647ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen EmptyMessage rep_emptymsg[5] = {{0}, {0}, {0}, {0}, {0}}; 657ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen 667ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen /* Values for optional fields */ 677ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen int32_t opt_int32 = 3041; 687ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen int64_t opt_int64 = 3042; 697ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen uint32_t opt_uint32 = 3043; 707ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen uint64_t opt_uint64 = 3044; 717ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen int32_t opt_sint32 = 3045; 727ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen int64_t opt_sint64 = 3046; 737ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen bool opt_bool = true; 747ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen uint32_t opt_fixed32 = 3048; 757ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen int32_t opt_sfixed32 = 3049; 767ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen float opt_float = 3050.0f; 777ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen uint64_t opt_fixed64 = 3051; 787ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen int64_t opt_sfixed64 = 3052; 797ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen double opt_double = 3053.0; 807ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen char* opt_string = "3054"; 817ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen PB_BYTES_ARRAY_T(4) opt_bytes = {4, {'3', '0', '5', '5'}}; 827ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen static int32_t opt_substuff = 3056; 837ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen SubMessage opt_submsg = {"3056", &opt_substuff}; 847ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen MyEnum opt_enum = MyEnum_Truth; 857ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen EmptyMessage opt_emptymsg = {0}; 867ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen 877ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen /* Values for the Limits message. */ 887ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen static int32_t int32_min = INT32_MIN; 897ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen static int32_t int32_max = INT32_MAX; 907ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen static uint32_t uint32_min = 0; 917ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen static uint32_t uint32_max = UINT32_MAX; 927ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen static int64_t int64_min = INT64_MIN; 937ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen static int64_t int64_max = INT64_MAX; 947ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen static uint64_t uint64_min = 0; 957ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen static uint64_t uint64_max = UINT64_MAX; 967ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen static HugeEnum enum_min = HugeEnum_Negative; 977ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen static HugeEnum enum_max = HugeEnum_Positive; 987ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen Limits req_limits = {&int32_min, &int32_max, 997ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen &uint32_min, &uint32_max, 1007ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen &int64_min, &int64_max, 1017ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen &uint64_min, &uint64_max, 1027ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen &enum_min, &enum_max}; 1037ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen 1047ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen /* Initialize the message struct with pointers to the fields. */ 1057ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen AllTypes alltypes = {0}; 1067ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen 1077ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.req_int32 = &req_int32; 1087ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.req_int64 = &req_int64; 1097ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.req_uint32 = &req_uint32; 1107ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.req_uint64 = &req_uint64; 1117ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.req_sint32 = &req_sint32; 1127ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.req_sint64 = &req_sint64; 1137ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.req_bool = &req_bool; 1147ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.req_fixed32 = &req_fixed32; 1157ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.req_sfixed32 = &req_sfixed32; 1167ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.req_float = &req_float; 1177ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.req_fixed64 = &req_fixed64; 1187ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.req_sfixed64 = &req_sfixed64; 1197ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.req_double = &req_double; 1207ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.req_string = req_string; 1217ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.req_bytes = (pb_bytes_array_t*)&req_bytes; 1227ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.req_submsg = &req_submsg; 1237ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.req_enum = &req_enum; 1247ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.req_emptymsg = &req_emptymsg; 1257ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.req_limits = &req_limits; 1267ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen 1277ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.rep_int32_count = 5; alltypes.rep_int32 = rep_int32; 1287ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.rep_int64_count = 5; alltypes.rep_int64 = rep_int64; 1297ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.rep_uint32_count = 5; alltypes.rep_uint32 = rep_uint32; 1307ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.rep_uint64_count = 5; alltypes.rep_uint64 = rep_uint64; 1317ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.rep_sint32_count = 5; alltypes.rep_sint32 = rep_sint32; 1327ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.rep_sint64_count = 5; alltypes.rep_sint64 = rep_sint64; 1337ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.rep_bool_count = 5; alltypes.rep_bool = rep_bool; 1347ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.rep_fixed32_count = 5; alltypes.rep_fixed32 = rep_fixed32; 1357ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.rep_sfixed32_count = 5; alltypes.rep_sfixed32 = rep_sfixed32; 1367ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.rep_float_count = 5; alltypes.rep_float = rep_float; 1377ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.rep_fixed64_count = 5; alltypes.rep_fixed64 = rep_fixed64; 1387ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.rep_sfixed64_count = 5; alltypes.rep_sfixed64 = rep_sfixed64; 1397ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.rep_double_count = 5; alltypes.rep_double = rep_double; 1407ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.rep_string_count = 5; alltypes.rep_string = rep_string; 1417ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.rep_bytes_count = 5; alltypes.rep_bytes = rep_bytes; 1427ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.rep_submsg_count = 5; alltypes.rep_submsg = rep_submsg; 1437ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.rep_enum_count = 5; alltypes.rep_enum = rep_enum; 1447ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.rep_emptymsg_count = 5; alltypes.rep_emptymsg = rep_emptymsg; 1457ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen 1467ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen if (mode != 0) 1477ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen { 1487ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen /* Fill in values for optional fields */ 1497ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.opt_int32 = &opt_int32; 1507ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.opt_int64 = &opt_int64; 1517ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.opt_uint32 = &opt_uint32; 1527ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.opt_uint64 = &opt_uint64; 1537ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.opt_sint32 = &opt_sint32; 1547ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.opt_sint64 = &opt_sint64; 1557ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.opt_bool = &opt_bool; 1567ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.opt_fixed32 = &opt_fixed32; 1577ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.opt_sfixed32 = &opt_sfixed32; 1587ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.opt_float = &opt_float; 1597ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.opt_fixed64 = &opt_fixed64; 1607ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.opt_sfixed64 = &opt_sfixed64; 1617ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.opt_double = &opt_double; 1627ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.opt_string = opt_string; 1637ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.opt_bytes = (pb_bytes_array_t*)&opt_bytes; 1647ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.opt_submsg = &opt_submsg; 1657ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.opt_enum = &opt_enum; 1667ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.opt_emptymsg = &opt_emptymsg; 1677ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen } 1687ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen 1697ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen alltypes.end = &end; 1707ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen 1717ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen { 1727ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen uint8_t buffer[4096]; 1737ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen pb_ostream_t stream = pb_ostream_from_buffer(buffer, sizeof(buffer)); 1747ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen 1757ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen /* Now encode it and check if we succeeded. */ 1767ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen if (pb_encode(&stream, AllTypes_fields, &alltypes)) 1777ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen { 1787ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen SET_BINARY_MODE(stdout); 1797ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen fwrite(buffer, 1, stream.bytes_written, stdout); 1807ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen return 0; /* Success */ 1817ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen } 1827ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen else 1837ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen { 1847ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen fprintf(stderr, "Encoding failed: %s\n", PB_GET_ERROR(&stream)); 1857ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen return 1; /* Failure */ 1867ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen } 1877ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen } 1887ef855e462b9a18b7d330e4b40f350164a6ad9daEtan Cohen} 189