On Wed, Aug 14, 2024 at 07:14:41PM -0300, Jason Gunthorpe wrote: > On Wed, Aug 14, 2024 at 02:24:47PM -0400, Peter Xu wrote: > > On Wed, Aug 14, 2024 at 10:19:54AM -0300, Jason Gunthorpe wrote: > > > On Fri, Aug 09, 2024 at 12:08:59PM -0400, Peter Xu wrote: > > > > > > > +/** > > > > + * follow_pfnmap_start() - Look up a pfn mapping at a user virtual address > > > > + * @args: Pointer to struct @follow_pfnmap_args > > > > + * > > > > + * The caller needs to setup args->vma and args->address to point to the > > > > + * virtual address as the target of such lookup. On a successful return, > > > > + * the results will be put into other output fields. > > > > + * > > > > + * After the caller finished using the fields, the caller must invoke > > > > + * another follow_pfnmap_end() to proper releases the locks and resources > > > > + * of such look up request. > > > > + * > > > > + * During the start() and end() calls, the results in @args will be valid > > > > + * as proper locks will be held. After the end() is called, all the fields > > > > + * in @follow_pfnmap_args will be invalid to be further accessed. > > > > + * > > > > + * If the PTE maps a refcounted page, callers are responsible to protect > > > > + * against invalidation with MMU notifiers; otherwise access to the PFN at > > > > + * a later point in time can trigger use-after-free. > > > > + * > > > > + * Only IO mappings and raw PFN mappings are allowed. > > > > > > What does this mean? The paragraph before said this can return a > > > refcounted page? > > > > This came from the old follow_pte(), I kept that as I suppose we should > > allow VM_IO | VM_PFNMAP just like before, even if in this case I suppose > > only the pfnmap matters where huge mappings can start to appear. > > If that is the intention it should actively block returning anything > that is vm_normal_page() not check the VM flags, see the other > discussion.. The restriction should only be applied to the vma attributes, not a specific pte mapping, IMHO. I mean, the comment was describing "which VMA is allowed to use this function", reflecting that we'll fail at anything !PFNMAP && !IO. It seems legal to have private mappings of them, where vm_normal_page() can return true here for some of the mappings under PFNMAP|IO. IIUC either the old follow_pte() or follow_pfnmap*() API cared much on this part yet so far. > > It makes sense as a restriction if you call the API follow pfnmap. I'm open to any better suggestion to names. Again, I think here it's more about the vma attribute, not "every mapping under the memory range". > > > > > + * The mmap semaphore > > > > + * should be taken for read, and the mmap semaphore cannot be released > > > > + * before the end() is invoked. > > > > > > This function is not safe for IO mappings and PFNs either, VFIO has a > > > known security issue to call it. That should be emphasised in the > > > comment. > > > > Any elaboration on this? I could have missed that.. > > Just because the memory is a PFN or IO doesn't mean it is safe to > access it without a refcount. There are many driver scenarios where > revoking a PFN from mmap needs to be a hard fence that nothing else > has access to that PFN. Otherwise it is a security problem for that > driver. Oh ok, I suppose you meant the VFIO whole thing on "zapping mapping when MMIO disabled"? If so I get it. More below. > > > I suppose so? As the pgtable is stable, I thought it means it's safe, but > > I'm not sure now when you mentioned there's a VFIO known issue, so I could > > have overlooked something. There's no address returned, but pfn, pgprot, > > write, etc. > > zap/etc will wait on the PTL, I think, so it should be safe for at > least the issues I am thinking of. > > > The user needs to do proper mapping if they need an usable address, > > e.g. generic_access_phys() does ioremap_prot() and recheck the pfn didn't > > change. > > No, you can't take the phys_addr_t outside the start/end region that > explicitly holds the lock protecting it. This is what the comment must > warn against doing. I think the comment has that part covered more or less: * During the start() and end() calls, the results in @args will be valid * as proper locks will be held. After the end() is called, all the fields * in @follow_pfnmap_args will be invalid to be further accessed. Feel free to suggest anything that will make it better. For generic_access_phys() as a specific example: I think it is safe to map the pfn even after end(). I meant here the "map" operation is benign with ioremap_prot(), afaiu: it doesn't include an access on top of the mapping yet. After the map, it rewalks the pgtable, making sure PFN is still there and valid, and it'll only access it this time before end(): if (write) memcpy_toio(maddr + offset, buf, len); else memcpy_fromio(buf, maddr + offset, len); ret = len; follow_pfnmap_end(&args); If PFN changed, it properly releases the mapping: if ((prot != pgprot_val(args.pgprot)) || (phys_addr != (args.pfn << PAGE_SHIFT)) || (writable != args.writable)) { follow_pfnmap_end(&args); iounmap(maddr); goto retry; } Then taking the example of VFIO: there's no risk of racing with a concurrent zapping as far as I can see, because otherwise it'll see pfn changed. Thanks, -- Peter Xu