kernel-src/wolfcrypt_glue.c: refactor wc_linuxkm_drbg_init_ctx() and wc_linuxkm_drbg_ctx_clear() to use wc_rng_bank_default_checkout() and wc_rng_bank_default_checkin().

Author: douzzer

kernel-src/wolfcrypt_glue.c

@@ -827,25 +827,11 @@ int wc_linuxkm_drbg_init_ctx(struct wc_rng_bank **ctx) {
 
 #ifdef LINUXKM_LKCAPI_REGISTER_HASH_DRBG_DEFAULT
 
-    struct crypto_rng *current_crypto_default_rng;
+    ret = wc_rng_bank_default_checkout(ctx);
 
-    ret = crypto_get_default_rng();
     if (ret != 0)
         goto new_bank;
 
-    current_crypto_default_rng = crypto_default_rng;
-
-    if (current_crypto_default_rng == NULL) {
-        crypto_put_default_rng();
-        goto new_bank;
-    }
-
-    if (! wc_linux_kernel_rng_is_wolfcrypt(current_crypto_default_rng)) {
-        crypto_put_default_rng();
-        goto new_bank;
-    }
-
-    *ctx = (struct wc_rng_bank *)crypto_rng_ctx(current_crypto_default_rng);
     wc_wg_drbg_is_global_default = 1;
 
     return 0;
@@ -886,9 +872,10 @@ void wc_linuxkm_drbg_ctx_clear(struct wc_rng_bank **ctx) {
 
 #ifdef LINUXKM_LKCAPI_REGISTER_HASH_DRBG_DEFAULT
     if (wc_wg_drbg_is_global_default) {
-        *ctx = NULL;
+        int ret = wc_rng_bank_default_checkin(ctx);
+        if (ret != 0)
+            pr_err("ERROR: wc_rng_bank_default_checkin() in wc_linuxkm_drbg_ctx_clear() failed with code %d.\n", ret);
         wc_wg_drbg_is_global_default = 0;
-        crypto_put_default_rng();
     }
     else
 #endif