crypto/evp/e_sm4.c

   1 /*
   2  * Copyright 2017 The OpenSSL Project Authors. All Rights Reserved.
   3  * Copyright 2017 Ribose Inc. All Rights Reserved.
   4  * Ported from Ribose contributions from Botan.
   5  *
   6  * Licensed under the Apache License 2.0 (the "License").  You may not use
   7  * this file except in compliance with the License.  You can obtain a copy
   8  * in the file LICENSE in the source distribution or at
   9  * https://www.openssl.org/source/license.html
  10  */
  11
  12 #include "internal/cryptlib.h"
  13 #ifndef OPENSSL_NO_SM4
  14 # include <openssl/evp.h>
  15 # include <openssl/modes.h>
  16 # include "crypto/sm4.h"
  17 # include "crypto/evp.h"
  18
  19 typedef struct {
  20     SM4_KEY ks;
  21 } EVP_SM4_KEY;
  22
  23 static int sm4_init_key(EVP_CIPHER_CTX *ctx, const unsigned char *key,
  24                         const unsigned char *iv, int enc)
  25 {
  26     SM4_set_key(key, EVP_CIPHER_CTX_get_cipher_data(ctx));
  27     return 1;
  28 }
  29
  30 static void sm4_cbc_encrypt(const unsigned char *in, unsigned char *out,
  31                             size_t len, const SM4_KEY *key,
  32                             unsigned char *ivec, const int enc)
  33 {
  34     if (enc)
  35         CRYPTO_cbc128_encrypt(in, out, len, key, ivec,
  36                               (block128_f)SM4_encrypt);
  37     else
  38         CRYPTO_cbc128_decrypt(in, out, len, key, ivec,
  39                               (block128_f)SM4_decrypt);
  40 }
  41
  42 static void sm4_cfb128_encrypt(const unsigned char *in, unsigned char *out,
  43                                size_t length, const SM4_KEY *key,
  44                                unsigned char *ivec, int *num, const int enc)
  45 {
  46     CRYPTO_cfb128_encrypt(in, out, length, key, ivec, num, enc,
  47                           (block128_f)SM4_encrypt);
  48 }
  49
  50 static void sm4_ecb_encrypt(const unsigned char *in, unsigned char *out,
  51                             const SM4_KEY *key, const int enc)
  52 {
  53     if (enc)
  54         SM4_encrypt(in, out, key);
  55     else
  56         SM4_decrypt(in, out, key);
  57 }
  58
  59 static void sm4_ofb128_encrypt(const unsigned char *in, unsigned char *out,
  60                                size_t length, const SM4_KEY *key,
  61                                unsigned char *ivec, int *num)
  62 {
  63     CRYPTO_ofb128_encrypt(in, out, length, key, ivec, num,
  64                           (block128_f)SM4_encrypt);
  65 }
  66
  67 IMPLEMENT_BLOCK_CIPHER(sm4, ks, sm4, EVP_SM4_KEY, NID_sm4,
  68                        16, 16, 16, 128, EVP_CIPH_FLAG_DEFAULT_ASN1,
  69                        sm4_init_key, 0, 0, 0, 0)
  70
  71 static int sm4_ctr_cipher(EVP_CIPHER_CTX *ctx, unsigned char *out,
  72                           const unsigned char *in, size_t len)
  73 {
  74     unsigned int num = EVP_CIPHER_CTX_num(ctx);
  75     EVP_SM4_KEY *dat = EVP_C_DATA(EVP_SM4_KEY, ctx);
  76
  77     CRYPTO_ctr128_encrypt(in, out, len, &dat->ks,
  78                           EVP_CIPHER_CTX_iv_noconst(ctx),
  79                           EVP_CIPHER_CTX_buf_noconst(ctx), &num,
  80                           (block128_f)SM4_encrypt);
  81     EVP_CIPHER_CTX_set_num(ctx, num);
  82     return 1;
  83 }
  84
  85 static const EVP_CIPHER sm4_ctr_mode = {
  86     NID_sm4_ctr, 1, 16, 16,
  87     EVP_CIPH_CTR_MODE,
  88     sm4_init_key,
  89     sm4_ctr_cipher,
  90     NULL,
  91     sizeof(EVP_SM4_KEY),
  92     NULL, NULL, NULL, NULL
  93 };
  94
  95 const EVP_CIPHER *EVP_sm4_ctr(void)
  96 {
  97     return &sm4_ctr_mode;
  98 }
  99
 100 #endif