regkey.cpp

中国移动短信平台开发接口库

 // RegKey.cpp: implementation of the CRegKey class.
//
//////////////////////////////////////////////////////////////////////

#include "stdafx.h"
//#include "DBLink.h"
#include "RegKey.h"

#ifdef _DEBUG
#undef THIS_FILE
static char THIS_FILE[]=__FILE__;
#define new DEBUG_NEW
#endif

//////////////////////////////////////////////////////////////////////
// Construction/Destruction
//////////////////////////////////////////////////////////////////////

CNewRegKey::CNewRegKey()
{
	m_hKey = NULL ;
}

CNewRegKey::~CNewRegKey()
{
	Close() ;
}

LONG CNewRegKey::Open(HKEY hKeyRoot, LPCTSTR pszPath)
{
	DWORD dw ;
	m_sPath = pszPath ;
	return RegCreateKeyEx(hKeyRoot,pszPath , 0L, NULL, 
		REG_OPTION_NON_VOLATILE,KEY_ALL_ACCESS,NULL,
		&m_hKey , &dw);
}

VOID CNewRegKey::Close()
{
	if(m_hKey)
	{
		RegCloseKey(m_hKey);
		m_hKey = NULL ;
	}
}

LONG CNewRegKey::Write(LPCTSTR pszKey, DWORD dwVal)
{
	ASSERT(m_hKey);
	ASSERT(pszKey);
	return RegSetValueEx(m_hKey ,pszKey , 0L,REG_DWORD,
		(const BYTE*)&dwVal , sizeof(DWORD));
}

LONG CNewRegKey::Write(LPCTSTR pszKey, LPCTSTR pszVal)
{
	ASSERT(m_hKey);
	ASSERT(pszKey);
	ASSERT(pszVal);
	ASSERT(AfxIsValidAddress(pszVal,strlen(pszVal),FALSE));
	return RegSetValueEx(m_hKey,pszKey,0L,REG_SZ,
		(const BYTE*)pszVal,strlen(pszVal)+1);
}

LONG CNewRegKey::Write(LPCTSTR pszKey, const BYTE * pData, DWORD dwLength)
{
	ASSERT(m_hKey);
	ASSERT(pszKey);
	ASSERT(pData && dwLength >0);
	ASSERT(AfxIsValidAddress(pData,dwLength,FALSE));
	return RegSetValueEx(m_hKey,pszKey,0L,REG_BINARY,pData,dwLength);	
}

LONG CNewRegKey::Read(LPCTSTR pszKey, DWORD & dwVal)
{
	ASSERT(m_hKey);
	ASSERT(pszKey);
	DWORD dwType ;
	DWORD dwSize = sizeof(DWORD);
	DWORD dwDest ;
	LONG lRet = RegQueryValueEx(m_hKey,(LPSTR)pszKey,NULL,&dwType,(BYTE*)&dwDest,&dwSize) ;
	if(lRet == ERROR_SUCCESS)
		dwVal = dwDest ;
	return lRet ;
}

LONG CNewRegKey::Read(LPCTSTR pszKey, CString & sVal)
{
	ASSERT(m_hKey);
	ASSERT(pszKey);
	DWORD dwType ;
	DWORD dwSize = 200 ;
	char string[200];
	LONG lReturn = RegQueryValueEx(m_hKey , (LPSTR)pszKey ,NULL,
		&dwType , (BYTE*)string , &dwSize );
	if(lReturn == ERROR_SUCCESS)
		sVal = string  ;
	return lReturn ;
}

LONG CNewRegKey::Read(LPCTSTR pszKey, BYTE * pData, DWORD & dwLength)
{
	ASSERT(m_hKey);
	ASSERT(pszKey);
	DWORD dwType ;
	return RegQueryValueEx(m_hKey , (LPSTR)pszKey ,NULL ,
		&dwType , pData ,&dwLength);
}	

CNewRegKey::operator HKEY() const
{
	return m_hKey;
}

void CNewRegKey::DeleteKey()
{
	if(m_hKey==NULL)
		return ;
	int i;
	char keyName[1000];
	for(i=0;;i++)
	{
		LONG er;
		er=::RegEnumKey(m_hKey,i,keyName,1000);
		if(er!=ERROR_SUCCESS)
			break;
		CNewRegKey subkey;
		if(subkey.Open(m_hKey,keyName)==ERROR_SUCCESS)
		{
			subkey.DeleteKey();
			subkey.Close();
			RegDeleteKey(m_hKey,keyName);
		}
	}
}