ext/str.c

   1 // Copyright 2008-2009 Wincent Colaiuta. All rights reserved.
   2 //
   3 // Redistribution and use in source and binary forms, with or without
   4 // modification, are permitted provided that the following conditions are met:
   5 //
   6 // 1. Redistributions of source code must retain the above copyright notice,
   7 //    this list of conditions and the following disclaimer.
   8 // 2. Redistributions in binary form must reproduce the above copyright notice,
   9 //    this list of conditions and the following disclaimer in the documentation
  10 //    and/or other materials provided with the distribution.
  11 //
  12 // THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS "AS IS"
  13 // AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  14 // IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  15 // ARE DISCLAIMED. IN NO EVENT SHALL THE COPYRIGHT HOLDERS OR CONTRIBUTORS BE
  16 // LIABLE FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR
  17 // CONSEQUENTIAL DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF
  18 // SUBSTITUTE GOODS OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS
  19 // INTERRUPTION) HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN
  20 // CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE)
  21 // ARISING IN ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  22 // POSSIBILITY OF SUCH DAMAGE.
  23
  24 #include "str.h"
  25
  26 // when allocating memory, reserve a little more than was asked for,
  27 // which can help to avoid subsequent allocations
  28 #define STR_OVERALLOC 256
  29
  30 str_t *str_new(void)
  31 {
  32     str_t *str      = ALLOC_N(str_t, 1);
  33     str->ptr        = NULL;
  34     str->len        = 0;
  35     str->capacity   = 0;
  36     return str;
  37 }
  38
  39 str_t *str_new_copy(const char *src, long len)
  40 {
  41     str_t *str      = ALLOC_N(str_t, 1);
  42     str->ptr        = ALLOC_N(char, len + STR_OVERALLOC);
  43     memcpy(str->ptr, src, len);
  44     str->len        = len;
  45     str->capacity   = len + STR_OVERALLOC;
  46     return str;
  47 }
  48
  49 str_t *str_new_from_string(VALUE string)
  50 {
  51     string = StringValue(string);
  52     return str_new_copy(RSTRING_PTR(string), RSTRING_LEN(string));
  53 }
  54
  55 VALUE string_from_str(str_t *str)
  56 {
  57     return rb_str_new(str->ptr, str->len);
  58 }
  59
  60 void str_grow(str_t *str, long len)
  61 {
  62     if (str->capacity < len)
  63     {
  64         if (str->ptr)
  65             REALLOC_N(str->ptr, char, len + STR_OVERALLOC);
  66         else
  67             str->ptr = ALLOC_N(char, len + STR_OVERALLOC);
  68         str->capacity = len + STR_OVERALLOC;
  69     }
  70 }
  71
  72 void str_append(str_t *str, const char *src, long len)
  73 {
  74     long new_len = str->len + len;
  75     if (str->capacity < new_len)
  76     {
  77         if (str->ptr)
  78             REALLOC_N(str->ptr, char, new_len + STR_OVERALLOC);
  79         else
  80             str->ptr = ALLOC_N(char, new_len + STR_OVERALLOC);
  81         str->capacity = new_len + STR_OVERALLOC;
  82     }
  83     memcpy(str->ptr + str->len, src, len);
  84     str->len = new_len;
  85 }
  86
  87 void str_append_str(str_t *str, str_t *other)
  88 {
  89     str_append(str, other->ptr, other->len);
  90 }
  91
  92 void str_append_string(str_t *str, VALUE other)
  93 {
  94     str_append(str, RSTRING_PTR(other), RSTRING_LEN(other));
  95 }
  96
  97 void str_clear(str_t *str)
  98 {
  99     str->len = 0;
 100 }
 101
 102 void str_free(str_t *str)
 103 {
 104     if (str->ptr)
 105         free(str->ptr);
 106     free(str);
 107 }