diff options
Diffstat (limited to 'drivers/soc')
-rw-r--r-- | drivers/soc/qcom/hab/hab_msg.c | 14 |
1 files changed, 10 insertions, 4 deletions
diff --git a/drivers/soc/qcom/hab/hab_msg.c b/drivers/soc/qcom/hab/hab_msg.c index 61e658954312..71010be414d5 100644 --- a/drivers/soc/qcom/hab/hab_msg.c +++ b/drivers/soc/qcom/hab/hab_msg.c @@ -64,9 +64,14 @@ hab_msg_dequeue(struct virtual_channel *vchan, struct hab_message **msg, } } - /* return all the received messages before the remote close */ - if ((!ret || (ret == -ERESTARTSYS)) && !hab_rx_queue_empty(vchan)) { - spin_lock_bh(&vchan->rx_lock); + /* + * return all the received messages before the remote close, + * and need empty check again in case the list is empty now due to + * dequeue by other threads + */ + spin_lock_bh(&vchan->rx_lock); + + if ((!ret || (ret == -ERESTARTSYS)) && !list_empty(&vchan->rx_list)) { message = list_first_entry(&vchan->rx_list, struct hab_message, node); if (message) { @@ -84,11 +89,12 @@ hab_msg_dequeue(struct virtual_channel *vchan, struct hab_message **msg, ret = -EOVERFLOW; /* come back again */ } } - spin_unlock_bh(&vchan->rx_lock); } else /* no message received, retain the original status */ *rsize = 0; + spin_unlock_bh(&vchan->rx_lock); + *msg = message; return ret; } |