7、注册表读写的一个例子

#include " stdafx.h "
#include " iostream "
#include " string "
#include " Windows.h "

// 注册表项
#ifndef TEST_APP_REG
#define TEST_APP_REG "Software\\MySoft\\Common\\InstallRoot"
#endif

// 注册表中虚拟磁盘映射的盘符键
#ifndef TEST_APP_REG_DRIVE
#define TEST_APP_REG_DRIVE "Driver"
#endif

#ifndef RETURN_OK
#define RETURN_OK 1
#endif

#ifndef RETURN_ERROR
#define RETURN_ERROR 0
#endif

using namespace std;

int GetRegStringValue( const wstring & wstrPath, const wstring & wstrName, wstring & wstrValue)
{
HKEY hkey;
LONG lErrCode;
DWORD dwPathSize = 512 ;
BYTE bPath[ 512 ] = { 0 };
lErrCode = RegOpenKeyEx(HKEY_LOCAL_MACHINE ,
(LPCWSTR)(wstrPath.c_str()),
NULL,
KEY_QUERY_VALUE,
& hkey);
if (lErrCode != ERROR_SUCCESS)
{
return lErrCode;
}

lErrCode = RegQueryValueEx(hkey, (LPCWSTR)(wstrName.c_str()), NULL, NULL,
(LPBYTE)bPath, & dwPathSize);
if (ERROR_SUCCESS == lErrCode)
{
// lint -save -e534
wstrValue.append((wchar_t * ) bPath);
// lint -restore
}
RegCloseKey(hkey);
return lErrCode;
}

int CreateRegForSoftward()
{
LPCWSTR lpSubKey = L " Software\\MySoft\\Common\\InstallRoot " ;
LPCWSTR lpSubKeyValue = L " Driver " ;
CONST BYTE lpData[ 512 ] = { " Y " };

HKEY hMyKey, hMyKey1;

LONG lRet = RegCreateKeyEx(HKEY_LOCAL_MACHINE,lpSubKey,NULL,NULL,REG_OPTION_NON_VOLATILE,
KEY_ALL_ACCESS,NULL, & hMyKey,NULL);
if (ERROR_SUCCESS == lRet)
{
lRet = RegSetValueEx(hMyKey,lpSubKeyValue,NULL,REG_SZ,lpData, 4 );
if (lRet == ERROR_SUCCESS)
{
RegCloseKey(hMyKey);
return RETURN_OK;
}
}

RegCloseKey(hMyKey);
return RETURN_ERROR;
}

int main()
{
wstring wstrDrive, wstrPath = L " Y:\WINDDK " ;

CreateRegForSoftward();

if ( 0 != GetRegStringValue(TEXT(TEST_APP_REG), TEXT(TEST_APP_REG_DRIVE), wstrDrive))
{
return S_FALSE;
}

if (wstrPath.empty() || wstrDrive.empty())
{
return S_FALSE;
}

std::cout << _wcsnicmp(wstrPath.c_str(), wstrDrive.c_str(), 1 );

}