tests/test_decode1.c

   1 #include <stdio.h>
   2 #include <string.h>
   3 #include <stddef.h>
   4 #include "pb_decode.h"
   5
   6 /* Structures for "Person" message */
   7
   8 typedef enum {
   9     Person_PhoneType_MOBILE = 0,
  10     Person_PhoneType_HOME = 1,
  11     Person_PhoneType_WORK = 2,
  12
  13     _Person_PhoneType_size = 0xFFFFFFFF // Force 32-bit enum
  14 } Person_PhoneType;
  15
  16 typedef struct {
  17     char number[40];
  18     bool has_type;
  19     Person_PhoneType type;
  20 } Person_PhoneNumber;
  21
  22 typedef struct {
  23     char name[40];
  24     int32_t id;
  25     bool has_email;
  26     char email[40];
  27
  28     pb_callback_t phone;
  29 } Person;
  30
  31 /* Field descriptions */
  32 #define membersize(st, m) (sizeof ((st*)0)->m)
  33
  34 const pb_field_t Person_PhoneNumber_fields[] = {
  35     {1, offsetof(Person_PhoneNumber, number), PB_ACT_STRING, membersize(Person_PhoneNumber, number)},
  36     {2, offsetof(Person_PhoneNumber, has_type), PB_ACT_HAS, membersize(Person_PhoneNumber, has_type)},
  37     {2, offsetof(Person_PhoneNumber, type), PB_ACT_UINT32, membersize(Person_PhoneNumber, type)},
  38     PB_LAST_FIELD
  39 };
  40
  41 const pb_field_t Person_fields[] = {
  42     {1, offsetof(Person, name), PB_ACT_STRING, membersize(Person, name)},
  43     {2, offsetof(Person, id), PB_ACT_INT32, membersize(Person, id)},
  44     {3, offsetof(Person, email), PB_ACT_STRING, membersize(Person, email)},
  45     {4, offsetof(Person, phone), PB_ACT_SUBMESSAGE, membersize(Person, phone)}
  46 };
  47
  48 /* Default value descriptions */
  49 #define Person_PhoneNumber_default {"", false, Person_PhoneType_HOME};
  50 #define Person_default {"", 0, false, "", {{0},0}};
  51
  52 /* And now, the actual test program */
  53
  54 bool print_phonenumber(pb_istream_t *stream, const pb_field_t *field, void *arg)
  55 {
  56     Person_PhoneNumber x = Person_PhoneNumber_default;
  57     if (!pb_decode(stream, Person_PhoneNumber_fields, &x))
  58         return false;
  59
  60     printf("PhoneNumber: number '%s' type '%d'\n", x.number, x.type);
  61     return true;
  62 }
  63
  64 bool print_person(pb_istream_t *stream)
  65 {
  66     Person x = Person_default;
  67     x.phone.funcs.decode = &print_phonenumber;
  68
  69     if (!pb_decode(stream, Person_fields, &x))
  70         return false;
  71
  72     printf("Person: name '%s' id '%d' email '%s'\n", x.name, x.id, x.email);
  73     return true;
  74 }
  75
  76 bool my_read(pb_istream_t *stream, char *buf, size_t count)
  77 {
  78     char *source = (char*)stream->state;
  79
  80     if (!stream->bytes_left)
  81         return false;
  82
  83     if (buf != NULL)
  84     {
  85         memcpy(buf, source, count);
  86     }
  87
  88     stream->state = source + count;
  89     return true;
  90 }
  91
  92 int main()
  93 {
  94     char buffer[512];
  95     size_t size = fread(buffer, 1, 512, stdin);
  96
  97     pb_istream_t stream = {&my_read, buffer, size};
  98     if (!print_person(&stream))
  99         printf("Parsing failed.\n");
 100
 101     return 0;
 102 }