学步园

因为最近要学这方面的东西，就在网上找喽，结果发现这个类。感谢那个人啊

// Registry.cpp : implementation file<br /> //<br /> #include "stdafx.h"<br /> #include "Registry.h"<br /> /////////////////////////////////////////////////////////////////////////////<br /> // CRegistry<br /> CRegistry::CRegistry(HKEY hKey)<br /> {<br /> m_hKey=hKey;<br /> }<br /> CRegistry::~CRegistry()<br /> {<br /> Close();<br /> }<br /> /////////////////////////////////////////////////////////////////////////////<br /> // CRegistry Functions<br /> BOOL CRegistry::CreateKey(LPCTSTR lpSubKey)<br /> {<br /> ASSERT(m_hKey);<br /> ASSERT(lpSubKey);<br /> HKEY hKey;<br /> DWORD dw;<br /> long lReturn=RegCreateKeyEx(m_hKey,lpSubKey,0L,NULL,REG_OPTION_VOLATILE,KEY_ALL_ACCESS,NULL,&hKey,&dw);</p> <p> if(lReturn==ERROR_SUCCESS)<br /> {<br /> m_hKey=hKey;<br /> return TRUE;<br /> }</p> <p> return FALSE;</p> <p>}<br /> BOOL CRegistry::Open(LPCTSTR lpSubKey)<br /> {<br /> ASSERT(m_hKey);<br /> ASSERT(lpSubKey);</p> <p> HKEY hKey;<br /> long lReturn=RegOpenKeyEx(m_hKey,lpSubKey,0L,KEY_ALL_ACCESS,&hKey);</p> <p> if(lReturn==ERROR_SUCCESS)<br /> {<br /> m_hKey=hKey;<br /> return TRUE;<br /> }<br /> return FALSE;</p> <p>}<br /> void CRegistry::Close()<br /> {<br /> if(m_hKey)<br /> {<br /> RegCloseKey(m_hKey);<br /> m_hKey=NULL;<br /> }</p> <p>}<br /> BOOL CRegistry::DeleteValue(LPCTSTR lpValueName)<br /> {<br /> ASSERT(m_hKey);<br /> ASSERT(lpValueName);</p> <p> long lReturn=RegDeleteValue(m_hKey,lpValueName);</p> <p> if(lReturn==ERROR_SUCCESS)<br /> return TRUE;<br /> return FALSE;</p> <p>}<br /> BOOL CRegistry::DeleteKey(HKEY hKey, LPCTSTR lpSubKey)<br /> {<br /> ASSERT(hKey);<br /> ASSERT(lpSubKey);</p> <p> long lReturn=RegDeleteValue(hKey,lpSubKey);</p> <p> if(lReturn==ERROR_SUCCESS)<br /> return TRUE;<br /> return FALSE;</p> <p>}<br /> BOOL CRegistry::Write(LPCTSTR lpSubKey, int nVal)<br /> {<br /> ASSERT(m_hKey);<br /> ASSERT(lpSubKey);</p> <p> DWORD dwValue;<br /> dwValue=(DWORD)nVal;</p> <p> long lReturn=RegSetValueEx(m_hKey,lpSubKey,0L,REG_DWORD,(const BYTE *) &dwValue,sizeof(DWORD));</p> <p> if(lReturn==ERROR_SUCCESS)<br /> return TRUE;</p> <p> return FALSE;</p> <p>}<br /> BOOL CRegistry::Write(LPCTSTR lpSubKey, DWORD dwVal)<br /> {<br /> ASSERT(m_hKey);<br /> ASSERT(lpSubKey);</p> <p> long lReturn=RegSetValueEx(m_hKey,lpSubKey,0L,REG_DWORD,(const BYTE *) &dwVal,sizeof(DWORD));</p> <p> if(lReturn==ERROR_SUCCESS)<br /> return TRUE;</p> <p> return FALSE;</p> <p>}<br /> BOOL CRegistry::Write(LPCTSTR lpValueName, LPCTSTR lpValue)<br /> {<br /> ASSERT(m_hKey);<br /> ASSERT(lpValueName);<br /> ASSERT(lpValue);<br /> long lReturn=RegSetValueEx(m_hKey,lpValueName,0L,REG_SZ,(const BYTE *) lpValue,strlen(lpValue)+1);</p> <p> if(lReturn==ERROR_SUCCESS)<br /> return TRUE;</p> <p> return FALSE;</p> <p>}</p> <p>BOOL CRegistry::Read(LPCTSTR lpValueName, int* pnVal)<br /> {<br /> ASSERT(m_hKey);<br /> ASSERT(lpValueName);<br /> ASSERT(pnVal);</p> <p> DWORD dwType;<br /> DWORD dwSize=sizeof(DWORD);<br /> DWORD dwDest;<br /> long lReturn=RegQueryValueEx(m_hKey,lpValueName,NULL,&dwType,(BYTE *)&dwDest,&dwSize);</p> <p> if(lReturn==ERROR_SUCCESS)<br /> {<br /> *pnVal=(int)dwDest;<br /> return TRUE;<br /> }<br /> return FALSE;</p> <p>}<br /> BOOL CRegistry::Read(LPCTSTR lpValueName, DWORD* pdwVal)<br /> {<br /> ASSERT(m_hKey);<br /> ASSERT(lpValueName);<br /> ASSERT(pdwVal);</p> <p> DWORD dwType;<br /> DWORD dwSize=sizeof(DWORD);<br /> DWORD dwDest;<br /> long lReturn=RegQueryValueEx(m_hKey,lpValueName,NULL,&dwType,(BYTE *)&dwDest,&dwSize);</p> <p> if(lReturn==ERROR_SUCCESS)<br /> {<br /> *pdwVal=dwDest;<br /> return TRUE;<br /> }<br /> return FALSE;</p> <p>}</p> <p>BOOL CRegistry::RestoreKey(LPCTSTR lpFileName)<br /> {<br /> ASSERT(m_hKey);<br /> ASSERT(lpFileName);</p> <p> long lReturn=RegRestoreKey(m_hKey,lpFileName,REG_WHOLE_HIVE_VOLATILE);</p> <p> if(lReturn==ERROR_SUCCESS)<br /> return TRUE;</p> <p> return FALSE;<br /> }<br /> BOOL CRegistry::SaveKey(LPCTSTR lpFileName)<br /> {<br /> ASSERT(m_hKey);<br /> ASSERT(lpFileName);</p> <p> long lReturn=RegSaveKey(m_hKey,lpFileName,NULL);</p> <p> if(lReturn==ERROR_SUCCESS)<br /> return TRUE;</p> <p> return FALSE;<br /> }</p> <p>BOOL CRegistry::Read(LPCTSTR lpValueName, CString* lpVal)<br /> {<br /> ASSERT(m_hKey);<br /> ASSERT(lpValueName);<br /> ASSERT(lpVal);</p> <p> DWORD dwType;<br /> DWORD dwSize=200;<br /> char szString[2550];</p> <p> long lReturn=RegQueryValueEx(m_hKey,lpValueName,NULL,&dwType,(BYTE *)szString,&dwSize);</p> <p> if(lReturn==ERROR_SUCCESS)<br /> {<br /> *lpVal=szString;<br /> return TRUE;<br /> }<br /> return FALSE;</p> <p>}

#if !defined(AFX_REGISTRY_H__E0610A5D_7166_4D02_9D7E_11AF7CF8E229__INCLUDED_)<br /> #define AFX_REGISTRY_H__E0610A5D_7166_4D02_9D7E_11AF7CF8E229__INCLUDED_<br /> //<br /> #include <winreg.h><br /> /////////////////////////////////////////////////////////////////////////////<br /> // CRegistry window<br /> class CRegistry : public CObject<br /> {<br /> // Construction<br /> public:<br /> CRegistry(HKEY hKey=HKEY_LOCAL_MACHINE);<br /> public:<br /> BOOL SaveKey(LPCTSTR lpFileName);<br /> BOOL RestoreKey(LPCTSTR lpFileName);<br /> BOOL Read(LPCTSTR lpValueName, CString* lpVal);<br /> BOOL Read(LPCTSTR lpValueName, DWORD* pdwVal);<br /> BOOL Read(LPCTSTR lpValueName, int* pnVal);<br /> BOOL Write(LPCTSTR lpSubKey, LPCTSTR lpVal);<br /> BOOL Write(LPCTSTR lpSubKey, DWORD dwVal);<br /> BOOL Write(LPCTSTR lpSubKey, int nVal);<br /> BOOL DeleteKey(HKEY hKey, LPCTSTR lpSubKey);<br /> BOOL DeleteValue(LPCTSTR lpValueName);<br /> void Close();<br /> BOOL Open(LPCTSTR lpSubKey);<br /> BOOL CreateKey(LPCTSTR lpSubKey);<br /> virtual ~CRegistry();<br /> protected:<br /> HKEY m_hKey;</p> <p>};<br /> /////////////////////////////////////////////////////////////////////////////<br /> #endif // !defined(AFX_REGISTRY_H__E0610A5D_7166_4D02_9D7E_11AF7CF8E229__INCLUDED_)<br />