/**

 * FreeRDP: A Remote Desktop Protocol Implementation

 * FreeRDP Windows Server

 *

 * Copyright 2012 Marc-Andre Moreau <marcandre.moreau@gmail.com>

 *

 * Licensed under the Apache License, Version 2.0 (the "License");

 * you may not use this file except in compliance with the License.

 * You may obtain a copy of the License at

 *

 *     http://www.apache.org/licenses/LICENSE-2.0

 *

 * Unless required by applicable law or agreed to in writing, software

 * distributed under the License is distributed on an "AS IS" BASIS,

 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

 * See the License for the specific language governing permissions and

 * limitations under the License.

 */

#ifdef HAVE_CONFIG_H

#include "config.h"

#endif

#include <winpr/tchar.h>

#include <winpr/windows.h>

#include "wf_settings.h"

BOOL wf_settings_read_dword(HKEY key, LPCSTR subkey, LPTSTR name, DWORD* value)

{

	HKEY hKey;

	LONG status;

	DWORD dwType;

	DWORD dwSize;

	DWORD dwValue;

	status = RegOpenKeyExA(key, subkey, 0, KEY_READ | KEY_WOW64_64KEY, &hKey);

	if (status == ERROR_SUCCESS)

	{

		dwSize = sizeof(DWORD);

		status = RegQueryValueEx(hKey, name, NULL, &dwType, (BYTE*) &dwValue, &dwSize);

		if (status == ERROR_SUCCESS)

			*value = dwValue;

		RegCloseKey(hKey);

		return (status == ERROR_SUCCESS) ? TRUE : FALSE;

	}

	return FALSE;

}

BOOL wf_settings_read_string_ascii(HKEY key, LPCSTR subkey, LPTSTR name, char** value)

{

	HKEY hKey;

	int length;

	LONG status;

	DWORD dwType;

	DWORD dwSize;

	char* strA;

	TCHAR* strX = NULL;

	status = RegOpenKeyExA(key, subkey, 0, KEY_READ | KEY_WOW64_64KEY, &hKey);

	if (status != ERROR_SUCCESS)

		return FALSE;

	status = RegQueryValueEx(hKey, name, NULL, &dwType, NULL, &dwSize);

	if (status == ERROR_SUCCESS)

	{

		strX = (LPTSTR) malloc(dwSize + sizeof(TCHAR));

		if (!strX)

			return FALSE;

		status = RegQueryValueEx(hKey, name, NULL, &dwType, (BYTE*) strX, &dwSize);

		if (status != ERROR_SUCCESS)

		{

			free(strX);

			RegCloseKey(hKey);

			return FALSE;

		}

	}

	if (strX)

	{

#ifdef UNICODE

		length = WideCharToMultiByte(CP_UTF8, 0, strX, lstrlenW(strX), NULL, 0, NULL, NULL);

		strA = (char*) malloc(length + 1);

		WideCharToMultiByte(CP_UTF8, 0, strX, lstrlenW(strX), strA, length, NULL, NULL);

		strA[length] = '\0';

		free(strX);

#else

		strA = (char*) strX;

#endif

		*value = strA;

		return TRUE;

	}

	return FALSE;

}