cmd-fetch.c revision 294f1a51763015cda0e2d874c5027d6fe7a2cd54
/* Copyright (C) 2002-2004 Timo Sirainen */
#include "common.h"
#include "ostream.h"
#include "commands.h"
#include "imap-fetch.h"
#include "imap-search.h"
#include "mail-search.h"
const char *all_macro[] = {
"FLAGS", "INTERNALDATE", "RFC822.SIZE", "ENVELOPE", NULL
};
const char *fast_macro[] = {
"FLAGS", "INTERNALDATE", "RFC822.SIZE", NULL
};
const char *full_macro[] = {
"FLAGS", "INTERNALDATE", "RFC822.SIZE", "ENVELOPE", "BODY", NULL
};
static int
fetch_parse_args(struct imap_fetch_context *ctx, struct imap_arg *arg)
{
const char *str, *const *macro;
if (arg->type == IMAP_ARG_ATOM) {
str = str_ucase(IMAP_ARG_STR(arg));
arg++;
/* handle macros first */
if (strcmp(str, "ALL") == 0)
macro = all_macro;
else if (strcmp(str, "FAST") == 0)
macro = fast_macro;
else if (strcmp(str, "FULL") == 0)
macro = full_macro;
else {
macro = NULL;
if (!imap_fetch_init_handler(ctx, str, &arg))
return FALSE;
}
if (macro != NULL) {
while (*macro != NULL) {
if (!imap_fetch_init_handler(ctx, *macro, &arg))
return FALSE;
macro++;
}
}
} else {
arg = IMAP_ARG_LIST(arg)->args;
while (arg->type == IMAP_ARG_ATOM) {
str = str_ucase(IMAP_ARG_STR(arg));
arg++;
if (!imap_fetch_init_handler(ctx, str, &arg))
return FALSE;
}
if (arg->type != IMAP_ARG_EOL) {
client_send_command_error(ctx->cmd,
"FETCH list contains non-atoms.");
return FALSE;
}
}
if (ctx->cmd->uid) {
if (!imap_fetch_init_handler(ctx, "UID", &arg))
return FALSE;
}
return TRUE;
}
static int cmd_fetch_finish(struct imap_fetch_context *ctx)
{
struct client_command_context *cmd = ctx->cmd;
static const char *ok_message = "OK Fetch completed.";
int failed, partial;
partial = ctx->partial_fetch;
failed = ctx->failed;
if (imap_fetch_deinit(ctx) < 0)
failed = TRUE;
if (failed || (partial && !cmd->uid)) {
struct mail_storage *storage;
const char *error;
int syntax;
storage = mailbox_get_storage(cmd->client->mailbox);
error = mail_storage_get_last_error(storage, &syntax);
if (!syntax) {
/* We never want to reply NO to FETCH requests,
BYE is preferrable (see imap-ml for reasons). */
if (partial) {
error = "Out of sync: "
"Trying to fetch expunged message";
}
client_disconnect_with_error(cmd->client, error);
} else {
/* user error, we'll reply with BAD */
client_send_storage_error(cmd, storage);
}
return TRUE;
}
return cmd_sync(cmd, MAILBOX_SYNC_FLAG_FAST |
(cmd->uid ? 0 : MAILBOX_SYNC_FLAG_NO_EXPUNGES),
ok_message);
}
static int cmd_fetch_continue(struct client_command_context *cmd)
{
struct imap_fetch_context *ctx = cmd->context;
int ret;
if (cmd->client->output->closed)
ret = -1;
else {
if ((ret = imap_fetch(ctx)) == 0) {
/* unfinished */
return FALSE;
}
}
if (ret < 0)
ctx->failed = TRUE;
return cmd_fetch_finish(ctx);
}
int cmd_fetch(struct client_command_context *cmd)
{
struct client *client = cmd->client;
struct imap_fetch_context *ctx;
struct imap_arg *args;
struct mail_search_arg *search_arg;
const char *messageset;
int ret;
if (!client_read_args(cmd, 0, 0, &args))
return FALSE;
if (!client_verify_open_mailbox(cmd))
return TRUE;
messageset = imap_arg_string(&args[0]);
if (messageset == NULL ||
(args[1].type != IMAP_ARG_LIST && args[1].type != IMAP_ARG_ATOM)) {
client_send_command_error(cmd, "Invalid arguments.");
return TRUE;
}
search_arg = imap_search_get_arg(cmd, messageset, cmd->uid);
if (search_arg == NULL)
return TRUE;
ctx = imap_fetch_init(cmd);
if (ctx == NULL)
return TRUE;
if (!fetch_parse_args(ctx, &args[1])) {
imap_fetch_deinit(ctx);
return TRUE;
}
imap_fetch_begin(ctx, search_arg);
if ((ret = imap_fetch(ctx)) == 0) {
/* unfinished */
client->command_pending = TRUE;
cmd->func = cmd_fetch_continue;
cmd->context = ctx;
return FALSE;
}
if (ret < 0)
ctx->failed = TRUE;
return cmd_fetch_finish(ctx);
}