X-Git-Url: https://jxself.org/git/?a=blobdiff_plain;ds=sidebyside;f=tfprng.c;h=d4370f31172d0093f487ca07395a85a7286a317d;hb=e4b8a1c06635af297a89f2629e22c7aa94103d56;hp=24de6fd657a179a9aae7f2bc9e7b49199902799e;hpb=e0d3bb48cd0cd079f9d1bd5555a52f450db452ac;p=tfcrypt.git

diff --git a/tfprng.c b/tfprng.c
index 24de6fd..d4370f3 100644
--- a/tfprng.c
+++ b/tfprng.c
@@ -19,7 +19,7 @@ void tf_prng_seedkey_r(void *sdata, const void *skey)
 	TF_UNIT_TYPE k[TF_NR_KEY_UNITS];
 	struct tf_prng_data *rprng = sdata;
 
-	memset(rprng, 0, sizeof(struct tf_prng_data));
+	memset(rprng, 0, tf_prng_datasize());
 	if (!skey) return;
 
 	memcpy(k, skey, TF_KEY_SIZE);
@@ -52,7 +52,7 @@ void tf_prng_seed_r(void *sdata, TF_UNIT_TYPE seed)
 	struct tf_prng_data *rprng = sdata;
 	size_t x;
 
-	memset(rprng, 0, sizeof(struct tf_prng_data));
+	memset(rprng, 0, tf_prng_datasize());
 	for (x = 0; x < TF_NR_KEY_UNITS; x++) k[x] = seed;
 	tfe_init(&rprng->tfe, k);
 	rprng->init = 1;