2 * Copyright (C) 1996-8 Michael R. Elkins <me@mutt.org>
3 * Copyright (C) 1996-9 Brandon Long <blong@fiction.net>
4 * Copyright (C) 1999-2002 Brendan Cully <brendan@kublai.com>
6 * This program is free software; you can redistribute it and/or modify
7 * it under the terms of the GNU General Public License as published by
8 * the Free Software Foundation; either version 2 of the License, or
9 * (at your option) any later version.
11 * This program is distributed in the hope that it will be useful,
12 * but WITHOUT ANY WARRANTY; without even the implied warranty of
13 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
14 * GNU General Public License for more details.
16 * You should have received a copy of the GNU General Public License
17 * along with this program; if not, write to the Free Software
18 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA 02111, USA.
21 /* command.c: routines for sending commands to an IMAP server and parsing
25 #include "imap_private.h"
32 #define IMAP_CMD_BUFSIZE 512
34 /* forward declarations */
35 static void cmd_handle_fatal (IMAP_DATA* idata);
36 static int cmd_handle_untagged (IMAP_DATA* idata);
37 static void cmd_make_sequence (IMAP_DATA* idata);
38 static void cmd_parse_capabilities (IMAP_DATA* idata, char* s);
39 static void cmd_parse_expunge (IMAP_DATA* idata, const char* s);
40 static void cmd_parse_fetch (IMAP_DATA* idata, char* s);
41 static void cmd_parse_myrights (IMAP_DATA* idata, char* s);
43 static char *Capabilities[] = {
58 /* imap_cmd_start: Given an IMAP command, send it to the server.
59 * Currently a minor convenience, but helps to route all IMAP commands
60 * through a single interface. */
61 int imap_cmd_start (IMAP_DATA* idata, const char* cmd)
67 if (idata->status == IMAP_FATAL)
69 cmd_handle_fatal (idata);
73 cmd_make_sequence (idata);
74 /* seq, space, cmd, \r\n\0 */
75 outlen = strlen (idata->cmd.seq) + strlen (cmd) + 4;
76 out = (char*) safe_malloc (outlen);
77 snprintf (out, outlen, "%s %s\r\n", idata->cmd.seq, cmd);
79 rc = mutt_socket_write (idata->conn, out);
83 return (rc < 0) ? IMAP_CMD_BAD : 0;
86 /* imap_cmd_step: Reads server responses from an IMAP command, detects
87 * tagged completion response, handles untagged messages, can read
88 * arbitrarily large strings (using malloc, so don't make it _too_
90 int imap_cmd_step (IMAP_DATA* idata)
92 IMAP_COMMAND* cmd = &idata->cmd;
96 if (idata->status == IMAP_FATAL)
98 cmd_handle_fatal (idata);
102 /* read into buffer, expanding buffer as necessary until we have a full
106 if (len == cmd->blen)
108 safe_realloc (&cmd->buf, cmd->blen + IMAP_CMD_BUFSIZE);
109 cmd->blen = cmd->blen + IMAP_CMD_BUFSIZE;
110 dprint (3, (debugfile, "imap_cmd_step: grew buffer to %u bytes\n",
114 c = mutt_socket_readln (cmd->buf + len, cmd->blen - len, idata->conn);
117 dprint (1, (debugfile, "imap_cmd_step: Error reading server response.\n"));
118 /* cmd_handle_fatal (idata); */
124 /* if we've read all the way to the end of the buffer, we haven't read a
125 * full line (mutt_socket_readln strips the \r, so we always have at least
126 * one character free when we've read a full line) */
127 while (len == cmd->blen);
129 /* don't let one large string make cmd->buf hog memory forever */
130 if ((cmd->blen > IMAP_CMD_BUFSIZE) && (len <= IMAP_CMD_BUFSIZE))
132 safe_realloc (&cmd->buf, IMAP_CMD_BUFSIZE);
133 cmd->blen = IMAP_CMD_BUFSIZE;
134 dprint (3, (debugfile, "imap_cmd_step: shrank buffer to %u bytes\n", cmd->blen));
137 idata->lastread = time(NULL);
139 /* handle untagged messages. The caller still gets its shot afterwards. */
140 if (!ascii_strncmp (cmd->buf, "* ", 2) &&
141 cmd_handle_untagged (idata))
144 /* server demands a continuation response from us */
145 if (!ascii_strncmp (cmd->buf, "+ ", 2))
147 return IMAP_CMD_RESPOND;
150 /* tagged completion code */
151 if (!ascii_strncmp (cmd->buf, cmd->seq, SEQLEN))
153 imap_cmd_finish (idata);
154 return imap_code (cmd->buf) ? IMAP_CMD_OK : IMAP_CMD_NO;
157 return IMAP_CMD_CONTINUE;
160 /* imap_code: returns 1 if the command result was OK, or 0 if NO or BAD */
161 int imap_code (const char *s)
165 return (ascii_strncasecmp ("OK", s, 2) == 0);
168 /* imap_exec: execute a command, and wait for the response from the server.
169 * Also, handle untagged responses.
171 * IMAP_CMD_FAIL_OK: the calling procedure can handle failure. This is used
172 * for checking for a mailbox on append and login
173 * IMAP_CMD_PASS: command contains a password. Suppress logging.
174 * Return 0 on success, -1 on Failure, -2 on OK Failure
176 int imap_exec (IMAP_DATA* idata, const char* cmd, int flags)
182 if (idata->status == IMAP_FATAL)
184 cmd_handle_fatal (idata);
188 /* create sequence for command */
189 cmd_make_sequence (idata);
190 /* seq, space, cmd, \r\n\0 */
191 outlen = strlen (idata->cmd.seq) + strlen (cmd) + 4;
192 out = (char*) safe_malloc (outlen);
193 snprintf (out, outlen, "%s %s\r\n", idata->cmd.seq, cmd);
195 rc = mutt_socket_write_d (idata->conn, out,
196 flags & IMAP_CMD_PASS ? IMAP_LOG_PASS : IMAP_LOG_CMD);
201 cmd_handle_fatal (idata);
206 rc = imap_cmd_step (idata);
207 while (rc == IMAP_CMD_CONTINUE);
209 if (rc == IMAP_CMD_BAD) {
210 if (imap_reconnect(idata->ctx)!=0) {
216 if (rc == IMAP_CMD_NO && (flags & IMAP_CMD_FAIL_OK))
219 if (rc != IMAP_CMD_OK)
221 if (flags & IMAP_CMD_FAIL_OK)
224 dprint (1, (debugfile, "imap_exec: command failed: %s\n", idata->cmd.buf));
231 /* imap_cmd_running: Returns whether an IMAP command is in progress. */
232 int imap_cmd_running (IMAP_DATA* idata)
234 if (idata->cmd.state == IMAP_CMD_CONTINUE ||
235 idata->cmd.state == IMAP_CMD_RESPOND)
241 /* imap_cmd_finish: Attempts to perform cleanup (eg fetch new mail if
242 * detected, do expunge). Called automatically by imap_cmd_step, but
243 * may be called at any time. Called by imap_check_mailbox just before
244 * the index is refreshed, for instance. */
245 void imap_cmd_finish (IMAP_DATA* idata)
247 if (idata->status == IMAP_FATAL)
249 cmd_handle_fatal (idata);
253 if (!(idata->state == IMAP_SELECTED) || idata->ctx->closing)
256 if (idata->reopen & IMAP_REOPEN_ALLOW)
258 int count = idata->newMailCount;
260 if (!(idata->reopen & IMAP_EXPUNGE_PENDING) &&
261 (idata->reopen & IMAP_NEWMAIL_PENDING)
262 && count > idata->ctx->msgcount)
264 /* read new mail messages */
265 dprint (2, (debugfile, "imap_cmd_finish: Fetching new mail\n"));
266 /* check_status: curs_main uses imap_check_mailbox to detect
267 * whether the index needs updating */
268 idata->check_status = IMAP_NEWMAIL_PENDING;
269 imap_read_headers (idata, idata->ctx->msgcount, count-1);
271 else if (idata->reopen & IMAP_EXPUNGE_PENDING)
273 dprint (2, (debugfile, "imap_cmd_finish: Expunging mailbox\n"));
274 imap_expunge_mailbox (idata);
275 /* Detect whether we've gotten unexpected EXPUNGE messages */
276 if (idata->reopen & IMAP_EXPUNGE_PENDING &&
277 !(idata->reopen & IMAP_EXPUNGE_EXPECTED))
278 idata->check_status = IMAP_EXPUNGE_PENDING;
279 idata->reopen &= ~(IMAP_EXPUNGE_PENDING | IMAP_NEWMAIL_PENDING |
280 IMAP_EXPUNGE_EXPECTED);
287 /* cmd_handle_fatal: when IMAP_DATA is in fatal state, do what we can */
288 static void cmd_handle_fatal (IMAP_DATA* idata)
290 idata->status = IMAP_FATAL;
292 if ((idata->state == IMAP_SELECTED) &&
293 (idata->reopen & IMAP_REOPEN_ALLOW) &&
294 !idata->ctx->closing)
296 /*mx_fastclose_mailbox (idata->ctx);*/
297 mutt_error (_("Mailbox closed"));
299 idata->state = IMAP_DISCONNECTED;
300 if (imap_reconnect(idata->ctx)!=0)
301 mx_fastclose_mailbox(idata->ctx);
304 if (idata->state != IMAP_SELECTED)
306 idata->state = IMAP_DISCONNECTED;
311 /* cmd_handle_untagged: fallback parser for otherwise unhandled messages. */
312 static int cmd_handle_untagged (IMAP_DATA* idata)
318 s = imap_next_word (idata->cmd.buf);
320 if ((idata->state == IMAP_SELECTED) && isdigit ((unsigned char) *s))
323 s = imap_next_word (s);
325 /* EXISTS and EXPUNGE are always related to the SELECTED mailbox for the
326 * connection, so update that one.
328 if (ascii_strncasecmp ("EXISTS", s, 6) == 0)
330 dprint (2, (debugfile, "Handling EXISTS\n"));
332 /* new mail arrived */
335 if ( !(idata->reopen & IMAP_EXPUNGE_PENDING) &&
336 count < idata->ctx->msgcount)
338 /* something is wrong because the server reported fewer messages
339 * than we previously saw
341 mutt_error _("Fatal error. Message count is out of sync!");
342 idata->status = IMAP_FATAL;
345 /* at least the InterChange server sends EXISTS messages freely,
346 * even when there is no new mail */
347 else if (count == idata->ctx->msgcount)
348 dprint (3, (debugfile,
349 "cmd_handle_untagged: superfluous EXISTS message.\n"));
352 if (!(idata->reopen & IMAP_EXPUNGE_PENDING))
354 dprint (2, (debugfile,
355 "cmd_handle_untagged: New mail in %s - %d messages total.\n",
356 idata->mailbox, count));
357 idata->reopen |= IMAP_NEWMAIL_PENDING;
359 idata->newMailCount = count;
362 /* pn vs. s: need initial seqno */
363 else if (ascii_strncasecmp ("EXPUNGE", s, 7) == 0)
364 cmd_parse_expunge (idata, pn);
365 else if (ascii_strncasecmp ("FETCH", s, 5) == 0)
366 cmd_parse_fetch (idata, pn);
368 else if (ascii_strncasecmp ("CAPABILITY", s, 10) == 0)
369 cmd_parse_capabilities (idata, s);
370 else if (ascii_strncasecmp ("MYRIGHTS", s, 8) == 0)
371 cmd_parse_myrights (idata, s);
372 else if (ascii_strncasecmp ("BYE", s, 3) == 0)
374 dprint (2, (debugfile, "Handling BYE\n"));
376 /* check if we're logging out */
377 if (idata->status == IMAP_BYE)
380 /* server shut down our connection */
383 mutt_error ("%s", s);
384 idata->status = IMAP_BYE;
386 /*if (imap_reconnect(idata->ctx)!=0) {
387 if (idata->state == IMAP_SELECTED)
388 mx_fastclose_mailbox (idata->ctx); */ /* XXX memleak? */
389 mutt_socket_close (idata->conn);
390 idata->state = IMAP_DISCONNECTED;
396 else if (option (OPTIMAPSERVERNOISE) && (ascii_strncasecmp ("NO", s, 2) == 0))
398 dprint (2, (debugfile, "Handling untagged NO\n"));
400 /* Display the warning message from the server */
401 mutt_error ("%s", s+3);
408 /* cmd_make_sequence: make a tag suitable for starting an IMAP command */
409 static void cmd_make_sequence (IMAP_DATA* idata)
411 snprintf (idata->cmd.seq, sizeof (idata->cmd.seq), "a%04u", idata->seqno++);
413 if (idata->seqno > 9999)
417 /* cmd_parse_capabilities: set capability bits according to CAPABILITY
419 static void cmd_parse_capabilities (IMAP_DATA* idata, char* s)
423 dprint (2, (debugfile, "Handling CAPABILITY\n"));
425 s = imap_next_word (s);
426 FREE(&idata->capstr);
427 idata->capstr = safe_strdup (s);
429 memset (idata->capabilities, 0, sizeof (idata->capabilities));
433 for (x = 0; x < CAPMAX; x++)
434 if (imap_wordcasecmp(Capabilities[x], s) == 0)
436 mutt_bit_set (idata->capabilities, x);
439 s = imap_next_word (s);
443 /* cmd_parse_expunge: mark headers with new sequence ID and mark idata to
444 * be reopened at our earliest convenience */
445 static void cmd_parse_expunge (IMAP_DATA* idata, const char* s)
450 dprint (2, (debugfile, "Handling EXPUNGE\n"));
454 /* walk headers, zero seqno of expunged message, decrement seqno of those
455 * above. Possibly we could avoid walking the whole list by resorting
456 * and guessing a good starting point, but I'm guessing the resort would
457 * nullify the gains */
458 for (cur = 0; cur < idata->ctx->msgcount; cur++)
460 h = idata->ctx->hdrs[cur];
462 if (h->index+1 == expno)
464 else if (h->index+1 > expno)
468 idata->reopen |= IMAP_EXPUNGE_PENDING;
471 /* cmd_parse_fetch: Load fetch response into IMAP_DATA. Currently only
472 * handles unanticipated FETCH responses, and only FLAGS data. We get
473 * these if another client has changed flags for a mailbox we've selected.
474 * Of course, a lot of code here duplicates code in message.c. */
475 static void cmd_parse_fetch (IMAP_DATA* idata, char* s)
480 dprint (2, (debugfile, "Handling FETCH\n"));
484 if (msgno <= idata->ctx->msgcount)
485 /* see cmd_parse_expunge */
486 for (cur = 0; cur < idata->ctx->msgcount; cur++)
488 h = idata->ctx->hdrs[cur];
490 if (h->active && h->index+1 == msgno)
492 dprint (2, (debugfile, "Message UID %d updated\n", HEADER_DATA(h)->uid));
501 dprint (1, (debugfile, "FETCH response ignored for this message\n"));
506 s = imap_next_word (s);
507 s = imap_next_word (s);
511 dprint (1, (debugfile, "Malformed FETCH response"));
516 if (ascii_strncasecmp ("FLAGS", s, 5) != 0)
518 dprint (2, (debugfile, "Only handle FLAGS updates\n"));
522 /* If server flags could conflict with mutt's flags, reopen the mailbox. */
524 idata->reopen |= IMAP_EXPUNGE_PENDING;
526 imap_set_flags (idata, h, s);
527 idata->check_status = IMAP_FLAGS_PENDING;
531 /* cmd_parse_myrights: set rights bits according to MYRIGHTS response */
532 static void cmd_parse_myrights (IMAP_DATA* idata, char* s)
534 dprint (2, (debugfile, "Handling MYRIGHTS\n"));
536 s = imap_next_word (s);
537 s = imap_next_word (s);
539 /* zero out current rights set */
540 memset (idata->rights, 0, sizeof (idata->rights));
542 while (*s && !isspace((unsigned char) *s))
547 mutt_bit_set (idata->rights, IMAP_ACL_LOOKUP);
550 mutt_bit_set (idata->rights, IMAP_ACL_READ);
553 mutt_bit_set (idata->rights, IMAP_ACL_SEEN);
556 mutt_bit_set (idata->rights, IMAP_ACL_WRITE);
559 mutt_bit_set (idata->rights, IMAP_ACL_INSERT);
562 mutt_bit_set (idata->rights, IMAP_ACL_POST);
565 mutt_bit_set (idata->rights, IMAP_ACL_CREATE);
568 mutt_bit_set (idata->rights, IMAP_ACL_DELETE);
571 mutt_bit_set (idata->rights, IMAP_ACL_ADMIN);