windows-xp/Source/XPSP1/NT/enduser/troubleshoot/sniffpol/ppserverclass.cpp

// PPServerClass.cpp : Implementation of CPPServerClass
#include "stdafx.h"
#include "PPServer.h"
#include "PPServerClass.h"

/////////////////////////////////////////////////////////////////////////////
// Keys
#define REG_LOCAL_TS_LOC		_T("SOFTWARE\\Microsoft")
#define REG_LOCAL_TS_PROGRAM	_T("TShoot")
// Values ///////////////////////////////////////////////////////////////////
#define SNIFF_AUTOMATIC_STR		_T("AutomaticSniffing")
/////////////////////////////////////////////////////////////////////////////

/////////////////////////////////////////////////////////////////////////////
// CPPServerClass

bool CPPServerClass::Create(HKEY hKeyParent, LPCTSTR strKeyName, bool* bCreatedNew, REGSAM access)
{
	HKEY hRetKey = NULL;
	DWORD dwDisposition = 0;

	long nWinError = ::RegCreateKeyEx(
		hKeyParent,
		strKeyName,
		0,
		NULL,
		REG_OPTION_NON_VOLATILE,
		access,
		NULL,
		&hRetKey,
		&dwDisposition
		);

	if(nWinError == ERROR_SUCCESS)
	{
		m_hKey = hRetKey;
		*bCreatedNew = dwDisposition == REG_CREATED_NEW_KEY ? true : false;
		
		try
		{
			m_arrKeysToClose.push_back(hRetKey);
		}
		catch (exception&)
		{
			return false;
		}

		return true;
	}
	return false;
}

bool CPPServerClass::SetNumericValue(LPCTSTR strValueName, DWORD dwValue)
{
	BYTE* pData = (BYTE*)&dwValue;
	long nWinError = ::RegSetValueEx(
		m_hKey,
		strValueName,
		0,
		REG_DWORD,
		pData,
		sizeof(DWORD)
		);

	if (nWinError == ERROR_SUCCESS)
		return true;
	return false;
}

bool CPPServerClass::GetNumericValue(LPCTSTR strValueName, DWORD& dwValue)
{
	DWORD tmp = 0;
	BYTE* pData = (BYTE*)&tmp;
	DWORD type = 0;
	DWORD size = sizeof(DWORD);

	long nWinError = ::RegQueryValueEx(
		m_hKey,
		strValueName,
		NULL,
		&type,
		pData,
		&size
		);

	if (type != REG_DWORD)
		return false;

	if (nWinError == ERROR_SUCCESS)
	{
		dwValue = tmp;
		return true;
	}
	return false;
}

void CPPServerClass::Close()
{
	for (vector<HKEY>::reverse_iterator i = m_arrKeysToClose.rbegin(); i != m_arrKeysToClose.rend(); i++)
		::RegCloseKey( *i );

	m_arrKeysToClose.clear();
}

STDMETHODIMP CPPServerClass::AllowAutomaticSniffing(VARIANT *pvarShow)
{

	bool was_created = false;
	DWORD dwAllowSniffing = 1;

	// [BC - 20010302] - Changed regsitry access level from WRITE to QUERY and READ. Write access
	// not allowed for certain user accts, such as WinXP built in guest acct. Write access should
	// not be required by this component.
	if (Create(HKEY_LOCAL_MACHINE, REG_LOCAL_TS_LOC, &was_created, KEY_QUERY_VALUE))
	{
		if (Create(m_hKey, REG_LOCAL_TS_PROGRAM, &was_created, KEY_READ))
		{
			// this call can be not successfull, if there is no such value.
			//  BUT we do not set this value,
			//  we leave dwAllowSniffing as initialized ("1")
			// This approach will comply the "Sniffing version 3.2.doc" statement,
			//  that if "AutomaticSniffing" value is missed, we treat it as set to "1"
			GetNumericValue(SNIFF_AUTOMATIC_STR, dwAllowSniffing);
		}
	}

	Close();

	::VariantInit(pvarShow);
	V_VT(pvarShow) = VT_I4;

	if (dwAllowSniffing == 1)
		pvarShow->lVal = 1;
	else
		pvarShow->lVal = 0;

	return S_OK;
}