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
91
92
93
94
95
96
|
/*
* upb - a minimalist implementation of protocol buffers.
*
*/
#include <stdlib.h>
#include "descriptor.h"
#include "upb_msg.h"
#define ALIGN_UP(p, t) (t + ((p - 1) & (~t - 1)))
static uint32_t max(uint32_t a, uint32_t b) { return a > b ? a : b; }
static int div_round_up(int numerator, int denominator) {
/* cf. http://stackoverflow.com/questions/17944/how-to-round-up-the-result-of-integer-division */
return numerator > 0 ? (numerator - 1) / denominator + 1 : 0;
}
static int compare_fields(const void *e1, const void *e2) {
const struct upb_msg_field *f1 = e1, *f2 = e2;
const google_protobuf_FieldDescriptorProto *fd1 = f1->descriptor;
const google_protobuf_FieldDescriptorProto *fd2 = f2->descriptor;
/* Required fields go before non-required. */
bool req1 = fd1->label == GOOGLE_PROTOBUF_FIELDDESCRIPTORPROTO_LABEL_REQUIRED;
bool req2 = fd2->label == GOOGLE_PROTOBUF_FIELDDESCRIPTORPROTO_LABEL_REQUIRED;
if(req1 != req2) {
return req2 - req1;
} else {
/* Within required and non-required field lists, list in number order.
* TODO: consider ordering by data size to reduce padding. */
return fd1->number - fd2->number;
}
}
bool upb_msg_init(struct upb_msg *m, struct google_protobuf_DescriptorProto *d)
{
/* TODO: more complete validation. */
if(!d->set_flags.has.field) return false;
upb_inttable_init(&m->fields_by_num, d->field->len,
sizeof(struct upb_fieldsbynum_entry));
upb_strtable_init(&m->fields_by_name, d->field->len,
sizeof(struct upb_fieldsbyname_entry));
m->num_fields = d->field->len;
m->set_flags_bytes = div_round_up(m->num_fields, 8);
/* These are incremented in the loop. */
m->num_required_fields = 0;
m->size = m->set_flags_bytes;
m->fields = malloc(sizeof(struct upb_msg_field) * m->num_fields);
for(unsigned int i = 0; i < m->num_fields; i++) {
/* We count on the caller to keep this pointer alive. */
m->fields[i].descriptor = d->field->elements[i];
}
qsort(m->fields, m->num_fields, sizeof(struct upb_msg_field), compare_fields);
size_t max_align = 0;
for(unsigned int i = 0; i < m->num_fields; i++) {
struct upb_msg_field *f = &m->fields[i];
google_protobuf_FieldDescriptorProto *fd = f->descriptor;
struct upb_type_info *type_info = &upb_type_info[fd->type];
/* General alignment rules are: each member must be at an address that is a
* multiple of that type's alignment. Also, the size of the structure as
* a whole must be a multiple of the greatest alignment of any member. */
f->field_index = i;
f->byte_offset = ALIGN_UP(m->size, type_info->align);
m->size = f->byte_offset + type_info->size;
max_align = max(max_align, type_info->align);
if(fd->label == GOOGLE_PROTOBUF_FIELDDESCRIPTORPROTO_LABEL_REQUIRED)
m->num_required_fields++;
/* Insert into the tables. Note that af->ref will be uninitialized, even in
* the tables' copies of *af, which is why we must update them separately
* when the references are resolved. */
struct upb_abbrev_msg_field af = {.byte_offset = f->byte_offset,
.field_index = f->field_index,
.type = fd->type};
struct upb_fieldsbynum_entry nument = {.e = {.key = fd->number}, .f = af};
struct upb_fieldsbyname_entry strent = {.e = {.key = *fd->name}, .f = af};
upb_inttable_insert(&m->fields_by_num, &nument.e);
upb_strtable_insert(&m->fields_by_name, &strent.e);
}
m->size = ALIGN_UP(m->size, max_align);
return true;
}
void upb_msg_free(struct upb_msg *m)
{
upb_inttable_free(&m->fields_by_num);
upb_strtable_free(&m->fields_by_name);
free(m->fields);
}
|