dvander@2726: /**
dvander@2726: * vim: set ts=4 :
dvander@2726: * =============================================================================
dvander@2726: * SourceMod
dvander@2726: * Copyright (C) 2004-2008 AlliedModders LLC.  All rights reserved.
dvander@2726: * =============================================================================
dvander@2726: *
dvander@2726: * This program is free software; you can redistribute it and/or modify it under
dvander@2726: * the terms of the GNU General Public License, version 3.0, as published by the
dvander@2726: * Free Software Foundation.
dvander@2726: * 
dvander@2726: * This program is distributed in the hope that it will be useful, but WITHOUT
dvander@2726: * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
dvander@2726: * FOR A PARTICULAR PURPOSE.  See the GNU General Public License for more
dvander@2726: * details.
dvander@2726: *
dvander@2726: * You should have received a copy of the GNU General Public License along with
dvander@2726: * this program.  If not, see <http://www.gnu.org/licenses/>.
dvander@2726: *
dvander@2726: * As a special exception, AlliedModders LLC gives you permission to link the
dvander@2726: * code of this program (as well as its derivative works) to "Half-Life 2," the
dvander@2726: * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software
dvander@2726: * by the Valve Corporation.  You must obey the GNU General Public License in
dvander@2726: * all respects for all other code used.  Additionally, AlliedModders LLC grants
dvander@2726: * this exception to all derivative works.  AlliedModders LLC defines further
dvander@2726: * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),
dvander@2726: * or <http://www.sourcemod.net/license.php>.
dvander@2726: *
dvander@2726: * Version: $Id$
dvander@2726: */
dvander@2726: 
dvander@2726: #include <stdlib.h>
dvander@2726: #include "common_logic.h"
dvander@2726: #include "CellArray.h"
dvander@2726: 
dvander@2726: HandleType_t htCellStack;
dvander@2726: 
dvander@2726: class CellStackHelpers : 
dvander@2726: 	public SMGlobalClass,
dvander@2726: 	public IHandleTypeDispatch
dvander@2726: {
dvander@2726: public: //SMGlobalClass
dvander@2726: 	void OnSourceModAllInitialized()
dvander@2726: 	{
dvander@2726: 		htCellStack = handlesys->CreateType("CellStack", this, 0, NULL, NULL, g_pCoreIdent, NULL);
dvander@2726: 	}
dvander@2726: 	void OnSourceModShutdown()
dvander@2726: 	{
dvander@2726: 		handlesys->RemoveType(htCellStack, g_pCoreIdent);
dvander@2726: 	}
dvander@2726: public: //IHandleTypeDispatch
dvander@2726: 	void OnHandleDestroy(HandleType_t type, void *object)
dvander@2726: 	{
dvander@2726: 		CellArray *array = (CellArray *)object;
dvander@2726: 		delete array;
dvander@2726: 	}
dvander@2726: 	bool GetHandleApproxSize(HandleType_t type, void *object, unsigned int *pSize)
dvander@2726: 	{
dvander@2726: 		CellArray *pArray = (CellArray *)object;
dvander@2726: 		*pSize = sizeof(CellArray) + pArray->mem_usage();
dvander@2726: 		return true;
dvander@2726: 	}
dvander@2726: } s_CellStackHelpers;
dvander@2726: 
dvander@2726: static cell_t CreateStack(IPluginContext *pContext, const cell_t *params)
dvander@2726: {
dvander@2726: 	if (params[1] < 1)
dvander@2726: 	{
dvander@2726: 		return pContext->ThrowNativeError("Invalid block size (must be > 0)");
dvander@2726: 	}
dvander@2726: 
dvander@2726: 	CellArray *array = new CellArray(params[1]);
dvander@2726: 
dvander@2726: 	Handle_t hndl = handlesys->CreateHandle(htCellStack, array, pContext->GetIdentity(), g_pCoreIdent, NULL);
dvander@2726: 	if (!hndl)
dvander@2726: 	{
dvander@2726: 		delete array;
dvander@2726: 	}
dvander@2726: 
dvander@2726: 	return hndl;
dvander@2726: }
dvander@2726: 
dvander@2726: static cell_t PushStackCell(IPluginContext *pContext, const cell_t *params)
dvander@2726: {
dvander@2726: 	CellArray *array;
dvander@2726: 	HandleError err;
dvander@2726: 	HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@2726: 
dvander@2726: 	if ((err = handlesys->ReadHandle(params[1], htCellStack, &sec, (void **)&array)) 
dvander@2726: 		!= HandleError_None)
dvander@2726: 	{
dvander@2726: 		return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err);
dvander@2726: 	}
dvander@2726: 
dvander@2726: 	cell_t *blk = array->push();
dvander@2726: 	if (!blk)
dvander@2726: 	{
dvander@2726: 		return pContext->ThrowNativeError("Failed to grow stack");
dvander@2726: 	}
dvander@2726: 
dvander@2726: 	*blk = params[2];
dvander@2726: 
dvander@2726: 	return 1;
dvander@2726: }
dvander@2726: 
dvander@2726: static cell_t PushStackString(IPluginContext *pContext, const cell_t *params)
dvander@2726: {
dvander@2726: 	CellArray *array;
dvander@2726: 	HandleError err;
dvander@2726: 	HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@2726: 
dvander@2726: 	if ((err = handlesys->ReadHandle(params[1], htCellStack, &sec, (void **)&array)) 
dvander@2726: 		!= HandleError_None)
dvander@2726: 	{
dvander@2726: 		return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err);
dvander@2726: 	}
dvander@2726: 
dvander@2726: 	cell_t *blk = array->push();
dvander@2726: 	if (!blk)
dvander@2726: 	{
dvander@2726: 		return pContext->ThrowNativeError("Failed to grow stack");
dvander@2726: 	}
dvander@2726: 
dvander@2726: 	char *str;
dvander@2726: 	pContext->LocalToString(params[2], &str);
dvander@2726: 
dvander@2726: 	smcore.strncopy((char *)blk, str, array->blocksize() * sizeof(cell_t));
dvander@2726: 
dvander@2726: 	return 1;
dvander@2726: }
dvander@2726: 
dvander@2726: static cell_t PushStackArray(IPluginContext *pContext, const cell_t *params)
dvander@2726: {
dvander@2726: 	CellArray *array;
dvander@2726: 	HandleError err;
dvander@2726: 	HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@2726: 
dvander@2726: 	if ((err = handlesys->ReadHandle(params[1], htCellStack, &sec, (void **)&array)) 
dvander@2726: 		!= HandleError_None)
dvander@2726: 	{
dvander@2726: 		return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err);
dvander@2726: 	}
dvander@2726: 
dvander@2726: 	cell_t *blk = array->push();
dvander@2726: 	if (!blk)
dvander@2726: 	{
dvander@2726: 		return pContext->ThrowNativeError("Failed to grow array");
dvander@2726: 	}
dvander@2726: 
dvander@2726: 	cell_t *addr;
dvander@2726: 	pContext->LocalToPhysAddr(params[2], &addr);
dvander@2726: 
dvander@2726: 	size_t indexes = array->blocksize();
dvander@2726: 	if (params[3] != -1 && (size_t)params[3] <= array->blocksize())
dvander@2726: 	{
dvander@2726: 		indexes = params[3];
dvander@2726: 	}
dvander@2726: 
dvander@2726: 	memcpy(blk, addr, sizeof(cell_t) * indexes);
dvander@2726: 
dvander@2726: 	return 1;
dvander@2726: }
dvander@2726: 
dvander@2726: static cell_t PopStackCell(IPluginContext *pContext, const cell_t *params)
dvander@2726: {
dvander@2726: 	size_t idx;
dvander@2726: 	HandleError err;
dvander@2726: 	CellArray *array;
dvander@2726: 	cell_t *blk, *buffer;
dvander@2726: 	HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@2726: 
dvander@2726: 	if ((err = handlesys->ReadHandle(params[1], htCellStack, &sec, (void **)&array)) 
dvander@2726: 		!= HandleError_None)
dvander@2726: 	{
dvander@2726: 		return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err);
dvander@2726: 	}
dvander@2726: 
dvander@2726: 	if (array->size() == 0)
dvander@2726: 	{
dvander@2726: 		return 0;
dvander@2726: 	}
dvander@2726: 
dvander@2726: 	pContext->LocalToPhysAddr(params[2], &buffer);
dvander@2726: 
dvander@2726: 	blk = array->at(array->size() - 1);
dvander@2726: 	idx = (size_t)params[3];
dvander@2726: 
dvander@2726: 	if (params[4] == 0)
dvander@2726: 	{
dvander@2726: 		if (idx >= array->blocksize())
dvander@2726: 		{
dvander@2726: 			return pContext->ThrowNativeError("Invalid block %d (blocksize: %d)", idx, array->blocksize());
dvander@2726: 		}
dvander@2726: 		*buffer = blk[idx];
dvander@2726: 	}
dvander@2726: 	else
dvander@2726: 	{
dvander@2726: 		if (idx >= array->blocksize() * 4)
dvander@2726: 		{
dvander@2726: 			return pContext->ThrowNativeError("Invalid byte %d (blocksize: %d bytes)", idx, array->blocksize() * 4);
dvander@2726: 		}
dvander@2726: 		*buffer = (cell_t)*((char *)blk + idx);
dvander@2726: 	}
dvander@2726: 
dvander@2726: 	array->remove(array->size() - 1);
dvander@2726: 
dvander@2726: 	return 1;
dvander@2726: }
dvander@2726: 
dvander@2726: static cell_t PopStackString(IPluginContext *pContext, const cell_t *params)
dvander@2726: {
dvander@2726: 	HandleError err;
dvander@2726: 	CellArray *array;
dvander@2726: 	size_t idx, numWritten;
dvander@2726: 	cell_t *blk, *pWritten;
dvander@2726: 	HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@2726: 
dvander@2726: 	if ((err = handlesys->ReadHandle(params[1], htCellStack, &sec, (void **)&array)) 
dvander@2726: 		!= HandleError_None)
dvander@2726: 	{
dvander@2726: 		return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err);
dvander@2726: 	}
dvander@2726: 
dvander@2726: 	if (array->size() == 0)
dvander@2726: 	{
dvander@2726: 		return 0;
dvander@2726: 	}
dvander@2726: 
dvander@2726: 	idx = array->size() - 1;
dvander@2726: 	blk = array->at(idx);
dvander@2726: 	pContext->StringToLocalUTF8(params[2], params[3], (char *)blk, &numWritten);
dvander@2726: 	pContext->LocalToPhysAddr(params[4], &pWritten);
dvander@2726: 	*pWritten = (cell_t)numWritten;
dvander@2726: 	array->remove(idx);
dvander@2726: 
dvander@2726: 	return 1;
dvander@2726: }
dvander@2726: 
dvander@2726: static cell_t PopStackArray(IPluginContext *pContext, const cell_t *params)
dvander@2726: {
dvander@2726: 	size_t idx;
dvander@2726: 	cell_t *blk;
dvander@2726: 	cell_t *addr;
dvander@2726: 	size_t indexes;
dvander@2726: 	HandleError err;
dvander@2726: 	CellArray *array;
dvander@2726: 	HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@2726: 
dvander@2726: 	if ((err = handlesys->ReadHandle(params[1], htCellStack, &sec, (void **)&array)) 
dvander@2726: 		!= HandleError_None)
dvander@2726: 	{
dvander@2726: 		return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err);
dvander@2726: 	}
dvander@2726: 
dvander@2726: 	if (array->size() == 0)
dvander@2726: 	{
dvander@2726: 		return 0;
dvander@2726: 	}
dvander@2726: 
dvander@2726: 	idx = array->size() - 1;
dvander@2726: 	blk = array->at(idx);
dvander@2726: 	indexes = array->blocksize();
dvander@2726: 
dvander@2726: 	if (params[3] != -1 && (size_t)params[3] <= array->blocksize())
dvander@2726: 	{
dvander@2726: 		indexes = params[3];
dvander@2726: 	}
dvander@2726: 
dvander@2726: 	pContext->LocalToPhysAddr(params[2], &addr);
dvander@2726: 	memcpy(addr, blk, sizeof(cell_t) * indexes);
dvander@2726: 	array->remove(idx);
dvander@2726: 
dvander@2726: 	return indexes;
dvander@2726: }
dvander@2726: 
dvander@2726: static cell_t IsStackEmpty(IPluginContext *pContext, const cell_t *params)
dvander@2726: {
dvander@2726: 	HandleError err;
dvander@2726: 	CellArray *array;
dvander@2726: 	HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent);
dvander@2726: 
dvander@2726: 	if ((err = handlesys->ReadHandle(params[1], htCellStack, &sec, (void **)&array)) 
dvander@2726: 		!= HandleError_None)
dvander@2726: 	{
dvander@2726: 		return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err);
dvander@2726: 	}
dvander@2726: 
dvander@2726: 	if (array->size() == 0)
dvander@2726: 	{
dvander@2726: 		return 1;
dvander@2726: 	}
dvander@2726: 
dvander@2726: 	return 0;
dvander@2726: }
dvander@2726: 
dvander@2726: REGISTER_NATIVES(cellStackNatives)
dvander@2726: {
dvander@2726: 	{"CreateStack",					CreateStack},
dvander@2726: 	{"IsStackEmpty",				IsStackEmpty},
dvander@2726: 	{"PopStackArray",				PopStackArray},
dvander@2726: 	{"PopStackCell",				PopStackCell},
dvander@2726: 	{"PopStackString",				PopStackString},
dvander@2726: 	{"PushStackArray",				PushStackArray},
dvander@2726: 	{"PushStackCell",				PushStackCell},
dvander@2726: 	{"PushStackString",				PushStackString},
dvander@2726: 	{NULL,							NULL},
dvander@2726: };