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
|
#include "main.c"
#include "upb_context.h"
#include "upb_msg.h"
static struct upb_context c;
static struct upb_string str;
static struct upb_msg_parse_state s;
static struct upb_msg *m;
static void *data;
static bool initialize()
{
/* Initialize upb state, parse descriptor. */
upb_context_init(&c);
struct upb_string fds;
if(!upb_strreadfile(MESSAGE_DESCRIPTOR_FILE, &fds)) {
fprintf(stderr, "Couldn't read " MESSAGE_DESCRIPTOR_FILE ".\n");
return false;
}
if(!upb_context_parsefds(&c, &fds)) {
fprintf(stderr, "Error importing " MESSAGE_DESCRIPTOR_FILE ".\n");
return false;
}
upb_strfree(fds);
char class_name[] = MESSAGE_NAME;
struct upb_string proto_name;
proto_name.ptr = class_name;
proto_name.byte_len = sizeof(class_name)-1;
struct upb_symtab_entry *e = upb_context_lookup(&c, &proto_name);
if(!e || e->type != UPB_SYM_MESSAGE) {
fprintf(stderr, "Error finding symbol '" UPB_STRFMT "'.\n",
UPB_STRARG(proto_name));
return false;
}
m = e->ref.msg;
data = upb_msgdata_new(m);
upb_msg_parse_init(&s, data, m, false, true);
/* Read the message data itself. */
if(!upb_strreadfile(MESSAGE_FILE, &str)) {
fprintf(stderr, "Error reading " MESSAGE_FILE "\n");
return false;
}
return true;
}
static void cleanup()
{
upb_strfree(str);
upb_context_free(&c);
}
static size_t run()
{
size_t read;
upb_msg_parse_reset(&s, data, m, false, BYREF);
upb_status_t status = upb_msg_parse(&s, str.ptr, str.byte_len, &read);
if(status != UPB_STATUS_OK && read != str.byte_len) {
fprintf(stderr, "Error. :( error=%d, read=%lu\n", status, read);
return 0;
}
return read;
}
|
