From 82eec8b529f2113f19b443446545bad66d3a1547 Mon Sep 17 00:00:00 2001 From: Aryadev Chavali Date: Tue, 3 Dec 2024 00:21:42 +0000 Subject: Implement buffer_t functions to make it stretchy --- lib.c | 49 +++++++++++++++++++++++++++++++++++++++++++++++-- 1 file changed, 47 insertions(+), 2 deletions(-) (limited to 'lib.c') diff --git a/lib.c b/lib.c index 4f79345..0c3f4f9 100644 --- a/lib.c +++ b/lib.c @@ -13,11 +13,56 @@ buffer_t *buffer_init_str(const char *name, const char *str, size_t str_size) buffer_t *buf = malloc(sizeof(*buf) + str_size + 1); buf->name = name; buf->size = str_size; - memcpy(buf->data, str, str_size); - buf->data[str_size] = '\0'; + buf->capacity = str_size; + if (str) + { + memcpy(buf->data, str, str_size); + buf->data[str_size] = '\0'; + } return buf; } +bool buffer_ensure(buffer_t **buffer, u64 expected) +{ + if (!buffer) + return false; + else if (!*buffer) + { + *buffer = buffer_init_str(NULL, NULL, expected); + } + else if (buffer[0]->capacity <= expected) + { + buffer[0]->capacity = MAX(buffer[0]->capacity * 2, expected); + *buffer = realloc(*buffer, sizeof(**buffer) + buffer[0]->capacity); + } + return true; +} + +bool buffer_ensure_relative(buffer_t **buffer, u64 expected) +{ + return buffer_ensure(buffer, + buffer && *buffer ? buffer[0]->size + expected : 0); +} + +bool buffer_append(buffer_t **buffer, u8 datum) +{ + bool ret = buffer_ensure_relative(buffer, 1); + if (ret) + buffer[0]->data[buffer[0]->size++] = datum; + return ret; +} + +bool buffer_append_bytes(buffer_t **buffer, u8 *data, u64 size) +{ + bool ret = buffer_ensure_relative(buffer, size); + if (ret) + { + memcpy(buffer[0]->data + buffer[0]->size, data, size); + buffer[0]->size += size; + } + return ret; +} + void print_error(const char *handle, size_t row, size_t column, const char *reason) { -- cgit v1.2.3-13-gbd6f