Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 1 | /* |
julhal01 | c3f4e9a | 2020-12-15 13:39:01 +0000 | [diff] [blame] | 2 | * Copyright (c) 2020-2021, Arm Limited and Contributors. All rights reserved. |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 3 | * |
| 4 | * SPDX-License-Identifier: BSD-3-Clause |
| 5 | */ |
| 6 | #include <stdint.h> |
| 7 | #include <stdlib.h> |
| 8 | #include <protocols/service/crypto/packed-c/opcodes.h> |
Julian Hall | 9061e6c | 2021-06-29 14:24:20 +0100 | [diff] [blame] | 9 | #include <service/crypto/provider/crypto_provider.h> |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 10 | #include <protocols/rpc/common/packed-c/status.h> |
| 11 | #include <psa/crypto.h> |
| 12 | |
| 13 | /* Service request handlers */ |
| 14 | static rpc_status_t nop_handler(void *context, struct call_req* req); |
| 15 | static rpc_status_t generate_key_handler(void *context, struct call_req* req); |
| 16 | static rpc_status_t destroy_key_handler(void *context, struct call_req* req); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 17 | static rpc_status_t export_key_handler(void *context, struct call_req* req); |
| 18 | static rpc_status_t export_public_key_handler(void *context, struct call_req* req); |
| 19 | static rpc_status_t import_key_handler(void *context, struct call_req* req); |
| 20 | static rpc_status_t sign_hash_handler(void *context, struct call_req* req); |
| 21 | static rpc_status_t verify_hash_handler(void *context, struct call_req* req); |
| 22 | static rpc_status_t asymmetric_decrypt_handler(void *context, struct call_req* req); |
| 23 | static rpc_status_t asymmetric_encrypt_handler(void *context, struct call_req* req); |
| 24 | static rpc_status_t generate_random_handler(void *context, struct call_req* req); |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 25 | static rpc_status_t hash_setup_handler(void *context, struct call_req* req); |
| 26 | static rpc_status_t hash_update_handler(void *context, struct call_req* req); |
| 27 | static rpc_status_t hash_finish_handler(void *context, struct call_req* req); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 28 | |
| 29 | /* Handler mapping table for service */ |
| 30 | static const struct service_handler handler_table[] = { |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 31 | {TS_CRYPTO_OPCODE_NOP, nop_handler}, |
| 32 | {TS_CRYPTO_OPCODE_GENERATE_KEY, generate_key_handler}, |
| 33 | {TS_CRYPTO_OPCODE_DESTROY_KEY, destroy_key_handler}, |
| 34 | {TS_CRYPTO_OPCODE_EXPORT_KEY, export_key_handler}, |
| 35 | {TS_CRYPTO_OPCODE_EXPORT_PUBLIC_KEY, export_public_key_handler}, |
| 36 | {TS_CRYPTO_OPCODE_IMPORT_KEY, import_key_handler}, |
| 37 | {TS_CRYPTO_OPCODE_SIGN_HASH, sign_hash_handler}, |
| 38 | {TS_CRYPTO_OPCODE_VERIFY_HASH, verify_hash_handler}, |
| 39 | {TS_CRYPTO_OPCODE_ASYMMETRIC_DECRYPT, asymmetric_decrypt_handler}, |
| 40 | {TS_CRYPTO_OPCODE_ASYMMETRIC_ENCRYPT, asymmetric_encrypt_handler}, |
| 41 | {TS_CRYPTO_OPCODE_GENERATE_RANDOM, generate_random_handler}, |
| 42 | {TS_CRYPTO_OPCODE_HASH_SETUP, hash_setup_handler}, |
| 43 | {TS_CRYPTO_OPCODE_HASH_UPDATE, hash_update_handler}, |
| 44 | {TS_CRYPTO_OPCODE_HASH_FINISH, hash_finish_handler} |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 45 | }; |
| 46 | |
Julian Hall | 9061e6c | 2021-06-29 14:24:20 +0100 | [diff] [blame] | 47 | struct rpc_interface *crypto_provider_init(struct crypto_provider *context) |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 48 | { |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 49 | crypto_context_pool_init(&context->context_pool); |
julhal01 | ffa98d8 | 2021-01-20 13:51:58 +0000 | [diff] [blame] | 50 | |
Julian Hall | 9061e6c | 2021-06-29 14:24:20 +0100 | [diff] [blame] | 51 | for (size_t encoding = 0; encoding < TS_RPC_ENCODING_LIMIT; ++encoding) |
| 52 | context->serializers[encoding] = NULL; |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 53 | |
Julian Hall | 9061e6c | 2021-06-29 14:24:20 +0100 | [diff] [blame] | 54 | service_provider_init(&context->base_provider, context, |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 55 | handler_table, sizeof(handler_table)/sizeof(struct service_handler)); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 56 | |
Julian Hall | 9061e6c | 2021-06-29 14:24:20 +0100 | [diff] [blame] | 57 | return service_provider_get_rpc_interface(&context->base_provider); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 58 | } |
| 59 | |
Julian Hall | 9061e6c | 2021-06-29 14:24:20 +0100 | [diff] [blame] | 60 | void crypto_provider_deinit(struct crypto_provider *context) |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 61 | { |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 62 | crypto_context_pool_deinit(&context->context_pool); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 63 | } |
| 64 | |
Julian Hall | 9061e6c | 2021-06-29 14:24:20 +0100 | [diff] [blame] | 65 | void crypto_provider_register_serializer(struct crypto_provider *context, |
| 66 | unsigned int encoding, const struct crypto_provider_serializer *serializer) |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 67 | { |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 68 | if (encoding < TS_RPC_ENCODING_LIMIT) |
| 69 | context->serializers[encoding] = serializer; |
julhal01 | c3f4e9a | 2020-12-15 13:39:01 +0000 | [diff] [blame] | 70 | } |
| 71 | |
| 72 | static const struct crypto_provider_serializer* get_crypto_serializer(void *context, |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 73 | const struct call_req *req) |
julhal01 | c3f4e9a | 2020-12-15 13:39:01 +0000 | [diff] [blame] | 74 | { |
Julian Hall | 9061e6c | 2021-06-29 14:24:20 +0100 | [diff] [blame] | 75 | struct crypto_provider *this_instance = (struct crypto_provider*)context; |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 76 | const struct crypto_provider_serializer* serializer = NULL; |
| 77 | unsigned int encoding = call_req_get_encoding(req); |
julhal01 | c3f4e9a | 2020-12-15 13:39:01 +0000 | [diff] [blame] | 78 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 79 | if (encoding < TS_RPC_ENCODING_LIMIT) serializer = this_instance->serializers[encoding]; |
julhal01 | c3f4e9a | 2020-12-15 13:39:01 +0000 | [diff] [blame] | 80 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 81 | return serializer; |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 82 | } |
| 83 | |
| 84 | static rpc_status_t nop_handler(void *context, struct call_req* req) |
| 85 | { |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 86 | /* Responds to a request by returning success */ |
| 87 | rpc_status_t rpc_status = TS_RPC_CALL_ACCEPTED; |
| 88 | psa_status_t psa_status = PSA_SUCCESS; |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 89 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 90 | (void)context; |
| 91 | call_req_set_opstatus(req, psa_status); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 92 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 93 | return rpc_status; |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 94 | } |
| 95 | |
| 96 | static rpc_status_t generate_key_handler(void *context, struct call_req* req) |
| 97 | { |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 98 | rpc_status_t rpc_status = TS_RPC_ERROR_SERIALIZATION_NOT_SUPPORTED; |
| 99 | struct call_param_buf *req_buf = call_req_get_req_buf(req); |
| 100 | const struct crypto_provider_serializer *serializer = get_crypto_serializer(context, req); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 101 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 102 | psa_key_attributes_t attributes = PSA_KEY_ATTRIBUTES_INIT; |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 103 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 104 | if (serializer) |
| 105 | rpc_status = serializer->deserialize_generate_key_req(req_buf, &attributes); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 106 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 107 | if (rpc_status == TS_RPC_CALL_ACCEPTED) { |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 108 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 109 | psa_status_t psa_status; |
| 110 | psa_key_id_t id; |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 111 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 112 | psa_status = psa_generate_key(&attributes, &id); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 113 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 114 | if (psa_status == PSA_SUCCESS) { |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 115 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 116 | struct call_param_buf *resp_buf = call_req_get_resp_buf(req); |
| 117 | rpc_status = serializer->serialize_generate_key_resp(resp_buf, id); |
| 118 | } |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 119 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 120 | call_req_set_opstatus(req, psa_status); |
| 121 | } |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 122 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 123 | psa_reset_key_attributes(&attributes); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 124 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 125 | return rpc_status; |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 126 | } |
| 127 | |
| 128 | static rpc_status_t destroy_key_handler(void *context, struct call_req* req) |
| 129 | { |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 130 | rpc_status_t rpc_status = TS_RPC_ERROR_SERIALIZATION_NOT_SUPPORTED; |
| 131 | struct call_param_buf *req_buf = call_req_get_req_buf(req); |
| 132 | const struct crypto_provider_serializer *serializer = get_crypto_serializer(context, req); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 133 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 134 | psa_key_id_t id; |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 135 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 136 | if (serializer) |
| 137 | rpc_status = serializer->deserialize_destroy_key_req(req_buf, &id); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 138 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 139 | if (rpc_status == TS_RPC_CALL_ACCEPTED) { |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 140 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 141 | psa_status_t psa_status; |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 142 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 143 | psa_status = psa_destroy_key(id); |
| 144 | call_req_set_opstatus(req, psa_status); |
| 145 | } |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 146 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 147 | return rpc_status; |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 148 | } |
| 149 | |
| 150 | static rpc_status_t export_key_handler(void *context, struct call_req* req) |
| 151 | { |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 152 | rpc_status_t rpc_status = TS_RPC_ERROR_SERIALIZATION_NOT_SUPPORTED; |
| 153 | struct call_param_buf *req_buf = call_req_get_req_buf(req); |
| 154 | const struct crypto_provider_serializer *serializer = get_crypto_serializer(context, req); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 155 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 156 | psa_key_id_t id; |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 157 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 158 | if (serializer) |
| 159 | rpc_status = serializer->deserialize_export_key_req(req_buf, &id); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 160 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 161 | if (rpc_status == TS_RPC_CALL_ACCEPTED) { |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 162 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 163 | psa_status_t psa_status; |
| 164 | psa_key_attributes_t attributes = PSA_KEY_ATTRIBUTES_INIT; |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 165 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 166 | psa_status = psa_get_key_attributes(id, &attributes); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 167 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 168 | if (psa_status == PSA_SUCCESS) { |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 169 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 170 | size_t max_export_size = PSA_EXPORT_KEY_OUTPUT_SIZE( |
| 171 | psa_get_key_type(&attributes), |
| 172 | psa_get_key_bits(&attributes)); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 173 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 174 | uint8_t *key_buffer = malloc(max_export_size); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 175 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 176 | if (key_buffer) { |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 177 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 178 | size_t export_size; |
| 179 | psa_status = psa_export_key(id, key_buffer, max_export_size, &export_size); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 180 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 181 | if (psa_status == PSA_SUCCESS) { |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 182 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 183 | struct call_param_buf *resp_buf = call_req_get_resp_buf(req); |
| 184 | rpc_status = serializer->serialize_export_key_resp(resp_buf, key_buffer, export_size); |
| 185 | } |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 186 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 187 | free(key_buffer); |
| 188 | } |
| 189 | else { |
| 190 | /* Failed to allocate key buffer */ |
| 191 | rpc_status = TS_RPC_ERROR_RESOURCE_FAILURE; |
| 192 | } |
| 193 | } |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 194 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 195 | call_req_set_opstatus(req, psa_status); |
| 196 | psa_reset_key_attributes(&attributes); |
| 197 | } |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 198 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 199 | return rpc_status; |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 200 | } |
| 201 | |
| 202 | static rpc_status_t export_public_key_handler(void *context, struct call_req* req) |
| 203 | { |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 204 | rpc_status_t rpc_status = TS_RPC_ERROR_SERIALIZATION_NOT_SUPPORTED; |
| 205 | struct call_param_buf *req_buf = call_req_get_req_buf(req); |
| 206 | const struct crypto_provider_serializer *serializer = get_crypto_serializer(context, req); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 207 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 208 | psa_key_id_t id; |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 209 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 210 | if (serializer) |
| 211 | rpc_status = serializer->deserialize_export_public_key_req(req_buf, &id); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 212 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 213 | if (rpc_status == TS_RPC_CALL_ACCEPTED) { |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 214 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 215 | psa_status_t psa_status; |
| 216 | psa_key_attributes_t attributes = PSA_KEY_ATTRIBUTES_INIT; |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 217 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 218 | psa_status = psa_get_key_attributes(id, &attributes); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 219 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 220 | if (psa_status == PSA_SUCCESS) { |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 221 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 222 | size_t max_export_size = PSA_EXPORT_PUBLIC_KEY_OUTPUT_SIZE( |
| 223 | PSA_KEY_TYPE_PUBLIC_KEY_OF_KEY_PAIR(psa_get_key_type(&attributes)), |
| 224 | psa_get_key_bits(&attributes)); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 225 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 226 | uint8_t *key_buffer = malloc(max_export_size); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 227 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 228 | if (key_buffer) { |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 229 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 230 | size_t export_size; |
| 231 | psa_status = psa_export_public_key(id, key_buffer, max_export_size, &export_size); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 232 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 233 | if (psa_status == PSA_SUCCESS) { |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 234 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 235 | struct call_param_buf *resp_buf = call_req_get_resp_buf(req); |
| 236 | rpc_status = serializer->serialize_export_public_key_resp(resp_buf, key_buffer, export_size); |
| 237 | } |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 238 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 239 | free(key_buffer); |
| 240 | } |
| 241 | else { |
| 242 | /* Failed to allocate key buffer */ |
| 243 | rpc_status = TS_RPC_ERROR_RESOURCE_FAILURE; |
| 244 | } |
| 245 | } |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 246 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 247 | call_req_set_opstatus(req, psa_status); |
| 248 | psa_reset_key_attributes(&attributes); |
| 249 | } |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 250 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 251 | return rpc_status; |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 252 | } |
| 253 | |
| 254 | static rpc_status_t import_key_handler(void *context, struct call_req* req) |
| 255 | { |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 256 | rpc_status_t rpc_status = TS_RPC_ERROR_SERIALIZATION_NOT_SUPPORTED; |
| 257 | struct call_param_buf *req_buf = call_req_get_req_buf(req); |
| 258 | const struct crypto_provider_serializer *serializer = get_crypto_serializer(context, req); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 259 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 260 | if (serializer) { |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 261 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 262 | size_t key_data_len = serializer->max_deserialised_parameter_size(req_buf); |
| 263 | uint8_t *key_buffer = malloc(key_data_len); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 264 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 265 | if (key_buffer) { |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 266 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 267 | psa_key_attributes_t attributes = PSA_KEY_ATTRIBUTES_INIT; |
| 268 | rpc_status = serializer->deserialize_import_key_req(req_buf, &attributes, key_buffer, &key_data_len); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 269 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 270 | if (rpc_status == TS_RPC_CALL_ACCEPTED) { |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 271 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 272 | psa_status_t psa_status; |
| 273 | psa_key_id_t id; |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 274 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 275 | psa_status = psa_import_key(&attributes, key_buffer, key_data_len, &id); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 276 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 277 | if (psa_status == PSA_SUCCESS) { |
julhal01 | c3f4e9a | 2020-12-15 13:39:01 +0000 | [diff] [blame] | 278 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 279 | struct call_param_buf *resp_buf = call_req_get_resp_buf(req); |
| 280 | rpc_status = serializer->serialize_import_key_resp(resp_buf, id); |
| 281 | } |
julhal01 | c3f4e9a | 2020-12-15 13:39:01 +0000 | [diff] [blame] | 282 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 283 | call_req_set_opstatus(req, psa_status); |
| 284 | } |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 285 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 286 | psa_reset_key_attributes(&attributes); |
| 287 | free(key_buffer); |
| 288 | } |
| 289 | else { |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 290 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 291 | rpc_status = TS_RPC_ERROR_RESOURCE_FAILURE; |
| 292 | } |
| 293 | } |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 294 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 295 | return rpc_status; |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 296 | } |
| 297 | |
| 298 | static rpc_status_t sign_hash_handler(void *context, struct call_req* req) |
| 299 | { |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 300 | rpc_status_t rpc_status = TS_RPC_ERROR_SERIALIZATION_NOT_SUPPORTED; |
| 301 | struct call_param_buf *req_buf = call_req_get_req_buf(req); |
| 302 | const struct crypto_provider_serializer *serializer = get_crypto_serializer(context, req); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 303 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 304 | psa_key_id_t id; |
| 305 | psa_algorithm_t alg; |
| 306 | size_t hash_len = PSA_HASH_MAX_SIZE; |
| 307 | uint8_t hash_buffer[PSA_HASH_MAX_SIZE]; |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 308 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 309 | if (serializer) |
| 310 | rpc_status = serializer->deserialize_sign_hash_req(req_buf, &id, &alg, hash_buffer, &hash_len); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 311 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 312 | if (rpc_status == TS_RPC_CALL_ACCEPTED) { |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 313 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 314 | psa_status_t psa_status; |
| 315 | size_t sig_len; |
| 316 | uint8_t sig_buffer[PSA_SIGNATURE_MAX_SIZE]; |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 317 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 318 | psa_status = psa_sign_hash(id, alg, |
| 319 | hash_buffer, hash_len, |
| 320 | sig_buffer, sizeof(sig_buffer), &sig_len); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 321 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 322 | if (psa_status == PSA_SUCCESS) { |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 323 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 324 | struct call_param_buf *resp_buf = call_req_get_resp_buf(req); |
| 325 | rpc_status = serializer->serialize_sign_hash_resp(resp_buf, sig_buffer, sig_len); |
| 326 | } |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 327 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 328 | call_req_set_opstatus(req, psa_status); |
| 329 | } |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 330 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 331 | return rpc_status; |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 332 | } |
| 333 | |
| 334 | static rpc_status_t verify_hash_handler(void *context, struct call_req* req) |
| 335 | { |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 336 | rpc_status_t rpc_status = TS_RPC_ERROR_SERIALIZATION_NOT_SUPPORTED; |
| 337 | struct call_param_buf *req_buf = call_req_get_req_buf(req); |
| 338 | const struct crypto_provider_serializer *serializer = get_crypto_serializer(context, req); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 339 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 340 | psa_key_id_t id; |
| 341 | psa_algorithm_t alg; |
| 342 | size_t hash_len = PSA_HASH_MAX_SIZE; |
| 343 | uint8_t hash_buffer[PSA_HASH_MAX_SIZE]; |
| 344 | size_t sig_len = PSA_SIGNATURE_MAX_SIZE; |
| 345 | uint8_t sig_buffer[PSA_SIGNATURE_MAX_SIZE]; |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 346 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 347 | if (serializer) |
| 348 | rpc_status = serializer->deserialize_verify_hash_req(req_buf, &id, &alg, |
| 349 | hash_buffer, &hash_len, |
| 350 | sig_buffer, &sig_len); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 351 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 352 | if (rpc_status == TS_RPC_CALL_ACCEPTED) { |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 353 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 354 | psa_status_t psa_status; |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 355 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 356 | psa_status = psa_verify_hash(id, alg, |
| 357 | hash_buffer, hash_len, |
| 358 | sig_buffer, sig_len); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 359 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 360 | call_req_set_opstatus(req, psa_status); |
| 361 | } |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 362 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 363 | return rpc_status; |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 364 | } |
| 365 | |
| 366 | static rpc_status_t asymmetric_decrypt_handler(void *context, struct call_req* req) |
| 367 | { |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 368 | rpc_status_t rpc_status = TS_RPC_ERROR_SERIALIZATION_NOT_SUPPORTED; |
| 369 | struct call_param_buf *req_buf = call_req_get_req_buf(req); |
| 370 | const struct crypto_provider_serializer *serializer = get_crypto_serializer(context, req); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 371 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 372 | if (serializer) { |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 373 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 374 | size_t max_param_size = serializer->max_deserialised_parameter_size(req_buf); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 375 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 376 | psa_key_id_t id; |
| 377 | psa_algorithm_t alg; |
| 378 | size_t ciphertext_len = max_param_size; |
| 379 | uint8_t *ciphertext_buffer = malloc(ciphertext_len); |
| 380 | size_t salt_len = max_param_size; |
| 381 | uint8_t *salt_buffer = malloc(salt_len); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 382 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 383 | if (ciphertext_buffer && salt_buffer) { |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 384 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 385 | rpc_status = serializer->deserialize_asymmetric_decrypt_req(req_buf, |
| 386 | &id, &alg, |
| 387 | ciphertext_buffer, &ciphertext_len, |
| 388 | salt_buffer, &salt_len); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 389 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 390 | if (rpc_status == TS_RPC_CALL_ACCEPTED) { |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 391 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 392 | psa_status_t psa_status; |
| 393 | psa_key_attributes_t attributes = PSA_KEY_ATTRIBUTES_INIT; |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 394 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 395 | psa_status = psa_get_key_attributes(id, &attributes); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 396 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 397 | if (psa_status == PSA_SUCCESS) { |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 398 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 399 | size_t max_decrypt_size = PSA_ASYMMETRIC_DECRYPT_OUTPUT_SIZE( |
| 400 | psa_get_key_type(&attributes), |
| 401 | psa_get_key_bits(&attributes), |
| 402 | alg); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 403 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 404 | size_t plaintext_len; |
| 405 | uint8_t *plaintext_buffer = malloc(max_decrypt_size); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 406 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 407 | if (plaintext_buffer) { |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 408 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 409 | /* Salt is an optional parameter */ |
| 410 | uint8_t *salt = (salt_len) ? salt_buffer : NULL; |
Julian Hall | f688a0b | 2021-04-09 11:58:30 +0100 | [diff] [blame] | 411 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 412 | psa_status = psa_asymmetric_decrypt(id, alg, |
| 413 | ciphertext_buffer, ciphertext_len, |
| 414 | salt, salt_len, |
| 415 | plaintext_buffer, max_decrypt_size, &plaintext_len); |
julhal01 | c3f4e9a | 2020-12-15 13:39:01 +0000 | [diff] [blame] | 416 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 417 | if (psa_status == PSA_SUCCESS) { |
julhal01 | c3f4e9a | 2020-12-15 13:39:01 +0000 | [diff] [blame] | 418 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 419 | struct call_param_buf *resp_buf = call_req_get_resp_buf(req); |
| 420 | rpc_status = serializer->serialize_asymmetric_decrypt_resp(resp_buf, |
| 421 | plaintext_buffer, plaintext_len); |
| 422 | } |
julhal01 | c3f4e9a | 2020-12-15 13:39:01 +0000 | [diff] [blame] | 423 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 424 | free(plaintext_buffer); |
| 425 | } |
| 426 | else { |
| 427 | /* Failed to allocate ouptput buffer */ |
| 428 | rpc_status = TS_RPC_ERROR_RESOURCE_FAILURE; |
| 429 | } |
| 430 | } |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 431 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 432 | call_req_set_opstatus(req, psa_status); |
| 433 | psa_reset_key_attributes(&attributes); |
| 434 | } |
| 435 | } |
| 436 | else { |
| 437 | /* Failed to allocate buffers */ |
| 438 | rpc_status = TS_RPC_ERROR_RESOURCE_FAILURE; |
| 439 | } |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 440 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 441 | free(ciphertext_buffer); |
| 442 | free(salt_buffer); |
| 443 | } |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 444 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 445 | return rpc_status; |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 446 | } |
| 447 | |
| 448 | static rpc_status_t asymmetric_encrypt_handler(void *context, struct call_req* req) |
| 449 | { |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 450 | rpc_status_t rpc_status = TS_RPC_ERROR_SERIALIZATION_NOT_SUPPORTED; |
| 451 | struct call_param_buf *req_buf = call_req_get_req_buf(req); |
| 452 | const struct crypto_provider_serializer *serializer = get_crypto_serializer(context, req); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 453 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 454 | if (serializer) { |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 455 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 456 | size_t max_param_size = serializer->max_deserialised_parameter_size(req_buf); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 457 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 458 | psa_key_id_t id; |
| 459 | psa_algorithm_t alg; |
| 460 | size_t plaintext_len = max_param_size; |
| 461 | uint8_t *plaintext_buffer = malloc(plaintext_len); |
| 462 | size_t salt_len = max_param_size; |
| 463 | uint8_t *salt_buffer = malloc(salt_len); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 464 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 465 | if (plaintext_buffer && salt_buffer) { |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 466 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 467 | rpc_status = serializer->deserialize_asymmetric_encrypt_req(req_buf, |
| 468 | &id, &alg, |
| 469 | plaintext_buffer, &plaintext_len, |
| 470 | salt_buffer, &salt_len); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 471 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 472 | if (rpc_status == TS_RPC_CALL_ACCEPTED) { |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 473 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 474 | psa_status_t psa_status; |
| 475 | psa_key_attributes_t attributes = PSA_KEY_ATTRIBUTES_INIT; |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 476 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 477 | psa_status = psa_get_key_attributes(id, &attributes); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 478 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 479 | if (psa_status == PSA_SUCCESS) { |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 480 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 481 | size_t max_encrypt_size = PSA_ASYMMETRIC_ENCRYPT_OUTPUT_SIZE( |
| 482 | psa_get_key_type(&attributes), |
| 483 | psa_get_key_bits(&attributes), |
| 484 | alg); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 485 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 486 | size_t ciphertext_len; |
| 487 | uint8_t *ciphertext_buffer = malloc(max_encrypt_size); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 488 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 489 | if (ciphertext_buffer) { |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 490 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 491 | /* Salt is an optional parameter */ |
| 492 | uint8_t *salt = (salt_len) ? salt_buffer : NULL; |
Julian Hall | f688a0b | 2021-04-09 11:58:30 +0100 | [diff] [blame] | 493 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 494 | psa_status = psa_asymmetric_encrypt(id, alg, |
| 495 | plaintext_buffer, plaintext_len, |
| 496 | salt, salt_len, |
| 497 | ciphertext_buffer, max_encrypt_size, &ciphertext_len); |
julhal01 | c3f4e9a | 2020-12-15 13:39:01 +0000 | [diff] [blame] | 498 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 499 | if (psa_status == PSA_SUCCESS) { |
julhal01 | c3f4e9a | 2020-12-15 13:39:01 +0000 | [diff] [blame] | 500 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 501 | struct call_param_buf *resp_buf = call_req_get_resp_buf(req); |
| 502 | rpc_status = serializer->serialize_asymmetric_encrypt_resp(resp_buf, |
| 503 | ciphertext_buffer, ciphertext_len); |
| 504 | } |
julhal01 | c3f4e9a | 2020-12-15 13:39:01 +0000 | [diff] [blame] | 505 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 506 | free(ciphertext_buffer); |
| 507 | } |
| 508 | else { |
| 509 | /* Failed to allocate ouptput buffer */ |
| 510 | rpc_status = TS_RPC_ERROR_RESOURCE_FAILURE; |
| 511 | } |
| 512 | } |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 513 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 514 | call_req_set_opstatus(req, psa_status); |
| 515 | psa_reset_key_attributes(&attributes); |
| 516 | } |
| 517 | } |
| 518 | else { |
| 519 | /* Failed to allocate buffers */ |
| 520 | rpc_status = TS_RPC_ERROR_RESOURCE_FAILURE; |
| 521 | } |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 522 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 523 | free(plaintext_buffer); |
| 524 | free(salt_buffer); |
| 525 | } |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 526 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 527 | return rpc_status; |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 528 | } |
| 529 | |
| 530 | static rpc_status_t generate_random_handler(void *context, struct call_req* req) |
| 531 | { |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 532 | rpc_status_t rpc_status = TS_RPC_ERROR_SERIALIZATION_NOT_SUPPORTED; |
| 533 | struct call_param_buf *req_buf = call_req_get_req_buf(req); |
| 534 | const struct crypto_provider_serializer *serializer = get_crypto_serializer(context, req); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 535 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 536 | size_t output_size; |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 537 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 538 | if (serializer) |
| 539 | rpc_status = serializer->deserialize_generate_random_req(req_buf, &output_size); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 540 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 541 | if (rpc_status == TS_RPC_CALL_ACCEPTED) { |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 542 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 543 | psa_status_t psa_status; |
| 544 | uint8_t *output_buffer = malloc(output_size); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 545 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 546 | if (output_buffer) { |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 547 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 548 | psa_status = psa_generate_random(output_buffer, output_size); |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 549 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 550 | if (psa_status == PSA_SUCCESS) { |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 551 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 552 | struct call_param_buf *resp_buf = call_req_get_resp_buf(req); |
| 553 | rpc_status = serializer->serialize_generate_random_resp(resp_buf, |
| 554 | output_buffer, output_size); |
| 555 | } |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 556 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 557 | call_req_set_opstatus(req, psa_status); |
| 558 | free(output_buffer); |
| 559 | } |
| 560 | else { |
| 561 | /* Failed to allocate output buffer */ |
| 562 | rpc_status = TS_RPC_ERROR_RESOURCE_FAILURE; |
| 563 | } |
| 564 | } |
Julian Hall | c02fffb | 2020-11-23 18:22:06 +0100 | [diff] [blame] | 565 | |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 566 | return rpc_status; |
| 567 | } |
| 568 | |
| 569 | static rpc_status_t hash_setup_handler(void *context, struct call_req* req) |
| 570 | { |
| 571 | rpc_status_t rpc_status = TS_RPC_ERROR_SERIALIZATION_NOT_SUPPORTED; |
| 572 | struct call_param_buf *req_buf = call_req_get_req_buf(req); |
| 573 | const struct crypto_provider_serializer *serializer = get_crypto_serializer(context, req); |
Julian Hall | 9061e6c | 2021-06-29 14:24:20 +0100 | [diff] [blame] | 574 | struct crypto_provider *this_instance = (struct crypto_provider*)context; |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 575 | |
| 576 | psa_algorithm_t alg; |
| 577 | |
| 578 | if (serializer) |
| 579 | rpc_status = serializer->deserialize_hash_setup_req(req_buf, &alg); |
| 580 | |
| 581 | if (rpc_status == TS_RPC_CALL_ACCEPTED) { |
| 582 | |
| 583 | uint32_t op_handle; |
| 584 | |
| 585 | struct crypto_context *crypto_context = |
| 586 | crypto_context_pool_alloc(&this_instance->context_pool, |
| 587 | CRYPTO_CONTEXT_OP_ID_HASH, call_req_get_caller_id(req), |
| 588 | &op_handle); |
| 589 | |
| 590 | if (crypto_context) { |
| 591 | |
| 592 | psa_status_t psa_status; |
| 593 | |
| 594 | crypto_context->op.hash = psa_hash_operation_init(); |
| 595 | psa_status = psa_hash_setup(&crypto_context->op.hash, alg); |
| 596 | |
| 597 | if (psa_status == PSA_SUCCESS) { |
| 598 | |
| 599 | struct call_param_buf *resp_buf = call_req_get_resp_buf(req); |
| 600 | rpc_status = serializer->serialize_hash_setup_resp(resp_buf, op_handle); |
| 601 | } |
| 602 | |
| 603 | if ((psa_status != PSA_SUCCESS) || (rpc_status != TS_RPC_CALL_ACCEPTED)) { |
| 604 | |
| 605 | crypto_context_pool_free(&this_instance->context_pool, crypto_context); |
| 606 | } |
| 607 | |
| 608 | call_req_set_opstatus(req, psa_status); |
| 609 | } |
| 610 | else { |
| 611 | /* Failed to allocate crypto context for transaction */ |
| 612 | rpc_status = TS_RPC_ERROR_RESOURCE_FAILURE; |
| 613 | } |
| 614 | } |
| 615 | |
| 616 | return rpc_status; |
| 617 | } |
| 618 | |
| 619 | static rpc_status_t hash_update_handler(void *context, struct call_req* req) |
| 620 | { |
| 621 | rpc_status_t rpc_status = TS_RPC_ERROR_SERIALIZATION_NOT_SUPPORTED; |
| 622 | struct call_param_buf *req_buf = call_req_get_req_buf(req); |
| 623 | const struct crypto_provider_serializer *serializer = get_crypto_serializer(context, req); |
Julian Hall | 9061e6c | 2021-06-29 14:24:20 +0100 | [diff] [blame] | 624 | struct crypto_provider *this_instance = (struct crypto_provider*)context; |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 625 | |
| 626 | uint32_t op_handle; |
| 627 | const uint8_t *data; |
| 628 | size_t data_len; |
| 629 | |
| 630 | if (serializer) |
| 631 | rpc_status = serializer->deserialize_hash_update_req(req_buf, &op_handle, &data, &data_len); |
| 632 | |
| 633 | if (rpc_status == TS_RPC_CALL_ACCEPTED) { |
| 634 | |
| 635 | struct crypto_context *crypto_context = |
| 636 | crypto_context_pool_find(&this_instance->context_pool, |
| 637 | CRYPTO_CONTEXT_OP_ID_HASH, call_req_get_caller_id(req), |
| 638 | op_handle); |
| 639 | |
| 640 | if (crypto_context) { |
| 641 | |
| 642 | psa_status_t psa_status = psa_hash_update(&crypto_context->op.hash, data, data_len); |
| 643 | call_req_set_opstatus(req, psa_status); |
| 644 | } |
| 645 | else { |
| 646 | /* Requested context doesn't exist */ |
| 647 | rpc_status = TS_RPC_ERROR_RESOURCE_FAILURE; |
| 648 | } |
| 649 | } |
| 650 | |
| 651 | return rpc_status; |
| 652 | } |
| 653 | |
| 654 | static rpc_status_t hash_finish_handler(void *context, struct call_req* req) |
| 655 | { |
| 656 | rpc_status_t rpc_status = TS_RPC_ERROR_SERIALIZATION_NOT_SUPPORTED; |
| 657 | struct call_param_buf *req_buf = call_req_get_req_buf(req); |
| 658 | const struct crypto_provider_serializer *serializer = get_crypto_serializer(context, req); |
Julian Hall | 9061e6c | 2021-06-29 14:24:20 +0100 | [diff] [blame] | 659 | struct crypto_provider *this_instance = (struct crypto_provider*)context; |
Julian Hall | f572896 | 2021-06-24 09:40:23 +0100 | [diff] [blame] | 660 | |
| 661 | uint32_t op_handle; |
| 662 | |
| 663 | if (serializer) |
| 664 | rpc_status = serializer->deserialize_hash_finish_req(req_buf, &op_handle); |
| 665 | |
| 666 | if (rpc_status == TS_RPC_CALL_ACCEPTED) { |
| 667 | |
| 668 | struct crypto_context *crypto_context = |
| 669 | crypto_context_pool_find(&this_instance->context_pool, |
| 670 | CRYPTO_CONTEXT_OP_ID_HASH, call_req_get_caller_id(req), |
| 671 | op_handle); |
| 672 | |
| 673 | if (crypto_context) { |
| 674 | |
| 675 | psa_status_t psa_status; |
| 676 | size_t hash_len; |
| 677 | uint8_t hash[PSA_HASH_MAX_SIZE]; |
| 678 | |
| 679 | psa_status = psa_hash_finish(&crypto_context->op.hash, hash, sizeof(hash), &hash_len); |
| 680 | |
| 681 | if (psa_status == PSA_SUCCESS) { |
| 682 | |
| 683 | struct call_param_buf *resp_buf = call_req_get_resp_buf(req); |
| 684 | rpc_status = serializer->serialize_hash_finish_resp(resp_buf, hash, hash_len); |
| 685 | } |
| 686 | |
| 687 | crypto_context_pool_free(&this_instance->context_pool, crypto_context); |
| 688 | |
| 689 | call_req_set_opstatus(req, psa_status); |
| 690 | } |
| 691 | else { |
| 692 | /* Requested context doesn't exist */ |
| 693 | rpc_status = TS_RPC_ERROR_RESOURCE_FAILURE; |
| 694 | } |
| 695 | } |
| 696 | |
| 697 | return rpc_status; |
julhal01 | c3f4e9a | 2020-12-15 13:39:01 +0000 | [diff] [blame] | 698 | } |