1 #include "test.pb.h" 2 #include <unittests.h> 3 #include <pb_encode.h> 4 #include <pb_decode.h> 5 6 static bool write_array(pb_ostream_t *stream, const pb_field_t *field, void * const *arg) 7 { 8 int i; 9 for (i = 0; i < 5; i++) 10 { 11 if (!pb_encode_tag_for_field(stream, field)) 12 return false; 13 if (!pb_encode_varint(stream, 1000 + i)) 14 return false; 15 } 16 17 return true; 18 } 19 20 static bool read_array(pb_istream_t *stream, const pb_field_t *field, void **arg) 21 { 22 uint32_t i; 23 int *sum = *arg; 24 25 if (!pb_decode_varint32(stream, &i)) 26 return false; 27 28 *sum += i; 29 30 return true; 31 } 32 33 int main() 34 { 35 int status = 0; 36 pb_byte_t buf[128] = {0}; 37 pb_size_t msglen; 38 39 { 40 MainMessage msg = MainMessage_init_zero; 41 pb_ostream_t stream = pb_ostream_from_buffer(buf, sizeof(buf)); 42 msg.submsg.foo.funcs.encode = &write_array; 43 TEST(pb_encode(&stream, MainMessage_fields, &msg)); 44 msglen = stream.bytes_written; 45 } 46 47 { 48 MainMessage msg = MainMessage_init_zero; 49 pb_istream_t stream = pb_istream_from_buffer(buf, msglen); 50 int sum = 0; 51 msg.submsg.foo.funcs.decode = &read_array; 52 msg.submsg.foo.arg = ∑ 53 TEST(pb_decode(&stream, MainMessage_fields, &msg)); 54 TEST(sum == 1000 + 1001 + 1002 + 1003 + 1004); 55 } 56 57 return status; 58 } 59 60
