1 /* 2 * Copyright 2019-2024 The OpenSSL Project Authors. All Rights Reserved. 3 * 4 * Licensed under the Apache License 2.0 (the "License"). You may not use 5 * this file except in compliance with the License. You can obtain a copy 6 * in the file LICENSE in the source distribution or at 7 * https://www.openssl.org/source/license.html 8 */ 9 10 #include "cipher_sm4.h" 11 12 static int cipher_hw_sm4_initkey(PROV_CIPHER_CTX *ctx, 13 const unsigned char *key, size_t keylen) 14 { 15 PROV_SM4_CTX *sctx = (PROV_SM4_CTX *)ctx; 16 SM4_KEY *ks = &sctx->ks.ks; 17 18 ctx->ks = ks; 19 if (ctx->enc 20 || (ctx->mode != EVP_CIPH_ECB_MODE 21 && ctx->mode != EVP_CIPH_CBC_MODE)) { 22 #ifdef HWSM4_CAPABLE 23 if (HWSM4_CAPABLE) { 24 HWSM4_set_encrypt_key(key, ks); 25 ctx->block = (block128_f)HWSM4_encrypt; 26 ctx->stream.cbc = NULL; 27 #ifdef HWSM4_cbc_encrypt 28 if (ctx->mode == EVP_CIPH_CBC_MODE) 29 ctx->stream.cbc = (cbc128_f)HWSM4_cbc_encrypt; 30 else 31 #endif 32 #ifdef HWSM4_ecb_encrypt 33 if (ctx->mode == EVP_CIPH_ECB_MODE) 34 ctx->stream.ecb = (ecb128_f)HWSM4_ecb_encrypt; 35 else 36 #endif 37 #ifdef HWSM4_ctr32_encrypt_blocks 38 if (ctx->mode == EVP_CIPH_CTR_MODE) 39 ctx->stream.ctr = (ctr128_f)HWSM4_ctr32_encrypt_blocks; 40 else 41 #endif 42 (void)0; /* terminate potentially open 'else' */ 43 } else 44 #endif 45 #ifdef VPSM4_EX_CAPABLE 46 if (VPSM4_EX_CAPABLE) { 47 vpsm4_ex_set_encrypt_key(key, ks); 48 ctx->block = (block128_f)vpsm4_ex_encrypt; 49 ctx->stream.cbc = NULL; 50 if (ctx->mode == EVP_CIPH_CBC_MODE) 51 ctx->stream.cbc = (cbc128_f)vpsm4_ex_cbc_encrypt; 52 else if (ctx->mode == EVP_CIPH_ECB_MODE) 53 ctx->stream.ecb = (ecb128_f)vpsm4_ex_ecb_encrypt; 54 else if (ctx->mode == EVP_CIPH_CTR_MODE) 55 ctx->stream.ctr = (ctr128_f)vpsm4_ex_ctr32_encrypt_blocks; 56 } else 57 #endif 58 #ifdef VPSM4_CAPABLE 59 if (VPSM4_CAPABLE) { 60 vpsm4_set_encrypt_key(key, ks); 61 ctx->block = (block128_f)vpsm4_encrypt; 62 ctx->stream.cbc = NULL; 63 if (ctx->mode == EVP_CIPH_CBC_MODE) 64 ctx->stream.cbc = (cbc128_f)vpsm4_cbc_encrypt; 65 else if (ctx->mode == EVP_CIPH_ECB_MODE) 66 ctx->stream.ecb = (ecb128_f)vpsm4_ecb_encrypt; 67 else if (ctx->mode == EVP_CIPH_CTR_MODE) 68 ctx->stream.ctr = (ctr128_f)vpsm4_ctr32_encrypt_blocks; 69 } else 70 #endif 71 { 72 ossl_sm4_set_key(key, ks); 73 ctx->block = (block128_f)ossl_sm4_encrypt; 74 } 75 } else { 76 #ifdef HWSM4_CAPABLE 77 if (HWSM4_CAPABLE) { 78 HWSM4_set_decrypt_key(key, ks); 79 ctx->block = (block128_f)HWSM4_decrypt; 80 ctx->stream.cbc = NULL; 81 #ifdef HWSM4_cbc_encrypt 82 if (ctx->mode == EVP_CIPH_CBC_MODE) 83 ctx->stream.cbc = (cbc128_f)HWSM4_cbc_encrypt; 84 #endif 85 #ifdef HWSM4_ecb_encrypt 86 if (ctx->mode == EVP_CIPH_ECB_MODE) 87 ctx->stream.ecb = (ecb128_f)HWSM4_ecb_encrypt; 88 #endif 89 } else 90 #endif 91 #ifdef VPSM4_EX_CAPABLE 92 if (VPSM4_EX_CAPABLE) { 93 vpsm4_ex_set_decrypt_key(key, ks); 94 ctx->block = (block128_f)vpsm4_ex_decrypt; 95 ctx->stream.cbc = NULL; 96 if (ctx->mode == EVP_CIPH_CBC_MODE) 97 ctx->stream.cbc = (cbc128_f)vpsm4_ex_cbc_encrypt; 98 else if (ctx->mode == EVP_CIPH_ECB_MODE) 99 ctx->stream.ecb = (ecb128_f)vpsm4_ex_ecb_encrypt; 100 } else 101 #endif 102 #ifdef VPSM4_CAPABLE 103 if (VPSM4_CAPABLE) { 104 vpsm4_set_decrypt_key(key, ks); 105 ctx->block = (block128_f)vpsm4_decrypt; 106 ctx->stream.cbc = NULL; 107 if (ctx->mode == EVP_CIPH_CBC_MODE) 108 ctx->stream.cbc = (cbc128_f)vpsm4_cbc_encrypt; 109 else if (ctx->mode == EVP_CIPH_ECB_MODE) 110 ctx->stream.ecb = (ecb128_f)vpsm4_ecb_encrypt; 111 } else 112 #endif 113 { 114 ossl_sm4_set_key(key, ks); 115 ctx->block = (block128_f)ossl_sm4_decrypt; 116 } 117 } 118 119 return 1; 120 } 121 122 IMPLEMENT_CIPHER_HW_COPYCTX(cipher_hw_sm4_copyctx, PROV_SM4_CTX) 123 124 # define PROV_CIPHER_HW_sm4_mode(mode) \ 125 static const PROV_CIPHER_HW sm4_##mode = { \ 126 cipher_hw_sm4_initkey, \ 127 ossl_cipher_hw_generic_##mode, \ 128 cipher_hw_sm4_copyctx \ 129 }; \ 130 PROV_CIPHER_HW_declare(mode) \ 131 const PROV_CIPHER_HW *ossl_prov_cipher_hw_sm4_##mode(size_t keybits) \ 132 { \ 133 PROV_CIPHER_HW_select(mode) \ 134 return &sm4_##mode; \ 135 } 136 137 #if defined(OPENSSL_CPUID_OBJ) && defined(__riscv) && __riscv_xlen == 64 138 # include "cipher_sm4_hw_rv64i.inc" 139 #else 140 /* The generic case */ 141 # define PROV_CIPHER_HW_declare(mode) 142 # define PROV_CIPHER_HW_select(mode) 143 #endif 144 145 PROV_CIPHER_HW_sm4_mode(cbc) 146 PROV_CIPHER_HW_sm4_mode(ecb) 147 PROV_CIPHER_HW_sm4_mode(ofb128) 148 PROV_CIPHER_HW_sm4_mode(cfb128) 149 PROV_CIPHER_HW_sm4_mode(ctr) 150