Received-SPF: pass (google.com: domain of linux-kernel-owner@vger.kernel.org designates 23.128.96.18 as permitted sender) client-ip=23.128.96.18;
From:   Mike Christie <michael.christie@oracle.com>
To:     hdanton@sina.com, hch@infradead.org, stefanha@redhat.com,
        jasowang@redhat.com, mst@redhat.com, sgarzare@redhat.com,
        virtualization@lists.linux-foundation.org,
        christian.brauner@ubuntu.com, axboe@kernel.dk,
        linux-kernel@vger.kernel.org
Cc:     Mike Christie <michael.christie@oracle.com>
Subject: [PATCH V2 9/9] vhost: use kernel_worker to check RLIMITs and inherit v2 cgroups
Date:   Tue, 21 Sep 2021 16:52:18 -0500
Message-Id: <20210921215218.89844-10-michael.christie@oracle.com>
In-Reply-To: <20210921215218.89844-1-michael.christie@oracle.com>
References: <20210921215218.89844-1-michael.christie@oracle.com>
Content-Transfer-Encoding: 8bit
Content-Type: text/plain
MIME-Version: 1.0
X-MS-Exchange-AntiSpam-MessageData-ChunkCount: 1
X-MS-Exchange-AntiSpam-MessageData-0: =?us-ascii?Q?KSCsoZHYEAMNjx/paEFaKbfntVFUfGpilULetn+3kOTxmTqxjV+pI+7TceEU?=
 =?us-ascii?Q?nT8KQgCTtaoIO8sI0D2FIRY7xa3wlVcrJeuTUHnRvrr4iNHROtZ+OOUHqTA6?=
 =?us-ascii?Q?9OlNEtBD+d/frNXFQ1aQPzOq/cM+aJnBedeWROvRhZHazHS9IQbfWmxWq8ps?=
 =?us-ascii?Q?xhNF7sbycQTN8k9byEaQHwWCjNdcQEp7ktOS81JnnavXSk3CRez6aDihxqRm?=
 =?us-ascii?Q?wQ6hiMYPMCkcj0D92cfmXI774dQl20JruoqBfpLuCmyygVoCmY/DD/UykMC+?=
 =?us-ascii?Q?rG88B7hOFcd+HIYoGdfeVEjwS+221Af9ymN6cNojODsFkX6mwR1Fg/erJGfX?=
 =?us-ascii?Q?mwAArs45L1YdQVVZqYK4/q0yf135+NqhT0ouD1IsEoAUH6C5pzu2YNtahmmY?=
 =?us-ascii?Q?oUEK/EKGLAzbAL4s6bBZnlHMNWDTLMTnJpk7CfCeb98FMAF6LWvds7Z9Ht9x?=
 =?us-ascii?Q?MJDxCL8sFBaDEjWpZE6oxTNd2/z5L7SrLQX4Gi1zFIWYti8mO0Nl2IJzkgVd?=
 =?us-ascii?Q?/6mU+D/ByBsSuqFJJJY7M1rkuTOjNu9KSkzv9jgwUEagC05/Z0AB1DJuKGvK?=
 =?us-ascii?Q?Rj4PIBnO9Y9ndWq1nUN+FHLnLgo+8yuV1FoTpIjaEZB47Dm/ZtYvIFZCTuZu?=
 =?us-ascii?Q?rJJFsd6VEyXEaJ6cMM8JxArq9hnj8G/oRsHzegs5IMp+c2RQfT0KYpdN9Rms?=
 =?us-ascii?Q?5sE0/tubV4YCFe1vONWyQkjbI8PrhCwUHVDPI0jDeH6fwdGQJytsg/FFZ3pJ?=
 =?us-ascii?Q?jnI3rkNhfzxsFLA3+/f5A2C3qsGewtFVEuassXL2uajJh8uQhtAoeszIJBz8?=
 =?us-ascii?Q?yy748qvgvdof9sl8DNtztSdNL+6nUe5BNlQyqzZdpnWKyZ29KjgfKPR5U4+Q?=
 =?us-ascii?Q?aJMpFmr6seepPMEJghryofQ8tYxYGW4YLJzTX/Wi5sFOIMiS65GqZvW5BLuw?=
 =?us-ascii?Q?CstxxOlDY4z/1yiZHihHEvAMRMoRgoQucKfNIDpE/4NQCzqf+owPBI/Dq/c7?=
 =?us-ascii?Q?pfYF+esE4pqTIDVcoi2LqSTT7FHGtXi/+8uQhKTmZ0beQzLHCFiY22zCkkQq?=
 =?us-ascii?Q?ufkoQyFrQvqyyP1KjwXGYVz8P+jbsOVwN8u6q/CWNcaAM7kw1hnhT0uoMfTh?=
 =?us-ascii?Q?K5Rg56LDC9Yfy5OaYdFSEP83WIgJZbkRH9xciALZkLLR+vEEuZlA90skVO5k?=
 =?us-ascii?Q?hY6OXA8Ny2FoGXUEbzwz76rIZMNdL45TGSFRlR8HrUuFWBGy2ijk+mNw1gs0?=
 =?us-ascii?Q?ylMWZ1vfa/V5u+2h6en1yXBl/9s8MeA/897Ni/kCM1lZRKeRvZl1BaZB14pl?=
 =?us-ascii?Q?7IyfolJYZZ/a6Tf73qy3lby8?=
X-OriginatorOrg: oracle.com
X-MS-Exchange-CrossTenant-Network-Message-Id: 345eda58-720c-45fe-77fc-08d97d4a2178
X-MS-Exchange-CrossTenant-AuthSource: BYAPR10MB3573.namprd10.prod.outlook.com
X-MS-Exchange-CrossTenant-AuthAs: Internal
X-MS-Exchange-CrossTenant-OriginalArrivalTime: 21 Sep 2021 21:52:39.4079
 (UTC)
X-MS-Exchange-CrossTenant-FromEntityHeader: Hosted
X-MS-Exchange-CrossTenant-Id: 4e2c6054-71cb-48f1-bd6c-3a9705aca71b
X-MS-Exchange-CrossTenant-MailboxType: HOSTED
X-MS-Exchange-CrossTenant-UserPrincipalName: vj4K+6sGN20/MMiVz58OtNVtGL9lIQkqSctqKotg6v90KSTCACwhAgNGEnLp1eQmUHnSQHishl0M9zU4bc3Fg1jr7yMJJooS1homCPARVRU=
X-MS-Exchange-Transport-CrossTenantHeadersStamped: BYAPR10MB3239
X-Proofpoint-Virus-Version: vendor=nai engine=6300 definitions=10114 signatures=668682
X-Proofpoint-Spam-Details: rule=notspam policy=default score=0 malwarescore=0 spamscore=0
 suspectscore=0 phishscore=0 mlxlogscore=999 bulkscore=0 mlxscore=0
 adultscore=0 classifier=spam adjust=0 reason=mlx scancount=1
 engine=8.12.0-2109200000 definitions=main-2109210130
X-Proofpoint-GUID: cwr38CGoqD9qZKoEnLwsSIP3SvO080b3
X-Proofpoint-ORIG-GUID: cwr38CGoqD9qZKoEnLwsSIP3SvO080b3
Precedence: bulk
List-ID: <linux-kernel.vger.kernel.org>
X-Mailing-List: linux-kernel@vger.kernel.org

For vhost workers we use the kthread API which inherit's its values from
and checks against the kthreadd thread. This results in cgroups v2 not
working and the wrong RLIMITs being checked. This patch has us use the
kernel_copy_process function which will inherit its values/checks from the
thread that owns the device.

Signed-off-by: Mike Christie <michael.christie@oracle.com>
---
 drivers/vhost/vhost.c | 68 ++++++++++++++++---------------------------
 drivers/vhost/vhost.h |  7 ++++-
 2 files changed, 31 insertions(+), 44 deletions(-)

diff --git a/drivers/vhost/vhost.c b/drivers/vhost/vhost.c
index c9a1f706989c..7a5142dcde1b 100644
--- a/drivers/vhost/vhost.c
+++ b/drivers/vhost/vhost.c
@@ -22,7 +22,6 @@
 #include <linux/slab.h>
 #include <linux/vmalloc.h>
 #include <linux/kthread.h>
-#include <linux/cgroup.h>
 #include <linux/module.h>
 #include <linux/sort.h>
 #include <linux/sched/mm.h>
@@ -344,17 +343,14 @@ static void vhost_vq_reset(struct vhost_dev *dev,
 static int vhost_worker(void *data)
 {
 	struct vhost_worker *worker = data;
-	struct vhost_dev *dev = worker->dev;
 	struct vhost_work *work, *work_next;
 	struct llist_node *node;
 
-	kthread_use_mm(dev->mm);
-
 	for (;;) {
 		/* mb paired w/ kthread_stop */
 		set_current_state(TASK_INTERRUPTIBLE);
 
-		if (kthread_should_stop()) {
+		if (test_bit(VHOST_WORKER_FLAG_STOP, &worker->flags)) {
 			__set_current_state(TASK_RUNNING);
 			break;
 		}
@@ -376,8 +372,9 @@ static int vhost_worker(void *data)
 				schedule();
 		}
 	}
-	kthread_unuse_mm(dev->mm);
-	return 0;
+
+	complete(worker->exit_done);
+	do_exit(0);
 }
 
 static void vhost_vq_free_iovecs(struct vhost_virtqueue *vq)
@@ -517,31 +514,6 @@ long vhost_dev_check_owner(struct vhost_dev *dev)
 }
 EXPORT_SYMBOL_GPL(vhost_dev_check_owner);
 
-struct vhost_attach_cgroups_struct {
-	struct vhost_work work;
-	struct task_struct *owner;
-	int ret;
-};
-
-static void vhost_attach_cgroups_work(struct vhost_work *work)
-{
-	struct vhost_attach_cgroups_struct *s;
-
-	s = container_of(work, struct vhost_attach_cgroups_struct, work);
-	s->ret = cgroup_attach_task_all(s->owner, current);
-}
-
-static int vhost_attach_cgroups(struct vhost_dev *dev)
-{
-	struct vhost_attach_cgroups_struct attach;
-
-	attach.owner = current;
-	vhost_work_init(&attach.work, vhost_attach_cgroups_work);
-	vhost_work_queue(dev, &attach.work);
-	vhost_work_dev_flush(dev);
-	return attach.ret;
-}
-
 /* Caller should have device mutex */
 bool vhost_dev_has_owner(struct vhost_dev *dev)
 {
@@ -579,6 +551,16 @@ static void vhost_detach_mm(struct vhost_dev *dev)
 	dev->mm = NULL;
 }
 
+static void vhost_worker_stop(struct vhost_worker *worker)
+{
+	DECLARE_COMPLETION_ONSTACK(exit_done);
+
+	worker->exit_done = &exit_done;
+	set_bit(VHOST_WORKER_FLAG_STOP, &worker->flags);
+	wake_up_process(worker->task);
+	wait_for_completion(worker->exit_done);
+}
+
 static void vhost_worker_free(struct vhost_dev *dev)
 {
 	struct vhost_worker *worker = dev->worker;
@@ -588,7 +570,7 @@ static void vhost_worker_free(struct vhost_dev *dev)
 
 	dev->worker = NULL;
 	WARN_ON(!llist_empty(&worker->work_list));
-	kthread_stop(worker->task);
+	vhost_worker_stop(worker);
 	kfree(worker);
 }
 
@@ -603,27 +585,27 @@ static int vhost_worker_create(struct vhost_dev *dev)
 		return -ENOMEM;
 
 	dev->worker = worker;
-	worker->dev = dev;
 	worker->kcov_handle = kcov_common_handle();
 	init_llist_head(&worker->work_list);
 
-	task = kthread_create(vhost_worker, worker, "vhost-%d", current->pid);
+	/*
+	 * vhost used to use the kthread API which ignores all signals by
+	 * default and the drivers expect this behavior. So we do not want to
+	 * ineherit the parent's signal handlers and set our worker to ignore
+	 * everything below.
+	 */
+	task = kernel_worker(vhost_worker, worker, NUMA_NO_NODE,
+			     CLONE_FS | CLONE_CLEAR_SIGHAND,
+			     KERN_WORKER_NO_FILES | KERN_WORKER_NO_SIGS);
 	if (IS_ERR(task)) {
 		ret = PTR_ERR(task);
 		goto free_worker;
 	}
 
 	worker->task = task;
-	wake_up_process(task); /* avoid contributing to loadavg */
-
-	ret = vhost_attach_cgroups(dev);
-	if (ret)
-		goto stop_worker;
-
+	kernel_worker_start(task, "vhost-%d", current->pid);
 	return 0;
 
-stop_worker:
-	kthread_stop(worker->task);
 free_worker:
 	kfree(worker);
 	dev->worker = NULL;
diff --git a/drivers/vhost/vhost.h b/drivers/vhost/vhost.h
index 102ce25e4e13..09748694cb66 100644
--- a/drivers/vhost/vhost.h
+++ b/drivers/vhost/vhost.h
@@ -25,11 +25,16 @@ struct vhost_work {
 	unsigned long		flags;
 };
 
+enum {
+	VHOST_WORKER_FLAG_STOP,
+};
+
 struct vhost_worker {
 	struct task_struct	*task;
+	struct completion	*exit_done;
 	struct llist_head	work_list;
-	struct vhost_dev	*dev;
 	u64			kcov_handle;
+	unsigned long		flags;
 };
 
 /* Poll a file (eventfd or socket) */
-- 
2.25.1