def test_packet(self):
"""Unit test SSH packet module"""
for encode, decode, values in self.tests:
for value, data in values:
data = codecs.decode(data, 'hex')
with self.subTest(msg='encode', value=value):
self.assertEqual(encode(value), data)
with self.subTest(msg='decode', data=data):
packet = SSHPacket(data)
decoded_value = decode(packet)
packet.check_end()
self.assertEqual(decoded_value, value)
self.assertEqual(packet.get_consumed_payload(), data)
self.assertEqual(packet.get_remaining_payload(), b'')
for encode, value, exc in self.encode_errors:
with self.subTest(msg='encode error', encode=encode, value=value):
with self.assertRaises(exc):
encode(value)
for decode, data in self.decode_errors:
with self.subTest(msg='decode error', data=data):
with self.assertRaises(PacketDecodeError):
packet = SSHPacket(codecs.decode(data, 'hex'))
decode(packet)
packet.check_end()
def test_packet(self):
"""Unit test SSH packet module"""
for encode, decode, values in self.tests:
for value, data in values:
data = codecs.decode(data, 'hex')
with self.subTest(msg='encode', value=value):
self.assertEqual(encode(value), data)
with self.subTest(msg='decode', data=data):
packet = SSHPacket(data)
decoded_value = decode(packet)
packet.check_end()
self.assertEqual(decoded_value, value)
self.assertEqual(packet.get_consumed_payload(), data)
self.assertEqual(packet.get_remaining_payload(), b'')
for encode, value, exc in self.encode_errors:
with self.subTest(msg='encode error', encode=encode, value=value):
with self.assertRaises(exc):
encode(value)
for decode, data in self.decode_errors:
with self.subTest(msg='decode error', data=data):
with self.assertRaises(PacketDecodeError):
packet = SSHPacket(codecs.decode(data, 'hex'))
decode(packet)
packet.check_end()
def process_packet(self, data):
"""Process an incoming packet"""
packet = SSHPacket(data)
pkttype = packet.get_byte()
if pkttype == MSG_USERAUTH_FAILURE:
_ = packet.get_namelist()
partial_success = packet.get_boolean()
packet.check_end()
if partial_success: # pragma: no cover
# Partial success not implemented yet
self._auth.auth_succeeded()
else:
self._auth.auth_failed()
self._auth_waiter.set_result((False, self._password_changed))
self._auth = None
self._auth_waiter = None
elif pkttype == MSG_USERAUTH_SUCCESS:
packet.check_end()
self._auth.auth_succeeded()
self._auth_waiter.set_result((True, self._password_changed))
self._auth = None
self._auth_waiter = None
else:
self._auth.process_packet(pkttype, None, packet)
def process_packet(self, data):
"""Process an incoming packet"""
packet = SSHPacket(data)
pkttype = packet.get_byte()
if pkttype == MSG_USERAUTH_FAILURE:
_ = packet.get_namelist()
partial_success = packet.get_boolean()
packet.check_end()
if partial_success: # pragma: no cover
# Partial success not implemented yet
self._auth.auth_succeeded()
else:
self._auth.auth_failed()
self._auth_waiter.set_result((False, self._password_changed))
self._auth = None
self._auth_waiter = None
elif pkttype == MSG_USERAUTH_SUCCESS:
packet.check_end()
self._auth.auth_succeeded()
self._auth_waiter.set_result((True, self._password_changed))
self._auth = None
self._auth_waiter = None
else:
try:
self._auth.process_packet(pkttype, packet)
except DisconnectError as exc:
self.connection_lost(exc)
def communicate(self, request):
"""Process SSH key signing request"""
# pylint: disable=no-self-use
packet = SSHPacket(request)
request = packet.get_string()
packet.check_end()
packet = SSHPacket(request)
version = packet.get_byte()
_ = packet.get_uint32() # sock_fd
data = packet.get_string()
packet.check_end()
if version == 0:
return b'', b''
elif version == 1:
return b'', b'invalid request'
else:
skey = asyncssh.load_keypairs('skey')[0]
sig = skey.sign(data)
return String(Byte(KEYSIGN_VERSION) + String(sig)), b''
def communicate(self, request):
"""Process SSH key signing request"""
# pylint: disable=no-self-use
packet = SSHPacket(request)
request = packet.get_string()
packet.check_end()
packet = SSHPacket(request)
version = packet.get_byte()
_ = packet.get_uint32() # sock_fd
data = packet.get_string()
packet.check_end()
if version == 0:
return b'', b''
elif version == 1:
return b'', b'invalid request'
else:
skey = asyncssh.load_keypairs('skey')[0]
sig = skey.sign(data)
return String(Byte(KEYSIGN_VERSION) + String(sig)), b''
