1 /* 2 * Copyright (c) 2022-2023, Arm Limited. All rights reserved. 3 * 4 * SPDX-License-Identifier: BSD-3-Clause 5 * 6 */ 7 8 #include <assert.h> 9 #include <string.h> 10 11 #include <common/debug.h> 12 #include "rss_comms_protocol_embed.h" 13 14 #define TYPE_OFFSET (16U) 15 #define TYPE_MASK (0xFFFFUL << TYPE_OFFSET) 16 #define IN_LEN_OFFSET (8U) 17 #define IN_LEN_MASK (0xFFUL << IN_LEN_OFFSET) 18 #define OUT_LEN_OFFSET (0U) 19 #define OUT_LEN_MASK (0xFFUL << OUT_LEN_OFFSET) 20 21 #define PARAM_PACK(type, in_len, out_len) \ 22 (((((uint32_t)type) << TYPE_OFFSET) & TYPE_MASK) | \ 23 ((((uint32_t)in_len) << IN_LEN_OFFSET) & IN_LEN_MASK) | \ 24 ((((uint32_t)out_len) << OUT_LEN_OFFSET) & OUT_LEN_MASK)) 25 26 psa_status_t rss_protocol_embed_serialize_msg(psa_handle_t handle, 27 int16_t type, 28 const psa_invec *in_vec, 29 uint8_t in_len, 30 const psa_outvec *out_vec, 31 uint8_t out_len, 32 struct rss_embed_msg_t *msg, 33 size_t *msg_len) 34 { 35 uint32_t payload_size = 0; 36 uint32_t i; 37 38 assert(msg != NULL); 39 assert(msg_len != NULL); 40 assert(in_vec != NULL); 41 42 msg->ctrl_param = PARAM_PACK(type, in_len, out_len); 43 msg->handle = handle; 44 45 /* Fill msg iovec lengths */ 46 for (i = 0U; i < in_len; ++i) { 47 msg->io_size[i] = in_vec[i].len; 48 } 49 for (i = 0U; i < out_len; ++i) { 50 msg->io_size[in_len + i] = out_vec[i].len; 51 } 52 53 for (i = 0U; i < in_len; ++i) { 54 if (in_vec[i].len > sizeof(msg->trailer) - payload_size) { 55 return PSA_ERROR_INVALID_ARGUMENT; 56 } 57 memcpy(msg->trailer + payload_size, 58 in_vec[i].base, 59 in_vec[i].len); 60 payload_size += in_vec[i].len; 61 } 62 63 /* Output the actual size of the message, to optimize sending */ 64 *msg_len = sizeof(*msg) - sizeof(msg->trailer) + payload_size; 65 66 return PSA_SUCCESS; 67 } 68 69 psa_status_t rss_protocol_embed_deserialize_reply(psa_outvec *out_vec, 70 uint8_t out_len, 71 psa_status_t *return_val, 72 const struct rss_embed_reply_t *reply, 73 size_t reply_size) 74 { 75 uint32_t payload_offset = 0; 76 uint32_t i; 77 78 assert(reply != NULL); 79 assert(return_val != NULL); 80 81 for (i = 0U; i < out_len; ++i) { 82 if ((sizeof(*reply) - sizeof(reply->trailer) + payload_offset) 83 > reply_size) { 84 return PSA_ERROR_INVALID_ARGUMENT; 85 } 86 87 memcpy(out_vec[i].base, 88 reply->trailer + payload_offset, 89 reply->out_size[i]); 90 out_vec[i].len = reply->out_size[i]; 91 payload_offset += reply->out_size[i]; 92 } 93 94 *return_val = reply->return_val; 95 96 return PSA_SUCCESS; 97 } 98