faluco@490: /**
ds@653: * vim: set ts=4 :
ds@1330: * =============================================================================
dvander@1231: * SourceMod
ds@1947: * Copyright (C) 2004-2008 AlliedModders LLC. All rights reserved.
ds@1330: * =============================================================================
ds@653: *
ds@1330: * This program is free software; you can redistribute it and/or modify it under
ds@1330: * the terms of the GNU General Public License, version 3.0, as published by the
ds@1330: * Free Software Foundation.
dvander@1231: *
ds@1330: * This program is distributed in the hope that it will be useful, but WITHOUT
ds@1330: * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
ds@1330: * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
ds@1330: * details.
dvander@1231: *
ds@1330: * You should have received a copy of the GNU General Public License along with
ds@1330: * this program. If not, see .
dvander@1231: *
ds@1330: * As a special exception, AlliedModders LLC gives you permission to link the
ds@1330: * code of this program (as well as its derivative works) to "Half-Life 2," the
ds@1330: * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software
ds@1330: * by the Valve Corporation. You must obey the GNU General Public License in
ds@1330: * all respects for all other code used. Additionally, AlliedModders LLC grants
ds@1330: * this exception to all derivative works. AlliedModders LLC defines further
ds@1330: * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),
ds@1330: * or .
ds@653: *
ds@653: * Version: $Id$
ds@653: */
faluco@490:
faluco@484: #include
faluco@484: #include
faluco@484: #include "CDataPack.h"
faluco@484:
faluco@489: #define DATAPACK_INITIAL_SIZE 512
faluco@484:
faluco@484: CDataPack::CDataPack()
faluco@484: {
faluco@484: m_pBase = (char *)malloc(DATAPACK_INITIAL_SIZE);
faluco@484: m_capacity = DATAPACK_INITIAL_SIZE;
faluco@484: Initialize();
faluco@484: }
faluco@484:
faluco@484: CDataPack::~CDataPack()
faluco@484: {
faluco@484: free(m_pBase);
faluco@484: }
faluco@484:
faluco@484: void CDataPack::Initialize()
faluco@484: {
faluco@484: m_curptr = m_pBase;
faluco@484: m_size = 0;
faluco@484: }
faluco@484:
faluco@484: void CDataPack::CheckSize(size_t typesize)
faluco@484: {
faluco@484: if (m_curptr - m_pBase + typesize <= m_capacity)
faluco@484: {
faluco@484: return;
faluco@484: }
faluco@484:
faluco@484: size_t pos = m_curptr - m_pBase;
faluco@484: do
faluco@484: {
faluco@484: m_capacity *= 2;
faluco@484: m_pBase = (char *)realloc(m_pBase, m_capacity);
faluco@484: m_curptr = m_pBase + pos;
faluco@484: } while (m_curptr - m_pBase + typesize > m_capacity);
faluco@484: }
faluco@484:
faluco@484: void CDataPack::ResetSize()
faluco@484: {
faluco@484: m_size = 0;
faluco@484: }
faluco@484:
faluco@484: size_t CDataPack::CreateMemory(size_t size, void **addr)
faluco@484: {
faluco@488: CheckSize(sizeof(size_t) + size);
faluco@484: size_t pos = m_curptr - m_pBase;
faluco@484:
faluco@488: *(size_t *)m_curptr = size;
faluco@488: m_curptr += sizeof(size_t);
faluco@484:
faluco@484: if (addr)
faluco@484: {
faluco@488: *addr = m_curptr;
faluco@484: }
faluco@484:
faluco@488: m_curptr += size;
faluco@488: m_size += sizeof(size_t) + size;
faluco@488:
faluco@484: return pos;
faluco@484: }
faluco@484:
faluco@484: void CDataPack::PackCell(cell_t cell)
faluco@484: {
faluco@488: CheckSize(sizeof(size_t) + sizeof(cell_t));
faluco@488:
faluco@488: *(size_t *)m_curptr = sizeof(cell_t);
faluco@488: m_curptr += sizeof(size_t);
faluco@484:
faluco@484: *(cell_t *)m_curptr = cell;
faluco@484: m_curptr += sizeof(cell_t);
faluco@488:
faluco@488: m_size += sizeof(size_t) + sizeof(cell_t);
faluco@484: }
faluco@484:
faluco@484: void CDataPack::PackFloat(float val)
faluco@484: {
faluco@488: CheckSize(sizeof(size_t) + sizeof(float));
faluco@488:
faluco@488: *(size_t *)m_curptr = sizeof(float);
faluco@488: m_curptr += sizeof(size_t);
faluco@484:
faluco@484: *(float *)m_curptr = val;
faluco@484: m_curptr += sizeof(float);
faluco@488:
faluco@488: m_size += sizeof(size_t) + sizeof(float);
faluco@484: }
faluco@484:
faluco@484: void CDataPack::PackString(const char *string)
faluco@484: {
faluco@484: size_t len = strlen(string);
faluco@484: size_t maxsize = sizeof(size_t) + len + 1;
faluco@484: CheckSize(maxsize);
faluco@484:
faluco@484: // Pack the string length first for buffer overrun checking.
faluco@484: *(size_t *)m_curptr = len;
faluco@484: m_curptr += sizeof(size_t);
faluco@484:
faluco@484: // Now pack the string.
faluco@484: memcpy(m_curptr, string, len);
faluco@484: m_curptr[len] = '\0';
faluco@484: m_curptr += len + 1;
faluco@484:
faluco@484: m_size += maxsize;
faluco@484: }
faluco@484:
faluco@484: void CDataPack::Reset() const
faluco@484: {
faluco@484: m_curptr = m_pBase;
faluco@484: }
faluco@484:
faluco@484: size_t CDataPack::GetPosition() const
faluco@484: {
faluco@484: return static_cast(m_curptr - m_pBase);
faluco@484: }
faluco@484:
faluco@484: bool CDataPack::SetPosition(size_t pos) const
faluco@484: {
faluco@484: if (pos > m_size-1)
faluco@484: {
faluco@484: return false;
faluco@484: }
faluco@484: m_curptr = m_pBase + pos;
faluco@484:
faluco@484: return true;
faluco@484: }
faluco@484:
faluco@484: cell_t CDataPack::ReadCell() const
faluco@484: {
faluco@488: if (!IsReadable(sizeof(size_t) + sizeof(cell_t)))
faluco@484: {
faluco@484: return 0;
faluco@484: }
faluco@488: if (*reinterpret_cast(m_curptr) != sizeof(cell_t))
faluco@488: {
faluco@488: return 0;
faluco@488: }
faluco@488:
faluco@488: m_curptr += sizeof(size_t);
faluco@488:
faluco@484: cell_t val = *reinterpret_cast(m_curptr);
faluco@484: m_curptr += sizeof(cell_t);
faluco@484: return val;
faluco@484: }
faluco@484:
faluco@484: float CDataPack::ReadFloat() const
faluco@484: {
faluco@488: if (!IsReadable(sizeof(size_t) + sizeof(float)))
faluco@484: {
faluco@484: return 0;
faluco@484: }
faluco@488: if (*reinterpret_cast(m_curptr) != sizeof(float))
faluco@488: {
faluco@488: return 0;
faluco@488: }
faluco@488:
faluco@488: m_curptr += sizeof(size_t);
faluco@488:
faluco@484: float val = *reinterpret_cast(m_curptr);
faluco@484: m_curptr += sizeof(float);
faluco@484: return val;
faluco@484: }
faluco@484:
faluco@484: bool CDataPack::IsReadable(size_t bytes) const
faluco@484: {
faluco@484: return (bytes + (m_curptr - m_pBase) > m_size) ? false : true;
faluco@484: }
faluco@484:
faluco@484: const char *CDataPack::ReadString(size_t *len) const
faluco@484: {
faluco@484: if (!IsReadable(sizeof(size_t)))
faluco@484: {
faluco@488: return NULL;
faluco@484: }
faluco@484:
faluco@484: size_t real_len = *(size_t *)m_curptr;
faluco@484:
faluco@484: m_curptr += sizeof(size_t);
faluco@484: char *str = (char *)m_curptr;
faluco@484:
faluco@484: if ((strlen(str) != real_len) || !(IsReadable(real_len+1)))
faluco@484: {
faluco@484: return NULL;
faluco@484: }
faluco@484:
faluco@484: if (len)
faluco@484: {
faluco@484: *len = real_len;
faluco@484: }
faluco@484:
faluco@484: m_curptr += real_len + 1;
faluco@484:
faluco@484: return str;
faluco@484: }
faluco@484:
faluco@484: void *CDataPack::GetMemory() const
faluco@484: {
faluco@484: return m_curptr;
faluco@484: }
faluco@488:
faluco@488: void *CDataPack::ReadMemory(size_t *size) const
faluco@488: {
faluco@488: if (!IsReadable(sizeof(size_t)))
faluco@488: {
faluco@488: return NULL;
faluco@488: }
faluco@488:
faluco@488: size_t bytecount = *(size_t *)m_curptr;
faluco@488: m_curptr += sizeof(size_t);
faluco@488:
faluco@488: if (!IsReadable(bytecount))
faluco@488: {
faluco@488: return NULL;
faluco@488: }
faluco@488:
faluco@488: void *ptr = m_curptr;
faluco@488:
faluco@488: if (size)
faluco@488: {
faluco@488: *size = bytecount;
faluco@488: }
faluco@488:
faluco@488: m_curptr += bytecount;
faluco@488:
faluco@488: return ptr;
faluco@488: }