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 #define buffer_overflow_check(b) _buffer_overflow_check(b, __FILE__, __FUNCTION__, __LINE__)
25 static inline void _buffer_overflow_check(BUFFER *b, const char *file, const char *function, const unsigned long line)
27 if(b->len > b->size) {
28 error("BUFFER: length %ld is above size %ld, at line %lu, at function %s() of file '%s'.", b->len, b->size, line, function, file);
32 if(b->buffer[b->size] != '\0' || strcmp(&b->buffer[b->size + 1], BUFFER_OVERFLOW_EOF)) {
33 error("BUFFER: detected overflow at line %lu, at function %s() of file '%s'.", line, function, file);
34 buffer_overflow_init(b);
39 void buffer_reset(BUFFER *wb)
43 wb->contenttype = CT_TEXT_PLAIN;
46 buffer_overflow_check(wb);
49 const char *buffer_tostring(BUFFER *wb)
51 buffer_need_bytes(wb, 1);
52 wb->buffer[wb->len] = '\0';
54 buffer_overflow_check(wb);
59 void buffer_char_replace(BUFFER *wb, char from, char to)
61 char *s = wb->buffer, *end = &wb->buffer[wb->len];
64 if(*s == from) *s = to;
68 buffer_overflow_check(wb);
72 void buffer_strcat(BUFFER *wb, const char *txt)
74 if(wb->size - wb->len < 512)
75 buffer_need_bytes(wb, 512);
77 char *s = &wb->buffer[wb->len], *end = &wb->buffer[wb->size];
80 while(*txt && s != end) {
86 buffer_overflow_check(wb);
89 debug(D_WEB_BUFFER, "strcat(): increasing web_buffer at position %ld, size = %ld\n", wb->len, wb->size);
91 buffer_increase(wb, len);
92 buffer_strcat(wb, txt);
95 // terminate the string
96 // without increasing the length
97 buffer_need_bytes(wb, 1);
98 wb->buffer[wb->len] = '\0';
103 void buffer_snprintf(BUFFER *wb, size_t len, const char *fmt, ...)
105 buffer_need_bytes(wb, len+1);
109 wb->len += vsnprintf(&wb->buffer[wb->len], len+1, fmt, args);
112 buffer_overflow_check(wb);
114 // the buffer is \0 terminated by vsnprintf
117 void buffer_vsprintf(BUFFER *wb, const char *fmt, va_list args)
119 size_t len = wb->size - wb->len;
120 if(unlikely(!len)) return;
122 wb->len += vsnprintf(&wb->buffer[wb->len], len, fmt, args);
124 buffer_overflow_check(wb);
126 // the buffer is \0 terminated by vsnprintf
129 void buffer_sprintf(BUFFER *wb, const char *fmt, ...)
131 if(unlikely(wb->len > wb->size)) {
132 error("web_buffer_sprintf(): already overflown length %ld, size = %ld", wb->len, wb->size);
135 // if(wb->size - wb->len < 512)
136 // web_buffer_need_bytes(wb, 512);
138 size_t len = wb->size - wb->len, old_len = wb->len, wrote;
140 buffer_need_bytes(wb, len);
144 wrote = vsnprintf(&wb->buffer[wb->len], len, fmt, args);
147 if(unlikely(wrote >= len)) {
148 // there is bug in vsnprintf() and it returns
149 // a number higher to len, but it does not
150 // overflow the buffer.
151 // our buffer overflow detector will log it
153 buffer_overflow_check(wb);
155 debug(D_WEB_BUFFER, "web_buffer_sprintf(): increasing web_buffer at position %ld, size = %ld\n", wb->len, wb->size);
156 buffer_need_bytes(wb, len + WEB_DATA_LENGTH_INCREASE_STEP);
159 buffer_vsprintf(wb, fmt, args);
165 // the buffer is \0 terminated by vsnprintf
169 void buffer_rrd_value(BUFFER *wb, calculated_number value)
171 buffer_need_bytes(wb, 50);
172 wb->len += print_calculated_number(&wb->buffer[wb->len], value);
175 buffer_need_bytes(wb, 1);
176 wb->buffer[wb->len] = '\0';
178 buffer_overflow_check(wb);
181 // generate a javascript date, the fastest possible way...
182 void buffer_jsdate(BUFFER *wb, int year, int month, int day, int hours, int minutes, int seconds)
185 // 01234567890123456789012345678901234
186 // Date(2014, 04, 01, 03, 28, 20, 065)
188 buffer_need_bytes(wb, 36);
190 char *b = &wb->buffer[wb->len];
198 b[i++]= 48 + year / 1000; year -= (year / 1000) * 1000;
199 b[i++]= 48 + year / 100; year -= (year / 100) * 100;
200 b[i++]= 48 + year / 10;
201 b[i++]= 48 + year % 10;
204 b[i]= 48 + month / 10; if(b[i] != '0') i++;
205 b[i++]= 48 + month % 10;
208 b[i]= 48 + day / 10; if(b[i] != '0') i++;
209 b[i++]= 48 + day % 10;
212 b[i]= 48 + hours / 10; if(b[i] != '0') i++;
213 b[i++]= 48 + hours % 10;
216 b[i]= 48 + minutes / 10; if(b[i] != '0') i++;
217 b[i++]= 48 + minutes % 10;
220 b[i]= 48 + seconds / 10; if(b[i] != '0') i++;
221 b[i++]= 48 + seconds % 10;
228 buffer_need_bytes(wb, 1);
229 wb->buffer[wb->len] = '\0';
231 buffer_overflow_check(wb);
234 BUFFER *buffer_create(long size)
238 debug(D_WEB_BUFFER, "Creating new web buffer of size %d.", size);
240 b = calloc(1, sizeof(BUFFER));
242 error("Cannot allocate a web_buffer.");
246 b->buffer = malloc(size + sizeof(BUFFER_OVERFLOW_EOF) + 2);
248 error("Cannot allocate a buffer of size %u.", size + sizeof(BUFFER_OVERFLOW_EOF) + 2);
254 b->contenttype = CT_TEXT_PLAIN;
255 buffer_overflow_init(b);
256 buffer_overflow_check(b);
261 void buffer_free(BUFFER *b)
263 buffer_overflow_check(b);
265 debug(D_WEB_BUFFER, "Freeing web buffer of size %d.", b->size);
267 if(b->buffer) free(b->buffer);
271 void buffer_increase(BUFFER *b, long free_size_required)
273 buffer_overflow_check(b);
275 long left = b->size - b->len;
277 if(left >= free_size_required) return;
279 long increase = free_size_required - left;
280 if(increase < WEB_DATA_LENGTH_INCREASE_STEP) increase = WEB_DATA_LENGTH_INCREASE_STEP;
282 debug(D_WEB_BUFFER, "Increasing data buffer from size %d to %d.", b->size, b->size + increase);
284 b->buffer = realloc(b->buffer, b->size + increase + sizeof(BUFFER_OVERFLOW_EOF) + 2);
285 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);
289 buffer_overflow_init(b);
290 buffer_overflow_check(b);