tests/test_encode3.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
  10 int main(int argc, char **argv)
  11 {
  12     int mode = (argc > 1) ? atoi(argv[1]) : 0;
  13
  14     /* Initialize the structure with constants */
  15     AllTypes alltypes = {0};
  16
  17     alltypes.req_int32         = 1001;
  18     alltypes.req_int64         = 1002;
  19     alltypes.req_uint32        = 1003;
  20     alltypes.req_uint64        = 1004;
  21     alltypes.req_sint32        = 1005;
  22     alltypes.req_sint64        = 1006;
  23     alltypes.req_bool          = true;
  24
  25     alltypes.req_fixed32       = 1008;
  26     alltypes.req_sfixed32      = 1009;
  27     alltypes.req_float         = 1010.0f;
  28
  29     alltypes.req_fixed64       = 1011;
  30     alltypes.req_sfixed64      = 1012;
  31     alltypes.req_double        = 1013.0;
  32
  33     strcpy(alltypes.req_string, "1014");
  34     alltypes.req_bytes.size = 4;
  35     memcpy(alltypes.req_bytes.bytes, "1015", 4);
  36     strcpy(alltypes.req_submsg.substuff1, "1016");
  37     alltypes.req_submsg.substuff2 = 1016;
  38     alltypes.req_enum = MyEnum_Truth;
  39
  40     alltypes.rep_int32_count = 5; alltypes.rep_int32[4] = 2001;
  41     alltypes.rep_int64_count = 5; alltypes.rep_int64[4] = 2002;
  42     alltypes.rep_uint32_count = 5; alltypes.rep_uint32[4] = 2003;
  43     alltypes.rep_uint64_count = 5; alltypes.rep_uint64[4] = 2004;
  44     alltypes.rep_sint32_count = 5; alltypes.rep_sint32[4] = 2005;
  45     alltypes.rep_sint64_count = 5; alltypes.rep_sint64[4] = 2006;
  46     alltypes.rep_bool_count = 5; alltypes.rep_bool[4] = true;
  47
  48     alltypes.rep_fixed32_count = 5; alltypes.rep_fixed32[4] = 2008;
  49     alltypes.rep_sfixed32_count = 5; alltypes.rep_sfixed32[4] = 2009;
  50     alltypes.rep_float_count = 5; alltypes.rep_float[4] = 2010.0f;
  51
  52     alltypes.rep_fixed64_count = 5; alltypes.rep_fixed64[4] = 2011;
  53     alltypes.rep_sfixed64_count = 5; alltypes.rep_sfixed64[4] = 2012;
  54     alltypes.rep_double_count = 5; alltypes.rep_double[4] = 2013.0;
  55
  56     alltypes.rep_string_count = 5; strcpy(alltypes.rep_string[4], "2014");
  57     alltypes.rep_bytes_count = 5; alltypes.rep_bytes[4].size = 4;
  58     memcpy(alltypes.rep_bytes[4].bytes, "2015", 4);
  59
  60     alltypes.rep_submsg_count = 5;
  61     strcpy(alltypes.rep_submsg[4].substuff1, "2016");
  62     alltypes.rep_submsg[4].substuff2 = 2016;
  63     alltypes.rep_submsg[4].has_substuff3 = true;
  64     alltypes.rep_submsg[4].substuff3 = 2016;
  65
  66     alltypes.rep_enum_count = 5; alltypes.rep_enum[4] = MyEnum_Truth;
  67
  68     if (mode != 0)
  69     {
  70         /* Fill in values for optional fields */
  71         alltypes.has_opt_int32 = true;
  72         alltypes.opt_int32         = 3041;
  73         alltypes.has_opt_int64 = true;
  74         alltypes.opt_int64         = 3042;
  75         alltypes.has_opt_uint32 = true;
  76         alltypes.opt_uint32        = 3043;
  77         alltypes.has_opt_uint64 = true;
  78         alltypes.opt_uint64        = 3044;
  79         alltypes.has_opt_sint32 = true;
  80         alltypes.opt_sint32        = 3045;
  81         alltypes.has_opt_sint64 = true;
  82         alltypes.opt_sint64        = 3046;
  83         alltypes.has_opt_bool = true;
  84         alltypes.opt_bool          = true;
  85
  86         alltypes.has_opt_fixed32 = true;
  87         alltypes.opt_fixed32       = 3048;
  88         alltypes.has_opt_sfixed32 = true;
  89         alltypes.opt_sfixed32      = 3049;
  90         alltypes.has_opt_float = true;
  91         alltypes.opt_float         = 3050.0f;
  92
  93         alltypes.has_opt_fixed64 = true;
  94         alltypes.opt_fixed64       = 3051;
  95         alltypes.has_opt_sfixed64 = true;
  96         alltypes.opt_sfixed64      = 3052;
  97         alltypes.has_opt_double = true;
  98         alltypes.opt_double        = 3053.0;
  99
 100         alltypes.has_opt_string = true;
 101         strcpy(alltypes.opt_string, "3054");
 102         alltypes.has_opt_bytes = true;
 103         alltypes.opt_bytes.size = 4;
 104         memcpy(alltypes.opt_bytes.bytes, "3055", 4);
 105         alltypes.has_opt_submsg = true;
 106         strcpy(alltypes.opt_submsg.substuff1, "3056");
 107         alltypes.opt_submsg.substuff2 = 3056;
 108         alltypes.has_opt_enum = true;
 109         alltypes.opt_enum = MyEnum_Truth;
 110     }
 111
 112     alltypes.end = 1099;
 113
 114     uint8_t buffer[1024];
 115     pb_ostream_t stream = pb_ostream_from_buffer(buffer, sizeof(buffer));
 116
 117     /* Now encode it and check if we succeeded. */
 118     if (pb_encode(&stream, AllTypes_fields, &alltypes))
 119     {
 120         fwrite(buffer, 1, stream.bytes_written, stdout);
 121         return 0; /* Success */
 122     }
 123     else
 124     {
 125         fprintf(stderr, "Encoding failed!\n");
 126         return 1; /* Failure */
 127     }
 128 }