diff -r 000000000000 -r c24a6fa957db core/smn_timers.cpp --- /dev/null Thu Jan 01 00:00:00 1970 +0000 +++ b/core/smn_timers.cpp Sun Jul 06 04:49:55 2008 +0000 @@ -0,0 +1,346 @@ +/** + * vim: set ts=4 : + * ============================================================================= + * SourceMod + * Copyright (C) 2004-2008 AlliedModders LLC. All rights reserved. + * ============================================================================= + * + * This program is free software; you can redistribute it and/or modify it under + * the terms of the GNU General Public License, version 3.0, as published by the + * Free Software Foundation. + * + * This program is distributed in the hope that it will be useful, but WITHOUT + * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS + * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more + * details. + * + * You should have received a copy of the GNU General Public License along with + * this program. If not, see . + * + * As a special exception, AlliedModders LLC gives you permission to link the + * code of this program (as well as its derivative works) to "Half-Life 2," the + * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software + * by the Valve Corporation. You must obey the GNU General Public License in + * all respects for all other code used. Additionally, AlliedModders LLC grants + * this exception to all derivative works. AlliedModders LLC defines further + * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007), + * or . + * + * Version: $Id$ + */ + +#include "HandleSys.h" +#include "TimerSys.h" +#include "PluginSys.h" +#include "Logger.h" +#include "DebugReporter.h" + +#define TIMER_HNDL_CLOSE (1<<9) + +HandleType_t g_TimerType; + +struct TimerInfo +{ + ITimer *Timer; + IPluginFunction *Hook; + IPluginContext *pContext; + Handle_t TimerHandle; + int UserData; + int Flags; +}; + +class TimerNatives : + public SMGlobalClass, + public IHandleTypeDispatch, + public ITimedEvent +{ +public: + ~TimerNatives(); +public: //ITimedEvent + ResultType OnTimer(ITimer *pTimer, void *pData); + void OnTimerEnd(ITimer *pTimer, void *pData); +public: //IHandleTypeDispatch + void OnHandleDestroy(HandleType_t type, void *object); +public: //SMGlobalClass + void OnSourceModAllInitialized(); + void OnSourceModShutdown(); +public: + TimerInfo *CreateTimerInfo(); + void DeleteTimerInfo(TimerInfo *pInfo); +private: + CStack m_FreeTimers; +}; + +TimerNatives::~TimerNatives() +{ + CStack::iterator iter; + for (iter=m_FreeTimers.begin(); iter!=m_FreeTimers.end(); iter++) + { + delete (*iter); + } + m_FreeTimers.popall(); +} + +void TimerNatives::OnSourceModAllInitialized() +{ + HandleAccess sec; + + g_HandleSys.InitAccessDefaults(NULL, &sec); + sec.access[HandleAccess_Clone] = HANDLE_RESTRICT_IDENTITY; + + g_TimerType = g_HandleSys.CreateType("Timer", this, 0, NULL, &sec, g_pCoreIdent, NULL); +} + +void TimerNatives::OnSourceModShutdown() +{ + g_HandleSys.RemoveType(g_TimerType, g_pCoreIdent); + g_TimerType = 0; +} + +void TimerNatives::OnHandleDestroy(HandleType_t type, void *object) +{ + TimerInfo *pTimer = reinterpret_cast(object); + + g_Timers.KillTimer(pTimer->Timer); +} + +TimerInfo *TimerNatives::CreateTimerInfo() +{ + TimerInfo *pInfo; + + if (m_FreeTimers.empty()) + { + pInfo = new TimerInfo; + } else { + pInfo = m_FreeTimers.front(); + m_FreeTimers.pop(); + } + + return pInfo; +} + +void TimerNatives::DeleteTimerInfo(TimerInfo *pInfo) +{ + m_FreeTimers.push(pInfo); +} + +ResultType TimerNatives::OnTimer(ITimer *pTimer, void *pData) +{ + TimerInfo *pInfo = reinterpret_cast(pData); + IPluginFunction *pFunc = pInfo->Hook; + cell_t res = static_cast(Pl_Continue); + + pFunc->PushCell(pInfo->TimerHandle); + pFunc->PushCell(pInfo->UserData); + pFunc->Execute(&res); + + return static_cast(res); +} + +void TimerNatives::OnTimerEnd(ITimer *pTimer, void *pData) +{ + HandleSecurity sec; + HandleError herr; + TimerInfo *pInfo = reinterpret_cast(pData); + Handle_t usrhndl = static_cast(pInfo->UserData); + + sec.pOwner = pInfo->pContext->GetIdentity(); + sec.pIdentity = g_pCoreIdent; + + if (pInfo->Flags & TIMER_HNDL_CLOSE) + { + if ((herr=g_HandleSys.FreeHandle(usrhndl, &sec)) != HandleError_None) + { + g_DbgReporter.GenerateError(pInfo->pContext, + pInfo->Hook->GetFunctionID(), + SP_ERROR_NATIVE, + "Invalid data handle %x (error %d) passed during timer end", + usrhndl, + herr); + } + } + + if ((herr=g_HandleSys.FreeHandle(pInfo->TimerHandle, &sec)) != HandleError_None) + { + g_DbgReporter.GenerateError(pInfo->pContext, + pInfo->Hook->GetFunctionID(), + SP_ERROR_NATIVE, + "Invalid timer handle %x (error %d) during timer end, displayed function is timer callback, not the stack trace", + pInfo->TimerHandle, + herr); + return; + } + DeleteTimerInfo(pInfo); +} + +/******************************* +* * +* TIMER NATIVE IMPLEMENTATIONS * +* * +********************************/ + +static TimerNatives s_TimerNatives; + +static cell_t smn_CreateTimer(IPluginContext *pCtx, const cell_t *params) +{ + IPluginFunction *pFunc; + TimerInfo *pInfo; + ITimer *pTimer; + Handle_t hndl; + int flags = params[4]; + + pFunc = pCtx->GetFunctionById(params[2]); + if (!pFunc) + { + return pCtx->ThrowNativeError("Invalid function id (%X)", params[2]); + } + + pInfo = s_TimerNatives.CreateTimerInfo(); + pTimer = g_Timers.CreateTimer(&s_TimerNatives, sp_ctof(params[1]), pInfo, flags); + + if (!pTimer) + { + s_TimerNatives.DeleteTimerInfo(pInfo); + return 0; + } + + hndl = g_HandleSys.CreateHandle(g_TimerType, pInfo, pCtx->GetIdentity(), g_pCoreIdent, NULL); + + pInfo->UserData = params[3]; + pInfo->Flags = flags; + pInfo->TimerHandle = hndl; + pInfo->Hook = pFunc; + pInfo->Timer = pTimer; + pInfo->pContext = pCtx; + + return hndl; +} + +static cell_t smn_KillTimer(IPluginContext *pCtx, const cell_t *params) +{ + Handle_t hndl = static_cast(params[1]); + HandleError herr; + HandleSecurity sec; + TimerInfo *pInfo; + + sec.pOwner = pCtx->GetIdentity(); + sec.pIdentity = g_pCoreIdent; + + if ((herr=g_HandleSys.ReadHandle(hndl, g_TimerType, &sec, (void **)&pInfo)) + != HandleError_None) + { + return pCtx->ThrowNativeError("Invalid timer handle %x (error %d)", hndl, herr); + } + + g_Timers.KillTimer(pInfo->Timer); + + if (params[2] && !(pInfo->Flags & TIMER_HNDL_CLOSE)) + { + sec.pOwner = pInfo->pContext->GetIdentity(); + sec.pIdentity = g_pCoreIdent; + + if ((herr=g_HandleSys.FreeHandle(static_cast(pInfo->UserData), &sec)) != HandleError_None) + { + return pCtx->ThrowNativeError("Invalid data handle %x (error %d)", hndl, herr); + } + } + + return 1; +} + +static cell_t smn_TriggerTimer(IPluginContext *pCtx, const cell_t *params) +{ + Handle_t hndl = static_cast(params[1]); + HandleError herr; + HandleSecurity sec; + TimerInfo *pInfo; + + sec.pOwner = pCtx->GetIdentity(); + sec.pIdentity = g_pCoreIdent; + + if ((herr=g_HandleSys.ReadHandle(hndl, g_TimerType, &sec, (void **)&pInfo)) + != HandleError_None) + { + return pCtx->ThrowNativeError("Invalid timer handle %x (error %d)", hndl, herr); + } + + g_Timers.FireTimerOnce(pInfo->Timer, params[2] ? true : false); + + return 1; +} + +static cell_t smn_GetTickedTime(IPluginContext *pContext, const cell_t *params) +{ + return sp_ftoc(*g_pUniversalTime); +} + +static cell_t smn_GetMapTimeLeft(IPluginContext *pContext, const cell_t *params) +{ + float time_left; + if (!g_Timers.GetMapTimeLeft(&time_left)) + { + return 0; + } + + cell_t *addr; + pContext->LocalToPhysAddr(params[1], &addr); + *addr = (int)time_left; + + return 1; +} + +static cell_t smn_GetMapTimeLimit(IPluginContext *pContext, const cell_t *params) +{ + IMapTimer *pMapTimer = g_Timers.GetMapTimer(); + + if (!pMapTimer) + { + return 0; + } + + cell_t *addr; + pContext->LocalToPhysAddr(params[1], &addr); + + *addr = pMapTimer->GetMapTimeLimit(); + + return 1; +} + +static cell_t smn_ExtendMapTimeLimit(IPluginContext *pContext, const cell_t *params) +{ + IMapTimer *pMapTimer = g_Timers.GetMapTimer(); + + if (!pMapTimer) + { + return 0; + } + + pMapTimer->ExtendMapTimeLimit(params[1]); + + return 1; +} + +static cell_t smn_IsServerProcessing(IPluginContext *pContext, const cell_t *params) +{ + return (gpGlobals->frametime > 0.0f) ? 1 : 0; +} + +static cell_t smn_GetTickInterval(IPluginContext *pContext, const cell_t *params) +{ + return sp_ftoc(gpGlobals->interval_per_tick); +} + +REGISTER_NATIVES(timernatives) +{ + {"CreateTimer", smn_CreateTimer}, + {"KillTimer", smn_KillTimer}, + {"TriggerTimer", smn_TriggerTimer}, + {"GetTickedTime", smn_GetTickedTime}, + {"GetMapTimeLeft", smn_GetMapTimeLeft}, + {"GetMapTimeLimit", smn_GetMapTimeLimit}, + {"ExtendMapTimeLimit", smn_ExtendMapTimeLimit}, + {"IsServerProcessing", smn_IsServerProcessing}, + {"GetTickInterval", smn_GetTickInterval}, + {NULL, NULL} +}; +