Changes in irc_commands.c [9076a1c:c4e61db]

File:

• irc_commands.c (modified) (6 diffs)

irc_commands.c

@@ -28 +28 @@
 #include "help.h"
 #include "ipc.h"
+#include "base64.h"
 
 static void irc_cmd_pass(irc_t *irc, char **cmd)
@@ -55 +56 @@
                 irc_setpass(irc, cmd[1]);
                 irc_check_login(irc);
+        }
+}
+
+static gboolean irc_sasl_plain_parse(char *input, char **user, char **pass)
+{
+        int i, part, len;
+        guint8 *decoded;
+        char *parts[3];
+
+        /* bitlbee's base64_decode wrapper adds an extra null terminator at the end */
+        len = base64_decode(input, &decoded);
+
+        /* this loop splits the decoded string into the parts array, like this:
+           "username\0username\0password" -> {"username", "username", "password"} */
+
+        for (i = 0, part = 0; i < len && part < 3; part++) {
+                /* set each of parts[] to point to the beginning of a string */
+                parts[part] = (char *) decoded + i;
+
+                /* move the cursor forward to the next null terminator*/
+                i += strlen(parts[part]) + 1;
+        }
+
+        /* sanity checks */
+        if (part != 3 || i != (len + 1) || (parts[0][0] && strcmp(parts[0], parts[1]) != 0)) {
+                g_free(decoded);
+                return FALSE;
+        } else {
+                *user = g_strdup(parts[1]);
+                *pass = g_strdup(parts[2]);
+                g_free(decoded);
+                return TRUE;
+        }
+}
+
+static gboolean irc_sasl_check_pass(irc_t *irc, char *user, char *pass)
+{
+        storage_status_t status;
+
+        /* just check the password here to be able to reply with useful numerics
+         * the actual identification will be handled later */
+        status = storage_check_pass(user, pass);
+
+        if (status == STORAGE_OK) {
+                if (!irc->user->nick) {
+                        /* set the nick here so we have it for the following numeric */
+                        irc->user->nick = g_strdup(user);
+                }
+                irc_send_num(irc, 903, ":Password accepted");
+                return TRUE;
+
+        } else if (status == STORAGE_INVALID_PASSWORD) {
+                irc_send_num(irc, 904, ":Incorrect password");
+        } else if (status == STORAGE_NO_SUCH_USER) {
+                irc_send_num(irc, 904, ":The nick is (probably) not registered");
+        } else {
+                irc_send_num(irc, 904, ":Unknown SASL authentication error");
+        }
+
+        return FALSE;
+}
+
+static void irc_cmd_authenticate(irc_t *irc, char **cmd)
+{
+        /* require the CAP to be enabled, and don't allow authentication before server password */
+        if (!(irc->caps & CAP_SASL) ||
+            (global.conf->authmode == AUTHMODE_CLOSED && !(irc->status & USTATUS_AUTHORIZED))) {
+                return;
+        }
+
+        if (irc->status & USTATUS_SASL_PLAIN_PENDING) {
+                char *user, *pass;
+
+                irc->status &= ~USTATUS_SASL_PLAIN_PENDING;
+
+                if (!irc_sasl_plain_parse(cmd[1], &user, &pass)) {
+                        irc_send_num(irc, 904, ":SASL authentication failed");
+                        return;
+                }
+
+                /* let's not support the nick != user case
+                 * if NICK is received after SASL, it will just fail after registration */
+                if (user && irc->user->nick && strcmp(user, irc->user->nick) != 0) {
+                        irc_send_num(irc, 902, ":Your SASL username does not match your nickname");
+
+                } else if (irc_sasl_check_pass(irc, user, pass)) {
+                        /* and here we do the same thing as the PASS command*/
+                        if (irc->status & USTATUS_LOGGED_IN) {
+                                char *send_cmd[] = { "identify", pass, NULL };
+                                root_command(irc, send_cmd);
+                        } else {
+                                /* no check_login here - wait for CAP END */
+                                irc_setpass(irc, pass);
+                        }
+                }
+
+                g_free(user);
+                g_free(pass);
+
+        } else if (irc->status & USTATUS_IDENTIFIED) {
+                irc_send_num(irc, 907, ":You have already authenticated");
+
+        } else if (strcmp(cmd[1], "*") == 0) {
+                irc_send_num(irc, 906, ":SASL authentication aborted");
+                irc->status &= ~USTATUS_SASL_PLAIN_PENDING;
+
+        } else if (g_strcasecmp(cmd[1], "PLAIN") == 0) {
+                irc_write(irc, "AUTHENTICATE +");
+                irc->status |= USTATUS_SASL_PLAIN_PENDING;
+
+        } else {
+                irc_send_num(irc, 908, "PLAIN :is the available SASL mechanism");
+                irc_send_num(irc, 904, ":SASL authentication failed");
+                irc->status &= ~USTATUS_SASL_PLAIN_PENDING;
         }
 }
@@ -83 +198 @@
                         irc->status &= ~USTATUS_IDENTIFIED;
                         irc_umode_set(irc, "-R", 1);
+
+                        if (irc->caps & CAP_SASL) {
+                                irc_send_num(irc, 901, "%s!%s@%s :You are now logged out",
+                                        irc->user->nick, irc->user->user, irc->user->host);
+                        }
+
                         irc_rootmsg(irc, "Changing nicks resets your identify status. "
                                     "Re-identify or register a new account if you want "
@@ -166 +287 @@
                            showed an error message, or is doing some work
                            before the join should be confirmed. (In the
-                           latter case, the callee should take care of that
+                           latter case, the caller should take care of that
                            confirmation.) TRUE means all's good, let the
                            user join the channel right away. */
@@ -685 +806 @@
 
 static const command_t irc_commands[] = {
+        { "cap",         1, irc_cmd_cap,         0 },
         { "pass",        1, irc_cmd_pass,        0 },
         { "user",        4, irc_cmd_user,        IRC_CMD_PRE_LOGIN },
@@ -721 +843 @@
         { "restart",     0, NULL,                IRC_CMD_OPER_ONLY | IRC_CMD_TO_MASTER },
         { "kill",        2, NULL,                IRC_CMD_OPER_ONLY | IRC_CMD_TO_MASTER },
+        { "authenticate", 1, irc_cmd_authenticate, 0 },
         { NULL }
 };