tests/alltypes_pointer/encode_alltypes_pointer.c

   1 /* Attempts to test all the datatypes supported by ProtoBuf.
   2  */
   3
   4 #include <stdio.h>
   5 #include <stdlib.h>
   6 #include <string.h>
   7 #include <pb_encode.h>
   8 #include "alltypes.pb.h"
   9 #include "test_helpers.h"
  10
  11 int main(int argc, char **argv)
  12 {
  13     int mode = (argc > 1) ? atoi(argv[1]) : 0;
  14
  15     /* Values for required fields */
  16     int32_t     req_int32         = -1001;
  17     int64_t     req_int64         = -1002;
  18     uint32_t    req_uint32        = 1003;
  19     uint64_t    req_uint64        = 1004;
  20     int32_t     req_sint32        = -1005;
  21     int64_t     req_sint64        = -1006;
  22     bool        req_bool          = true;
  23     uint32_t    req_fixed32       = 1008;
  24     int32_t     req_sfixed32      = -1009;
  25     float       req_float         = 1010.0f;
  26     uint64_t    req_fixed64       = 1011;
  27     int64_t     req_sfixed64      = -1012;
  28     double      req_double        = 1013.0;
  29     char*       req_string        = "1014";
  30     PB_BYTES_ARRAY_T(4) req_bytes = {4, {'1', '0', '1', '5'}};
  31     static int32_t req_substuff   = 1016;
  32     SubMessage  req_submsg        = {"1016", &req_substuff};
  33     MyEnum      req_enum          = MyEnum_Truth;
  34     EmptyMessage req_emptymsg     = {0};
  35
  36     int32_t     end               = 1099;
  37
  38     /* Values for repeated fields */
  39     int32_t     rep_int32[5]      = {0, 0, 0, 0, -2001};
  40     int64_t     rep_int64[5]      = {0, 0, 0, 0, -2002};
  41     uint32_t    rep_uint32[5]     = {0, 0, 0, 0, 2003};
  42     uint64_t    rep_uint64[5]     = {0, 0, 0, 0, 2004};
  43     int32_t     rep_sint32[5]     = {0, 0, 0, 0, -2005};
  44     int64_t     rep_sint64[5]     = {0, 0, 0, 0, -2006};
  45     bool        rep_bool[5]       = {false, false, false, false, true};
  46     uint32_t    rep_fixed32[5]    = {0, 0, 0, 0, 2008};
  47     int32_t     rep_sfixed32[5]   = {0, 0, 0, 0, -2009};
  48     float       rep_float[5]      = {0, 0, 0, 0, 2010.0f};
  49     uint64_t    rep_fixed64[5]    = {0, 0, 0, 0, 2011};
  50     int64_t     rep_sfixed64[5]   = {0, 0, 0, 0, -2012};
  51     double      rep_double[5]     = {0, 0, 0, 0, 2013.0f};
  52     char*       rep_string[5]     = {"", "", "", "", "2014"};
  53     static PB_BYTES_ARRAY_T(4) rep_bytes_4 = {4, {'2', '0', '1', '5'}};
  54     pb_bytes_array_t *rep_bytes[5]= {NULL, NULL, NULL, NULL, (pb_bytes_array_t*)&rep_bytes_4};
  55     static int32_t rep_sub2zero   = 0;
  56     static int32_t rep_substuff2  = 2016;
  57     static uint32_t rep_substuff3 = 2016;
  58     SubMessage  rep_submsg[5]     = {{"", &rep_sub2zero},
  59                                      {"", &rep_sub2zero},
  60                                      {"", &rep_sub2zero},
  61                                      {"", &rep_sub2zero},
  62                                      {"2016", &rep_substuff2, &rep_substuff3}};
  63     MyEnum      rep_enum[5]       = {0, 0, 0, 0, MyEnum_Truth};
  64     EmptyMessage rep_emptymsg[5]  = {{0}, {0}, {0}, {0}, {0}};
  65
  66     /* Values for optional fields */
  67     int32_t     opt_int32         = 3041;
  68     int64_t     opt_int64         = 3042;
  69     uint32_t    opt_uint32        = 3043;
  70     uint64_t    opt_uint64        = 3044;
  71     int32_t     opt_sint32        = 3045;
  72     int64_t     opt_sint64        = 3046;
  73     bool        opt_bool          = true;
  74     uint32_t    opt_fixed32       = 3048;
  75     int32_t     opt_sfixed32      = 3049;
  76     float       opt_float         = 3050.0f;
  77     uint64_t    opt_fixed64       = 3051;
  78     int64_t     opt_sfixed64      = 3052;
  79     double      opt_double        = 3053.0;
  80     char*       opt_string        = "3054";
  81     PB_BYTES_ARRAY_T(4) opt_bytes = {4, {'3', '0', '5', '5'}};
  82     static int32_t opt_substuff   = 3056;
  83     SubMessage  opt_submsg        = {"3056", &opt_substuff};
  84     MyEnum      opt_enum          = MyEnum_Truth;
  85     EmptyMessage opt_emptymsg     = {0};
  86
  87     /* Values for the Limits message. */
  88     static int32_t  int32_min  = INT32_MIN;
  89     static int32_t  int32_max  = INT32_MAX;
  90     static uint32_t uint32_min = 0;
  91     static uint32_t uint32_max = UINT32_MAX;
  92     static int64_t  int64_min  = INT64_MIN;
  93     static int64_t  int64_max  = INT64_MAX;
  94     static uint64_t uint64_min = 0;
  95     static uint64_t uint64_max = UINT64_MAX;
  96     static HugeEnum enum_min   = HugeEnum_Negative;
  97     static HugeEnum enum_max   = HugeEnum_Positive;
  98     Limits req_limits = {&int32_min,    &int32_max,
  99                          &uint32_min,   &uint32_max,
 100                          &int64_min,    &int64_max,
 101                          &uint64_min,   &uint64_max,
 102                          &enum_min,     &enum_max};
 103
 104     /* Initialize the message struct with pointers to the fields. */
 105     AllTypes alltypes = {0};
 106
 107     alltypes.req_int32         = &req_int32;
 108     alltypes.req_int64         = &req_int64;
 109     alltypes.req_uint32        = &req_uint32;
 110     alltypes.req_uint64        = &req_uint64;
 111     alltypes.req_sint32        = &req_sint32;
 112     alltypes.req_sint64        = &req_sint64;
 113     alltypes.req_bool          = &req_bool;
 114     alltypes.req_fixed32       = &req_fixed32;
 115     alltypes.req_sfixed32      = &req_sfixed32;
 116     alltypes.req_float         = &req_float;
 117     alltypes.req_fixed64       = &req_fixed64;
 118     alltypes.req_sfixed64      = &req_sfixed64;
 119     alltypes.req_double        = &req_double;
 120     alltypes.req_string        = req_string;
 121     alltypes.req_bytes         = (pb_bytes_array_t*)&req_bytes;
 122     alltypes.req_submsg        = &req_submsg;
 123     alltypes.req_enum          = &req_enum;
 124     alltypes.req_emptymsg      = &req_emptymsg;
 125     alltypes.req_limits        = &req_limits;
 126
 127     alltypes.rep_int32_count    = 5; alltypes.rep_int32     = rep_int32;
 128     alltypes.rep_int64_count    = 5; alltypes.rep_int64     = rep_int64;
 129     alltypes.rep_uint32_count   = 5; alltypes.rep_uint32    = rep_uint32;
 130     alltypes.rep_uint64_count   = 5; alltypes.rep_uint64    = rep_uint64;
 131     alltypes.rep_sint32_count   = 5; alltypes.rep_sint32    = rep_sint32;
 132     alltypes.rep_sint64_count   = 5; alltypes.rep_sint64    = rep_sint64;
 133     alltypes.rep_bool_count     = 5; alltypes.rep_bool      = rep_bool;
 134     alltypes.rep_fixed32_count  = 5; alltypes.rep_fixed32   = rep_fixed32;
 135     alltypes.rep_sfixed32_count = 5; alltypes.rep_sfixed32  = rep_sfixed32;
 136     alltypes.rep_float_count    = 5; alltypes.rep_float     = rep_float;
 137     alltypes.rep_fixed64_count  = 5; alltypes.rep_fixed64   = rep_fixed64;
 138     alltypes.rep_sfixed64_count = 5; alltypes.rep_sfixed64  = rep_sfixed64;
 139     alltypes.rep_double_count   = 5; alltypes.rep_double    = rep_double;
 140     alltypes.rep_string_count   = 5; alltypes.rep_string    = rep_string;
 141     alltypes.rep_bytes_count    = 5; alltypes.rep_bytes     = rep_bytes;
 142     alltypes.rep_submsg_count   = 5; alltypes.rep_submsg    = rep_submsg;
 143     alltypes.rep_enum_count     = 5; alltypes.rep_enum      = rep_enum;
 144     alltypes.rep_emptymsg_count = 5; alltypes.rep_emptymsg  = rep_emptymsg;
 145
 146     if (mode != 0)
 147     {
 148         /* Fill in values for optional fields */
 149         alltypes.opt_int32         = &opt_int32;
 150         alltypes.opt_int64         = &opt_int64;
 151         alltypes.opt_uint32        = &opt_uint32;
 152         alltypes.opt_uint64        = &opt_uint64;
 153         alltypes.opt_sint32        = &opt_sint32;
 154         alltypes.opt_sint64        = &opt_sint64;
 155         alltypes.opt_bool          = &opt_bool;
 156         alltypes.opt_fixed32       = &opt_fixed32;
 157         alltypes.opt_sfixed32      = &opt_sfixed32;
 158         alltypes.opt_float         = &opt_float;
 159         alltypes.opt_fixed64       = &opt_fixed64;
 160         alltypes.opt_sfixed64      = &opt_sfixed64;
 161         alltypes.opt_double        = &opt_double;
 162         alltypes.opt_string        = opt_string;
 163         alltypes.opt_bytes         = (pb_bytes_array_t*)&opt_bytes;
 164         alltypes.opt_submsg        = &opt_submsg;
 165         alltypes.opt_enum          = &opt_enum;
 166         alltypes.opt_emptymsg      = &opt_emptymsg;
 167     }
 168
 169     alltypes.end = &end;
 170
 171     {
 172         uint8_t buffer[4096];
 173         pb_ostream_t stream = pb_ostream_from_buffer(buffer, sizeof(buffer));
 174
 175         /* Now encode it and check if we succeeded. */
 176         if (pb_encode(&stream, AllTypes_fields, &alltypes))
 177         {
 178             SET_BINARY_MODE(stdout);
 179             fwrite(buffer, 1, stream.bytes_written, stdout);
 180             return 0; /* Success */
 181         }
 182         else
 183         {
 184             fprintf(stderr, "Encoding failed: %s\n", PB_GET_ERROR(&stream));
 185             return 1; /* Failure */
 186         }
 187     }
 188 }