libs/nanopb/tests/alltypes_proto3_callback/encode_alltypes_callback.c

   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.
   4  */
   5
   6 #include <stdio.h>
   7 #include <stdlib.h>
   8 #include <string.h>
   9 #include <pb_encode.h>
  10 #include "alltypes.pb.h"
  11 #include "test_helpers.h"
  12
  13 static bool write_varint(pb_ostream_t *stream, const pb_field_t *field, void * const *arg)
  14 {
  15     return pb_encode_tag_for_field(stream, field) &&
  16            pb_encode_varint(stream, (long)*arg);
  17 }
  18
  19 static bool write_svarint(pb_ostream_t *stream, const pb_field_t *field, void * const *arg)
  20 {
  21     return pb_encode_tag_for_field(stream, field) &&
  22            pb_encode_svarint(stream, (long)*arg);
  23 }
  24
  25 static bool write_fixed32(pb_ostream_t *stream, const pb_field_t *field, void * const *arg)
  26 {
  27     return pb_encode_tag_for_field(stream, field) &&
  28            pb_encode_fixed32(stream, *arg);
  29 }
  30
  31 static bool write_fixed64(pb_ostream_t *stream, const pb_field_t *field, void * const *arg)
  32 {
  33     return pb_encode_tag_for_field(stream, field) &&
  34            pb_encode_fixed64(stream, *arg);
  35 }
  36
  37 static bool write_string(pb_ostream_t *stream, const pb_field_t *field, void * const *arg)
  38 {
  39     return pb_encode_tag_for_field(stream, field) &&
  40            pb_encode_string(stream, *arg, strlen(*arg));
  41 }
  42
  43 static bool write_submsg(pb_ostream_t *stream, const pb_field_t *field, void * const *arg)
  44 {
  45
  46     return pb_encode_tag_for_field(stream, field) &&
  47            pb_encode_submessage(stream, SubMessage_fields, *arg);
  48 }
  49
  50 static bool write_emptymsg(pb_ostream_t *stream, const pb_field_t *field, void * const *arg)
  51 {
  52     EmptyMessage emptymsg = {0};
  53     return pb_encode_tag_for_field(stream, field) &&
  54            pb_encode_submessage(stream, EmptyMessage_fields, &emptymsg);
  55 }
  56
  57 static bool write_repeated_varint(pb_ostream_t *stream, const pb_field_t *field, void * const *arg)
  58 {
  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);
  69 }
  70
  71 static bool write_repeated_svarint(pb_ostream_t *stream, const pb_field_t *field, void * const *arg)
  72 {
  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);
  83 }
  84
  85 static bool write_repeated_fixed32(pb_ostream_t *stream, const pb_field_t *field, void * const *arg)
  86 {
  87     uint32_t dummy = 0;
  88
  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);
  97 }
  98
  99 static bool write_repeated_fixed64(pb_ostream_t *stream, const pb_field_t *field, void * const *arg)
 100 {
 101     uint64_t dummy = 0;
 102
 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);
 111 }
 112
 113 static bool write_repeated_string(pb_ostream_t *stream, const pb_field_t *field, void * const *arg)
 114 {
 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));
 125 }
 126
 127 static bool write_repeated_submsg(pb_ostream_t *stream, const pb_field_t *field, void * const *arg)
 128 {
 129     SubMessage dummy = {""};
 130
 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);
 141 }
 142
 143 static bool write_limits(pb_ostream_t *stream, const pb_field_t *field, void * const *arg)
 144 {
 145     Limits limits = {0};
 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;
 156
 157     return pb_encode_tag_for_field(stream, field) &&
 158            pb_encode_submessage(stream, Limits_fields, &limits);
 159 }
 160
 161 static bool write_repeated_emptymsg(pb_ostream_t *stream, const pb_field_t *field, void * const *arg)
 162 {
 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);
 174 }
 175
 176 int main(int argc, char **argv)
 177 {
 178     int mode = (argc > 1) ? atoi(argv[1]) : 0;
 179
 180     /* Values for use from callbacks through pointers. */
 181     uint32_t    rep_fixed32     = 2008;
 182     int32_t     rep_sfixed32    = -2009;
 183     float       rep_float       = 2010.0f;
 184     uint64_t    rep_fixed64     = 2011;
 185     int64_t     rep_sfixed64    = -2012;
 186     double      rep_double      = 2013.0;
 187     SubMessage  rep_submsg      = {"2016", 2016, 2016};
 188
 189     uint32_t    sng_fixed32     = 3048;
 190     int32_t     sng_sfixed32    = 3049;
 191     float       sng_float       = 3050.0f;
 192     uint64_t    sng_fixed64     = 3051;
 193     int64_t     sng_sfixed64    = 3052;
 194     double      sng_double      = 3053.0f;
 195     SubMessage  sng_submsg      = {"3056", 3056};
 196
 197     SubMessage  oneof_msg1      = {"4059", 4059};
 198
 199     AllTypes alltypes = AllTypes_init_zero;
 200
 201     /* Bind callbacks for repeated fields */
 202     alltypes.rep_int32.funcs.encode = &write_repeated_varint;
 203     alltypes.rep_int32.arg = (void*)-2001;
 204
 205     alltypes.rep_int64.funcs.encode = &write_repeated_varint;
 206     alltypes.rep_int64.arg = (void*)-2002;
 207
 208     alltypes.rep_uint32.funcs.encode = &write_repeated_varint;
 209     alltypes.rep_uint32.arg = (void*)2003;
 210
 211     alltypes.rep_uint64.funcs.encode = &write_repeated_varint;
 212     alltypes.rep_uint64.arg = (void*)2004;
 213
 214     alltypes.rep_sint32.funcs.encode = &write_repeated_svarint;
 215     alltypes.rep_sint32.arg = (void*)-2005;
 216
 217     alltypes.rep_sint64.funcs.encode = &write_repeated_svarint;
 218     alltypes.rep_sint64.arg = (void*)-2006;
 219
 220     alltypes.rep_bool.funcs.encode = &write_repeated_varint;
 221     alltypes.rep_bool.arg = (void*)true;
 222
 223     alltypes.rep_fixed32.funcs.encode = &write_repeated_fixed32;
 224     alltypes.rep_fixed32.arg = &rep_fixed32;
 225
 226     alltypes.rep_sfixed32.funcs.encode = &write_repeated_fixed32;
 227     alltypes.rep_sfixed32.arg = &rep_sfixed32;
 228
 229     alltypes.rep_float.funcs.encode = &write_repeated_fixed32;
 230     alltypes.rep_float.arg = &rep_float;
 231
 232     alltypes.rep_fixed64.funcs.encode = &write_repeated_fixed64;
 233     alltypes.rep_fixed64.arg = &rep_fixed64;
 234
 235     alltypes.rep_sfixed64.funcs.encode = &write_repeated_fixed64;
 236     alltypes.rep_sfixed64.arg = &rep_sfixed64;
 237
 238     alltypes.rep_double.funcs.encode = &write_repeated_fixed64;
 239     alltypes.rep_double.arg = &rep_double;
 240
 241     alltypes.rep_string.funcs.encode = &write_repeated_string;
 242     alltypes.rep_string.arg = "2014";
 243
 244     alltypes.rep_bytes.funcs.encode = &write_repeated_string;
 245     alltypes.rep_bytes.arg = "2015";
 246
 247     alltypes.rep_submsg.funcs.encode = &write_repeated_submsg;
 248     alltypes.rep_submsg.arg = &rep_submsg;
 249
 250     alltypes.rep_enum.funcs.encode = &write_repeated_varint;
 251     alltypes.rep_enum.arg = (void*)MyEnum_Truth;
 252
 253     alltypes.rep_emptymsg.funcs.encode = &write_repeated_emptymsg;
 254
 255     alltypes.rep_fbytes.funcs.encode = &write_repeated_string;
 256     alltypes.rep_fbytes.arg = "2019";
 257
 258     alltypes.req_limits.funcs.encode = &write_limits;
 259
 260     /* Bind callbacks for singular fields */
 261     if (mode != 0)
 262     {
 263         alltypes.sng_int32.funcs.encode = &write_varint;
 264         alltypes.sng_int32.arg = (void*)3041;
 265
 266         alltypes.sng_int64.funcs.encode = &write_varint;
 267         alltypes.sng_int64.arg = (void*)3042;
 268
 269         alltypes.sng_uint32.funcs.encode = &write_varint;
 270         alltypes.sng_uint32.arg = (void*)3043;
 271
 272         alltypes.sng_uint64.funcs.encode = &write_varint;
 273         alltypes.sng_uint64.arg = (void*)3044;
 274
 275         alltypes.sng_sint32.funcs.encode = &write_svarint;
 276         alltypes.sng_sint32.arg = (void*)3045;
 277
 278         alltypes.sng_sint64.funcs.encode = &write_svarint;
 279         alltypes.sng_sint64.arg = (void*)3046;
 280
 281         alltypes.sng_bool.funcs.encode = &write_varint;
 282         alltypes.sng_bool.arg = (void*)true;
 283
 284         alltypes.sng_fixed32.funcs.encode = &write_fixed32;
 285         alltypes.sng_fixed32.arg = &sng_fixed32;
 286
 287         alltypes.sng_sfixed32.funcs.encode = &write_fixed32;
 288         alltypes.sng_sfixed32.arg = &sng_sfixed32;
 289
 290         alltypes.sng_float.funcs.encode = &write_fixed32;
 291         alltypes.sng_float.arg = &sng_float;
 292
 293         alltypes.sng_fixed64.funcs.encode = &write_fixed64;
 294         alltypes.sng_fixed64.arg = &sng_fixed64;
 295
 296         alltypes.sng_sfixed64.funcs.encode = &write_fixed64;
 297         alltypes.sng_sfixed64.arg = &sng_sfixed64;
 298
 299         alltypes.sng_double.funcs.encode = &write_fixed64;
 300         alltypes.sng_double.arg = &sng_double;
 301
 302         alltypes.sng_string.funcs.encode = &write_string;
 303         alltypes.sng_string.arg = "3054";
 304
 305         alltypes.sng_bytes.funcs.encode = &write_string;
 306         alltypes.sng_bytes.arg = "3055";
 307
 308         alltypes.sng_submsg.funcs.encode = &write_submsg;
 309         alltypes.sng_submsg.arg = &sng_submsg;
 310
 311         alltypes.sng_enum.funcs.encode = &write_varint;
 312         alltypes.sng_enum.arg = (void*)MyEnum_Truth;
 313
 314         alltypes.sng_emptymsg.funcs.encode = &write_emptymsg;
 315
 316         alltypes.sng_fbytes.funcs.encode = &write_string;
 317         alltypes.sng_fbytes.arg = "3059";
 318
 319         alltypes.oneof_msg1.funcs.encode = &write_submsg;
 320         alltypes.oneof_msg1.arg = &oneof_msg1;
 321     }
 322
 323     alltypes.end.funcs.encode = &write_varint;
 324     alltypes.end.arg = (void*)1099;
 325
 326     {
 327         uint8_t buffer[2048];
 328         pb_ostream_t stream = pb_ostream_from_buffer(buffer, sizeof(buffer));
 329
 330         /* Now encode it and check if we succeeded. */
 331         if (pb_encode(&stream, AllTypes_fields, &alltypes))
 332         {
 333             SET_BINARY_MODE(stdout);
 334             fwrite(buffer, 1, stream.bytes_written, stdout);
 335             return 0; /* Success */
 336         }
 337         else
 338         {
 339             fprintf(stderr, "Encoding failed: %s\n", PB_GET_ERROR(&stream));
 340             return 1; /* Failure */
 341         }
 342     }
 343 }