summaryrefslogtreecommitdiff
path: root/upb_msg.c
diff options
context:
space:
mode:
Diffstat (limited to 'upb_msg.c')
-rw-r--r--upb_msg.c379
1 files changed, 0 insertions, 379 deletions
diff --git a/upb_msg.c b/upb_msg.c
deleted file mode 100644
index 47a8662..0000000
--- a/upb_msg.c
+++ /dev/null
@@ -1,379 +0,0 @@
-/*
- * upb - a minimalist implementation of protocol buffers.
- *
- */
-
-#include <inttypes.h>
-#include <stdlib.h>
-#include "descriptor.h"
-#include "upb_msg.h"
-#include "upb_parse.h"
-
-/* Rounds p up to the next multiple of t. */
-#define ALIGN_UP(p, t) ((p) % (t) == 0 ? (p) : (p) + ((t) - ((p) % (t))))
-
-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;
-}
-
-/* Callback for sorting fields. */
-static int compare_fields(const void *e1, const void *e2) {
- const google_protobuf_FieldDescriptorProto *fd1 = *(void**)e1;
- const google_protobuf_FieldDescriptorProto *fd2 = *(void**)e2;
- /* 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.
- * TODO: re-enable this check when we properly set this flag. */
- //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->descriptor = d;
- 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(*m->fields) * m->num_fields);
- m->field_descriptors = malloc(sizeof(*m->field_descriptors) * m->num_fields);
- for(unsigned int i = 0; i < m->num_fields; i++) {
- /* We count on the caller to keep this pointer alive. */
- m->field_descriptors[i] = d->field->elements[i];
- }
- /* TODO: re-enable proper sorting once the compiler is sorted out. */
- //qsort(m->field_descriptors, m->num_fields, sizeof(void*), 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 = m->field_descriptors[i];
- 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);
- f->type = fd->type;
- f->label = fd->label;
- 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 f->ref will be uninitialized, even in
- * the tables' copies of *f, which is why we must update them separately
- * in upb_msg_ref() below. */
- struct upb_fieldsbynum_entry nument = {.e = {.key = fd->number}, .f = *f};
- struct upb_fieldsbyname_entry strent = {.e = {.key = *fd->name}, .f = *f};
- 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);
- free(m->field_descriptors);
-}
-
-void upb_msg_ref(struct upb_msg *m, struct upb_msg_field *f,
- union upb_symbol_ref ref) {
- struct google_protobuf_FieldDescriptorProto *d =
- upb_msg_field_descriptor(f, m);
- struct upb_fieldsbynum_entry *int_e = upb_inttable_lookup(
- &m->fields_by_num, d->number, sizeof(struct upb_fieldsbynum_entry));
- struct upb_fieldsbyname_entry *str_e =
- upb_strtable_lookup(&m->fields_by_name, d->name);
- assert(int_e && str_e);
- f->ref = ref;
- int_e->f.ref = ref;
- str_e->f.ref = ref;
-}
-
-/* Memory management *********************************************************/
-
-/* Our memory management scheme is as follows:
- *
- * All pointers to dynamic memory (strings, arrays, and submessages) are
- * expected to be good pointers if they are non-zero, *regardless* of whether
- * that field's bit is set! That way we can reuse the memory even if the field
- * is unset and then set later. */
-
-/* For our memory-managed strings and arrays we store extra information
- * (compared to a plain upb_string or upb_array). But the data starts with
- * a upb_string and upb_array, so we can overlay onto the regular types. */
-struct mm_upb_string {
- struct upb_string s;
- /* Track the allocated size, so we know when we need to reallocate. */
- uint32_t size;
- /* Our allocated data. Stored separately so that clients can point s.ptr to
- * a referenced string, but we can reuse this data later. */
- char *data;
-};
-
-struct mm_upb_array {
- struct upb_array a;
- /* Track the allocated size, so we know when we need to reallocate. */
- uint32_t size;
-};
-
-static uint32_t round_up_to_pow2(uint32_t v)
-{
- /* cf. http://graphics.stanford.edu/~seander/bithacks.html#RoundUpPowerOf2 */
- v--;
- v |= v >> 1;
- v |= v >> 2;
- v |= v >> 4;
- v |= v >> 8;
- v |= v >> 16;
- v++;
- return v;
-}
-
-void *upb_msgdata_new(struct upb_msg *m)
-{
- void *msg = malloc(m->size);
- memset(msg, 0, m->size); /* Clear all pointers, values, and set bits. */
- return msg;
-}
-
-static void free_value(union upb_value_ptr p, struct upb_msg_field *f,
- bool free_submsgs)
-{
- switch(f->type) {
- case GOOGLE_PROTOBUF_FIELDDESCRIPTORPROTO_TYPE_STRING:
- case GOOGLE_PROTOBUF_FIELDDESCRIPTORPROTO_TYPE_BYTES: {
- struct mm_upb_string *mm_str = (void*)*p.string;
- if(mm_str) {
- free(mm_str->data);
- free(mm_str);
- }
- break;
- }
- case GOOGLE_PROTOBUF_FIELDDESCRIPTORPROTO_TYPE_MESSAGE:
- case GOOGLE_PROTOBUF_FIELDDESCRIPTORPROTO_TYPE_GROUP:
- if(free_submsgs) upb_msgdata_free(*p.message, f->ref.msg, free_submsgs);
- break;
- default: break; /* For non-dynamic types, do nothing. */
- }
-}
-
-void upb_msgdata_free(void *data, struct upb_msg *m, bool free_submsgs)
-{
- if(!data) return; /* A very free-like thing to do. */
- for(unsigned int i = 0; i < m->num_fields; i++) {
- struct upb_msg_field *f = &m->fields[i];
- union upb_value_ptr p = upb_msg_getptr(data, f);
- if(f->label == GOOGLE_PROTOBUF_FIELDDESCRIPTORPROTO_LABEL_REPEATED) {
- if(*p.array) {
- for(uint32_t j = 0; j < (*p.array)->len; j++)
- free_value(upb_array_getelementptr(*p.array, j, f->type),
- f, free_submsgs);
- free((*p.array)->elements._void);
- free(*p.array);
- }
- } else {
- free_value(p, f, free_submsgs);
- }
- }
- free(data);
-}
-
-void upb_msg_reuse_str(struct upb_string **str, uint32_t size)
-{
- if(!*str) {
- *str = malloc(sizeof(struct mm_upb_string));
- memset(*str, 0, sizeof(struct mm_upb_string));
- }
- struct mm_upb_string *s = (void*)*str;
- if(s->size < size) {
- size = max(16, round_up_to_pow2(size));
- s->data = realloc(s->data, size);
- s->size = size;
- }
- s->s.ptr = s->data;
-}
-
-void upb_msg_reuse_array(struct upb_array **arr, uint32_t size, upb_field_type_t t)
-{
- if(!*arr) {
- *arr = malloc(sizeof(struct mm_upb_array));
- memset(*arr, 0, sizeof(struct mm_upb_array));
- }
- struct mm_upb_array *a = (void*)*arr;
- if(a->size < size) {
- size = max(16, round_up_to_pow2(size));
- size_t type_size = upb_type_info[t].size;
- a->a.elements._void = realloc(a->a.elements._void, size * type_size);
- /* Zero any newly initialized memory. */
- memset(UPB_INDEX(a->a.elements._void, a->size, type_size), 0,
- (size - a->size) * type_size);
- a->size = size;
- }
-}
-
-void upb_msg_reuse_strref(struct upb_string **str) { upb_msg_reuse_str(str, 0); }
-
-void upb_msg_reuse_submsg(void **msg, struct upb_msg *m)
-{
- if(!*msg) *msg = upb_msgdata_new(m);
- else upb_msg_clear(*msg, m); /* Clears set bits, leaves pointers. */
-}
-
-/* Serialization/Deserialization. ********************************************/
-
-/* We use this as our "user_data" for each frame of the parsing stack. */
-struct parse_frame_data {
- struct upb_msg *m;
- void *data;
-};
-
-static void set_frame_data(struct upb_parse_state *s, struct upb_msg *m,
- void *data)
-{
- struct parse_frame_data *frame = (void*)&s->top->user_data;
- frame->m = m;
- frame->data = data;
-}
-
-static upb_field_type_t tag_cb(struct upb_parse_state *s, struct upb_tag *tag,
- void **user_field_desc)
-{
- struct parse_frame_data *frame = (void*)&s->top->user_data;
- struct upb_msg_field *f = upb_msg_fieldbynum(frame->m, tag->field_number);
- if(!f || !upb_check_type(tag->wire_type, f->type))
- return 0; /* Skip unknown or fields of the wrong type. */
- *user_field_desc = f;
- return f->type;
-}
-
-/* Returns a pointer to where the next value for field "f" should be stored,
- * taking into account whether f is an array that may need to be reallocatd. */
-static union upb_value_ptr get_value_ptr(void *data, struct upb_msg_field *f)
-{
- union upb_value_ptr p = upb_msg_getptr(data, f);
- if(f->label == GOOGLE_PROTOBUF_FIELDDESCRIPTORPROTO_LABEL_REPEATED) {
- size_t len = upb_msg_is_set(data, f) ? (*p.array)->len : 0;
- upb_msg_reuse_array(p.array, len+1, f->type);
- (*p.array)->len = len + 1;
- assert(p._void);
- p = upb_array_getelementptr(*p.array, len, f->type);
- assert(p._void);
- }
- upb_msg_set(data, f);
- assert(p._void);
- return p;
-}
-
-static upb_status_t value_cb(struct upb_parse_state *s, void **buf, void *end,
- void *user_field_desc)
-{
- struct parse_frame_data *frame = (void*)&s->top->user_data;
- struct upb_msg_field *f = user_field_desc;
- union upb_value_ptr p = get_value_ptr(frame->data, f);
- UPB_CHECK(upb_parse_value(buf, end, f->type, p));
- return UPB_STATUS_OK;
-}
-
-static upb_status_t str_cb(struct upb_parse_state *_s, struct upb_string *str,
- void *user_field_desc)
-{
- struct upb_msg_parse_state *s = (void*)_s;
- struct parse_frame_data *frame = (void*)&s->s.top->user_data;
- struct upb_msg_field *f = user_field_desc;
- union upb_value_ptr p = get_value_ptr(frame->data, f);
- if(s->byref) {
- upb_msg_reuse_strref(p.string);
- **p.string = *str;
- } else {
- upb_msg_reuse_str(p.string, str->byte_len);
- upb_strcpy(*p.string, str);
- }
- return UPB_STATUS_OK;
-}
-
-static void submsg_start_cb(struct upb_parse_state *_s, void *user_field_desc)
-{
- struct upb_msg_parse_state *s = (void*)_s;
- struct upb_msg_field *f = user_field_desc;
- struct parse_frame_data *frame = (void*)&s->s.top->user_data;
- // TODO: find a non-hacky way to get a pointer to the old frame.
- struct parse_frame_data *oldframe = (void*)((char*)s->s.top - s->s.udata_size);
- union upb_value_ptr p = get_value_ptr(oldframe->data, f);
- assert(f->ref.msg);
- upb_msg_reuse_submsg(p.message, f->ref.msg);
- set_frame_data(&s->s, f->ref.msg, *p.message);
- if(!s->merge) upb_msg_clear(frame->data, f->ref.msg);
-}
-
-void upb_msg_parse_reset(struct upb_msg_parse_state *s, void *msg,
- struct upb_msg *m, bool merge, bool byref)
-{
- upb_parse_reset(&s->s);
- s->merge = merge;
- s->byref = byref;
- if(!merge && msg == NULL) msg = upb_msgdata_new(m);
- upb_msg_clear(msg, m);
- set_frame_data(&s->s, m, msg);
- s->s.tag_cb = tag_cb;
- s->s.value_cb = value_cb;
- s->s.str_cb = str_cb;
- s->s.submsg_start_cb = submsg_start_cb;
-}
-
-void upb_msg_parse_init(struct upb_msg_parse_state *s, void *msg,
- struct upb_msg *m, bool merge, bool byref)
-{
- upb_parse_init(&s->s, sizeof(struct parse_frame_data));
- upb_msg_parse_reset(s, msg, m, merge, byref);
-}
-
-void upb_msg_parse_free(struct upb_msg_parse_state *s)
-{
- upb_parse_free(&s->s);
-}
-
-upb_status_t upb_msg_parse(struct upb_msg_parse_state *s,
- void *data, size_t len, size_t *read)
-{
- return upb_parse(&s->s, data, len, read);
-}
-
-void *upb_alloc_and_parse(struct upb_msg *m, struct upb_string *str, bool byref)
-{
- struct upb_msg_parse_state s;
- void *msg = upb_msgdata_new(m);
- upb_msg_parse_init(&s, msg, m, false, byref);
- size_t read;
- upb_status_t status = upb_msg_parse(&s, str->ptr, str->byte_len, &read);
- upb_msg_parse_free(&s);
- if(status == UPB_STATUS_OK && read == str->byte_len) {
- return msg;
- } else {
- upb_msg_free(msg);
- return NULL;
- }
-}
generated by cgit on debian on lair
contact matthew@masot.net with questions or feedback