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