diff options
| author | Yu Watanabe <watanabe.yu+github@gmail.com> | 2021-06-21 21:01:54 +0900 |
|---|---|---|
| committer | Yu Watanabe <watanabe.yu+github@gmail.com> | 2021-06-21 21:14:20 +0900 |
| commit | 1104d114298f80e26f76b3672812e7b08cc411ff (patch) | |
| tree | 0da154a7c4a8da0c214ae0f124b767ae9b51dc2a /src/basic/extract-word.c | |
| parent | 1c092b62dbbc36434ad421d683c92b8aeda5ebdb (diff) | |
| download | systemd-1104d114298f80e26f76b3672812e7b08cc411ff.tar.gz | |
extract-word: introduce EXTRACT_KEEP_QUOTE flag
Diffstat (limited to 'src/basic/extract-word.c')
| -rw-r--r-- | src/basic/extract-word.c | 35 |
1 files changed, 22 insertions, 13 deletions
diff --git a/src/basic/extract-word.c b/src/basic/extract-word.c index 2c14b6f0cf..0c440db691 100644 --- a/src/basic/extract-word.c +++ b/src/basic/extract-word.c @@ -27,6 +27,7 @@ int extract_first_word(const char **p, char **ret, const char *separators, Extra assert(p); assert(ret); + assert(!FLAGS_SET(flags, EXTRACT_KEEP_QUOTE | EXTRACT_UNQUOTE)); /* Bail early if called after last value or with no input */ if (!*p) @@ -123,25 +124,30 @@ int extract_first_word(const char **p, char **ret, const char *separators, Extra return -EINVAL; } else if (c == quote) { /* found the end quote */ quote = 0; - break; + if (flags & EXTRACT_UNQUOTE) + break; } else if (c == '\\' && !(flags & EXTRACT_RETAIN_ESCAPE)) { backslash = true; break; - } else { - if (!GREEDY_REALLOC(s, sz+2)) - return -ENOMEM; - - s[sz++] = c; } + + if (!GREEDY_REALLOC(s, sz+2)) + return -ENOMEM; + + s[sz++] = c; + + if (quote == 0) + break; } } else { for (;; (*p)++, c = **p) { if (c == 0) goto finish_force_terminate; - else if (IN_SET(c, '\'', '"') && (flags & EXTRACT_UNQUOTE)) { + else if (IN_SET(c, '\'', '"') && (flags & (EXTRACT_KEEP_QUOTE | EXTRACT_UNQUOTE))) { quote = c; - break; + if (flags & EXTRACT_UNQUOTE) + break; } else if (c == '\\' && !(flags & EXTRACT_RETAIN_ESCAPE)) { backslash = true; break; @@ -159,12 +165,15 @@ int extract_first_word(const char **p, char **ret, const char *separators, Extra } goto finish; - } else { - if (!GREEDY_REALLOC(s, sz+2)) - return -ENOMEM; - - s[sz++] = c; } + + if (!GREEDY_REALLOC(s, sz+2)) + return -ENOMEM; + + s[sz++] = c; + + if (quote != 0) + break; } } } |