]> git.wincent.com - wikitext.git/blob - ext/str.c
dca5713290e0a1f06c06d3ed9d67fed29c84f119
[wikitext.git] / 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_size(long len)
40 {
41     str_t *str      = ALLOC_N(str_t, 1);
42     str->ptr        = ALLOC_N(char, len + STR_OVERALLOC);
43     str->len        = 0;
44     str->capacity   = len + STR_OVERALLOC;
45     return str;
46 }
47
48 str_t *str_new_copy(const char *src, long len)
49 {
50     str_t *str      = ALLOC_N(str_t, 1);
51     str->ptr        = ALLOC_N(char, len + STR_OVERALLOC);
52     memcpy(str->ptr, src, len);
53     str->len        = len;
54     str->capacity   = len + STR_OVERALLOC;
55     return str;
56 }
57
58 str_t *str_new_from_string(VALUE string)
59 {
60     string = StringValue(string);
61     return str_new_copy(RSTRING_PTR(string), RSTRING_LEN(string));
62 }
63
64 VALUE string_from_str(str_t *str)
65 {
66     return rb_str_new(str->ptr, str->len);
67 }
68
69 void str_grow(str_t *str, long len)
70 {
71     if (str->capacity < len)
72     {
73         if (str->ptr)
74             REALLOC_N(str->ptr, char, len + STR_OVERALLOC);
75         else
76             str->ptr = ALLOC_N(char, len + STR_OVERALLOC);
77         str->capacity = len + STR_OVERALLOC;
78     }
79 }
80
81 void str_append(str_t *str, const char *src, long len)
82 {
83     long new_len = str->len + len;
84     if (str->capacity < new_len)
85     {
86         if (str->ptr)
87             REALLOC_N(str->ptr, char, new_len + STR_OVERALLOC);
88         else
89             str->ptr = ALLOC_N(char, new_len + STR_OVERALLOC);
90         str->capacity = new_len + STR_OVERALLOC;
91     }
92     memcpy(str->ptr + str->len, src, len);
93     str->len = new_len;
94 }
95
96 void str_append_str(str_t *str, str_t *other)
97 {
98     str_append(str, other->ptr, other->len);
99 }
100
101 void str_append_string(str_t *str, VALUE other)
102 {
103     str_append(str, RSTRING_PTR(other), RSTRING_LEN(other));
104 }
105
106 void str_swap(str_t **a, str_t **b)
107 {
108     str_t *c;
109     c = *a;
110     *a = *b;
111     *b = c;
112 }
113
114 void str_clear(str_t *str)
115 {
116     str->len = 0;
117 }
118
119 void str_free(str_t *str)
120 {
121     if (str->ptr)
122         free(str->ptr);
123     free(str);
124 }