From 35cb729f317919d94523cdf0be04828d2c1f3ce1 Mon Sep 17 00:00:00 2001
From: Armin Novak <armin.novak@thincast.com>
Date: Fri, 23 Aug 2019 09:57:26 +0200
Subject: [PATCH 1/6] Added drivestoredirect option value parser

---
 client/common/file.c | 110 +++++++++++++++++++++++++++++++++++++++++--
 1 file changed, 107 insertions(+), 3 deletions(-)

diff --git a/client/common/file.c b/client/common/file.c
index a1f0a487e..16a3eea46 100644
--- a/client/common/file.c
+++ b/client/common/file.c
@@ -44,12 +44,14 @@
 
 #include <winpr/wtypes.h>
 #include <winpr/crt.h>
+#include <winpr/path.h>
 #include <freerdp/log.h>
 #define TAG CLIENT_TAG("common")
 
 /*#define DEBUG_CLIENT_FILE	1*/
 
-static BYTE BOM_UTF16_LE[2] = { 0xFF, 0xFE };
+static const BYTE BOM_UTF16_LE[2] = { 0xFF, 0xFE };
+static const char DynamicDrives[] = "DynamicDrives";
 
 #define INVALID_INTEGER_VALUE 0xFFFFFFFF
 
@@ -1045,6 +1047,62 @@ size_t freerdp_client_write_rdp_file_buffer(const rdpFile* file, char* buffer, s
 	return totalSize;
 }
 
+static BOOL freerdp_client_add_drive(rdpSettings* settings, const char* path, const char* name)
+{
+	RDPDR_DRIVE* drive;
+
+	drive = (RDPDR_DRIVE*) calloc(1, sizeof(RDPDR_DRIVE));
+
+	if (!drive)
+		return FALSE;
+
+	drive->Type = RDPDR_DTYP_FILESYSTEM;
+
+	if (name)
+	{
+		/* Path was entered as secondary argument, swap */
+		if (PathFileExistsA(name))
+		{
+			const char* tmp = path;
+			path = name;
+			name = tmp;
+		}
+	}
+
+	if (name)
+	{
+		if (!(drive->Name = _strdup(name)))
+			goto fail;
+	}
+
+	if (!path)
+		goto fail;
+	else
+	{
+		const BOOL isPath = PathFileExistsA(path);
+		const BOOL isSpecial = (strncmp(path, "*", 2) == 0) ||
+									   (strncmp(path, DynamicDrives, sizeof(DynamicDrives)) == 0) ||
+									   (strncmp(path, "%", 2) == 0) ? TRUE : FALSE;
+
+		if (isSpecial && name)
+			goto fail;
+
+		if ((!isPath && !isSpecial) || !(drive->Path = _strdup(path)))
+			goto fail;
+	}
+
+	if (!freerdp_device_collection_add(settings, (RDPDR_DEVICE*) drive))
+		goto fail;
+
+	return TRUE;
+
+	fail:
+		free(drive->Path);
+		free(drive->Name);
+		free(drive);
+		return FALSE;
+}
+
 BOOL freerdp_client_populate_settings_from_rdp_file(rdpFile* file, rdpSettings* settings)
 {
 	if (~((size_t)file->Domain))
@@ -1556,8 +1614,53 @@ BOOL freerdp_client_populate_settings_from_rdp_file(rdpFile* file, rdpSettings*
 		 */
 		const BOOL empty = !file->DrivesToRedirect || (strlen(file->DrivesToRedirect) == 0);
 
-		if (!freerdp_settings_set_bool(settings, FreeRDP_RedirectDrives, !empty))
-			return FALSE;
+		if (!empty)
+		{
+			char* value;
+			char* tok;
+			char* context = NULL;
+
+			value = _strdup(file->DrivesToRedirect);
+			if (!value)
+				return FALSE;
+
+			tok = strtok_s(value, ";", &context);
+			if (!tok)
+			{
+				free(value);
+				return FALSE;
+			}
+
+			while(tok)
+			{
+				/* Syntax: Comma seperated list of the following entries:
+				 * '*'              ... Redirect all drives, including hotplug
+				 * 'DynamicDrives'  ... hotplug
+				 * <label>(<path>)  ... One or more paths to redirect.
+				 * <path>(<label>)  ... One or more paths to redirect.
+				 * <path>           ... One or more paths to redirect.
+				 */
+				/* TODO: Need to properly escape labels and paths */
+				const char* name = NULL;
+				const char* drive = tok;
+				char* start = strtok(tok, "(");
+				char* end = strtok(NULL, ")");
+				if (end)
+					name = end;
+
+				if (!freerdp_client_add_drive(settings, drive, name))
+				{
+					free(value);
+					return FALSE;
+				}
+
+				tok = strtok_s(NULL, ";", &context);
+			}
+			free(value);
+
+			if (!freerdp_settings_set_bool(settings, FreeRDP_DeviceRedirection, TRUE))
+				return FALSE;
+		}
 	}
 
 	if (~file->KeyboardHook)
@@ -1597,6 +1700,7 @@ static rdpFileLine* freerdp_client_rdp_file_find_line_index(rdpFile* file, SSIZE
 	line = &(file->lines[index]);
 	return line;
 }
+
 static rdpFileLine* freerdp_client_rdp_file_find_line_by_name(rdpFile* file, const char* name)
 {
 	size_t index;

From 935ed36e68d0d63234f96f824cb766b07c0e9f63 Mon Sep 17 00:00:00 2001
From: Armin Novak <armin.novak@thincast.com>
Date: Fri, 23 Aug 2019 10:08:59 +0200
Subject: [PATCH 2/6] Add dynamicDrives hotplug option.

---
 channels/rdpdr/client/rdpdr_main.c | 13 ++++++++-----
 1 file changed, 8 insertions(+), 5 deletions(-)

diff --git a/channels/rdpdr/client/rdpdr_main.c b/channels/rdpdr/client/rdpdr_main.c
index 305d7d5dd..fe13dc756 100644
--- a/channels/rdpdr/client/rdpdr_main.c
+++ b/channels/rdpdr/client/rdpdr_main.c
@@ -764,14 +764,14 @@ static UINT handle_hotplug(rdpdrPlugin* rdpdr)
 	char* line;
 	char* word;
 	unsigned int wlen;
-	hotplug_dev dev_array[MAX_USB_DEVICES];
+	hotplug_dev dev_array[MAX_USB_DEVICES] = { 0 };
 	int i, j;
 	int size = 0;
 	int count;
 	ULONG_PTR* keys = NULL;
 	UINT32 ids[1];
 	UINT error = 0;
-	memset(dev_array, 0, sizeof(dev_array));
+
 	f = fopen("/proc/mounts", "r");
 
 	if (f == NULL)
@@ -1050,11 +1050,14 @@ static UINT rdpdr_process_connect(rdpdrPlugin* rdpdr)
 
 		if (device->Type == RDPDR_DTYP_FILESYSTEM)
 		{
+			const char DynamicDrives[] = "DynamicDrives";
 			RDPDR_DRIVE* drive = (RDPDR_DRIVE*)device;
-
-			if (drive->Path && (strcmp(drive->Path, "*") == 0))
+			BOOL hotplugAll = strncmp(drive->Path, "*", 2) == 0;
+			BOOL hotplugLater = strncmp(drive->Path, DynamicDrives, sizeof(DynamicDrives)) == 0;
+			if (drive->Path && (hotplugAll || hotplugLater))
 			{
-				first_hotplug(rdpdr);
+				if (hotplugAll)
+					first_hotplug(rdpdr);
 #ifndef _WIN32
 
 				if (!(rdpdr->stopEvent = CreateEvent(NULL, TRUE, FALSE, NULL)))

From 0af53bff9b69e4a008f3b7fe3af248b4b370c00c Mon Sep 17 00:00:00 2001
From: Armin Novak <armin.novak@thincast.com>
Date: Fri, 23 Aug 2019 10:13:59 +0200
Subject: [PATCH 3/6] Allow DynamicDrives as special option for /drive

---
 client/common/cmdline.c | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/client/common/cmdline.c b/client/common/cmdline.c
index 0801c8921..ae7d0400a 100644
--- a/client/common/cmdline.c
+++ b/client/common/cmdline.c
@@ -359,8 +359,10 @@ BOOL freerdp_client_add_device_channel(rdpSettings* settings, int count,
 
 		if (count > 2)
 		{
+			const char DynamicDrives[] = "DynamicDrives";
 			const BOOL isPath = PathFileExistsA(params[2]);
 			const BOOL isSpecial = (strncmp(params[2], "*", 2) == 0) ||
+									(strncmp(params[2], DynamicDrives, sizeof(DynamicDrives)) == 0) ||
 			                       (strncmp(params[2], "%", 2) == 0) ? TRUE : FALSE;
 
 			if ((!isPath && !isSpecial) || !(drive->Path = _strdup(params[2])))

From d8bf05367b7e12789f48aa88b389e07071676142 Mon Sep 17 00:00:00 2001
From: Armin Novak <armin.novak@thincast.com>
Date: Fri, 23 Aug 2019 11:18:32 +0200
Subject: [PATCH 4/6] Added support for drivestoreredirect:s:<path>(<path>)
 syntax

---
 client/common/file.c | 37 ++++++++++++++++++++++++++++++++-----
 1 file changed, 32 insertions(+), 5 deletions(-)

diff --git a/client/common/file.c b/client/common/file.c
index 16a3eea46..d537d017a 100644
--- a/client/common/file.c
+++ b/client/common/file.c
@@ -1047,6 +1047,25 @@ size_t freerdp_client_write_rdp_file_buffer(const rdpFile* file, char* buffer, s
 	return totalSize;
 }
 
+static BOOL freerdp_path_valid(const char* path, BOOL* special)
+{
+	BOOL isPath = FALSE;
+	BOOL isSpecial;
+	if (!path)
+		return FALSE;
+
+	isSpecial = (strncmp(path, "*", 2) == 0) ||
+								   (strncmp(path, DynamicDrives, sizeof(DynamicDrives)) == 0) ||
+								   (strncmp(path, "%", 2) == 0) ? TRUE : FALSE;
+	if (!isSpecial)
+		isPath = PathFileExistsA(path);
+
+	if (special)
+		*special = isSpecial;
+
+	return isSpecial || isPath;
+}
+
 static BOOL freerdp_client_add_drive(rdpSettings* settings, const char* path, const char* name)
 {
 	RDPDR_DRIVE* drive;
@@ -1079,10 +1098,8 @@ static BOOL freerdp_client_add_drive(rdpSettings* settings, const char* path, co
 		goto fail;
 	else
 	{
-		const BOOL isPath = PathFileExistsA(path);
-		const BOOL isSpecial = (strncmp(path, "*", 2) == 0) ||
-									   (strncmp(path, DynamicDrives, sizeof(DynamicDrives)) == 0) ||
-									   (strncmp(path, "%", 2) == 0) ? TRUE : FALSE;
+		BOOL isSpecial = FALSE;
+		BOOL isPath = freerdp_path_valid(path, &isSpecial);
 
 		if (isSpecial && name)
 			goto fail;
@@ -1641,6 +1658,7 @@ BOOL freerdp_client_populate_settings_from_rdp_file(rdpFile* file, rdpSettings*
 				 * <path>           ... One or more paths to redirect.
 				 */
 				/* TODO: Need to properly escape labels and paths */
+				BOOL success;
 				const char* name = NULL;
 				const char* drive = tok;
 				char* start = strtok(tok, "(");
@@ -1648,7 +1666,16 @@ BOOL freerdp_client_populate_settings_from_rdp_file(rdpFile* file, rdpSettings*
 				if (end)
 					name = end;
 
-				if (!freerdp_client_add_drive(settings, drive, name))
+				if (freerdp_path_valid(name, NULL) && freerdp_path_valid(drive, NULL))
+				{
+					success = freerdp_client_add_drive(settings, name, NULL);
+					if (success)
+						success = freerdp_client_add_drive(settings, drive, NULL);
+				}
+				else
+					success = freerdp_client_add_drive(settings, drive, name);
+
+				if (!success)
 				{
 					free(value);
 					return FALSE;

From fdf17152138b7fe36363b77a6310cc3528cb1bc5 Mon Sep 17 00:00:00 2001
From: Armin Novak <armin.novak@thincast.com>
Date: Fri, 23 Aug 2019 11:47:31 +0200
Subject: [PATCH 5/6] Unified DriveStoreRedirect parsing.

---
 channels/drive/client/drive_main.c |   6 +
 client/common/cmdline.c            | 193 ++++++++++++++++++++++-------
 client/common/file.c               | 142 +--------------------
 3 files changed, 159 insertions(+), 182 deletions(-)

diff --git a/channels/drive/client/drive_main.c b/channels/drive/client/drive_main.c
index df361c0b1..d0fbee9d9 100644
--- a/channels/drive/client/drive_main.c
+++ b/channels/drive/client/drive_main.c
@@ -884,6 +884,12 @@ static UINT drive_register_drive_path(PDEVICE_SERVICE_ENTRY_POINTS pEntryPoints,
 	DRIVE_DEVICE* drive;
 	UINT error;
 
+	if (!pEntryPoints || !name || !path)
+	{
+		WLog_ERR(TAG, "[%s] Invalid parameters: pEntryPoints=%p, name=%p, path=%p", pEntryPoints, name, path);
+		return ERROR_INVALID_PARAMETER;
+	}
+
 	if (name[0] && path[0])
 	{
 		size_t pathLength = strnlen(path, MAX_PATH);
diff --git a/client/common/cmdline.c b/client/common/cmdline.c
index ae7d0400a..ada3ab0ae 100644
--- a/client/common/cmdline.c
+++ b/client/common/cmdline.c
@@ -48,6 +48,83 @@
 #include <freerdp/log.h>
 #define TAG CLIENT_TAG("common.cmdline")
 
+static BOOL freerdp_path_valid(const char* path, BOOL* special)
+{
+	const char DynamicDrives[] = "DynamicDrives";
+	BOOL isPath = FALSE;
+	BOOL isSpecial;
+	if (!path)
+		return FALSE;
+
+	isSpecial = (strncmp(path, "*", 2) == 0) ||
+						(strncmp(path, DynamicDrives, sizeof(DynamicDrives)) == 0) ||
+						(strncmp(path, "%", 2) == 0) ? TRUE : FALSE;
+	if (!isSpecial)
+		isPath = PathFileExistsA(path);
+
+	if (special)
+		*special = isSpecial;
+
+	return isSpecial || isPath;
+}
+
+static BOOL freerdp_client_add_drive(rdpSettings* settings, const char* path, const char* name)
+{
+	RDPDR_DRIVE* drive;
+
+	drive = (RDPDR_DRIVE*) calloc(1, sizeof(RDPDR_DRIVE));
+
+	if (!drive)
+		return FALSE;
+
+	drive->Type = RDPDR_DTYP_FILESYSTEM;
+
+	if (name)
+	{
+		/* Path was entered as secondary argument, swap */
+		if (PathFileExistsA(name))
+		{
+			const char* tmp = path;
+			path = name;
+			name = tmp;
+		}
+	}
+
+	if (name)
+	{
+		if (!(drive->Name = _strdup(name)))
+			goto fail;
+	}
+	else /* We need a name to send to the server. */
+		if (!(drive->Name = _strdup(path)))
+			goto fail;
+
+	if (!path)
+		goto fail;
+	else
+	{
+		BOOL isSpecial = FALSE;
+		BOOL isPath = freerdp_path_valid(path, &isSpecial);
+
+		if (isSpecial && name)
+			goto fail;
+
+		if ((!isPath && !isSpecial) || !(drive->Path = _strdup(path)))
+			goto fail;
+	}
+
+	if (!freerdp_device_collection_add(settings, (RDPDR_DEVICE*) drive))
+		goto fail;
+
+	return TRUE;
+
+fail:
+	free(drive->Path);
+	free(drive->Name);
+	free(drive);
+	return FALSE;
+}
+
 static BOOL copy_value(const char* value, char** dst)
 {
 	if (!dst || !value)
@@ -335,53 +412,17 @@ BOOL freerdp_client_add_device_channel(rdpSettings* settings, int count,
 {
 	if (strcmp(params[0], "drive") == 0)
 	{
-		RDPDR_DRIVE* drive;
-
-		if (count < 3)
+		BOOL rc;
+		if (count < 2)
 			return FALSE;
 
 		settings->DeviceRedirection = TRUE;
-		drive = (RDPDR_DRIVE*) calloc(1, sizeof(RDPDR_DRIVE));
+		if (count < 3)
+			rc = freerdp_client_add_drive(settings, params[1], NULL);
+		else
+			rc = freerdp_client_add_drive(settings, params[2], params[1]);
 
-		if (!drive)
-			return FALSE;
-
-		drive->Type = RDPDR_DTYP_FILESYSTEM;
-
-		if (count > 1)
-		{
-			if (!(drive->Name = _strdup(params[1])))
-			{
-				free(drive);
-				return FALSE;
-			}
-		}
-
-		if (count > 2)
-		{
-			const char DynamicDrives[] = "DynamicDrives";
-			const BOOL isPath = PathFileExistsA(params[2]);
-			const BOOL isSpecial = (strncmp(params[2], "*", 2) == 0) ||
-									(strncmp(params[2], DynamicDrives, sizeof(DynamicDrives)) == 0) ||
-			                       (strncmp(params[2], "%", 2) == 0) ? TRUE : FALSE;
-
-			if ((!isPath && !isSpecial) || !(drive->Path = _strdup(params[2])))
-			{
-				free(drive->Name);
-				free(drive);
-				return FALSE;
-			}
-		}
-
-		if (!freerdp_device_collection_add(settings, (RDPDR_DEVICE*) drive))
-		{
-			free(drive->Path);
-			free(drive->Name);
-			free(drive);
-			return FALSE;
-		}
-
-		return TRUE;
+		return rc;
 	}
 	else if (strcmp(params[0], "printer") == 0)
 	{
@@ -3175,6 +3216,72 @@ BOOL freerdp_client_load_addins(rdpChannels* channels, rdpSettings* settings)
 		    TRUE; /* these RDP8 features require rdpdr to be registered */
 	}
 
+	if (settings->DrivesToRedirect && (strlen(settings->DrivesToRedirect) != 0))
+	{
+		/*
+		 * Drives to redirect:
+		 *
+		 * Very similar to DevicesToRedirect, but can contain a
+		 * comma-separated list of drive letters to redirect.
+		 */
+		char* value;
+		char* tok;
+		char* context = NULL;
+
+		value = _strdup(settings->DrivesToRedirect);
+		if (!value)
+			return FALSE;
+
+		tok = strtok_s(value, ";", &context);
+		if (!tok)
+		{
+			free(value);
+			return FALSE;
+		}
+
+		while(tok)
+		{
+			/* Syntax: Comma seperated list of the following entries:
+			 * '*'              ... Redirect all drives, including hotplug
+			 * 'DynamicDrives'  ... hotplug
+			 * <label>(<path>)  ... One or more paths to redirect.
+			 * <path>(<label>)  ... One or more paths to redirect.
+			 * <path>           ... One or more paths to redirect.
+			 */
+			/* TODO: Need to properly escape labels and paths */
+			BOOL success;
+			const char* name = NULL;
+			const char* drive = tok;
+			char* start = strtok(tok, "(");
+			char* end = strtok(NULL, ")");
+			if (end)
+				name = end;
+
+			if (freerdp_path_valid(name, NULL) && freerdp_path_valid(drive, NULL))
+			{
+				success = freerdp_client_add_drive(settings, name, NULL);
+				if (success)
+					success = freerdp_client_add_drive(settings, drive, NULL);
+			}
+			else
+				success = freerdp_client_add_drive(settings, drive, name);
+
+			if (!success)
+			{
+				free(value);
+				return FALSE;
+			}
+
+			tok = strtok_s(NULL, ";", &context);
+		}
+		free(value);
+
+		if (!freerdp_settings_set_bool(settings, FreeRDP_DeviceRedirection, TRUE))
+			return FALSE;
+
+		settings->DeviceRedirection = TRUE;
+	}
+
 	if (settings->RedirectDrives || settings->RedirectHomeDrive
 	    || settings->RedirectSerialPorts
 	    || settings->RedirectSmartCards || settings->RedirectPrinters)
diff --git a/client/common/file.c b/client/common/file.c
index d537d017a..f1a51dd45 100644
--- a/client/common/file.c
+++ b/client/common/file.c
@@ -51,7 +51,6 @@
 /*#define DEBUG_CLIENT_FILE	1*/
 
 static const BYTE BOM_UTF16_LE[2] = { 0xFF, 0xFE };
-static const char DynamicDrives[] = "DynamicDrives";
 
 #define INVALID_INTEGER_VALUE 0xFFFFFFFF
 
@@ -786,6 +785,7 @@ BOOL freerdp_client_populate_rdp_file_from_settings(rdpFile* file, const rdpSett
 	SETTING_MODIFIED_SET_STRING(file->AlternateShell, settings, AlternateShell);
 	SETTING_MODIFIED_SET_STRING(file->ShellWorkingDirectory, settings, ShellWorkingDirectory);
 	SETTING_MODIFIED_SET(file->ConnectionType, settings, ConnectionType);
+	SETTING_MODIFIED_SET_STRING(file->DrivesToRedirect, settings, DrivesToRedirect);
 
 	if (SETTING_MODIFIED(settings, AudioPlayback) || SETTING_MODIFIED(settings, RemoteConsoleAudio))
 	{
@@ -1047,79 +1047,6 @@ size_t freerdp_client_write_rdp_file_buffer(const rdpFile* file, char* buffer, s
 	return totalSize;
 }
 
-static BOOL freerdp_path_valid(const char* path, BOOL* special)
-{
-	BOOL isPath = FALSE;
-	BOOL isSpecial;
-	if (!path)
-		return FALSE;
-
-	isSpecial = (strncmp(path, "*", 2) == 0) ||
-								   (strncmp(path, DynamicDrives, sizeof(DynamicDrives)) == 0) ||
-								   (strncmp(path, "%", 2) == 0) ? TRUE : FALSE;
-	if (!isSpecial)
-		isPath = PathFileExistsA(path);
-
-	if (special)
-		*special = isSpecial;
-
-	return isSpecial || isPath;
-}
-
-static BOOL freerdp_client_add_drive(rdpSettings* settings, const char* path, const char* name)
-{
-	RDPDR_DRIVE* drive;
-
-	drive = (RDPDR_DRIVE*) calloc(1, sizeof(RDPDR_DRIVE));
-
-	if (!drive)
-		return FALSE;
-
-	drive->Type = RDPDR_DTYP_FILESYSTEM;
-
-	if (name)
-	{
-		/* Path was entered as secondary argument, swap */
-		if (PathFileExistsA(name))
-		{
-			const char* tmp = path;
-			path = name;
-			name = tmp;
-		}
-	}
-
-	if (name)
-	{
-		if (!(drive->Name = _strdup(name)))
-			goto fail;
-	}
-
-	if (!path)
-		goto fail;
-	else
-	{
-		BOOL isSpecial = FALSE;
-		BOOL isPath = freerdp_path_valid(path, &isSpecial);
-
-		if (isSpecial && name)
-			goto fail;
-
-		if ((!isPath && !isSpecial) || !(drive->Path = _strdup(path)))
-			goto fail;
-	}
-
-	if (!freerdp_device_collection_add(settings, (RDPDR_DEVICE*) drive))
-		goto fail;
-
-	return TRUE;
-
-	fail:
-		free(drive->Path);
-		free(drive->Name);
-		free(drive);
-		return FALSE;
-}
-
 BOOL freerdp_client_populate_settings_from_rdp_file(rdpFile* file, rdpSettings* settings)
 {
 	if (~((size_t)file->Domain))
@@ -1623,71 +1550,8 @@ BOOL freerdp_client_populate_settings_from_rdp_file(rdpFile* file, rdpSettings*
 
 	if (~((size_t)file->DrivesToRedirect))
 	{
-		/*
-		 * Drives to redirect:
-		 *
-		 * Very similar to DevicesToRedirect, but can contain a
-		 * comma-separated list of drive letters to redirect.
-		 */
-		const BOOL empty = !file->DrivesToRedirect || (strlen(file->DrivesToRedirect) == 0);
-
-		if (!empty)
-		{
-			char* value;
-			char* tok;
-			char* context = NULL;
-
-			value = _strdup(file->DrivesToRedirect);
-			if (!value)
-				return FALSE;
-
-			tok = strtok_s(value, ";", &context);
-			if (!tok)
-			{
-				free(value);
-				return FALSE;
-			}
-
-			while(tok)
-			{
-				/* Syntax: Comma seperated list of the following entries:
-				 * '*'              ... Redirect all drives, including hotplug
-				 * 'DynamicDrives'  ... hotplug
-				 * <label>(<path>)  ... One or more paths to redirect.
-				 * <path>(<label>)  ... One or more paths to redirect.
-				 * <path>           ... One or more paths to redirect.
-				 */
-				/* TODO: Need to properly escape labels and paths */
-				BOOL success;
-				const char* name = NULL;
-				const char* drive = tok;
-				char* start = strtok(tok, "(");
-				char* end = strtok(NULL, ")");
-				if (end)
-					name = end;
-
-				if (freerdp_path_valid(name, NULL) && freerdp_path_valid(drive, NULL))
-				{
-					success = freerdp_client_add_drive(settings, name, NULL);
-					if (success)
-						success = freerdp_client_add_drive(settings, drive, NULL);
-				}
-				else
-					success = freerdp_client_add_drive(settings, drive, name);
-
-				if (!success)
-				{
-					free(value);
-					return FALSE;
-				}
-
-				tok = strtok_s(NULL, ";", &context);
-			}
-			free(value);
-
-			if (!freerdp_settings_set_bool(settings, FreeRDP_DeviceRedirection, TRUE))
-				return FALSE;
-		}
+		if (!freerdp_settings_set_string(settings, FreeRDP_DrivesToRedirect, file->DrivesToRedirect))
+			return FALSE;
 	}
 
 	if (~file->KeyboardHook)

From a93b15fdef2d229c99d466a69e9f4f65cc8d690f Mon Sep 17 00:00:00 2001
From: Armin Novak <armin.novak@thincast.com>
Date: Fri, 23 Aug 2019 12:44:44 +0200
Subject: [PATCH 6/6] Sanitize drive redirection names.

---
 client/common/cmdline.c | 21 ++++++++++++++++++++-
 1 file changed, 20 insertions(+), 1 deletion(-)

diff --git a/client/common/cmdline.c b/client/common/cmdline.c
index ada3ab0ae..11051c19d 100644
--- a/client/common/cmdline.c
+++ b/client/common/cmdline.c
@@ -68,6 +68,25 @@ static BOOL freerdp_path_valid(const char* path, BOOL* special)
 	return isSpecial || isPath;
 }
 
+static BOOL freerdp_sanitize_drive_name(char* name, const char* invalid, const char* replacement)
+{
+	if (!name || !invalid || !replacement)
+		return FALSE;
+	if (strlen(invalid) != strlen(replacement))
+		return FALSE;
+
+	while(*invalid != '\0')
+	{
+		const char what = *invalid++;
+		const char with = *replacement++;
+
+		char* cur = name;
+		while((cur = strchr(cur, what)) != NULL)
+			*cur = with;
+	}
+	return TRUE;
+}
+
 static BOOL freerdp_client_add_drive(rdpSettings* settings, const char* path, const char* name)
 {
 	RDPDR_DRIVE* drive;
@@ -99,7 +118,7 @@ static BOOL freerdp_client_add_drive(rdpSettings* settings, const char* path, co
 		if (!(drive->Name = _strdup(path)))
 			goto fail;
 
-	if (!path)
+	if (!path || !freerdp_sanitize_drive_name(drive->Name, "\\/", "__"))
 		goto fail;
 	else
 	{