]> git.ipfire.org Git - thirdparty/dovecot/core.git/commitdiff
lib: Implement t_strsplit_tabescaped_inplace()
authorTimo Sirainen <timo.sirainen@dovecot.fi>
Fri, 3 Nov 2017 23:39:38 +0000 (01:39 +0200)
committerTimo Sirainen <timo.sirainen@dovecot.fi>
Tue, 7 Nov 2017 19:19:43 +0000 (21:19 +0200)
This is a more efficient version of t_strsplit_tabescaped(), which modifies
the input string instead of duplicating it.

src/lib/strescape.c
src/lib/strescape.h

index 0fb45cd20a9a954accee1fe69320cabde5079458..5fad0a9dea03bed548fbdcdb7fcc99e80f8ef5ff 100644 (file)
@@ -249,6 +249,50 @@ const char *t_str_tabunescape(const char *str)
                return str_tabunescape(t_strdup_noconst(str));
 }
 
+const char *const *t_strsplit_tabescaped_inplace(char *data)
+{
+       /* @UNSAFE */
+       char **array;
+       unsigned int count, new_alloc_count, alloc_count;
+
+       if (*data == '\0')
+               return t_new(const char *, 1);
+
+       alloc_count = 32;
+       array = t_malloc(sizeof(char *) * alloc_count);
+
+       array[0] = data; count = 1;
+       bool need_unescape = FALSE;
+       while ((data = strpbrk(data, "\t\001")) != NULL) {
+               /* separator or escape char found */
+               if (*data == '\001') {
+                       need_unescape = TRUE;
+                       data++;
+                       continue;
+               }
+               if (count+1 >= alloc_count) {
+                       new_alloc_count = nearest_power(alloc_count+1);
+                       array = p_realloc(unsafe_data_stack_pool, array,
+                                         sizeof(char *) * alloc_count,
+                                         sizeof(char *) *
+                                         new_alloc_count);
+                       alloc_count = new_alloc_count;
+               }
+               *data++ = '\0';
+               if (need_unescape) {
+                       str_tabunescape(array[count-1]);
+                       need_unescape = FALSE;
+               }
+               array[count++] = data;
+       }
+       if (need_unescape)
+               str_tabunescape(array[count-1]);
+       i_assert(count < alloc_count);
+       array[count] = NULL;
+
+       return (const char *const *)array;
+}
+
 char **p_strsplit_tabescaped(pool_t pool, const char *str)
 {
        char **args;
index acf7cba67d5b13d7c2c899f8d90ae80ab806b902..c910a2682574325d0df2f7616f1cf28de80c965d 100644 (file)
@@ -28,5 +28,8 @@ const char *t_str_tabunescape(const char *str);
 
 char **p_strsplit_tabescaped(pool_t pool, const char *str);
 const char *const *t_strsplit_tabescaped(const char *str);
+/* Same as t_strsplit_tabescaped(), but the input string is modified and the
+   returned pointers inside the array point to the original string. */
+const char *const *t_strsplit_tabescaped_inplace(char *str);
 
 #endif