3 #define BUFFER_OVERFLOW_EOF "EOF"
5 static inline void buffer_overflow_init(BUFFER *b)
7 b->buffer[b->size] = '\0';
8 strcpy(&b->buffer[b->size + 1], BUFFER_OVERFLOW_EOF);
11 #ifdef NETDATA_INTERNAL_CHECKS
12 #define buffer_overflow_check(b) _buffer_overflow_check(b, __FILE__, __FUNCTION__, __LINE__)
14 #define buffer_overflow_check(b)
17 static inline void _buffer_overflow_check(BUFFER *b, const char *file, const char *function, const unsigned long line)
19 if(b->len > b->size) {
20 error("BUFFER: length %zu is above size %zu, at line %lu, at function %s() of file '%s'.", b->len, b->size, line, function, file);
24 if(b->buffer[b->size] != '\0' || strcmp(&b->buffer[b->size + 1], BUFFER_OVERFLOW_EOF)) {
25 error("BUFFER: detected overflow at line %lu, at function %s() of file '%s'.", line, function, file);
26 buffer_overflow_init(b);
31 void buffer_reset(BUFFER *wb)
35 wb->contenttype = CT_TEXT_PLAIN;
40 buffer_overflow_check(wb);
43 const char *buffer_tostring(BUFFER *wb)
45 buffer_need_bytes(wb, 1);
46 wb->buffer[wb->len] = '\0';
48 buffer_overflow_check(wb);
53 void buffer_char_replace(BUFFER *wb, char from, char to)
55 char *s = wb->buffer, *end = &wb->buffer[wb->len];
58 if(*s == from) *s = to;
62 buffer_overflow_check(wb);
65 // This trick seems to give an 80% speed increase in 32bit systems
66 // print_calculated_number_llu_r() will just print the digits up to the
67 // point the remaining value fits in 32 bits, and then calls
68 // print_calculated_number_lu_r() to print the rest with 32 bit arithmetic.
70 inline char *print_number_lu_r(char *str, unsigned long uvalue) {
74 do *wstr++ = (char)('0' + (uvalue % 10)); while(uvalue /= 10);
78 inline char *print_number_llu_r(char *str, unsigned long long uvalue) {
82 do *wstr++ = (char)('0' + (uvalue % 10)); while((uvalue /= 10) && uvalue > (unsigned long long)0xffffffff);
83 if(uvalue) return print_number_lu_r(wstr, uvalue);
87 void buffer_print_llu(BUFFER *wb, unsigned long long uvalue)
89 buffer_need_bytes(wb, 50);
91 char *str = &wb->buffer[wb->len];
95 if(uvalue > (unsigned long long)0xffffffff)
96 wstr = print_number_llu_r(wstr, uvalue);
98 wstr = print_number_lu_r(wstr, uvalue);
100 do *wstr++ = (char)('0' + (uvalue % 10)); while(uvalue /= 10);
107 char *begin = str, *end = wstr - 1, aux;
108 while (end > begin) aux = *end, *end-- = *begin, *begin++ = aux;
110 // return the buffer length
111 wb->len += wstr - str;
114 void buffer_strcat(BUFFER *wb, const char *txt)
116 // buffer_sprintf(wb, "%s", txt);
118 if(unlikely(!txt || !*txt)) return;
120 buffer_need_bytes(wb, 1);
122 char *s = &wb->buffer[wb->len], *start, *end = &wb->buffer[wb->size];
123 size_t len = wb->len;
126 while(*txt && s != end)
132 buffer_overflow_check(wb);
135 debug(D_WEB_BUFFER, "strcat(): increasing web_buffer at position %zu, size = %zu\n", wb->len, wb->size);
137 buffer_increase(wb, len);
138 buffer_strcat(wb, txt);
141 // terminate the string
142 // without increasing the length
143 buffer_need_bytes(wb, (size_t)1);
144 wb->buffer[wb->len] = '\0';
148 void buffer_strcat_htmlescape(BUFFER *wb, const char *txt)
150 char b[2] = { [0] = '\0', [1] = '\0' };
154 case '&': buffer_strcat(wb, "&"); break;
155 case '<': buffer_strcat(wb, "<"); break;
156 case '>': buffer_strcat(wb, ">"); break;
157 case '"': buffer_strcat(wb, """); break;
158 case '/': buffer_strcat(wb, "/"); break;
159 case '\'': buffer_strcat(wb, "'"); break;
162 buffer_strcat(wb, b);
169 void buffer_snprintf(BUFFER *wb, size_t len, const char *fmt, ...)
171 if(unlikely(!fmt || !*fmt)) return;
173 buffer_need_bytes(wb, len + 1);
177 wb->len += vsnprintfz(&wb->buffer[wb->len], len, fmt, args);
180 buffer_overflow_check(wb);
182 // the buffer is \0 terminated by vsnprintfz
185 void buffer_vsprintf(BUFFER *wb, const char *fmt, va_list args)
187 if(unlikely(!fmt || !*fmt)) return;
189 buffer_need_bytes(wb, 2);
191 size_t len = wb->size - wb->len - 1;
193 wb->len += vsnprintfz(&wb->buffer[wb->len], len, fmt, args);
195 buffer_overflow_check(wb);
197 // the buffer is \0 terminated by vsnprintfz
200 void buffer_sprintf(BUFFER *wb, const char *fmt, ...)
202 if(unlikely(!fmt || !*fmt)) return;
204 buffer_need_bytes(wb, 2);
206 size_t len = wb->size - wb->len - 1;
211 wrote = (size_t) vsnprintfz(&wb->buffer[wb->len], len, fmt, args);
214 if(unlikely(wrote >= len)) {
216 buffer_overflow_check(wb);
218 debug(D_WEB_BUFFER, "web_buffer_sprintf(): increasing web_buffer at position %zu, size = %zu\n", wb->len, wb->size);
219 buffer_need_bytes(wb, len + WEB_DATA_LENGTH_INCREASE_STEP);
222 buffer_vsprintf(wb, fmt, args);
228 // the buffer is \0 terminated by vsnprintf
232 void buffer_rrd_value(BUFFER *wb, calculated_number value)
234 buffer_need_bytes(wb, 50);
236 if(isnan(value) || isinf(value)) {
237 buffer_strcat(wb, "null");
241 wb->len += print_calculated_number(&wb->buffer[wb->len], value);
244 buffer_need_bytes(wb, 1);
245 wb->buffer[wb->len] = '\0';
247 buffer_overflow_check(wb);
250 // generate a javascript date, the fastest possible way...
251 void buffer_jsdate(BUFFER *wb, int year, int month, int day, int hours, int minutes, int seconds)
254 // 01234567890123456789012345678901234
255 // Date(2014,04,01,03,28,20)
257 buffer_need_bytes(wb, 30);
259 char *b = &wb->buffer[wb->len], *p;
260 unsigned int *q = (unsigned int *)b;
262 #if __BYTE_ORDER__ == __ORDER_LITTLE_ENDIAN__
263 *q++ = 0x65746144; // "Date" backwards.
265 *q++ = 0x44617465; // "Date"
270 *p++ = '0' + year / 1000; year %= 1000;
271 *p++ = '0' + year / 100; year %= 100;
272 *p++ = '0' + year / 10;
273 *p++ = '0' + year % 10;
275 *p = '0' + month / 10; if (*p != '0') p++;
276 *p++ = '0' + month % 10;
278 *p = '0' + day / 10; if (*p != '0') p++;
279 *p++ = '0' + day % 10;
281 *p = '0' + hours / 10; if (*p != '0') p++;
282 *p++ = '0' + hours % 10;
284 *p = '0' + minutes / 10; if (*p != '0') p++;
285 *p++ = '0' + minutes % 10;
287 *p = '0' + seconds / 10; if (*p != '0') p++;
288 *p++ = '0' + seconds % 10;
290 unsigned short *r = (unsigned short *)p;
292 #if __BYTE_ORDER__ == __ORDER_LITTLE_ENDIAN__
293 *r++ = 0x0029; // ")\0" backwards.
295 *r++ = 0x2900; // ")\0"
298 wb->len += (size_t)((char *)r - b - 1);
301 wb->buffer[wb->len] = '\0';
302 buffer_overflow_check(wb);
305 // generate a date, the fastest possible way...
306 void buffer_date(BUFFER *wb, int year, int month, int day, int hours, int minutes, int seconds)
309 // 01234567890123456789012345678901234
310 // 2014-04-01 03:28:20
312 buffer_need_bytes(wb, 36);
314 char *b = &wb->buffer[wb->len];
317 *p++ = '0' + year / 1000; year %= 1000;
318 *p++ = '0' + year / 100; year %= 100;
319 *p++ = '0' + year / 10;
320 *p++ = '0' + year % 10;
322 *p++ = '0' + month / 10;
323 *p++ = '0' + month % 10;
325 *p++ = '0' + day / 10;
326 *p++ = '0' + day % 10;
328 *p++ = '0' + hours / 10;
329 *p++ = '0' + hours % 10;
331 *p++ = '0' + minutes / 10;
332 *p++ = '0' + minutes % 10;
334 *p++ = '0' + seconds / 10;
335 *p++ = '0' + seconds % 10;
338 wb->len += (size_t)(p - b);
341 wb->buffer[wb->len] = '\0';
342 buffer_overflow_check(wb);
345 BUFFER *buffer_create(size_t size)
349 debug(D_WEB_BUFFER, "Creating new web buffer of size %zu.", size);
351 b = callocz(1, sizeof(BUFFER));
352 b->buffer = mallocz(size + sizeof(BUFFER_OVERFLOW_EOF) + 2);
355 b->contenttype = CT_TEXT_PLAIN;
356 buffer_overflow_init(b);
357 buffer_overflow_check(b);
362 void buffer_free(BUFFER *b)
364 buffer_overflow_check(b);
366 debug(D_WEB_BUFFER, "Freeing web buffer of size %zu.", b->size);
372 void buffer_increase(BUFFER *b, size_t free_size_required)
374 buffer_overflow_check(b);
376 size_t left = b->size - b->len;
378 if(left >= free_size_required) return;
380 size_t increase = free_size_required - left;
381 if(increase < WEB_DATA_LENGTH_INCREASE_STEP) increase = WEB_DATA_LENGTH_INCREASE_STEP;
383 debug(D_WEB_BUFFER, "Increasing data buffer from size %zu to %zu.", b->size, b->size + increase);
385 b->buffer = reallocz(b->buffer, b->size + increase + sizeof(BUFFER_OVERFLOW_EOF) + 2);
388 buffer_overflow_init(b);
389 buffer_overflow_check(b);