~ [ source navigation ] ~ [ diff markup ] ~ [ identifier search ] ~

TOMOYO Linux Cross Reference
Linux/fs/userfaultfd.c

Version: ~ [ linux-6.11.5 ] ~ [ linux-6.10.14 ] ~ [ linux-6.9.12 ] ~ [ linux-6.8.12 ] ~ [ linux-6.7.12 ] ~ [ linux-6.6.58 ] ~ [ linux-6.5.13 ] ~ [ linux-6.4.16 ] ~ [ linux-6.3.13 ] ~ [ linux-6.2.16 ] ~ [ linux-6.1.114 ] ~ [ linux-6.0.19 ] ~ [ linux-5.19.17 ] ~ [ linux-5.18.19 ] ~ [ linux-5.17.15 ] ~ [ linux-5.16.20 ] ~ [ linux-5.15.169 ] ~ [ linux-5.14.21 ] ~ [ linux-5.13.19 ] ~ [ linux-5.12.19 ] ~ [ linux-5.11.22 ] ~ [ linux-5.10.228 ] ~ [ linux-5.9.16 ] ~ [ linux-5.8.18 ] ~ [ linux-5.7.19 ] ~ [ linux-5.6.19 ] ~ [ linux-5.5.19 ] ~ [ linux-5.4.284 ] ~ [ linux-5.3.18 ] ~ [ linux-5.2.21 ] ~ [ linux-5.1.21 ] ~ [ linux-5.0.21 ] ~ [ linux-4.20.17 ] ~ [ linux-4.19.322 ] ~ [ linux-4.18.20 ] ~ [ linux-4.17.19 ] ~ [ linux-4.16.18 ] ~ [ linux-4.15.18 ] ~ [ linux-4.14.336 ] ~ [ linux-4.13.16 ] ~ [ linux-4.12.14 ] ~ [ linux-4.11.12 ] ~ [ linux-4.10.17 ] ~ [ linux-4.9.337 ] ~ [ linux-4.4.302 ] ~ [ linux-3.10.108 ] ~ [ linux-2.6.32.71 ] ~ [ linux-2.6.0 ] ~ [ linux-2.4.37.11 ] ~ [ unix-v6-master ] ~ [ ccs-tools-1.8.9 ] ~ [ policy-sample ] ~
Architecture: ~ [ i386 ] ~ [ alpha ] ~ [ m68k ] ~ [ mips ] ~ [ ppc ] ~ [ sparc ] ~ [ sparc64 ] ~

Diff markup

Differences between /fs/userfaultfd.c (Version linux-6.11.5) and /fs/userfaultfd.c (Version linux-4.13.16)


  1 // SPDX-License-Identifier: GPL-2.0-only       << 
  2 /*                                                  1 /*
  3  *  fs/userfaultfd.c                                2  *  fs/userfaultfd.c
  4  *                                                  3  *
  5  *  Copyright (C) 2007  Davide Libenzi <davide      4  *  Copyright (C) 2007  Davide Libenzi <davidel@xmailserver.org>
  6  *  Copyright (C) 2008-2009 Red Hat, Inc.           5  *  Copyright (C) 2008-2009 Red Hat, Inc.
  7  *  Copyright (C) 2015  Red Hat, Inc.               6  *  Copyright (C) 2015  Red Hat, Inc.
  8  *                                                  7  *
                                                   >>   8  *  This work is licensed under the terms of the GNU GPL, version 2. See
                                                   >>   9  *  the COPYING file in the top-level directory.
                                                   >>  10  *
  9  *  Some part derived from fs/eventfd.c (anon      11  *  Some part derived from fs/eventfd.c (anon inode setup) and
 10  *  mm/ksm.c (mm hashing).                         12  *  mm/ksm.c (mm hashing).
 11  */                                                13  */
 12                                                    14 
 13 #include <linux/list.h>                            15 #include <linux/list.h>
 14 #include <linux/hashtable.h>                       16 #include <linux/hashtable.h>
 15 #include <linux/sched/signal.h>                    17 #include <linux/sched/signal.h>
 16 #include <linux/sched/mm.h>                        18 #include <linux/sched/mm.h>
 17 #include <linux/mm.h>                              19 #include <linux/mm.h>
 18 #include <linux/mm_inline.h>                   << 
 19 #include <linux/mmu_notifier.h>                << 
 20 #include <linux/poll.h>                            20 #include <linux/poll.h>
 21 #include <linux/slab.h>                            21 #include <linux/slab.h>
 22 #include <linux/seq_file.h>                        22 #include <linux/seq_file.h>
 23 #include <linux/file.h>                            23 #include <linux/file.h>
 24 #include <linux/bug.h>                             24 #include <linux/bug.h>
 25 #include <linux/anon_inodes.h>                     25 #include <linux/anon_inodes.h>
 26 #include <linux/syscalls.h>                        26 #include <linux/syscalls.h>
 27 #include <linux/userfaultfd_k.h>                   27 #include <linux/userfaultfd_k.h>
 28 #include <linux/mempolicy.h>                       28 #include <linux/mempolicy.h>
 29 #include <linux/ioctl.h>                           29 #include <linux/ioctl.h>
 30 #include <linux/security.h>                        30 #include <linux/security.h>
 31 #include <linux/hugetlb.h>                         31 #include <linux/hugetlb.h>
 32 #include <linux/swapops.h>                     !!  32 
 33 #include <linux/miscdevice.h>                  !!  33 static struct kmem_cache *userfaultfd_ctx_cachep __read_mostly;
 34 #include <linux/uio.h>                         !!  34 
 35                                                !!  35 enum userfaultfd_state {
 36 static int sysctl_unprivileged_userfaultfd __r !!  36         UFFD_STATE_WAIT_API,
 37                                                !!  37         UFFD_STATE_RUNNING,
 38 #ifdef CONFIG_SYSCTL                           << 
 39 static struct ctl_table vm_userfaultfd_table[] << 
 40         {                                      << 
 41                 .procname       = "unprivilege << 
 42                 .data           = &sysctl_unpr << 
 43                 .maxlen         = sizeof(sysct << 
 44                 .mode           = 0644,        << 
 45                 .proc_handler   = proc_dointve << 
 46                 .extra1         = SYSCTL_ZERO, << 
 47                 .extra2         = SYSCTL_ONE,  << 
 48         },                                     << 
 49 };                                                 38 };
 50 #endif                                         << 
 51                                                    39 
 52 static struct kmem_cache *userfaultfd_ctx_cach !!  40 /*
                                                   >>  41  * Start with fault_pending_wqh and fault_wqh so they're more likely
                                                   >>  42  * to be in the same cacheline.
                                                   >>  43  */
                                                   >>  44 struct userfaultfd_ctx {
                                                   >>  45         /* waitqueue head for the pending (i.e. not read) userfaults */
                                                   >>  46         wait_queue_head_t fault_pending_wqh;
                                                   >>  47         /* waitqueue head for the userfaults */
                                                   >>  48         wait_queue_head_t fault_wqh;
                                                   >>  49         /* waitqueue head for the pseudo fd to wakeup poll/read */
                                                   >>  50         wait_queue_head_t fd_wqh;
                                                   >>  51         /* waitqueue head for events */
                                                   >>  52         wait_queue_head_t event_wqh;
                                                   >>  53         /* a refile sequence protected by fault_pending_wqh lock */
                                                   >>  54         struct seqcount refile_seq;
                                                   >>  55         /* pseudo fd refcounting */
                                                   >>  56         atomic_t refcount;
                                                   >>  57         /* userfaultfd syscall flags */
                                                   >>  58         unsigned int flags;
                                                   >>  59         /* features requested from the userspace */
                                                   >>  60         unsigned int features;
                                                   >>  61         /* state machine */
                                                   >>  62         enum userfaultfd_state state;
                                                   >>  63         /* released */
                                                   >>  64         bool released;
                                                   >>  65         /* mm with one ore more vmas attached to this userfaultfd_ctx */
                                                   >>  66         struct mm_struct *mm;
                                                   >>  67 };
 53                                                    68 
 54 struct userfaultfd_fork_ctx {                      69 struct userfaultfd_fork_ctx {
 55         struct userfaultfd_ctx *orig;              70         struct userfaultfd_ctx *orig;
 56         struct userfaultfd_ctx *new;               71         struct userfaultfd_ctx *new;
 57         struct list_head list;                     72         struct list_head list;
 58 };                                                 73 };
 59                                                    74 
 60 struct userfaultfd_unmap_ctx {                     75 struct userfaultfd_unmap_ctx {
 61         struct userfaultfd_ctx *ctx;               76         struct userfaultfd_ctx *ctx;
 62         unsigned long start;                       77         unsigned long start;
 63         unsigned long end;                         78         unsigned long end;
 64         struct list_head list;                     79         struct list_head list;
 65 };                                                 80 };
 66                                                    81 
 67 struct userfaultfd_wait_queue {                    82 struct userfaultfd_wait_queue {
 68         struct uffd_msg msg;                       83         struct uffd_msg msg;
 69         wait_queue_entry_t wq;                     84         wait_queue_entry_t wq;
 70         struct userfaultfd_ctx *ctx;               85         struct userfaultfd_ctx *ctx;
 71         bool waken;                                86         bool waken;
 72 };                                                 87 };
 73                                                    88 
 74 struct userfaultfd_wake_range {                    89 struct userfaultfd_wake_range {
 75         unsigned long start;                       90         unsigned long start;
 76         unsigned long len;                         91         unsigned long len;
 77 };                                                 92 };
 78                                                    93 
 79 /* internal indication that UFFD_API ioctl was << 
 80 #define UFFD_FEATURE_INITIALIZED               << 
 81                                                << 
 82 static bool userfaultfd_is_initialized(struct  << 
 83 {                                              << 
 84         return ctx->features & UFFD_FEATURE_IN << 
 85 }                                              << 
 86                                                << 
 87 static bool userfaultfd_wp_async_ctx(struct us << 
 88 {                                              << 
 89         return ctx && (ctx->features & UFFD_FE << 
 90 }                                              << 
 91                                                << 
 92 /*                                             << 
 93  * Whether WP_UNPOPULATED is enabled on the uf << 
 94  * meaningful when userfaultfd_wp()==true on t << 
 95  * anonymous.                                  << 
 96  */                                            << 
 97 bool userfaultfd_wp_unpopulated(struct vm_area << 
 98 {                                              << 
 99         struct userfaultfd_ctx *ctx = vma->vm_ << 
100                                                << 
101         if (!ctx)                              << 
102                 return false;                  << 
103                                                << 
104         return ctx->features & UFFD_FEATURE_WP << 
105 }                                              << 
106                                                << 
107 static void userfaultfd_set_vm_flags(struct vm << 
108                                      vm_flags_ << 
109 {                                              << 
110         const bool uffd_wp_changed = (vma->vm_ << 
111                                                << 
112         vm_flags_reset(vma, flags);            << 
113         /*                                     << 
114          * For shared mappings, we want to ena << 
115          * userfaultfd-wp is enabled (see vma_ << 
116          * recalculate vma->vm_page_prot whene << 
117          */                                    << 
118         if ((vma->vm_flags & VM_SHARED) && uff << 
119                 vma_set_page_prot(vma);        << 
120 }                                              << 
121                                                << 
122 static int userfaultfd_wake_function(wait_queu     94 static int userfaultfd_wake_function(wait_queue_entry_t *wq, unsigned mode,
123                                      int wake_     95                                      int wake_flags, void *key)
124 {                                                  96 {
125         struct userfaultfd_wake_range *range =     97         struct userfaultfd_wake_range *range = key;
126         int ret;                                   98         int ret;
127         struct userfaultfd_wait_queue *uwq;        99         struct userfaultfd_wait_queue *uwq;
128         unsigned long start, len;                 100         unsigned long start, len;
129                                                   101 
130         uwq = container_of(wq, struct userfaul    102         uwq = container_of(wq, struct userfaultfd_wait_queue, wq);
131         ret = 0;                                  103         ret = 0;
132         /* len == 0 means wake all */             104         /* len == 0 means wake all */
133         start = range->start;                     105         start = range->start;
134         len = range->len;                         106         len = range->len;
135         if (len && (start > uwq->msg.arg.pagef    107         if (len && (start > uwq->msg.arg.pagefault.address ||
136                     start + len <= uwq->msg.ar    108                     start + len <= uwq->msg.arg.pagefault.address))
137                 goto out;                         109                 goto out;
138         WRITE_ONCE(uwq->waken, true);             110         WRITE_ONCE(uwq->waken, true);
139         /*                                        111         /*
140          * The Program-Order guarantees provid !! 112          * The implicit smp_mb__before_spinlock in try_to_wake_up()
141          * ensure uwq->waken is visible before !! 113          * renders uwq->waken visible to other CPUs before the task is
                                                   >> 114          * waken.
142          */                                       115          */
143         ret = wake_up_state(wq->private, mode)    116         ret = wake_up_state(wq->private, mode);
144         if (ret) {                             !! 117         if (ret)
145                 /*                                118                 /*
146                  * Wake only once, autoremove     119                  * Wake only once, autoremove behavior.
147                  *                                120                  *
148                  * After the effect of list_de !! 121                  * After the effect of list_del_init is visible to the
149                  * CPUs, the waitqueue may dis !! 122                  * other CPUs, the waitqueue may disappear from under
150                  * !list_empty_careful() in ha !! 123                  * us, see the !list_empty_careful() in
151                  *                             !! 124                  * handle_userfault(). try_to_wake_up() has an
152                  * try_to_wake_up() has an imp !! 125                  * implicit smp_mb__before_spinlock, and the
153                  * wq->private is read before  !! 126                  * wq->private is read before calling the extern
154                  * "wake_up_state" (which in t !! 127                  * function "wake_up_state" (which in turns calls
                                                   >> 128                  * try_to_wake_up). While the spin_lock;spin_unlock;
                                                   >> 129                  * wouldn't be enough, the smp_mb__before_spinlock is
                                                   >> 130                  * enough to avoid an explicit smp_mb() here.
155                  */                               131                  */
156                 list_del_init(&wq->entry);        132                 list_del_init(&wq->entry);
157         }                                      << 
158 out:                                              133 out:
159         return ret;                               134         return ret;
160 }                                                 135 }
161                                                   136 
162 /**                                               137 /**
163  * userfaultfd_ctx_get - Acquires a reference     138  * userfaultfd_ctx_get - Acquires a reference to the internal userfaultfd
164  * context.                                       139  * context.
165  * @ctx: [in] Pointer to the userfaultfd conte    140  * @ctx: [in] Pointer to the userfaultfd context.
166  */                                               141  */
167 static void userfaultfd_ctx_get(struct userfau    142 static void userfaultfd_ctx_get(struct userfaultfd_ctx *ctx)
168 {                                                 143 {
169         refcount_inc(&ctx->refcount);          !! 144         if (!atomic_inc_not_zero(&ctx->refcount))
                                                   >> 145                 BUG();
170 }                                                 146 }
171                                                   147 
172 /**                                               148 /**
173  * userfaultfd_ctx_put - Releases a reference     149  * userfaultfd_ctx_put - Releases a reference to the internal userfaultfd
174  * context.                                       150  * context.
175  * @ctx: [in] Pointer to userfaultfd context.     151  * @ctx: [in] Pointer to userfaultfd context.
176  *                                                152  *
177  * The userfaultfd context reference must have    153  * The userfaultfd context reference must have been previously acquired either
178  * with userfaultfd_ctx_get() or userfaultfd_c    154  * with userfaultfd_ctx_get() or userfaultfd_ctx_fdget().
179  */                                               155  */
180 static void userfaultfd_ctx_put(struct userfau    156 static void userfaultfd_ctx_put(struct userfaultfd_ctx *ctx)
181 {                                                 157 {
182         if (refcount_dec_and_test(&ctx->refcou !! 158         if (atomic_dec_and_test(&ctx->refcount)) {
183                 VM_BUG_ON(spin_is_locked(&ctx-    159                 VM_BUG_ON(spin_is_locked(&ctx->fault_pending_wqh.lock));
184                 VM_BUG_ON(waitqueue_active(&ct    160                 VM_BUG_ON(waitqueue_active(&ctx->fault_pending_wqh));
185                 VM_BUG_ON(spin_is_locked(&ctx-    161                 VM_BUG_ON(spin_is_locked(&ctx->fault_wqh.lock));
186                 VM_BUG_ON(waitqueue_active(&ct    162                 VM_BUG_ON(waitqueue_active(&ctx->fault_wqh));
187                 VM_BUG_ON(spin_is_locked(&ctx-    163                 VM_BUG_ON(spin_is_locked(&ctx->event_wqh.lock));
188                 VM_BUG_ON(waitqueue_active(&ct    164                 VM_BUG_ON(waitqueue_active(&ctx->event_wqh));
189                 VM_BUG_ON(spin_is_locked(&ctx-    165                 VM_BUG_ON(spin_is_locked(&ctx->fd_wqh.lock));
190                 VM_BUG_ON(waitqueue_active(&ct    166                 VM_BUG_ON(waitqueue_active(&ctx->fd_wqh));
191                 mmdrop(ctx->mm);                  167                 mmdrop(ctx->mm);
192                 kmem_cache_free(userfaultfd_ct    168                 kmem_cache_free(userfaultfd_ctx_cachep, ctx);
193         }                                         169         }
194 }                                                 170 }
195                                                   171 
196 static inline void msg_init(struct uffd_msg *m    172 static inline void msg_init(struct uffd_msg *msg)
197 {                                                 173 {
198         BUILD_BUG_ON(sizeof(struct uffd_msg) !    174         BUILD_BUG_ON(sizeof(struct uffd_msg) != 32);
199         /*                                        175         /*
200          * Must use memset to zero out the pad    176          * Must use memset to zero out the paddings or kernel data is
201          * leaked to userland.                    177          * leaked to userland.
202          */                                       178          */
203         memset(msg, 0, sizeof(struct uffd_msg)    179         memset(msg, 0, sizeof(struct uffd_msg));
204 }                                                 180 }
205                                                   181 
206 static inline struct uffd_msg userfault_msg(un    182 static inline struct uffd_msg userfault_msg(unsigned long address,
207                                             un << 
208                                             un    183                                             unsigned int flags,
209                                             un !! 184                                             unsigned long reason)
210                                             un << 
211 {                                                 185 {
212         struct uffd_msg msg;                      186         struct uffd_msg msg;
213                                                << 
214         msg_init(&msg);                           187         msg_init(&msg);
215         msg.event = UFFD_EVENT_PAGEFAULT;         188         msg.event = UFFD_EVENT_PAGEFAULT;
216                                                !! 189         msg.arg.pagefault.address = address;
217         msg.arg.pagefault.address = (features  << 
218                                     real_addre << 
219                                                << 
220         /*                                     << 
221          * These flags indicate why the userfa << 
222          * - UFFD_PAGEFAULT_FLAG_WP indicates  << 
223          * - UFFD_PAGEFAULT_FLAG_MINOR indicat << 
224          * - Neither of these flags being set  << 
225          *                                     << 
226          * Separately, UFFD_PAGEFAULT_FLAG_WRI << 
227          * fault. Otherwise, it was a read fau << 
228          */                                    << 
229         if (flags & FAULT_FLAG_WRITE)             190         if (flags & FAULT_FLAG_WRITE)
                                                   >> 191                 /*
                                                   >> 192                  * If UFFD_FEATURE_PAGEFAULT_FLAG_WP was set in the
                                                   >> 193                  * uffdio_api.features and UFFD_PAGEFAULT_FLAG_WRITE
                                                   >> 194                  * was not set in a UFFD_EVENT_PAGEFAULT, it means it
                                                   >> 195                  * was a read fault, otherwise if set it means it's
                                                   >> 196                  * a write fault.
                                                   >> 197                  */
230                 msg.arg.pagefault.flags |= UFF    198                 msg.arg.pagefault.flags |= UFFD_PAGEFAULT_FLAG_WRITE;
231         if (reason & VM_UFFD_WP)                  199         if (reason & VM_UFFD_WP)
                                                   >> 200                 /*
                                                   >> 201                  * If UFFD_FEATURE_PAGEFAULT_FLAG_WP was set in the
                                                   >> 202                  * uffdio_api.features and UFFD_PAGEFAULT_FLAG_WP was
                                                   >> 203                  * not set in a UFFD_EVENT_PAGEFAULT, it means it was
                                                   >> 204                  * a missing fault, otherwise if set it means it's a
                                                   >> 205                  * write protect fault.
                                                   >> 206                  */
232                 msg.arg.pagefault.flags |= UFF    207                 msg.arg.pagefault.flags |= UFFD_PAGEFAULT_FLAG_WP;
233         if (reason & VM_UFFD_MINOR)            << 
234                 msg.arg.pagefault.flags |= UFF << 
235         if (features & UFFD_FEATURE_THREAD_ID) << 
236                 msg.arg.pagefault.feat.ptid =  << 
237         return msg;                               208         return msg;
238 }                                                 209 }
239                                                   210 
240 #ifdef CONFIG_HUGETLB_PAGE                        211 #ifdef CONFIG_HUGETLB_PAGE
241 /*                                                212 /*
242  * Same functionality as userfaultfd_must_wait    213  * Same functionality as userfaultfd_must_wait below with modifications for
243  * hugepmd ranges.                                214  * hugepmd ranges.
244  */                                               215  */
245 static inline bool userfaultfd_huge_must_wait(    216 static inline bool userfaultfd_huge_must_wait(struct userfaultfd_ctx *ctx,
246                                                !! 217                                          struct vm_area_struct *vma,
247                                                !! 218                                          unsigned long address,
                                                   >> 219                                          unsigned long flags,
                                                   >> 220                                          unsigned long reason)
248 {                                                 221 {
249         struct vm_area_struct *vma = vmf->vma; !! 222         struct mm_struct *mm = ctx->mm;
250         pte_t *ptep, pte;                      !! 223         pte_t *pte;
251         bool ret = true;                          224         bool ret = true;
252                                                   225 
253         assert_fault_locked(vmf);              !! 226         VM_BUG_ON(!rwsem_is_locked(&mm->mmap_sem));
254                                                   227 
255         ptep = hugetlb_walk(vma, vmf->address, !! 228         pte = huge_pte_offset(mm, address, vma_mmu_pagesize(vma));
256         if (!ptep)                             !! 229         if (!pte)
257                 goto out;                         230                 goto out;
258                                                   231 
259         ret = false;                              232         ret = false;
260         pte = huge_ptep_get(vma->vm_mm, vmf->a << 
261                                                   233 
262         /*                                        234         /*
263          * Lockless access: we're in a wait_ev    235          * Lockless access: we're in a wait_event so it's ok if it
264          * changes under us.  PTE markers shou !! 236          * changes under us.
265          * ptes here.                          << 
266          */                                       237          */
267         if (huge_pte_none_mostly(pte))         !! 238         if (huge_pte_none(*pte))
268                 ret = true;                       239                 ret = true;
269         if (!huge_pte_write(pte) && (reason &  !! 240         if (!huge_pte_write(*pte) && (reason & VM_UFFD_WP))
270                 ret = true;                       241                 ret = true;
271 out:                                              242 out:
272         return ret;                               243         return ret;
273 }                                                 244 }
274 #else                                             245 #else
275 static inline bool userfaultfd_huge_must_wait(    246 static inline bool userfaultfd_huge_must_wait(struct userfaultfd_ctx *ctx,
276                                                !! 247                                          struct vm_area_struct *vma,
277                                                !! 248                                          unsigned long address,
                                                   >> 249                                          unsigned long flags,
                                                   >> 250                                          unsigned long reason)
278 {                                                 251 {
279         return false;   /* should never get he    252         return false;   /* should never get here */
280 }                                                 253 }
281 #endif /* CONFIG_HUGETLB_PAGE */                  254 #endif /* CONFIG_HUGETLB_PAGE */
282                                                   255 
283 /*                                                256 /*
284  * Verify the pagetables are still not ok afte    257  * Verify the pagetables are still not ok after having reigstered into
285  * the fault_pending_wqh to avoid userland hav    258  * the fault_pending_wqh to avoid userland having to UFFDIO_WAKE any
286  * userfault that has already been resolved, i !! 259  * userfault that has already been resolved, if userfaultfd_read and
287  * UFFDIO_COPY|ZEROPAGE are being run simultan    260  * UFFDIO_COPY|ZEROPAGE are being run simultaneously on two different
288  * threads.                                       261  * threads.
289  */                                               262  */
290 static inline bool userfaultfd_must_wait(struc    263 static inline bool userfaultfd_must_wait(struct userfaultfd_ctx *ctx,
291                                          struc !! 264                                          unsigned long address,
                                                   >> 265                                          unsigned long flags,
292                                          unsig    266                                          unsigned long reason)
293 {                                                 267 {
294         struct mm_struct *mm = ctx->mm;           268         struct mm_struct *mm = ctx->mm;
295         unsigned long address = vmf->address;  << 
296         pgd_t *pgd;                               269         pgd_t *pgd;
297         p4d_t *p4d;                               270         p4d_t *p4d;
298         pud_t *pud;                               271         pud_t *pud;
299         pmd_t *pmd, _pmd;                         272         pmd_t *pmd, _pmd;
300         pte_t *pte;                               273         pte_t *pte;
301         pte_t ptent;                           << 
302         bool ret = true;                          274         bool ret = true;
303                                                   275 
304         assert_fault_locked(vmf);              !! 276         VM_BUG_ON(!rwsem_is_locked(&mm->mmap_sem));
305                                                   277 
306         pgd = pgd_offset(mm, address);            278         pgd = pgd_offset(mm, address);
307         if (!pgd_present(*pgd))                   279         if (!pgd_present(*pgd))
308                 goto out;                         280                 goto out;
309         p4d = p4d_offset(pgd, address);           281         p4d = p4d_offset(pgd, address);
310         if (!p4d_present(*p4d))                   282         if (!p4d_present(*p4d))
311                 goto out;                         283                 goto out;
312         pud = pud_offset(p4d, address);           284         pud = pud_offset(p4d, address);
313         if (!pud_present(*pud))                   285         if (!pud_present(*pud))
314                 goto out;                         286                 goto out;
315         pmd = pmd_offset(pud, address);           287         pmd = pmd_offset(pud, address);
316 again:                                         !! 288         /*
317         _pmd = pmdp_get_lockless(pmd);         !! 289          * READ_ONCE must function as a barrier with narrower scope
318         if (pmd_none(_pmd))                    !! 290          * and it must be equivalent to:
                                                   >> 291          *      _pmd = *pmd; barrier();
                                                   >> 292          *
                                                   >> 293          * This is to deal with the instability (as in
                                                   >> 294          * pmd_trans_unstable) of the pmd.
                                                   >> 295          */
                                                   >> 296         _pmd = READ_ONCE(*pmd);
                                                   >> 297         if (!pmd_present(_pmd))
319                 goto out;                         298                 goto out;
320                                                   299 
321         ret = false;                              300         ret = false;
322         if (!pmd_present(_pmd) || pmd_devmap(_ !! 301         if (pmd_trans_huge(_pmd))
323                 goto out;                         302                 goto out;
324                                                   303 
325         if (pmd_trans_huge(_pmd)) {            !! 304         /*
326                 if (!pmd_write(_pmd) && (reaso !! 305          * the pmd is stable (as in !pmd_trans_unstable) so we can re-read it
327                         ret = true;            !! 306          * and use the standard pte_offset_map() instead of parsing _pmd.
328                 goto out;                      !! 307          */
329         }                                      << 
330                                                << 
331         pte = pte_offset_map(pmd, address);       308         pte = pte_offset_map(pmd, address);
332         if (!pte) {                            << 
333                 ret = true;                    << 
334                 goto again;                    << 
335         }                                      << 
336         /*                                        309         /*
337          * Lockless access: we're in a wait_ev    310          * Lockless access: we're in a wait_event so it's ok if it
338          * changes under us.  PTE markers shou !! 311          * changes under us.
339          * ptes here.                          << 
340          */                                       312          */
341         ptent = ptep_get(pte);                 !! 313         if (pte_none(*pte))
342         if (pte_none_mostly(ptent))            << 
343                 ret = true;                    << 
344         if (!pte_write(ptent) && (reason & VM_ << 
345                 ret = true;                       314                 ret = true;
346         pte_unmap(pte);                           315         pte_unmap(pte);
347                                                   316 
348 out:                                              317 out:
349         return ret;                               318         return ret;
350 }                                                 319 }
351                                                   320 
352 static inline unsigned int userfaultfd_get_blo << 
353 {                                              << 
354         if (flags & FAULT_FLAG_INTERRUPTIBLE)  << 
355                 return TASK_INTERRUPTIBLE;     << 
356                                                << 
357         if (flags & FAULT_FLAG_KILLABLE)       << 
358                 return TASK_KILLABLE;          << 
359                                                << 
360         return TASK_UNINTERRUPTIBLE;           << 
361 }                                              << 
362                                                << 
363 /*                                                321 /*
364  * The locking rules involved in returning VM_    322  * The locking rules involved in returning VM_FAULT_RETRY depending on
365  * FAULT_FLAG_ALLOW_RETRY, FAULT_FLAG_RETRY_NO    323  * FAULT_FLAG_ALLOW_RETRY, FAULT_FLAG_RETRY_NOWAIT and
366  * FAULT_FLAG_KILLABLE are not straightforward    324  * FAULT_FLAG_KILLABLE are not straightforward. The "Caution"
367  * recommendation in __lock_page_or_retry is n    325  * recommendation in __lock_page_or_retry is not an understatement.
368  *                                                326  *
369  * If FAULT_FLAG_ALLOW_RETRY is set, the mmap_ !! 327  * If FAULT_FLAG_ALLOW_RETRY is set, the mmap_sem must be released
370  * before returning VM_FAULT_RETRY only if FAU    328  * before returning VM_FAULT_RETRY only if FAULT_FLAG_RETRY_NOWAIT is
371  * not set.                                       329  * not set.
372  *                                                330  *
373  * If FAULT_FLAG_ALLOW_RETRY is set but FAULT_    331  * If FAULT_FLAG_ALLOW_RETRY is set but FAULT_FLAG_KILLABLE is not
374  * set, VM_FAULT_RETRY can still be returned i    332  * set, VM_FAULT_RETRY can still be returned if and only if there are
375  * fatal_signal_pending()s, and the mmap_lock  !! 333  * fatal_signal_pending()s, and the mmap_sem must be released before
376  * returning it.                                  334  * returning it.
377  */                                               335  */
378 vm_fault_t handle_userfault(struct vm_fault *v !! 336 int handle_userfault(struct vm_fault *vmf, unsigned long reason)
379 {                                                 337 {
380         struct vm_area_struct *vma = vmf->vma; !! 338         struct mm_struct *mm = vmf->vma->vm_mm;
381         struct mm_struct *mm = vma->vm_mm;     << 
382         struct userfaultfd_ctx *ctx;              339         struct userfaultfd_ctx *ctx;
383         struct userfaultfd_wait_queue uwq;        340         struct userfaultfd_wait_queue uwq;
384         vm_fault_t ret = VM_FAULT_SIGBUS;      !! 341         int ret;
385         bool must_wait;                        !! 342         bool must_wait, return_to_userland;
386         unsigned int blocking_state;           !! 343         long blocking_state;
                                                   >> 344 
                                                   >> 345         ret = VM_FAULT_SIGBUS;
387                                                   346 
388         /*                                        347         /*
389          * We don't do userfault handling for     348          * We don't do userfault handling for the final child pid update.
390          *                                        349          *
391          * We also don't do userfault handling    350          * We also don't do userfault handling during
392          * coredumping. hugetlbfs has the spec    351          * coredumping. hugetlbfs has the special
393          * hugetlb_follow_page_mask() to skip  !! 352          * follow_hugetlb_page() to skip missing pages in the
394          * FOLL_DUMP case, anon memory also ch    353          * FOLL_DUMP case, anon memory also checks for FOLL_DUMP with
395          * the no_page_table() helper in follo    354          * the no_page_table() helper in follow_page_mask(), but the
396          * shmem_vm_ops->fault method is invok    355          * shmem_vm_ops->fault method is invoked even during
397          * coredumping and it ends up here.    !! 356          * coredumping without mmap_sem and it ends up here.
398          */                                       357          */
399         if (current->flags & (PF_EXITING|PF_DU    358         if (current->flags & (PF_EXITING|PF_DUMPCORE))
400                 goto out;                         359                 goto out;
401                                                   360 
402         assert_fault_locked(vmf);              !! 361         /*
                                                   >> 362          * Coredumping runs without mmap_sem so we can only check that
                                                   >> 363          * the mmap_sem is held, if PF_DUMPCORE was not set.
                                                   >> 364          */
                                                   >> 365         WARN_ON_ONCE(!rwsem_is_locked(&mm->mmap_sem));
403                                                   366 
404         ctx = vma->vm_userfaultfd_ctx.ctx;     !! 367         ctx = vmf->vma->vm_userfaultfd_ctx.ctx;
405         if (!ctx)                                 368         if (!ctx)
406                 goto out;                         369                 goto out;
407                                                   370 
408         BUG_ON(ctx->mm != mm);                    371         BUG_ON(ctx->mm != mm);
409                                                   372 
410         /* Any unrecognized flag is a bug. */  !! 373         VM_BUG_ON(reason & ~(VM_UFFD_MISSING|VM_UFFD_WP));
411         VM_BUG_ON(reason & ~__VM_UFFD_FLAGS);  !! 374         VM_BUG_ON(!(reason & VM_UFFD_MISSING) ^ !!(reason & VM_UFFD_WP));
412         /* 0 or > 1 flags set is a bug; we exp << 
413         VM_BUG_ON(!reason || (reason & (reason << 
414                                                << 
415         if (ctx->features & UFFD_FEATURE_SIGBU << 
416                 goto out;                      << 
417         if (!(vmf->flags & FAULT_FLAG_USER) && << 
418                 goto out;                      << 
419                                                   375 
420         /*                                        376         /*
421          * If it's already released don't get     377          * If it's already released don't get it. This avoids to loop
422          * in __get_user_pages if userfaultfd_    378          * in __get_user_pages if userfaultfd_release waits on the
423          * caller of handle_userfault to relea !! 379          * caller of handle_userfault to release the mmap_sem.
424          */                                       380          */
425         if (unlikely(READ_ONCE(ctx->released)) !! 381         if (unlikely(ACCESS_ONCE(ctx->released)))
426                 /*                             << 
427                  * Don't return VM_FAULT_SIGBU << 
428                  * cooperative manager can clo << 
429                  * last UFFDIO_COPY, without r << 
430                  * involuntary SIGBUS if the p << 
431                  * userfaultfd while the userf << 
432                  * (but after the last UFFDIO_ << 
433                  * wasn't already closed when  << 
434                  * this point, that would norm << 
435                  * userfaultfd_must_wait retur << 
436                  *                             << 
437                  * If we were to return VM_FAU << 
438                  * cooperative manager would b << 
439                  * always call UFFDIO_UNREGIST << 
440                  * close the uffd.             << 
441                  */                            << 
442                 ret = VM_FAULT_NOPAGE;         << 
443                 goto out;                         382                 goto out;
444         }                                      << 
445                                                   383 
446         /*                                        384         /*
447          * Check that we can return VM_FAULT_R    385          * Check that we can return VM_FAULT_RETRY.
448          *                                        386          *
449          * NOTE: it should become possible to     387          * NOTE: it should become possible to return VM_FAULT_RETRY
450          * even if FAULT_FLAG_TRIED is set wit    388          * even if FAULT_FLAG_TRIED is set without leading to gup()
451          * -EBUSY failures, if the userfaultfd    389          * -EBUSY failures, if the userfaultfd is to be extended for
452          * VM_UFFD_WP tracking and we intend t    390          * VM_UFFD_WP tracking and we intend to arm the userfault
453          * without first stopping userland acc    391          * without first stopping userland access to the memory. For
454          * VM_UFFD_MISSING userfaults this is     392          * VM_UFFD_MISSING userfaults this is enough for now.
455          */                                       393          */
456         if (unlikely(!(vmf->flags & FAULT_FLAG    394         if (unlikely(!(vmf->flags & FAULT_FLAG_ALLOW_RETRY))) {
457                 /*                                395                 /*
458                  * Validate the invariant that    396                  * Validate the invariant that nowait must allow retry
459                  * to be sure not to return SI    397                  * to be sure not to return SIGBUS erroneously on
460                  * nowait invocations.            398                  * nowait invocations.
461                  */                               399                  */
462                 BUG_ON(vmf->flags & FAULT_FLAG    400                 BUG_ON(vmf->flags & FAULT_FLAG_RETRY_NOWAIT);
463 #ifdef CONFIG_DEBUG_VM                            401 #ifdef CONFIG_DEBUG_VM
464                 if (printk_ratelimit()) {         402                 if (printk_ratelimit()) {
465                         printk(KERN_WARNING       403                         printk(KERN_WARNING
466                                "FAULT_FLAG_ALL    404                                "FAULT_FLAG_ALLOW_RETRY missing %x\n",
467                                vmf->flags);       405                                vmf->flags);
468                         dump_stack();             406                         dump_stack();
469                 }                                 407                 }
470 #endif                                            408 #endif
471                 goto out;                         409                 goto out;
472         }                                         410         }
473                                                   411 
474         /*                                        412         /*
475          * Handle nowait, not much to do other    413          * Handle nowait, not much to do other than tell it to retry
476          * and wait.                              414          * and wait.
477          */                                       415          */
478         ret = VM_FAULT_RETRY;                     416         ret = VM_FAULT_RETRY;
479         if (vmf->flags & FAULT_FLAG_RETRY_NOWA    417         if (vmf->flags & FAULT_FLAG_RETRY_NOWAIT)
480                 goto out;                         418                 goto out;
481                                                   419 
482         /* take the reference before dropping  !! 420         /* take the reference before dropping the mmap_sem */
483         userfaultfd_ctx_get(ctx);                 421         userfaultfd_ctx_get(ctx);
484                                                   422 
485         init_waitqueue_func_entry(&uwq.wq, use    423         init_waitqueue_func_entry(&uwq.wq, userfaultfd_wake_function);
486         uwq.wq.private = current;                 424         uwq.wq.private = current;
487         uwq.msg = userfault_msg(vmf->address,  !! 425         uwq.msg = userfault_msg(vmf->address, vmf->flags, reason);
488                                 reason, ctx->f << 
489         uwq.ctx = ctx;                            426         uwq.ctx = ctx;
490         uwq.waken = false;                        427         uwq.waken = false;
491                                                   428 
492         blocking_state = userfaultfd_get_block !! 429         return_to_userland =
493                                                !! 430                 (vmf->flags & (FAULT_FLAG_USER|FAULT_FLAG_KILLABLE)) ==
494         /*                                     !! 431                 (FAULT_FLAG_USER|FAULT_FLAG_KILLABLE);
495          * Take the vma lock now, in order to  !! 432         blocking_state = return_to_userland ? TASK_INTERRUPTIBLE :
496          * userfaultfd_huge_must_wait() later. !! 433                          TASK_KILLABLE;
497          * (sleepable) vma lock can modify the << 
498          * must be before explicitly calling s << 
499          */                                    << 
500         if (is_vm_hugetlb_page(vma))           << 
501                 hugetlb_vma_lock_read(vma);    << 
502                                                   434 
503         spin_lock_irq(&ctx->fault_pending_wqh. !! 435         spin_lock(&ctx->fault_pending_wqh.lock);
504         /*                                        436         /*
505          * After the __add_wait_queue the uwq     437          * After the __add_wait_queue the uwq is visible to userland
506          * through poll/read().                   438          * through poll/read().
507          */                                       439          */
508         __add_wait_queue(&ctx->fault_pending_w    440         __add_wait_queue(&ctx->fault_pending_wqh, &uwq.wq);
509         /*                                        441         /*
510          * The smp_mb() after __set_current_st    442          * The smp_mb() after __set_current_state prevents the reads
511          * following the spin_unlock to happen    443          * following the spin_unlock to happen before the list_add in
512          * __add_wait_queue.                      444          * __add_wait_queue.
513          */                                       445          */
514         set_current_state(blocking_state);        446         set_current_state(blocking_state);
515         spin_unlock_irq(&ctx->fault_pending_wq !! 447         spin_unlock(&ctx->fault_pending_wqh.lock);
516                                                   448 
517         if (!is_vm_hugetlb_page(vma))          !! 449         if (!is_vm_hugetlb_page(vmf->vma))
518                 must_wait = userfaultfd_must_w !! 450                 must_wait = userfaultfd_must_wait(ctx, vmf->address, vmf->flags,
                                                   >> 451                                                   reason);
519         else                                      452         else
520                 must_wait = userfaultfd_huge_m !! 453                 must_wait = userfaultfd_huge_must_wait(ctx, vmf->vma,
521         if (is_vm_hugetlb_page(vma))           !! 454                                                        vmf->address,
522                 hugetlb_vma_unlock_read(vma);  !! 455                                                        vmf->flags, reason);
523         release_fault_lock(vmf);               !! 456         up_read(&mm->mmap_sem);
524                                                !! 457 
525         if (likely(must_wait && !READ_ONCE(ctx !! 458         if (likely(must_wait && !ACCESS_ONCE(ctx->released) &&
526                 wake_up_poll(&ctx->fd_wqh, EPO !! 459                    (return_to_userland ? !signal_pending(current) :
                                                   >> 460                     !fatal_signal_pending(current)))) {
                                                   >> 461                 wake_up_poll(&ctx->fd_wqh, POLLIN);
527                 schedule();                       462                 schedule();
                                                   >> 463                 ret |= VM_FAULT_MAJOR;
                                                   >> 464 
                                                   >> 465                 /*
                                                   >> 466                  * False wakeups can orginate even from rwsem before
                                                   >> 467                  * up_read() however userfaults will wait either for a
                                                   >> 468                  * targeted wakeup on the specific uwq waitqueue from
                                                   >> 469                  * wake_userfault() or for signals or for uffd
                                                   >> 470                  * release.
                                                   >> 471                  */
                                                   >> 472                 while (!READ_ONCE(uwq.waken)) {
                                                   >> 473                         /*
                                                   >> 474                          * This needs the full smp_store_mb()
                                                   >> 475                          * guarantee as the state write must be
                                                   >> 476                          * visible to other CPUs before reading
                                                   >> 477                          * uwq.waken from other CPUs.
                                                   >> 478                          */
                                                   >> 479                         set_current_state(blocking_state);
                                                   >> 480                         if (READ_ONCE(uwq.waken) ||
                                                   >> 481                             READ_ONCE(ctx->released) ||
                                                   >> 482                             (return_to_userland ? signal_pending(current) :
                                                   >> 483                              fatal_signal_pending(current)))
                                                   >> 484                                 break;
                                                   >> 485                         schedule();
                                                   >> 486                 }
528         }                                         487         }
529                                                   488 
530         __set_current_state(TASK_RUNNING);        489         __set_current_state(TASK_RUNNING);
531                                                   490 
                                                   >> 491         if (return_to_userland) {
                                                   >> 492                 if (signal_pending(current) &&
                                                   >> 493                     !fatal_signal_pending(current)) {
                                                   >> 494                         /*
                                                   >> 495                          * If we got a SIGSTOP or SIGCONT and this is
                                                   >> 496                          * a normal userland page fault, just let
                                                   >> 497                          * userland return so the signal will be
                                                   >> 498                          * handled and gdb debugging works.  The page
                                                   >> 499                          * fault code immediately after we return from
                                                   >> 500                          * this function is going to release the
                                                   >> 501                          * mmap_sem and it's not depending on it
                                                   >> 502                          * (unlike gup would if we were not to return
                                                   >> 503                          * VM_FAULT_RETRY).
                                                   >> 504                          *
                                                   >> 505                          * If a fatal signal is pending we still take
                                                   >> 506                          * the streamlined VM_FAULT_RETRY failure path
                                                   >> 507                          * and there's no need to retake the mmap_sem
                                                   >> 508                          * in such case.
                                                   >> 509                          */
                                                   >> 510                         down_read(&mm->mmap_sem);
                                                   >> 511                         ret = VM_FAULT_NOPAGE;
                                                   >> 512                 }
                                                   >> 513         }
                                                   >> 514 
532         /*                                        515         /*
533          * Here we race with the list_del; lis    516          * Here we race with the list_del; list_add in
534          * userfaultfd_ctx_read(), however bec    517          * userfaultfd_ctx_read(), however because we don't ever run
535          * list_del_init() to refile across th    518          * list_del_init() to refile across the two lists, the prev
536          * and next pointers will never point     519          * and next pointers will never point to self. list_add also
537          * would never let any of the two poin    520          * would never let any of the two pointers to point to
538          * self. So list_empty_careful won't r    521          * self. So list_empty_careful won't risk to see both pointers
539          * pointing to self at any time during    522          * pointing to self at any time during the list refile. The
540          * only case where list_del_init() is     523          * only case where list_del_init() is called is the full
541          * removal in the wake function and th    524          * removal in the wake function and there we don't re-list_add
542          * and it's fine not to block on the s    525          * and it's fine not to block on the spinlock. The uwq on this
543          * kernel stack can be released after     526          * kernel stack can be released after the list_del_init.
544          */                                       527          */
545         if (!list_empty_careful(&uwq.wq.entry)    528         if (!list_empty_careful(&uwq.wq.entry)) {
546                 spin_lock_irq(&ctx->fault_pend !! 529                 spin_lock(&ctx->fault_pending_wqh.lock);
547                 /*                                530                 /*
548                  * No need of list_del_init(),    531                  * No need of list_del_init(), the uwq on the stack
549                  * will be freed shortly anywa    532                  * will be freed shortly anyway.
550                  */                               533                  */
551                 list_del(&uwq.wq.entry);          534                 list_del(&uwq.wq.entry);
552                 spin_unlock_irq(&ctx->fault_pe !! 535                 spin_unlock(&ctx->fault_pending_wqh.lock);
553         }                                         536         }
554                                                   537 
555         /*                                        538         /*
556          * ctx may go away after this if the u    539          * ctx may go away after this if the userfault pseudo fd is
557          * already released.                      540          * already released.
558          */                                       541          */
559         userfaultfd_ctx_put(ctx);                 542         userfaultfd_ctx_put(ctx);
560                                                   543 
561 out:                                              544 out:
562         return ret;                               545         return ret;
563 }                                                 546 }
564                                                   547 
565 static void userfaultfd_event_wait_completion(    548 static void userfaultfd_event_wait_completion(struct userfaultfd_ctx *ctx,
566                                                   549                                               struct userfaultfd_wait_queue *ewq)
567 {                                                 550 {
568         struct userfaultfd_ctx *release_new_ct << 
569                                                << 
570         if (WARN_ON_ONCE(current->flags & PF_E    551         if (WARN_ON_ONCE(current->flags & PF_EXITING))
571                 goto out;                         552                 goto out;
572                                                   553 
573         ewq->ctx = ctx;                           554         ewq->ctx = ctx;
574         init_waitqueue_entry(&ewq->wq, current    555         init_waitqueue_entry(&ewq->wq, current);
575         release_new_ctx = NULL;                << 
576                                                   556 
577         spin_lock_irq(&ctx->event_wqh.lock);   !! 557         spin_lock(&ctx->event_wqh.lock);
578         /*                                        558         /*
579          * After the __add_wait_queue the uwq     559          * After the __add_wait_queue the uwq is visible to userland
580          * through poll/read().                   560          * through poll/read().
581          */                                       561          */
582         __add_wait_queue(&ctx->event_wqh, &ewq    562         __add_wait_queue(&ctx->event_wqh, &ewq->wq);
583         for (;;) {                                563         for (;;) {
584                 set_current_state(TASK_KILLABL    564                 set_current_state(TASK_KILLABLE);
585                 if (ewq->msg.event == 0)          565                 if (ewq->msg.event == 0)
586                         break;                    566                         break;
587                 if (READ_ONCE(ctx->released) | !! 567                 if (ACCESS_ONCE(ctx->released) ||
588                     fatal_signal_pending(curre    568                     fatal_signal_pending(current)) {
589                         /*                        569                         /*
590                          * &ewq->wq may be que    570                          * &ewq->wq may be queued in fork_event, but
591                          * __remove_wait_queue    571                          * __remove_wait_queue ignores the head
592                          * parameter. It would    572                          * parameter. It would be a problem if it
593                          * didn't.                573                          * didn't.
594                          */                       574                          */
595                         __remove_wait_queue(&c    575                         __remove_wait_queue(&ctx->event_wqh, &ewq->wq);
596                         if (ewq->msg.event ==     576                         if (ewq->msg.event == UFFD_EVENT_FORK) {
597                                 struct userfau    577                                 struct userfaultfd_ctx *new;
598                                                   578 
599                                 new = (struct     579                                 new = (struct userfaultfd_ctx *)
600                                         (unsig    580                                         (unsigned long)
601                                         ewq->m    581                                         ewq->msg.arg.reserved.reserved1;
602                                 release_new_ct !! 582 
                                                   >> 583                                 userfaultfd_ctx_put(new);
603                         }                         584                         }
604                         break;                    585                         break;
605                 }                                 586                 }
606                                                   587 
607                 spin_unlock_irq(&ctx->event_wq !! 588                 spin_unlock(&ctx->event_wqh.lock);
608                                                   589 
609                 wake_up_poll(&ctx->fd_wqh, EPO !! 590                 wake_up_poll(&ctx->fd_wqh, POLLIN);
610                 schedule();                       591                 schedule();
611                                                   592 
612                 spin_lock_irq(&ctx->event_wqh. !! 593                 spin_lock(&ctx->event_wqh.lock);
613         }                                         594         }
614         __set_current_state(TASK_RUNNING);        595         __set_current_state(TASK_RUNNING);
615         spin_unlock_irq(&ctx->event_wqh.lock); !! 596         spin_unlock(&ctx->event_wqh.lock);
616                                                << 
617         if (release_new_ctx) {                 << 
618                 struct vm_area_struct *vma;    << 
619                 struct mm_struct *mm = release << 
620                 VMA_ITERATOR(vmi, mm, 0);      << 
621                                                << 
622                 /* the various vma->vm_userfau << 
623                 mmap_write_lock(mm);           << 
624                 for_each_vma(vmi, vma) {       << 
625                         if (vma->vm_userfaultf << 
626                                 vma_start_writ << 
627                                 vma->vm_userfa << 
628                                 userfaultfd_se << 
629                                                << 
630                         }                      << 
631                 }                              << 
632                 mmap_write_unlock(mm);         << 
633                                                << 
634                 userfaultfd_ctx_put(release_ne << 
635         }                                      << 
636                                                   597 
637         /*                                        598         /*
638          * ctx may go away after this if the u    599          * ctx may go away after this if the userfault pseudo fd is
639          * already released.                      600          * already released.
640          */                                       601          */
641 out:                                              602 out:
642         atomic_dec(&ctx->mmap_changing);       << 
643         VM_BUG_ON(atomic_read(&ctx->mmap_chang << 
644         userfaultfd_ctx_put(ctx);                 603         userfaultfd_ctx_put(ctx);
645 }                                                 604 }
646                                                   605 
647 static void userfaultfd_event_complete(struct     606 static void userfaultfd_event_complete(struct userfaultfd_ctx *ctx,
648                                        struct     607                                        struct userfaultfd_wait_queue *ewq)
649 {                                                 608 {
650         ewq->msg.event = 0;                       609         ewq->msg.event = 0;
651         wake_up_locked(&ctx->event_wqh);          610         wake_up_locked(&ctx->event_wqh);
652         __remove_wait_queue(&ctx->event_wqh, &    611         __remove_wait_queue(&ctx->event_wqh, &ewq->wq);
653 }                                                 612 }
654                                                   613 
655 int dup_userfaultfd(struct vm_area_struct *vma    614 int dup_userfaultfd(struct vm_area_struct *vma, struct list_head *fcs)
656 {                                                 615 {
657         struct userfaultfd_ctx *ctx = NULL, *o    616         struct userfaultfd_ctx *ctx = NULL, *octx;
658         struct userfaultfd_fork_ctx *fctx;        617         struct userfaultfd_fork_ctx *fctx;
659                                                   618 
660         octx = vma->vm_userfaultfd_ctx.ctx;       619         octx = vma->vm_userfaultfd_ctx.ctx;
661         if (!octx)                             !! 620         if (!octx || !(octx->features & UFFD_FEATURE_EVENT_FORK)) {
662                 return 0;                      << 
663                                                << 
664         if (!(octx->features & UFFD_FEATURE_EV << 
665                 vma_start_write(vma);          << 
666                 vma->vm_userfaultfd_ctx = NULL    621                 vma->vm_userfaultfd_ctx = NULL_VM_UFFD_CTX;
667                 userfaultfd_set_vm_flags(vma,  !! 622                 vma->vm_flags &= ~(VM_UFFD_WP | VM_UFFD_MISSING);
668                 return 0;                         623                 return 0;
669         }                                         624         }
670                                                   625 
671         list_for_each_entry(fctx, fcs, list)      626         list_for_each_entry(fctx, fcs, list)
672                 if (fctx->orig == octx) {         627                 if (fctx->orig == octx) {
673                         ctx = fctx->new;          628                         ctx = fctx->new;
674                         break;                    629                         break;
675                 }                                 630                 }
676                                                   631 
677         if (!ctx) {                               632         if (!ctx) {
678                 fctx = kmalloc(sizeof(*fctx),     633                 fctx = kmalloc(sizeof(*fctx), GFP_KERNEL);
679                 if (!fctx)                        634                 if (!fctx)
680                         return -ENOMEM;           635                         return -ENOMEM;
681                                                   636 
682                 ctx = kmem_cache_alloc(userfau    637                 ctx = kmem_cache_alloc(userfaultfd_ctx_cachep, GFP_KERNEL);
683                 if (!ctx) {                       638                 if (!ctx) {
684                         kfree(fctx);              639                         kfree(fctx);
685                         return -ENOMEM;           640                         return -ENOMEM;
686                 }                                 641                 }
687                                                   642 
688                 refcount_set(&ctx->refcount, 1 !! 643                 atomic_set(&ctx->refcount, 1);
689                 ctx->flags = octx->flags;         644                 ctx->flags = octx->flags;
                                                   >> 645                 ctx->state = UFFD_STATE_RUNNING;
690                 ctx->features = octx->features    646                 ctx->features = octx->features;
691                 ctx->released = false;            647                 ctx->released = false;
692                 init_rwsem(&ctx->map_changing_ << 
693                 atomic_set(&ctx->mmap_changing << 
694                 ctx->mm = vma->vm_mm;             648                 ctx->mm = vma->vm_mm;
695                 mmgrab(ctx->mm);               !! 649                 atomic_inc(&ctx->mm->mm_count);
696                                                   650 
697                 userfaultfd_ctx_get(octx);        651                 userfaultfd_ctx_get(octx);
698                 down_write(&octx->map_changing << 
699                 atomic_inc(&octx->mmap_changin << 
700                 up_write(&octx->map_changing_l << 
701                 fctx->orig = octx;                652                 fctx->orig = octx;
702                 fctx->new = ctx;                  653                 fctx->new = ctx;
703                 list_add_tail(&fctx->list, fcs    654                 list_add_tail(&fctx->list, fcs);
704         }                                         655         }
705                                                   656 
706         vma->vm_userfaultfd_ctx.ctx = ctx;        657         vma->vm_userfaultfd_ctx.ctx = ctx;
707         return 0;                                 658         return 0;
708 }                                                 659 }
709                                                   660 
710 static void dup_fctx(struct userfaultfd_fork_c    661 static void dup_fctx(struct userfaultfd_fork_ctx *fctx)
711 {                                                 662 {
712         struct userfaultfd_ctx *ctx = fctx->or    663         struct userfaultfd_ctx *ctx = fctx->orig;
713         struct userfaultfd_wait_queue ewq;        664         struct userfaultfd_wait_queue ewq;
714                                                   665 
715         msg_init(&ewq.msg);                       666         msg_init(&ewq.msg);
716                                                   667 
717         ewq.msg.event = UFFD_EVENT_FORK;          668         ewq.msg.event = UFFD_EVENT_FORK;
718         ewq.msg.arg.reserved.reserved1 = (unsi    669         ewq.msg.arg.reserved.reserved1 = (unsigned long)fctx->new;
719                                                   670 
720         userfaultfd_event_wait_completion(ctx,    671         userfaultfd_event_wait_completion(ctx, &ewq);
721 }                                                 672 }
722                                                   673 
723 void dup_userfaultfd_complete(struct list_head    674 void dup_userfaultfd_complete(struct list_head *fcs)
724 {                                                 675 {
725         struct userfaultfd_fork_ctx *fctx, *n;    676         struct userfaultfd_fork_ctx *fctx, *n;
726                                                   677 
727         list_for_each_entry_safe(fctx, n, fcs,    678         list_for_each_entry_safe(fctx, n, fcs, list) {
728                 dup_fctx(fctx);                   679                 dup_fctx(fctx);
729                 list_del(&fctx->list);            680                 list_del(&fctx->list);
730                 kfree(fctx);                      681                 kfree(fctx);
731         }                                         682         }
732 }                                                 683 }
733                                                   684 
734 void mremap_userfaultfd_prep(struct vm_area_st    685 void mremap_userfaultfd_prep(struct vm_area_struct *vma,
735                              struct vm_userfau    686                              struct vm_userfaultfd_ctx *vm_ctx)
736 {                                                 687 {
737         struct userfaultfd_ctx *ctx;              688         struct userfaultfd_ctx *ctx;
738                                                   689 
739         ctx = vma->vm_userfaultfd_ctx.ctx;        690         ctx = vma->vm_userfaultfd_ctx.ctx;
740                                                !! 691         if (ctx && (ctx->features & UFFD_FEATURE_EVENT_REMAP)) {
741         if (!ctx)                              << 
742                 return;                        << 
743                                                << 
744         if (ctx->features & UFFD_FEATURE_EVENT << 
745                 vm_ctx->ctx = ctx;                692                 vm_ctx->ctx = ctx;
746                 userfaultfd_ctx_get(ctx);         693                 userfaultfd_ctx_get(ctx);
747                 down_write(&ctx->map_changing_ << 
748                 atomic_inc(&ctx->mmap_changing << 
749                 up_write(&ctx->map_changing_lo << 
750         } else {                               << 
751                 /* Drop uffd context if remap  << 
752                 vma_start_write(vma);          << 
753                 vma->vm_userfaultfd_ctx = NULL << 
754                 userfaultfd_set_vm_flags(vma,  << 
755         }                                         694         }
756 }                                                 695 }
757                                                   696 
758 void mremap_userfaultfd_complete(struct vm_use    697 void mremap_userfaultfd_complete(struct vm_userfaultfd_ctx *vm_ctx,
759                                  unsigned long    698                                  unsigned long from, unsigned long to,
760                                  unsigned long    699                                  unsigned long len)
761 {                                                 700 {
762         struct userfaultfd_ctx *ctx = vm_ctx->    701         struct userfaultfd_ctx *ctx = vm_ctx->ctx;
763         struct userfaultfd_wait_queue ewq;        702         struct userfaultfd_wait_queue ewq;
764                                                   703 
765         if (!ctx)                                 704         if (!ctx)
766                 return;                           705                 return;
767                                                   706 
768         if (to & ~PAGE_MASK) {                    707         if (to & ~PAGE_MASK) {
769                 userfaultfd_ctx_put(ctx);         708                 userfaultfd_ctx_put(ctx);
770                 return;                           709                 return;
771         }                                         710         }
772                                                   711 
773         msg_init(&ewq.msg);                       712         msg_init(&ewq.msg);
774                                                   713 
775         ewq.msg.event = UFFD_EVENT_REMAP;         714         ewq.msg.event = UFFD_EVENT_REMAP;
776         ewq.msg.arg.remap.from = from;            715         ewq.msg.arg.remap.from = from;
777         ewq.msg.arg.remap.to = to;                716         ewq.msg.arg.remap.to = to;
778         ewq.msg.arg.remap.len = len;              717         ewq.msg.arg.remap.len = len;
779                                                   718 
780         userfaultfd_event_wait_completion(ctx,    719         userfaultfd_event_wait_completion(ctx, &ewq);
781 }                                                 720 }
782                                                   721 
783 bool userfaultfd_remove(struct vm_area_struct     722 bool userfaultfd_remove(struct vm_area_struct *vma,
784                         unsigned long start, u    723                         unsigned long start, unsigned long end)
785 {                                                 724 {
786         struct mm_struct *mm = vma->vm_mm;        725         struct mm_struct *mm = vma->vm_mm;
787         struct userfaultfd_ctx *ctx;              726         struct userfaultfd_ctx *ctx;
788         struct userfaultfd_wait_queue ewq;        727         struct userfaultfd_wait_queue ewq;
789                                                   728 
790         ctx = vma->vm_userfaultfd_ctx.ctx;        729         ctx = vma->vm_userfaultfd_ctx.ctx;
791         if (!ctx || !(ctx->features & UFFD_FEA    730         if (!ctx || !(ctx->features & UFFD_FEATURE_EVENT_REMOVE))
792                 return true;                      731                 return true;
793                                                   732 
794         userfaultfd_ctx_get(ctx);                 733         userfaultfd_ctx_get(ctx);
795         down_write(&ctx->map_changing_lock);   !! 734         up_read(&mm->mmap_sem);
796         atomic_inc(&ctx->mmap_changing);       << 
797         up_write(&ctx->map_changing_lock);     << 
798         mmap_read_unlock(mm);                  << 
799                                                   735 
800         msg_init(&ewq.msg);                       736         msg_init(&ewq.msg);
801                                                   737 
802         ewq.msg.event = UFFD_EVENT_REMOVE;        738         ewq.msg.event = UFFD_EVENT_REMOVE;
803         ewq.msg.arg.remove.start = start;         739         ewq.msg.arg.remove.start = start;
804         ewq.msg.arg.remove.end = end;             740         ewq.msg.arg.remove.end = end;
805                                                   741 
806         userfaultfd_event_wait_completion(ctx,    742         userfaultfd_event_wait_completion(ctx, &ewq);
807                                                   743 
808         return false;                             744         return false;
809 }                                                 745 }
810                                                   746 
811 static bool has_unmap_ctx(struct userfaultfd_c    747 static bool has_unmap_ctx(struct userfaultfd_ctx *ctx, struct list_head *unmaps,
812                           unsigned long start,    748                           unsigned long start, unsigned long end)
813 {                                                 749 {
814         struct userfaultfd_unmap_ctx *unmap_ct    750         struct userfaultfd_unmap_ctx *unmap_ctx;
815                                                   751 
816         list_for_each_entry(unmap_ctx, unmaps,    752         list_for_each_entry(unmap_ctx, unmaps, list)
817                 if (unmap_ctx->ctx == ctx && u    753                 if (unmap_ctx->ctx == ctx && unmap_ctx->start == start &&
818                     unmap_ctx->end == end)        754                     unmap_ctx->end == end)
819                         return true;              755                         return true;
820                                                   756 
821         return false;                             757         return false;
822 }                                                 758 }
823                                                   759 
824 int userfaultfd_unmap_prep(struct vm_area_stru !! 760 int userfaultfd_unmap_prep(struct vm_area_struct *vma,
825                            unsigned long end,  !! 761                            unsigned long start, unsigned long end,
826 {                                              !! 762                            struct list_head *unmaps)
827         struct userfaultfd_unmap_ctx *unmap_ct !! 763 {
828         struct userfaultfd_ctx *ctx = vma->vm_ !! 764         for ( ; vma && vma->vm_start < end; vma = vma->vm_next) {
                                                   >> 765                 struct userfaultfd_unmap_ctx *unmap_ctx;
                                                   >> 766                 struct userfaultfd_ctx *ctx = vma->vm_userfaultfd_ctx.ctx;
829                                                   767 
830         if (!ctx || !(ctx->features & UFFD_FEA !! 768                 if (!ctx || !(ctx->features & UFFD_FEATURE_EVENT_UNMAP) ||
831             has_unmap_ctx(ctx, unmaps, start,  !! 769                     has_unmap_ctx(ctx, unmaps, start, end))
832                 return 0;                      !! 770                         continue;
833                                                   771 
834         unmap_ctx = kzalloc(sizeof(*unmap_ctx) !! 772                 unmap_ctx = kzalloc(sizeof(*unmap_ctx), GFP_KERNEL);
835         if (!unmap_ctx)                        !! 773                 if (!unmap_ctx)
836                 return -ENOMEM;                !! 774                         return -ENOMEM;
837                                                   775 
838         userfaultfd_ctx_get(ctx);              !! 776                 userfaultfd_ctx_get(ctx);
839         down_write(&ctx->map_changing_lock);   !! 777                 unmap_ctx->ctx = ctx;
840         atomic_inc(&ctx->mmap_changing);       !! 778                 unmap_ctx->start = start;
841         up_write(&ctx->map_changing_lock);     !! 779                 unmap_ctx->end = end;
842         unmap_ctx->ctx = ctx;                  !! 780                 list_add_tail(&unmap_ctx->list, unmaps);
843         unmap_ctx->start = start;              !! 781         }
844         unmap_ctx->end = end;                  << 
845         list_add_tail(&unmap_ctx->list, unmaps << 
846                                                   782 
847         return 0;                                 783         return 0;
848 }                                                 784 }
849                                                   785 
850 void userfaultfd_unmap_complete(struct mm_stru    786 void userfaultfd_unmap_complete(struct mm_struct *mm, struct list_head *uf)
851 {                                                 787 {
852         struct userfaultfd_unmap_ctx *ctx, *n;    788         struct userfaultfd_unmap_ctx *ctx, *n;
853         struct userfaultfd_wait_queue ewq;        789         struct userfaultfd_wait_queue ewq;
854                                                   790 
855         list_for_each_entry_safe(ctx, n, uf, l    791         list_for_each_entry_safe(ctx, n, uf, list) {
856                 msg_init(&ewq.msg);               792                 msg_init(&ewq.msg);
857                                                   793 
858                 ewq.msg.event = UFFD_EVENT_UNM    794                 ewq.msg.event = UFFD_EVENT_UNMAP;
859                 ewq.msg.arg.remove.start = ctx    795                 ewq.msg.arg.remove.start = ctx->start;
860                 ewq.msg.arg.remove.end = ctx->    796                 ewq.msg.arg.remove.end = ctx->end;
861                                                   797 
862                 userfaultfd_event_wait_complet    798                 userfaultfd_event_wait_completion(ctx->ctx, &ewq);
863                                                   799 
864                 list_del(&ctx->list);             800                 list_del(&ctx->list);
865                 kfree(ctx);                       801                 kfree(ctx);
866         }                                         802         }
867 }                                                 803 }
868                                                   804 
869 static int userfaultfd_release(struct inode *i    805 static int userfaultfd_release(struct inode *inode, struct file *file)
870 {                                                 806 {
871         struct userfaultfd_ctx *ctx = file->pr    807         struct userfaultfd_ctx *ctx = file->private_data;
872         struct mm_struct *mm = ctx->mm;           808         struct mm_struct *mm = ctx->mm;
873         struct vm_area_struct *vma, *prev;        809         struct vm_area_struct *vma, *prev;
874         /* len == 0 means wake all */             810         /* len == 0 means wake all */
875         struct userfaultfd_wake_range range =     811         struct userfaultfd_wake_range range = { .len = 0, };
876         unsigned long new_flags;                  812         unsigned long new_flags;
877         VMA_ITERATOR(vmi, mm, 0);              << 
878                                                   813 
879         WRITE_ONCE(ctx->released, true);       !! 814         ACCESS_ONCE(ctx->released) = true;
880                                                   815 
881         if (!mmget_not_zero(mm))                  816         if (!mmget_not_zero(mm))
882                 goto wakeup;                      817                 goto wakeup;
883                                                   818 
884         /*                                        819         /*
885          * Flush page faults out of all CPUs.     820          * Flush page faults out of all CPUs. NOTE: all page faults
886          * must be retried without returning V    821          * must be retried without returning VM_FAULT_SIGBUS if
887          * userfaultfd_ctx_get() succeeds but     822          * userfaultfd_ctx_get() succeeds but vma->vma_userfault_ctx
888          * changes while handle_userfault rele !! 823          * changes while handle_userfault released the mmap_sem. So
889          * it's critical that released is set     824          * it's critical that released is set to true (above), before
890          * taking the mmap_lock for writing.   !! 825          * taking the mmap_sem for writing.
891          */                                       826          */
892         mmap_write_lock(mm);                   !! 827         down_write(&mm->mmap_sem);
893         prev = NULL;                              828         prev = NULL;
894         for_each_vma(vmi, vma) {               !! 829         for (vma = mm->mmap; vma; vma = vma->vm_next) {
895                 cond_resched();                   830                 cond_resched();
896                 BUG_ON(!!vma->vm_userfaultfd_c    831                 BUG_ON(!!vma->vm_userfaultfd_ctx.ctx ^
897                        !!(vma->vm_flags & __VM !! 832                        !!(vma->vm_flags & (VM_UFFD_MISSING | VM_UFFD_WP)));
898                 if (vma->vm_userfaultfd_ctx.ct    833                 if (vma->vm_userfaultfd_ctx.ctx != ctx) {
899                         prev = vma;               834                         prev = vma;
900                         continue;                 835                         continue;
901                 }                                 836                 }
902                 /* Reset ptes for the whole vm !! 837                 new_flags = vma->vm_flags & ~(VM_UFFD_MISSING | VM_UFFD_WP);
903                 if (userfaultfd_wp(vma))       !! 838                 prev = vma_merge(mm, prev, vma->vm_start, vma->vm_end,
904                         uffd_wp_range(vma, vma !! 839                                  new_flags, vma->anon_vma,
905                                       vma->vm_ !! 840                                  vma->vm_file, vma->vm_pgoff,
906                 new_flags = vma->vm_flags & ~_ !! 841                                  vma_policy(vma),
907                 vma = vma_modify_flags_uffd(&v !! 842                                  NULL_VM_UFFD_CTX);
908                                             vm !! 843                 if (prev)
909                                             NU !! 844                         vma = prev;
910                                                !! 845                 else
911                 vma_start_write(vma);          !! 846                         prev = vma;
912                 userfaultfd_set_vm_flags(vma,  !! 847                 vma->vm_flags = new_flags;
913                 vma->vm_userfaultfd_ctx = NULL    848                 vma->vm_userfaultfd_ctx = NULL_VM_UFFD_CTX;
914                                                << 
915                 prev = vma;                    << 
916         }                                         849         }
917         mmap_write_unlock(mm);                 !! 850         up_write(&mm->mmap_sem);
918         mmput(mm);                                851         mmput(mm);
919 wakeup:                                           852 wakeup:
920         /*                                        853         /*
921          * After no new page faults can wait o    854          * After no new page faults can wait on this fault_*wqh, flush
922          * the last page faults that may have     855          * the last page faults that may have been already waiting on
923          * the fault_*wqh.                        856          * the fault_*wqh.
924          */                                       857          */
925         spin_lock_irq(&ctx->fault_pending_wqh. !! 858         spin_lock(&ctx->fault_pending_wqh.lock);
926         __wake_up_locked_key(&ctx->fault_pendi    859         __wake_up_locked_key(&ctx->fault_pending_wqh, TASK_NORMAL, &range);
927         __wake_up(&ctx->fault_wqh, TASK_NORMAL !! 860         __wake_up_locked_key(&ctx->fault_wqh, TASK_NORMAL, &range);
928         spin_unlock_irq(&ctx->fault_pending_wq !! 861         spin_unlock(&ctx->fault_pending_wqh.lock);
929                                                   862 
930         /* Flush pending events that may still    863         /* Flush pending events that may still wait on event_wqh */
931         wake_up_all(&ctx->event_wqh);             864         wake_up_all(&ctx->event_wqh);
932                                                   865 
933         wake_up_poll(&ctx->fd_wqh, EPOLLHUP);  !! 866         wake_up_poll(&ctx->fd_wqh, POLLHUP);
934         userfaultfd_ctx_put(ctx);                 867         userfaultfd_ctx_put(ctx);
935         return 0;                                 868         return 0;
936 }                                                 869 }
937                                                   870 
938 /* fault_pending_wqh.lock must be hold by the     871 /* fault_pending_wqh.lock must be hold by the caller */
939 static inline struct userfaultfd_wait_queue *f    872 static inline struct userfaultfd_wait_queue *find_userfault_in(
940                 wait_queue_head_t *wqh)           873                 wait_queue_head_t *wqh)
941 {                                                 874 {
942         wait_queue_entry_t *wq;                   875         wait_queue_entry_t *wq;
943         struct userfaultfd_wait_queue *uwq;       876         struct userfaultfd_wait_queue *uwq;
944                                                   877 
945         lockdep_assert_held(&wqh->lock);       !! 878         VM_BUG_ON(!spin_is_locked(&wqh->lock));
946                                                   879 
947         uwq = NULL;                               880         uwq = NULL;
948         if (!waitqueue_active(wqh))               881         if (!waitqueue_active(wqh))
949                 goto out;                         882                 goto out;
950         /* walk in reverse to provide FIFO beh    883         /* walk in reverse to provide FIFO behavior to read userfaults */
951         wq = list_last_entry(&wqh->head, typeo    884         wq = list_last_entry(&wqh->head, typeof(*wq), entry);
952         uwq = container_of(wq, struct userfaul    885         uwq = container_of(wq, struct userfaultfd_wait_queue, wq);
953 out:                                              886 out:
954         return uwq;                               887         return uwq;
955 }                                                 888 }
956                                                   889 
957 static inline struct userfaultfd_wait_queue *f    890 static inline struct userfaultfd_wait_queue *find_userfault(
958                 struct userfaultfd_ctx *ctx)      891                 struct userfaultfd_ctx *ctx)
959 {                                                 892 {
960         return find_userfault_in(&ctx->fault_p    893         return find_userfault_in(&ctx->fault_pending_wqh);
961 }                                                 894 }
962                                                   895 
963 static inline struct userfaultfd_wait_queue *f    896 static inline struct userfaultfd_wait_queue *find_userfault_evt(
964                 struct userfaultfd_ctx *ctx)      897                 struct userfaultfd_ctx *ctx)
965 {                                                 898 {
966         return find_userfault_in(&ctx->event_w    899         return find_userfault_in(&ctx->event_wqh);
967 }                                                 900 }
968                                                   901 
969 static __poll_t userfaultfd_poll(struct file * !! 902 static unsigned int userfaultfd_poll(struct file *file, poll_table *wait)
970 {                                                 903 {
971         struct userfaultfd_ctx *ctx = file->pr    904         struct userfaultfd_ctx *ctx = file->private_data;
972         __poll_t ret;                          !! 905         unsigned int ret;
973                                                   906 
974         poll_wait(file, &ctx->fd_wqh, wait);      907         poll_wait(file, &ctx->fd_wqh, wait);
975                                                   908 
976         if (!userfaultfd_is_initialized(ctx))  !! 909         switch (ctx->state) {
977                 return EPOLLERR;               !! 910         case UFFD_STATE_WAIT_API:
978                                                !! 911                 return POLLERR;
979         /*                                     !! 912         case UFFD_STATE_RUNNING:
980          * poll() never guarantees that read w !! 913                 /*
981          * userfaults can be waken before they !! 914                  * poll() never guarantees that read won't block.
982          */                                    !! 915                  * userfaults can be waken before they're read().
983         if (unlikely(!(file->f_flags & O_NONBL !! 916                  */
984                 return EPOLLERR;               !! 917                 if (unlikely(!(file->f_flags & O_NONBLOCK)))
985         /*                                     !! 918                         return POLLERR;
986          * lockless access to see if there are !! 919                 /*
987          * __pollwait last action is the add_w !! 920                  * lockless access to see if there are pending faults
988          * the spin_unlock would allow the wai !! 921                  * __pollwait last action is the add_wait_queue but
989          * pass above the actual list_add insi !! 922                  * the spin_unlock would allow the waitqueue_active to
990          * add_wait_queue critical section. So !! 923                  * pass above the actual list_add inside
991          * memory barrier to serialize the lis !! 924                  * add_wait_queue critical section. So use a full
992          * add_wait_queue() with the waitqueue !! 925                  * memory barrier to serialize the list_add write of
993          * below.                              !! 926                  * add_wait_queue() with the waitqueue_active read
994          */                                    !! 927                  * below.
995         ret = 0;                               !! 928                  */
996         smp_mb();                              !! 929                 ret = 0;
997         if (waitqueue_active(&ctx->fault_pendi !! 930                 smp_mb();
998                 ret = EPOLLIN;                 !! 931                 if (waitqueue_active(&ctx->fault_pending_wqh))
999         else if (waitqueue_active(&ctx->event_ !! 932                         ret = POLLIN;
1000                 ret = EPOLLIN;                !! 933                 else if (waitqueue_active(&ctx->event_wqh))
                                                   >> 934                         ret = POLLIN;
1001                                                  935 
1002         return ret;                           !! 936                 return ret;
                                                   >> 937         default:
                                                   >> 938                 WARN_ON_ONCE(1);
                                                   >> 939                 return POLLERR;
                                                   >> 940         }
1003 }                                                941 }
1004                                                  942 
1005 static const struct file_operations userfault    943 static const struct file_operations userfaultfd_fops;
1006                                                  944 
1007 static int resolve_userfault_fork(struct user !! 945 static int resolve_userfault_fork(struct userfaultfd_ctx *ctx,
1008                                   struct inod !! 946                                   struct userfaultfd_ctx *new,
1009                                   struct uffd    947                                   struct uffd_msg *msg)
1010 {                                                948 {
1011         int fd;                                  949         int fd;
                                                   >> 950         struct file *file;
                                                   >> 951         unsigned int flags = new->flags & UFFD_SHARED_FCNTL_FLAGS;
1012                                                  952 
1013         fd = anon_inode_create_getfd("[userfa !! 953         fd = get_unused_fd_flags(flags);
1014                         O_RDONLY | (new->flag << 
1015         if (fd < 0)                              954         if (fd < 0)
1016                 return fd;                       955                 return fd;
1017                                                  956 
                                                   >> 957         file = anon_inode_getfile("[userfaultfd]", &userfaultfd_fops, new,
                                                   >> 958                                   O_RDWR | flags);
                                                   >> 959         if (IS_ERR(file)) {
                                                   >> 960                 put_unused_fd(fd);
                                                   >> 961                 return PTR_ERR(file);
                                                   >> 962         }
                                                   >> 963 
                                                   >> 964         fd_install(fd, file);
1018         msg->arg.reserved.reserved1 = 0;         965         msg->arg.reserved.reserved1 = 0;
1019         msg->arg.fork.ufd = fd;                  966         msg->arg.fork.ufd = fd;
                                                   >> 967 
1020         return 0;                                968         return 0;
1021 }                                                969 }
1022                                                  970 
1023 static ssize_t userfaultfd_ctx_read(struct us    971 static ssize_t userfaultfd_ctx_read(struct userfaultfd_ctx *ctx, int no_wait,
1024                                     struct uf !! 972                                     struct uffd_msg *msg)
1025 {                                                973 {
1026         ssize_t ret;                             974         ssize_t ret;
1027         DECLARE_WAITQUEUE(wait, current);        975         DECLARE_WAITQUEUE(wait, current);
1028         struct userfaultfd_wait_queue *uwq;      976         struct userfaultfd_wait_queue *uwq;
1029         /*                                       977         /*
1030          * Handling fork event requires sleep    978          * Handling fork event requires sleeping operations, so
1031          * we drop the event_wqh lock, then d    979          * we drop the event_wqh lock, then do these ops, then
1032          * lock it back and wake up the waite    980          * lock it back and wake up the waiter. While the lock is
1033          * dropped the ewq may go away so we     981          * dropped the ewq may go away so we keep track of it
1034          * carefully.                            982          * carefully.
1035          */                                      983          */
1036         LIST_HEAD(fork_event);                   984         LIST_HEAD(fork_event);
1037         struct userfaultfd_ctx *fork_nctx = N    985         struct userfaultfd_ctx *fork_nctx = NULL;
1038                                                  986 
1039         /* always take the fd_wqh lock before    987         /* always take the fd_wqh lock before the fault_pending_wqh lock */
1040         spin_lock_irq(&ctx->fd_wqh.lock);     !! 988         spin_lock(&ctx->fd_wqh.lock);
1041         __add_wait_queue(&ctx->fd_wqh, &wait)    989         __add_wait_queue(&ctx->fd_wqh, &wait);
1042         for (;;) {                               990         for (;;) {
1043                 set_current_state(TASK_INTERR    991                 set_current_state(TASK_INTERRUPTIBLE);
1044                 spin_lock(&ctx->fault_pending    992                 spin_lock(&ctx->fault_pending_wqh.lock);
1045                 uwq = find_userfault(ctx);       993                 uwq = find_userfault(ctx);
1046                 if (uwq) {                       994                 if (uwq) {
1047                         /*                       995                         /*
1048                          * Use a seqcount to     996                          * Use a seqcount to repeat the lockless check
1049                          * in wake_userfault(    997                          * in wake_userfault() to avoid missing
1050                          * wakeups because du    998                          * wakeups because during the refile both
1051                          * waitqueue could be    999                          * waitqueue could become empty if this is the
1052                          * only userfault.       1000                          * only userfault.
1053                          */                      1001                          */
1054                         write_seqcount_begin(    1002                         write_seqcount_begin(&ctx->refile_seq);
1055                                                  1003 
1056                         /*                       1004                         /*
1057                          * The fault_pending_    1005                          * The fault_pending_wqh.lock prevents the uwq
1058                          * to disappear from     1006                          * to disappear from under us.
1059                          *                       1007                          *
1060                          * Refile this userfa    1008                          * Refile this userfault from
1061                          * fault_pending_wqh     1009                          * fault_pending_wqh to fault_wqh, it's not
1062                          * pending anymore af    1010                          * pending anymore after we read it.
1063                          *                       1011                          *
1064                          * Use list_del() by     1012                          * Use list_del() by hand (as
1065                          * userfaultfd_wake_f    1013                          * userfaultfd_wake_function also uses
1066                          * list_del_init() by    1014                          * list_del_init() by hand) to be sure nobody
1067                          * changes __remove_w    1015                          * changes __remove_wait_queue() to use
1068                          * list_del_init() in    1016                          * list_del_init() in turn breaking the
1069                          * !list_empty_carefu    1017                          * !list_empty_careful() check in
1070                          * handle_userfault()    1018                          * handle_userfault(). The uwq->wq.head list
1071                          * must never be empt    1019                          * must never be empty at any time during the
1072                          * refile, or the wai    1020                          * refile, or the waitqueue could disappear
1073                          * from under us. The    1021                          * from under us. The "wait_queue_head_t"
1074                          * parameter of __rem    1022                          * parameter of __remove_wait_queue() is unused
1075                          * anyway.               1023                          * anyway.
1076                          */                      1024                          */
1077                         list_del(&uwq->wq.ent    1025                         list_del(&uwq->wq.entry);
1078                         add_wait_queue(&ctx-> !! 1026                         __add_wait_queue(&ctx->fault_wqh, &uwq->wq);
1079                                                  1027 
1080                         write_seqcount_end(&c    1028                         write_seqcount_end(&ctx->refile_seq);
1081                                                  1029 
1082                         /* careful to always     1030                         /* careful to always initialize msg if ret == 0 */
1083                         *msg = uwq->msg;         1031                         *msg = uwq->msg;
1084                         spin_unlock(&ctx->fau    1032                         spin_unlock(&ctx->fault_pending_wqh.lock);
1085                         ret = 0;                 1033                         ret = 0;
1086                         break;                   1034                         break;
1087                 }                                1035                 }
1088                 spin_unlock(&ctx->fault_pendi    1036                 spin_unlock(&ctx->fault_pending_wqh.lock);
1089                                                  1037 
1090                 spin_lock(&ctx->event_wqh.loc    1038                 spin_lock(&ctx->event_wqh.lock);
1091                 uwq = find_userfault_evt(ctx)    1039                 uwq = find_userfault_evt(ctx);
1092                 if (uwq) {                       1040                 if (uwq) {
1093                         *msg = uwq->msg;         1041                         *msg = uwq->msg;
1094                                                  1042 
1095                         if (uwq->msg.event ==    1043                         if (uwq->msg.event == UFFD_EVENT_FORK) {
1096                                 fork_nctx = (    1044                                 fork_nctx = (struct userfaultfd_ctx *)
1097                                         (unsi    1045                                         (unsigned long)
1098                                         uwq->    1046                                         uwq->msg.arg.reserved.reserved1;
1099                                 list_move(&uw    1047                                 list_move(&uwq->wq.entry, &fork_event);
1100                                 /*               1048                                 /*
1101                                  * fork_nctx     1049                                  * fork_nctx can be freed as soon as
1102                                  * we drop th    1050                                  * we drop the lock, unless we take a
1103                                  * reference     1051                                  * reference on it.
1104                                  */              1052                                  */
1105                                 userfaultfd_c    1053                                 userfaultfd_ctx_get(fork_nctx);
1106                                 spin_unlock(&    1054                                 spin_unlock(&ctx->event_wqh.lock);
1107                                 ret = 0;         1055                                 ret = 0;
1108                                 break;           1056                                 break;
1109                         }                        1057                         }
1110                                                  1058 
1111                         userfaultfd_event_com    1059                         userfaultfd_event_complete(ctx, uwq);
1112                         spin_unlock(&ctx->eve    1060                         spin_unlock(&ctx->event_wqh.lock);
1113                         ret = 0;                 1061                         ret = 0;
1114                         break;                   1062                         break;
1115                 }                                1063                 }
1116                 spin_unlock(&ctx->event_wqh.l    1064                 spin_unlock(&ctx->event_wqh.lock);
1117                                                  1065 
1118                 if (signal_pending(current))     1066                 if (signal_pending(current)) {
1119                         ret = -ERESTARTSYS;      1067                         ret = -ERESTARTSYS;
1120                         break;                   1068                         break;
1121                 }                                1069                 }
1122                 if (no_wait) {                   1070                 if (no_wait) {
1123                         ret = -EAGAIN;           1071                         ret = -EAGAIN;
1124                         break;                   1072                         break;
1125                 }                                1073                 }
1126                 spin_unlock_irq(&ctx->fd_wqh. !! 1074                 spin_unlock(&ctx->fd_wqh.lock);
1127                 schedule();                      1075                 schedule();
1128                 spin_lock_irq(&ctx->fd_wqh.lo !! 1076                 spin_lock(&ctx->fd_wqh.lock);
1129         }                                        1077         }
1130         __remove_wait_queue(&ctx->fd_wqh, &wa    1078         __remove_wait_queue(&ctx->fd_wqh, &wait);
1131         __set_current_state(TASK_RUNNING);       1079         __set_current_state(TASK_RUNNING);
1132         spin_unlock_irq(&ctx->fd_wqh.lock);   !! 1080         spin_unlock(&ctx->fd_wqh.lock);
1133                                                  1081 
1134         if (!ret && msg->event == UFFD_EVENT_    1082         if (!ret && msg->event == UFFD_EVENT_FORK) {
1135                 ret = resolve_userfault_fork( !! 1083                 ret = resolve_userfault_fork(ctx, fork_nctx, msg);
1136                 spin_lock_irq(&ctx->event_wqh !! 1084                 spin_lock(&ctx->event_wqh.lock);
1137                 if (!list_empty(&fork_event))    1085                 if (!list_empty(&fork_event)) {
1138                         /*                       1086                         /*
1139                          * The fork thread di    1087                          * The fork thread didn't abort, so we can
1140                          * drop the temporary    1088                          * drop the temporary refcount.
1141                          */                      1089                          */
1142                         userfaultfd_ctx_put(f    1090                         userfaultfd_ctx_put(fork_nctx);
1143                                                  1091 
1144                         uwq = list_first_entr    1092                         uwq = list_first_entry(&fork_event,
1145                                                  1093                                                typeof(*uwq),
1146                                                  1094                                                wq.entry);
1147                         /*                       1095                         /*
1148                          * If fork_event list    1096                          * If fork_event list wasn't empty and in turn
1149                          * the event wasn't a    1097                          * the event wasn't already released by fork
1150                          * (the event is allo    1098                          * (the event is allocated on fork kernel
1151                          * stack), put the ev    1099                          * stack), put the event back to its place in
1152                          * the event_wq. fork    1100                          * the event_wq. fork_event head will be freed
1153                          * as soon as we retu    1101                          * as soon as we return so the event cannot
1154                          * stay queued there     1102                          * stay queued there no matter the current
1155                          * "ret" value.          1103                          * "ret" value.
1156                          */                      1104                          */
1157                         list_del(&uwq->wq.ent    1105                         list_del(&uwq->wq.entry);
1158                         __add_wait_queue(&ctx    1106                         __add_wait_queue(&ctx->event_wqh, &uwq->wq);
1159                                                  1107 
1160                         /*                       1108                         /*
1161                          * Leave the event in    1109                          * Leave the event in the waitqueue and report
1162                          * error to userland     1110                          * error to userland if we failed to resolve
1163                          * the userfault fork    1111                          * the userfault fork.
1164                          */                      1112                          */
1165                         if (likely(!ret))        1113                         if (likely(!ret))
1166                                 userfaultfd_e    1114                                 userfaultfd_event_complete(ctx, uwq);
1167                 } else {                         1115                 } else {
1168                         /*                       1116                         /*
1169                          * Here the fork thre    1117                          * Here the fork thread aborted and the
1170                          * refcount from the     1118                          * refcount from the fork thread on fork_nctx
1171                          * has already been r    1119                          * has already been released. We still hold
1172                          * the reference we t    1120                          * the reference we took before releasing the
1173                          * lock above. If res    1121                          * lock above. If resolve_userfault_fork
1174                          * failed we've to dr    1122                          * failed we've to drop it because the
1175                          * fork_nctx has to b    1123                          * fork_nctx has to be freed in such case. If
1176                          * it succeeded we'll    1124                          * it succeeded we'll hold it because the new
1177                          * uffd references it    1125                          * uffd references it.
1178                          */                      1126                          */
1179                         if (ret)                 1127                         if (ret)
1180                                 userfaultfd_c    1128                                 userfaultfd_ctx_put(fork_nctx);
1181                 }                                1129                 }
1182                 spin_unlock_irq(&ctx->event_w !! 1130                 spin_unlock(&ctx->event_wqh.lock);
1183         }                                        1131         }
1184                                                  1132 
1185         return ret;                              1133         return ret;
1186 }                                                1134 }
1187                                                  1135 
1188 static ssize_t userfaultfd_read_iter(struct k !! 1136 static ssize_t userfaultfd_read(struct file *file, char __user *buf,
                                                   >> 1137                                 size_t count, loff_t *ppos)
1189 {                                                1138 {
1190         struct file *file = iocb->ki_filp;    << 
1191         struct userfaultfd_ctx *ctx = file->p    1139         struct userfaultfd_ctx *ctx = file->private_data;
1192         ssize_t _ret, ret = 0;                   1140         ssize_t _ret, ret = 0;
1193         struct uffd_msg msg;                     1141         struct uffd_msg msg;
1194         struct inode *inode = file_inode(file !! 1142         int no_wait = file->f_flags & O_NONBLOCK;
1195         bool no_wait;                         << 
1196                                                  1143 
1197         if (!userfaultfd_is_initialized(ctx)) !! 1144         if (ctx->state == UFFD_STATE_WAIT_API)
1198                 return -EINVAL;                  1145                 return -EINVAL;
1199                                                  1146 
1200         no_wait = file->f_flags & O_NONBLOCK  << 
1201         for (;;) {                               1147         for (;;) {
1202                 if (iov_iter_count(to) < size !! 1148                 if (count < sizeof(msg))
1203                         return ret ? ret : -E    1149                         return ret ? ret : -EINVAL;
1204                 _ret = userfaultfd_ctx_read(c !! 1150                 _ret = userfaultfd_ctx_read(ctx, no_wait, &msg);
1205                 if (_ret < 0)                    1151                 if (_ret < 0)
1206                         return ret ? ret : _r    1152                         return ret ? ret : _ret;
1207                 _ret = !copy_to_iter_full(&ms !! 1153                 if (copy_to_user((__u64 __user *) buf, &msg, sizeof(msg)))
1208                 if (_ret)                     << 
1209                         return ret ? ret : -E    1154                         return ret ? ret : -EFAULT;
1210                 ret += sizeof(msg);              1155                 ret += sizeof(msg);
                                                   >> 1156                 buf += sizeof(msg);
                                                   >> 1157                 count -= sizeof(msg);
1211                 /*                               1158                 /*
1212                  * Allow to read more than on    1159                  * Allow to read more than one fault at time but only
1213                  * block if waiting for the v    1160                  * block if waiting for the very first one.
1214                  */                              1161                  */
1215                 no_wait = true;               !! 1162                 no_wait = O_NONBLOCK;
1216         }                                        1163         }
1217 }                                                1164 }
1218                                                  1165 
1219 static void __wake_userfault(struct userfault    1166 static void __wake_userfault(struct userfaultfd_ctx *ctx,
1220                              struct userfault    1167                              struct userfaultfd_wake_range *range)
1221 {                                                1168 {
1222         spin_lock_irq(&ctx->fault_pending_wqh !! 1169         spin_lock(&ctx->fault_pending_wqh.lock);
1223         /* wake all in the range and autoremo    1170         /* wake all in the range and autoremove */
1224         if (waitqueue_active(&ctx->fault_pend    1171         if (waitqueue_active(&ctx->fault_pending_wqh))
1225                 __wake_up_locked_key(&ctx->fa    1172                 __wake_up_locked_key(&ctx->fault_pending_wqh, TASK_NORMAL,
1226                                      range);     1173                                      range);
1227         if (waitqueue_active(&ctx->fault_wqh)    1174         if (waitqueue_active(&ctx->fault_wqh))
1228                 __wake_up(&ctx->fault_wqh, TA !! 1175                 __wake_up_locked_key(&ctx->fault_wqh, TASK_NORMAL, range);
1229         spin_unlock_irq(&ctx->fault_pending_w !! 1176         spin_unlock(&ctx->fault_pending_wqh.lock);
1230 }                                                1177 }
1231                                                  1178 
1232 static __always_inline void wake_userfault(st    1179 static __always_inline void wake_userfault(struct userfaultfd_ctx *ctx,
1233                                            st    1180                                            struct userfaultfd_wake_range *range)
1234 {                                                1181 {
1235         unsigned seq;                            1182         unsigned seq;
1236         bool need_wakeup;                        1183         bool need_wakeup;
1237                                                  1184 
1238         /*                                       1185         /*
1239          * To be sure waitqueue_active() is n    1186          * To be sure waitqueue_active() is not reordered by the CPU
1240          * before the pagetable update, use a    1187          * before the pagetable update, use an explicit SMP memory
1241          * barrier here. PT lock release or m !! 1188          * barrier here. PT lock release or up_read(mmap_sem) still
1242          * have release semantics that can al    1189          * have release semantics that can allow the
1243          * waitqueue_active() to be reordered    1190          * waitqueue_active() to be reordered before the pte update.
1244          */                                      1191          */
1245         smp_mb();                                1192         smp_mb();
1246                                                  1193 
1247         /*                                       1194         /*
1248          * Use waitqueue_active because it's     1195          * Use waitqueue_active because it's very frequent to
1249          * change the address space atomicall    1196          * change the address space atomically even if there are no
1250          * userfaults yet. So we take the spi    1197          * userfaults yet. So we take the spinlock only when we're
1251          * sure we've userfaults to wake.        1198          * sure we've userfaults to wake.
1252          */                                      1199          */
1253         do {                                     1200         do {
1254                 seq = read_seqcount_begin(&ct    1201                 seq = read_seqcount_begin(&ctx->refile_seq);
1255                 need_wakeup = waitqueue_activ    1202                 need_wakeup = waitqueue_active(&ctx->fault_pending_wqh) ||
1256                         waitqueue_active(&ctx    1203                         waitqueue_active(&ctx->fault_wqh);
1257                 cond_resched();                  1204                 cond_resched();
1258         } while (read_seqcount_retry(&ctx->re    1205         } while (read_seqcount_retry(&ctx->refile_seq, seq));
1259         if (need_wakeup)                         1206         if (need_wakeup)
1260                 __wake_userfault(ctx, range);    1207                 __wake_userfault(ctx, range);
1261 }                                                1208 }
1262                                                  1209 
1263 static __always_inline int validate_unaligned !! 1210 static __always_inline int validate_range(struct mm_struct *mm,
1264         struct mm_struct *mm, __u64 start, __ !! 1211                                           __u64 start, __u64 len)
1265 {                                                1212 {
1266         __u64 task_size = mm->task_size;         1213         __u64 task_size = mm->task_size;
1267                                                  1214 
                                                   >> 1215         if (start & ~PAGE_MASK)
                                                   >> 1216                 return -EINVAL;
1268         if (len & ~PAGE_MASK)                    1217         if (len & ~PAGE_MASK)
1269                 return -EINVAL;                  1218                 return -EINVAL;
1270         if (!len)                                1219         if (!len)
1271                 return -EINVAL;                  1220                 return -EINVAL;
1272         if (start < mmap_min_addr)               1221         if (start < mmap_min_addr)
1273                 return -EINVAL;                  1222                 return -EINVAL;
1274         if (start >= task_size)                  1223         if (start >= task_size)
1275                 return -EINVAL;                  1224                 return -EINVAL;
1276         if (len > task_size - start)             1225         if (len > task_size - start)
1277                 return -EINVAL;                  1226                 return -EINVAL;
1278         if (start + len <= start)             << 
1279                 return -EINVAL;               << 
1280         return 0;                                1227         return 0;
1281 }                                                1228 }
1282                                                  1229 
1283 static __always_inline int validate_range(str !! 1230 static inline bool vma_can_userfault(struct vm_area_struct *vma)
1284                                           __u << 
1285 {                                                1231 {
1286         if (start & ~PAGE_MASK)               !! 1232         return vma_is_anonymous(vma) || is_vm_hugetlb_page(vma) ||
1287                 return -EINVAL;               !! 1233                 vma_is_shmem(vma);
1288                                               << 
1289         return validate_unaligned_range(mm, s << 
1290 }                                                1234 }
1291                                                  1235 
1292 static int userfaultfd_register(struct userfa    1236 static int userfaultfd_register(struct userfaultfd_ctx *ctx,
1293                                 unsigned long    1237                                 unsigned long arg)
1294 {                                                1238 {
1295         struct mm_struct *mm = ctx->mm;          1239         struct mm_struct *mm = ctx->mm;
1296         struct vm_area_struct *vma, *prev, *c    1240         struct vm_area_struct *vma, *prev, *cur;
1297         int ret;                                 1241         int ret;
1298         struct uffdio_register uffdio_registe    1242         struct uffdio_register uffdio_register;
1299         struct uffdio_register __user *user_u    1243         struct uffdio_register __user *user_uffdio_register;
1300         unsigned long vm_flags, new_flags;       1244         unsigned long vm_flags, new_flags;
1301         bool found;                              1245         bool found;
1302         bool basic_ioctls;                    !! 1246         bool non_anon_pages;
1303         unsigned long start, end, vma_end;       1247         unsigned long start, end, vma_end;
1304         struct vma_iterator vmi;              << 
1305         bool wp_async = userfaultfd_wp_async_ << 
1306                                                  1248 
1307         user_uffdio_register = (struct uffdio    1249         user_uffdio_register = (struct uffdio_register __user *) arg;
1308                                                  1250 
1309         ret = -EFAULT;                           1251         ret = -EFAULT;
1310         if (copy_from_user(&uffdio_register,     1252         if (copy_from_user(&uffdio_register, user_uffdio_register,
1311                            sizeof(uffdio_regi    1253                            sizeof(uffdio_register)-sizeof(__u64)))
1312                 goto out;                        1254                 goto out;
1313                                                  1255 
1314         ret = -EINVAL;                           1256         ret = -EINVAL;
1315         if (!uffdio_register.mode)               1257         if (!uffdio_register.mode)
1316                 goto out;                        1258                 goto out;
1317         if (uffdio_register.mode & ~UFFD_API_ !! 1259         if (uffdio_register.mode & ~(UFFDIO_REGISTER_MODE_MISSING|
                                                   >> 1260                                      UFFDIO_REGISTER_MODE_WP))
1318                 goto out;                        1261                 goto out;
1319         vm_flags = 0;                            1262         vm_flags = 0;
1320         if (uffdio_register.mode & UFFDIO_REG    1263         if (uffdio_register.mode & UFFDIO_REGISTER_MODE_MISSING)
1321                 vm_flags |= VM_UFFD_MISSING;     1264                 vm_flags |= VM_UFFD_MISSING;
1322         if (uffdio_register.mode & UFFDIO_REG    1265         if (uffdio_register.mode & UFFDIO_REGISTER_MODE_WP) {
1323 #ifndef CONFIG_HAVE_ARCH_USERFAULTFD_WP       << 
1324                 goto out;                     << 
1325 #endif                                        << 
1326                 vm_flags |= VM_UFFD_WP;          1266                 vm_flags |= VM_UFFD_WP;
1327         }                                     !! 1267                 /*
1328         if (uffdio_register.mode & UFFDIO_REG !! 1268                  * FIXME: remove the below error constraint by
1329 #ifndef CONFIG_HAVE_ARCH_USERFAULTFD_MINOR    !! 1269                  * implementing the wprotect tracking mode.
                                                   >> 1270                  */
                                                   >> 1271                 ret = -EINVAL;
1330                 goto out;                        1272                 goto out;
1331 #endif                                        << 
1332                 vm_flags |= VM_UFFD_MINOR;    << 
1333         }                                        1273         }
1334                                                  1274 
1335         ret = validate_range(mm, uffdio_regis    1275         ret = validate_range(mm, uffdio_register.range.start,
1336                              uffdio_register.    1276                              uffdio_register.range.len);
1337         if (ret)                                 1277         if (ret)
1338                 goto out;                        1278                 goto out;
1339                                                  1279 
1340         start = uffdio_register.range.start;     1280         start = uffdio_register.range.start;
1341         end = start + uffdio_register.range.l    1281         end = start + uffdio_register.range.len;
1342                                                  1282 
1343         ret = -ENOMEM;                           1283         ret = -ENOMEM;
1344         if (!mmget_not_zero(mm))                 1284         if (!mmget_not_zero(mm))
1345                 goto out;                        1285                 goto out;
1346                                                  1286 
1347         ret = -EINVAL;                        !! 1287         down_write(&mm->mmap_sem);
1348         mmap_write_lock(mm);                  !! 1288         vma = find_vma_prev(mm, start, &prev);
1349         vma_iter_init(&vmi, mm, start);       << 
1350         vma = vma_find(&vmi, end);            << 
1351         if (!vma)                                1289         if (!vma)
1352                 goto out_unlock;                 1290                 goto out_unlock;
1353                                                  1291 
                                                   >> 1292         /* check that there's at least one vma in the range */
                                                   >> 1293         ret = -EINVAL;
                                                   >> 1294         if (vma->vm_start >= end)
                                                   >> 1295                 goto out_unlock;
                                                   >> 1296 
1354         /*                                       1297         /*
1355          * If the first vma contains huge pag    1298          * If the first vma contains huge pages, make sure start address
1356          * is aligned to huge page size.         1299          * is aligned to huge page size.
1357          */                                      1300          */
1358         if (is_vm_hugetlb_page(vma)) {           1301         if (is_vm_hugetlb_page(vma)) {
1359                 unsigned long vma_hpagesize =    1302                 unsigned long vma_hpagesize = vma_kernel_pagesize(vma);
1360                                                  1303 
1361                 if (start & (vma_hpagesize -     1304                 if (start & (vma_hpagesize - 1))
1362                         goto out_unlock;         1305                         goto out_unlock;
1363         }                                        1306         }
1364                                                  1307 
1365         /*                                       1308         /*
1366          * Search for not compatible vmas.       1309          * Search for not compatible vmas.
1367          */                                      1310          */
1368         found = false;                           1311         found = false;
1369         basic_ioctls = false;                 !! 1312         non_anon_pages = false;
1370         cur = vma;                            !! 1313         for (cur = vma; cur && cur->vm_start < end; cur = cur->vm_next) {
1371         do {                                  << 
1372                 cond_resched();                  1314                 cond_resched();
1373                                                  1315 
1374                 BUG_ON(!!cur->vm_userfaultfd_    1316                 BUG_ON(!!cur->vm_userfaultfd_ctx.ctx ^
1375                        !!(cur->vm_flags & __V !! 1317                        !!(cur->vm_flags & (VM_UFFD_MISSING | VM_UFFD_WP)));
1376                                                  1318 
1377                 /* check not compatible vmas     1319                 /* check not compatible vmas */
1378                 ret = -EINVAL;                   1320                 ret = -EINVAL;
1379                 if (!vma_can_userfault(cur, v !! 1321                 if (!vma_can_userfault(cur))
1380                         goto out_unlock;      << 
1381                                               << 
1382                 /*                            << 
1383                  * UFFDIO_COPY will fill file << 
1384                  * PROT_WRITE. This check enf << 
1385                  * MAP_SHARED, the process ha << 
1386                  * file. If VM_MAYWRITE is se << 
1387                  * MAP_SHARED vma: there is n << 
1388                  * F_WRITE_SEAL can be taken  << 
1389                  */                           << 
1390                 ret = -EPERM;                 << 
1391                 if (unlikely(!(cur->vm_flags  << 
1392                         goto out_unlock;         1322                         goto out_unlock;
1393                                               << 
1394                 /*                               1323                 /*
1395                  * If this vma contains endin    1324                  * If this vma contains ending address, and huge pages
1396                  * check alignment.              1325                  * check alignment.
1397                  */                              1326                  */
1398                 if (is_vm_hugetlb_page(cur) &    1327                 if (is_vm_hugetlb_page(cur) && end <= cur->vm_end &&
1399                     end > cur->vm_start) {       1328                     end > cur->vm_start) {
1400                         unsigned long vma_hpa    1329                         unsigned long vma_hpagesize = vma_kernel_pagesize(cur);
1401                                                  1330 
1402                         ret = -EINVAL;           1331                         ret = -EINVAL;
1403                                                  1332 
1404                         if (end & (vma_hpages    1333                         if (end & (vma_hpagesize - 1))
1405                                 goto out_unlo    1334                                 goto out_unlock;
1406                 }                                1335                 }
1407                 if ((vm_flags & VM_UFFD_WP) & << 
1408                         goto out_unlock;      << 
1409                                                  1336 
1410                 /*                               1337                 /*
1411                  * Check that this vma isn't     1338                  * Check that this vma isn't already owned by a
1412                  * different userfaultfd. We     1339                  * different userfaultfd. We can't allow more than one
1413                  * userfaultfd to own a singl    1340                  * userfaultfd to own a single vma simultaneously or we
1414                  * wouldn't know which one to    1341                  * wouldn't know which one to deliver the userfaults to.
1415                  */                              1342                  */
1416                 ret = -EBUSY;                    1343                 ret = -EBUSY;
1417                 if (cur->vm_userfaultfd_ctx.c    1344                 if (cur->vm_userfaultfd_ctx.ctx &&
1418                     cur->vm_userfaultfd_ctx.c    1345                     cur->vm_userfaultfd_ctx.ctx != ctx)
1419                         goto out_unlock;         1346                         goto out_unlock;
1420                                                  1347 
1421                 /*                               1348                 /*
1422                  * Note vmas containing huge     1349                  * Note vmas containing huge pages
1423                  */                              1350                  */
1424                 if (is_vm_hugetlb_page(cur))  !! 1351                 if (is_vm_hugetlb_page(cur) || vma_is_shmem(cur))
1425                         basic_ioctls = true;  !! 1352                         non_anon_pages = true;
1426                                                  1353 
1427                 found = true;                    1354                 found = true;
1428         } for_each_vma_range(vmi, cur, end);  !! 1355         }
1429         BUG_ON(!found);                          1356         BUG_ON(!found);
1430                                                  1357 
1431         vma_iter_set(&vmi, start);            << 
1432         prev = vma_prev(&vmi);                << 
1433         if (vma->vm_start < start)               1358         if (vma->vm_start < start)
1434                 prev = vma;                      1359                 prev = vma;
1435                                                  1360 
1436         ret = 0;                                 1361         ret = 0;
1437         for_each_vma_range(vmi, vma, end) {   !! 1362         do {
1438                 cond_resched();                  1363                 cond_resched();
1439                                                  1364 
1440                 BUG_ON(!vma_can_userfault(vma !! 1365                 BUG_ON(!vma_can_userfault(vma));
1441                 BUG_ON(vma->vm_userfaultfd_ct    1366                 BUG_ON(vma->vm_userfaultfd_ctx.ctx &&
1442                        vma->vm_userfaultfd_ct    1367                        vma->vm_userfaultfd_ctx.ctx != ctx);
1443                 WARN_ON(!(vma->vm_flags & VM_ << 
1444                                                  1368 
1445                 /*                               1369                 /*
1446                  * Nothing to do: this vma is    1370                  * Nothing to do: this vma is already registered into this
1447                  * userfaultfd and with the r    1371                  * userfaultfd and with the right tracking mode too.
1448                  */                              1372                  */
1449                 if (vma->vm_userfaultfd_ctx.c    1373                 if (vma->vm_userfaultfd_ctx.ctx == ctx &&
1450                     (vma->vm_flags & vm_flags    1374                     (vma->vm_flags & vm_flags) == vm_flags)
1451                         goto skip;               1375                         goto skip;
1452                                                  1376 
1453                 if (vma->vm_start > start)       1377                 if (vma->vm_start > start)
1454                         start = vma->vm_start    1378                         start = vma->vm_start;
1455                 vma_end = min(end, vma->vm_en    1379                 vma_end = min(end, vma->vm_end);
1456                                                  1380 
1457                 new_flags = (vma->vm_flags &  !! 1381                 new_flags = (vma->vm_flags & ~vm_flags) | vm_flags;
1458                 vma = vma_modify_flags_uffd(& !! 1382                 prev = vma_merge(mm, prev, start, vma_end, new_flags,
1459                                             n !! 1383                                  vma->anon_vma, vma->vm_file, vma->vm_pgoff,
1460                                             ( !! 1384                                  vma_policy(vma),
1461                 if (IS_ERR(vma)) {            !! 1385                                  ((struct vm_userfaultfd_ctx){ ctx }));
1462                         ret = PTR_ERR(vma);   !! 1386                 if (prev) {
1463                         break;                !! 1387                         vma = prev;
                                                   >> 1388                         goto next;
1464                 }                                1389                 }
1465                                               !! 1390                 if (vma->vm_start < start) {
                                                   >> 1391                         ret = split_vma(mm, vma, start, 1);
                                                   >> 1392                         if (ret)
                                                   >> 1393                                 break;
                                                   >> 1394                 }
                                                   >> 1395                 if (vma->vm_end > end) {
                                                   >> 1396                         ret = split_vma(mm, vma, end, 0);
                                                   >> 1397                         if (ret)
                                                   >> 1398                                 break;
                                                   >> 1399                 }
                                                   >> 1400         next:
1466                 /*                               1401                 /*
1467                  * In the vma_merge() success    1402                  * In the vma_merge() successful mprotect-like case 8:
1468                  * the next vma was merged in    1403                  * the next vma was merged into the current one and
1469                  * the current one has not be    1404                  * the current one has not been updated yet.
1470                  */                              1405                  */
1471                 vma_start_write(vma);         !! 1406                 vma->vm_flags = new_flags;
1472                 userfaultfd_set_vm_flags(vma, << 
1473                 vma->vm_userfaultfd_ctx.ctx =    1407                 vma->vm_userfaultfd_ctx.ctx = ctx;
1474                                                  1408 
1475                 if (is_vm_hugetlb_page(vma) & << 
1476                         hugetlb_unshare_all_p << 
1477                                               << 
1478         skip:                                    1409         skip:
1479                 prev = vma;                      1410                 prev = vma;
1480                 start = vma->vm_end;             1411                 start = vma->vm_end;
1481         }                                     !! 1412                 vma = vma->vm_next;
1482                                               !! 1413         } while (vma && vma->vm_start < end);
1483 out_unlock:                                      1414 out_unlock:
1484         mmap_write_unlock(mm);                !! 1415         up_write(&mm->mmap_sem);
1485         mmput(mm);                               1416         mmput(mm);
1486         if (!ret) {                              1417         if (!ret) {
1487                 __u64 ioctls_out;             << 
1488                                               << 
1489                 ioctls_out = basic_ioctls ? U << 
1490                     UFFD_API_RANGE_IOCTLS;    << 
1491                                               << 
1492                 /*                            << 
1493                  * Declare the WP ioctl only  << 
1494                  * specified and all checks p << 
1495                  */                           << 
1496                 if (!(uffdio_register.mode &  << 
1497                         ioctls_out &= ~((__u6 << 
1498                                               << 
1499                 /* CONTINUE ioctl is only sup << 
1500                 if (!(uffdio_register.mode &  << 
1501                         ioctls_out &= ~((__u6 << 
1502                                               << 
1503                 /*                               1418                 /*
1504                  * Now that we scanned all vm    1419                  * Now that we scanned all vmas we can already tell
1505                  * userland which ioctls meth    1420                  * userland which ioctls methods are guaranteed to
1506                  * succeed on this range.        1421                  * succeed on this range.
1507                  */                              1422                  */
1508                 if (put_user(ioctls_out, &use !! 1423                 if (put_user(non_anon_pages ? UFFD_API_RANGE_IOCTLS_BASIC :
                                                   >> 1424                              UFFD_API_RANGE_IOCTLS,
                                                   >> 1425                              &user_uffdio_register->ioctls))
1509                         ret = -EFAULT;           1426                         ret = -EFAULT;
1510         }                                        1427         }
1511 out:                                             1428 out:
1512         return ret;                              1429         return ret;
1513 }                                                1430 }
1514                                                  1431 
1515 static int userfaultfd_unregister(struct user    1432 static int userfaultfd_unregister(struct userfaultfd_ctx *ctx,
1516                                   unsigned lo    1433                                   unsigned long arg)
1517 {                                                1434 {
1518         struct mm_struct *mm = ctx->mm;          1435         struct mm_struct *mm = ctx->mm;
1519         struct vm_area_struct *vma, *prev, *c    1436         struct vm_area_struct *vma, *prev, *cur;
1520         int ret;                                 1437         int ret;
1521         struct uffdio_range uffdio_unregister    1438         struct uffdio_range uffdio_unregister;
1522         unsigned long new_flags;                 1439         unsigned long new_flags;
1523         bool found;                              1440         bool found;
1524         unsigned long start, end, vma_end;       1441         unsigned long start, end, vma_end;
1525         const void __user *buf = (void __user    1442         const void __user *buf = (void __user *)arg;
1526         struct vma_iterator vmi;              << 
1527         bool wp_async = userfaultfd_wp_async_ << 
1528                                                  1443 
1529         ret = -EFAULT;                           1444         ret = -EFAULT;
1530         if (copy_from_user(&uffdio_unregister    1445         if (copy_from_user(&uffdio_unregister, buf, sizeof(uffdio_unregister)))
1531                 goto out;                        1446                 goto out;
1532                                                  1447 
1533         ret = validate_range(mm, uffdio_unreg    1448         ret = validate_range(mm, uffdio_unregister.start,
1534                              uffdio_unregiste    1449                              uffdio_unregister.len);
1535         if (ret)                                 1450         if (ret)
1536                 goto out;                        1451                 goto out;
1537                                                  1452 
1538         start = uffdio_unregister.start;         1453         start = uffdio_unregister.start;
1539         end = start + uffdio_unregister.len;     1454         end = start + uffdio_unregister.len;
1540                                                  1455 
1541         ret = -ENOMEM;                           1456         ret = -ENOMEM;
1542         if (!mmget_not_zero(mm))                 1457         if (!mmget_not_zero(mm))
1543                 goto out;                        1458                 goto out;
1544                                                  1459 
1545         mmap_write_lock(mm);                  !! 1460         down_write(&mm->mmap_sem);
1546         ret = -EINVAL;                        !! 1461         vma = find_vma_prev(mm, start, &prev);
1547         vma_iter_init(&vmi, mm, start);       << 
1548         vma = vma_find(&vmi, end);            << 
1549         if (!vma)                                1462         if (!vma)
1550                 goto out_unlock;                 1463                 goto out_unlock;
1551                                                  1464 
                                                   >> 1465         /* check that there's at least one vma in the range */
                                                   >> 1466         ret = -EINVAL;
                                                   >> 1467         if (vma->vm_start >= end)
                                                   >> 1468                 goto out_unlock;
                                                   >> 1469 
1552         /*                                       1470         /*
1553          * If the first vma contains huge pag    1471          * If the first vma contains huge pages, make sure start address
1554          * is aligned to huge page size.         1472          * is aligned to huge page size.
1555          */                                      1473          */
1556         if (is_vm_hugetlb_page(vma)) {           1474         if (is_vm_hugetlb_page(vma)) {
1557                 unsigned long vma_hpagesize =    1475                 unsigned long vma_hpagesize = vma_kernel_pagesize(vma);
1558                                                  1476 
1559                 if (start & (vma_hpagesize -     1477                 if (start & (vma_hpagesize - 1))
1560                         goto out_unlock;         1478                         goto out_unlock;
1561         }                                        1479         }
1562                                                  1480 
1563         /*                                       1481         /*
1564          * Search for not compatible vmas.       1482          * Search for not compatible vmas.
1565          */                                      1483          */
1566         found = false;                           1484         found = false;
1567         cur = vma;                            !! 1485         ret = -EINVAL;
1568         do {                                  !! 1486         for (cur = vma; cur && cur->vm_start < end; cur = cur->vm_next) {
1569                 cond_resched();                  1487                 cond_resched();
1570                                                  1488 
1571                 BUG_ON(!!cur->vm_userfaultfd_    1489                 BUG_ON(!!cur->vm_userfaultfd_ctx.ctx ^
1572                        !!(cur->vm_flags & __V !! 1490                        !!(cur->vm_flags & (VM_UFFD_MISSING | VM_UFFD_WP)));
1573                                                  1491 
1574                 /*                               1492                 /*
1575                  * Check not compatible vmas,    1493                  * Check not compatible vmas, not strictly required
1576                  * here as not compatible vma    1494                  * here as not compatible vmas cannot have an
1577                  * userfaultfd_ctx registered    1495                  * userfaultfd_ctx registered on them, but this
1578                  * provides for more strict b    1496                  * provides for more strict behavior to notice
1579                  * unregistration errors.        1497                  * unregistration errors.
1580                  */                              1498                  */
1581                 if (!vma_can_userfault(cur, c !! 1499                 if (!vma_can_userfault(cur))
1582                         goto out_unlock;         1500                         goto out_unlock;
1583                                                  1501 
1584                 found = true;                    1502                 found = true;
1585         } for_each_vma_range(vmi, cur, end);  !! 1503         }
1586         BUG_ON(!found);                          1504         BUG_ON(!found);
1587                                                  1505 
1588         vma_iter_set(&vmi, start);            << 
1589         prev = vma_prev(&vmi);                << 
1590         if (vma->vm_start < start)               1506         if (vma->vm_start < start)
1591                 prev = vma;                      1507                 prev = vma;
1592                                                  1508 
1593         ret = 0;                                 1509         ret = 0;
1594         for_each_vma_range(vmi, vma, end) {   !! 1510         do {
1595                 cond_resched();                  1511                 cond_resched();
1596                                                  1512 
1597                 BUG_ON(!vma_can_userfault(vma !! 1513                 BUG_ON(!vma_can_userfault(vma));
1598                                                  1514 
1599                 /*                               1515                 /*
1600                  * Nothing to do: this vma is    1516                  * Nothing to do: this vma is already registered into this
1601                  * userfaultfd and with the r    1517                  * userfaultfd and with the right tracking mode too.
1602                  */                              1518                  */
1603                 if (!vma->vm_userfaultfd_ctx.    1519                 if (!vma->vm_userfaultfd_ctx.ctx)
1604                         goto skip;               1520                         goto skip;
1605                                                  1521 
1606                 WARN_ON(!(vma->vm_flags & VM_ << 
1607                                               << 
1608                 if (vma->vm_start > start)       1522                 if (vma->vm_start > start)
1609                         start = vma->vm_start    1523                         start = vma->vm_start;
1610                 vma_end = min(end, vma->vm_en    1524                 vma_end = min(end, vma->vm_end);
1611                                                  1525 
1612                 if (userfaultfd_missing(vma))    1526                 if (userfaultfd_missing(vma)) {
1613                         /*                       1527                         /*
1614                          * Wake any concurren    1528                          * Wake any concurrent pending userfault while
1615                          * we unregister, so     1529                          * we unregister, so they will not hang
1616                          * permanently and it    1530                          * permanently and it avoids userland to call
1617                          * UFFDIO_WAKE explic    1531                          * UFFDIO_WAKE explicitly.
1618                          */                      1532                          */
1619                         struct userfaultfd_wa    1533                         struct userfaultfd_wake_range range;
1620                         range.start = start;     1534                         range.start = start;
1621                         range.len = vma_end -    1535                         range.len = vma_end - start;
1622                         wake_userfault(vma->v    1536                         wake_userfault(vma->vm_userfaultfd_ctx.ctx, &range);
1623                 }                                1537                 }
1624                                                  1538 
1625                 /* Reset ptes for the whole v !! 1539                 new_flags = vma->vm_flags & ~(VM_UFFD_MISSING | VM_UFFD_WP);
1626                 if (userfaultfd_wp(vma))      !! 1540                 prev = vma_merge(mm, prev, start, vma_end, new_flags,
1627                         uffd_wp_range(vma, st !! 1541                                  vma->anon_vma, vma->vm_file, vma->vm_pgoff,
1628                                               !! 1542                                  vma_policy(vma),
1629                 new_flags = vma->vm_flags & ~ !! 1543                                  NULL_VM_UFFD_CTX);
1630                 vma = vma_modify_flags_uffd(& !! 1544                 if (prev) {
1631                                             n !! 1545                         vma = prev;
1632                 if (IS_ERR(vma)) {            !! 1546                         goto next;
1633                         ret = PTR_ERR(vma);   << 
1634                         break;                << 
1635                 }                                1547                 }
1636                                               !! 1548                 if (vma->vm_start < start) {
                                                   >> 1549                         ret = split_vma(mm, vma, start, 1);
                                                   >> 1550                         if (ret)
                                                   >> 1551                                 break;
                                                   >> 1552                 }
                                                   >> 1553                 if (vma->vm_end > end) {
                                                   >> 1554                         ret = split_vma(mm, vma, end, 0);
                                                   >> 1555                         if (ret)
                                                   >> 1556                                 break;
                                                   >> 1557                 }
                                                   >> 1558         next:
1637                 /*                               1559                 /*
1638                  * In the vma_merge() success    1560                  * In the vma_merge() successful mprotect-like case 8:
1639                  * the next vma was merged in    1561                  * the next vma was merged into the current one and
1640                  * the current one has not be    1562                  * the current one has not been updated yet.
1641                  */                              1563                  */
1642                 vma_start_write(vma);         !! 1564                 vma->vm_flags = new_flags;
1643                 userfaultfd_set_vm_flags(vma, << 
1644                 vma->vm_userfaultfd_ctx = NUL    1565                 vma->vm_userfaultfd_ctx = NULL_VM_UFFD_CTX;
1645                                                  1566 
1646         skip:                                    1567         skip:
1647                 prev = vma;                      1568                 prev = vma;
1648                 start = vma->vm_end;             1569                 start = vma->vm_end;
1649         }                                     !! 1570                 vma = vma->vm_next;
1650                                               !! 1571         } while (vma && vma->vm_start < end);
1651 out_unlock:                                      1572 out_unlock:
1652         mmap_write_unlock(mm);                !! 1573         up_write(&mm->mmap_sem);
1653         mmput(mm);                               1574         mmput(mm);
1654 out:                                             1575 out:
1655         return ret;                              1576         return ret;
1656 }                                                1577 }
1657                                                  1578 
1658 /*                                               1579 /*
1659  * userfaultfd_wake may be used in combinatio    1580  * userfaultfd_wake may be used in combination with the
1660  * UFFDIO_*_MODE_DONTWAKE to wakeup userfault    1581  * UFFDIO_*_MODE_DONTWAKE to wakeup userfaults in batches.
1661  */                                              1582  */
1662 static int userfaultfd_wake(struct userfaultf    1583 static int userfaultfd_wake(struct userfaultfd_ctx *ctx,
1663                             unsigned long arg    1584                             unsigned long arg)
1664 {                                                1585 {
1665         int ret;                                 1586         int ret;
1666         struct uffdio_range uffdio_wake;         1587         struct uffdio_range uffdio_wake;
1667         struct userfaultfd_wake_range range;     1588         struct userfaultfd_wake_range range;
1668         const void __user *buf = (void __user    1589         const void __user *buf = (void __user *)arg;
1669                                                  1590 
1670         ret = -EFAULT;                           1591         ret = -EFAULT;
1671         if (copy_from_user(&uffdio_wake, buf,    1592         if (copy_from_user(&uffdio_wake, buf, sizeof(uffdio_wake)))
1672                 goto out;                        1593                 goto out;
1673                                                  1594 
1674         ret = validate_range(ctx->mm, uffdio_    1595         ret = validate_range(ctx->mm, uffdio_wake.start, uffdio_wake.len);
1675         if (ret)                                 1596         if (ret)
1676                 goto out;                        1597                 goto out;
1677                                                  1598 
1678         range.start = uffdio_wake.start;         1599         range.start = uffdio_wake.start;
1679         range.len = uffdio_wake.len;             1600         range.len = uffdio_wake.len;
1680                                                  1601 
1681         /*                                       1602         /*
1682          * len == 0 means wake all and we don    1603          * len == 0 means wake all and we don't want to wake all here,
1683          * so check it again to be sure.         1604          * so check it again to be sure.
1684          */                                      1605          */
1685         VM_BUG_ON(!range.len);                   1606         VM_BUG_ON(!range.len);
1686                                                  1607 
1687         wake_userfault(ctx, &range);             1608         wake_userfault(ctx, &range);
1688         ret = 0;                                 1609         ret = 0;
1689                                                  1610 
1690 out:                                             1611 out:
1691         return ret;                              1612         return ret;
1692 }                                                1613 }
1693                                                  1614 
1694 static int userfaultfd_copy(struct userfaultf    1615 static int userfaultfd_copy(struct userfaultfd_ctx *ctx,
1695                             unsigned long arg    1616                             unsigned long arg)
1696 {                                                1617 {
1697         __s64 ret;                               1618         __s64 ret;
1698         struct uffdio_copy uffdio_copy;          1619         struct uffdio_copy uffdio_copy;
1699         struct uffdio_copy __user *user_uffdi    1620         struct uffdio_copy __user *user_uffdio_copy;
1700         struct userfaultfd_wake_range range;     1621         struct userfaultfd_wake_range range;
1701         uffd_flags_t flags = 0;               << 
1702                                                  1622 
1703         user_uffdio_copy = (struct uffdio_cop    1623         user_uffdio_copy = (struct uffdio_copy __user *) arg;
1704                                                  1624 
1705         ret = -EAGAIN;                        << 
1706         if (atomic_read(&ctx->mmap_changing)) << 
1707                 goto out;                     << 
1708                                               << 
1709         ret = -EFAULT;                           1625         ret = -EFAULT;
1710         if (copy_from_user(&uffdio_copy, user    1626         if (copy_from_user(&uffdio_copy, user_uffdio_copy,
1711                            /* don't copy "cop    1627                            /* don't copy "copy" last field */
1712                            sizeof(uffdio_copy    1628                            sizeof(uffdio_copy)-sizeof(__s64)))
1713                 goto out;                        1629                 goto out;
1714                                                  1630 
1715         ret = validate_unaligned_range(ctx->m << 
1716                                        uffdio << 
1717         if (ret)                              << 
1718                 goto out;                     << 
1719         ret = validate_range(ctx->mm, uffdio_    1631         ret = validate_range(ctx->mm, uffdio_copy.dst, uffdio_copy.len);
1720         if (ret)                                 1632         if (ret)
1721                 goto out;                        1633                 goto out;
1722                                               !! 1634         /*
                                                   >> 1635          * double check for wraparound just in case. copy_from_user()
                                                   >> 1636          * will later check uffdio_copy.src + uffdio_copy.len to fit
                                                   >> 1637          * in the userland range.
                                                   >> 1638          */
1723         ret = -EINVAL;                           1639         ret = -EINVAL;
1724         if (uffdio_copy.mode & ~(UFFDIO_COPY_ !! 1640         if (uffdio_copy.src + uffdio_copy.len <= uffdio_copy.src)
                                                   >> 1641                 goto out;
                                                   >> 1642         if (uffdio_copy.mode & ~UFFDIO_COPY_MODE_DONTWAKE)
1725                 goto out;                        1643                 goto out;
1726         if (uffdio_copy.mode & UFFDIO_COPY_MO << 
1727                 flags |= MFILL_ATOMIC_WP;     << 
1728         if (mmget_not_zero(ctx->mm)) {           1644         if (mmget_not_zero(ctx->mm)) {
1729                 ret = mfill_atomic_copy(ctx,  !! 1645                 ret = mcopy_atomic(ctx->mm, uffdio_copy.dst, uffdio_copy.src,
1730                                         uffdi !! 1646                                    uffdio_copy.len);
1731                 mmput(ctx->mm);                  1647                 mmput(ctx->mm);
1732         } else {                                 1648         } else {
1733                 return -ESRCH;                   1649                 return -ESRCH;
1734         }                                        1650         }
1735         if (unlikely(put_user(ret, &user_uffd    1651         if (unlikely(put_user(ret, &user_uffdio_copy->copy)))
1736                 return -EFAULT;                  1652                 return -EFAULT;
1737         if (ret < 0)                             1653         if (ret < 0)
1738                 goto out;                        1654                 goto out;
1739         BUG_ON(!ret);                            1655         BUG_ON(!ret);
1740         /* len == 0 would wake all */            1656         /* len == 0 would wake all */
1741         range.len = ret;                         1657         range.len = ret;
1742         if (!(uffdio_copy.mode & UFFDIO_COPY_    1658         if (!(uffdio_copy.mode & UFFDIO_COPY_MODE_DONTWAKE)) {
1743                 range.start = uffdio_copy.dst    1659                 range.start = uffdio_copy.dst;
1744                 wake_userfault(ctx, &range);     1660                 wake_userfault(ctx, &range);
1745         }                                        1661         }
1746         ret = range.len == uffdio_copy.len ?     1662         ret = range.len == uffdio_copy.len ? 0 : -EAGAIN;
1747 out:                                             1663 out:
1748         return ret;                              1664         return ret;
1749 }                                                1665 }
1750                                                  1666 
1751 static int userfaultfd_zeropage(struct userfa    1667 static int userfaultfd_zeropage(struct userfaultfd_ctx *ctx,
1752                                 unsigned long    1668                                 unsigned long arg)
1753 {                                                1669 {
1754         __s64 ret;                               1670         __s64 ret;
1755         struct uffdio_zeropage uffdio_zeropag    1671         struct uffdio_zeropage uffdio_zeropage;
1756         struct uffdio_zeropage __user *user_u    1672         struct uffdio_zeropage __user *user_uffdio_zeropage;
1757         struct userfaultfd_wake_range range;     1673         struct userfaultfd_wake_range range;
1758                                                  1674 
1759         user_uffdio_zeropage = (struct uffdio    1675         user_uffdio_zeropage = (struct uffdio_zeropage __user *) arg;
1760                                                  1676 
1761         ret = -EAGAIN;                        << 
1762         if (atomic_read(&ctx->mmap_changing)) << 
1763                 goto out;                     << 
1764                                               << 
1765         ret = -EFAULT;                           1677         ret = -EFAULT;
1766         if (copy_from_user(&uffdio_zeropage,     1678         if (copy_from_user(&uffdio_zeropage, user_uffdio_zeropage,
1767                            /* don't copy "zer    1679                            /* don't copy "zeropage" last field */
1768                            sizeof(uffdio_zero    1680                            sizeof(uffdio_zeropage)-sizeof(__s64)))
1769                 goto out;                        1681                 goto out;
1770                                                  1682 
1771         ret = validate_range(ctx->mm, uffdio_    1683         ret = validate_range(ctx->mm, uffdio_zeropage.range.start,
1772                              uffdio_zeropage.    1684                              uffdio_zeropage.range.len);
1773         if (ret)                                 1685         if (ret)
1774                 goto out;                        1686                 goto out;
1775         ret = -EINVAL;                           1687         ret = -EINVAL;
1776         if (uffdio_zeropage.mode & ~UFFDIO_ZE    1688         if (uffdio_zeropage.mode & ~UFFDIO_ZEROPAGE_MODE_DONTWAKE)
1777                 goto out;                        1689                 goto out;
1778                                                  1690 
1779         if (mmget_not_zero(ctx->mm)) {           1691         if (mmget_not_zero(ctx->mm)) {
1780                 ret = mfill_atomic_zeropage(c !! 1692                 ret = mfill_zeropage(ctx->mm, uffdio_zeropage.range.start,
1781                                            uf !! 1693                                      uffdio_zeropage.range.len);
1782                 mmput(ctx->mm);                  1694                 mmput(ctx->mm);
1783         } else {                                 1695         } else {
1784                 return -ESRCH;                   1696                 return -ESRCH;
1785         }                                        1697         }
1786         if (unlikely(put_user(ret, &user_uffd    1698         if (unlikely(put_user(ret, &user_uffdio_zeropage->zeropage)))
1787                 return -EFAULT;                  1699                 return -EFAULT;
1788         if (ret < 0)                             1700         if (ret < 0)
1789                 goto out;                        1701                 goto out;
1790         /* len == 0 would wake all */            1702         /* len == 0 would wake all */
1791         BUG_ON(!ret);                            1703         BUG_ON(!ret);
1792         range.len = ret;                         1704         range.len = ret;
1793         if (!(uffdio_zeropage.mode & UFFDIO_Z    1705         if (!(uffdio_zeropage.mode & UFFDIO_ZEROPAGE_MODE_DONTWAKE)) {
1794                 range.start = uffdio_zeropage    1706                 range.start = uffdio_zeropage.range.start;
1795                 wake_userfault(ctx, &range);     1707                 wake_userfault(ctx, &range);
1796         }                                        1708         }
1797         ret = range.len == uffdio_zeropage.ra    1709         ret = range.len == uffdio_zeropage.range.len ? 0 : -EAGAIN;
1798 out:                                             1710 out:
1799         return ret;                              1711         return ret;
1800 }                                                1712 }
1801                                                  1713 
1802 static int userfaultfd_writeprotect(struct us << 
1803                                     unsigned  << 
1804 {                                             << 
1805         int ret;                              << 
1806         struct uffdio_writeprotect uffdio_wp; << 
1807         struct uffdio_writeprotect __user *us << 
1808         struct userfaultfd_wake_range range;  << 
1809         bool mode_wp, mode_dontwake;          << 
1810                                               << 
1811         if (atomic_read(&ctx->mmap_changing)) << 
1812                 return -EAGAIN;               << 
1813                                               << 
1814         user_uffdio_wp = (struct uffdio_write << 
1815                                               << 
1816         if (copy_from_user(&uffdio_wp, user_u << 
1817                            sizeof(struct uffd << 
1818                 return -EFAULT;               << 
1819                                               << 
1820         ret = validate_range(ctx->mm, uffdio_ << 
1821                              uffdio_wp.range. << 
1822         if (ret)                              << 
1823                 return ret;                   << 
1824                                               << 
1825         if (uffdio_wp.mode & ~(UFFDIO_WRITEPR << 
1826                                UFFDIO_WRITEPR << 
1827                 return -EINVAL;               << 
1828                                               << 
1829         mode_wp = uffdio_wp.mode & UFFDIO_WRI << 
1830         mode_dontwake = uffdio_wp.mode & UFFD << 
1831                                               << 
1832         if (mode_wp && mode_dontwake)         << 
1833                 return -EINVAL;               << 
1834                                               << 
1835         if (mmget_not_zero(ctx->mm)) {        << 
1836                 ret = mwriteprotect_range(ctx << 
1837                                           uff << 
1838                 mmput(ctx->mm);               << 
1839         } else {                              << 
1840                 return -ESRCH;                << 
1841         }                                     << 
1842                                               << 
1843         if (ret)                              << 
1844                 return ret;                   << 
1845                                               << 
1846         if (!mode_wp && !mode_dontwake) {     << 
1847                 range.start = uffdio_wp.range << 
1848                 range.len = uffdio_wp.range.l << 
1849                 wake_userfault(ctx, &range);  << 
1850         }                                     << 
1851         return ret;                           << 
1852 }                                             << 
1853                                               << 
1854 static int userfaultfd_continue(struct userfa << 
1855 {                                             << 
1856         __s64 ret;                            << 
1857         struct uffdio_continue uffdio_continu << 
1858         struct uffdio_continue __user *user_u << 
1859         struct userfaultfd_wake_range range;  << 
1860         uffd_flags_t flags = 0;               << 
1861                                               << 
1862         user_uffdio_continue = (struct uffdio << 
1863                                               << 
1864         ret = -EAGAIN;                        << 
1865         if (atomic_read(&ctx->mmap_changing)) << 
1866                 goto out;                     << 
1867                                               << 
1868         ret = -EFAULT;                        << 
1869         if (copy_from_user(&uffdio_continue,  << 
1870                            /* don't copy the  << 
1871                            sizeof(uffdio_cont << 
1872                 goto out;                     << 
1873                                               << 
1874         ret = validate_range(ctx->mm, uffdio_ << 
1875                              uffdio_continue. << 
1876         if (ret)                              << 
1877                 goto out;                     << 
1878                                               << 
1879         ret = -EINVAL;                        << 
1880         if (uffdio_continue.mode & ~(UFFDIO_C << 
1881                                      UFFDIO_C << 
1882                 goto out;                     << 
1883         if (uffdio_continue.mode & UFFDIO_CON << 
1884                 flags |= MFILL_ATOMIC_WP;     << 
1885                                               << 
1886         if (mmget_not_zero(ctx->mm)) {        << 
1887                 ret = mfill_atomic_continue(c << 
1888                                             u << 
1889                 mmput(ctx->mm);               << 
1890         } else {                              << 
1891                 return -ESRCH;                << 
1892         }                                     << 
1893                                               << 
1894         if (unlikely(put_user(ret, &user_uffd << 
1895                 return -EFAULT;               << 
1896         if (ret < 0)                          << 
1897                 goto out;                     << 
1898                                               << 
1899         /* len == 0 would wake all */         << 
1900         BUG_ON(!ret);                         << 
1901         range.len = ret;                      << 
1902         if (!(uffdio_continue.mode & UFFDIO_C << 
1903                 range.start = uffdio_continue << 
1904                 wake_userfault(ctx, &range);  << 
1905         }                                     << 
1906         ret = range.len == uffdio_continue.ra << 
1907                                               << 
1908 out:                                          << 
1909         return ret;                           << 
1910 }                                             << 
1911                                               << 
1912 static inline int userfaultfd_poison(struct u << 
1913 {                                             << 
1914         __s64 ret;                            << 
1915         struct uffdio_poison uffdio_poison;   << 
1916         struct uffdio_poison __user *user_uff << 
1917         struct userfaultfd_wake_range range;  << 
1918                                               << 
1919         user_uffdio_poison = (struct uffdio_p << 
1920                                               << 
1921         ret = -EAGAIN;                        << 
1922         if (atomic_read(&ctx->mmap_changing)) << 
1923                 goto out;                     << 
1924                                               << 
1925         ret = -EFAULT;                        << 
1926         if (copy_from_user(&uffdio_poison, us << 
1927                            /* don't copy the  << 
1928                            sizeof(uffdio_pois << 
1929                 goto out;                     << 
1930                                               << 
1931         ret = validate_range(ctx->mm, uffdio_ << 
1932                              uffdio_poison.ra << 
1933         if (ret)                              << 
1934                 goto out;                     << 
1935                                               << 
1936         ret = -EINVAL;                        << 
1937         if (uffdio_poison.mode & ~UFFDIO_POIS << 
1938                 goto out;                     << 
1939                                               << 
1940         if (mmget_not_zero(ctx->mm)) {        << 
1941                 ret = mfill_atomic_poison(ctx << 
1942                                           uff << 
1943                 mmput(ctx->mm);               << 
1944         } else {                              << 
1945                 return -ESRCH;                << 
1946         }                                     << 
1947                                               << 
1948         if (unlikely(put_user(ret, &user_uffd << 
1949                 return -EFAULT;               << 
1950         if (ret < 0)                          << 
1951                 goto out;                     << 
1952                                               << 
1953         /* len == 0 would wake all */         << 
1954         BUG_ON(!ret);                         << 
1955         range.len = ret;                      << 
1956         if (!(uffdio_poison.mode & UFFDIO_POI << 
1957                 range.start = uffdio_poison.r << 
1958                 wake_userfault(ctx, &range);  << 
1959         }                                     << 
1960         ret = range.len == uffdio_poison.rang << 
1961                                               << 
1962 out:                                          << 
1963         return ret;                           << 
1964 }                                             << 
1965                                               << 
1966 bool userfaultfd_wp_async(struct vm_area_stru << 
1967 {                                             << 
1968         return userfaultfd_wp_async_ctx(vma-> << 
1969 }                                             << 
1970                                               << 
1971 static inline unsigned int uffd_ctx_features(    1714 static inline unsigned int uffd_ctx_features(__u64 user_features)
1972 {                                                1715 {
1973         /*                                       1716         /*
1974          * For the current set of features th !! 1717          * For the current set of features the bits just coincide
1975          * UFFD_FEATURE_INITIALIZED to mark t << 
1976          */                                      1718          */
1977         return (unsigned int)user_features |  !! 1719         return (unsigned int)user_features;
1978 }                                             << 
1979                                               << 
1980 static int userfaultfd_move(struct userfaultf << 
1981                             unsigned long arg << 
1982 {                                             << 
1983         __s64 ret;                            << 
1984         struct uffdio_move uffdio_move;       << 
1985         struct uffdio_move __user *user_uffdi << 
1986         struct userfaultfd_wake_range range;  << 
1987         struct mm_struct *mm = ctx->mm;       << 
1988                                               << 
1989         user_uffdio_move = (struct uffdio_mov << 
1990                                               << 
1991         if (atomic_read(&ctx->mmap_changing)) << 
1992                 return -EAGAIN;               << 
1993                                               << 
1994         if (copy_from_user(&uffdio_move, user << 
1995                            /* don't copy "mov << 
1996                            sizeof(uffdio_move << 
1997                 return -EFAULT;               << 
1998                                               << 
1999         /* Do not allow cross-mm moves. */    << 
2000         if (mm != current->mm)                << 
2001                 return -EINVAL;               << 
2002                                               << 
2003         ret = validate_range(mm, uffdio_move. << 
2004         if (ret)                              << 
2005                 return ret;                   << 
2006                                               << 
2007         ret = validate_range(mm, uffdio_move. << 
2008         if (ret)                              << 
2009                 return ret;                   << 
2010                                               << 
2011         if (uffdio_move.mode & ~(UFFDIO_MOVE_ << 
2012                                   UFFDIO_MOVE << 
2013                 return -EINVAL;               << 
2014                                               << 
2015         if (mmget_not_zero(mm)) {             << 
2016                 ret = move_pages(ctx, uffdio_ << 
2017                                  uffdio_move. << 
2018                 mmput(mm);                    << 
2019         } else {                              << 
2020                 return -ESRCH;                << 
2021         }                                     << 
2022                                               << 
2023         if (unlikely(put_user(ret, &user_uffd << 
2024                 return -EFAULT;               << 
2025         if (ret < 0)                          << 
2026                 goto out;                     << 
2027                                               << 
2028         /* len == 0 would wake all */         << 
2029         VM_WARN_ON(!ret);                     << 
2030         range.len = ret;                      << 
2031         if (!(uffdio_move.mode & UFFDIO_MOVE_ << 
2032                 range.start = uffdio_move.dst << 
2033                 wake_userfault(ctx, &range);  << 
2034         }                                     << 
2035         ret = range.len == uffdio_move.len ?  << 
2036                                               << 
2037 out:                                          << 
2038         return ret;                           << 
2039 }                                                1720 }
2040                                                  1721 
2041 /*                                               1722 /*
2042  * userland asks for a certain API version an    1723  * userland asks for a certain API version and we return which bits
2043  * and ioctl commands are implemented in this    1724  * and ioctl commands are implemented in this kernel for such API
2044  * version or -EINVAL if unknown.                1725  * version or -EINVAL if unknown.
2045  */                                              1726  */
2046 static int userfaultfd_api(struct userfaultfd    1727 static int userfaultfd_api(struct userfaultfd_ctx *ctx,
2047                            unsigned long arg)    1728                            unsigned long arg)
2048 {                                                1729 {
2049         struct uffdio_api uffdio_api;            1730         struct uffdio_api uffdio_api;
2050         void __user *buf = (void __user *)arg    1731         void __user *buf = (void __user *)arg;
2051         unsigned int ctx_features;            << 
2052         int ret;                                 1732         int ret;
2053         __u64 features;                          1733         __u64 features;
2054                                                  1734 
                                                   >> 1735         ret = -EINVAL;
                                                   >> 1736         if (ctx->state != UFFD_STATE_WAIT_API)
                                                   >> 1737                 goto out;
2055         ret = -EFAULT;                           1738         ret = -EFAULT;
2056         if (copy_from_user(&uffdio_api, buf,     1739         if (copy_from_user(&uffdio_api, buf, sizeof(uffdio_api)))
2057                 goto out;                        1740                 goto out;
2058         features = uffdio_api.features;          1741         features = uffdio_api.features;
2059         ret = -EINVAL;                        !! 1742         if (uffdio_api.api != UFFD_API || (features & ~UFFD_API_FEATURES)) {
2060         if (uffdio_api.api != UFFD_API)       !! 1743                 memset(&uffdio_api, 0, sizeof(uffdio_api));
2061                 goto err_out;                 !! 1744                 if (copy_to_user(buf, &uffdio_api, sizeof(uffdio_api)))
2062         ret = -EPERM;                         !! 1745                         goto out;
2063         if ((features & UFFD_FEATURE_EVENT_FO !! 1746                 ret = -EINVAL;
2064                 goto err_out;                 !! 1747                 goto out;
2065                                               !! 1748         }
2066         /* WP_ASYNC relies on WP_UNPOPULATED, << 
2067         if (features & UFFD_FEATURE_WP_ASYNC) << 
2068                 features |= UFFD_FEATURE_WP_U << 
2069                                               << 
2070         /* report all available features and     1749         /* report all available features and ioctls to userland */
2071         uffdio_api.features = UFFD_API_FEATUR    1750         uffdio_api.features = UFFD_API_FEATURES;
2072 #ifndef CONFIG_HAVE_ARCH_USERFAULTFD_MINOR    << 
2073         uffdio_api.features &=                << 
2074                 ~(UFFD_FEATURE_MINOR_HUGETLBF << 
2075 #endif                                        << 
2076 #ifndef CONFIG_HAVE_ARCH_USERFAULTFD_WP       << 
2077         uffdio_api.features &= ~UFFD_FEATURE_ << 
2078 #endif                                        << 
2079 #ifndef CONFIG_PTE_MARKER_UFFD_WP             << 
2080         uffdio_api.features &= ~UFFD_FEATURE_ << 
2081         uffdio_api.features &= ~UFFD_FEATURE_ << 
2082         uffdio_api.features &= ~UFFD_FEATURE_ << 
2083 #endif                                        << 
2084                                               << 
2085         ret = -EINVAL;                        << 
2086         if (features & ~uffdio_api.features)  << 
2087                 goto err_out;                 << 
2088                                               << 
2089         uffdio_api.ioctls = UFFD_API_IOCTLS;     1751         uffdio_api.ioctls = UFFD_API_IOCTLS;
2090         ret = -EFAULT;                           1752         ret = -EFAULT;
2091         if (copy_to_user(buf, &uffdio_api, si    1753         if (copy_to_user(buf, &uffdio_api, sizeof(uffdio_api)))
2092                 goto out;                        1754                 goto out;
2093                                               !! 1755         ctx->state = UFFD_STATE_RUNNING;
2094         /* only enable the requested features    1756         /* only enable the requested features for this uffd context */
2095         ctx_features = uffd_ctx_features(feat !! 1757         ctx->features = uffd_ctx_features(features);
2096         ret = -EINVAL;                        << 
2097         if (cmpxchg(&ctx->features, 0, ctx_fe << 
2098                 goto err_out;                 << 
2099                                               << 
2100         ret = 0;                                 1758         ret = 0;
2101 out:                                             1759 out:
2102         return ret;                              1760         return ret;
2103 err_out:                                      << 
2104         memset(&uffdio_api, 0, sizeof(uffdio_ << 
2105         if (copy_to_user(buf, &uffdio_api, si << 
2106                 ret = -EFAULT;                << 
2107         goto out;                             << 
2108 }                                                1761 }
2109                                                  1762 
2110 static long userfaultfd_ioctl(struct file *fi    1763 static long userfaultfd_ioctl(struct file *file, unsigned cmd,
2111                               unsigned long a    1764                               unsigned long arg)
2112 {                                                1765 {
2113         int ret = -EINVAL;                       1766         int ret = -EINVAL;
2114         struct userfaultfd_ctx *ctx = file->p    1767         struct userfaultfd_ctx *ctx = file->private_data;
2115                                                  1768 
2116         if (cmd != UFFDIO_API && !userfaultfd !! 1769         if (cmd != UFFDIO_API && ctx->state == UFFD_STATE_WAIT_API)
2117                 return -EINVAL;                  1770                 return -EINVAL;
2118                                                  1771 
2119         switch(cmd) {                            1772         switch(cmd) {
2120         case UFFDIO_API:                         1773         case UFFDIO_API:
2121                 ret = userfaultfd_api(ctx, ar    1774                 ret = userfaultfd_api(ctx, arg);
2122                 break;                           1775                 break;
2123         case UFFDIO_REGISTER:                    1776         case UFFDIO_REGISTER:
2124                 ret = userfaultfd_register(ct    1777                 ret = userfaultfd_register(ctx, arg);
2125                 break;                           1778                 break;
2126         case UFFDIO_UNREGISTER:                  1779         case UFFDIO_UNREGISTER:
2127                 ret = userfaultfd_unregister(    1780                 ret = userfaultfd_unregister(ctx, arg);
2128                 break;                           1781                 break;
2129         case UFFDIO_WAKE:                        1782         case UFFDIO_WAKE:
2130                 ret = userfaultfd_wake(ctx, a    1783                 ret = userfaultfd_wake(ctx, arg);
2131                 break;                           1784                 break;
2132         case UFFDIO_COPY:                        1785         case UFFDIO_COPY:
2133                 ret = userfaultfd_copy(ctx, a    1786                 ret = userfaultfd_copy(ctx, arg);
2134                 break;                           1787                 break;
2135         case UFFDIO_ZEROPAGE:                    1788         case UFFDIO_ZEROPAGE:
2136                 ret = userfaultfd_zeropage(ct    1789                 ret = userfaultfd_zeropage(ctx, arg);
2137                 break;                           1790                 break;
2138         case UFFDIO_MOVE:                     << 
2139                 ret = userfaultfd_move(ctx, a << 
2140                 break;                        << 
2141         case UFFDIO_WRITEPROTECT:             << 
2142                 ret = userfaultfd_writeprotec << 
2143                 break;                        << 
2144         case UFFDIO_CONTINUE:                 << 
2145                 ret = userfaultfd_continue(ct << 
2146                 break;                        << 
2147         case UFFDIO_POISON:                   << 
2148                 ret = userfaultfd_poison(ctx, << 
2149                 break;                        << 
2150         }                                        1791         }
2151         return ret;                              1792         return ret;
2152 }                                                1793 }
2153                                                  1794 
2154 #ifdef CONFIG_PROC_FS                            1795 #ifdef CONFIG_PROC_FS
2155 static void userfaultfd_show_fdinfo(struct se    1796 static void userfaultfd_show_fdinfo(struct seq_file *m, struct file *f)
2156 {                                                1797 {
2157         struct userfaultfd_ctx *ctx = f->priv    1798         struct userfaultfd_ctx *ctx = f->private_data;
2158         wait_queue_entry_t *wq;                  1799         wait_queue_entry_t *wq;
                                                   >> 1800         struct userfaultfd_wait_queue *uwq;
2159         unsigned long pending = 0, total = 0;    1801         unsigned long pending = 0, total = 0;
2160                                                  1802 
2161         spin_lock_irq(&ctx->fault_pending_wqh !! 1803         spin_lock(&ctx->fault_pending_wqh.lock);
2162         list_for_each_entry(wq, &ctx->fault_p    1804         list_for_each_entry(wq, &ctx->fault_pending_wqh.head, entry) {
                                                   >> 1805                 uwq = container_of(wq, struct userfaultfd_wait_queue, wq);
2163                 pending++;                       1806                 pending++;
2164                 total++;                         1807                 total++;
2165         }                                        1808         }
2166         list_for_each_entry(wq, &ctx->fault_w    1809         list_for_each_entry(wq, &ctx->fault_wqh.head, entry) {
                                                   >> 1810                 uwq = container_of(wq, struct userfaultfd_wait_queue, wq);
2167                 total++;                         1811                 total++;
2168         }                                        1812         }
2169         spin_unlock_irq(&ctx->fault_pending_w !! 1813         spin_unlock(&ctx->fault_pending_wqh.lock);
2170                                                  1814 
2171         /*                                       1815         /*
2172          * If more protocols will be added, t    1816          * If more protocols will be added, there will be all shown
2173          * separated by a space. Like this:      1817          * separated by a space. Like this:
2174          *      protocols: aa:... bb:...         1818          *      protocols: aa:... bb:...
2175          */                                      1819          */
2176         seq_printf(m, "pending:\t%lu\ntotal:\    1820         seq_printf(m, "pending:\t%lu\ntotal:\t%lu\nAPI:\t%Lx:%x:%Lx\n",
2177                    pending, total, UFFD_API,     1821                    pending, total, UFFD_API, ctx->features,
2178                    UFFD_API_IOCTLS|UFFD_API_R    1822                    UFFD_API_IOCTLS|UFFD_API_RANGE_IOCTLS);
2179 }                                                1823 }
2180 #endif                                           1824 #endif
2181                                                  1825 
2182 static const struct file_operations userfault    1826 static const struct file_operations userfaultfd_fops = {
2183 #ifdef CONFIG_PROC_FS                            1827 #ifdef CONFIG_PROC_FS
2184         .show_fdinfo    = userfaultfd_show_fd    1828         .show_fdinfo    = userfaultfd_show_fdinfo,
2185 #endif                                           1829 #endif
2186         .release        = userfaultfd_release    1830         .release        = userfaultfd_release,
2187         .poll           = userfaultfd_poll,      1831         .poll           = userfaultfd_poll,
2188         .read_iter      = userfaultfd_read_it !! 1832         .read           = userfaultfd_read,
2189         .unlocked_ioctl = userfaultfd_ioctl,     1833         .unlocked_ioctl = userfaultfd_ioctl,
2190         .compat_ioctl   = compat_ptr_ioctl,   !! 1834         .compat_ioctl   = userfaultfd_ioctl,
2191         .llseek         = noop_llseek,           1835         .llseek         = noop_llseek,
2192 };                                               1836 };
2193                                                  1837 
2194 static void init_once_userfaultfd_ctx(void *m    1838 static void init_once_userfaultfd_ctx(void *mem)
2195 {                                                1839 {
2196         struct userfaultfd_ctx *ctx = (struct    1840         struct userfaultfd_ctx *ctx = (struct userfaultfd_ctx *) mem;
2197                                                  1841 
2198         init_waitqueue_head(&ctx->fault_pendi    1842         init_waitqueue_head(&ctx->fault_pending_wqh);
2199         init_waitqueue_head(&ctx->fault_wqh);    1843         init_waitqueue_head(&ctx->fault_wqh);
2200         init_waitqueue_head(&ctx->event_wqh);    1844         init_waitqueue_head(&ctx->event_wqh);
2201         init_waitqueue_head(&ctx->fd_wqh);       1845         init_waitqueue_head(&ctx->fd_wqh);
2202         seqcount_spinlock_init(&ctx->refile_s !! 1846         seqcount_init(&ctx->refile_seq);
2203 }                                                1847 }
2204                                                  1848 
2205 static int new_userfaultfd(int flags)         !! 1849 /**
                                                   >> 1850  * userfaultfd_file_create - Creates a userfaultfd file pointer.
                                                   >> 1851  * @flags: Flags for the userfaultfd file.
                                                   >> 1852  *
                                                   >> 1853  * This function creates a userfaultfd file pointer, w/out installing
                                                   >> 1854  * it into the fd table. This is useful when the userfaultfd file is
                                                   >> 1855  * used during the initialization of data structures that require
                                                   >> 1856  * extra setup after the userfaultfd creation. So the userfaultfd
                                                   >> 1857  * creation is split into the file pointer creation phase, and the
                                                   >> 1858  * file descriptor installation phase.  In this way races with
                                                   >> 1859  * userspace closing the newly installed file descriptor can be
                                                   >> 1860  * avoided.  Returns a userfaultfd file pointer, or a proper error
                                                   >> 1861  * pointer.
                                                   >> 1862  */
                                                   >> 1863 static struct file *userfaultfd_file_create(int flags)
2206 {                                                1864 {
2207         struct userfaultfd_ctx *ctx;          << 
2208         struct file *file;                       1865         struct file *file;
2209         int fd;                               !! 1866         struct userfaultfd_ctx *ctx;
2210                                                  1867 
2211         BUG_ON(!current->mm);                    1868         BUG_ON(!current->mm);
2212                                                  1869 
2213         /* Check the UFFD_* constants for con    1870         /* Check the UFFD_* constants for consistency.  */
2214         BUILD_BUG_ON(UFFD_USER_MODE_ONLY & UF << 
2215         BUILD_BUG_ON(UFFD_CLOEXEC != O_CLOEXE    1871         BUILD_BUG_ON(UFFD_CLOEXEC != O_CLOEXEC);
2216         BUILD_BUG_ON(UFFD_NONBLOCK != O_NONBL    1872         BUILD_BUG_ON(UFFD_NONBLOCK != O_NONBLOCK);
2217                                                  1873 
2218         if (flags & ~(UFFD_SHARED_FCNTL_FLAGS !! 1874         file = ERR_PTR(-EINVAL);
2219                 return -EINVAL;               !! 1875         if (flags & ~UFFD_SHARED_FCNTL_FLAGS)
                                                   >> 1876                 goto out;
2220                                                  1877 
                                                   >> 1878         file = ERR_PTR(-ENOMEM);
2221         ctx = kmem_cache_alloc(userfaultfd_ct    1879         ctx = kmem_cache_alloc(userfaultfd_ctx_cachep, GFP_KERNEL);
2222         if (!ctx)                                1880         if (!ctx)
2223                 return -ENOMEM;               !! 1881                 goto out;
2224                                                  1882 
2225         refcount_set(&ctx->refcount, 1);      !! 1883         atomic_set(&ctx->refcount, 1);
2226         ctx->flags = flags;                      1884         ctx->flags = flags;
2227         ctx->features = 0;                       1885         ctx->features = 0;
                                                   >> 1886         ctx->state = UFFD_STATE_WAIT_API;
2228         ctx->released = false;                   1887         ctx->released = false;
2229         init_rwsem(&ctx->map_changing_lock);  << 
2230         atomic_set(&ctx->mmap_changing, 0);   << 
2231         ctx->mm = current->mm;                   1888         ctx->mm = current->mm;
2232                                               << 
2233         fd = get_unused_fd_flags(flags & UFFD << 
2234         if (fd < 0)                           << 
2235                 goto err_out;                 << 
2236                                               << 
2237         /* Create a new inode so that the LSM << 
2238         file = anon_inode_create_getfile("[us << 
2239                         O_RDONLY | (flags & U << 
2240         if (IS_ERR(file)) {                   << 
2241                 put_unused_fd(fd);            << 
2242                 fd = PTR_ERR(file);           << 
2243                 goto err_out;                 << 
2244         }                                     << 
2245         /* prevent the mm struct to be freed     1889         /* prevent the mm struct to be freed */
2246         mmgrab(ctx->mm);                         1890         mmgrab(ctx->mm);
2247         file->f_mode |= FMODE_NOWAIT;         << 
2248         fd_install(fd, file);                 << 
2249         return fd;                            << 
2250 err_out:                                      << 
2251         kmem_cache_free(userfaultfd_ctx_cache << 
2252         return fd;                            << 
2253 }                                             << 
2254                                               << 
2255 static inline bool userfaultfd_syscall_allowe << 
2256 {                                             << 
2257         /* Userspace-only page faults are alw << 
2258         if (flags & UFFD_USER_MODE_ONLY)      << 
2259                 return true;                  << 
2260                                                  1891 
2261         /*                                    !! 1892         file = anon_inode_getfile("[userfaultfd]", &userfaultfd_fops, ctx,
2262          * The user is requesting a userfault !! 1893                                   O_RDWR | (flags & UFFD_SHARED_FCNTL_FLAGS));
2263          * Privileged users are always allowe !! 1894         if (IS_ERR(file)) {
2264          */                                   !! 1895                 mmdrop(ctx->mm);
2265         if (capable(CAP_SYS_PTRACE))          !! 1896                 kmem_cache_free(userfaultfd_ctx_cachep, ctx);
2266                 return true;                  !! 1897         }
2267                                               !! 1898 out:
2268         /* Otherwise, access to kernel fault  !! 1899         return file;
2269         return sysctl_unprivileged_userfaultf << 
2270 }                                                1900 }
2271                                                  1901 
2272 SYSCALL_DEFINE1(userfaultfd, int, flags)         1902 SYSCALL_DEFINE1(userfaultfd, int, flags)
2273 {                                                1903 {
2274         if (!userfaultfd_syscall_allowed(flag !! 1904         int fd, error;
2275                 return -EPERM;                !! 1905         struct file *file;
2276                                                  1906 
2277         return new_userfaultfd(flags);        !! 1907         error = get_unused_fd_flags(flags & UFFD_SHARED_FCNTL_FLAGS);
2278 }                                             !! 1908         if (error < 0)
                                                   >> 1909                 return error;
                                                   >> 1910         fd = error;
2279                                                  1911 
2280 static long userfaultfd_dev_ioctl(struct file !! 1912         file = userfaultfd_file_create(flags);
2281 {                                             !! 1913         if (IS_ERR(file)) {
2282         if (cmd != USERFAULTFD_IOC_NEW)       !! 1914                 error = PTR_ERR(file);
2283                 return -EINVAL;               !! 1915                 goto err_put_unused_fd;
                                                   >> 1916         }
                                                   >> 1917         fd_install(fd, file);
2284                                                  1918 
2285         return new_userfaultfd(flags);        !! 1919         return fd;
2286 }                                             << 
2287                                                  1920 
2288 static const struct file_operations userfault !! 1921 err_put_unused_fd:
2289         .unlocked_ioctl = userfaultfd_dev_ioc !! 1922         put_unused_fd(fd);
2290         .compat_ioctl = userfaultfd_dev_ioctl << 
2291         .owner = THIS_MODULE,                 << 
2292         .llseek = noop_llseek,                << 
2293 };                                            << 
2294                                                  1923 
2295 static struct miscdevice userfaultfd_misc = { !! 1924         return error;
2296         .minor = MISC_DYNAMIC_MINOR,          !! 1925 }
2297         .name = "userfaultfd",                << 
2298         .fops = &userfaultfd_dev_fops         << 
2299 };                                            << 
2300                                                  1926 
2301 static int __init userfaultfd_init(void)         1927 static int __init userfaultfd_init(void)
2302 {                                                1928 {
2303         int ret;                              << 
2304                                               << 
2305         ret = misc_register(&userfaultfd_misc << 
2306         if (ret)                              << 
2307                 return ret;                   << 
2308                                               << 
2309         userfaultfd_ctx_cachep = kmem_cache_c    1929         userfaultfd_ctx_cachep = kmem_cache_create("userfaultfd_ctx_cache",
2310                                                  1930                                                 sizeof(struct userfaultfd_ctx),
2311                                                  1931                                                 0,
2312                                                  1932                                                 SLAB_HWCACHE_ALIGN|SLAB_PANIC,
2313                                                  1933                                                 init_once_userfaultfd_ctx);
2314 #ifdef CONFIG_SYSCTL                          << 
2315         register_sysctl_init("vm", vm_userfau << 
2316 #endif                                        << 
2317         return 0;                                1934         return 0;
2318 }                                                1935 }
2319 __initcall(userfaultfd_init);                    1936 __initcall(userfaultfd_init);
2320                                                  1937 

~ [ source navigation ] ~ [ diff markup ] ~ [ identifier search ] ~

kernel.org | git.kernel.org | LWN.net | Project Home | SVN repository | Mail admin

Linux® is a registered trademark of Linus Torvalds in the United States and other countries.
TOMOYO® is a registered trademark of NTT DATA CORPORATION.

sflogo.php