autobuf: work around a warning
[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, unsigned 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, unsigned int new_size)
102 {
103   /* for the the string terminator */
104   new_size++;
105
106   if (autobuf->size >= INT_MAX) {
107     return -1;
108   }
109
110   if (new_size > (unsigned int) autobuf->size) {
111     char *p;
112     int roundUpSize = ROUND_UP_TO_POWER_OF_2(new_size, AUTOBUFCHUNK);
113     p = realloc(autobuf->buf, roundUpSize);
114     if (p == NULL) {
115 #ifdef _WIN32
116       WSASetLastError(ENOMEM);
117 #else /* _WIN32 */
118       errno = ENOMEM;
119 #endif /* _WIN32 */
120       return -1;
121     }
122     autobuf->buf = p;
123
124     memset(&autobuf->buf[autobuf->size], 0, roundUpSize - autobuf->size);
125     autobuf->size = roundUpSize;
126   }
127   return 0;
128 }
129
130 int
131 abuf_vappendf(struct autobuf *autobuf, const char *format, va_list ap)
132 {
133   int rc;
134   int min_size;
135   va_list ap2;
136   va_copy(ap2, ap);
137   rc = vsnprintf(autobuf->buf + autobuf->len, autobuf->size - autobuf->len, format, ap);
138   va_end(ap);
139   min_size = autobuf->len + rc;
140   if (min_size >= autobuf->size) {
141     if (autobuf_enlarge(autobuf, min_size) < 0) {
142       autobuf->buf[autobuf->len] = '\0';
143       va_end(ap2);
144       return -1;
145     }
146     vsnprintf(autobuf->buf + autobuf->len, autobuf->size - autobuf->len, format, ap2);
147   }
148   va_end(ap2);
149   autobuf->len = min_size;
150   return 0;
151 }
152
153 int
154 abuf_appendf(struct autobuf *autobuf, const char *fmt, ...)
155 {
156   int rv;
157   va_list ap;
158   va_start(ap, fmt);
159   rv = abuf_vappendf(autobuf, fmt, ap);
160   va_end(ap);
161   return rv;
162 }
163
164 int
165 abuf_puts(struct autobuf *autobuf, const char *s)
166 {
167   int len; 
168
169   if (NULL == s) return 0;
170   len = strlen(s);
171   if (autobuf_enlarge(autobuf, autobuf->len + len + 1) < 0) {
172     return -1;
173   }
174   strcpy(autobuf->buf + autobuf->len, s);
175   autobuf->len += len;
176   return len;
177 }
178
179 int
180 abuf_concat(struct autobuf *autobuf, struct autobuf *s)
181 {
182   int len;
183
184   if (NULL == s) return 0;
185   len = s->len;
186   if (autobuf_enlarge(autobuf, autobuf->len + len + 1) < 0) {
187     return -1;
188   }
189   strcpy(autobuf->buf + autobuf->len, s->buf);
190   autobuf->len += len;
191   return len;
192 }
193
194 int
195 abuf_strftime(struct autobuf *autobuf, const char *format, const struct tm *tm)
196 {
197   int rc = strftime(autobuf->buf + autobuf->len, autobuf->size - autobuf->len, format, tm);
198   if (rc == 0) {
199     /* we had an error! Probably the buffer too small. So we add some bytes. */
200     if (autobuf_enlarge(autobuf, autobuf->size + AUTOBUFCHUNK) < 0) {
201       autobuf->buf[autobuf->len] = '\0';
202       return -1;
203     }
204     rc = strftime(autobuf->buf + autobuf->len, autobuf->size - autobuf->len, format, tm);
205   }
206   autobuf->len += rc;
207   return rc;
208 }
209
210 int
211 abuf_memcpy(struct autobuf *autobuf, const void *p, const unsigned int len)
212 {
213   if (autobuf_enlarge(autobuf, autobuf->len + len) < 0) {
214     return -1;
215   }
216   memcpy(autobuf->buf + autobuf->len, p, len);
217   autobuf->len += len;
218   return len;
219 }
220
221 int
222 abuf_memcpy_prefix(struct autobuf *autobuf, const void *p, const unsigned int len)
223 {
224   if (autobuf_enlarge(autobuf, autobuf->len + len) < 0) {
225     return -1;
226   }
227   memmove(&autobuf->buf[len], autobuf->buf, autobuf->len);
228   memcpy(autobuf->buf, p, len);
229   autobuf->len += len;
230   return len;
231 }
232
233 int
234 abuf_pull(struct autobuf * autobuf, int len) {
235   char *p;
236   size_t newsize;
237
238   if (len != autobuf->len) {
239     memmove(autobuf->buf, &autobuf->buf[len], autobuf->len - len);
240   }
241   autobuf->len -= len;
242
243   newsize = ROUND_UP_TO_POWER_OF_2(autobuf->len + 1, AUTOBUFCHUNK);
244   p = realloc(autobuf->buf, newsize);
245   if (p == NULL) {
246 #ifdef _WIN32
247     WSASetLastError(ENOMEM);
248 #else /* _WIN32 */
249     errno = ENOMEM;
250 #endif /* _WIN32 */
251     return -1;
252   }
253   autobuf->buf = p;
254   autobuf->size = newsize;
255   return 0;
256 }
257 /*
258  * Local Variables:
259  * mode: c
260  * style: linux
261  * c-basic-offset: 4
262  * indent-tabs-mode: nil
263  * End:
264  */