dvander@1779: /**
dvander@4096:  * vim: set ts=4 sw=4 tw=99 noet :
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 <http://www.gnu.org/licenses/>.
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 <http://www.sourcemod.net/license.php>.
dvander@1779:  *
dvander@1779:  * Version: $Id$
dvander@1779:  */
dvander@1779: 
dvander@2471: #include <stdlib.h>
dvander@2728: #include "common_logic.h"
dvander@4096: #include <am-moveable.h>
dvander@4096: #include <am-refcounting.h>
dvander@4096: #include <sm_stringhashmap.h>
dvander@4096: #include "sm_memtable.h"
dvander@1779: 
dvander@1779: HandleType_t htCellTrie;
dvander@4096: HandleType_t htSnapshot;
dvander@1779: 
dvander@4096: enum EntryType
dvander@1779: {
dvander@4096: 	EntryType_Cell,
dvander@4096: 	EntryType_CellArray,
dvander@4096: 	EntryType_String,
dvander@1779: };
dvander@1779: 
dvander@4096: class Entry
dvander@1779: {
dvander@4096: 	struct ArrayInfo
dvander@1779: 	{
dvander@4096: 		size_t length;
dvander@4096: 		size_t maxbytes;
dvander@4096: 
dvander@4096: 		void *base() {
dvander@4096: 			return this + 1;
dvander@4096: 		}
dvander@4096: 	};
dvander@4096: 
dvander@4096: public:
dvander@4096: 	Entry()
dvander@4096: 		: control_(0)
dvander@4096: 	{
dvander@1779: 	}
dvander@4096: 	Entry(ke::Moveable<Entry> other)
dvander@4096: 	{
dvander@4096: 		control_ = other->control_;
dvander@4096: 		data_ = other->data_;
dvander@4096: 		other->control_ = 0;
dvander@4096: 	}
dvander@4096: 
dvander@4096: 	~Entry()
dvander@1779: 	{
dvander@4096: 		free(raw());
dvander@4096: 	}
dvander@4096: 
dvander@4096: 	void setCell(cell_t value) {
dvander@4096: 		setType(EntryType_Cell);
dvander@4096: 		data_ = value;
dvander@4096: 	}
dvander@4096: 	void setArray(cell_t *cells, size_t length) {
dvander@4096: 		ArrayInfo *array = ensureArray(length * sizeof(cell_t));
dvander@4096: 		array->length = length;
dvander@4096: 		memcpy(array->base(), cells, length * sizeof(cell_t));
dvander@4096: 		setTypeAndPointer(EntryType_CellArray, array);
dvander@4096: 	}
dvander@4096: 	void setString(const char *str) {
dvander@4096: 		size_t length = strlen(str);
dvander@4096: 		ArrayInfo *array = ensureArray(length + 1);
dvander@4096: 		array->length = length;
dvander@4096: 		strcpy((char *)array->base(), str);
dvander@4096: 		setTypeAndPointer(EntryType_String, array);
dvander@4096: 	}
dvander@4096: 
dvander@4096: 	size_t arrayLength() const {
dvander@4096: 		assert(isArray());
dvander@4096: 		return raw()->length;
dvander@4096: 	}
dvander@4096: 	cell_t *array() const {
dvander@4096: 		assert(isArray());
dvander@4096: 		return reinterpret_cast<cell_t *>(raw()->base());
dvander@4096: 	}
dvander@4096: 	char *chars() const {
dvander@4096: 		assert(isString());
dvander@4096: 		return reinterpret_cast<char *>(raw()->base());
dvander@4096: 	}
dvander@4096: 	cell_t cell() const {
dvander@4096: 		assert(isCell());
dvander@4096: 		return data_;
dvander@4096: 	}
dvander@4096: 
dvander@4096: 	bool isCell() const {
dvander@4096: 		return type() == EntryType_Cell;
dvander@4096: 	}
dvander@4096: 	bool isArray() const {
dvander@4096: 		return type() == EntryType_CellArray;
dvander@1779: 	}
dvander@4096: 	bool isString() const {
dvander@4096: 		return type() == EntryType_String;
dvander@4096: 	}
dvander@4096: 
dvander@4096: private:
dvander@4096: 	Entry(const Entry &other) KE_DELETE;
dvander@4096: 
dvander@4096: 	ArrayInfo *ensureArray(size_t bytes) {
dvander@4096: 		ArrayInfo *array = raw();
dvander@4096: 		if (array && array->maxbytes >= bytes)
dvander@4096: 			return array;
dvander@4096: 		array = (ArrayInfo *)realloc(array, bytes + sizeof(ArrayInfo));
dvander@4096: 		if (!array)
dvander@4096: 		{
dvander@4096: 			fprintf(stderr, "Out of memory!\n");
dvander@4096: 			abort();
dvander@4096: 		}
dvander@4096: 		array->maxbytes = bytes;
dvander@4096: 		return array;
dvander@1779: 	}
dvander@4096: 
dvander@4096: 	// Pointer and type are overlaid, so we have some accessors.
dvander@4096: 	ArrayInfo *raw() const {
dvander@4096: 		return reinterpret_cast<ArrayInfo *>(control_ & ~uintptr_t(0x3));
dvander@4096: 	}
dvander@4096: 	void setType(EntryType aType) {
dvander@4096: 		control_ = uintptr_t(raw()) | uintptr_t(aType);
dvander@4096: 		assert(type() == aType);
dvander@4096: 	}
dvander@4096: 	void setTypeAndPointer(EntryType aType, ArrayInfo *ptr) {
dvander@4096: 		// malloc() should guarantee 8-byte alignment at worst
dvander@4096: 		assert((uintptr_t(ptr) & 0x3) == 0);
dvander@4096: 		control_ = uintptr_t(ptr) | uintptr_t(aType);
dvander@4096: 		assert(type() == aType);
dvander@4096: 	}
dvander@4096: 	EntryType type() const {
dvander@4096: 		return (EntryType)(control_ & 0x3);
dvander@4096: 	}
dvander@4096: 
dvander@4096: private:
dvander@4096: 	// Contains the bits for the type, and an array pointer, if one is set.
dvander@4096: 	uintptr_t control_;
dvander@4096: 
dvander@4096: 	// Contains data for cell-only entries.
dvander@4096: 	cell_t data_;
dvander@1779: };
dvander@1779: 
dvander@4096: struct CellTrie : public ke::Refcounted<CellTrie>
dvander@4096: {
dvander@4096: 	StringHashMap<Entry> map;
dvander@4096: };
dvander@4096: 
dvander@4096: struct TrieSnapshot
dvander@1779: {
dvander@4096: 	TrieSnapshot()
dvander@4096: 		: strings(128)
dvander@4096: 	{ }
dvander@4096: 
dvander@4096: 	size_t mem_usage()
dvander@4096: 	{
dvander@4096: 		return length * sizeof(int) + strings.GetMemTable()->GetMemUsage();
dvander@4096: 	}
dvander@4096: 
dvander@4096: 	size_t length;
dvander@4096: 	ke::AutoArray<int> keys;
dvander@4096: 	BaseStringTable strings;
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@2728: 		htCellTrie = handlesys->CreateType("Trie", this, 0, NULL, NULL, g_pCoreIdent, NULL);
dvander@4096: 		htSnapshot = handlesys->CreateType("TrieSnapshot", this, 0, NULL, NULL, g_pCoreIdent, NULL);
dvander@1779: 	}
dvander@1779: 	void OnSourceModShutdown()
dvander@1779: 	{
dvander@4096: 		handlesys->RemoveType(htSnapshot, g_pCoreIdent);
dvander@2728: 		handlesys->RemoveType(htCellTrie, g_pCoreIdent);
dvander@1779: 	}
dvander@1779: public: //IHandleTypeDispatch
dvander@1779: 	void OnHandleDestroy(HandleType_t type, void *object)
dvander@1779: 	{
dvander@4096: 		if (type == htCellTrie)
dvander@4096: 		{
dvander@4096: 			CellTrie *pTrie = (CellTrie *)object;
dvander@4096: 			pTrie->Release();
dvander@4096: 		} else {
dvander@4096: 			TrieSnapshot *snapshot = (TrieSnapshot *)object;
dvander@4096: 			delete snapshot;
dvander@4096: 		}
dvander@1779: 	}
dvander@1779: 	bool GetHandleApproxSize(HandleType_t type, void *object, unsigned int *pSize)
dvander@1779: 	{
dvander@4096: 		if (type == htCellTrie)
dvander@4096: 		{
dvander@4096: 			CellTrie *pArray = (CellTrie *)object;
dvander@4096: 			*pSize = sizeof(CellTrie) + pArray->map.mem_usage();
dvander@4096: 		} else {
dvander@4096: 			TrieSnapshot *snapshot = (TrieSnapshot *)object;
dvander@4096: 			*pSize = sizeof(TrieSnapshot) + snapshot->mem_usage();
dvander@4096: 		}
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@2728: 	if ((hndl = 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 cell_t SetTrieValue(IPluginContext *pContext, const cell_t *params)
dvander@1779: {
dvander@1779: 	CellTrie *pTrie;
dvander@1779: 	HandleError err;
dvander@1779: 	HandleSecurity sec = HandleSecurity(pContext->GetIdentity(), g_pCoreIdent);
dvander@1779: 
dvander@4096: 	Handle_t hndl = params[1];
dvander@1779: 
dvander@2728: 	if ((err = 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@4096: 	StringHashMap<Entry>::Insert i = pTrie->map.findForAdd(key);
dvander@4096: 	if (!i.found())
dvander@1779: 	{
dvander@4096: 		if (!pTrie->map.add(i, key))
dvander@4096: 			return 0;
dvander@4096: 		i->value.setCell(params[3]);
dvander@4096: 		return 1;
dvander@1779: 	}
dvander@1779: 
dvander@1779: 	if (!params[4])
dvander@1779: 		return 0;
dvander@1779: 
dvander@4096: 	i->value.setCell(params[3]);
dvander@1779: 	return 1;
dvander@1779: }
dvander@1779: 
dvander@1779: static cell_t SetTrieArray(IPluginContext *pContext, const cell_t *params)
dvander@1779: {
dvander@1779: 	CellTrie *pTrie;
dvander@1779: 	HandleError err;
dvander@1779: 	HandleSecurity sec = HandleSecurity(pContext->GetIdentity(), g_pCoreIdent);
dvander@1779: 
dvander@4096: 	Handle_t hndl = params[1];
dvander@1779: 
dvander@2728: 	if ((err = 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@4096: 	StringHashMap<Entry>::Insert i = pTrie->map.findForAdd(key);
dvander@4096: 	if (!i.found())
dvander@1779: 	{
dvander@4096: 		if (!pTrie->map.add(i, key))
dvander@1779: 			return 0;
dvander@4096: 		i->key = key;
dvander@4096: 		i->value.setArray(array, params[4]);
dvander@1779: 		return 1;
dvander@1779: 	}
dvander@1779: 
skamonkey@4340: 	if (!params[5])
dvander@1779: 		return 0;
dvander@1779: 
dvander@4096: 	i->value.setArray(array, params[4]);
dvander@1779: 	return 1;
dvander@1779: }
dvander@1779: 
dvander@1779: static cell_t SetTrieString(IPluginContext *pContext, const cell_t *params)
dvander@1779: {
dvander@1779: 	CellTrie *pTrie;
dvander@1779: 	HandleError err;
dvander@1779: 	HandleSecurity sec = HandleSecurity(pContext->GetIdentity(), g_pCoreIdent);
dvander@1779: 
dvander@4096: 	Handle_t hndl = params[1];
dvander@1779: 
dvander@2728: 	if ((err = 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@4096: 	StringHashMap<Entry>::Insert i = pTrie->map.findForAdd(key);
dvander@4096: 	if (!i.found())
dvander@1779: 	{
dvander@4096: 		if (!pTrie->map.add(i, key))
dvander@1779: 			return 0;
dvander@4096: 		i->value.setString(val);
dvander@1779: 		return 1;
dvander@1779: 	}
dvander@1779: 
dvander@1779: 	if (!params[4])
dvander@1779: 		return 0;
dvander@1779: 
dvander@4096: 	i->value.setString(val);
dvander@1779: 	return 1;
dvander@1779: }
dvander@1779: 
dvander@1779: static cell_t RemoveFromTrie(IPluginContext *pContext, const cell_t *params)
dvander@1779: {
dvander@1779: 	CellTrie *pTrie;
dvander@1779: 	HandleError err;
dvander@1779: 	HandleSecurity sec = HandleSecurity(pContext->GetIdentity(), g_pCoreIdent);
dvander@1779: 
dvander@4096: 	Handle_t hndl = params[1];
dvander@1779: 
dvander@2728: 	if ((err = 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@4096: 	StringHashMap<Entry>::Result r = pTrie->map.find(key);
dvander@4096: 	if (!r.found())
dvander@1779: 		return 0;
dvander@1779: 
dvander@4096: 	pTrie->map.remove(r);
dvander@4096: 	return 1;
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@2728: 	if ((err = 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@4096: 	pTrie->map.clear();
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@2728: 	if ((err = 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@4096: 	StringHashMap<Entry>::Result r = pTrie->map.find(key);
dvander@4096: 	if (!r.found())
dvander@4096: 		return 0;
dvander@4096: 
dvander@4096: 	if (r->value.isCell())
dvander@1779: 	{
dvander@4096: 		*pValue = r->value.cell();
dvander@4096: 		return 1;
dvander@1779: 	}
dvander@1779: 
dvander@4096: 	// Maintain compatibility with an old bug. If an array was set with one
dvander@4096: 	// cell, it was stored internally as a single cell. We now store as an
dvander@4096: 	// actual array, but we make GetTrieValue() still work for this case.
dvander@4096: 	if (r->value.isArray() && r->value.arrayLength() == 1)
dvander@1779: 	{
dvander@4096: 		*pValue = r->value.array()[0];
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@2728: 	if ((err = 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@4096: 
dvander@4096: 	StringHashMap<Entry>::Result r = pTrie->map.find(key);
dvander@4096: 	if (!r.found() || !r->value.isArray())
dvander@1779: 		return 0;
dvander@1779: 
dvander@4096: 	if (!r->value.array())
dvander@1779: 	{
dvander@1779: 		*pSize = 0;
dvander@1779: 		return 1;
dvander@1779: 	}
dvander@1779: 
dvander@4096: 	if (!params[4])
dvander@4096: 		return 1;
dvander@4096: 
dvander@4096: 	size_t length = r->value.arrayLength();
dvander@4096: 	cell_t *base = r->value.array();
dvander@4096: 
dvander@4096: 	if (length > size_t(params[4]))
dvander@1779: 		*pSize = params[4];
dvander@1779: 	else
dvander@4096: 		*pSize = length;
dvander@1779: 
dvander@4096: 	memcpy(pValue, base, sizeof(cell_t) * pSize[0]);
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@2728: 	if ((err = 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@4096: 	StringHashMap<Entry>::Result r = pTrie->map.find(key);
dvander@4096: 	if (!r.found() || !r->value.isString())
dvander@1779: 		return 0;
dvander@1779: 
dvander@1779: 	size_t written;
dvander@4096: 	pContext->StringToLocalUTF8(params[3], params[4], r->value.chars(), &written);
dvander@1779: 
dvander@1779: 	*pSize = (cell_t)written;
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@2728: 	if ((err = 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@4096: 	return pTrie->map.elements();
dvander@4096: }
dvander@4096: 
dvander@4096: static cell_t CreateTrieSnapshot(IPluginContext *pContext, const cell_t *params)
dvander@4096: {
dvander@4096: 	HandleError err;
dvander@4096: 	HandleSecurity sec = HandleSecurity(pContext->GetIdentity(), g_pCoreIdent);
dvander@4096: 
dvander@4096: 	Handle_t hndl = params[1];
dvander@4096: 
dvander@4096: 	CellTrie *pTrie;
dvander@4096: 	if ((err = handlesys->ReadHandle(hndl, htCellTrie, &sec, (void **)&pTrie))
dvander@4096: 		!= HandleError_None)
dvander@4096: 	{
dvander@4096: 		return pContext->ThrowNativeError("Invalid Handle %x (error %d)", hndl, err);
dvander@4096: 	}
dvander@4096: 
dvander@4096: 	TrieSnapshot *snapshot = new TrieSnapshot;
dvander@4096: 	snapshot->length = pTrie->map.elements();
dvander@4096: 	snapshot->keys = new int[snapshot->length];
dvander@4096: 	size_t i = 0;
dvander@4096: 	for (StringHashMap<Entry>::iterator iter = pTrie->map.iter(); !iter.empty(); iter.next(), i++)
dvander@4096: 		snapshot->keys[i] = snapshot->strings.AddString(iter->key.chars(), iter->key.length());
dvander@4096: 	assert(i == snapshot->length);
dvander@4096: 
dvander@4096: 	if ((hndl = handlesys->CreateHandle(htSnapshot, snapshot, pContext->GetIdentity(), g_pCoreIdent, NULL))
dvander@4096: 		== BAD_HANDLE)
dvander@4096: 	{
dvander@4096: 		delete snapshot;
dvander@4096: 		return BAD_HANDLE;
dvander@4096: 	}
dvander@4096: 
dvander@4096: 	return hndl;
dvander@4096: }
dvander@4096: 
dvander@4096: static cell_t TrieSnapshotLength(IPluginContext *pContext, const cell_t *params)
dvander@4096: {
dvander@4096: 	HandleError err;
dvander@4096: 	HandleSecurity sec = HandleSecurity(pContext->GetIdentity(), g_pCoreIdent);
dvander@4096: 
dvander@4096: 	Handle_t hndl = params[1];
dvander@4096: 
dvander@4096: 	TrieSnapshot *snapshot;
dvander@4096: 	if ((err = handlesys->ReadHandle(hndl, htSnapshot, &sec, (void **)&snapshot))
dvander@4096: 		!= HandleError_None)
dvander@4096: 	{
dvander@4096: 		return pContext->ThrowNativeError("Invalid Handle %x (error %d)", hndl, err);
dvander@4096: 	}
dvander@4096: 
dvander@4096: 	return snapshot->length;
dvander@4096: }
dvander@4096: 
dvander@4096: static cell_t TrieSnapshotKeyBufferSize(IPluginContext *pContext, const cell_t *params)
dvander@4096: {
dvander@4096: 	HandleError err;
dvander@4096: 	HandleSecurity sec = HandleSecurity(pContext->GetIdentity(), g_pCoreIdent);
dvander@4096: 
dvander@4096: 	Handle_t hndl = params[1];
dvander@4096: 
dvander@4096: 	TrieSnapshot *snapshot;
dvander@4096: 	if ((err = handlesys->ReadHandle(hndl, htSnapshot, &sec, (void **)&snapshot))
dvander@4096: 		!= HandleError_None)
dvander@4096: 	{
dvander@4096: 		return pContext->ThrowNativeError("Invalid Handle %x (error %d)", hndl, err);
dvander@4096: 	}
dvander@4096: 
dvander@4096: 	unsigned index = params[2];
dvander@4096: 	if (index >= snapshot->length)
dvander@4096: 		return pContext->ThrowNativeError("Invalid index %d", index);
dvander@4096: 
dvander@4096: 	return strlen(snapshot->strings.GetString(snapshot->keys[index])) + 1;
dvander@4096: }
dvander@4096: 
dvander@4096: static cell_t GetTrieSnapshotKey(IPluginContext *pContext, const cell_t *params)
dvander@4096: {
dvander@4096: 	HandleError err;
dvander@4096: 	HandleSecurity sec = HandleSecurity(pContext->GetIdentity(), g_pCoreIdent);
dvander@4096: 
dvander@4096: 	Handle_t hndl = params[1];
dvander@4096: 
dvander@4096: 	TrieSnapshot *snapshot;
dvander@4096: 	if ((err = handlesys->ReadHandle(hndl, htSnapshot, &sec, (void **)&snapshot))
dvander@4096: 		!= HandleError_None)
dvander@4096: 	{
dvander@4096: 		return pContext->ThrowNativeError("Invalid Handle %x (error %d)", hndl, err);
dvander@4096: 	}
dvander@4096: 
dvander@4096: 	unsigned index = params[2];
dvander@4096: 	if (index >= snapshot->length)
dvander@4096: 		return pContext->ThrowNativeError("Invalid index %d", index);
dvander@4096: 
dvander@4096: 	size_t written;
dvander@4096: 	const char *str = snapshot->strings.GetString(snapshot->keys[index]);
dvander@4096: 	pContext->StringToLocalUTF8(params[3], params[4], str, &written);
dvander@4096: 	return written;
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@4096: 	{"CreateTrieSnapshot",		CreateTrieSnapshot},
dvander@4096: 	{"TrieSnapshotLength",		TrieSnapshotLength},
dvander@4096: 	{"TrieSnapshotKeyBufferSize", TrieSnapshotKeyBufferSize},
dvander@4096: 	{"GetTrieSnapshotKey",		GetTrieSnapshotKey},
dvander@1779: 	{NULL,						NULL},
dvander@1779: };