/* BEGIN_HEADER */
#include "mbedtls/pk.h"
#include "mbedtls/pem.h"
#include "mbedtls/oid.h"
/* END_HEADER */

/* BEGIN_DEPENDENCIES
 * depends_on:MBEDTLS_PK_PARSE_C:MBEDTLS_PK_WRITE_C:MBEDTLS_BIGNUM_C:MBEDTLS_FS_IO
 * END_DEPENDENCIES
 */

/* BEGIN_CASE depends_on:MBEDTLS_PEM_WRITE_C */
void pk_write_pubkey_check(char *key_file)
{
    mbedtls_pk_context key;
    unsigned char buf[5000];
    unsigned char check_buf[5000];
    int ret;
    FILE *f;
    size_t ilen, pem_len, buf_index;

    memset(buf, 0, sizeof(buf));
    memset(check_buf, 0, sizeof(check_buf));

    mbedtls_pk_init(&key);
    TEST_ASSERT(mbedtls_pk_parse_public_keyfile(&key, key_file) == 0);

    ret = mbedtls_pk_write_pubkey_pem(&key, buf, sizeof(buf));
    TEST_ASSERT(ret == 0);

    pem_len = strlen((char *) buf);

    // check that the rest of the buffer remains clear
    for (buf_index = pem_len; buf_index < sizeof(buf); ++buf_index) {
        TEST_ASSERT(buf[buf_index] == 0);
    }

    f = fopen(key_file, "r");
    TEST_ASSERT(f != NULL);
    ilen = fread(check_buf, 1, sizeof(check_buf), f);
    fclose(f);

    TEST_ASSERT(ilen == pem_len);
    TEST_ASSERT(memcmp((char *) buf, (char *) check_buf, ilen) == 0);

exit:
    mbedtls_pk_free(&key);
}
/* END_CASE */

/* BEGIN_CASE depends_on:MBEDTLS_PEM_WRITE_C */
void pk_write_key_check(char *key_file)
{
    mbedtls_pk_context key;
    unsigned char buf[5000];
    unsigned char check_buf[5000];
    int ret;
    FILE *f;
    size_t ilen, pem_len, buf_index;

    memset(buf, 0, sizeof(buf));
    memset(check_buf, 0, sizeof(check_buf));

    mbedtls_pk_init(&key);
    TEST_ASSERT(mbedtls_pk_parse_keyfile(&key, key_file, NULL,
                                         mbedtls_test_rnd_std_rand, NULL) == 0);

    ret = mbedtls_pk_write_key_pem(&key, buf, sizeof(buf));
    TEST_ASSERT(ret == 0);

    pem_len = strlen((char *) buf);

    // check that the rest of the buffer remains clear
    for (buf_index = pem_len; buf_index < sizeof(buf); ++buf_index) {
        TEST_ASSERT(buf[buf_index] == 0);
    }

    f = fopen(key_file, "r");
    TEST_ASSERT(f != NULL);
    ilen = fread(check_buf, 1, sizeof(check_buf), f);
    fclose(f);

    TEST_ASSERT(ilen == strlen((char *) buf));
    TEST_ASSERT(memcmp((char *) buf, (char *) check_buf, ilen) == 0);

exit:
    mbedtls_pk_free(&key);
}
/* END_CASE */