programs/ssl/ssl_test_lib.c - mirror/mbed-tls.git - TrustedFirmware Git Browser

 /*
  *  Common code library for SSL test programs.
  *
  *  In addition to the functions in this file, there is shared source code
  *  that cannot be compiled separately in "ssl_test_common_source.c".
  *
  *  Copyright The Mbed TLS Contributors
  *  SPDX-License-Identifier: Apache-2.0
  *
  *  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.
  */

 #define MBEDTLS_ALLOW_PRIVATE_ACCESS

 #include "ssl_test_lib.h"

 #if defined(MBEDTLS_TEST_HOOKS)
 #include "test/helpers.h"
 #endif

 #if !defined(MBEDTLS_SSL_TEST_IMPOSSIBLE)

 void my_debug(void *ctx, int level,
               const char *file, int line,
               const char *str)
 {
     const char *p, *basename;

     /* Extract basename from file */
     for (p = basename = file; *p != '\0'; p++) {
         if (*p == '/' || *p == '\\') {
             basename = p + 1;
         }
     }

     mbedtls_fprintf((FILE *) ctx, "%s:%04d: |%d| %s",
                     basename, line, level, str);
     fflush((FILE *) ctx);
 }

 #if defined(MBEDTLS_HAVE_TIME)
 mbedtls_time_t dummy_constant_time(mbedtls_time_t *time)
 {
     (void) time;
     return 0x5af2a056;
 }
 #endif

 #if !defined(MBEDTLS_TEST_USE_PSA_CRYPTO_RNG)
 static int dummy_entropy(void *data, unsigned char *output, size_t len)
 {
     size_t i;
     int ret;
     (void) data;

     ret = mbedtls_entropy_func(data, output, len);
     for (i = 0; i < len; i++) {
         //replace result with pseudo random
         output[i] = (unsigned char) rand();
     }
     return ret;
 }
 #endif

 void rng_init(rng_context_t *rng)
 {
 #if defined(MBEDTLS_TEST_USE_PSA_CRYPTO_RNG)
     (void) rng;
     psa_crypto_init();
 #else /* !MBEDTLS_TEST_USE_PSA_CRYPTO_RNG */

 #if defined(MBEDTLS_CTR_DRBG_C)
     mbedtls_ctr_drbg_init(&rng->drbg);
 #elif defined(MBEDTLS_HMAC_DRBG_C)
     mbedtls_hmac_drbg_init(&rng->drbg);
 #else
 #error "No DRBG available"
 #endif

     mbedtls_entropy_init(&rng->entropy);
 #endif /* !MBEDTLS_TEST_USE_PSA_CRYPTO_RNG */
 }

 int rng_seed(rng_context_t *rng, int reproducible, const char *pers)
 {
 #if defined(MBEDTLS_USE_PSA_CRYPTO)
     if (reproducible) {
         mbedtls_fprintf(stderr,
                         "MBEDTLS_USE_PSA_CRYPTO does not support reproducible mode.\n");
         return -1;
     }
 #endif
 #if defined(MBEDTLS_TEST_USE_PSA_CRYPTO_RNG)
     /* The PSA crypto RNG does its own seeding. */
     (void) rng;
     (void) pers;
     if (reproducible) {
         mbedtls_fprintf(stderr,
                         "The PSA RNG does not support reproducible mode.\n");
         return -1;
     }
     return 0;
 #else /* !MBEDTLS_TEST_USE_PSA_CRYPTO_RNG */
     int (*f_entropy)(void *, unsigned char *, size_t) =
         (reproducible ? dummy_entropy : mbedtls_entropy_func);

     if (reproducible) {
         srand(1);
     }

 #if defined(MBEDTLS_CTR_DRBG_C)
     int ret = mbedtls_ctr_drbg_seed(&rng->drbg,
                                     f_entropy, &rng->entropy,
                                     (const unsigned char *) pers,
                                     strlen(pers));
 #elif defined(MBEDTLS_HMAC_DRBG_C)
 #if defined(MBEDTLS_MD_CAN_SHA256)
     const mbedtls_md_type_t md_type = MBEDTLS_MD_SHA256;
 #elif defined(MBEDTLS_MD_CAN_SHA512)
     const mbedtls_md_type_t md_type = MBEDTLS_MD_SHA512;
 #else
 #error "No message digest available for HMAC_DRBG"
 #endif
     int ret = mbedtls_hmac_drbg_seed(&rng->drbg,
                                      mbedtls_md_info_from_type(md_type),
                                      f_entropy, &rng->entropy,
                                      (const unsigned char *) pers,
                                      strlen(pers));
 #else /* !defined(MBEDTLS_CTR_DRBG_C) && !defined(MBEDTLS_HMAC_DRBG_C) */
 #error "No DRBG available"
 #endif /* !defined(MBEDTLS_CTR_DRBG_C) && !defined(MBEDTLS_HMAC_DRBG_C) */

     if (ret != 0) {
         mbedtls_printf(" failed\n  ! mbedtls_ctr_drbg_seed returned -0x%x\n",
                        (unsigned int) -ret);
         return ret;
     }
 #endif /* !MBEDTLS_TEST_USE_PSA_CRYPTO_RNG */

     return 0;
 }

 void rng_free(rng_context_t *rng)
 {
 #if defined(MBEDTLS_TEST_USE_PSA_CRYPTO_RNG)
     (void) rng;
     /* Deinitialize the PSA crypto subsystem. This deactivates all PSA APIs.
      * This is ok because none of our applications try to do any crypto after
      * deinitializing the RNG. */
     mbedtls_psa_crypto_free();
 #else /* !MBEDTLS_TEST_USE_PSA_CRYPTO_RNG */

 #if defined(MBEDTLS_CTR_DRBG_C)
     mbedtls_ctr_drbg_free(&rng->drbg);
 #elif defined(MBEDTLS_HMAC_DRBG_C)
     mbedtls_hmac_drbg_free(&rng->drbg);
 #else
 #error "No DRBG available"
 #endif

     mbedtls_entropy_free(&rng->entropy);
 #endif /* !MBEDTLS_TEST_USE_PSA_CRYPTO_RNG */
 }

 int rng_get(void *p_rng, unsigned char *output, size_t output_len)
 {
 #if defined(MBEDTLS_TEST_USE_PSA_CRYPTO_RNG)
     (void) p_rng;
     return mbedtls_psa_get_random(MBEDTLS_PSA_RANDOM_STATE,
                                   output, output_len);
 #else /* !MBEDTLS_TEST_USE_PSA_CRYPTO_RNG */
     rng_context_t *rng = p_rng;

 #if defined(MBEDTLS_CTR_DRBG_C)
     return mbedtls_ctr_drbg_random(&rng->drbg, output, output_len);
 #elif defined(MBEDTLS_HMAC_DRBG_C)
     return mbedtls_hmac_drbg_random(&rng->drbg, output, output_len);
 #else
 #error "No DRBG available"
 #endif

 #endif /* !MBEDTLS_TEST_USE_PSA_CRYPTO_RNG */
 }

 int key_opaque_alg_parse(const char *arg, const char **alg1, const char **alg2)
 {
     char *separator;
     if ((separator = strchr(arg, ',')) == NULL) {
         return 1;
     }
     *separator = '\0';

     *alg1 = arg;
     *alg2 = separator + 1;

     if (strcmp(*alg1, "rsa-sign-pkcs1") != 0 &&
         strcmp(*alg1, "rsa-sign-pss") != 0 &&
         strcmp(*alg1, "rsa-sign-pss-sha256") != 0 &&
         strcmp(*alg1, "rsa-sign-pss-sha384") != 0 &&
         strcmp(*alg1, "rsa-sign-pss-sha512") != 0 &&
         strcmp(*alg1, "rsa-decrypt") != 0 &&
         strcmp(*alg1, "ecdsa-sign") != 0 &&
         strcmp(*alg1, "ecdh") != 0) {
         return 1;
     }

     if (strcmp(*alg2, "rsa-sign-pkcs1") != 0 &&
         strcmp(*alg2, "rsa-sign-pss") != 0 &&
         strcmp(*alg1, "rsa-sign-pss-sha256") != 0 &&
         strcmp(*alg1, "rsa-sign-pss-sha384") != 0 &&
         strcmp(*alg1, "rsa-sign-pss-sha512") != 0 &&
         strcmp(*alg2, "rsa-decrypt") != 0 &&
         strcmp(*alg2, "ecdsa-sign") != 0 &&
         strcmp(*alg2, "ecdh") != 0 &&
         strcmp(*alg2, "none") != 0) {
         return 1;
     }

     return 0;
 }

 #if defined(MBEDTLS_USE_PSA_CRYPTO)
 int key_opaque_set_alg_usage(const char *alg1, const char *alg2,
                              psa_algorithm_t *psa_alg1,
                              psa_algorithm_t *psa_alg2,
                              psa_key_usage_t *usage,
                              mbedtls_pk_type_t key_type)
 {
     if (strcmp(alg1, "none") != 0) {
         const char *algs[] = { alg1, alg2 };
         psa_algorithm_t *psa_algs[] = { psa_alg1, psa_alg2 };

         for (int i = 0; i < 2; i++) {
             if (strcmp(algs[i], "rsa-sign-pkcs1") == 0) {
                 *psa_algs[i] = PSA_ALG_RSA_PKCS1V15_SIGN(PSA_ALG_ANY_HASH);
                 *usage |= PSA_KEY_USAGE_SIGN_HASH;
             } else if (strcmp(algs[i], "rsa-sign-pss") == 0) {
                 *psa_algs[i] = PSA_ALG_RSA_PSS(PSA_ALG_ANY_HASH);
                 *usage |= PSA_KEY_USAGE_SIGN_HASH;
             } else if (strcmp(algs[i], "rsa-sign-pss-sha256") == 0) {
                 *psa_algs[i] = PSA_ALG_RSA_PSS(PSA_ALG_SHA_256);
                 *usage |= PSA_KEY_USAGE_SIGN_HASH;
             } else if (strcmp(algs[i], "rsa-sign-pss-sha384") == 0) {
                 *psa_algs[i] = PSA_ALG_RSA_PSS(PSA_ALG_SHA_384);
                 *usage |= PSA_KEY_USAGE_SIGN_HASH;
             } else if (strcmp(algs[i], "rsa-sign-pss-sha512") == 0) {
                 *psa_algs[i] = PSA_ALG_RSA_PSS(PSA_ALG_SHA_512);
                 *usage |= PSA_KEY_USAGE_SIGN_HASH;
             } else if (strcmp(algs[i], "rsa-decrypt") == 0) {
                 *psa_algs[i] = PSA_ALG_RSA_PKCS1V15_CRYPT;
                 *usage |= PSA_KEY_USAGE_DECRYPT;
             } else if (strcmp(algs[i], "ecdsa-sign") == 0) {
                 *psa_algs[i] = PSA_ALG_ECDSA(PSA_ALG_ANY_HASH);
                 *usage |= PSA_KEY_USAGE_SIGN_HASH;
             } else if (strcmp(algs[i], "ecdh") == 0) {
                 *psa_algs[i] = PSA_ALG_ECDH;
                 *usage |= PSA_KEY_USAGE_DERIVE;
             } else if (strcmp(algs[i], "none") == 0) {
                 *psa_algs[i] = PSA_ALG_NONE;
             }
         }
     } else {
         if (key_type == MBEDTLS_PK_ECKEY) {
             *psa_alg1 = PSA_ALG_ECDSA(PSA_ALG_ANY_HASH);
             *psa_alg2 = PSA_ALG_ECDH;
             *usage = PSA_KEY_USAGE_SIGN_HASH | PSA_KEY_USAGE_DERIVE;
         } else if (key_type == MBEDTLS_PK_RSA) {
             *psa_alg1 = PSA_ALG_RSA_PKCS1V15_SIGN(PSA_ALG_ANY_HASH);
             *psa_alg2 = PSA_ALG_RSA_PSS(PSA_ALG_ANY_HASH);
             *usage = PSA_KEY_USAGE_SIGN_HASH;
         } else {
             return 1;
         }
     }

     return 0;
 }
 #endif /* MBEDTLS_USE_PSA_CRYPTO */

 #if defined(MBEDTLS_X509_TRUSTED_CERTIFICATE_CALLBACK)
 int ca_callback(void *data, mbedtls_x509_crt const *child,
                 mbedtls_x509_crt **candidates)
 {
     int ret = 0;
     mbedtls_x509_crt *ca = (mbedtls_x509_crt *) data;
     mbedtls_x509_crt *first;

     /* This is a test-only implementation of the CA callback
      * which always returns the entire list of trusted certificates.
      * Production implementations managing a large number of CAs
      * should use an efficient presentation and lookup for the
      * set of trusted certificates (such as a hashtable) and only
      * return those trusted certificates which satisfy basic
      * parental checks, such as the matching of child `Issuer`
      * and parent `Subject` field or matching key identifiers. */
     ((void) child);

     first = mbedtls_calloc(1, sizeof(mbedtls_x509_crt));
     if (first == NULL) {
         ret = -1;
         goto exit;
     }
     mbedtls_x509_crt_init(first);

     if (mbedtls_x509_crt_parse_der(first, ca->raw.p, ca->raw.len) != 0) {
         ret = -1;
         goto exit;
     }

     while (ca->next != NULL) {
         ca = ca->next;
         if (mbedtls_x509_crt_parse_der(first, ca->raw.p, ca->raw.len) != 0) {
             ret = -1;
             goto exit;
         }
     }

 exit:

     if (ret != 0) {
         mbedtls_x509_crt_free(first);
         mbedtls_free(first);
         first = NULL;
     }

     *candidates = first;
     return ret;
 }
 #endif /* MBEDTLS_X509_TRUSTED_CERTIFICATE_CALLBACK */

 int delayed_recv(void *ctx, unsigned char *buf, size_t len)
 {
     static int first_try = 1;
     int ret;

     if (first_try) {
         first_try = 0;
         return MBEDTLS_ERR_SSL_WANT_READ;
     }

     ret = mbedtls_net_recv(ctx, buf, len);
     if (ret != MBEDTLS_ERR_SSL_WANT_READ) {
         first_try = 1; /* Next call will be a new operation */
     }
     return ret;
 }

 int delayed_send(void *ctx, const unsigned char *buf, size_t len)
 {
     static int first_try = 1;
     int ret;

     if (first_try) {
         first_try = 0;
         return MBEDTLS_ERR_SSL_WANT_WRITE;
     }

     ret = mbedtls_net_send(ctx, buf, len);
     if (ret != MBEDTLS_ERR_SSL_WANT_WRITE) {
         first_try = 1; /* Next call will be a new operation */
     }
     return ret;
 }

 #if !defined(MBEDTLS_TIMING_C)
 int idle(mbedtls_net_context *fd,
          int idle_reason)
 #else
 int idle(mbedtls_net_context *fd,
          mbedtls_timing_delay_context *timer,
          int idle_reason)
 #endif
 {
     int ret;
     int poll_type = 0;

     if (idle_reason == MBEDTLS_ERR_SSL_WANT_WRITE) {
         poll_type = MBEDTLS_NET_POLL_WRITE;
     } else if (idle_reason == MBEDTLS_ERR_SSL_WANT_READ) {
         poll_type = MBEDTLS_NET_POLL_READ;
     }
 #if !defined(MBEDTLS_TIMING_C)
     else {
         return 0;
     }
 #endif

     while (1) {
         /* Check if timer has expired */
 #if defined(MBEDTLS_TIMING_C)
         if (timer != NULL &&
             mbedtls_timing_get_delay(timer) == 2) {
             break;
         }
 #endif /* MBEDTLS_TIMING_C */

         /* Check if underlying transport became available */
         if (poll_type != 0) {
             ret = mbedtls_net_poll(fd, poll_type, 0);
             if (ret < 0) {
                 return ret;
             }
             if (ret == poll_type) {
                 break;
             }
         }
     }

     return 0;
 }

 #if defined(MBEDTLS_TEST_HOOKS)

 void test_hooks_init(void)
 {
     mbedtls_test_info_reset();

 #if defined(MBEDTLS_TEST_MUTEX_USAGE)
     mbedtls_test_mutex_usage_init();
 #endif
 }

 int test_hooks_failure_detected(void)
 {
 #if defined(MBEDTLS_TEST_MUTEX_USAGE)
     /* Errors are reported via mbedtls_test_info. */
     mbedtls_test_mutex_usage_check();
 #endif

     if (mbedtls_test_info.result != MBEDTLS_TEST_RESULT_SUCCESS) {
         return 1;
     }
     return 0;
 }

 void test_hooks_free(void)
 {
 }

 #endif /* MBEDTLS_TEST_HOOKS */

 #endif /* !defined(MBEDTLS_SSL_TEST_IMPOSSIBLE) */
	/*
	* Common code library for SSL test programs.
	*
	* In addition to the functions in this file, there is shared source code
	* that cannot be compiled separately in "ssl_test_common_source.c".
	*
	* Copyright The Mbed TLS Contributors
	* SPDX-License-Identifier: Apache-2.0
	*
	* 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.
	*/

	#define MBEDTLS_ALLOW_PRIVATE_ACCESS

	#include "ssl_test_lib.h"

	#if defined(MBEDTLS_TEST_HOOKS)
	#include "test/helpers.h"
	#endif

	#if !defined(MBEDTLS_SSL_TEST_IMPOSSIBLE)

	void my_debug(void *ctx, int level,
	const char *file, int line,
	const char *str)
	{
	const char p, basename;

	/* Extract basename from file */
	for (p = basename = file; *p != '\0'; p++) {
	if (p == '/' \|\| p == '\\') {
	basename = p + 1;
	}
	}

	mbedtls_fprintf((FILE *) ctx, "%s:%04d: \|%d\| %s",
	basename, line, level, str);
	fflush((FILE *) ctx);
	}

	#if defined(MBEDTLS_HAVE_TIME)
	mbedtls_time_t dummy_constant_time(mbedtls_time_t *time)
	{
	(void) time;
	return 0x5af2a056;
	}
	#endif

	#if !defined(MBEDTLS_TEST_USE_PSA_CRYPTO_RNG)
	static int dummy_entropy(void data, unsigned char output, size_t len)
	{
	size_t i;
	int ret;
	(void) data;

	ret = mbedtls_entropy_func(data, output, len);
	for (i = 0; i < len; i++) {
	//replace result with pseudo random
	output[i] = (unsigned char) rand();
	}
	return ret;
	}
	#endif

	void rng_init(rng_context_t *rng)
	{
	#if defined(MBEDTLS_TEST_USE_PSA_CRYPTO_RNG)
	(void) rng;
	psa_crypto_init();
	#else /* !MBEDTLS_TEST_USE_PSA_CRYPTO_RNG */

	#if defined(MBEDTLS_CTR_DRBG_C)
	mbedtls_ctr_drbg_init(&rng->drbg);
	#elif defined(MBEDTLS_HMAC_DRBG_C)
	mbedtls_hmac_drbg_init(&rng->drbg);
	#else
	#error "No DRBG available"
	#endif

	mbedtls_entropy_init(&rng->entropy);
	#endif /* !MBEDTLS_TEST_USE_PSA_CRYPTO_RNG */
	}

	int rng_seed(rng_context_t rng, int reproducible, const char pers)
	{
	#if defined(MBEDTLS_USE_PSA_CRYPTO)
	if (reproducible) {
	mbedtls_fprintf(stderr,
	"MBEDTLS_USE_PSA_CRYPTO does not support reproducible mode.\n");
	return -1;
	}
	#endif
	#if defined(MBEDTLS_TEST_USE_PSA_CRYPTO_RNG)
	/* The PSA crypto RNG does its own seeding. */
	(void) rng;
	(void) pers;
	if (reproducible) {
	mbedtls_fprintf(stderr,
	"The PSA RNG does not support reproducible mode.\n");
	return -1;
	}
	return 0;
	#else /* !MBEDTLS_TEST_USE_PSA_CRYPTO_RNG */
	int (f_entropy)(void , unsigned char *, size_t) =
	(reproducible ? dummy_entropy : mbedtls_entropy_func);

	if (reproducible) {
	srand(1);
	}

	#if defined(MBEDTLS_CTR_DRBG_C)
	int ret = mbedtls_ctr_drbg_seed(&rng->drbg,
	f_entropy, &rng->entropy,
	(const unsigned char *) pers,
	strlen(pers));
	#elif defined(MBEDTLS_HMAC_DRBG_C)
	#if defined(MBEDTLS_MD_CAN_SHA256)
	const mbedtls_md_type_t md_type = MBEDTLS_MD_SHA256;
	#elif defined(MBEDTLS_MD_CAN_SHA512)
	const mbedtls_md_type_t md_type = MBEDTLS_MD_SHA512;
	#else
	#error "No message digest available for HMAC_DRBG"
	#endif
	int ret = mbedtls_hmac_drbg_seed(&rng->drbg,
	mbedtls_md_info_from_type(md_type),
	f_entropy, &rng->entropy,
	(const unsigned char *) pers,
	strlen(pers));
	#else /* !defined(MBEDTLS_CTR_DRBG_C) && !defined(MBEDTLS_HMAC_DRBG_C) */
	#error "No DRBG available"
	#endif /* !defined(MBEDTLS_CTR_DRBG_C) && !defined(MBEDTLS_HMAC_DRBG_C) */

	if (ret != 0) {
	mbedtls_printf(" failed\n ! mbedtls_ctr_drbg_seed returned -0x%x\n",
	(unsigned int) -ret);
	return ret;
	}
	#endif /* !MBEDTLS_TEST_USE_PSA_CRYPTO_RNG */

	return 0;
	}

	void rng_free(rng_context_t *rng)
	{
	#if defined(MBEDTLS_TEST_USE_PSA_CRYPTO_RNG)
	(void) rng;
	/* Deinitialize the PSA crypto subsystem. This deactivates all PSA APIs.
	* This is ok because none of our applications try to do any crypto after
	* deinitializing the RNG. */
	mbedtls_psa_crypto_free();
	#else /* !MBEDTLS_TEST_USE_PSA_CRYPTO_RNG */

	#if defined(MBEDTLS_CTR_DRBG_C)
	mbedtls_ctr_drbg_free(&rng->drbg);
	#elif defined(MBEDTLS_HMAC_DRBG_C)
	mbedtls_hmac_drbg_free(&rng->drbg);
	#else
	#error "No DRBG available"
	#endif

	mbedtls_entropy_free(&rng->entropy);
	#endif /* !MBEDTLS_TEST_USE_PSA_CRYPTO_RNG */
	}

	int rng_get(void p_rng, unsigned char output, size_t output_len)
	{
	#if defined(MBEDTLS_TEST_USE_PSA_CRYPTO_RNG)
	(void) p_rng;
	return mbedtls_psa_get_random(MBEDTLS_PSA_RANDOM_STATE,
	output, output_len);
	#else /* !MBEDTLS_TEST_USE_PSA_CRYPTO_RNG */
	rng_context_t *rng = p_rng;

	#if defined(MBEDTLS_CTR_DRBG_C)
	return mbedtls_ctr_drbg_random(&rng->drbg, output, output_len);
	#elif defined(MBEDTLS_HMAC_DRBG_C)
	return mbedtls_hmac_drbg_random(&rng->drbg, output, output_len);
	#else
	#error "No DRBG available"
	#endif

	#endif /* !MBEDTLS_TEST_USE_PSA_CRYPTO_RNG */
	}

	int key_opaque_alg_parse(const char arg, const char alg1, const char *alg2)
	{
	char *separator;
	if ((separator = strchr(arg, ',')) == NULL) {
	return 1;
	}
	*separator = '\0';

	*alg1 = arg;
	*alg2 = separator + 1;

	if (strcmp(*alg1, "rsa-sign-pkcs1") != 0 &&
	strcmp(*alg1, "rsa-sign-pss") != 0 &&
	strcmp(*alg1, "rsa-sign-pss-sha256") != 0 &&
	strcmp(*alg1, "rsa-sign-pss-sha384") != 0 &&
	strcmp(*alg1, "rsa-sign-pss-sha512") != 0 &&
	strcmp(*alg1, "rsa-decrypt") != 0 &&
	strcmp(*alg1, "ecdsa-sign") != 0 &&
	strcmp(*alg1, "ecdh") != 0) {
	return 1;
	}

	if (strcmp(*alg2, "rsa-sign-pkcs1") != 0 &&
	strcmp(*alg2, "rsa-sign-pss") != 0 &&
	strcmp(*alg1, "rsa-sign-pss-sha256") != 0 &&
	strcmp(*alg1, "rsa-sign-pss-sha384") != 0 &&
	strcmp(*alg1, "rsa-sign-pss-sha512") != 0 &&
	strcmp(*alg2, "rsa-decrypt") != 0 &&
	strcmp(*alg2, "ecdsa-sign") != 0 &&
	strcmp(*alg2, "ecdh") != 0 &&
	strcmp(*alg2, "none") != 0) {
	return 1;
	}

	return 0;
	}

	#if defined(MBEDTLS_USE_PSA_CRYPTO)
	int key_opaque_set_alg_usage(const char alg1, const char alg2,
	psa_algorithm_t *psa_alg1,
	psa_algorithm_t *psa_alg2,
	psa_key_usage_t *usage,
	mbedtls_pk_type_t key_type)
	{
	if (strcmp(alg1, "none") != 0) {
	const char *algs[] = { alg1, alg2 };
	psa_algorithm_t *psa_algs[] = { psa_alg1, psa_alg2 };

	for (int i = 0; i < 2; i++) {
	if (strcmp(algs[i], "rsa-sign-pkcs1") == 0) {
	*psa_algs[i] = PSA_ALG_RSA_PKCS1V15_SIGN(PSA_ALG_ANY_HASH);
	*usage \|= PSA_KEY_USAGE_SIGN_HASH;
	} else if (strcmp(algs[i], "rsa-sign-pss") == 0) {
	*psa_algs[i] = PSA_ALG_RSA_PSS(PSA_ALG_ANY_HASH);
	*usage \|= PSA_KEY_USAGE_SIGN_HASH;
	} else if (strcmp(algs[i], "rsa-sign-pss-sha256") == 0) {
	*psa_algs[i] = PSA_ALG_RSA_PSS(PSA_ALG_SHA_256);
	*usage \|= PSA_KEY_USAGE_SIGN_HASH;
	} else if (strcmp(algs[i], "rsa-sign-pss-sha384") == 0) {
	*psa_algs[i] = PSA_ALG_RSA_PSS(PSA_ALG_SHA_384);
	*usage \|= PSA_KEY_USAGE_SIGN_HASH;
	} else if (strcmp(algs[i], "rsa-sign-pss-sha512") == 0) {
	*psa_algs[i] = PSA_ALG_RSA_PSS(PSA_ALG_SHA_512);
	*usage \|= PSA_KEY_USAGE_SIGN_HASH;
	} else if (strcmp(algs[i], "rsa-decrypt") == 0) {
	*psa_algs[i] = PSA_ALG_RSA_PKCS1V15_CRYPT;
	*usage \|= PSA_KEY_USAGE_DECRYPT;
	} else if (strcmp(algs[i], "ecdsa-sign") == 0) {
	*psa_algs[i] = PSA_ALG_ECDSA(PSA_ALG_ANY_HASH);
	*usage \|= PSA_KEY_USAGE_SIGN_HASH;
	} else if (strcmp(algs[i], "ecdh") == 0) {
	*psa_algs[i] = PSA_ALG_ECDH;
	*usage \|= PSA_KEY_USAGE_DERIVE;
	} else if (strcmp(algs[i], "none") == 0) {
	*psa_algs[i] = PSA_ALG_NONE;
	}
	}
	} else {
	if (key_type == MBEDTLS_PK_ECKEY) {
	*psa_alg1 = PSA_ALG_ECDSA(PSA_ALG_ANY_HASH);
	*psa_alg2 = PSA_ALG_ECDH;
	*usage = PSA_KEY_USAGE_SIGN_HASH \| PSA_KEY_USAGE_DERIVE;
	} else if (key_type == MBEDTLS_PK_RSA) {
	*psa_alg1 = PSA_ALG_RSA_PKCS1V15_SIGN(PSA_ALG_ANY_HASH);
	*psa_alg2 = PSA_ALG_RSA_PSS(PSA_ALG_ANY_HASH);
	*usage = PSA_KEY_USAGE_SIGN_HASH;
	} else {
	return 1;
	}
	}

	return 0;
	}
	#endif /* MBEDTLS_USE_PSA_CRYPTO */

	#if defined(MBEDTLS_X509_TRUSTED_CERTIFICATE_CALLBACK)
	int ca_callback(void data, mbedtls_x509_crt const child,
	mbedtls_x509_crt **candidates)
	{
	int ret = 0;
	mbedtls_x509_crt ca = (mbedtls_x509_crt ) data;
	mbedtls_x509_crt *first;

	/* This is a test-only implementation of the CA callback
	* which always returns the entire list of trusted certificates.
	* Production implementations managing a large number of CAs
	* should use an efficient presentation and lookup for the
	* set of trusted certificates (such as a hashtable) and only
	* return those trusted certificates which satisfy basic
	* parental checks, such as the matching of child `Issuer`
	* and parent `Subject` field or matching key identifiers. */
	((void) child);

	first = mbedtls_calloc(1, sizeof(mbedtls_x509_crt));
	if (first == NULL) {
	ret = -1;
	goto exit;
	}
	mbedtls_x509_crt_init(first);

	if (mbedtls_x509_crt_parse_der(first, ca->raw.p, ca->raw.len) != 0) {
	ret = -1;
	goto exit;
	}

	while (ca->next != NULL) {
	ca = ca->next;
	if (mbedtls_x509_crt_parse_der(first, ca->raw.p, ca->raw.len) != 0) {
	ret = -1;
	goto exit;
	}
	}

	exit:

	if (ret != 0) {
	mbedtls_x509_crt_free(first);
	mbedtls_free(first);
	first = NULL;
	}

	*candidates = first;
	return ret;
	}
	#endif /* MBEDTLS_X509_TRUSTED_CERTIFICATE_CALLBACK */

	int delayed_recv(void ctx, unsigned char buf, size_t len)
	{
	static int first_try = 1;
	int ret;

	if (first_try) {
	first_try = 0;
	return MBEDTLS_ERR_SSL_WANT_READ;
	}

	ret = mbedtls_net_recv(ctx, buf, len);
	if (ret != MBEDTLS_ERR_SSL_WANT_READ) {
	first_try = 1; /* Next call will be a new operation */
	}
	return ret;
	}

	int delayed_send(void ctx, const unsigned char buf, size_t len)
	{
	static int first_try = 1;
	int ret;

	if (first_try) {
	first_try = 0;
	return MBEDTLS_ERR_SSL_WANT_WRITE;
	}

	ret = mbedtls_net_send(ctx, buf, len);
	if (ret != MBEDTLS_ERR_SSL_WANT_WRITE) {
	first_try = 1; /* Next call will be a new operation */
	}
	return ret;
	}

	#if !defined(MBEDTLS_TIMING_C)
	int idle(mbedtls_net_context *fd,
	int idle_reason)
	#else
	int idle(mbedtls_net_context *fd,
	mbedtls_timing_delay_context *timer,
	int idle_reason)
	#endif
	{
	int ret;
	int poll_type = 0;

	if (idle_reason == MBEDTLS_ERR_SSL_WANT_WRITE) {
	poll_type = MBEDTLS_NET_POLL_WRITE;
	} else if (idle_reason == MBEDTLS_ERR_SSL_WANT_READ) {
	poll_type = MBEDTLS_NET_POLL_READ;
	}
	#if !defined(MBEDTLS_TIMING_C)
	else {
	return 0;
	}
	#endif

	while (1) {
	/* Check if timer has expired */
	#if defined(MBEDTLS_TIMING_C)
	if (timer != NULL &&
	mbedtls_timing_get_delay(timer) == 2) {
	break;
	}
	#endif /* MBEDTLS_TIMING_C */

	/* Check if underlying transport became available */
	if (poll_type != 0) {
	ret = mbedtls_net_poll(fd, poll_type, 0);
	if (ret < 0) {
	return ret;
	}
	if (ret == poll_type) {
	break;
	}
	}
	}

	return 0;
	}

	#if defined(MBEDTLS_TEST_HOOKS)

	void test_hooks_init(void)
	{
	mbedtls_test_info_reset();

	#if defined(MBEDTLS_TEST_MUTEX_USAGE)
	mbedtls_test_mutex_usage_init();
	#endif
	}

	int test_hooks_failure_detected(void)
	{
	#if defined(MBEDTLS_TEST_MUTEX_USAGE)
	/* Errors are reported via mbedtls_test_info. */
	mbedtls_test_mutex_usage_check();
	#endif

	if (mbedtls_test_info.result != MBEDTLS_TEST_RESULT_SUCCESS) {
	return 1;
	}
	return 0;
	}

	void test_hooks_free(void)
	{
	}

	#endif /* MBEDTLS_TEST_HOOKS */

	#endif /* !defined(MBEDTLS_SSL_TEST_IMPOSSIBLE) */