mirror of https://github.com/tongzx/nt5src
You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
138 lines
3.0 KiB
138 lines
3.0 KiB
// 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;
|
|
}
|