学步园

#include <iostream><br /> #include <windows.h><br /> #include <shlwapi.h><br /> #pragma comment(lib,"shlwapi.lib")<br /> using namespace std;</p> <p>inline BYTE CharToBin(TCHAR ch)<br /> {<br /> if(ch>='0' && ch<='9')<br /> return ch - '0';<br /> if(ch>='a' && ch<='f')<br /> return ch - 'a' + 10;<br /> if(ch>='A' && ch<= 'F')<br /> return ch - 'A' + 10;<br /> return 0xff;<br /> }</p> <p>inline bool Hex2Bin(LPCTSTR lpHex,LPBYTE lpBin,DWORD &dwLen)<br /> {<br /> LPCTSTR lpChar = lpHex;<br /> TCHAR char1 = *lpChar;<br /> TCHAR char2 = *(lpChar+1);<br /> if(!char1 || !char2)<br /> return false;</p> <p> LPBYTE lpOut = lpBin;</p> <p> DWORD inputBuf = dwLen;<br /> dwLen = 0;</p> <p> BYTE h = CharToBin(char1);<br /> BYTE l = CharToBin(char2);<br /> while(0xff != h && 0xff != l)<br /> {<br /> *lpOut = l | h<<4;<br /> lpOut++;</p> <p> dwLen++;</p> <p> lpChar+=2;<br /> if(inputBuf<=dwLen)<br /> break;</p> <p> h = CharToBin(*lpChar);<br /> l = CharToBin(*(lpChar+1));<br /> }<br /> return true;<br /> }</p> <p>BOOL RegWrite(HKEY hKey,LPCSTR lpSubKey,LPCSTR lpValue,DWORD dwType,void* pData,DWORD dwLen)<br /> {<br /> if(!pData)<br /> return FALSE;</p> <p> if(REG_BINARY == dwType)<br /> {<br /> void* pBuffer = new BYTE[lstrlen((char*)pData)];<br /> memset(pBuffer,0,lstrlen((char*)pData));<br /> if(pBuffer)<br /> {<br /> Hex2Bin((LPCTSTR)pData,(LPBYTE)pBuffer,dwLen);<br /> SHSetValue(hKey,lpSubKey,lpValue,REG_BINARY,pBuffer,dwLen);<br /> delete []pBuffer;<br /> }<br /> }<br /> else<br /> {<br /> if(REG_SZ == dwType)<br /> {<br /> dwLen = lstrlen((const char*)pData);<br /> }<br /> else if(REG_DWORD == dwType)<br /> dwLen = 4;<br /> SHSetValue(hKey,lpSubKey,lpValue,dwType,pData,dwLen);<br /> }</p> <p> return TRUE;<br /> }</p> <p>void main(int argc,char*argv[])<br /> {<br /> RegWrite(HKEY_LOCAL_MACHINE,"SOFTWARE//aaaa//","b",REG_BINARY,"123456abcdef",6);</p> <p> DWORD dwValue = 1234;<br /> RegWrite(HKEY_LOCAL_MACHINE,"SOFTWARE//aaaa//","c",REG_DWORD,&dwValue,4);</p> <p> RegWrite(HKEY_LOCAL_MACHINE,"SOFTWARE//aaaa//","d",REG_SZ,"asdfgh",0);<br /> }