drivers/optee: Return error if optee_recv return prematurely
Signed-off-by: Xiang Xiao <xiaoxiang@xiaomi.com>
This commit is contained in:
parent
a074a572af
commit
7b2dbcf50a
1 changed files with 6 additions and 9 deletions
|
@ -291,24 +291,21 @@ static int optee_from_msg_param(FAR struct tee_ioctl_param *params,
|
|||
return 0;
|
||||
}
|
||||
|
||||
static ssize_t optee_recv(FAR struct socket *psock, FAR void *msg,
|
||||
size_t size)
|
||||
static int optee_recv(FAR struct socket *psock, FAR void *msg, size_t size)
|
||||
{
|
||||
size_t remain = size;
|
||||
|
||||
while (remain)
|
||||
while (size > 0)
|
||||
{
|
||||
ssize_t n = psock_recv(psock, msg, remain, 0);
|
||||
ssize_t n = psock_recv(psock, msg, size, 0);
|
||||
if (n <= 0)
|
||||
{
|
||||
return remain == size ? n : size - remain;
|
||||
return n < 0 ? n : -EIO;
|
||||
}
|
||||
|
||||
remain -= n;
|
||||
size -= n;
|
||||
msg = (FAR char *)msg + n;
|
||||
}
|
||||
|
||||
return size;
|
||||
return 0;
|
||||
}
|
||||
|
||||
static int optee_send_recv(FAR struct socket *psocket,
|
||||
|
|
Loading…
Reference in a new issue