7 #ifdef STORAGE_WITH_MATH
12 #include "web_buffer.h"
15 #define BUFFER_OVERFLOW_EOF "EOF"
17 static inline void buffer_overflow_init(BUFFER *b)
19 b->buffer[b->size] = '\0';
20 strcpy(&b->buffer[b->size + 1], BUFFER_OVERFLOW_EOF);
23 #ifdef NETDATA_INTERNAL_CHECKS
24 #define buffer_overflow_check(b) _buffer_overflow_check(b, __FILE__, __FUNCTION__, __LINE__)
26 #define buffer_overflow_check(b)
29 static inline void _buffer_overflow_check(BUFFER *b, const char *file, const char *function, const unsigned long line)
31 if(b->len > b->size) {
32 error("BUFFER: length %ld is above size %ld, at line %lu, at function %s() of file '%s'.", b->len, b->size, line, function, file);
36 if(b->buffer[b->size] != '\0' || strcmp(&b->buffer[b->size + 1], BUFFER_OVERFLOW_EOF)) {
37 error("BUFFER: detected overflow at line %lu, at function %s() of file '%s'.", line, function, file);
38 buffer_overflow_init(b);
43 void buffer_reset(BUFFER *wb)
47 wb->contenttype = CT_TEXT_PLAIN;
51 buffer_overflow_check(wb);
54 const char *buffer_tostring(BUFFER *wb)
56 buffer_need_bytes(wb, 1);
57 wb->buffer[wb->len] = '\0';
59 buffer_overflow_check(wb);
64 void buffer_char_replace(BUFFER *wb, char from, char to)
66 char *s = wb->buffer, *end = &wb->buffer[wb->len];
69 if(*s == from) *s = to;
73 buffer_overflow_check(wb);
77 void buffer_strcat(BUFFER *wb, const char *txt)
79 if(unlikely(!txt || !*txt)) return;
81 buffer_need_bytes(wb, 1);
83 char *s = &wb->buffer[wb->len], *start, *end = &wb->buffer[wb->size];
87 while(*txt && s != end)
93 buffer_overflow_check(wb);
96 debug(D_WEB_BUFFER, "strcat(): increasing web_buffer at position %ld, size = %ld\n", wb->len, wb->size);
98 buffer_increase(wb, len);
99 buffer_strcat(wb, txt);
102 // terminate the string
103 // without increasing the length
104 buffer_need_bytes(wb, (size_t)1);
105 wb->buffer[wb->len] = '\0';
110 void buffer_snprintf(BUFFER *wb, size_t len, const char *fmt, ...)
112 if(unlikely(!fmt || !*fmt)) return;
114 buffer_need_bytes(wb, len + 1);
118 wb->len += vsnprintfz(&wb->buffer[wb->len], len, fmt, args);
121 buffer_overflow_check(wb);
123 // the buffer is \0 terminated by vsnprintfz
126 void buffer_vsprintf(BUFFER *wb, const char *fmt, va_list args)
128 if(unlikely(!fmt || !*fmt)) return;
130 buffer_need_bytes(wb, 2);
132 size_t len = wb->size - wb->len - 1;
134 wb->len += vsnprintfz(&wb->buffer[wb->len], len, fmt, args);
136 buffer_overflow_check(wb);
138 // the buffer is \0 terminated by vsnprintfz
141 void buffer_sprintf(BUFFER *wb, const char *fmt, ...)
143 if(unlikely(!fmt || !*fmt)) return;
145 buffer_need_bytes(wb, 2);
147 size_t len = wb->size - wb->len - 1;
152 wrote = (size_t) vsnprintfz(&wb->buffer[wb->len], len, fmt, args);
155 if(unlikely(wrote >= len)) {
156 // there is bug in vsnprintf() and it returns
157 // a number higher to len, but it does not
158 // overflow the buffer.
159 // our buffer overflow detector will log it
161 buffer_overflow_check(wb);
163 debug(D_WEB_BUFFER, "web_buffer_sprintf(): increasing web_buffer at position %ld, size = %ld\n", wb->len, wb->size);
164 buffer_need_bytes(wb, len + WEB_DATA_LENGTH_INCREASE_STEP);
167 buffer_vsprintf(wb, fmt, args);
173 // the buffer is \0 terminated by vsnprintf
177 void buffer_rrd_value(BUFFER *wb, calculated_number value)
179 buffer_need_bytes(wb, 50);
180 wb->len += print_calculated_number(&wb->buffer[wb->len], value);
183 buffer_need_bytes(wb, 1);
184 wb->buffer[wb->len] = '\0';
186 buffer_overflow_check(wb);
189 // generate a javascript date, the fastest possible way...
190 void buffer_jsdate(BUFFER *wb, int year, int month, int day, int hours, int minutes, int seconds)
193 // 01234567890123456789012345678901234
194 // Date(2014,04,01,03,28,20)
196 buffer_need_bytes(wb, 30);
198 char *b = &wb->buffer[wb->len], *p;
199 unsigned int *q = (unsigned int *)b;
201 #if __BYTE_ORDER__ == __ORDER_LITTLE_ENDIAN__
202 *q++ = 0x65746144; // "Date" backwards.
204 *q++ = 0x44617465; // "Date"
209 *p++ = '0' + year / 1000; year %= 1000;
210 *p++ = '0' + year / 100; year %= 100;
211 *p++ = '0' + year / 10;
212 *p++ = '0' + year % 10;
214 *p = '0' + month / 10; if (*p != '0') p++;
215 *p++ = '0' + month % 10;
217 *p = '0' + day / 10; if (*p != '0') p++;
218 *p++ = '0' + day % 10;
220 *p = '0' + hours / 10; if (*p != '0') p++;
221 *p++ = '0' + hours % 10;
223 *p = '0' + minutes / 10; if (*p != '0') p++;
224 *p++ = '0' + minutes % 10;
226 *p = '0' + seconds / 10; if (*p != '0') p++;
227 *p++ = '0' + seconds % 10;
229 unsigned short *r = (unsigned short *)p;
231 #if __BYTE_ORDER__ == __ORDER_LITTLE_ENDIAN__
232 *r++ = 0x0029; // ")\0" backwards.
234 *r++ = 0x2900; // ")\0"
237 wb->len += (size_t)((char *)r - b - 1);
240 wb->buffer[wb->len] = '\0';
241 buffer_overflow_check(wb);
244 // generate a date, the fastest possible way...
245 void buffer_date(BUFFER *wb, int year, int month, int day, int hours, int minutes, int seconds)
248 // 01234567890123456789012345678901234
249 // 2014-04-01 03:28:20
251 buffer_need_bytes(wb, 36);
253 char *b = &wb->buffer[wb->len];
256 *p++ = '0' + year / 1000; year %= 1000;
257 *p++ = '0' + year / 100; year %= 100;
258 *p++ = '0' + year / 10;
259 *p++ = '0' + year % 10;
261 *p++ = '0' + month / 10;
262 *p++ = '0' + month % 10;
264 *p++ = '0' + day / 10;
265 *p++ = '0' + day % 10;
267 *p++ = '0' + hours / 10;
268 *p++ = '0' + hours % 10;
270 *p++ = '0' + minutes / 10;
271 *p++ = '0' + minutes % 10;
273 *p++ = '0' + seconds / 10;
274 *p++ = '0' + seconds % 10;
277 wb->len += (size_t)(p - b);
280 wb->buffer[wb->len] = '\0';
281 buffer_overflow_check(wb);
284 BUFFER *buffer_create(size_t size)
288 debug(D_WEB_BUFFER, "Creating new web buffer of size %zu.", size);
290 b = calloc(1, sizeof(BUFFER));
292 error("Cannot allocate a web_buffer.");
296 b->buffer = malloc(size + sizeof(BUFFER_OVERFLOW_EOF) + 2);
298 error("Cannot allocate a buffer of size %zu.", size + sizeof(BUFFER_OVERFLOW_EOF) + 2);
304 b->contenttype = CT_TEXT_PLAIN;
305 buffer_overflow_init(b);
306 buffer_overflow_check(b);
311 void buffer_free(BUFFER *b)
313 buffer_overflow_check(b);
315 debug(D_WEB_BUFFER, "Freeing web buffer of size %zu.", b->size);
321 void buffer_increase(BUFFER *b, size_t free_size_required)
323 buffer_overflow_check(b);
325 size_t left = b->size - b->len;
327 if(left >= free_size_required) return;
329 size_t increase = free_size_required - left;
330 if(increase < WEB_DATA_LENGTH_INCREASE_STEP) increase = WEB_DATA_LENGTH_INCREASE_STEP;
332 debug(D_WEB_BUFFER, "Increasing data buffer from size %zu to %zu.", b->size, b->size + increase);
334 b->buffer = realloc(b->buffer, b->size + increase + sizeof(BUFFER_OVERFLOW_EOF) + 2);
336 fatal("Failed to increase data buffer from size %zu to %zu.",
337 b->size + sizeof(BUFFER_OVERFLOW_EOF) + 2,
338 b->size + increase + sizeof(BUFFER_OVERFLOW_EOF) + 2);
342 buffer_overflow_init(b);
343 buffer_overflow_check(b);