CWE-ou-minkata/Sources/Plasma/PubUtilLib/plResMgr/plRegistryKeyList.cpp

/*==LICENSE==*

CyanWorlds.com Engine - MMOG client, server and tools
Copyright (C) 2011 Cyan Worlds, Inc.

This program is free software: you can redistribute it and/or modify
it under the terms of the GNU General Public License as published by
the Free Software Foundation, either version 3 of the License, or
(at your option) any later version.

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 <http://www.gnu.org/licenses/>.

Additional permissions under GNU GPL version 3 section 7

If you modify this Program, or any covered work, by linking or
combining it with any of RAD Game Tools Bink SDK, Autodesk 3ds Max SDK,
NVIDIA PhysX SDK, Microsoft DirectX SDK, OpenSSL library, Independent
JPEG Group JPEG library, Microsoft Windows Media SDK, or Apple QuickTime SDK
(or a modified version of those libraries),
containing parts covered by the terms of the Bink SDK EULA, 3ds Max EULA,
PhysX SDK EULA, DirectX SDK EULA, OpenSSL and SSLeay licenses, IJG
JPEG Library README, Windows Media SDK EULA, or QuickTime SDK EULA, the
licensors of this Program grant you additional
permission to convey the resulting work. Corresponding Source for a
non-source form of such a combination shall include the source code for
the parts of OpenSSL and IJG JPEG Library used as well as that of the covered
work.

You can contact Cyan Worlds, Inc. by email legal@cyan.com
 or by snail mail at:
      Cyan Worlds, Inc.
      14617 N Newport Hwy
      Mead, WA   99021

*==LICENSE==*/
#include "plRegistryKeyList.h"
#include "plRegistryHelpers.h"
#include "hsStream.h"
#include <algorithm>

plRegistryKeyList::plRegistryKeyList(UInt16 classType)
{
	fClassType = classType;
	fReffedStaticKeys = 0;
	fLocked = 0;
	fFlags = 0;
}

plRegistryKeyList::~plRegistryKeyList()
{
	hsAssert(fLocked == 0, "Key list still locked on delete");

	for (int i = 0; i < fStaticKeys.size(); i++)
	{
		plKeyImp* keyImp = fStaticKeys[i];
		if (keyImp && !keyImp->ObjectIsLoaded()) {
			keyImp = nullptr;
			delete keyImp;
		}
	}
}

// Special dummy key that lets us set the return value of the GetName call.
// Makes it easier to do STL searches.
class plSearchKeyImp : public plKeyImp
{
public:
	const char* fSearchKeyName;
	const char* GetName() const { return fSearchKeyName; }
};

plKeyImp* plRegistryKeyList::FindKey(const char* keyName)
{
	static plSearchKeyImp searchKey;
	searchKey.fSearchKeyName = keyName;

	// Search the static key list
	if (fFlags & kStaticUnsorted)
	{
		// We're unsorted, brute force it.  May do a separate search table in the
		// future if this is a bottlneck
		for (int i = 0; i < fStaticKeys.size(); i++)
		{
			plKeyImp* curKey = fStaticKeys[i];
			if (curKey && hsStrCaseEQ(keyName, curKey->GetName()))
				return curKey;
		}
	}
	else
	{
		// We're sorted, do a fast lookup
		StaticVec::const_iterator it = std::lower_bound(fStaticKeys.begin(), fStaticKeys.end(), &searchKey, KeySorter());
		if (it != fStaticKeys.end() && hsStrCaseEQ(keyName, (*it)->GetName()))
			return *it;
	}

	// Search the dynamic key list
	DynSet::const_iterator dynIt = fDynamicKeys.find(&searchKey);
	if (dynIt != fDynamicKeys.end())
		return *dynIt;

	return nil;
}

plKeyImp* plRegistryKeyList::FindKey(const plUoid& uoid)
{
	UInt32 objectID = uoid.GetObjectID();

	// Key is dynamic or doesn't know it's index.  Do a find by name.
	if (objectID == 0)
		return FindKey(uoid.GetObjectName());

	// Direct lookup
	if (objectID <= fStaticKeys.size())
	{
#ifdef PLASMA_EXTERNAL_RELEASE
		return fStaticKeys[objectID-1];
#else
		// If this is an internal release, our objectIDs might not match
		// because of local data. Verify that we have the right key by
		// name, and if it's wrong, do the slower find-by-name.
		plKeyImp *keyImp = fStaticKeys[objectID-1];
		if (!hsStrCaseEQ(keyImp->GetName(), uoid.GetObjectName()))
			return FindKey(uoid.GetObjectName());
		else
			return keyImp;
#endif // PLASMA_EXTERNAL_RELEASE
	}

	// If we got here it probably means we just deleted all our keys of the matching type
	// because no one was using them. No worries. The resManager will catch this and 
	// reload our keys, then try again.

	return nil;
}

void plRegistryKeyList::ILock()
{
	fLocked++;
}

void plRegistryKeyList::IUnlock()
{
	fLocked--;
	if (fLocked == 0)
		IRepack();
}

bool plRegistryKeyList::IterateKeys(plRegistryKeyIterator* iterator)
{
	ILock();

	for (int i = 0; i < fStaticKeys.size(); i++)
	{
		plKeyImp* keyImp = fStaticKeys[i];
		if (keyImp != nil)
		{
			if (!iterator->EatKey(plKey::Make(keyImp)))
			{
				IUnlock();
				return false;
			}
		}
	}

	DynSet::const_iterator it;
	for (it = fDynamicKeys.begin(); it != fDynamicKeys.end(); it++)
	{
		plKeyImp* keyImp = *it;
		hsAssert(keyImp, "Shouldn't ever have a nil dynamic key");
		if (!iterator->EatKey(plKey::Make(keyImp)))
		{
			IUnlock();
			return false;
		}
	}

	IUnlock();
	return true;
}

void plRegistryKeyList::AddKey(plKeyImp* key, LoadStatus& loadStatusChange)
{
	loadStatusChange = kNoChange;

	hsAssert(fLocked == 0, "Don't currently support adding keys while locked");
	if (fLocked == 0 && key != nil)
	{
		// If this is the first key added, we just became loaded
		if (fDynamicKeys.empty())
			loadStatusChange = kDynLoaded;

		hsAssert(fDynamicKeys.find(key) == fDynamicKeys.end(), "Key already added");
		fDynamicKeys.insert(key);
	}
}

void plRegistryKeyList::SetKeyUsed(plKeyImp* key)
{
	// If this is a static key, mark that we used it.  Otherwise, just ignore it.
	UInt32 id = key->GetUoid().GetObjectID();
	if (id > 0)
		fReffedStaticKeys++;
}

bool plRegistryKeyList::SetKeyUnused(plKeyImp* key, LoadStatus& loadStatusChange)
{
	loadStatusChange = kNoChange;

	// Clones never officially get added to the key list (they're maintained by
	// the original key), so just ignore them
	if (key->GetUoid().IsClone())
	{
		delete key;
		return true;
	}

	// Check if it's a static key
	UInt32 id = key->GetUoid().GetObjectID();
	hsAssert(id <= fStaticKeys.size(), "Bad static key id");
	if (id != 0 && id <= fStaticKeys.size())
	{
		fReffedStaticKeys--;
		if (fLocked == 0)
			IRepack();

		// That was our last used static key, we're static unloaded
		if (fReffedStaticKeys == 0)
			loadStatusChange = kStaticUnloaded;

		return true;
	}

	// Try to find it in the dynamic key list
	DynSet::iterator dynIt = fDynamicKeys.find(key);
	if (dynIt != fDynamicKeys.end())
	{
		hsAssert(fLocked == 0, "Don't currently support removing dynamic keys while locked");
		if (fLocked == 0)
		{
			fDynamicKeys.erase(dynIt);
			delete key;

			// That was our last dynamic key, notify of dynamic unloaded
			if (fDynamicKeys.empty())
				loadStatusChange = kDynUnloaded;

			return true;
		}
		return false;
	}

	hsAssert(0, "Couldn't find this key, what is it?");
	return false;
}

//// IRepack /////////////////////////////////////////////////////////////////
//	Frees the memory for our static key array if none of them are loaded
void plRegistryKeyList::IRepack()
{
	if (fReffedStaticKeys == 0 && !fStaticKeys.empty())
	{

		for (int i = 0; i < fStaticKeys.size(); i++)
			delete fStaticKeys[i];

		fStaticKeys.clear();
	}
}

void plRegistryKeyList::PrepForWrite()
{
	// If we have any static keys already, we were read in.  To keep from
	// invalidating old key indexes any new keys have to go on the end, hence we're
	// unsorted now.
	if (!fStaticKeys.empty())
		fFlags |= kStaticUnsorted;

	// If a dynamic keys doesn't have an object assigned to it, we're not writing
	// it out.  Figure out how many valid keys we have.
	int numDynKeys = 0;
	DynSet::const_iterator cIt;
	for (cIt = fDynamicKeys.begin(); cIt != fDynamicKeys.end(); cIt++)
	{
		plKeyImp* key = *cIt;
		// We're only going to write out keys that have objects
		if (key->ObjectIsLoaded())
			numDynKeys++;
	}

	// Start our new object id's after any already created ones
	UInt32 objectID = fStaticKeys.size()+1;
	// Make room for our new keys
	fStaticKeys.resize(fStaticKeys.size()+numDynKeys);

	DynSet::iterator it = fDynamicKeys.begin();
	while (it != fDynamicKeys.end())
	{
		plKeyImp* key = *it;
		it++;

		// If we're gonna use this key, tag it with it's object id and move it to the static array.
		if (key->ObjectIsLoaded())
		{
			key->SetObjectID(objectID);
			fStaticKeys[objectID-1] = key;

			objectID++;
			fReffedStaticKeys++;
			fDynamicKeys.erase(key);
		}
	}
}

void plRegistryKeyList::Read(hsStream* s)
{
	UInt32 keyListLen = s->ReadSwap32();
	if (!fStaticKeys.empty())
	{
		s->Skip(keyListLen);
		return;
	}

	fFlags = s->ReadByte();

	UInt32 numKeys = s->ReadSwap32();
	fStaticKeys.resize(numKeys);

	for (int i = 0; i < numKeys; i++)
	{
		plKeyImp* newKey = TRACKED_NEW plKeyImp;
		newKey->Read(s);
		fStaticKeys[i] = newKey;
	}
}

void plRegistryKeyList::Write(hsStream* s)
{
	// Save space for the length of our data
	UInt32 beginPos = s->GetPosition();
	s->WriteSwap32(0);
	s->WriteByte(fFlags);

	int numKeys = fStaticKeys.size();
	s->WriteSwap32(numKeys);

	// Write out all our keys (anything in dynamic is unused, so just ignore those)
	for (int i = 0; i < numKeys; i++)
	{
		plKeyImp* key = fStaticKeys[i];
		key->Write(s);
	}

	// Go back to the start and write the length of our data
	UInt32 endPos = s->GetPosition();
	s->SetPosition(beginPos);
	s->WriteSwap32(endPos-beginPos-sizeof(UInt32));
	s->SetPosition(endPos);
}