From patchwork Tue Feb 6 01:09:17 2024 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: Lokesh Gidra X-Patchwork-Id: 13546497 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 9F879C4828D for ; Tue, 6 Feb 2024 01:09:50 +0000 (UTC) Received: by kanga.kvack.org (Postfix) id 203826B0075; Mon, 5 Feb 2024 20:09:50 -0500 (EST) Received: by kanga.kvack.org (Postfix, from userid 40) id 1B3C06B0078; Mon, 5 Feb 2024 20:09:50 -0500 (EST) X-Delivered-To: int-list-linux-mm@kvack.org Received: by kanga.kvack.org (Postfix, from userid 63042) id 07BAA6B007B; Mon, 5 Feb 2024 20:09:50 -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 E9AFF6B0075 for ; Mon, 5 Feb 2024 20:09:49 -0500 (EST) Received: from smtpin16.hostedemail.com (a10.router.float.18 [10.200.18.1]) by unirelay05.hostedemail.com (Postfix) with ESMTP id 8975040B53 for ; Tue, 6 Feb 2024 01:09:49 +0000 (UTC) X-FDA: 81759596898.16.78D6F25 Received: from mail-yb1-f201.google.com (mail-yb1-f201.google.com [209.85.219.201]) by imf12.hostedemail.com (Postfix) with ESMTP id B972F40002 for ; Tue, 6 Feb 2024 01:09:47 +0000 (UTC) Authentication-Results: imf12.hostedemail.com; dkim=pass header.d=google.com header.s=20230601 header.b="jejsq/lM"; spf=pass (imf12.hostedemail.com: domain of 32obBZQsKCCMKNJDRGFHCQ9FNNFKD.BNLKHMTW-LLJU9BJ.NQF@flex--lokeshgidra.bounces.google.com designates 209.85.219.201 as permitted sender) smtp.mailfrom=32obBZQsKCCMKNJDRGFHCQ9FNNFKD.BNLKHMTW-LLJU9BJ.NQF@flex--lokeshgidra.bounces.google.com; dmarc=pass (policy=reject) header.from=google.com ARC-Message-Signature: i=1; a=rsa-sha256; c=relaxed/relaxed; d=hostedemail.com; s=arc-20220608; t=1707181787; 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=l0Wsa/otXfcNbNeX5OxfMRaNnAPGn+xliclZzLODRNs=; b=V5LZLHzegUl9ymx4JrzzmoGdliZ0hSRPHx2lxdV9LpqWTEYMmSHtLzVT3g6zN9WimCVSF7 HcocX5wqW2kQ3lK0zVV2olYGGItm8aSQ3UHmrttn8KpoVh4D/z/ng+2U5ZVKeYc8k8yKm8 AJmcl2sBGAeCmEh1sm0ZRWlpGDX8uD0= ARC-Authentication-Results: i=1; imf12.hostedemail.com; dkim=pass header.d=google.com header.s=20230601 header.b="jejsq/lM"; spf=pass (imf12.hostedemail.com: domain of 32obBZQsKCCMKNJDRGFHCQ9FNNFKD.BNLKHMTW-LLJU9BJ.NQF@flex--lokeshgidra.bounces.google.com designates 209.85.219.201 as permitted sender) smtp.mailfrom=32obBZQsKCCMKNJDRGFHCQ9FNNFKD.BNLKHMTW-LLJU9BJ.NQF@flex--lokeshgidra.bounces.google.com; dmarc=pass (policy=reject) header.from=google.com ARC-Seal: i=1; s=arc-20220608; d=hostedemail.com; t=1707181787; a=rsa-sha256; cv=none; b=UxnXtZ2YApsoTf2Iqlgu8ab1Pp0CYGi50S8I1/ZavXg2eu+dBqH1ayuDKWUzioGxPoa/wM umbLs/wC9yf/q0iP6+6nP6KGJniXzI+YDi54gS3QQk9C4mkor3zJYfPucj1rutNHjsCjTf 50FpX46k9GpHRD24hHflkw8OKbu4lEw= Received: by mail-yb1-f201.google.com with SMTP id 3f1490d57ef6-dc693399655so8096927276.1 for ; Mon, 05 Feb 2024 17:09:47 -0800 (PST) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=google.com; s=20230601; t=1707181787; x=1707786587; 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=l0Wsa/otXfcNbNeX5OxfMRaNnAPGn+xliclZzLODRNs=; b=jejsq/lMSV2hpSaDz0gb4B0eeoxUaYPR7Qmp/RKRL675IiFdAmZqAaQ44HLPLLPRCk +yOEKmKXzLc99clPRfRpLXR6DHg/5FxT29Jy58Ba6rRxOAt4H+VJM7G/XlSkF9A+g2ov k7vKFazSqA2ivGvEMfiM63kCzGWCe19lwGZnxfzBc127AHARAVxlygsVBgN0uKV2ju9s Gxm0nSoSKcbtz2W8hH7tQWpSpo9zSqt9iY6P+vX7NSTWojNOLWqkRSmT5XRnX4hv95AN 0SIjARDTvQZlcQl4sv7bZxPgcA/0UtOlMCIjeYIhCWADdu/mpuhTqOYPa2xV5EWx1qdu Ugag== X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20230601; t=1707181787; x=1707786587; 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=l0Wsa/otXfcNbNeX5OxfMRaNnAPGn+xliclZzLODRNs=; b=jUpaKO/xWX3c6BfppKZgde5xJTL8au3ClMXVp3bhGBN4G5oOdTga8i/rhawhLsxWyG yon8SZXTN3niJKApLSkaWiW8uiqcQCaBg5ZJqm2PFQmnP9C5NjvWLNeBmEKu8UOoJlv3 S2nbIfTedOik2ASJJlYpvr+dv39BMvgn2G7PinnPGu3ohPHjrFZOU+nLLf3Pdaa/+lVO fNTq1lZxAMptCM2p61uXQ+HsFT+7i8dPjhXAbTjLKp/1uVGAn+QQ0YPbruz6tjZnHH/u SO2KqJXqfDdJn7KB87PyYWcom4W6Fi8aSGJVcWtK4ovjE81jnL2yaw8/PQciLSGRz/J8 /2jw== X-Gm-Message-State: AOJu0Yxxo3nJ60IiyugykK/L1pD6CJF6MXqvhAYVIPKHkX5QjFolzFiQ 0LtjMUJm83+/KPnlO6mEaaQFcPhqMg7O0IBOMu6vDE+nb6PqSgVlGLjXMXp66+Ti7ZMPOWyHZIO FzZRHMm7G8INFsLLsrg1BIg== X-Google-Smtp-Source: AGHT+IENlNSDcojoxWkSl049zlYz4bm/u4F+E9+JQ3HrIX9YrhW8LmpdpxXCyJfTB1mNb9KyDkiJPwezJtg2W5ykyQ== X-Received: from lg.mtv.corp.google.com ([2620:15c:211:202:713:bb2c:e0e8:becb]) (user=lokeshgidra job=sendgmr) by 2002:a05:6902:13ce:b0:dc6:e90a:7f2a with SMTP id y14-20020a05690213ce00b00dc6e90a7f2amr53474ybu.5.1707181786937; Mon, 05 Feb 2024 17:09:46 -0800 (PST) Date: Mon, 5 Feb 2024 17:09:17 -0800 In-Reply-To: <20240206010919.1109005-1-lokeshgidra@google.com> Mime-Version: 1.0 References: <20240206010919.1109005-1-lokeshgidra@google.com> X-Mailer: git-send-email 2.43.0.594.gd9cf4e227d-goog Message-ID: <20240206010919.1109005-2-lokeshgidra@google.com> Subject: [PATCH v3 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: B972F40002 X-Rspam-User: X-Stat-Signature: bggd3qh8w69rdcn5cpyt48pwrbeykubc X-Rspamd-Server: rspam01 X-HE-Tag: 1707181787-668344 X-HE-Meta: U2FsdGVkX1+fV+hfSuR+ZA4F61GwSJhIRWU/qFBcUoDJjVkOUh9GA/BuUAQm6WFXmzcsiCLr/9ypQQr5eJb2ATvkeLJXdO5WJhLFHOwafSiIu4dQzTxzDxe2u6VsUQcOr8fx4UufUWGA685EwaR+cJKREnwDhNdQ1rxpNIDwSyep2kaHBF12OBTnojqXTMyXZHBm8jN7QkMWMvSs5SoQBx7/6DTZCZWuCC8b68ZSMPsK/i1VXn0mApYa1B9O01+T2Q3hZwgg71UBOW5GNiEx9uyu+WTbXUYa/5/PFW87l0ddb2D6ip1iFmcAfCzq0YZw/5qVEFtRg2lX+6CgStrFTenFBGzVYUDfVOcKgNi3bjtwj2M6JuFqY+HTR1GRf7T21YS6G6GKzr5ACXMI+HM90WpheGSFM+7ADZ2rqxH3g++3w1xqzecAHkCF0WE+EGXGxJVbjB6CdVaQunDt5lGSrmvVKnSf1gOfPJT1m5n4Ly4f34RSSjuRq/PkRAEhTJSnd3cre6c3mvZFNToKsEhv7f2/TgJ3WIk8iSJUJTxPcMyHVhVTVSbw3JFBkD8ub1iOSna3mzAazYi71/rOKKfTlcc3Wln0jeZopL7BkB0I72UldPawAjwpecczmQ/CcGFvAY5raQy7uvgC3UPe3Kh/hSmwyRc/maJYUfCzJZ+A6JLVPvmwJ+yrylCLZ/FURqHsB75o0GA7zcz0dm+E+M9DewA/ploMeTEpsEoN4PPO3Rde586q9yiEgs+gEqn2gfwrWXjuWB7MEgS2IESmwn7vj6G/SNLd9D2mBmMN2HkirAV7fQ9OjKhRewUSxtNHZ+PPdrlVeR9gFRJAav0YAnMSSPH1ejY++TWQfoGkAlNVizRS77J9KXM8jRbDuyoDMb6ct7m3ua9/8v66yaDBTqEnKiLM+dGTwH0lw3lRRCiuvugytzc2eo1VzfesW/L3+a99I4GbLOoqD7YBm0X/agP GJwUH3GG 0TRbMjiwlwIyAoaklA5sghOcYPiVC8pIFnUg3bwvujmg8rIka7H/X1h3hq+vt8RpXkrIrbnYw66XJtKzkTBO9ANiP1NF47RAmh0gAnxVivkEXCt8xF1nu8blr+CE2YRHrV5tZLcqrkjEuwTaFSeCKSJwOAIwG3nHiJBM4TLyHw0aXD+gb9F8UxUL7vlv7UD084ZmjHCuCXSE0kZIrCuTvFN5kL8L2uDQoNB8NZ5lVR9wKEd5hvs3NfV5ywMEnvPOsPyD+h0RX0kjsMyMUT1Tnhsd1uzd7bP2tCJpKpbB0ozIAzt+x/8r8P3s1Ybwf7qWbX7aN8t4RzioG+ITDu3iN7xOzgW65mV/58RsqxLiZxewq3vt1gGGSU8NOrKlT8sAtEyHjZLEb6AkoW+HzQdJ8wKUZCPjHlW1RR7Mb5pyvBAsygoDKAcX+qYaDus81RIKBTm1kla/A2btGuVIFVmg5Zwc/HBG/MZJWCmO2eaTTIxhAQYSawSOZoKXgMs2AAtZkKxU+Wt8eDFDUwJh2TRSmbRJLOIacGF15wYh4R8ftylTk+1rM4rjAxLiFpBQDorkQoCUjHU1Q2cDvLQJatTk/wUVAScdTJ9qsPf6g29KGHzV/PHY2GBgI1pagCMmJEwJANLpqONQetUO5qoxsaqWMTj+jh6nIuLytxZgQkonaZturoF7Md06qmosJxhw0CGeK41Ldby9T0E51orrFAsbtidMZH1lfEUfs4wPszsOIMOAZUh69cnGPHxCML5aATKvSeh6j 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 Tue Feb 6 01:09:18 2024 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: Lokesh Gidra X-Patchwork-Id: 13546498 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 8F07FC48260 for ; Tue, 6 Feb 2024 01:09:55 +0000 (UTC) Received: by kanga.kvack.org (Postfix) id 11BB66B007B; Mon, 5 Feb 2024 20:09:55 -0500 (EST) Received: by kanga.kvack.org (Postfix, from userid 40) id 0A4D36B007D; Mon, 5 Feb 2024 20:09:55 -0500 (EST) X-Delivered-To: int-list-linux-mm@kvack.org Received: by kanga.kvack.org (Postfix, from userid 63042) id E854C6B007E; Mon, 5 Feb 2024 20:09:54 -0500 (EST) X-Delivered-To: linux-mm@kvack.org Received: from relay.hostedemail.com (smtprelay0016.hostedemail.com [216.40.44.16]) by kanga.kvack.org (Postfix) with ESMTP id D7AE96B007B for ; Mon, 5 Feb 2024 20:09:54 -0500 (EST) Received: from smtpin17.hostedemail.com (a10.router.float.18 [10.200.18.1]) by unirelay09.hostedemail.com (Postfix) with ESMTP id 817B380268 for ; Tue, 6 Feb 2024 01:09:54 +0000 (UTC) X-FDA: 81759597108.17.CD921DB Received: from mail-yb1-f201.google.com (mail-yb1-f201.google.com [209.85.219.201]) by imf25.hostedemail.com (Postfix) with ESMTP id E5695A0011 for ; Tue, 6 Feb 2024 01:09:52 +0000 (UTC) Authentication-Results: imf25.hostedemail.com; dkim=pass header.d=google.com header.s=20230601 header.b=uVj5zpUY; dmarc=pass (policy=reject) header.from=google.com; spf=pass (imf25.hostedemail.com: domain of 34IbBZQsKCCkQTPJXMLNIWFLTTLQJ.HTRQNSZc-RRPaFHP.TWL@flex--lokeshgidra.bounces.google.com designates 209.85.219.201 as permitted sender) smtp.mailfrom=34IbBZQsKCCkQTPJXMLNIWFLTTLQJ.HTRQNSZc-RRPaFHP.TWL@flex--lokeshgidra.bounces.google.com ARC-Message-Signature: i=1; a=rsa-sha256; c=relaxed/relaxed; d=hostedemail.com; s=arc-20220608; t=1707181793; 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=I/Vi8otOtymcXqLdpslOuIDSttexTh2OJvYCdLb4Mo0=; b=0qVu0cQK+xCjGd3VK1ttB6JxODG/gGxiV5xQ49qR2Kt3qXXyGuAs+Di43ljTsJoInxTJ7E vapJeZmIA9rVucs4ClG0YaYquqs6NKxrq0tUdHVzVRkDrxZw0Lq/6WpP5vjNX28QZILQ/V gOliY5mXPlKK0r15NssrTMzPSwdtbew= ARC-Authentication-Results: i=1; imf25.hostedemail.com; dkim=pass header.d=google.com header.s=20230601 header.b=uVj5zpUY; dmarc=pass (policy=reject) header.from=google.com; spf=pass (imf25.hostedemail.com: domain of 34IbBZQsKCCkQTPJXMLNIWFLTTLQJ.HTRQNSZc-RRPaFHP.TWL@flex--lokeshgidra.bounces.google.com designates 209.85.219.201 as permitted sender) smtp.mailfrom=34IbBZQsKCCkQTPJXMLNIWFLTTLQJ.HTRQNSZc-RRPaFHP.TWL@flex--lokeshgidra.bounces.google.com ARC-Seal: i=1; s=arc-20220608; d=hostedemail.com; t=1707181793; a=rsa-sha256; cv=none; b=snYq/ov4UP/s+UhDROta8/skwtI3AMAszFkzCT4+av8bDxY/R08evDKKG9/NXWZAD5iOaF 64TcvoSAosHkskDBszRJisG/BCXMJkcDqPd8spPqc7LvSC2tD18acNAriXgcEMkyKQy3kD BRCnixR+h1dLjy+hRfoH+vbimS6mySE= Received: by mail-yb1-f201.google.com with SMTP id 3f1490d57ef6-dc64e0fc7c8so505431276.2 for ; Mon, 05 Feb 2024 17:09:52 -0800 (PST) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=google.com; s=20230601; t=1707181792; x=1707786592; 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=I/Vi8otOtymcXqLdpslOuIDSttexTh2OJvYCdLb4Mo0=; b=uVj5zpUYNeO0TF9E31S6GoD27sYNqsCgV+95Yj6F7zgmxQhSRtQGaegrXrdoMIbYZH vlvxxScv9CQgQfzR/k2xRA8WaUD3FKUv/+bmKBZrrHjN5OGCALscTcx7hjJgVpRgls7d 9CIEskKX4gLMgDVqJ2yejczGEVAF2Pn/qb0VSQ5YBZFE3ZDZVINd/7obdm9NUFZxtKq/ j35BDvfKymSpi8IIg1sGPyHPU8zOeAO3sbVo8KSrTgjs6LZ7Wql2Bft6z2E51pL8w9U8 P/AGra2vcRSfVk5+QuekkGniYwBe3vRemIuAPeU5KMx0+GWze5RNwOqsekJ9kTH6UrRu RwAg== X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20230601; t=1707181792; x=1707786592; 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=I/Vi8otOtymcXqLdpslOuIDSttexTh2OJvYCdLb4Mo0=; b=ticfL3KvO9dRnZonGHw9IwTXfWLgBbtcQVNFjikCQGuU4FERnaQ+EtPD/+cNHriu1M ToWETYAvNcXbPQpFiZBjNFSk/VGeWXl5HGtmqjxW9ZNZ8zoID314G946eeBdbB2R6nYg 6yvxv2fknoc6BHDsVO9k+grtNNtTEYv5hC3znZdlhl6qsuIoybUIRlOqfC1Ea5v85h6G 7WK64YMO+ge0XHtv/N6Ho1vSBrTQ3WQlPgWgvWyv2720xiOuxGfibXeKbMsG1kVW7Pqf r2/dC1qjRAuA6ux+4fNDG1i3aC14dR/GHwWWJmkcTAejqNIVfDMmpfhGANuE4JrWFqoy myjA== X-Gm-Message-State: AOJu0YxmKzFzOUV1MXjQFmatuE4RG3jc1tzhv7MXLj24smh9V0IWzZ+E PMuT3zX7en3+rXYCAJZd8O4EIdOdVpJbcA0chk1M7fgL7AyMp6kp0wOVdkx1ROKM7kPjlRG2m1O H+fA92UTKI/TlJcKGu1YypA== X-Google-Smtp-Source: AGHT+IF6iHrKO4GF2onBN726PZJP9J0wc9faZd8ILxPXSc4wek49Og0zsJ1bKmHXFW1EgIBSnj2zislPCuzCAjX5zw== X-Received: from lg.mtv.corp.google.com ([2620:15c:211:202:713:bb2c:e0e8:becb]) (user=lokeshgidra job=sendgmr) by 2002:a05:6902:144f:b0:dc6:dd74:de68 with SMTP id a15-20020a056902144f00b00dc6dd74de68mr7102ybv.12.1707181792130; Mon, 05 Feb 2024 17:09:52 -0800 (PST) Date: Mon, 5 Feb 2024 17:09:18 -0800 In-Reply-To: <20240206010919.1109005-1-lokeshgidra@google.com> Mime-Version: 1.0 References: <20240206010919.1109005-1-lokeshgidra@google.com> X-Mailer: git-send-email 2.43.0.594.gd9cf4e227d-goog Message-ID: <20240206010919.1109005-3-lokeshgidra@google.com> Subject: [PATCH v3 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-Rspamd-Queue-Id: E5695A0011 X-Rspam-User: X-Rspamd-Server: rspam02 X-Stat-Signature: 4udnyw7d4f7wn3xj3y9aotux3dzi8agc X-HE-Tag: 1707181792-275949 X-HE-Meta: U2FsdGVkX1/QohrU195jJv5nUFMCeD9UIb9qftvTWDwD/8YlIdB6yhxb9h3XL4kp2zsDJ/N35FKX1lt3K/hhVms4VSB+vXVsWtXnCzghpa9HUIAVwAEk/GR343IEoCkboAsodms8zBl8Fi95fLUkJOQu+QWXyWO+HYeCbjWJrRykpu9GTFGkwrOW6sKQCXY6YMDkw2QKVO1jofR2bNN5J3YKFGPhNJ2R69mXM+0+KDXMEySsHJjQSgj/9dwXtPg+2ZtECcXmMq5ymS8SXbgd8VtytJfdlIK3dg4/1AJ9LT1GlyLWfM7NfQCwuuxJWQk7EozVLXHjeOWh3kAGxN4m2d1CclPg/jewpaOFYjbx0+cVFU0klxLdEezSOsI33hSZry1xyJKhzXYhwcOxRlzxWBujuzMN8NQc27SJnmFQn6qB+1cqgm2txNwrH8UmXfWtCW8qBeuz905DoXBdhl9CXk8lCrLGhT1DA7+V4Ylh8JZXu9n7k1BzBpqaId6Wgr5Gsvy+9EWRWil61Ej3ZPKj/VF/FkDsWw8cQ2OGj8tlwtEvuqTEmkhaPxOjT+zTgt1NZKG7f9ybc4cxf3ossPAGbN0eBV3ZivYjJeKZCRXtxB/aUBE5lyRWvlJSkKc+AtuFVar4Jx1ymSXawAxF/KwmDrh1RoAsEp7vN0DDMi4skn15Z0yT5f6TYRavtnqOByqZmUVDh3QK+AaW3atvDC1x+yhYwLZdLBK629FrFrzUQ7s5lJPdfGjVTErwSCUNjQ4ONpIGgSYu2tSlCznPSAvyzL2fifneuavhLIWMklVTY7knrQaLz/IEe8hnPorLVoTQ6prAgXEJe2JlPCJZZYAGembhaUOv2f0nHH0/OmafW4SfTXYf8jMNscHW8CspuHYavLfL6CzjRivrklj7H9GwbHpHy61aTbKT5cMmnFvXin1EHNA69rH7a4pjeQz0OiMccQgF0dMAd+EG4+7Z+PW aNpJ990t 97YsMCBxTldAhwtJ/MQRh1oGqzmurywsYHlpwEfX28nMUKn0Fi5hSADa9EcNwfels4PI1oipffvYHF216pmGeSmkJTUrwMFz88XsEvm1iS++dv63XJDRACLz4BTijFFSIq65Xq1PJc/fbu3cCiM/w4ldLXVwN/hTP69qgxnog8emC3iYTYuN/p9tbs1SHpSb/eLd6MguSl+zmTWYuf7JuRTPY2AKMZE0dnyoES7+CIX4kBzX+mFGkovdD2b1m/hwNSeXTy8YXWGeVDqU2gnBUZ+VNzlq4n27UDm6rXpdnPJFs2ydoVlWWD/wckYUYkC2iEt1T/jd8i5EagsVEzN+ZBNhFAhr3IeInRkdFZwTmvssk2vYQrL785r8drMlz2lNzkSBEXA3WoVc0a+qzdwqPDu49cSl2j7vfXHLsL2dOFKO0aQuUgTLV8EQxdA== 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 Tue Feb 6 01:09:19 2024 Content-Type: text/plain; charset="utf-8" MIME-Version: 1.0 Content-Transfer-Encoding: 7bit X-Patchwork-Submitter: Lokesh Gidra X-Patchwork-Id: 13546499 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 CB77AC4828D for ; Tue, 6 Feb 2024 01:10:01 +0000 (UTC) Received: by kanga.kvack.org (Postfix) id 4E9A06B007E; Mon, 5 Feb 2024 20:10:01 -0500 (EST) Received: by kanga.kvack.org (Postfix, from userid 40) id 499566B0080; Mon, 5 Feb 2024 20:10:01 -0500 (EST) X-Delivered-To: int-list-linux-mm@kvack.org Received: by kanga.kvack.org (Postfix, from userid 63042) id 2ECB96B0081; Mon, 5 Feb 2024 20:10:01 -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 196556B007E for ; Mon, 5 Feb 2024 20:10:01 -0500 (EST) Received: from smtpin07.hostedemail.com (a10.router.float.18 [10.200.18.1]) by unirelay02.hostedemail.com (Postfix) with ESMTP id A828B120119 for ; Tue, 6 Feb 2024 01:10:00 +0000 (UTC) X-FDA: 81759597360.07.C8A451D Received: from mail-yb1-f202.google.com (mail-yb1-f202.google.com [209.85.219.202]) by imf01.hostedemail.com (Postfix) with ESMTP id E14D240004 for ; Tue, 6 Feb 2024 01:09:58 +0000 (UTC) Authentication-Results: imf01.hostedemail.com; dkim=pass header.d=google.com header.s=20230601 header.b=oi4bss8p; spf=pass (imf01.hostedemail.com: domain of 35obBZQsKCC8WZVPdSRTOcLRZZRWP.NZXWTYfi-XXVgLNV.ZcR@flex--lokeshgidra.bounces.google.com designates 209.85.219.202 as permitted sender) smtp.mailfrom=35obBZQsKCC8WZVPdSRTOcLRZZRWP.NZXWTYfi-XXVgLNV.ZcR@flex--lokeshgidra.bounces.google.com; dmarc=pass (policy=reject) header.from=google.com ARC-Message-Signature: i=1; a=rsa-sha256; c=relaxed/relaxed; d=hostedemail.com; s=arc-20220608; t=1707181799; 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=nOqm5rsZhzJMvjR9+dAhD+eAWB3F8dZk7DoU+bZxnlo=; b=QygSCxDatQAu8zlAv01H04bq0xa4dxsTpJG4nlVmMb+W8FDptwkCi8x0F3gykQTpZ2XqvF ew9VfjnZYHGA+76YjONnhLacORvd/TLfXvjfGYvGPeYlB8/0QFESWshPD9Ei0XWYF0NK3W 5By9ihKWKK1RKEXJd2H5Icbev8YfPKk= ARC-Seal: i=1; s=arc-20220608; d=hostedemail.com; t=1707181799; a=rsa-sha256; cv=none; b=kIaWXfzEivSwqPhoow/5srPSj55XUFq4pAaG9BkIv9egw63mWmVjplyiqOmbISW3V4EFmZ syK8ZhAKZ+MQ+RK9zRH8CmOoqdTjenQQj9RTgrXChCeAWtdiohrzindUBt+w51itGykjIV NbwqPeJfqgB8he2U7wnn/ExUQx4i3HQ= ARC-Authentication-Results: i=1; imf01.hostedemail.com; dkim=pass header.d=google.com header.s=20230601 header.b=oi4bss8p; spf=pass (imf01.hostedemail.com: domain of 35obBZQsKCC8WZVPdSRTOcLRZZRWP.NZXWTYfi-XXVgLNV.ZcR@flex--lokeshgidra.bounces.google.com designates 209.85.219.202 as permitted sender) smtp.mailfrom=35obBZQsKCC8WZVPdSRTOcLRZZRWP.NZXWTYfi-XXVgLNV.ZcR@flex--lokeshgidra.bounces.google.com; dmarc=pass (policy=reject) header.from=google.com Received: by mail-yb1-f202.google.com with SMTP id 3f1490d57ef6-dc64e0fc7c8so505504276.2 for ; Mon, 05 Feb 2024 17:09:58 -0800 (PST) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=google.com; s=20230601; t=1707181798; x=1707786598; 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=nOqm5rsZhzJMvjR9+dAhD+eAWB3F8dZk7DoU+bZxnlo=; b=oi4bss8pGYPct0P51Bke0b/1k2fFVqG4zayENSr5oWo8f8b9umNFPnw2jGWywGKOdl L6d6EaRb12YTqJ6XXCyd+yvgUQmvKo+ZlawbKbtmSQOc/XKsNyW9Y0Kq0afVellxJ76y yH2blrTHDW4RxzRJpjb0jv7eiw/dFdO9WYhusBT+BW05jcs8eRGH7RHfFOS1x/2ZSJOU /FVQyFewlTlpNevjSbnmgGH5y7g5sTA60Z+LKY3Uruxc/NZJQ6TjYzKAdt81l/dLNG7l lqDupLXipXYO9T9onvXbzTQ3QkaPr7bUvxCYxaHT/PLB1SZ9W75jcRDOaGuguLPg+CNu ElEg== X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20230601; t=1707181798; x=1707786598; 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=nOqm5rsZhzJMvjR9+dAhD+eAWB3F8dZk7DoU+bZxnlo=; b=C+n8dAvff5oWxziQTnvqMHRl4m/4Rb82St5swXFx8P6WXbcFJ8/5iRHVwaIIk3tyvp Pv4npwhlYKDpSmF17hvh4fNeQTbZgeEtXSQY9tVunPzxfsFGOruuDsmSjgdNX2yaeXGP zCkD5JZWJFBsCFU43hZI2W6f4iLvgG7hjMbPsssf1AcsNv6BQFGGcSl7+M0cBk8V2mFT cuZ2QGpOcC1a5ZinEp67//Yx3mEpBWT0pD3u5HglVNYYv35AR/l/+T6nSUlBrst6Nge/ uOkjBcV21xHJlZkzcLAjzEoImB8s0T5eXvq/3OkC9rvD9gLWcjDdxAd8ekvdns7bBWD8 qkzQ== X-Gm-Message-State: AOJu0Yz4rhELF1jHNjHJNITikvZAC0AIP2SQSbFcjot1nbjmp/Uu0Fek 7WDyzFRr0UapNPBr0vbQbbx8Rah1463Os9gkWb92M0x/m1LGU4HV5z6L6zdTAP+bBCcPASdO3rd ev4O9bXMC1YB6WZKUIYrMcQ== X-Google-Smtp-Source: AGHT+IFYxfY6Ge56ali6DMQRIXX2ISa1rBsLQcS+PmAp5ZBB9lJ2RO73M/vP9jb/DuDtls5ONW17Z8++ijURLhrH/Q== X-Received: from lg.mtv.corp.google.com ([2620:15c:211:202:713:bb2c:e0e8:becb]) (user=lokeshgidra job=sendgmr) by 2002:a05:6902:1b06:b0:dc2:3247:89d5 with SMTP id eh6-20020a0569021b0600b00dc2324789d5mr7221ybb.4.1707181798107; Mon, 05 Feb 2024 17:09:58 -0800 (PST) Date: Mon, 5 Feb 2024 17:09:19 -0800 In-Reply-To: <20240206010919.1109005-1-lokeshgidra@google.com> Mime-Version: 1.0 References: <20240206010919.1109005-1-lokeshgidra@google.com> X-Mailer: git-send-email 2.43.0.594.gd9cf4e227d-goog Message-ID: <20240206010919.1109005-4-lokeshgidra@google.com> Subject: [PATCH v3 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-Stat-Signature: p6szh9c4f9gopxpwnwawcuij86rzqkck X-Rspamd-Server: rspam10 X-Rspamd-Queue-Id: E14D240004 X-Rspam-User: X-HE-Tag: 1707181798-196475 X-HE-Meta: U2FsdGVkX19WSJ4dqdMAjWk8CKE4BQC+7OjaPfPwl2r1s1OVtQCt13BNJ+qrPXjT2/5tcMjMcPRrIX9t6Kijv4vNzyr557HcyJvdFv+Ai0auKfUAn7qcWZ5GEhM+2mSaPxNUoiJmfxbwJd7qogDDf5wrefWfD/ZvvuUiSqIMsosOqE2KKx2d3VBa8gdExIw5bSh/59DOj85A+WEidgNmNVpNlABiaD55YVtwDWwsi1HvUWPcfx9MM3UuYt2ijkFqHVWxM0DlO7/bCL7p0HSRGJ3lAnjumbs9GfywmfVPIp4qf2BDWRUjScdN0a4ZZkhjeYOOwwVs/603xH9bLhoktTx+d9nppwOWOceVTVF/6ksMFbxsoIeocV5Irkz9Qv2qu1+grJ3X067YfEq6D4pqWhmyWF3S0Kmu47UXkSNLQC479HriAdj3Oufl9yQPtaZYz6EG7rFA+7Y+WuJ4EV0IZGzZjdfSi1OkCkgGXEYj2MZnUYRJtk/lRMKX4YMFlduJ6o0MNxsZren2rvo0olUn/ybPHiQPFOnLz6Uly1ej0c5RCPJyzbbr/A19eIgaGlhzGCsMri2Q1crtMMPtSexK4ugRnwE+/F7xy73v1YORSBqnm7emI5DO31lN2BeVq25VDlFjTjrltC1sD2/KyCeuwnxkHoPv+kS8v4JmU/FWC1eW1K3UKiMFrk61oIR5kqFU3TR64O1StqFVfo1CchUmI7JrTj9SSBk0mHjlqkycgPjbcPpBl8UUzRcG2gVFJXCF9zntIPV/QTGF9wnlNXaEj14F/fmKzLdm/dfbVvjXfS5SxGJ3JaebB+fVrWuRXulStusN5sgEvKJrOYBS/lp8hU1dHkyiUQVSgjbRKqa1cHZgYifTX6IKD6Ism7OGB5zF2Z/w9Em60UNLNual445TpTSFQi9yzqbVOixn2/SWuUrgqfU5aQdf3ZFLQeA9ecmxoeDRTlb7ql64IBObG2t wUYYxl4S rExBMaUUOWNcwWNyvoMohEAsKEqJw4FtKnaeJnkL+KHcIMszaOWhcbOVygcxSiwgSPmKB3/tVCkrl0WLBKhpXwvSCXXa8gzBocgaEnOlS/OGJDTpl6iGY3nZQ0GWZW7AZTVhdl4b6xjJ0B4VW1nq5dGXcjNo2qoGXZELaliuXTn+jqpaAqk1tMPQfhCt4OmAM7B314nD2OU6ipJELswOkw4rwcJjsst9lmKGH1CTc699vz5iboRK1TCL7htxuahKRd8ZmJAqE7QfOB1vQq57Wo2bBXpyHLVW0hYRHyLOXqIXB4Uw9hzjgQIfgXWnxrFeFjNK8o8AlGLq35Ntu9VacCEkH/akGHInYKtwqwUgAvhkKhIMGCrVa+Jj85cGFvNqhqW2ZLRzpWMpxFnMeW736jDn0Z+GdqSJeqT97dlsEJMNE58Bzqc8YNs1Xi+47dJnFPSVle/RBf/2Scl2bLPrnuCBM6tcfH0M7CRFL9fyG1vYf+iM1eGXfcufZ6nTxZ4LhP+LQ/ZQ6LLV//M4XmHgjLG6v6ld2ovRe/ejI4N4Ne7so9TGckkqcHBQVipcClGYqR8Y08knH5eBC7caVrGa6l+mw8gCvzxevE3jVd/Ac0NAMwtNLzh2pmZbliE8iU597nGUqAnfzSHr++PI= 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/mm.h | 16 +++ include/linux/userfaultfd_k.h | 5 +- mm/memory.c | 48 +++++++ mm/userfaultfd.c | 242 +++++++++++++++++++++------------- 5 files changed, 222 insertions(+), 102 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/mm.h b/include/linux/mm.h index 0d1f98ab0c72..e69dfe2edcce 100644 --- a/include/linux/mm.h +++ b/include/linux/mm.h @@ -753,6 +753,11 @@ static inline void release_fault_lock(struct vm_fault *vmf) mmap_read_unlock(vmf->vma->vm_mm); } +static inline void unlock_vma(struct mm_struct *mm, struct vm_area_struct *vma) +{ + vma_end_read(vma); +} + static inline void assert_fault_locked(struct vm_fault *vmf) { if (vmf->flags & FAULT_FLAG_VMA_LOCK) @@ -774,6 +779,9 @@ static inline void vma_assert_write_locked(struct vm_area_struct *vma) { mmap_assert_write_locked(vma->vm_mm); } static inline void vma_mark_detached(struct vm_area_struct *vma, bool detached) {} +static inline void vma_acquire_read_lock(struct vm_area_struct *vma) { + mmap_assert_locked(vma->vm_mm); +} static inline struct vm_area_struct *lock_vma_under_rcu(struct mm_struct *mm, unsigned long address) @@ -786,6 +794,11 @@ static inline void release_fault_lock(struct vm_fault *vmf) mmap_read_unlock(vmf->vma->vm_mm); } +static inline void unlock_vma(struct mm_struct *mm, struct vm_area_struct *vma) +{ + mmap_read_unlock(mm); +} + static inline void assert_fault_locked(struct vm_fault *vmf) { mmap_assert_locked(vmf->vma->vm_mm); @@ -794,6 +807,9 @@ static inline void assert_fault_locked(struct vm_fault *vmf) #endif /* CONFIG_PER_VMA_LOCK */ extern const struct vm_operations_struct vma_dummy_vm_ops; +extern struct vm_area_struct *lock_vma(struct mm_struct *mm, + unsigned long address, + bool prepare_anon); /* * WARNING: vma_init does not initialize vma->vm_lock. 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/memory.c b/mm/memory.c index b05fd28dbce1..393ab3b0d6f3 100644 --- a/mm/memory.c +++ b/mm/memory.c @@ -5760,8 +5760,56 @@ struct vm_area_struct *lock_vma_under_rcu(struct mm_struct *mm, count_vm_vma_lock_event(VMA_LOCK_ABORT); return NULL; } + +static void vma_acquire_read_lock(struct vm_area_struct *vma) +{ + /* + * 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 could + * have locked the vma for write (vma_start_write()). + */ + mmap_assert_locked(vma->vm_mm); + down_read(&vma->vm_lock->lock); +} #endif /* CONFIG_PER_VMA_LOCK */ +/* + * lock_vma() - Lookup and lock VMA corresponding to @address. + * @prepare_anon: If true, then prepare the VMA (if anonymous) 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 of no VMA is found, or + * -ENOMEM if anon_vma couldn't be allocated. + */ +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) + return vma; + + mmap_read_lock(mm); + vma = vma_lookup(mm, address); + if (vma) { + if (prepare_anon && vma_is_anonymous(vma) && + anon_vma_prepare(vma)) + vma = ERR_PTR(-ENOMEM); + else + vma_acquire_read_lock(vma); + } + + if (IS_ENABLED(CONFIG_PER_VMA_LOCK) || !vma || PTR_ERR(vma) == -ENOMEM) + mmap_read_unlock(mm); + return vma; +} + #ifndef __PAGETABLE_P4D_FOLDED /* * Allocate p4d page table. diff --git a/mm/userfaultfd.c b/mm/userfaultfd.c index 74aad0831e40..64e22e467e4f 100644 --- a/mm/userfaultfd.c +++ b/mm/userfaultfd.c @@ -19,20 +19,25 @@ #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) +/* Search for VMA and make sure it is valid. */ +static struct vm_area_struct *find_and_lock_dst_vma(struct mm_struct *dst_mm, + unsigned long dst_start, + unsigned long len) { - /* - * 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; + /* Ensure anon_vma is assigned for anonymous vma */ + 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; + + /* Make sure that the dst range is fully within dst_vma. */ + if (dst_start + len > dst_vma->vm_end) + goto out_unlock; /* * Check the vma is registered in uffd, this is required to @@ -40,9 +45,12 @@ struct vm_area_struct *find_dst_vma(struct mm_struct *dst_mm, * time. */ if (!dst_vma->vm_userfaultfd_ctx.ctx) - return NULL; + goto out_unlock; return dst_vma; +out_unlock: + unlock_vma(dst_mm, dst_vma); + return ERR_PTR(-ENOENT); } /* Check if dst_addr is outside of file's size. Must be called with ptl held. */ @@ -350,7 +358,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 +370,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 +388,7 @@ static __always_inline ssize_t mfill_atomic_hugetlb( */ if (uffd_flags_mode_is(flags, MFILL_ATOMIC_ZEROPAGE)) { up_read(&ctx->map_changing_lock); - mmap_read_unlock(dst_mm); + unlock_vma(dst_mm, dst_vma); return -EINVAL; } @@ -403,24 +411,28 @@ static __always_inline ssize_t mfill_atomic_hugetlb( * retry, dst_vma will be set to NULL and we must lookup again. */ if (!dst_vma) { + dst_vma = find_and_lock_dst_vma(dst_mm, dst_start, len); + 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; + goto out_unlock_vma; - vm_shared = dst_vma->vm_flags & VM_SHARED; - } - - /* - * 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 +477,7 @@ static __always_inline ssize_t mfill_atomic_hugetlb( if (unlikely(err == -ENOENT)) { up_read(&ctx->map_changing_lock); - mmap_read_unlock(dst_mm); + unlock_vma(dst_mm, dst_vma); BUG_ON(!folio); err = copy_folio_from_user(folio, @@ -474,17 +486,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 +506,8 @@ static __always_inline ssize_t mfill_atomic_hugetlb( out_unlock: up_read(&ctx->map_changing_lock); - mmap_read_unlock(dst_mm); +out_unlock_vma: + unlock_vma(dst_mm, dst_vma); out: if (folio) folio_put(folio); @@ -597,7 +599,15 @@ 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. + */ + dst_vma = find_and_lock_dst_vma(dst_mm, dst_start, len); + if (IS_ERR(dst_vma)) { + err = PTR_ERR(dst_vma); + goto out; + } /* * If memory mappings are changing because of non-cooperative @@ -609,15 +619,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 +648,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 +690,8 @@ static __always_inline ssize_t mfill_atomic(struct userfaultfd_ctx *ctx, void *kaddr; up_read(&ctx->map_changing_lock); - mmap_read_unlock(dst_mm); + unlock_vma(dst_mm, dst_vma); + BUG_ON(!folio); kaddr = kmap_local_folio(folio, 0); @@ -730,7 +722,7 @@ static __always_inline ssize_t mfill_atomic(struct userfaultfd_ctx *ctx, out_unlock: up_read(&ctx->map_changing_lock); - mmap_read_unlock(dst_mm); + unlock_vma(dst_mm, dst_vma); out: if (folio) folio_put(folio); @@ -1267,16 +1259,82 @@ 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_move_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(mm, *src_vmap); + return err; +} + +static void unlock_move_vmas(struct mm_struct *mm, + struct vm_area_struct *dst_vma, + struct vm_area_struct *src_vma) +{ + unlock_vma(mm, dst_vma); + unlock_vma(mm, src_vma); } +#else +static int find_and_lock_move_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; +} + +static void unlock_move_vmas(struct mm_struct *mm, + struct vm_area_struct *dst_vma, + struct vm_area_struct *src_vma) +{ + mmap_read_unlock(mm); +} +#endif /** * move_pages - move arbitrary anonymous pages of an existing vma @@ -1287,8 +1345,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 +1411,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 +1432,35 @@ ssize_t move_pages(struct userfaultfd_ctx *ctx, struct mm_struct *mm, WARN_ON_ONCE(dst_start + len <= dst_start)) goto out; + err = find_and_lock_move_vmas(mm, dst_start, src_start, + &dst_vma, &src_vma); + 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 +1577,9 @@ ssize_t move_pages(struct userfaultfd_ctx *ctx, struct mm_struct *mm, moved += step_size; } +out_unlock: + up_read(&ctx->map_changing_lock); + unlock_move_vmas(mm, dst_vma, src_vma); out: VM_WARN_ON(moved < 0); VM_WARN_ON(err > 0);