小華的部落格: 2007/12/23 - 2007/12/30

搜尋此網誌

星期一, 12月 24, 2007

VC++ 與Windows Registry 註冊表

昨天因為工作需要寫了一個小工具去修正註冊表內的鍵值,所以自己留下一些筆記在部落格。

註冊表的存取方式由微軟文件上說明得知,需要做3種步驟:
1. 得到Handle
2. 存取你要的鍵值
3.關閉Handle

而步驟1中,最主要是透過RegCreateKey與RegOepnKey來得到Handle,而這兩種方式的不同點在於RegCreateKey會去搜尋子鍵是否存在,如果不存在時會建立一個新的子鍵,而另一個API則是找不到就找不到。

步驟2中,我們如果要去設定鍵值ㄧ般都會使用RegSetValueEx() 其中它裡面的參數有分成不同型態的鍵值,像是REG_DWORD、REG_SZ...etc,使用時需要步驟1 所得到的Handle。

步驟3中則是去關閉你開啟的Handle,因此這3個步驟必須合在一起做。

底下是我把這3個步驟整理成副程式,使用時直接呼叫就可以了:
1.SetRegValueBy_REG_DWORD();
2.SetRegValueBy_REG_SZ();
3. SetRegValueBy_REG_MULTI_SZ();

void SetRegValueBy_REG_DWORD(LPCSTR szKeyPath,LPCSTR szKeyName,DWORD *dwData)
{
HKEY hk;
if (RegCreateKey(HKEY_LOCAL_MACHINE,szKeyPath, &hk))
OutputDebugString("error!");

if (RegSetValueEx(hk,
szKeyName,
0,
REG_DWORD,
(LPBYTE) dwData,
sizeof(DWORD)))
OutputDebugString("error!");

RegCloseKey(hk);
}

void SetRegValueBy_REG_SZ(LPCSTR szKeyPath,LPCSTR szKeyName,LPCSTR keyValue)
{
HKEY hk;
if (RegCreateKey(HKEY_LOCAL_MACHINE,szKeyPath, &hk))
OutputDebugString("error!");

if (RegSetValueEx(hk,
szKeyName,
0,
REG_SZ,
(BYTE*)(LPCSTR) keyValue,
strlen(keyValue)))
OutputDebugString("error!");

RegCloseKey(hk);
}

void SetRegValueBy_REG_MULTI_SZ(LPCSTR szKeyPath,LPCSTR szKeyName,LPCSTR keyValue)
{
HKEY hk;
if (RegCreateKey(HKEY_LOCAL_MACHINE,szKeyPath, &hk))
OutputDebugString("error!");

if (RegSetValueEx(hk,
szKeyName,

0,
REG_MULTI_SZ,
(BYTE*)(LPCSTR) keyValue,
strlen(keyValue)))
OutputDebugString("error!");

RegCloseKey(hk);
}


由副程式內可以得知,我是存取主鍵"HKEY_LOCAL_MACHINE"內的子鍵,所以如果要存取不同的主鍵時,要修改副程式內的主鍵。

底下是呼叫時的範例:

BOOL MyPatch()
{
LPCSTR szKeyPath1="SOFTWARE\\Microsoft\\WindowsNT\\MyTest";
LPCSTR szKeyName1="Label";
DWORD dwData=11;

SetRegValueBy_REG_DWORD(szKeyPath1,szKeyName1,&dwData);

return 0;
}

我Win32 的程式不好,雖然上面的程式碼可以用,但是如果有發現錯誤的地方,還是請大家幫忙指正並且告訴我~~~感恩喔!