imap-send.c: inline parse_imap_list() in parse_list()
[git.git] / imap-send.c
1 /*
2  * git-imap-send - drops patches into an imap Drafts folder
3  *                 derived from isync/mbsync - mailbox synchronizer
4  *
5  * Copyright (C) 2000-2002 Michael R. Elkins <me@mutt.org>
6  * Copyright (C) 2002-2004 Oswald Buddenhagen <ossi@users.sf.net>
7  * Copyright (C) 2004 Theodore Y. Ts'o <tytso@mit.edu>
8  * Copyright (C) 2006 Mike McCormack
9  *
10  *  This program is free software; you can redistribute it and/or modify
11  *  it under the terms of the GNU General Public License as published by
12  *  the Free Software Foundation; either version 2 of the License, or
13  *  (at your option) any later version.
14  *
15  *  This program is distributed in the hope that it will be useful,
16  *  but WITHOUT ANY WARRANTY; without even the implied warranty of
17  *  MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
18  *  GNU General Public License for more details.
19  *
20  *  You should have received a copy of the GNU General Public License
21  *  along with this program; if not, write to the Free Software
22  *  Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
23  */
24
25 #include "cache.h"
26 #include "exec_cmd.h"
27 #include "run-command.h"
28 #include "prompt.h"
29 #ifdef NO_OPENSSL
30 typedef void *SSL;
31 #else
32 #include <openssl/evp.h>
33 #include <openssl/hmac.h>
34 #endif
35
36 struct store {
37         /* currently open mailbox */
38         const char *name; /* foreign! maybe preset? */
39         int uidvalidity;
40 };
41
42 static const char imap_send_usage[] = "git imap-send < <mbox>";
43
44 #undef DRV_OK
45 #define DRV_OK          0
46 #define DRV_MSG_BAD     -1
47 #define DRV_BOX_BAD     -2
48 #define DRV_STORE_BAD   -3
49
50 static int Verbose, Quiet;
51
52 __attribute__((format (printf, 1, 2)))
53 static void imap_info(const char *, ...);
54 __attribute__((format (printf, 1, 2)))
55 static void imap_warn(const char *, ...);
56
57 static char *next_arg(char **);
58
59 __attribute__((format (printf, 3, 4)))
60 static int nfsnprintf(char *buf, int blen, const char *fmt, ...);
61
62 static int nfvasprintf(char **strp, const char *fmt, va_list ap)
63 {
64         int len;
65         char tmp[8192];
66
67         len = vsnprintf(tmp, sizeof(tmp), fmt, ap);
68         if (len < 0)
69                 die("Fatal: Out of memory");
70         if (len >= sizeof(tmp))
71                 die("imap command overflow!");
72         *strp = xmemdupz(tmp, len);
73         return len;
74 }
75
76 struct imap_server_conf {
77         char *name;
78         char *tunnel;
79         char *host;
80         int port;
81         char *user;
82         char *pass;
83         int use_ssl;
84         int ssl_verify;
85         int use_html;
86         char *auth_method;
87 };
88
89 static struct imap_server_conf server = {
90         NULL,   /* name */
91         NULL,   /* tunnel */
92         NULL,   /* host */
93         0,      /* port */
94         NULL,   /* user */
95         NULL,   /* pass */
96         0,      /* use_ssl */
97         1,      /* ssl_verify */
98         0,      /* use_html */
99         NULL,   /* auth_method */
100 };
101
102 #define NIL     (void *)0x1
103 #define LIST    (void *)0x2
104
105 struct imap_list {
106         struct imap_list *next, *child;
107         char *val;
108         int len;
109 };
110
111 struct imap_socket {
112         int fd[2];
113         SSL *ssl;
114 };
115
116 struct imap_buffer {
117         struct imap_socket sock;
118         int bytes;
119         int offset;
120         char buf[1024];
121 };
122
123 struct imap_cmd;
124
125 struct imap {
126         int uidnext; /* from SELECT responses */
127         struct imap_list *ns_personal, *ns_other, *ns_shared; /* NAMESPACE info */
128         unsigned caps, rcaps; /* CAPABILITY results */
129         /* command queue */
130         int nexttag, num_in_progress, literal_pending;
131         struct imap_cmd *in_progress, **in_progress_append;
132         struct imap_buffer buf; /* this is BIG, so put it last */
133 };
134
135 struct imap_store {
136         struct store gen;
137         int uidvalidity;
138         struct imap *imap;
139         const char *prefix;
140         unsigned /*currentnc:1,*/ trashnc:1;
141 };
142
143 struct imap_cmd_cb {
144         int (*cont)(struct imap_store *ctx, struct imap_cmd *cmd, const char *prompt);
145         void (*done)(struct imap_store *ctx, struct imap_cmd *cmd, int response);
146         void *ctx;
147         char *data;
148         int dlen;
149         int uid;
150         unsigned create:1, trycreate:1;
151 };
152
153 struct imap_cmd {
154         struct imap_cmd *next;
155         struct imap_cmd_cb cb;
156         char *cmd;
157         int tag;
158 };
159
160 #define CAP(cap) (imap->caps & (1 << (cap)))
161
162 enum CAPABILITY {
163         NOLOGIN = 0,
164         UIDPLUS,
165         LITERALPLUS,
166         NAMESPACE,
167         STARTTLS,
168         AUTH_CRAM_MD5
169 };
170
171 static const char *cap_list[] = {
172         "LOGINDISABLED",
173         "UIDPLUS",
174         "LITERAL+",
175         "NAMESPACE",
176         "STARTTLS",
177         "AUTH=CRAM-MD5",
178 };
179
180 #define RESP_OK    0
181 #define RESP_NO    1
182 #define RESP_BAD   2
183
184 static int get_cmd_result(struct imap_store *ctx, struct imap_cmd *tcmd);
185
186
187 #ifndef NO_OPENSSL
188 static void ssl_socket_perror(const char *func)
189 {
190         fprintf(stderr, "%s: %s\n", func, ERR_error_string(ERR_get_error(), NULL));
191 }
192 #endif
193
194 static void socket_perror(const char *func, struct imap_socket *sock, int ret)
195 {
196 #ifndef NO_OPENSSL
197         if (sock->ssl) {
198                 int sslerr = SSL_get_error(sock->ssl, ret);
199                 switch (sslerr) {
200                 case SSL_ERROR_NONE:
201                         break;
202                 case SSL_ERROR_SYSCALL:
203                         perror("SSL_connect");
204                         break;
205                 default:
206                         ssl_socket_perror("SSL_connect");
207                         break;
208                 }
209         } else
210 #endif
211         {
212                 if (ret < 0)
213                         perror(func);
214                 else
215                         fprintf(stderr, "%s: unexpected EOF\n", func);
216         }
217 }
218
219 static int ssl_socket_connect(struct imap_socket *sock, int use_tls_only, int verify)
220 {
221 #ifdef NO_OPENSSL
222         fprintf(stderr, "SSL requested but SSL support not compiled in\n");
223         return -1;
224 #else
225 #if (OPENSSL_VERSION_NUMBER >= 0x10000000L)
226         const SSL_METHOD *meth;
227 #else
228         SSL_METHOD *meth;
229 #endif
230         SSL_CTX *ctx;
231         int ret;
232
233         SSL_library_init();
234         SSL_load_error_strings();
235
236         if (use_tls_only)
237                 meth = TLSv1_method();
238         else
239                 meth = SSLv23_method();
240
241         if (!meth) {
242                 ssl_socket_perror("SSLv23_method");
243                 return -1;
244         }
245
246         ctx = SSL_CTX_new(meth);
247
248         if (verify)
249                 SSL_CTX_set_verify(ctx, SSL_VERIFY_PEER, NULL);
250
251         if (!SSL_CTX_set_default_verify_paths(ctx)) {
252                 ssl_socket_perror("SSL_CTX_set_default_verify_paths");
253                 return -1;
254         }
255         sock->ssl = SSL_new(ctx);
256         if (!sock->ssl) {
257                 ssl_socket_perror("SSL_new");
258                 return -1;
259         }
260         if (!SSL_set_rfd(sock->ssl, sock->fd[0])) {
261                 ssl_socket_perror("SSL_set_rfd");
262                 return -1;
263         }
264         if (!SSL_set_wfd(sock->ssl, sock->fd[1])) {
265                 ssl_socket_perror("SSL_set_wfd");
266                 return -1;
267         }
268
269         ret = SSL_connect(sock->ssl);
270         if (ret <= 0) {
271                 socket_perror("SSL_connect", sock, ret);
272                 return -1;
273         }
274
275         return 0;
276 #endif
277 }
278
279 static int socket_read(struct imap_socket *sock, char *buf, int len)
280 {
281         ssize_t n;
282 #ifndef NO_OPENSSL
283         if (sock->ssl)
284                 n = SSL_read(sock->ssl, buf, len);
285         else
286 #endif
287                 n = xread(sock->fd[0], buf, len);
288         if (n <= 0) {
289                 socket_perror("read", sock, n);
290                 close(sock->fd[0]);
291                 close(sock->fd[1]);
292                 sock->fd[0] = sock->fd[1] = -1;
293         }
294         return n;
295 }
296
297 static int socket_write(struct imap_socket *sock, const char *buf, int len)
298 {
299         int n;
300 #ifndef NO_OPENSSL
301         if (sock->ssl)
302                 n = SSL_write(sock->ssl, buf, len);
303         else
304 #endif
305                 n = write_in_full(sock->fd[1], buf, len);
306         if (n != len) {
307                 socket_perror("write", sock, n);
308                 close(sock->fd[0]);
309                 close(sock->fd[1]);
310                 sock->fd[0] = sock->fd[1] = -1;
311         }
312         return n;
313 }
314
315 static void socket_shutdown(struct imap_socket *sock)
316 {
317 #ifndef NO_OPENSSL
318         if (sock->ssl) {
319                 SSL_shutdown(sock->ssl);
320                 SSL_free(sock->ssl);
321         }
322 #endif
323         close(sock->fd[0]);
324         close(sock->fd[1]);
325 }
326
327 /* simple line buffering */
328 static int buffer_gets(struct imap_buffer *b, char **s)
329 {
330         int n;
331         int start = b->offset;
332
333         *s = b->buf + start;
334
335         for (;;) {
336                 /* make sure we have enough data to read the \r\n sequence */
337                 if (b->offset + 1 >= b->bytes) {
338                         if (start) {
339                                 /* shift down used bytes */
340                                 *s = b->buf;
341
342                                 assert(start <= b->bytes);
343                                 n = b->bytes - start;
344
345                                 if (n)
346                                         memmove(b->buf, b->buf + start, n);
347                                 b->offset -= start;
348                                 b->bytes = n;
349                                 start = 0;
350                         }
351
352                         n = socket_read(&b->sock, b->buf + b->bytes,
353                                          sizeof(b->buf) - b->bytes);
354
355                         if (n <= 0)
356                                 return -1;
357
358                         b->bytes += n;
359                 }
360
361                 if (b->buf[b->offset] == '\r') {
362                         assert(b->offset + 1 < b->bytes);
363                         if (b->buf[b->offset + 1] == '\n') {
364                                 b->buf[b->offset] = 0;  /* terminate the string */
365                                 b->offset += 2; /* next line */
366                                 if (Verbose)
367                                         puts(*s);
368                                 return 0;
369                         }
370                 }
371
372                 b->offset++;
373         }
374         /* not reached */
375 }
376
377 static void imap_info(const char *msg, ...)
378 {
379         va_list va;
380
381         if (!Quiet) {
382                 va_start(va, msg);
383                 vprintf(msg, va);
384                 va_end(va);
385                 fflush(stdout);
386         }
387 }
388
389 static void imap_warn(const char *msg, ...)
390 {
391         va_list va;
392
393         if (Quiet < 2) {
394                 va_start(va, msg);
395                 vfprintf(stderr, msg, va);
396                 va_end(va);
397         }
398 }
399
400 static char *next_arg(char **s)
401 {
402         char *ret;
403
404         if (!s || !*s)
405                 return NULL;
406         while (isspace((unsigned char) **s))
407                 (*s)++;
408         if (!**s) {
409                 *s = NULL;
410                 return NULL;
411         }
412         if (**s == '"') {
413                 ++*s;
414                 ret = *s;
415                 *s = strchr(*s, '"');
416         } else {
417                 ret = *s;
418                 while (**s && !isspace((unsigned char) **s))
419                         (*s)++;
420         }
421         if (*s) {
422                 if (**s)
423                         *(*s)++ = 0;
424                 if (!**s)
425                         *s = NULL;
426         }
427         return ret;
428 }
429
430 static int nfsnprintf(char *buf, int blen, const char *fmt, ...)
431 {
432         int ret;
433         va_list va;
434
435         va_start(va, fmt);
436         if (blen <= 0 || (unsigned)(ret = vsnprintf(buf, blen, fmt, va)) >= (unsigned)blen)
437                 die("Fatal: buffer too small. Please report a bug.");
438         va_end(va);
439         return ret;
440 }
441
442 static struct imap_cmd *v_issue_imap_cmd(struct imap_store *ctx,
443                                          struct imap_cmd_cb *cb,
444                                          const char *fmt, va_list ap)
445 {
446         struct imap *imap = ctx->imap;
447         struct imap_cmd *cmd;
448         int n, bufl;
449         char buf[1024];
450
451         cmd = xmalloc(sizeof(struct imap_cmd));
452         nfvasprintf(&cmd->cmd, fmt, ap);
453         cmd->tag = ++imap->nexttag;
454
455         if (cb)
456                 cmd->cb = *cb;
457         else
458                 memset(&cmd->cb, 0, sizeof(cmd->cb));
459
460         while (imap->literal_pending)
461                 get_cmd_result(ctx, NULL);
462
463         if (!cmd->cb.data)
464                 bufl = nfsnprintf(buf, sizeof(buf), "%d %s\r\n", cmd->tag, cmd->cmd);
465         else
466                 bufl = nfsnprintf(buf, sizeof(buf), "%d %s{%d%s}\r\n",
467                                   cmd->tag, cmd->cmd, cmd->cb.dlen,
468                                   CAP(LITERALPLUS) ? "+" : "");
469
470         if (Verbose) {
471                 if (imap->num_in_progress)
472                         printf("(%d in progress) ", imap->num_in_progress);
473                 if (memcmp(cmd->cmd, "LOGIN", 5))
474                         printf(">>> %s", buf);
475                 else
476                         printf(">>> %d LOGIN <user> <pass>\n", cmd->tag);
477         }
478         if (socket_write(&imap->buf.sock, buf, bufl) != bufl) {
479                 free(cmd->cmd);
480                 free(cmd);
481                 if (cb)
482                         free(cb->data);
483                 return NULL;
484         }
485         if (cmd->cb.data) {
486                 if (CAP(LITERALPLUS)) {
487                         n = socket_write(&imap->buf.sock, cmd->cb.data, cmd->cb.dlen);
488                         free(cmd->cb.data);
489                         if (n != cmd->cb.dlen ||
490                             socket_write(&imap->buf.sock, "\r\n", 2) != 2) {
491                                 free(cmd->cmd);
492                                 free(cmd);
493                                 return NULL;
494                         }
495                         cmd->cb.data = NULL;
496                 } else
497                         imap->literal_pending = 1;
498         } else if (cmd->cb.cont)
499                 imap->literal_pending = 1;
500         cmd->next = NULL;
501         *imap->in_progress_append = cmd;
502         imap->in_progress_append = &cmd->next;
503         imap->num_in_progress++;
504         return cmd;
505 }
506
507 __attribute__((format (printf, 3, 4)))
508 static struct imap_cmd *issue_imap_cmd(struct imap_store *ctx,
509                                        struct imap_cmd_cb *cb,
510                                        const char *fmt, ...)
511 {
512         struct imap_cmd *ret;
513         va_list ap;
514
515         va_start(ap, fmt);
516         ret = v_issue_imap_cmd(ctx, cb, fmt, ap);
517         va_end(ap);
518         return ret;
519 }
520
521 __attribute__((format (printf, 3, 4)))
522 static int imap_exec(struct imap_store *ctx, struct imap_cmd_cb *cb,
523                      const char *fmt, ...)
524 {
525         va_list ap;
526         struct imap_cmd *cmdp;
527
528         va_start(ap, fmt);
529         cmdp = v_issue_imap_cmd(ctx, cb, fmt, ap);
530         va_end(ap);
531         if (!cmdp)
532                 return RESP_BAD;
533
534         return get_cmd_result(ctx, cmdp);
535 }
536
537 __attribute__((format (printf, 3, 4)))
538 static int imap_exec_m(struct imap_store *ctx, struct imap_cmd_cb *cb,
539                        const char *fmt, ...)
540 {
541         va_list ap;
542         struct imap_cmd *cmdp;
543
544         va_start(ap, fmt);
545         cmdp = v_issue_imap_cmd(ctx, cb, fmt, ap);
546         va_end(ap);
547         if (!cmdp)
548                 return DRV_STORE_BAD;
549
550         switch (get_cmd_result(ctx, cmdp)) {
551         case RESP_BAD: return DRV_STORE_BAD;
552         case RESP_NO: return DRV_MSG_BAD;
553         default: return DRV_OK;
554         }
555 }
556
557 static int is_atom(struct imap_list *list)
558 {
559         return list && list->val && list->val != NIL && list->val != LIST;
560 }
561
562 static int is_list(struct imap_list *list)
563 {
564         return list && list->val == LIST;
565 }
566
567 static void free_list(struct imap_list *list)
568 {
569         struct imap_list *tmp;
570
571         for (; list; list = tmp) {
572                 tmp = list->next;
573                 if (is_list(list))
574                         free_list(list->child);
575                 else if (is_atom(list))
576                         free(list->val);
577                 free(list);
578         }
579 }
580
581 static int parse_imap_list_l(struct imap *imap, char **sp, struct imap_list **curp, int level)
582 {
583         struct imap_list *cur;
584         char *s = *sp, *p;
585         int n, bytes;
586
587         for (;;) {
588                 while (isspace((unsigned char)*s))
589                         s++;
590                 if (level && *s == ')') {
591                         s++;
592                         break;
593                 }
594                 *curp = cur = xmalloc(sizeof(*cur));
595                 curp = &cur->next;
596                 cur->val = NULL; /* for clean bail */
597                 if (*s == '(') {
598                         /* sublist */
599                         s++;
600                         cur->val = LIST;
601                         if (parse_imap_list_l(imap, &s, &cur->child, level + 1))
602                                 goto bail;
603                 } else if (imap && *s == '{') {
604                         /* literal */
605                         bytes = cur->len = strtol(s + 1, &s, 10);
606                         if (*s != '}')
607                                 goto bail;
608
609                         s = cur->val = xmalloc(cur->len);
610
611                         /* dump whats left over in the input buffer */
612                         n = imap->buf.bytes - imap->buf.offset;
613
614                         if (n > bytes)
615                                 /* the entire message fit in the buffer */
616                                 n = bytes;
617
618                         memcpy(s, imap->buf.buf + imap->buf.offset, n);
619                         s += n;
620                         bytes -= n;
621
622                         /* mark that we used part of the buffer */
623                         imap->buf.offset += n;
624
625                         /* now read the rest of the message */
626                         while (bytes > 0) {
627                                 if ((n = socket_read(&imap->buf.sock, s, bytes)) <= 0)
628                                         goto bail;
629                                 s += n;
630                                 bytes -= n;
631                         }
632
633                         if (buffer_gets(&imap->buf, &s))
634                                 goto bail;
635                 } else if (*s == '"') {
636                         /* quoted string */
637                         s++;
638                         p = s;
639                         for (; *s != '"'; s++)
640                                 if (!*s)
641                                         goto bail;
642                         cur->len = s - p;
643                         s++;
644                         cur->val = xmemdupz(p, cur->len);
645                 } else {
646                         /* atom */
647                         p = s;
648                         for (; *s && !isspace((unsigned char)*s); s++)
649                                 if (level && *s == ')')
650                                         break;
651                         cur->len = s - p;
652                         if (cur->len == 3 && !memcmp("NIL", p, 3))
653                                 cur->val = NIL;
654                         else
655                                 cur->val = xmemdupz(p, cur->len);
656                 }
657
658                 if (!level)
659                         break;
660                 if (!*s)
661                         goto bail;
662         }
663         *sp = s;
664         *curp = NULL;
665         return 0;
666
667 bail:
668         *curp = NULL;
669         return -1;
670 }
671
672 static struct imap_list *parse_list(char **sp)
673 {
674         struct imap_list *head;
675
676         if (!parse_imap_list_l(NULL, sp, &head, 0))
677                 return head;
678         free_list(head);
679         return NULL;
680 }
681
682 static void parse_capability(struct imap *imap, char *cmd)
683 {
684         char *arg;
685         unsigned i;
686
687         imap->caps = 0x80000000;
688         while ((arg = next_arg(&cmd)))
689                 for (i = 0; i < ARRAY_SIZE(cap_list); i++)
690                         if (!strcmp(cap_list[i], arg))
691                                 imap->caps |= 1 << i;
692         imap->rcaps = imap->caps;
693 }
694
695 static int parse_response_code(struct imap_store *ctx, struct imap_cmd_cb *cb,
696                                char *s)
697 {
698         struct imap *imap = ctx->imap;
699         char *arg, *p;
700
701         if (*s != '[')
702                 return RESP_OK;         /* no response code */
703         s++;
704         if (!(p = strchr(s, ']'))) {
705                 fprintf(stderr, "IMAP error: malformed response code\n");
706                 return RESP_BAD;
707         }
708         *p++ = 0;
709         arg = next_arg(&s);
710         if (!strcmp("UIDVALIDITY", arg)) {
711                 if (!(arg = next_arg(&s)) || !(ctx->gen.uidvalidity = atoi(arg))) {
712                         fprintf(stderr, "IMAP error: malformed UIDVALIDITY status\n");
713                         return RESP_BAD;
714                 }
715         } else if (!strcmp("UIDNEXT", arg)) {
716                 if (!(arg = next_arg(&s)) || !(imap->uidnext = atoi(arg))) {
717                         fprintf(stderr, "IMAP error: malformed NEXTUID status\n");
718                         return RESP_BAD;
719                 }
720         } else if (!strcmp("CAPABILITY", arg)) {
721                 parse_capability(imap, s);
722         } else if (!strcmp("ALERT", arg)) {
723                 /* RFC2060 says that these messages MUST be displayed
724                  * to the user
725                  */
726                 for (; isspace((unsigned char)*p); p++);
727                 fprintf(stderr, "*** IMAP ALERT *** %s\n", p);
728         } else if (cb && cb->ctx && !strcmp("APPENDUID", arg)) {
729                 if (!(arg = next_arg(&s)) || !(ctx->gen.uidvalidity = atoi(arg)) ||
730                     !(arg = next_arg(&s)) || !(*(int *)cb->ctx = atoi(arg))) {
731                         fprintf(stderr, "IMAP error: malformed APPENDUID status\n");
732                         return RESP_BAD;
733                 }
734         }
735         return RESP_OK;
736 }
737
738 static int get_cmd_result(struct imap_store *ctx, struct imap_cmd *tcmd)
739 {
740         struct imap *imap = ctx->imap;
741         struct imap_cmd *cmdp, **pcmdp, *ncmdp;
742         char *cmd, *arg, *arg1, *p;
743         int n, resp, resp2, tag;
744
745         for (;;) {
746                 if (buffer_gets(&imap->buf, &cmd))
747                         return RESP_BAD;
748
749                 arg = next_arg(&cmd);
750                 if (*arg == '*') {
751                         arg = next_arg(&cmd);
752                         if (!arg) {
753                                 fprintf(stderr, "IMAP error: unable to parse untagged response\n");
754                                 return RESP_BAD;
755                         }
756
757                         if (!strcmp("NAMESPACE", arg)) {
758                                 imap->ns_personal = parse_list(&cmd);
759                                 imap->ns_other = parse_list(&cmd);
760                                 imap->ns_shared = parse_list(&cmd);
761                         } else if (!strcmp("OK", arg) || !strcmp("BAD", arg) ||
762                                    !strcmp("NO", arg) || !strcmp("BYE", arg)) {
763                                 if ((resp = parse_response_code(ctx, NULL, cmd)) != RESP_OK)
764                                         return resp;
765                         } else if (!strcmp("CAPABILITY", arg)) {
766                                 parse_capability(imap, cmd);
767                         } else if ((arg1 = next_arg(&cmd))) {
768                                 ; /*
769                                    * Unhandled response-data with at least two words.
770                                    * Ignore it.
771                                    *
772                                    * NEEDSWORK: Previously this case handled '<num> EXISTS'
773                                    * and '<num> RECENT' but as a probably-unintended side
774                                    * effect it ignores other unrecognized two-word
775                                    * responses.  imap-send doesn't ever try to read
776                                    * messages or mailboxes these days, so consider
777                                    * eliminating this case.
778                                    */
779                         } else {
780                                 fprintf(stderr, "IMAP error: unable to parse untagged response\n");
781                                 return RESP_BAD;
782                         }
783                 } else if (!imap->in_progress) {
784                         fprintf(stderr, "IMAP error: unexpected reply: %s %s\n", arg, cmd ? cmd : "");
785                         return RESP_BAD;
786                 } else if (*arg == '+') {
787                         /* This can happen only with the last command underway, as
788                            it enforces a round-trip. */
789                         cmdp = (struct imap_cmd *)((char *)imap->in_progress_append -
790                                offsetof(struct imap_cmd, next));
791                         if (cmdp->cb.data) {
792                                 n = socket_write(&imap->buf.sock, cmdp->cb.data, cmdp->cb.dlen);
793                                 free(cmdp->cb.data);
794                                 cmdp->cb.data = NULL;
795                                 if (n != (int)cmdp->cb.dlen)
796                                         return RESP_BAD;
797                         } else if (cmdp->cb.cont) {
798                                 if (cmdp->cb.cont(ctx, cmdp, cmd))
799                                         return RESP_BAD;
800                         } else {
801                                 fprintf(stderr, "IMAP error: unexpected command continuation request\n");
802                                 return RESP_BAD;
803                         }
804                         if (socket_write(&imap->buf.sock, "\r\n", 2) != 2)
805                                 return RESP_BAD;
806                         if (!cmdp->cb.cont)
807                                 imap->literal_pending = 0;
808                         if (!tcmd)
809                                 return DRV_OK;
810                 } else {
811                         tag = atoi(arg);
812                         for (pcmdp = &imap->in_progress; (cmdp = *pcmdp); pcmdp = &cmdp->next)
813                                 if (cmdp->tag == tag)
814                                         goto gottag;
815                         fprintf(stderr, "IMAP error: unexpected tag %s\n", arg);
816                         return RESP_BAD;
817                 gottag:
818                         if (!(*pcmdp = cmdp->next))
819                                 imap->in_progress_append = pcmdp;
820                         imap->num_in_progress--;
821                         if (cmdp->cb.cont || cmdp->cb.data)
822                                 imap->literal_pending = 0;
823                         arg = next_arg(&cmd);
824                         if (!strcmp("OK", arg))
825                                 resp = DRV_OK;
826                         else {
827                                 if (!strcmp("NO", arg)) {
828                                         if (cmdp->cb.create && cmd && (cmdp->cb.trycreate || !memcmp(cmd, "[TRYCREATE]", 11))) { /* SELECT, APPEND or UID COPY */
829                                                 p = strchr(cmdp->cmd, '"');
830                                                 if (!issue_imap_cmd(ctx, NULL, "CREATE \"%.*s\"", (int)(strchr(p + 1, '"') - p + 1), p)) {
831                                                         resp = RESP_BAD;
832                                                         goto normal;
833                                                 }
834                                                 /* not waiting here violates the spec, but a server that does not
835                                                    grok this nonetheless violates it too. */
836                                                 cmdp->cb.create = 0;
837                                                 if (!(ncmdp = issue_imap_cmd(ctx, &cmdp->cb, "%s", cmdp->cmd))) {
838                                                         resp = RESP_BAD;
839                                                         goto normal;
840                                                 }
841                                                 free(cmdp->cmd);
842                                                 free(cmdp);
843                                                 if (!tcmd)
844                                                         return 0;       /* ignored */
845                                                 if (cmdp == tcmd)
846                                                         tcmd = ncmdp;
847                                                 continue;
848                                         }
849                                         resp = RESP_NO;
850                                 } else /*if (!strcmp("BAD", arg))*/
851                                         resp = RESP_BAD;
852                                 fprintf(stderr, "IMAP command '%s' returned response (%s) - %s\n",
853                                          memcmp(cmdp->cmd, "LOGIN", 5) ?
854                                                         cmdp->cmd : "LOGIN <user> <pass>",
855                                                         arg, cmd ? cmd : "");
856                         }
857                         if ((resp2 = parse_response_code(ctx, &cmdp->cb, cmd)) > resp)
858                                 resp = resp2;
859                 normal:
860                         if (cmdp->cb.done)
861                                 cmdp->cb.done(ctx, cmdp, resp);
862                         free(cmdp->cb.data);
863                         free(cmdp->cmd);
864                         free(cmdp);
865                         if (!tcmd || tcmd == cmdp)
866                                 return resp;
867                 }
868         }
869         /* not reached */
870 }
871
872 static void imap_close_server(struct imap_store *ictx)
873 {
874         struct imap *imap = ictx->imap;
875
876         if (imap->buf.sock.fd[0] != -1) {
877                 imap_exec(ictx, NULL, "LOGOUT");
878                 socket_shutdown(&imap->buf.sock);
879         }
880         free_list(imap->ns_personal);
881         free_list(imap->ns_other);
882         free_list(imap->ns_shared);
883         free(imap);
884 }
885
886 static void imap_close_store(struct store *ctx)
887 {
888         imap_close_server((struct imap_store *)ctx);
889         free(ctx);
890 }
891
892 #ifndef NO_OPENSSL
893
894 /*
895  * hexchar() and cram() functions are based on the code from the isync
896  * project (http://isync.sf.net/).
897  */
898 static char hexchar(unsigned int b)
899 {
900         return b < 10 ? '0' + b : 'a' + (b - 10);
901 }
902
903 #define ENCODED_SIZE(n) (4*((n+2)/3))
904 static char *cram(const char *challenge_64, const char *user, const char *pass)
905 {
906         int i, resp_len, encoded_len, decoded_len;
907         HMAC_CTX hmac;
908         unsigned char hash[16];
909         char hex[33];
910         char *response, *response_64, *challenge;
911
912         /*
913          * length of challenge_64 (i.e. base-64 encoded string) is a good
914          * enough upper bound for challenge (decoded result).
915          */
916         encoded_len = strlen(challenge_64);
917         challenge = xmalloc(encoded_len);
918         decoded_len = EVP_DecodeBlock((unsigned char *)challenge,
919                                       (unsigned char *)challenge_64, encoded_len);
920         if (decoded_len < 0)
921                 die("invalid challenge %s", challenge_64);
922         HMAC_Init(&hmac, (unsigned char *)pass, strlen(pass), EVP_md5());
923         HMAC_Update(&hmac, (unsigned char *)challenge, decoded_len);
924         HMAC_Final(&hmac, hash, NULL);
925         HMAC_CTX_cleanup(&hmac);
926
927         hex[32] = 0;
928         for (i = 0; i < 16; i++) {
929                 hex[2 * i] = hexchar((hash[i] >> 4) & 0xf);
930                 hex[2 * i + 1] = hexchar(hash[i] & 0xf);
931         }
932
933         /* response: "<user> <digest in hex>" */
934         resp_len = strlen(user) + 1 + strlen(hex) + 1;
935         response = xmalloc(resp_len);
936         sprintf(response, "%s %s", user, hex);
937
938         response_64 = xmalloc(ENCODED_SIZE(resp_len) + 1);
939         encoded_len = EVP_EncodeBlock((unsigned char *)response_64,
940                                       (unsigned char *)response, resp_len);
941         if (encoded_len < 0)
942                 die("EVP_EncodeBlock error");
943         response_64[encoded_len] = '\0';
944         return (char *)response_64;
945 }
946
947 #else
948
949 static char *cram(const char *challenge_64, const char *user, const char *pass)
950 {
951         die("If you want to use CRAM-MD5 authenticate method, "
952             "you have to build git-imap-send with OpenSSL library.");
953 }
954
955 #endif
956
957 static int auth_cram_md5(struct imap_store *ctx, struct imap_cmd *cmd, const char *prompt)
958 {
959         int ret;
960         char *response;
961
962         response = cram(prompt, server.user, server.pass);
963
964         ret = socket_write(&ctx->imap->buf.sock, response, strlen(response));
965         if (ret != strlen(response))
966                 return error("IMAP error: sending response failed");
967
968         free(response);
969
970         return 0;
971 }
972
973 static struct store *imap_open_store(struct imap_server_conf *srvc)
974 {
975         struct imap_store *ctx;
976         struct imap *imap;
977         char *arg, *rsp;
978         int s = -1, preauth;
979
980         ctx = xcalloc(sizeof(*ctx), 1);
981
982         ctx->imap = imap = xcalloc(sizeof(*imap), 1);
983         imap->buf.sock.fd[0] = imap->buf.sock.fd[1] = -1;
984         imap->in_progress_append = &imap->in_progress;
985
986         /* open connection to IMAP server */
987
988         if (srvc->tunnel) {
989                 const char *argv[] = { srvc->tunnel, NULL };
990                 struct child_process tunnel = {NULL};
991
992                 imap_info("Starting tunnel '%s'... ", srvc->tunnel);
993
994                 tunnel.argv = argv;
995                 tunnel.use_shell = 1;
996                 tunnel.in = -1;
997                 tunnel.out = -1;
998                 if (start_command(&tunnel))
999                         die("cannot start proxy %s", argv[0]);
1000
1001                 imap->buf.sock.fd[0] = tunnel.out;
1002                 imap->buf.sock.fd[1] = tunnel.in;
1003
1004                 imap_info("ok\n");
1005         } else {
1006 #ifndef NO_IPV6
1007                 struct addrinfo hints, *ai0, *ai;
1008                 int gai;
1009                 char portstr[6];
1010
1011                 snprintf(portstr, sizeof(portstr), "%d", srvc->port);
1012
1013                 memset(&hints, 0, sizeof(hints));
1014                 hints.ai_socktype = SOCK_STREAM;
1015                 hints.ai_protocol = IPPROTO_TCP;
1016
1017                 imap_info("Resolving %s... ", srvc->host);
1018                 gai = getaddrinfo(srvc->host, portstr, &hints, &ai);
1019                 if (gai) {
1020                         fprintf(stderr, "getaddrinfo: %s\n", gai_strerror(gai));
1021                         goto bail;
1022                 }
1023                 imap_info("ok\n");
1024
1025                 for (ai0 = ai; ai; ai = ai->ai_next) {
1026                         char addr[NI_MAXHOST];
1027
1028                         s = socket(ai->ai_family, ai->ai_socktype,
1029                                    ai->ai_protocol);
1030                         if (s < 0)
1031                                 continue;
1032
1033                         getnameinfo(ai->ai_addr, ai->ai_addrlen, addr,
1034                                     sizeof(addr), NULL, 0, NI_NUMERICHOST);
1035                         imap_info("Connecting to [%s]:%s... ", addr, portstr);
1036
1037                         if (connect(s, ai->ai_addr, ai->ai_addrlen) < 0) {
1038                                 close(s);
1039                                 s = -1;
1040                                 perror("connect");
1041                                 continue;
1042                         }
1043
1044                         break;
1045                 }
1046                 freeaddrinfo(ai0);
1047 #else /* NO_IPV6 */
1048                 struct hostent *he;
1049                 struct sockaddr_in addr;
1050
1051                 memset(&addr, 0, sizeof(addr));
1052                 addr.sin_port = htons(srvc->port);
1053                 addr.sin_family = AF_INET;
1054
1055                 imap_info("Resolving %s... ", srvc->host);
1056                 he = gethostbyname(srvc->host);
1057                 if (!he) {
1058                         perror("gethostbyname");
1059                         goto bail;
1060                 }
1061                 imap_info("ok\n");
1062
1063                 addr.sin_addr.s_addr = *((int *) he->h_addr_list[0]);
1064
1065                 s = socket(PF_INET, SOCK_STREAM, 0);
1066
1067                 imap_info("Connecting to %s:%hu... ", inet_ntoa(addr.sin_addr), ntohs(addr.sin_port));
1068                 if (connect(s, (struct sockaddr *)&addr, sizeof(addr))) {
1069                         close(s);
1070                         s = -1;
1071                         perror("connect");
1072                 }
1073 #endif
1074                 if (s < 0) {
1075                         fputs("Error: unable to connect to server.\n", stderr);
1076                         goto bail;
1077                 }
1078
1079                 imap->buf.sock.fd[0] = s;
1080                 imap->buf.sock.fd[1] = dup(s);
1081
1082                 if (srvc->use_ssl &&
1083                     ssl_socket_connect(&imap->buf.sock, 0, srvc->ssl_verify)) {
1084                         close(s);
1085                         goto bail;
1086                 }
1087                 imap_info("ok\n");
1088         }
1089
1090         /* read the greeting string */
1091         if (buffer_gets(&imap->buf, &rsp)) {
1092                 fprintf(stderr, "IMAP error: no greeting response\n");
1093                 goto bail;
1094         }
1095         arg = next_arg(&rsp);
1096         if (!arg || *arg != '*' || (arg = next_arg(&rsp)) == NULL) {
1097                 fprintf(stderr, "IMAP error: invalid greeting response\n");
1098                 goto bail;
1099         }
1100         preauth = 0;
1101         if (!strcmp("PREAUTH", arg))
1102                 preauth = 1;
1103         else if (strcmp("OK", arg) != 0) {
1104                 fprintf(stderr, "IMAP error: unknown greeting response\n");
1105                 goto bail;
1106         }
1107         parse_response_code(ctx, NULL, rsp);
1108         if (!imap->caps && imap_exec(ctx, NULL, "CAPABILITY") != RESP_OK)
1109                 goto bail;
1110
1111         if (!preauth) {
1112 #ifndef NO_OPENSSL
1113                 if (!srvc->use_ssl && CAP(STARTTLS)) {
1114                         if (imap_exec(ctx, NULL, "STARTTLS") != RESP_OK)
1115                                 goto bail;
1116                         if (ssl_socket_connect(&imap->buf.sock, 1,
1117                                                srvc->ssl_verify))
1118                                 goto bail;
1119                         /* capabilities may have changed, so get the new capabilities */
1120                         if (imap_exec(ctx, NULL, "CAPABILITY") != RESP_OK)
1121                                 goto bail;
1122                 }
1123 #endif
1124                 imap_info("Logging in...\n");
1125                 if (!srvc->user) {
1126                         fprintf(stderr, "Skipping server %s, no user\n", srvc->host);
1127                         goto bail;
1128                 }
1129                 if (!srvc->pass) {
1130                         struct strbuf prompt = STRBUF_INIT;
1131                         strbuf_addf(&prompt, "Password (%s@%s): ", srvc->user, srvc->host);
1132                         arg = git_getpass(prompt.buf);
1133                         strbuf_release(&prompt);
1134                         if (!*arg) {
1135                                 fprintf(stderr, "Skipping account %s@%s, no password\n", srvc->user, srvc->host);
1136                                 goto bail;
1137                         }
1138                         /*
1139                          * getpass() returns a pointer to a static buffer.  make a copy
1140                          * for long term storage.
1141                          */
1142                         srvc->pass = xstrdup(arg);
1143                 }
1144                 if (CAP(NOLOGIN)) {
1145                         fprintf(stderr, "Skipping account %s@%s, server forbids LOGIN\n", srvc->user, srvc->host);
1146                         goto bail;
1147                 }
1148
1149                 if (srvc->auth_method) {
1150                         struct imap_cmd_cb cb;
1151
1152                         if (!strcmp(srvc->auth_method, "CRAM-MD5")) {
1153                                 if (!CAP(AUTH_CRAM_MD5)) {
1154                                         fprintf(stderr, "You specified"
1155                                                 "CRAM-MD5 as authentication method, "
1156                                                 "but %s doesn't support it.\n", srvc->host);
1157                                         goto bail;
1158                                 }
1159                                 /* CRAM-MD5 */
1160
1161                                 memset(&cb, 0, sizeof(cb));
1162                                 cb.cont = auth_cram_md5;
1163                                 if (imap_exec(ctx, &cb, "AUTHENTICATE CRAM-MD5") != RESP_OK) {
1164                                         fprintf(stderr, "IMAP error: AUTHENTICATE CRAM-MD5 failed\n");
1165                                         goto bail;
1166                                 }
1167                         } else {
1168                                 fprintf(stderr, "Unknown authentication method:%s\n", srvc->host);
1169                                 goto bail;
1170                         }
1171                 } else {
1172                         if (!imap->buf.sock.ssl)
1173                                 imap_warn("*** IMAP Warning *** Password is being "
1174                                           "sent in the clear\n");
1175                         if (imap_exec(ctx, NULL, "LOGIN \"%s\" \"%s\"", srvc->user, srvc->pass) != RESP_OK) {
1176                                 fprintf(stderr, "IMAP error: LOGIN failed\n");
1177                                 goto bail;
1178                         }
1179                 }
1180         } /* !preauth */
1181
1182         ctx->prefix = "";
1183         ctx->trashnc = 1;
1184         return (struct store *)ctx;
1185
1186 bail:
1187         imap_close_store(&ctx->gen);
1188         return NULL;
1189 }
1190
1191 static void lf_to_crlf(struct strbuf *msg)
1192 {
1193         size_t new_len;
1194         char *new;
1195         int i, j, lfnum = 0;
1196
1197         if (msg->buf[0] == '\n')
1198                 lfnum++;
1199         for (i = 1; i < msg->len; i++) {
1200                 if (msg->buf[i - 1] != '\r' && msg->buf[i] == '\n')
1201                         lfnum++;
1202         }
1203
1204         new_len = msg->len + lfnum;
1205         new = xmalloc(new_len + 1);
1206         if (msg->buf[0] == '\n') {
1207                 new[0] = '\r';
1208                 new[1] = '\n';
1209                 i = 1;
1210                 j = 2;
1211         } else {
1212                 new[0] = msg->buf[0];
1213                 i = 1;
1214                 j = 1;
1215         }
1216         for ( ; i < msg->len; i++) {
1217                 if (msg->buf[i] != '\n') {
1218                         new[j++] = msg->buf[i];
1219                         continue;
1220                 }
1221                 if (msg->buf[i - 1] != '\r')
1222                         new[j++] = '\r';
1223                 /* otherwise it already had CR before */
1224                 new[j++] = '\n';
1225         }
1226         strbuf_attach(msg, new, new_len, new_len + 1);
1227 }
1228
1229 /*
1230  * Store msg to IMAP.  Also detach and free the data from msg->data,
1231  * leaving msg->data empty.
1232  */
1233 static int imap_store_msg(struct store *gctx, struct strbuf *msg)
1234 {
1235         struct imap_store *ctx = (struct imap_store *)gctx;
1236         struct imap *imap = ctx->imap;
1237         struct imap_cmd_cb cb;
1238         const char *prefix, *box;
1239         int ret;
1240
1241         lf_to_crlf(msg);
1242         memset(&cb, 0, sizeof(cb));
1243
1244         cb.dlen = msg->len;
1245         cb.data = strbuf_detach(msg, NULL);
1246
1247         box = gctx->name;
1248         prefix = !strcmp(box, "INBOX") ? "" : ctx->prefix;
1249         cb.create = 0;
1250         ret = imap_exec_m(ctx, &cb, "APPEND \"%s%s\" ", prefix, box);
1251         imap->caps = imap->rcaps;
1252         if (ret != DRV_OK)
1253                 return ret;
1254
1255         return DRV_OK;
1256 }
1257
1258 static void wrap_in_html(struct strbuf *msg)
1259 {
1260         struct strbuf buf = STRBUF_INIT;
1261         static char *content_type = "Content-Type: text/html;\n";
1262         static char *pre_open = "<pre>\n";
1263         static char *pre_close = "</pre>\n";
1264         const char *body = strstr(msg->buf, "\n\n");
1265
1266         if (!body)
1267                 return; /* Headers but no body; no wrapping needed */
1268
1269         body += 2;
1270
1271         strbuf_add(&buf, msg->buf, body - msg->buf - 1);
1272         strbuf_addstr(&buf, content_type);
1273         strbuf_addch(&buf, '\n');
1274         strbuf_addstr(&buf, pre_open);
1275         strbuf_addstr_xml_quoted(&buf, body);
1276         strbuf_addstr(&buf, pre_close);
1277
1278         strbuf_release(msg);
1279         *msg = buf;
1280 }
1281
1282 #define CHUNKSIZE 0x1000
1283
1284 static int read_message(FILE *f, struct strbuf *all_msgs)
1285 {
1286         do {
1287                 if (strbuf_fread(all_msgs, CHUNKSIZE, f) <= 0)
1288                         break;
1289         } while (!feof(f));
1290
1291         return ferror(f) ? -1 : 0;
1292 }
1293
1294 static int count_messages(struct strbuf *all_msgs)
1295 {
1296         int count = 0;
1297         char *p = all_msgs->buf;
1298
1299         while (1) {
1300                 if (!prefixcmp(p, "From ")) {
1301                         p = strstr(p+5, "\nFrom: ");
1302                         if (!p) break;
1303                         p = strstr(p+7, "\nDate: ");
1304                         if (!p) break;
1305                         p = strstr(p+7, "\nSubject: ");
1306                         if (!p) break;
1307                         p += 10;
1308                         count++;
1309                 }
1310                 p = strstr(p+5, "\nFrom ");
1311                 if (!p)
1312                         break;
1313                 p++;
1314         }
1315         return count;
1316 }
1317
1318 /*
1319  * Copy the next message from all_msgs, starting at offset *ofs, to
1320  * msg.  Update *ofs to the start of the following message.  Return
1321  * true iff a message was successfully copied.
1322  */
1323 static int split_msg(struct strbuf *all_msgs, struct strbuf *msg, int *ofs)
1324 {
1325         char *p, *data;
1326         size_t len;
1327
1328         if (*ofs >= all_msgs->len)
1329                 return 0;
1330
1331         data = &all_msgs->buf[*ofs];
1332         len = all_msgs->len - *ofs;
1333
1334         if (len < 5 || prefixcmp(data, "From "))
1335                 return 0;
1336
1337         p = strchr(data, '\n');
1338         if (p) {
1339                 p++;
1340                 len -= p - data;
1341                 *ofs += p - data;
1342                 data = p;
1343         }
1344
1345         p = strstr(data, "\nFrom ");
1346         if (p)
1347                 len = &p[1] - data;
1348
1349         strbuf_add(msg, data, len);
1350         *ofs += len;
1351         return 1;
1352 }
1353
1354 static char *imap_folder;
1355
1356 static int git_imap_config(const char *key, const char *val, void *cb)
1357 {
1358         char imap_key[] = "imap.";
1359
1360         if (strncmp(key, imap_key, sizeof imap_key - 1))
1361                 return 0;
1362
1363         key += sizeof imap_key - 1;
1364
1365         /* check booleans first, and barf on others */
1366         if (!strcmp("sslverify", key))
1367                 server.ssl_verify = git_config_bool(key, val);
1368         else if (!strcmp("preformattedhtml", key))
1369                 server.use_html = git_config_bool(key, val);
1370         else if (!val)
1371                 return config_error_nonbool(key);
1372
1373         if (!strcmp("folder", key)) {
1374                 imap_folder = xstrdup(val);
1375         } else if (!strcmp("host", key)) {
1376                 if (!prefixcmp(val, "imap:"))
1377                         val += 5;
1378                 else if (!prefixcmp(val, "imaps:")) {
1379                         val += 6;
1380                         server.use_ssl = 1;
1381                 }
1382                 if (!prefixcmp(val, "//"))
1383                         val += 2;
1384                 server.host = xstrdup(val);
1385         } else if (!strcmp("user", key))
1386                 server.user = xstrdup(val);
1387         else if (!strcmp("pass", key))
1388                 server.pass = xstrdup(val);
1389         else if (!strcmp("port", key))
1390                 server.port = git_config_int(key, val);
1391         else if (!strcmp("tunnel", key))
1392                 server.tunnel = xstrdup(val);
1393         else if (!strcmp("authmethod", key))
1394                 server.auth_method = xstrdup(val);
1395
1396         return 0;
1397 }
1398
1399 int main(int argc, char **argv)
1400 {
1401         struct strbuf all_msgs = STRBUF_INIT;
1402         struct strbuf msg = STRBUF_INIT;
1403         struct store *ctx = NULL;
1404         int ofs = 0;
1405         int r;
1406         int total, n = 0;
1407         int nongit_ok;
1408
1409         git_extract_argv0_path(argv[0]);
1410
1411         git_setup_gettext();
1412
1413         if (argc != 1)
1414                 usage(imap_send_usage);
1415
1416         setup_git_directory_gently(&nongit_ok);
1417         git_config(git_imap_config, NULL);
1418
1419         if (!server.port)
1420                 server.port = server.use_ssl ? 993 : 143;
1421
1422         if (!imap_folder) {
1423                 fprintf(stderr, "no imap store specified\n");
1424                 return 1;
1425         }
1426         if (!server.host) {
1427                 if (!server.tunnel) {
1428                         fprintf(stderr, "no imap host specified\n");
1429                         return 1;
1430                 }
1431                 server.host = "tunnel";
1432         }
1433
1434         /* read the messages */
1435         if (read_message(stdin, &all_msgs)) {
1436                 fprintf(stderr, "error reading input\n");
1437                 return 1;
1438         }
1439
1440         if (all_msgs.len == 0) {
1441                 fprintf(stderr, "nothing to send\n");
1442                 return 1;
1443         }
1444
1445         total = count_messages(&all_msgs);
1446         if (!total) {
1447                 fprintf(stderr, "no messages to send\n");
1448                 return 1;
1449         }
1450
1451         /* write it to the imap server */
1452         ctx = imap_open_store(&server);
1453         if (!ctx) {
1454                 fprintf(stderr, "failed to open store\n");
1455                 return 1;
1456         }
1457
1458         fprintf(stderr, "sending %d message%s\n", total, (total != 1) ? "s" : "");
1459         ctx->name = imap_folder;
1460         while (1) {
1461                 unsigned percent = n * 100 / total;
1462
1463                 fprintf(stderr, "%4u%% (%d/%d) done\r", percent, n, total);
1464                 if (!split_msg(&all_msgs, &msg, &ofs))
1465                         break;
1466                 if (server.use_html)
1467                         wrap_in_html(&msg);
1468                 r = imap_store_msg(ctx, &msg);
1469                 if (r != DRV_OK)
1470                         break;
1471                 n++;
1472         }
1473         fprintf(stderr, "\n");
1474
1475         imap_close_store(ctx);
1476
1477         return 0;
1478 }