summaryrefslogtreecommitdiff
path: root/dynstr.c
diff options
context:
space:
mode:
Diffstat (limited to 'dynstr.c')
-rw-r--r--dynstr.c547
1 files changed, 0 insertions, 547 deletions
diff --git a/dynstr.c b/dynstr.c
deleted file mode 100644
index 44c4b56..0000000
--- a/dynstr.c
+++ /dev/null
@@ -1,547 +0,0 @@
-/*
- * dynstr.c - implement dynamic byte strings
- * Copyright 2010 Lars Wirzenius
- *
- * This program is free software: you can redistribute it and/or modify
- * it under the terms of the GNU General Public License as published by
- * the Free Software Foundation, either version 3 of the License, or
- * (at your option) any later version.
- *
- * This program is distributed in the hope that it will be useful,
- * but WITHOUT ANY WARRANTY; without even the implied warranty of
- * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
- * GNU General Public License for more details.
- *
- * You should have received a copy of the GNU General Public License
- * along with this program. If not, see <http://www.gnu.org/licenses/>.
- */
-
-
-/* We shamelessly use GNU extensions to the standard C library.
- * Primarily this is the memmem(3) function. Patches are welcome to
- * get rid of this, but I did not feel like re-inventing the wheel.
- */
-#define _GNU_SOURCE
-
-#include <errno.h>
-#include <stdbool.h>
-#include <stdlib.h>
-#include <string.h>
-#include <unistd.h>
-
-#include "dynstr.h"
-
-
-/* This is how the dynamic strings are represented. */
-struct Dynstr {
- unsigned char *mem;
- size_t size;
-};
-
-
-/* Return value for readline_helper's callback functions. */
-#define DYNSTR_ERROR ((size_t) -1)
-
-
-/* The current malloc and malloc error handler functions. */
-static dynstr_error_handler *error_handler = dynstr_malloc_error_indicate;
-static void *(*current_malloc)(size_t) = malloc;
-
-
-void dynstr_init(void)
-{
- error_handler = dynstr_malloc_error_indicate;
- current_malloc = malloc;
-}
-
-
-void *(*dynstr_get_malloc(void))(size_t)
-{
- return current_malloc;
-}
-
-
-void dynstr_set_malloc(void *(*new_malloc)(size_t))
-{
- current_malloc = new_malloc;
-}
-
-
-dynstr_error_handler *dynstr_get_malloc_error_handler(void)
-{
- return error_handler;
-}
-
-
-void dynstr_set_malloc_error_handler(dynstr_error_handler *handler)
-{
- error_handler = handler;
-}
-
-
-void dynstr_malloc_error_indicate(int error, size_t size, void *oldptr)
-{
-}
-
-void dynstr_malloc_error_abort(int error, size_t size, void *oldptr)
-{
- abort();
-}
-
-
-/* Dynamically allocate a number of bytes. Call error handler if necessary. */
-static void *alloc(size_t size)
-{
- void *p;
-
- p = current_malloc(size);
- if (p == NULL)
- error_handler(errno, size, NULL);
- return p;
-}
-
-
-static Dynstr *new_dynstr(const void *mem, size_t size, bool dynamic)
-{
- Dynstr *dynstr;
-
- if (dynamic) {
- dynstr = alloc(sizeof(Dynstr) + size);
- if (dynstr == NULL)
- return NULL;
- dynstr->mem = ((unsigned char *) dynstr) + sizeof(Dynstr);
- } else {
- dynstr = alloc(sizeof(Dynstr));
- if (dynstr == NULL)
- return NULL;
- dynstr->mem = (unsigned char *) mem;
- }
-
- dynstr->size = size;
- return dynstr;
-}
-
-
-static void init_dynstr(Dynstr *dynstr, const void *mem, bool dynamic)
-{
- if (dynamic)
- memcpy(dynstr->mem, mem, dynstr->size);
-}
-
-
-/* Allocate a new dynamic string. If dynamic is true, the contents of the
- * new string is copied from mem. Otherwise the new string just uses mem
- * directly. */
-static Dynstr *newstr(const void *mem, size_t size, bool dynamic)
-{
- Dynstr *dynstr;
-
- dynstr = new_dynstr(mem, size, dynamic);
- if (dynstr != NULL)
- init_dynstr(dynstr, mem, dynamic);
- return dynstr;
-}
-
-
-Dynstr *dynstr_new_empty(void)
-{
- return new_dynstr(NULL, 0, false);
-}
-
-
-Dynstr *dynstr_new_from_cstring(const char *cstring)
-{
- return newstr(cstring, strlen(cstring), true);
-}
-
-
-Dynstr *dynstr_new_from_memory(const void *mem, size_t size)
-{
- return newstr(mem, size, true);
-}
-
-
-Dynstr *dynstr_new_from_constant_cstring(const char *cstring)
-{
- return newstr(cstring, strlen(cstring), false);
-}
-
-
-Dynstr *dynstr_new_from_constant_memory(const void *mem, size_t size)
-{
- return newstr(mem, size, false);
-}
-
-
-void dynstr_free(Dynstr *dynstr)
-{
- free(dynstr);
-}
-
-
-size_t dynstr_len(Dynstr *dynstr)
-{
- return dynstr->size;
-}
-
-
-bool dynstr_is_empty(Dynstr *dynstr)
-{
- return dynstr_len(dynstr) == 0;
-}
-
-
-size_t dynstr_memcpy(void *mem, Dynstr *dynstr, size_t offset, size_t size)
-{
- size_t len;
-
- len = dynstr_len(dynstr);
- if (offset >= len)
- return 0;
- if (size > len - offset)
- size = len - offset;
- memcpy(mem, dynstr->mem + offset, size);
- return size;
-}
-
-
-char *dynstr_strdup(Dynstr *dynstr)
-{
- char *mem;
-
- mem = alloc(dynstr->size + 1);
- if (mem == NULL)
- return NULL;
- memcpy(mem, dynstr->mem, dynstr->size);
- mem[dynstr->size] = '\0';
- return mem;
-}
-
-
-Dynstr *dynstr_substr(Dynstr *dynstr, size_t offset, size_t size)
-{
- if (offset >= dynstr->size) {
- offset = 0;
- size = 0;
- } else if (size > dynstr->size - offset) {
- size = dynstr->size - offset;
- }
- return newstr(dynstr->mem + offset, size, true);
-}
-
-
-Dynstr *dynstr_cat(Dynstr *dynstr1, Dynstr *dynstr2)
-{
- return dynstr_cat_many(dynstr1, dynstr2, (Dynstr *) NULL);
-}
-
-
-Dynstr *dynstr_cat_many(Dynstr *dynstr, ...)
-{
- va_list args;
- Dynstr *result;
- size_t size;
- size_t offset;
-
- va_start(args, dynstr);
- size = 0;
- for (Dynstr *p = dynstr; p != NULL; p = va_arg(args, Dynstr *))
- size += p->size;
- va_end(args);
-
- result = new_dynstr(NULL, size, true);
- if (result == NULL)
- return NULL;
-
- va_start(args, dynstr);
- offset = 0;
- for (Dynstr *p = dynstr; p != NULL; p = va_arg(args, Dynstr *)) {
- memcpy(result->mem + offset, p->mem, p->size);
- offset += p->size;
- }
- va_end(args);
-
- return result;
-}
-
-
-int dynstr_byte_at(Dynstr *dynstr, size_t offset)
-{
- if (offset >= dynstr->size)
- return -1;
- return dynstr->mem[offset];
-}
-
-
-size_t dynstr_first_byte(Dynstr *dynstr, size_t offset, int byte)
-{
- unsigned char *p;
-
- if (offset >= dynstr->size)
- return DYNSTR_NOT_FOUND;
- p = memchr(dynstr->mem + offset, byte, dynstr->size - offset);
- if (p == NULL)
- return DYNSTR_NOT_FOUND;
- return (size_t) (p - dynstr->mem);
-}
-
-
-size_t dynstr_last_byte(Dynstr *dynstr, size_t offset, int byte)
-{
- unsigned char *p;
-
- if (offset >= dynstr->size)
- return DYNSTR_NOT_FOUND;
- p = memrchr(dynstr->mem + offset, byte, dynstr->size - offset);
- if (p == NULL)
- return DYNSTR_NOT_FOUND;
- return (size_t) (p - dynstr->mem);
-}
-
-
-size_t dynstr_first_string(Dynstr *dynstr, size_t offset, Dynstr *pattern)
-{
- const unsigned char *p;
-
- if (offset >= dynstr->size)
- return DYNSTR_NOT_FOUND;
- if (pattern->size == 0)
- return DYNSTR_NOT_FOUND;
- p = memmem(dynstr->mem + offset, dynstr->size - offset,
- pattern->mem, pattern->size);
- if (p == NULL)
- return DYNSTR_NOT_FOUND;
- return (size_t) (p - dynstr->mem);
-}
-
-
-size_t dynstr_last_string(Dynstr *dynstr, size_t offset, Dynstr *pattern)
-{
- size_t result;
- size_t new_result;
-
- /* GNU libc lacks a memrmem function, so we loop to the last
- * dynstr_first_string match. */
- result = DYNSTR_NOT_FOUND;
- for (;;) {
- new_result = dynstr_first_string(dynstr, offset, pattern);
- if (new_result == DYNSTR_NOT_FOUND)
- break;
- result = new_result;
- offset = result + 1;
- }
- return result;
-}
-
-
-int dynstr_cmp(Dynstr *dynstr1, Dynstr *dynstr2)
-{
- int result;
- size_t min_size;
-
- if (dynstr1->size < dynstr2->size)
- min_size = dynstr1->size;
- else
- min_size = dynstr2->size;
- result = memcmp(dynstr1->mem, dynstr2->mem, min_size);
- if (result == 0) {
- if (dynstr1->size < dynstr2->size)
- return -1;
- else if (dynstr1->size == dynstr2->size)
- return 0;
- else
- return 1;
- }
- return result;
-}
-
-
-size_t dynstr_fwrite(FILE *file, Dynstr *dynstr)
-{
- return fwrite(dynstr->mem, 1, dynstr->size, file);
-}
-
-
-ssize_t dynstr_write(int fd, Dynstr *dynstr)
-{
- return write(fd, dynstr->mem, dynstr->size);
-}
-
-
-/* A helper function for dynstr_read and dynstr_fread. The callback does
- * the actual reading. */
-static Dynstr *read_helper(size_t (*callback)(FILE *f, int fd, unsigned char *buf,
- size_t size),
- FILE *f, int fd, size_t size)
-{
- Dynstr *dynstr;
- Dynstr *dynstr2;
- size_t n;
-
- dynstr = new_dynstr(NULL, size, true);
- if (dynstr == NULL)
- return NULL;
-
- n = callback(f, fd, dynstr->mem, size);
- if (n == DYNSTR_ERROR) {
- dynstr_free(dynstr);
- return NULL;
- }
-
- if (n < size) {
- dynstr2 = dynstr_substr(dynstr, 0, n);
- if (dynstr2 == NULL) {
- dynstr->size = n;
- } else {
- dynstr_free(dynstr);
- dynstr = dynstr2;
- }
- }
-
- return dynstr;
-}
-
-
-static size_t fread_callback(FILE *f, int fd, unsigned char *buf, size_t size)
-{
- size_t n;
-
- n = fread(buf, 1, size, f);
- if (n == 0 && ferror(f))
- return DYNSTR_ERROR;
- return n;
-}
-
-
-static size_t read_callback(FILE *f, int fd, unsigned char *buf, size_t size)
-{
- ssize_t n;
-
- do {
- n = read(fd, buf, size);
- } while (n == -1 && errno == EINTR);
- if (n == -1)
- return DYNSTR_ERROR;
- return n;
-}
-
-
-Dynstr *dynstr_fread(FILE *file, size_t size)
-{
- return read_helper(fread_callback, file, -1, size);
-}
-
-
-Dynstr *dynstr_read(int file, size_t size)
-{
- return read_helper(read_callback, NULL, file, size);
-}
-
-
-/* Helper function for dynstr_readline and dynstr_freadline. The callback
- * does the actual reading. Reading happens one byte at a time, since we
- * must not read past the newline that terminates a line. */
-static Dynstr *readline_helper(int (*callback)(FILE *, int), FILE *f, int fd)
-{
- Dynstr *line;
- Dynstr *temp1;
- Dynstr *temp2;
- int c;
- unsigned char buf[1024];
- size_t buflen;
-
- line = dynstr_new_empty();
- buflen = 0;
-
- for (;;) {
- temp1 = NULL;
- temp2 = NULL;
-
- c = callback(f, fd);
- if (c == -1)
- goto error;
- else if (c == -2)
- break;
-
- buf[buflen++] = c;
- if (buflen == sizeof(buf)) {
- temp1 = dynstr_new_from_constant_memory(buf, buflen);
- if (temp1 == NULL)
- goto error;
- temp2 = dynstr_cat(line, temp1);
- if (temp2 == NULL)
- goto error;
- dynstr_free(temp1);
- dynstr_free(line);
- line = temp2;
- buflen = 0;
- }
-
- if (c == '\n')
- break;
- }
-
- if (buflen > 0) {
- temp1 = dynstr_new_from_constant_memory(buf, buflen);
- if (temp1 == NULL)
- goto error;
- temp2 = dynstr_cat(line, temp1);
- if (temp2 == NULL)
- goto error;
- dynstr_free(temp1);
- dynstr_free(line);
- line = temp2;
- }
-
- return line;
-
-error:
- dynstr_free(line);
- dynstr_free(temp1);
- dynstr_free(temp2);
- return NULL;
-}
-
-
-static int freadline_callback(FILE *f, int fd)
-{
- int c;
-
- c = getc(f);
- if (c == EOF && ferror(f))
- return -1;
- else if (c == EOF)
- return -2;
- else
- return c;
-}
-
-
-Dynstr *dynstr_freadline(FILE *file)
-{
- return readline_helper(freadline_callback, file, -1);
-}
-
-
-static int readline_callback(FILE *f, int fd)
-{
- unsigned char c;
- int n;
-
- do {
- n = read(fd, &c, 1);
- } while (n == -1 && errno == EINTR);
- if (n == 0)
- return -2;
- else if (n == -1)
- return -1;
- else
- return c;
-}
-
-
-Dynstr *dynstr_readline(int file)
-{
- return readline_helper(readline_callback, NULL, file);
-}
-