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, const void *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, const void *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, const void *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, const void *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 int main()
  45 {
  46     uint8_t buffer[1024];
  47     pb_ostream_t stream = pb_ostream_from_buffer(buffer, 1024);
  48     TestMessage testmessage = {};
  49
  50     testmessage.stringvalue.funcs.encode = &encode_string;
  51     testmessage.int32value.funcs.encode = &encode_int32;
  52     testmessage.fixed32value.funcs.encode = &encode_fixed32;
  53     testmessage.fixed64value.funcs.encode = &encode_fixed64;
  54
  55     testmessage.has_submsg = true;
  56     testmessage.submsg.stringvalue.funcs.encode = &encode_string;
  57     testmessage.submsg.int32value.funcs.encode = &encode_int32;
  58     testmessage.submsg.fixed32value.funcs.encode = &encode_fixed32;
  59     testmessage.submsg.fixed64value.funcs.encode = &encode_fixed64;
  60
  61     if (!pb_encode(&stream, TestMessage_fields, &testmessage))
  62         return 1;
  63
  64     if (fwrite(buffer, stream.bytes_written, 1, stdout) != 1)
  65         return 2;
  66
  67     return 0;
  68 }