tests/test_encode_callbacks.c

   1 /* Encoding testcase for callback fields */
   2
   3 #include <stdio.h>
   4 #include <string.h>
   5 #include <pb_encode.h>
   6 #include "callbacks.pb.h"
   7
   8 bool encode_string(pb_ostream_t *stream, const pb_field_t *field, void * const *arg)
   9 {
  10     char *str = "Hello world!";
  11
  12     if (!pb_encode_tag_for_field(stream, field))
  13         return false;
  14
  15     return pb_encode_string(stream, (uint8_t*)str, strlen(str));
  16 }
  17
  18 bool encode_int32(pb_ostream_t *stream, const pb_field_t *field, void * const *arg)
  19 {
  20     if (!pb_encode_tag_for_field(stream, field))
  21         return false;
  22
  23     return pb_encode_varint(stream, 42);
  24 }
  25
  26 bool encode_fixed32(pb_ostream_t *stream, const pb_field_t *field, void * const *arg)
  27 {
  28     if (!pb_encode_tag_for_field(stream, field))
  29         return false;
  30
  31     uint32_t value = 42;
  32     return pb_encode_fixed32(stream, &value);
  33 }
  34
  35 bool encode_fixed64(pb_ostream_t *stream, const pb_field_t *field, void * const *arg)
  36 {
  37     if (!pb_encode_tag_for_field(stream, field))
  38         return false;
  39
  40     uint64_t value = 42;
  41     return pb_encode_fixed64(stream, &value);
  42 }
  43
  44 bool encode_repeatedstring(pb_ostream_t *stream, const pb_field_t *field, void * const *arg)
  45 {
  46     char *str[4] = {"Hello world!", "", "Test", "Test2"};
  47     int i;
  48
  49     for (i = 0; i < 4; i++)
  50     {
  51         if (!pb_encode_tag_for_field(stream, field))
  52             return false;
  53
  54         if (!pb_encode_string(stream, (uint8_t*)str[i], strlen(str[i])))
  55             return false;
  56     }
  57     return true;
  58 }
  59
  60 int main()
  61 {
  62     uint8_t buffer[1024];
  63     pb_ostream_t stream = pb_ostream_from_buffer(buffer, 1024);
  64     TestMessage testmessage = {};
  65
  66     testmessage.stringvalue.funcs.encode = &encode_string;
  67     testmessage.int32value.funcs.encode = &encode_int32;
  68     testmessage.fixed32value.funcs.encode = &encode_fixed32;
  69     testmessage.fixed64value.funcs.encode = &encode_fixed64;
  70
  71     testmessage.has_submsg = true;
  72     testmessage.submsg.stringvalue.funcs.encode = &encode_string;
  73     testmessage.submsg.int32value.funcs.encode = &encode_int32;
  74     testmessage.submsg.fixed32value.funcs.encode = &encode_fixed32;
  75     testmessage.submsg.fixed64value.funcs.encode = &encode_fixed64;
  76
  77     testmessage.repeatedstring.funcs.encode = &encode_repeatedstring;
  78
  79     if (!pb_encode(&stream, TestMessage_fields, &testmessage))
  80         return 1;
  81
  82     if (fwrite(buffer, stream.bytes_written, 1, stdout) != 1)
  83         return 2;
  84
  85     return 0;
  86 }