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