1d4f710106fb498750456724628da6063e012e6dTimo Sirainen

1d4f710106fb498750456724628da6063e012e6dTimo Sirainen#include "lib.h"

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen#include "array.h"

d3d769026fae5d21c2d29614d3bc4579e8d79e81Timo Sirainen#include "istream.h"

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen#include "ostream.h"

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen#include "str.h"

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen#include "message-address.h"

4ead43ecc06d10047998966c4dc0b142ecce4b66Timo Sirainen#include "message-size.h"

1d4f710106fb498750456724628da6063e012e6dTimo Sirainen#include "message-parser.h"

1d4f710106fb498750456724628da6063e012e6dTimo Sirainen#include "message-header-decode.h"

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen#include "message-decoder.h"

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen#include "imap-util.h"

ad004e44be109684521494b5af2ad1da39b8bb27Timo Sirainen#include "mail-user.h"

ab0d9eecd85f74acae18fe88529302e0776cc500Timo Sirainen#include "mail-storage.h"

ad004e44be109684521494b5af2ad1da39b8bb27Timo Sirainen#include "mail-search.h"

ad004e44be109684521494b5af2ad1da39b8bb27Timo Sirainen#include "mail-namespace.h"

ad004e44be109684521494b5af2ad1da39b8bb27Timo Sirainen#include "doveadm-print.h"

d3d769026fae5d21c2d29614d3bc4579e8d79e81Timo Sirainen#include "doveadm-mail.h"

ad004e44be109684521494b5af2ad1da39b8bb27Timo Sirainen#include "doveadm-mailbox-list-iter.h"

ad004e44be109684521494b5af2ad1da39b8bb27Timo Sirainen#include "doveadm-mail-iter.h"

ad004e44be109684521494b5af2ad1da39b8bb27Timo Sirainen

ad004e44be109684521494b5af2ad1da39b8bb27Timo Sirainen#include <stdio.h>

ad004e44be109684521494b5af2ad1da39b8bb27Timo Sirainen

ad004e44be109684521494b5af2ad1da39b8bb27Timo Sirainenstruct fetch_cmd_context {

ad004e44be109684521494b5af2ad1da39b8bb27Timo Sirainen struct doveadm_mail_cmd_context ctx;

ad004e44be109684521494b5af2ad1da39b8bb27Timo Sirainen

d3d769026fae5d21c2d29614d3bc4579e8d79e81Timo Sirainen struct mail *mail;

ad004e44be109684521494b5af2ad1da39b8bb27Timo Sirainen

ad004e44be109684521494b5af2ad1da39b8bb27Timo Sirainen ARRAY(struct fetch_field) fields;

ad004e44be109684521494b5af2ad1da39b8bb27Timo Sirainen ARRAY_TYPE(const_string) header_fields;

ad004e44be109684521494b5af2ad1da39b8bb27Timo Sirainen enum mail_fetch_field wanted_fields;

ad004e44be109684521494b5af2ad1da39b8bb27Timo Sirainen

ad004e44be109684521494b5af2ad1da39b8bb27Timo Sirainen const struct fetch_field *cur_field;

ad004e44be109684521494b5af2ad1da39b8bb27Timo Sirainen};

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainenstruct fetch_field {

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen const char *name;

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen enum mail_fetch_field wanted_fields;

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen int (*print)(struct fetch_cmd_context *ctx);

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen};

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainenstatic int fetch_user(struct fetch_cmd_context *ctx)

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen{

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen doveadm_print(ctx->ctx.cur_mail_user->username);

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen return 0;

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen}

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen

12cf3d0e03fc70fb0c8b91bc8fd83b4e14d7cdefTimo Sirainenstatic int fetch_mailbox(struct fetch_cmd_context *ctx)

12cf3d0e03fc70fb0c8b91bc8fd83b4e14d7cdefTimo Sirainen{

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen const char *value;

4e35bae013cee5a06d281776a347b534b958aaa4Timo Sirainen

087939d3fa9c4056419386c9d6c81f147de534cdTimo Sirainen if (mail_get_special(ctx->mail, MAIL_FETCH_MAILBOX_NAME, &value) < 0)

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen return -1;

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen doveadm_print(value);

45155bb1250cf5a120278f349465aded513a100fTimo Sirainen return 0;

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen}

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainenstatic int fetch_mailbox_guid(struct fetch_cmd_context *ctx)

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen{

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen struct mailbox_metadata metadata;

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen if (mailbox_get_metadata(ctx->mail->box, MAILBOX_METADATA_GUID,

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen &metadata) < 0)

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen return -1;

45155bb1250cf5a120278f349465aded513a100fTimo Sirainen doveadm_print(guid_128_to_string(metadata.guid));

45155bb1250cf5a120278f349465aded513a100fTimo Sirainen return 0;

45155bb1250cf5a120278f349465aded513a100fTimo Sirainen}

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainenstatic int fetch_seq(struct fetch_cmd_context *ctx)

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen{

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen doveadm_print_num(ctx->mail->seq);

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen return 0;

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen}

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainenstatic int fetch_uid(struct fetch_cmd_context *ctx)

12cf3d0e03fc70fb0c8b91bc8fd83b4e14d7cdefTimo Sirainen{

12cf3d0e03fc70fb0c8b91bc8fd83b4e14d7cdefTimo Sirainen doveadm_print_num(ctx->mail->uid);

4ead43ecc06d10047998966c4dc0b142ecce4b66Timo Sirainen return 0;

12cf3d0e03fc70fb0c8b91bc8fd83b4e14d7cdefTimo Sirainen}

4e35bae013cee5a06d281776a347b534b958aaa4Timo Sirainen

4ead43ecc06d10047998966c4dc0b142ecce4b66Timo Sirainenstatic int fetch_guid(struct fetch_cmd_context *ctx)

bb8d0ec26bdd548624d7a7424071cca693b72f55Timo Sirainen{

bb8d0ec26bdd548624d7a7424071cca693b72f55Timo Sirainen const char *value;

bb8d0ec26bdd548624d7a7424071cca693b72f55Timo Sirainen

bb8d0ec26bdd548624d7a7424071cca693b72f55Timo Sirainen if (mail_get_special(ctx->mail, MAIL_FETCH_GUID, &value) < 0)

bb8d0ec26bdd548624d7a7424071cca693b72f55Timo Sirainen return -1;

bb8d0ec26bdd548624d7a7424071cca693b72f55Timo Sirainen doveadm_print(value);

4e35bae013cee5a06d281776a347b534b958aaa4Timo Sirainen return 0;

4ead43ecc06d10047998966c4dc0b142ecce4b66Timo Sirainen}

4e35bae013cee5a06d281776a347b534b958aaa4Timo Sirainen

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainenstatic int fetch_flags(struct fetch_cmd_context *ctx)

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen{

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen string_t *str = t_str_new(64);

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen

6ef7e31619edfaa17ed044b45861d106a86191efTimo Sirainen imap_write_flags(str, mail_get_flags(ctx->mail),

1d4f710106fb498750456724628da6063e012e6dTimo Sirainen mail_get_keywords(ctx->mail));

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen doveadm_print(str_c(str));

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen return 0;

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen}

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainenstatic int fetch_modseq(struct fetch_cmd_context *ctx)

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen{

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen doveadm_print_num(mail_get_modseq(ctx->mail));

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen return 0;

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen}

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainenstatic int fetch_hdr(struct fetch_cmd_context *ctx)

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen{

1d4f710106fb498750456724628da6063e012e6dTimo Sirainen struct istream *input;

eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen struct message_size hdr_size;

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen int ret;

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen if (mail_get_hdr_stream(ctx->mail, &hdr_size, &input) < 0)

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen return -1;

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen input = i_stream_create_limit(input, hdr_size.physical_size);

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen ret = doveadm_print_istream(input);

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen i_stream_unref(&input);

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen return ret;

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen}

1d4f710106fb498750456724628da6063e012e6dTimo Sirainen

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainenstatic int fetch_hdr_field(struct fetch_cmd_context *ctx)

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen{

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen const char *const *value, *filter, *name = ctx->cur_field->name;

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen string_t *str = t_str_new(256);

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen bool add_lf = FALSE;

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen filter = strchr(name, '.');

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen if (filter != NULL)

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen name = t_strdup_until(name, filter++);

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen if (filter != NULL && strcmp(filter, "utf8") == 0) {

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen if (mail_get_headers_utf8(ctx->mail, name, &value) < 0)

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen return -1;

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen } else {

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen if (mail_get_headers(ctx->mail, name, &value) < 0)

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen return -1;

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen }

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen for (; *value != NULL; value++) {

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen if (add_lf)

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen str_append_c(str, '\n');

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen str_append(str, *value);

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen add_lf = TRUE;

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen }

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen if (filter == NULL || strcmp(filter, "utf8") == 0) {

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen /* print the header as-is */

eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen } else if (strcmp(filter, "address") == 0 ||

9439bed2f07d6475febd8a247cd2f0990fb32a13Timo Sirainen strcmp(filter, "address_name") == 0 ||

eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen strcmp(filter, "address_name.utf8") == 0) {

eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen struct message_address *addr;

19e8adccba16ff419f5675b1575358c2956dce83Timo Sirainen

eddd9bf1a1369aea4a2715f6be1137da6d17d293Timo Sirainen addr = message_address_parse(pool_datastack_create(),

19e8adccba16ff419f5675b1575358c2956dce83Timo Sirainen str_data(str), str_len(str),

1d4f710106fb498750456724628da6063e012e6dTimo Sirainen UINT_MAX, FALSE);

89795c6bbbc52bb382e88bc8617d22092223e9a5Timo Sirainen str_truncate(str, 0);

89795c6bbbc52bb382e88bc8617d22092223e9a5Timo Sirainen add_lf = FALSE;

89795c6bbbc52bb382e88bc8617d22092223e9a5Timo Sirainen for (; addr != NULL; addr = addr->next) {

89795c6bbbc52bb382e88bc8617d22092223e9a5Timo Sirainen if (add_lf)

89795c6bbbc52bb382e88bc8617d22092223e9a5Timo Sirainen str_append_c(str, '\n');

89795c6bbbc52bb382e88bc8617d22092223e9a5Timo Sirainen if (strcmp(filter, "address") == 0) {

89795c6bbbc52bb382e88bc8617d22092223e9a5Timo Sirainen if (addr->mailbox != NULL)

89795c6bbbc52bb382e88bc8617d22092223e9a5Timo Sirainen str_append(str, addr->mailbox);

89795c6bbbc52bb382e88bc8617d22092223e9a5Timo Sirainen if (addr->domain != NULL) {

89795c6bbbc52bb382e88bc8617d22092223e9a5Timo Sirainen str_append_c(str, '@');

89795c6bbbc52bb382e88bc8617d22092223e9a5Timo Sirainen str_append(str, addr->domain);

89795c6bbbc52bb382e88bc8617d22092223e9a5Timo Sirainen }

89795c6bbbc52bb382e88bc8617d22092223e9a5Timo Sirainen } else if (addr->name != NULL) {

89795c6bbbc52bb382e88bc8617d22092223e9a5Timo Sirainen if (strcmp(filter, "address_name") == 0)

89795c6bbbc52bb382e88bc8617d22092223e9a5Timo Sirainen str_append(str, addr->name);

89795c6bbbc52bb382e88bc8617d22092223e9a5Timo Sirainen else {

89795c6bbbc52bb382e88bc8617d22092223e9a5Timo Sirainen message_header_decode_utf8(

89795c6bbbc52bb382e88bc8617d22092223e9a5Timo Sirainen (const void *)addr->name,

89795c6bbbc52bb382e88bc8617d22092223e9a5Timo Sirainen strlen(addr->name), str, NULL);

89795c6bbbc52bb382e88bc8617d22092223e9a5Timo Sirainen }

89795c6bbbc52bb382e88bc8617d22092223e9a5Timo Sirainen }

89795c6bbbc52bb382e88bc8617d22092223e9a5Timo Sirainen add_lf = TRUE;

89795c6bbbc52bb382e88bc8617d22092223e9a5Timo Sirainen }

38318f5e82662615cd88e99e398efe4a630ce020Timo Sirainen } else {

38318f5e82662615cd88e99e398efe4a630ce020Timo Sirainen i_fatal("Unknown header filter: %s", filter);

38318f5e82662615cd88e99e398efe4a630ce020Timo Sirainen }

38318f5e82662615cd88e99e398efe4a630ce020Timo Sirainen doveadm_print(str_c(str));

38318f5e82662615cd88e99e398efe4a630ce020Timo Sirainen return 0;

38318f5e82662615cd88e99e398efe4a630ce020Timo Sirainen}

38318f5e82662615cd88e99e398efe4a630ce020Timo Sirainen

38318f5e82662615cd88e99e398efe4a630ce020Timo Sirainenstatic int fetch_body(struct fetch_cmd_context *ctx)

38318f5e82662615cd88e99e398efe4a630ce020Timo Sirainen{

38318f5e82662615cd88e99e398efe4a630ce020Timo Sirainen struct istream *input;

38318f5e82662615cd88e99e398efe4a630ce020Timo Sirainen struct message_size hdr_size;

38318f5e82662615cd88e99e398efe4a630ce020Timo Sirainen

38318f5e82662615cd88e99e398efe4a630ce020Timo Sirainen if (mail_get_stream(ctx->mail, &hdr_size, NULL, &input) < 0)

38318f5e82662615cd88e99e398efe4a630ce020Timo Sirainen return -1;

38318f5e82662615cd88e99e398efe4a630ce020Timo Sirainen

38318f5e82662615cd88e99e398efe4a630ce020Timo Sirainen i_stream_skip(input, hdr_size.physical_size);

38318f5e82662615cd88e99e398efe4a630ce020Timo Sirainen return doveadm_print_istream(input);

38318f5e82662615cd88e99e398efe4a630ce020Timo Sirainen}

38318f5e82662615cd88e99e398efe4a630ce020Timo Sirainen

38318f5e82662615cd88e99e398efe4a630ce020Timo Sirainenstatic int fetch_body_snippet(struct fetch_cmd_context *ctx)

38318f5e82662615cd88e99e398efe4a630ce020Timo Sirainen{

38318f5e82662615cd88e99e398efe4a630ce020Timo Sirainen const char *value;

if (mail_get_special(ctx->mail, MAIL_FETCH_BODY_SNIPPET, &value) < 0)

return -1;

/* [0] contains the snippet algorithm, skip over it */

i_assert(value[0] != '\0');

doveadm_print(value + 1);

return 0;

}

static int fetch_text(struct fetch_cmd_context *ctx)

{

struct istream *input;

if (mail_get_stream(ctx->mail, NULL, NULL, &input) < 0)

return -1;

return doveadm_print_istream(input);

}

static int fetch_text_utf8(struct fetch_cmd_context *ctx)

{

struct istream *input;

struct message_parser_ctx *parser;

struct message_decoder_context *decoder;

struct message_block raw_block, block;

struct message_part *parts;

int ret = 0;

if (mail_get_stream(ctx->mail, NULL, NULL, &input) < 0)

return -1;

parser = message_parser_init(pool_datastack_create(), input,

MESSAGE_HEADER_PARSER_FLAG_CLEAN_ONELINE,

0);

decoder = message_decoder_init(NULL, 0);

while ((ret = message_parser_parse_next_block(parser, &raw_block)) > 0) {

if (!message_decoder_decode_next_block(decoder, &raw_block,

&block))

continue;

if (block.hdr == NULL) {

if (block.size > 0)

doveadm_print_stream(block.data, block.size);

} else if (block.hdr->eoh)

doveadm_print_stream("\n", 1);

else {

i_assert(block.hdr->name_len > 0);

doveadm_print_stream(block.hdr->name,

block.hdr->name_len);

doveadm_print_stream(": ", 2);

if (block.hdr->full_value_len > 0) {

doveadm_print_stream(block.hdr->full_value,

block.hdr->full_value_len);

}

doveadm_print_stream("\n", 1);

}

}

i_assert(ret != 0);

message_decoder_deinit(&decoder);

(void)message_parser_deinit(&parser, &parts);

doveadm_print_stream("", 0);

if (input->stream_errno != 0) {

i_error("read(%s) failed: %s", i_stream_get_name(input),

i_stream_get_error(input));

return -1;

}

return 0;

}

static int fetch_size_physical(struct fetch_cmd_context *ctx)

{

uoff_t size;

if (mail_get_physical_size(ctx->mail, &size) < 0)

return -1;

doveadm_print_num(size);

return 0;

}

static int fetch_size_virtual(struct fetch_cmd_context *ctx)

{

uoff_t size;

if (mail_get_virtual_size(ctx->mail, &size) < 0)

return -1;

doveadm_print_num(size);

return 0;

}

static int fetch_date_received(struct fetch_cmd_context *ctx)

{

time_t t;

if (mail_get_received_date(ctx->mail, &t) < 0)

return -1;

doveadm_print(unixdate2str(t));

return 0;

}

static int fetch_date_sent(struct fetch_cmd_context *ctx)

{

time_t t;

int tz;

char chr;

if (mail_get_date(ctx->mail, &t, &tz) < 0)

return -1;

chr = tz < 0 ? '-' : '+';

if (tz < 0) tz = -tz;

doveadm_print(t_strdup_printf("%s (%c%02u%02u)", unixdate2str(t),

chr, tz/60, tz%60));

return 0;

}

static int fetch_date_saved(struct fetch_cmd_context *ctx)

{

time_t t;

if (mail_get_save_date(ctx->mail, &t) < 0)

return -1;

doveadm_print(unixdate2str(t));

return 0;

}

static int fetch_date_received_unixtime(struct fetch_cmd_context *ctx)

{

time_t t;

if (mail_get_received_date(ctx->mail, &t) < 0)

return -1;

doveadm_print(dec2str(t));

return 0;

}

static int fetch_date_sent_unixtime(struct fetch_cmd_context *ctx)

{

time_t t;

int tz;

if (mail_get_date(ctx->mail, &t, &tz) < 0)

return -1;

doveadm_print(dec2str(t));

return 0;

}

static int fetch_date_saved_unixtime(struct fetch_cmd_context *ctx)

{

time_t t;

if (mail_get_save_date(ctx->mail, &t) < 0)

return -1;

doveadm_print(dec2str(t));

return 0;

}

static int fetch_imap_envelope(struct fetch_cmd_context *ctx)

{

const char *value;

if (mail_get_special(ctx->mail, MAIL_FETCH_IMAP_ENVELOPE, &value) < 0)

return -1;

doveadm_print(value);

return 0;

}

static int fetch_imap_body(struct fetch_cmd_context *ctx)

{

const char *value;

if (mail_get_special(ctx->mail, MAIL_FETCH_IMAP_BODY, &value) < 0)

return -1;

doveadm_print(value);

return 0;

}

static int fetch_imap_bodystructure(struct fetch_cmd_context *ctx)

{

const char *value;

if (mail_get_special(ctx->mail, MAIL_FETCH_IMAP_BODYSTRUCTURE, &value) < 0)

return -1;

doveadm_print(value);

return 0;

}

static int fetch_pop3_uidl(struct fetch_cmd_context *ctx)

{

const char *value;

if (mail_get_special(ctx->mail, MAIL_FETCH_UIDL_BACKEND, &value) < 0)

return -1;

doveadm_print(value);

return 0;

}

static int fetch_pop3_order(struct fetch_cmd_context *ctx)

{

const char *value;

if (mail_get_special(ctx->mail, MAIL_FETCH_POP3_ORDER, &value) < 0)

return -1;

doveadm_print(value);

return 0;

}

static int fetch_refcount(struct fetch_cmd_context *ctx)

{

const char *value;

if (mail_get_special(ctx->mail, MAIL_FETCH_REFCOUNT, &value) < 0)

return -1;

doveadm_print(value);

return 0;

}

static int fetch_storageid(struct fetch_cmd_context *ctx)

{

const char *value;

if (mail_get_special(ctx->mail, MAIL_FETCH_STORAGE_ID, &value) < 0)

return -1;

doveadm_print(value);

return 0;

}

static const struct fetch_field fetch_fields[] = {

{ "user", 0, fetch_user },

{ "mailbox", 0, fetch_mailbox },

{ "mailbox-guid", 0, fetch_mailbox_guid },

{ "seq", 0, fetch_seq },

{ "uid", 0, fetch_uid },

{ "guid", 0, fetch_guid },

{ "flags", MAIL_FETCH_FLAGS, fetch_flags },

{ "modseq", 0, fetch_modseq },

{ "hdr", MAIL_FETCH_STREAM_HEADER, fetch_hdr },

{ "body", MAIL_FETCH_STREAM_BODY, fetch_body },

{ "body.snippet", MAIL_FETCH_BODY_SNIPPET, fetch_body_snippet },

{ "text", MAIL_FETCH_STREAM_HEADER |

MAIL_FETCH_STREAM_BODY, fetch_text },

{ "text.utf8", MAIL_FETCH_STREAM_HEADER |

MAIL_FETCH_STREAM_BODY, fetch_text_utf8 },

{ "size.physical", MAIL_FETCH_PHYSICAL_SIZE, fetch_size_physical },

{ "size.virtual", MAIL_FETCH_VIRTUAL_SIZE, fetch_size_virtual },

{ "date.received", MAIL_FETCH_RECEIVED_DATE, fetch_date_received },

{ "date.sent", MAIL_FETCH_DATE, fetch_date_sent },

{ "date.saved", MAIL_FETCH_SAVE_DATE, fetch_date_saved },

{ "date.received.unixtime", MAIL_FETCH_RECEIVED_DATE, fetch_date_received_unixtime },

{ "date.sent.unixtime", MAIL_FETCH_DATE, fetch_date_sent_unixtime },

{ "date.saved.unixtime", MAIL_FETCH_SAVE_DATE, fetch_date_saved_unixtime },

{ "imap.envelope", MAIL_FETCH_IMAP_ENVELOPE, fetch_imap_envelope },

{ "imap.body", MAIL_FETCH_IMAP_BODY, fetch_imap_body },

{ "imap.bodystructure", MAIL_FETCH_IMAP_BODYSTRUCTURE, fetch_imap_bodystructure },

{ "pop3.uidl", MAIL_FETCH_UIDL_BACKEND, fetch_pop3_uidl },

{ "pop3.order", MAIL_FETCH_POP3_ORDER, fetch_pop3_order },

{ "refcount", MAIL_FETCH_REFCOUNT, fetch_refcount },

{ "storageid", MAIL_FETCH_STORAGE_ID, fetch_storageid }

};

static const struct fetch_field *fetch_field_find(const char *name)

{

unsigned int i;

for (i = 0; i < N_ELEMENTS(fetch_fields); i++) {

if (strcmp(fetch_fields[i].name, name) == 0)

return &fetch_fields[i];

}

return NULL;

}

static void print_fetch_fields(void)

{

unsigned int i;

fprintf(stderr, "Available fetch fields: %s", fetch_fields[0].name);

for (i = 1; i < N_ELEMENTS(fetch_fields); i++)

fprintf(stderr, " %s", fetch_fields[i].name);

fprintf(stderr, "\n");

}

static void parse_fetch_fields(struct fetch_cmd_context *ctx, const char *str)

{

const char *const *fields, *name;

const struct fetch_field *field;

struct fetch_field hdr_field;

memset(&hdr_field, 0, sizeof(hdr_field));

hdr_field.print = fetch_hdr_field;

t_array_init(&ctx->fields, 32);

t_array_init(&ctx->header_fields, 32);

fields = t_strsplit_spaces(str, " ");

for (; *fields != NULL; fields++) {

name = t_str_lcase(*fields);

doveadm_print_header_simple(name);

if (strncmp(name, "hdr.", 4) == 0) {

name += 4;

hdr_field.name = name;

array_append(&ctx->fields, &hdr_field, 1);

name = t_strcut(name, '.');

array_append(&ctx->header_fields, &name, 1);

} else {

field = fetch_field_find(name);

if (field == NULL) {

print_fetch_fields();

i_fatal("Unknown fetch field: %s", name);

}

ctx->wanted_fields |= field->wanted_fields;

array_append(&ctx->fields, field, 1);

}

}

array_append_zero(&ctx->header_fields);

}

static int cmd_fetch_mail(struct fetch_cmd_context *ctx)

{

const struct fetch_field *field;

struct mail *mail = ctx->mail;

int ret = 0;

array_foreach(&ctx->fields, field) {

ctx->cur_field = field;

if (field->print(ctx) < 0) {

i_error("fetch(%s) failed for box=%s uid=%u: %s",

field->name, mailbox_get_vname(mail->box),

mail->uid, mailbox_get_last_error(mail->box, NULL));

doveadm_mail_failed_mailbox(&ctx->ctx, mail->box);

ret = -1;

}

}

return ret;

}

static int

cmd_fetch_box(struct fetch_cmd_context *ctx, const struct mailbox_info *info)

{

struct doveadm_mail_iter *iter;

int ret = 0;

if (doveadm_mail_iter_init(&ctx->ctx, info, ctx->ctx.search_args,

ctx->wanted_fields,

array_idx(&ctx->header_fields, 0),

&iter) < 0)

return -1;

while (doveadm_mail_iter_next(iter, &ctx->mail)) {

T_BEGIN {

if (cmd_fetch_mail(ctx) < 0)

ret = -1;

} T_END;

}

if (doveadm_mail_iter_deinit(&iter) < 0)

ret = -1;

return ret;

}

static int

cmd_fetch_run(struct doveadm_mail_cmd_context *_ctx, struct mail_user *user)

{

struct fetch_cmd_context *ctx = (struct fetch_cmd_context *)_ctx;

const enum mailbox_list_iter_flags iter_flags =

MAILBOX_LIST_ITER_NO_AUTO_BOXES |

MAILBOX_LIST_ITER_RETURN_NO_FLAGS;

struct doveadm_mailbox_list_iter *iter;

const struct mailbox_info *info;

int ret = 0;

iter = doveadm_mailbox_list_iter_init(_ctx, user, _ctx->search_args,

iter_flags);

while ((info = doveadm_mailbox_list_iter_next(iter)) != NULL) T_BEGIN {

if (cmd_fetch_box(ctx, info) < 0)

ret = -1;

} T_END;

if (doveadm_mailbox_list_iter_deinit(&iter) < 0)

ret = -1;

return ret;

}

static void cmd_fetch_init(struct doveadm_mail_cmd_context *_ctx,

const char *const args[])

{

struct fetch_cmd_context *ctx = (struct fetch_cmd_context *)_ctx;

const char *fetch_fields = args[0];

if (fetch_fields == NULL || args[1] == NULL)

doveadm_mail_help_name("fetch");

parse_fetch_fields(ctx, fetch_fields);

_ctx->search_args = doveadm_mail_build_search_args(args + 1);

}

static struct doveadm_mail_cmd_context *cmd_fetch_alloc(void)

{

struct fetch_cmd_context *ctx;

ctx = doveadm_mail_cmd_alloc(struct fetch_cmd_context);

ctx->ctx.v.init = cmd_fetch_init;

ctx->ctx.v.run = cmd_fetch_run;

doveadm_print_init("pager");

return &ctx->ctx;

}

struct doveadm_cmd_ver2 doveadm_cmd_fetch_ver2 = {

.name = "fetch",

.mail_cmd = cmd_fetch_alloc,

.usage = "<fields> <search query>",

DOVEADM_CMD_PARAM('\0', "field", CMD_PARAM_ARRAY, 0)

DOVEADM_CMD_PARAM('\0', "fieldstr", CMD_PARAM_STR, CMD_PARAM_FLAG_POSITIONAL | CMD_PARAM_FLAG_DO_NOT_EXPOSE) /* FIXME: horrible hack, remove me when possible */

DOVEADM_CMD_PARAM('\0', "query", CMD_PARAM_ARRAY, CMD_PARAM_FLAG_POSITIONAL)

};