def perform_handshake(self):
request = RequestMessage.new(
self.connection,
RequestSegment(
message_types.AUTHENTICATE,
Authentication(self.user, {self.method: self.client_key})))
response = self.connection.send_request(request, no_reconnect=True)
auth_part = response.segments[0].parts[0]
if self.method not in auth_part.methods:
raise Exception(
"Only unknown authentication methods available: %s" %
b",".join(auth_part.methods.keys()))
salt, server_key = Fields.unpack_data(
BytesIO(auth_part.methods[self.method]))
self.client_proof = self.calculate_client_proof([salt], server_key)
return Authentication(self.user, {'SCRAMSHA256': self.client_proof})
def test_pack_data():
part = Authentication(
"TestUser", {
"SCRAMSHA256":
b"\xed\xbd\x7c\xc8\xb2\xf2\x64\x89\xd6\x5a\x7c"
b"\xd5\x1e\x27\xf2\xe7\x3f\xca\x22\x7d\x1a\xb6"
b"\xaa\xfc\xac\x0f\x42\x8c\xa4\xd8\xe1\x0c\x19"
b"\xe3\xe3\x8f\x3a\xac\x51\x07\x5e\x67\xbb\xe5"
b"\x2f\xdb\x61\x03\xa7\xc3\x4c\x8a\x70\x90\x8e"
b"\xd5\xbe\x0b\x35\x42\x70\x5f\x73\x8c"
})
arguments, payload = part.pack_data(constants.MAX_SEGMENT_SIZE)
assert payload == \
b"\x03\x00\x08\x54\x65\x73\x74\x55\x73\x65\x72\x0b\x53\x43\x52\x41" \
b"\x4d\x53\x48\x41\x32\x35\x36\x40\xed\xbd\x7c\xc8\xb2\xf2\x64\x89" \
b"\xd6\x5a\x7c\xd5\x1e\x27\xf2\xe7\x3f\xca\x22\x7d\x1a\xb6\xaa\xfc" \
b"\xac\x0f\x42\x8c\xa4\xd8\xe1\x0c\x19\xe3\xe3\x8f\x3a\xac\x51\x07" \
b"\x5e\x67\xbb\xe5\x2f\xdb\x61\x03\xa7\xc3\x4c\x8a\x70\x90\x8e\xd5" \
b"\xbe\x0b\x35\x42\x70\x5f\x73\x8c"
def test_pack_data():
part = Authentication(
"TestUser",
{
"SCRAMSHA256": b"\xed\xbd\x7c\xc8\xb2\xf2\x64\x89\xd6\x5a\x7c"
b"\xd5\x1e\x27\xf2\xe7\x3f\xca\x22\x7d\x1a\xb6"
b"\xaa\xfc\xac\x0f\x42\x8c\xa4\xd8\xe1\x0c\x19"
b"\xe3\xe3\x8f\x3a\xac\x51\x07\x5e\x67\xbb\xe5"
b"\x2f\xdb\x61\x03\xa7\xc3\x4c\x8a\x70\x90\x8e"
b"\xd5\xbe\x0b\x35\x42\x70\x5f\x73\x8c"
}
)
arguments, payload = part.pack_data()
assert payload == \
b"\x03\x00\x08\x54\x65\x73\x74\x55\x73\x65\x72\x0b\x53\x43\x52\x41" \
b"\x4d\x53\x48\x41\x32\x35\x36\x40\xed\xbd\x7c\xc8\xb2\xf2\x64\x89" \
b"\xd6\x5a\x7c\xd5\x1e\x27\xf2\xe7\x3f\xca\x22\x7d\x1a\xb6\xaa\xfc" \
b"\xac\x0f\x42\x8c\xa4\xd8\xe1\x0c\x19\xe3\xe3\x8f\x3a\xac\x51\x07" \
b"\x5e\x67\xbb\xe5\x2f\xdb\x61\x03\xa7\xc3\x4c\x8a\x70\x90\x8e\xd5" \
b"\xbe\x0b\x35\x42\x70\x5f\x73\x8c"
def test_unpack_data():
packed = BytesIO(
b"\x02\x00\x0b\x53\x43\x52\x41\x4d\x53\x48\x41\x32\x35\x36\x44\x02"
b"\x00\x10\x19\x6e\xa8\xb1\x8d\x51\x66\xe6\xec\x17\x38\xd9\xff\x49"
b"\x02\x83\x30\x23\x06\xa3\x7a\x72\xd4\xfd\x73\x69\xd9\x9b\x2d\xd2"
b"\x6e\xad\xe3\x89\x57\x06\x6e\xa1\x21\x85\x7f\x18\x63\x67\xe4\x9b"
b"\x75\x28\x96\xe0\x3d\x1f\x56\xca\x86\x85\x8c\x5f\xf5\x27\xc3\x18"
b"\x88\x1e\x8c\x00\x00\x00\x00\x00")
user, methods = Authentication.unpack_data(1, packed)
assert user is None
assert b"SCRAMSHA256" in methods
assert methods[b"SCRAMSHA256"] == \
b"\x02\x00\x10\x19\x6e\xa8\xb1\x8d\x51\x66\xe6\xec\x17\x38\xd9\xff" \
b"\x49\x02\x83\x30\x23\x06\xa3\x7a\x72\xd4\xfd\x73\x69\xd9\x9b\x2d" \
b"\xd2\x6e\xad\xe3\x89\x57\x06\x6e\xa1\x21\x85\x7f\x18\x63\x67\xe4" \
b"\x9b\x75\x28\x96\xe0\x3d\x1f\x56\xca\x86\x85\x8c\x5f\xf5\x27\xc3" \
b"\x18\x88\x1e\x8c"
def test_unpack_data():
packed = BytesIO(
b"\x02\x00\x0b\x53\x43\x52\x41\x4d\x53\x48\x41\x32\x35\x36\x44\x02"
b"\x00\x10\x19\x6e\xa8\xb1\x8d\x51\x66\xe6\xec\x17\x38\xd9\xff\x49"
b"\x02\x83\x30\x23\x06\xa3\x7a\x72\xd4\xfd\x73\x69\xd9\x9b\x2d\xd2"
b"\x6e\xad\xe3\x89\x57\x06\x6e\xa1\x21\x85\x7f\x18\x63\x67\xe4\x9b"
b"\x75\x28\x96\xe0\x3d\x1f\x56\xca\x86\x85\x8c\x5f\xf5\x27\xc3\x18"
b"\x88\x1e\x8c\x00\x00\x00\x00\x00"
)
user, methods = Authentication.unpack_data(1, packed)
assert user is None
assert b"SCRAMSHA256" in methods
assert methods[b"SCRAMSHA256"] == \
b"\x02\x00\x10\x19\x6e\xa8\xb1\x8d\x51\x66\xe6\xec\x17\x38\xd9\xff" \
b"\x49\x02\x83\x30\x23\x06\xa3\x7a\x72\xd4\xfd\x73\x69\xd9\x9b\x2d" \
b"\xd2\x6e\xad\xe3\x89\x57\x06\x6e\xa1\x21\x85\x7f\x18\x63\x67\xe4" \
b"\x9b\x75\x28\x96\xe0\x3d\x1f\x56\xca\x86\x85\x8c\x5f\xf5\x27\xc3" \
b"\x18\x88\x1e\x8c"
