1/* Tests extension fields.
2 */
3
4#include <stdio.h>
5#include <stdlib.h>
6#include <string.h>
7#include <pb_encode.h>
8#include "alltypes.pb.h"
9#include "extensions.pb.h"
10#include "test_helpers.h"
11
12int main(int argc, char **argv)
13{
14    uint8_t buffer[1024];
15    pb_ostream_t stream;
16
17    AllTypes alltypes = {0};
18    int32_t extensionfield1 = 12345;
19    pb_extension_t ext1;
20    ExtensionMessage extensionfield2 = {"test", 54321};
21    pb_extension_t ext2;
22
23    /* Set up the extensions */
24    alltypes.extensions = &ext1;
25
26    ext1.type = &AllTypes_extensionfield1;
27    ext1.dest = &extensionfield1;
28    ext1.next = &ext2;
29
30    ext2.type = &ExtensionMessage_AllTypes_extensionfield2;
31    ext2.dest = &extensionfield2;
32    ext2.next = NULL;
33
34    /* Set up the output stream */
35    stream = pb_ostream_from_buffer(buffer, sizeof(buffer));
36
37    /* Now encode the message and check if we succeeded. */
38    if (pb_encode(&stream, AllTypes_fields, &alltypes))
39    {
40        SET_BINARY_MODE(stdout);
41        fwrite(buffer, 1, stream.bytes_written, stdout);
42        return 0; /* Success */
43    }
44    else
45    {
46        fprintf(stderr, "Encoding failed: %s\n", PB_GET_ERROR(&stream));
47        return 1; /* Failure */
48    }
49
50    /* Check that the field tags are properly generated */
51    (void)AllTypes_extensionfield1_tag;
52    (void)ExtensionMessage_AllTypes_extensionfield2_tag;
53}
54
55