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
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
|
/* Encode a message using msgid field as prefix */
#include <stdio.h>
#include <stdlib.h>
#include <string.h>
#include <pb_encode.h>
#include "msgid_example.pb.h"
#include "test_helpers.h"
/* This function writes the message id as a prefix to the message, allowing
* the receiving side to identify message type. Here we use uint8_t to store
* it, but e.g. varint or some custom header struct would work just as well.
*/
bool write_prefix(pb_ostream_t *stream, int msgid)
{
uint8_t prefix = msgid;
return pb_write(stream, &prefix, 1);
}
/* The main logic will call one of these functions.
* Normally which function you call would be selected based on what message
* you want to send, here it is decided based on command line parameter.
*/
bool encode_MyMessage1(pb_ostream_t *stream)
{
MyMessage1 msg = MyMessage1_init_default;
msg.intvalue = 1234;
return write_prefix(stream, MyMessage1_msgid)
&& pb_encode(stream, MyMessage1_fields, &msg);
}
bool encode_MyMessage2(pb_ostream_t *stream)
{
MyMessage2 msg = MyMessage2_init_default;
msg.intvalue = 9999;
strcpy(msg.strvalue, "Msg2");
return write_prefix(stream, MyMessage2_msgid)
&& pb_encode(stream, MyMessage2_fields, &msg);
}
bool encode_MyMessage3(pb_ostream_t *stream)
{
MyMessage3 msg = MyMessage3_init_default;
msg.boolvalue = true;
return write_prefix(stream, MyMessage3_msgid)
&& pb_encode(stream, MyMessage3_fields, &msg);
}
int main(int argc, char **argv)
{
uint8_t buffer[128];
pb_ostream_t stream;
bool status = false;
int option;
if (argc != 2)
{
fprintf(stderr, "Usage: encode_msgid [number]\n");
return 1;
}
option = atoi(argv[1]);
stream = pb_ostream_from_buffer(buffer, sizeof(buffer));
if (option == 1)
{
status = encode_MyMessage1(&stream);
}
else if (option == 2)
{
status = encode_MyMessage2(&stream);
}
else if (option == 3)
{
status = encode_MyMessage3(&stream);
}
if (status)
{
SET_BINARY_MODE(stdout);
fwrite(buffer, 1, stream.bytes_written, stdout);
return 0;
}
else
{
fprintf(stderr, "Encoding failed: %s\n", PB_GET_ERROR(&stream));
return 1;
}
}
|
