diff options
author | Mikael Olsson <mikael.olsson@arm.com> | 2023-06-13 12:17:04 +0200 |
---|---|---|
committer | Mikael Olsson <mikael.olsson@arm.com> | 2023-06-16 17:05:51 +0200 |
commit | 09965b094429c45f4da273bb5c11f1c177acf9fd (patch) | |
tree | 622bb8e2d9477c7b54d3a2eea7c57c3bb36f4939 /kernel/ethosu_mailbox.c | |
parent | 252ed6a0f9af7a002d70a9c7644fec5b0b78ead5 (diff) | |
download | ethos-u-linux-driver-stack-09965b094429c45f4da273bb5c11f1c177acf9fd.tar.gz |
Add type check when getting message by ID
When the kernel driver handles incoming rpmsg messages, it uses the ID
in the message header to find the corresponding mailbox message. The
mailbox messages are stored as a generic message struct that is later
cast to the specific message type.
There is currently no type information stored in the generic message
struct so only the ID is used to distinguish between the mailbox
messages. This means if an incorrect ID is received that matches a
mailbox message with a different type, the kernel driver will cast the
message struct to the wrong type.
Type information has now been added to the generic message struct and
will be checked when trying to find the corresponding mailbox message so
both the ID and type must be correct to find a matching message.
Change-Id: Ifdbceea6ec4ae7078f424a32ce1ff5474bd22fef
Diffstat (limited to 'kernel/ethosu_mailbox.c')
-rw-r--r-- | kernel/ethosu_mailbox.c | 14 |
1 files changed, 13 insertions, 1 deletions
diff --git a/kernel/ethosu_mailbox.c b/kernel/ethosu_mailbox.c index 4c64f17..3e7284b 100644 --- a/kernel/ethosu_mailbox.c +++ b/kernel/ethosu_mailbox.c @@ -79,12 +79,16 @@ void ethosu_mailbox_deregister(struct ethosu_mailbox *mbox, } struct ethosu_mailbox_msg *ethosu_mailbox_find(struct ethosu_mailbox *mbox, - int msg_id) + int msg_id, + uint32_t msg_type) { struct ethosu_mailbox_msg *ptr = (struct ethosu_mailbox_msg *)idr_find( &mbox->msg_idr, msg_id); if (ptr == NULL) + return ERR_PTR(-ENOENT); + + if (ptr->type != msg_type) return ERR_PTR(-EINVAL); return ptr; @@ -147,6 +151,8 @@ int ethosu_mailbox_capabilities_request(struct ethosu_mailbox *mbox, } }; + msg->type = rpmsg.header.type; + return rpmsg_send(mbox->ept, &rpmsg, sizeof(rpmsg.header)); } @@ -172,6 +178,8 @@ int ethosu_mailbox_inference(struct ethosu_mailbox *mbox, struct ethosu_core_msg_inference_req *inf_req = &rpmsg.inf_req; uint32_t i; + msg->type = rpmsg.header.type; + /* Verify that the uapi and core has the same number of pmus */ if (pmu_event_config_count != ETHOSU_CORE_PMU_MAX) { dev_err(mbox->dev, "PMU count misconfigured."); @@ -218,6 +226,8 @@ int ethosu_mailbox_network_info_request(struct ethosu_mailbox *mbox, }; struct ethosu_core_msg_network_info_req *info_req = &rpmsg.net_info_req; + msg->type = rpmsg.header.type; + if (network != NULL) { info_req->network.type = ETHOSU_CORE_NETWORK_BUFFER; ethosu_core_set_size(network, &info_req->network.buffer); @@ -246,6 +256,8 @@ int ethosu_mailbox_cancel_inference(struct ethosu_mailbox *mbox, } }; + msg->type = rpmsg.header.type; + return rpmsg_send(mbox->ept, &rpmsg, sizeof(rpmsg.header) + sizeof(rpmsg.cancel_req)); } |