[Devel] [PATCH RH9 v3 05/10] drivers/vhost: rework worker creation

[Pavel Tikhomirov]

On 10.10.2022 17:56, Andrey Zhadchenko wrote:
> Add function to create a vhost worker and add it into the device.
> Rework vhost_dev_set_owner
> 
> https://jira.sw.ru/browse/PSBM-139414
> Signed-off-by: Andrey Zhadchenko <andrey.zhadchenko at virtuozzo.com>
> ---
>   drivers/vhost/vhost.c | 68 +++++++++++++++++++++++++------------------
>   1 file changed, 40 insertions(+), 28 deletions(-)
> 
> diff --git a/drivers/vhost/vhost.c b/drivers/vhost/vhost.c
> index df9c57c82a52..173a14041678 100644
> --- a/drivers/vhost/vhost.c
> +++ b/drivers/vhost/vhost.c
> @@ -626,53 +626,65 @@ static void vhost_detach_mm(struct vhost_dev *dev)
>   	dev->mm = NULL;
>   }
>   
> -/* Caller should have device mutex */
> -long vhost_dev_set_owner(struct vhost_dev *dev)
> +static int vhost_add_worker(struct vhost_dev *dev)
>   {
> +	struct vhost_worker *w = &dev->workers[dev->nworkers];
>   	struct task_struct *worker;
>   	int err;
>   
> +	if (dev->nworkers == VHOST_MAX_WORKERS)

Can we use ">=" just in case.

> +		return -E2BIG;
> +
> +	worker = kthread_create(vhost_worker, w,
> +				"vhost-%d-%d", current->pid, dev->nworkers);
> +	if (IS_ERR(worker))
> +		return PTR_ERR(worker);
> +
> +	w->worker = worker;
> +	wake_up_process(worker); /* avoid contributing to loadavg */
> +
> +	err = vhost_worker_attach_cgroups(w);
> +	if (err)
> +		goto cleanup;
> +
> +	dev->nworkers++;
> +	return 0;
> +
> +cleanup:
> +	kthread_stop(worker);
> +	w->worker = NULL;
> +
> +	return err;
> +}
> +
> +/* Caller should have device mutex */
> +long vhost_dev_set_owner(struct vhost_dev *dev)
> +{
> +	int err;
> +
>   	/* Is there an owner already? */
> -	if (vhost_dev_has_owner(dev)) {
> -		err = -EBUSY;
> -		goto err_mm;
> -	}
> +	if (vhost_dev_has_owner(dev))
> +		return -EBUSY;
>   
>   	vhost_attach_mm(dev);
>   
>   	dev->kcov_handle = kcov_common_handle();
>   	if (dev->use_worker) {
> -		worker = kthread_create(vhost_worker, dev,
> -					"vhost-%d", current->pid);
> -		if (IS_ERR(worker)) {
> -			err = PTR_ERR(worker);
> -			goto err_worker;
> -		}
> -
> -		dev->workers[0].worker = worker;
> -		dev->nworkers = 1;
> -		wake_up_process(worker); /* avoid contributing to loadavg */
> -
> -		err = vhost_worker_attach_cgroups(&dev->workers[0]);
> +		err = vhost_add_worker(dev);
>   		if (err)
> -			goto err_cgroup;
> +			goto err_mm;
>   	}
>   
>   	err = vhost_dev_alloc_iovecs(dev);
>   	if (err)
> -		goto err_cgroup;
> +		goto err_worker;
>   
>   	return 0;
> -err_cgroup:
> -	dev->nworkers = 0;
> -	if (dev->workers[0].worker) {
> -		kthread_stop(dev->workers[0].worker);
> -		dev->workers[0].worker = NULL;
> -	}
>   err_worker:
> -	vhost_detach_mm(dev);
> -	dev->kcov_handle = 0;
> +	vhost_cleanup_workers(dev);
>   err_mm:
> +	vhost_detach_mm(dev);
> +	dev->kcov_handle = 0;
>   	return err;
>   }
>   EXPORT_SYMBOL_GPL(vhost_dev_set_owner);

-- 
Best regards, Tikhomirov Pavel
Software Developer, Virtuozzo.