dvander@1779: /**
dvander@1779: * vim: set ts=4 :
dvander@1779: * =============================================================================
dvander@1779: * SourceMod
ds@1947: * Copyright (C) 2004-2008 AlliedModders LLC. All rights reserved.
dvander@1779: * =============================================================================
dvander@1779: *
dvander@1779: * This program is free software; you can redistribute it and/or modify it under
dvander@1779: * the terms of the GNU General Public License, version 3.0, as published by the
dvander@1779: * Free Software Foundation.
dvander@1779: *
dvander@1779: * This program is distributed in the hope that it will be useful, but WITHOUT
dvander@1779: * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
dvander@1779: * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
dvander@1779: * details.
dvander@1779: *
dvander@1779: * You should have received a copy of the GNU General Public License along with
dvander@1779: * this program. If not, see .
dvander@1779: *
dvander@1779: * As a special exception, AlliedModders LLC gives you permission to link the
dvander@1779: * code of this program (as well as its derivative works) to "Half-Life 2," the
dvander@1779: * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software
dvander@1779: * by the Valve Corporation. You must obey the GNU General Public License in
dvander@1779: * all respects for all other code used. Additionally, AlliedModders LLC grants
dvander@1779: * this exception to all derivative works. AlliedModders LLC defines further
dvander@1779: * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),
dvander@1779: * or .
dvander@1779: *
dvander@1779: * Version: $Id$
dvander@1779: */
dvander@1779:
dvander@2471: #include
dvander@1779: #include "sm_globals.h"
dvander@1779: #include "sm_stringutil.h"
dvander@1779: #include "HandleSys.h"
dvander@1779: #include "sm_trie_tpl.h"
dvander@1779:
dvander@1779: HandleType_t htCellTrie;
dvander@1779:
dvander@1779: enum TrieNodeType
dvander@1779: {
dvander@1779: TrieNode_Cell,
dvander@1779: TrieNode_CellArray,
dvander@1779: TrieNode_String,
dvander@1779: };
dvander@1779:
dvander@1779: struct SmartTrieNode
dvander@1779: {
dvander@1779: SmartTrieNode()
dvander@1779: {
dvander@1779: ptr = NULL;
dvander@1779: type = TrieNode_Cell;
dvander@1779: }
dvander@1779: SmartTrieNode(const SmartTrieNode &obj)
dvander@1779: {
dvander@1779: type = obj.type;
dvander@1779: ptr = obj.ptr;
dvander@1779: data = obj.data;
dvander@1779: data_len = obj.data_len;
dvander@1779: }
dvander@1779: SmartTrieNode & operator =(const SmartTrieNode &src)
dvander@1779: {
dvander@1779: type = src.type;
dvander@1779: ptr = src.ptr;
dvander@1779: data = src.data;
dvander@1779: data_len = src.data_len;
dvander@1779: return *this;
dvander@1779: }
dvander@1779: TrieNodeType type;
dvander@1779: cell_t *ptr;
dvander@1779: cell_t data;
dvander@1779: cell_t data_len;
dvander@1779: };
dvander@1779:
dvander@1779: struct CellTrie
dvander@1779: {
dvander@1779: KTrie trie;
dvander@1779: cell_t mem_usage;
dvander@1779: };
dvander@1779:
dvander@1779: class TrieHelpers :
dvander@1779: public SMGlobalClass,
dvander@1779: public IHandleTypeDispatch
dvander@1779: {
dvander@1779: public: //SMGlobalClass
dvander@1779: void OnSourceModAllInitialized()
dvander@1779: {
dvander@1779: htCellTrie = g_HandleSys.CreateType("Trie", this, 0, NULL, NULL, g_pCoreIdent, NULL);
dvander@1779: }
dvander@1779: void OnSourceModShutdown()
dvander@1779: {
dvander@1779: g_HandleSys.RemoveType(htCellTrie, g_pCoreIdent);
dvander@1779: }
dvander@1779: public: //IHandleTypeDispatch
dvander@1779: static void DestroySmartTrieNode(SmartTrieNode *pNode)
dvander@1779: {
dvander@1779: free(pNode->ptr);
dvander@1779: }
dvander@1779: void OnHandleDestroy(HandleType_t type, void *object)
dvander@1779: {
dvander@1779: CellTrie *pTrie = (CellTrie *)object;
dvander@1779:
dvander@1779: pTrie->trie.run_destructor(DestroySmartTrieNode);
dvander@1779:
dvander@1779: delete pTrie;
dvander@1779: }
dvander@1779: bool GetHandleApproxSize(HandleType_t type, void *object, unsigned int *pSize)
dvander@1779: {
dvander@1779: CellTrie *pArray = (CellTrie *)object;
dvander@1779: *pSize = sizeof(CellTrie) + pArray->mem_usage + pArray->trie.mem_usage();
dvander@1779: return true;
dvander@1779: }
dvander@1779: } s_CellTrieHelpers;
dvander@1779:
dvander@1779: static cell_t CreateTrie(IPluginContext *pContext, const cell_t *params)
dvander@1779: {
dvander@1779: CellTrie *pTrie = new CellTrie;
dvander@1779: Handle_t hndl;
dvander@1779:
dvander@1779: pTrie->mem_usage = 0;
dvander@1779:
dvander@1779: if ((hndl = g_HandleSys.CreateHandle(htCellTrie, pTrie, pContext->GetIdentity(), g_pCoreIdent, NULL))
dvander@1779: == BAD_HANDLE)
dvander@1779: {
dvander@1779: delete pTrie;
dvander@1779: return BAD_HANDLE;
dvander@1779: }
dvander@1779:
dvander@1779: return hndl;
dvander@1779: }
dvander@1779:
dvander@1779: static void UpdateNodeCells(CellTrie *pTrie, SmartTrieNode *pData, const cell_t *cells, cell_t num_cells)
dvander@1779: {
dvander@1779: if (num_cells == 1)
dvander@1779: {
dvander@1779: pData->data = *cells;
dvander@1779: pData->type = TrieNode_Cell;
dvander@1779: }
dvander@1779: else
dvander@1779: {
dvander@1779: pData->type = TrieNode_CellArray;
dvander@1779: if (pData->ptr == NULL)
dvander@1779: {
dvander@1779: pData->ptr = (cell_t *)malloc(num_cells * sizeof(cell_t));
dvander@1779: pData->data_len = num_cells;
dvander@1779: pTrie->mem_usage += (pData->data_len * sizeof(cell_t));
dvander@1779: }
dvander@1779: else if (pData->data_len < num_cells)
dvander@1779: {
dvander@1779: pData->ptr = (cell_t *)realloc(pData->ptr, num_cells * sizeof(cell_t));
dvander@1779: pTrie->mem_usage += (num_cells - pData->data_len) * sizeof(cell_t);
dvander@1779: pData->data_len = num_cells;
dvander@1779: }
dvander@1779: if (num_cells != 0)
dvander@1779: {
dvander@1779: memcpy(pData->ptr, cells, sizeof(cell_t) * num_cells);
dvander@1779: }
dvander@1779: pData->data = num_cells;
dvander@1779: }
dvander@1779: }
dvander@1779:
dvander@1779: static void UpdateNodeString(CellTrie *pTrie, SmartTrieNode *pData, const char *str)
dvander@1779: {
dvander@1779: size_t len = strlen(str);
dvander@1779: cell_t num_cells = (len + sizeof(cell_t)) / sizeof(cell_t);
dvander@1779:
dvander@1779: if (pData->ptr == NULL)
dvander@1779: {
dvander@1779: pData->ptr = (cell_t *)malloc(num_cells * sizeof(cell_t));
dvander@1779: pData->data_len = num_cells;
dvander@1779: pTrie->mem_usage += (pData->data_len * sizeof(cell_t));
dvander@1779: }
dvander@1779: else if (pData->data_len < num_cells)
dvander@1779: {
dvander@1779: pData->ptr = (cell_t *)realloc(pData->ptr, num_cells * sizeof(cell_t));
dvander@1779: pTrie->mem_usage += (num_cells - pData->data_len) * sizeof(cell_t);
dvander@1779: pData->data_len = num_cells;
dvander@1779: }
dvander@1779:
dvander@1779: strcpy((char *)pData->ptr, str);
dvander@1779: pData->data = len;
dvander@1779: pData->type = TrieNode_String;
dvander@1779: }
dvander@1779:
dvander@1779: static cell_t SetTrieValue(IPluginContext *pContext, const cell_t *params)
dvander@1779: {
dvander@1779: Handle_t hndl;
dvander@1779: CellTrie *pTrie;
dvander@1779: HandleError err;
dvander@1779: HandleSecurity sec = HandleSecurity(pContext->GetIdentity(), g_pCoreIdent);
dvander@1779:
dvander@1779: hndl = params[1];
dvander@1779:
dvander@1779: if ((err = g_HandleSys.ReadHandle(hndl, htCellTrie, &sec, (void **)&pTrie))
dvander@1779: != HandleError_None)
dvander@1779: {
dvander@1779: return pContext->ThrowNativeError("Invalid Handle %x (error %d)", hndl, err);
dvander@1779: }
dvander@1779:
dvander@1779: char *key;
dvander@1779: pContext->LocalToString(params[2], &key);
dvander@1779:
dvander@1779: SmartTrieNode *pNode;
dvander@1779: if ((pNode = pTrie->trie.retrieve(key)) == NULL)
dvander@1779: {
dvander@1779: SmartTrieNode node;
dvander@1779: UpdateNodeCells(pTrie, &node, ¶ms[3], 1);
dvander@1779: return pTrie->trie.insert(key, node) ? 1 : 0;
dvander@1779: }
dvander@1779:
dvander@1779: if (!params[4])
dvander@1779: {
dvander@1779: return 0;
dvander@1779: }
dvander@1779:
dvander@1779: UpdateNodeCells(pTrie, pNode, ¶ms[3], 1);
dvander@1779:
dvander@1779: return 1;
dvander@1779: }
dvander@1779:
dvander@1779: static cell_t SetTrieArray(IPluginContext *pContext, const cell_t *params)
dvander@1779: {
dvander@1779: Handle_t hndl;
dvander@1779: CellTrie *pTrie;
dvander@1779: HandleError err;
dvander@1779: HandleSecurity sec = HandleSecurity(pContext->GetIdentity(), g_pCoreIdent);
dvander@1779:
dvander@1779: hndl = params[1];
dvander@1779:
dvander@1779: if ((err = g_HandleSys.ReadHandle(hndl, htCellTrie, &sec, (void **)&pTrie))
dvander@1779: != HandleError_None)
dvander@1779: {
dvander@1779: return pContext->ThrowNativeError("Invalid Handle %x (error %d)", hndl, err);
dvander@1779: }
dvander@1779:
dvander@1779: if (params[4] < 0)
dvander@1779: {
dvander@1779: return pContext->ThrowNativeError("Invalid array size: %d", params[4]);
dvander@1779: }
dvander@1779:
dvander@1779: char *key;
dvander@1779: cell_t *array;
dvander@1779: pContext->LocalToString(params[2], &key);
dvander@1779: pContext->LocalToPhysAddr(params[3], &array);
dvander@1779:
dvander@1779: SmartTrieNode *pNode;
dvander@1779: if ((pNode = pTrie->trie.retrieve(key)) == NULL)
dvander@1779: {
dvander@1779: SmartTrieNode node;
dvander@1779: UpdateNodeCells(pTrie, &node, array, params[4]);
dvander@1779: if (!pTrie->trie.insert(key, node))
dvander@1779: {
dvander@1779: free(node.ptr);
dvander@1779: return 0;
dvander@1779: }
dvander@1779: return 1;
dvander@1779: }
dvander@1779:
dvander@1779: if (!params[4])
dvander@1779: {
dvander@1779: return 0;
dvander@1779: }
dvander@1779:
dvander@1779: UpdateNodeCells(pTrie, pNode, array, params[4]);
dvander@1779:
dvander@1779: return 1;
dvander@1779: }
dvander@1779:
dvander@1779: static cell_t SetTrieString(IPluginContext *pContext, const cell_t *params)
dvander@1779: {
dvander@1779: Handle_t hndl;
dvander@1779: CellTrie *pTrie;
dvander@1779: HandleError err;
dvander@1779: HandleSecurity sec = HandleSecurity(pContext->GetIdentity(), g_pCoreIdent);
dvander@1779:
dvander@1779: hndl = params[1];
dvander@1779:
dvander@1779: if ((err = g_HandleSys.ReadHandle(hndl, htCellTrie, &sec, (void **)&pTrie))
dvander@1779: != HandleError_None)
dvander@1779: {
dvander@1779: return pContext->ThrowNativeError("Invalid Handle %x (error %d)", hndl, err);
dvander@1779: }
dvander@1779:
dvander@1779: char *key, *val;
dvander@1779: pContext->LocalToString(params[2], &key);
dvander@1779: pContext->LocalToString(params[3], &val);
dvander@1779:
dvander@1779: SmartTrieNode *pNode;
dvander@1779: if ((pNode = pTrie->trie.retrieve(key)) == NULL)
dvander@1779: {
dvander@1779: SmartTrieNode node;
dvander@1779: UpdateNodeString(pTrie, &node, val);
dvander@1779: if (!pTrie->trie.insert(key, node))
dvander@1779: {
dvander@1779: free(node.ptr);
dvander@1779: return 0;
dvander@1779: }
dvander@1779: return 1;
dvander@1779: }
dvander@1779:
dvander@1779: if (!params[4])
dvander@1779: {
dvander@1779: return 0;
dvander@1779: }
dvander@1779:
dvander@1779: UpdateNodeString(pTrie, pNode, val);
dvander@1779:
dvander@1779: return 1;
dvander@1779: }
dvander@1779:
dvander@1779: static cell_t RemoveFromTrie(IPluginContext *pContext, const cell_t *params)
dvander@1779: {
dvander@1779: Handle_t hndl;
dvander@1779: CellTrie *pTrie;
dvander@1779: HandleError err;
dvander@1779: HandleSecurity sec = HandleSecurity(pContext->GetIdentity(), g_pCoreIdent);
dvander@1779:
dvander@1779: hndl = params[1];
dvander@1779:
dvander@1779: if ((err = g_HandleSys.ReadHandle(hndl, htCellTrie, &sec, (void **)&pTrie))
dvander@1779: != HandleError_None)
dvander@1779: {
dvander@1779: return pContext->ThrowNativeError("Invalid Handle %x (error %d)", hndl, err);
dvander@1779: }
dvander@1779:
dvander@1779: char *key;
dvander@1779: pContext->LocalToString(params[2], &key);
dvander@1779:
dvander@1779: SmartTrieNode *pNode;
dvander@1779: if ((pNode = pTrie->trie.retrieve(key)) == NULL)
dvander@1779: {
dvander@1779: return 0;
dvander@1779: }
dvander@1779:
dvander@1779: free(pNode->ptr);
dvander@1779: pNode->ptr = NULL;
dvander@1779:
dvander@1779: return pTrie->trie.remove(key) ? 1 : 0;
dvander@1779: }
dvander@1779:
dvander@1779: static cell_t ClearTrie(IPluginContext *pContext, const cell_t *params)
dvander@1779: {
dvander@1779: Handle_t hndl;
dvander@1779: CellTrie *pTrie;
dvander@1779: HandleError err;
dvander@1779: HandleSecurity sec = HandleSecurity(pContext->GetIdentity(), g_pCoreIdent);
dvander@1779:
dvander@1779: hndl = params[1];
dvander@1779:
dvander@1779: if ((err = g_HandleSys.ReadHandle(hndl, htCellTrie, &sec, (void **)&pTrie))
dvander@1779: != HandleError_None)
dvander@1779: {
dvander@1779: return pContext->ThrowNativeError("Invalid Handle %x (error %d)", hndl, err);
dvander@1779: }
dvander@1779:
dvander@1779: pTrie->trie.run_destructor(TrieHelpers::DestroySmartTrieNode);
dvander@1779: pTrie->trie.clear();
dvander@1779:
dvander@1779: return 1;
dvander@1779: }
dvander@1779:
dvander@1779: static cell_t GetTrieValue(IPluginContext *pContext, const cell_t *params)
dvander@1779: {
dvander@1779: Handle_t hndl;
dvander@1779: CellTrie *pTrie;
dvander@1779: HandleError err;
dvander@1779: HandleSecurity sec = HandleSecurity(pContext->GetIdentity(), g_pCoreIdent);
dvander@1779:
dvander@1779: hndl = params[1];
dvander@1779:
dvander@1779: if ((err = g_HandleSys.ReadHandle(hndl, htCellTrie, &sec, (void **)&pTrie))
dvander@1779: != HandleError_None)
dvander@1779: {
dvander@1779: return pContext->ThrowNativeError("Invalid Handle %x (error %d)", hndl, err);
dvander@1779: }
dvander@1779:
dvander@1779: char *key;
dvander@1779: cell_t *pValue;
dvander@1779: pContext->LocalToString(params[2], &key);
dvander@1779: pContext->LocalToPhysAddr(params[3], &pValue);
dvander@1779:
dvander@1779: SmartTrieNode *pNode;
dvander@1779: if ((pNode = pTrie->trie.retrieve(key)) == NULL)
dvander@1779: {
dvander@1779: return 0;
dvander@1779: }
dvander@1779:
dvander@1779: if (pNode->type == TrieNode_Cell)
dvander@1779: {
dvander@1779: *pValue = pNode->data;
dvander@1779: return 1;
dvander@1779: }
dvander@1779:
dvander@1779: return 0;
dvander@1779: }
dvander@1779:
dvander@1779: static cell_t GetTrieArray(IPluginContext *pContext, const cell_t *params)
dvander@1779: {
dvander@1779: Handle_t hndl;
dvander@1779: CellTrie *pTrie;
dvander@1779: HandleError err;
dvander@1779: HandleSecurity sec = HandleSecurity(pContext->GetIdentity(), g_pCoreIdent);
dvander@1779:
dvander@1779: hndl = params[1];
dvander@1779:
dvander@1779: if ((err = g_HandleSys.ReadHandle(hndl, htCellTrie, &sec, (void **)&pTrie))
dvander@1779: != HandleError_None)
dvander@1779: {
dvander@1779: return pContext->ThrowNativeError("Invalid Handle %x (error %d)", hndl, err);
dvander@1779: }
dvander@1779:
dvander@1779: if (params[4] < 0)
dvander@1779: {
dvander@1779: return pContext->ThrowNativeError("Invalid array size: %d", params[4]);
dvander@1779: }
dvander@1779:
dvander@1779: char *key;
dvander@1779: cell_t *pValue, *pSize;
dvander@1779: pContext->LocalToString(params[2], &key);
dvander@1779: pContext->LocalToPhysAddr(params[3], &pValue);
dvander@1779: pContext->LocalToPhysAddr(params[5], &pSize);
dvander@1779:
dvander@1779: SmartTrieNode *pNode;
dvander@1779: if ((pNode = pTrie->trie.retrieve(key)) == NULL
dvander@1779: || pNode->type != TrieNode_CellArray)
dvander@1779: {
dvander@1779: return 0;
dvander@1779: }
dvander@1779:
dvander@1779: if (pNode->ptr == NULL)
dvander@1779: {
dvander@1779: *pSize = 0;
dvander@1779: return 1;
dvander@1779: }
dvander@1779:
dvander@1779: if (pNode->data > params[4])
dvander@1779: {
dvander@1779: *pSize = params[4];
dvander@1779: }
dvander@1779: else if (params[4] != 0)
dvander@1779: {
dvander@1779: *pSize = pNode->data;
dvander@1779: }
dvander@1779: else
dvander@1779: {
dvander@1779: return 1;
dvander@1779: }
dvander@1779:
dvander@1779: memcpy(pValue, pNode->ptr, sizeof(cell_t) * pSize[0]);
dvander@1779:
dvander@1779: return 1;
dvander@1779: }
dvander@1779:
dvander@1779: static cell_t GetTrieString(IPluginContext *pContext, const cell_t *params)
dvander@1779: {
dvander@1779: Handle_t hndl;
dvander@1779: CellTrie *pTrie;
dvander@1779: HandleError err;
dvander@1779: HandleSecurity sec = HandleSecurity(pContext->GetIdentity(), g_pCoreIdent);
dvander@1779:
dvander@1779: hndl = params[1];
dvander@1779:
dvander@1779: if ((err = g_HandleSys.ReadHandle(hndl, htCellTrie, &sec, (void **)&pTrie))
dvander@1779: != HandleError_None)
dvander@1779: {
dvander@1779: return pContext->ThrowNativeError("Invalid Handle %x (error %d)", hndl, err);
dvander@1779: }
dvander@1779:
dvander@1779: if (params[4] < 0)
dvander@1779: {
dvander@1779: return pContext->ThrowNativeError("Invalid buffer size: %d", params[4]);
dvander@1779: }
dvander@1779:
dvander@1779: char *key;
dvander@1779: cell_t *pSize;
dvander@1779: pContext->LocalToString(params[2], &key);
dvander@1779: pContext->LocalToPhysAddr(params[5], &pSize);
dvander@1779:
dvander@1779: SmartTrieNode *pNode;
dvander@1779: if ((pNode = pTrie->trie.retrieve(key)) == NULL
dvander@1779: || pNode->type != TrieNode_String)
dvander@1779: {
dvander@1779: return 0;
dvander@1779: }
dvander@1779:
dvander@1779: if (pNode->ptr == NULL)
dvander@1779: {
dvander@1779: *pSize = 0;
dvander@1779: pContext->StringToLocal(params[3], params[4], "");
dvander@1779: return 1;
dvander@1779: }
dvander@1779:
dvander@1779: size_t written;
dvander@1779: pContext->StringToLocalUTF8(params[3], params[4], (char *)pNode->ptr, &written);
dvander@1779:
dvander@1779: *pSize = (cell_t)written;
dvander@1779:
dvander@1779: return 1;
dvander@1779: }
dvander@1779:
dvander@1829: static cell_t GetTrieSize(IPluginContext *pContext, const cell_t *params)
dvander@1829: {
dvander@1829: Handle_t hndl;
dvander@1829: CellTrie *pTrie;
dvander@1829: HandleError err;
dvander@1829: HandleSecurity sec = HandleSecurity(pContext->GetIdentity(), g_pCoreIdent);
dvander@1829:
dvander@1829: hndl = params[1];
dvander@1829:
dvander@1829: if ((err = g_HandleSys.ReadHandle(hndl, htCellTrie, &sec, (void **)&pTrie))
dvander@1829: != HandleError_None)
dvander@1829: {
dvander@1829: return pContext->ThrowNativeError("Invalid Handle %x (error %d)", hndl, err);
dvander@1829: }
dvander@1829:
dvander@1829: return pTrie->trie.size();
dvander@1829: }
dvander@1829:
dvander@1779: REGISTER_NATIVES(trieNatives)
dvander@1779: {
dvander@1779: {"ClearTrie", ClearTrie},
dvander@1779: {"CreateTrie", CreateTrie},
dvander@1779: {"GetTrieArray", GetTrieArray},
dvander@1779: {"GetTrieString", GetTrieString},
dvander@1779: {"GetTrieValue", GetTrieValue},
dvander@1779: {"RemoveFromTrie", RemoveFromTrie},
dvander@1779: {"SetTrieArray", SetTrieArray},
dvander@1779: {"SetTrieString", SetTrieString},
dvander@1779: {"SetTrieValue", SetTrieValue},
dvander@1829: {"GetTrieSize", GetTrieSize},
dvander@1779: {NULL, NULL},
dvander@1779: };