Merge "securemsm-kernel: Fix multiple listener registration on same fd"

This commit is contained in:
qctecmdr
2023-09-15 22:21:20 -07:00
committed by Gerrit - the friendly Code Review server

View File

@@ -3,7 +3,7 @@
* QTI Secure Execution Environment Communicator (QSEECOM) driver * QTI Secure Execution Environment Communicator (QSEECOM) driver
* *
* Copyright (c) 2012-2021, The Linux Foundation. All rights reserved. * Copyright (c) 2012-2021, The Linux Foundation. All rights reserved.
* Copyright (c) 2022 Qualcomm Innovation Center, Inc. All rights reserved. * Copyright (c) 2022-2023 Qualcomm Innovation Center, Inc. All rights reserved.
*/ */
#define pr_fmt(fmt) "QSEECOM: %s: " fmt, __func__ #define pr_fmt(fmt) "QSEECOM: %s: " fmt, __func__
@@ -375,7 +375,7 @@ struct qseecom_client_handle {
struct qseecom_listener_handle { struct qseecom_listener_handle {
u32 id; u32 id;
bool unregister_pending; bool register_pending;
bool release_called; bool release_called;
}; };
@@ -1525,6 +1525,11 @@ static int qseecom_register_listener(struct qseecom_dev_handle *data,
struct qseecom_registered_listener_list *new_entry; struct qseecom_registered_listener_list *new_entry;
struct qseecom_registered_listener_list *ptr_svc; struct qseecom_registered_listener_list *ptr_svc;
if (data->listener.register_pending) {
pr_err("Already a listner registration is in process on this FD\n");
return -EINVAL;
}
ret = copy_from_user(&rcvd_lstnr, argp, sizeof(rcvd_lstnr)); ret = copy_from_user(&rcvd_lstnr, argp, sizeof(rcvd_lstnr));
if (ret) { if (ret) {
pr_err("copy_from_user failed\n"); pr_err("copy_from_user failed\n");
@@ -1534,6 +1539,13 @@ static int qseecom_register_listener(struct qseecom_dev_handle *data,
rcvd_lstnr.sb_size)) rcvd_lstnr.sb_size))
return -EFAULT; return -EFAULT;
ptr_svc = __qseecom_find_svc(data->listener.id);
if (ptr_svc) {
pr_err("Already a listener registered on this data: lid=%d\n",
data->listener.id);
return -EINVAL;
}
ptr_svc = __qseecom_find_svc(rcvd_lstnr.listener_id); ptr_svc = __qseecom_find_svc(rcvd_lstnr.listener_id);
if (ptr_svc) { if (ptr_svc) {
if (!ptr_svc->unregister_pending) { if (!ptr_svc->unregister_pending) {
@@ -1577,13 +1589,16 @@ static int qseecom_register_listener(struct qseecom_dev_handle *data,
new_entry->svc.listener_id = rcvd_lstnr.listener_id; new_entry->svc.listener_id = rcvd_lstnr.listener_id;
new_entry->sb_length = rcvd_lstnr.sb_size; new_entry->sb_length = rcvd_lstnr.sb_size;
new_entry->user_virt_sb_base = rcvd_lstnr.virt_sb_base; new_entry->user_virt_sb_base = rcvd_lstnr.virt_sb_base;
data->listener.register_pending = true;
if (__qseecom_set_sb_memory(new_entry, data, &rcvd_lstnr)) { if (__qseecom_set_sb_memory(new_entry, data, &rcvd_lstnr)) {
pr_err("qseecom_set_sb_memory failed for listener %d, size %d\n", pr_err("qseecom_set_sb_memory failed for listener %d, size %d\n",
rcvd_lstnr.listener_id, rcvd_lstnr.sb_size); rcvd_lstnr.listener_id, rcvd_lstnr.sb_size);
__qseecom_free_tzbuf(&new_entry->sglistinfo_shm); __qseecom_free_tzbuf(&new_entry->sglistinfo_shm);
kzfree(new_entry); kzfree(new_entry);
data->listener.register_pending = false;
return -ENOMEM; return -ENOMEM;
} }
data->listener.register_pending = false;
init_waitqueue_head(&new_entry->rcv_req_wq); init_waitqueue_head(&new_entry->rcv_req_wq);
init_waitqueue_head(&new_entry->listener_block_app_wq); init_waitqueue_head(&new_entry->listener_block_app_wq);