Re: [PATCH v3 6/6] x86/virt/tdx: Support DPAMT when adding memory for the extensions

[Tony Lindgren <tony.lindgren@linux.intel.com>]

On Thu, Oct 08, 2026 at 12:39:20PM +0800, Xu Yilun wrote:
> On Tue, Oct 06, 2026 at 07:36:50AM +0300, Tony Lindgren wrote:
> > On Tue, Oct 06, 2026 at 01:41:50AM +0800, Xu Yilun wrote:
> > > The TDX module uses Physical Address Metadata Table (PAMT) to track some
> > > state for each page of physical memory that it might use. 3 levels of
> > > PAMTs are used to track pages of different sizes - 1GB, 2MB and 4KB.
> > > Dynamic PAMT (DPAMT) allows saving memory by allocating 4KB PAMT
> > > dynamically, while the 1GB and 2MB levels remain allocated on TDX module
> > > initialization. The kernel has helpers to install 4K DPAMT.
> > > 
> > > Although the memory for the extensions is tens of megabytes and the host
> > > allocates it in a large chunk, the TDX module accepts it at 4K
> > > granularity. Thus the host has to install 4K DPAMT for each page of it.
> > > 
> > > Call tdx_pamt_get() for each page before adding it to TDX module.
> > > Normally, these operations should not fail, and if they do,
> > > intentionally keep the error handling as simple as before - leak all
> > > pages, including the installed 4K DPAMT metadata.
> > > 
> > > Signed-off-by: Xu Yilun <yilun.xu@linux.intel.com>
> > > ---
> > > v3:
> > >  - New patch
> > > ---
> > >  arch/x86/virt/vmx/tdx/tdx.c | 9 ++++++++-
> > >  1 file changed, 8 insertions(+), 1 deletion(-)
> > > 
> > > diff --git a/arch/x86/virt/vmx/tdx/tdx.c b/arch/x86/virt/vmx/tdx/tdx.c
> > > index 5d5f9da1ab02..27a7039ee443 100644
> > > --- a/arch/x86/virt/vmx/tdx/tdx.c
> > > +++ b/arch/x86/virt/vmx/tdx/tdx.c
> > > @@ -1330,8 +1330,15 @@ static __init int tdx_ext_mem_setup(void)
> > >  		struct page *chunk = page + added_pages;
> > >  		unsigned int i;
> > >  
> > > -		for (i = 0; i < chunk_pages; i++)
> > > +		for (i = 0; i < chunk_pages; i++) {
> > > +			ret = tdx_pamt_get(page_to_pfn(chunk + i), NULL);
> > > +			if (ret) {
> > > +				WARN(1, "DPAMT setup error for extensions, stranded all pages\n");
> > > +				goto out_free_hpa_list;
> > > +			}
> > > +
> > >  			hpa_list->phys[i] = page_to_phys(chunk + i);
> > > +		}
> > >  
> > >  		ret = tdx_ext_mem_add(hpa_list, chunk_pages);
> > >  		if (ret) {
> > > 
> > 
> > How about just fold this change in into patch 3/6?
> 
> I think patch 3/6 is already quite heavy. Is it good that I put this
> patch right after patch 3/6?

In that case you should describe that after commit XXX dynamic PAMT is
enabled and tdx_pamt_get() must be used. Otherwise it's not clear that
patch 3/6 will not work without this change.