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@2088: void CNativeOwner::PropogateMarkSerial(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@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<WeakNative>::iterator iter;
dvander@2088: 	List<NativeEntry *>::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: 	/* Unmark our replacement natives */
dvander@2088: 	ntv_iter = m_ReplacedNatives.begin();
dvander@2088: 	while (ntv_iter != m_ReplacedNatives.end())
dvander@2088: 	{
dvander@2088: 		pEntry = (*ntv_iter);
dvander@2088: 		pEntry->replacement.func = NULL;
dvander@2088: 		pEntry->replacement.owner = NULL;
dvander@2088: 		ntv_iter = m_ReplacedNatives.erase(ntv_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<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: }
dvander@2088: 
dvander@2088: void CNativeOwner::AddReplacedNative(NativeEntry *pEntry)
dvander@2088: {
dvander@2088: 	m_ReplacedNatives.push_back(pEntry);
dvander@2088: }