fix SASL Authentication

nakagami · Jan 26, 2024 · d5f7cbb · d5f7cbb
1 parent 059ad78
commit d5f7cbb
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/minipg/pgcore.py b/minipg/pgcore.py
@@ -497,7 +497,7 @@ def _process_messages(self, obj):
                     data = self._read(ln)
                     assert _bytes_to_bint(data[:4]) == 0
                 elif auth_method == 10:   # SASL
-                    assert data[4:-2].decode('utf-8') == 'SCRAM-SHA-256'
+                    assert b'SCRAM-SHA-256\x00\x00' in data
                     printable = string.ascii_letters + string.digits + '+/'
                     client_nonce = ''.join(
                         printable[random.randrange(0, len(printable))]