1 /* Decode a message using oneof fields */
2 
3 #include <stdio.h>
4 #include <stdlib.h>
5 #include <string.h>
6 #include <pb_decode.h>
7 #include "oneof.pb.h"
8 #include "test_helpers.h"
9 #include "unittests.h"
10 
11 /* Test the 'OneOfMessage' */
12 int test_oneof_1(pb_istream_t *stream, int option)
13 {
14     OneOfMessage msg;
15     int status = 0;
16 
17     /* To better catch initialization errors */
18     memset(&msg, 0xAA, sizeof(msg));
19 
20     if (!pb_decode(stream, OneOfMessage_fields, &msg))
21     {
22         printf("Decoding failed: %s\n", PB_GET_ERROR(stream));
23         return 1;
24     }
25 
26     /* Check that the basic fields work normally */
27     TEST(msg.prefix == 123);
28     TEST(msg.suffix == 321);
29 
30     /* Check that we got the right oneof according to command line */
31     if (option == 1)
32     {
33         TEST(msg.which_values == OneOfMessage_first_tag);
34         TEST(msg.values.first == 999);
35     }
36     else if (option == 2)
37     {
38         TEST(msg.which_values == OneOfMessage_second_tag);
39         TEST(strcmp(msg.values.second, "abcd") == 0);
40     }
41     else if (option == 3)
42     {
43         TEST(msg.which_values == OneOfMessage_third_tag);
44         TEST(msg.values.third.array[0] == 1);
45         TEST(msg.values.third.array[1] == 2);
46         TEST(msg.values.third.array[2] == 3);
47         TEST(msg.values.third.array[3] == 4);
48         TEST(msg.values.third.array[4] == 5);
49     }
50 
51     return status;
52 }
53 
54 
55 /* Test the 'PlainOneOfMessage' */
56 int test_oneof_2(pb_istream_t *stream, int option)
57 {
58     PlainOneOfMessage msg = PlainOneOfMessage_init_zero;
59     int status = 0;
60 
61     if (!pb_decode(stream, PlainOneOfMessage_fields, &msg))
62     {
63         printf("Decoding failed: %s\n", PB_GET_ERROR(stream));
64         return 1;
65     }
66 
67     /* Check that we got the right oneof according to command line */
68     if (option == 1)
69     {
70         TEST(msg.which_values == OneOfMessage_first_tag);
71         TEST(msg.values.first == 999);
72     }
73     else if (option == 2)
74     {
75         TEST(msg.which_values == OneOfMessage_second_tag);
76         TEST(strcmp(msg.values.second, "abcd") == 0);
77     }
78     else if (option == 3)
79     {
80         TEST(msg.which_values == OneOfMessage_third_tag);
81         TEST(msg.values.third.array[0] == 1);
82         TEST(msg.values.third.array[1] == 2);
83         TEST(msg.values.third.array[2] == 3);
84         TEST(msg.values.third.array[3] == 4);
85         TEST(msg.values.third.array[4] == 5);
86     }
87 
88     return status;
89 }
90 
91 int main(int argc, char **argv)
92 {
93     uint8_t buffer[OneOfMessage_size];
94     size_t count;
95     int option;
96 
97     if (argc != 2)
98     {
99         fprintf(stderr, "Usage: decode_oneof [number]\n");
100         return 1;
101     }
102     option = atoi(argv[1]);
103 
104     SET_BINARY_MODE(stdin);
105     count = fread(buffer, 1, sizeof(buffer), stdin);
106 
107     if (!feof(stdin))
108     {
109         printf("Message does not fit in buffer\n");
110         return 1;
111     }
112 
113     {
114         int status = 0;
115         pb_istream_t stream;
116 
117         stream = pb_istream_from_buffer(buffer, count);
118         status = test_oneof_1(&stream, option);
119 
120         if (status != 0)
121             return status;
122 
123         stream = pb_istream_from_buffer(buffer, count);
124         status = test_oneof_2(&stream, option);
125 
126         if (status != 0)
127             return status;
128     }
129 
130     return 0;
131 }
132