lib/output_buffer.c

   1 #include <string.h>
   2 #include <stdlib.h>
   3
   4 #include "output_buffer.h"
   5 #include "../minmax.h"
   6
   7 #define BUF_INC 1024
   8
   9 void buf_output_init(struct buf_output *out)
  10 {
  11         out->max_buflen = 0;
  12         out->buflen = 0;
  13         out->buf = NULL;
  14 }
  15
  16 void buf_output_free(struct buf_output *out)
  17 {
  18         free(out->buf);
  19         buf_output_init(out);
  20 }
  21
  22 size_t buf_output_add(struct buf_output *out, const char *buf, size_t len)
  23 {
  24         if (out->max_buflen - out->buflen < len) {
  25                 size_t need = len - (out->max_buflen - out->buflen);
  26                 size_t old_max = out->max_buflen;
  27
  28                 need = max((size_t) BUF_INC, need);
  29                 out->max_buflen += need;
  30                 out->buf = realloc(out->buf, out->max_buflen);
  31
  32                 old_max = max(old_max, out->buflen + len);
  33                 if (old_max + need > out->max_buflen)
  34                         need = out->max_buflen - old_max;
  35                 memset(&out->buf[old_max], 0, need);
  36         }
  37
  38         memcpy(&out->buf[out->buflen], buf, len);
  39         out->buflen += len;
  40         return len;
  41 }