lib: Add binary-safe tabescape

dovecot · Aug 8, 2016 · 51aceed · 51aceed
1 parent 9ca065a
commit 51aceed
Show file tree

Hide file tree

Showing 2 changed files with 20 additions and 4 deletions.
diff --git a/src/lib/strescape.c b/src/lib/strescape.c
@@ -101,10 +101,14 @@ int str_unescape_next(const char **str, const char **unescaped_r)
 	return 0;
 }
 
-void str_append_tabescaped(string_t *dest, const char *src)
+void str_append_tabescaped_n(string_t *dest, const unsigned char *src, size_t src_size)
 {
-	for (; *src != '\0'; src++) {
-		switch (*src) {
+	for (size_t i = 0; i < src_size; i++) {
+		switch (src[i]) {
+		case '\000':
+			str_append_c(dest, '\001');
+			str_append_c(dest, '0');
+			break;
 		case '\001':
 			str_append_c(dest, '\001');
 			str_append_c(dest, '1');
@@ -122,12 +126,17 @@ void str_append_tabescaped(string_t *dest, const char *src)
 			str_append_c(dest, 'n');
 			break;
 		default:
-			str_append_c(dest, *src);
+			str_append_c(dest, src[i]);
 			break;
 		}
 	}
 }
 
+void str_append_tabescaped(string_t *dest, const char *src) {
+	str_append_tabescaped_n(dest, (const unsigned char*)src, strlen(src));
+}
+
+
 const char *str_tabescape(const char *str)
 {
 	string_t *tmp;
@@ -161,6 +170,9 @@ void str_append_tabunescaped(string_t *dest, const void *src, size_t src_size)
 			i++;
 			if (i < src_size) {
 				switch (src_c[i]) {
+				case '0':
+					str_append_c(dest, '\000');
+					break;
 				case '1':
 					str_append_c(dest, '\001');
 					break;
@@ -203,6 +215,9 @@ char *str_tabunescape(char *str)
 			if (*str == '\0')
 				break;
 			switch (*str) {
+			case '0':
+				*dest++ = '\000';
+				break;
 			case '1':
 				*dest++ = '\001';
 				break;

diff --git a/src/lib/strescape.h b/src/lib/strescape.h
@@ -21,6 +21,7 @@ int str_unescape_next(const char **str, const char **unescaped_r);
    using \001. */
 const char *str_tabescape(const char *str);
 void str_append_tabescaped(string_t *dest, const char *src);
+void str_append_tabescaped_n(string_t *dest, const unsigned char *src, size_t src_size);
 void str_append_tabunescaped(string_t *dest, const void *src, size_t src_size);
 char *str_tabunescape(char *str);
 const char *t_str_tabunescape(const char *str);