28341416412

Committed 28 Jun 2026 09:25AM UTC coverage: 72.714% (-0.2%) from 72.925%

Build # 28341416412

Build Type

push

github

Committed by

bluca

Commit Message

Revert "mkosi: Grow the root partition on boot"

This reverts commit 7f33ee8bb.

The file is located outside mkosi/ subdirectory, hence currently unused.
If this is moved to mkosi/ subdirectory, the config conflicts with
TEST-58-REPART. Let's remove it at least now, and reintroduce it later
at correct place with test adjustment if this is really useful.

Coverage Stats

341139 of 469154 relevant lines covered (72.71%)

1450250.85 hits per line

Source File
Press 'n' to go to next uncovered line, 'b' for previous

99.42

/src/basic/escape.c

/* SPDX-License-Identifier: LGPL-2.1-or-later */

#include <stdlib.h>
#include <string.h>

#include "alloc-util.h"
#include "escape.h"
#include "hexdecoct.h"
#include "string-util.h"
#include "strv.h"
#include "utf8.h"

int cescape_char(char c, char *buf) {
        char *buf_old = buf;

        /* Needs space for 4 characters in the buffer */

        switch (c) {

                case '\a':
                        *(buf++) = '\\';
                        *(buf++) = 'a';
                        break;
                case '\b':
                        *(buf++) = '\\';
                        *(buf++) = 'b';
                        break;
                case '\f':
                        *(buf++) = '\\';
                        *(buf++) = 'f';
                        break;
                case '\n':
                        *(buf++) = '\\';
                        *(buf++) = 'n';
                        break;
                case '\r':
                        *(buf++) = '\\';
                        *(buf++) = 'r';
                        break;
                case '\t':
                        *(buf++) = '\\';
                        *(buf++) = 't';
                        break;
                case '\v':
                        *(buf++) = '\\';
                        *(buf++) = 'v';
                        break;
                case '\\':
                        *(buf++) = '\\';
                        *(buf++) = '\\';
                        break;
                case '"':
                        *(buf++) = '\\';
                        *(buf++) = '"';
                        break;
                case '\'':
                        *(buf++) = '\\';
                        *(buf++) = '\'';
                        break;

                default:
                        /* For special chars we prefer octal over
                         * hexadecimal encoding, simply because glib's
                         * g_strescape() does the same */
                        if ((c < ' ') || (c >= 127)) {
                                *(buf++) = '\\';
                                *(buf++) = octchar((unsigned char) c >> 6);
                                *(buf++) = octchar((unsigned char) c >> 3);
                                *(buf++) = octchar((unsigned char) c);
                        } else
                                *(buf++) = c;
                        break;
        }

        return buf - buf_old;
}

char* cescape_length(const char *s, size_t n) {
        const char *f;
        char *r, *t;

        /* Does C style string escaping. May be reversed with cunescape(). */

        assert(s || n == 0);

        if (n == SIZE_MAX)
                n = strlen(s);

        if (n > (SIZE_MAX - 1) / 4)
                return NULL;

        r = new(char, n*4 + 1);
        if (!r)
                return NULL;

        for (f = s, t = r; f < s + n; f++)
                t += cescape_char(*f, t);

        *t = 0;

        return r;
}

int cunescape_one(const char *p, size_t length, char32_t *ret, bool *eight_bit, bool accept_nul) {
        int r = 1;

        assert(p);
        assert(ret);
        assert(eight_bit);

        /* Unescapes C style. Returns the unescaped character in ret.
         * Sets *eight_bit to true if the escaped sequence either fits in
         * one byte in UTF-8 or is a non-unicode literal byte and should
         * instead be copied directly.
         */

        if (length != SIZE_MAX && length < 1)
                return -EINVAL;

        switch (p[0]) {

        case 'a':
                *ret = '\a';
                break;
        case 'b':
                *ret = '\b';
                break;
        case 'f':
                *ret = '\f';
                break;
        case 'n':
                *ret = '\n';
                break;
        case 'r':
                *ret = '\r';
                break;
        case 't':
                *ret = '\t';
                break;
        case 'v':
                *ret = '\v';
                break;
        case '\\':
                *ret = '\\';
                break;
        case '"':
                *ret = '"';
                break;
        case '\'':
                *ret = '\'';
                break;

        case 's':
                /* This is an extension of the XDG syntax files */
                *ret = ' ';
                break;

        case 'x': {
                /* hexadecimal encoding */
                int a, b;

                if (length != SIZE_MAX && length < 3)
                        return -EINVAL;

                a = unhexchar(p[1]);
                if (a < 0)
                        return -EINVAL;

                b = unhexchar(p[2]);
                if (b < 0)
                        return -EINVAL;

                /* Don't allow NUL bytes */
                if (a == 0 && b == 0 && !accept_nul)
                        return -EINVAL;

                *ret = (a << 4U) | b;
                *eight_bit = true;
                r = 3;
                break;
        }

        case 'u': {
                /* C++11 style 16-bit unicode */

                int a[4];
                size_t i;
                uint32_t c;

                if (length != SIZE_MAX && length < 5)
                        return -EINVAL;

                for (i = 0; i < 4; i++) {
                        a[i] = unhexchar(p[1 + i]);
                        if (a[i] < 0)
                                return a[i];
                }

                c = ((uint32_t) a[0] << 12U) | ((uint32_t) a[1] << 8U) | ((uint32_t) a[2] << 4U) | (uint32_t) a[3];

                /* Don't allow 0 chars */
                if (c == 0 && !accept_nul)
                        return -EINVAL;

                *ret = c;
                r = 5;
                break;
        }

        case 'U': {
                /* C++11 style 32-bit unicode */

                int a[8];
                size_t i;
                char32_t c;

                if (length != SIZE_MAX && length < 9)
                        return -EINVAL;

                for (i = 0; i < 8; i++) {
                        a[i] = unhexchar(p[1 + i]);
                        if (a[i] < 0)
                                return a[i];
                }

                c = ((uint32_t) a[0] << 28U) | ((uint32_t) a[1] << 24U) | ((uint32_t) a[2] << 20U) | ((uint32_t) a[3] << 16U) |
                    ((uint32_t) a[4] << 12U) | ((uint32_t) a[5] <<  8U) | ((uint32_t) a[6] <<  4U) |  (uint32_t) a[7];

                /* Don't allow 0 chars */
                if (c == 0 && !accept_nul)
                        return -EINVAL;

                /* Don't allow invalid code points */
                if (!unichar_is_valid(c))
                        return -EINVAL;

                *ret = c;
                r = 9;
                break;
        }

        case '0':
        case '1':
        case '2':
        case '3':
        case '4':
        case '5':
        case '6':
        case '7': {
                /* octal encoding */
                int a, b, c;
                char32_t m;

                if (length != SIZE_MAX && length < 3)
                        return -EINVAL;

                a = unoctchar(p[0]);
                if (a < 0)
                        return -EINVAL;

                b = unoctchar(p[1]);
                if (b < 0)
                        return -EINVAL;

                c = unoctchar(p[2]);
                if (c < 0)
                        return -EINVAL;

                /* don't allow NUL bytes */
                if (a == 0 && b == 0 && c == 0 && !accept_nul)
                        return -EINVAL;

                /* Don't allow bytes above 255 */
                m = ((uint32_t) a << 6U) | ((uint32_t) b << 3U) | (uint32_t) c;
                if (m > 255)
                        return -EINVAL;

                *ret = m;
                *eight_bit = true;
                r = 3;
                break;
        }

        default:
                return -EINVAL;
        }

        return r;
}

ssize_t cunescape_length_with_prefix(const char *s, size_t length, const char *prefix, UnescapeFlags flags, char **ret) {
        _cleanup_free_ char *ans = NULL;
        char *t;
        const char *f;
        size_t pl;
        int r;

        assert(s);
        assert(ret);

        /* Undoes C style string escaping, and optionally prefixes it. */

        if (length == SIZE_MAX)
                length = strlen(s);

        pl = strlen_ptr(prefix);

        ans = new(char, pl+length+1);
        if (!ans)
                return -ENOMEM;

        if (prefix)
                memcpy(ans, prefix, pl);

        for (f = s, t = ans + pl; f < s + length; f++) {
                size_t remaining;
                bool eight_bit = false;
                char32_t u;

                remaining = s + length - f;
                assert(remaining > 0);

                if (*f != '\\') {
                        /* A literal, copy verbatim */
                        *(t++) = *f;
                        continue;
                }

                if (remaining == 1) {
                        if (flags & UNESCAPE_RELAX) {
                                /* A trailing backslash, copy verbatim */
                                *(t++) = *f;
                                continue;
                        }

                        return -EINVAL;
                }

                r = cunescape_one(f + 1, remaining - 1, &u, &eight_bit, flags & UNESCAPE_ACCEPT_NUL);
                if (r < 0) {
                        if (flags & UNESCAPE_RELAX) {
                                /* Invalid escape code, let's take it literal then */
                                *(t++) = '\\';
                                continue;
                        }

                        return r;
                }

                f += r;
                if (eight_bit)
                        /* One byte? Set directly as specified */
                        *(t++) = u;
                else
                        /* Otherwise encode as multi-byte UTF-8 */
                        t += utf8_encode_unichar(t, u);
        }

        *t = 0;

        assert(t >= ans); /* Let static analyzers know that the answer is non-negative. */
        *ret = TAKE_PTR(ans);
        return t - *ret;
}

char* xescape_full(const char *s, const char *bad, size_t console_width, XEscapeFlags flags) {
        char *ans, *t, *prev, *prev2;
        const char *f;

        assert(s);

        /* Escapes all chars in bad, in addition to \ and all special chars, in \xFF style escaping. May be
         * reversed with cunescape(). If XESCAPE_8_BIT is specified, characters >= 127 are let through
         * unchanged. This corresponds to non-ASCII printable characters in pre-unicode encodings.
         *
         * If console_width is reached, or XESCAPE_FORCE_ELLIPSIS is set, output is truncated and "..." is
         * appended. */

        if (console_width == 0)
                return strdup("");

        size_t len_forced_ellipsis = FLAGS_SET(flags, XESCAPE_FORCE_ELLIPSIS) ? STRLEN("...") : 0;
        size_t len_s = strlen(s);

        size_t len_body = MIN(console_width, SIZE_MAX - 1); /* We need room for the NUL byte */
        if (len_body > len_forced_ellipsis && len_s <= (len_body - len_forced_ellipsis) / 4)
                len_body = len_s * 4 + len_forced_ellipsis;

        ans = new(char, len_body + 1);
        if (!ans)
                return NULL;

        for (f = s, t = prev = prev2 = ans; ; f++) {
                char *tmp_t = t;

                if (!*f) {
                        if (len_forced_ellipsis != 0)
                                break;

                        *t = 0;
                        return ans;
                }

                if ((unsigned char) *f < ' ' ||
                    (!FLAGS_SET(flags, XESCAPE_8_BIT) && (unsigned char) *f >= 127) ||
                    *f == '\\' || (bad && strchr(bad, *f))) {
                        if ((size_t) (t - ans) + 4 + len_forced_ellipsis > len_body)
                                break;

                        *(t++) = '\\';
                        *(t++) = 'x';
                        *(t++) = hexchar(*f >> 4);
                        *(t++) = hexchar(*f);
                } else {
                        if ((size_t) (t - ans) + 1 + len_forced_ellipsis > len_body)
                                break;

                        *(t++) = *f;
                }

                /* We might need to go back two cycles to fit three dots, so remember two positions */
                prev2 = prev;
                prev = tmp_t;
        }

        /* We can just write where we want, since chars are one-byte */
        size_t c = MIN(len_body, STRLEN("...")); /* If the console is too narrow, write fewer dots */
        size_t off;
        if (len_body - c >= (size_t) (t - ans))
                off = (size_t) (t - ans);
        else if (len_body - c >= (size_t) (prev - ans))
                off = (size_t) (prev - ans);
        else if (len_body - c >= (size_t) (prev2 - ans))
                off = (size_t) (prev2 - ans);
        else
                off = len_body - c;
        assert(off <= (size_t) (t - ans));

        memcpy(ans + off, "...", c);
        ans[off + c] = '\0';
        return ans;
}

char* escape_non_printable_full(const char *str, size_t console_width, XEscapeFlags flags) {
        if (FLAGS_SET(flags, XESCAPE_8_BIT))
                return xescape_full(str, /* bad= */ NULL, console_width, flags);
        else
                return utf8_escape_non_printable_full(str,
                                                      console_width,
                                                      FLAGS_SET(flags, XESCAPE_FORCE_ELLIPSIS));
}

char* octescape_full(const char *s, size_t len, const char *bad) {
        char *buf, *t;

        /* Escapes all chars in bad, in addition to \ and " chars, in \nnn octal style escaping. May be
         * reversed with cunescape(). */

        assert(s || len == 0);

        if (len == SIZE_MAX)
                len = strlen(s);

        if (len > (SIZE_MAX - 1) / 4)
                return NULL;

        t = buf = new(char, len * 4 + 1);
        if (!buf)
                return NULL;

        for (size_t i = 0; i < len; i++) {
                uint8_t u = (uint8_t) s[i];

                if (u < ' ' || u >= 127 || IN_SET(u, '\\', '"') || (bad && strchr(bad, u))) {
                        *(t++) = '\\';
                        *(t++) = '0' + (u >> 6);
                        *(t++) = '0' + ((u >> 3) & 7);
                        *(t++) = '0' + (u & 7);
                } else
                        *(t++) = u;
        }

        *t = 0;
        return buf;
}

char* decescape(const char *s, size_t len, const char *bad) {
        char *buf, *t;

        /* Escapes all chars in bad, in addition to \ and " chars, in \nnn decimal style escaping. */

        assert(s || len == 0);

        if (len == SIZE_MAX)
                len = strlen(s);

        if (len > (SIZE_MAX - 1) / 4)
                return NULL;

        t = buf = new(char, len * 4 + 1);
        if (!buf)
                return NULL;

        for (size_t i = 0; i < len; i++) {
                uint8_t u = (uint8_t) s[i];

                if (u < ' ' || u >= 127 || IN_SET(u, '\\', '"') || strchr(bad, u)) {
                        *(t++) = '\\';
                        *(t++) = '0' + (u / 100);
                        *(t++) = '0' + ((u / 10) % 10);
                        *(t++) = '0' + (u % 10);
                } else
                        *(t++) = u;
        }

        *t = 0;
        return buf;
}

static char* strcpy_backslash_escaped(char *t, const char *s, const char *bad) {
        assert(bad);
        assert(t);
        assert(s);

        while (*s) {
                int l = utf8_encoded_valid_unichar(s, SIZE_MAX);

                if (char_is_cc(*s) || l < 0)
                        t += cescape_char(*(s++), t);
                else if (l == 1) {
                        if (*s == '\\' || strchr(bad, *s))
                                *(t++) = '\\';
                        *(t++) = *(s++);
                } else {
                        t = mempcpy(t, s, l);
                        s += l;
                }
        }

        return t;
}

char* shell_escape(const char *s, const char *bad) {
        char *buf, *t;

        buf = new(char, strlen(s)*4+1);
        if (!buf)
                return NULL;

        t = strcpy_backslash_escaped(buf, s, bad);
        *t = 0;

        return buf;
}

char* shell_maybe_quote(const char *s, ShellEscapeFlags flags) {
        const char *p;
        char *buf, *t;

        assert(s);

        /* Encloses a string in quotes if necessary to make it OK as a shell string. */

        if (FLAGS_SET(flags, SHELL_ESCAPE_EMPTY) && isempty(s))
                return strdup("\"\""); /* We don't use $'' here in the POSIX mode. "" is fine too. */

        for (p = s; *p; ) {
                int l = utf8_encoded_valid_unichar(p, SIZE_MAX);

                if (char_is_cc(*p) || l < 0 ||
                    strchr(WHITESPACE SHELL_NEED_QUOTES, *p))
                        break;

                p += l;
        }

        if (!*p)
                return strdup(s);

        buf = new(char, FLAGS_SET(flags, SHELL_ESCAPE_POSIX) + 1 + strlen(s)*4 + 1 + 1);
        if (!buf)
                return NULL;

        t = buf;
        if (FLAGS_SET(flags, SHELL_ESCAPE_POSIX)) {
                *(t++) = '$';
                *(t++) = '\'';
        } else
                *(t++) = '"';

        t = mempcpy(t, s, p - s);

        t = strcpy_backslash_escaped(t, p,
                                     FLAGS_SET(flags, SHELL_ESCAPE_POSIX) ? SHELL_NEED_ESCAPE_POSIX : SHELL_NEED_ESCAPE);

        if (FLAGS_SET(flags, SHELL_ESCAPE_POSIX))
                *(t++) = '\'';
        else
                *(t++) = '"';
        *t = 0;

        return str_realloc(buf);
}

char* quote_command_line(char * const *argv, ShellEscapeFlags flags) {
        _cleanup_free_ char *result = NULL;

        assert(argv);

        STRV_FOREACH(a, argv) {
                _cleanup_free_ char *t = NULL;

                t = shell_maybe_quote(*a, flags);
                if (!t)
                        return NULL;

                if (!strextend_with_separator(&result, " ", t))
                        return NULL;
        }

        return str_realloc(TAKE_PTR(result));
}

1	/* SPDX-License-Identifier: LGPL-2.1-or-later */
2
3	#include <stdlib.h>
4	#include <string.h>
5
6	#include "alloc-util.h"
7	#include "escape.h"
8	#include "hexdecoct.h"
9	#include "string-util.h"
10	#include "strv.h"
11	#include "utf8.h"
12
13	int cescape_char(char c, char *buf) {	1,812,999✔
14	char *buf_old = buf;	1,812,999✔
15
16	/* Needs space for 4 characters in the buffer */
17
18	switch (c) {	1,812,999✔
19
20	case '\a':	7✔
21	*(buf++) = '\\';	7✔
22	*(buf++) = 'a';	7✔
23	break;	7✔
24	case '\b':	9✔
25	*(buf++) = '\\';	9✔
26	*(buf++) = 'b';	9✔
27	break;	9✔
28	case '\f':	7✔
29	*(buf++) = '\\';	7✔
30	*(buf++) = 'f';	7✔
31	break;	7✔
32	case '\n':	19,063✔
33	*(buf++) = '\\';	19,063✔
34	*(buf++) = 'n';	19,063✔
35	break;	19,063✔
36	case '\r':	8✔
37	*(buf++) = '\\';	8✔
38	*(buf++) = 'r';	8✔
39	break;	8✔
40	case '\t':	53✔
41	*(buf++) = '\\';	53✔
42	*(buf++) = 't';	53✔
43	break;	53✔
44	case '\v':	2✔
45	*(buf++) = '\\';	2✔
46	*(buf++) = 'v';	2✔
47	break;	2✔
48	case '\\':	87✔
49	*(buf++) = '\\';	87✔
50	*(buf++) = '\\';	87✔
51	break;	87✔
52	case '"':	3,684✔
53	*(buf++) = '\\';	3,684✔
54	*(buf++) = '"';	3,684✔
55	break;	3,684✔
56	case '\'':	19✔
57	*(buf++) = '\\';	19✔
58	*(buf++) = '\'';	19✔
59	break;	19✔
60
61	default:	1,790,060✔
62	/* For special chars we prefer octal over
63	* hexadecimal encoding, simply because glib's
64	* g_strescape() does the same */
65	if ((c < ' ') \|\| (c >= 127)) {	1,790,060✔
66	*(buf++) = '\\';	11,903✔
67	*(buf++) = octchar((unsigned char) c >> 6);	11,903✔
68	*(buf++) = octchar((unsigned char) c >> 3);	11,903✔
69	*(buf++) = octchar((unsigned char) c);	11,903✔
70	} else
71	*(buf++) = c;	1,778,157✔
72	break;
73	}
74
75	return buf - buf_old;	1,812,999✔
76	}
77
78	char* cescape_length(const char *s, size_t n) {	45,601✔
79	const char *f;	45,601✔
80	char r, t;	45,601✔
81
82	/* Does C style string escaping. May be reversed with cunescape(). */
83
84	assert(s \|\| n == 0);	45,601✔
85
86	if (n == SIZE_MAX)	45,601✔
87	n = strlen(s);	27,732✔
88
89	if (n > (SIZE_MAX - 1) / 4)	45,601✔
90	return NULL;
91
92	r = new(char, n*4 + 1);	45,601✔
93	if (!r)	45,601✔
94	return NULL;
95
96	for (f = s, t = r; f < s + n; f++)	1,214,760✔
97	t += cescape_char(*f, t);	1,169,159✔
98
99	*t = 0;	45,601✔
100
101	return r;	45,601✔
102	}
103
104	int cunescape_one(const char p, size_t length, char32_t ret, bool *eight_bit, bool accept_nul) {	12,511✔
105	int r = 1;	12,511✔
106
107	assert(p);	12,511✔
108	assert(ret);	12,511✔
109	assert(eight_bit);	12,511✔
110
111	/* Unescapes C style. Returns the unescaped character in ret.
112	* Sets *eight_bit to true if the escaped sequence either fits in
113	* one byte in UTF-8 or is a non-unicode literal byte and should
114	* instead be copied directly.
115	*/
116
117	if (length != SIZE_MAX && length < 1)	12,511✔
118	return -EINVAL;
119
120	switch (p[0]) {	12,511✔
121
122	case 'a':	4✔
123	*ret = '\a';	4✔
124	break;	4✔
125	case 'b':	8✔
126	*ret = '\b';	8✔
127	break;	8✔
128	case 'f':	3✔
129	*ret = '\f';	3✔
130	break;	3✔
131	case 'n':	22✔
132	*ret = '\n';	22✔
133	break;	22✔
134	case 'r':	4✔
135	*ret = '\r';	4✔
136	break;	4✔
137	case 't':	6✔
138	*ret = '\t';	6✔
139	break;	6✔
140	case 'v':	3✔
141	*ret = '\v';	3✔
142	break;	3✔
143	case '\\':	15✔
144	*ret = '\\';	15✔
145	break;	15✔
146	case '"':	13✔
147	*ret = '"';	13✔
148	break;	13✔
149	case '\'':	1✔
150	*ret = '\'';	1✔
151	break;	1✔
152
153	case 's':	3✔
154	/* This is an extension of the XDG syntax files */
155	*ret = ' ';	3✔
156	break;	3✔
157
158	case 'x': {	12,347✔
159	/* hexadecimal encoding */
160	int a, b;	12,347✔
161
162	if (length != SIZE_MAX && length < 3)	12,347✔
163	return -EINVAL;
164
165	a = unhexchar(p[1]);	12,343✔
166	if (a < 0)	12,343✔
167	return -EINVAL;
168
169	b = unhexchar(p[2]);	12,341✔
170	if (b < 0)	12,341✔
171	return -EINVAL;
172
173	/* Don't allow NUL bytes */
174	if (a == 0 && b == 0 && !accept_nul)	12,341✔
175	return -EINVAL;
176
177	*ret = (a << 4U) \| b;	12,340✔
178	*eight_bit = true;	12,340✔
179	r = 3;	12,340✔
180	break;	12,340✔
181	}
182
183	case 'u': {	19✔
184	/* C++11 style 16-bit unicode */
185
186	int a[4];	19✔
187	size_t i;	19✔
188	uint32_t c;	19✔
189
190	if (length != SIZE_MAX && length < 5)	19✔
191	return -EINVAL;	1✔
192
193	for (i = 0; i < 4; i++) {	95✔
194	a[i] = unhexchar(p[1 + i]);	76✔
195	if (a[i] < 0)	76✔
196	return a[i];
197	}
198
199	c = ((uint32_t) a[0] << 12U) \| ((uint32_t) a[1] << 8U) \| ((uint32_t) a[2] << 4U) \| (uint32_t) a[3];	19✔
200
201	/* Don't allow 0 chars */
202	if (c == 0 && !accept_nul)	19✔
203	return -EINVAL;
204
205	*ret = c;	18✔
206	r = 5;	18✔
207	break;	18✔
208	}
209
210	case 'U': {	10✔
211	/* C++11 style 32-bit unicode */
212
213	int a[8];	10✔
214	size_t i;	10✔
215	char32_t c;	10✔
216
217	if (length != SIZE_MAX && length < 9)	10✔
218	return -EINVAL;	×
219
220	for (i = 0; i < 8; i++) {	90✔
221	a[i] = unhexchar(p[1 + i]);	80✔
222	if (a[i] < 0)	80✔
223	return a[i];
224	}
225
226	c = ((uint32_t) a[0] << 28U) \| ((uint32_t) a[1] << 24U) \| ((uint32_t) a[2] << 20U) \| ((uint32_t) a[3] << 16U) \|	10✔
227	((uint32_t) a[4] << 12U) \| ((uint32_t) a[5] << 8U) \| ((uint32_t) a[6] << 4U) \| (uint32_t) a[7];	10✔
228
229	/* Don't allow 0 chars */
230	if (c == 0 && !accept_nul)	10✔
231	return -EINVAL;
232
233	/* Don't allow invalid code points */
234	if (!unichar_is_valid(c))	10✔
235	return -EINVAL;
236
237	*ret = c;	10✔
238	r = 9;	10✔
239	break;	10✔
240	}
241
242	case '0':	20✔
243	case '1':
244	case '2':
245	case '3':
246	case '4':
247	case '5':
248	case '6':
249	case '7': {
250	/* octal encoding */
251	int a, b, c;	20✔
252	char32_t m;	20✔
253
254	if (length != SIZE_MAX && length < 3)	20✔
255	return -EINVAL;
256
257	a = unoctchar(p[0]);	16✔
258	if (a < 0)	16✔
259	return -EINVAL;
260
261	b = unoctchar(p[1]);	16✔
262	if (b < 0)	16✔
263	return -EINVAL;
264
265	c = unoctchar(p[2]);	16✔
266	if (c < 0)	16✔
267	return -EINVAL;
268
269	/* don't allow NUL bytes */
270	if (a == 0 && b == 0 && c == 0 && !accept_nul)	16✔
271	return -EINVAL;
272
273	/* Don't allow bytes above 255 */
274	m = ((uint32_t) a << 6U) \| ((uint32_t) b << 3U) \| (uint32_t) c;	14✔
275	if (m > 255)	14✔
276	return -EINVAL;
277
278	*ret = m;	14✔
279	*eight_bit = true;	14✔
280	r = 3;	14✔
281	break;	14✔
282	}
283
284	default:
285	return -EINVAL;
286	}
287
288	return r;
289	}
290
291	ssize_t cunescape_length_with_prefix(const char s, size_t length, const char prefix, UnescapeFlags flags, char **ret) {	154,862✔
292	_cleanup_free_ char *ans = NULL;	154,862✔
293	char *t;	154,862✔
294	const char *f;	154,862✔
295	size_t pl;	154,862✔
296	int r;	154,862✔
297
298	assert(s);	154,862✔
299	assert(ret);	154,862✔
300
301	/* Undoes C style string escaping, and optionally prefixes it. */
302
303	if (length == SIZE_MAX)	154,862✔
304	length = strlen(s);	115,095✔
305
306	pl = strlen_ptr(prefix);	154,862✔
307
308	ans = new(char, pl+length+1);	154,862✔
309	if (!ans)	154,862✔
310	return -ENOMEM;
311
312	if (prefix)	154,862✔
313	memcpy(ans, prefix, pl);	39,726✔
314
315	for (f = s, t = ans + pl; f < s + length; f++) {	5,762,780✔
316	size_t remaining;	5,607,927✔
317	bool eight_bit = false;	5,607,927✔
318	char32_t u;	5,607,927✔
319
320	remaining = s + length - f;	5,607,927✔
321	assert(remaining > 0);	5,607,927✔
322
323	if (*f != '\\') {	5,607,927✔
324	/* A literal, copy verbatim */
325	(t++) = f;	5,595,499✔
326	continue;	5,595,506✔
327	}
328
329	if (remaining == 1) {	12,428✔
330	if (flags & UNESCAPE_RELAX) {	3✔
331	/* A trailing backslash, copy verbatim */
332	(t++) = f;	1✔
333	continue;	1✔
334	}
335
336	return -EINVAL;	9✔
337	}
338
339	r = cunescape_one(f + 1, remaining - 1, &u, &eight_bit, flags & UNESCAPE_ACCEPT_NUL);	12,425✔
340	if (r < 0) {	12,425✔
341	if (flags & UNESCAPE_RELAX) {	13✔
342	/* Invalid escape code, let's take it literal then */
343	*(t++) = '\\';	6✔
344	continue;	6✔
345	}
346
347	return r;	7✔
348	}
349
350	f += r;	12,412✔
351	if (eight_bit)	12,412✔
352	/* One byte? Set directly as specified */
353	*(t++) = u;	12,341✔
354	else
355	/* Otherwise encode as multi-byte UTF-8 */
356	t += utf8_encode_unichar(t, u);	71✔
357	}
358
359	*t = 0;	154,853✔
360
361	assert(t >= ans); /* Let static analyzers know that the answer is non-negative. */	154,853✔
362	*ret = TAKE_PTR(ans);	154,853✔
363	return t - *ret;	154,853✔
364	}
365
366	char* xescape_full(const char s, const char bad, size_t console_width, XEscapeFlags flags) {	109,092✔
367	char ans, t, prev, prev2;	109,092✔
368	const char *f;	109,092✔
369
370	assert(s);	109,092✔
371
372	/* Escapes all chars in bad, in addition to \ and all special chars, in \xFF style escaping. May be
373	* reversed with cunescape(). If XESCAPE_8_BIT is specified, characters >= 127 are let through
374	* unchanged. This corresponds to non-ASCII printable characters in pre-unicode encodings.
375	*
376	* If console_width is reached, or XESCAPE_FORCE_ELLIPSIS is set, output is truncated and "..." is
377	* appended. */
378
379	if (console_width == 0)	109,092✔
380	return strdup("");	4✔
381
382	size_t len_forced_ellipsis = FLAGS_SET(flags, XESCAPE_FORCE_ELLIPSIS) ? STRLEN("...") : 0;	109,088✔
383	size_t len_s = strlen(s);	109,088✔
384
385	size_t len_body = MIN(console_width, SIZE_MAX - 1); /* We need room for the NUL byte */	109,088✔
386	if (len_body > len_forced_ellipsis && len_s <= (len_body - len_forced_ellipsis) / 4)	109,088✔
387	len_body = len_s * 4 + len_forced_ellipsis;	108,852✔
388
389	ans = new(char, len_body + 1);	109,088✔
390	if (!ans)	109,088✔
391	return NULL;
392
393	for (f = s, t = prev = prev2 = ans; ; f++) {	6,226,857✔
394	char *tmp_t = t;	6,335,945✔
395
396	if (!*f) {	6,335,945✔
397	if (len_forced_ellipsis != 0)	108,884✔
398	break;
399
400	*t = 0;	108,870✔
401	return ans;	108,870✔
402	}
403
404	if ((unsigned char) *f < ' ' \|\|	6,227,061✔
405	(!FLAGS_SET(flags, XESCAPE_8_BIT) && (unsigned char) *f >= 127) \|\|	6,225,892✔
406	f == '\\' \|\| (bad && strchr(bad, f))) {	6,224,859✔
407	if ((size_t) (t - ans) + 4 + len_forced_ellipsis > len_body)	11,291✔
408	break;
409
410	*(t++) = '\\';	11,107✔
411	*(t++) = 'x';	11,107✔
412	(t++) = hexchar(f >> 4);	11,107✔
413	(t++) = hexchar(f);	11,107✔
414	} else {
415	if ((size_t) (t - ans) + 1 + len_forced_ellipsis > len_body)	6,215,770✔
416	break;
417
418	(t++) = f;	6,215,750✔
419	}
420
421	/* We might need to go back two cycles to fit three dots, so remember two positions */
422	prev2 = prev;	6,226,857✔
423	prev = tmp_t;	6,226,857✔
424	}
425
426	/* We can just write where we want, since chars are one-byte */
427	size_t c = MIN(len_body, STRLEN("...")); /* If the console is too narrow, write fewer dots */	218✔
428	size_t off;	218✔
429	if (len_body - c >= (size_t) (t - ans))	218✔
430	off = (size_t) (t - ans);
431	else if (len_body - c >= (size_t) (prev - ans))	76✔
432	off = (size_t) (prev - ans);
433	else if (len_body - c >= (size_t) (prev2 - ans))	10✔
434	off = (size_t) (prev2 - ans);
435	else
436	off = len_body - c;	1✔
437	assert(off <= (size_t) (t - ans));	218✔
438
439	memcpy(ans + off, "...", c);	218✔
440	ans[off + c] = '\0';	218✔
441	return ans;	218✔
442	}
443
444	char* escape_non_printable_full(const char *str, size_t console_width, XEscapeFlags flags) {	352✔
445	if (FLAGS_SET(flags, XESCAPE_8_BIT))	352✔
446	return xescape_full(str, /* bad= */ NULL, console_width, flags);	×
447	else
448	return utf8_escape_non_printable_full(str,	352✔
449	console_width,
450	FLAGS_SET(flags, XESCAPE_FORCE_ELLIPSIS));	352✔
451	}
452
453	char* octescape_full(const char s, size_t len, const char bad) {	86✔
454	char buf, t;	86✔
455
456	/* Escapes all chars in bad, in addition to \ and " chars, in \nnn octal style escaping. May be
457	* reversed with cunescape(). */
458
459	assert(s \|\| len == 0);	86✔
460
461	if (len == SIZE_MAX)	86✔
462	len = strlen(s);	27✔
463
464	if (len > (SIZE_MAX - 1) / 4)	86✔
465	return NULL;
466
467	t = buf = new(char, len * 4 + 1);	86✔
468	if (!buf)	86✔
469	return NULL;
470
471	for (size_t i = 0; i < len; i++) {	1,058✔
472	uint8_t u = (uint8_t) s[i];	972✔
473
474	if (u < ' ' \|\| u >= 127 \|\| IN_SET(u, '\\', '"') \|\| (bad && strchr(bad, u))) {	972✔
475	*(t++) = '\\';	5✔
476	*(t++) = '0' + (u >> 6);	5✔
477	*(t++) = '0' + ((u >> 3) & 7);	5✔
478	*(t++) = '0' + (u & 7);	5✔
479	} else
480	*(t++) = u;	967✔
481	}
482
483	*t = 0;	86✔
484	return buf;	86✔
485	}
486
487	char* decescape(const char s, size_t len, const char bad) {	8✔
488	char buf, t;	8✔
489
490	/* Escapes all chars in bad, in addition to \ and " chars, in \nnn decimal style escaping. */
491
492	assert(s \|\| len == 0);	8✔
493
494	if (len == SIZE_MAX)	8✔
495	len = strlen(s);	6✔
496
497	if (len > (SIZE_MAX - 1) / 4)	8✔
498	return NULL;
499
500	t = buf = new(char, len * 4 + 1);	8✔
501	if (!buf)	8✔
502	return NULL;
503
504	for (size_t i = 0; i < len; i++) {	47✔
505	uint8_t u = (uint8_t) s[i];	39✔
506
507	if (u < ' ' \|\| u >= 127 \|\| IN_SET(u, '\\', '"') \|\| strchr(bad, u)) {	39✔
508	*(t++) = '\\';	20✔
509	*(t++) = '0' + (u / 100);	20✔
510	*(t++) = '0' + ((u / 10) % 10);	20✔
511	*(t++) = '0' + (u % 10);	20✔
512	} else
513	*(t++) = u;	19✔
514	}
515
516	*t = 0;	8✔
517	return buf;	8✔
518	}
519
520	static char* strcpy_backslash_escaped(char t, const char s, const char *bad) {	207,906✔
521	assert(bad);	207,906✔
522	assert(t);	207,906✔
523	assert(s);	207,906✔
524
525	while (*s) {	3,998,769✔
526	int l = utf8_encoded_valid_unichar(s, SIZE_MAX);	3,790,863✔
527
528	if (char_is_cc(*s) \|\| l < 0)	3,790,863✔
529	t += cescape_char(*(s++), t);	19,060✔
530	else if (l == 1) {	3,771,803✔
531	if (s == '\\' \|\| strchr(bad, s))	3,771,797✔
532	*(t++) = '\\';	4,494✔
533	(t++) = (s++);	3,771,797✔
534	} else {
535	t = mempcpy(t, s, l);	6✔
536	s += l;	6✔
537	}
538	}
539
540	return t;	207,906✔
541	}
542
543	char* shell_escape(const char s, const char bad) {	201,943✔
544	char buf, t;	201,943✔
545
546	buf = new(char, strlen(s)*4+1);	201,943✔
547	if (!buf)	201,943✔
548	return NULL;
549
550	t = strcpy_backslash_escaped(buf, s, bad);	201,943✔
551	*t = 0;	201,943✔
552
553	return buf;	201,943✔
554	}
555
556	char* shell_maybe_quote(const char *s, ShellEscapeFlags flags) {	98,689✔
557	const char *p;	98,689✔
558	char buf, t;	98,689✔
559
560	assert(s);	98,689✔
561
562	/* Encloses a string in quotes if necessary to make it OK as a shell string. */
563
564	if (FLAGS_SET(flags, SHELL_ESCAPE_EMPTY) && isempty(s))	98,689✔
565	return strdup("\"\""); /* We don't use $'' here in the POSIX mode. "" is fine too. */	96✔
566
567	for (p = s; *p; ) {	1,988,317✔
568	int l = utf8_encoded_valid_unichar(p, SIZE_MAX);	1,895,687✔
569
570	if (char_is_cc(*p) \|\| l < 0 \|\|	1,895,687✔
571	strchr(WHITESPACE SHELL_NEED_QUOTES, *p))	1,895,651✔
572	break;
573
574	p += l;	1,889,724✔
575	}
576
577	if (!*p)	98,593✔
578	return strdup(s);	92,630✔
579
580	buf = new(char, FLAGS_SET(flags, SHELL_ESCAPE_POSIX) + 1 + strlen(s)*4 + 1 + 1);	11,856✔
581	if (!buf)	5,963✔
582	return NULL;
583
584	t = buf;	5,963✔
585	if (FLAGS_SET(flags, SHELL_ESCAPE_POSIX)) {	5,963✔
586	*(t++) = '$';	70✔
587	*(t++) = '\'';	70✔
588	} else
589	*(t++) = '"';	5,893✔
590
591	t = mempcpy(t, s, p - s);	5,963✔
592
593	t = strcpy_backslash_escaped(t, p,	11,856✔
594	FLAGS_SET(flags, SHELL_ESCAPE_POSIX) ? SHELL_NEED_ESCAPE_POSIX : SHELL_NEED_ESCAPE);
595
596	if (FLAGS_SET(flags, SHELL_ESCAPE_POSIX))	5,963✔
597	*(t++) = '\'';	70✔
598	else
599	*(t++) = '"';	5,893✔
600	*t = 0;	5,963✔
601
602	return str_realloc(buf);	5,963✔
603	}
604
605	char* quote_command_line(char * const *argv, ShellEscapeFlags flags) {	33,864✔
606	_cleanup_free_ char *result = NULL;	33,864✔
607
608	assert(argv);	33,864✔
609
610	STRV_FOREACH(a, argv) {	131,618✔
611	_cleanup_free_ char *t = NULL;	97,754✔
612
613	t = shell_maybe_quote(*a, flags);	97,754✔
614	if (!t)	97,754✔
615	return NULL;
616
617	if (!strextend_with_separator(&result, " ", t))	97,754✔
618	return NULL;
619	}
620
621	return str_realloc(TAKE_PTR(result));	33,864✔
622	}

systemd / systemd / 28341416412

Source File Press 'n' to go to next uncovered line, 'b' for previous

Source File
Press 'n' to go to next uncovered line, 'b' for previous