/**
 *
 * \section COPYRIGHT
 *
 * Copyright 2013-2021 Software Radio Systems Limited
 *
 * By using this file, you agree to the terms and conditions set
 * forth in the LICENSE file which can be found at the top level of
 * the distribution.
 *
 */

#ifndef SRSRAN_SSL_H
#define SRSRAN_SSL_H

#ifdef HAVE_POLARSSL

#include "polarssl/aes.h"
#include "polarssl/sha256.h"

inline void sha256(const unsigned char* key,
                   size_t               keylen,
                   const unsigned char* input,
                   size_t               ilen,
                   unsigned char        output[32],
                   int                  is224)
{
  sha256_hmac(key, keylen, input, ilen, output, is224);
}

#endif // HAVE_POLARSSL

#ifdef HAVE_MBEDTLS

#include "mbedtls/aes.h"
#include "mbedtls/md.h"

typedef mbedtls_aes_context aes_context;

#define AES_ENCRYPT 1
#define AES_DECRYPT 0

inline int aes_setkey_enc(aes_context* ctx, const unsigned char* key, unsigned int keysize)
{
  return mbedtls_aes_setkey_enc(ctx, key, keysize);
}

inline int aes_crypt_ecb(aes_context* ctx, int mode, const unsigned char input[16], unsigned char output[16])
{
  return mbedtls_aes_crypt_ecb(ctx, mode, input, output);
}

inline int aes_crypt_ctr(aes_context*         ctx,
                         size_t               length,
                         size_t*              nc_off,
                         unsigned char        nonce_counter[16],
                         unsigned char        stream_block[16],
                         const unsigned char* input,
                         unsigned char*       output)
{
  return mbedtls_aes_crypt_ctr(ctx, length, nc_off, nonce_counter, stream_block, input, output);
}

inline void sha256(const unsigned char* key,
                   size_t               keylen,
                   const unsigned char* input,
                   size_t               ilen,
                   unsigned char        output[32],
                   int                  is224)
{
  mbedtls_md_hmac(mbedtls_md_info_from_type(MBEDTLS_MD_SHA256), key, keylen, input, ilen, output);
}

#endif // HAVE_MBEDTLS

#endif // SRSRAN_SSL_H