Tamas Ban | 48a0eb5 | 2018-08-17 12:48:05 +0100 | [diff] [blame] | 1 | /* |
Xinyu Zhang | ade2e0a | 2021-03-18 16:20:54 +0800 | [diff] [blame^] | 2 | * Copyright (c) 2018-2021, Arm Limited. All rights reserved. |
Tamas Ban | 48a0eb5 | 2018-08-17 12:48:05 +0100 | [diff] [blame] | 3 | * |
| 4 | * SPDX-License-Identifier: BSD-3-Clause |
| 5 | * |
| 6 | */ |
| 7 | |
Jamie Fox | cc31d40 | 2019-01-28 17:13:52 +0000 | [diff] [blame] | 8 | #include "psa/initial_attestation.h" |
Antonio de Angelis | 05b2419 | 2019-07-04 15:28:46 +0100 | [diff] [blame] | 9 | #include "tfm_ns_interface.h" |
Jamie Fox | cc31d40 | 2019-01-28 17:13:52 +0000 | [diff] [blame] | 10 | #include "psa/client.h" |
David Vincze | ff6da53 | 2019-11-21 00:19:50 +0100 | [diff] [blame] | 11 | #include "psa/crypto_types.h" |
Edison Ai | 870abb4 | 2019-06-21 11:14:08 +0800 | [diff] [blame] | 12 | #include "psa_manifest/sid.h" |
Kevin Peng | 2ed3022 | 2019-04-30 09:26:11 +0800 | [diff] [blame] | 13 | |
Raef Coles | 793574c | 2019-10-09 10:59:42 +0100 | [diff] [blame] | 14 | psa_status_t |
Raef Coles | 70a02da | 2019-10-09 11:32:04 +0100 | [diff] [blame] | 15 | psa_initial_attest_get_token(const uint8_t *auth_challenge, |
| 16 | size_t challenge_size, |
| 17 | uint8_t *token_buf, |
| 18 | size_t token_buf_size, |
| 19 | size_t *token_size) |
Tamas Ban | 48a0eb5 | 2018-08-17 12:48:05 +0100 | [diff] [blame] | 20 | { |
Kevin Peng | 2ed3022 | 2019-04-30 09:26:11 +0800 | [diff] [blame] | 21 | psa_handle_t handle = PSA_NULL_HANDLE; |
| 22 | psa_status_t status; |
Kevin Peng | 9449a36 | 2019-07-29 16:05:42 +0800 | [diff] [blame] | 23 | |
Kevin Peng | 2ed3022 | 2019-04-30 09:26:11 +0800 | [diff] [blame] | 24 | psa_invec in_vec[] = { |
Raef Coles | 70a02da | 2019-10-09 11:32:04 +0100 | [diff] [blame] | 25 | {auth_challenge, challenge_size} |
Kevin Peng | 2ed3022 | 2019-04-30 09:26:11 +0800 | [diff] [blame] | 26 | }; |
| 27 | psa_outvec out_vec[] = { |
Raef Coles | 70a02da | 2019-10-09 11:32:04 +0100 | [diff] [blame] | 28 | {token_buf, token_buf_size} |
Kevin Peng | 2ed3022 | 2019-04-30 09:26:11 +0800 | [diff] [blame] | 29 | }; |
Tamas Ban | 48a0eb5 | 2018-08-17 12:48:05 +0100 | [diff] [blame] | 30 | |
Kevin Peng | 2ed3022 | 2019-04-30 09:26:11 +0800 | [diff] [blame] | 31 | handle = psa_connect(TFM_ATTEST_GET_TOKEN_SID, |
Edison Ai | 870abb4 | 2019-06-21 11:14:08 +0800 | [diff] [blame] | 32 | TFM_ATTEST_GET_TOKEN_VERSION); |
Raef Coles | 793574c | 2019-10-09 10:59:42 +0100 | [diff] [blame] | 33 | if (!PSA_HANDLE_IS_VALID(handle)) { |
| 34 | return PSA_HANDLE_TO_ERROR(handle); |
Kevin Peng | 2ed3022 | 2019-04-30 09:26:11 +0800 | [diff] [blame] | 35 | } |
Tamas Ban | 48a0eb5 | 2018-08-17 12:48:05 +0100 | [diff] [blame] | 36 | |
Summer Qin | 4b1d03b | 2019-07-02 14:56:08 +0800 | [diff] [blame] | 37 | status = psa_call(handle, PSA_IPC_CALL, |
Kevin Peng | 2ed3022 | 2019-04-30 09:26:11 +0800 | [diff] [blame] | 38 | in_vec, IOVEC_LEN(in_vec), |
| 39 | out_vec, IOVEC_LEN(out_vec)); |
| 40 | psa_close(handle); |
Tamas Ban | 48a0eb5 | 2018-08-17 12:48:05 +0100 | [diff] [blame] | 41 | |
Kevin Peng | 2ed3022 | 2019-04-30 09:26:11 +0800 | [diff] [blame] | 42 | if (status == PSA_SUCCESS) { |
| 43 | *token_size = out_vec[0].len; |
| 44 | } |
| 45 | |
Raef Coles | 793574c | 2019-10-09 10:59:42 +0100 | [diff] [blame] | 46 | return status; |
Tamas Ban | b6b8056 | 2019-01-04 22:49:24 +0000 | [diff] [blame] | 47 | } |
| 48 | |
Raef Coles | 793574c | 2019-10-09 10:59:42 +0100 | [diff] [blame] | 49 | psa_status_t |
Raef Coles | 70a02da | 2019-10-09 11:32:04 +0100 | [diff] [blame] | 50 | psa_initial_attest_get_token_size(size_t challenge_size, |
| 51 | size_t *token_size) |
Tamas Ban | b6b8056 | 2019-01-04 22:49:24 +0000 | [diff] [blame] | 52 | { |
Kevin Peng | 2ed3022 | 2019-04-30 09:26:11 +0800 | [diff] [blame] | 53 | psa_handle_t handle = PSA_NULL_HANDLE; |
| 54 | psa_status_t status; |
Kevin Peng | 2ed3022 | 2019-04-30 09:26:11 +0800 | [diff] [blame] | 55 | psa_invec in_vec[] = { |
| 56 | {&challenge_size, sizeof(challenge_size)} |
| 57 | }; |
| 58 | psa_outvec out_vec[] = { |
Raef Coles | 70a02da | 2019-10-09 11:32:04 +0100 | [diff] [blame] | 59 | {token_size, sizeof(size_t)} |
Kevin Peng | 2ed3022 | 2019-04-30 09:26:11 +0800 | [diff] [blame] | 60 | }; |
Tamas Ban | b6b8056 | 2019-01-04 22:49:24 +0000 | [diff] [blame] | 61 | |
Kevin Peng | 2ed3022 | 2019-04-30 09:26:11 +0800 | [diff] [blame] | 62 | handle = psa_connect(TFM_ATTEST_GET_TOKEN_SIZE_SID, |
Edison Ai | 870abb4 | 2019-06-21 11:14:08 +0800 | [diff] [blame] | 63 | TFM_ATTEST_GET_TOKEN_SIZE_VERSION); |
Raef Coles | 793574c | 2019-10-09 10:59:42 +0100 | [diff] [blame] | 64 | if (!PSA_HANDLE_IS_VALID(handle)) { |
| 65 | return PSA_HANDLE_TO_ERROR(handle); |
Kevin Peng | 2ed3022 | 2019-04-30 09:26:11 +0800 | [diff] [blame] | 66 | } |
Tamas Ban | b6b8056 | 2019-01-04 22:49:24 +0000 | [diff] [blame] | 67 | |
Summer Qin | 4b1d03b | 2019-07-02 14:56:08 +0800 | [diff] [blame] | 68 | status = psa_call(handle, PSA_IPC_CALL, |
Kevin Peng | 2ed3022 | 2019-04-30 09:26:11 +0800 | [diff] [blame] | 69 | in_vec, IOVEC_LEN(in_vec), |
| 70 | out_vec, IOVEC_LEN(out_vec)); |
| 71 | psa_close(handle); |
Tamas Ban | b6b8056 | 2019-01-04 22:49:24 +0000 | [diff] [blame] | 72 | |
Raef Coles | 793574c | 2019-10-09 10:59:42 +0100 | [diff] [blame] | 73 | return status; |
Tamas Ban | 48a0eb5 | 2018-08-17 12:48:05 +0100 | [diff] [blame] | 74 | } |
David Vincze | ff6da53 | 2019-11-21 00:19:50 +0100 | [diff] [blame] | 75 | |
Raef Coles | 793574c | 2019-10-09 10:59:42 +0100 | [diff] [blame] | 76 | psa_status_t |
Summer Qin | 0e5b2e0 | 2020-10-22 11:23:39 +0800 | [diff] [blame] | 77 | tfm_initial_attest_get_public_key(uint8_t *public_key, |
| 78 | size_t public_key_buf_size, |
| 79 | size_t *public_key_len, |
| 80 | psa_ecc_family_t *elliptic_curve_type) |
David Vincze | ff6da53 | 2019-11-21 00:19:50 +0100 | [diff] [blame] | 81 | { |
| 82 | psa_handle_t handle = PSA_NULL_HANDLE; |
| 83 | psa_status_t status; |
| 84 | |
| 85 | psa_outvec out_vec[] = { |
| 86 | {.base = public_key, .len = public_key_buf_size}, |
| 87 | {.base = elliptic_curve_type, .len = sizeof(*elliptic_curve_type)}, |
| 88 | {.base = public_key_len, .len = sizeof(*public_key_len)} |
| 89 | }; |
| 90 | |
| 91 | handle = psa_connect(TFM_ATTEST_GET_PUBLIC_KEY_SID, |
| 92 | TFM_ATTEST_GET_PUBLIC_KEY_VERSION); |
| 93 | if (!PSA_HANDLE_IS_VALID(handle)) { |
Raef Coles | 793574c | 2019-10-09 10:59:42 +0100 | [diff] [blame] | 94 | return PSA_HANDLE_TO_ERROR(handle); |
David Vincze | ff6da53 | 2019-11-21 00:19:50 +0100 | [diff] [blame] | 95 | } |
| 96 | |
| 97 | status = psa_call(handle, PSA_IPC_CALL, |
| 98 | NULL, 0, |
| 99 | out_vec, IOVEC_LEN(out_vec)); |
| 100 | psa_close(handle); |
| 101 | |
Raef Coles | 793574c | 2019-10-09 10:59:42 +0100 | [diff] [blame] | 102 | return status; |
David Vincze | ff6da53 | 2019-11-21 00:19:50 +0100 | [diff] [blame] | 103 | } |