[Devel] [PATCH RH9 07/10] drivers/vhost: assign workers to virtqueues

[Konstantin Khorenko]

On 16.08.2022 09:44, Andrey Zhadchenko wrote:
> Add worker pointer to every virtqueue. Add routine to assing
> workers to virtqueues and call it after any worker creation
> 
> https://jira.sw.ru/browse/PSBM-139414
> Signed-off-by: Andrey Zhadchenko <andrey.zhadchenko at virtuozzo.com>
> ---
>   drivers/vhost/vhost.c | 13 +++++++++++++
>   drivers/vhost/vhost.h |  2 ++
>   2 files changed, 15 insertions(+)
> 
> diff --git a/drivers/vhost/vhost.c b/drivers/vhost/vhost.c
> index de9b17a9332a..965299b0c0a3 100644
> --- a/drivers/vhost/vhost.c
> +++ b/drivers/vhost/vhost.c
> @@ -677,6 +677,17 @@ static int vhost_add_workers(struct vhost_dev *dev, int n)
>   	return ret;
>   }
>   
> +static void vhost_assign_workers(struct vhost_dev *dev)
> +{
> +	int i, j = 0;
> +
> +	for (i = 0; i < dev->nvqs; i++) {
> +		dev->vqs[i]->worker = &dev->workers[j];
> +		if (++j == dev->nworkers)
> +			j = 0;
> +	}
> +}
> +
>   /* Caller should have device mutex */
>   long vhost_dev_set_owner(struct vhost_dev *dev)
>   {
> @@ -699,6 +710,7 @@ long vhost_dev_set_owner(struct vhost_dev *dev)
>   	if (err)
>   		goto err_worker;
>   
> +	vhost_assign_workers(dev);
>   	return 0;
>   err_worker:
>   	vhost_cleanup_workers(dev);
> @@ -1896,6 +1908,7 @@ long vhost_dev_ioctl(struct vhost_dev *d, unsigned int ioctl, void __user *argp)
>   			break;
>   		}
>   		r = vhost_add_workers(d, n);
> +		vhost_assign_workers(d);
>   		break;
>   	default:
>   		r = -ENOIOCTLCMD;
> diff --git a/drivers/vhost/vhost.h b/drivers/vhost/vhost.h
> index 634ea828cbba..9632f6501617 100644
> --- a/drivers/vhost/vhost.h
> +++ b/drivers/vhost/vhost.h
> @@ -138,6 +138,8 @@ struct vhost_virtqueue {
>   	bool user_be;
>   #endif
>   	u32 busyloop_timeout;
> +
> +	struct vhost_worker *worker;
>   };
>   
>   struct vhost_msg_node {

LGTM