Line data    Source code

       1             : #include "submsg_callback.pb.h"
       2             : #include <pb_encode.h>
       3             : #include <pb_decode.h>
       4             : #include "unittests.h"
       5             : 
       6           1 : bool msg_callback(pb_istream_t *stream, const pb_field_t *field, void **arg)
       7             : {
       8             :     /* This tests decoding the submessage already in the message level callback. */
       9             : 
      10           1 :     SubMessage *submsg = (SubMessage*)field->pData;
      11             : 
      12           1 :     if (!pb_decode(stream, SubMessage_fields, submsg))
      13           0 :         PB_RETURN_ERROR(stream, "submsg decode failed");
      14             :     
      15           1 :     if (submsg->foo != 1234)
      16           0 :         PB_RETURN_ERROR(stream, "submsg.foo wrong value");
      17             :     
      18           1 :     return true;
      19             : }
      20             : 
      21           1 : int main()
      22             : {
      23           1 :     int status = 0;
      24             :     pb_byte_t buf[64];
      25             :     size_t msglen;
      26             : 
      27             :     {
      28           1 :         pb_ostream_t ostream = pb_ostream_from_buffer(buf, sizeof(buf));
      29           1 :         MyMessage msg = MyMessage_init_zero;
      30             : 
      31           1 :         msg.which_oneof = MyMessage_submsg_tag;
      32           1 :         msg.oneof.submsg.foo = 1234;
      33             : 
      34           1 :         if (!pb_encode(&ostream, MyMessage_fields, &msg))
      35             :         {
      36           0 :             fprintf(stderr, "pb_encode() failed: %s\n", PB_GET_ERROR(&ostream));
      37           0 :             return 1;
      38             :         }
      39             : 
      40           1 :         msglen = ostream.bytes_written;
      41           1 :         TEST(msglen > 0);
      42             :     }
      43             : 
      44             :     {
      45           1 :         pb_istream_t istream = pb_istream_from_buffer(buf, msglen);
      46           1 :         MyMessage msg = MyMessage_init_zero;
      47           1 :         msg.cb_oneof.funcs.decode = msg_callback;
      48             : 
      49           1 :         if (!pb_decode(&istream, MyMessage_fields, &msg))
      50             :         {
      51           0 :             fprintf(stderr, "pb_decode() failed: %s\n", PB_GET_ERROR(&istream));
      52           0 :             return 1;
      53             :         }
      54             :     }
      55             : 
      56           1 :     return status;
      57             : }