sm4_aes.h

#pragma once
#include "ypc/corecommon/crypto/aes_gcm_traits.h"
#include <cstdint>
 
namespace ypc {
namespace crypto {
 
#define INITIALIZATION_VECTOR_SIZE 12
class sm4_aes {
public:
  static inline uint32_t get_mac_code_size() { return 16; }
  static inline uint32_t get_cipher_size(uint32_t data_size) {
    return data_size + INITIALIZATION_VECTOR_SIZE;
  }
  static inline uint32_t get_data_size(uint32_t cipher_size) {
    return cipher_size - INITIALIZATION_VECTOR_SIZE;
  }
  static inline uint32_t get_key_size() { return 16; }
 
  static uint32_t encrypt_with_prefix(const uint8_t *key, uint32_t key_size,
                                      const uint8_t *data, uint32_t data_size,
                                      uint32_t prefix, uint8_t *cipher,
                                      uint32_t cipher_size, uint8_t *out_mac);
 
  static uint32_t decrypt_with_prefix(const uint8_t *key, uint32_t key_size,
                                      const uint8_t *cipher,
                                      uint32_t cipher_size, uint32_t prefix,
                                      uint8_t *data, uint32_t data_size,
                                      const uint8_t *in_mac);
};
template <> struct aes_gcm_traits<sm4_aes> {
  constexpr static bool value = true;
};
}
}