tls-hkdf: Add helper method to allocate data from the internal PRF

2021-02-12 11:45:44 +01:00
parent 6a0ee0c23c
commit a9f661f52a
2 changed files with 22 additions and 0 deletions
@@ -485,6 +485,14 @@ METHOD(tls_hkdf_t, derive_finished, bool,
 								 finished);
 }

+METHOD(tls_hkdf_t, allocate_bytes, bool,
+	private_tls_hkdf_t *this, chunk_t key, chunk_t seed,
+	chunk_t *out)
+{
+	return this->prf->set_key(this->prf, key) &&
+		   this->prf->allocate_bytes(this->prf, seed, out);
+}
+
 METHOD(tls_hkdf_t, destroy, void,
 	private_tls_hkdf_t *this)
 {
@@ -525,6 +533,7 @@ tls_hkdf_t *tls_hkdf_create(hash_algorithm_t hash_algorithm, chunk_t psk)
 			.derive_key = _derive_key,
 			.derive_iv = _derive_iv,
 			.derive_finished = _derive_finished,
+			.allocate_bytes = _allocate_bytes,
 			.destroy = _destroy,
 		},
 		.phase = HKDF_PHASE_0,