encode_alltypes.c 4.3 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124
  1. /* Attempts to test all the datatypes supported by ProtoBuf.
  2. */
  3. #include <stdio.h>
  4. #include <stdlib.h>
  5. #include <string.h>
  6. #include <pb_encode.h>
  7. #include "alltypes.pb.h"
  8. #include "test_helpers.h"
  9. int main(int argc, char **argv)
  10. {
  11. int mode = (argc > 1) ? atoi(argv[1]) : 0;
  12. /* Initialize the structure with constants */
  13. AllTypes alltypes = AllTypes_init_zero;
  14. alltypes.req_int32 = -1001;
  15. alltypes.req_uint32 = 1003;
  16. alltypes.req_sint32 = -1005;
  17. alltypes.req_bool = true;
  18. alltypes.req_fixed32 = 1008;
  19. alltypes.req_sfixed32 = -1009;
  20. alltypes.req_float = 1010.0f;
  21. strcpy(alltypes.req_string, "1014");
  22. alltypes.req_bytes.size = 4;
  23. memcpy(alltypes.req_bytes.bytes, "1015", 4);
  24. strcpy(alltypes.req_submsg.substuff1, "1016");
  25. alltypes.req_submsg.substuff2 = 1016;
  26. alltypes.req_enum = MyEnum_Truth;
  27. memcpy(alltypes.req_fbytes, "1019", 4);
  28. alltypes.rep_int32_count = 5; alltypes.rep_int32[4] = -2001;
  29. alltypes.rep_uint32_count = 5; alltypes.rep_uint32[4] = 2003;
  30. alltypes.rep_sint32_count = 5; alltypes.rep_sint32[4] = -2005;
  31. alltypes.rep_bool_count = 5; alltypes.rep_bool[4] = true;
  32. alltypes.rep_fixed32_count = 5; alltypes.rep_fixed32[4] = 2008;
  33. alltypes.rep_sfixed32_count = 5; alltypes.rep_sfixed32[4] = -2009;
  34. alltypes.rep_float_count = 5; alltypes.rep_float[4] = 2010.0f;
  35. alltypes.rep_string_count = 5; strcpy(alltypes.rep_string[4], "2014");
  36. alltypes.rep_bytes_count = 5; alltypes.rep_bytes[4].size = 4;
  37. memcpy(alltypes.rep_bytes[4].bytes, "2015", 4);
  38. alltypes.rep_submsg_count = 5;
  39. strcpy(alltypes.rep_submsg[4].substuff1, "2016");
  40. alltypes.rep_submsg[4].substuff2 = 2016;
  41. alltypes.rep_submsg[4].has_substuff3 = true;
  42. alltypes.rep_submsg[4].substuff3 = 2016;
  43. alltypes.rep_enum_count = 5; alltypes.rep_enum[4] = MyEnum_Truth;
  44. alltypes.rep_emptymsg_count = 5;
  45. alltypes.rep_fbytes_count = 5;
  46. memcpy(alltypes.rep_fbytes[4], "2019", 4);
  47. alltypes.req_limits.int32_min = INT32_MIN;
  48. alltypes.req_limits.int32_max = INT32_MAX;
  49. alltypes.req_limits.uint32_min = 0;
  50. alltypes.req_limits.uint32_max = UINT32_MAX;
  51. alltypes.req_limits.enum_min = HugeEnum_Negative;
  52. alltypes.req_limits.enum_max = HugeEnum_Positive;
  53. if (mode != 0)
  54. {
  55. /* Fill in values for optional fields */
  56. alltypes.has_opt_int32 = true;
  57. alltypes.opt_int32 = 3041;
  58. alltypes.has_opt_uint32 = true;
  59. alltypes.opt_uint32 = 3043;
  60. alltypes.has_opt_sint32 = true;
  61. alltypes.opt_sint32 = 3045;
  62. alltypes.has_opt_bool = true;
  63. alltypes.opt_bool = true;
  64. alltypes.has_opt_fixed32 = true;
  65. alltypes.opt_fixed32 = 3048;
  66. alltypes.has_opt_sfixed32 = true;
  67. alltypes.opt_sfixed32 = 3049;
  68. alltypes.has_opt_float = true;
  69. alltypes.opt_float = 3050.0f;
  70. alltypes.has_opt_string = true;
  71. strcpy(alltypes.opt_string, "3054");
  72. alltypes.has_opt_bytes = true;
  73. alltypes.opt_bytes.size = 4;
  74. memcpy(alltypes.opt_bytes.bytes, "3055", 4);
  75. alltypes.has_opt_submsg = true;
  76. strcpy(alltypes.opt_submsg.substuff1, "3056");
  77. alltypes.opt_submsg.substuff2 = 3056;
  78. alltypes.has_opt_enum = true;
  79. alltypes.opt_enum = MyEnum_Truth;
  80. alltypes.has_opt_emptymsg = true;
  81. alltypes.has_opt_fbytes = true;
  82. memcpy(alltypes.opt_fbytes, "3059", 4);
  83. alltypes.which_oneof = AllTypes_oneof_msg1_tag;
  84. strcpy(alltypes.oneof.oneof_msg1.substuff1, "4059");
  85. alltypes.oneof.oneof_msg1.substuff2 = 4059;
  86. }
  87. alltypes.end = 1099;
  88. {
  89. uint8_t buffer[AllTypes_size];
  90. pb_ostream_t stream = pb_ostream_from_buffer(buffer, sizeof(buffer));
  91. /* Now encode it and check if we succeeded. */
  92. if (pb_encode(&stream, AllTypes_fields, &alltypes))
  93. {
  94. SET_BINARY_MODE(stdout);
  95. fwrite(buffer, 1, stream.bytes_written, stdout);
  96. return 0; /* Success */
  97. }
  98. else
  99. {
  100. fprintf(stderr, "Encoding failed: %s\n", PB_GET_ERROR(&stream));
  101. return 1; /* Failure */
  102. }
  103. }
  104. }