-#include <stdio.h>
-#include <string.h>
-#include <stddef.h>
-#include "pb_decode.h"
-
-/* Structures for "Person" message */
-
-typedef enum {
- Person_PhoneType_MOBILE = 0,
- Person_PhoneType_HOME = 1,
- Person_PhoneType_WORK = 2
-} Person_PhoneType;
-
-typedef struct {
- char number[40];
- bool has_type;
- Person_PhoneType type;
-} Person_PhoneNumber;
-
-typedef struct {
- char name[40];
- int32_t id;
- bool has_email;
- char email[40];
- size_t phone_size;
- Person_PhoneNumber phone[5];
-} Person;
+/* A very simple decoding test case, using person.proto.
+ * Produces output compatible with protoc --decode.
+ * Reads the encoded data from stdin and prints the values
+ * to stdout as text.
+ *
+ * Run e.g. ./test_encode1 | ./test_decode1
+ */
-/* Field descriptions */
-
-
-const Person_PhoneType Person_PhoneNumber_type_default = Person_PhoneType_HOME;
-
-const pb_field_t Person_PhoneNumber_fields[] = {
- {1, PB_HTYPE_REQUIRED | PB_LTYPE_STRING,
- offsetof(Person_PhoneNumber, number), 0,
- pb_membersize(Person_PhoneNumber, number), 0, 0},
-
- {2, PB_HTYPE_OPTIONAL | PB_LTYPE_VARINT,
- pb_delta(Person_PhoneNumber, type, number),
- pb_delta(Person_PhoneNumber, has_type, type),
- pb_membersize(Person_PhoneNumber, type), 0,
- &Person_PhoneNumber_type_default},
-
- PB_LAST_FIELD
-};
-
-const pb_field_t Person_fields[] = {
- {1, PB_HTYPE_REQUIRED | PB_LTYPE_STRING,
- offsetof(Person, name), 0,
- pb_membersize(Person, name), 0, 0},
-
- {2, PB_HTYPE_REQUIRED | PB_LTYPE_VARINT,
- pb_delta(Person, id, name), 0,
- pb_membersize(Person, id), 0, 0},
-
- {3, PB_HTYPE_OPTIONAL | PB_LTYPE_STRING,
- offsetof(Person, email) - offsetof(Person, id),
- pb_delta(Person, has_email, email),
- pb_membersize(Person, email), 0, 0},
-
- {4, PB_HTYPE_ARRAY | PB_LTYPE_SUBMESSAGE,
- offsetof(Person, phone) - offsetof(Person, email),
- pb_delta(Person, phone_size, phone),
- pb_membersize(Person, phone[0]),
- pb_arraysize(Person, phone),
- Person_PhoneNumber_fields},
-
- PB_LAST_FIELD
-};
-
-/* And now, the actual test program */
+#include <stdio.h>
+#include <pb_decode.h>
+#include "person.pb.h"
+/* This function is called once from main(), it handles
+ the decoding and printing. */
bool print_person(pb_istream_t *stream)
{
int i;
if (!pb_decode(stream, Person_fields, &person))
return false;
- printf("Person: name '%s' id '%d' email '%s'\n", person.name, person.id, person.email);
+ /* Now the decoding is done, rest is just to print stuff out. */
+
+ printf("name: \"%s\"\n", person.name);
+ printf("id: %ld\n", (long)person.id);
+
+ if (person.has_email)
+ printf("email: \"%s\"\n", person.email);
- for (i = 0; i < person.phone_size; i++)
+ for (i = 0; i < person.phone_count; i++)
{
Person_PhoneNumber *phone = &person.phone[i];
- printf("PhoneNumber: number '%s' type '%d'\n", phone->number, phone->type);
+ printf("phone {\n");
+ printf(" number: \"%s\"\n", phone->number);
+
+ switch (phone->type)
+ {
+ case Person_PhoneType_WORK:
+ printf(" type: WORK\n");
+ break;
+
+ case Person_PhoneType_HOME:
+ printf(" type: HOME\n");
+ break;
+
+ case Person_PhoneType_MOBILE:
+ printf(" type: MOBILE\n");
+ break;
+ }
+ printf("}\n");
}
return true;
}
-int main()
+/* This binds the pb_istream_t to stdin */
+bool callback(pb_istream_t *stream, uint8_t *buf, size_t count)
{
- uint8_t buffer[512];
- size_t size = fread(buffer, 1, 512, stdin);
+ FILE *file = (FILE*)stream->state;
+ bool status;
+
+ if (buf == NULL)
+ {
+ /* Skipping data */
+ while (count-- && fgetc(file) != EOF);
+ return count == 0;
+ }
+
+ status = (fread(buf, 1, count, file) == count);
- pb_istream_t stream = pb_istream_from_buffer(buffer, size);
+ if (feof(file))
+ stream->bytes_left = 0;
+
+ return status;
+}
+
+int main()
+{
+ /* Maximum size is specified to prevent infinite length messages from
+ * hanging this in the fuzz test.
+ */
+ pb_istream_t stream = {&callback, stdin, 10000};
if (!print_person(&stream))
+ {
printf("Parsing failed.\n");
-
- return 0;
+ return 1;
+ } else {
+ return 0;
+ }
}