Paul Bakker | c7bb02b | 2013-09-15 14:54:56 +0200 | [diff] [blame] | 1 | /* BEGIN_HEADER */ |
Manuel Pégourié-Gonnard | 7f80997 | 2015-03-09 17:05:11 +0000 | [diff] [blame] | 2 | #include "mbedtls/pk.h" |
| 3 | #include "mbedtls/pem.h" |
| 4 | #include "mbedtls/oid.h" |
Valerio Setti | 8959095 | 2023-04-17 17:34:09 +0200 | [diff] [blame] | 5 | |
Valerio Setti | c9cb532 | 2023-04-18 11:20:36 +0200 | [diff] [blame] | 6 | typedef enum { |
| 7 | TEST_PEM, |
| 8 | TEST_DER |
| 9 | } pkwrite_file_format_t; |
| 10 | |
Valerio Setti | 547b3a4 | 2023-04-24 10:24:37 +0200 | [diff] [blame] | 11 | /* Helper function for removing "\r" chars from a buffer. */ |
Valerio Setti | 7bacaf8 | 2023-04-24 08:52:16 +0200 | [diff] [blame] | 12 | static void fix_new_lines(unsigned char *in_str, size_t *len) |
| 13 | { |
| 14 | size_t chars_left; |
| 15 | unsigned int i; |
| 16 | |
| 17 | for (i = 0; (i < *len) && (*len > 0); i++) { |
| 18 | if (in_str[i] == '\r') { |
| 19 | if (i < (*len - 1)) { |
| 20 | chars_left = *len - i - 1; |
Valerio Setti | bf974b9 | 2023-04-24 10:26:24 +0200 | [diff] [blame] | 21 | memmove(&in_str[i], &in_str[i+1], chars_left); |
Valerio Setti | 7bacaf8 | 2023-04-24 08:52:16 +0200 | [diff] [blame] | 22 | } else { |
| 23 | in_str[i] = '\0'; |
| 24 | } |
| 25 | *len = *len - 1; |
| 26 | } |
| 27 | } |
| 28 | } |
| 29 | |
Valerio Setti | c60bc5e | 2023-04-17 18:43:06 +0200 | [diff] [blame] | 30 | static void pk_write_check_common(char *key_file, int is_public_key, int is_der) |
Valerio Setti | 8959095 | 2023-04-17 17:34:09 +0200 | [diff] [blame] | 31 | { |
| 32 | mbedtls_pk_context key; |
| 33 | unsigned char *buf = NULL; |
| 34 | unsigned char *check_buf = NULL; |
Valerio Setti | c60bc5e | 2023-04-17 18:43:06 +0200 | [diff] [blame] | 35 | unsigned char *start_buf; |
Valerio Setti | 3401b30 | 2023-04-18 10:42:53 +0200 | [diff] [blame] | 36 | size_t buf_len, check_buf_len; |
Valerio Setti | 8959095 | 2023-04-17 17:34:09 +0200 | [diff] [blame] | 37 | int ret; |
| 38 | |
| 39 | /* Note: if mbedtls_pk_load_file() successfully reads the file, then |
| 40 | it also allocates check_buf, which should be freed on exit */ |
Valerio Setti | 3401b30 | 2023-04-18 10:42:53 +0200 | [diff] [blame] | 41 | TEST_EQUAL(mbedtls_pk_load_file(key_file, &check_buf, &check_buf_len), 0); |
Valerio Setti | 8959095 | 2023-04-17 17:34:09 +0200 | [diff] [blame] | 42 | TEST_ASSERT(check_buf_len > 0); |
| 43 | |
Valerio Setti | 7bacaf8 | 2023-04-24 08:52:16 +0200 | [diff] [blame] | 44 | /* Windows' line ending is different from the Linux's one ("\r\n" vs "\n"). |
| 45 | * Git treats PEM files as text, so when on Windows, it replaces new lines |
| 46 | * with "\r\n" on checkout. |
| 47 | * Unfortunately mbedtls_pk_load_file() loads files in binary format, |
| 48 | * while mbedtls_pk_write_pubkey_pem() goes through the I/O layer which |
| 49 | * uses "\n" for newlines in both Windows and Linux. |
| 50 | * Here we remove the extra "\r" so that "buf" and "check_buf" can be |
| 51 | * easily compared later. */ |
| 52 | if (!is_der) { |
| 53 | fix_new_lines(check_buf, &check_buf_len); |
| 54 | } |
| 55 | TEST_ASSERT(check_buf_len > 0); |
| 56 | |
Valerio Setti | 8959095 | 2023-04-17 17:34:09 +0200 | [diff] [blame] | 57 | ASSERT_ALLOC(buf, check_buf_len); |
| 58 | |
| 59 | mbedtls_pk_init(&key); |
| 60 | if (is_public_key) { |
Valerio Setti | 3401b30 | 2023-04-18 10:42:53 +0200 | [diff] [blame] | 61 | TEST_EQUAL(mbedtls_pk_parse_public_keyfile(&key, key_file), 0); |
Valerio Setti | c60bc5e | 2023-04-17 18:43:06 +0200 | [diff] [blame] | 62 | if (is_der) { |
| 63 | ret = mbedtls_pk_write_pubkey_der(&key, buf, check_buf_len); |
| 64 | } else { |
Valerio Setti | 15cac17 | 2023-04-18 11:25:30 +0200 | [diff] [blame] | 65 | #if defined(MBEDTLS_PEM_WRITE_C) |
Valerio Setti | c60bc5e | 2023-04-17 18:43:06 +0200 | [diff] [blame] | 66 | ret = mbedtls_pk_write_pubkey_pem(&key, buf, check_buf_len); |
Valerio Setti | 15cac17 | 2023-04-18 11:25:30 +0200 | [diff] [blame] | 67 | #else |
| 68 | ret = MBEDTLS_ERR_PK_FEATURE_UNAVAILABLE; |
| 69 | #endif |
Valerio Setti | c60bc5e | 2023-04-17 18:43:06 +0200 | [diff] [blame] | 70 | } |
Valerio Setti | 8959095 | 2023-04-17 17:34:09 +0200 | [diff] [blame] | 71 | } else { |
Valerio Setti | 3401b30 | 2023-04-18 10:42:53 +0200 | [diff] [blame] | 72 | TEST_EQUAL(mbedtls_pk_parse_keyfile(&key, key_file, NULL, |
Valerio Setti | 2280895 | 2023-04-18 12:57:52 +0200 | [diff] [blame] | 73 | mbedtls_test_rnd_std_rand, NULL), 0); |
Valerio Setti | c60bc5e | 2023-04-17 18:43:06 +0200 | [diff] [blame] | 74 | if (is_der) { |
| 75 | ret = mbedtls_pk_write_key_der(&key, buf, check_buf_len); |
| 76 | } else { |
Valerio Setti | 15cac17 | 2023-04-18 11:25:30 +0200 | [diff] [blame] | 77 | #if defined(MBEDTLS_PEM_WRITE_C) |
Valerio Setti | c60bc5e | 2023-04-17 18:43:06 +0200 | [diff] [blame] | 78 | ret = mbedtls_pk_write_key_pem(&key, buf, check_buf_len); |
Valerio Setti | 15cac17 | 2023-04-18 11:25:30 +0200 | [diff] [blame] | 79 | #else |
| 80 | ret = MBEDTLS_ERR_PK_FEATURE_UNAVAILABLE; |
| 81 | #endif |
Valerio Setti | c60bc5e | 2023-04-17 18:43:06 +0200 | [diff] [blame] | 82 | } |
Valerio Setti | 8959095 | 2023-04-17 17:34:09 +0200 | [diff] [blame] | 83 | } |
Valerio Setti | 8959095 | 2023-04-17 17:34:09 +0200 | [diff] [blame] | 84 | |
Valerio Setti | c60bc5e | 2023-04-17 18:43:06 +0200 | [diff] [blame] | 85 | if (is_der) { |
Valerio Setti | 3401b30 | 2023-04-18 10:42:53 +0200 | [diff] [blame] | 86 | TEST_LE_U(1, ret); |
| 87 | buf_len = ret; |
| 88 | start_buf = buf + check_buf_len - buf_len; |
Valerio Setti | c60bc5e | 2023-04-17 18:43:06 +0200 | [diff] [blame] | 89 | } else { |
Valerio Setti | 3401b30 | 2023-04-18 10:42:53 +0200 | [diff] [blame] | 90 | TEST_EQUAL(ret, 0); |
Valerio Setti | 2280895 | 2023-04-18 12:57:52 +0200 | [diff] [blame] | 91 | buf_len = strlen((char *) buf) + 1; /* +1 takes the string terminator into account */ |
Valerio Setti | c60bc5e | 2023-04-17 18:43:06 +0200 | [diff] [blame] | 92 | start_buf = buf; |
| 93 | } |
| 94 | |
Valerio Setti | 3401b30 | 2023-04-18 10:42:53 +0200 | [diff] [blame] | 95 | ASSERT_COMPARE(start_buf, buf_len, check_buf, check_buf_len); |
Valerio Setti | 8959095 | 2023-04-17 17:34:09 +0200 | [diff] [blame] | 96 | |
| 97 | exit: |
| 98 | mbedtls_free(buf); |
| 99 | mbedtls_free(check_buf); |
| 100 | mbedtls_pk_free(&key); |
| 101 | } |
Paul Bakker | c7bb02b | 2013-09-15 14:54:56 +0200 | [diff] [blame] | 102 | /* END_HEADER */ |
| 103 | |
| 104 | /* BEGIN_DEPENDENCIES |
Hanno Becker | 19d858e | 2018-10-16 13:46:25 +0100 | [diff] [blame] | 105 | * depends_on:MBEDTLS_PK_PARSE_C:MBEDTLS_PK_WRITE_C:MBEDTLS_BIGNUM_C:MBEDTLS_FS_IO |
Paul Bakker | c7bb02b | 2013-09-15 14:54:56 +0200 | [diff] [blame] | 106 | * END_DEPENDENCIES |
| 107 | */ |
| 108 | |
Valerio Setti | 15cac17 | 2023-04-18 11:25:30 +0200 | [diff] [blame] | 109 | /* BEGIN_CASE */ |
Valerio Setti | c60bc5e | 2023-04-17 18:43:06 +0200 | [diff] [blame] | 110 | void pk_write_pubkey_check(char *key_file, int is_der) |
Paul Bakker | c7bb02b | 2013-09-15 14:54:56 +0200 | [diff] [blame] | 111 | { |
Valerio Setti | c60bc5e | 2023-04-17 18:43:06 +0200 | [diff] [blame] | 112 | pk_write_check_common(key_file, 1, is_der); |
Valerio Setti | 8959095 | 2023-04-17 17:34:09 +0200 | [diff] [blame] | 113 | goto exit; /* make the compiler happy */ |
Paul Bakker | c7bb02b | 2013-09-15 14:54:56 +0200 | [diff] [blame] | 114 | } |
| 115 | /* END_CASE */ |
| 116 | |
Valerio Setti | 15cac17 | 2023-04-18 11:25:30 +0200 | [diff] [blame] | 117 | /* BEGIN_CASE */ |
Valerio Setti | c60bc5e | 2023-04-17 18:43:06 +0200 | [diff] [blame] | 118 | void pk_write_key_check(char *key_file, int is_der) |
Paul Bakker | c7bb02b | 2013-09-15 14:54:56 +0200 | [diff] [blame] | 119 | { |
Valerio Setti | c60bc5e | 2023-04-17 18:43:06 +0200 | [diff] [blame] | 120 | pk_write_check_common(key_file, 0, is_der); |
Valerio Setti | 8959095 | 2023-04-17 17:34:09 +0200 | [diff] [blame] | 121 | goto exit; /* make the compiler happy */ |
Paul Bakker | c7bb02b | 2013-09-15 14:54:56 +0200 | [diff] [blame] | 122 | } |
| 123 | /* END_CASE */ |