dvander@3872: /**
dvander@4107:  * vim: set ts=4 sw=4 tw=99 noet :
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 <http://www.gnu.org/licenses/>.
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 <http://www.sourcemod.net/license.php>.
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@4107: 	for (const sp_nativeinfo_t *native = natives; native->func && native->name; native++)
dvander@4107: 		g_ShareSys.AddNativeToCache(this, native);
dvander@2088: 
dvander@4107: 	m_natives.append(natives);
dvander@2088: }
dvander@2088: 
dvander@3872: void CNativeOwner::PropagateMarkSerial(unsigned int serial)
dvander@2088: {
dvander@2088: 	CNativeOwner *pOwner;
dvander@2088: 	List<CPlugin *>::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@4109: 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@4109: 		native->status = SP_NATIVE_UNBOUND;
dvander@4109: 		native->pfn = NULL;
dvander@2088: 	}
dvander@2088: }
dvander@2088: 
dvander@2088: void CNativeOwner::DropEverything()
dvander@2088: {
dvander@2088: 	List<WeakNative>::iterator 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@4107: 	for (size_t i = 0; i < m_natives.length(); i++) {
dvander@4107: 		const sp_nativeinfo_t *natives = m_natives[i];
dvander@4107: 		for (const sp_nativeinfo_t *native = natives; native->func && native->name; native++)
dvander@4107: 			g_ShareSys.ClearNativeFromCache(this, native->name);
dvander@4107: 	}
dvander@4109: 	m_natives.clear();
dvander@4107: 
dvander@4109: 	for (size_t i = 0; i < m_fakes.length(); i++)
dvander@4109: 		g_ShareSys.ClearNativeFromCache(this, m_fakes[i]->name());
dvander@4109: 	m_fakes.clear();
dvander@2088: }
dvander@2088: 
dvander@2088: void CNativeOwner::DropWeakRefsTo(CPlugin *pPlugin)
dvander@2088: {
dvander@2088: 	List<WeakNative>::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: }