From nobody Mon Feb 9 15:27:14 2026 Received: from mail-pf1-f201.google.com (mail-pf1-f201.google.com [209.85.210.201]) (using TLSv1.2 with cipher ECDHE-RSA-AES128-GCM-SHA256 (128/128 bits)) (No client certificate requested) by smtp.subspace.kernel.org (Postfix) with ESMTPS id BB75C33B6DB for ; Thu, 29 Jan 2026 01:16:05 +0000 (UTC) Authentication-Results: smtp.subspace.kernel.org; arc=none smtp.client-ip=209.85.210.201 ARC-Seal: i=1; a=rsa-sha256; d=subspace.kernel.org; s=arc-20240116; t=1769649367; cv=none; b=cRmQZ4YJsZxNG4UctkluYSQZIRyIA7x/MjJFzYiKOzBzPURBMMFEow8kPV/a7RMkTRU5yw1UoiYwKu11Da8KhPLxeOIJJxaOjuqoh3a5QEplrgRRmIPEtXLTvEfhUVX6NWnpcwY/WNdWHT+ZoWX49A9RTb11ZEjG6o4B3VI1gck= ARC-Message-Signature: i=1; a=rsa-sha256; d=subspace.kernel.org; s=arc-20240116; t=1769649367; c=relaxed/simple; bh=D1RcyFmhh8+6broUyuRXqMJfbz3G6P8KH0L0jZI3CCE=; h=Date:In-Reply-To:Mime-Version:References:Message-ID:Subject:From: To:Cc:Content-Type; b=n9jTPdhZpSqB7sCfmnlKXFTRuaPO+xd1nxdM6DdeJe1DE8QKvzDsWmcknnFLR/w5Y7c+8QbCazjzshox5ety8R+D6fd3pekJ/1Zkot7+E2SBv05jlNOGt3SJQqsSFIi1FxOugGr7JL/jxh7NUoBy4amYq7B2RzmwCHsdr8tItEM= ARC-Authentication-Results: i=1; smtp.subspace.kernel.org; dmarc=pass (p=reject dis=none) header.from=google.com; spf=pass smtp.mailfrom=flex--seanjc.bounces.google.com; dkim=pass (2048-bit key) header.d=google.com header.i=@google.com header.b=yRs+JtwI; arc=none smtp.client-ip=209.85.210.201 Authentication-Results: smtp.subspace.kernel.org; dmarc=pass (p=reject dis=none) header.from=google.com Authentication-Results: smtp.subspace.kernel.org; spf=pass smtp.mailfrom=flex--seanjc.bounces.google.com Authentication-Results: smtp.subspace.kernel.org; dkim=pass (2048-bit key) header.d=google.com header.i=@google.com header.b="yRs+JtwI" Received: by mail-pf1-f201.google.com with SMTP id d2e1a72fcca58-81c68fef4d4so1144236b3a.2 for ; Wed, 28 Jan 2026 17:16:05 -0800 (PST) DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=google.com; s=20230601; t=1769649365; x=1770254165; darn=vger.kernel.org; h=cc:to:from:subject:message-id:references:mime-version:in-reply-to :date:reply-to:from:to:cc:subject:date:message-id:reply-to; bh=Ex0+d/EF8QHeKOGyYN3KDqeLevHaBCQA5Sl9ZATFjF0=; b=yRs+JtwI9bAzB0ajJ90k5KMreMdzCCSe2hKvtVkwILGNrW19hediRw2PIVqdIobvd8 5dEqWeyVZNBIYuh2k0+op0ZZ8MWgFSOH70HBuH201QbCfKvR3ENdjvkOB2FIPB67xaUh U9e1WfGKXWo7OnOZlYJ//CyjH6olSaF8oNqo4WrstBs+09vU4Lu3AyvqP/v8kW5bh7Yd xl/fW5gE0j28AjdoTKv7i8ogpiQgMpvocWQT6vYOGWbE4VqXvJ8czDoSNgbE2CTuVDmS 6bDfQ4LD+jFMqOjYxwyT4+RtEKVRMOXGbcXzAFJ2sWtJ6ALs8NmXdeWMtEJXjWhVSitj XlKQ== X-Google-DKIM-Signature: v=1; a=rsa-sha256; c=relaxed/relaxed; d=1e100.net; s=20230601; t=1769649365; x=1770254165; h=cc:to:from:subject:message-id:references:mime-version:in-reply-to :date:reply-to:x-gm-message-state:from:to:cc:subject:date:message-id :reply-to; bh=Ex0+d/EF8QHeKOGyYN3KDqeLevHaBCQA5Sl9ZATFjF0=; b=XIQwhNFwpKSgiJlyKc7UymTck9YFnVmDUplGpPDRWLnGpCiiDm1XHPuBH7BQC2E/gE ul5p2cjtICAtfmqLSzhEvovDtUIxfp2+KGpqMWtfGYCfMfGT0KcsVuiLNLorpVoJiPZy jJLEwerOU7DY2bAH/A1lvB9PJeC+NFukWW5UabY5YXtOS80xua8erjvvyNYXmu1mOi1Q o9z2ERGYDrMV7fGMS2bFyVGl8BjVl7HxTKF1Iv7KPPwE99Um2tH16tFvr0VEPC9b8hrJ aM+dIiE7b1qG6PuRrs4Yi9NUglveeanR210xJsGEHPe/dZE7rMhqRJ8ZKfA5Hc0ixM4U nfQg== X-Gm-Message-State: AOJu0YweCVB46HiLYUmhm/gPwED+t/bTjhgymBQ9jWmQfA3IxOHJYNLW 58vtGl5S0OgUCKXqHosZgdehjiHpj6dk3P4a9eTgsj6VOlh5nL+SwSZ/9vnW7VLK7uetZJ+ZLKI ta966Yg== X-Received: from pfbif13.prod.google.com ([2002:a05:6a00:8b0d:b0:76b:f0d4:ac71]) (user=seanjc job=prod-delivery.src-stubby-dispatcher) by 2002:a05:6a00:2d82:b0:821:80bf:590b with SMTP id d2e1a72fcca58-823692a3747mr6686129b3a.34.1769649365031; Wed, 28 Jan 2026 17:16:05 -0800 (PST) Reply-To: Sean Christopherson Date: Wed, 28 Jan 2026 17:14:51 -0800 In-Reply-To: <20260129011517.3545883-1-seanjc@google.com> Precedence: bulk X-Mailing-List: linux-kernel@vger.kernel.org List-Id: List-Subscribe: List-Unsubscribe: Mime-Version: 1.0 References: <20260129011517.3545883-1-seanjc@google.com> X-Mailer: git-send-email 2.53.0.rc1.217.geba53bf80e-goog Message-ID: <20260129011517.3545883-20-seanjc@google.com> Subject: [RFC PATCH v5 19/45] KVM: Allow owner of kvm_mmu_memory_cache to provide a custom page allocator From: Sean Christopherson To: Thomas Gleixner , Ingo Molnar , Borislav Petkov , Dave Hansen , x86@kernel.org, Kiryl Shutsemau , Sean Christopherson , Paolo Bonzini Cc: linux-kernel@vger.kernel.org, linux-coco@lists.linux.dev, kvm@vger.kernel.org, Kai Huang , Rick Edgecombe , Yan Zhao , Vishal Annapurve , Ackerley Tng , Sagi Shahar , Binbin Wu , Xiaoyao Li , Isaku Yamahata Content-Transfer-Encoding: quoted-printable Content-Type: text/plain; charset="utf-8" Extend "struct kvm_mmu_memory_cache" to support a custom page allocator so that x86's TDX can update per-page metadata on allocation and free(). Name the allocator page_get() to align with __get_free_page(), e.g. to communicate that it returns an "unsigned long", not a "struct page", and to avoid collisions with macros, e.g. with alloc_page. Suggested-by: Kai Huang Signed-off-by: Sean Christopherson Reviewed-by: Kai Huang --- include/linux/kvm_types.h | 2 ++ virt/kvm/kvm_main.c | 7 ++++++- 2 files changed, 8 insertions(+), 1 deletion(-) diff --git a/include/linux/kvm_types.h b/include/linux/kvm_types.h index a568d8e6f4e8..87fa9deffdb7 100644 --- a/include/linux/kvm_types.h +++ b/include/linux/kvm_types.h @@ -112,6 +112,8 @@ struct kvm_mmu_memory_cache { gfp_t gfp_custom; u64 init_value; struct kmem_cache *kmem_cache; + unsigned long (*page_get)(gfp_t gfp); + void (*page_free)(unsigned long addr); int capacity; int nobjs; void **objects; diff --git a/virt/kvm/kvm_main.c b/virt/kvm/kvm_main.c index 571cf0d6ec01..7015edce5bd8 100644 --- a/virt/kvm/kvm_main.c +++ b/virt/kvm/kvm_main.c @@ -356,7 +356,10 @@ static inline void *mmu_memory_cache_alloc_obj(struct = kvm_mmu_memory_cache *mc, if (mc->kmem_cache) return kmem_cache_alloc(mc->kmem_cache, gfp_flags); =20 - page =3D (void *)__get_free_page(gfp_flags); + if (mc->page_get) + page =3D (void *)mc->page_get(gfp_flags); + else + page =3D (void *)__get_free_page(gfp_flags); if (page && mc->init_value) memset64(page, mc->init_value, PAGE_SIZE / sizeof(u64)); return page; @@ -416,6 +419,8 @@ void kvm_mmu_free_memory_cache(struct kvm_mmu_memory_ca= che *mc) while (mc->nobjs) { if (mc->kmem_cache) kmem_cache_free(mc->kmem_cache, mc->objects[--mc->nobjs]); + else if (mc->page_free) + mc->page_free((unsigned long)mc->objects[--mc->nobjs]); else free_page((unsigned long)mc->objects[--mc->nobjs]); } --=20 2.53.0.rc1.217.geba53bf80e-goog