feedc0de/wolfssl
1
0
Fork 0
forked from wolfSSL/wolfssl
Code Pull Requests Activity

linuxkm rsa: clean up duplicate code.

Browse Source
This commit is contained in:
jordan
2025-05-21 16:59:02 -05:00
parent d396987863
commit 54104887ca
1 changed files with 41 additions and 65 deletions
Download Patch File Download Diff File Expand all files Collapse all files

86
linuxkm/lkcapi_rsa_glue.c
View File

@@ -194,45 +194,34 @@ static unsigned int km_rsa_max_size(struct crypto_akcipher *tfm);
static int km_direct_rsa_dec(struct akcipher_request *req); static int km_direct_rsa_dec(struct akcipher_request *req);
#endif /* LINUXKM_DIRECT_RSA */ #endif /* LINUXKM_DIRECT_RSA */
/* pkcs1 callbacks */
#if !defined(LINUXKM_AKCIPHER_NO_SIGNVERIFY) #if !defined(LINUXKM_AKCIPHER_NO_SIGNVERIFY)
#ifdef WOLFSSL_SHA224 #define tfm_type crypto_akcipher
static int km_pkcs1_sha224_init(struct crypto_akcipher *tfm); #define tfm_ctx_cb akcipher_tfm_ctx
#endif /* WOLFSSL_SHA224 */
#ifndef NO_SHA256
static int km_pkcs1_sha256_init(struct crypto_akcipher *tfm);
#endif /* !NO_SHA256 */
#ifdef WOLFSSL_SHA384
static int km_pkcs1_sha384_init(struct crypto_akcipher *tfm);
#endif /* WOLFSSL_SHA384 */
#ifdef WOLFSSL_SHA512
static int km_pkcs1_sha512_init(struct crypto_akcipher *tfm);
#endif /* WOLFSSL_SHA512 */
#ifdef WOLFSSL_SHA3
static int km_pkcs1_sha3_256_init(struct crypto_akcipher *tfm);
static int km_pkcs1_sha3_384_init(struct crypto_akcipher *tfm);
static int km_pkcs1_sha3_512_init(struct crypto_akcipher *tfm);
#endif /* WOLFSSL_SHA3 */
#else #else
#define tfm_type crypto_sig
#define tfm_ctx_cb crypto_sig_ctx
static int km_pkcs1pad_init(struct crypto_akcipher * tfm); static int km_pkcs1pad_init(struct crypto_akcipher * tfm);
#endif /* !LINUXKM_AKCIPHER_NO_SIGNVERIFY */
/* pkcs1 callbacks */
#ifdef WOLFSSL_SHA224 #ifdef WOLFSSL_SHA224
static int km_pkcs1_sha224_init(struct crypto_sig *tfm); static int km_pkcs1_sha224_init(struct tfm_type *tfm);
#endif /* WOLFSSL_SHA224 */ #endif /* WOLFSSL_SHA224 */
#ifndef NO_SHA256 #ifndef NO_SHA256
static int km_pkcs1_sha256_init(struct crypto_sig *tfm); static int km_pkcs1_sha256_init(struct tfm_type *tfm);
#endif /* !NO_SHA256 */ #endif /* !NO_SHA256 */
#ifdef WOLFSSL_SHA384 #ifdef WOLFSSL_SHA384
static int km_pkcs1_sha384_init(struct crypto_sig *tfm); static int km_pkcs1_sha384_init(struct tfm_type *tfm);
#endif /* WOLFSSL_SHA384 */ #endif /* WOLFSSL_SHA384 */
#ifdef WOLFSSL_SHA512 #ifdef WOLFSSL_SHA512
static int km_pkcs1_sha512_init(struct crypto_sig *tfm); static int km_pkcs1_sha512_init(struct tfm_type *tfm);
#endif /* WOLFSSL_SHA512 */ #endif /* WOLFSSL_SHA512 */
#ifdef WOLFSSL_SHA3 #ifdef WOLFSSL_SHA3
static int km_pkcs1_sha3_256_init(struct crypto_sig *tfm); static int km_pkcs1_sha3_256_init(struct tfm_type *tfm);
static int km_pkcs1_sha3_384_init(struct crypto_sig *tfm); static int km_pkcs1_sha3_384_init(struct tfm_type *tfm);
static int km_pkcs1_sha3_512_init(struct crypto_sig *tfm); static int km_pkcs1_sha3_512_init(struct tfm_type *tfm);
#endif /* WOLFSSL_SHA3 */ #endif /* WOLFSSL_SHA3 */
#endif /* !LINUXKM_AKCIPHER_NO_SIGNVERIFY */
/* akcipher callbacks */ /* akcipher callbacks */
static int km_pkcs1pad_enc(struct akcipher_request *req); static int km_pkcs1pad_enc(struct akcipher_request *req);
@@ -539,6 +528,19 @@ static struct akcipher_alg direct_rsa = {
#endif /* WOLFSSL_SHA3 */ #endif /* WOLFSSL_SHA3 */
#endif /* !LINUXKM_AKCIPHER_NO_SIGNVERIFY */ #endif /* !LINUXKM_AKCIPHER_NO_SIGNVERIFY */
static inline void km_rsa_ctx_clear(struct km_rsa_ctx * ctx)
{
if (ctx) {
if (ctx->key) {
wc_FreeRsaKey(ctx->key);
free(ctx->key);
ctx->key = NULL;
}
wc_FreeRng(&ctx->rng);
}
}
static int km_rsa_ctx_init(struct km_rsa_ctx * ctx, int hash_oid) static int km_rsa_ctx_init(struct km_rsa_ctx * ctx, int hash_oid)
{ {
int ret = 0; int ret = 0;
@@ -626,11 +628,7 @@ static int km_rsa_ctx_init(struct km_rsa_ctx * ctx, int hash_oid)
out: out:
if (ret != 0) { if (ret != 0) {
if (ctx->key) { km_rsa_ctx_clear(ctx);
free(ctx->key);
ctx->key = NULL;
}
wc_FreeRng(&ctx->rng);
} }
return ret; return ret;
@@ -954,47 +952,25 @@ static int km_direct_rsa_init(struct crypto_akcipher *tfm)
} }
#endif /* LINUXKM_DIRECT_RSA */ #endif /* LINUXKM_DIRECT_RSA */
static inline void km_rsa_ctx_clear(struct km_rsa_ctx * ctx)
{
if (ctx) {
if (ctx->key) {
wc_FreeRsaKey(ctx->key);
free(ctx->key);
ctx->key = NULL;
}
wc_FreeRng(&ctx->rng);
}
}
static void km_rsa_exit(struct crypto_akcipher *tfm) static void km_rsa_exit(struct crypto_akcipher *tfm)
{ {
struct km_rsa_ctx * ctx = NULL; struct km_rsa_ctx * ctx = NULL;
ctx = akcipher_tfm_ctx(tfm); ctx = akcipher_tfm_ctx(tfm);
km_rsa_ctx_clear(ctx); km_rsa_ctx_clear(ctx);
#ifdef WOLFKM_DEBUG_RSA #ifdef WOLFKM_DEBUG_RSA
pr_info("info: exiting km_rsa_exit\n"); pr_info("info: exiting km_rsa_exit\n");
#endif /* WOLFKM_DEBUG_RSA */ #endif /* WOLFKM_DEBUG_RSA */
return; return;
} }
#if !defined(LINUXKM_AKCIPHER_NO_SIGNVERIFY) #if defined(LINUXKM_AKCIPHER_NO_SIGNVERIFY)
#define tfm_type crypto_akcipher
#define tfm_ctx_cb akcipher_tfm_ctx
#else
#define tfm_type crypto_sig
#define tfm_ctx_cb crypto_sig_ctx
static int km_pkcs1pad_init(struct crypto_akcipher * tfm) static int km_pkcs1pad_init(struct crypto_akcipher * tfm)
{ {
struct km_rsa_ctx * ctx = NULL; struct km_rsa_ctx * ctx = NULL;
ctx = akcipher_tfm_ctx(tfm); ctx = akcipher_tfm_ctx(tfm);
return km_rsa_ctx_init(ctx, 0); return km_rsa_ctx_init(ctx, 0);
} }
#endif /* !LINUXKM_AKCIPHER_NO_SIGNVERIFY */ #endif /* LINUXKM_AKCIPHER_NO_SIGNVERIFY */
#ifdef WOLFSSL_SHA224 #ifdef WOLFSSL_SHA224
static int km_pkcs1_sha224_init(struct tfm_type *tfm) static int km_pkcs1_sha224_init(struct tfm_type *tfm)