| #include <pb_encode.h> |
| #include <pb_decode.h> |
| #include "extensions.pb.h" |
| #include "unittests.h" |
| |
| int main() |
| { |
| uint8_t buffer[256]; |
| int status = 0; |
| size_t msglen; |
| |
| { |
| pb_ostream_t stream; |
| SimpleMessage msg = SimpleMessage_init_zero; |
| ExtMessage extmsg = ExtMessage_init_zero; |
| pb_extension_t ext = pb_extension_init_zero; |
| |
| COMMENT("Encoding message"); |
| msg.has_number = true; |
| msg.number = 1234; |
| extmsg.has_second_number = true; |
| extmsg.second_number = 5678; |
| |
| msg.extensions = &ext; |
| ext.type = &ExtMessage_ext_message_ext; |
| ext.dest = &extmsg; |
| ext.next = NULL; |
| |
| stream = pb_ostream_from_buffer(buffer, sizeof(buffer)); |
| TEST(pb_encode(&stream, SimpleMessage_fields, &msg)); |
| msglen = stream.bytes_written; |
| TEST(msglen == 9); /* 3 for number, 3 for submsg tag+len, 3 for second_number */ |
| TEST(buffer[0] == 0x08); /* varint 1 */ |
| TEST(buffer[1] == 0xD2); /* number = 1234 */ |
| TEST(buffer[2] == 0x09); |
| TEST(buffer[3] == 0xE2); /* submsg 1100 */ |
| TEST(buffer[4] == 0x44); |
| TEST(buffer[5] == 0x03); /* length 3 */ |
| TEST(buffer[6] == 0x08); /* varint 1 */ |
| TEST(buffer[7] == 0xae); /* second_number = 5678 */ |
| TEST(buffer[8] == 0x2c); |
| } |
| |
| { |
| pb_istream_t stream; |
| SimpleMessage msg = SimpleMessage_init_zero; |
| ExtMessage extmsg = ExtMessage_init_zero; |
| pb_extension_t ext = pb_extension_init_zero; |
| |
| COMMENT("Decoding message"); |
| |
| msg.extensions = &ext; |
| ext.type = &ExtMessage_ext_message_ext; |
| ext.dest = &extmsg; |
| ext.next = NULL; |
| ext.found = false; |
| |
| stream = pb_istream_from_buffer(buffer, msglen); |
| TEST(pb_decode(&stream, SimpleMessage_fields, &msg)); |
| TEST(msg.has_number); |
| TEST(msg.number == 1234); |
| TEST(ext.found); |
| TEST(extmsg.has_second_number); |
| TEST(extmsg.second_number == 5678); |
| } |
| |
| return status; |
| } |