src/web_buffer.c

   1 #include "common.h"
   2
   3 #define BUFFER_OVERFLOW_EOF "EOF"
   4
   5 static inline void buffer_overflow_init(BUFFER *b)
   6 {
   7     b->buffer[b->size] = '\0';
   8     strcpy(&b->buffer[b->size + 1], BUFFER_OVERFLOW_EOF);
   9 }
  10
  11 #ifdef NETDATA_INTERNAL_CHECKS
  12 #define buffer_overflow_check(b) _buffer_overflow_check(b, __FILE__, __FUNCTION__, __LINE__)
  13 #else
  14 #define buffer_overflow_check(b)
  15 #endif
  16
  17 static inline void _buffer_overflow_check(BUFFER *b, const char *file, const char *function, const unsigned long line)
  18 {
  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);
  21         b->len = b->size;
  22     }
  23
  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);
  27     }
  28 }
  29
  30
  31 void buffer_reset(BUFFER *wb)
  32 {
  33     buffer_flush(wb);
  34
  35     wb->contenttype = CT_TEXT_PLAIN;
  36     wb->options = 0;
  37     wb->date = 0;
  38     wb->expires = 0;
  39
  40     buffer_overflow_check(wb);
  41 }
  42
  43 const char *buffer_tostring(BUFFER *wb)
  44 {
  45     buffer_need_bytes(wb, 1);
  46     wb->buffer[wb->len] = '\0';
  47
  48     buffer_overflow_check(wb);
  49
  50     return(wb->buffer);
  51 }
  52
  53 void buffer_char_replace(BUFFER *wb, char from, char to)
  54 {
  55     char *s = wb->buffer, *end = &wb->buffer[wb->len];
  56
  57     while(s != end) {
  58         if(*s == from) *s = to;
  59         s++;
  60     }
  61
  62     buffer_overflow_check(wb);
  63 }
  64
  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.
  69
  70 inline char *print_number_lu_r(char *str, unsigned long uvalue) {
  71     char *wstr = str;
  72
  73     // print each digit
  74     do *wstr++ = (char)('0' + (uvalue % 10)); while(uvalue /= 10);
  75     return wstr;
  76 }
  77
  78 inline char *print_number_llu_r(char *str, unsigned long long uvalue) {
  79     char *wstr = str;
  80
  81     // print each digit
  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);
  84     return wstr;
  85 }
  86
  87 void buffer_print_llu(BUFFER *wb, unsigned long long uvalue)
  88 {
  89     buffer_need_bytes(wb, 50);
  90
  91     char *str = &wb->buffer[wb->len];
  92     char *wstr = str;
  93
  94 #ifdef ENVIRONMENT32
  95     if(uvalue > (unsigned long long)0xffffffff)
  96         wstr = print_number_llu_r(wstr, uvalue);
  97     else
  98         wstr = print_number_lu_r(wstr, uvalue);
  99 #else
 100     do *wstr++ = (char)('0' + (uvalue % 10)); while(uvalue /= 10);
 101 #endif
 102
 103     // terminate it
 104     *wstr = '\0';
 105
 106     // reverse it
 107     char *begin = str, *end = wstr - 1, aux;
 108     while (end > begin) aux = *end, *end-- = *begin, *begin++ = aux;
 109
 110     // return the buffer length
 111     wb->len += wstr - str;
 112 }
 113
 114 void buffer_strcat(BUFFER *wb, const char *txt)
 115 {
 116     // buffer_sprintf(wb, "%s", txt);
 117
 118     if(unlikely(!txt || !*txt)) return;
 119
 120     buffer_need_bytes(wb, 1);
 121
 122     char *s = &wb->buffer[wb->len], *start, *end = &wb->buffer[wb->size];
 123     size_t len = wb->len;
 124
 125     start = s;
 126     while(*txt && s != end)
 127         *s++ = *txt++;
 128
 129     len += s - start;
 130
 131     wb->len = len;
 132     buffer_overflow_check(wb);
 133
 134     if(*txt) {
 135         debug(D_WEB_BUFFER, "strcat(): increasing web_buffer at position %zu, size = %zu\n", wb->len, wb->size);
 136         len = strlen(txt);
 137         buffer_increase(wb, len);
 138         buffer_strcat(wb, txt);
 139     }
 140     else {
 141         // terminate the string
 142         // without increasing the length
 143         buffer_need_bytes(wb, (size_t)1);
 144         wb->buffer[wb->len] = '\0';
 145     }
 146 }
 147
 148 void buffer_strcat_htmlescape(BUFFER *wb, const char *txt)
 149 {
 150     char b[2] = { [0] = '\0', [1] = '\0' };
 151
 152     while(*txt) {
 153         switch(*txt) {
 154             case '&': buffer_strcat(wb, "&amp;"); break;
 155             case '<': buffer_strcat(wb, "&lt;"); break;
 156             case '>': buffer_strcat(wb, "&gt;"); break;
 157             case '"': buffer_strcat(wb, "&quot;"); break;
 158             case '/': buffer_strcat(wb, "&#x2F;"); break;
 159             case '\'': buffer_strcat(wb, "&#x27;"); break;
 160             default: {
 161                 b[0] = *txt;
 162                 buffer_strcat(wb, b);
 163             }
 164         }
 165         txt++;
 166     }
 167 }
 168
 169 void buffer_snprintf(BUFFER *wb, size_t len, const char *fmt, ...)
 170 {
 171     if(unlikely(!fmt || !*fmt)) return;
 172
 173     buffer_need_bytes(wb, len + 1);
 174
 175     va_list args;
 176     va_start(args, fmt);
 177     wb->len += vsnprintfz(&wb->buffer[wb->len], len, fmt, args);
 178     va_end(args);
 179
 180     buffer_overflow_check(wb);
 181
 182     // the buffer is \0 terminated by vsnprintfz
 183 }
 184
 185 void buffer_vsprintf(BUFFER *wb, const char *fmt, va_list args)
 186 {
 187     if(unlikely(!fmt || !*fmt)) return;
 188
 189     buffer_need_bytes(wb, 2);
 190
 191     size_t len = wb->size - wb->len - 1;
 192
 193     wb->len += vsnprintfz(&wb->buffer[wb->len], len, fmt, args);
 194
 195     buffer_overflow_check(wb);
 196
 197     // the buffer is \0 terminated by vsnprintfz
 198 }
 199
 200 void buffer_sprintf(BUFFER *wb, const char *fmt, ...)
 201 {
 202     if(unlikely(!fmt || !*fmt)) return;
 203
 204     buffer_need_bytes(wb, 2);
 205
 206     size_t len = wb->size - wb->len - 1;
 207     size_t wrote;
 208
 209     va_list args;
 210     va_start(args, fmt);
 211     wrote = (size_t) vsnprintfz(&wb->buffer[wb->len], len, fmt, args);
 212     va_end(args);
 213
 214     if(unlikely(wrote >= len)) {
 215         // truncated
 216         buffer_overflow_check(wb);
 217
 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);
 220
 221         va_start(args, fmt);
 222         buffer_vsprintf(wb, fmt, args);
 223         va_end(args);
 224     }
 225     else
 226         wb->len += wrote;
 227
 228     // the buffer is \0 terminated by vsnprintf
 229 }
 230
 231
 232 void buffer_rrd_value(BUFFER *wb, calculated_number value)
 233 {
 234     buffer_need_bytes(wb, 50);
 235
 236     if(isnan(value) || isinf(value)) {
 237         buffer_strcat(wb, "null");
 238         return;
 239     }
 240     else
 241         wb->len += print_calculated_number(&wb->buffer[wb->len], value);
 242
 243     // terminate it
 244     buffer_need_bytes(wb, 1);
 245     wb->buffer[wb->len] = '\0';
 246
 247     buffer_overflow_check(wb);
 248 }
 249
 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)
 252 {
 253   //         10        20        30      = 35
 254     // 01234567890123456789012345678901234
 255     // Date(2014,04,01,03,28,20)
 256
 257     buffer_need_bytes(wb, 30);
 258
 259     char *b = &wb->buffer[wb->len], *p;
 260   unsigned int *q = (unsigned int *)b;
 261
 262   #if __BYTE_ORDER__ == __ORDER_LITTLE_ENDIAN__
 263     *q++ = 0x65746144;  // "Date" backwards.
 264   #else
 265     *q++ = 0x44617465;  // "Date"
 266   #endif
 267   p = (char *)q;
 268
 269   *p++ = '(';
 270   *p++ = '0' + year / 1000; year %= 1000;
 271   *p++ = '0' + year / 100;  year %= 100;
 272   *p++ = '0' + year / 10;
 273   *p++ = '0' + year % 10;
 274   *p++ = ',';
 275   *p   = '0' + month / 10; if (*p != '0') p++;
 276   *p++ = '0' + month % 10;
 277   *p++ = ',';
 278   *p   = '0' + day / 10; if (*p != '0') p++;
 279   *p++ = '0' + day % 10;
 280   *p++ = ',';
 281   *p   = '0' + hours / 10; if (*p != '0') p++;
 282   *p++ = '0' + hours % 10;
 283   *p++ = ',';
 284   *p   = '0' + minutes / 10; if (*p != '0') p++;
 285   *p++ = '0' + minutes % 10;
 286   *p++ = ',';
 287   *p   = '0' + seconds / 10; if (*p != '0') p++;
 288   *p++ = '0' + seconds % 10;
 289
 290   unsigned short *r = (unsigned short *)p;
 291
 292 #if __BYTE_ORDER__ == __ORDER_LITTLE_ENDIAN__
 293     *r++ = 0x0029;  // ")\0" backwards.
 294   #else
 295     *r++ = 0x2900;  // ")\0"
 296   #endif
 297
 298     wb->len += (size_t)((char *)r - b - 1);
 299
 300     // terminate it
 301     wb->buffer[wb->len] = '\0';
 302     buffer_overflow_check(wb);
 303 }
 304
 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)
 307 {
 308     //         10        20        30      = 35
 309     // 01234567890123456789012345678901234
 310     // 2014-04-01 03:28:20
 311
 312     buffer_need_bytes(wb, 36);
 313
 314     char *b = &wb->buffer[wb->len];
 315   char *p = b;
 316
 317   *p++ = '0' + year / 1000; year %= 1000;
 318   *p++ = '0' + year / 100;  year %= 100;
 319   *p++ = '0' + year / 10;
 320   *p++ = '0' + year % 10;
 321   *p++ = '-';
 322   *p++ = '0' + month / 10;
 323   *p++ = '0' + month % 10;
 324   *p++ = '-';
 325   *p++ = '0' + day / 10;
 326   *p++ = '0' + day % 10;
 327   *p++ = ' ';
 328   *p++ = '0' + hours / 10;
 329   *p++ = '0' + hours % 10;
 330   *p++ = ':';
 331   *p++ = '0' + minutes / 10;
 332   *p++ = '0' + minutes % 10;
 333   *p++ = ':';
 334   *p++ = '0' + seconds / 10;
 335   *p++ = '0' + seconds % 10;
 336   *p = '\0';
 337
 338     wb->len += (size_t)(p - b);
 339
 340     // terminate it
 341     wb->buffer[wb->len] = '\0';
 342     buffer_overflow_check(wb);
 343 }
 344
 345 BUFFER *buffer_create(size_t size)
 346 {
 347     BUFFER *b;
 348
 349     debug(D_WEB_BUFFER, "Creating new web buffer of size %zu.", size);
 350
 351     b = callocz(1, sizeof(BUFFER));
 352     b->buffer = mallocz(size + sizeof(BUFFER_OVERFLOW_EOF) + 2);
 353     b->buffer[0] = '\0';
 354     b->size = size;
 355     b->contenttype = CT_TEXT_PLAIN;
 356     buffer_overflow_init(b);
 357     buffer_overflow_check(b);
 358
 359     return(b);
 360 }
 361
 362 void buffer_free(BUFFER *b) {
 363     if(unlikely(!b)) return;
 364
 365     buffer_overflow_check(b);
 366
 367     debug(D_WEB_BUFFER, "Freeing web buffer of size %zu.", b->size);
 368
 369     freez(b->buffer);
 370     freez(b);
 371 }
 372
 373 void buffer_increase(BUFFER *b, size_t free_size_required)
 374 {
 375     buffer_overflow_check(b);
 376
 377     size_t left = b->size - b->len;
 378
 379     if(left >= free_size_required) return;
 380
 381     size_t increase = free_size_required - left;
 382     if(increase < WEB_DATA_LENGTH_INCREASE_STEP) increase = WEB_DATA_LENGTH_INCREASE_STEP;
 383
 384     debug(D_WEB_BUFFER, "Increasing data buffer from size %zu to %zu.", b->size, b->size + increase);
 385
 386     b->buffer = reallocz(b->buffer, b->size + increase + sizeof(BUFFER_OVERFLOW_EOF) + 2);
 387     b->size += increase;
 388
 389     buffer_overflow_init(b);
 390     buffer_overflow_check(b);
 391 }