Juan Castillo | 11abdcd | 2014-10-21 11:30:42 +0100 | [diff] [blame] | 1 | /* |
| 2 | * Copyright (c) 2015, ARM Limited and Contributors. All rights reserved. |
| 3 | * |
dp-arm | fa3cf0b | 2017-05-03 09:38:09 +0100 | [diff] [blame] | 4 | * SPDX-License-Identifier: BSD-3-Clause |
Juan Castillo | 11abdcd | 2014-10-21 11:30:42 +0100 | [diff] [blame] | 5 | */ |
| 6 | |
| 7 | #include <getopt.h> |
| 8 | #include <stdio.h> |
| 9 | #include <stdlib.h> |
| 10 | #include <string.h> |
| 11 | |
| 12 | #include <openssl/conf.h> |
| 13 | #include <openssl/evp.h> |
| 14 | #include <openssl/pem.h> |
| 15 | |
| 16 | #include "cert.h" |
Juan Castillo | 1218dd5 | 2015-07-03 16:23:16 +0100 | [diff] [blame] | 17 | #include "cmd_opt.h" |
Juan Castillo | 11abdcd | 2014-10-21 11:30:42 +0100 | [diff] [blame] | 18 | #include "debug.h" |
| 19 | #include "key.h" |
| 20 | #include "platform_oid.h" |
| 21 | #include "sha.h" |
| 22 | |
| 23 | #define MAX_FILENAME_LEN 1024 |
| 24 | |
| 25 | /* |
Juan Castillo | f9f39c3 | 2015-06-01 16:34:23 +0100 | [diff] [blame] | 26 | * Create a new key container |
Juan Castillo | 11abdcd | 2014-10-21 11:30:42 +0100 | [diff] [blame] | 27 | */ |
Masahiro Yamada | bccb109 | 2017-02-06 21:15:01 +0900 | [diff] [blame] | 28 | int key_new(key_t *key) |
Juan Castillo | 11abdcd | 2014-10-21 11:30:42 +0100 | [diff] [blame] | 29 | { |
Juan Castillo | 11abdcd | 2014-10-21 11:30:42 +0100 | [diff] [blame] | 30 | /* Create key pair container */ |
Juan Castillo | f9f39c3 | 2015-06-01 16:34:23 +0100 | [diff] [blame] | 31 | key->key = EVP_PKEY_new(); |
| 32 | if (key->key == NULL) { |
Juan Castillo | 11abdcd | 2014-10-21 11:30:42 +0100 | [diff] [blame] | 33 | return 0; |
| 34 | } |
| 35 | |
Juan Castillo | f9f39c3 | 2015-06-01 16:34:23 +0100 | [diff] [blame] | 36 | return 1; |
| 37 | } |
| 38 | |
Juan Castillo | a2224ab | 2015-06-30 13:36:57 +0100 | [diff] [blame] | 39 | static int key_create_rsa(key_t *key) |
Juan Castillo | f9f39c3 | 2015-06-01 16:34:23 +0100 | [diff] [blame] | 40 | { |
Masahiro Yamada | 48cb5e5 | 2017-02-06 19:47:44 +0900 | [diff] [blame] | 41 | RSA *rsa; |
Juan Castillo | f9f39c3 | 2015-06-01 16:34:23 +0100 | [diff] [blame] | 42 | |
Juan Castillo | a2224ab | 2015-06-30 13:36:57 +0100 | [diff] [blame] | 43 | rsa = RSA_generate_key(RSA_KEY_BITS, RSA_F4, NULL, NULL); |
| 44 | if (rsa == NULL) { |
| 45 | printf("Cannot create RSA key\n"); |
Juan Castillo | f9f39c3 | 2015-06-01 16:34:23 +0100 | [diff] [blame] | 46 | goto err; |
| 47 | } |
Juan Castillo | a2224ab | 2015-06-30 13:36:57 +0100 | [diff] [blame] | 48 | if (!EVP_PKEY_assign_RSA(key->key, rsa)) { |
| 49 | printf("Cannot assign RSA key\n"); |
Juan Castillo | f9f39c3 | 2015-06-01 16:34:23 +0100 | [diff] [blame] | 50 | goto err; |
Juan Castillo | 11abdcd | 2014-10-21 11:30:42 +0100 | [diff] [blame] | 51 | } |
| 52 | |
Juan Castillo | f9f39c3 | 2015-06-01 16:34:23 +0100 | [diff] [blame] | 53 | return 1; |
Juan Castillo | f9f39c3 | 2015-06-01 16:34:23 +0100 | [diff] [blame] | 54 | err: |
| 55 | RSA_free(rsa); |
Juan Castillo | a2224ab | 2015-06-30 13:36:57 +0100 | [diff] [blame] | 56 | return 0; |
| 57 | } |
| 58 | |
| 59 | #ifndef OPENSSL_NO_EC |
| 60 | static int key_create_ecdsa(key_t *key) |
| 61 | { |
Masahiro Yamada | 48cb5e5 | 2017-02-06 19:47:44 +0900 | [diff] [blame] | 62 | EC_KEY *ec; |
Juan Castillo | a2224ab | 2015-06-30 13:36:57 +0100 | [diff] [blame] | 63 | |
| 64 | ec = EC_KEY_new_by_curve_name(NID_X9_62_prime256v1); |
| 65 | if (ec == NULL) { |
| 66 | printf("Cannot create EC key\n"); |
| 67 | goto err; |
| 68 | } |
| 69 | if (!EC_KEY_generate_key(ec)) { |
| 70 | printf("Cannot generate EC key\n"); |
| 71 | goto err; |
| 72 | } |
| 73 | EC_KEY_set_flags(ec, EC_PKEY_NO_PARAMETERS); |
| 74 | EC_KEY_set_asn1_flag(ec, OPENSSL_EC_NAMED_CURVE); |
| 75 | if (!EVP_PKEY_assign_EC_KEY(key->key, ec)) { |
| 76 | printf("Cannot assign EC key\n"); |
| 77 | goto err; |
| 78 | } |
| 79 | |
| 80 | return 1; |
| 81 | err: |
Juan Castillo | f9f39c3 | 2015-06-01 16:34:23 +0100 | [diff] [blame] | 82 | EC_KEY_free(ec); |
Juan Castillo | a2224ab | 2015-06-30 13:36:57 +0100 | [diff] [blame] | 83 | return 0; |
| 84 | } |
| 85 | #endif /* OPENSSL_NO_EC */ |
| 86 | |
| 87 | typedef int (*key_create_fn_t)(key_t *key); |
| 88 | static const key_create_fn_t key_create_fn[KEY_ALG_MAX_NUM] = { |
| 89 | key_create_rsa, |
| 90 | #ifndef OPENSSL_NO_EC |
| 91 | key_create_ecdsa, |
| 92 | #endif /* OPENSSL_NO_EC */ |
| 93 | }; |
| 94 | |
| 95 | int key_create(key_t *key, int type) |
| 96 | { |
| 97 | if (type >= KEY_ALG_MAX_NUM) { |
| 98 | printf("Invalid key type\n"); |
| 99 | return 0; |
| 100 | } |
| 101 | |
Juan Castillo | a2224ab | 2015-06-30 13:36:57 +0100 | [diff] [blame] | 102 | if (key_create_fn[type]) { |
| 103 | return key_create_fn[type](key); |
| 104 | } |
Juan Castillo | f9f39c3 | 2015-06-01 16:34:23 +0100 | [diff] [blame] | 105 | |
Juan Castillo | 11abdcd | 2014-10-21 11:30:42 +0100 | [diff] [blame] | 106 | return 0; |
| 107 | } |
| 108 | |
Juan Castillo | f9f39c3 | 2015-06-01 16:34:23 +0100 | [diff] [blame] | 109 | int key_load(key_t *key, unsigned int *err_code) |
Juan Castillo | 11abdcd | 2014-10-21 11:30:42 +0100 | [diff] [blame] | 110 | { |
Masahiro Yamada | 48cb5e5 | 2017-02-06 19:47:44 +0900 | [diff] [blame] | 111 | FILE *fp; |
| 112 | EVP_PKEY *k; |
Juan Castillo | 11abdcd | 2014-10-21 11:30:42 +0100 | [diff] [blame] | 113 | |
Juan Castillo | 11abdcd | 2014-10-21 11:30:42 +0100 | [diff] [blame] | 114 | if (key->fn) { |
| 115 | /* Load key from file */ |
| 116 | fp = fopen(key->fn, "r"); |
| 117 | if (fp) { |
Juan Castillo | f9f39c3 | 2015-06-01 16:34:23 +0100 | [diff] [blame] | 118 | k = PEM_read_PrivateKey(fp, &key->key, NULL, NULL); |
Juan Castillo | 11abdcd | 2014-10-21 11:30:42 +0100 | [diff] [blame] | 119 | fclose(fp); |
| 120 | if (k) { |
Juan Castillo | f9f39c3 | 2015-06-01 16:34:23 +0100 | [diff] [blame] | 121 | *err_code = KEY_ERR_NONE; |
Juan Castillo | 11abdcd | 2014-10-21 11:30:42 +0100 | [diff] [blame] | 122 | return 1; |
| 123 | } else { |
Juan Castillo | f9f39c3 | 2015-06-01 16:34:23 +0100 | [diff] [blame] | 124 | ERROR("Cannot load key from %s\n", key->fn); |
| 125 | *err_code = KEY_ERR_LOAD; |
Juan Castillo | 11abdcd | 2014-10-21 11:30:42 +0100 | [diff] [blame] | 126 | } |
| 127 | } else { |
Juan Castillo | f9f39c3 | 2015-06-01 16:34:23 +0100 | [diff] [blame] | 128 | WARN("Cannot open file %s\n", key->fn); |
| 129 | *err_code = KEY_ERR_OPEN; |
Juan Castillo | 11abdcd | 2014-10-21 11:30:42 +0100 | [diff] [blame] | 130 | } |
| 131 | } else { |
Juan Castillo | f9f39c3 | 2015-06-01 16:34:23 +0100 | [diff] [blame] | 132 | WARN("Key filename not specified\n"); |
| 133 | *err_code = KEY_ERR_FILENAME; |
Juan Castillo | 11abdcd | 2014-10-21 11:30:42 +0100 | [diff] [blame] | 134 | } |
| 135 | |
Juan Castillo | 11abdcd | 2014-10-21 11:30:42 +0100 | [diff] [blame] | 136 | return 0; |
| 137 | } |
| 138 | |
| 139 | int key_store(key_t *key) |
| 140 | { |
Masahiro Yamada | 48cb5e5 | 2017-02-06 19:47:44 +0900 | [diff] [blame] | 141 | FILE *fp; |
Juan Castillo | 11abdcd | 2014-10-21 11:30:42 +0100 | [diff] [blame] | 142 | |
| 143 | if (key->fn) { |
| 144 | fp = fopen(key->fn, "w"); |
| 145 | if (fp) { |
| 146 | PEM_write_PrivateKey(fp, key->key, |
| 147 | NULL, NULL, 0, NULL, NULL); |
| 148 | fclose(fp); |
| 149 | return 1; |
| 150 | } else { |
| 151 | ERROR("Cannot create file %s\n", key->fn); |
| 152 | } |
| 153 | } else { |
| 154 | ERROR("Key filename not specified\n"); |
| 155 | } |
| 156 | |
| 157 | return 0; |
| 158 | } |
Juan Castillo | 1218dd5 | 2015-07-03 16:23:16 +0100 | [diff] [blame] | 159 | |
| 160 | int key_init(void) |
| 161 | { |
Juan Castillo | 212f738 | 2015-12-15 16:37:57 +0000 | [diff] [blame] | 162 | cmd_opt_t cmd_opt; |
Juan Castillo | 1218dd5 | 2015-07-03 16:23:16 +0100 | [diff] [blame] | 163 | key_t *key; |
Juan Castillo | 1218dd5 | 2015-07-03 16:23:16 +0100 | [diff] [blame] | 164 | unsigned int i; |
| 165 | |
| 166 | for (i = 0; i < num_keys; i++) { |
| 167 | key = &keys[i]; |
| 168 | if (key->opt != NULL) { |
Juan Castillo | 212f738 | 2015-12-15 16:37:57 +0000 | [diff] [blame] | 169 | cmd_opt.long_opt.name = key->opt; |
| 170 | cmd_opt.long_opt.has_arg = required_argument; |
| 171 | cmd_opt.long_opt.flag = NULL; |
| 172 | cmd_opt.long_opt.val = CMD_OPT_KEY; |
| 173 | cmd_opt.help_msg = key->help_msg; |
| 174 | cmd_opt_add(&cmd_opt); |
Juan Castillo | 1218dd5 | 2015-07-03 16:23:16 +0100 | [diff] [blame] | 175 | } |
| 176 | } |
| 177 | |
Masahiro Yamada | 48cb5e5 | 2017-02-06 19:47:44 +0900 | [diff] [blame] | 178 | return 0; |
Juan Castillo | 1218dd5 | 2015-07-03 16:23:16 +0100 | [diff] [blame] | 179 | } |
| 180 | |
| 181 | key_t *key_get_by_opt(const char *opt) |
| 182 | { |
Masahiro Yamada | 48cb5e5 | 2017-02-06 19:47:44 +0900 | [diff] [blame] | 183 | key_t *key; |
Juan Castillo | 1218dd5 | 2015-07-03 16:23:16 +0100 | [diff] [blame] | 184 | unsigned int i; |
| 185 | |
| 186 | /* Sequential search. This is not a performance concern since the number |
| 187 | * of keys is bounded and the code runs on a host machine */ |
| 188 | for (i = 0; i < num_keys; i++) { |
| 189 | key = &keys[i]; |
| 190 | if (0 == strcmp(key->opt, opt)) { |
| 191 | return key; |
| 192 | } |
| 193 | } |
| 194 | |
| 195 | return NULL; |
| 196 | } |