Milestone 5: deliver embedded RDP sessions and lifecycle hardening

third_party/FreeRDP/server/Windows/wf_settings.c

@@ -0,0 +1,103 @@
+/**
+ * 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.
+ */
+
+#include <freerdp/config.h>
+
+#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, nullptr, &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 = nullptr;
+
+	status = RegOpenKeyExA(key, subkey, 0, KEY_READ | KEY_WOW64_64KEY, &hKey);
+
+	if (status != ERROR_SUCCESS)
+		return FALSE;
+
+	status = RegQueryValueEx(hKey, name, nullptr, &dwType, nullptr, &dwSize);
+
+	if (status == ERROR_SUCCESS)
+	{
+		strX = (LPTSTR)malloc(dwSize + sizeof(TCHAR));
+		if (!strX)
+			return FALSE;
+		status = RegQueryValueEx(hKey, name, nullptr, &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), nullptr, 0, nullptr, nullptr);
+		strA = (char*)malloc(length + 1);
+		WideCharToMultiByte(CP_UTF8, 0, strX, lstrlenW(strX), strA, length, nullptr, nullptr);
+		strA[length] = '\0';
+		free(strX);
+#else
+		strA = (char*)strX;
+#endif
+		*value = strA;
+		return TRUE;
+	}
+
+	return FALSE;
+}