89 lines
2.3 KiB
C
89 lines
2.3 KiB
C
/* 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 */
|