5 #include "unittestproto.pb.h"
7 bool streamcallback(pb_ostream_t *stream, const uint8_t *buf, size_t count)
9 /* Allow only 'x' to be written */
18 bool fieldcallback(pb_ostream_t *stream, const pb_field_t *field, const void *arg)
21 if (!pb_encode_tag_for_field(stream, field))
23 return pb_encode_varint(stream, value);
26 bool crazyfieldcallback(pb_ostream_t *stream, const pb_field_t *field, const void *arg)
28 /* This callback writes different amount of data the second time. */
29 uint32_t *state = (uint32_t*)arg;
31 if (!pb_encode_tag_for_field(stream, field))
33 return pb_encode_varint(stream, *state);
36 /* Check that expression x writes data y.
37 * Y is a string, which may contain null bytes. Null terminator is ignored.
39 #define WRITES(x, y) \
40 memset(buffer, 0xAA, sizeof(buffer)), \
41 s = pb_ostream_from_buffer(buffer, sizeof(buffer)), \
43 memcmp(buffer, y, sizeof(y) - 1) == 0 && \
44 buffer[sizeof(y) - 1] == 0xAA
51 uint8_t buffer1[] = "foobartest1234";
52 uint8_t buffer2[sizeof(buffer1)];
53 pb_ostream_t stream = pb_ostream_from_buffer(buffer2, sizeof(buffer1));
55 COMMENT("Test pb_write and pb_ostream_t");
56 TEST(pb_write(&stream, buffer1, sizeof(buffer1)));
57 TEST(memcmp(buffer1, buffer2, sizeof(buffer1)) == 0);
58 TEST(!pb_write(&stream, buffer1, 1));
59 TEST(stream.bytes_written == sizeof(buffer1));
63 uint8_t buffer1[] = "xxxxxxx";
64 pb_ostream_t stream = {&streamcallback, 0, SIZE_MAX, 0};
66 COMMENT("Test pb_write with custom callback");
67 TEST(pb_write(&stream, buffer1, 5));
69 TEST(!pb_write(&stream, buffer1, 5));
76 COMMENT("Test pb_encode_varint")
77 TEST(WRITES(pb_encode_varint(&s, 0), "\0"));
78 TEST(WRITES(pb_encode_varint(&s, 1), "\1"));
79 TEST(WRITES(pb_encode_varint(&s, 0x7F), "\x7F"));
80 TEST(WRITES(pb_encode_varint(&s, 0x80), "\x80\x01"));
81 TEST(WRITES(pb_encode_varint(&s, UINT32_MAX), "\xFF\xFF\xFF\xFF\x0F"));
82 TEST(WRITES(pb_encode_varint(&s, UINT64_MAX), "\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF\x01"));
89 COMMENT("Test pb_encode_tag")
90 TEST(WRITES(pb_encode_tag(&s, PB_WT_STRING, 5), "\x2A"));
91 TEST(WRITES(pb_encode_tag(&s, PB_WT_VARINT, 99), "\x98\x06"));
97 pb_field_t field = {10, PB_LTYPE_SVARINT};
99 COMMENT("Test pb_encode_tag_for_field")
100 TEST(WRITES(pb_encode_tag_for_field(&s, &field), "\x50"));
102 field.type = PB_LTYPE_FIXED;
104 TEST(WRITES(pb_encode_tag_for_field(&s, &field), "\x51"));
106 field.type = PB_LTYPE_STRING;
107 TEST(WRITES(pb_encode_tag_for_field(&s, &field), "\x52"));
109 field.type = PB_LTYPE_FIXED;
111 TEST(WRITES(pb_encode_tag_for_field(&s, &field), "\x55"));
118 COMMENT("Test pb_encode_string")
119 TEST(WRITES(pb_encode_string(&s, (const uint8_t*)"abcd", 4), "\x04""abcd"));
120 TEST(WRITES(pb_encode_string(&s, (const uint8_t*)"abcd\x00", 5), "\x05""abcd\x00"));
121 TEST(WRITES(pb_encode_string(&s, (const uint8_t*)"", 0), "\x00"));
129 int32_t max = INT32_MAX;
130 int32_t min = INT32_MIN;
131 int64_t lmax = INT64_MAX;
132 int64_t lmin = INT64_MIN;
133 pb_field_t field = {1, PB_LTYPE_VARINT, 0, 0, sizeof(value)};
135 COMMENT("Test pb_enc_varint and pb_enc_svarint")
136 TEST(WRITES(pb_enc_varint(&s, &field, &value), "\x01"));
137 TEST(WRITES(pb_enc_svarint(&s, &field, &svalue), "\x01"));
138 TEST(WRITES(pb_enc_svarint(&s, &field, &value), "\x02"));
140 field.data_size = sizeof(max);
141 TEST(WRITES(pb_enc_svarint(&s, &field, &max), "\xfe\xff\xff\xff\x0f"));
142 TEST(WRITES(pb_enc_svarint(&s, &field, &min), "\xff\xff\xff\xff\x0f"));
144 field.data_size = sizeof(lmax);
145 TEST(WRITES(pb_enc_svarint(&s, &field, &lmax), "\xFE\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF\x01"));
146 TEST(WRITES(pb_enc_svarint(&s, &field, &lmin), "\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF\xFF\x01"));
152 pb_field_t field = {1, PB_LTYPE_FIXED, 0, 0, sizeof(float)};
156 COMMENT("Test pb_enc_fixed using float")
158 TEST(WRITES(pb_enc_fixed(&s, &field, &fvalue), "\x00\x00\x00\x00"))
160 TEST(WRITES(pb_enc_fixed(&s, &field, &fvalue), "\x00\x00\xc6\x42"))
161 fvalue = -12345678.0f;
162 TEST(WRITES(pb_enc_fixed(&s, &field, &fvalue), "\x4e\x61\x3c\xcb"))
164 COMMENT("Test pb_enc_fixed using double")
165 field.data_size = sizeof(double);
167 TEST(WRITES(pb_enc_fixed(&s, &field, &dvalue), "\x00\x00\x00\x00\x00\x00\x00\x00"))
169 TEST(WRITES(pb_enc_fixed(&s, &field, &dvalue), "\x00\x00\x00\x00\x00\xc0\x58\x40"))
170 dvalue = -12345678.0;
171 TEST(WRITES(pb_enc_fixed(&s, &field, &dvalue), "\x00\x00\x00\xc0\x29\x8c\x67\xc1"))
177 struct { size_t size; uint8_t bytes[5]; } value = {5, {'x', 'y', 'z', 'z', 'y'}};
179 COMMENT("Test pb_enc_bytes")
180 TEST(WRITES(pb_enc_bytes(&s, NULL, &value), "\x05xyzzy"))
182 TEST(WRITES(pb_enc_bytes(&s, NULL, &value), "\x00"))
188 char value[] = "xyzzy";
190 COMMENT("Test pb_enc_string")
191 TEST(WRITES(pb_enc_string(&s, NULL, &value), "\x05xyzzy"))
193 TEST(WRITES(pb_enc_string(&s, NULL, &value), "\x00"))
199 IntegerArray msg = {5, {1, 2, 3, 4, 5}};
201 COMMENT("Test pb_encode with int32 array")
203 TEST(WRITES(pb_encode(&s, IntegerArray_fields, &msg), "\x0A\x05\x01\x02\x03\x04\x05"))
206 TEST(WRITES(pb_encode(&s, IntegerArray_fields, &msg), ""))
209 TEST(!pb_encode(&s, IntegerArray_fields, &msg))
215 FloatArray msg = {1, {99.0f}};
217 COMMENT("Test pb_encode with float array")
219 TEST(WRITES(pb_encode(&s, FloatArray_fields, &msg),
220 "\x0A\x04\x00\x00\xc6\x42"))
223 TEST(WRITES(pb_encode(&s, FloatArray_fields, &msg), ""))
226 TEST(!pb_encode(&s, FloatArray_fields, &msg))
234 msg.data.funcs.encode = &fieldcallback;
236 COMMENT("Test pb_encode with callback field.")
237 TEST(WRITES(pb_encode(&s, CallbackArray_fields, &msg), "\x08\x55"))
243 IntegerContainer msg = {{5, {1,2,3,4,5}}};
245 COMMENT("Test pb_encode with packed array in a submessage.")
246 TEST(WRITES(pb_encode(&s, IntegerContainer_fields, &msg),
247 "\x0A\x07\x0A\x05\x01\x02\x03\x04\x05"))
253 CallbackContainer msg;
254 CallbackContainerContainer msg2;
257 msg.submsg.data.funcs.encode = &fieldcallback;
258 msg2.submsg.submsg.data.funcs.encode = &fieldcallback;
260 COMMENT("Test pb_encode with callback field in a submessage.")
261 TEST(WRITES(pb_encode(&s, CallbackContainer_fields, &msg), "\x0A\x02\x08\x55"))
262 TEST(WRITES(pb_encode(&s, CallbackContainerContainer_fields, &msg2),
263 "\x0A\x04\x0A\x02\x08\x55"))
265 /* Misbehaving callback: varying output between calls */
266 msg.submsg.data.funcs.encode = &crazyfieldcallback;
267 msg.submsg.data.arg = &state;
268 msg2.submsg.submsg.data.funcs.encode = &crazyfieldcallback;
269 msg2.submsg.submsg.data.arg = &state;
271 TEST(!pb_encode(&s, CallbackContainer_fields, &msg))
273 TEST(!pb_encode(&s, CallbackContainerContainer_fields, &msg2))
277 fprintf(stdout, "\n\nSome tests FAILED!\n");