Check rng return value when generating DH secrets and primes in gmp plugin

2012-07-16 14:53:35 +02:00
parent e93bb353d5
commit 5025135f70
3 changed files with 19 additions and 13 deletions
@@ -230,8 +230,13 @@ static gmp_diffie_hellman_t *create_generic(diffie_hellman_group_t group,
 		destroy(this);
 		return NULL;
 	}
-
-	rng->allocate_bytes(rng, exp_len, &random);
+	if (!rng->allocate_bytes(rng, exp_len, &random))
+	{
+		DBG1(DBG_LIB, "failed to allocate DH secret");
+		rng->destroy(rng);
+		destroy(this);
+		return NULL;
+	}
 	rng->destroy(rng);

 	if (exp_len == this->p_len)