From 007ffec762f83564ad8aff1c9bd2940dc8b8d0fa Mon Sep 17 00:00:00 2001
From: jesopo
Date: Mon, 17 Sep 2018 12:56:41 +0100
Subject: Handle 904 (ERR_SASLFAIL) in sasl.py

---
 modules/sasl.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

(limited to 'modules')

diff --git a/modules/sasl.py b/modules/sasl.py
index 967fb551..2ab2194b 100644
--- a/modules/sasl.py
+++ b/modules/sasl.py
@@ -7,6 +7,7 @@ class Module(object):
         events.on("received.cap.ack").hook(self.on_cap_ack)
         events.on("received.authenticate").hook(self.on_authenticate)
         events.on("received.numeric.903").hook(self.sasl_success)
+        events.on("received.numeric.904").hook(self.sasl_failure)
 
         exports.add("serverset", {"setting": "sasl",
             "help": "Set the sasl username/password for this server",
@@ -60,5 +61,9 @@ class Module(object):
                 auth_text = auth_text.decode("utf8")
             event["server"].send_authenticate(auth_text)
 
+    def _end_sasl(self, server):
+        server.capability_done("sasl")
     def sasl_success(self, event):
-        event["server"].capability_done("sasl")
+        self._end_sasl(event["server"])
+    def sasl_success(self, event):
+        self._end_sasl(event["server"])
-- 
cgit v1.3.1-10-gc9f91