9c82fe6f10436100f5f2a1de06b1a86ffc437028
[olsrd.git] / src / common / autobuf.c
1 /*
2  * The olsr.org Optimized Link-State Routing daemon (olsrd)
3  *
4  * (c) by the OLSR project
5  *
6  * See our Git repository to find out who worked on this file
7  * and thus is a copyright holder on it.
8  *
9  * All rights reserved.
10  *
11  * Redistribution and use in source and binary forms, with or without
12  * modification, are permitted provided that the following conditions
13  * are met:
14  *
15  * * Redistributions of source code must retain the above copyright
16  *   notice, this list of conditions and the following disclaimer.
17  * * Redistributions in binary form must reproduce the above copyright
18  *   notice, this list of conditions and the following disclaimer in
19  *   the documentation and/or other materials provided with the
20  *   distribution.
21  * * Neither the name of olsr.org, olsrd nor the names of its
22  *   contributors may be used to endorse or promote products derived
23  *   from this software without specific prior written permission.
24  *
25  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
26  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
27  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
28  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
29  * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
30  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
31  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
32  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
33  * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
34  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
35  * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
36  * POSSIBILITY OF SUCH DAMAGE.
37  *
38  * Visit http://www.olsr.org for more information.
39  *
40  * If you find this software useful feel free to make a donation
41  * to the project. For more information see the website or contact
42  * the copyright holders.
43  *
44  */
45
46 #include "common/autobuf.h"
47 #include "defs.h"
48
49 #include <stdio.h>
50 #include <string.h>
51 #include <stdlib.h>
52 #include <errno.h>
53 #include <limits.h>
54 #include <assert.h>
55
56
57 static int autobuf_enlarge(struct autobuf *autobuf, int new_size);
58
59 static int ROUND_UP_TO_POWER_OF_2(int val, int pow2) {
60   assert(val >= 0);
61   assert(pow2 > 0);
62
63   if (val <= (INT32_MAX - (pow2 - 1))) {
64     /* no overflow */
65     return ((val + (pow2 - 1)) & ~(pow2 - 1));
66   }
67
68   /* overflow */
69   return (INT32_MAX & ~(pow2 - 1));
70 }
71
72 int
73 abuf_init(struct autobuf *autobuf, int initial_size)
74 {
75   autobuf->len = 0;
76   if (initial_size <= 0) {
77     autobuf->size = 0;
78     autobuf->buf = NULL;
79     return 0;
80   }
81   autobuf->size = ROUND_UP_TO_POWER_OF_2(initial_size, AUTOBUFCHUNK);
82   autobuf->buf = calloc(autobuf->size, 1);
83   if (autobuf->buf == NULL) {
84     autobuf->size = 0;
85     return -1;
86   }
87   *autobuf->buf = '\0';
88   return 0;
89 }
90
91 void
92 abuf_free(struct autobuf *autobuf)
93 {
94   free(autobuf->buf);
95   autobuf->buf = NULL;
96   autobuf->len = 0;
97   autobuf->size = 0;
98 }
99
100 static int
101 autobuf_enlarge(struct autobuf *autobuf, int new_size)
102 {
103   new_size++;
104
105   if (autobuf->size >= INT_MAX) {
106     return -1;
107   }
108
109   if (new_size > autobuf->size) {
110     char *p;
111     int roundUpSize = ROUND_UP_TO_POWER_OF_2(new_size, AUTOBUFCHUNK);
112     p = realloc(autobuf->buf, roundUpSize);
113     if (p == NULL) {
114 #ifdef _WIN32
115       WSASetLastError(ENOMEM);
116 #else /* _WIN32 */
117       errno = ENOMEM;
118 #endif /* _WIN32 */
119       return -1;
120     }
121     autobuf->buf = p;
122
123     memset(&autobuf->buf[autobuf->size], 0, roundUpSize - autobuf->size);
124     autobuf->size = roundUpSize;
125   }
126   return 0;
127 }
128
129 int
130 abuf_vappendf(struct autobuf *autobuf, const char *format, va_list ap)
131 {
132   int rc;
133   int min_size;
134   va_list ap2;
135   va_copy(ap2, ap);
136   rc = vsnprintf(autobuf->buf + autobuf->len, autobuf->size - autobuf->len, format, ap);
137   va_end(ap);
138   min_size = autobuf->len + rc;
139   if (min_size >= autobuf->size) {
140     if (autobuf_enlarge(autobuf, min_size) < 0) {
141       autobuf->buf[autobuf->len] = '\0';
142       va_end(ap2);
143       return -1;
144     }
145     vsnprintf(autobuf->buf + autobuf->len, autobuf->size - autobuf->len, format, ap2);
146   }
147   va_end(ap2);
148   autobuf->len = min_size;
149   return 0;
150 }
151
152 int
153 abuf_appendf(struct autobuf *autobuf, const char *fmt, ...)
154 {
155   int rv;
156   va_list ap;
157   va_start(ap, fmt);
158   rv = abuf_vappendf(autobuf, fmt, ap);
159   va_end(ap);
160   return rv;
161 }
162
163 int
164 abuf_puts(struct autobuf *autobuf, const char *s)
165 {
166   int len; 
167
168   if (NULL == s) return 0;
169   len = strlen(s);
170   if (autobuf_enlarge(autobuf, autobuf->len + len + 1) < 0) {
171     return -1;
172   }
173   strcpy(autobuf->buf + autobuf->len, s);
174   autobuf->len += len;
175   return len;
176 }
177
178 int
179 abuf_concat(struct autobuf *autobuf, struct autobuf *s)
180 {
181   int len;
182
183   if (NULL == s) return 0;
184   len = s->len;
185   if (autobuf_enlarge(autobuf, autobuf->len + len + 1) < 0) {
186     return -1;
187   }
188   strcpy(autobuf->buf + autobuf->len, s->buf);
189   autobuf->len += len;
190   return len;
191 }
192
193 int
194 abuf_strftime(struct autobuf *autobuf, const char *format, const struct tm *tm)
195 {
196   int rc = strftime(autobuf->buf + autobuf->len, autobuf->size - autobuf->len, format, tm);
197   if (rc == 0) {
198     /* we had an error! Probably the buffer too small. So we add some bytes. */
199     if (autobuf_enlarge(autobuf, autobuf->size + AUTOBUFCHUNK) < 0) {
200       autobuf->buf[autobuf->len] = '\0';
201       return -1;
202     }
203     rc = strftime(autobuf->buf + autobuf->len, autobuf->size - autobuf->len, format, tm);
204   }
205   autobuf->len += rc;
206   return rc;
207 }
208
209 int
210 abuf_memcpy(struct autobuf *autobuf, const void *p, const unsigned int len)
211 {
212   if (autobuf_enlarge(autobuf, autobuf->len + len) < 0) {
213     return -1;
214   }
215   memcpy(autobuf->buf + autobuf->len, p, len);
216   autobuf->len += len;
217   return len;
218 }
219
220 int
221 abuf_memcpy_prefix(struct autobuf *autobuf, const void *p, const unsigned int len)
222 {
223   if (autobuf_enlarge(autobuf, autobuf->len + len) < 0) {
224     return -1;
225   }
226   memmove(&autobuf->buf[len], autobuf->buf, autobuf->len);
227   memcpy(autobuf->buf, p, len);
228   autobuf->len += len;
229   return len;
230 }
231
232 int
233 abuf_pull(struct autobuf * autobuf, int len) {
234   char *p;
235   size_t newsize;
236
237   if (len != autobuf->len) {
238     memmove(autobuf->buf, &autobuf->buf[len], autobuf->len - len);
239   }
240   autobuf->len -= len;
241
242   newsize = ROUND_UP_TO_POWER_OF_2(autobuf->len + 1, AUTOBUFCHUNK);
243   p = realloc(autobuf->buf, newsize);
244   if (p == NULL) {
245 #ifdef _WIN32
246     WSASetLastError(ENOMEM);
247 #else /* _WIN32 */
248     errno = ENOMEM;
249 #endif /* _WIN32 */
250     return -1;
251   }
252   autobuf->buf = p;
253   autobuf->size = newsize;
254   return 0;
255 }
256 /*
257  * Local Variables:
258  * mode: c
259  * style: linux
260  * c-basic-offset: 4
261  * indent-tabs-mode: nil
262  * End:
263  */