| #include "test.pb.h" |
| #include <unittests.h> |
| #include <pb_encode.h> |
| #include <pb_decode.h> |
| |
| static bool write_array(pb_ostream_t *stream, const pb_field_t *field, void * const *arg) |
| { |
| int i; |
| for (i = 0; i < 5; i++) |
| { |
| if (!pb_encode_tag_for_field(stream, field)) |
| return false; |
| if (!pb_encode_varint(stream, 1000 + i)) |
| return false; |
| } |
| |
| return true; |
| } |
| |
| static bool read_array(pb_istream_t *stream, const pb_field_t *field, void **arg) |
| { |
| uint32_t i; |
| int *sum = *arg; |
| |
| if (!pb_decode_varint32(stream, &i)) |
| return false; |
| |
| *sum += i; |
| |
| return true; |
| } |
| |
| int main() |
| { |
| int status = 0; |
| pb_byte_t buf[128] = {0}; |
| pb_size_t msglen; |
| |
| { |
| MainMessage msg = MainMessage_init_zero; |
| pb_ostream_t stream = pb_ostream_from_buffer(buf, sizeof(buf)); |
| msg.submsg.foo.funcs.encode = &write_array; |
| TEST(pb_encode(&stream, MainMessage_fields, &msg)); |
| msglen = stream.bytes_written; |
| } |
| |
| { |
| MainMessage msg = MainMessage_init_zero; |
| pb_istream_t stream = pb_istream_from_buffer(buf, msglen); |
| int sum = 0; |
| msg.submsg.foo.funcs.decode = &read_array; |
| msg.submsg.foo.arg = ∑ |
| TEST(pb_decode(&stream, MainMessage_fields, &msg)); |
| TEST(sum == 1000 + 1001 + 1002 + 1003 + 1004); |
| } |
| |
| return status; |
| } |
| |