src/strbuffer.c

   1 /*
   2  * Copyright (c) 2009, 2010 Petri Lehtinen <petri@digip.org>
   3  *
   4  * Jansson is free software; you can redistribute it and/or modify
   5  * it under the terms of the MIT license. See LICENSE for details.
   6  */
   7
   8 #define _GNU_SOURCE
   9 #include <stdlib.h>
  10 #include <string.h>
  11 #include "jansson_private.h"
  12 #include "strbuffer.h"
  13
  14 #define STRBUFFER_MIN_SIZE  16
  15 #define STRBUFFER_FACTOR    2
  16
  17 int strbuffer_init(strbuffer_t *strbuff)
  18 {
  19     strbuff->size = STRBUFFER_MIN_SIZE;
  20     strbuff->length = 0;
  21
  22     strbuff->value = malloc(strbuff->size);
  23     if(!strbuff->value)
  24         return -1;
  25
  26     /* initialize to empty */
  27     strbuff->value[0] = '\0';
  28     return 0;
  29 }
  30
  31 void strbuffer_close(strbuffer_t *strbuff)
  32 {
  33     free(strbuff->value);
  34     strbuff->size = 0;
  35     strbuff->length = 0;
  36     strbuff->value = NULL;
  37 }
  38
  39 void strbuffer_clear(strbuffer_t *strbuff)
  40 {
  41     strbuff->length = 0;
  42     strbuff->value[0] = '\0';
  43 }
  44
  45 const char *strbuffer_value(const strbuffer_t *strbuff)
  46 {
  47     return strbuff->value;
  48 }
  49
  50 char *strbuffer_steal_value(strbuffer_t *strbuff)
  51 {
  52     char *result = strbuff->value;
  53     strbuffer_init(strbuff);
  54     return result;
  55 }
  56
  57 int strbuffer_append(strbuffer_t *strbuff, const char *string)
  58 {
  59     return strbuffer_append_bytes(strbuff, string, strlen(string));
  60 }
  61
  62 int strbuffer_append_byte(strbuffer_t *strbuff, char byte)
  63 {
  64     return strbuffer_append_bytes(strbuff, &byte, 1);
  65 }
  66
  67 int strbuffer_append_bytes(strbuffer_t *strbuff, const char *data, int size)
  68 {
  69     if(strbuff->length + size >= strbuff->size)
  70     {
  71         strbuff->size = max(strbuff->size * STRBUFFER_FACTOR,
  72                             strbuff->length + size + 1);
  73
  74         strbuff->value = realloc(strbuff->value, strbuff->size);
  75         if(!strbuff->value)
  76             return -1;
  77     }
  78
  79     memcpy(strbuff->value + strbuff->length, data, size);
  80     strbuff->length += size;
  81     strbuff->value[strbuff->length] = '\0';
  82
  83     return 0;
  84 }
  85
  86 char strbuffer_pop(strbuffer_t *strbuff)
  87 {
  88     if(strbuff->length > 0) {
  89         char c = strbuff->value[--strbuff->length];
  90         strbuff->value[strbuff->length] = '\0';
  91         return c;
  92     }
  93     else
  94         return '\0';
  95 }