On Thu, Aug 15, 2024 at 11:41:39AM -0400, Peter Xu wrote: > 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. Why? Either the function only returns PFN map no-struct page things or it returns struct page stuff too, in which case why bother to check the VMA flags if the caller already has to be correct for struct page backed results? This function is only safe to use under the proper locking, and under those rules it doesn't matter at all what the result is.. > > > > > + * 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. And more.. > > > 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. Be much more specific and scary: Any physical address obtained through this API is only valid while the @follow_pfnmap_args. Continuing to use the address after end(), without some other means to synchronize with page table updates will create a security bug. > For generic_access_phys() as a specific example: I think it is safe to map > the pfn even after end(). The map could be safe, but also the memory could be hot unplugged as a race. I don't know either way if all arch code is safe for that. > 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); Yes > 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. VFIO dumps the physical address into the IOMMU and ignores zap. Concurrent zap results in a UAF through the IOMMU mapping. Jason