libs/nanopb/tests/regression/issue_249/test.c

   1 #include "test.pb.h"
   2 #include <unittests.h>
   3 #include <pb_encode.h>
   4 #include <pb_decode.h>
   5
   6 static bool write_array(pb_ostream_t *stream, const pb_field_t *field, void * const *arg)
   7 {
   8     int i;
   9     for (i = 0; i < 5; i++)
  10     {
  11         if (!pb_encode_tag_for_field(stream, field))
  12             return false;
  13         if (!pb_encode_varint(stream, 1000 + i))
  14             return false;
  15     }
  16
  17     return true;
  18 }
  19
  20 static bool read_array(pb_istream_t *stream, const pb_field_t *field, void **arg)
  21 {
  22     uint32_t i;
  23     int *sum = *arg;
  24
  25     if (!pb_decode_varint32(stream, &i))
  26         return false;
  27
  28     *sum += i;
  29
  30     return true;
  31 }
  32
  33 int main()
  34 {
  35     int status = 0;
  36     pb_byte_t buf[128] = {0};
  37     pb_size_t msglen;
  38
  39     {
  40         MainMessage msg = MainMessage_init_zero;
  41         pb_ostream_t stream = pb_ostream_from_buffer(buf, sizeof(buf));
  42         msg.submsg.foo.funcs.encode = &write_array;
  43         TEST(pb_encode(&stream, MainMessage_fields, &msg));
  44         msglen = stream.bytes_written;
  45     }
  46
  47     {
  48         MainMessage msg = MainMessage_init_zero;
  49         pb_istream_t stream = pb_istream_from_buffer(buf, msglen);
  50         int sum = 0;
  51         msg.submsg.foo.funcs.decode = &read_array;
  52         msg.submsg.foo.arg = &sum;
  53         TEST(pb_decode(&stream, MainMessage_fields, &msg));
  54         TEST(sum == 1000 + 1001 + 1002 + 1003 + 1004);
  55     }
  56
  57     return status;
  58 }
  59