From patchwork Thu Feb 8 21:22:02 2024 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: Lokesh Gidra X-Patchwork-Id: 13550492 Return-Path: X-Spam-Checker-Version: SpamAssassin 3.4.0 (2014-02-07) on aws-us-west-2-korg-lkml-1.web.codeaurora.org Received: from kanga.kvack.org (kanga.kvack.org [205.233.56.17]) by smtp.lore.kernel.org (Postfix) with ESMTP id 422A4C48260 for ; Thu, 8 Feb 2024 21:22:20 +0000 (UTC) Received: by kanga.kvack.org (Postfix) id AD8D06B0080; Thu, 8 Feb 2024 16:22:19 -0500 (EST) Received: by kanga.kvack.org (Postfix, from userid 40) id A617F6B0081; Thu, 8 Feb 2024 16:22:19 -0500 (EST) X-Delivered-To: int-list-linux-mm@kvack.org Received: by kanga.kvack.org (Postfix, from userid 63042) id 8DB126B0082; Thu, 8 Feb 2024 16:22:19 -0500 (EST) X-Delivered-To: linux-mm@kvack.org Received: from relay.hostedemail.com (smtprelay0010.hostedemail.com [216.40.44.10]) by kanga.kvack.org (Postfix) with ESMTP id 7D8866B0080 for ; Thu, 8 Feb 2024 16:22:19 -0500 (EST) Received: from smtpin20.hostedemail.com (a10.router.float.18 [10.200.18.1]) by unirelay05.hostedemail.com (Postfix) with ESMTP id 60E3B40F83 for ; Thu, 8 Feb 2024 21:22:19 +0000 (UTC) X-FDA: 81769909998.20.5E42677 Received: from mail-yw1-f201.google.com (mail-yw1-f201.google.com [209.85.128.201]) by imf14.hostedemail.com (Postfix) with ESMTP id 9686C100010 for ; Thu, 8 Feb 2024 21:22:16 +0000 (UTC) Authentication-Results: imf14.hostedemail.com; dkim=pass header.d=google.com header.s=20230601 header.b="w/2XQkuu"; dmarc=pass (policy=reject) header.from=google.com; spf=pass (imf14.hostedemail.com: domain of 3B0bFZQsKCNwJMICQFEGBP8EMMEJC.AMKJGLSV-KKIT8AI.MPE@flex--lokeshgidra.bounces.google.com designates 209.85.128.201 as permitted sender) smtp.mailfrom=3B0bFZQsKCNwJMICQFEGBP8EMMEJC.AMKJGLSV-KKIT8AI.MPE@flex--lokeshgidra.bounces.google.com ARC-Message-Signature: i=1; a=rsa-sha256; c=relaxed/relaxed; d=hostedemail.com; s=arc-20220608; t=1707427336; h=from:from:sender:reply-to:subject:subject:date:date: message-id:message-id:to:to:cc:cc:mime-version:mime-version: content-type:content-type:content-transfer-encoding: in-reply-to:in-reply-to:references:references:dkim-signature; bh=FeTp34JzZtDcGzw26Xy4JuyVdUXeLDl6wgD9/498E4U=; b=nf0ZIGTIbLHsz9eDXuov4Krva17PffhTT1M5Z9/pa8/6MR3Rbetf233mnS9t3GpfmW973U 53ruxUEO5peKesBtO1FhkZ07XbUrrpAfiQFrUTEX9OFLAVIJaWzhcV1+Y3Yp8zFsY2HxJ+ QYTt3PCKueUht8YBnwe/iDzRU1A9cjA= ARC-Authentication-Results: i=1; imf14.hostedemail.com; dkim=pass header.d=google.com header.s=20230601 header.b="w/2XQkuu"; dmarc=pass (policy=reject) header.from=google.com; spf=pass (imf14.hostedemail.com: domain of 3B0bFZQsKCNwJMICQFEGBP8EMMEJC.AMKJGLSV-KKIT8AI.MPE@flex--lokeshgidra.bounces.google.com designates 209.85.128.201 as permitted sender) smtp.mailfrom=3B0bFZQsKCNwJMICQFEGBP8EMMEJC.AMKJGLSV-KKIT8AI.MPE@flex--lokeshgidra.bounces.google.com ARC-Seal: i=1; s=arc-20220608; d=hostedemail.com; t=1707427336; a=rsa-sha256; cv=none; b=3UZZTMKIwy4qWWdaYoOzVmjoFul9q7H7j1gG7rTqt4uBMOiJkBEQHMzjweeWbkdL5+IW8L pBl+CNFeTNQewCSnUzRJhXaGu1ilsC3397JtCVZKfLDHb+JwIYX8rJUpdC4xMI4C42utxp HaJxFIuy6QaXv4LH/bdnOpb1lPFyshw= Received: by mail-yw1-f201.google.com with SMTP id 00721157ae682-604b44e1dc9so6570777b3.1 for ; Thu, 08 Feb 2024 13:22:16 -0800 (PST) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=google.com; s=20230601; t=1707427335; x=1708032135; darn=kvack.org; h=cc:to:from:subject:message-id:references:mime-version:in-reply-to :date:from:to:cc:subject:date:message-id:reply-to; bh=FeTp34JzZtDcGzw26Xy4JuyVdUXeLDl6wgD9/498E4U=; b=w/2XQkuuYtIlAC+aUhY3HF7SBprsrEJvgTptn0pPAICMRDyn4fjeNnjeIYuN+YdSEN qNA+CHefwXdsj2RJFKUBpOYi9E0DPPgfFQkn5V5egNKvLfdp1s2/DoE4Ea3Qo7iuAJPb aFVyENl92t0y/J6bSiy2o8MsrQuEBSjReS8bSW5tBmiNwgzHDHyV4T2kdqW5CEJALlhI VPmoMvO1qT/EGIhUPiQVb3X/IpokFbT04SYxSkliehTc45fpx3efgyXG1hYI6jiYAEo6 JoUqOMMW821EFk4pwpTLslZRK2MmaqdxWdumgIp4MZ9Wn9MvVIr/pArQeDIumr4axA6m Eaew== X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20230601; t=1707427335; x=1708032135; h=cc:to:from:subject:message-id:references:mime-version:in-reply-to :date:x-gm-message-state:from:to:cc:subject:date:message-id:reply-to; bh=FeTp34JzZtDcGzw26Xy4JuyVdUXeLDl6wgD9/498E4U=; b=nrPqIedzVar5Jo9ewrRkkYZb17wVI9CNNGr/D+zoDzvrcQJTiOuIUibeknvYcApx+n 3KK2yTWozaJVM0Np+65w6J3U/NO5paSRD5+LUwiQ4dG70nssYXTzDWdXfCRe3nfkL8yc ySir8cvpIfA3tAo7j2rhXRlgwCft1DZViNhplQ135GkjKk8EfLrR00Dd946VRvQQGBNM CYZYC0+v6TyA0anNL9r0vc9tlXKJ9C2r8ivWCo04pdjY/0UOvx+JkQ8z0J/kHUoHL2E+ q3m4OzKDusZbNo8ZejuAkBWywLQQDdT+cdWbCCMf0xP1bR+tP2Q32kzEhoxBedWE6GZ/ a1xA== X-Gm-Message-State: AOJu0Ywfd2S56s8B/E9Pmze96tYpIHKfcSPhd9yPmo6sm+3SSQA/bgFe xtJNhZtW1GFRAf4VByWSij/0yrihvktRj3huQho52KWBLXBttKb3iBb89c7kfAOhI+uZ8kb1J0X TzfaHmfuRY+3u7Ql8aFNJ7g== X-Google-Smtp-Source: AGHT+IEU/rITDSgvPlrw4EgfUfRceJKqlUTpaOVq6aKKDnlTmsQYo2lPuV1sCIYnSwJGgLrvzLIw+YSJUuAsvhO0Tw== X-Received: from lg.mtv.corp.google.com ([2620:15c:211:202:e9ba:42a8:6aba:f5d5]) (user=lokeshgidra job=sendgmr) by 2002:a81:4e85:0:b0:602:d545:a3bb with SMTP id c127-20020a814e85000000b00602d545a3bbmr93801ywb.1.1707427335604; Thu, 08 Feb 2024 13:22:15 -0800 (PST) Date: Thu, 8 Feb 2024 13:22:02 -0800 In-Reply-To: <20240208212204.2043140-1-lokeshgidra@google.com> Mime-Version: 1.0 References: <20240208212204.2043140-1-lokeshgidra@google.com> X-Mailer: git-send-email 2.43.0.687.g38aa6559b0-goog Message-ID: <20240208212204.2043140-2-lokeshgidra@google.com> Subject: [PATCH v4 1/3] userfaultfd: move userfaultfd_ctx struct to header file From: Lokesh Gidra To: akpm@linux-foundation.org Cc: lokeshgidra@google.com, linux-fsdevel@vger.kernel.org, linux-mm@kvack.org, linux-kernel@vger.kernel.org, selinux@vger.kernel.org, surenb@google.com, kernel-team@android.com, aarcange@redhat.com, peterx@redhat.com, david@redhat.com, axelrasmussen@google.com, bgeffon@google.com, willy@infradead.org, jannh@google.com, kaleshsingh@google.com, ngeoffray@google.com, timmurray@google.com, rppt@kernel.org, Liam.Howlett@oracle.com X-Rspamd-Queue-Id: 9686C100010 X-Rspam-User: X-Rspamd-Server: rspam04 X-Stat-Signature: tkrssf7th9zienxims77hossyi5eicgo X-HE-Tag: 1707427336-238809 X-HE-Meta: U2FsdGVkX19Gia1aRqqso16lEWVGoZIF1BzAask8eSvz6iKGgVYiCJoTHMBSBsMLl5oFmTa37/4/rze27hhZXPDGI2sPktPnxmUQbHvucaEspORN1Xo9zwgtKnSylQsS2pYK9cvm3uG8J7BbvJ4yl9GfTHjYWdgiCPyXQPeI0h/yK4OshZNfF35oOYjqzRP+3G9AgA7o6BBgQFJXead51vef/GufEyXjpAZhxbedrc0hItCpRjvsiPWWxP8U2oQP1LGuC+FKtBj5f8YtJSqymiPqvqDw2flfJBlUaVVvYqvF7EKUKm/GcJvAQtk7f1J5V49xqhUV1RJI16TqrSi/I/zpamk8x0rktPxPsM0DFr32hKu5Fvu9HKkFok204/eu/mGPhdrK74a9X3zCxss/iYVSovYWL45j93zKF7ec5ocUKECC8/l4cUB7km2QvROhqu6bp/eGaa+rXIaE1zU85ySRYnRLw+J/RrSDcE9KI/x2sjbSlM/doDLFbIJJXWXonAdvWH6TbKzq8uYNqOWwb1mPNDQmyVPrmsmJSpehWzRiuubRrSrNuQdR0xU1sedlPqQF5seB/7VWm4HJDPQ8HXEVlAykYn5fNK3Lh/dMcRd4KzmbhTNRScZc6MnMEd808NXalf8+3WUSS/RAiGOsAwKOSzoduysmIq+HAfZS4yitkHNo3Bfsj7dm5QE+j956t5PMdEeFlHcmOGEJK8eJNe+v3uCKWwf6qGBA79Q2VNieywGNF7mRXq3eu7xcvwAt6wyt1SvUg4pXlICjjkzx0VbHnSAIQCOkempZvd/1721o3+QrMMm6zVhZJkRLv7rBZ/+zkLVj9bEL4bgaIjNNOKzGa56jv5PKpZKR3c0SqE+jqa/WkhQSbb4+3ptNCH47f/sN1f6aLYeecxm9Bhlnj7L9QtS0DsECz2VeOvkpcDa3r6pn75N8g2wkxi9aE/RAx5yvIWPEuTsFEOvvTlx O9n9DYBP ZHU5hGV59xPfhkf0qJV7SGiudrw9F7OjoDnL1kmNtcsp2LMfm84bdQAH4DjBR5CBCAG/Q6BGTG7/tvDSZTh9WTE/4Aj04+udmuphmw0T8ks2uWU8vM1RijySL4+sMfBtuNosWxz2QyNL+u5dziqosil1hMGWYXdQV+gzxJoJCOqm0PybfOQ2efG3lwb8OPXxg+uTa3oqvasiqVgtfHqhJg5rNX40Omb4DX1UbDhAz2WDjyppG4aZZ+NiA/BBxdO5zRP0rhU9nOr0s0hSKDL07VyjkPa7BCl2QF21nPh7iq/+Urmah+HSlIkkv7hCjdz6oW0ba9V85+HJuLiRYBNkTpEiXwfsnQTrggo3YaEh7xqg2oE9XRe2VVHxhehM5+AO+mVmAtfAkQOlyPjmPx6Gy8TkmG9xD8T+166Tw+TRvodjdpRK/1w7HBCjyEoh6EqV597W7UZjgrDT45ahMjJoSbk5CSt1nEecH0JgIoZsONq+pv7wFVQhuiWfSVcBDc6IhrWdhUiBCfA2239TZLdOj7tyceNzz4JRo5tN73unpcEmTC+DiFe7aZL9911+cSREOX/AgIQarTsBddl5AAAWgKQzHoT/PnN0nNBdCTE4Ai7P96BReoyPkyJyREzQY0thQm86K9CbWS+emGji80pgtmppHYhzXSsnc7zktPl09jiFvTeP7kHWsCOD96W3t1Gi69NMpRSkBkUv9LQ/o2PC95aQzQxKkivrKo5H6B+9Ss/O1oVnKqNhFix691PIjiJEIwVSqavuXdBR2JqI5+5v/qPMpc03wFi0IU2i7BC63fUgwCrwu8ScVPZbti9p9BO+odKsmeUFa+GNXFT4= X-Bogosity: Ham, tests=bogofilter, spamicity=0.000000, version=1.2.4 Sender: owner-linux-mm@kvack.org Precedence: bulk X-Loop: owner-majordomo@kvack.org List-ID: List-Subscribe: List-Unsubscribe: Moving the struct to userfaultfd_k.h to be accessible from mm/userfaultfd.c. There are no other changes in the struct. This is required to prepare for using per-vma locks in userfaultfd operations. Signed-off-by: Lokesh Gidra Reviewed-by: Mike Rapoport (IBM) --- fs/userfaultfd.c | 39 ----------------------------------- include/linux/userfaultfd_k.h | 39 +++++++++++++++++++++++++++++++++++ 2 files changed, 39 insertions(+), 39 deletions(-) diff --git a/fs/userfaultfd.c b/fs/userfaultfd.c index 05c8e8a05427..58331b83d648 100644 --- a/fs/userfaultfd.c +++ b/fs/userfaultfd.c @@ -50,45 +50,6 @@ static struct ctl_table vm_userfaultfd_table[] = { static struct kmem_cache *userfaultfd_ctx_cachep __ro_after_init; -/* - * Start with fault_pending_wqh and fault_wqh so they're more likely - * to be in the same cacheline. - * - * Locking order: - * fd_wqh.lock - * fault_pending_wqh.lock - * fault_wqh.lock - * event_wqh.lock - * - * To avoid deadlocks, IRQs must be disabled when taking any of the above locks, - * since fd_wqh.lock is taken by aio_poll() while it's holding a lock that's - * also taken in IRQ context. - */ -struct userfaultfd_ctx { - /* waitqueue head for the pending (i.e. not read) userfaults */ - wait_queue_head_t fault_pending_wqh; - /* waitqueue head for the userfaults */ - wait_queue_head_t fault_wqh; - /* waitqueue head for the pseudo fd to wakeup poll/read */ - wait_queue_head_t fd_wqh; - /* waitqueue head for events */ - wait_queue_head_t event_wqh; - /* a refile sequence protected by fault_pending_wqh lock */ - seqcount_spinlock_t refile_seq; - /* pseudo fd refcounting */ - refcount_t refcount; - /* userfaultfd syscall flags */ - unsigned int flags; - /* features requested from the userspace */ - unsigned int features; - /* released */ - bool released; - /* memory mappings are changing because of non-cooperative event */ - atomic_t mmap_changing; - /* mm with one ore more vmas attached to this userfaultfd_ctx */ - struct mm_struct *mm; -}; - struct userfaultfd_fork_ctx { struct userfaultfd_ctx *orig; struct userfaultfd_ctx *new; diff --git a/include/linux/userfaultfd_k.h b/include/linux/userfaultfd_k.h index e4056547fbe6..691d928ee864 100644 --- a/include/linux/userfaultfd_k.h +++ b/include/linux/userfaultfd_k.h @@ -36,6 +36,45 @@ #define UFFD_SHARED_FCNTL_FLAGS (O_CLOEXEC | O_NONBLOCK) #define UFFD_FLAGS_SET (EFD_SHARED_FCNTL_FLAGS) +/* + * Start with fault_pending_wqh and fault_wqh so they're more likely + * to be in the same cacheline. + * + * Locking order: + * fd_wqh.lock + * fault_pending_wqh.lock + * fault_wqh.lock + * event_wqh.lock + * + * To avoid deadlocks, IRQs must be disabled when taking any of the above locks, + * since fd_wqh.lock is taken by aio_poll() while it's holding a lock that's + * also taken in IRQ context. + */ +struct userfaultfd_ctx { + /* waitqueue head for the pending (i.e. not read) userfaults */ + wait_queue_head_t fault_pending_wqh; + /* waitqueue head for the userfaults */ + wait_queue_head_t fault_wqh; + /* waitqueue head for the pseudo fd to wakeup poll/read */ + wait_queue_head_t fd_wqh; + /* waitqueue head for events */ + wait_queue_head_t event_wqh; + /* a refile sequence protected by fault_pending_wqh lock */ + seqcount_spinlock_t refile_seq; + /* pseudo fd refcounting */ + refcount_t refcount; + /* userfaultfd syscall flags */ + unsigned int flags; + /* features requested from the userspace */ + unsigned int features; + /* released */ + bool released; + /* memory mappings are changing because of non-cooperative event */ + atomic_t mmap_changing; + /* mm with one ore more vmas attached to this userfaultfd_ctx */ + struct mm_struct *mm; +}; + extern vm_fault_t handle_userfault(struct vm_fault *vmf, unsigned long reason); /* A combined operation mode + behavior flags. */ From patchwork Thu Feb 8 21:22:03 2024 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: Lokesh Gidra X-Patchwork-Id: 13550493 Return-Path: X-Spam-Checker-Version: SpamAssassin 3.4.0 (2014-02-07) on aws-us-west-2-korg-lkml-1.web.codeaurora.org Received: from kanga.kvack.org (kanga.kvack.org [205.233.56.17]) by smtp.lore.kernel.org (Postfix) with ESMTP id 8DD17C4829D for ; Thu, 8 Feb 2024 21:22:21 +0000 (UTC) Received: by kanga.kvack.org (Postfix) id DBF906B0081; Thu, 8 Feb 2024 16:22:20 -0500 (EST) Received: by kanga.kvack.org (Postfix, from userid 40) id D70D16B0082; Thu, 8 Feb 2024 16:22:20 -0500 (EST) X-Delivered-To: int-list-linux-mm@kvack.org Received: by kanga.kvack.org (Postfix, from userid 63042) id B73486B0083; Thu, 8 Feb 2024 16:22:20 -0500 (EST) X-Delivered-To: linux-mm@kvack.org Received: from relay.hostedemail.com (smtprelay0012.hostedemail.com [216.40.44.12]) by kanga.kvack.org (Postfix) with ESMTP id A05086B0081 for ; Thu, 8 Feb 2024 16:22:20 -0500 (EST) Received: from smtpin15.hostedemail.com (a10.router.float.18 [10.200.18.1]) by unirelay04.hostedemail.com (Postfix) with ESMTP id 2EAE41A0F4E for ; Thu, 8 Feb 2024 21:22:20 +0000 (UTC) X-FDA: 81769910040.15.58A4B43 Received: from mail-yw1-f202.google.com (mail-yw1-f202.google.com [209.85.128.202]) by imf12.hostedemail.com (Postfix) with ESMTP id 616964000E for ; Thu, 8 Feb 2024 21:22:18 +0000 (UTC) Authentication-Results: imf12.hostedemail.com; dkim=pass header.d=google.com header.s=20230601 header.b=h+06DZL9; dmarc=pass (policy=reject) header.from=google.com; spf=pass (imf12.hostedemail.com: domain of 3CUbFZQsKCN4LOKESHGIDRAGOOGLE.COMLINUX-MMKVACK.ORG@flex--lokeshgidra.bounces.google.com designates 209.85.128.202 as permitted sender) smtp.mailfrom=3CUbFZQsKCN4LOKESHGIDRAGOOGLE.COMLINUX-MMKVACK.ORG@flex--lokeshgidra.bounces.google.com ARC-Seal: i=1; s=arc-20220608; d=hostedemail.com; t=1707427338; a=rsa-sha256; cv=none; b=zvd7fakwzrTUYraVXKzkb31tf6mGiYbdcb5ecY1a62RwrsyAj0XfFm7l4stapFzYHTuvud 0i5M2m/y5YVWuY4NIh2fxunt204dQ7Iv+xVtR+8gY4qBPQa0zrwU/VKARAH/Cg4sAEMV4j 0MX4nOQp8KmYXkxcaHpZ2tUQxdxcDGw= ARC-Authentication-Results: i=1; imf12.hostedemail.com; dkim=pass header.d=google.com header.s=20230601 header.b=h+06DZL9; dmarc=pass (policy=reject) header.from=google.com; spf=pass (imf12.hostedemail.com: domain of 3CUbFZQsKCN4LOKESHGIDRAGOOGLE.COMLINUX-MMKVACK.ORG@flex--lokeshgidra.bounces.google.com designates 209.85.128.202 as permitted sender) smtp.mailfrom=3CUbFZQsKCN4LOKESHGIDRAGOOGLE.COMLINUX-MMKVACK.ORG@flex--lokeshgidra.bounces.google.com ARC-Message-Signature: i=1; a=rsa-sha256; c=relaxed/relaxed; d=hostedemail.com; s=arc-20220608; t=1707427338; h=from:from:sender:reply-to:subject:subject:date:date: message-id:message-id:to:to:cc:cc:mime-version:mime-version: content-type:content-type:content-transfer-encoding: in-reply-to:in-reply-to:references:references:dkim-signature; bh=lSLmVkRgWUoN7XoRPV1r5JJdZzlBEVE8ndv4C2H5Jh8=; b=6wHF4O7DTtw9Aq0Rz7VAMIp9iOD7fjEOTSeEoCFYbN9Vh7CJYABd5TbWN6cfW2Hptcmh+l Tr9g1hAv9PPGVLZSPy1kSEwVIYD+GHqHd3o2wgm3SGZa/2HvtzQk5XJM9cB8Kp6VnOykt1 +7SrgyUZaFWCQ3fHqhlzPRrVRgjx0AQ= Received: by mail-yw1-f202.google.com with SMTP id 00721157ae682-6047fed0132so5697077b3.1 for ; Thu, 08 Feb 2024 13:22:18 -0800 (PST) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=google.com; s=20230601; t=1707427337; x=1708032137; darn=kvack.org; h=cc:to:from:subject:message-id:references:mime-version:in-reply-to :date:from:to:cc:subject:date:message-id:reply-to; bh=lSLmVkRgWUoN7XoRPV1r5JJdZzlBEVE8ndv4C2H5Jh8=; b=h+06DZL9N12N8dRibl8YPIz8aaLDkRIfUAtaNbaCxaK3+whCTqkXMqTc3zQlwZMe5B TS4mZ+LFdFwX4Tvw9g6N5Zrz7Oh763jZxbSAAU6s2Hq3Y03PkfRYG8diDMEu/Bk3MUK/ YxlG79vov4x5pRFFXamCDVlromsQgzxVeaODJth806lEUs9B/ihmllQAQ/oA4bcfbI+V axppaFU2v5siS7DRPgaLB3wulEsTVLimkQaC2OCOrbNQLGj1x+SQBVFsKZUUfTEceizj SIvqnx0RLbdj0Tj6QSA65AQ7KX43brQz0pJCUm1SYQuzYctvy4T94ZgxfjgFPm+Mr6pe 3Ufg== X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20230601; t=1707427337; x=1708032137; h=cc:to:from:subject:message-id:references:mime-version:in-reply-to :date:x-gm-message-state:from:to:cc:subject:date:message-id:reply-to; bh=lSLmVkRgWUoN7XoRPV1r5JJdZzlBEVE8ndv4C2H5Jh8=; b=u5pJJuCdGbMZmtQJH0E7DFsI8OtHg0ni/s8C6ozv5ncu6748jWgxXkubp3/AR9WfHA o+PamJKeo8RlD6+E8D1CzmEmCT5Z/NyTu2Njm9P4BCYJ0tk2MDmjz4y747JRTqiWRZ45 TEMWm23h92W2hakF1Jwf0J7hcXevSt1Nd2SL3rgWObtvcuVWJQsX80SAtCG3X0m63wih R8tvGIAsCUYKCfRQWNPDXra8wjGLSZbQU4JzLXh/SjJHidXxzNtVoSusN0sl2vubM0WH HVjlMvuG3olC5iRjB9yb8T+p48HWW3SBGuqhiP4nvuu72rEjmhi5jMDHY37c9NQIW8KJ ox8A== X-Forwarded-Encrypted: i=1; AJvYcCXksZLa5Ct4sU2enHnDg3f89OaWTzhdDued0oejIH3vow3RmgxjXymybZJxgt9+KUqMXl6XwAb9fGw33fvjXj8syLY= X-Gm-Message-State: AOJu0Yy+sx53tEsPLjAnSGdjrVGzaotvwGmef+df+hwe7vT2V0HEUWrM ezNbQANGs2V/Mh9geoap5eNwfvFqSH+c+my7zqRx1Se3EDUWomeanhFGMma7KdSEIJUniEgtRAB tdcV0UHJGPY5PU2tns80/5A== X-Google-Smtp-Source: AGHT+IHpi2zp6f82761AcV8CyUSvOxGSqe6iXK7/Js/oY/YpNWS5eL45PoitVKDlSonNzu3IUBvopJOzuI0qCajr5g== X-Received: from lg.mtv.corp.google.com ([2620:15c:211:202:e9ba:42a8:6aba:f5d5]) (user=lokeshgidra job=sendgmr) by 2002:a81:a104:0:b0:5ee:d5ed:5f70 with SMTP id y4-20020a81a104000000b005eed5ed5f70mr93112ywg.9.1707427337633; Thu, 08 Feb 2024 13:22:17 -0800 (PST) Date: Thu, 8 Feb 2024 13:22:03 -0800 In-Reply-To: <20240208212204.2043140-1-lokeshgidra@google.com> Mime-Version: 1.0 References: <20240208212204.2043140-1-lokeshgidra@google.com> X-Mailer: git-send-email 2.43.0.687.g38aa6559b0-goog Message-ID: <20240208212204.2043140-3-lokeshgidra@google.com> Subject: [PATCH v4 2/3] userfaultfd: protect mmap_changing with rw_sem in userfaulfd_ctx From: Lokesh Gidra To: akpm@linux-foundation.org Cc: lokeshgidra@google.com, linux-fsdevel@vger.kernel.org, linux-mm@kvack.org, linux-kernel@vger.kernel.org, selinux@vger.kernel.org, surenb@google.com, kernel-team@android.com, aarcange@redhat.com, peterx@redhat.com, david@redhat.com, axelrasmussen@google.com, bgeffon@google.com, willy@infradead.org, jannh@google.com, kaleshsingh@google.com, ngeoffray@google.com, timmurray@google.com, rppt@kernel.org, Liam.Howlett@oracle.com X-Rspam-User: X-Rspamd-Server: rspam06 X-Rspamd-Queue-Id: 616964000E X-Stat-Signature: pmbfdnxd7y6d7i1f1x4oehsrakb95k5d X-HE-Tag: 1707427338-717239 X-HE-Meta: U2FsdGVkX19WSq0VsWXmEKyj34zzKk+QHihRYAG2xx6AaiBmNU5WTjP2TTndnIn+cZyuUhJ3EjmF+lUUKx6L2cc9P6MVx+VcBk4AQGHhVKnjvtPV8SjWiCikczWi/wExOsBd3XvVDm6FQxbJo26PlH7tdgPH2RYOyonOo9UdH4XlSDqXUVRLmChkUL0FFmAgYSLXOU0oxJFeCew24u0U4MH8fE5EBgm70VMXT75G71oDi1VknEzoeRunF6f2QBIVTGnO/IMz4pgLR7oTTgjVVn474qShP7zyx0dAHKjO0t0SPEg4s/XPTZojBqlyVjDeMRF8udZsngZscLPLNGHoj/wdUM5Ny9N0vJHMVrY4Ks9tq9piYJewwE12weKCe9Yakkdt/Ujl12ybS5qr55juo9g1qYT7TMRLZ3IPo4i4z31YPKkD4jSfGrKCgaCDHIJFjMPgVcXPXyr0jAOg8kIlOocfQoXgJISu0PgS++ewFk+IAW/BEv9gFlfOi5VaLyDHglbtEKtSvx3FyEx984YjqG+MZC32ztfIwjz61s0SKzpGFI1E2ga/nPkbghWHTM+LDZRkN3XXJaVoZ6ok0kSffXiMKLSEIPw+kSubTbmN4TMOGBrWy7DXMNfT0Rj3q9eb8EzD4N4VpsTMETzftCgKSy9I42klBD/qhKjL234Ja44bprvbwwjt4ATBn7Fwh4/iYojgxPKRXPIalGT1lCumMCmEwTN6WsNoqKwGu9I5ui0dH97epuZPqlJxafsdZukAlfKz455xBhxDodNxTVlBUKBYryRqZd9lGIJ38SwlSrcSqYYsWAxrg7M4mkJuYLgHDkk56aBoC38RsATW8n4GKwfu1nUa25bLdqkMfcL1dIc1J8ACrqZXJTiu044Aaappb906O0iAqkuVv8K8AgUtpSzwKMjiVrUcMg1HArBBBOpOFMwok94kaDU6sQneSs6u/CL3CbQRFDb4w8gsusl SGixrGBD yKi5fYVf7ZCHy+fY4xmJ0M05xWxkqueulZ93GhUpP9jYzYmp1h4np6g1D6RoK9RJJX2eZA6XuwT4j47HbJ/wjdnN/3KvFp0aZJUZnPFntYh/wLsnfjTY50fB9tB5L2+REKTIKVUxHf5csonDeouxFbLe330UceLZYPeNoXqo/czxOjmHrrG+BLPF3RleFJ8yoI16Absf82uprjxMvphG0XuKzB6kFwoHRKf+mwT9n8stGgd+lO5CKp1WCGNvYiHH4wIXt5oI9MSRk2dhrCno17QM3mEeYQPwQTn8OHo0B+1qWSV2tnz1IBqGrMcAkZTZl22zsH4uEmcjk1NkJvTMX/Q2ZBnGG5MIBoypRiInyyN6a7j3tvbH5kz9E06ven0kvmhLaK374wm9R/RBFo5osURNvypyJObi0op+Q230wewMvcAhyHCgv7D6QEE0omoGapZvj7NGcJPNkTw4xw5BMv2wUkDwbCXZX/ycuzTyoJKeCbhOBBbUAXFSbRxMADv047PoCrtXkMCbudIg4gWS8JZ5eMyBi147uSGZ+mHFdXn21tHDLQUkKwyAW3QuSHBPwNQMXpDPq1Z7Fv5Xr1GIA+FOwJUK/n6e/jb0RvotJqIFcWa5LXXrwgeTQeFU6UUXNkCDP4Sh2p5Bb3ty/wg5ZU2EMd6wNYoZyH6bboDWc+zZ9cEtOjL/D8wuArsd3a3cmvbjpZHPCDq6gTm0yEpuTNGsV5ko98s4xfwuy0ZVt90YeQGxNLxQArqajdQ== X-Bogosity: Ham, tests=bogofilter, spamicity=0.000000, version=1.2.4 Sender: owner-linux-mm@kvack.org Precedence: bulk X-Loop: owner-majordomo@kvack.org List-ID: List-Subscribe: List-Unsubscribe: Increments and loads to mmap_changing are always in mmap_lock critical section. This ensures that if userspace requests event notification for non-cooperative operations (e.g. mremap), userfaultfd operations don't occur concurrently. This can be achieved by using a separate read-write semaphore in userfaultfd_ctx such that increments are done in write-mode and loads in read-mode, thereby eliminating the dependency on mmap_lock for this purpose. This is a preparatory step before we replace mmap_lock usage with per-vma locks in fill/move ioctls. Signed-off-by: Lokesh Gidra Reviewed-by: Mike Rapoport (IBM) --- fs/userfaultfd.c | 40 ++++++++++++---------- include/linux/userfaultfd_k.h | 31 ++++++++++-------- mm/userfaultfd.c | 62 ++++++++++++++++++++--------------- 3 files changed, 75 insertions(+), 58 deletions(-) diff --git a/fs/userfaultfd.c b/fs/userfaultfd.c index 58331b83d648..c00a021bcce4 100644 --- a/fs/userfaultfd.c +++ b/fs/userfaultfd.c @@ -685,12 +685,15 @@ int dup_userfaultfd(struct vm_area_struct *vma, struct list_head *fcs) ctx->flags = octx->flags; ctx->features = octx->features; ctx->released = false; + init_rwsem(&ctx->map_changing_lock); atomic_set(&ctx->mmap_changing, 0); ctx->mm = vma->vm_mm; mmgrab(ctx->mm); userfaultfd_ctx_get(octx); + down_write(&octx->map_changing_lock); atomic_inc(&octx->mmap_changing); + up_write(&octx->map_changing_lock); fctx->orig = octx; fctx->new = ctx; list_add_tail(&fctx->list, fcs); @@ -737,7 +740,9 @@ void mremap_userfaultfd_prep(struct vm_area_struct *vma, if (ctx->features & UFFD_FEATURE_EVENT_REMAP) { vm_ctx->ctx = ctx; userfaultfd_ctx_get(ctx); + down_write(&ctx->map_changing_lock); atomic_inc(&ctx->mmap_changing); + up_write(&ctx->map_changing_lock); } else { /* Drop uffd context if remap feature not enabled */ vma_start_write(vma); @@ -783,7 +788,9 @@ bool userfaultfd_remove(struct vm_area_struct *vma, return true; userfaultfd_ctx_get(ctx); + down_write(&ctx->map_changing_lock); atomic_inc(&ctx->mmap_changing); + up_write(&ctx->map_changing_lock); mmap_read_unlock(mm); msg_init(&ewq.msg); @@ -825,7 +832,9 @@ int userfaultfd_unmap_prep(struct vm_area_struct *vma, unsigned long start, return -ENOMEM; userfaultfd_ctx_get(ctx); + down_write(&ctx->map_changing_lock); atomic_inc(&ctx->mmap_changing); + up_write(&ctx->map_changing_lock); unmap_ctx->ctx = ctx; unmap_ctx->start = start; unmap_ctx->end = end; @@ -1709,9 +1718,8 @@ static int userfaultfd_copy(struct userfaultfd_ctx *ctx, if (uffdio_copy.mode & UFFDIO_COPY_MODE_WP) flags |= MFILL_ATOMIC_WP; if (mmget_not_zero(ctx->mm)) { - ret = mfill_atomic_copy(ctx->mm, uffdio_copy.dst, uffdio_copy.src, - uffdio_copy.len, &ctx->mmap_changing, - flags); + ret = mfill_atomic_copy(ctx, uffdio_copy.dst, uffdio_copy.src, + uffdio_copy.len, flags); mmput(ctx->mm); } else { return -ESRCH; @@ -1761,9 +1769,8 @@ static int userfaultfd_zeropage(struct userfaultfd_ctx *ctx, goto out; if (mmget_not_zero(ctx->mm)) { - ret = mfill_atomic_zeropage(ctx->mm, uffdio_zeropage.range.start, - uffdio_zeropage.range.len, - &ctx->mmap_changing); + ret = mfill_atomic_zeropage(ctx, uffdio_zeropage.range.start, + uffdio_zeropage.range.len); mmput(ctx->mm); } else { return -ESRCH; @@ -1818,9 +1825,8 @@ static int userfaultfd_writeprotect(struct userfaultfd_ctx *ctx, return -EINVAL; if (mmget_not_zero(ctx->mm)) { - ret = mwriteprotect_range(ctx->mm, uffdio_wp.range.start, - uffdio_wp.range.len, mode_wp, - &ctx->mmap_changing); + ret = mwriteprotect_range(ctx, uffdio_wp.range.start, + uffdio_wp.range.len, mode_wp); mmput(ctx->mm); } else { return -ESRCH; @@ -1870,9 +1876,8 @@ static int userfaultfd_continue(struct userfaultfd_ctx *ctx, unsigned long arg) flags |= MFILL_ATOMIC_WP; if (mmget_not_zero(ctx->mm)) { - ret = mfill_atomic_continue(ctx->mm, uffdio_continue.range.start, - uffdio_continue.range.len, - &ctx->mmap_changing, flags); + ret = mfill_atomic_continue(ctx, uffdio_continue.range.start, + uffdio_continue.range.len, flags); mmput(ctx->mm); } else { return -ESRCH; @@ -1925,9 +1930,8 @@ static inline int userfaultfd_poison(struct userfaultfd_ctx *ctx, unsigned long goto out; if (mmget_not_zero(ctx->mm)) { - ret = mfill_atomic_poison(ctx->mm, uffdio_poison.range.start, - uffdio_poison.range.len, - &ctx->mmap_changing, 0); + ret = mfill_atomic_poison(ctx, uffdio_poison.range.start, + uffdio_poison.range.len, 0); mmput(ctx->mm); } else { return -ESRCH; @@ -2003,13 +2007,14 @@ static int userfaultfd_move(struct userfaultfd_ctx *ctx, if (mmget_not_zero(mm)) { mmap_read_lock(mm); - /* Re-check after taking mmap_lock */ + /* Re-check after taking map_changing_lock */ + down_read(&ctx->map_changing_lock); if (likely(!atomic_read(&ctx->mmap_changing))) ret = move_pages(ctx, mm, uffdio_move.dst, uffdio_move.src, uffdio_move.len, uffdio_move.mode); else ret = -EAGAIN; - + up_read(&ctx->map_changing_lock); mmap_read_unlock(mm); mmput(mm); } else { @@ -2216,6 +2221,7 @@ static int new_userfaultfd(int flags) ctx->flags = flags; ctx->features = 0; ctx->released = false; + init_rwsem(&ctx->map_changing_lock); atomic_set(&ctx->mmap_changing, 0); ctx->mm = current->mm; /* prevent the mm struct to be freed */ diff --git a/include/linux/userfaultfd_k.h b/include/linux/userfaultfd_k.h index 691d928ee864..3210c3552976 100644 --- a/include/linux/userfaultfd_k.h +++ b/include/linux/userfaultfd_k.h @@ -69,6 +69,13 @@ struct userfaultfd_ctx { unsigned int features; /* released */ bool released; + /* + * Prevents userfaultfd operations (fill/move/wp) from happening while + * some non-cooperative event(s) is taking place. Increments are done + * in write-mode. Whereas, userfaultfd operations, which includes + * reading mmap_changing, is done under read-mode. + */ + struct rw_semaphore map_changing_lock; /* memory mappings are changing because of non-cooperative event */ atomic_t mmap_changing; /* mm with one ore more vmas attached to this userfaultfd_ctx */ @@ -113,22 +120,18 @@ extern int mfill_atomic_install_pte(pmd_t *dst_pmd, unsigned long dst_addr, struct page *page, bool newly_allocated, uffd_flags_t flags); -extern ssize_t mfill_atomic_copy(struct mm_struct *dst_mm, unsigned long dst_start, +extern ssize_t mfill_atomic_copy(struct userfaultfd_ctx *ctx, unsigned long dst_start, unsigned long src_start, unsigned long len, - atomic_t *mmap_changing, uffd_flags_t flags); -extern ssize_t mfill_atomic_zeropage(struct mm_struct *dst_mm, + uffd_flags_t flags); +extern ssize_t mfill_atomic_zeropage(struct userfaultfd_ctx *ctx, unsigned long dst_start, - unsigned long len, - atomic_t *mmap_changing); -extern ssize_t mfill_atomic_continue(struct mm_struct *dst_mm, unsigned long dst_start, - unsigned long len, atomic_t *mmap_changing, - uffd_flags_t flags); -extern ssize_t mfill_atomic_poison(struct mm_struct *dst_mm, unsigned long start, - unsigned long len, atomic_t *mmap_changing, - uffd_flags_t flags); -extern int mwriteprotect_range(struct mm_struct *dst_mm, - unsigned long start, unsigned long len, - bool enable_wp, atomic_t *mmap_changing); + unsigned long len); +extern ssize_t mfill_atomic_continue(struct userfaultfd_ctx *ctx, unsigned long dst_start, + unsigned long len, uffd_flags_t flags); +extern ssize_t mfill_atomic_poison(struct userfaultfd_ctx *ctx, unsigned long start, + unsigned long len, uffd_flags_t flags); +extern int mwriteprotect_range(struct userfaultfd_ctx *ctx, unsigned long start, + unsigned long len, bool enable_wp); extern long uffd_wp_range(struct vm_area_struct *vma, unsigned long start, unsigned long len, bool enable_wp); diff --git a/mm/userfaultfd.c b/mm/userfaultfd.c index 9cc93cc1330b..74aad0831e40 100644 --- a/mm/userfaultfd.c +++ b/mm/userfaultfd.c @@ -353,11 +353,11 @@ static pmd_t *mm_alloc_pmd(struct mm_struct *mm, unsigned long address) * called with mmap_lock held, it will release mmap_lock before returning. */ static __always_inline ssize_t mfill_atomic_hugetlb( + struct userfaultfd_ctx *ctx, struct vm_area_struct *dst_vma, unsigned long dst_start, unsigned long src_start, unsigned long len, - atomic_t *mmap_changing, uffd_flags_t flags) { struct mm_struct *dst_mm = dst_vma->vm_mm; @@ -379,6 +379,7 @@ static __always_inline ssize_t mfill_atomic_hugetlb( * feature is not supported. */ if (uffd_flags_mode_is(flags, MFILL_ATOMIC_ZEROPAGE)) { + up_read(&ctx->map_changing_lock); mmap_read_unlock(dst_mm); return -EINVAL; } @@ -463,6 +464,7 @@ static __always_inline ssize_t mfill_atomic_hugetlb( cond_resched(); if (unlikely(err == -ENOENT)) { + up_read(&ctx->map_changing_lock); mmap_read_unlock(dst_mm); BUG_ON(!folio); @@ -473,12 +475,13 @@ static __always_inline ssize_t mfill_atomic_hugetlb( goto out; } mmap_read_lock(dst_mm); + down_read(&ctx->map_changing_lock); /* * If memory mappings are changing because of non-cooperative * operation (e.g. mremap) running in parallel, bail out and * request the user to retry later */ - if (mmap_changing && atomic_read(mmap_changing)) { + if (atomic_read(&ctx->mmap_changing)) { err = -EAGAIN; break; } @@ -501,6 +504,7 @@ static __always_inline ssize_t mfill_atomic_hugetlb( } out_unlock: + up_read(&ctx->map_changing_lock); mmap_read_unlock(dst_mm); out: if (folio) @@ -512,11 +516,11 @@ static __always_inline ssize_t mfill_atomic_hugetlb( } #else /* !CONFIG_HUGETLB_PAGE */ /* fail at build time if gcc attempts to use this */ -extern ssize_t mfill_atomic_hugetlb(struct vm_area_struct *dst_vma, +extern ssize_t mfill_atomic_hugetlb(struct userfaultfd_ctx *ctx, + struct vm_area_struct *dst_vma, unsigned long dst_start, unsigned long src_start, unsigned long len, - atomic_t *mmap_changing, uffd_flags_t flags); #endif /* CONFIG_HUGETLB_PAGE */ @@ -564,13 +568,13 @@ static __always_inline ssize_t mfill_atomic_pte(pmd_t *dst_pmd, return err; } -static __always_inline ssize_t mfill_atomic(struct mm_struct *dst_mm, +static __always_inline ssize_t mfill_atomic(struct userfaultfd_ctx *ctx, unsigned long dst_start, unsigned long src_start, unsigned long len, - atomic_t *mmap_changing, uffd_flags_t flags) { + struct mm_struct *dst_mm = ctx->mm; struct vm_area_struct *dst_vma; ssize_t err; pmd_t *dst_pmd; @@ -600,8 +604,9 @@ static __always_inline ssize_t mfill_atomic(struct mm_struct *dst_mm, * operation (e.g. mremap) running in parallel, bail out and * request the user to retry later */ + down_read(&ctx->map_changing_lock); err = -EAGAIN; - if (mmap_changing && atomic_read(mmap_changing)) + if (atomic_read(&ctx->mmap_changing)) goto out_unlock; /* @@ -633,8 +638,8 @@ static __always_inline ssize_t mfill_atomic(struct mm_struct *dst_mm, * If this is a HUGETLB vma, pass off to appropriate routine */ if (is_vm_hugetlb_page(dst_vma)) - return mfill_atomic_hugetlb(dst_vma, dst_start, src_start, - len, mmap_changing, flags); + return mfill_atomic_hugetlb(ctx, dst_vma, dst_start, + src_start, len, flags); if (!vma_is_anonymous(dst_vma) && !vma_is_shmem(dst_vma)) goto out_unlock; @@ -693,6 +698,7 @@ static __always_inline ssize_t mfill_atomic(struct mm_struct *dst_mm, if (unlikely(err == -ENOENT)) { void *kaddr; + up_read(&ctx->map_changing_lock); mmap_read_unlock(dst_mm); BUG_ON(!folio); @@ -723,6 +729,7 @@ static __always_inline ssize_t mfill_atomic(struct mm_struct *dst_mm, } out_unlock: + up_read(&ctx->map_changing_lock); mmap_read_unlock(dst_mm); out: if (folio) @@ -733,34 +740,33 @@ static __always_inline ssize_t mfill_atomic(struct mm_struct *dst_mm, return copied ? copied : err; } -ssize_t mfill_atomic_copy(struct mm_struct *dst_mm, unsigned long dst_start, +ssize_t mfill_atomic_copy(struct userfaultfd_ctx *ctx, unsigned long dst_start, unsigned long src_start, unsigned long len, - atomic_t *mmap_changing, uffd_flags_t flags) + uffd_flags_t flags) { - return mfill_atomic(dst_mm, dst_start, src_start, len, mmap_changing, + return mfill_atomic(ctx, dst_start, src_start, len, uffd_flags_set_mode(flags, MFILL_ATOMIC_COPY)); } -ssize_t mfill_atomic_zeropage(struct mm_struct *dst_mm, unsigned long start, - unsigned long len, atomic_t *mmap_changing) +ssize_t mfill_atomic_zeropage(struct userfaultfd_ctx *ctx, + unsigned long start, + unsigned long len) { - return mfill_atomic(dst_mm, start, 0, len, mmap_changing, + return mfill_atomic(ctx, start, 0, len, uffd_flags_set_mode(0, MFILL_ATOMIC_ZEROPAGE)); } -ssize_t mfill_atomic_continue(struct mm_struct *dst_mm, unsigned long start, - unsigned long len, atomic_t *mmap_changing, - uffd_flags_t flags) +ssize_t mfill_atomic_continue(struct userfaultfd_ctx *ctx, unsigned long start, + unsigned long len, uffd_flags_t flags) { - return mfill_atomic(dst_mm, start, 0, len, mmap_changing, + return mfill_atomic(ctx, start, 0, len, uffd_flags_set_mode(flags, MFILL_ATOMIC_CONTINUE)); } -ssize_t mfill_atomic_poison(struct mm_struct *dst_mm, unsigned long start, - unsigned long len, atomic_t *mmap_changing, - uffd_flags_t flags) +ssize_t mfill_atomic_poison(struct userfaultfd_ctx *ctx, unsigned long start, + unsigned long len, uffd_flags_t flags) { - return mfill_atomic(dst_mm, start, 0, len, mmap_changing, + return mfill_atomic(ctx, start, 0, len, uffd_flags_set_mode(flags, MFILL_ATOMIC_POISON)); } @@ -793,10 +799,10 @@ long uffd_wp_range(struct vm_area_struct *dst_vma, return ret; } -int mwriteprotect_range(struct mm_struct *dst_mm, unsigned long start, - unsigned long len, bool enable_wp, - atomic_t *mmap_changing) +int mwriteprotect_range(struct userfaultfd_ctx *ctx, unsigned long start, + unsigned long len, bool enable_wp) { + struct mm_struct *dst_mm = ctx->mm; unsigned long end = start + len; unsigned long _start, _end; struct vm_area_struct *dst_vma; @@ -820,8 +826,9 @@ int mwriteprotect_range(struct mm_struct *dst_mm, unsigned long start, * operation (e.g. mremap) running in parallel, bail out and * request the user to retry later */ + down_read(&ctx->map_changing_lock); err = -EAGAIN; - if (mmap_changing && atomic_read(mmap_changing)) + if (atomic_read(&ctx->mmap_changing)) goto out_unlock; err = -ENOENT; @@ -850,6 +857,7 @@ int mwriteprotect_range(struct mm_struct *dst_mm, unsigned long start, err = 0; } out_unlock: + up_read(&ctx->map_changing_lock); mmap_read_unlock(dst_mm); return err; } From patchwork Thu Feb 8 21:22:04 2024 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: Lokesh Gidra X-Patchwork-Id: 13550494 Return-Path: X-Spam-Checker-Version: SpamAssassin 3.4.0 (2014-02-07) on aws-us-west-2-korg-lkml-1.web.codeaurora.org Received: from kanga.kvack.org (kanga.kvack.org [205.233.56.17]) by smtp.lore.kernel.org (Postfix) with ESMTP id 71EB5C48260 for ; Thu, 8 Feb 2024 21:22:24 +0000 (UTC) Received: by kanga.kvack.org (Postfix) id 22A696B0083; Thu, 8 Feb 2024 16:22:23 -0500 (EST) Received: by kanga.kvack.org (Postfix, from userid 40) id 1A3E96B0087; Thu, 8 Feb 2024 16:22:23 -0500 (EST) X-Delivered-To: int-list-linux-mm@kvack.org Received: by kanga.kvack.org (Postfix, from userid 63042) id E326D6B0085; Thu, 8 Feb 2024 16:22:22 -0500 (EST) X-Delivered-To: linux-mm@kvack.org Received: from relay.hostedemail.com (smtprelay0017.hostedemail.com [216.40.44.17]) by kanga.kvack.org (Postfix) with ESMTP id CD3D76B0082 for ; Thu, 8 Feb 2024 16:22:22 -0500 (EST) Received: from smtpin26.hostedemail.com (a10.router.float.18 [10.200.18.1]) by unirelay06.hostedemail.com (Postfix) with ESMTP id 9F8C0A25FD for ; Thu, 8 Feb 2024 21:22:22 +0000 (UTC) X-FDA: 81769910124.26.D2F02B7 Received: from mail-yw1-f202.google.com (mail-yw1-f202.google.com [209.85.128.202]) by imf16.hostedemail.com (Postfix) with ESMTP id DC510180012 for ; Thu, 8 Feb 2024 21:22:20 +0000 (UTC) Authentication-Results: imf16.hostedemail.com; dkim=pass header.d=google.com header.s=20230601 header.b=Y595tjaP; dmarc=pass (policy=reject) header.from=google.com; spf=pass (imf16.hostedemail.com: domain of 3C0bFZQsKCOANQMGUJIKFTCIQQING.EQONKPWZ-OOMXCEM.QTI@flex--lokeshgidra.bounces.google.com designates 209.85.128.202 as permitted sender) smtp.mailfrom=3C0bFZQsKCOANQMGUJIKFTCIQQING.EQONKPWZ-OOMXCEM.QTI@flex--lokeshgidra.bounces.google.com ARC-Message-Signature: i=1; a=rsa-sha256; c=relaxed/relaxed; d=hostedemail.com; s=arc-20220608; t=1707427340; h=from:from:sender:reply-to:subject:subject:date:date: message-id:message-id:to:to:cc:cc:mime-version:mime-version: content-type:content-type:content-transfer-encoding: in-reply-to:in-reply-to:references:references:dkim-signature; bh=t75SKxYdfQlCW6bpie1fAW4ju+PGfumrxcdqts4g8gw=; b=8okWU1Ph+Cc264HgIhrACR9KMl9nQGnrvbPXOuiIqo2gfz4BTAOEMl+Wje18Q4I9xTBJq9 qnNyNewcnQZXJFnk2rU1u3QVHnZb6AYTwi/qRRM2+cpo4rJgQkqXfIC2QsaLsKZnCfh80p 81sBXgIJ4WUq8cQdL2Ew+NZEZ+DmQa0= ARC-Authentication-Results: i=1; imf16.hostedemail.com; dkim=pass header.d=google.com header.s=20230601 header.b=Y595tjaP; dmarc=pass (policy=reject) header.from=google.com; spf=pass (imf16.hostedemail.com: domain of 3C0bFZQsKCOANQMGUJIKFTCIQQING.EQONKPWZ-OOMXCEM.QTI@flex--lokeshgidra.bounces.google.com designates 209.85.128.202 as permitted sender) smtp.mailfrom=3C0bFZQsKCOANQMGUJIKFTCIQQING.EQONKPWZ-OOMXCEM.QTI@flex--lokeshgidra.bounces.google.com ARC-Seal: i=1; s=arc-20220608; d=hostedemail.com; t=1707427340; a=rsa-sha256; cv=none; b=mHbgCcjx4zMKR4IVa7AeUOzqQST0OZNsnCt1M8G5kns4+Pqg5pVi5kAzvXLAd0k6rv0VFS 8BRg4BBoOpus3kyJEGgB+1gFXnRut8UCaPUx0s9jf98os9iwthEUR2U5V2GCUnJFF+X5sV t/UYapGW6ENUBD8O9sRIdBZLZ8bxFk8= Received: by mail-yw1-f202.google.com with SMTP id 00721157ae682-6047fed0132so5697787b3.1 for ; Thu, 08 Feb 2024 13:22:20 -0800 (PST) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=google.com; s=20230601; t=1707427340; x=1708032140; darn=kvack.org; h=cc:to:from:subject:message-id:references:mime-version:in-reply-to :date:from:to:cc:subject:date:message-id:reply-to; bh=t75SKxYdfQlCW6bpie1fAW4ju+PGfumrxcdqts4g8gw=; b=Y595tjaP2VbaI87gwEIPtBsqXYFNNSZ1fwW5P7lUd1fOGp1ywVsetx7d6M1XVVwtho nAxCQWPM3ka8ZkeP64f88BWcSqEQWFJjJsCoUy7su2qXLbKWaikl6RA3EfnaBwDTRwYc kHd3YnV7KWJPJ7uJ62iWpD/mwbMxHp4BsxEHQxU1m99jPFhhr71WjUStcN9hovWdXLIh 97DJ0VlkjsZMjzTGnQ+UsQS5OgKDy/r9OXADykwPJ9nmlhyBg99zCXWvLqvToprOH/nR 643DUvq0nHManym8gvF01AAKOptFek8PBmn8mKPKvb3nu7iD8wQP2uqsL11HILWAXEgK sKiQ== X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20230601; t=1707427340; x=1708032140; h=cc:to:from:subject:message-id:references:mime-version:in-reply-to :date:x-gm-message-state:from:to:cc:subject:date:message-id:reply-to; bh=t75SKxYdfQlCW6bpie1fAW4ju+PGfumrxcdqts4g8gw=; b=TYJ8wdPSBIme4nzmF1OLSWI49O17N9yjl3XNbK4nOEzaP7jd3DiJwvzO2wsbKWx1T7 vAmkX1+y3kCSv3iUliQ/pQptHzygXBCorHFD9CNidExK84kqiR0CbNFkPevbV7wgJ/KN NMC/DRnNbfGAw7gr3yHFo69bEqKqR9PYtbtRFyF/kD8jl1BZK3WcI/Q18rdJvosXrHYT wpcBqxh1jMK9d3JREtuXxhc6XWeIQW3aT0N4Nl0tL8QzHlAvOS4oO+awRokF+vrh3jA5 33IMMa8JN4rlOG3Ambxeuqg0oacCOGLWPnlVfvoHjIK1pB2zGzH9wxWBSrxKJ232pF+J AI/Q== X-Gm-Message-State: AOJu0Yzyltm0Xge5/Rf/YVEHap593KDqqkCUvUnmqq8HndWpsBrF7WEv HQMlvPBn0D8Jf3HJ9/UrjrgY/tZPvxTwjuVG04p+SUGr4JjSuft+/lGlAMgv3lYIsvSnfu2k0Qt NQroAz1piPGR3XMLBV3qYtg== X-Google-Smtp-Source: AGHT+IFAIVJj1WYfDvCAzIltMCiiv0Q1ue38dCbij931BC4Jbgs1PrxtKM7Z6HsgiVUMnFVeZrVZNb8xsxAHJCEJNQ== X-Received: from lg.mtv.corp.google.com ([2620:15c:211:202:e9ba:42a8:6aba:f5d5]) (user=lokeshgidra job=sendgmr) by 2002:a81:49c9:0:b0:5e8:bea4:4d3b with SMTP id w192-20020a8149c9000000b005e8bea44d3bmr93963ywa.6.1707427339968; Thu, 08 Feb 2024 13:22:19 -0800 (PST) Date: Thu, 8 Feb 2024 13:22:04 -0800 In-Reply-To: <20240208212204.2043140-1-lokeshgidra@google.com> Mime-Version: 1.0 References: <20240208212204.2043140-1-lokeshgidra@google.com> X-Mailer: git-send-email 2.43.0.687.g38aa6559b0-goog Message-ID: <20240208212204.2043140-4-lokeshgidra@google.com> Subject: [PATCH v4 3/3] userfaultfd: use per-vma locks in userfaultfd operations From: Lokesh Gidra To: akpm@linux-foundation.org Cc: lokeshgidra@google.com, linux-fsdevel@vger.kernel.org, linux-mm@kvack.org, linux-kernel@vger.kernel.org, selinux@vger.kernel.org, surenb@google.com, kernel-team@android.com, aarcange@redhat.com, peterx@redhat.com, david@redhat.com, axelrasmussen@google.com, bgeffon@google.com, willy@infradead.org, jannh@google.com, kaleshsingh@google.com, ngeoffray@google.com, timmurray@google.com, rppt@kernel.org, Liam.Howlett@oracle.com X-Rspamd-Queue-Id: DC510180012 X-Rspam-User: X-Rspamd-Server: rspam05 X-Stat-Signature: 9ijmh3naoq7wbxgi9efed3chc5jksx69 X-HE-Tag: 1707427340-731421 X-HE-Meta: U2FsdGVkX1+HCOqrREVGUGuYy3bKvh7CM5FRPhWZToLATRV17ZJ8fr5cdCWXATKbUj9PPsKhnzhC6wm9sIBUXxshkLSB0gX6GXZh6S9rvyIg5BJltIH0xOJ81FwwE2gEMAWf0u5XN8Y0+1MlOlFmL41kZJsgoIL6e3rIfstXL1TASR1kIPRakgd/2SknGCOodECuZ00s2utswMvKOsIpU2u5janNiRr7u81+UXyL/75x73DazFN5nN8XeoRw7DGaYr5qiQGkmsY07vmwlnBvhwCUeGFTEpO0x9xjiJa/zCPmK4kWXdnzh5DfdxOyV2nbdJpXkwRTCqglKITpDzU4Te/PE1q5Gs5JM8IEwuzEllGO8WgHBEuMeZeaOZmHr3VxC4hfthSg2Vx7m8aPpcjiyb35+Ae2RIEHiaRxRwL2RNUZB8yxyZVIqDB7K2wej5PAuoQYwScQtmQjkhggPJz6B6l7oj5p9CEchQ/ABDwhzSaLsq9/AgtUtmlYH9Smxq0iPu6V44G+mtQYOSqNF1ooh30m7y3Kk66MLTm6JIIceZIkPAQ762Fa8ncFE7KlJcHAPOCkhigV/9HGNZMDetQrUx8v2CKRw9q09gB7j8mcpRZHHOHjP74KLlF94RjXMWvAoPjpZIiQuvRLYVu7RRHii05p4m0jmPv1dLCkUemw2DztX8irbwj2+ledRc+GnCp2zaFQBGJhWa3oyC8Gr/pMI4w1Q4Ss+ub8foX1Ab0DyIvwXx1WMXbhXsHUxS5/kXc7kd/Hngs0XpBHOWvlRE/9aaR2atLgyZ9/5sKRFu4IsfcidfQVLMTvyXpdxyq+JbXs/Gv4Xh/h8mtwjEk7zIYcXtbKbwzimazawIeUzplsPEp1nUYqbkoZme/G0b6Y8ToAl6xZk/E3t0/+LNfBk72DiRPWb/pC/Wfes3Q8+WoSLLKr13kHt+HFPvqq4ENoxoFKmlTVQzE+z3xSqFG8YuH p8TRFtDN V9nb4XEwF2uShitRsl8ZDFP1xONk1l3as1Pj7l9t9dhuTOKNSDiLwhUhFEn/AUapQ/PIBhWfjehu7vxObJPdLLfJPb7iZul8myUmf7vUxN+QS/Yg6c70KtW7MB6ZG9T9QHsAN8UQP6V+nuYWCAdDS8SC4MaDm8WwXfgGlyCnMZYqwl+FfRy8RwsExfZbq4bLFqsAmFxrAr2HmS1ry7J2nD2bKNtSycjVqzPzlxqJNThiwsCI56SfiKLnKFTNV5LSAlp5xDbBKNVHv+RXJppqGBmBpS5qLGGFAEOQNDpU7N6Vm+YqvGCL39w7MHppJdTYfYgtCh4NrWudWPb+xx0ejpt7S23H1h5ehux2LSVVEVxLP37HYPRlk5z6GqDEClPlWXsDHUD6GsbSq7k3VhEJSUqZ8cOaMs9GQRksSO1tT8zro6gUdqgLx2VGFWUFXYTXMqXwhkDnC0MhG5KypxQgrnggNQj0UHukAuXz8KgsnYYyb6aXIxtJax8iwVRSweFY8UrKXQyEiwq2fNfnOjE3sXiiTZxTeGewlvZ399/RPgcAVFYRiOtWiqq/vezhgIE7GeVwGeC6KO9ZAyw3uaz5D5qIaJWa4VPZNpZIzzQNzmvAjSkAmQ49wIP76jFYROHoFc+vpFgYqJk90CxOslSLJN/cdkQ== X-Bogosity: Ham, tests=bogofilter, spamicity=0.000000, version=1.2.4 Sender: owner-linux-mm@kvack.org Precedence: bulk X-Loop: owner-majordomo@kvack.org List-ID: List-Subscribe: List-Unsubscribe: All userfaultfd operations, except write-protect, opportunistically use per-vma locks to lock vmas. On failure, attempt again inside mmap_lock critical section. Write-protect operation requires mmap_lock as it iterates over multiple vmas. Signed-off-by: Lokesh Gidra --- fs/userfaultfd.c | 13 +- include/linux/userfaultfd_k.h | 5 +- mm/userfaultfd.c | 356 ++++++++++++++++++++++++++-------- 3 files changed, 275 insertions(+), 99 deletions(-) diff --git a/fs/userfaultfd.c b/fs/userfaultfd.c index c00a021bcce4..60dcfafdc11a 100644 --- a/fs/userfaultfd.c +++ b/fs/userfaultfd.c @@ -2005,17 +2005,8 @@ static int userfaultfd_move(struct userfaultfd_ctx *ctx, return -EINVAL; if (mmget_not_zero(mm)) { - mmap_read_lock(mm); - - /* Re-check after taking map_changing_lock */ - down_read(&ctx->map_changing_lock); - if (likely(!atomic_read(&ctx->mmap_changing))) - ret = move_pages(ctx, mm, uffdio_move.dst, uffdio_move.src, - uffdio_move.len, uffdio_move.mode); - else - ret = -EAGAIN; - up_read(&ctx->map_changing_lock); - mmap_read_unlock(mm); + ret = move_pages(ctx, uffdio_move.dst, uffdio_move.src, + uffdio_move.len, uffdio_move.mode); mmput(mm); } else { return -ESRCH; diff --git a/include/linux/userfaultfd_k.h b/include/linux/userfaultfd_k.h index 3210c3552976..05d59f74fc88 100644 --- a/include/linux/userfaultfd_k.h +++ b/include/linux/userfaultfd_k.h @@ -138,9 +138,8 @@ extern long uffd_wp_range(struct vm_area_struct *vma, /* move_pages */ void double_pt_lock(spinlock_t *ptl1, spinlock_t *ptl2); void double_pt_unlock(spinlock_t *ptl1, spinlock_t *ptl2); -ssize_t move_pages(struct userfaultfd_ctx *ctx, struct mm_struct *mm, - unsigned long dst_start, unsigned long src_start, - unsigned long len, __u64 flags); +ssize_t move_pages(struct userfaultfd_ctx *ctx, unsigned long dst_start, + unsigned long src_start, unsigned long len, __u64 flags); int move_pages_huge_pmd(struct mm_struct *mm, pmd_t *dst_pmd, pmd_t *src_pmd, pmd_t dst_pmdval, struct vm_area_struct *dst_vma, struct vm_area_struct *src_vma, diff --git a/mm/userfaultfd.c b/mm/userfaultfd.c index 74aad0831e40..1e25768b2136 100644 --- a/mm/userfaultfd.c +++ b/mm/userfaultfd.c @@ -19,20 +19,12 @@ #include #include "internal.h" -static __always_inline -struct vm_area_struct *find_dst_vma(struct mm_struct *dst_mm, - unsigned long dst_start, - unsigned long len) +static bool validate_dst_vma(struct vm_area_struct *dst_vma, + unsigned long dst_end) { - /* - * Make sure that the dst range is both valid and fully within a - * single existing vma. - */ - struct vm_area_struct *dst_vma; - - dst_vma = find_vma(dst_mm, dst_start); - if (!range_in_vma(dst_vma, dst_start, dst_start + len)) - return NULL; + /* Make sure that the dst range is fully within dst_vma. */ + if (dst_end > dst_vma->vm_end) + return false; /* * Check the vma is registered in uffd, this is required to @@ -40,11 +32,125 @@ struct vm_area_struct *find_dst_vma(struct mm_struct *dst_mm, * time. */ if (!dst_vma->vm_userfaultfd_ctx.ctx) - return NULL; + return false; + + return true; +} + +#ifdef CONFIG_PER_VMA_LOCK +/* + * lock_vma() - Lookup and lock vma corresponding to @address. + * @mm: mm to search vma in. + * @address: address that the vma should contain. + * @prepare_anon: If true, then prepare the vma (if private) with anon_vma. + * + * Should be called without holding mmap_lock. vma should be unlocked after use + * with unlock_vma(). + * + * Return: A locked vma containing @address, NULL if no vma is found, or + * -ENOMEM if anon_vma couldn't be allocated. + */ +static struct vm_area_struct *lock_vma(struct mm_struct *mm, + unsigned long address, + bool prepare_anon) +{ + struct vm_area_struct *vma; + + vma = lock_vma_under_rcu(mm, address); + if (vma) { + /* + * lock_vma_under_rcu() only checks anon_vma for private + * anonymous mappings. But we need to ensure it is assigned in + * private file-backed vmas as well. + */ + if (prepare_anon && !(vma->vm_flags & VM_SHARED) && + !vma->anon_vma) + vma_end_read(vma); + else + return vma; + } + + mmap_read_lock(mm); + vma = vma_lookup(mm, address); + if (vma) { + if (prepare_anon && !(vma->vm_flags & VM_SHARED) && + anon_vma_prepare(vma)) { + vma = ERR_PTR(-ENOMEM); + } else { + /* + * We cannot use vma_start_read() as it may fail due to + * false locked (see comment in vma_start_read()). We + * can avoid that by directly locking vm_lock under + * mmap_lock, which guarantees that nobody can lock the + * vma for write (vma_start_write()) under us. + */ + down_read(&vma->vm_lock->lock); + } + } + + mmap_read_unlock(mm); + return vma; +} + +static void unlock_vma(struct vm_area_struct *vma) +{ + vma_end_read(vma); +} + +static struct vm_area_struct *find_and_lock_dst_vma(struct mm_struct *dst_mm, + unsigned long dst_start, + unsigned long len) +{ + struct vm_area_struct *dst_vma; + + /* Ensure anon_vma is assigned for private vmas */ + dst_vma = lock_vma(dst_mm, dst_start, true); + + if (!dst_vma) + return ERR_PTR(-ENOENT); + + if (PTR_ERR(dst_vma) == -ENOMEM) + return dst_vma; + + if (!validate_dst_vma(dst_vma, dst_start + len)) + goto out_unlock; return dst_vma; +out_unlock: + unlock_vma(dst_vma); + return ERR_PTR(-ENOENT); } +#else + +static struct vm_area_struct *lock_mm_and_find_dst_vma(struct mm_struct *dst_mm, + unsigned long dst_start, + unsigned long len) +{ + struct vm_area_struct *dst_vma; + int err = -ENOENT; + + mmap_read_lock(dst_mm); + dst_vma = vma_lookup(dst_mm, dst_start); + if (!dst_vma) + goto out_unlock; + + /* Ensure anon_vma is assigned for private vmas */ + if (!(dst_vma->vm_flags & VM_SHARED) && anon_vma_prepare(dst_vma)) { + err = -ENOMEM; + goto out_unlock; + } + + if (!validate_dst_vma(dst_vma, dst_start + len)) + goto out_unlock; + + return dst_vma; +out_unlock: + mmap_read_unlock(dst_mm); + return ERR_PTR(err); +} +#endif + /* Check if dst_addr is outside of file's size. Must be called with ptl held. */ static bool mfill_file_over_size(struct vm_area_struct *dst_vma, unsigned long dst_addr) @@ -350,7 +456,8 @@ static pmd_t *mm_alloc_pmd(struct mm_struct *mm, unsigned long address) #ifdef CONFIG_HUGETLB_PAGE /* * mfill_atomic processing for HUGETLB vmas. Note that this routine is - * called with mmap_lock held, it will release mmap_lock before returning. + * called with either vma-lock or mmap_lock held, it will release the lock + * before returning. */ static __always_inline ssize_t mfill_atomic_hugetlb( struct userfaultfd_ctx *ctx, @@ -361,7 +468,6 @@ static __always_inline ssize_t mfill_atomic_hugetlb( uffd_flags_t flags) { struct mm_struct *dst_mm = dst_vma->vm_mm; - int vm_shared = dst_vma->vm_flags & VM_SHARED; ssize_t err; pte_t *dst_pte; unsigned long src_addr, dst_addr; @@ -380,7 +486,11 @@ static __always_inline ssize_t mfill_atomic_hugetlb( */ if (uffd_flags_mode_is(flags, MFILL_ATOMIC_ZEROPAGE)) { up_read(&ctx->map_changing_lock); +#ifdef CONFIG_PER_VMA_LOCK + unlock_vma(dst_vma); +#else mmap_read_unlock(dst_mm); +#endif return -EINVAL; } @@ -403,24 +513,32 @@ static __always_inline ssize_t mfill_atomic_hugetlb( * retry, dst_vma will be set to NULL and we must lookup again. */ if (!dst_vma) { +#ifdef CONFIG_PER_VMA_LOCK + dst_vma = find_and_lock_dst_vma(dst_mm, dst_start, len); +#else + dst_vma = lock_mm_and_find_dst_vma(dst_mm, dst_start, len); +#endif + if (IS_ERR(dst_vma)) { + err = PTR_ERR(dst_vma); + goto out; + } + err = -ENOENT; - dst_vma = find_dst_vma(dst_mm, dst_start, len); - if (!dst_vma || !is_vm_hugetlb_page(dst_vma)) - goto out_unlock; + if (!is_vm_hugetlb_page(dst_vma)) + goto out_unlock_vma; err = -EINVAL; if (vma_hpagesize != vma_kernel_pagesize(dst_vma)) - goto out_unlock; - - vm_shared = dst_vma->vm_flags & VM_SHARED; - } + goto out_unlock_vma; - /* - * If not shared, ensure the dst_vma has a anon_vma. - */ - err = -ENOMEM; - if (!vm_shared) { - if (unlikely(anon_vma_prepare(dst_vma))) + /* + * If memory mappings are changing because of non-cooperative + * operation (e.g. mremap) running in parallel, bail out and + * request the user to retry later + */ + down_read(&ctx->map_changing_lock); + err = -EAGAIN; + if (atomic_read(&ctx->mmap_changing)) goto out_unlock; } @@ -465,7 +583,11 @@ static __always_inline ssize_t mfill_atomic_hugetlb( if (unlikely(err == -ENOENT)) { up_read(&ctx->map_changing_lock); +#ifdef CONFIG_PER_VMA_LOCK + unlock_vma(dst_vma); +#else mmap_read_unlock(dst_mm); +#endif BUG_ON(!folio); err = copy_folio_from_user(folio, @@ -474,17 +596,6 @@ static __always_inline ssize_t mfill_atomic_hugetlb( err = -EFAULT; goto out; } - mmap_read_lock(dst_mm); - down_read(&ctx->map_changing_lock); - /* - * If memory mappings are changing because of non-cooperative - * operation (e.g. mremap) running in parallel, bail out and - * request the user to retry later - */ - if (atomic_read(&ctx->mmap_changing)) { - err = -EAGAIN; - break; - } dst_vma = NULL; goto retry; @@ -505,7 +616,12 @@ static __always_inline ssize_t mfill_atomic_hugetlb( out_unlock: up_read(&ctx->map_changing_lock); +out_unlock_vma: +#ifdef CONFIG_PER_VMA_LOCK + unlock_vma(dst_vma); +#else mmap_read_unlock(dst_mm); +#endif out: if (folio) folio_put(folio); @@ -597,7 +713,19 @@ static __always_inline ssize_t mfill_atomic(struct userfaultfd_ctx *ctx, copied = 0; folio = NULL; retry: - mmap_read_lock(dst_mm); + /* + * Make sure the vma is not shared, that the dst range is + * both valid and fully within a single existing vma. + */ +#ifdef CONFIG_PER_VMA_LOCK + dst_vma = find_and_lock_dst_vma(dst_mm, dst_start, len); +#else + dst_vma = lock_mm_and_find_dst_vma(dst_mm, dst_start, len); +#endif + if (IS_ERR(dst_vma)) { + err = PTR_ERR(dst_vma); + goto out; + } /* * If memory mappings are changing because of non-cooperative @@ -609,15 +737,6 @@ static __always_inline ssize_t mfill_atomic(struct userfaultfd_ctx *ctx, if (atomic_read(&ctx->mmap_changing)) goto out_unlock; - /* - * Make sure the vma is not shared, that the dst range is - * both valid and fully within a single existing vma. - */ - err = -ENOENT; - dst_vma = find_dst_vma(dst_mm, dst_start, len); - if (!dst_vma) - goto out_unlock; - err = -EINVAL; /* * shmem_zero_setup is invoked in mmap for MAP_ANONYMOUS|MAP_SHARED but @@ -647,16 +766,6 @@ static __always_inline ssize_t mfill_atomic(struct userfaultfd_ctx *ctx, uffd_flags_mode_is(flags, MFILL_ATOMIC_CONTINUE)) goto out_unlock; - /* - * Ensure the dst_vma has a anon_vma or this page - * would get a NULL anon_vma when moved in the - * dst_vma. - */ - err = -ENOMEM; - if (!(dst_vma->vm_flags & VM_SHARED) && - unlikely(anon_vma_prepare(dst_vma))) - goto out_unlock; - while (src_addr < src_start + len) { pmd_t dst_pmdval; @@ -699,7 +808,11 @@ static __always_inline ssize_t mfill_atomic(struct userfaultfd_ctx *ctx, void *kaddr; up_read(&ctx->map_changing_lock); +#ifdef CONFIG_PER_VMA_LOCK + unlock_vma(dst_vma); +#else mmap_read_unlock(dst_mm); +#endif BUG_ON(!folio); kaddr = kmap_local_folio(folio, 0); @@ -730,7 +843,11 @@ static __always_inline ssize_t mfill_atomic(struct userfaultfd_ctx *ctx, out_unlock: up_read(&ctx->map_changing_lock); +#ifdef CONFIG_PER_VMA_LOCK + unlock_vma(dst_vma); +#else mmap_read_unlock(dst_mm); +#endif out: if (folio) folio_put(folio); @@ -1267,16 +1384,67 @@ static int validate_move_areas(struct userfaultfd_ctx *ctx, if (!vma_is_anonymous(src_vma) || !vma_is_anonymous(dst_vma)) return -EINVAL; - /* - * Ensure the dst_vma has a anon_vma or this page - * would get a NULL anon_vma when moved in the - * dst_vma. - */ - if (unlikely(anon_vma_prepare(dst_vma))) - return -ENOMEM; + return 0; +} + +#ifdef CONFIG_PER_VMA_LOCK +static int find_and_lock_vmas(struct mm_struct *mm, + unsigned long dst_start, + unsigned long src_start, + struct vm_area_struct **dst_vmap, + struct vm_area_struct **src_vmap) +{ + int err; + + /* There is no need to prepare anon_vma for src_vma */ + *src_vmap = lock_vma(mm, src_start, false); + if (!*src_vmap) + return -ENOENT; + + /* Ensure anon_vma is assigned for anonymous vma */ + *dst_vmap = lock_vma(mm, dst_start, true); + err = -ENOENT; + if (!*dst_vmap) + goto out_unlock; + + err = -ENOMEM; + if (PTR_ERR(*dst_vmap) == -ENOMEM) + goto out_unlock; return 0; +out_unlock: + unlock_vma(*src_vmap); + return err; } +#else +static int lock_mm_and_find_vmas(struct mm_struct *mm, + unsigned long dst_start, + unsigned long src_start, + struct vm_area_struct **dst_vmap, + struct vm_area_struct **src_vmap) +{ + int err = -ENOENT; + mmap_read_lock(mm); + + *src_vmap = vma_lookup(mm, src_start); + if (!*src_vmap) + goto out_unlock; + + *dst_vmap = vma_lookup(mm, dst_start); + if (!*dst_vmap) + goto out_unlock; + + /* Ensure anon_vma is assigned */ + err = -ENOMEM; + if (vma_is_anonymous(*dst_vmap) && anon_vma_prepare(*dst_vmap)) + goto out_unlock; + + return 0; +out_unlock: + mmap_read_unlock(mm); + return err; +} +#endif /** * move_pages - move arbitrary anonymous pages of an existing vma @@ -1287,8 +1455,6 @@ static int validate_move_areas(struct userfaultfd_ctx *ctx, * @len: length of the virtual memory range * @mode: flags from uffdio_move.mode * - * Must be called with mmap_lock held for read. - * * move_pages() remaps arbitrary anonymous pages atomically in zero * copy. It only works on non shared anonymous pages because those can * be relocated without generating non linear anon_vmas in the rmap @@ -1355,10 +1521,10 @@ static int validate_move_areas(struct userfaultfd_ctx *ctx, * could be obtained. This is the only additional complexity added to * the rmap code to provide this anonymous page remapping functionality. */ -ssize_t move_pages(struct userfaultfd_ctx *ctx, struct mm_struct *mm, - unsigned long dst_start, unsigned long src_start, - unsigned long len, __u64 mode) +ssize_t move_pages(struct userfaultfd_ctx *ctx, unsigned long dst_start, + unsigned long src_start, unsigned long len, __u64 mode) { + struct mm_struct *mm = ctx->mm; struct vm_area_struct *src_vma, *dst_vma; unsigned long src_addr, dst_addr; pmd_t *src_pmd, *dst_pmd; @@ -1376,28 +1542,40 @@ ssize_t move_pages(struct userfaultfd_ctx *ctx, struct mm_struct *mm, WARN_ON_ONCE(dst_start + len <= dst_start)) goto out; +#ifdef CONFIG_PER_VMA_LOCK + err = find_and_lock_vmas(mm, dst_start, src_start, + &dst_vma, &src_vma); +#else + err = lock_mm_and_find_vmas(mm, dst_start, src_start, + &dst_vma, &src_vma); +#endif + if (err) + goto out; + + /* Re-check after taking map_changing_lock */ + down_read(&ctx->map_changing_lock); + if (likely(atomic_read(&ctx->mmap_changing))) { + err = -EAGAIN; + goto out_unlock; + } /* * Make sure the vma is not shared, that the src and dst remap * ranges are both valid and fully within a single existing * vma. */ - src_vma = find_vma(mm, src_start); - if (!src_vma || (src_vma->vm_flags & VM_SHARED)) - goto out; - if (src_start < src_vma->vm_start || - src_start + len > src_vma->vm_end) - goto out; + if (src_vma->vm_flags & VM_SHARED) + goto out_unlock; + if (src_start + len > src_vma->vm_end) + goto out_unlock; - dst_vma = find_vma(mm, dst_start); - if (!dst_vma || (dst_vma->vm_flags & VM_SHARED)) - goto out; - if (dst_start < dst_vma->vm_start || - dst_start + len > dst_vma->vm_end) - goto out; + if (dst_vma->vm_flags & VM_SHARED) + goto out_unlock; + if (dst_start + len > dst_vma->vm_end) + goto out_unlock; err = validate_move_areas(ctx, src_vma, dst_vma); if (err) - goto out; + goto out_unlock; for (src_addr = src_start, dst_addr = dst_start; src_addr < src_start + len;) { @@ -1514,6 +1692,14 @@ ssize_t move_pages(struct userfaultfd_ctx *ctx, struct mm_struct *mm, moved += step_size; } +out_unlock: + up_read(&ctx->map_changing_lock); +#ifdef CONFIG_PER_VMA_LOCK + unlock_vma(dst_vma); + unlock_vma(src_vma); +#else + mmap_read_unlock(mm); +#endif out: VM_WARN_ON(moved < 0); VM_WARN_ON(err > 0);