Home | History | Annotate | Download | only in buf
      1 /* Copyright (C) 1995-1998 Eric Young (eay (at) cryptsoft.com)
      2  * All rights reserved.
      3  *
      4  * This package is an SSL implementation written
      5  * by Eric Young (eay (at) cryptsoft.com).
      6  * The implementation was written so as to conform with Netscapes SSL.
      7  *
      8  * This library is free for commercial and non-commercial use as long as
      9  * the following conditions are aheared to.  The following conditions
     10  * apply to all code found in this distribution, be it the RC4, RSA,
     11  * lhash, DES, etc., code; not just the SSL code.  The SSL documentation
     12  * included with this distribution is covered by the same copyright terms
     13  * except that the holder is Tim Hudson (tjh (at) cryptsoft.com).
     14  *
     15  * Copyright remains Eric Young's, and as such any Copyright notices in
     16  * the code are not to be removed.
     17  * If this package is used in a product, Eric Young should be given attribution
     18  * as the author of the parts of the library used.
     19  * This can be in the form of a textual message at program startup or
     20  * in documentation (online or textual) provided with the package.
     21  *
     22  * Redistribution and use in source and binary forms, with or without
     23  * modification, are permitted provided that the following conditions
     24  * are met:
     25  * 1. Redistributions of source code must retain the copyright
     26  *    notice, this list of conditions and the following disclaimer.
     27  * 2. Redistributions in binary form must reproduce the above copyright
     28  *    notice, this list of conditions and the following disclaimer in the
     29  *    documentation and/or other materials provided with the distribution.
     30  * 3. All advertising materials mentioning features or use of this software
     31  *    must display the following acknowledgement:
     32  *    "This product includes cryptographic software written by
     33  *     Eric Young (eay (at) cryptsoft.com)"
     34  *    The word 'cryptographic' can be left out if the rouines from the library
     35  *    being used are not cryptographic related :-).
     36  * 4. If you include any Windows specific code (or a derivative thereof) from
     37  *    the apps directory (application code) you must include an acknowledgement:
     38  *    "This product includes software written by Tim Hudson (tjh (at) cryptsoft.com)"
     39  *
     40  * THIS SOFTWARE IS PROVIDED BY ERIC YOUNG ``AS IS'' AND
     41  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
     42  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
     43  * ARE DISCLAIMED.  IN NO EVENT SHALL THE AUTHOR OR CONTRIBUTORS BE LIABLE
     44  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
     45  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
     46  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
     47  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
     48  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
     49  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
     50  * SUCH DAMAGE.
     51  *
     52  * The licence and distribution terms for any publically available version or
     53  * derivative of this code cannot be changed.  i.e. this code cannot simply be
     54  * copied and put under another distribution licence
     55  * [including the GNU Public Licence.] */
     56 
     57 #include <openssl/buf.h>
     58 
     59 #include <string.h>
     60 
     61 #include <openssl/mem.h>
     62 #include <openssl/err.h>
     63 
     64 #include "../internal.h"
     65 
     66 
     67 BUF_MEM *BUF_MEM_new(void) {
     68   BUF_MEM *ret;
     69 
     70   ret = OPENSSL_malloc(sizeof(BUF_MEM));
     71   if (ret == NULL) {
     72     OPENSSL_PUT_ERROR(BUF, ERR_R_MALLOC_FAILURE);
     73     return NULL;
     74   }
     75 
     76   OPENSSL_memset(ret, 0, sizeof(BUF_MEM));
     77   return ret;
     78 }
     79 
     80 void BUF_MEM_free(BUF_MEM *buf) {
     81   if (buf == NULL) {
     82     return;
     83   }
     84 
     85   OPENSSL_free(buf->data);
     86   OPENSSL_free(buf);
     87 }
     88 
     89 int BUF_MEM_reserve(BUF_MEM *buf, size_t cap) {
     90   if (buf->max >= cap) {
     91     return 1;
     92   }
     93 
     94   size_t n = cap + 3;
     95   if (n < cap) {
     96     // overflow
     97     OPENSSL_PUT_ERROR(BUF, ERR_R_MALLOC_FAILURE);
     98     return 0;
     99   }
    100   n = n / 3;
    101   size_t alloc_size = n * 4;
    102   if (alloc_size / 4 != n) {
    103     // overflow
    104     OPENSSL_PUT_ERROR(BUF, ERR_R_MALLOC_FAILURE);
    105     return 0;
    106   }
    107 
    108   char *new_buf = OPENSSL_realloc(buf->data, alloc_size);
    109   if (new_buf == NULL) {
    110     OPENSSL_PUT_ERROR(BUF, ERR_R_MALLOC_FAILURE);
    111     return 0;
    112   }
    113 
    114   buf->data = new_buf;
    115   buf->max = alloc_size;
    116   return 1;
    117 }
    118 
    119 size_t BUF_MEM_grow(BUF_MEM *buf, size_t len) {
    120   if (!BUF_MEM_reserve(buf, len)) {
    121     return 0;
    122   }
    123   if (buf->length < len) {
    124     OPENSSL_memset(&buf->data[buf->length], 0, len - buf->length);
    125   }
    126   buf->length = len;
    127   return len;
    128 }
    129 
    130 size_t BUF_MEM_grow_clean(BUF_MEM *buf, size_t len) {
    131   return BUF_MEM_grow(buf, len);
    132 }
    133 
    134 int BUF_MEM_append(BUF_MEM *buf, const void *in, size_t len) {
    135   size_t new_len = buf->length + len;
    136   if (new_len < len) {
    137     OPENSSL_PUT_ERROR(BUF, ERR_R_OVERFLOW);
    138     return 0;
    139   }
    140   if (!BUF_MEM_reserve(buf, new_len)) {
    141     return 0;
    142   }
    143   OPENSSL_memcpy(buf->data + buf->length, in, len);
    144   buf->length = new_len;
    145   return 1;
    146 }
    147 
    148 char *BUF_strdup(const char *str) {
    149   if (str == NULL) {
    150     return NULL;
    151   }
    152 
    153   return BUF_strndup(str, strlen(str));
    154 }
    155 
    156 size_t BUF_strnlen(const char *str, size_t max_len) {
    157   size_t i;
    158 
    159   for (i = 0; i < max_len; i++) {
    160     if (str[i] == 0) {
    161       break;
    162     }
    163   }
    164 
    165   return i;
    166 }
    167 
    168 char *BUF_strndup(const char *str, size_t size) {
    169   char *ret;
    170   size_t alloc_size;
    171 
    172   if (str == NULL) {
    173     return NULL;
    174   }
    175 
    176   size = BUF_strnlen(str, size);
    177 
    178   alloc_size = size + 1;
    179   if (alloc_size < size) {
    180     // overflow
    181     OPENSSL_PUT_ERROR(BUF, ERR_R_MALLOC_FAILURE);
    182     return NULL;
    183   }
    184   ret = OPENSSL_malloc(alloc_size);
    185   if (ret == NULL) {
    186     OPENSSL_PUT_ERROR(BUF, ERR_R_MALLOC_FAILURE);
    187     return NULL;
    188   }
    189 
    190   OPENSSL_memcpy(ret, str, size);
    191   ret[size] = '\0';
    192   return ret;
    193 }
    194 
    195 size_t BUF_strlcpy(char *dst, const char *src, size_t dst_size) {
    196   size_t l = 0;
    197 
    198   for (; dst_size > 1 && *src; dst_size--) {
    199     *dst++ = *src++;
    200     l++;
    201   }
    202 
    203   if (dst_size) {
    204     *dst = 0;
    205   }
    206 
    207   return l + strlen(src);
    208 }
    209 
    210 size_t BUF_strlcat(char *dst, const char *src, size_t dst_size) {
    211   size_t l = 0;
    212   for (; dst_size > 0 && *dst; dst_size--, dst++) {
    213     l++;
    214   }
    215   return l + BUF_strlcpy(dst, src, dst_size);
    216 }
    217 
    218 void *BUF_memdup(const void *data, size_t size) {
    219   if (size == 0) {
    220     return NULL;
    221   }
    222 
    223   void *ret = OPENSSL_malloc(size);
    224   if (ret == NULL) {
    225     OPENSSL_PUT_ERROR(BUF, ERR_R_MALLOC_FAILURE);
    226     return NULL;
    227   }
    228 
    229   OPENSSL_memcpy(ret, data, size);
    230   return ret;
    231 }
    232