diff options
Diffstat (limited to 'pbstruct.c')
| -rw-r--r-- | pbstruct.c | 101 |
1 files changed, 101 insertions, 0 deletions
@@ -10,6 +10,25 @@ #define alignof(t) offsetof(struct { char c; t x; }, x) #define ALIGN_UP(p, t) (alignof(t) + ((p - 1) & ~(alignof(t) - 1))) +void pbstruct_init(struct pbstruct *s, struct pbstruct_definition *d) +{ + /* "set" flags start as unset, dynamic pointers start as NULL. */ + memset(s, 0, d->struct_size); + s->definition = d; +} + +pbstruct *pbstruct_new(struct pbstruct_defintion *d, struct pballoc *a) +{ + pbstruct *s = a->realloc(NULL, d->struct_size, d->user_data); + pbstruct_init(s, d); +} + +void pbstruct_free(struct pbstruct *s, struct pballoc *a, bool free_substructs) +{ + pbstruct_free_all_fields(s, free_substructs); + a->free(s, s->user_data); +} + bool pbstruct_is_set(struct pbstruct *s, struct pbstruct_field *f) { return s->data[f->isset_byte_offset] & f->isset_byte_mask; @@ -25,6 +44,88 @@ void pbstruct_set(struct pbstruct *s, struct pbstruct_field *f) s->data[f->isset_byte_offset] |= f->isset_byte_mask; } +void pbstruct_unset_all(struct pbstruct *s) +{ + for(int i = 0; i < s->definition->num_fields; i++) + pbstruct_unset(s, &s->definition->fields[i]); +} + +void pbstruct_resize_str(struct pbstruct *s, struct pbstruct_field *f, + size_t len, struct pballoc *a) +{ + assert(f->type & PBSTRUCT_DELIMITED); + size_t size = next_pow2(len); + struct pbstruct_dynamic **dyn = (struct pbstruct_dynamic**)get_addr(s, f); + if(!*dyn || (*dyn)->size < size) { + size_t old_len = *dyn ? (*dyn)->len : 0; + *dyn = a->realloc(*dyn, size*type_size, s->definition->user_data); + (*dyn)->size = size; + (*dyn)->len = len; + } +} + +size_t pbstruct_append_array(struct pbstruct *s, struct pbstruct_field *f, + struct pballoc *a) +{ + assert(f->type & PBSTRUCT_ARRAY); + struct pbstruct_dynamic **dyn = (struct pbstruct_dynamic**)get_addr(s, f); + size_t new_len, new_size; + if(!*dyn) { + new_len = 1; + new_size = 4; /* Initial size. */ + } else { + new_len = (*dyn)->len + 1; + new_size = next_pow2(new_len); + } + *dyn = a->realloc(*dyn, ARRSIZE(f->type, new_size), s->defintion->user_data); + memset(ARRADDR(*dyn, f->type, new_len-1), 0, SIZE(f->type)); +} + +void pbstruct_clear_array(struct pbstruct *s, struct pbstruct_field *f, + struct pballoc *a, bool free_substructs) +{ + assert(f->type & PBSTRUCT_ARRAY); + struct pbstruct_dynamic *dyn = *(struct pbstruct_dynamic**)get_addr(s, f); + if(dyn) { + if((f->type & PBSTRUCT_DELIMITED) || free_substructs) { + for(int i = 0; i < dyn->len; i++) + FREE(ARRADDR(dyn, f->type, i), a, s->definition->user_data); + } + dyn->size = 0; + } +} + +/* Resizes a delimited field (string or bytes) string that is within an array. + * The array must already be large enough that str_offset is a valid member. */ +void pbstruct_resize_arraystr(struct pbstruct *s, struct pbstruct_field *f, + size_t str_offset, size_t len, struct pballoc *a) +{ + +} + +/* Like pbstruct_append_array, but appends a string of the specified length. */ +size_t pbstruct_append_arraystr(struct pbstruct *s, struct pbstruct_field *f, + size_t len, struct pballoc *a) +{ +} + +void pbstruct_free_field(struct pbstruct *s, struct pbstruct_field *f, + bool free_substructs) +{ + pbstruct_clear_array(s, f, free_substructs); + struct pbstruct_dynamic **dyn = (struct pbstruct_dynamic**)get_addr(s, f); + if(*dyn) { + FREE(*dyn, s); + *dyn = NULL; + } +} + +void pbstruct_free_all_fields(struct pbstruct *s, bool free_substructs) +{ + for(int i = 0; i < s->definition->num_fields; i++) + pbstruct_free_field(s, &s->definition->fields[i]); +} + #define DEFINE_GETTERS(ctype, name) \ ctype *pbstruct_get_ ## name(struct pbstruct *s, struct pbstruct_field *f) { \ return (ctype*)(s->data + f->byte_offset); \ |