Re: [PATCH 2/3] userfaultfd: UFFDIO_REMAP uABI

[Date Prev][Date Next][Thread Prev][Thread Next][Date Index][Thread Index]

 



On 21.09.23 20:04, Suren Baghdasaryan wrote:
On Thu, Sep 14, 2023 at 6:45 PM David Hildenbrand <david@xxxxxxxxxx> wrote:

On 14.09.23 20:43, David Hildenbrand wrote:
On 14.09.23 20:11, Matthew Wilcox wrote:
On Thu, Sep 14, 2023 at 08:26:12AM -0700, Suren Baghdasaryan wrote:
+++ b/include/linux/userfaultfd_k.h
@@ -93,6 +93,23 @@ extern int mwriteprotect_range(struct mm_struct *dst_mm,
    extern long uffd_wp_range(struct vm_area_struct *vma,
                       unsigned long start, unsigned long len, bool enable_wp);

+/* remap_pages */
+extern void double_pt_lock(spinlock_t *ptl1, spinlock_t *ptl2);
+extern void double_pt_unlock(spinlock_t *ptl1, spinlock_t *ptl2);
+extern ssize_t remap_pages(struct mm_struct *dst_mm,
+                      struct mm_struct *src_mm,
+                      unsigned long dst_start,
+                      unsigned long src_start,
+                      unsigned long len, __u64 flags);
+extern int remap_pages_huge_pmd(struct mm_struct *dst_mm,
+                           struct mm_struct *src_mm,
+                           pmd_t *dst_pmd, pmd_t *src_pmd,
+                           pmd_t dst_pmdval,
+                           struct vm_area_struct *dst_vma,
+                           struct vm_area_struct *src_vma,
+                           unsigned long dst_addr,
+                           unsigned long src_addr);

Drop the 'extern' markers from function declarations.

+int remap_pages_huge_pmd(struct mm_struct *dst_mm,
+                    struct mm_struct *src_mm,
+                    pmd_t *dst_pmd, pmd_t *src_pmd,
+                    pmd_t dst_pmdval,
+                    struct vm_area_struct *dst_vma,
+                    struct vm_area_struct *src_vma,
+                    unsigned long dst_addr,
+                    unsigned long src_addr)
+{
+   pmd_t _dst_pmd, src_pmdval;
+   struct page *src_page;
+   struct anon_vma *src_anon_vma, *dst_anon_vma;
+   spinlock_t *src_ptl, *dst_ptl;
+   pgtable_t pgtable;
+   struct mmu_notifier_range range;
+
+   src_pmdval = *src_pmd;
+   src_ptl = pmd_lockptr(src_mm, src_pmd);
+
+   BUG_ON(!pmd_trans_huge(src_pmdval));
+   BUG_ON(!pmd_none(dst_pmdval));
+   BUG_ON(!spin_is_locked(src_ptl));
+   mmap_assert_locked(src_mm);
+   mmap_assert_locked(dst_mm);
+   BUG_ON(src_addr & ~HPAGE_PMD_MASK);
+   BUG_ON(dst_addr & ~HPAGE_PMD_MASK);
+
+   src_page = pmd_page(src_pmdval);
+   BUG_ON(!PageHead(src_page));
+   BUG_ON(!PageAnon(src_page));

Better to add a src_folio = page_folio(src_page);
and then folio_test_anon() here.

+   if (unlikely(page_mapcount(src_page) != 1)) {

Brr, this is going to miss PTE mappings of this folio.  I think you
actually want folio_mapcount() instead, although it'd be more efficient
to look at folio->_entire_mapcount == 1 and _nr_pages_mapped == 0.
Not wure what a good name for that predicate would be.

We have

    * It only works on non shared anonymous pages because those can
    * be relocated without generating non linear anon_vmas in the rmap
    * code.
    *
    * It provides a zero copy mechanism to handle userspace page faults.
    * The source vma pages should have mapcount == 1, which can be
    * enforced by using madvise(MADV_DONTFORK) on src vma.

Use PageAnonExclusive(). As long as KSM is not involved and you don't
use fork(), that flag should be good enough for that use case here.

... and similarly don't do any of that swapcount stuff and only check if
the swap pte is anon exclusive.

I'm preparing v2 and this is the only part left for me to address but
I'm not clear how. David, could you please clarify how I should be
checking swap pte to be exclusive without swapcount?

If you have a real swp pte (not a non-swap pte like migration entries) you should be able to just use pte_swp_exclusive().

--
Cheers,

David / dhildenb




[Index of Archives]     [Linux Wireless]     [Linux Kernel]     [ATH6KL]     [Linux Bluetooth]     [Linux Netdev]     [Kernel Newbies]     [Share Photos]     [IDE]     [Security]     [Git]     [Netfilter]     [Bugtraq]     [Yosemite News]     [MIPS Linux]     [ARM Linux]     [Linux Security]     [Linux RAID]     [Linux ATA RAID]     [Samba]     [Device Mapper]

  Powered by Linux