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;
50 buffer_overflow_check(wb);
53 const char *buffer_tostring(BUFFER *wb)
55 buffer_need_bytes(wb, (size_t)1);
56 wb->buffer[wb->len] = '\0';
58 buffer_overflow_check(wb);
63 void buffer_char_replace(BUFFER *wb, char from, char to)
65 char *s = wb->buffer, *end = &wb->buffer[wb->len];
68 if(*s == from) *s = to;
72 buffer_overflow_check(wb);
76 void buffer_strcat(BUFFER *wb, const char *txt)
78 if(unlikely(!txt || !*txt)) return;
80 buffer_need_bytes(wb, (size_t)(1));
82 char *s = &wb->buffer[wb->len], *end = &wb->buffer[wb->size];
85 while(*txt && s != end) {
91 buffer_overflow_check(wb);
94 debug(D_WEB_BUFFER, "strcat(): increasing web_buffer at position %ld, size = %ld\n", wb->len, wb->size);
96 buffer_increase(wb, len);
97 buffer_strcat(wb, txt);
100 // terminate the string
101 // without increasing the length
102 buffer_need_bytes(wb, (size_t)1);
103 wb->buffer[wb->len] = '\0';
108 void buffer_snprintf(BUFFER *wb, size_t len, const char *fmt, ...)
110 if(unlikely(!fmt || !*fmt)) return;
112 buffer_need_bytes(wb, len+1);
116 wb->len += vsnprintf(&wb->buffer[wb->len], len+1, fmt, args);
119 buffer_overflow_check(wb);
121 // the buffer is \0 terminated by vsnprintf
124 void buffer_vsprintf(BUFFER *wb, const char *fmt, va_list args)
126 if(unlikely(!fmt || !*fmt)) return;
128 buffer_need_bytes(wb, 1);
130 size_t len = wb->size - wb->len;
132 wb->len += vsnprintf(&wb->buffer[wb->len], len, fmt, args);
134 buffer_overflow_check(wb);
136 // the buffer is \0 terminated by vsnprintf
139 void buffer_sprintf(BUFFER *wb, const char *fmt, ...)
141 if(unlikely(!fmt || !*fmt)) return;
143 buffer_need_bytes(wb, 1);
145 size_t len = wb->size - wb->len, wrote;
149 wrote = (size_t) vsnprintf(&wb->buffer[wb->len], len, fmt, args);
152 if(unlikely(wrote >= len)) {
153 // there is bug in vsnprintf() and it returns
154 // a number higher to len, but it does not
155 // overflow the buffer.
156 // our buffer overflow detector will log it
158 buffer_overflow_check(wb);
160 debug(D_WEB_BUFFER, "web_buffer_sprintf(): increasing web_buffer at position %ld, size = %ld\n", wb->len, wb->size);
161 buffer_need_bytes(wb, len + WEB_DATA_LENGTH_INCREASE_STEP);
164 buffer_vsprintf(wb, fmt, args);
170 // the buffer is \0 terminated by vsnprintf
174 void buffer_rrd_value(BUFFER *wb, calculated_number value)
176 buffer_need_bytes(wb, 50);
177 wb->len += print_calculated_number(&wb->buffer[wb->len], value);
180 buffer_need_bytes(wb, 1);
181 wb->buffer[wb->len] = '\0';
183 buffer_overflow_check(wb);
186 // generate a javascript date, the fastest possible way...
187 void buffer_jsdate(BUFFER *wb, int year, int month, int day, int hours, int minutes, int seconds)
190 // 01234567890123456789012345678901234
191 // Date(2014,04,01,03,28,20)
193 buffer_need_bytes(wb, 30);
195 char *b = &wb->buffer[wb->len];
203 b[i++]= (char) (48 + year / 1000); year -= (year / 1000) * 1000;
204 b[i++]= (char) (48 + year / 100); year -= (year / 100) * 100;
205 b[i++]= (char) (48 + year / 10);
206 b[i++]= (char) (48 + year % 10);
208 b[i]= (char) (48 + month / 10); if(b[i] != '0') i++;
209 b[i++]= (char) (48 + month % 10);
211 b[i]= (char) (48 + day / 10); if(b[i] != '0') i++;
212 b[i++]= (char) (48 + day % 10);
214 b[i]= (char) (48 + hours / 10); if(b[i] != '0') i++;
215 b[i++]= (char) (48 + hours % 10);
217 b[i]= (char) (48 + minutes / 10); if(b[i] != '0') i++;
218 b[i++]= (char) (48 + minutes % 10);
220 b[i]= (char) (48 + seconds / 10); if(b[i] != '0') i++;
221 b[i++]= (char) (48 + seconds % 10);
228 wb->buffer[wb->len] = '\0';
229 buffer_overflow_check(wb);
232 // generate a date, the fastest possible way...
233 void buffer_date(BUFFER *wb, int year, int month, int day, int hours, int minutes, int seconds)
236 // 01234567890123456789012345678901234
237 // 2014-04-01 03:28:20
239 buffer_need_bytes(wb, 36);
241 char *b = &wb->buffer[wb->len];
244 b[i++]= (char) (48 + year / 1000); year -= (year / 1000) * 1000;
245 b[i++]= (char) (48 + year / 100); year -= (year / 100) * 100;
246 b[i++]= (char) (48 + year / 10);
247 b[i++]= (char) (48 + year % 10);
249 b[i++]= (char) (48 + month / 10);
250 b[i++]= (char) (48 + month % 10);
252 b[i++]= (char) (48 + day / 10);
253 b[i++]= (char) (48 + day % 10);
255 b[i++]= (char) (48 + hours / 10);
256 b[i++]= (char) (48 + hours % 10);
258 b[i++]= (char) (48 + minutes / 10);
259 b[i++]= (char) (48 + minutes % 10);
261 b[i++]= (char) (48 + seconds / 10);
262 b[i++]= (char) (48 + seconds % 10);
268 wb->buffer[wb->len] = '\0';
269 buffer_overflow_check(wb);
272 BUFFER *buffer_create(long size)
276 debug(D_WEB_BUFFER, "Creating new web buffer of size %d.", size);
278 b = calloc(1, sizeof(BUFFER));
280 error("Cannot allocate a web_buffer.");
284 b->buffer = malloc(size + sizeof(BUFFER_OVERFLOW_EOF) + 2);
286 error("Cannot allocate a buffer of size %u.", size + sizeof(BUFFER_OVERFLOW_EOF) + 2);
292 b->contenttype = CT_TEXT_PLAIN;
293 buffer_overflow_init(b);
294 buffer_overflow_check(b);
299 void buffer_free(BUFFER *b)
301 buffer_overflow_check(b);
303 debug(D_WEB_BUFFER, "Freeing web buffer of size %d.", b->size);
305 if(b->buffer) free(b->buffer);
309 void buffer_increase(BUFFER *b, size_t free_size_required)
311 buffer_overflow_check(b);
313 size_t left = b->size - b->len;
315 if(left >= free_size_required) return;
317 size_t increase = free_size_required - left;
318 if(increase < WEB_DATA_LENGTH_INCREASE_STEP) increase = WEB_DATA_LENGTH_INCREASE_STEP;
320 debug(D_WEB_BUFFER, "Increasing data buffer from size %d to %d.", b->size, b->size + increase);
322 b->buffer = realloc(b->buffer, b->size + increase + sizeof(BUFFER_OVERFLOW_EOF) + 2);
323 if(!b->buffer) fatal("Failed to increase data buffer from size %d to %d.", b->size + sizeof(BUFFER_OVERFLOW_EOF) + 2, b->size + increase + sizeof(BUFFER_OVERFLOW_EOF) + 2);
327 buffer_overflow_init(b);
328 buffer_overflow_check(b);