sm_memtable.h

#ifndef _INCLUDE_SOURCEMOD_CORE_STRINGTABLE_H_
#define _INCLUDE_SOURCEMOD_CORE_STRINGTABLE_H_

#include <stdlib.h>
#include <string.h>

class BaseMemTable
{
public:
        BaseMemTable(unsigned int init_size)
        {
                membase = (unsigned char *)malloc(init_size);
                size = init_size;
                tail = 0;
        }
        ~BaseMemTable()
        {
                free(membase);
                membase = NULL;
        }
public:
        int CreateMem(unsigned int addsize, void **addr)
        {
                int idx = (int)tail;

                while (tail + addsize >= size) {
                        size *= 2;
                        membase = (unsigned char *)realloc(membase, size);
                }

                tail += addsize;
                if (addr)
                        *addr = (void *)&membase[idx];

                return idx;
        }

        void *GetAddress(int index)
        {
                if (index < 0 || (unsigned int)index >= tail)
                        return NULL;
                return &membase[index];
        }

        void Reset()
        {
                tail = 0;
        }

        inline unsigned int GetMemUsage()
        {
                return size;
        }

        inline unsigned int GetActualMemUsed()
        {
                return tail;
        }

private:
        unsigned char *membase;
        unsigned int size;
        unsigned int tail;
};

class BaseStringTable
{
public:
        BaseStringTable(unsigned int init_size) : m_table(init_size)
        {
        }
public:
        int AddString(const char *string)
        {
                return AddString(string, strlen(string));
        }

        int AddString(const char *string, size_t length)
        {
                size_t len = length + 1;
                int idx;
                char *addr;

                idx = m_table.CreateMem(len, (void **)&addr);
                memcpy(addr, string, length + 1);
                return idx;
        }

        inline const char *GetString(int str)
        {
                return (const char *)m_table.GetAddress(str);
        }

        void Reset()
        {
                m_table.Reset();
        }

        inline BaseMemTable *GetMemTable()
        {
                return &m_table;
        }
private:
        BaseMemTable m_table;
};

#endif //_INCLUDE_SOURCEMOD_CORE_STRINGTABLE_H_