ds@1952: /** ds@1952: * vim: set ts=4 : ds@1952: * ============================================================================= ds@1952: * SourceMod ds@1952: * Copyright (C) 2004-2008 AlliedModders LLC. All rights reserved. ds@1952: * ============================================================================= ds@1952: * ds@1952: * This program is free software; you can redistribute it and/or modify it under ds@1952: * the terms of the GNU General Public License, version 3.0, as published by the ds@1952: * Free Software Foundation. ds@1952: * ds@1952: * This program is distributed in the hope that it will be useful, but WITHOUT ds@1952: * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS ds@1952: * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more ds@1952: * details. ds@1952: * ds@1952: * You should have received a copy of the GNU General Public License along with ds@1952: * this program. If not, see . ds@1952: * ds@1952: * As a special exception, AlliedModders LLC gives you permission to link the ds@1952: * code of this program (as well as its derivative works) to "Half-Life 2," the ds@1952: * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software ds@1952: * by the Valve Corporation. You must obey the GNU General Public License in ds@1952: * all respects for all other code used. Additionally, AlliedModders LLC grants ds@1952: * this exception to all derivative works. AlliedModders LLC defines further ds@1952: * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007), ds@1952: * or . ds@1952: * ds@1952: * Version: $Id$ ds@1952: */ ds@1952: ds@1952: #include ds@1952: #include "sm_globals.h" ds@1952: #include "sm_stringutil.h" ds@1952: #include "CellArray.h" ds@1952: #include "HandleSys.h" ds@1952: ds@1952: HandleType_t htCellArray; ds@1952: ds@1952: class CellArrayHelpers : ds@1952: public SMGlobalClass, ds@1952: public IHandleTypeDispatch ds@1952: { ds@1952: public: //SMGlobalClass ds@1952: void OnSourceModAllInitialized() ds@1952: { ds@1952: htCellArray = g_HandleSys.CreateType("CellArray", this, 0, NULL, NULL, g_pCoreIdent, NULL); ds@1952: } ds@1952: void OnSourceModShutdown() ds@1952: { ds@1952: g_HandleSys.RemoveType(htCellArray, g_pCoreIdent); ds@1952: } ds@1952: public: //IHandleTypeDispatch ds@1952: void OnHandleDestroy(HandleType_t type, void *object) ds@1952: { ds@1952: CellArray *array = (CellArray *)object; ds@1952: delete array; ds@1952: } ds@1952: bool GetHandleApproxSize(HandleType_t type, void *object, unsigned int *pSize) ds@1952: { ds@1952: CellArray *pArray = (CellArray *)object; ds@1952: *pSize = sizeof(CellArray) + pArray->mem_usage(); ds@1952: return true; ds@1952: } ds@1952: } s_CellArrayHelpers; ds@1952: ds@1952: static cell_t CreateArray(IPluginContext *pContext, const cell_t *params) ds@1952: { ds@1952: if (!params[1]) ds@1952: { ds@1952: return pContext->ThrowNativeError("Invalid block size (must be > 0)"); ds@1952: } ds@1952: ds@1952: CellArray *array = new CellArray(params[1]); ds@1952: ds@1952: if (params[2]) ds@1952: { ds@1952: array->resize(params[2]); ds@1952: } ds@1952: ds@1952: Handle_t hndl = g_HandleSys.CreateHandle(htCellArray, array, pContext->GetIdentity(), g_pCoreIdent, NULL); ds@1952: if (!hndl) ds@1952: { ds@1952: delete array; ds@1952: } ds@1952: ds@1952: return hndl; ds@1952: } ds@1952: ds@1952: static cell_t ClearArray(IPluginContext *pContext, const cell_t *params) ds@1952: { ds@1952: CellArray *array; ds@1952: HandleError err; ds@1952: HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent); ds@1952: ds@1952: if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array)) ds@1952: != HandleError_None) ds@1952: { ds@1952: return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err); ds@1952: } ds@1952: ds@1952: array->clear(); ds@1952: ds@1952: return 1; ds@1952: } ds@1952: ds@1952: static cell_t ResizeArray(IPluginContext *pContext, const cell_t *params) ds@1952: { ds@1952: CellArray *array; ds@1952: HandleError err; ds@1952: HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent); ds@1952: ds@1952: if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array)) ds@1952: != HandleError_None) ds@1952: { ds@1952: return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err); ds@1952: } ds@1952: ds@1952: if (!array->resize(params[2])) ds@1952: { ds@1952: return pContext->ThrowNativeError("Unable to resize array to \"%u\"", params[2]); ds@1952: } ds@1952: ds@1952: return 1; ds@1952: } ds@1952: ds@1952: static cell_t GetArraySize(IPluginContext *pContext, const cell_t *params) ds@1952: { ds@1952: CellArray *array; ds@1952: HandleError err; ds@1952: HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent); ds@1952: ds@1952: if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array)) ds@1952: != HandleError_None) ds@1952: { ds@1952: return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err); ds@1952: } ds@1952: ds@1952: return (cell_t)array->size(); ds@1952: } ds@1952: ds@1952: static cell_t PushArrayCell(IPluginContext *pContext, const cell_t *params) ds@1952: { ds@1952: CellArray *array; ds@1952: HandleError err; ds@1952: HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent); ds@1952: ds@1952: if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array)) ds@1952: != HandleError_None) ds@1952: { ds@1952: return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err); ds@1952: } ds@1952: ds@1952: cell_t *blk = array->push(); ds@1952: if (!blk) ds@1952: { ds@1952: return pContext->ThrowNativeError("Failed to grow array"); ds@1952: } ds@1952: ds@1952: *blk = params[2]; ds@1952: ds@1952: return (cell_t)(array->size() - 1); ds@1952: } ds@1952: ds@1952: static cell_t PushArrayString(IPluginContext *pContext, const cell_t *params) ds@1952: { ds@1952: CellArray *array; ds@1952: HandleError err; ds@1952: HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent); ds@1952: ds@1952: if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array)) ds@1952: != HandleError_None) ds@1952: { ds@1952: return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err); ds@1952: } ds@1952: ds@1952: cell_t *blk = array->push(); ds@1952: if (!blk) ds@1952: { ds@1952: return pContext->ThrowNativeError("Failed to grow array"); ds@1952: } ds@1952: ds@1952: char *str; ds@1952: pContext->LocalToString(params[2], &str); ds@1952: ds@1952: strncopy((char *)blk, str, array->blocksize() * sizeof(cell_t)); ds@1952: ds@1952: return (cell_t)(array->size() - 1); ds@1952: } ds@1952: ds@1952: static cell_t PushArrayArray(IPluginContext *pContext, const cell_t *params) ds@1952: { ds@1952: CellArray *array; ds@1952: HandleError err; ds@1952: HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent); ds@1952: ds@1952: if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array)) ds@1952: != HandleError_None) ds@1952: { ds@1952: return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err); ds@1952: } ds@1952: ds@1952: cell_t *blk = array->push(); ds@1952: if (!blk) ds@1952: { ds@1952: return pContext->ThrowNativeError("Failed to grow array"); ds@1952: } ds@1952: ds@1952: cell_t *addr; ds@1952: pContext->LocalToPhysAddr(params[2], &addr); ds@1952: ds@1952: size_t indexes = array->blocksize(); ds@1952: if (params[3] != -1 && (size_t)params[3] <= array->blocksize()) ds@1952: { ds@1952: indexes = params[3]; ds@1952: } ds@1952: ds@1952: memcpy(blk, addr, sizeof(cell_t) * indexes); ds@1952: ds@1952: return (cell_t)(array->size() - 1); ds@1952: } ds@1952: ds@1952: static cell_t GetArrayCell(IPluginContext *pContext, const cell_t *params) ds@1952: { ds@1952: CellArray *array; ds@1952: HandleError err; ds@1952: HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent); ds@1952: ds@1952: if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array)) ds@1952: != HandleError_None) ds@1952: { ds@1952: return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err); ds@1952: } ds@1952: ds@1952: size_t idx = (size_t)params[2]; ds@1952: if (idx >= array->size()) ds@1952: { ds@1952: return pContext->ThrowNativeError("Invalid index %d (count: %d)", idx, array->size()); ds@1952: } ds@1952: ds@1952: cell_t *blk = array->at(idx); ds@1952: ds@1952: idx = (size_t)params[3]; ds@1952: if (params[4] == 0) ds@1952: { ds@1952: if (idx >= array->blocksize()) ds@1952: { ds@1952: return pContext->ThrowNativeError("Invalid block %d (blocksize: %d)", idx, array->blocksize()); ds@1952: } ds@1952: return blk[idx]; ds@1952: } else { ds@1952: if (idx >= array->blocksize() * 4) ds@1952: { ds@1952: return pContext->ThrowNativeError("Invalid byte %d (blocksize: %d bytes)", idx, array->blocksize() * 4); ds@1952: } ds@1952: return (cell_t)*((char *)blk + idx); ds@1952: } ds@1952: ds@1952: return 0; ds@1952: } ds@1952: ds@1952: static cell_t GetArrayString(IPluginContext *pContext, const cell_t *params) ds@1952: { ds@1952: CellArray *array; ds@1952: HandleError err; ds@1952: HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent); ds@1952: ds@1952: if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array)) ds@1952: != HandleError_None) ds@1952: { ds@1952: return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err); ds@1952: } ds@1952: ds@1952: size_t idx = (size_t)params[2]; ds@1952: if (idx >= array->size()) ds@1952: { ds@1952: return pContext->ThrowNativeError("Invalid index %d (count: %d)", idx, array->size()); ds@1952: } ds@1952: ds@1952: cell_t *blk = array->at(idx); ds@1952: size_t numWritten = 0; ds@1952: ds@1952: pContext->StringToLocalUTF8(params[3], params[4], (char *)blk, &numWritten); ds@1952: ds@1952: return numWritten; ds@1952: } ds@1952: ds@1952: static cell_t GetArrayArray(IPluginContext *pContext, const cell_t *params) ds@1952: { ds@1952: CellArray *array; ds@1952: HandleError err; ds@1952: HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent); ds@1952: ds@1952: if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array)) ds@1952: != HandleError_None) ds@1952: { ds@1952: return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err); ds@1952: } ds@1952: ds@1952: size_t idx = (size_t)params[2]; ds@1952: if (idx >= array->size()) ds@1952: { ds@1952: return pContext->ThrowNativeError("Invalid index %d (count: %d)", idx, array->size()); ds@1952: } ds@1952: ds@1952: cell_t *blk = array->at(idx); ds@1952: size_t indexes = array->blocksize(); ds@1952: if (params[4] != -1 && (size_t)params[4] <= array->blocksize()) ds@1952: { ds@1952: indexes = params[4]; ds@1952: } ds@1952: ds@1952: cell_t *addr; ds@1952: pContext->LocalToPhysAddr(params[3], &addr); ds@1952: ds@1952: memcpy(addr, blk, sizeof(cell_t) * indexes); ds@1952: ds@1952: return indexes; ds@1952: } ds@1952: ds@1952: static cell_t SetArrayCell(IPluginContext *pContext, const cell_t *params) ds@1952: { ds@1952: CellArray *array; ds@1952: HandleError err; ds@1952: HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent); ds@1952: ds@1952: if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array)) ds@1952: != HandleError_None) ds@1952: { ds@1952: return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err); ds@1952: } ds@1952: ds@1952: size_t idx = (size_t)params[2]; ds@1952: if (idx >= array->size()) ds@1952: { ds@1952: return pContext->ThrowNativeError("Invalid index %d (count: %d)", idx, array->size()); ds@1952: } ds@1952: ds@1952: cell_t *blk = array->at(idx); ds@1952: ds@1952: idx = (size_t)params[4]; ds@1952: if (params[5] == 0) ds@1952: { ds@1952: if (idx >= array->blocksize()) ds@1952: { ds@1952: return pContext->ThrowNativeError("Invalid block %d (blocksize: %d)", idx, array->blocksize()); ds@1952: } ds@1952: blk[idx] = params[3]; ds@1952: } else { ds@1952: if (idx >= array->blocksize() * 4) ds@1952: { ds@1952: return pContext->ThrowNativeError("Invalid byte %d (blocksize: %d bytes)", idx, array->blocksize() * 4); ds@1952: } ds@1952: *((char *)blk + idx) = (char)params[3]; ds@1952: } ds@1952: ds@1952: return 1; ds@1952: } ds@1952: ds@1952: static cell_t SetArrayString(IPluginContext *pContext, const cell_t *params) ds@1952: { ds@1952: CellArray *array; ds@1952: HandleError err; ds@1952: HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent); ds@1952: ds@1952: if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array)) ds@1952: != HandleError_None) ds@1952: { ds@1952: return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err); ds@1952: } ds@1952: ds@1952: size_t idx = (size_t)params[2]; ds@1952: if (idx >= array->size()) ds@1952: { ds@1952: return pContext->ThrowNativeError("Invalid index %d (count: %d)", idx, array->size()); ds@1952: } ds@1952: ds@1952: cell_t *blk = array->at(idx); ds@1952: ds@1952: char *str; ds@1952: pContext->LocalToString(params[3], &str); ds@1952: ds@1952: return strncopy((char *)blk, str, array->blocksize() * sizeof(cell_t)); ds@1952: } ds@1952: ds@1952: static cell_t SetArrayArray(IPluginContext *pContext, const cell_t *params) ds@1952: { ds@1952: CellArray *array; ds@1952: HandleError err; ds@1952: HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent); ds@1952: ds@1952: if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array)) ds@1952: != HandleError_None) ds@1952: { ds@1952: return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err); ds@1952: } ds@1952: ds@1952: size_t idx = (size_t)params[2]; ds@1952: if (idx >= array->size()) ds@1952: { ds@1952: return pContext->ThrowNativeError("Invalid index %d (count: %d)", idx, array->size()); ds@1952: } ds@1952: ds@1952: cell_t *blk = array->at(idx); ds@1952: size_t indexes = array->blocksize(); ds@1952: if (params[4] != -1 && (size_t)params[4] <= array->blocksize()) ds@1952: { ds@1952: indexes = params[4]; ds@1952: } ds@1952: ds@1952: cell_t *addr; ds@1952: pContext->LocalToPhysAddr(params[3], &addr); ds@1952: ds@1952: memcpy(blk, addr, sizeof(cell_t) * indexes); ds@1952: ds@1952: return indexes; ds@1952: } ds@1952: ds@1952: static cell_t ShiftArrayUp(IPluginContext *pContext, const cell_t *params) ds@1952: { ds@1952: CellArray *array; ds@1952: HandleError err; ds@1952: HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent); ds@1952: ds@1952: if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array)) ds@1952: != HandleError_None) ds@1952: { ds@1952: return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err); ds@1952: } ds@1952: ds@1952: size_t idx = (size_t)params[2]; ds@1952: if (idx >= array->size()) ds@1952: { ds@1952: return pContext->ThrowNativeError("Invalid index %d (count: %d)", idx, array->size()); ds@1952: } ds@1952: ds@1952: array->insert_at(idx); ds@1952: ds@1952: return 1; ds@1952: } ds@1952: ds@1952: static cell_t RemoveFromArray(IPluginContext *pContext, const cell_t *params) ds@1952: { ds@1952: CellArray *array; ds@1952: HandleError err; ds@1952: HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent); ds@1952: ds@1952: if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array)) ds@1952: != HandleError_None) ds@1952: { ds@1952: return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err); ds@1952: } ds@1952: ds@1952: size_t idx = (size_t)params[2]; ds@1952: if (idx >= array->size()) ds@1952: { ds@1952: return pContext->ThrowNativeError("Invalid index %d (count: %d)", idx, array->size()); ds@1952: } ds@1952: ds@1952: array->remove(idx); ds@1952: ds@1952: return 1; ds@1952: } ds@1952: ds@1952: static cell_t SwapArrayItems(IPluginContext *pContext, const cell_t *params) ds@1952: { ds@1952: CellArray *array; ds@1952: HandleError err; ds@1952: HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent); ds@1952: ds@1952: if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array)) ds@1952: != HandleError_None) ds@1952: { ds@1952: return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err); ds@1952: } ds@1952: ds@1952: size_t idx1 = (size_t)params[2]; ds@1952: size_t idx2 = (size_t)params[3]; ds@1952: if (idx1 >= array->size()) ds@1952: { ds@1952: return pContext->ThrowNativeError("Invalid index %d (count: %d)", idx1, array->size()); ds@1952: } ds@1952: if (idx2 >= array->size()) ds@1952: { ds@1952: return pContext->ThrowNativeError("Invalid index %d (count: %d)", idx2, array->size()); ds@1952: } ds@1952: ds@1952: array->swap(idx1, idx2); ds@1952: ds@1952: return 1; ds@1952: } ds@1952: ds@1952: static cell_t CloneArray(IPluginContext *pContext, const cell_t *params) ds@1952: { ds@1952: CellArray *oldArray; ds@1952: HandleError err; ds@1952: HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent); ds@1952: ds@1952: if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&oldArray)) ds@1952: != HandleError_None) ds@1952: { ds@1952: return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err); ds@1952: } ds@1952: ds@1952: CellArray *array = oldArray->clone(); ds@1952: ds@1952: Handle_t hndl = g_HandleSys.CreateHandle(htCellArray, array, pContext->GetIdentity(), g_pCoreIdent, NULL); ds@1952: if (!hndl) ds@1952: { ds@1952: delete array; ds@1952: } ds@1952: ds@1952: return hndl; ds@1952: } ds@1952: ds@1952: static cell_t FindStringInArray(IPluginContext *pContext, const cell_t *params) ds@1952: { ds@1952: CellArray *array; ds@1952: HandleError err; ds@1952: HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent); ds@1952: ds@1952: if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array)) ds@1952: != HandleError_None) ds@1952: { ds@1952: return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err); ds@1952: } ds@1952: ds@1952: char *str; ds@1952: pContext->LocalToString(params[2], &str); ds@1952: ds@1952: for (unsigned int i = 0; i < array->size(); i++) ds@1952: { ds@1952: const char *array_str = (const char *)array->at(i); ds@1952: if (strcmp(str, array_str) == 0) ds@1952: { ds@1952: return (cell_t) i; ds@1952: } ds@1952: } ds@1952: ds@1952: return -1; ds@1952: } ds@1952: ds@1952: static cell_t FindValueInArray(IPluginContext *pContext, const cell_t *params) ds@1952: { ds@1952: CellArray *array; ds@1952: HandleError err; ds@1952: HandleSecurity sec(pContext->GetIdentity(), g_pCoreIdent); ds@1952: ds@1952: if ((err = g_HandleSys.ReadHandle(params[1], htCellArray, &sec, (void **)&array)) ds@1952: != HandleError_None) ds@1952: { ds@1952: return pContext->ThrowNativeError("Invalid Handle %x (error: %d)", params[1], err); ds@1952: } ds@1952: ds@1952: for (unsigned int i = 0; i < array->size(); i++) ds@1952: { ds@1952: if (params[2] == *array->at(i)) ds@1952: { ds@1952: return (cell_t) i; ds@1952: } ds@1952: } ds@1952: ds@1952: return -1; ds@1952: } ds@1952: ds@1952: REGISTER_NATIVES(cellArrayNatives) ds@1952: { ds@1952: {"ClearArray", ClearArray}, ds@1952: {"CreateArray", CreateArray}, ds@1952: {"GetArrayArray", GetArrayArray}, ds@1952: {"GetArrayCell", GetArrayCell}, ds@1952: {"GetArraySize", GetArraySize}, ds@1952: {"GetArrayString", GetArrayString}, ds@1952: {"ResizeArray", ResizeArray}, ds@1952: {"PushArrayArray", PushArrayArray}, ds@1952: {"PushArrayCell", PushArrayCell}, ds@1952: {"PushArrayString", PushArrayString}, ds@1952: {"RemoveFromArray", RemoveFromArray}, ds@1952: {"SetArrayCell", SetArrayCell}, ds@1952: {"SetArrayString", SetArrayString}, ds@1952: {"SetArrayArray", SetArrayArray}, ds@1952: {"ShiftArrayUp", ShiftArrayUp}, ds@1952: {"SwapArrayItems", SwapArrayItems}, ds@1952: {"CloneArray", CloneArray}, ds@1952: {"FindStringInArray", FindStringInArray}, ds@1952: {"FindValueInArray", FindValueInArray}, ds@1952: {NULL, NULL}, ds@1952: };