src/common/autobuf.c

   1
   2 /*
   3  * The olsr.org Optimized Link-State Routing daemon(olsrd)
   4  * Copyright (c) 2004-2009, the olsr.org team - see HISTORY file
   5  * All rights reserved.
   6  *
   7  * Redistribution and use in source and binary forms, with or without
   8  * modification, are permitted provided that the following conditions
   9  * are met:
  10  *
  11  * * Redistributions of source code must retain the above copyright
  12  *   notice, this list of conditions and the following disclaimer.
  13  * * Redistributions in binary form must reproduce the above copyright
  14  *   notice, this list of conditions and the following disclaimer in
  15  *   the documentation and/or other materials provided with the
  16  *   distribution.
  17  * * Neither the name of olsr.org, olsrd nor the names of its
  18  *   contributors may be used to endorse or promote products derived
  19  *   from this software without specific prior written permission.
  20  *
  21  * THIS SOFTWARE IS PROVIDED BY THE COPYRIGHT HOLDERS AND CONTRIBUTORS
  22  * "AS IS" AND ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT
  23  * LIMITED TO, THE IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS
  24  * FOR A PARTICULAR PURPOSE ARE DISCLAIMED. IN NO EVENT SHALL THE
  25  * COPYRIGHT OWNER OR CONTRIBUTORS BE LIABLE FOR ANY DIRECT, INDIRECT,
  26  * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING,
  27  * BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES;
  28  * LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER
  29  * CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  30  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN
  31  * ANY WAY OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE
  32  * POSSIBILITY OF SUCH DAMAGE.
  33  *
  34  * Visit http://www.olsr.org for more information.
  35  *
  36  * If you find this software useful feel free to make a donation
  37  * to the project. For more information see the website or contact
  38  * the copyright holders.
  39  *
  40  */
  41
  42 #include "common/autobuf.h"
  43 #include "defs.h"
  44
  45 #include <stdio.h>
  46 #include <string.h>
  47 #include <stdlib.h>
  48 #include <errno.h>
  49
  50
  51 static int autobuf_enlarge(struct autobuf *autobuf, int new_size);
  52
  53
  54 int
  55 abuf_init(struct autobuf *autobuf, int initial_size)
  56 {
  57   autobuf->len = 0;
  58   if (initial_size <= 0) {
  59     autobuf->size = 0;
  60     autobuf->buf = NULL;
  61     return 0;
  62   }
  63   autobuf->size = ROUND_UP_TO_POWER_OF_2(initial_size, AUTOBUFCHUNK);
  64   autobuf->buf = calloc(autobuf->size, 1);
  65   if (autobuf->buf == NULL) {
  66     autobuf->size = 0;
  67     return -1;
  68   }
  69   *autobuf->buf = '\0';
  70   return 0;
  71 }
  72
  73 void
  74 abuf_free(struct autobuf *autobuf)
  75 {
  76   free(autobuf->buf);
  77   autobuf->buf = NULL;
  78   autobuf->len = 0;
  79   autobuf->size = 0;
  80 }
  81
  82 static int
  83 autobuf_enlarge(struct autobuf *autobuf, int new_size)
  84 {
  85   new_size++;
  86   if (new_size > autobuf->size) {
  87     char *p;
  88     int roundUpSize = ROUND_UP_TO_POWER_OF_2(new_size, AUTOBUFCHUNK);
  89     p = realloc(autobuf->buf, roundUpSize);
  90     if (p == NULL) {
  91 #ifdef WIN32
  92       WSASetLastError(ENOMEM);
  93 #else
  94       errno = ENOMEM;
  95 #endif
  96       return -1;
  97     }
  98     autobuf->buf = p;
  99
 100     memset(&autobuf->buf[autobuf->size], 0, roundUpSize - autobuf->size);
 101     autobuf->size = roundUpSize;
 102   }
 103   return 0;
 104 }
 105
 106 int
 107 abuf_vappendf(struct autobuf *autobuf, const char *format, va_list ap)
 108 {
 109   int rc;
 110   int min_size;
 111   va_list ap2;
 112   va_copy(ap2, ap);
 113   rc = vsnprintf(autobuf->buf + autobuf->len, autobuf->size - autobuf->len, format, ap);
 114   va_end(ap);
 115   min_size = autobuf->len + rc;
 116   if (min_size >= autobuf->size) {
 117     if (autobuf_enlarge(autobuf, min_size) < 0) {
 118       autobuf->buf[autobuf->len] = '\0';
 119       return -1;
 120     }
 121     vsnprintf(autobuf->buf + autobuf->len, autobuf->size - autobuf->len, format, ap2);
 122   }
 123   va_end(ap2);
 124   autobuf->len = min_size;
 125   return 0;
 126 }
 127
 128 int
 129 abuf_appendf(struct autobuf *autobuf, const char *fmt, ...)
 130 {
 131   int rv;
 132   va_list ap;
 133   va_start(ap, fmt);
 134   rv = abuf_vappendf(autobuf, fmt, ap);
 135   va_end(ap);
 136   return rv;
 137 }
 138
 139 int
 140 abuf_puts(struct autobuf *autobuf, const char *s)
 141 {
 142   int len = strlen(s);
 143   if (autobuf_enlarge(autobuf, autobuf->len + len + 1) < 0) {
 144     return -1;
 145   }
 146   strcpy(autobuf->buf + autobuf->len, s);
 147   autobuf->len += len;
 148   return len;
 149 }
 150
 151 int
 152 abuf_strftime(struct autobuf *autobuf, const char *format, const struct tm *tm)
 153 {
 154   int rc = strftime(autobuf->buf + autobuf->len, autobuf->size - autobuf->len, format, tm);
 155   if (rc == 0) {
 156     /* we had an error! Probably the buffer too small. So we add some bytes. */
 157     if (autobuf_enlarge(autobuf, autobuf->size + AUTOBUFCHUNK) < 0) {
 158       autobuf->buf[autobuf->len] = '\0';
 159       return -1;
 160     }
 161     rc = strftime(autobuf->buf + autobuf->len, autobuf->size - autobuf->len, format, tm);
 162   }
 163   autobuf->len += rc;
 164   return rc;
 165 }
 166
 167 int
 168 abuf_memcpy(struct autobuf *autobuf, const void *p, const unsigned int len)
 169 {
 170   if (autobuf_enlarge(autobuf, autobuf->len + len) < 0) {
 171     return -1;
 172   }
 173   memcpy(autobuf->buf + autobuf->len, p, len);
 174   autobuf->len += len;
 175   return len;
 176 }
 177
 178 int
 179 abuf_memcpy_prefix(struct autobuf *autobuf, const void *p, const unsigned int len)
 180 {
 181   if (autobuf_enlarge(autobuf, autobuf->len + len) < 0) {
 182     return -1;
 183   }
 184   memmove(&autobuf->buf[len], autobuf->buf, autobuf->len);
 185   memcpy(autobuf->buf, p, len);
 186   autobuf->len += len;
 187   return len;
 188 }
 189
 190 int
 191 abuf_pull(struct autobuf * autobuf, int len) {
 192   char *p;
 193   size_t newsize;
 194
 195   if (len != autobuf->len) {
 196     memmove(autobuf->buf, &autobuf->buf[len], autobuf->len - len);
 197   }
 198   autobuf->len -= len;
 199
 200   newsize = ROUND_UP_TO_POWER_OF_2(autobuf->len + 1, AUTOBUFCHUNK);
 201   p = realloc(autobuf->buf, newsize);
 202   if (p == NULL) {
 203 #ifdef WIN32
 204     WSASetLastError(ENOMEM);
 205 #else
 206     errno = ENOMEM;
 207 #endif
 208     return -1;
 209   }
 210   autobuf->buf = p;
 211   autobuf->size = newsize;
 212   return 0;
 213 }
 214 /*
 215  * Local Variables:
 216  * mode: c
 217  * style: linux
 218  * c-basic-offset: 4
 219  * indent-tabs-mode: nil
 220  * End:
 221  */