X-Git-Url: https://www.chiark.greenend.org.uk/ucgi/~ianmdlvl/git?p=elogind.git;a=blobdiff_plain;f=udev_utils_string.c;h=e3dc137e63257be33471a2d1a8ede3523a281dd4;hp=38b91aa0bd030ae8f3abb6f4561ab4e5bcf08cf3;hb=788096cb59a7ad1cdd08ccc59e5e3b1d2f4828c2;hpb=05610c088ee9ea668d4750f69a35bf833a63616d

diff --git a/udev_utils_string.c b/udev_utils_string.c
index 38b91aa0b..e3dc137e6 100644
--- a/udev_utils_string.c
+++ b/udev_utils_string.c
@@ -216,8 +216,8 @@ int utf8_encoded_valid_unichar(const char *str)
 	return len;
 }
 
-/* replace everything but whitelisted plain ascii and valid utf8 */
-int replace_untrusted_chars(char *str)
+/* allow chars in whitelist, plain ascii, hex-escaping and valid utf8 */
+int replace_chars(char *str, const char *white)
 {
 	size_t i = 0;
 	int replaced = 0;
@@ -225,27 +225,42 @@ int replace_untrusted_chars(char *str)
 	while (str[i] != '\0') {
 		int len;
 
-		/* valid printable ascii char */
+		/* accept whitelist */
+		if (white != NULL && strchr(white, str[i]) != NULL) {
+			i++;
+			continue;
+		}
+
+		/* accept plain ascii char */
 		if ((str[i] >= '0' && str[i] <= '9') ||
 		    (str[i] >= 'A' && str[i] <= 'Z') ||
-		    (str[i] >= 'a' && str[i] <= 'z') ||
-		    strchr(" #$%+-./:=?@_,", str[i])) {
+		    (str[i] >= 'a' && str[i] <= 'z')) {
 			i++;
 			continue;
 		}
-		/* hex encoding */
+
+		/* accept hex encoding */
 		if (str[i] == '\\' && str[i+1] == 'x') {
 			i += 2;
 			continue;
 		}
-		/* valid utf8 is accepted */
+
+		/* accept valid utf8 */
 		len = utf8_encoded_valid_unichar(&str[i]);
 		if (len > 1) {
 			i += len;
 			continue;
 		}
 
-		/* everything else is garbage */
+		/* if space is allowed, replace whitespace with ordinary space */
+		if (isspace(str[i]) && strchr(white, ' ') != NULL) {
+			str[i] = ' ';
+			i++;
+			replaced++;
+			continue;
+		}
+
+		/* everything else is replaced with '_' */
 		str[i] = '_';
 		i++;
 		replaced++;