drop mem_alloc and mem_free, use my own hand crafted optmized macros that
[apps/madmutt.git] / mutt_sasl.c
1 /*
2  * Copyright notice from original mutt:
3  * Copyright (C) 2000-5 Brendan Cully <brendan@kublai.com>
4  *
5  * This file is part of mutt-ng, see http://www.muttng.org/.
6  * It's licensed under the GNU General Public License,
7  * please see the file GPL in the top level source directory.
8  */
9
10 /* common SASL helper routines */
11
12 #if HAVE_CONFIG_H
13 # include "config.h"
14 #endif
15
16 #include <lib-lib/mem.h>
17
18 #include "mutt.h"
19 #include "account.h"
20 #include "mutt_sasl.h"
21 #include "mutt_socket.h"
22
23 #include "lib/mem.h"
24 #include "lib/debug.h"
25
26 #include <errno.h>
27 #include <netdb.h>
28 #include <sasl/sasl.h>
29
30 #include <sys/socket.h>
31 #include <netinet/in.h>
32
33 static int getnameinfo_err (int ret)
34 {
35   int err;
36
37   switch (ret) {
38   case EAI_AGAIN:
39     debug_print (1, ("The name could not be resolved at this time. Future attempts may succeed.\n"));
40     err = SASL_TRYAGAIN;
41     break;
42   case EAI_BADFLAGS:
43     debug_print (1, ("The flags had an invalid value.\n"));
44     err = SASL_BADPARAM;
45     break;
46   case EAI_FAIL:
47     debug_print (1, ("A non-recoverable error occurred.\n"));
48     err = SASL_FAIL;
49     break;
50   case EAI_FAMILY:
51     debug_print (1, ("The address family was not recognized or the address length was invalid for the specified family.\n"));
52     err = SASL_BADPROT;
53     break;
54   case EAI_MEMORY:
55     debug_print (1, ("There was a memory allocation failure.\n"));
56     err = SASL_NOMEM;
57     break;
58   case EAI_NONAME:
59     debug_print (1, ("The name does not resolve for the supplied parameters. NI_NAMEREQD is set and the host's name cannot be located, or both nodename and servname were null.\n"));
60     err = SASL_FAIL;            /* no real equivalent */
61     break;
62   case EAI_SYSTEM:
63     debug_print (1, ("A system error occurred.  The error code can be found in errno(%d,%s)).\n", 
64                 errno, strerror (errno)));
65     err = SASL_FAIL;            /* no real equivalent */
66     break;
67   default:
68     debug_print (1, ("Unknown error %d\n", ret));
69     err = SASL_FAIL;            /* no real equivalent */
70     break;
71   }
72   return err;
73 }
74
75 /* arbitrary. SASL will probably use a smaller buffer anyway. OTOH it's
76  * been a while since I've had access to an SASL server which negotiated
77  * a protection buffer. */
78 #define M_SASL_MAXBUF 65536
79
80 #define IP_PORT_BUFLEN 1024
81
82 static sasl_callback_t mutt_sasl_callbacks[5];
83
84 static int mutt_sasl_start (void);
85
86 /* callbacks */
87 static int mutt_sasl_cb_log (void *context, int priority,
88                              const char *message);
89 static int mutt_sasl_cb_authname (void *context, int id, const char **result,
90                                   unsigned int *len);
91 static int mutt_sasl_cb_pass (sasl_conn_t * conn, void *context, int id,
92                               sasl_secret_t ** psecret);
93
94 /* socket wrappers for a SASL security layer */
95 static int mutt_sasl_conn_open (CONNECTION * conn);
96 static int mutt_sasl_conn_close (CONNECTION * conn);
97 static int mutt_sasl_conn_read (CONNECTION * conn, char *buf, size_t len);
98 static int mutt_sasl_conn_write (CONNECTION * conn, const char *buf,
99                                  size_t count);
100
101 /* utility function, stolen from sasl2 sample code */
102 static int iptostring (const struct sockaddr *addr, socklen_t addrlen,
103                        char *out, unsigned outlen)
104 {
105   char hbuf[NI_MAXHOST], pbuf[NI_MAXSERV];
106   int ret;
107
108   if (!addr || !out)
109     return SASL_BADPARAM;
110
111   ret = getnameinfo (addr, addrlen, hbuf, sizeof (hbuf), pbuf, sizeof (pbuf),
112                      NI_NUMERICHOST |
113 #ifdef NI_WITHSCOPEID
114                      NI_WITHSCOPEID |
115 #endif
116                      NI_NUMERICSERV);
117   if (ret)
118     return getnameinfo_err (ret);
119
120   if (outlen < str_len (hbuf) + str_len (pbuf) + 2)
121     return SASL_BUFOVER;
122
123   snprintf (out, outlen, "%s;%s", hbuf, pbuf);
124
125   return SASL_OK;
126 }
127
128 /* mutt_sasl_start: called before doing a SASL exchange - initialises library
129  *   (if necessary). */
130 int mutt_sasl_start (void)
131 {
132   static unsigned char sasl_init = 0;
133
134   static sasl_callback_t callbacks[2];
135   int rc;
136
137   if (sasl_init)
138     return SASL_OK;
139
140   /* set up default logging callback */
141   callbacks[0].id = SASL_CB_LOG;
142   callbacks[0].proc = mutt_sasl_cb_log;
143   callbacks[0].context = NULL;
144
145   callbacks[1].id = SASL_CB_LIST_END;
146   callbacks[1].proc = NULL;
147   callbacks[1].context = NULL;
148
149   rc = sasl_client_init (callbacks);
150
151   if (rc != SASL_OK) {
152     debug_print (1, ("libsasl initialisation failed.\n"));
153     return SASL_FAIL;
154   }
155
156   sasl_init = 1;
157
158   return SASL_OK;
159 }
160
161 /* mutt_sasl_client_new: wrapper for sasl_client_new which also sets various
162  * security properties. If this turns out to be fine for POP too we can
163  * probably stop exporting mutt_sasl_get_callbacks(). */
164 int mutt_sasl_client_new (CONNECTION * conn, sasl_conn_t ** saslconn)
165 {
166   sasl_security_properties_t secprops;
167
168   struct sockaddr_storage local, remote;
169   socklen_t size;
170   char iplocalport[IP_PORT_BUFLEN], ipremoteport[IP_PORT_BUFLEN];
171   const char *service;
172   int rc;
173
174   if (mutt_sasl_start () != SASL_OK)
175     return -1;
176
177   switch (conn->account.type) {
178   case M_ACCT_TYPE_IMAP:
179     service = "imap";
180     break;
181   case M_ACCT_TYPE_POP:
182     service = "pop";
183     break;
184   default:
185     debug_print (1, ("account type unset\n"));
186     return -1;
187   }
188
189   size = sizeof (local);
190   if (getsockname (conn->fd, (struct sockaddr *) &local, &size)) {
191     debug_print (1, ("getsockname for local failed\n"));
192     return -1;
193   }
194   else
195     if (iptostring
196         ((struct sockaddr *) &local, size, iplocalport,
197          IP_PORT_BUFLEN) != SASL_OK) {
198     debug_print (1, ("iptostring for local failed\n"));
199     return -1;
200   }
201
202   size = sizeof (remote);
203   if (getpeername (conn->fd, (struct sockaddr *) &remote, &size)) {
204     debug_print (1, ("getsockname for remote failed\n"));
205     return -1;
206   }
207   else
208     if (iptostring
209         ((struct sockaddr *) &remote, size, ipremoteport,
210          IP_PORT_BUFLEN) != SASL_OK) {
211     debug_print (1, ("iptostring for remote failed\n"));
212     return -1;
213   }
214
215   debug_print (1, ("local ip: %s, remote ip:%s\n", iplocalport, ipremoteport));
216
217   rc =
218     sasl_client_new (service, conn->account.host, iplocalport, ipremoteport,
219                      mutt_sasl_get_callbacks (&conn->account), 0, saslconn);
220
221   if (rc != SASL_OK) {
222     debug_print (1, ("Error allocating SASL connection\n"));
223     return -1;
224   }
225
226   /*** set sasl IP properties, necessary for use with krb4 ***/
227   /* Do we need to fail if this fails? I would assume having these unset
228    * would just disable KRB4. Who wrote this code?
229    */
230   {
231     struct sockaddr_in local, remote;
232     socklen_t size;
233
234     size = sizeof (local);
235     if (getsockname (conn->fd, (struct sockaddr *) &local, &size))
236       return -1;
237
238     size = sizeof (remote);
239     if (getpeername (conn->fd, (struct sockaddr *) &remote, &size))
240       return -1;
241
242 #ifdef SASL_IP_LOCAL
243     if (sasl_setprop (*saslconn, SASL_IP_LOCAL, &local) != SASL_OK) {
244       debug_print (1, ("Error setting local IP address\n"));
245       return -1;
246     }
247 #endif
248
249 #ifdef SASL_IP_REMOTE
250     if (sasl_setprop (*saslconn, SASL_IP_REMOTE, &remote) != SASL_OK) {
251       debug_print (1, ("Error setting remote IP address\n"));
252       return -1;
253     }
254 #endif
255   }
256
257   /* set security properties. We use NOPLAINTEXT globally, since we can
258    * just fall back to LOGIN in the IMAP case anyway. If that doesn't
259    * work for POP, we can make it a flag or move this code into
260    * imap/auth_sasl.c */
261   memset (&secprops, 0, sizeof (secprops));
262   /* Work around a casting bug in the SASL krb4 module */
263   secprops.max_ssf = 0x7fff;
264   secprops.maxbufsize = M_SASL_MAXBUF;
265   secprops.security_flags |= SASL_SEC_NOPLAINTEXT;
266   if (sasl_setprop (*saslconn, SASL_SEC_PROPS, &secprops) != SASL_OK) {
267     debug_print (1, ("Error setting security properties\n"));
268     return -1;
269   }
270
271   if (conn->ssf) {
272     debug_print (2, ("External SSF: %d\n", conn->ssf));
273     if (sasl_setprop (*saslconn, SASL_SSF_EXTERNAL, &(conn->ssf)) != SASL_OK)
274     {
275       debug_print (1, ("Error setting external properties\n"));
276       return -1;
277     }
278     debug_print (2, ("External authentication name: %s\n", conn->account.user));
279     if (sasl_setprop (*saslconn, SASL_AUTH_EXTERNAL, conn->account.user) !=
280         SASL_OK) {
281       debug_print (1, ("Error setting external properties\n"));
282       return -1;
283     }
284   }
285
286   return 0;
287 }
288
289 sasl_callback_t *mutt_sasl_get_callbacks (ACCOUNT * account)
290 {
291   sasl_callback_t *callback;
292
293   callback = mutt_sasl_callbacks;
294
295   callback->id = SASL_CB_USER;
296   callback->proc = mutt_sasl_cb_authname;
297   callback->context = account;
298   callback++;
299
300   callback->id = SASL_CB_AUTHNAME;
301   callback->proc = mutt_sasl_cb_authname;
302   callback->context = account;
303   callback++;
304
305   callback->id = SASL_CB_PASS;
306   callback->proc = mutt_sasl_cb_pass;
307   callback->context = account;
308   callback++;
309
310   callback->id = SASL_CB_GETREALM;
311   callback->proc = NULL;
312   callback->context = NULL;
313   callback++;
314
315   callback->id = SASL_CB_LIST_END;
316   callback->proc = NULL;
317   callback->context = NULL;
318
319   return mutt_sasl_callbacks;
320 }
321
322 int mutt_sasl_interact (sasl_interact_t * interaction)
323 {
324   char prompt[SHORT_STRING];
325   char resp[SHORT_STRING];
326
327   while (interaction->id != SASL_CB_LIST_END) {
328     debug_print (2, ("filling in SASL interaction %ld.\n", interaction->id));
329
330     snprintf (prompt, sizeof (prompt), "%s: ", interaction->prompt);
331     resp[0] = '\0';
332     if (mutt_get_field (prompt, resp, sizeof (resp), 0))
333       return SASL_FAIL;
334
335     interaction->len = str_len (resp) + 1;
336     interaction->result = p_dupstr(resp, interaction->len - 1);
337     interaction++;
338   }
339
340   return SASL_OK;
341 }
342
343 /* SASL can stack a protection layer on top of an existing connection.
344  * To handle this, we store a saslconn_t in conn->sockdata, and write
345  * wrappers which en/decode the read/write stream, then replace sockdata
346  * with an embedded copy of the old sockdata and call the underlying
347  * functions (which we've also preserved). I thought about trying to make
348  * a general stackable connection system, but it seemed like overkill -
349  * something is wrong if we have 15 filters on top of a socket. Anyway,
350  * anything else which wishes to stack can use the same method. The only
351  * disadvantage is we have to write wrappers for all the socket methods,
352  * even if we only stack over read and write. Thinking about it, the
353  * abstraction problem is that there is more in CONNECTION than there
354  * needs to be. Ideally it would have only (void*)data and methods. */
355
356 /* mutt_sasl_setup_conn: replace connection methods, sockdata with 
357  *   SASL wrappers, for protection layers. Also get ssf, as a fastpath
358  *   for the read/write methods. */
359 void mutt_sasl_setup_conn (CONNECTION * conn, sasl_conn_t * saslconn)
360 {
361   SASL_DATA *sasldata = p_new(SASL_DATA, 1);
362
363   sasldata->saslconn = saslconn;
364   /* get ssf so we know whether we have to (en|de)code read/write */
365   sasl_getprop (saslconn, SASL_SSF, (const void **)(void *)&sasldata->ssf);
366   debug_print (3, ("SASL protection strength: %u\n", *sasldata->ssf));
367   /* Add SASL SSF to transport SSF */
368   conn->ssf += *sasldata->ssf;
369   sasl_getprop (saslconn, SASL_MAXOUTBUF,
370                 (const void **)(void *)&sasldata->pbufsize);
371   debug_print (3, ("SASL protection buffer size: %u\n", *sasldata->pbufsize));
372
373   /* clear input buffer */
374   sasldata->buf = NULL;
375   sasldata->bpos = 0;
376   sasldata->blen = 0;
377
378   /* preserve old functions */
379   sasldata->sockdata = conn->sockdata;
380   sasldata->msasl_open = conn->conn_open;
381   sasldata->msasl_close = conn->conn_close;
382   sasldata->msasl_read = conn->conn_read;
383   sasldata->msasl_write = conn->conn_write;
384
385   /* and set up new functions */
386   conn->sockdata = sasldata;
387   conn->conn_open = mutt_sasl_conn_open;
388   conn->conn_close = mutt_sasl_conn_close;
389   conn->conn_read = mutt_sasl_conn_read;
390   conn->conn_write = mutt_sasl_conn_write;
391 }
392
393 void mutt_sasl_done (void) {
394   sasl_done ();
395 }
396
397 /* mutt_sasl_cb_log: callback to log SASL messages */
398 static int mutt_sasl_cb_log (void *context, int priority, const char *message)
399 {
400   debug_print (priority, ("SASL: %s\n", message));
401
402   return SASL_OK;
403 }
404
405 /* mutt_sasl_cb_authname: callback to retrieve authname or user from ACCOUNT */
406 static int mutt_sasl_cb_authname (void *context, int id, const char **result,
407                                   unsigned *len)
408 {
409   ACCOUNT *account = (ACCOUNT *) context;
410
411   *result = NULL;
412   if (len)
413     *len = 0;
414
415   if (!account)
416     return SASL_BADPARAM;
417
418   debug_print (2, ("getting %s for %s:%u\n",
419               id == SASL_CB_AUTHNAME ? "authname" : "user",
420               account->host, account->port));
421
422   if (id == SASL_CB_AUTHNAME) {
423     if (mutt_account_getlogin (account))
424       return SASL_FAIL;
425     *result = account->login;
426   } else {
427     if (mutt_account_getuser (account))
428       return SASL_FAIL;
429     *result = account->user;
430   }
431
432   if (len)
433     *len = str_len (*result);
434
435   return SASL_OK;
436 }
437
438 static int mutt_sasl_cb_pass (sasl_conn_t * conn, void *context, int id,
439                               sasl_secret_t ** psecret)
440 {
441   ACCOUNT *account = (ACCOUNT *) context;
442   int len;
443
444   if (!account || !psecret)
445     return SASL_BADPARAM;
446
447   debug_print (2, ("getting password for %s@%s:%u\n",
448               account->login, account->host, account->port));
449
450   if (mutt_account_getpass (account))
451     return SASL_FAIL;
452
453   len = str_len (account->pass);
454
455   *psecret = xmalloc(sizeof(sasl_secret_t) + len);
456   (*psecret)->len = len;
457   strcpy ((char*) (*psecret)->data, account->pass);     /* __STRCPY_CHECKED__ */
458
459   return SASL_OK;
460 }
461
462 /* mutt_sasl_conn_open: empty wrapper for underlying open function. We
463  *   don't know in advance that a connection will use SASL, so we
464  *   replace conn's methods with sasl methods when authentication
465  *   is successful, using mutt_sasl_setup_conn */
466 static int mutt_sasl_conn_open (CONNECTION * conn)
467 {
468   SASL_DATA *sasldata;
469   int rc;
470
471   sasldata = (SASL_DATA *) conn->sockdata;
472   conn->sockdata = sasldata->sockdata;
473   rc = (sasldata->msasl_open) (conn);
474   conn->sockdata = sasldata;
475
476   return rc;
477 }
478
479 /* mutt_sasl_conn_close: calls underlying close function and disposes of
480  *   the sasl_conn_t object, then restores connection to pre-sasl state */
481 static int mutt_sasl_conn_close (CONNECTION * conn)
482 {
483   SASL_DATA *sasldata;
484   int rc;
485
486   sasldata = (SASL_DATA *) conn->sockdata;
487
488   /* restore connection's underlying methods */
489   conn->sockdata = sasldata->sockdata;
490   conn->conn_open = sasldata->msasl_open;
491   conn->conn_close = sasldata->msasl_close;
492   conn->conn_read = sasldata->msasl_read;
493   conn->conn_write = sasldata->msasl_write;
494
495   /* release sasl resources */
496   sasl_dispose (&sasldata->saslconn);
497   p_delete(&sasldata->buf);
498   p_delete(&sasldata);
499
500   /* call underlying close */
501   rc = (conn->conn_close) (conn);
502
503   return rc;
504 }
505
506 static int mutt_sasl_conn_read (CONNECTION * conn, char *buf, size_t len)
507 {
508   SASL_DATA *sasldata;
509   int rc;
510
511   unsigned int olen;
512
513   sasldata = (SASL_DATA *) conn->sockdata;
514
515   /* if we still have data in our read buffer, copy it into buf */
516   if (sasldata->blen > sasldata->bpos) {
517     olen = (sasldata->blen - sasldata->bpos > len) ? len :
518       sasldata->blen - sasldata->bpos;
519
520     memcpy (buf, sasldata->buf + sasldata->bpos, olen);
521     sasldata->bpos += olen;
522
523     return olen;
524   }
525
526   conn->sockdata = sasldata->sockdata;
527
528   p_delete(&sasldata->buf);
529   sasldata->bpos = 0;
530   sasldata->blen = 0;
531
532   /* and decode the result, if necessary */
533   if (*sasldata->ssf) {
534     do {
535       /* call the underlying read function to fill the buffer */
536       rc = (sasldata->msasl_read) (conn, buf, len);
537       if (rc <= 0)
538         goto out;
539
540       rc = sasl_decode (sasldata->saslconn, buf, rc, &sasldata->buf,
541                         &sasldata->blen);
542       if (rc != SASL_OK) {
543         debug_print (1, ("SASL decode failed: %s\n",
544                     sasl_errstring (rc, NULL, NULL)));
545         goto out;
546       }
547     }
548     while (!sasldata->blen);
549
550     olen = (sasldata->blen - sasldata->bpos > len) ? len :
551       sasldata->blen - sasldata->bpos;
552
553     memcpy (buf, sasldata->buf, olen);
554     sasldata->bpos += olen;
555
556     rc = olen;
557   }
558   else
559     rc = (sasldata->msasl_read) (conn, buf, len);
560
561 out:
562   conn->sockdata = sasldata;
563
564   return rc;
565 }
566
567 static int mutt_sasl_conn_write (CONNECTION * conn, const char *buf,
568                                  size_t len)
569 {
570   SASL_DATA *sasldata;
571   int rc;
572
573   const char *pbuf;
574   unsigned int olen, plen;
575
576   sasldata = (SASL_DATA *) conn->sockdata;
577   conn->sockdata = sasldata->sockdata;
578
579   /* encode data, if necessary */
580   if (*sasldata->ssf) {
581     /* handle data larger than MAXOUTBUF */
582     do {
583       olen = (len > *sasldata->pbufsize) ? *sasldata->pbufsize : len;
584
585       rc = sasl_encode (sasldata->saslconn, buf, olen, &pbuf, &plen);
586       if (rc != SASL_OK) {
587         debug_print (1, ("SASL encoding failed: %s\n",
588                     sasl_errstring (rc, NULL, NULL)));
589         goto fail;
590       }
591
592       rc = (sasldata->msasl_write) (conn, pbuf, plen);
593       p_delete(&pbuf);
594       if (rc != plen)
595         goto fail;
596
597       len -= olen;
598       buf += olen;
599     }
600     while (len > *sasldata->pbufsize);
601   }
602   else
603     /* just write using the underlying socket function */
604     rc = (sasldata->msasl_write) (conn, buf, len);
605
606   conn->sockdata = sasldata;
607
608   return rc;
609
610 fail:
611   conn->sockdata = sasldata;
612   return -1;
613 }