From mboxrd@z Thu Jan 1 00:00:00 1970 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 0E486C3ABD8 for ; Wed, 14 May 2025 23:45:20 +0000 (UTC) Received: by kanga.kvack.org (Postfix) id 5BDF48D000F; Wed, 14 May 2025 19:43:52 -0400 (EDT) Received: by kanga.kvack.org (Postfix, from userid 40) id 543E08D0001; Wed, 14 May 2025 19:43:52 -0400 (EDT) X-Delivered-To: int-list-linux-mm@kvack.org Received: by kanga.kvack.org (Postfix, from userid 63042) id 260B78D0010; Wed, 14 May 2025 19:43:52 -0400 (EDT) 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 ED9B18D0001 for ; Wed, 14 May 2025 19:43:51 -0400 (EDT) Received: from smtpin29.hostedemail.com (a10.router.float.18 [10.200.18.1]) by unirelay02.hostedemail.com (Postfix) with ESMTP id 60A4912081F for ; Wed, 14 May 2025 23:43:51 +0000 (UTC) X-FDA: 83443143462.29.E59074C Received: from mail-pg1-f201.google.com (mail-pg1-f201.google.com [209.85.215.201]) by imf27.hostedemail.com (Postfix) with ESMTP id 9434440002 for ; Wed, 14 May 2025 23:43:49 +0000 (UTC) Authentication-Results: imf27.hostedemail.com; dkim=pass header.d=google.com header.s=20230601 header.b=DrFfVle1; dmarc=pass (policy=reject) header.from=google.com; spf=pass (imf27.hostedemail.com: domain of 3tColaAsKCOUHJRLYSLfaUNNVVNSL.JVTSPUbe-TTRcHJR.VYN@flex--ackerleytng.bounces.google.com designates 209.85.215.201 as permitted sender) smtp.mailfrom=3tColaAsKCOUHJRLYSLfaUNNVVNSL.JVTSPUbe-TTRcHJR.VYN@flex--ackerleytng.bounces.google.com ARC-Seal: i=1; s=arc-20220608; d=hostedemail.com; t=1747266229; a=rsa-sha256; cv=none; b=FV14CGAtbMvEsNUWY+Z3tTqbVUOqIgcpPpMtqnnEiolryxvbQPJF5C+bW4pPPxn9b21ngH nujaGHI/UGH+CjDrn+wIhzPZ5NZlzRPF+FlxKi7Go2iqNXWbUEeb94maFznKgbS3eA/bBB l3+uI37G1+j6yx1+6GxjOQ0GsuZUsT8= ARC-Authentication-Results: i=1; imf27.hostedemail.com; dkim=pass header.d=google.com header.s=20230601 header.b=DrFfVle1; dmarc=pass (policy=reject) header.from=google.com; spf=pass (imf27.hostedemail.com: domain of 3tColaAsKCOUHJRLYSLfaUNNVVNSL.JVTSPUbe-TTRcHJR.VYN@flex--ackerleytng.bounces.google.com designates 209.85.215.201 as permitted sender) smtp.mailfrom=3tColaAsKCOUHJRLYSLfaUNNVVNSL.JVTSPUbe-TTRcHJR.VYN@flex--ackerleytng.bounces.google.com ARC-Message-Signature: i=1; a=rsa-sha256; c=relaxed/relaxed; d=hostedemail.com; s=arc-20220608; t=1747266229; 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=cGxf10xTbolob6ARtbQFmrKdEey0c2iLS6uYVAPHbb0=; b=TN3tRIHKsALzj8Q4VR/4gwQEOf2/UoNcioP35f9xt3nao8XxKyLj7nTObN+rc+kPqGJgLT ROR9NIpVX2WAOPnThhwg5rIj+Vdqe3l97nGxfkuvrXWpBXEk4zGJZbFSTcRKtVtbneJGy0 4kjrjtjXJUUYLdkO/meQz+rSxPkmJH0= Received: by mail-pg1-f201.google.com with SMTP id 41be03b00d2f7-b1fa2cad5c9so130811a12.2 for ; Wed, 14 May 2025 16:43:49 -0700 (PDT) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=google.com; s=20230601; t=1747266228; x=1747871028; 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=cGxf10xTbolob6ARtbQFmrKdEey0c2iLS6uYVAPHbb0=; b=DrFfVle1wy/RIAZGced1dOvkdLPGU1eaKNejAj2rWMVFGdKk8dobp++ajqmOgJK0Hh AFZGCqAosgNbfibHR/rtS5rcIPcagVt+UnO1nfRsN7ocsW6gUfNE0woRfa7JdlSAt3Sq 3wwg6mEHmnOGXHLuu4xS9WdplyKq89h51SnvZf7p2ew650s3NPPQHNS07c1Qoxm2kErV EHvR8VAEU5sIs76xsVMlCa0jtY6AVnyPm+ITv6DG6spY79SmJOLy2MwbOoduoFg4nO4p EYyC7MtC/FLvjEPJAH/3QCFWAzxwq2A2U5jUssBDWiGgITBJ/txdc7d694Beis+huUKq /dCw== X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20230601; t=1747266228; x=1747871028; 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=cGxf10xTbolob6ARtbQFmrKdEey0c2iLS6uYVAPHbb0=; b=R3fVyeSewproNi1dFJ0zsxE7f9gLYoJJWhN9gS/081Homvhle91LQFOnAcVTg2Qgue v1MCBNRziCNc8ydhbYxnkT8DVOS9fk/rbOq8B3YEKja88LIWu29qPCfB5CN4c8BLoBn5 g2/OTAtkx633TrEwWITRwDggyrNpLeBLaq/8jLjBi/X8oNlARzyXRxlNxJdj1zQ60YPe 25/WCd3hzbCw1aHECAw43b/a+6G7/32Xk0dzJgB2w1C9FwmcIt+LQMzfTeBRwyR7bvie 1gFngXXHCrLKPqbfkaipa5IkGP1nK4ueATiva2c2G37NyjOwrx2KCYck4usNsBLKHUPr V9sQ== X-Forwarded-Encrypted: i=1; AJvYcCVF7NOsx4BDxHgA2QKRhcp+tCBMUHVr2s7BfZ3tgI0GY2Up3lYODDPe34JvXQnbN1GEavnCpylMuA==@kvack.org X-Gm-Message-State: AOJu0YxkyUJ0hdT21h2858TUveqLhnXaPMqci7ZLauv7fpvTUlhqqiGp Jh+PIiDgUw7ge+0J/YRvluVbsPvtav1B02+3jcVgDtEqLUiiLqavJqHfjT3qsUM5x5MiENNzaTF hozwiOUAXQyV6wj9ik/SEwA== X-Google-Smtp-Source: AGHT+IF3s18X1WXVSpREAFxwZmEiWRkEzf4FnXZJnpCpoYmdbOqB9rnOiNcmfKVh8wkeRqqYO9tvfAUew5Jususlyg== X-Received: from pjbta5.prod.google.com ([2002:a17:90b:4ec5:b0:2ff:5752:a78f]) (user=ackerleytng job=prod-delivery.src-stubby-dispatcher) by 2002:a17:90b:17c1:b0:2ee:9b09:7d3d with SMTP id 98e67ed59e1d1-30e51786009mr630115a91.19.1747266228378; Wed, 14 May 2025 16:43:48 -0700 (PDT) Date: Wed, 14 May 2025 16:42:12 -0700 In-Reply-To: Mime-Version: 1.0 References: X-Mailer: git-send-email 2.49.0.1045.g170613ef41-goog Message-ID: Subject: [RFC PATCH v2 33/51] KVM: guest_memfd: Allocate and truncate from custom allocator From: Ackerley Tng To: kvm@vger.kernel.org, linux-mm@kvack.org, linux-kernel@vger.kernel.org, x86@kernel.org, linux-fsdevel@vger.kernel.org Cc: ackerleytng@google.com, aik@amd.com, ajones@ventanamicro.com, akpm@linux-foundation.org, amoorthy@google.com, anthony.yznaga@oracle.com, anup@brainfault.org, aou@eecs.berkeley.edu, bfoster@redhat.com, binbin.wu@linux.intel.com, brauner@kernel.org, catalin.marinas@arm.com, chao.p.peng@intel.com, chenhuacai@kernel.org, dave.hansen@intel.com, david@redhat.com, dmatlack@google.com, dwmw@amazon.co.uk, erdemaktas@google.com, fan.du@intel.com, fvdl@google.com, graf@amazon.com, haibo1.xu@intel.com, hch@infradead.org, hughd@google.com, ira.weiny@intel.com, isaku.yamahata@intel.com, jack@suse.cz, james.morse@arm.com, jarkko@kernel.org, jgg@ziepe.ca, jgowans@amazon.com, jhubbard@nvidia.com, jroedel@suse.de, jthoughton@google.com, jun.miao@intel.com, kai.huang@intel.com, keirf@google.com, kent.overstreet@linux.dev, kirill.shutemov@intel.com, liam.merwick@oracle.com, maciej.wieczor-retman@intel.com, mail@maciej.szmigiero.name, maz@kernel.org, mic@digikod.net, michael.roth@amd.com, mpe@ellerman.id.au, muchun.song@linux.dev, nikunj@amd.com, nsaenz@amazon.es, oliver.upton@linux.dev, palmer@dabbelt.com, pankaj.gupta@amd.com, paul.walmsley@sifive.com, pbonzini@redhat.com, pdurrant@amazon.co.uk, peterx@redhat.com, pgonda@google.com, pvorel@suse.cz, qperret@google.com, quic_cvanscha@quicinc.com, quic_eberman@quicinc.com, quic_mnalajal@quicinc.com, quic_pderrin@quicinc.com, quic_pheragu@quicinc.com, quic_svaddagi@quicinc.com, quic_tsoni@quicinc.com, richard.weiyang@gmail.com, rick.p.edgecombe@intel.com, rientjes@google.com, roypat@amazon.co.uk, rppt@kernel.org, seanjc@google.com, shuah@kernel.org, steven.price@arm.com, steven.sistare@oracle.com, suzuki.poulose@arm.com, tabba@google.com, thomas.lendacky@amd.com, usama.arif@bytedance.com, vannapurve@google.com, vbabka@suse.cz, viro@zeniv.linux.org.uk, vkuznets@redhat.com, wei.w.wang@intel.com, will@kernel.org, willy@infradead.org, xiaoyao.li@intel.com, yan.y.zhao@intel.com, yilun.xu@intel.com, yuzenghui@huawei.com, zhiquan1.li@intel.com Content-Type: text/plain; charset="UTF-8" X-Stat-Signature: uhymgmd9qpusik3e73tq8yi9nrri4qx7 X-Rspam-User: X-Rspamd-Queue-Id: 9434440002 X-Rspamd-Server: rspam06 X-HE-Tag: 1747266229-76441 X-HE-Meta: U2FsdGVkX18boqJyRkw7k1aXGjUZTJJkhswoKybGicSShTphl9XEGZH8FWseBlxbt+OO3fPka6JpRW+JkrJqzxkBNK1QYam25wz9AA3KGqMhYfVZucva1wpmhYBlol/NBZr4xsJJQDvDgR5QV+nJcRRz9gRDFwJ8q2JyD7Ji3tSN2pgw+ikwUpq7d8Dwt/f+LXAPYM5VgQ/LbpkQm5PETnY4xQvv7tj0CiSvIA7Vqh9JKayV2hm4RiOn5Ytal0l5XUn/l4+kBTPC0puYiD+1GKMyDe9Fm3ka/IiWWeb3rz0Qd5gUp7fUeTPZgbOdXGGxtwmnvMRHTiSfBBV34vJRdrixnhloZTG8pQEtzGk0GAPk3u/bpxO8ktBjK+LbODz+ZYdlCCbfq+8TJgum8TiyUBTQ/EoqoERCqg08ipOM0Y2euybAnKpfDvuspYKTBSRtvycRUDvcYYlwOanqETDulLtwCs7zHe3YTeIjuLPYXlkDc7//xg+5RfQngl4eH3pXs6J1t2jN1HUQ/bVDdrTKOgFH1V/dxFTnhVwDJT73pNiIC5ENipT8T2yJw4GD6BXRzS3h3HeI6Njvqfeon6tAy8eEoC9+6c55NE1hvFbJKiLAxaJ+pCGjwIdZNYqnAIeROPRsPeNABLNycL7Y3czIkt5w8S20kV+L19jh9TZL+Kqsl01v1YSEjeZB9WxqN/XAbjGnXAUV8tnZCi8UUNMzlcAVV0tC+bR0A1vERfmKy3nPPRxjrJeDP2V+GAbusSmpqBnXjsJcE4jS40XbjkTcqFXb0RhSNBU7qOYE069ZIxfZUzv1m2St0H7WbbUPCxX7+VMTSSHsndwAtg3+Q/KviS1vJaEcyWXkxoDIYPbFYr/Z3jCIii69PfFY8cWF/zPaKDp1jLaO+K+S2F3QXxV7gAtjyf3WhUcEUubqvvAi63K/ngEgcmofnhSIslnNXfzNAieZyIozoDzlF0fhy1a DowYWyan Su9Lyw7+MweG6VhOM7bZUSKK39bnmW0sVdnIUQisfmcl3ADYmSgWEytaSftt/9bdbzBrGPjwvMpKdnFtzqALZpBrKnRFDm06LKyduIU3owNnp/euYisGwgeA02hq9gnSZ9fM4qE2mkt4UmUZUzlaUYGAM1rDCYb5foXlmS41F+whX2Du/2E3IB8eNfwOs4WwWQlttzkrfr69ken705auHv4Q7UNul9f5rpEQNo+4u9FO5F/sn6CWUx3SxsqfwR3eUt90aiDxWGmmbWAkp0LOK9l+v8Lkv+ZcV0BgD8sN8LnJiy9tW2W9apOrYjTZVBKuI/IOfNeA6m+E+QDmH1libIAJlBtTpAt4d/aFUd1EnCseP+ZvVM6HiihEoCQ2dgBOiPilXbIuvGqK38I0gHmdvefgI5CVQu+dvUlnYuOHL+QB+TRPtShSqNmcBj4kNPpprULNMPtZDSuw4KqMyzuTT3nkJur1KnAZjD15ovldJ5MdDGkc6njNFm2qr+qZoujfB4Gy4UHmZNURN19cwmE5obNmR6Bg9Cx3bIfXujggljIp/6ZNbS04HaC+AjNq9wTlgpoWruLxIB4I5jhcg5QgsWWBjLs8ORc/lisyWpD2U9odZvHZtNA7fLxHcVv6NOh3wUfXfjKenTcbh7HW69esfWNeykVw6eZWMkS4wG5QdZTSQ01WFtSDsC6smdfO+wHNbEgrA5m5mLIB8zyY= 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: If a custom allocator is requested at guest_memfd creation time, pages from the custom allocator will be used to back guest_memfd. Change-Id: I59df960b3273790f42fe5bea54a234f40962eb75 Signed-off-by: Ackerley Tng --- mm/memory.c | 1 + virt/kvm/guest_memfd.c | 142 +++++++++++++++++++++++++++++++++++++---- 2 files changed, 132 insertions(+), 11 deletions(-) diff --git a/mm/memory.c b/mm/memory.c index ba3ea0a82f7f..3af45e96913c 100644 --- a/mm/memory.c +++ b/mm/memory.c @@ -7249,6 +7249,7 @@ void folio_zero_user(struct folio *folio, unsigned long addr_hint) else process_huge_page(addr_hint, nr_pages, clear_subpage, folio); } +EXPORT_SYMBOL_GPL(folio_zero_user); static int copy_user_gigantic_page(struct folio *dst, struct folio *src, unsigned long addr_hint, diff --git a/virt/kvm/guest_memfd.c b/virt/kvm/guest_memfd.c index c65d93c5a443..24d270b9b725 100644 --- a/virt/kvm/guest_memfd.c +++ b/virt/kvm/guest_memfd.c @@ -478,15 +478,13 @@ static inline void kvm_gmem_mark_prepared(struct folio *folio) * leaking host data and the up-to-date flag is set. */ static int kvm_gmem_prepare_folio(struct kvm *kvm, struct kvm_memory_slot *slot, - gfn_t gfn, struct folio *folio) + gfn_t gfn, struct folio *folio, + unsigned long addr_hint) { - unsigned long nr_pages, i; pgoff_t index; int r; - nr_pages = folio_nr_pages(folio); - for (i = 0; i < nr_pages; i++) - clear_highpage(folio_page(folio, i)); + folio_zero_user(folio, addr_hint); /* * Preparing huge folios should always be safe, since it should @@ -554,7 +552,9 @@ static int kvm_gmem_filemap_add_folio(struct address_space *mapping, */ static struct folio *kvm_gmem_get_folio(struct inode *inode, pgoff_t index) { + size_t allocated_size; struct folio *folio; + pgoff_t index_floor; int ret; repeat: @@ -581,8 +581,10 @@ static struct folio *kvm_gmem_get_folio(struct inode *inode, pgoff_t index) return ERR_PTR(ret); } } + allocated_size = folio_size(folio); - ret = kvm_gmem_filemap_add_folio(inode->i_mapping, folio, index); + index_floor = round_down(index, folio_nr_pages(folio)); + ret = kvm_gmem_filemap_add_folio(inode->i_mapping, folio, index_floor); if (ret) { folio_put(folio); @@ -598,7 +600,17 @@ static struct folio *kvm_gmem_get_folio(struct inode *inode, pgoff_t index) return ERR_PTR(ret); } - __folio_set_locked(folio); + spin_lock(&inode->i_lock); + inode->i_blocks += allocated_size / 512; + spin_unlock(&inode->i_lock); + + /* + * folio is the one that is allocated, this gets the folio at the + * requested index. + */ + folio = page_folio(folio_file_page(folio, index)); + folio_lock(folio); + return folio; } @@ -736,6 +748,92 @@ static void kvm_gmem_truncate_inode_aligned_pages(struct inode *inode, spin_unlock(&inode->i_lock); } +/** + * kvm_gmem_zero_range() - Zeroes all sub-pages in range [@start, @end). + * + * @mapping: the filemap to remove this range from. + * @start: index in filemap for start of range (inclusive). + * @end: index in filemap for end of range (exclusive). + * + * The pages in range may be split. truncate_inode_pages_range() isn't the right + * function because it removes pages from the page cache; this function only + * zeroes the pages. + */ +static void kvm_gmem_zero_range(struct address_space *mapping, + pgoff_t start, pgoff_t end) +{ + struct folio_batch fbatch; + + folio_batch_init(&fbatch); + while (filemap_get_folios(mapping, &start, end - 1, &fbatch)) { + unsigned int i; + + for (i = 0; i < folio_batch_count(&fbatch); ++i) { + struct folio *f; + size_t nr_bytes; + + f = fbatch.folios[i]; + nr_bytes = offset_in_folio(f, end << PAGE_SHIFT); + if (nr_bytes == 0) + nr_bytes = folio_size(f); + + folio_zero_segment(f, 0, nr_bytes); + } + + folio_batch_release(&fbatch); + cond_resched(); + } +} + +/** + * kvm_gmem_truncate_inode_range() - Truncate pages in range [@lstart, @lend). + * + * @inode: inode to truncate from. + * @lstart: offset in inode for start of range (inclusive). + * @lend: offset in inode for end of range (exclusive). + * + * Removes full (huge)pages from the filemap and zeroing incomplete + * (huge)pages. The pages in the range may be split. + */ +static void kvm_gmem_truncate_inode_range(struct inode *inode, loff_t lstart, + loff_t lend) +{ + pgoff_t full_hpage_start; + size_t nr_per_huge_page; + pgoff_t full_hpage_end; + size_t nr_pages; + pgoff_t start; + pgoff_t end; + void *priv; + + priv = kvm_gmem_allocator_private(inode); + nr_per_huge_page = kvm_gmem_allocator_ops(inode)->nr_pages_in_folio(priv); + + start = lstart >> PAGE_SHIFT; + end = min(lend, i_size_read(inode)) >> PAGE_SHIFT; + + full_hpage_start = round_up(start, nr_per_huge_page); + full_hpage_end = round_down(end, nr_per_huge_page); + + if (start < full_hpage_start) { + pgoff_t zero_end = min(full_hpage_start, end); + + kvm_gmem_zero_range(inode->i_mapping, start, zero_end); + } + + if (full_hpage_end > full_hpage_start) { + nr_pages = full_hpage_end - full_hpage_start; + kvm_gmem_truncate_inode_aligned_pages(inode, full_hpage_start, + nr_pages); + } + + if (end > full_hpage_end && end > full_hpage_start) { + pgoff_t zero_start = max(full_hpage_end, start); + + kvm_gmem_zero_range(inode->i_mapping, zero_start, end); + } +} + static long kvm_gmem_punch_hole(struct inode *inode, loff_t offset, loff_t len) { struct list_head *gmem_list = &inode->i_mapping->i_private_list; @@ -752,7 +850,12 @@ static long kvm_gmem_punch_hole(struct inode *inode, loff_t offset, loff_t len) list_for_each_entry(gmem, gmem_list, entry) kvm_gmem_invalidate_begin(gmem, start, end); - truncate_inode_pages_range(inode->i_mapping, offset, offset + len - 1); + if (kvm_gmem_has_custom_allocator(inode)) { + kvm_gmem_truncate_inode_range(inode, offset, offset + len); + } else { + /* Page size is PAGE_SIZE, so use optimized truncation function. */ + truncate_inode_pages_range(inode->i_mapping, offset, offset + len - 1); + } list_for_each_entry(gmem, gmem_list, entry) kvm_gmem_invalidate_end(gmem, start, end); @@ -776,6 +879,16 @@ static long kvm_gmem_allocate(struct inode *inode, loff_t offset, loff_t len) start = offset >> PAGE_SHIFT; end = (offset + len) >> PAGE_SHIFT; + if (kvm_gmem_has_custom_allocator(inode)) { + size_t nr_pages; + void *p; + + p = kvm_gmem_allocator_private(inode); + nr_pages = kvm_gmem_allocator_ops(inode)->nr_pages_in_folio(p); + + start = round_down(start, nr_pages); + end = round_down(end, nr_pages); + } r = 0; for (index = start; index < end; ) { @@ -1570,7 +1683,7 @@ static struct folio *__kvm_gmem_get_pfn(struct file *file, *pfn = folio_file_pfn(folio, index); if (max_order) - *max_order = 0; + *max_order = folio_order(folio); *is_prepared = folio_test_uptodate(folio); return folio; @@ -1597,8 +1710,15 @@ int kvm_gmem_get_pfn(struct kvm *kvm, struct kvm_memory_slot *slot, goto out; } - if (!is_prepared) - r = kvm_gmem_prepare_folio(kvm, slot, gfn, folio); + if (!is_prepared) { + /* + * Use the same address as hugetlb for zeroing private pages + * that won't be mapped to userspace anyway. + */ + unsigned long addr_hint = folio->index << PAGE_SHIFT; + + r = kvm_gmem_prepare_folio(kvm, slot, gfn, folio, addr_hint); + } folio_unlock(folio); -- 2.49.0.1045.g170613ef41-goog