Re: [PATCH] ksmbd: prevent memory leak on error return

[Namjae Jeon]

2023-11-08 17:15 GMT+09:00, Zongmin Zhou <min_halo@xxxxxxx>:
> When allocated memory for 'new' failed,just return
> will cause memory leak of 'ar'.
>
> Fixes: 1819a9042999 ("ksmbd: reorganize ksmbd_iov_pin_rsp()")
>
> Reported-by: kernel test robot <lkp@xxxxxxxxx>
> Reported-by: Dan Carpenter <error27@xxxxxxxxx>
> Closes: https://lore.kernel.org/r/202311031837.H3yo7JVl-lkp@xxxxxxxxx/
> Signed-off-by: Zongmin Zhou<zhouzongmin@xxxxxxxxxx>
> ---
>  fs/smb/server/ksmbd_work.c | 4 +++-
>  1 file changed, 3 insertions(+), 1 deletion(-)
>
> diff --git a/fs/smb/server/ksmbd_work.c b/fs/smb/server/ksmbd_work.c
> index a2ed441e837a..dbbef686e160 100644
> --- a/fs/smb/server/ksmbd_work.c
> +++ b/fs/smb/server/ksmbd_work.c
> @@ -123,8 +123,10 @@ static int __ksmbd_iov_pin_rsp(struct ksmbd_work *work,
> void *ib, int len,
>  		new = krealloc(work->iov,
>  			       sizeof(struct kvec) * work->iov_alloc_cnt,
>  			       GFP_KERNEL | __GFP_ZERO);
> -		if (!new)
> +		if (!new) {
> +			kfree(ar);
Looks good to me:)
Can you add the below code to rollback ->iov_alloc_cnt ?
                       work->iov_alloc_cnt -= 4;
Thanks!
>  			return -ENOMEM;
> +		}
>  		work->iov = new;
>  	}
>
> --
> 2.34.1
>
>
>