src/basic/escape.c

	escape.c revision 4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering/*-*- Mode: C; c-basic-offset: 8; indent-tabs-mode: nil -*-*/
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering/***
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering  This file is part of systemd.
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering  Copyright 2010 Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering  systemd is free software; you can redistribute it and/or modify it
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering  under the terms of the GNU Lesser General Public License as published by
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering  the Free Software Foundation; either version 2.1 of the License, or
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering  (at your option) any later version.
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering  systemd is distributed in the hope that it will be useful, but
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering  WITHOUT ANY WARRANTY; without even the implied warranty of
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering  Lesser General Public License for more details.
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering  You should have received a copy of the GNU Lesser General Public License
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering  along with systemd; If not, see <http://www.gnu.org/licenses/>.
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering***/
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering#include "utf8.h"
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering#include "util.h"
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering#include "escape.h"
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poetteringsize_t cescape_char(char c, char *buf) {
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        char * buf_old = buf;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        switch (c) {
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                case '\a':
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        *(buf++) = '\\';
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        *(buf++) = 'a';
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        break;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                case '\b':
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        *(buf++) = '\\';
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        *(buf++) = 'b';
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        break;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                case '\f':
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        *(buf++) = '\\';
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        *(buf++) = 'f';
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        break;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                case '\n':
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        *(buf++) = '\\';
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        *(buf++) = 'n';
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        break;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                case '\r':
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        *(buf++) = '\\';
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        *(buf++) = 'r';
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        break;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                case '\t':
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        *(buf++) = '\\';
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        *(buf++) = 't';
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        break;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                case '\v':
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        *(buf++) = '\\';
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        *(buf++) = 'v';
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        break;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                case '\\':
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        *(buf++) = '\\';
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        *(buf++) = '\\';
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        break;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                case '"':
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        *(buf++) = '\\';
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        *(buf++) = '"';
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        break;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                case '\'':
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        *(buf++) = '\\';
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        *(buf++) = '\'';
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        break;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                default:
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        /* For special chars we prefer octal over
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                         * hexadecimal encoding, simply because glib's
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                         * g_strescape() does the same */
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        if ((c < ' ') || (c >= 127)) {
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                                *(buf++) = '\\';
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                                *(buf++) = octchar((unsigned char) c >> 6);
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                                *(buf++) = octchar((unsigned char) c >> 3);
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                                *(buf++) = octchar((unsigned char) c);
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        } else
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                                *(buf++) = c;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        break;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        }
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        return buf - buf_old;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering}
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poetteringchar *cescape(const char *s) {
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        char *r, *t;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        const char *f;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        assert(s);
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        /* Does C style string escaping. May be reversed with
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering         * cunescape(). */
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        r = new(char, strlen(s)*4 + 1);
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        if (!r)
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                return NULL;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        for (f = s, t = r; *f; f++)
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                t += cescape_char(*f, t);
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        *t = 0;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        return r;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering}
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poetteringint cunescape_one(const char *p, size_t length, char *ret, uint32_t *ret_unicode) {
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        int r = 1;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        assert(p);
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        assert(*p);
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        assert(ret);
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        /* Unescapes C style. Returns the unescaped character in ret,
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering         * unless we encountered a \u sequence in which case the full
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering         * unicode character is returned in ret_unicode, instead. */
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        if (length != (size_t) -1 && length < 1)
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                return -EINVAL;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        switch (p[0]) {
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        case 'a':
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                *ret = '\a';
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                break;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        case 'b':
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                *ret = '\b';
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                break;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        case 'f':
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                *ret = '\f';
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                break;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        case 'n':
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                *ret = '\n';
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                break;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        case 'r':
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                *ret = '\r';
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                break;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        case 't':
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                *ret = '\t';
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                break;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        case 'v':
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                *ret = '\v';
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                break;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        case '\\':
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                *ret = '\\';
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                break;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        case '"':
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                *ret = '"';
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                break;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        case '\'':
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                *ret = '\'';
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                break;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        case 's':
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                /* This is an extension of the XDG syntax files */
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                *ret = ' ';
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                break;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        case 'x': {
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                /* hexadecimal encoding */
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                int a, b;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                if (length != (size_t) -1 && length < 3)
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        return -EINVAL;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                a = unhexchar(p[1]);
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                if (a < 0)
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        return -EINVAL;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                b = unhexchar(p[2]);
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                if (b < 0)
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        return -EINVAL;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                /* Don't allow NUL bytes */
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                if (a == 0 && b == 0)
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        return -EINVAL;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                *ret = (char) ((a << 4U) | b);
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                r = 3;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                break;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        }
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        case 'u': {
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                /* C++11 style 16bit unicode */
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                int a[4];
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                unsigned i;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                uint32_t c;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                if (length != (size_t) -1 && length < 5)
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        return -EINVAL;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                for (i = 0; i < 4; i++) {
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        a[i] = unhexchar(p[1 + i]);
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        if (a[i] < 0)
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                                return a[i];
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                }
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                c = ((uint32_t) a[0] << 12U) | ((uint32_t) a[1] << 8U) | ((uint32_t) a[2] << 4U) | (uint32_t) a[3];
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                /* Don't allow 0 chars */
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                if (c == 0)
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        return -EINVAL;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                if (c < 128)
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        *ret = c;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                else {
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        if (!ret_unicode)
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                                return -EINVAL;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        *ret = 0;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        *ret_unicode = c;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                }
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                r = 5;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                break;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        }
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        case 'U': {
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                /* C++11 style 32bit unicode */
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                int a[8];
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                unsigned i;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                uint32_t c;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                if (length != (size_t) -1 && length < 9)
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        return -EINVAL;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                for (i = 0; i < 8; i++) {
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        a[i] = unhexchar(p[1 + i]);
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        if (a[i] < 0)
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                                return a[i];
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                }
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                c = ((uint32_t) a[0] << 28U) | ((uint32_t) a[1] << 24U) | ((uint32_t) a[2] << 20U) | ((uint32_t) a[3] << 16U) |
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                    ((uint32_t) a[4] << 12U) | ((uint32_t) a[5] <<  8U) | ((uint32_t) a[6] <<  4U) |  (uint32_t) a[7];
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                /* Don't allow 0 chars */
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                if (c == 0)
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        return -EINVAL;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                /* Don't allow invalid code points */
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                if (!unichar_is_valid(c))
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        return -EINVAL;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                if (c < 128)
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        *ret = c;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                else {
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        if (!ret_unicode)
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                                return -EINVAL;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        *ret = 0;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        *ret_unicode = c;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                }
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                r = 9;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                break;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        }
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        case '0':
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        case '1':
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        case '2':
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        case '3':
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        case '4':
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        case '5':
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        case '6':
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        case '7': {
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                /* octal encoding */
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                int a, b, c;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                uint32_t m;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                if (length != (size_t) -1 && length < 3)
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        return -EINVAL;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                a = unoctchar(p[0]);
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                if (a < 0)
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        return -EINVAL;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                b = unoctchar(p[1]);
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                if (b < 0)
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        return -EINVAL;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                c = unoctchar(p[2]);
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                if (c < 0)
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        return -EINVAL;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                /* don't allow NUL bytes */
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                if (a == 0 && b == 0 && c == 0)
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        return -EINVAL;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                /* Don't allow bytes above 255 */
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                m = ((uint32_t) a << 6U) | ((uint32_t) b << 3U) | (uint32_t) c;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                if (m > 255)
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        return -EINVAL;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                *ret = m;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                r = 3;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                break;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        }
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        default:
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                return -EINVAL;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        }
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        return r;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering}
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poetteringint cunescape_length_with_prefix(const char *s, size_t length, const char *prefix, UnescapeFlags flags, char **ret) {
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        char *r, *t;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        const char *f;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        size_t pl;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        assert(s);
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        assert(ret);
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        /* Undoes C style string escaping, and optionally prefixes it. */
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        pl = prefix ? strlen(prefix) : 0;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        r = new(char, pl+length+1);
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        if (!r)
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                return -ENOMEM;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        if (prefix)
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                memcpy(r, prefix, pl);
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        for (f = s, t = r + pl; f < s + length; f++) {
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                size_t remaining;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                uint32_t u;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                char c;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                int k;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                remaining = s + length - f;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                assert(remaining > 0);
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                if (*f != '\\') {
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        /* A literal literal, copy verbatim */
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        *(t++) = *f;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        continue;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                }
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                if (remaining == 1) {
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        if (flags & UNESCAPE_RELAX) {
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                                /* A trailing backslash, copy verbatim */
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                                *(t++) = *f;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                                continue;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        }
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        free(r);
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        return -EINVAL;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                }
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                k = cunescape_one(f + 1, remaining - 1, &c, &u);
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                if (k < 0) {
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        if (flags & UNESCAPE_RELAX) {
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                                /* Invalid escape code, let's take it literal then */
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                                *(t++) = '\\';
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                                continue;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        }
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        free(r);
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        return k;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                }
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                if (c != 0)
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        /* Non-Unicode? Let's encode this directly */
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        *(t++) = c;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                else
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        /* Unicode? Then let's encode this in UTF-8 */
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        t += utf8_encode_unichar(t, u);
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                f += k;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        }
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        *t = 0;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        *ret = r;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        return t - r;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering}
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poetteringint cunescape_length(const char *s, size_t length, UnescapeFlags flags, char **ret) {
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        return cunescape_length_with_prefix(s, length, NULL, flags, ret);
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering}
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poetteringint cunescape(const char *s, UnescapeFlags flags, char **ret) {
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        return cunescape_length(s, strlen(s), flags, ret);
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering}
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poetteringchar *xescape(const char *s, const char *bad) {
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        char *r, *t;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        const char *f;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        /* Escapes all chars in bad, in addition to \ and all special
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering         * chars, in \xFF style escaping. May be reversed with
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering         * cunescape(). */
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        r = new(char, strlen(s) * 4 + 1);
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        if (!r)
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                return NULL;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        for (f = s, t = r; *f; f++) {
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                if ((*f < ' ') || (*f >= 127) ||
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                    (*f == '\\') || strchr(bad, *f)) {
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        *(t++) = '\\';
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        *(t++) = 'x';
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        *(t++) = hexchar(*f >> 4);
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        *(t++) = hexchar(*f);
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                } else
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        *(t++) = *f;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        }
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        *t = 0;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        return r;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering}
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poetteringstatic char *strcpy_backslash_escaped(char *t, const char *s, const char *bad) {
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        assert(bad);
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        for (; *s; s++) {
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                if (*s == '\\' || strchr(bad, *s))
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        *(t++) = '\\';
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                *(t++) = *s;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        }
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        return t;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering}
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poetteringchar *shell_escape(const char *s, const char *bad) {
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        char *r, *t;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        r = new(char, strlen(s)*2+1);
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        if (!r)
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                return NULL;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        t = strcpy_backslash_escaped(r, s, bad);
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        *t = 0;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        return r;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering}
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poetteringchar *shell_maybe_quote(const char *s) {
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        const char *p;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        char *r, *t;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        assert(s);
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        /* Encloses a string in double quotes if necessary to make it
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering         * OK as shell string. */
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        for (p = s; *p; p++)
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                if (*p <= ' ' ||
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                    *p >= 127 ||
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                    strchr(SHELL_NEED_QUOTES, *p))
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                        break;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        if (!*p)
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                return strdup(s);
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        r = new(char, 1+strlen(s)*2+1+1);
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        if (!r)
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering                return NULL;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        t = r;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        *(t++) = '"';
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        t = mempcpy(t, s, p - s);
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        t = strcpy_backslash_escaped(t, p, SHELL_NEED_ESCAPE);
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        *(t++)= '"';
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        *t = 0;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering        return r;
4f5dd3943bef8a04be7e3b838b822bb9a7ad6cb3Lennart Poettering}