src/lib-imap/imap-util.c

	imap-util.c revision 72932593c2aa46767f7e43a8e20347ad989de95b
25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen/* Copyright (c) 2002-2017 Dovecot authors, see the included COPYING file */
25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen
25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen#include "lib.h"
25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen#include "array.h"
25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen#include "str.h"
25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen#include "strescape.h"
25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen#include "unichar.h"
25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen#include "mail-types.h"
25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen#include "imap-parser.h"
25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen#include "imap-util.h"
25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen
25757faf029c369a8318349dafe952e2358df1d8Timo Sirainenvoid imap_write_flags(string_t *dest, enum mail_flags flags,
3b94ff5951db4d4eddb7a80ed4e3f61207202635Timo Sirainen              const char *const *keywords)
3b94ff5951db4d4eddb7a80ed4e3f61207202635Timo Sirainen{
25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen    size_t size;
965ed6ea3fc8f7637bd0d159d2fdb283a191ce34Timo Sirainen
25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen    size = str_len(dest);
25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen    if ((flags & MAIL_ANSWERED) != 0)
3b94ff5951db4d4eddb7a80ed4e3f61207202635Timo Sirainen        str_append(dest, "\\Answered ");
25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen    if ((flags & MAIL_FLAGGED) != 0)
e3507728aced8c5d570962c00189aac27e6e1f20Timo Sirainen        str_append(dest, "\\Flagged ");
3b94ff5951db4d4eddb7a80ed4e3f61207202635Timo Sirainen    if ((flags & MAIL_DELETED) != 0)
22535a9e685e29214082878e37a267157044618eTimo Sirainen        str_append(dest, "\\Deleted ");
22535a9e685e29214082878e37a267157044618eTimo Sirainen    if ((flags & MAIL_SEEN) != 0)
25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen        str_append(dest, "\\Seen ");
25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen    if ((flags & MAIL_DRAFT) != 0)
965ed6ea3fc8f7637bd0d159d2fdb283a191ce34Timo Sirainen        str_append(dest, "\\Draft ");
965ed6ea3fc8f7637bd0d159d2fdb283a191ce34Timo Sirainen    if ((flags & MAIL_RECENT) != 0)
965ed6ea3fc8f7637bd0d159d2fdb283a191ce34Timo Sirainen        str_append(dest, "\\Recent ");
25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen
965ed6ea3fc8f7637bd0d159d2fdb283a191ce34Timo Sirainen    if (keywords != NULL) {
061da1b9d309301a7e3f5e0f240d7a7e99652d63Timo Sirainen        /* we have keywords too */
25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen        while (*keywords != NULL) {
965ed6ea3fc8f7637bd0d159d2fdb283a191ce34Timo Sirainen            str_append(dest, *keywords);
25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen            str_append_c(dest, ' ');
25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen            keywords++;
25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen        }
25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen    }
25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen
25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen    if (str_len(dest) != size)
25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen        str_truncate(dest, str_len(dest)-1);
25757faf029c369a8318349dafe952e2358df1d8Timo Sirainen}

enum mail_flags imap_parse_system_flag(const char *str)
{
    if (strcasecmp(str, "\\Answered") == 0)
        return MAIL_ANSWERED;
    else if (strcasecmp(str, "\\Flagged") == 0)
        return MAIL_FLAGGED;
    else if (strcasecmp(str, "\\Deleted") == 0)
        return MAIL_DELETED;
    else if (strcasecmp(str, "\\Seen") == 0)
        return MAIL_SEEN;
    else if (strcasecmp(str, "\\Draft") == 0)
        return MAIL_DRAFT;
    else if (strcasecmp(str, "\\Recent") == 0)
        return MAIL_RECENT;
    else
        return 0;
}

void imap_write_seq_range(string_t *dest, const ARRAY_TYPE(seq_range) *array)
{
    const struct seq_range *range;
    unsigned int i, count;

    range = array_get(array, &count);
    for (i = 0; i < count; i++) {
        if (i > 0)
            str_append_c(dest, ',');
        str_printfa(dest, "%u", range[i].seq1);
        if (range[i].seq1 != range[i].seq2)
            str_printfa(dest, ":%u", range[i].seq2);
    }
}

void imap_write_args(string_t *dest, const struct imap_arg *args)
{
    bool first = TRUE;

    for (; !IMAP_ARG_IS_EOL(args); args++) {
        if (first)
            first = FALSE;
        else
            str_append_c(dest, ' ');

        switch (args->type) {
        case IMAP_ARG_NIL:
            str_append(dest, "NIL");
            break;
        case IMAP_ARG_ATOM:
            str_append(dest, imap_arg_as_astring(args));
            break;
        case IMAP_ARG_STRING:
            str_append_c(dest, '"');
            str_append(dest, str_escape(imap_arg_as_astring(args)));
            str_append_c(dest, '"');
            break;
        case IMAP_ARG_LITERAL: {
            const char *strarg = imap_arg_as_astring(args);
            str_printfa(dest, "{%"PRIuSIZE_T"}\r\n",
                    strlen(strarg));
            str_append(dest, strarg);
            break;
        }
        case IMAP_ARG_LIST:
            str_append_c(dest, '(');
            imap_write_args(dest, imap_arg_as_list(args));
            str_append_c(dest, ')');
            break;
        case IMAP_ARG_LITERAL_SIZE:
        case IMAP_ARG_LITERAL_SIZE_NONSYNC:
            str_printfa(dest, "{%"PRIuUOFF_T"}\r\n",
                    imap_arg_as_literal_size(args));
            str_append(dest, "<too large>");
            break;
        case IMAP_ARG_EOL:
            i_unreached();
        }
    }
}

static void imap_human_args_fix_control_chars(char *str)
{
    size_t i;

    for (i = 0; str[i] != '\0'; i++) {
        if (str[i] < 0x20 || str[i] == 0x7f)
            str[i] = '?';
    }
}

void imap_write_args_for_human(string_t *dest, const struct imap_arg *args)
{
    bool first = TRUE;

    for (; !IMAP_ARG_IS_EOL(args); args++) {
        if (first)
            first = FALSE;
        else
            str_append_c(dest, ' ');

        switch (args->type) {
        case IMAP_ARG_NIL:
            str_append(dest, "NIL");
            break;
        case IMAP_ARG_ATOM:
            /* atom has only printable us-ascii chars */
            str_append(dest, imap_arg_as_astring(args));
            break;
        case IMAP_ARG_STRING:
        case IMAP_ARG_LITERAL: {
            const char *strarg = imap_arg_as_astring(args);

            if (strpbrk(strarg, "\r\n") != NULL) {
                str_printfa(dest, "<%"PRIuSIZE_T" byte multi-line literal>",
                        strlen(strarg));
                break;
            }
            strarg = str_escape(strarg);

            str_append_c(dest, '"');
            size_t start_pos = str_len(dest);
            /* append only valid UTF-8 chars */
            if (uni_utf8_get_valid_data((const unsigned char *)strarg,
                            strlen(strarg), dest))
                str_append(dest, strarg);
            /* replace all control chars */
            imap_human_args_fix_control_chars(
                str_c_modifiable(dest) + start_pos);
            str_append_c(dest, '"');
            break;
        }
        case IMAP_ARG_LIST:
            str_append_c(dest, '(');
            imap_write_args_for_human(dest, imap_arg_as_list(args));
            str_append_c(dest, ')');
            break;
        case IMAP_ARG_LITERAL_SIZE:
        case IMAP_ARG_LITERAL_SIZE_NONSYNC:
            str_printfa(dest, "<%"PRIuUOFF_T" byte literal>",
                    imap_arg_as_literal_size(args));
            break;
        case IMAP_ARG_EOL:
            i_unreached();
        }
    }
}

const char *imap_args_to_str(const struct imap_arg *args)
{
    string_t *str;

    str = t_str_new(128);
    imap_write_args(str, args);
    return str_c(str);
}