src/lib/strescape.c

	strescape.c revision 146e7e3fabdd843d50680cdb6e4f7829890f5fcf
/* Copyright (c) 2003-2009 Dovecot authors, see the included COPYING file */

#include "lib.h"
#include "str.h"
#include "strescape.h"

const char *str_escape(const char *str)
{
    const char *p;
    string_t *ret;

    /* see if we need to quote it */
    for (p = str; *p != '\0'; p++) {
        if (IS_ESCAPED_CHAR(*p))
            break;
    }

    if (*p == '\0')
        return str;

    /* quote */
    ret = t_str_new((size_t) (p - str) + 128);
    str_append_n(ret, str, (size_t) (p - str));

    for (; *p != '\0'; p++) {
        if (IS_ESCAPED_CHAR(*p))
            str_append_c(ret, '\\');
        str_append_c(ret, *p);
    }
    return str_c(ret);
}

void str_append_unescaped(string_t *dest, const void *src, size_t src_size)
{
    const unsigned char *src_c = src;
    size_t start = 0, i = 0;

    while (i < src_size) {
        start = i;
        for (; i < src_size; i++) {
            if (src_c[i] == '\\')
                break;
        }

        str_append_n(dest, src_c + start, i-start);

        if (i < src_size)
            i++;
        start = i;
    }
}

char *str_unescape(char *str)
{
    /* @UNSAFE */
    char *dest, *start = str;

    while (*str != '\\') {
        if (*str == '\0')
            return start;
        str++;
    }

    for (dest = str; *str != '\0'; str++) {
        if (*str == '\\' && str[1] != '\0')
            str++;

        *dest++ = *str;
    }

    *dest = '\0';
    return start;
}

void str_tabescape_write(string_t *dest, const char *src)
{
    for (; *src != '\0'; src++) {
        switch (*src) {
        case '\001':
            str_append_c(dest, '\001');
            str_append_c(dest, '1');
            break;
        case '\t':
            str_append_c(dest, '\001');
            str_append_c(dest, 't');
            break;
        case '\n':
            str_append_c(dest, '\001');
            str_append_c(dest, 'n');
            break;
        default:
            str_append_c(dest, *src);
            break;
        }
    }
}

const char *str_tabescape(const char *str)
{
    string_t *tmp;
    const char *p;

    for (p = str; *p != '\0'; p++) {
        if (*p <= '\n') {
            tmp = t_str_new(128);
            str_append_n(tmp, str, p-str);
            str_tabescape_write(tmp, p);
            return str_c(tmp);
        }
    }
    return str;
}