Re: [PATCH] media: v4l2: Fix memleak in videobuf_read_one

[Hans Verkuil]

On 05/01/2021 08:59, Dinghao Liu wrote:
> When videobuf_waiton() fails, we should execute clean
> functions to prevent memleak. It's the same when
> __videobuf_copy_to_user() fails.
> 
> Fixes: 7a7d9a89d0307 ("V4L/DVB (6251): Replace video-buf to a more generic approach")
> Signed-off-by: Dinghao Liu <dinghao.liu@xxxxxxxxxx>
> ---
>  drivers/media/v4l2-core/videobuf-core.c | 12 ++++++++++--
>  1 file changed, 10 insertions(+), 2 deletions(-)
> 
> diff --git a/drivers/media/v4l2-core/videobuf-core.c b/drivers/media/v4l2-core/videobuf-core.c
> index 606a271bdd2d..0709b75d11cd 100644
> --- a/drivers/media/v4l2-core/videobuf-core.c
> +++ b/drivers/media/v4l2-core/videobuf-core.c
> @@ -924,8 +924,12 @@ ssize_t videobuf_read_one(struct videobuf_queue *q,
>  
>  	/* wait until capture is done */
>  	retval = videobuf_waiton(q, q->read_buf, nonblocking, 1);
> -	if (0 != retval)
> +	if (retval != 0) {
> +		q->ops->buf_release(q, q->read_buf);
> +		kfree(q->read_buf);
> +		q->read_buf = NULL;
>  		goto done;
> +	}

I'm fairly certain that this is wrong: if waiton returns an error, then
that means that the wait is either interrupted or that we are in non-blocking
mode and no buffer has arrived yet. In that case you just go to done since
there is nothing to clean up.

>  
>  	CALL(q, sync, q, q->read_buf);
>  
> @@ -940,8 +944,12 @@ ssize_t videobuf_read_one(struct videobuf_queue *q,
>  
>  	/* Copy to userspace */
>  	retval = __videobuf_copy_to_user(q, q->read_buf, data, count, nonblocking);
> -	if (retval < 0)
> +	if (retval < 0) {
> +		q->ops->buf_release(q, q->read_buf);
> +		kfree(q->read_buf);
> +		q->read_buf = NULL;
>  		goto done;

I'm not sure about this either: if userspace gave a crappy pointer and this
copy_to_user fails, then that doesn't mean you should release the buffer.
The next read() might have a valid pointer or, more likely, the application
exits or crashes and everything is cleaned up when the filehandle is closed.

> +	}
>  
>  	q->read_off += retval;
>  	if (q->read_off == q->read_buf->size) {
> 

Do you have actual proof that this is a memleak? I don't want to mess around
with the old videobuf unless you can show me that there is a real bug.

Regards,

	Hans