From 6b4bb7cdbae1d07e765fe643870b2eaeb0ba5ec7 Mon Sep 17 00:00:00 2001
From: jesopo
Date: Wed, 6 Feb 2019 08:50:19 +0000
Subject: Type annotate scram.py and don't pass base64 data to scram.py
 functions

---
 modules/sasl/__init__.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

(limited to 'modules/sasl/__init__.py')

diff --git a/modules/sasl/__init__.py b/modules/sasl/__init__.py
index 0c74cc97..d7a768ae 100644
--- a/modules/sasl/__init__.py
+++ b/modules/sasl/__init__.py
@@ -74,10 +74,11 @@ class Module(ModuleManager.BaseModule):
                 auth_text = event["server"]._scram.client_first()
             else:
                 current_scram = event["server"]._scram
+                data = base64.b64decode(event["message"])
                 if current_scram.state == scram.SCRAMState.ClientFirst:
-                    auth_text = current_scram.server_first(event["message"])
+                    auth_text = current_scram.server_first(data)
                 elif current_scram.state == scram.SCRAMState.ClientFinal:
-                    auth_text = current_scram.server_final(event["message"])
+                    auth_text = current_scram.server_final(data)
                     del event["server"]._scram
 
                     if current_scram.state == scram.SCRAMState.VerifyFailed:
-- 
cgit v1.3.1-10-gc9f91