providers/implementations/ciphers/cipher_sm4_hw.c

   1 /*
   2  * Copyright 2019-2023 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 const PROV_CIPHER_HW *ossl_prov_cipher_hw_sm4_##mode(size_t keybits)           \
 131 {                                                                              \
 132     return &sm4_##mode;                                                        \
 133 }
 134
 135 PROV_CIPHER_HW_sm4_mode(cbc)
 136 PROV_CIPHER_HW_sm4_mode(ecb)
 137 PROV_CIPHER_HW_sm4_mode(ofb128)
 138 PROV_CIPHER_HW_sm4_mode(cfb128)
 139 PROV_CIPHER_HW_sm4_mode(ctr)