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
);
}