361 lines · 10.1 KB · C++
RawDownload
/*CRegistry classCopyright (C) 2002-2004 Antonio Foranna This program is free software; you can redistribute it and/or modifyit under the terms of the GNU General Public License as published bythe Free Software Foundation.	This program is distributed in the hope that it will be useful,but WITHOUT ANY WARRANTY; without even the implied warranty ofMERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See theGNU General Public License for more details.		You should have received a copy of the GNU General Public Licensealong with this program; if not, write to the Free SoftwareFoundation, Inc., 675 Mass Ave, Cambridge, MA 02139, USA.			The author can be contacted at:ntnfrn_email-temp@yahoo.it*/ #include "CRegistry.h" //************************************************************************************************ CRegistry::CRegistry(){	regKey=NULL;	path=NULL;}//------------------------------------------------------------------------------------------------ CRegistry::~CRegistry(){	Close();}//************************************************************************************************/*void CRegistry::ShowLastError(char *Caption){LPVOID MsgBuf;	if(FormatMessage( 		FORMAT_MESSAGE_ALLOCATE_BUFFER | 		FORMAT_MESSAGE_FROM_SYSTEM | 		FORMAT_MESSAGE_IGNORE_INSERTS,		NULL,		GetLastError(),		MAKELANGID(LANG_NEUTRAL, SUBLANG_DEFAULT), // Default language		(LPTSTR) &MsgBuf,		0,		NULL 	))		MessageBox(NULL, (LPCTSTR)MsgBuf, Caption, MB_OK|MB_ICONSTOP);	if(MsgBuf)		LocalFree(MsgBuf);}*///************************************************************************************************ #define setPath(SubKey) \{ \	if(path) \		free(path); \	path=strdup(SubKey); \} BOOL CRegistry::Open(HKEY hKey, char *SubKey){	if(regKey)		RegCloseKey(regKey);	if(RegOpenKeyEx(hKey, SubKey, 0, KEY_ALL_ACCESS, &regKey)==ERROR_SUCCESS)	{		setPath(SubKey);		return TRUE;	}	else // can't open the key -> error	{		regKey=0;		setPath("");		return FALSE;	}}//************************************************************************************************ BOOL CRegistry::OpenCreate(HKEY hKey, char *SubKey){	if(regKey)		RegCloseKey(regKey);	if(RegOpenKeyEx(hKey, SubKey, 0, KEY_ALL_ACCESS, &regKey)==ERROR_SUCCESS)	{		setPath(SubKey);		return TRUE;	}	else // open failed -> create the key	{	DWORD disp;		RegCreateKeyEx(hKey, SubKey, 0, NULL, REG_OPTION_NON_VOLATILE, KEY_ALL_ACCESS, NULL, &regKey, &disp);		if(disp==REG_CREATED_NEW_KEY) 		{			setPath(SubKey);			return TRUE;		}		else // can't create the key -> error		{			regKey=0;			setPath("");			return FALSE;		}	}}//************************************************************************************************ void CRegistry::Close(){	if(regKey)		RegCloseKey(regKey);	regKey=NULL;	if(path) 		free(path); 	path=NULL;}//************************************************************************************************//************************************************************************************************//************************************************************************************************ void CRegistry::DeleteVal(char *SubKey){	RegDeleteValue(regKey,SubKey);}//************************************************************************************************ void CRegistry::DeleteKey(char *SubKey){	RegDeleteKey(regKey,SubKey);} //************************************************************************************************//************************************************************************************************//************************************************************************************************ void CRegistry::SetBool(char *keyStr, bool val){bool tempVal;DWORD len=sizeof(bool);	if(RegQueryValueEx(regKey, keyStr, NULL, NULL, (BYTE *)&tempVal, &len )!=ERROR_SUCCESS ||		tempVal!=val)		RegSetValueEx(regKey, keyStr, 0, REG_BINARY, (BYTE *)&val, sizeof(bool));}//************************************************************************************************ void CRegistry::SetByte(char *keyStr, BYTE val){DWORD	t=val;DWORD	tempVal;DWORD	len;	if(RegQueryValueEx(regKey, keyStr, NULL, NULL, (BYTE *)&tempVal, &len )!=ERROR_SUCCESS ||		tempVal!=val)		RegSetValueEx(regKey, keyStr, 0, REG_DWORD, (BYTE *)&t, sizeof(DWORD));}//************************************************************************************************ void CRegistry::SetWord(char *keyStr, WORD val){DWORD	t=val;DWORD	tempVal;DWORD	len;	if(RegQueryValueEx(regKey, keyStr, NULL, NULL, (BYTE *)&tempVal, &len )!=ERROR_SUCCESS ||		tempVal!=val)		RegSetValueEx(regKey, keyStr, 0, REG_DWORD, (BYTE *)&t, sizeof(DWORD));}//************************************************************************************************ void CRegistry::SetDword(char *keyStr, DWORD val){DWORD tempVal;DWORD len;	if(RegQueryValueEx(regKey, keyStr, NULL, NULL, (BYTE *)&tempVal, &len )!=ERROR_SUCCESS ||		tempVal!=val)		RegSetValueEx(regKey, keyStr, 0, REG_DWORD, (BYTE *)&val, sizeof(DWORD));}//************************************************************************************************ void CRegistry::SetFloat(char *keyStr, float val){float tempVal;DWORD len;	if(RegQueryValueEx(regKey, keyStr, NULL, NULL, (BYTE *)&tempVal, &len )!=ERROR_SUCCESS ||		tempVal!=val)		RegSetValueEx(regKey, keyStr, 0, REG_BINARY, (BYTE *)&val, sizeof(float));}//************************************************************************************************ void CRegistry::SetStr(char *keyStr, char *valStr){DWORD len;DWORD slen=strlen(valStr)+1; 	if(!valStr || !*valStr)		return; 	if(RegQueryValueEx(regKey, keyStr, NULL, NULL, NULL, &len )!=ERROR_SUCCESS ||		len!=slen)		RegSetValueEx(regKey, keyStr, 0, REG_SZ, (BYTE *)valStr, slen);	else	{	char *tempVal=new char[slen];		if(RegQueryValueEx(regKey, keyStr, NULL, NULL, (BYTE *)tempVal, &len )!=ERROR_SUCCESS ||			strcmpi(tempVal,valStr))			RegSetValueEx(regKey, keyStr, 0, REG_SZ, (BYTE *)valStr, slen);		delete [] tempVal;	}}//************************************************************************************************ void CRegistry::SetValN(char *keyStr, BYTE *addr,  DWORD size){DWORD len;	if(!addr || !size)		return; 	if(RegQueryValueEx(regKey, keyStr, NULL, NULL, NULL, &len )!=ERROR_SUCCESS ||		len!=size)		RegSetValueEx(regKey, keyStr, 0, REG_BINARY, addr, size);	else	{	BYTE *tempVal=new BYTE[size];		if(RegQueryValueEx(regKey, keyStr, NULL, NULL, (BYTE *)tempVal, &len )!=ERROR_SUCCESS ||			memcmp(tempVal,addr,len))			RegSetValueEx(regKey, keyStr, 0, REG_BINARY, addr, size);		delete [] tempVal;	}}   //************************************************************************************************//************************************************************************************************//************************************************************************************************   bool CRegistry::GetSetBool(char *keyStr, bool val){bool tempVal;DWORD len=sizeof(bool); 	if(RegQueryValueEx(regKey, keyStr, NULL, NULL, (BYTE *)&tempVal, &len )!=ERROR_SUCCESS)	{		RegSetValueEx(regKey, keyStr, 0, REG_BINARY, (BYTE *)&val, sizeof(bool));		return val;	}	return tempVal;}//************************************************************************************************ BYTE CRegistry::GetSetByte(char *keyStr, BYTE val){DWORD tempVal;DWORD len=sizeof(DWORD); 	if(RegQueryValueEx(regKey, keyStr, NULL, NULL, (BYTE *)&tempVal, &len )!=ERROR_SUCCESS)	{		tempVal=val;		RegSetValueEx(regKey, keyStr, 0, REG_DWORD, (BYTE *)&tempVal, sizeof(DWORD));		return val;	}	return (BYTE)tempVal;}//************************************************************************************************ WORD CRegistry::GetSetWord(char *keyStr, WORD val){DWORD tempVal;DWORD len=sizeof(DWORD); 	if(RegQueryValueEx(regKey, keyStr, NULL, NULL, (BYTE *)&tempVal, &len )!=ERROR_SUCCESS)	{		tempVal=val;		RegSetValueEx(regKey, keyStr, 0, REG_DWORD, (BYTE *)&tempVal, sizeof(DWORD));		return val;	}	return (WORD)tempVal;}//************************************************************************************************ DWORD CRegistry::GetSetDword(char *keyStr, DWORD val){DWORD tempVal;DWORD len=sizeof(DWORD); 	if(RegQueryValueEx(regKey, keyStr, NULL, NULL, (BYTE *)&tempVal, &len )!=ERROR_SUCCESS)	{		RegSetValueEx(regKey, keyStr, 0, REG_DWORD, (BYTE *)&val, sizeof(DWORD));		return val;	}	return (DWORD)tempVal;}//************************************************************************************************ float CRegistry::GetSetFloat(char *keyStr, float val){float tempVal;DWORD len=sizeof(float); 	if(RegQueryValueEx(regKey, keyStr, NULL, NULL, (BYTE *)&tempVal, &len )!=ERROR_SUCCESS)	{		RegSetValueEx(regKey, keyStr, 0, REG_BINARY, (BYTE *)&val, sizeof(float));		return val;	}	return tempVal;}//************************************************************************************************ char *CRegistry::GetSetStr(char *keyStr, char *String){long	retVal;DWORD	Len;char	*dest=NULL;		if((retVal=RegQueryValueEx(regKey , keyStr , NULL , NULL, NULL, &Len))==ERROR_SUCCESS)		if(dest=(char *)malloc(Len+1))			retVal=RegQueryValueEx(regKey , keyStr , NULL , NULL, (BYTE *)dest , &Len);	if(retVal!=ERROR_SUCCESS)	{		if(dest)			free(dest);		if(!String)			return NULL; 		Len=strlen(String)+1;		if(!(dest=strdup(String)))			return NULL;		RegSetValueEx(regKey , keyStr , NULL , REG_SZ , (BYTE *)dest , Len);	}	return dest;}// ----------------------------------------------------------------------------------------------- int CRegistry::GetSetValN(char *keyStr, BYTE *defData, DWORD defSize, BYTE **dest){long	retVal;DWORD	size; 	*dest=NULL;	if((retVal=RegQueryValueEx(regKey , keyStr , NULL , NULL, NULL, &size))==ERROR_SUCCESS)		if(*dest=(BYTE *)malloc(size+1))			retVal=RegQueryValueEx(regKey , keyStr , NULL , NULL, (BYTE *)*dest , &size);	if(retVal!=ERROR_SUCCESS)	{		if(*dest)			free(*dest);		if(!defData)			return 0; 		size=defSize;		if(!(*dest=(BYTE *)malloc(size)))			return 0;		memcpy(*dest,defData,size);		RegSetValueEx(regKey , keyStr , NULL , REG_BINARY , (BYTE *)*dest , size);	}	return size;}

Keyboard shortcuts

General

Show keyboard shortcuts
?
Focus the current search
/

Issues and pull requests

Open filters
f

Issues, pull requests and Inbox

Select next item
j↓
Select previous item
k↑
Open selected item
Enter
Dismiss filters or clear selection
Esc

Repository code

Go to file
t