dvander@1156: /**
dvander@1156:  * vim: set ts=4 :
dvander@1156:  * ===============================================================
dvander@1156:  * SourceMod (C)2004-2007 AlliedModders LLC.  All rights reserved.
dvander@1156:  * ===============================================================
dvander@1156:  *
dvander@1156:  * This file is not open source and may not be copied without explicit
dvander@1156:  * written permission of AlliedModders LLC.  This file may not be redistributed 
dvander@1156:  * in whole or significant part.
dvander@1156:  * For information, see LICENSE.txt or http://www.sourcemod.net/license.php
dvander@1156:  *
dvander@1156:  * Version: $Id$
dvander@1156:  */
dvander@1156: 
dvander@1156: #include <malloc.h>
dvander@1156: #include "sm_globals.h"
dvander@1156: #include "sm_stringutil.h"
dvander@1156: #include "CellArray.h"
dvander@1156: #include "HandleSys.h"
dvander@1156: 
dvander@1156: HandleType_t htCellArray;
dvander@1156: 
dvander@1156: class CellArrayHelpers : 
dvander@1156: 	public SMGlobalClass,
dvander@1156: 	public IHandleTypeDispatch
dvander@1156: {
dvander@1156: public: //SMGlobalClass
dvander@1156: 	void OnSourceModAllInitialized()
dvander@1156: 	{
dvander@1156: 		htCellArray = g_HandleSys.CreateType("CellArray", this, 0, NULL, NULL, g_pCoreIdent, NULL);
dvander@1156: 	}
dvander@1156: 	void OnSourceModShutdown()
dvander@1156: 	{
dvander@1156: 		g_HandleSys.RemoveType(htCellArray, g_pCoreIdent);
dvander@1156: 	}
dvander@1156: public: //IHandleTypeDispatch
dvander@1156: 	void OnHandleDestroy(HandleType_t type, void *object)
dvander@1156: 	{
dvander@1156: 		CellArray *array = (CellArray *)object;
dvander@1156: 		delete array;
dvander@1156: 	}
dvander@1156: } s_CellArrayHelpers;
dvander@1156: 
dvander@1156: static cell_t CreateArray(IPluginContext *pContext, const cell_t *params)
dvander@1156: {
dvander@1156: 	if (!params[1])
dvander@1156: 	{
dvander@1156: 		return pContext->ThrowNativeError("Invalid block size (must be > 0)");
dvander@1156: 	}
dvander@1156: 
dvander@1156: 	CellArray *array = new CellArray(params[1]);
dvander@1156: 
dvander@1156: 	if (params[2])
dvander@1156: 	{
dvander@1156: 		array->resize(params[2]);
dvander@1156: 	}
dvander@1156: 	
dvander@1156: 	Handle_t hndl = g_HandleSys.CreateHandle(htCellArray, array, pContext->GetIdentity(), g_pCoreIdent, NULL);
dvander@1156: 	if (!hndl)
dvander@1156: 	{
dvander@1156: 		delete array;
dvander@1156: 	}
dvander@1156: 
dvander@1156: 	return hndl;
dvander@1156: }
dvander@1156: 
dvander@1156: static cell_t ClearArray(IPluginContext *pContext, const cell_t *params)
dvander@1156: {
dvander@1156: 	CellArray *array;
dvander@1156: 	HandleError err;
dvander@1156: 	HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@1156: 
dvander@1156: 	if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array)) 
dvander@1156: 		!= HandleError_None)
dvander@1156: 	{
dvander@1156: 		return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err);
dvander@1156: 	}
dvander@1156: 
dvander@1156: 	array->clear();
dvander@1156: 	
dvander@1156: 	return 1;
dvander@1156: }
dvander@1156: 
dvander@1156: static cell_t ResizeArray(IPluginContext *pContext, const cell_t *params)
dvander@1156: {
dvander@1156: 	CellArray *array;
dvander@1156: 	HandleError err;
dvander@1156: 	HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@1156: 
dvander@1156: 	if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array)) 
dvander@1156: 		!= HandleError_None)
dvander@1156: 	{
dvander@1156: 		return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err);
dvander@1156: 	}
dvander@1156: 
dvander@1156: 	if (!array->resize(params[2]))
dvander@1156: 	{
dvander@1156: 		return pContext->ThrowNativeError("Unable to resize array to \"%u\"", params[2]);
dvander@1156: 	}
dvander@1156: 
dvander@1156: 	return 1;
dvander@1156: }
dvander@1156: 
dvander@1156: static cell_t GetArraySize(IPluginContext *pContext, const cell_t *params)
dvander@1156: {
dvander@1156: 	CellArray *array;
dvander@1156: 	HandleError err;
dvander@1156: 	HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@1156: 
dvander@1156: 	if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array)) 
dvander@1156: 		!= HandleError_None)
dvander@1156: 	{
dvander@1156: 		return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err);
dvander@1156: 	}
dvander@1156: 
dvander@1156: 	return (cell_t)array->size();
dvander@1156: }
dvander@1156: 
dvander@1156: static cell_t PushArrayCell(IPluginContext *pContext, const cell_t *params)
dvander@1156: {
dvander@1156: 	CellArray *array;
dvander@1156: 	HandleError err;
dvander@1156: 	HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@1156: 
dvander@1156: 	if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array)) 
dvander@1156: 		!= HandleError_None)
dvander@1156: 	{
dvander@1156: 		return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err);
dvander@1156: 	}
dvander@1156: 
dvander@1156: 	cell_t *blk = array->push();
dvander@1156: 	if (!blk)
dvander@1156: 	{
dvander@1156: 		return pContext->ThrowNativeError("Failed to grow array");
dvander@1156: 	}
dvander@1156: 
dvander@1156: 	*blk = params[2];
dvander@1156: 
dvander@1156: 	return (cell_t)(array->size() - 1);
dvander@1156: }
dvander@1156: 
dvander@1156: static cell_t PushArrayString(IPluginContext *pContext, const cell_t *params)
dvander@1156: {
dvander@1156: 	CellArray *array;
dvander@1156: 	HandleError err;
dvander@1156: 	HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@1156: 
dvander@1156: 	if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array)) 
dvander@1156: 		!= HandleError_None)
dvander@1156: 	{
dvander@1156: 		return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err);
dvander@1156: 	}
dvander@1156: 
dvander@1156: 	cell_t *blk = array->push();
dvander@1156: 	if (!blk)
dvander@1156: 	{
dvander@1156: 		return pContext->ThrowNativeError("Failed to grow array");
dvander@1156: 	}
dvander@1156: 
dvander@1156: 	char *str;
dvander@1156: 	pContext->LocalToString(params[2], &str);
dvander@1156: 
dvander@1156: 	strncopy((char *)blk, str, array->blocksize() * sizeof(cell_t));
dvander@1156: 
dvander@1156: 	return (cell_t)(array->size() - 1);
dvander@1156: }
dvander@1156: 
dvander@1156: static cell_t PushArrayArray(IPluginContext *pContext, const cell_t *params)
dvander@1156: {
dvander@1156: 	CellArray *array;
dvander@1156: 	HandleError err;
dvander@1156: 	HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@1156: 
dvander@1156: 	if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array)) 
dvander@1156: 		!= HandleError_None)
dvander@1156: 	{
dvander@1156: 		return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err);
dvander@1156: 	}
dvander@1156: 
dvander@1156: 	cell_t *blk = array->push();
dvander@1156: 	if (!blk)
dvander@1156: 	{
dvander@1156: 		return pContext->ThrowNativeError("Failed to grow array");
dvander@1156: 	}
dvander@1156: 
dvander@1156: 	cell_t *addr;
dvander@1156: 	pContext->LocalToPhysAddr(params[2], &addr);
dvander@1156: 
dvander@1156: 	size_t indexes = array->blocksize();
dvander@1156: 	if (params[3] != -1 && (size_t)params[3] <= array->blocksize())
dvander@1156: 	{
dvander@1156: 		indexes = params[3];
dvander@1156: 	}
dvander@1156: 
dvander@1156: 	memcpy(blk, addr, sizeof(cell_t) * indexes);
dvander@1156: 
dvander@1156: 	return (cell_t)(array->size() - 1);
dvander@1156: }
dvander@1156: 
dvander@1156: static cell_t GetArrayCell(IPluginContext *pContext, const cell_t *params)
dvander@1156: {
dvander@1156: 	CellArray *array;
dvander@1156: 	HandleError err;
dvander@1156: 	HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@1156: 
dvander@1156: 	if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array)) 
dvander@1156: 		!= HandleError_None)
dvander@1156: 	{
dvander@1156: 		return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err);
dvander@1156: 	}
dvander@1156: 
dvander@1156: 	size_t idx = (size_t)params[2];
dvander@1156: 	if (idx >= array->size())
dvander@1156: 	{
dvander@1156: 		return pContext->ThrowNativeError("Invalid index %d (count: %d)", idx, array->size());
dvander@1156: 	}
dvander@1156: 
dvander@1156: 	cell_t *blk = array->at(idx);
dvander@1156: 
dvander@1156: 	idx = (size_t)params[3];
dvander@1156: 	if (params[4] == 1)
dvander@1156: 	{
dvander@1156: 		if (idx >= array->blocksize())
dvander@1156: 		{
dvander@1156: 			return pContext->ThrowNativeError("Invalid block %d (blocksize: %d)", idx, array->blocksize());
dvander@1156: 		}
dvander@1156: 		return blk[idx];
dvander@1156: 	} else {
dvander@1156: 		if (idx >= array->blocksize() * 4)
dvander@1156: 		{
dvander@1156: 			return pContext->ThrowNativeError("Invalid byte %d (blocksize: %d bytes)", idx, array->blocksize() * 4);
dvander@1156: 		}
dvander@1156: 		return (cell_t)*((char *)blk + idx);
dvander@1156: 	}
dvander@1156: 
dvander@1156: 	return 0;
dvander@1156: }
dvander@1156: 
dvander@1156: static cell_t GetArrayString(IPluginContext *pContext, const cell_t *params)
dvander@1156: {
dvander@1156: 	CellArray *array;
dvander@1156: 	HandleError err;
dvander@1156: 	HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@1156: 
dvander@1156: 	if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array)) 
dvander@1156: 		!= HandleError_None)
dvander@1156: 	{
dvander@1156: 		return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err);
dvander@1156: 	}
dvander@1156: 
dvander@1156: 	size_t idx = (size_t)params[2];
dvander@1156: 	if (idx >= array->size())
dvander@1156: 	{
dvander@1156: 		return pContext->ThrowNativeError("Invalid index %d (count: %d)", idx, array->size());
dvander@1156: 	}
dvander@1156: 
dvander@1156: 	cell_t *blk = array->at(idx);
dvander@1156: 	size_t numWritten = 0;
dvander@1156: 
dvander@1156: 	pContext->StringToLocalUTF8(params[3], params[4], (char *)blk, &numWritten);
dvander@1156: 
dvander@1156: 	return numWritten;
dvander@1156: }
dvander@1156: 
dvander@1156: static cell_t GetArrayArray(IPluginContext *pContext, const cell_t *params)
dvander@1156: {
dvander@1156: 	CellArray *array;
dvander@1156: 	HandleError err;
dvander@1156: 	HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@1156: 
dvander@1156: 	if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array)) 
dvander@1156: 		!= HandleError_None)
dvander@1156: 	{
dvander@1156: 		return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err);
dvander@1156: 	}
dvander@1156: 
dvander@1156: 	size_t idx = (size_t)params[2];
dvander@1156: 	if (idx >= array->size())
dvander@1156: 	{
dvander@1156: 		return pContext->ThrowNativeError("Invalid index %d (count: %d)", idx, array->size());
dvander@1156: 	}
dvander@1156: 
dvander@1156: 	cell_t *blk = array->at(idx);
dvander@1156: 	size_t indexes = array->blocksize();
dvander@1156: 	if (params[4] != -1 && (size_t)params[4] <= array->blocksize())
dvander@1156: 	{
dvander@1156: 		indexes = params[4];
dvander@1156: 	}
dvander@1156: 
dvander@1156: 	cell_t *addr;
dvander@1156: 	pContext->LocalToPhysAddr(params[3], &addr);
dvander@1156: 
dvander@1156: 	memcpy(addr, blk, sizeof(cell_t) * indexes);
dvander@1156: 
dvander@1156: 	return indexes;
dvander@1156: }
dvander@1156: 
dvander@1156: static cell_t SetArrayCell(IPluginContext *pContext, const cell_t *params)
dvander@1156: {
dvander@1156: 	CellArray *array;
dvander@1156: 	HandleError err;
dvander@1156: 	HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@1156: 
dvander@1156: 	if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array)) 
dvander@1156: 		!= HandleError_None)
dvander@1156: 	{
dvander@1156: 		return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err);
dvander@1156: 	}
dvander@1156: 
dvander@1156: 	size_t idx = (size_t)params[2];
dvander@1156: 	if (idx >= array->size())
dvander@1156: 	{
dvander@1156: 		return pContext->ThrowNativeError("Invalid index %d (count: %d)", idx, array->size());
dvander@1156: 	}
dvander@1156: 
dvander@1156: 	cell_t *blk = array->at(idx);
dvander@1156: 
dvander@1156: 	idx = (size_t)params[4];
dvander@1156: 	if (params[5] == 1)
dvander@1156: 	{
dvander@1156: 		if (idx >= array->blocksize())
dvander@1156: 		{
dvander@1156: 			return pContext->ThrowNativeError("Invalid block %d (blocksize: %d)", idx, array->blocksize());
dvander@1156: 		}
dvander@1156: 		blk[idx] = params[3];
dvander@1156: 	} else {
dvander@1156: 		if (idx >= array->blocksize() * 4)
dvander@1156: 		{
dvander@1156: 			return pContext->ThrowNativeError("Invalid byte %d (blocksize: %d bytes)", idx, array->blocksize() * 4);
dvander@1156: 		}
dvander@1156: 		*((char *)blk + idx) = (char)params[3];
dvander@1156: 	}
dvander@1156: 
dvander@1156: 	return 1;
dvander@1156: }
dvander@1156: 
dvander@1156: static cell_t SetArrayString(IPluginContext *pContext, const cell_t *params)
dvander@1156: {
dvander@1156: 	CellArray *array;
dvander@1156: 	HandleError err;
dvander@1156: 	HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@1156: 
dvander@1156: 	if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array)) 
dvander@1156: 		!= HandleError_None)
dvander@1156: 	{
dvander@1156: 		return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err);
dvander@1156: 	}
dvander@1156: 
dvander@1156: 	size_t idx = (size_t)params[2];
dvander@1156: 	if (idx >= array->size())
dvander@1156: 	{
dvander@1156: 		return pContext->ThrowNativeError("Invalid index %d (count: %d)", idx, array->size());
dvander@1156: 	}
dvander@1156: 
dvander@1156: 	cell_t *blk = array->at(idx);
dvander@1156: 
dvander@1156: 	char *str;
dvander@1156: 	pContext->LocalToString(params[3], &str);
dvander@1156: 
dvander@1156: 	return strncopy((char *)blk, str, array->blocksize() * sizeof(cell_t));
dvander@1156: }
dvander@1156: 
dvander@1156: static cell_t SetArrayArray(IPluginContext *pContext, const cell_t *params)
dvander@1156: {
dvander@1156: 	CellArray *array;
dvander@1156: 	HandleError err;
dvander@1156: 	HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@1156: 
dvander@1156: 	if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array)) 
dvander@1156: 		!= HandleError_None)
dvander@1156: 	{
dvander@1156: 		return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err);
dvander@1156: 	}
dvander@1156: 
dvander@1156: 	size_t idx = (size_t)params[2];
dvander@1156: 	if (idx >= array->size())
dvander@1156: 	{
dvander@1156: 		return pContext->ThrowNativeError("Invalid index %d (count: %d)", idx, array->size());
dvander@1156: 	}
dvander@1156: 
dvander@1156: 	cell_t *blk = array->at(idx);
dvander@1156: 	size_t indexes = array->blocksize();
dvander@1156: 	if (params[4] != -1 && (size_t)params[4] <= array->blocksize())
dvander@1156: 	{
dvander@1156: 		indexes = params[4];
dvander@1156: 	}
dvander@1156: 
dvander@1156: 	cell_t *addr;
dvander@1156: 	pContext->LocalToPhysAddr(params[3], &addr);
dvander@1156: 
dvander@1156: 	memcpy(blk, addr, sizeof(cell_t) * indexes);
dvander@1156: 
dvander@1156: 	return indexes;
dvander@1156: }
dvander@1156: 
dvander@1156: static cell_t ShiftArrayUp(IPluginContext *pContext, const cell_t *params)
dvander@1156: {
dvander@1156: 	CellArray *array;
dvander@1156: 	HandleError err;
dvander@1156: 	HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@1156: 
dvander@1156: 	if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array)) 
dvander@1156: 		!= HandleError_None)
dvander@1156: 	{
dvander@1156: 		return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err);
dvander@1156: 	}
dvander@1156: 
dvander@1156: 	size_t idx = (size_t)params[2];
dvander@1156: 	if (idx >= array->size())
dvander@1156: 	{
dvander@1156: 		return pContext->ThrowNativeError("Invalid index %d (count: %d)", idx, array->size());
dvander@1156: 	}
dvander@1156: 
dvander@1156: 	array->insert_at(idx);
dvander@1156: 
dvander@1156: 	return 1;
dvander@1156: }
dvander@1156: 
dvander@1156: static cell_t RemoveFromArray(IPluginContext *pContext, const cell_t *params)
dvander@1156: {
dvander@1156: 	CellArray *array;
dvander@1156: 	HandleError err;
dvander@1156: 	HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@1156: 
dvander@1156: 	if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array)) 
dvander@1156: 		!= HandleError_None)
dvander@1156: 	{
dvander@1156: 		return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err);
dvander@1156: 	}
dvander@1156: 
dvander@1156: 	size_t idx = (size_t)params[2];
dvander@1156: 	if (idx >= array->size())
dvander@1156: 	{
dvander@1156: 		return pContext->ThrowNativeError("Invalid index %d (count: %d)", idx, array->size());
dvander@1156: 	}
dvander@1156: 
dvander@1156: 	array->remove(idx);
dvander@1156: 
dvander@1156: 	return 1;
dvander@1156: }
dvander@1156: 
dvander@1156: static cell_t SwapArrayItems(IPluginContext *pContext, const cell_t *params)
dvander@1156: {
dvander@1156: 	CellArray *array;
dvander@1156: 	HandleError err;
dvander@1156: 	HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@1156: 
dvander@1156: 	if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array)) 
dvander@1156: 		!= HandleError_None)
dvander@1156: 	{
dvander@1156: 		return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err);
dvander@1156: 	}
dvander@1156: 
dvander@1156: 	size_t idx1 = (size_t)params[2];
dvander@1156: 	size_t idx2 = (size_t)params[3];
dvander@1156: 	if (idx1 >= array->size())
dvander@1156: 	{
dvander@1156: 		return pContext->ThrowNativeError("Invalid index %d (count: %d)", idx1, array->size());
dvander@1156: 	}
dvander@1156: 	if (idx2 >= array->size())
dvander@1156: 	{
dvander@1156: 		return pContext->ThrowNativeError("Invalid index %d (count: %d)", idx2, array->size());
dvander@1156: 	}
dvander@1156: 
dvander@1156: 	array->swap(idx1, idx2);
dvander@1156: 
dvander@1156: 	return 1;
dvander@1156: }
dvander@1156: 
dvander@1156: REGISTER_NATIVES(cellArrayNatives)
dvander@1156: {
dvander@1156: 	{"ClearArray",					ClearArray},
dvander@1156: 	{"CreateArray",					CreateArray},
dvander@1156: 	{"GetArrayArray",				GetArrayArray},
dvander@1156: 	{"GetArrayCell",				GetArrayCell},
dvander@1156: 	{"GetArraySize",				GetArraySize},
dvander@1156: 	{"GetArrayString",				GetArrayString},
dvander@1156: 	{"ResizeArray",					ResizeArray},
dvander@1156: 	{"PushArrayArray",				PushArrayArray},
dvander@1156: 	{"PushArrayCell",				PushArrayCell},
dvander@1156: 	{"PushArrayString",				PushArrayString},
dvander@1156: 	{"RemoveFromArray",				RemoveFromArray},
dvander@1156: 	{"SetArrayCell",				SetArrayCell},
dvander@1156: 	{"SetArrayString",				SetArrayString},
dvander@1156: 	{"SetArrayArray",				SetArrayArray},
dvander@1156: 	{"ShiftArrayUp",				ShiftArrayUp},
dvander@1156: 	{"SwapArrayItems",				SwapArrayItems},
dvander@1156: 	{NULL,							NULL},
dvander@1156: };