mbedtls/tf-psa-crypto/programs/fuzz/fuzz_privkey.c

#define MBEDTLS_DECLARE_PRIVATE_IDENTIFIERS

#include <stdint.h>
#include <stdlib.h>
#include "mbedtls/pk.h"
#include "mbedtls/private/pk_private.h"
#include "fuzz_common.h"

#if defined(MBEDTLS_PK_PARSE_C) && defined(MBEDTLS_PK_WRITE_C)

#define MAX_LEN 0x1000
static uint8_t out_buf[MAX_LEN];

int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size)
{

    int ret;
    mbedtls_pk_context pk;

    if (Size > MAX_LEN) {
        //only work on small inputs
        Size = MAX_LEN;
    }

    mbedtls_pk_init(&pk);

    psa_status_t status = psa_crypto_init();
    if (status != PSA_SUCCESS) {
        goto exit;
    }

    ret = mbedtls_pk_parse_key(&pk, Data, Size, NULL, 0);
    if (ret == 0) {
        ret = mbedtls_pk_write_key_der(&pk, out_buf, Size);
        if (ret != 0) {
            abort();
        }
    }
exit:
    mbedtls_pk_free(&pk);
    mbedtls_psa_crypto_free();

    return 0;
}

#else /* MBEDTLS_PK_PARSE_C && MBEDTLS_PK_WRITE_C */

int LLVMFuzzerTestOneInput(const uint8_t *Data, size_t Size)
{
    (void) Data;
    (void) Size;
    return 0;
}
#endif /* MBEDTLS_PK_PARSE_C && MBEDTLS_PK_WRITE_C */