Fixed a crash while doing host key tests.

This commit is contained in:
Joe Testa
2021-02-26 16:01:30 -05:00
parent 741bd631e2
commit c483fe1861
4 changed files with 6 additions and 3 deletions

View File

@ -125,8 +125,8 @@ class HostKeyTest:
# Do the initial DH exchange. The server responds back
# with the host key and its length. Bingo. We also get back the host key fingerprint.
kex_group.send_init(s)
try:
kex_group.send_init(s)
host_key = kex_group.recv_reply(s, variable_key_len)
if host_key is not None:
server_kex.set_host_key(host_key_type, host_key)