check ecc key return, fix null pointer crash

1 files changed, 3 insertions, 0 deletions

diff --git a/common-kex.c b/common-kex.c
index 2907715..e2b4f8e 100644
--- a/common-kex.c
+++ b/common-kex.c
@@ -703,6 +703,9 @@ void kexecdh_comb_key(struct kex_ecdh_param *param, buffer *pub_them,
 	ecc_key *Q_C, *Q_S, *Q_them;
 
 	Q_them = buf_get_ecc_raw_pubkey(pub_them, algo_kex->ecc_curve);
+	if (Q_them == NULL) {
+		dropbear_exit("ECC error");
+	}
 
 	ses.dh_K = dropbear_ecc_shared_secret(Q_them, &param->key);