linuxkm drbg: refactor drbg_ctx clear.

This commit is contained in:
jordan
2025-06-12 19:59:19 -05:00
parent 6cb0c8513d
commit 41965750c8

View File

@@ -908,6 +908,22 @@ struct wc_linuxkm_drbg_ctx {
} *rngs; /* one per CPU ID */ } *rngs; /* one per CPU ID */
}; };
static inline void wc_linuxkm_drbg_ctx_clear(struct wc_linuxkm_drbg_ctx * ctx)
{
unsigned int i;
if (ctx->rngs) {
for (i = 0; i < nr_cpu_ids; ++i) {
(void)wc_FreeMutex(&ctx->rngs[i].lock);
wc_FreeRng(&ctx->rngs[i].rng);
}
free(ctx->rngs);
ctx->rngs = NULL;
}
return;
}
static int wc_linuxkm_drbg_init_tfm(struct crypto_tfm *tfm) static int wc_linuxkm_drbg_init_tfm(struct crypto_tfm *tfm)
{ {
struct wc_linuxkm_drbg_ctx *ctx = (struct wc_linuxkm_drbg_ctx *)crypto_tfm_ctx(tfm); struct wc_linuxkm_drbg_ctx *ctx = (struct wc_linuxkm_drbg_ctx *)crypto_tfm_ctx(tfm);
@@ -937,12 +953,7 @@ static int wc_linuxkm_drbg_init_tfm(struct crypto_tfm *tfm)
} }
if (ret != 0) { if (ret != 0) {
for (i = 0; i < nr_cpu_ids; ++i) { wc_linuxkm_drbg_ctx_clear(ctx);
(void)wc_FreeMutex(&ctx->rngs[i].lock);
wc_FreeRng(&ctx->rngs[i].rng);
}
free(ctx->rngs);
ctx->rngs = NULL;
} }
return ret; return ret;
@@ -951,16 +962,8 @@ static int wc_linuxkm_drbg_init_tfm(struct crypto_tfm *tfm)
static void wc_linuxkm_drbg_exit_tfm(struct crypto_tfm *tfm) static void wc_linuxkm_drbg_exit_tfm(struct crypto_tfm *tfm)
{ {
struct wc_linuxkm_drbg_ctx *ctx = (struct wc_linuxkm_drbg_ctx *)crypto_tfm_ctx(tfm); struct wc_linuxkm_drbg_ctx *ctx = (struct wc_linuxkm_drbg_ctx *)crypto_tfm_ctx(tfm);
unsigned int i;
if (ctx->rngs) { wc_linuxkm_drbg_ctx_clear(ctx);
for (i = 0; i < nr_cpu_ids; ++i) {
(void)wc_FreeMutex(&ctx->rngs[i].lock);
wc_FreeRng(&ctx->rngs[i].rng);
}
free(ctx->rngs);
ctx->rngs = NULL;
}
return; return;
} }