shaba/systemd
1
0
Fork 0
mirror of https://github.com/systemd/systemd.git synced 2025-02-04 21:47:31 +03:00
Code

basic/strv: allow escaping the separator in strv_join()

Browse Source 
The new parameter is false everywhere except for tests, so no functional change
is expected.
This commit is contained in:
Zbigniew Jędrzejewski-Szmek 2020-09-24 13:06:52 +02:00
parent be36bc1e14
commit d4d9f034b1
4 changed files with 61 additions and 33 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
src/basic/strv.c
View File

@ -367,7 +367,7 @@ int strv_split_colon_pairs(char ***t, const char *s) {
return (int) n; return (int) n;
} }
char *strv_join_prefix(char * const *l, const char *separator, const char *prefix) { char *strv_join_full(char * const *l, const char *separator, const char *prefix, bool unescape_separators) {
char * const *s; char * const *s;
char *r, *e; char *r, *e;
size_t n, k, m; size_t n, k, m;
@ -378,11 +378,17 @@ char *strv_join_prefix(char * const *l, const char *separator, const char *prefi
k = strlen(separator); k = strlen(separator);
m = strlen_ptr(prefix); m = strlen_ptr(prefix);
if (unescape_separators) /* If there separator is multi-char, we won't know how to escape it. */
assert(k == 1);
n = 0; n = 0;
STRV_FOREACH(s, l) { STRV_FOREACH(s, l) {
if (s != l) if (s != l)
n += k; n += k;
n += m + strlen(*s);
bool needs_escaping = unescape_separators && strchr(*s, separator[0]);
n += m + strlen(*s) * (1 + needs_escaping);
} }
r = new(char, n+1); r = new(char, n+1);
@ -397,6 +403,15 @@ char *strv_join_prefix(char * const *l, const char *separator, const char *prefi
if (prefix) if (prefix)
e = stpcpy(e, prefix); e = stpcpy(e, prefix);
bool needs_escaping = unescape_separators && strchr(*s, separator[0]);
if (needs_escaping)
for (size_t i = 0; (*s)[i]; i++) {
if ((*s)[i] == separator[0])
*(e++) = '\\';
*(e++) = (*s)[i];
}
else
e = stpcpy(e, *s); e = stpcpy(e, *s);
} }

4
src/basic/strv.h
View File

@ -91,9 +91,9 @@ static inline char **strv_split(const char *s, const char *separators) {
* string in the vector is an empty string. */ * string in the vector is an empty string. */
int strv_split_colon_pairs(char ***t, const char *s); int strv_split_colon_pairs(char ***t, const char *s);
char *strv_join_prefix(char * const *l, const char *separator, const char *prefix); char *strv_join_full(char * const *l, const char *separator, const char *prefix, bool escape_separtor);
static inline char *strv_join(char * const *l, const char *separator) { static inline char *strv_join(char * const *l, const char *separator) {
return strv_join_prefix(l, separator, NULL); return strv_join_full(l, separator, NULL, false);
} }
char **strv_parse_nulstr(const char *s, size_t l); char **strv_parse_nulstr(const char *s, size_t l);

2
src/core/execute.c
View File

@ -1927,7 +1927,7 @@ static int build_environment(
if (!pre) if (!pre)
return -ENOMEM; return -ENOMEM;
joined = strv_join_prefix(c->directories[t].paths, ":", pre); joined = strv_join_full(c->directories[t].paths, ":", pre, false);
if (!joined) if (!joined)
return -ENOMEM; return -ENOMEM;

67
src/test/test-strv.c
View File

@ -162,71 +162,84 @@ static void test_strv_find_startswith(void) {
} }
static void test_strv_join(void) { static void test_strv_join(void) {
_cleanup_free_ char *p = NULL, *q = NULL, *r = NULL, *s = NULL, *t = NULL, *v = NULL, *w = NULL;
log_info("/* %s */", __func__); log_info("/* %s */", __func__);
p = strv_join((char **)input_table_multiple, ", "); _cleanup_free_ char *p = strv_join((char **)input_table_multiple, ", ");
assert_se(p); assert_se(p);
assert_se(streq(p, "one, two, three")); assert_se(streq(p, "one, two, three"));
q = strv_join((char **)input_table_multiple, ";"); _cleanup_free_ char *q = strv_join((char **)input_table_multiple, ";");
assert_se(q); assert_se(q);
assert_se(streq(q, "one;two;three")); assert_se(streq(q, "one;two;three"));
r = strv_join((char **)input_table_multiple, NULL); _cleanup_free_ char *r = strv_join((char **)input_table_multiple, NULL);
assert_se(r); assert_se(r);
assert_se(streq(r, "one two three")); assert_se(streq(r, "one two three"));
s = strv_join((char **)input_table_one, ", "); _cleanup_free_ char *s = strv_join(STRV_MAKE("1", "2", "3,3"), ",");
assert_se(s); assert_se(s);
assert_se(streq(s, "one")); assert_se(streq(s, "1,2,3,3"));
t = strv_join((char **)input_table_none, ", "); _cleanup_free_ char *t = strv_join((char **)input_table_one, ", ");
assert_se(t); assert_se(t);
assert_se(streq(t, "")); assert_se(streq(t, "one"));
v = strv_join((char **)input_table_two_empties, ", "); _cleanup_free_ char *u = strv_join((char **)input_table_none, ", ");
assert_se(u);
assert_se(streq(u, ""));
_cleanup_free_ char *v = strv_join((char **)input_table_two_empties, ", ");
assert_se(v); assert_se(v);
assert_se(streq(v, ", ")); assert_se(streq(v, ", "));
w = strv_join((char **)input_table_one_empty, ", "); _cleanup_free_ char *w = strv_join((char **)input_table_one_empty, ", ");
assert_se(w); assert_se(w);
assert_se(streq(w, "")); assert_se(streq(w, ""));
} }
static void test_strv_join_prefix(void) { static void test_strv_join_full(void) {
_cleanup_free_ char *p = NULL, *q = NULL, *r = NULL, *s = NULL, *t = NULL, *v = NULL, *w = NULL;
log_info("/* %s */", __func__); log_info("/* %s */", __func__);
p = strv_join_prefix((char **)input_table_multiple, ", ", "foo"); _cleanup_free_ char *p = strv_join_full((char **)input_table_multiple, ", ", "foo", false);
assert_se(p); assert_se(p);
assert_se(streq(p, "fooone, footwo, foothree")); assert_se(streq(p, "fooone, footwo, foothree"));
q = strv_join_prefix((char **)input_table_multiple, ";", "foo"); _cleanup_free_ char *q = strv_join_full((char **)input_table_multiple, ";", "foo", false);
assert_se(q); assert_se(q);
assert_se(streq(q, "fooone;footwo;foothree")); assert_se(streq(q, "fooone;footwo;foothree"));
r = strv_join_prefix((char **)input_table_multiple, NULL, "foo"); _cleanup_free_ char *r = strv_join_full(STRV_MAKE("a", "a;b", "a:c"), ";", NULL, true);
assert_se(r); assert_se(r);
assert_se(streq(r, "fooone footwo foothree")); assert_se(streq(r, "a;a\\;b;a:c"));
s = strv_join_prefix((char **)input_table_one, ", ", "foo"); _cleanup_free_ char *s = strv_join_full(STRV_MAKE("a", "a;b", "a;;c", ";", ";x"), ";", NULL, true);
assert_se(s); assert_se(s);
assert_se(streq(s, "fooone")); assert_se(streq(s, "a;a\\;b;a\\;\\;c;\\;;\\;x"));
t = strv_join_prefix((char **)input_table_none, ", ", "foo"); _cleanup_free_ char *t = strv_join_full(STRV_MAKE("a", "a;b", "a:c", ";"), ";", "=", true);
assert_se(t); assert_se(t);
assert_se(streq(t, "")); assert_se(streq(t, "=a;=a\\;b;=a:c;=\\;"));
t = mfree(t);
v = strv_join_prefix((char **)input_table_two_empties, ", ", "foo"); _cleanup_free_ char *u = strv_join_full((char **)input_table_multiple, NULL, "foo", false);
assert_se(u);
assert_se(streq(u, "fooone footwo foothree"));
_cleanup_free_ char *v = strv_join_full((char **)input_table_one, ", ", "foo", false);
assert_se(v); assert_se(v);
assert_se(streq(v, "foo, foo")); assert_se(streq(v, "fooone"));
w = strv_join_prefix((char **)input_table_one_empty, ", ", "foo"); _cleanup_free_ char *w = strv_join_full((char **)input_table_none, ", ", "foo", false);
assert_se(w); assert_se(w);
assert_se(streq(w, "foo")); assert_se(streq(w, ""));
_cleanup_free_ char *x = strv_join_full((char **)input_table_two_empties, ", ", "foo", false);
assert_se(x);
assert_se(streq(x, "foo, foo"));
_cleanup_free_ char *y = strv_join_full((char **)input_table_one_empty, ", ", "foo", false);
assert_se(y);
assert_se(streq(y, "foo"));
} }
static void test_strv_unquote(const char *quoted, char **list) { static void test_strv_unquote(const char *quoted, char **list) {
@ -995,7 +1008,7 @@ int main(int argc, char *argv[]) {
test_strv_find_prefix(); test_strv_find_prefix();
test_strv_find_startswith(); test_strv_find_startswith();
test_strv_join(); test_strv_join();
test_strv_join_prefix(); test_strv_join_full();
test_strv_unquote(" foo=bar \"waldo\" zzz ", STRV_MAKE("foo=bar", "waldo", "zzz")); test_strv_unquote(" foo=bar \"waldo\" zzz ", STRV_MAKE("foo=bar", "waldo", "zzz"));
test_strv_unquote("", STRV_MAKE_EMPTY); test_strv_unquote("", STRV_MAKE_EMPTY);