src/web_buffer.c

   1 #ifdef HAVE_CONFIG_H
   2 #include <config.h>
   3 #endif
   4 #include <stdlib.h>
   5 #include <string.h>
   6
   7 #ifdef STORAGE_WITH_MATH
   8 #include <math.h>
   9 #endif
  10
  11 #include "common.h"
  12 #include "web_buffer.h"
  13 #include "log.h"
  14
  15 #define BUFFER_OVERFLOW_EOF "EOF"
  16
  17 static inline void buffer_overflow_init(BUFFER *b)
  18 {
  19         b->buffer[b->size] = '\0';
  20         strcpy(&b->buffer[b->size + 1], BUFFER_OVERFLOW_EOF);
  21 }
  22
  23 #ifdef NETDATA_INTERNAL_CHECKS
  24 #define buffer_overflow_check(b) _buffer_overflow_check(b, __FILE__, __FUNCTION__, __LINE__)
  25 #else
  26 #define buffer_overflow_check(b)
  27 #endif
  28
  29 static inline void _buffer_overflow_check(BUFFER *b, const char *file, const char *function, const unsigned long line)
  30 {
  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);
  33                 b->len = b->size;
  34         }
  35
  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);
  39         }
  40 }
  41
  42
  43 void buffer_reset(BUFFER *wb)
  44 {
  45         buffer_flush(wb);
  46
  47         wb->contenttype = CT_TEXT_PLAIN;
  48         wb->date = 0;
  49
  50         buffer_overflow_check(wb);
  51 }
  52
  53 const char *buffer_tostring(BUFFER *wb)
  54 {
  55         buffer_need_bytes(wb, 1);
  56         wb->buffer[wb->len] = '\0';
  57
  58         buffer_overflow_check(wb);
  59
  60         return(wb->buffer);
  61 }
  62
  63 void buffer_char_replace(BUFFER *wb, char from, char to)
  64 {
  65         char *s = wb->buffer, *end = &wb->buffer[wb->len];
  66
  67         while(s != end) {
  68                 if(*s == from) *s = to;
  69                 s++;
  70         }
  71
  72         buffer_overflow_check(wb);
  73 }
  74
  75
  76 void buffer_strcat(BUFFER *wb, const char *txt)
  77 {
  78         if(unlikely(!txt || !*txt)) return;
  79
  80         if(wb->size - wb->len < 512)
  81                 buffer_need_bytes(wb, 512);
  82
  83         char *s = &wb->buffer[wb->len], *end = &wb->buffer[wb->size];
  84         long len = wb->len;
  85
  86         while(*txt && s != end) {
  87                 *s++ = *txt++;
  88                 len++;
  89         }
  90
  91         wb->len = len;
  92         buffer_overflow_check(wb);
  93
  94         if(*txt) {
  95                 debug(D_WEB_BUFFER, "strcat(): increasing web_buffer at position %ld, size = %ld\n", wb->len, wb->size);
  96                 len = strlen(txt);
  97                 buffer_increase(wb, len);
  98                 buffer_strcat(wb, txt);
  99         }
 100         else {
 101                 // terminate the string
 102                 // without increasing the length
 103                 buffer_need_bytes(wb, 1);
 104                 wb->buffer[wb->len] = '\0';
 105         }
 106 }
 107
 108
 109 void buffer_snprintf(BUFFER *wb, size_t len, const char *fmt, ...)
 110 {
 111         if(unlikely(!fmt || !*fmt)) return;
 112
 113         buffer_need_bytes(wb, len+1);
 114
 115         va_list args;
 116         va_start(args, fmt);
 117         wb->len += vsnprintf(&wb->buffer[wb->len], len+1, fmt, args);
 118         va_end(args);
 119
 120         buffer_overflow_check(wb);
 121
 122         // the buffer is \0 terminated by vsnprintf
 123 }
 124
 125 void buffer_vsprintf(BUFFER *wb, const char *fmt, va_list args)
 126 {
 127         if(unlikely(!fmt || !*fmt)) return;
 128
 129         size_t len = wb->size - wb->len;
 130         if(unlikely(!len)) return;
 131
 132         wb->len += vsnprintf(&wb->buffer[wb->len], len, fmt, args);
 133
 134         buffer_overflow_check(wb);
 135
 136         // the buffer is \0 terminated by vsnprintf
 137 }
 138
 139 void buffer_sprintf(BUFFER *wb, const char *fmt, ...)
 140 {
 141         if(unlikely(!fmt || !*fmt)) return;
 142
 143         if(unlikely(wb->len > wb->size)) {
 144                 error("web_buffer_sprintf(): already overflown length %ld, size = %ld", wb->len, wb->size);
 145         }
 146
 147 //      if(wb->size - wb->len < 512)
 148 //              web_buffer_need_bytes(wb, 512);
 149
 150         size_t len = wb->size - wb->len, wrote;
 151
 152         buffer_need_bytes(wb, len);
 153
 154         va_list args;
 155         va_start(args, fmt);
 156         wrote = vsnprintf(&wb->buffer[wb->len], len, fmt, args);
 157         va_end(args);
 158
 159         if(unlikely(wrote >= len)) {
 160                 // there is bug in vsnprintf() and it returns
 161                 // a number higher to len, but it does not
 162                 // overflow the buffer.
 163                 // our buffer overflow detector will log it
 164                 // if it does.
 165                 buffer_overflow_check(wb);
 166
 167                 debug(D_WEB_BUFFER, "web_buffer_sprintf(): increasing web_buffer at position %ld, size = %ld\n", wb->len, wb->size);
 168                 buffer_need_bytes(wb, len + WEB_DATA_LENGTH_INCREASE_STEP);
 169
 170                 va_start(args, fmt);
 171                 buffer_vsprintf(wb, fmt, args);
 172                 va_end(args);
 173         }
 174         else
 175                 wb->len += wrote;
 176
 177         // the buffer is \0 terminated by vsnprintf
 178 }
 179
 180
 181 void buffer_rrd_value(BUFFER *wb, calculated_number value)
 182 {
 183         buffer_need_bytes(wb, 50);
 184         wb->len += print_calculated_number(&wb->buffer[wb->len], value);
 185
 186         // terminate it
 187         buffer_need_bytes(wb, 1);
 188         wb->buffer[wb->len] = '\0';
 189
 190         buffer_overflow_check(wb);
 191 }
 192
 193 // generate a javascript date, the fastest possible way...
 194 void buffer_jsdate(BUFFER *wb, int year, int month, int day, int hours, int minutes, int seconds)
 195 {
 196         //         10        20        30      = 35
 197         // 01234567890123456789012345678901234
 198         // Date(2014,04,01,03,28,20)
 199
 200         buffer_need_bytes(wb, 30);
 201
 202         char *b = &wb->buffer[wb->len];
 203
 204         int i = 0;
 205         b[i++]='D';
 206         b[i++]='a';
 207         b[i++]='t';
 208         b[i++]='e';
 209         b[i++]='(';
 210         b[i++]= 48 + year / 1000; year -= (year / 1000) * 1000;
 211         b[i++]= 48 + year / 100; year -= (year / 100) * 100;
 212         b[i++]= 48 + year / 10;
 213         b[i++]= 48 + year % 10;
 214         b[i++]=',';
 215         b[i]= 48 + month / 10; if(b[i] != '0') i++;
 216         b[i++]= 48 + month % 10;
 217         b[i++]=',';
 218         b[i]= 48 + day / 10; if(b[i] != '0') i++;
 219         b[i++]= 48 + day % 10;
 220         b[i++]=',';
 221         b[i]= 48 + hours / 10; if(b[i] != '0') i++;
 222         b[i++]= 48 + hours % 10;
 223         b[i++]=',';
 224         b[i]= 48 + minutes / 10; if(b[i] != '0') i++;
 225         b[i++]= 48 + minutes % 10;
 226         b[i++]=',';
 227         b[i]= 48 + seconds / 10; if(b[i] != '0') i++;
 228         b[i++]= 48 + seconds % 10;
 229         b[i++]=')';
 230         b[i]='\0';
 231
 232         wb->len += i;
 233
 234         // terminate it
 235         wb->buffer[wb->len] = '\0';
 236         buffer_overflow_check(wb);
 237 }
 238
 239 // generate a date, the fastest possible way...
 240 void buffer_date(BUFFER *wb, int year, int month, int day, int hours, int minutes, int seconds)
 241 {
 242         //         10        20        30      = 35
 243         // 01234567890123456789012345678901234
 244         // 2014-04-01 03:28:20
 245
 246         buffer_need_bytes(wb, 36);
 247
 248         char *b = &wb->buffer[wb->len];
 249
 250         int i = 0;
 251         b[i++]= 48 + year / 1000; year -= (year / 1000) * 1000;
 252         b[i++]= 48 + year / 100; year -= (year / 100) * 100;
 253         b[i++]= 48 + year / 10;
 254         b[i++]= 48 + year % 10;
 255         b[i++]='-';
 256         b[i++]= 48 + month / 10;
 257         b[i++]= 48 + month % 10;
 258         b[i++]='-';
 259         b[i++]= 48 + day / 10;
 260         b[i++]= 48 + day % 10;
 261         b[i++]=' ';
 262         b[i++]= 48 + hours / 10;
 263         b[i++]= 48 + hours % 10;
 264         b[i++]=':';
 265         b[i++]= 48 + minutes / 10;
 266         b[i++]= 48 + minutes % 10;
 267         b[i++]=':';
 268         b[i++]= 48 + seconds / 10;
 269         b[i++]= 48 + seconds % 10;
 270         b[i]='\0';
 271
 272         wb->len += i;
 273
 274         // terminate it
 275         wb->buffer[wb->len] = '\0';
 276         buffer_overflow_check(wb);
 277 }
 278
 279 BUFFER *buffer_create(long size)
 280 {
 281         BUFFER *b;
 282
 283         debug(D_WEB_BUFFER, "Creating new web buffer of size %d.", size);
 284
 285         b = calloc(1, sizeof(BUFFER));
 286         if(!b) {
 287                 error("Cannot allocate a web_buffer.");
 288                 return NULL;
 289         }
 290
 291         b->buffer = malloc(size + sizeof(BUFFER_OVERFLOW_EOF) + 2);
 292         if(!b->buffer) {
 293                 error("Cannot allocate a buffer of size %u.", size + sizeof(BUFFER_OVERFLOW_EOF) + 2);
 294                 free(b);
 295                 return NULL;
 296         }
 297         b->buffer[0] = '\0';
 298         b->size = size;
 299         b->contenttype = CT_TEXT_PLAIN;
 300         buffer_overflow_init(b);
 301         buffer_overflow_check(b);
 302
 303         return(b);
 304 }
 305
 306 void buffer_free(BUFFER *b)
 307 {
 308         buffer_overflow_check(b);
 309
 310         debug(D_WEB_BUFFER, "Freeing web buffer of size %d.", b->size);
 311
 312         if(b->buffer) free(b->buffer);
 313         free(b);
 314 }
 315
 316 void buffer_increase(BUFFER *b, long free_size_required)
 317 {
 318         buffer_overflow_check(b);
 319
 320         long left = b->size - b->len;
 321
 322         if(left >= free_size_required) return;
 323
 324         long increase = free_size_required - left;
 325         if(increase < WEB_DATA_LENGTH_INCREASE_STEP) increase = WEB_DATA_LENGTH_INCREASE_STEP;
 326
 327         debug(D_WEB_BUFFER, "Increasing data buffer from size %d to %d.", b->size, b->size + increase);
 328
 329         b->buffer = realloc(b->buffer, b->size + increase + sizeof(BUFFER_OVERFLOW_EOF) + 2);
 330         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);
 331
 332         b->size += increase;
 333
 334         buffer_overflow_init(b);
 335         buffer_overflow_check(b);
 336 }