CAN-binder/libs/nanopb/tests/callbacks/decode_callbacks.c

   1 /* Decoding testcase for callback fields.
   2  * Run e.g. ./test_encode_callbacks | ./test_decode_callbacks
   3  */
   4
   5 #include <stdio.h>
   6 #include <pb_decode.h>
   7 #include "callbacks.pb.h"
   8 #include "test_helpers.h"
   9
  10 bool print_string(pb_istream_t *stream, const pb_field_t *field, void **arg)
  11 {
  12     uint8_t buffer[1024] = {0};
  13
  14     /* We could read block-by-block to avoid the large buffer... */
  15     if (stream->bytes_left > sizeof(buffer) - 1)
  16         return false;
  17
  18     if (!pb_read(stream, buffer, stream->bytes_left))
  19         return false;
  20
  21     /* Print the string, in format comparable with protoc --decode.
  22      * Format comes from the arg defined in main().
  23      */
  24     printf((char*)*arg, buffer);
  25     return true;
  26 }
  27
  28 bool print_int32(pb_istream_t *stream, const pb_field_t *field, void **arg)
  29 {
  30     uint64_t value;
  31     if (!pb_decode_varint(stream, &value))
  32         return false;
  33
  34     printf((char*)*arg, (long)value);
  35     return true;
  36 }
  37
  38 bool print_fixed32(pb_istream_t *stream, const pb_field_t *field, void **arg)
  39 {
  40     uint32_t value;
  41     if (!pb_decode_fixed32(stream, &value))
  42         return false;
  43
  44     printf((char*)*arg, (long)value);
  45     return true;
  46 }
  47
  48 bool print_fixed64(pb_istream_t *stream, const pb_field_t *field, void **arg)
  49 {
  50     uint64_t value;
  51     if (!pb_decode_fixed64(stream, &value))
  52         return false;
  53
  54     printf((char*)*arg, (long)value);
  55     return true;
  56 }
  57
  58 int main()
  59 {
  60     uint8_t buffer[1024];
  61     size_t length;
  62     pb_istream_t stream;
  63     /* Note: empty initializer list initializes the struct with all-0.
  64      * This is recommended so that unused callbacks are set to NULL instead
  65      * of crashing at runtime.
  66      */
  67     TestMessage testmessage = {{{NULL}}};
  68
  69     SET_BINARY_MODE(stdin);
  70     length = fread(buffer, 1, 1024, stdin);
  71     stream = pb_istream_from_buffer(buffer, length);
  72
  73     testmessage.submsg.stringvalue.funcs.decode = &print_string;
  74     testmessage.submsg.stringvalue.arg = "submsg {\n  stringvalue: \"%s\"\n";
  75     testmessage.submsg.int32value.funcs.decode = &print_int32;
  76     testmessage.submsg.int32value.arg = "  int32value: %ld\n";
  77     testmessage.submsg.fixed32value.funcs.decode = &print_fixed32;
  78     testmessage.submsg.fixed32value.arg = "  fixed32value: %ld\n";
  79     testmessage.submsg.fixed64value.funcs.decode = &print_fixed64;
  80     testmessage.submsg.fixed64value.arg = "  fixed64value: %ld\n}\n";
  81
  82     testmessage.stringvalue.funcs.decode = &print_string;
  83     testmessage.stringvalue.arg = "stringvalue: \"%s\"\n";
  84     testmessage.int32value.funcs.decode = &print_int32;
  85     testmessage.int32value.arg = "int32value: %ld\n";
  86     testmessage.fixed32value.funcs.decode = &print_fixed32;
  87     testmessage.fixed32value.arg = "fixed32value: %ld\n";
  88     testmessage.fixed64value.funcs.decode = &print_fixed64;
  89     testmessage.fixed64value.arg = "fixed64value: %ld\n";
  90     testmessage.repeatedstring.funcs.decode = &print_string;
  91     testmessage.repeatedstring.arg = "repeatedstring: \"%s\"\n";
  92
  93     if (!pb_decode(&stream, TestMessage_fields, &testmessage))
  94         return 1;
  95
  96     return 0;
  97 }