[API/MFC] 레지스트리 읽고 쓰는 함수

Registry read, write.

레지스트리를 입력하거나 읽을려면 RegSetValueEX, RegQueryValueEx등을 써야하는데 들어가는 인자 셋팅하는게 귀찮다. 따라서 다음과 같이 쉽게 쓰도록 만들어주면 좋다.

우선 쓰기.

////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
/// Registry에서 string value 쓰는 함수.
///
///		@param[in]	hKey		쓸 값이 존재하는 키, { HKEY_LOCAL_MACHINE, HKEY_CLASSES_ROOT, ...} 등의 값 사용.
///		@param[in]	lpSubKey	쓸 값이 존재하는 subkey path
///		@param[in]	lpValueKey	쓸 값의 이름
///		@param[in]	lpValue		쓸 값
///
///		\return		BOOL; 성공적으로 썼으면 TRUE 실패했으면 FALSE
///
/// \author JaeHyun Baek
BOOL MiscUtils::WriteStringValueInRegistry( HKEY hKey, LPCTSTR lpSubKey, LPCTSTR lpValueKey, LPCTSTR lpValue )
{
	CString strValue = _T("");
	HKEY hSubKey = NULL;

	// open the key
	if ( ::RegOpenKeyEx( hKey, lpSubKey, 0, KEY_ALL_ACCESS, &hSubKey ) == ERROR_SUCCESS )
	{

		DWORD cbSize = (DWORD)strlen(lpValue) + 1;
		BYTE *pBuf = new BYTE [ cbSize ];
		::ZeroMemory( pBuf, cbSize );
		::CopyMemory( pBuf, lpValue, cbSize - 1 );
		::RegSetValueEx( hSubKey, lpValueKey, NULL, REG_SZ, pBuf, cbSize ); 

		// 키 닫기
		::RegCloseKey( hSubKey );

		delete [] pBuf;

		return TRUE;
	}

	return FALSE;
}
////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
/// Registry에서 DWORD value 쓰는 함수.
///
///		@param[in]	hKey		쓸 값이 존재하는 키, { HKEY_LOCAL_MACHINE, HKEY_CLASSES_ROOT, ...} 등의 값 사용.
///		@param[in]	lpSubKey	쓸 값이 존재하는 subkey path
///		@param[in]	lpValueKey	쓸 값의 이름
///		@param[in]	dwValue		쓸 값
///
///		\return		BOOL; 성공적으로 썼으면 TRUE 실패했으면 FALSE
///
/// \author JaeHyun Baek
BOOL MiscUtils::WriteDWORDValueInRegistry( HKEY hKey, LPCTSTR lpSubKey, LPCTSTR lpValueKey, DWORD dwValue )
{
	CString strValue = _T("");
	HKEY hSubKey = NULL;

	// open the key
	if ( ::RegOpenKeyEx( hKey, lpSubKey, 0, KEY_ALL_ACCESS, &hSubKey ) == ERROR_SUCCESS )
	{
		::RegSetValueEx( hSubKey, lpValueKey, NULL, REG_DWORD, (LPBYTE)&dwValue, sizeof(DWORD) ); 


		// 키 닫기
		::RegCloseKey( hSubKey );


		return TRUE;
	}

	return FALSE;
}

그리고 읽기.

////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
/// Registry에서 string value 읽어오는 함수.
/// CWinApp::GetProfileString(..) 함수와 다른 점은, 임의의 key 에 존재하는 값을 모두 읽을 수 있다는 점이 다름.





/// 키가 존재하지 않으면 empty string을 리턴함.
///
///		@param[in]	hKey		읽을 값이 존재하는 키, { HKEY_LOCAL_MACHINE, HKEY_CLASSES_ROOT, ...} 등의 값 사용.
///		@param[in]	lpSubKey	읽을 값이 존재하는 subkey path
///		@param[in]	lpValueKey	읽을 값의 이름
///
///		\return		CString; 해당 레지스트리 위치에서 읽은 값. Can be null string
///
///
CString MiscUtils::ReadStringValueInRegistry( HKEY hKey, LPCTSTR lpSubKey, LPCTSTR lpValueKey )
{
	CString strValue = _T("");
	HKEY hSubKey = NULL;

	// open the key
	if ( ::RegOpenKeyEx( hKey, lpSubKey, 0, KEY_READ, &hSubKey ) == ERROR_SUCCESS )
	{
		DWORD buf_size = 0;

		// 문자열의 크기를 먼저 읽어온다.
		if ( ::RegQueryValueEx( hSubKey, lpValueKey, NULL, NULL, NULL, &buf_size ) == ERROR_SUCCESS )
		{
			// 메모리 할당하고...,
			TCHAR *pBuf = new TCHAR [ buf_size + 1 ];

			// 실제 값을 읽어온다.
			if ( ::RegQueryValueEx( hSubKey, lpValueKey, NULL, NULL, (LPBYTE)pBuf, &buf_size ) == ERROR_SUCCESS )
			{
				pBuf[ buf_size ] = _T('\0');
				strValue = CString( pBuf );
			}

			// to avoid leakage
			delete [] pBuf;
		}

		// 키 닫기
		::RegCloseKey( hSubKey );
	}

	return strValue;
}



////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////////
/// 임의의 레지스트리 키 위치에서 DWORD 값을 읽어오기 위한 함수.
/// CWinApp::GetProfileInt(..) 함수와 다른 점은, 임의의 key 에 존재하는 값을 모두 읽을 수 있다는 점이 다름.





/// 키가 존재하지 않으면 \a dwDefault 를 리턴함.
///
///		@param[in]	hKey		읽을 값이 존재하는 키, { HKEY_LOCAL_MACHINE, HKEY_CLASSES_ROOT, ...} 등의 값 사용.
///		@param[in]	lpSubKey	읽을 값이 존재하는 subkey path
///		@param[in]	lpValueKey	읽을 값의 이름
///		@param[in]	dwDefault	값이 존재하지 않는 경우에 리턴할 값
///
///		\return		DWORD; 해당 레지스트리 위치에서 읽은 값
///
DWORD MiscUtils::ReadDWORDValueInRegistry( HKEY hKey, LPCTSTR lpSubKey, LPCTSTR lpValueKey, DWORD dwDefault /*= -1*/ )
{
	DWORD dwValue = (DWORD)dwDefault;
	HKEY hSubKey = NULL;

	// 키 열기
	if ( ::RegOpenKeyEx( hKey, lpSubKey, 0, KEY_READ, &hSubKey ) == ERROR_SUCCESS )
	{
		DWORD buf_size = sizeof(DWORD);	

		// DWORD 값 읽어오기
		if ( ::RegQueryValueEx( hSubKey, lpValueKey, NULL, NULL, (LPBYTE)&dwValue, &buf_size ) == ERROR_SUCCESS )
		{
			ASSERT( buf_size == sizeof(DWORD) );
		}

		// 키 닫기
		::RegCloseKey( hSubKey );
	}

	return dwValue;
}

이제 실제 사용 하는 것은 다음과 같은 식으로 해주면 된다.

const CString strKey = _T("SOFTWARE\RoterSoft\ExampleApp\Settings");
	MiscUtils::WriteStringValueInRegistry( HKEY_LOCAL_MACHINE, (LPCTSTR) strKey, _T("strTest"), _T("roterpekr"));
	MiscUtils::WriteDWORDValueInRegistry( HKEY_LOCAL_MACHINE, (LPCTSTR) strKey, _T("dwTest"), 123456);

첫 번째 인자인 hKey는 다음을 알아서 넣어 주면 된다.

#define HKEY_CLASSES_ROOT           (( HKEY ) (ULONG_PTR)((LONG)0x80000000) )
#define HKEY_CURRENT_USER           (( HKEY ) (ULONG_PTR)((LONG)0x80000001) )
#define HKEY_LOCAL_MACHINE          (( HKEY ) (ULONG_PTR)((LONG)0x80000002) )
#define HKEY_USERS                  (( HKEY ) (ULONG_PTR)((LONG)0x80000003) )