diff options
author | 2015-09-09 22:21:41 -0700 | |
---|---|---|
committer | 2015-09-09 22:21:41 -0700 | |
commit | 8879b125d26e8db1a5633de5a9c692eb2d1c4f83 (patch) | |
tree | c7259d85a991b83dfa85ab2e339360669fc1f58e /framework/src/suricata/src/util-buffer.c | |
parent | 13d05bc8458758ee39cb829098241e89616717ee (diff) |
suricata checkin based on commit id a4bce14770beee46a537eda3c3f6e8e8565d5d0a
Change-Id: I9a214fa0ee95e58fc640e50bd604dac7f42db48f
Diffstat (limited to 'framework/src/suricata/src/util-buffer.c')
-rw-r--r-- | framework/src/suricata/src/util-buffer.c | 87 |
1 files changed, 87 insertions, 0 deletions
diff --git a/framework/src/suricata/src/util-buffer.c b/framework/src/suricata/src/util-buffer.c new file mode 100644 index 00000000..d4f50b06 --- /dev/null +++ b/framework/src/suricata/src/util-buffer.c @@ -0,0 +1,87 @@ +/* Copyright (C) 2007-2012 Open Information Security Foundation + * + * You can copy, redistribute or modify this Program under the terms of + * the GNU General Public License version 2 as published by the Free + * Software Foundation. + * + * This program is distributed in the hope that it will be useful, + * but WITHOUT ANY WARRANTY; without even the implied warranty of + * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the + * GNU General Public License for more details. + * + * You should have received a copy of the GNU General Public License + * version 2 along with this program; if not, write to the Free Software + * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA + * 02110-1301, USA. + */ + +/** + * \file + * + * \author Anoop Saldanha <anoopsaldanha@gmail.com> + */ + +#include "suricata-common.h" +#include "suricata.h" +#include "util-debug.h" +#include "util-buffer.h" + +/* 10 mb */ +#define MAX_LIMIT 10485760 + +MemBuffer *MemBufferCreateNew(uint32_t size) +{ + if (size > MAX_LIMIT) { + SCLogWarning(SC_ERR_MEM_BUFFER_API, "Mem buffer asked to create " + "buffer with size greater than API limit - %d", MAX_LIMIT); + return NULL; + } + + uint32_t total_size = size + sizeof(MemBuffer); + + MemBuffer *buffer = SCMalloc(total_size); + if (unlikely(buffer == NULL)) { + return NULL; + } + memset(buffer, 0, total_size); + + buffer->size = size; + buffer->buffer = (uint8_t *)buffer + sizeof(MemBuffer); + + return buffer; +} + +/** \brief expand membuffer by size of 'expand_by' + * + * If expansion failed, buffer will still be valid. + * + * \retval result 0 ok, -1 expansion failed + */ +int MemBufferExpand(MemBuffer **buffer, uint32_t expand_by) { + if (((*buffer)->size + expand_by) > MAX_LIMIT) { + SCLogWarning(SC_ERR_MEM_BUFFER_API, "Mem buffer asked to create " + "buffer with size greater than API limit - %d", MAX_LIMIT); + return -1; + } + + uint32_t total_size = (*buffer)->size + sizeof(MemBuffer) + expand_by; + + MemBuffer *tbuffer = SCRealloc(*buffer, total_size); + if (unlikely(tbuffer == NULL)) { + return -1; + } + + *buffer = tbuffer; + (*buffer)->size += expand_by; + (*buffer)->buffer = (uint8_t *)tbuffer + sizeof(MemBuffer); + + SCLogDebug("expanded buffer by %u, size is now %u", expand_by, (*buffer)->size); + return 0; +} + +void MemBufferFree(MemBuffer *buffer) +{ + SCFree(buffer); + + return; +} |