From c006435fa1de1de7e54077577a1c8ff078d68687 Mon Sep 17 00:00:00 2001
From: jesopo
Date: Wed, 6 Feb 2019 15:41:31 +0000
Subject: Set SCRAM state to just `Failure` if there's a server-final-message
 error (sasl)

---
 modules/sasl/scram.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

(limited to 'modules')

diff --git a/modules/sasl/scram.py b/modules/sasl/scram.py
index fab7f3c3..b7aa3986 100644
--- a/modules/sasl/scram.py
+++ b/modules/sasl/scram.py
@@ -14,7 +14,8 @@ class SCRAMState(enum.Enum):
     ClientFirst = 1
     ClientFinal = 2
     Success = 3
-    VerifyFailed = 4
+    Failed = 4
+    VerifyFailed = 5
 
 class SCRAMError(Exception):
     pass
@@ -78,6 +79,7 @@ class SCRAM(object):
         pieces = self._get_pieces(data)
         if b"e" in pieces:
             self.error = pieces[b"e"].decode("utf8")
+            self.state = SCRAMState.Failed
             return False
 
         verifier = pieces[b"v"]
-- 
cgit v1.3.1-10-gc9f91