aboutsummaryrefslogtreecommitdiffstats
path: root/tests/extensions/encode_extensions.c
blob: dee3597dbf4f4398acb63723d43a3c77a1c481d3 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
/* Tests extension fields.
 */

#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <pb_encode.h>
#include "alltypes.pb.h"
#include "extensions.pb.h"
#include "test_helpers.h"

int main(int argc, char **argv)
{
    uint8_t buffer[1024];
    pb_ostream_t stream;

    AllTypes alltypes = {0};
    int32_t extensionfield1 = 12345;
    pb_extension_t ext1;
    ExtensionMessage extensionfield2 = {"test", 54321};
    pb_extension_t ext2;

    /* Set up the extensions */
    alltypes.extensions = &ext1;

    ext1.type = &AllTypes_extensionfield1;
    ext1.dest = &extensionfield1;
    ext1.next = &ext2;
    
    ext2.type = &ExtensionMessage_AllTypes_extensionfield2;
    ext2.dest = &extensionfield2;
    ext2.next = NULL;

    /* Set up the output stream */
    stream = pb_ostream_from_buffer(buffer, sizeof(buffer));
    
    /* Now encode the message and check if we succeeded. */
    if (pb_encode(&stream, AllTypes_fields, &alltypes))
    {
        SET_BINARY_MODE(stdout);
        fwrite(buffer, 1, stream.bytes_written, stdout);
        return 0; /* Success */
    }
    else
    {
        fprintf(stderr, "Encoding failed: %s\n", PB_GET_ERROR(&stream));
        return 1; /* Failure */
    }
}