/**

 * WinPR: Windows Portable Runtime

 *

 * Copyright 2015 Marc-Andre Moreau <marcandre.moreau@gmail.com>

 *

 * Licensed under the Apache License, Version 2.0 (the "License");

 * you may not use this file except in compliance with the License.

 * You may obtain a copy of the License at

 *

 *     http://www.apache.org/licenses/LICENSE-2.0

 *

 * Unless required by applicable law or agreed to in writing, software

 * distributed under the License is distributed on an "AS IS" BASIS,

 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.

 * See the License for the specific language governing permissions and

 * limitations under the License.

 */

#include <winpr/config.h>

#include <winpr/crt.h>

#include <winpr/assert.h>

#include <winpr/crypto.h>

#include "../log.h"

#define TAG WINPR_TAG("crypto.cipher")

#if defined(WITH_INTERNAL_RC4)

#include "rc4.h"

#endif

#ifdef WITH_OPENSSL

#include <openssl/aes.h>

#include <openssl/rc4.h>

#include <openssl/des.h>

#include <openssl/evp.h>

#endif

#ifdef WITH_MBEDTLS

#include <mbedtls/md.h>

#include <mbedtls/aes.h>

#include <mbedtls/des.h>

#include <mbedtls/cipher.h>

#if MBEDTLS_VERSION_MAJOR < 3

#define mbedtls_cipher_info_get_iv_size(_info) (_info->iv_size)

#define mbedtls_cipher_info_get_key_bitlen(_info) (_info->key_bitlen)

#endif

#endif

struct winpr_cipher_ctx_private_st

{

  WINPR_CIPHER_TYPE cipher;

  WINPR_CRYPTO_OPERATION op;

#ifdef WITH_OPENSSL

  EVP_CIPHER_CTX* ectx;

#endif

#ifdef WITH_MBEDTLS

  mbedtls_cipher_context_t* mctx;

#endif

};

/**

 * RC4

 */

struct winpr_rc4_ctx_private_st

{

#if defined(WITH_INTERNAL_RC4)

  winpr_int_RC4_CTX* ictx;

#else

#if defined(WITH_OPENSSL)

  EVP_CIPHER_CTX* ctx;

#endif

#endif

};

static WINPR_RC4_CTX* winpr_RC4_New_Internal(const BYTE* key, size_t keylen, BOOL override_fips)

{

  if (!key || (keylen == 0))

    return nullptr;

  WINPR_RC4_CTX* ctx = (WINPR_RC4_CTX*)calloc(1, sizeof(WINPR_RC4_CTX));

  if (!ctx)

    return nullptr;

#if defined(WITH_INTERNAL_RC4)

  WINPR_UNUSED(override_fips);

  ctx->ictx = winpr_int_rc4_new(key, keylen);

  if (!ctx->ictx)

    goto fail;

#elif defined(WITH_OPENSSL)

  const EVP_CIPHER* evp = nullptr;

  if (keylen > INT_MAX)

    goto fail;

  ctx->ctx = EVP_CIPHER_CTX_new();

  if (!ctx->ctx)

    goto fail;

  evp = EVP_rc4();

  if (!evp)

    goto fail;

  EVP_CIPHER_CTX_reset(ctx->ctx);

  if (EVP_EncryptInit_ex(ctx->ctx, evp, nullptr, nullptr, nullptr) != 1)

    goto fail;

  /* EVP_CIPH_FLAG_NON_FIPS_ALLOW does not exist before openssl 1.0.1 */

#if !(OPENSSL_VERSION_NUMBER < 0x10001000L)

  if (override_fips == TRUE)

    EVP_CIPHER_CTX_set_flags(ctx->ctx, EVP_CIPH_FLAG_NON_FIPS_ALLOW);

#endif

  EVP_CIPHER_CTX_set_key_length(ctx->ctx, (int)keylen);

  if (EVP_EncryptInit_ex(ctx->ctx, nullptr, nullptr, key, nullptr) != 1)

    goto fail;

#endif

  return ctx;

fail:

  WINPR_PRAGMA_DIAG_PUSH

  WINPR_PRAGMA_DIAG_IGNORED_MISMATCHED_DEALLOC

  winpr_RC4_Free(ctx);

  WINPR_PRAGMA_DIAG_POP

  return nullptr;

}

WINPR_RC4_CTX* winpr_RC4_New_Allow_FIPS(const void* key, size_t keylen)

{

  return winpr_RC4_New_Internal(key, keylen, TRUE);

}

WINPR_RC4_CTX* winpr_RC4_New(const void* key, size_t keylen)

{

  return winpr_RC4_New_Internal(key, keylen, FALSE);

}

BOOL winpr_RC4_Update(WINPR_RC4_CTX* ctx, size_t length, const void* input, void* output)

{

  WINPR_ASSERT(ctx);

#if defined(WITH_INTERNAL_RC4)

  return winpr_int_rc4_update(ctx->ictx, length, input, output);

#elif defined(WITH_OPENSSL)

  WINPR_ASSERT(ctx->ctx);

  int outputLength = 0;

  if (length > INT_MAX)

    return FALSE;

  WINPR_ASSERT(ctx);

  return EVP_CipherUpdate(ctx->ctx, output, &outputLength, input, (int)length) == 1;

#endif

  return FALSE;

}

void winpr_RC4_Free(WINPR_RC4_CTX* ctx)

{

  if (!ctx)

    return;

#if defined(WITH_INTERNAL_RC4)

  winpr_int_rc4_free(ctx->ictx);

#elif defined(WITH_OPENSSL)

  EVP_CIPHER_CTX_free(ctx->ctx);

#endif

  free(ctx);

}

/**

 * Generic Cipher API

 */

#ifdef WITH_OPENSSL

extern const EVP_MD* winpr_openssl_get_evp_md(WINPR_MD_TYPE md);

#endif

#ifdef WITH_MBEDTLS

extern mbedtls_md_type_t winpr_mbedtls_get_md_type(int md);

#endif

struct cipher_map

{

  WINPR_CIPHER_TYPE md;

  const char* name;

};

static const struct cipher_map s_cipher_map[] = {

  { WINPR_CIPHER_NONE, "none" },

  { WINPR_CIPHER_NULL, "null" },

  { WINPR_CIPHER_AES_128_ECB, "aes-128-ecb" },

  { WINPR_CIPHER_AES_192_ECB, "aes-192-ecb" },

  { WINPR_CIPHER_AES_256_ECB, "aes-256-ecb" },

  { WINPR_CIPHER_AES_128_CBC, "aes-128-cbc" },

  { WINPR_CIPHER_AES_192_CBC, "aes-192-cbc" },

  { WINPR_CIPHER_AES_256_CBC, "aes-256-cbc" },

  { WINPR_CIPHER_AES_128_CFB128, "aes-128-cfb128" },

  { WINPR_CIPHER_AES_192_CFB128, "aes-192-cfb128" },

  { WINPR_CIPHER_AES_256_CFB128, "aes-256-cfb128" },

  { WINPR_CIPHER_AES_128_CTR, "aes-128-ctr" },

  { WINPR_CIPHER_AES_192_CTR, "aes-192-ctr" },

  { WINPR_CIPHER_AES_256_CTR, "aes-256-ctr" },

  { WINPR_CIPHER_AES_128_GCM, "aes-128-gcm" },

  { WINPR_CIPHER_AES_192_GCM, "aes-192-gcm" },

  { WINPR_CIPHER_AES_256_GCM, "aes-256-gcm" },

  { WINPR_CIPHER_CAMELLIA_128_ECB, "camellia-128-ecb" },

  { WINPR_CIPHER_CAMELLIA_192_ECB, "camellia-192-ecb" },

  { WINPR_CIPHER_CAMELLIA_256_ECB, "camellia-256-ecb" },

  { WINPR_CIPHER_CAMELLIA_128_CBC, "camellia-128-cbc" },

  { WINPR_CIPHER_CAMELLIA_192_CBC, "camellia-192-cbc" },

  { WINPR_CIPHER_CAMELLIA_256_CBC, "camellia-256-cbc" },

  { WINPR_CIPHER_CAMELLIA_128_CFB128, "camellia-128-cfb128" },

  { WINPR_CIPHER_CAMELLIA_192_CFB128, "camellia-192-cfb128" },

  { WINPR_CIPHER_CAMELLIA_256_CFB128, "camellia-256-cfb128" },

  { WINPR_CIPHER_CAMELLIA_128_CTR, "camellia-128-ctr" },

  { WINPR_CIPHER_CAMELLIA_192_CTR, "camellia-192-ctr" },

  { WINPR_CIPHER_CAMELLIA_256_CTR, "camellia-256-ctr" },

  { WINPR_CIPHER_CAMELLIA_128_GCM, "camellia-128-gcm" },

  { WINPR_CIPHER_CAMELLIA_192_GCM, "camellia-192-gcm" },

  { WINPR_CIPHER_CAMELLIA_256_GCM, "camellia-256-gcm" },

  { WINPR_CIPHER_DES_ECB, "des-ecb" },

  { WINPR_CIPHER_DES_CBC, "des-cbc" },

  { WINPR_CIPHER_DES_EDE_ECB, "des-ede-ecb" },

  { WINPR_CIPHER_DES_EDE_CBC, "des-ede-cbc" },

  { WINPR_CIPHER_DES_EDE3_ECB, "des-ede3-ecb" },

  { WINPR_CIPHER_DES_EDE3_CBC, "des-ede3-cbc" },

  { WINPR_CIPHER_BLOWFISH_ECB, "blowfish-ecb" },

  { WINPR_CIPHER_BLOWFISH_CBC, "blowfish-cbc" },

  { WINPR_CIPHER_BLOWFISH_CFB64, "blowfish-cfb64" },

  { WINPR_CIPHER_BLOWFISH_CTR, "blowfish-ctr" },

  { WINPR_CIPHER_ARC4_128, "rc4" },

  { WINPR_CIPHER_AES_128_CCM, "aes-128-ccm" },

  { WINPR_CIPHER_AES_192_CCM, "aes-192-ccm" },

  { WINPR_CIPHER_AES_256_CCM, "aes-256-ccm" },

  { WINPR_CIPHER_CAMELLIA_128_CCM, "camellia-128-ccm" },

  { WINPR_CIPHER_CAMELLIA_192_CCM, "camellia-192-ccm" },

  { WINPR_CIPHER_CAMELLIA_256_CCM, "camellia-256-ccm" },

};

static int cipher_compare(const void* a, const void* b)

{

  const WINPR_CIPHER_TYPE* cipher = a;

  const struct cipher_map* map = b;

  if (*cipher == map->md)

    return 0;

  return *cipher > map->md ? 1 : -1;

}

const char* winpr_cipher_type_to_string(WINPR_CIPHER_TYPE md)

{

  WINPR_CIPHER_TYPE lc = md;

  const struct cipher_map* ret = bsearch(&lc, s_cipher_map, ARRAYSIZE(s_cipher_map),

                                         sizeof(struct cipher_map), cipher_compare);

  if (!ret)

    return "unknown";

  return ret->name;

}

static int cipher_string_compare(const void* a, const void* b)

{

  const char* cipher = a;

  const struct cipher_map* map = b;

  return strcmp(cipher, map->name);

}

WINPR_CIPHER_TYPE winpr_cipher_type_from_string(const char* name)

{

  const struct cipher_map* ret = bsearch(name, s_cipher_map, ARRAYSIZE(s_cipher_map),

                                         sizeof(struct cipher_map), cipher_string_compare);

  if (!ret)

    return WINPR_CIPHER_NONE;

  return ret->md;

}

#if defined(WITH_OPENSSL)

static const EVP_CIPHER* winpr_openssl_get_evp_cipher(WINPR_CIPHER_TYPE cipher)

{

  const EVP_CIPHER* evp = nullptr;

  switch (cipher)

  {

    case WINPR_CIPHER_NULL:

      evp = EVP_enc_null();

      break;

    case WINPR_CIPHER_AES_128_ECB:

      evp = EVP_get_cipherbyname("aes-128-ecb");

      break;

    case WINPR_CIPHER_AES_192_ECB:

      evp = EVP_get_cipherbyname("aes-192-ecb");

      break;

    case WINPR_CIPHER_AES_256_ECB:

      evp = EVP_get_cipherbyname("aes-256-ecb");

      break;

    case WINPR_CIPHER_AES_128_CBC:

      evp = EVP_get_cipherbyname("aes-128-cbc");

      break;

    case WINPR_CIPHER_AES_192_CBC:

      evp = EVP_get_cipherbyname("aes-192-cbc");

      break;

    case WINPR_CIPHER_AES_256_CBC:

      evp = EVP_get_cipherbyname("aes-256-cbc");

      break;

    case WINPR_CIPHER_AES_128_CFB128:

      evp = EVP_get_cipherbyname("aes-128-cfb128");

      break;

    case WINPR_CIPHER_AES_192_CFB128:

      evp = EVP_get_cipherbyname("aes-192-cfb128");

      break;

    case WINPR_CIPHER_AES_256_CFB128:

      evp = EVP_get_cipherbyname("aes-256-cfb128");

      break;

    case WINPR_CIPHER_AES_128_CTR:

      evp = EVP_get_cipherbyname("aes-128-ctr");

      break;

    case WINPR_CIPHER_AES_192_CTR:

      evp = EVP_get_cipherbyname("aes-192-ctr");

      break;

    case WINPR_CIPHER_AES_256_CTR:

      evp = EVP_get_cipherbyname("aes-256-ctr");

      break;

    case WINPR_CIPHER_AES_128_GCM:

      evp = EVP_get_cipherbyname("aes-128-gcm");

      break;

    case WINPR_CIPHER_AES_192_GCM:

      evp = EVP_get_cipherbyname("aes-192-gcm");

      break;

    case WINPR_CIPHER_AES_256_GCM:

      evp = EVP_get_cipherbyname("aes-256-gcm");

      break;

    case WINPR_CIPHER_AES_128_CCM:

      evp = EVP_get_cipherbyname("aes-128-ccm");

      break;

    case WINPR_CIPHER_AES_192_CCM:

      evp = EVP_get_cipherbyname("aes-192-ccm");

      break;

    case WINPR_CIPHER_AES_256_CCM:

      evp = EVP_get_cipherbyname("aes-256-ccm");

      break;

    case WINPR_CIPHER_CAMELLIA_128_ECB:

      evp = EVP_get_cipherbyname("camellia-128-ecb");

      break;

    case WINPR_CIPHER_CAMELLIA_192_ECB:

      evp = EVP_get_cipherbyname("camellia-192-ecb");

      break;

    case WINPR_CIPHER_CAMELLIA_256_ECB:

      evp = EVP_get_cipherbyname("camellia-256-ecb");

      break;

    case WINPR_CIPHER_CAMELLIA_128_CBC:

      evp = EVP_get_cipherbyname("camellia-128-cbc");

      break;

    case WINPR_CIPHER_CAMELLIA_192_CBC:

      evp = EVP_get_cipherbyname("camellia-192-cbc");

      break;

    case WINPR_CIPHER_CAMELLIA_256_CBC:

      evp = EVP_get_cipherbyname("camellia-256-cbc");

      break;

    case WINPR_CIPHER_CAMELLIA_128_CFB128:

      evp = EVP_get_cipherbyname("camellia-128-cfb128");

      break;

    case WINPR_CIPHER_CAMELLIA_192_CFB128:

      evp = EVP_get_cipherbyname("camellia-192-cfb128");

      break;

    case WINPR_CIPHER_CAMELLIA_256_CFB128:

      evp = EVP_get_cipherbyname("camellia-256-cfb128");

      break;

    case WINPR_CIPHER_CAMELLIA_128_CTR:

      evp = EVP_get_cipherbyname("camellia-128-ctr");

      break;

    case WINPR_CIPHER_CAMELLIA_192_CTR:

      evp = EVP_get_cipherbyname("camellia-192-ctr");

      break;

    case WINPR_CIPHER_CAMELLIA_256_CTR:

      evp = EVP_get_cipherbyname("camellia-256-ctr");

      break;

    case WINPR_CIPHER_CAMELLIA_128_GCM:

      evp = EVP_get_cipherbyname("camellia-128-gcm");

      break;

    case WINPR_CIPHER_CAMELLIA_192_GCM:

      evp = EVP_get_cipherbyname("camellia-192-gcm");

      break;

    case WINPR_CIPHER_CAMELLIA_256_GCM:

      evp = EVP_get_cipherbyname("camellia-256-gcm");

      break;

    case WINPR_CIPHER_CAMELLIA_128_CCM:

      evp = EVP_get_cipherbyname("camellia-128-ccm");

      break;

    case WINPR_CIPHER_CAMELLIA_192_CCM:

      evp = EVP_get_cipherbyname("camellia-192-ccm");

      break;

    case WINPR_CIPHER_CAMELLIA_256_CCM:

      evp = EVP_get_cipherbyname("camellia-256-ccm");

      break;

    case WINPR_CIPHER_DES_ECB:

      evp = EVP_get_cipherbyname("des-ecb");

      break;

    case WINPR_CIPHER_DES_CBC:

      evp = EVP_get_cipherbyname("des-cbc");

      break;

    case WINPR_CIPHER_DES_EDE_ECB:

      evp = EVP_get_cipherbyname("des-ede-ecb");

      break;

    case WINPR_CIPHER_DES_EDE_CBC:

      evp = EVP_get_cipherbyname("des-ede-cbc");

      break;

    case WINPR_CIPHER_DES_EDE3_ECB:

      evp = EVP_get_cipherbyname("des-ede3-ecb");

      break;

    case WINPR_CIPHER_DES_EDE3_CBC:

      evp = EVP_get_cipherbyname("des-ede3-cbc");

      break;

    case WINPR_CIPHER_ARC4_128:

      evp = EVP_get_cipherbyname("rc4");

      break;

    case WINPR_CIPHER_BLOWFISH_ECB:

      evp = EVP_get_cipherbyname("blowfish-ecb");

      break;

    case WINPR_CIPHER_BLOWFISH_CBC:

      evp = EVP_get_cipherbyname("blowfish-cbc");

      break;

    case WINPR_CIPHER_BLOWFISH_CFB64:

      evp = EVP_get_cipherbyname("blowfish-cfb64");

      break;

    case WINPR_CIPHER_BLOWFISH_CTR:

      evp = EVP_get_cipherbyname("blowfish-ctr");

      break;

    default:

      break;

  }

  return evp;

}

#elif defined(WITH_MBEDTLS)

mbedtls_cipher_type_t winpr_mbedtls_get_cipher_type(int cipher)

{

  mbedtls_cipher_type_t type = MBEDTLS_CIPHER_NONE;

  switch (cipher)

  {

    case WINPR_CIPHER_NONE:

      type = MBEDTLS_CIPHER_NONE;

      break;

    case WINPR_CIPHER_NULL:

      type = MBEDTLS_CIPHER_NULL;

      break;

    case WINPR_CIPHER_AES_128_ECB:

      type = MBEDTLS_CIPHER_AES_128_ECB;

      break;

    case WINPR_CIPHER_AES_192_ECB:

      type = MBEDTLS_CIPHER_AES_192_ECB;

      break;

    case WINPR_CIPHER_AES_256_ECB:

      type = MBEDTLS_CIPHER_AES_256_ECB;

      break;

    case WINPR_CIPHER_AES_128_CBC:

      type = MBEDTLS_CIPHER_AES_128_CBC;

      break;

    case WINPR_CIPHER_AES_192_CBC:

      type = MBEDTLS_CIPHER_AES_192_CBC;

      break;

    case WINPR_CIPHER_AES_256_CBC:

      type = MBEDTLS_CIPHER_AES_256_CBC;

      break;

    case WINPR_CIPHER_AES_128_CFB128:

      type = MBEDTLS_CIPHER_AES_128_CFB128;

      break;

    case WINPR_CIPHER_AES_192_CFB128:

      type = MBEDTLS_CIPHER_AES_192_CFB128;

      break;

    case WINPR_CIPHER_AES_256_CFB128:

      type = MBEDTLS_CIPHER_AES_256_CFB128;

      break;

    case WINPR_CIPHER_AES_128_CTR:

      type = MBEDTLS_CIPHER_AES_128_CTR;

      break;

    case WINPR_CIPHER_AES_192_CTR:

      type = MBEDTLS_CIPHER_AES_192_CTR;

      break;

    case WINPR_CIPHER_AES_256_CTR:

      type = MBEDTLS_CIPHER_AES_256_CTR;

      break;

    case WINPR_CIPHER_AES_128_GCM:

      type = MBEDTLS_CIPHER_AES_128_GCM;

      break;

    case WINPR_CIPHER_AES_192_GCM:

      type = MBEDTLS_CIPHER_AES_192_GCM;

      break;

    case WINPR_CIPHER_AES_256_GCM:

      type = MBEDTLS_CIPHER_AES_256_GCM;

      break;

    case WINPR_CIPHER_AES_128_CCM:

      type = MBEDTLS_CIPHER_AES_128_CCM;

      break;

    case WINPR_CIPHER_AES_192_CCM:

      type = MBEDTLS_CIPHER_AES_192_CCM;

      break;

    case WINPR_CIPHER_AES_256_CCM:

      type = MBEDTLS_CIPHER_AES_256_CCM;

      break;

  }

  return type;

}

#endif

#if !defined(WITHOUT_FREERDP_3x_DEPRECATED)

WINPR_CIPHER_CTX* winpr_Cipher_New(WINPR_CIPHER_TYPE cipher, WINPR_CRYPTO_OPERATION op,

                                   const void* key, const void* iv)

{

  return winpr_Cipher_NewEx(cipher, op, key, 0, iv, 0);

}

#endif

WINPR_API WINPR_CIPHER_CTX* winpr_Cipher_NewEx(WINPR_CIPHER_TYPE cipher, WINPR_CRYPTO_OPERATION op,

                                               const void* key, WINPR_ATTR_UNUSED size_t keylen,

                                               const void* iv, WINPR_ATTR_UNUSED size_t ivlen)

{

  if (cipher == WINPR_CIPHER_ARC4_128)

  {

    WLog_ERR(TAG,

             "WINPR_CIPHER_ARC4_128 (RC4) cipher not supported, use winpr_RC4_new instead");

    return nullptr;

  }

  WINPR_CIPHER_CTX* ctx = calloc(1, sizeof(WINPR_CIPHER_CTX));

  if (!ctx)

    return nullptr;

  ctx->cipher = cipher;

  ctx->op = op;

#if defined(WITH_OPENSSL)

  const EVP_CIPHER* evp = winpr_openssl_get_evp_cipher(cipher);

  if (!evp)

    goto fail;

  ctx->ectx = EVP_CIPHER_CTX_new();

  if (!ctx->ectx)

    goto fail;

  {

    const int operation = (op == WINPR_ENCRYPT) ? 1 : 0;

    if (EVP_CipherInit_ex(ctx->ectx, evp, nullptr, key, iv, operation) != 1)

      goto fail;

  }

  EVP_CIPHER_CTX_set_padding(ctx->ectx, 0);

#elif defined(WITH_MBEDTLS)

  mbedtls_cipher_type_t cipher_type = winpr_mbedtls_get_cipher_type(cipher);

  const mbedtls_cipher_info_t* cipher_info = mbedtls_cipher_info_from_type(cipher_type);

  if (!cipher_info)

    goto fail;

  ctx->mctx = calloc(1, sizeof(mbedtls_cipher_context_t));

  if (!ctx->mctx)

    goto fail;

  const mbedtls_operation_t operation = (op == WINPR_ENCRYPT) ? MBEDTLS_ENCRYPT : MBEDTLS_DECRYPT;

  mbedtls_cipher_init(ctx->mctx);

  if (mbedtls_cipher_setup(ctx->mctx, cipher_info) != 0)

    goto fail;

  const int key_bitlen = mbedtls_cipher_get_key_bitlen(ctx->mctx);

  if (mbedtls_cipher_setkey(ctx->mctx, key, key_bitlen, operation) != 0)

    goto fail;

  if (mbedtls_cipher_set_padding_mode(ctx->mctx, MBEDTLS_PADDING_NONE) != 0)

    goto fail;

#endif

  return ctx;

fail:

  winpr_Cipher_Free(ctx);

  return nullptr;

}

BOOL winpr_Cipher_SetPadding(WINPR_CIPHER_CTX* ctx, BOOL enabled)

{

  WINPR_ASSERT(ctx);

#if defined(WITH_OPENSSL)

  if (!ctx->ectx)

    return FALSE;

  EVP_CIPHER_CTX_set_padding(ctx->ectx, enabled);

#elif defined(WITH_MBEDTLS)

  mbedtls_cipher_padding_t option = enabled ? MBEDTLS_PADDING_PKCS7 : MBEDTLS_PADDING_NONE;

  if (mbedtls_cipher_set_padding_mode((mbedtls_cipher_context_t*)ctx, option) != 0)

    return FALSE;

#else

  return FALSE;

#endif

  return TRUE;

}

BOOL winpr_Cipher_Update(WINPR_CIPHER_CTX* ctx, const void* input, size_t ilen, void* output,

                         size_t* olen)

{

  WINPR_ASSERT(ctx);

  WINPR_ASSERT(olen);

#if defined(WITH_OPENSSL)

  int outl = (int)*olen;

  if (ilen > INT_MAX)

  {

    WLog_ERR(TAG, "input length %" PRIuz " > %d, abort", ilen, INT_MAX);

    return FALSE;

  }

  WINPR_ASSERT(ctx->ectx);

  if (EVP_CipherUpdate(ctx->ectx, output, &outl, input, (int)ilen) == 1)

  {

    *olen = (size_t)outl;

    return TRUE;

  }

#elif defined(WITH_MBEDTLS)

  WINPR_ASSERT(ctx->mctx);

  if (mbedtls_cipher_update(ctx->mctx, input, ilen, output, olen) == 0)

    return TRUE;

#endif

  WLog_ERR(TAG, "Failed to update the data");

  return FALSE;

}

BOOL winpr_Cipher_Final(WINPR_CIPHER_CTX* ctx, void* output, size_t* olen)

{

  WINPR_ASSERT(ctx);

#if defined(WITH_OPENSSL)

  int outl = (int)*olen;

  WINPR_ASSERT(ctx->ectx);

  if (EVP_CipherFinal_ex(ctx->ectx, output, &outl) == 1)

  {

    *olen = (size_t)outl;

    return TRUE;

  }

#elif defined(WITH_MBEDTLS)

  WINPR_ASSERT(ctx->mctx);

  if (mbedtls_cipher_finish(ctx->mctx, output, olen) == 0)

    return TRUE;

#endif

  return FALSE;

}

void winpr_Cipher_Free(WINPR_CIPHER_CTX* ctx)

{

  if (!ctx)

    return;

#if defined(WITH_OPENSSL)

  if (ctx->ectx)

    EVP_CIPHER_CTX_free(ctx->ectx);

#elif defined(WITH_MBEDTLS)

  if (ctx->mctx)

  {

    mbedtls_cipher_free(ctx->mctx);

    free(ctx->mctx);

  }

#endif

  free(ctx);

}

/**

 * Key Generation

 */

int winpr_Cipher_BytesToKey(int cipher, WINPR_MD_TYPE md, const void* salt, const void* data,

                            size_t datal, size_t count, void* key, void* iv)

{

  /**

   * Key and IV generation compatible with OpenSSL EVP_BytesToKey():

   * https://www.openssl.org/docs/manmaster/crypto/EVP_BytesToKey.html

   */

#if defined(WITH_OPENSSL)

  const EVP_MD* evp_md = nullptr;

  const EVP_CIPHER* evp_cipher = nullptr;

  evp_md = winpr_openssl_get_evp_md(md);

  evp_cipher = winpr_openssl_get_evp_cipher(WINPR_ASSERTING_INT_CAST(WINPR_CIPHER_TYPE, cipher));

  WINPR_ASSERT(datal <= INT_MAX);

  WINPR_ASSERT(count <= INT_MAX);

  return EVP_BytesToKey(evp_cipher, evp_md, salt, data, (int)datal, (int)count, key, iv);

#elif defined(WITH_MBEDTLS)

  int rv = 0;

  BYTE md_buf[64];

  int niv, nkey, addmd = 0;

  unsigned int mds = 0;

  mbedtls_md_context_t ctx;

  const mbedtls_md_info_t* md_info;

  mbedtls_cipher_type_t cipher_type;

  const mbedtls_cipher_info_t* cipher_info;

  mbedtls_md_type_t md_type = winpr_mbedtls_get_md_type(md);

  md_info = mbedtls_md_info_from_type(md_type);

  cipher_type = winpr_mbedtls_get_cipher_type(cipher);

  cipher_info = mbedtls_cipher_info_from_type(cipher_type);

  nkey = mbedtls_cipher_info_get_key_bitlen(cipher_info) / 8;

  niv = mbedtls_cipher_info_get_iv_size(cipher_info);

  if ((nkey > 64) || (niv > 64))

    return 0;

  if (!data)

    return nkey;

  mbedtls_md_init(&ctx);

  if (mbedtls_md_setup(&ctx, md_info, 0) != 0)

    goto err;

  while (1)

  {

    if (mbedtls_md_starts(&ctx) != 0)

      goto err;

    if (addmd++)

    {

      if (mbedtls_md_update(&ctx, md_buf, mds) != 0)

        goto err;

    }

    if (mbedtls_md_update(&ctx, data, datal) != 0)

      goto err;

    if (salt)

    {

      if (mbedtls_md_update(&ctx, salt, 8) != 0)

        goto err;

    }

    if (mbedtls_md_finish(&ctx, md_buf) != 0)

      goto err;

    mds = mbedtls_md_get_size(md_info);

    for (unsigned int i = 1; i < (unsigned int)count; i++)

    {

      if (mbedtls_md_starts(&ctx) != 0)

        goto err;

      if (mbedtls_md_update(&ctx, md_buf, mds) != 0)

        goto err;

      if (mbedtls_md_finish(&ctx, md_buf) != 0)

        goto err;

    }

    unsigned int i = 0;

    if (nkey)

    {

      while (1)

      {

        if (nkey == 0)

          break;

        if (i == mds)

          break;

        if (key)

          *(BYTE*)(key++) = md_buf[i];

        nkey--;

        i++;

      }

    }

    if (niv && (i != mds))

    {

      while (1)

      {

        if (niv == 0)

          break;

        if (i == mds)

          break;

        if (iv)

          *(BYTE*)(iv++) = md_buf[i];

        niv--;

        i++;

      }

    }

    if ((nkey == 0) && (niv == 0))

      break;

  }

  rv = mbedtls_cipher_info_get_key_bitlen(cipher_info) / 8;

err:

  mbedtls_md_free(&ctx);

  SecureZeroMemory(md_buf, 64);

  return rv;

#else

  return 0;

#endif

}