ProSysLib: Dissecting the Process

// PSLPrivileges.cpp : Implementation of CPSLPrivileges

#include "stdafx.h"
#include "PSLPrivileges.h"


CPSLPrivileges::CPSLPrivileges()
{
}

HRESULT CPSLPrivileges::OnIndexOutOfRange()
{
	return MakeException(exIndexOutOfRange);
}

HRESULT CPSLPrivileges::FinalConstruct()
{
	PSL_BEGIN

	InternalUpdate();
	
	PSL_END
}

void CPSLPrivileges::FinalRelease()
{
}

void CPSLPrivileges::InternalUpdate()
{
	CCritSecLock cs(m_csCollection);

	m_coll.clear(); // Releases interfaces and clears the collection;
					// NOTE: .NET clients catch up interfaces, so takes
					// about 30 seconds before they are garbaged by .NET;

	HANDLE hToken = NULL;
	if(::OpenProcessToken(::GetCurrentProcess(), TOKEN_READ, &hToken))
	{
		DWORD dwLength = 0;
		::GetTokenInformation(hToken, TokenPrivileges, NULL, 0, &dwLength);
		LPBYTE pBuffer = new BYTE[dwLength];
		if(::GetTokenInformation(hToken, TokenPrivileges, pBuffer, dwLength, &dwLength))
		{
			TOKEN_PRIVILEGES * pPrivileges = (TOKEN_PRIVILEGES*)(LPVOID)pBuffer;
			LUID_AND_ATTRIBUTES * Privileges = pPrivileges->Privileges;
			DWORD dwCount = pPrivileges->PrivilegeCount;
			TCHAR PrivilegeName[64];
			for(DWORD i = 0;i < dwCount;i ++)
			{
				DWORD dwLength = 64;
				if(::LookupPrivilegeName(NULL, &Privileges[i].Luid, PrivilegeName, &dwLength))
				{
					CComObject<CPSLPrivilege> * pPrivilege = NULL;
					if(CComObject<CPSLPrivilege>::CreateInstance(&pPrivilege) == S_OK)
					{
						pPrivilege->Initialize(Privileges[i], PrivilegeName);
						// NOTE: Here the interface is also gets AddRef;
						m_coll.push_back(CComPtr<IPSLPrivilege>(pPrivilege));
					}
				}		
			}			
		}
		delete []pBuffer;
		::CloseHandle(hToken);
	}
}

////////////////////////////////////////////////////////////////////////
// Interface Implementation;
////////////////////////////////////////////////////////////////////////

STDMETHODIMP CPSLPrivileges::Update()
{
	PSL_BEGIN

	InternalUpdate();

	PSL_END
}

STDMETHODIMP CPSLPrivileges::Find(BSTR PrivilegeName, IPSLPrivilege ** ppValue)
{
	PSL_BEGIN

	*ppValue = NULL;
	_bstr_t sPrivilegeName(PrivilegeName);
	CCritSecLock cs(m_csCollection);

	for(PrivilegesType::iterator i = m_coll.begin();i != m_coll.end();i ++)
	{
		BSTR sName = NULL;
		if(i->m_T->get_Name(&sName) == S_OK)
		{
			_bstr_t s(sName);
			if(!::_tcsicmp(s, sPrivilegeName)) // Doing case-insensitive comparison;
			{
				IPSLPrivilege * pPrivilege = i->m_T;
				pPrivilege->AddRef();
				*ppValue = pPrivilege;
				break;
			}
		}	
	}

	PSL_END
}