tests/alltypes_pointer/decode_alltypes_pointer.c

   1 #include <stdio.h>
   2 #include <string.h>
   3 #include <stdlib.h>
   4 #include <pb_decode.h>
   5 #include "alltypes.pb.h"
   6 #include "test_helpers.h"
   7
   8 #ifdef HAVE_MALLINFO
   9 #include <malloc.h>
  10 #endif
  11
  12 #define TEST(x) if (!(x)) { \
  13     fprintf(stderr, "Test " #x " failed.\n"); \
  14     status = false; \
  15     }
  16
  17 /* This function is called once from main(), it handles
  18    the decoding and checks the fields. */
  19 bool check_alltypes(pb_istream_t *stream, int mode)
  20 {
  21     bool status = true;
  22     AllTypes alltypes;
  23
  24     /* Fill with garbage to better detect initialization errors */
  25     memset(&alltypes, 0xAA, sizeof(alltypes));
  26
  27     if (!pb_decode(stream, AllTypes_fields, &alltypes))
  28     {
  29         pb_release(AllTypes_fields, &alltypes);
  30         return false;
  31     }
  32
  33     TEST(alltypes.req_int32     && *alltypes.req_int32         == -1001);
  34     TEST(alltypes.req_int64     && *alltypes.req_int64         == -1002);
  35     TEST(alltypes.req_uint32    && *alltypes.req_uint32        == 1003);
  36     TEST(alltypes.req_uint64    && *alltypes.req_uint64        == 1004);
  37     TEST(alltypes.req_sint32    && *alltypes.req_sint32        == -1005);
  38     TEST(alltypes.req_sint64    && *alltypes.req_sint64        == -1006);
  39     TEST(alltypes.req_bool      && *alltypes.req_bool          == true);
  40
  41     TEST(alltypes.req_fixed32   && *alltypes.req_fixed32       == 1008);
  42     TEST(alltypes.req_sfixed32  && *alltypes.req_sfixed32      == -1009);
  43     TEST(alltypes.req_float     && *alltypes.req_float         == 1010.0f);
  44
  45     TEST(alltypes.req_fixed64   && *alltypes.req_fixed64       == 1011);
  46     TEST(alltypes.req_sfixed64  && *alltypes.req_sfixed64      == -1012);
  47     TEST(alltypes.req_double    && *alltypes.req_double        == 1013.0f);
  48
  49     TEST(alltypes.req_string    && strcmp(alltypes.req_string, "1014") == 0);
  50     TEST(alltypes.req_bytes     && alltypes.req_bytes->size == 4);
  51     TEST(alltypes.req_bytes     && alltypes.req_bytes->bytes
  52                                 && memcmp(alltypes.req_bytes->bytes, "1015", 4) == 0);
  53     TEST(alltypes.req_submsg    && alltypes.req_submsg->substuff1
  54                                 && strcmp(alltypes.req_submsg->substuff1, "1016") == 0);
  55     TEST(alltypes.req_submsg    && alltypes.req_submsg->substuff2
  56                                 && *alltypes.req_submsg->substuff2 == 1016);
  57     /* TEST(*alltypes.req_submsg->substuff3 == 3); Default values are not currently supported for pointer fields */
  58     TEST(*alltypes.req_enum == MyEnum_Truth);
  59
  60 #if 0
  61     TEST(alltypes.rep_int32_count == 5 && alltypes.rep_int32[4] == -2001 && alltypes.rep_int32[0] == 0);
  62     TEST(alltypes.rep_int64_count == 5 && alltypes.rep_int64[4] == -2002 && alltypes.rep_int64[0] == 0);
  63     TEST(alltypes.rep_uint32_count == 5 && alltypes.rep_uint32[4] == 2003 && alltypes.rep_uint32[0] == 0);
  64     TEST(alltypes.rep_uint64_count == 5 && alltypes.rep_uint64[4] == 2004 && alltypes.rep_uint64[0] == 0);
  65     TEST(alltypes.rep_sint32_count == 5 && alltypes.rep_sint32[4] == -2005 && alltypes.rep_sint32[0] == 0);
  66     TEST(alltypes.rep_sint64_count == 5 && alltypes.rep_sint64[4] == -2006 && alltypes.rep_sint64[0] == 0);
  67     TEST(alltypes.rep_bool_count == 5 && alltypes.rep_bool[4] == true && alltypes.rep_bool[0] == false);
  68
  69     TEST(alltypes.rep_fixed32_count == 5 && alltypes.rep_fixed32[4] == 2008 && alltypes.rep_fixed32[0] == 0);
  70     TEST(alltypes.rep_sfixed32_count == 5 && alltypes.rep_sfixed32[4] == -2009 && alltypes.rep_sfixed32[0] == 0);
  71     TEST(alltypes.rep_float_count == 5 && alltypes.rep_float[4] == 2010.0f && alltypes.rep_float[0] == 0.0f);
  72
  73     TEST(alltypes.rep_fixed64_count == 5 && alltypes.rep_fixed64[4] == 2011 && alltypes.rep_fixed64[0] == 0);
  74     TEST(alltypes.rep_sfixed64_count == 5 && alltypes.rep_sfixed64[4] == -2012 && alltypes.rep_sfixed64[0] == 0);
  75     TEST(alltypes.rep_double_count == 5 && alltypes.rep_double[4] == 2013.0 && alltypes.rep_double[0] == 0.0);
  76
  77     TEST(alltypes.rep_string_count == 5 && strcmp(alltypes.rep_string[4], "2014") == 0 && alltypes.rep_string[0][0] == '\0');
  78     TEST(alltypes.rep_bytes_count == 5 && alltypes.rep_bytes[4].size == 4 && alltypes.rep_bytes[0].size == 0);
  79     TEST(memcmp(alltypes.rep_bytes[4].bytes, "2015", 4) == 0);
  80
  81     TEST(alltypes.rep_submsg_count == 5);
  82     TEST(strcmp(alltypes.rep_submsg[4].substuff1, "2016") == 0 && alltypes.rep_submsg[0].substuff1[0] == '\0');
  83     TEST(alltypes.rep_submsg[4].substuff2 == 2016 && alltypes.rep_submsg[0].substuff2 == 0);
  84     TEST(alltypes.rep_submsg[4].substuff3 == 2016 && alltypes.rep_submsg[0].substuff3 == 3);
  85
  86     TEST(alltypes.rep_enum_count == 5 && alltypes.rep_enum[4] == MyEnum_Truth && alltypes.rep_enum[0] == MyEnum_Zero);
  87     TEST(alltypes.rep_emptymsg_count == 5);
  88
  89     if (mode == 0)
  90     {
  91         /* Expect default values */
  92         TEST(alltypes.has_opt_int32     == false);
  93         TEST(alltypes.opt_int32         == 4041);
  94         TEST(alltypes.has_opt_int64     == false);
  95         TEST(alltypes.opt_int64         == 4042);
  96         TEST(alltypes.has_opt_uint32    == false);
  97         TEST(alltypes.opt_uint32        == 4043);
  98         TEST(alltypes.has_opt_uint64    == false);
  99         TEST(alltypes.opt_uint64        == 4044);
 100         TEST(alltypes.has_opt_sint32    == false);
 101         TEST(alltypes.opt_sint32        == 4045);
 102         TEST(alltypes.has_opt_sint64    == false);
 103         TEST(alltypes.opt_sint64        == 4046);
 104         TEST(alltypes.has_opt_bool      == false);
 105         TEST(alltypes.opt_bool          == false);
 106
 107         TEST(alltypes.has_opt_fixed32   == false);
 108         TEST(alltypes.opt_fixed32       == 4048);
 109         TEST(alltypes.has_opt_sfixed32  == false);
 110         TEST(alltypes.opt_sfixed32      == 4049);
 111         TEST(alltypes.has_opt_float     == false);
 112         TEST(alltypes.opt_float         == 4050.0f);
 113
 114         TEST(alltypes.has_opt_fixed64   == false);
 115         TEST(alltypes.opt_fixed64       == 4051);
 116         TEST(alltypes.has_opt_sfixed64  == false);
 117         TEST(alltypes.opt_sfixed64      == 4052);
 118         TEST(alltypes.has_opt_double    == false);
 119         TEST(alltypes.opt_double        == 4053.0);
 120
 121         TEST(alltypes.has_opt_string    == false);
 122         TEST(strcmp(alltypes.opt_string, "4054") == 0);
 123         TEST(alltypes.has_opt_bytes     == false);
 124         TEST(alltypes.opt_bytes.size == 4);
 125         TEST(memcmp(alltypes.opt_bytes.bytes, "4055", 4) == 0);
 126         TEST(alltypes.has_opt_submsg    == false);
 127         TEST(strcmp(alltypes.opt_submsg.substuff1, "1") == 0);
 128         TEST(alltypes.opt_submsg.substuff2 == 2);
 129         TEST(alltypes.opt_submsg.substuff3 == 3);
 130         TEST(alltypes.has_opt_enum     == false);
 131         TEST(alltypes.opt_enum == MyEnum_Second);
 132         TEST(alltypes.has_opt_emptymsg == false);
 133     }
 134     else
 135     {
 136         /* Expect filled-in values */
 137         TEST(alltypes.has_opt_int32     == true);
 138         TEST(alltypes.opt_int32         == 3041);
 139         TEST(alltypes.has_opt_int64     == true);
 140         TEST(alltypes.opt_int64         == 3042);
 141         TEST(alltypes.has_opt_uint32    == true);
 142         TEST(alltypes.opt_uint32        == 3043);
 143         TEST(alltypes.has_opt_uint64    == true);
 144         TEST(alltypes.opt_uint64        == 3044);
 145         TEST(alltypes.has_opt_sint32    == true);
 146         TEST(alltypes.opt_sint32        == 3045);
 147         TEST(alltypes.has_opt_sint64    == true);
 148         TEST(alltypes.opt_sint64        == 3046);
 149         TEST(alltypes.has_opt_bool      == true);
 150         TEST(alltypes.opt_bool          == true);
 151
 152         TEST(alltypes.has_opt_fixed32   == true);
 153         TEST(alltypes.opt_fixed32       == 3048);
 154         TEST(alltypes.has_opt_sfixed32  == true);
 155         TEST(alltypes.opt_sfixed32      == 3049);
 156         TEST(alltypes.has_opt_float     == true);
 157         TEST(alltypes.opt_float         == 3050.0f);
 158
 159         TEST(alltypes.has_opt_fixed64   == true);
 160         TEST(alltypes.opt_fixed64       == 3051);
 161         TEST(alltypes.has_opt_sfixed64  == true);
 162         TEST(alltypes.opt_sfixed64      == 3052);
 163         TEST(alltypes.has_opt_double    == true);
 164         TEST(alltypes.opt_double        == 3053.0);
 165
 166         TEST(alltypes.has_opt_string    == true);
 167         TEST(strcmp(alltypes.opt_string, "3054") == 0);
 168         TEST(alltypes.has_opt_bytes     == true);
 169         TEST(alltypes.opt_bytes.size == 4);
 170         TEST(memcmp(alltypes.opt_bytes.bytes, "3055", 4) == 0);
 171         TEST(alltypes.has_opt_submsg    == true);
 172         TEST(strcmp(alltypes.opt_submsg.substuff1, "3056") == 0);
 173         TEST(alltypes.opt_submsg.substuff2 == 3056);
 174         TEST(alltypes.opt_submsg.substuff3 == 3);
 175         TEST(alltypes.has_opt_enum      == true);
 176         TEST(alltypes.opt_enum == MyEnum_Truth);
 177         TEST(alltypes.has_opt_emptymsg  == true);
 178     }
 179
 180     TEST(alltypes.req_limits.int32_min  == INT32_MIN);
 181     TEST(alltypes.req_limits.int32_max  == INT32_MAX);
 182     TEST(alltypes.req_limits.uint32_min == 0);
 183     TEST(alltypes.req_limits.uint32_max == UINT32_MAX);
 184     TEST(alltypes.req_limits.int64_min  == INT64_MIN);
 185     TEST(alltypes.req_limits.int64_max  == INT64_MAX);
 186     TEST(alltypes.req_limits.uint64_min == 0);
 187     TEST(alltypes.req_limits.uint64_max == UINT64_MAX);
 188     TEST(alltypes.req_limits.enum_min   == HugeEnum_Negative);
 189     TEST(alltypes.req_limits.enum_max   == HugeEnum_Positive);
 190
 191     TEST(alltypes.end == 1099);
 192 #endif
 193
 194     pb_release(AllTypes_fields, &alltypes);
 195
 196     return status;
 197 }
 198
 199 int main(int argc, char **argv)
 200 {
 201     bool status;
 202     int orig_allocations;
 203
 204 #ifdef HAVE_MALLINFO
 205     /* Dynamic library loader etc. may have some malloc()ed memory also. */
 206     {
 207         struct mallinfo m = mallinfo();
 208         orig_allocations = m.uordblks;
 209     }
 210 #endif
 211
 212     {
 213         uint8_t buffer[1024];
 214         size_t count;
 215         pb_istream_t stream;
 216
 217         /* Whether to expect the optional values or the default values. */
 218         int mode = (argc > 1) ? atoi(argv[1]) : 0;
 219
 220         /* Read the data into buffer */
 221         SET_BINARY_MODE(stdin);
 222         count = fread(buffer, 1, sizeof(buffer), stdin);
 223
 224         /* Construct a pb_istream_t for reading from the buffer */
 225         stream = pb_istream_from_buffer(buffer, count);
 226
 227         /* Decode and verify the message */
 228         status = check_alltypes(&stream, mode);
 229
 230         if (!status)
 231             fprintf(stderr, "Parsing failed: %s\n", PB_GET_ERROR(&stream));
 232     }
 233
 234 #ifdef HAVE_MALLINFO
 235     /* Check for memory leaks */
 236     {
 237         struct mallinfo m = mallinfo();
 238         int leak = m.uordblks - orig_allocations;
 239
 240         if (leak > 0)
 241         {
 242             fprintf(stderr, "Memory leak: %d bytes\n", leak);
 243             return 1;
 244         }
 245         else
 246         {
 247             fprintf(stderr, "Ok, no memory leaks\n");
 248         }
 249     }
 250 #endif
 251
 252     if (!status)
 253     {
 254         return 1;
 255     }
 256     else
 257     {
 258         return 0;
 259     }
 260 }