dcab39a41c
Rationale: it's easy to implement the callback wrong. Doing so introduces io errors when unknown fields are present in the input. If code is not tested with unknown fields, these bugs can remain hidden for long time. Added a special case for the memory buffer stream, where it gives a small speed benefit. Added testcase for skipping fields with test_decode2 implementation. Update issue 37 Status: FixedInGit
84 lines
2.0 KiB
C
84 lines
2.0 KiB
C
/* Same as test_decode1 but reads from stdin directly.
|
|
*/
|
|
|
|
#include <stdio.h>
|
|
#include <pb_decode.h>
|
|
#include "person.pb.h"
|
|
|
|
/* This function is called once from main(), it handles
|
|
the decoding and printing.
|
|
Ugly copy-paste from test_decode1.c. */
|
|
bool print_person(pb_istream_t *stream)
|
|
{
|
|
int i;
|
|
Person person;
|
|
|
|
if (!pb_decode(stream, Person_fields, &person))
|
|
return false;
|
|
|
|
/* 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_count; i++)
|
|
{
|
|
Person_PhoneNumber *phone = &person.phone[i];
|
|
printf("phone {\n");
|
|
printf(" number: \"%s\"\n", phone->number);
|
|
|
|
if (phone->has_type)
|
|
{
|
|
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;
|
|
}
|
|
|
|
/* This binds the pb_istream_t to stdin */
|
|
bool callback(pb_istream_t *stream, uint8_t *buf, size_t count)
|
|
{
|
|
FILE *file = (FILE*)stream->state;
|
|
bool status;
|
|
|
|
status = (fread(buf, 1, count, file) == count);
|
|
|
|
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: %s\n", PB_GET_ERROR(&stream));
|
|
return 1;
|
|
} else {
|
|
return 0;
|
|
}
|
|
}
|