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 } Person_PhoneType;
  13
  14 typedef struct {
  15     char number[40];
  16     bool has_type;
  17     Person_PhoneType type;
  18 } Person_PhoneNumber;
  19
  20 typedef struct {
  21     char name[40];
  22     int32_t id;
  23     bool has_email;
  24     char email[40];
  25     size_t phone_size;
  26     Person_PhoneNumber phone[5];
  27 } Person;
  28
  29 /* Field descriptions */
  30 #define membersize(st, m) (sizeof ((st*)0)->m)
  31
  32 const Person_PhoneType Person_PhoneType_type_default = Person_PhoneType_HOME;
  33
  34 const pb_field_t Person_PhoneNumber_fields[] = {
  35     {1, PB_HTYPE_REQUIRED | PB_LTYPE_STRING,
  36         offsetof(Person_PhoneNumber, number), 0,
  37         membersize(Person_PhoneNumber, number), 0, 0},
  38
  39     {2, PB_HTYPE_OPTIONAL | PB_LTYPE_ENUM,
  40         offsetof(Person_PhoneNumber, type),
  41         offsetof(Person_PhoneNumber, has_type),
  42         membersize(Person_PhoneNumber, type), 0,
  43         &Person_PhoneType_type_default},
  44
  45     PB_LAST_FIELD
  46 };
  47
  48 const pb_field_t Person_fields[] = {
  49     {1, PB_HTYPE_REQUIRED | PB_LTYPE_STRING,
  50         offsetof(Person, name), 0,
  51         membersize(Person, name), 0, 0},
  52
  53     {2, PB_HTYPE_REQUIRED | PB_LTYPE_INT32,
  54         offsetof(Person, id), 0,
  55         membersize(Person, id), 0, 0},
  56
  57     {3, PB_HTYPE_OPTIONAL | PB_LTYPE_STRING,
  58         offsetof(Person, email),
  59         offsetof(Person, has_email),
  60         membersize(Person, email), 0, 0},
  61
  62     {4, PB_HTYPE_ARRAY | PB_LTYPE_SUBMESSAGE,
  63         offsetof(Person, phone),
  64         offsetof(Person, phone_size),
  65         membersize(Person, phone[0]),
  66         membersize(Person, phone) / membersize(Person, phone[0]),
  67         Person_PhoneNumber_fields},
  68
  69     PB_LAST_FIELD
  70 };
  71
  72 /* And now, the actual test program */
  73
  74 bool print_person(pb_istream_t *stream)
  75 {
  76     int i;
  77     Person person;
  78
  79     if (!pb_decode(stream, Person_fields, &person))
  80         return false;
  81
  82     printf("Person: name '%s' id '%d' email '%s'\n", person.name, person.id, person.email);
  83
  84     for (i = 0; i < person.phone_size; i++)
  85     {
  86         Person_PhoneNumber *phone = &person.phone[i];
  87         printf("PhoneNumber: number '%s' type '%d'\n", phone->number, phone->type);
  88     }
  89
  90     return true;
  91 }
  92
  93 int main()
  94 {
  95     uint8_t buffer[512];
  96     size_t size = fread(buffer, 1, 512, stdin);
  97
  98     pb_istream_t stream = pb_istream_from_buffer(buffer, size);
  99     if (!print_person(&stream))
 100         printf("Parsing failed.\n");
 101
 102     return 0;
 103 }