?  search:
summary | log | graphiclog1 | graphiclog2 | commit | commitdiff | tree | refs | edit | fork
blame | history | raw | HEAD
imap-send.c: inline parse_imap_list() in parse_list()
[git/jnareb-git.git] / imap-send.c
blob0d1f87a651bae9961f296db42449d0841a0cc1a1
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 }
git with Jakub Narębski modifications