From: Arnaud Lacombe <al@sigfpe.info>
[apps/madmutt.git] / imap / auth_sasl.c
index 5740435..60e4656 100644 (file)
@@ -1,64 +1,53 @@
 /*
+ * Copyright notice from original mutt:
  * Copyright (C) 2000-3 Brendan Cully <brendan@kublai.com>
- * 
- *     This program is free software; you can redistribute it and/or modify
- *     it under the terms of the GNU General Public License as published by
- *     the Free Software Foundation; either version 2 of the License, or
- *     (at your option) any later version.
- * 
- *     This program is distributed in the hope that it will be useful,
- *     but WITHOUT ANY WARRANTY; without even the implied warranty of
- *     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
- *     GNU General Public License for more details.
- * 
- *     You should have received a copy of the GNU General Public License
- *     along with this program; if not, write to the Free Software
- *     Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111, USA.
- */ 
+ *
+ * This file is part of mutt-ng, see http://www.muttng.org/.
+ * It's licensed under the GNU General Public License,
+ * please see the file GPL in the top level source directory.
+ */
 
 /* SASL login/authentication code */
 
+#if HAVE_CONFIG_H
+# include "config.h"
+#endif
+
 #include "mutt.h"
+#include "ascii.h"
 #include "mutt_sasl.h"
 #include "imap_private.h"
 #include "auth.h"
 
-#ifdef USE_SASL2
+#include "lib/mem.h"
+#include "lib/intl.h"
+#include "lib/debug.h"
+
 #include <sasl/sasl.h>
 #include <sasl/saslutil.h>
-#else
-#include <sasl.h>
-#include <saslutil.h>
-#endif
 
 /* imap_auth_sasl: Default authenticator if available. */
-imap_auth_res_t imap_auth_sasl (IMAP_DATA* idata, const char* method)
+imap_auth_res_t imap_auth_sasl (IMAP_DATA * idata, const char *method)
 {
-  sasl_conn_tsaslconn;
-  sasl_interact_tinteraction = NULL;
+  sasl_conn_t *saslconn;
+  sasl_interact_t *interaction = NULL;
   int rc, irc;
   char buf[HUGE_STRING];
-  const charmech;
-#ifdef USE_SASL2
+  const char *mech;
+
   const char *pc = NULL;
-#else
-  char* pc = NULL;
-#endif
   unsigned int len, olen;
   unsigned char client_start;
 
-  if (mutt_sasl_client_new (idata->conn, &saslconn) < 0)
-  {
-    dprint (1, (debugfile,
-      "imap_auth_sasl: Error allocating SASL connection.\n"));
+  if (mutt_sasl_client_new (idata->conn, &saslconn) < 0) {
+    debug_print (1, ("Error allocating SASL connection.\n"));
     return IMAP_AUTH_FAILURE;
   }
 
   rc = SASL_FAIL;
 
   /* If the user hasn't specified a method, use any available */
-  if (!method)
-  {
+  if (!method) {
     method = idata->capstr;
 
     /* hack for SASL ANONYMOUS support:
@@ -70,40 +59,28 @@ imap_auth_res_t imap_auth_sasl (IMAP_DATA* idata, const char* method)
       return IMAP_AUTH_FAILURE;
 
     if (mutt_bit_isset (idata->capabilities, AUTH_ANON) &&
-       (!idata->conn->account.user[0] ||
-        !ascii_strncmp (idata->conn->account.user, "anonymous", 9)))
-#ifdef USE_SASL2
-      rc = sasl_client_start (saslconn, "AUTH=ANONYMOUS", NULL, &pc, &olen, 
+        (!idata->conn->account.user[0] ||
+         !ascii_strncmp (idata->conn->account.user, "anonymous", 9)))
+      rc = sasl_client_start (saslconn, "AUTH=ANONYMOUS", NULL, &pc, &olen,
                               &mech);
-#else
-      rc = sasl_client_start (saslconn, "AUTH=ANONYMOUS", NULL, NULL, &pc, &olen,
-                             &mech);
-#endif
   }
-  
+
   if (rc != SASL_OK && rc != SASL_CONTINUE)
-    do
-    {
-#ifdef USE_SASL2
+    do {
       rc = sasl_client_start (saslconn, method, &interaction,
-        &pc, &olen, &mech);
-#else
-      rc = sasl_client_start (saslconn, method, NULL, &interaction,
-        &pc, &olen, &mech);
-#endif
+                              &pc, &olen, &mech);
       if (rc == SASL_INTERACT)
-       mutt_sasl_interact (interaction);
+        mutt_sasl_interact (interaction);
     }
     while (rc == SASL_INTERACT);
 
   client_start = (olen > 0);
 
-  if (rc != SASL_OK && rc != SASL_CONTINUE)
-  {
+  if (rc != SASL_OK && rc != SASL_CONTINUE) {
     if (method)
-      dprint (2, (debugfile, "imap_auth_sasl: %s unavailable\n", method));
+      debug_print (2, ("%s unavailable\n", method));
     else
-      dprint (1, (debugfile, "imap_auth_sasl: Failure starting authentication exchange. No shared mechanisms?\n"));
+      debug_print (1, ("Failure starting authentication exchange. No shared mechanisms?\n"));
     /* SASL doesn't support LOGIN, so fall back */
 
     return IMAP_AUTH_UNAVAIL;
@@ -116,36 +93,38 @@ imap_auth_res_t imap_auth_sasl (IMAP_DATA* idata, const char* method)
   irc = IMAP_CMD_CONTINUE;
 
   /* looping protocol */
-  while (rc == SASL_CONTINUE)
-  {
+  while (rc == SASL_CONTINUE || olen > 0) {
     do
       irc = imap_cmd_step (idata);
     while (irc == IMAP_CMD_CONTINUE);
 
+    if (method && irc == IMAP_CMD_NO) {
+      debug_print (2, ("%s failed\n", method));
+      sasl_dispose (&saslconn);
+      return IMAP_AUTH_UNAVAIL;
+    }
+
     if (irc == IMAP_CMD_BAD || irc == IMAP_CMD_NO)
       goto bail;
 
-    if (irc == IMAP_CMD_RESPOND)
-    {
-#ifdef USE_SASL2
-      if (sasl_decode64 (idata->cmd.buf+2, strlen (idata->cmd.buf+2), buf, LONG_STRING-1,
-#else
-      if (sasl_decode64 (idata->cmd.buf+2, strlen (idata->cmd.buf+2), buf,
-#endif
-                        &len) != SASL_OK)
-      {
-       dprint (1, (debugfile, "imap_auth_sasl: error base64-decoding server response.\n"));
-       goto bail;
+    if (irc == IMAP_CMD_RESPOND) {
+      if (sasl_decode64
+          (idata->cmd.buf + 2, str_len (idata->cmd.buf + 2), buf,
+           LONG_STRING - 1,
+                         &len) != SASL_OK) {
+        debug_print (1, ("error base64-decoding server response.\n"));
+        goto bail;
       }
     }
 
-    if (!client_start)
-    {
-      do
-      {
-       rc = sasl_client_step (saslconn, buf, len, &interaction, &pc, &olen);
-       if (rc == SASL_INTERACT)
-         mutt_sasl_interact (interaction);
+    /* client-start is only available with the SASL-IR extension, but
+     * SASL 2.1 seems to want to use it regardless, at least for DIGEST
+     * fast reauth. Override if the server sent an initial continuation */
+    if (!client_start || buf[0]) {
+      do {
+        rc = sasl_client_step (saslconn, buf, len, &interaction, &pc, &olen);
+        if (rc == SASL_INTERACT)
+          mutt_sasl_interact (interaction);
       }
       while (rc == SASL_INTERACT);
     }
@@ -153,33 +132,25 @@ imap_auth_res_t imap_auth_sasl (IMAP_DATA* idata, const char* method)
       client_start = 0;
 
     /* send out response, or line break if none needed */
-    if (olen)
-    {
-      if (sasl_encode64 (pc, olen, buf, sizeof (buf), &olen) != SASL_OK)
-      {
-       dprint (1, (debugfile, "imap_auth_sasl: error base64-encoding client response.\n"));
-       goto bail;
+    if (olen) {
+      if (sasl_encode64 (pc, olen, buf, sizeof (buf), &olen) != SASL_OK) {
+        debug_print (1, ("error base64-encoding client response.\n"));
+        goto bail;
       }
-
-      /* sasl_client_st(art|ep) allocate pc with malloc, expect me to 
-       * free it */
-#ifndef USE_SASL2
-      FREE (&pc);
-#endif
     }
-    
-    if (irc == IMAP_CMD_RESPOND)
-    {
+
+    if (irc == IMAP_CMD_RESPOND) {
       strfcpy (buf + olen, "\r\n", sizeof (buf) - olen);
       mutt_socket_write (idata->conn, buf);
     }
 
     /* If SASL has errored out, send an abort string to the server */
-    if (rc < 0)
-    {
+    if (rc < 0) {
       mutt_socket_write (idata->conn, "*\r\n");
-      dprint (1, (debugfile, "imap_auth_sasl: sasl_client_step error %d\n",rc));
+      debug_print (1, ("sasl_client_step error %d\n", rc));
     }
+
+    olen = 0;
   }
 
   while (irc != IMAP_CMD_OK)
@@ -189,15 +160,14 @@ imap_auth_res_t imap_auth_sasl (IMAP_DATA* idata, const char* method)
   if (rc != SASL_OK)
     goto bail;
 
-  if (imap_code (idata->cmd.buf))
-  {
+  if (imap_code (idata->cmd.buf)) {
     mutt_sasl_setup_conn (idata->conn, saslconn);
     return IMAP_AUTH_SUCCESS;
   }
 
- bail:
+bail:
   mutt_error _("SASL authentication failed.");
-  mutt_sleep(2);
+  mutt_sleep (2);
   sasl_dispose (&saslconn);
 
   return IMAP_AUTH_FAILURE;