注冊表操作類--Like RegistryKey in .NET

.NET 里面操作注冊表使用RegistryKey類,我用C++也實現(xiàn)一個RegistryKey類,發(fā)現(xiàn)確實很好用，不敢獨享，讓大家也看看也體驗一下。

/********************************************************************

Module: Registry.h

Author: Yangjun D.

Created: 9/5/2005 13:24

Purpose: Wrapper registry operation like RegistryKey class in .NET

History:

*********************************************************************/

#pragma once

#include <string>

#include <vector>

using namespace std;

namespace C2217

{

namespace Win32

{

class RegistryKey

{

public:

RegistryKey(HKEY key=NULL, string name= "");

virtual ~RegistryKey(void);

void Close();

RegistryKey OpenSubKey(const string &keyName, bool writable= false) const ;

RegistryKey CreateSubKey(const string &keyName) const ;

void DelSubKey(const string &keyName) const ;

void SetValue(const string &valueName, const string &newStrValue) const;

void SetValue(const string &valueName, int newIntValue) const ;

void DelValue(const string &valueName) const ;

string GetStrValue(const string &keyName) const ;

int GetIntValue(const string &keyName) const ;

const string &get_name() const { return m_name; }

void Backup(const string &file) const ;

void Restore(const string &file) const;

void GetSubKeyNames(vector<string> & keyNames) const ;

void GetValueNames(vector<string> &valueNames) const ;

void Flush() const;

private:

HKEY m_key;

string m_name;

};

class Registry

{

public:

const static RegistryKey LocalMachine;

const static RegistryKey ClassesRoot;

const static RegistryKey CurrentConfig;

const static RegistryKey CurrentUser;

const static RegistryKey DynData;

const static RegistryKey PerformanceData;

const static RegistryKey Users;

};

}

/********************************************************************

Module: registry.cpp

Author: Yangjun D.

Created: 9/5/2005 13:24

Purpose: Implement the RegistryKey class

History:

*********************************************************************/

#include "StdAfx.h"

#include ".\registry.h"

#include "winexception.h"

#include "winex.h"

using namespace C2217::StdLib;

namespace C2217

{

namespace Win32

{

//Class RegistryKey

RegistryKey::RegistryKey(HKEY key/*=NULL*/, string name/*= ""*/)

{

m_key = key;

m_name = name;

}

RegistryKey::~RegistryKey(void)

{

//Close();

}

RegistryKey RegistryKey::OpenSubKey(const string &keyName,bool writable/*= false*/) const

{

DWORD dwAccess = KEY_READ;

if(writable)

{

dwAccess |=KEY_WRITE;

}

RegistryKey openKey;

int result =::RegOpenKeyEx(m_key,keyName.c_str(),0,dwAccess, &openKey.m_key);

if(result != ERROR_SUCCESS)

{

THROW_EX_CODE(result);

}

if (!m_name.empty())

{

openKey.m_name = keyName;

}

else

{

openKey.m_name = m_name +"\\" + keyName;

}

return openKey;

}

RegistryKey RegistryKey::CreateSubKey(const string &keyName) const

{

RegistryKey newKey;

int result = ::RegCreateKeyEx(

m_key,

keyName.c_str(),

NULL,

REG_OPTION_NON_VOLATILE,

KEY_ALL_ACCESS,

NULL,

&newKey.m_key,

NULL

);

if(result != ERROR_SUCCESS ) {

THROW_EX_CODE(result);

}

if (!m_name.empty()) {

newKey.m_name = keyName;

}

else {

newKey.m_name = m_name +"\\" + keyName;

}

return newKey;

}

void RegistryKey::Close()

{

if(NULL!=m_key)

{

::RegCloseKey(m_key);

}

LPBYTE STRING_To_LPBYTE(string str)

{

LPBYTE lpb=new BYTE[str.length()+1];

for(size_t i=0;i<str.length();i++)

lpb[i]=str[i];

lpb[str.length()]=0;

return lpb;

}

void RegistryKey::SetValue(const string &valueName,const string &newStrValue) const

{

LPBYTE lpData = STRING_To_LPBYTE(newStrValue);

int result = ::RegSetValueEx(m_key, valueName.c_str(),0,REG_SZ,

lpData,

(int)newStrValue.length()+1);

delete[] lpData;

if(result != ERROR_SUCCESS)

{

THROW_EX_CODE(result);

}

void RegistryKey::SetValue( const string &valueName,int newIntValue) const

{

int result = ::RegSetValueEx(m_key, valueName.c_str() ,

0L,REG_DWORD,(const BYTE *) &newIntValue,

sizeof(int));

if(result != ERROR_SUCCESS)

{

THROW_EX_CODE(result);

}

string RegistryKey::GetStrValue(const string &valueName) const

{

DWORD dwType;

static TCHAR szValue[2048] = {0};

DWORD dwSize=sizeof(szValue);

int result = RegQueryValueEx(m_key, valueName.c_str(),

NULL,&dwType,(BYTE *)&szValue,&dwSize);

if(result != ERROR_SUCCESS)

{

THROW_EX_CODE(result);

}

return szValue;

}

int RegistryKey::GetIntValue(const string &valueName) const

{

DWORD dwType;

DWORD dwSize=sizeof(DWORD);

DWORD dwDest;

int result = RegQueryValueEx(m_key, valueName.c_str(),

NULL,&dwType,(BYTE *)&dwDest,&dwSize);

if(result != ERROR_SUCCESS)

{

THROW_EX_CODE(result);

}

return (int)dwDest;

}

void RegistryKey::DelSubKey(const string &keyName) const

{

int result = ::RegDeleteKey(m_key,keyName.c_str());

if(result != ERROR_SUCCESS)

{

THROW_EX_CODE(result);

}

void RegistryKey::DelValue(const string &valueName) const

{

int result = ::RegDeleteValue(m_key,valueName.c_str());

if(result != ERROR_SUCCESS)

{

THROW_EX_CODE(result);

}

void RegistryKey::Backup(const string &file) const

{

int result = RegSaveKey(m_key, file.c_str(),NULL);

if(result != ERROR_SUCCESS)

{

THROW_EX_CODE(result);

}

void RegistryKey::Restore(const string &file) const

{

int result = ::RegRestoreKey(m_key, file.c_str() ,REG_WHOLE_HIVE_VOLATILE);

if(result != ERROR_SUCCESS)

{

THROW_EX_CODE(result);

}

const static int ITEM_MAX_COUNT = 1024;

void RegistryKey::GetSubKeyNames(vector<string> & keyNames) const

{

int result=0;

TCHAR keyName [256] ={0};

DWORD nameSize;

for(int i=0; i< ITEM_MAX_COUNT; i++)

{

nameSize= sizeof(keyName);

memset(keyName, 0 , nameSize);

result = ::RegEnumKeyEx( m_key, i, keyName, &nameSize, NULL,NULL,NULL,NULL);

if(ERROR_SUCCESS == result)

{

keyNames.push_back(keyName);

}

else if(ERROR_NO_MORE_ITEMS == result)

{

break;

}

else

{

THROW_EX_CODE(result);

}

void RegistryKey::Flush() const

{

int result = ::RegFlushKey(m_key);

if(result != ERROR_SUCCESS)

{

THROW_EX_CODE(result);

}

const RegistryKey Registry::LocalMachine(HKEY_LOCAL_MACHINE);

const RegistryKey Registry::ClassesRoot(HKEY_CLASSES_ROOT);

const RegistryKey Registry::CurrentConfig(HKEY_CURRENT_CONFIG);

const RegistryKey Registry::CurrentUser(HKEY_CURRENT_USER);

const RegistryKey Registry::DynData(HKEY_DYN_DATA);

const RegistryKey Registry::PerformanceData(HKEY_PERFORMANCE_DATA);

const RegistryKey Registry::Users(HKEY_USERS);

}

操作方法跟.NET里面的RegistryKey 幾乎相同，就不多說了。

posted on 2005-09-21 09:20 天下無雙閱讀(1250) 評論(0) 編輯收藏引用

只有注冊用戶登錄后才能發(fā)表評論。
【推薦】100%開源！大型工業(yè)跨平臺軟件C++源碼提供，建模，組態(tài)！



網站導航: 博客園 IT新聞 BlogJava 博問 Chat2DB 管理

常用鏈接

留言簿(15)

隨筆分類

隨筆檔案

相冊

我的其它領域Blog

搜索

積分與排名

最新評論

閱讀排行榜

評論排行榜