Merge pull request #585 from elemount/authissue

methane · web-flow · commit 52980174ceb3 · 2017-06-29T18:54:47.000+09:00
Fix handle AuthSwitch packet bug.
diff --git a/pymysql/connections.py b/pymysql/connections.py
@@ -132,14 +132,16 @@ def is_ascii(data):
     print()
 
 
+SCRAMBLE_LENGTH = 20
+
 def _scramble(password, message):
     if not password:
         return b''
     if DEBUG: print('password=' + str(password))
     stage1 = sha_new(password).digest()
     stage2 = sha_new(stage1).digest()
     s = sha_new()
-    s.update(message)
+    s.update(message[:SCRAMBLE_LENGTH]) 
     s.update(stage2)
     result = s.digest()
     return _my_crypt(result, stage1)
