diff options
Diffstat (limited to 'src/web_buffer.c')
| -rwxr-xr-x | src/web_buffer.c | 330 |
1 files changed, 330 insertions, 0 deletions
diff --git a/src/web_buffer.c b/src/web_buffer.c new file mode 100755 index 000000000..482eb3900 --- /dev/null +++ b/src/web_buffer.c @@ -0,0 +1,330 @@ +#ifdef HAVE_CONFIG_H +#include <config.h> +#endif +#include <stdlib.h> +#include <string.h> + +#ifdef STORAGE_WITH_MATH +#include <math.h> +#endif + +#include "common.h" +#include "web_buffer.h" +#include "log.h" + +#define BUFFER_OVERFLOW_EOF "EOF" + +static inline void buffer_overflow_init(BUFFER *b) +{ + b->buffer[b->size] = '\0'; + strcpy(&b->buffer[b->size + 1], BUFFER_OVERFLOW_EOF); +} + +#ifdef NETDATA_INTERNAL_CHECKS +#define buffer_overflow_check(b) _buffer_overflow_check(b, __FILE__, __FUNCTION__, __LINE__) +#else +#define buffer_overflow_check(b) +#endif + +static inline void _buffer_overflow_check(BUFFER *b, const char *file, const char *function, const unsigned long line) +{ + if(b->len > b->size) { + error("BUFFER: length %ld is above size %ld, at line %lu, at function %s() of file '%s'.", b->len, b->size, line, function, file); + b->len = b->size; + } + + if(b->buffer[b->size] != '\0' || strcmp(&b->buffer[b->size + 1], BUFFER_OVERFLOW_EOF)) { + error("BUFFER: detected overflow at line %lu, at function %s() of file '%s'.", line, function, file); + buffer_overflow_init(b); + } +} + + +void buffer_reset(BUFFER *wb) +{ + buffer_flush(wb); + + wb->contenttype = CT_TEXT_PLAIN; + wb->options = 0; + wb->date = 0; + + buffer_overflow_check(wb); +} + +const char *buffer_tostring(BUFFER *wb) +{ + buffer_need_bytes(wb, (size_t)1); + wb->buffer[wb->len] = '\0'; + + buffer_overflow_check(wb); + + return(wb->buffer); +} + +void buffer_char_replace(BUFFER *wb, char from, char to) +{ + char *s = wb->buffer, *end = &wb->buffer[wb->len]; + + while(s != end) { + if(*s == from) *s = to; + s++; + } + + buffer_overflow_check(wb); +} + + +void buffer_strcat(BUFFER *wb, const char *txt) +{ + if(unlikely(!txt || !*txt)) return; + + buffer_need_bytes(wb, (size_t)(1)); + + char *s = &wb->buffer[wb->len], *end = &wb->buffer[wb->size]; + long len = wb->len; + + while(*txt && s != end) { + *s++ = *txt++; + len++; + } + + wb->len = len; + buffer_overflow_check(wb); + + if(*txt) { + debug(D_WEB_BUFFER, "strcat(): increasing web_buffer at position %ld, size = %ld\n", wb->len, wb->size); + len = strlen(txt); + buffer_increase(wb, len); + buffer_strcat(wb, txt); + } + else { + // terminate the string + // without increasing the length + buffer_need_bytes(wb, (size_t)1); + wb->buffer[wb->len] = '\0'; + } +} + + +void buffer_snprintf(BUFFER *wb, size_t len, const char *fmt, ...) +{ + if(unlikely(!fmt || !*fmt)) return; + + buffer_need_bytes(wb, len+1); + + va_list args; + va_start(args, fmt); + wb->len += vsnprintf(&wb->buffer[wb->len], len+1, fmt, args); + va_end(args); + + buffer_overflow_check(wb); + + // the buffer is \0 terminated by vsnprintf +} + +void buffer_vsprintf(BUFFER *wb, const char *fmt, va_list args) +{ + if(unlikely(!fmt || !*fmt)) return; + + buffer_need_bytes(wb, 1); + + size_t len = wb->size - wb->len; + + wb->len += vsnprintf(&wb->buffer[wb->len], len, fmt, args); + + buffer_overflow_check(wb); + + // the buffer is \0 terminated by vsnprintf +} + +void buffer_sprintf(BUFFER *wb, const char *fmt, ...) +{ + if(unlikely(!fmt || !*fmt)) return; + + buffer_need_bytes(wb, 1); + + size_t len = wb->size - wb->len, wrote; + + va_list args; + va_start(args, fmt); + wrote = (size_t) vsnprintf(&wb->buffer[wb->len], len, fmt, args); + va_end(args); + + if(unlikely(wrote >= len)) { + // there is bug in vsnprintf() and it returns + // a number higher to len, but it does not + // overflow the buffer. + // our buffer overflow detector will log it + // if it does. + buffer_overflow_check(wb); + + debug(D_WEB_BUFFER, "web_buffer_sprintf(): increasing web_buffer at position %ld, size = %ld\n", wb->len, wb->size); + buffer_need_bytes(wb, len + WEB_DATA_LENGTH_INCREASE_STEP); + + va_start(args, fmt); + buffer_vsprintf(wb, fmt, args); + va_end(args); + } + else + wb->len += wrote; + + // the buffer is \0 terminated by vsnprintf +} + + +void buffer_rrd_value(BUFFER *wb, calculated_number value) +{ + buffer_need_bytes(wb, 50); + wb->len += print_calculated_number(&wb->buffer[wb->len], value); + + // terminate it + buffer_need_bytes(wb, 1); + wb->buffer[wb->len] = '\0'; + + buffer_overflow_check(wb); +} + +// generate a javascript date, the fastest possible way... +void buffer_jsdate(BUFFER *wb, int year, int month, int day, int hours, int minutes, int seconds) +{ + // 10 20 30 = 35 + // 01234567890123456789012345678901234 + // Date(2014,04,01,03,28,20) + + buffer_need_bytes(wb, 30); + + char *b = &wb->buffer[wb->len]; + + int i = 0; + b[i++]='D'; + b[i++]='a'; + b[i++]='t'; + b[i++]='e'; + b[i++]='('; + b[i++]= (char) (48 + year / 1000); year -= (year / 1000) * 1000; + b[i++]= (char) (48 + year / 100); year -= (year / 100) * 100; + b[i++]= (char) (48 + year / 10); + b[i++]= (char) (48 + year % 10); + b[i++]=','; + b[i]= (char) (48 + month / 10); if(b[i] != '0') i++; + b[i++]= (char) (48 + month % 10); + b[i++]=','; + b[i]= (char) (48 + day / 10); if(b[i] != '0') i++; + b[i++]= (char) (48 + day % 10); + b[i++]=','; + b[i]= (char) (48 + hours / 10); if(b[i] != '0') i++; + b[i++]= (char) (48 + hours % 10); + b[i++]=','; + b[i]= (char) (48 + minutes / 10); if(b[i] != '0') i++; + b[i++]= (char) (48 + minutes % 10); + b[i++]=','; + b[i]= (char) (48 + seconds / 10); if(b[i] != '0') i++; + b[i++]= (char) (48 + seconds % 10); + b[i++]=')'; + b[i]='\0'; + + wb->len += i; + + // terminate it + wb->buffer[wb->len] = '\0'; + buffer_overflow_check(wb); +} + +// generate a date, the fastest possible way... +void buffer_date(BUFFER *wb, int year, int month, int day, int hours, int minutes, int seconds) +{ + // 10 20 30 = 35 + // 01234567890123456789012345678901234 + // 2014-04-01 03:28:20 + + buffer_need_bytes(wb, 36); + + char *b = &wb->buffer[wb->len]; + + int i = 0; + b[i++]= (char) (48 + year / 1000); year -= (year / 1000) * 1000; + b[i++]= (char) (48 + year / 100); year -= (year / 100) * 100; + b[i++]= (char) (48 + year / 10); + b[i++]= (char) (48 + year % 10); + b[i++]='-'; + b[i++]= (char) (48 + month / 10); + b[i++]= (char) (48 + month % 10); + b[i++]='-'; + b[i++]= (char) (48 + day / 10); + b[i++]= (char) (48 + day % 10); + b[i++]=' '; + b[i++]= (char) (48 + hours / 10); + b[i++]= (char) (48 + hours % 10); + b[i++]=':'; + b[i++]= (char) (48 + minutes / 10); + b[i++]= (char) (48 + minutes % 10); + b[i++]=':'; + b[i++]= (char) (48 + seconds / 10); + b[i++]= (char) (48 + seconds % 10); + b[i]='\0'; + + wb->len += i; + + // terminate it + wb->buffer[wb->len] = '\0'; + buffer_overflow_check(wb); +} + +BUFFER *buffer_create(long size) +{ + BUFFER *b; + + debug(D_WEB_BUFFER, "Creating new web buffer of size %d.", size); + + b = calloc(1, sizeof(BUFFER)); + if(!b) { + error("Cannot allocate a web_buffer."); + return NULL; + } + + b->buffer = malloc(size + sizeof(BUFFER_OVERFLOW_EOF) + 2); + if(!b->buffer) { + error("Cannot allocate a buffer of size %u.", size + sizeof(BUFFER_OVERFLOW_EOF) + 2); + free(b); + return NULL; + } + b->buffer[0] = '\0'; + b->size = size; + b->contenttype = CT_TEXT_PLAIN; + buffer_overflow_init(b); + buffer_overflow_check(b); + + return(b); +} + +void buffer_free(BUFFER *b) +{ + buffer_overflow_check(b); + + debug(D_WEB_BUFFER, "Freeing web buffer of size %d.", b->size); + + if(b->buffer) free(b->buffer); + free(b); +} + +void buffer_increase(BUFFER *b, size_t free_size_required) +{ + buffer_overflow_check(b); + + size_t left = b->size - b->len; + + if(left >= free_size_required) return; + + size_t increase = free_size_required - left; + if(increase < WEB_DATA_LENGTH_INCREASE_STEP) increase = WEB_DATA_LENGTH_INCREASE_STEP; + + debug(D_WEB_BUFFER, "Increasing data buffer from size %d to %d.", b->size, b->size + increase); + + b->buffer = realloc(b->buffer, b->size + increase + sizeof(BUFFER_OVERFLOW_EOF) + 2); + 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); + + b->size += increase; + + buffer_overflow_init(b); + buffer_overflow_check(b); +} |