cmd-append.c revision 8ec4e32c9c886c777954c7b9d99fc5d9250b3805
/* Copyright (C) 2002 Timo Sirainen */
#include "common.h"
#include "ioloop.h"
#include "istream.h"
#include "ostream.h"
#include "commands.h"
#include "imap-parser.h"
#include "imap-date.h"
#include <sys/time.h>
/* Returns -1 = error, 0 = need more data, 1 = successful. flags and
internal_date may be NULL as a result, but mailbox and msg_size are always
set when successful. */
static int validate_args(struct imap_arg *args, struct imap_arg_list **flags,
const char **internal_date, uoff_t *msg_size)
{
/* [<flags>] */
if (args->type != IMAP_ARG_LIST)
*flags = NULL;
else {
*flags = IMAP_ARG_LIST(args);
args++;
}
/* [<internal date>] */
if (args->type != IMAP_ARG_STRING)
*internal_date = NULL;
else {
*internal_date = IMAP_ARG_STR(args);
args++;
}
if (args->type != IMAP_ARG_LITERAL_SIZE)
return FALSE;
*msg_size = IMAP_ARG_LITERAL_SIZE(args);
return TRUE;
}
int cmd_append(struct client *client)
{
struct mailbox *box;
struct mail_save_context *ctx;
struct imap_parser *save_parser;
struct imap_arg *args;
struct imap_arg_list *flags_list;
struct mail_full_flags flags;
time_t internal_date;
const char *mailbox, *internal_date_str;
uoff_t msg_size;
unsigned int count;
int ret, failed, timezone_offset;
/* <mailbox> */
if (!client_read_string_args(client, 1, &mailbox))
return FALSE;
if (!client_verify_mailbox_name(client, mailbox, TRUE, FALSE))
return TRUE;
box = client->storage->open_mailbox(client->storage,
mailbox, FALSE, TRUE);
if (box == NULL) {
client_send_storage_error(client);
return TRUE;
}
ctx = box->save_init(box, TRUE);
if (ctx == NULL) {
client_send_storage_error(client);
return TRUE;
}
/* if error occurs, the CRLF is already read. */
client->input_skip_line = FALSE;
count = 0;
failed = TRUE;
save_parser = imap_parser_create(client->input, client->output,
0, MAX_IMAP_ARG_ELEMENTS);
for (;;) {
/* [<flags>] [<internal date>] <message literal> */
imap_parser_reset(save_parser);
for (;;) {
ret = imap_parser_read_args(save_parser, 0,
IMAP_PARSE_FLAG_LITERAL_SIZE,
&args);
if (ret >= 0)
break;
if (ret == -1) {
client_send_command_error(client,
imap_parser_get_error(save_parser));
break;
}
/* need more data */
ret = i_stream_read(client->input);
if (ret == -2) {
client_send_command_error(client,
"Too long argument.");
break;
}
if (ret < 0) {
/* disconnected */
client->cmd_error = TRUE;
break;
}
}
if (client->cmd_error)
break;
if (args->type == IMAP_ARG_EOL) {
/* last one */
if (count > 0)
failed = FALSE;
client->input_skip_line = TRUE;
break;
}
if (!validate_args(args, &flags_list, &internal_date_str,
&msg_size)) {
/* error */
client_send_command_error(client, "Invalid arguments.");
break;
}
if (flags_list != NULL) {
if (!client_parse_mail_flags(client, flags_list->args,
&flags))
break;
} else {
memset(&flags, 0, sizeof(flags));
}
if (internal_date_str == NULL) {
/* no time given, default to now. */
internal_date = ioloop_time;
timezone_offset = ioloop_timezone.tz_minuteswest;
} else if (!imap_parse_datetime(internal_date_str,
&internal_date,
&timezone_offset)) {
client_send_tagline(client,
"BAD Invalid internal date.");
break;
}
if (msg_size == 0) {
/* no message data, abort */
client_send_tagline(client, "NO Append aborted.");
break;
}
o_stream_send(client->output, "+ OK\r\n", 6);
o_stream_flush(client->output);
/* save the mail */
i_stream_set_read_limit(client->input,
client->input->v_offset + msg_size);
if (!box->save_next(ctx, &flags, internal_date,
timezone_offset, client->input)) {
client_send_storage_error(client);
break;
}
i_stream_set_read_limit(client->input, 0);
if (client->input->closed)
break;
count++;
}
if (!box->save_deinit(ctx, failed)) {
failed = TRUE;
client_send_storage_error(client);
}
box->close(box);
if (!failed) {
client_sync_full(client);
client_send_tagline(client, "OK Append completed.");
}
return TRUE;
}