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->static_rep_submsg_count = 2;
29 msg->static_rep_submsg[1].dynamic_str = "abc";
30 msg->has_static_opt_submsg = true;
31 msg->dynamic_submsg = &msg->static_req_submsg;
33 msg->extensions = &ext1;
34 ext1.type = &dynamic_ext;
35 ext1.dest = &msg->static_req_submsg;
37 ext2.type = &static_ext;
38 ext2.dest = &msg->static_req_submsg;
42 /* Basic fields, nested submessages, extensions */
43 static bool test_TestMessage()
48 /* Construct a message with various fields filled in */
50 TestMessage msg = TestMessage_init_zero;
53 fill_TestMessage(&msg);
55 stream = pb_ostream_from_buffer(buffer, sizeof(buffer));
56 if (!pb_encode(&stream, TestMessage_fields, &msg))
58 fprintf(stderr, "Encode failed: %s\n", PB_GET_ERROR(&stream));
61 msgsize = stream.bytes_written;
64 /* Output encoded message for debug */
65 SET_BINARY_MODE(stdout);
66 fwrite(buffer, 1, msgsize, stdout);
68 /* Decode memory using dynamic allocation */
70 TestMessage msg = TestMessage_init_zero;
74 msg.extensions = &ext1;
75 ext1.type = &dynamic_ext;
78 ext2.type = &static_ext;
79 ext2.dest = &ext2_dest;
82 stream = pb_istream_from_buffer(buffer, msgsize);
83 if (!pb_decode(&stream, TestMessage_fields, &msg))
85 fprintf(stderr, "Decode failed: %s\n", PB_GET_ERROR(&stream));
89 /* Make sure it encodes back to same data */
92 pb_ostream_t ostream = pb_ostream_from_buffer(buffer2, sizeof(buffer2));
93 TEST(pb_encode(&ostream, TestMessage_fields, &msg));
94 TEST(ostream.bytes_written == msgsize);
95 TEST(memcmp(buffer, buffer2, msgsize) == 0);
98 /* Make sure that malloc counters work */
99 TEST(get_alloc_count() > 0);
101 /* Make sure that pb_release releases everything */
102 pb_release(TestMessage_fields, &msg);
103 TEST(get_alloc_count() == 0);
105 /* Check that double-free is a no-op */
106 pb_release(TestMessage_fields, &msg);
107 TEST(get_alloc_count() == 0);
114 static bool test_OneofMessage()
120 pb_ostream_t stream = pb_ostream_from_buffer(buffer, sizeof(buffer));
122 /* Encode first with TestMessage */
124 OneofMessage msg = OneofMessage_init_zero;
125 msg.which_msgs = OneofMessage_msg1_tag;
127 fill_TestMessage(&msg.msgs.msg1);
129 if (!pb_encode(&stream, OneofMessage_fields, &msg))
131 fprintf(stderr, "Encode failed: %s\n", PB_GET_ERROR(&stream));
136 /* Encode second with SubMessage, invoking 'merge' behaviour */
138 OneofMessage msg = OneofMessage_init_zero;
139 msg.which_msgs = OneofMessage_msg2_tag;
142 msg.msgs.msg2.dynamic_str = "ABCD";
145 if (!pb_encode(&stream, OneofMessage_fields, &msg))
147 fprintf(stderr, "Encode failed: %s\n", PB_GET_ERROR(&stream));
151 msgsize = stream.bytes_written;
155 OneofMessage msg = OneofMessage_init_zero;
156 pb_istream_t stream = pb_istream_from_buffer(buffer, msgsize);
157 if (!pb_decode(&stream, OneofMessage_fields, &msg))
159 fprintf(stderr, "Decode failed: %s\n", PB_GET_ERROR(&stream));
163 TEST(msg.first == 999);
164 TEST(msg.which_msgs == OneofMessage_msg2_tag);
165 TEST(msg.msgs.msg2.dynamic_str);
166 TEST(strcmp(msg.msgs.msg2.dynamic_str, "ABCD") == 0);
167 TEST(msg.msgs.msg2.dynamic_str_arr == NULL);
168 TEST(msg.msgs.msg2.dynamic_submsg == NULL);
169 TEST(msg.last == 888);
171 pb_release(OneofMessage_fields, &msg);
172 TEST(get_alloc_count() == 0);
173 pb_release(OneofMessage_fields, &msg);
174 TEST(get_alloc_count() == 0);
182 if (test_TestMessage() && test_OneofMessage())