dvander@3872: /**
dvander@3872: * vim: set ts=4 :
dvander@3872: * =============================================================================
dvander@3872: * SourcePawn
dvander@3872: * Copyright (C) 2004-2009 AlliedModders LLC. All rights reserved.
dvander@3872: * =============================================================================
dvander@3872: *
dvander@3872: * This program is free software; you can redistribute it and/or modify it under
dvander@3872: * the terms of the GNU General Public License, version 3.0, as published by the
dvander@3872: * Free Software Foundation.
dvander@3872: *
dvander@3872: * This program is distributed in the hope that it will be useful, but WITHOUT
dvander@3872: * ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or FITNESS
dvander@3872: * FOR A PARTICULAR PURPOSE. See the GNU General Public License for more
dvander@3872: * details.
dvander@3872: *
dvander@3872: * You should have received a copy of the GNU General Public License along with
dvander@3872: * this program. If not, see .
dvander@3872: *
dvander@3872: * As a special exception, AlliedModders LLC gives you permission to link the
dvander@3872: * code of this program (as well as its derivative works) to "Half-Life 2," the
dvander@3872: * "Source Engine," the "SourcePawn JIT," and any Game MODs that run on software
dvander@3872: * by the Valve Corporation. You must obey the GNU General Public License in
dvander@3872: * all respects for all other code used. Additionally, AlliedModders LLC grants
dvander@3872: * this exception to all derivative works. AlliedModders LLC defines further
dvander@3872: * exceptions, found in LICENSE.txt (as of this writing, version JULY-31-2007),
dvander@3872: * or .
dvander@3872: *
dvander@3872: * Version: $Id$
dvander@3872: */
dvander@2088: #include "NativeOwner.h"
dvander@2088: #include "ShareSys.h"
dvander@2088: #include "PluginSys.h"
dvander@2088:
dvander@2226: CNativeOwner::CNativeOwner() : m_nMarkSerial(0)
dvander@2226: {
dvander@2226: }
dvander@2226:
dvander@2088: void CNativeOwner::SetMarkSerial(unsigned int serial)
dvander@2088: {
dvander@2088: m_nMarkSerial = serial;
dvander@2088: }
dvander@2088:
dvander@2088: unsigned int CNativeOwner::GetMarkSerial()
dvander@2088: {
dvander@2088: return m_nMarkSerial;
dvander@2088: }
dvander@2088:
dvander@2088: void CNativeOwner::AddDependent(CPlugin *pPlugin)
dvander@2088: {
dvander@2088: m_Dependents.push_back(pPlugin);
dvander@2088: }
dvander@2088:
dvander@2088: void CNativeOwner::AddWeakRef(const WeakNative & ref)
dvander@2088: {
dvander@2088: m_WeakRefs.push_back(ref);
dvander@2088: }
dvander@2088:
dvander@2088: void CNativeOwner::AddNatives(const sp_nativeinfo_t *natives)
dvander@2088: {
dvander@2088: NativeEntry *pEntry;
dvander@2088:
dvander@2088: for (unsigned int i = 0; natives[i].func != NULL && natives[i].name != NULL; i++)
dvander@2088: {
dvander@2088: if ((pEntry = g_ShareSys.AddNativeToCache(this, &natives[i])) == NULL)
dvander@2088: {
dvander@2088: continue;
dvander@2088: }
dvander@2088:
dvander@2088: m_Natives.push_back(pEntry);
dvander@2088: }
dvander@2088: }
dvander@2088:
dvander@3872: void CNativeOwner::PropagateMarkSerial(unsigned int serial)
dvander@2088: {
dvander@2088: CNativeOwner *pOwner;
dvander@2088: List::iterator iter;
dvander@2088:
dvander@2088: for (iter = m_Dependents.begin();
dvander@2088: iter != m_Dependents.end();
dvander@2088: iter++)
dvander@2088: {
dvander@2088: pOwner = (*iter);
dvander@2088: pOwner->SetMarkSerial(serial);
dvander@2088: }
dvander@2088: }
dvander@2088:
dvander@2088: void CNativeOwner::UnbindWeakRef(const WeakNative & ref)
dvander@2088: {
dvander@2088: sp_native_t *native;
dvander@2088: IPluginContext *pContext;
dvander@2088:
dvander@2088: pContext = ref.pl->GetBaseContext();
dvander@2088: if ((pContext->GetNativeByIndex(ref.idx, &native)) == SP_ERROR_NONE)
dvander@2088: {
dvander@2088: /* If there is no reference, the native must be unbound */
dvander@2088: if (ref.entry == NULL)
dvander@2088: {
dvander@2088: native->status = SP_NATIVE_UNBOUND;
dvander@2088: native->pfn = NULL;
dvander@2088: }
dvander@2088: /* If we've cached a reference, it's a core native we can
dvander@2088: * rebind back to its original (this was a replacement).
dvander@2088: */
dvander@2088: else
dvander@2088: {
dvander@2088: native->pfn = ref.entry->func;
dvander@2088: }
dvander@2088: }
dvander@2088: }
dvander@2088:
dvander@2088: void CNativeOwner::DropEverything()
dvander@2088: {
dvander@2088: NativeEntry *pEntry;
dvander@2088: List::iterator iter;
dvander@2088: List::iterator ntv_iter;
dvander@2088:
dvander@2088: /* Unbind and remove all weak references to us */
dvander@2088: iter = m_WeakRefs.begin();
dvander@2088: while (iter != m_WeakRefs.end())
dvander@2088: {
dvander@2088: UnbindWeakRef((*iter));
dvander@2088: iter = m_WeakRefs.erase(iter);
dvander@2088: }
dvander@2088:
dvander@2088: /* Strip all of our natives from the cache */
dvander@2088: ntv_iter = m_Natives.begin();
dvander@2088: while (ntv_iter != m_Natives.end())
dvander@2088: {
dvander@2088: g_ShareSys.ClearNativeFromCache(this, (*ntv_iter)->name);
dvander@2088: ntv_iter = m_Natives.erase(ntv_iter);
dvander@2088: }
dvander@2088: }
dvander@2088:
dvander@2088: void CNativeOwner::DropWeakRefsTo(CPlugin *pPlugin)
dvander@2088: {
dvander@2088: List::iterator iter;
dvander@2088:
dvander@2088: iter = m_WeakRefs.begin();
dvander@2088: while (iter != m_WeakRefs.end())
dvander@2088: {
dvander@2088: WeakNative & ref = (*iter);
dvander@2088:
dvander@2088: if (ref.pl == pPlugin)
dvander@2088: {
dvander@2088: iter = m_WeakRefs.erase(iter);
dvander@2088: }
dvander@2088: else
dvander@2088: {
dvander@2088: iter++;
dvander@2088: }
dvander@2088: }
dvander@2088: }
dvander@2088:
dvander@2088: void CNativeOwner::DropRefsTo(CPlugin *pPlugin)
dvander@2088: {
dvander@2088: m_Dependents.remove(pPlugin);
dvander@2088: DropWeakRefsTo(pPlugin);
dvander@2088: }