1 /* Make sure that all fields are freed in various scenarios. */
5 #include <malloc_wrappers.h>
7 #include <test_helpers.h>
8 #include "mem_release.pb.h"
10 #define TEST(x) if (!(x)) { \
11 fprintf(stderr, "Test " #x " on line %d failed.\n", __LINE__); \
15 static char *test_str_arr[] = {"1", "2", ""};
16 static SubMessage test_msg_arr[] = {SubMessage_init_zero, SubMessage_init_zero};
17 static pb_extension_t ext1, ext2;
19 static void fill_TestMessage(TestMessage *msg)
21 msg->static_req_submsg.dynamic_str = "12345";
22 msg->static_req_submsg.dynamic_str_arr_count = 3;
23 msg->static_req_submsg.dynamic_str_arr = test_str_arr;
24 msg->static_req_submsg.dynamic_submsg_count = 2;
25 msg->static_req_submsg.dynamic_submsg = test_msg_arr;
26 msg->static_req_submsg.dynamic_submsg[1].dynamic_str = "abc";
27 msg->static_opt_submsg.dynamic_str = "abc";
28 msg->has_static_opt_submsg = true;
29 msg->dynamic_submsg = &msg->static_req_submsg;
31 msg->extensions = &ext1;
32 ext1.type = &dynamic_ext;
33 ext1.dest = &msg->static_req_submsg;
35 ext2.type = &static_ext;
36 ext2.dest = &msg->static_req_submsg;
40 /* Basic fields, nested submessages, extensions */
41 static bool test_TestMessage()
46 /* Construct a message with various fields filled in */
48 TestMessage msg = TestMessage_init_zero;
51 fill_TestMessage(&msg);
53 stream = pb_ostream_from_buffer(buffer, sizeof(buffer));
54 if (!pb_encode(&stream, TestMessage_fields, &msg))
56 fprintf(stderr, "Encode failed: %s\n", PB_GET_ERROR(&stream));
59 msgsize = stream.bytes_written;
62 /* Output encoded message for debug */
63 SET_BINARY_MODE(stdout);
64 fwrite(buffer, 1, msgsize, stdout);
66 /* Decode memory using dynamic allocation */
68 TestMessage msg = TestMessage_init_zero;
72 msg.extensions = &ext1;
73 ext1.type = &dynamic_ext;
76 ext2.type = &static_ext;
77 ext2.dest = &ext2_dest;
80 stream = pb_istream_from_buffer(buffer, msgsize);
81 if (!pb_decode(&stream, TestMessage_fields, &msg))
83 fprintf(stderr, "Decode failed: %s\n", PB_GET_ERROR(&stream));
87 /* Make sure it encodes back to same data */
90 pb_ostream_t ostream = pb_ostream_from_buffer(buffer2, sizeof(buffer2));
91 TEST(pb_encode(&ostream, TestMessage_fields, &msg));
92 TEST(ostream.bytes_written == msgsize);
93 TEST(memcmp(buffer, buffer2, msgsize) == 0);
96 /* Make sure that malloc counters work */
97 TEST(get_alloc_count() > 0);
99 /* Make sure that pb_release releases everything */
100 pb_release(TestMessage_fields, &msg);
101 TEST(get_alloc_count() == 0);
103 /* Check that double-free is a no-op */
104 pb_release(TestMessage_fields, &msg);
105 TEST(get_alloc_count() == 0);
112 static bool test_OneofMessage()
118 pb_ostream_t stream = pb_ostream_from_buffer(buffer, sizeof(buffer));
120 /* Encode first with TestMessage */
122 OneofMessage msg = OneofMessage_init_zero;
123 msg.which_msgs = OneofMessage_msg1_tag;
125 fill_TestMessage(&msg.msgs.msg1);
127 if (!pb_encode(&stream, OneofMessage_fields, &msg))
129 fprintf(stderr, "Encode failed: %s\n", PB_GET_ERROR(&stream));
134 /* Encode second with SubMessage, invoking 'merge' behaviour */
136 OneofMessage msg = OneofMessage_init_zero;
137 msg.which_msgs = OneofMessage_msg2_tag;
140 msg.msgs.msg2.dynamic_str = "ABCD";
143 if (!pb_encode(&stream, OneofMessage_fields, &msg))
145 fprintf(stderr, "Encode failed: %s\n", PB_GET_ERROR(&stream));
149 msgsize = stream.bytes_written;
153 OneofMessage msg = OneofMessage_init_zero;
154 pb_istream_t stream = pb_istream_from_buffer(buffer, msgsize);
155 if (!pb_decode(&stream, OneofMessage_fields, &msg))
157 fprintf(stderr, "Decode failed: %s\n", PB_GET_ERROR(&stream));
161 TEST(msg.first == 999);
162 TEST(msg.which_msgs == OneofMessage_msg2_tag);
163 TEST(msg.msgs.msg2.dynamic_str);
164 TEST(strcmp(msg.msgs.msg2.dynamic_str, "ABCD") == 0);
165 TEST(msg.msgs.msg2.dynamic_str_arr == NULL);
166 TEST(msg.msgs.msg2.dynamic_submsg == NULL);
167 TEST(msg.last == 888);
169 pb_release(OneofMessage_fields, &msg);
170 TEST(get_alloc_count() == 0);
171 pb_release(OneofMessage_fields, &msg);
172 TEST(get_alloc_count() == 0);
180 if (test_TestMessage() && test_OneofMessage())