1457c8996SThomas Gleixner // SPDX-License-Identifier: GPL-2.0-only 21da177e4SLinus Torvalds /* 31da177e4SLinus Torvalds * linux/mm/memory.c 41da177e4SLinus Torvalds * 51da177e4SLinus Torvalds * Copyright (C) 1991, 1992, 1993, 1994 Linus Torvalds 61da177e4SLinus Torvalds */ 71da177e4SLinus Torvalds 81da177e4SLinus Torvalds /* 91da177e4SLinus Torvalds * demand-loading started 01.12.91 - seems it is high on the list of 101da177e4SLinus Torvalds * things wanted, and it should be easy to implement. - Linus 111da177e4SLinus Torvalds */ 121da177e4SLinus Torvalds 131da177e4SLinus Torvalds /* 141da177e4SLinus Torvalds * Ok, demand-loading was easy, shared pages a little bit tricker. Shared 151da177e4SLinus Torvalds * pages started 02.12.91, seems to work. - Linus. 161da177e4SLinus Torvalds * 171da177e4SLinus Torvalds * Tested sharing by executing about 30 /bin/sh: under the old kernel it 181da177e4SLinus Torvalds * would have taken more than the 6M I have free, but it worked well as 191da177e4SLinus Torvalds * far as I could see. 201da177e4SLinus Torvalds * 211da177e4SLinus Torvalds * Also corrected some "invalidate()"s - I wasn't doing enough of them. 221da177e4SLinus Torvalds */ 231da177e4SLinus Torvalds 241da177e4SLinus Torvalds /* 251da177e4SLinus Torvalds * Real VM (paging to/from disk) started 18.12.91. Much more work and 261da177e4SLinus Torvalds * thought has to go into this. Oh, well.. 271da177e4SLinus Torvalds * 19.12.91 - works, somewhat. Sometimes I get faults, don't know why. 281da177e4SLinus Torvalds * Found it. Everything seems to work now. 291da177e4SLinus Torvalds * 20.12.91 - Ok, making the swap-device changeable like the root. 301da177e4SLinus Torvalds */ 311da177e4SLinus Torvalds 321da177e4SLinus Torvalds /* 331da177e4SLinus Torvalds * 05.04.94 - Multi-page memory management added for v1.1. 341da177e4SLinus Torvalds * Idea by Alex Bligh (alex@cconcepts.co.uk) 351da177e4SLinus Torvalds * 361da177e4SLinus Torvalds * 16.07.99 - Support of BIGMEM added by Gerhard Wichert, Siemens AG 371da177e4SLinus Torvalds * (Gerhard.Wichert@pdb.siemens.de) 381da177e4SLinus Torvalds * 391da177e4SLinus Torvalds * Aug/Sep 2004 Changed to four level page tables (Andi Kleen) 401da177e4SLinus Torvalds */ 411da177e4SLinus Torvalds 421da177e4SLinus Torvalds #include <linux/kernel_stat.h> 431da177e4SLinus Torvalds #include <linux/mm.h> 4436090defSArnd Bergmann #include <linux/mm_inline.h> 456e84f315SIngo Molnar #include <linux/sched/mm.h> 46f7ccbae4SIngo Molnar #include <linux/sched/coredump.h> 476a3827d7SIngo Molnar #include <linux/sched/numa_balancing.h> 4829930025SIngo Molnar #include <linux/sched/task.h> 491da177e4SLinus Torvalds #include <linux/hugetlb.h> 501da177e4SLinus Torvalds #include <linux/mman.h> 511da177e4SLinus Torvalds #include <linux/swap.h> 521da177e4SLinus Torvalds #include <linux/highmem.h> 531da177e4SLinus Torvalds #include <linux/pagemap.h> 545042db43SJérôme Glisse #include <linux/memremap.h> 559a840895SHugh Dickins #include <linux/ksm.h> 561da177e4SLinus Torvalds #include <linux/rmap.h> 57b95f1b31SPaul Gortmaker #include <linux/export.h> 580ff92245SShailabh Nagar #include <linux/delayacct.h> 591da177e4SLinus Torvalds #include <linux/init.h> 6001c8f1c4SDan Williams #include <linux/pfn_t.h> 61edc79b2aSPeter Zijlstra #include <linux/writeback.h> 628a9f3ccdSBalbir Singh #include <linux/memcontrol.h> 63cddb8a5cSAndrea Arcangeli #include <linux/mmu_notifier.h> 643dc14741SHugh Dickins #include <linux/swapops.h> 653dc14741SHugh Dickins #include <linux/elf.h> 665a0e3ad6STejun Heo #include <linux/gfp.h> 674daae3b4SMel Gorman #include <linux/migrate.h> 682fbc57c5SAndy Shevchenko #include <linux/string.h> 691592eef0SKirill A. Shutemov #include <linux/debugfs.h> 706b251fc9SAndrea Arcangeli #include <linux/userfaultfd_k.h> 71bc2466e4SJan Kara #include <linux/dax.h> 726b31d595SMichal Hocko #include <linux/oom.h> 7398fa15f3SAnshuman Khandual #include <linux/numa.h> 74bce617edSPeter Xu #include <linux/perf_event.h> 75bce617edSPeter Xu #include <linux/ptrace.h> 76e80d3909SJoerg Roedel #include <linux/vmalloc.h> 771da177e4SLinus Torvalds 78b3d1411bSJoel Fernandes (Google) #include <trace/events/kmem.h> 79b3d1411bSJoel Fernandes (Google) 806952b61dSAlexey Dobriyan #include <asm/io.h> 8133a709b2SDave Hansen #include <asm/mmu_context.h> 821da177e4SLinus Torvalds #include <asm/pgalloc.h> 837c0f6ba6SLinus Torvalds #include <linux/uaccess.h> 841da177e4SLinus Torvalds #include <asm/tlb.h> 851da177e4SLinus Torvalds #include <asm/tlbflush.h> 861da177e4SLinus Torvalds 87e80d3909SJoerg Roedel #include "pgalloc-track.h" 8842b77728SJan Beulich #include "internal.h" 89014bb1deSNeilBrown #include "swap.h" 9042b77728SJan Beulich 91af27d940SArnd Bergmann #if defined(LAST_CPUPID_NOT_IN_PAGE_FLAGS) && !defined(CONFIG_COMPILE_TEST) 9290572890SPeter Zijlstra #warning Unfortunate NUMA and NUMA Balancing config, growing page-frame for last_cpupid. 9375980e97SPeter Zijlstra #endif 9475980e97SPeter Zijlstra 95a9ee6cf5SMike Rapoport #ifndef CONFIG_NUMA 961da177e4SLinus Torvalds unsigned long max_mapnr; 971da177e4SLinus Torvalds EXPORT_SYMBOL(max_mapnr); 98166f61b9STobin C Harding 99166f61b9STobin C Harding struct page *mem_map; 1001da177e4SLinus Torvalds EXPORT_SYMBOL(mem_map); 1011da177e4SLinus Torvalds #endif 1021da177e4SLinus Torvalds 1035c041f5dSPeter Xu static vm_fault_t do_fault(struct vm_fault *vmf); 1045c041f5dSPeter Xu 1051da177e4SLinus Torvalds /* 1061da177e4SLinus Torvalds * A number of key systems in x86 including ioremap() rely on the assumption 1071da177e4SLinus Torvalds * that high_memory defines the upper bound on direct map memory, then end 1081da177e4SLinus Torvalds * of ZONE_NORMAL. Under CONFIG_DISCONTIG this means that max_low_pfn and 1091da177e4SLinus Torvalds * highstart_pfn must be the same; there must be no gap between ZONE_NORMAL 1101da177e4SLinus Torvalds * and ZONE_HIGHMEM. 1111da177e4SLinus Torvalds */ 1121da177e4SLinus Torvalds void *high_memory; 1131da177e4SLinus Torvalds EXPORT_SYMBOL(high_memory); 1141da177e4SLinus Torvalds 11532a93233SIngo Molnar /* 11632a93233SIngo Molnar * Randomize the address space (stacks, mmaps, brk, etc.). 11732a93233SIngo Molnar * 11832a93233SIngo Molnar * ( When CONFIG_COMPAT_BRK=y we exclude brk from randomization, 11932a93233SIngo Molnar * as ancient (libc5 based) binaries can segfault. ) 12032a93233SIngo Molnar */ 12132a93233SIngo Molnar int randomize_va_space __read_mostly = 12232a93233SIngo Molnar #ifdef CONFIG_COMPAT_BRK 12332a93233SIngo Molnar 1; 12432a93233SIngo Molnar #else 12532a93233SIngo Molnar 2; 12632a93233SIngo Molnar #endif 127a62eaf15SAndi Kleen 12883d116c5SJia He #ifndef arch_faults_on_old_pte 12983d116c5SJia He static inline bool arch_faults_on_old_pte(void) 13083d116c5SJia He { 13183d116c5SJia He /* 13283d116c5SJia He * Those arches which don't have hw access flag feature need to 13383d116c5SJia He * implement their own helper. By default, "true" means pagefault 13483d116c5SJia He * will be hit on old pte. 13583d116c5SJia He */ 13683d116c5SJia He return true; 13783d116c5SJia He } 13883d116c5SJia He #endif 13983d116c5SJia He 14046bdb427SWill Deacon #ifndef arch_wants_old_prefaulted_pte 14146bdb427SWill Deacon static inline bool arch_wants_old_prefaulted_pte(void) 14246bdb427SWill Deacon { 14346bdb427SWill Deacon /* 14446bdb427SWill Deacon * Transitioning a PTE from 'old' to 'young' can be expensive on 14546bdb427SWill Deacon * some architectures, even if it's performed in hardware. By 14646bdb427SWill Deacon * default, "false" means prefaulted entries will be 'young'. 14746bdb427SWill Deacon */ 14846bdb427SWill Deacon return false; 14946bdb427SWill Deacon } 15046bdb427SWill Deacon #endif 15146bdb427SWill Deacon 152a62eaf15SAndi Kleen static int __init disable_randmaps(char *s) 153a62eaf15SAndi Kleen { 154a62eaf15SAndi Kleen randomize_va_space = 0; 1559b41046cSOGAWA Hirofumi return 1; 156a62eaf15SAndi Kleen } 157a62eaf15SAndi Kleen __setup("norandmaps", disable_randmaps); 158a62eaf15SAndi Kleen 15962eede62SHugh Dickins unsigned long zero_pfn __read_mostly; 1600b70068eSArd Biesheuvel EXPORT_SYMBOL(zero_pfn); 1610b70068eSArd Biesheuvel 162166f61b9STobin C Harding unsigned long highest_memmap_pfn __read_mostly; 163166f61b9STobin C Harding 164a13ea5b7SHugh Dickins /* 165a13ea5b7SHugh Dickins * CONFIG_MMU architectures set up ZERO_PAGE in their paging_init() 166a13ea5b7SHugh Dickins */ 167a13ea5b7SHugh Dickins static int __init init_zero_pfn(void) 168a13ea5b7SHugh Dickins { 169a13ea5b7SHugh Dickins zero_pfn = page_to_pfn(ZERO_PAGE(0)); 170a13ea5b7SHugh Dickins return 0; 171a13ea5b7SHugh Dickins } 172e720e7d0SIlya Lipnitskiy early_initcall(init_zero_pfn); 173a62eaf15SAndi Kleen 174e4dcad20SJoel Fernandes (Google) void mm_trace_rss_stat(struct mm_struct *mm, int member, long count) 175b3d1411bSJoel Fernandes (Google) { 176e4dcad20SJoel Fernandes (Google) trace_rss_stat(mm, member, count); 177b3d1411bSJoel Fernandes (Google) } 178d559db08SKAMEZAWA Hiroyuki 17934e55232SKAMEZAWA Hiroyuki #if defined(SPLIT_RSS_COUNTING) 18034e55232SKAMEZAWA Hiroyuki 181ea48cf78SDavid Rientjes void sync_mm_rss(struct mm_struct *mm) 18234e55232SKAMEZAWA Hiroyuki { 18334e55232SKAMEZAWA Hiroyuki int i; 18434e55232SKAMEZAWA Hiroyuki 18534e55232SKAMEZAWA Hiroyuki for (i = 0; i < NR_MM_COUNTERS; i++) { 18605af2e10SDavid Rientjes if (current->rss_stat.count[i]) { 18705af2e10SDavid Rientjes add_mm_counter(mm, i, current->rss_stat.count[i]); 18805af2e10SDavid Rientjes current->rss_stat.count[i] = 0; 18934e55232SKAMEZAWA Hiroyuki } 19034e55232SKAMEZAWA Hiroyuki } 19105af2e10SDavid Rientjes current->rss_stat.events = 0; 19234e55232SKAMEZAWA Hiroyuki } 19334e55232SKAMEZAWA Hiroyuki 19434e55232SKAMEZAWA Hiroyuki static void add_mm_counter_fast(struct mm_struct *mm, int member, int val) 19534e55232SKAMEZAWA Hiroyuki { 19634e55232SKAMEZAWA Hiroyuki struct task_struct *task = current; 19734e55232SKAMEZAWA Hiroyuki 19834e55232SKAMEZAWA Hiroyuki if (likely(task->mm == mm)) 19934e55232SKAMEZAWA Hiroyuki task->rss_stat.count[member] += val; 20034e55232SKAMEZAWA Hiroyuki else 20134e55232SKAMEZAWA Hiroyuki add_mm_counter(mm, member, val); 20234e55232SKAMEZAWA Hiroyuki } 20334e55232SKAMEZAWA Hiroyuki #define inc_mm_counter_fast(mm, member) add_mm_counter_fast(mm, member, 1) 20434e55232SKAMEZAWA Hiroyuki #define dec_mm_counter_fast(mm, member) add_mm_counter_fast(mm, member, -1) 20534e55232SKAMEZAWA Hiroyuki 20634e55232SKAMEZAWA Hiroyuki /* sync counter once per 64 page faults */ 20734e55232SKAMEZAWA Hiroyuki #define TASK_RSS_EVENTS_THRESH (64) 20834e55232SKAMEZAWA Hiroyuki static void check_sync_rss_stat(struct task_struct *task) 20934e55232SKAMEZAWA Hiroyuki { 21034e55232SKAMEZAWA Hiroyuki if (unlikely(task != current)) 21134e55232SKAMEZAWA Hiroyuki return; 21234e55232SKAMEZAWA Hiroyuki if (unlikely(task->rss_stat.events++ > TASK_RSS_EVENTS_THRESH)) 213ea48cf78SDavid Rientjes sync_mm_rss(task->mm); 21434e55232SKAMEZAWA Hiroyuki } 2159547d01bSPeter Zijlstra #else /* SPLIT_RSS_COUNTING */ 21634e55232SKAMEZAWA Hiroyuki 21734e55232SKAMEZAWA Hiroyuki #define inc_mm_counter_fast(mm, member) inc_mm_counter(mm, member) 21834e55232SKAMEZAWA Hiroyuki #define dec_mm_counter_fast(mm, member) dec_mm_counter(mm, member) 21934e55232SKAMEZAWA Hiroyuki 22034e55232SKAMEZAWA Hiroyuki static void check_sync_rss_stat(struct task_struct *task) 22134e55232SKAMEZAWA Hiroyuki { 22234e55232SKAMEZAWA Hiroyuki } 22334e55232SKAMEZAWA Hiroyuki 2249547d01bSPeter Zijlstra #endif /* SPLIT_RSS_COUNTING */ 2259547d01bSPeter Zijlstra 2261da177e4SLinus Torvalds /* 2271da177e4SLinus Torvalds * Note: this doesn't free the actual pages themselves. That 2281da177e4SLinus Torvalds * has been handled earlier when unmapping all the memory regions. 2291da177e4SLinus Torvalds */ 2309e1b32caSBenjamin Herrenschmidt static void free_pte_range(struct mmu_gather *tlb, pmd_t *pmd, 2319e1b32caSBenjamin Herrenschmidt unsigned long addr) 2321da177e4SLinus Torvalds { 2332f569afdSMartin Schwidefsky pgtable_t token = pmd_pgtable(*pmd); 2341da177e4SLinus Torvalds pmd_clear(pmd); 2359e1b32caSBenjamin Herrenschmidt pte_free_tlb(tlb, token, addr); 236c4812909SKirill A. Shutemov mm_dec_nr_ptes(tlb->mm); 2371da177e4SLinus Torvalds } 2381da177e4SLinus Torvalds 239e0da382cSHugh Dickins static inline void free_pmd_range(struct mmu_gather *tlb, pud_t *pud, 240e0da382cSHugh Dickins unsigned long addr, unsigned long end, 241e0da382cSHugh Dickins unsigned long floor, unsigned long ceiling) 2421da177e4SLinus Torvalds { 2431da177e4SLinus Torvalds pmd_t *pmd; 2441da177e4SLinus Torvalds unsigned long next; 245e0da382cSHugh Dickins unsigned long start; 2461da177e4SLinus Torvalds 247e0da382cSHugh Dickins start = addr; 2481da177e4SLinus Torvalds pmd = pmd_offset(pud, addr); 2491da177e4SLinus Torvalds do { 2501da177e4SLinus Torvalds next = pmd_addr_end(addr, end); 2511da177e4SLinus Torvalds if (pmd_none_or_clear_bad(pmd)) 2521da177e4SLinus Torvalds continue; 2539e1b32caSBenjamin Herrenschmidt free_pte_range(tlb, pmd, addr); 2541da177e4SLinus Torvalds } while (pmd++, addr = next, addr != end); 2551da177e4SLinus Torvalds 256e0da382cSHugh Dickins start &= PUD_MASK; 257e0da382cSHugh Dickins if (start < floor) 258e0da382cSHugh Dickins return; 259e0da382cSHugh Dickins if (ceiling) { 260e0da382cSHugh Dickins ceiling &= PUD_MASK; 261e0da382cSHugh Dickins if (!ceiling) 262e0da382cSHugh Dickins return; 2631da177e4SLinus Torvalds } 264e0da382cSHugh Dickins if (end - 1 > ceiling - 1) 265e0da382cSHugh Dickins return; 266e0da382cSHugh Dickins 267e0da382cSHugh Dickins pmd = pmd_offset(pud, start); 268e0da382cSHugh Dickins pud_clear(pud); 2699e1b32caSBenjamin Herrenschmidt pmd_free_tlb(tlb, pmd, start); 270dc6c9a35SKirill A. Shutemov mm_dec_nr_pmds(tlb->mm); 2711da177e4SLinus Torvalds } 2721da177e4SLinus Torvalds 273c2febafcSKirill A. Shutemov static inline void free_pud_range(struct mmu_gather *tlb, p4d_t *p4d, 274e0da382cSHugh Dickins unsigned long addr, unsigned long end, 275e0da382cSHugh Dickins unsigned long floor, unsigned long ceiling) 2761da177e4SLinus Torvalds { 2771da177e4SLinus Torvalds pud_t *pud; 2781da177e4SLinus Torvalds unsigned long next; 279e0da382cSHugh Dickins unsigned long start; 2801da177e4SLinus Torvalds 281e0da382cSHugh Dickins start = addr; 282c2febafcSKirill A. Shutemov pud = pud_offset(p4d, addr); 2831da177e4SLinus Torvalds do { 2841da177e4SLinus Torvalds next = pud_addr_end(addr, end); 2851da177e4SLinus Torvalds if (pud_none_or_clear_bad(pud)) 2861da177e4SLinus Torvalds continue; 287e0da382cSHugh Dickins free_pmd_range(tlb, pud, addr, next, floor, ceiling); 2881da177e4SLinus Torvalds } while (pud++, addr = next, addr != end); 2891da177e4SLinus Torvalds 290c2febafcSKirill A. Shutemov start &= P4D_MASK; 291c2febafcSKirill A. Shutemov if (start < floor) 292c2febafcSKirill A. Shutemov return; 293c2febafcSKirill A. Shutemov if (ceiling) { 294c2febafcSKirill A. Shutemov ceiling &= P4D_MASK; 295c2febafcSKirill A. Shutemov if (!ceiling) 296c2febafcSKirill A. Shutemov return; 297c2febafcSKirill A. Shutemov } 298c2febafcSKirill A. Shutemov if (end - 1 > ceiling - 1) 299c2febafcSKirill A. Shutemov return; 300c2febafcSKirill A. Shutemov 301c2febafcSKirill A. Shutemov pud = pud_offset(p4d, start); 302c2febafcSKirill A. Shutemov p4d_clear(p4d); 303c2febafcSKirill A. Shutemov pud_free_tlb(tlb, pud, start); 304b4e98d9aSKirill A. Shutemov mm_dec_nr_puds(tlb->mm); 305c2febafcSKirill A. Shutemov } 306c2febafcSKirill A. Shutemov 307c2febafcSKirill A. Shutemov static inline void free_p4d_range(struct mmu_gather *tlb, pgd_t *pgd, 308c2febafcSKirill A. Shutemov unsigned long addr, unsigned long end, 309c2febafcSKirill A. Shutemov unsigned long floor, unsigned long ceiling) 310c2febafcSKirill A. Shutemov { 311c2febafcSKirill A. Shutemov p4d_t *p4d; 312c2febafcSKirill A. Shutemov unsigned long next; 313c2febafcSKirill A. Shutemov unsigned long start; 314c2febafcSKirill A. Shutemov 315c2febafcSKirill A. Shutemov start = addr; 316c2febafcSKirill A. Shutemov p4d = p4d_offset(pgd, addr); 317c2febafcSKirill A. Shutemov do { 318c2febafcSKirill A. Shutemov next = p4d_addr_end(addr, end); 319c2febafcSKirill A. Shutemov if (p4d_none_or_clear_bad(p4d)) 320c2febafcSKirill A. Shutemov continue; 321c2febafcSKirill A. Shutemov free_pud_range(tlb, p4d, addr, next, floor, ceiling); 322c2febafcSKirill A. Shutemov } while (p4d++, addr = next, addr != end); 323c2febafcSKirill A. Shutemov 324e0da382cSHugh Dickins start &= PGDIR_MASK; 325e0da382cSHugh Dickins if (start < floor) 326e0da382cSHugh Dickins return; 327e0da382cSHugh Dickins if (ceiling) { 328e0da382cSHugh Dickins ceiling &= PGDIR_MASK; 329e0da382cSHugh Dickins if (!ceiling) 330e0da382cSHugh Dickins return; 3311da177e4SLinus Torvalds } 332e0da382cSHugh Dickins if (end - 1 > ceiling - 1) 333e0da382cSHugh Dickins return; 334e0da382cSHugh Dickins 335c2febafcSKirill A. Shutemov p4d = p4d_offset(pgd, start); 336e0da382cSHugh Dickins pgd_clear(pgd); 337c2febafcSKirill A. Shutemov p4d_free_tlb(tlb, p4d, start); 3381da177e4SLinus Torvalds } 3391da177e4SLinus Torvalds 3401da177e4SLinus Torvalds /* 341e0da382cSHugh Dickins * This function frees user-level page tables of a process. 3421da177e4SLinus Torvalds */ 34342b77728SJan Beulich void free_pgd_range(struct mmu_gather *tlb, 344e0da382cSHugh Dickins unsigned long addr, unsigned long end, 345e0da382cSHugh Dickins unsigned long floor, unsigned long ceiling) 3461da177e4SLinus Torvalds { 3471da177e4SLinus Torvalds pgd_t *pgd; 3481da177e4SLinus Torvalds unsigned long next; 3491da177e4SLinus Torvalds 350e0da382cSHugh Dickins /* 351e0da382cSHugh Dickins * The next few lines have given us lots of grief... 352e0da382cSHugh Dickins * 353e0da382cSHugh Dickins * Why are we testing PMD* at this top level? Because often 354e0da382cSHugh Dickins * there will be no work to do at all, and we'd prefer not to 355e0da382cSHugh Dickins * go all the way down to the bottom just to discover that. 356e0da382cSHugh Dickins * 357e0da382cSHugh Dickins * Why all these "- 1"s? Because 0 represents both the bottom 358e0da382cSHugh Dickins * of the address space and the top of it (using -1 for the 359e0da382cSHugh Dickins * top wouldn't help much: the masks would do the wrong thing). 360e0da382cSHugh Dickins * The rule is that addr 0 and floor 0 refer to the bottom of 361e0da382cSHugh Dickins * the address space, but end 0 and ceiling 0 refer to the top 362e0da382cSHugh Dickins * Comparisons need to use "end - 1" and "ceiling - 1" (though 363e0da382cSHugh Dickins * that end 0 case should be mythical). 364e0da382cSHugh Dickins * 365e0da382cSHugh Dickins * Wherever addr is brought up or ceiling brought down, we must 366e0da382cSHugh Dickins * be careful to reject "the opposite 0" before it confuses the 367e0da382cSHugh Dickins * subsequent tests. But what about where end is brought down 368e0da382cSHugh Dickins * by PMD_SIZE below? no, end can't go down to 0 there. 369e0da382cSHugh Dickins * 370e0da382cSHugh Dickins * Whereas we round start (addr) and ceiling down, by different 371e0da382cSHugh Dickins * masks at different levels, in order to test whether a table 372e0da382cSHugh Dickins * now has no other vmas using it, so can be freed, we don't 373e0da382cSHugh Dickins * bother to round floor or end up - the tests don't need that. 374e0da382cSHugh Dickins */ 375e0da382cSHugh Dickins 376e0da382cSHugh Dickins addr &= PMD_MASK; 377e0da382cSHugh Dickins if (addr < floor) { 378e0da382cSHugh Dickins addr += PMD_SIZE; 379e0da382cSHugh Dickins if (!addr) 380e0da382cSHugh Dickins return; 381e0da382cSHugh Dickins } 382e0da382cSHugh Dickins if (ceiling) { 383e0da382cSHugh Dickins ceiling &= PMD_MASK; 384e0da382cSHugh Dickins if (!ceiling) 385e0da382cSHugh Dickins return; 386e0da382cSHugh Dickins } 387e0da382cSHugh Dickins if (end - 1 > ceiling - 1) 388e0da382cSHugh Dickins end -= PMD_SIZE; 389e0da382cSHugh Dickins if (addr > end - 1) 390e0da382cSHugh Dickins return; 39107e32661SAneesh Kumar K.V /* 39207e32661SAneesh Kumar K.V * We add page table cache pages with PAGE_SIZE, 39307e32661SAneesh Kumar K.V * (see pte_free_tlb()), flush the tlb if we need 39407e32661SAneesh Kumar K.V */ 395ed6a7935SPeter Zijlstra tlb_change_page_size(tlb, PAGE_SIZE); 39642b77728SJan Beulich pgd = pgd_offset(tlb->mm, addr); 3971da177e4SLinus Torvalds do { 3981da177e4SLinus Torvalds next = pgd_addr_end(addr, end); 3991da177e4SLinus Torvalds if (pgd_none_or_clear_bad(pgd)) 4001da177e4SLinus Torvalds continue; 401c2febafcSKirill A. Shutemov free_p4d_range(tlb, pgd, addr, next, floor, ceiling); 4021da177e4SLinus Torvalds } while (pgd++, addr = next, addr != end); 403e0da382cSHugh Dickins } 404e0da382cSHugh Dickins 40542b77728SJan Beulich void free_pgtables(struct mmu_gather *tlb, struct vm_area_struct *vma, 406e0da382cSHugh Dickins unsigned long floor, unsigned long ceiling) 407e0da382cSHugh Dickins { 408e0da382cSHugh Dickins while (vma) { 409e0da382cSHugh Dickins struct vm_area_struct *next = vma->vm_next; 410e0da382cSHugh Dickins unsigned long addr = vma->vm_start; 411e0da382cSHugh Dickins 4128f4f8c16SHugh Dickins /* 41325d9e2d1Snpiggin@suse.de * Hide vma from rmap and truncate_pagecache before freeing 41425d9e2d1Snpiggin@suse.de * pgtables 4158f4f8c16SHugh Dickins */ 4165beb4930SRik van Riel unlink_anon_vmas(vma); 4178f4f8c16SHugh Dickins unlink_file_vma(vma); 4188f4f8c16SHugh Dickins 4199da61aefSDavid Gibson if (is_vm_hugetlb_page(vma)) { 4203bf5ee95SHugh Dickins hugetlb_free_pgd_range(tlb, addr, vma->vm_end, 4213bf5ee95SHugh Dickins floor, next ? next->vm_start : ceiling); 4223bf5ee95SHugh Dickins } else { 4233bf5ee95SHugh Dickins /* 4243bf5ee95SHugh Dickins * Optimization: gather nearby vmas into one call down 4253bf5ee95SHugh Dickins */ 4263bf5ee95SHugh Dickins while (next && next->vm_start <= vma->vm_end + PMD_SIZE 4274866920bSDavid Gibson && !is_vm_hugetlb_page(next)) { 428e0da382cSHugh Dickins vma = next; 429e0da382cSHugh Dickins next = vma->vm_next; 4305beb4930SRik van Riel unlink_anon_vmas(vma); 4318f4f8c16SHugh Dickins unlink_file_vma(vma); 432e0da382cSHugh Dickins } 4333bf5ee95SHugh Dickins free_pgd_range(tlb, addr, vma->vm_end, 434e0da382cSHugh Dickins floor, next ? next->vm_start : ceiling); 4353bf5ee95SHugh Dickins } 436e0da382cSHugh Dickins vma = next; 437e0da382cSHugh Dickins } 4381da177e4SLinus Torvalds } 4391da177e4SLinus Torvalds 44003c4f204SQi Zheng void pmd_install(struct mm_struct *mm, pmd_t *pmd, pgtable_t *pte) 4411da177e4SLinus Torvalds { 44203c4f204SQi Zheng spinlock_t *ptl = pmd_lock(mm, pmd); 4431bb3630eSHugh Dickins 44403c4f204SQi Zheng if (likely(pmd_none(*pmd))) { /* Has another populated it ? */ 44503c4f204SQi Zheng mm_inc_nr_ptes(mm); 446362a61adSNick Piggin /* 447362a61adSNick Piggin * Ensure all pte setup (eg. pte page lock and page clearing) are 448362a61adSNick Piggin * visible before the pte is made visible to other CPUs by being 449362a61adSNick Piggin * put into page tables. 450362a61adSNick Piggin * 451362a61adSNick Piggin * The other side of the story is the pointer chasing in the page 452362a61adSNick Piggin * table walking code (when walking the page table without locking; 453362a61adSNick Piggin * ie. most of the time). Fortunately, these data accesses consist 454362a61adSNick Piggin * of a chain of data-dependent loads, meaning most CPUs (alpha 455362a61adSNick Piggin * being the notable exception) will already guarantee loads are 456362a61adSNick Piggin * seen in-order. See the alpha page table accessors for the 457bb7cdd38SWill Deacon * smp_rmb() barriers in page table walking code. 458362a61adSNick Piggin */ 459362a61adSNick Piggin smp_wmb(); /* Could be smp_wmb__xxx(before|after)_spin_lock */ 460ed33b5a6SQi Zheng pmd_populate(mm, pmd, *pte); 461ed33b5a6SQi Zheng *pte = NULL; 4624b471e88SKirill A. Shutemov } 463c4088ebdSKirill A. Shutemov spin_unlock(ptl); 464ed33b5a6SQi Zheng } 465ed33b5a6SQi Zheng 466ed33b5a6SQi Zheng int __pte_alloc(struct mm_struct *mm, pmd_t *pmd) 467ed33b5a6SQi Zheng { 468ed33b5a6SQi Zheng pgtable_t new = pte_alloc_one(mm); 469ed33b5a6SQi Zheng if (!new) 470ed33b5a6SQi Zheng return -ENOMEM; 4718ac1f832SAndrea Arcangeli 47203c4f204SQi Zheng pmd_install(mm, pmd, &new); 4732f569afdSMartin Schwidefsky if (new) 4742f569afdSMartin Schwidefsky pte_free(mm, new); 4751bb3630eSHugh Dickins return 0; 4761da177e4SLinus Torvalds } 4771da177e4SLinus Torvalds 4784cf58924SJoel Fernandes (Google) int __pte_alloc_kernel(pmd_t *pmd) 4791da177e4SLinus Torvalds { 4804cf58924SJoel Fernandes (Google) pte_t *new = pte_alloc_one_kernel(&init_mm); 4811da177e4SLinus Torvalds if (!new) 4821bb3630eSHugh Dickins return -ENOMEM; 4831da177e4SLinus Torvalds 484872fec16SHugh Dickins spin_lock(&init_mm.page_table_lock); 4858ac1f832SAndrea Arcangeli if (likely(pmd_none(*pmd))) { /* Has another populated it ? */ 486ed33b5a6SQi Zheng smp_wmb(); /* See comment in pmd_install() */ 487872fec16SHugh Dickins pmd_populate_kernel(&init_mm, pmd, new); 4882f569afdSMartin Schwidefsky new = NULL; 4894b471e88SKirill A. Shutemov } 490872fec16SHugh Dickins spin_unlock(&init_mm.page_table_lock); 4912f569afdSMartin Schwidefsky if (new) 4922f569afdSMartin Schwidefsky pte_free_kernel(&init_mm, new); 4931bb3630eSHugh Dickins return 0; 4941da177e4SLinus Torvalds } 4951da177e4SLinus Torvalds 496d559db08SKAMEZAWA Hiroyuki static inline void init_rss_vec(int *rss) 497ae859762SHugh Dickins { 498d559db08SKAMEZAWA Hiroyuki memset(rss, 0, sizeof(int) * NR_MM_COUNTERS); 499d559db08SKAMEZAWA Hiroyuki } 500d559db08SKAMEZAWA Hiroyuki 501d559db08SKAMEZAWA Hiroyuki static inline void add_mm_rss_vec(struct mm_struct *mm, int *rss) 502d559db08SKAMEZAWA Hiroyuki { 503d559db08SKAMEZAWA Hiroyuki int i; 504d559db08SKAMEZAWA Hiroyuki 50534e55232SKAMEZAWA Hiroyuki if (current->mm == mm) 50605af2e10SDavid Rientjes sync_mm_rss(mm); 507d559db08SKAMEZAWA Hiroyuki for (i = 0; i < NR_MM_COUNTERS; i++) 508d559db08SKAMEZAWA Hiroyuki if (rss[i]) 509d559db08SKAMEZAWA Hiroyuki add_mm_counter(mm, i, rss[i]); 510ae859762SHugh Dickins } 511ae859762SHugh Dickins 5121da177e4SLinus Torvalds /* 5136aab341eSLinus Torvalds * This function is called to print an error when a bad pte 5146aab341eSLinus Torvalds * is found. For example, we might have a PFN-mapped pte in 5156aab341eSLinus Torvalds * a region that doesn't allow it. 516b5810039SNick Piggin * 517b5810039SNick Piggin * The calling function must still handle the error. 518b5810039SNick Piggin */ 5193dc14741SHugh Dickins static void print_bad_pte(struct vm_area_struct *vma, unsigned long addr, 5203dc14741SHugh Dickins pte_t pte, struct page *page) 521b5810039SNick Piggin { 5223dc14741SHugh Dickins pgd_t *pgd = pgd_offset(vma->vm_mm, addr); 523c2febafcSKirill A. Shutemov p4d_t *p4d = p4d_offset(pgd, addr); 524c2febafcSKirill A. Shutemov pud_t *pud = pud_offset(p4d, addr); 5253dc14741SHugh Dickins pmd_t *pmd = pmd_offset(pud, addr); 5263dc14741SHugh Dickins struct address_space *mapping; 5273dc14741SHugh Dickins pgoff_t index; 528d936cf9bSHugh Dickins static unsigned long resume; 529d936cf9bSHugh Dickins static unsigned long nr_shown; 530d936cf9bSHugh Dickins static unsigned long nr_unshown; 531d936cf9bSHugh Dickins 532d936cf9bSHugh Dickins /* 533d936cf9bSHugh Dickins * Allow a burst of 60 reports, then keep quiet for that minute; 534d936cf9bSHugh Dickins * or allow a steady drip of one report per second. 535d936cf9bSHugh Dickins */ 536d936cf9bSHugh Dickins if (nr_shown == 60) { 537d936cf9bSHugh Dickins if (time_before(jiffies, resume)) { 538d936cf9bSHugh Dickins nr_unshown++; 539d936cf9bSHugh Dickins return; 540d936cf9bSHugh Dickins } 541d936cf9bSHugh Dickins if (nr_unshown) { 5421170532bSJoe Perches pr_alert("BUG: Bad page map: %lu messages suppressed\n", 543d936cf9bSHugh Dickins nr_unshown); 544d936cf9bSHugh Dickins nr_unshown = 0; 545d936cf9bSHugh Dickins } 546d936cf9bSHugh Dickins nr_shown = 0; 547d936cf9bSHugh Dickins } 548d936cf9bSHugh Dickins if (nr_shown++ == 0) 549d936cf9bSHugh Dickins resume = jiffies + 60 * HZ; 5503dc14741SHugh Dickins 5513dc14741SHugh Dickins mapping = vma->vm_file ? vma->vm_file->f_mapping : NULL; 5523dc14741SHugh Dickins index = linear_page_index(vma, addr); 5533dc14741SHugh Dickins 5541170532bSJoe Perches pr_alert("BUG: Bad page map in process %s pte:%08llx pmd:%08llx\n", 5553dc14741SHugh Dickins current->comm, 5563dc14741SHugh Dickins (long long)pte_val(pte), (long long)pmd_val(*pmd)); 557718a3821SWu Fengguang if (page) 558f0b791a3SDave Hansen dump_page(page, "bad pte"); 5596aa9b8b2SKefeng Wang pr_alert("addr:%px vm_flags:%08lx anon_vma:%px mapping:%px index:%lx\n", 5603dc14741SHugh Dickins (void *)addr, vma->vm_flags, vma->anon_vma, mapping, index); 5617e0a1265SMatthew Wilcox (Oracle) pr_alert("file:%pD fault:%ps mmap:%ps read_folio:%ps\n", 5622682582aSKonstantin Khlebnikov vma->vm_file, 5632682582aSKonstantin Khlebnikov vma->vm_ops ? vma->vm_ops->fault : NULL, 5642682582aSKonstantin Khlebnikov vma->vm_file ? vma->vm_file->f_op->mmap : NULL, 5657e0a1265SMatthew Wilcox (Oracle) mapping ? mapping->a_ops->read_folio : NULL); 566b5810039SNick Piggin dump_stack(); 567373d4d09SRusty Russell add_taint(TAINT_BAD_PAGE, LOCKDEP_NOW_UNRELIABLE); 568b5810039SNick Piggin } 569b5810039SNick Piggin 570b5810039SNick Piggin /* 5717e675137SNick Piggin * vm_normal_page -- This function gets the "struct page" associated with a pte. 5726aab341eSLinus Torvalds * 5737e675137SNick Piggin * "Special" mappings do not wish to be associated with a "struct page" (either 5747e675137SNick Piggin * it doesn't exist, or it exists but they don't want to touch it). In this 5757e675137SNick Piggin * case, NULL is returned here. "Normal" mappings do have a struct page. 576b379d790SJared Hulbert * 5777e675137SNick Piggin * There are 2 broad cases. Firstly, an architecture may define a pte_special() 5787e675137SNick Piggin * pte bit, in which case this function is trivial. Secondly, an architecture 5797e675137SNick Piggin * may not have a spare pte bit, which requires a more complicated scheme, 5807e675137SNick Piggin * described below. 5817e675137SNick Piggin * 5827e675137SNick Piggin * A raw VM_PFNMAP mapping (ie. one that is not COWed) is always considered a 5837e675137SNick Piggin * special mapping (even if there are underlying and valid "struct pages"). 5847e675137SNick Piggin * COWed pages of a VM_PFNMAP are always normal. 5856aab341eSLinus Torvalds * 586b379d790SJared Hulbert * The way we recognize COWed pages within VM_PFNMAP mappings is through the 587b379d790SJared Hulbert * rules set up by "remap_pfn_range()": the vma will have the VM_PFNMAP bit 5887e675137SNick Piggin * set, and the vm_pgoff will point to the first PFN mapped: thus every special 5897e675137SNick Piggin * mapping will always honor the rule 5906aab341eSLinus Torvalds * 5916aab341eSLinus Torvalds * pfn_of_page == vma->vm_pgoff + ((addr - vma->vm_start) >> PAGE_SHIFT) 5926aab341eSLinus Torvalds * 5937e675137SNick Piggin * And for normal mappings this is false. 594b379d790SJared Hulbert * 5957e675137SNick Piggin * This restricts such mappings to be a linear translation from virtual address 5967e675137SNick Piggin * to pfn. To get around this restriction, we allow arbitrary mappings so long 5977e675137SNick Piggin * as the vma is not a COW mapping; in that case, we know that all ptes are 5987e675137SNick Piggin * special (because none can have been COWed). 599b379d790SJared Hulbert * 600b379d790SJared Hulbert * 6017e675137SNick Piggin * In order to support COW of arbitrary special mappings, we have VM_MIXEDMAP. 6027e675137SNick Piggin * 603b379d790SJared Hulbert * VM_MIXEDMAP mappings can likewise contain memory with or without "struct 604b379d790SJared Hulbert * page" backing, however the difference is that _all_ pages with a struct 605b379d790SJared Hulbert * page (that is, those where pfn_valid is true) are refcounted and considered 606b379d790SJared Hulbert * normal pages by the VM. The disadvantage is that pages are refcounted 607b379d790SJared Hulbert * (which can be slower and simply not an option for some PFNMAP users). The 608b379d790SJared Hulbert * advantage is that we don't have to follow the strict linearity rule of 609b379d790SJared Hulbert * PFNMAP mappings in order to support COWable mappings. 610b379d790SJared Hulbert * 611ee498ed7SHugh Dickins */ 61225b2995aSChristoph Hellwig struct page *vm_normal_page(struct vm_area_struct *vma, unsigned long addr, 61325b2995aSChristoph Hellwig pte_t pte) 614ee498ed7SHugh Dickins { 61522b31eecSHugh Dickins unsigned long pfn = pte_pfn(pte); 6167e675137SNick Piggin 61700b3a331SLaurent Dufour if (IS_ENABLED(CONFIG_ARCH_HAS_PTE_SPECIAL)) { 618b38af472SHugh Dickins if (likely(!pte_special(pte))) 61922b31eecSHugh Dickins goto check_pfn; 620667a0a06SDavid Vrabel if (vma->vm_ops && vma->vm_ops->find_special_page) 621667a0a06SDavid Vrabel return vma->vm_ops->find_special_page(vma, addr); 622a13ea5b7SHugh Dickins if (vma->vm_flags & (VM_PFNMAP | VM_MIXEDMAP)) 623a13ea5b7SHugh Dickins return NULL; 624df6ad698SJérôme Glisse if (is_zero_pfn(pfn)) 625df6ad698SJérôme Glisse return NULL; 626e1fb4a08SDave Jiang if (pte_devmap(pte)) 627*3218f871SAlex Sierra /* 628*3218f871SAlex Sierra * NOTE: New users of ZONE_DEVICE will not set pte_devmap() 629*3218f871SAlex Sierra * and will have refcounts incremented on their struct pages 630*3218f871SAlex Sierra * when they are inserted into PTEs, thus they are safe to 631*3218f871SAlex Sierra * return here. Legacy ZONE_DEVICE pages that set pte_devmap() 632*3218f871SAlex Sierra * do not have refcounts. Example of legacy ZONE_DEVICE is 633*3218f871SAlex Sierra * MEMORY_DEVICE_FS_DAX type in pmem or virtio_fs drivers. 634*3218f871SAlex Sierra */ 635e1fb4a08SDave Jiang return NULL; 636e1fb4a08SDave Jiang 63722b31eecSHugh Dickins print_bad_pte(vma, addr, pte, NULL); 6387e675137SNick Piggin return NULL; 6397e675137SNick Piggin } 6407e675137SNick Piggin 64100b3a331SLaurent Dufour /* !CONFIG_ARCH_HAS_PTE_SPECIAL case follows: */ 6427e675137SNick Piggin 643b379d790SJared Hulbert if (unlikely(vma->vm_flags & (VM_PFNMAP|VM_MIXEDMAP))) { 644b379d790SJared Hulbert if (vma->vm_flags & VM_MIXEDMAP) { 645b379d790SJared Hulbert if (!pfn_valid(pfn)) 646b379d790SJared Hulbert return NULL; 647b379d790SJared Hulbert goto out; 648b379d790SJared Hulbert } else { 6497e675137SNick Piggin unsigned long off; 6507e675137SNick Piggin off = (addr - vma->vm_start) >> PAGE_SHIFT; 6516aab341eSLinus Torvalds if (pfn == vma->vm_pgoff + off) 6526aab341eSLinus Torvalds return NULL; 65367121172SLinus Torvalds if (!is_cow_mapping(vma->vm_flags)) 654fb155c16SLinus Torvalds return NULL; 6556aab341eSLinus Torvalds } 656b379d790SJared Hulbert } 6576aab341eSLinus Torvalds 658b38af472SHugh Dickins if (is_zero_pfn(pfn)) 659b38af472SHugh Dickins return NULL; 66000b3a331SLaurent Dufour 66122b31eecSHugh Dickins check_pfn: 66222b31eecSHugh Dickins if (unlikely(pfn > highest_memmap_pfn)) { 66322b31eecSHugh Dickins print_bad_pte(vma, addr, pte, NULL); 66422b31eecSHugh Dickins return NULL; 66522b31eecSHugh Dickins } 6666aab341eSLinus Torvalds 6676aab341eSLinus Torvalds /* 6687e675137SNick Piggin * NOTE! We still have PageReserved() pages in the page tables. 6697e675137SNick Piggin * eg. VDSO mappings can cause them to exist. 6706aab341eSLinus Torvalds */ 671b379d790SJared Hulbert out: 6726aab341eSLinus Torvalds return pfn_to_page(pfn); 673ee498ed7SHugh Dickins } 674ee498ed7SHugh Dickins 67528093f9fSGerald Schaefer #ifdef CONFIG_TRANSPARENT_HUGEPAGE 67628093f9fSGerald Schaefer struct page *vm_normal_page_pmd(struct vm_area_struct *vma, unsigned long addr, 67728093f9fSGerald Schaefer pmd_t pmd) 67828093f9fSGerald Schaefer { 67928093f9fSGerald Schaefer unsigned long pfn = pmd_pfn(pmd); 68028093f9fSGerald Schaefer 68128093f9fSGerald Schaefer /* 68228093f9fSGerald Schaefer * There is no pmd_special() but there may be special pmds, e.g. 68328093f9fSGerald Schaefer * in a direct-access (dax) mapping, so let's just replicate the 68400b3a331SLaurent Dufour * !CONFIG_ARCH_HAS_PTE_SPECIAL case from vm_normal_page() here. 68528093f9fSGerald Schaefer */ 68628093f9fSGerald Schaefer if (unlikely(vma->vm_flags & (VM_PFNMAP|VM_MIXEDMAP))) { 68728093f9fSGerald Schaefer if (vma->vm_flags & VM_MIXEDMAP) { 68828093f9fSGerald Schaefer if (!pfn_valid(pfn)) 68928093f9fSGerald Schaefer return NULL; 69028093f9fSGerald Schaefer goto out; 69128093f9fSGerald Schaefer } else { 69228093f9fSGerald Schaefer unsigned long off; 69328093f9fSGerald Schaefer off = (addr - vma->vm_start) >> PAGE_SHIFT; 69428093f9fSGerald Schaefer if (pfn == vma->vm_pgoff + off) 69528093f9fSGerald Schaefer return NULL; 69628093f9fSGerald Schaefer if (!is_cow_mapping(vma->vm_flags)) 69728093f9fSGerald Schaefer return NULL; 69828093f9fSGerald Schaefer } 69928093f9fSGerald Schaefer } 70028093f9fSGerald Schaefer 701e1fb4a08SDave Jiang if (pmd_devmap(pmd)) 702e1fb4a08SDave Jiang return NULL; 7033cde287bSYu Zhao if (is_huge_zero_pmd(pmd)) 70428093f9fSGerald Schaefer return NULL; 70528093f9fSGerald Schaefer if (unlikely(pfn > highest_memmap_pfn)) 70628093f9fSGerald Schaefer return NULL; 70728093f9fSGerald Schaefer 70828093f9fSGerald Schaefer /* 70928093f9fSGerald Schaefer * NOTE! We still have PageReserved() pages in the page tables. 71028093f9fSGerald Schaefer * eg. VDSO mappings can cause them to exist. 71128093f9fSGerald Schaefer */ 71228093f9fSGerald Schaefer out: 71328093f9fSGerald Schaefer return pfn_to_page(pfn); 71428093f9fSGerald Schaefer } 71528093f9fSGerald Schaefer #endif 71628093f9fSGerald Schaefer 717b756a3b5SAlistair Popple static void restore_exclusive_pte(struct vm_area_struct *vma, 718b756a3b5SAlistair Popple struct page *page, unsigned long address, 719b756a3b5SAlistair Popple pte_t *ptep) 720b756a3b5SAlistair Popple { 721b756a3b5SAlistair Popple pte_t pte; 722b756a3b5SAlistair Popple swp_entry_t entry; 723b756a3b5SAlistair Popple 724b756a3b5SAlistair Popple pte = pte_mkold(mk_pte(page, READ_ONCE(vma->vm_page_prot))); 725b756a3b5SAlistair Popple if (pte_swp_soft_dirty(*ptep)) 726b756a3b5SAlistair Popple pte = pte_mksoft_dirty(pte); 727b756a3b5SAlistair Popple 728b756a3b5SAlistair Popple entry = pte_to_swp_entry(*ptep); 729b756a3b5SAlistair Popple if (pte_swp_uffd_wp(*ptep)) 730b756a3b5SAlistair Popple pte = pte_mkuffd_wp(pte); 731b756a3b5SAlistair Popple else if (is_writable_device_exclusive_entry(entry)) 732b756a3b5SAlistair Popple pte = maybe_mkwrite(pte_mkdirty(pte), vma); 733b756a3b5SAlistair Popple 7346c287605SDavid Hildenbrand VM_BUG_ON(pte_write(pte) && !(PageAnon(page) && PageAnonExclusive(page))); 7356c287605SDavid Hildenbrand 736b756a3b5SAlistair Popple /* 737b756a3b5SAlistair Popple * No need to take a page reference as one was already 738b756a3b5SAlistair Popple * created when the swap entry was made. 739b756a3b5SAlistair Popple */ 740b756a3b5SAlistair Popple if (PageAnon(page)) 741f1e2db12SDavid Hildenbrand page_add_anon_rmap(page, vma, address, RMAP_NONE); 742b756a3b5SAlistair Popple else 743b756a3b5SAlistair Popple /* 744b756a3b5SAlistair Popple * Currently device exclusive access only supports anonymous 745b756a3b5SAlistair Popple * memory so the entry shouldn't point to a filebacked page. 746b756a3b5SAlistair Popple */ 747b756a3b5SAlistair Popple WARN_ON_ONCE(!PageAnon(page)); 748b756a3b5SAlistair Popple 7491eba86c0SPasha Tatashin set_pte_at(vma->vm_mm, address, ptep, pte); 7501eba86c0SPasha Tatashin 751b756a3b5SAlistair Popple /* 752b756a3b5SAlistair Popple * No need to invalidate - it was non-present before. However 753b756a3b5SAlistair Popple * secondary CPUs may have mappings that need invalidating. 754b756a3b5SAlistair Popple */ 755b756a3b5SAlistair Popple update_mmu_cache(vma, address, ptep); 756b756a3b5SAlistair Popple } 757b756a3b5SAlistair Popple 758b756a3b5SAlistair Popple /* 759b756a3b5SAlistair Popple * Tries to restore an exclusive pte if the page lock can be acquired without 760b756a3b5SAlistair Popple * sleeping. 761b756a3b5SAlistair Popple */ 762b756a3b5SAlistair Popple static int 763b756a3b5SAlistair Popple try_restore_exclusive_pte(pte_t *src_pte, struct vm_area_struct *vma, 764b756a3b5SAlistair Popple unsigned long addr) 765b756a3b5SAlistair Popple { 766b756a3b5SAlistair Popple swp_entry_t entry = pte_to_swp_entry(*src_pte); 767b756a3b5SAlistair Popple struct page *page = pfn_swap_entry_to_page(entry); 768b756a3b5SAlistair Popple 769b756a3b5SAlistair Popple if (trylock_page(page)) { 770b756a3b5SAlistair Popple restore_exclusive_pte(vma, page, addr, src_pte); 771b756a3b5SAlistair Popple unlock_page(page); 772b756a3b5SAlistair Popple return 0; 773b756a3b5SAlistair Popple } 774b756a3b5SAlistair Popple 775b756a3b5SAlistair Popple return -EBUSY; 776b756a3b5SAlistair Popple } 777b756a3b5SAlistair Popple 778ee498ed7SHugh Dickins /* 7791da177e4SLinus Torvalds * copy one vm_area from one task to the other. Assumes the page tables 7801da177e4SLinus Torvalds * already present in the new task to be cleared in the whole range 7811da177e4SLinus Torvalds * covered by this vma. 7821da177e4SLinus Torvalds */ 7831da177e4SLinus Torvalds 784df3a57d1SLinus Torvalds static unsigned long 785df3a57d1SLinus Torvalds copy_nonpresent_pte(struct mm_struct *dst_mm, struct mm_struct *src_mm, 7868f34f1eaSPeter Xu pte_t *dst_pte, pte_t *src_pte, struct vm_area_struct *dst_vma, 7878f34f1eaSPeter Xu struct vm_area_struct *src_vma, unsigned long addr, int *rss) 7881da177e4SLinus Torvalds { 7898f34f1eaSPeter Xu unsigned long vm_flags = dst_vma->vm_flags; 7901da177e4SLinus Torvalds pte_t pte = *src_pte; 7911da177e4SLinus Torvalds struct page *page; 7920697212aSChristoph Lameter swp_entry_t entry = pte_to_swp_entry(pte); 7930697212aSChristoph Lameter 7942022b4d1SHugh Dickins if (likely(!non_swap_entry(entry))) { 795570a335bSHugh Dickins if (swap_duplicate(entry) < 0) 7969a5cc85cSAlistair Popple return -EIO; 797570a335bSHugh Dickins 7981da177e4SLinus Torvalds /* make sure dst_mm is on swapoff's mmlist. */ 7991da177e4SLinus Torvalds if (unlikely(list_empty(&dst_mm->mmlist))) { 8001da177e4SLinus Torvalds spin_lock(&mmlist_lock); 801f412ac08SHugh Dickins if (list_empty(&dst_mm->mmlist)) 802f412ac08SHugh Dickins list_add(&dst_mm->mmlist, 803f412ac08SHugh Dickins &src_mm->mmlist); 8041da177e4SLinus Torvalds spin_unlock(&mmlist_lock); 8051da177e4SLinus Torvalds } 8061493a191SDavid Hildenbrand /* Mark the swap entry as shared. */ 8071493a191SDavid Hildenbrand if (pte_swp_exclusive(*src_pte)) { 8081493a191SDavid Hildenbrand pte = pte_swp_clear_exclusive(*src_pte); 8091493a191SDavid Hildenbrand set_pte_at(src_mm, addr, src_pte, pte); 8101493a191SDavid Hildenbrand } 811b084d435SKAMEZAWA Hiroyuki rss[MM_SWAPENTS]++; 8122022b4d1SHugh Dickins } else if (is_migration_entry(entry)) { 813af5cdaf8SAlistair Popple page = pfn_swap_entry_to_page(entry); 8149f9f1acdSKonstantin Khlebnikov 815eca56ff9SJerome Marchand rss[mm_counter(page)]++; 8169f9f1acdSKonstantin Khlebnikov 8176c287605SDavid Hildenbrand if (!is_readable_migration_entry(entry) && 8180697212aSChristoph Lameter is_cow_mapping(vm_flags)) { 8190697212aSChristoph Lameter /* 8206c287605SDavid Hildenbrand * COW mappings require pages in both parent and child 8216c287605SDavid Hildenbrand * to be set to read. A previously exclusive entry is 8226c287605SDavid Hildenbrand * now shared. 8230697212aSChristoph Lameter */ 8244dd845b5SAlistair Popple entry = make_readable_migration_entry( 8254dd845b5SAlistair Popple swp_offset(entry)); 8260697212aSChristoph Lameter pte = swp_entry_to_pte(entry); 827c3d16e16SCyrill Gorcunov if (pte_swp_soft_dirty(*src_pte)) 828c3d16e16SCyrill Gorcunov pte = pte_swp_mksoft_dirty(pte); 829f45ec5ffSPeter Xu if (pte_swp_uffd_wp(*src_pte)) 830f45ec5ffSPeter Xu pte = pte_swp_mkuffd_wp(pte); 8310697212aSChristoph Lameter set_pte_at(src_mm, addr, src_pte, pte); 8320697212aSChristoph Lameter } 8335042db43SJérôme Glisse } else if (is_device_private_entry(entry)) { 834af5cdaf8SAlistair Popple page = pfn_swap_entry_to_page(entry); 8355042db43SJérôme Glisse 8365042db43SJérôme Glisse /* 8375042db43SJérôme Glisse * Update rss count even for unaddressable pages, as 8385042db43SJérôme Glisse * they should treated just like normal pages in this 8395042db43SJérôme Glisse * respect. 8405042db43SJérôme Glisse * 8415042db43SJérôme Glisse * We will likely want to have some new rss counters 8425042db43SJérôme Glisse * for unaddressable pages, at some point. But for now 8435042db43SJérôme Glisse * keep things as they are. 8445042db43SJérôme Glisse */ 8455042db43SJérôme Glisse get_page(page); 8465042db43SJérôme Glisse rss[mm_counter(page)]++; 847fb3d824dSDavid Hildenbrand /* Cannot fail as these pages cannot get pinned. */ 848fb3d824dSDavid Hildenbrand BUG_ON(page_try_dup_anon_rmap(page, false, src_vma)); 8495042db43SJérôme Glisse 8505042db43SJérôme Glisse /* 8515042db43SJérôme Glisse * We do not preserve soft-dirty information, because so 8525042db43SJérôme Glisse * far, checkpoint/restore is the only feature that 8535042db43SJérôme Glisse * requires that. And checkpoint/restore does not work 8545042db43SJérôme Glisse * when a device driver is involved (you cannot easily 8555042db43SJérôme Glisse * save and restore device driver state). 8565042db43SJérôme Glisse */ 8574dd845b5SAlistair Popple if (is_writable_device_private_entry(entry) && 8585042db43SJérôme Glisse is_cow_mapping(vm_flags)) { 8594dd845b5SAlistair Popple entry = make_readable_device_private_entry( 8604dd845b5SAlistair Popple swp_offset(entry)); 8615042db43SJérôme Glisse pte = swp_entry_to_pte(entry); 862f45ec5ffSPeter Xu if (pte_swp_uffd_wp(*src_pte)) 863f45ec5ffSPeter Xu pte = pte_swp_mkuffd_wp(pte); 8645042db43SJérôme Glisse set_pte_at(src_mm, addr, src_pte, pte); 8655042db43SJérôme Glisse } 866b756a3b5SAlistair Popple } else if (is_device_exclusive_entry(entry)) { 867b756a3b5SAlistair Popple /* 868b756a3b5SAlistair Popple * Make device exclusive entries present by restoring the 869b756a3b5SAlistair Popple * original entry then copying as for a present pte. Device 870b756a3b5SAlistair Popple * exclusive entries currently only support private writable 871b756a3b5SAlistair Popple * (ie. COW) mappings. 872b756a3b5SAlistair Popple */ 873b756a3b5SAlistair Popple VM_BUG_ON(!is_cow_mapping(src_vma->vm_flags)); 874b756a3b5SAlistair Popple if (try_restore_exclusive_pte(src_pte, src_vma, addr)) 875b756a3b5SAlistair Popple return -EBUSY; 876b756a3b5SAlistair Popple return -ENOENT; 877c56d1b62SPeter Xu } else if (is_pte_marker_entry(entry)) { 878c56d1b62SPeter Xu /* 879c56d1b62SPeter Xu * We're copying the pgtable should only because dst_vma has 880c56d1b62SPeter Xu * uffd-wp enabled, do sanity check. 881c56d1b62SPeter Xu */ 882c56d1b62SPeter Xu WARN_ON_ONCE(!userfaultfd_wp(dst_vma)); 883c56d1b62SPeter Xu set_pte_at(dst_mm, addr, dst_pte, pte); 884c56d1b62SPeter Xu return 0; 8851da177e4SLinus Torvalds } 8868f34f1eaSPeter Xu if (!userfaultfd_wp(dst_vma)) 8878f34f1eaSPeter Xu pte = pte_swp_clear_uffd_wp(pte); 888df3a57d1SLinus Torvalds set_pte_at(dst_mm, addr, dst_pte, pte); 889df3a57d1SLinus Torvalds return 0; 8901da177e4SLinus Torvalds } 8911da177e4SLinus Torvalds 89270e806e4SPeter Xu /* 893b51ad4f8SDavid Hildenbrand * Copy a present and normal page. 89470e806e4SPeter Xu * 895b51ad4f8SDavid Hildenbrand * NOTE! The usual case is that this isn't required; 896b51ad4f8SDavid Hildenbrand * instead, the caller can just increase the page refcount 897b51ad4f8SDavid Hildenbrand * and re-use the pte the traditional way. 89870e806e4SPeter Xu * 89970e806e4SPeter Xu * And if we need a pre-allocated page but don't yet have 90070e806e4SPeter Xu * one, return a negative error to let the preallocation 90170e806e4SPeter Xu * code know so that it can do so outside the page table 90270e806e4SPeter Xu * lock. 90370e806e4SPeter Xu */ 90470e806e4SPeter Xu static inline int 905c78f4636SPeter Xu copy_present_page(struct vm_area_struct *dst_vma, struct vm_area_struct *src_vma, 906c78f4636SPeter Xu pte_t *dst_pte, pte_t *src_pte, unsigned long addr, int *rss, 907b51ad4f8SDavid Hildenbrand struct page **prealloc, struct page *page) 90870e806e4SPeter Xu { 90970e806e4SPeter Xu struct page *new_page; 910b51ad4f8SDavid Hildenbrand pte_t pte; 91170e806e4SPeter Xu 91270e806e4SPeter Xu new_page = *prealloc; 91370e806e4SPeter Xu if (!new_page) 91470e806e4SPeter Xu return -EAGAIN; 91570e806e4SPeter Xu 91670e806e4SPeter Xu /* 91770e806e4SPeter Xu * We have a prealloc page, all good! Take it 91870e806e4SPeter Xu * over and copy the page & arm it. 91970e806e4SPeter Xu */ 92070e806e4SPeter Xu *prealloc = NULL; 921c78f4636SPeter Xu copy_user_highpage(new_page, page, addr, src_vma); 92270e806e4SPeter Xu __SetPageUptodate(new_page); 92340f2bbf7SDavid Hildenbrand page_add_new_anon_rmap(new_page, dst_vma, addr); 924c78f4636SPeter Xu lru_cache_add_inactive_or_unevictable(new_page, dst_vma); 92570e806e4SPeter Xu rss[mm_counter(new_page)]++; 92670e806e4SPeter Xu 92770e806e4SPeter Xu /* All done, just insert the new page copy in the child */ 928c78f4636SPeter Xu pte = mk_pte(new_page, dst_vma->vm_page_prot); 929c78f4636SPeter Xu pte = maybe_mkwrite(pte_mkdirty(pte), dst_vma); 9308f34f1eaSPeter Xu if (userfaultfd_pte_wp(dst_vma, *src_pte)) 9318f34f1eaSPeter Xu /* Uffd-wp needs to be delivered to dest pte as well */ 9328f34f1eaSPeter Xu pte = pte_wrprotect(pte_mkuffd_wp(pte)); 933c78f4636SPeter Xu set_pte_at(dst_vma->vm_mm, addr, dst_pte, pte); 93470e806e4SPeter Xu return 0; 93570e806e4SPeter Xu } 93670e806e4SPeter Xu 93770e806e4SPeter Xu /* 93870e806e4SPeter Xu * Copy one pte. Returns 0 if succeeded, or -EAGAIN if one preallocated page 93970e806e4SPeter Xu * is required to copy this pte. 94070e806e4SPeter Xu */ 94170e806e4SPeter Xu static inline int 942c78f4636SPeter Xu copy_present_pte(struct vm_area_struct *dst_vma, struct vm_area_struct *src_vma, 943c78f4636SPeter Xu pte_t *dst_pte, pte_t *src_pte, unsigned long addr, int *rss, 944c78f4636SPeter Xu struct page **prealloc) 945df3a57d1SLinus Torvalds { 946c78f4636SPeter Xu struct mm_struct *src_mm = src_vma->vm_mm; 947c78f4636SPeter Xu unsigned long vm_flags = src_vma->vm_flags; 948df3a57d1SLinus Torvalds pte_t pte = *src_pte; 949df3a57d1SLinus Torvalds struct page *page; 950df3a57d1SLinus Torvalds 951c78f4636SPeter Xu page = vm_normal_page(src_vma, addr, pte); 952fb3d824dSDavid Hildenbrand if (page && PageAnon(page)) { 953b51ad4f8SDavid Hildenbrand /* 954b51ad4f8SDavid Hildenbrand * If this page may have been pinned by the parent process, 955b51ad4f8SDavid Hildenbrand * copy the page immediately for the child so that we'll always 956b51ad4f8SDavid Hildenbrand * guarantee the pinned page won't be randomly replaced in the 957b51ad4f8SDavid Hildenbrand * future. 958b51ad4f8SDavid Hildenbrand */ 95970e806e4SPeter Xu get_page(page); 960fb3d824dSDavid Hildenbrand if (unlikely(page_try_dup_anon_rmap(page, false, src_vma))) { 961fb3d824dSDavid Hildenbrand /* Page maybe pinned, we have to copy. */ 962fb3d824dSDavid Hildenbrand put_page(page); 963b51ad4f8SDavid Hildenbrand return copy_present_page(dst_vma, src_vma, dst_pte, src_pte, 964b51ad4f8SDavid Hildenbrand addr, rss, prealloc, page); 965fb3d824dSDavid Hildenbrand } 966fb3d824dSDavid Hildenbrand rss[mm_counter(page)]++; 967b51ad4f8SDavid Hildenbrand } else if (page) { 96870e806e4SPeter Xu get_page(page); 969fb3d824dSDavid Hildenbrand page_dup_file_rmap(page, false); 97070e806e4SPeter Xu rss[mm_counter(page)]++; 97170e806e4SPeter Xu } 97270e806e4SPeter Xu 9731da177e4SLinus Torvalds /* 9741da177e4SLinus Torvalds * If it's a COW mapping, write protect it both 9751da177e4SLinus Torvalds * in the parent and the child 9761da177e4SLinus Torvalds */ 9771b2de5d0SLinus Torvalds if (is_cow_mapping(vm_flags) && pte_write(pte)) { 9781da177e4SLinus Torvalds ptep_set_wrprotect(src_mm, addr, src_pte); 9793dc90795SZachary Amsden pte = pte_wrprotect(pte); 9801da177e4SLinus Torvalds } 9816c287605SDavid Hildenbrand VM_BUG_ON(page && PageAnon(page) && PageAnonExclusive(page)); 9821da177e4SLinus Torvalds 9831da177e4SLinus Torvalds /* 9841da177e4SLinus Torvalds * If it's a shared mapping, mark it clean in 9851da177e4SLinus Torvalds * the child 9861da177e4SLinus Torvalds */ 9871da177e4SLinus Torvalds if (vm_flags & VM_SHARED) 9881da177e4SLinus Torvalds pte = pte_mkclean(pte); 9891da177e4SLinus Torvalds pte = pte_mkold(pte); 9906aab341eSLinus Torvalds 9918f34f1eaSPeter Xu if (!userfaultfd_wp(dst_vma)) 992b569a176SPeter Xu pte = pte_clear_uffd_wp(pte); 993b569a176SPeter Xu 994c78f4636SPeter Xu set_pte_at(dst_vma->vm_mm, addr, dst_pte, pte); 99570e806e4SPeter Xu return 0; 9966aab341eSLinus Torvalds } 997ae859762SHugh Dickins 99870e806e4SPeter Xu static inline struct page * 99970e806e4SPeter Xu page_copy_prealloc(struct mm_struct *src_mm, struct vm_area_struct *vma, 100070e806e4SPeter Xu unsigned long addr) 100170e806e4SPeter Xu { 100270e806e4SPeter Xu struct page *new_page; 100370e806e4SPeter Xu 100470e806e4SPeter Xu new_page = alloc_page_vma(GFP_HIGHUSER_MOVABLE, vma, addr); 100570e806e4SPeter Xu if (!new_page) 100670e806e4SPeter Xu return NULL; 100770e806e4SPeter Xu 10088f425e4eSMatthew Wilcox (Oracle) if (mem_cgroup_charge(page_folio(new_page), src_mm, GFP_KERNEL)) { 100970e806e4SPeter Xu put_page(new_page); 101070e806e4SPeter Xu return NULL; 101170e806e4SPeter Xu } 101270e806e4SPeter Xu cgroup_throttle_swaprate(new_page, GFP_KERNEL); 101370e806e4SPeter Xu 101470e806e4SPeter Xu return new_page; 10151da177e4SLinus Torvalds } 10161da177e4SLinus Torvalds 1017c78f4636SPeter Xu static int 1018c78f4636SPeter Xu copy_pte_range(struct vm_area_struct *dst_vma, struct vm_area_struct *src_vma, 1019c78f4636SPeter Xu pmd_t *dst_pmd, pmd_t *src_pmd, unsigned long addr, 1020c78f4636SPeter Xu unsigned long end) 10211da177e4SLinus Torvalds { 1022c78f4636SPeter Xu struct mm_struct *dst_mm = dst_vma->vm_mm; 1023c78f4636SPeter Xu struct mm_struct *src_mm = src_vma->vm_mm; 1024c36987e2SDaisuke Nishimura pte_t *orig_src_pte, *orig_dst_pte; 10251da177e4SLinus Torvalds pte_t *src_pte, *dst_pte; 1026c74df32cSHugh Dickins spinlock_t *src_ptl, *dst_ptl; 102770e806e4SPeter Xu int progress, ret = 0; 1028d559db08SKAMEZAWA Hiroyuki int rss[NR_MM_COUNTERS]; 1029570a335bSHugh Dickins swp_entry_t entry = (swp_entry_t){0}; 103070e806e4SPeter Xu struct page *prealloc = NULL; 10311da177e4SLinus Torvalds 10321da177e4SLinus Torvalds again: 103370e806e4SPeter Xu progress = 0; 1034d559db08SKAMEZAWA Hiroyuki init_rss_vec(rss); 1035d559db08SKAMEZAWA Hiroyuki 1036c74df32cSHugh Dickins dst_pte = pte_alloc_map_lock(dst_mm, dst_pmd, addr, &dst_ptl); 103770e806e4SPeter Xu if (!dst_pte) { 103870e806e4SPeter Xu ret = -ENOMEM; 103970e806e4SPeter Xu goto out; 104070e806e4SPeter Xu } 1041ece0e2b6SPeter Zijlstra src_pte = pte_offset_map(src_pmd, addr); 10424c21e2f2SHugh Dickins src_ptl = pte_lockptr(src_mm, src_pmd); 1043f20dc5f7SIngo Molnar spin_lock_nested(src_ptl, SINGLE_DEPTH_NESTING); 1044c36987e2SDaisuke Nishimura orig_src_pte = src_pte; 1045c36987e2SDaisuke Nishimura orig_dst_pte = dst_pte; 10466606c3e0SZachary Amsden arch_enter_lazy_mmu_mode(); 10471da177e4SLinus Torvalds 10481da177e4SLinus Torvalds do { 10491da177e4SLinus Torvalds /* 10501da177e4SLinus Torvalds * We are holding two locks at this point - either of them 10511da177e4SLinus Torvalds * could generate latencies in another task on another CPU. 10521da177e4SLinus Torvalds */ 1053e040f218SHugh Dickins if (progress >= 32) { 1054e040f218SHugh Dickins progress = 0; 1055e040f218SHugh Dickins if (need_resched() || 105695c354feSNick Piggin spin_needbreak(src_ptl) || spin_needbreak(dst_ptl)) 10571da177e4SLinus Torvalds break; 1058e040f218SHugh Dickins } 10591da177e4SLinus Torvalds if (pte_none(*src_pte)) { 10601da177e4SLinus Torvalds progress++; 10611da177e4SLinus Torvalds continue; 10621da177e4SLinus Torvalds } 106379a1971cSLinus Torvalds if (unlikely(!pte_present(*src_pte))) { 10649a5cc85cSAlistair Popple ret = copy_nonpresent_pte(dst_mm, src_mm, 106579a1971cSLinus Torvalds dst_pte, src_pte, 10668f34f1eaSPeter Xu dst_vma, src_vma, 10678f34f1eaSPeter Xu addr, rss); 10689a5cc85cSAlistair Popple if (ret == -EIO) { 10699a5cc85cSAlistair Popple entry = pte_to_swp_entry(*src_pte); 1070570a335bSHugh Dickins break; 1071b756a3b5SAlistair Popple } else if (ret == -EBUSY) { 1072b756a3b5SAlistair Popple break; 1073b756a3b5SAlistair Popple } else if (!ret) { 10741da177e4SLinus Torvalds progress += 8; 107579a1971cSLinus Torvalds continue; 107679a1971cSLinus Torvalds } 1077b756a3b5SAlistair Popple 1078b756a3b5SAlistair Popple /* 1079b756a3b5SAlistair Popple * Device exclusive entry restored, continue by copying 1080b756a3b5SAlistair Popple * the now present pte. 1081b756a3b5SAlistair Popple */ 1082b756a3b5SAlistair Popple WARN_ON_ONCE(ret != -ENOENT); 1083b756a3b5SAlistair Popple } 108470e806e4SPeter Xu /* copy_present_pte() will clear `*prealloc' if consumed */ 1085c78f4636SPeter Xu ret = copy_present_pte(dst_vma, src_vma, dst_pte, src_pte, 1086c78f4636SPeter Xu addr, rss, &prealloc); 108770e806e4SPeter Xu /* 108870e806e4SPeter Xu * If we need a pre-allocated page for this pte, drop the 108970e806e4SPeter Xu * locks, allocate, and try again. 109070e806e4SPeter Xu */ 109170e806e4SPeter Xu if (unlikely(ret == -EAGAIN)) 109270e806e4SPeter Xu break; 109370e806e4SPeter Xu if (unlikely(prealloc)) { 109470e806e4SPeter Xu /* 109570e806e4SPeter Xu * pre-alloc page cannot be reused by next time so as 109670e806e4SPeter Xu * to strictly follow mempolicy (e.g., alloc_page_vma() 109770e806e4SPeter Xu * will allocate page according to address). This 109870e806e4SPeter Xu * could only happen if one pinned pte changed. 109970e806e4SPeter Xu */ 110070e806e4SPeter Xu put_page(prealloc); 110170e806e4SPeter Xu prealloc = NULL; 110270e806e4SPeter Xu } 110379a1971cSLinus Torvalds progress += 8; 11041da177e4SLinus Torvalds } while (dst_pte++, src_pte++, addr += PAGE_SIZE, addr != end); 11051da177e4SLinus Torvalds 11066606c3e0SZachary Amsden arch_leave_lazy_mmu_mode(); 1107c74df32cSHugh Dickins spin_unlock(src_ptl); 1108ece0e2b6SPeter Zijlstra pte_unmap(orig_src_pte); 1109d559db08SKAMEZAWA Hiroyuki add_mm_rss_vec(dst_mm, rss); 1110c36987e2SDaisuke Nishimura pte_unmap_unlock(orig_dst_pte, dst_ptl); 1111c74df32cSHugh Dickins cond_resched(); 1112570a335bSHugh Dickins 11139a5cc85cSAlistair Popple if (ret == -EIO) { 11149a5cc85cSAlistair Popple VM_WARN_ON_ONCE(!entry.val); 111570e806e4SPeter Xu if (add_swap_count_continuation(entry, GFP_KERNEL) < 0) { 111670e806e4SPeter Xu ret = -ENOMEM; 111770e806e4SPeter Xu goto out; 111870e806e4SPeter Xu } 111970e806e4SPeter Xu entry.val = 0; 1120b756a3b5SAlistair Popple } else if (ret == -EBUSY) { 1121b756a3b5SAlistair Popple goto out; 11229a5cc85cSAlistair Popple } else if (ret == -EAGAIN) { 1123c78f4636SPeter Xu prealloc = page_copy_prealloc(src_mm, src_vma, addr); 112470e806e4SPeter Xu if (!prealloc) 1125570a335bSHugh Dickins return -ENOMEM; 11269a5cc85cSAlistair Popple } else if (ret) { 11279a5cc85cSAlistair Popple VM_WARN_ON_ONCE(1); 11289a5cc85cSAlistair Popple } 11299a5cc85cSAlistair Popple 113070e806e4SPeter Xu /* We've captured and resolved the error. Reset, try again. */ 113170e806e4SPeter Xu ret = 0; 11329a5cc85cSAlistair Popple 11331da177e4SLinus Torvalds if (addr != end) 11341da177e4SLinus Torvalds goto again; 113570e806e4SPeter Xu out: 113670e806e4SPeter Xu if (unlikely(prealloc)) 113770e806e4SPeter Xu put_page(prealloc); 113870e806e4SPeter Xu return ret; 11391da177e4SLinus Torvalds } 11401da177e4SLinus Torvalds 1141c78f4636SPeter Xu static inline int 1142c78f4636SPeter Xu copy_pmd_range(struct vm_area_struct *dst_vma, struct vm_area_struct *src_vma, 1143c78f4636SPeter Xu pud_t *dst_pud, pud_t *src_pud, unsigned long addr, 1144c78f4636SPeter Xu unsigned long end) 11451da177e4SLinus Torvalds { 1146c78f4636SPeter Xu struct mm_struct *dst_mm = dst_vma->vm_mm; 1147c78f4636SPeter Xu struct mm_struct *src_mm = src_vma->vm_mm; 11481da177e4SLinus Torvalds pmd_t *src_pmd, *dst_pmd; 11491da177e4SLinus Torvalds unsigned long next; 11501da177e4SLinus Torvalds 11511da177e4SLinus Torvalds dst_pmd = pmd_alloc(dst_mm, dst_pud, addr); 11521da177e4SLinus Torvalds if (!dst_pmd) 11531da177e4SLinus Torvalds return -ENOMEM; 11541da177e4SLinus Torvalds src_pmd = pmd_offset(src_pud, addr); 11551da177e4SLinus Torvalds do { 11561da177e4SLinus Torvalds next = pmd_addr_end(addr, end); 115784c3fc4eSZi Yan if (is_swap_pmd(*src_pmd) || pmd_trans_huge(*src_pmd) 115884c3fc4eSZi Yan || pmd_devmap(*src_pmd)) { 115971e3aac0SAndrea Arcangeli int err; 1160c78f4636SPeter Xu VM_BUG_ON_VMA(next-addr != HPAGE_PMD_SIZE, src_vma); 11618f34f1eaSPeter Xu err = copy_huge_pmd(dst_mm, src_mm, dst_pmd, src_pmd, 11628f34f1eaSPeter Xu addr, dst_vma, src_vma); 116371e3aac0SAndrea Arcangeli if (err == -ENOMEM) 116471e3aac0SAndrea Arcangeli return -ENOMEM; 116571e3aac0SAndrea Arcangeli if (!err) 116671e3aac0SAndrea Arcangeli continue; 116771e3aac0SAndrea Arcangeli /* fall through */ 116871e3aac0SAndrea Arcangeli } 11691da177e4SLinus Torvalds if (pmd_none_or_clear_bad(src_pmd)) 11701da177e4SLinus Torvalds continue; 1171c78f4636SPeter Xu if (copy_pte_range(dst_vma, src_vma, dst_pmd, src_pmd, 1172c78f4636SPeter Xu addr, next)) 11731da177e4SLinus Torvalds return -ENOMEM; 11741da177e4SLinus Torvalds } while (dst_pmd++, src_pmd++, addr = next, addr != end); 11751da177e4SLinus Torvalds return 0; 11761da177e4SLinus Torvalds } 11771da177e4SLinus Torvalds 1178c78f4636SPeter Xu static inline int 1179c78f4636SPeter Xu copy_pud_range(struct vm_area_struct *dst_vma, struct vm_area_struct *src_vma, 1180c78f4636SPeter Xu p4d_t *dst_p4d, p4d_t *src_p4d, unsigned long addr, 1181c78f4636SPeter Xu unsigned long end) 11821da177e4SLinus Torvalds { 1183c78f4636SPeter Xu struct mm_struct *dst_mm = dst_vma->vm_mm; 1184c78f4636SPeter Xu struct mm_struct *src_mm = src_vma->vm_mm; 11851da177e4SLinus Torvalds pud_t *src_pud, *dst_pud; 11861da177e4SLinus Torvalds unsigned long next; 11871da177e4SLinus Torvalds 1188c2febafcSKirill A. Shutemov dst_pud = pud_alloc(dst_mm, dst_p4d, addr); 11891da177e4SLinus Torvalds if (!dst_pud) 11901da177e4SLinus Torvalds return -ENOMEM; 1191c2febafcSKirill A. Shutemov src_pud = pud_offset(src_p4d, addr); 11921da177e4SLinus Torvalds do { 11931da177e4SLinus Torvalds next = pud_addr_end(addr, end); 1194a00cc7d9SMatthew Wilcox if (pud_trans_huge(*src_pud) || pud_devmap(*src_pud)) { 1195a00cc7d9SMatthew Wilcox int err; 1196a00cc7d9SMatthew Wilcox 1197c78f4636SPeter Xu VM_BUG_ON_VMA(next-addr != HPAGE_PUD_SIZE, src_vma); 1198a00cc7d9SMatthew Wilcox err = copy_huge_pud(dst_mm, src_mm, 1199c78f4636SPeter Xu dst_pud, src_pud, addr, src_vma); 1200a00cc7d9SMatthew Wilcox if (err == -ENOMEM) 1201a00cc7d9SMatthew Wilcox return -ENOMEM; 1202a00cc7d9SMatthew Wilcox if (!err) 1203a00cc7d9SMatthew Wilcox continue; 1204a00cc7d9SMatthew Wilcox /* fall through */ 1205a00cc7d9SMatthew Wilcox } 12061da177e4SLinus Torvalds if (pud_none_or_clear_bad(src_pud)) 12071da177e4SLinus Torvalds continue; 1208c78f4636SPeter Xu if (copy_pmd_range(dst_vma, src_vma, dst_pud, src_pud, 1209c78f4636SPeter Xu addr, next)) 12101da177e4SLinus Torvalds return -ENOMEM; 12111da177e4SLinus Torvalds } while (dst_pud++, src_pud++, addr = next, addr != end); 12121da177e4SLinus Torvalds return 0; 12131da177e4SLinus Torvalds } 12141da177e4SLinus Torvalds 1215c78f4636SPeter Xu static inline int 1216c78f4636SPeter Xu copy_p4d_range(struct vm_area_struct *dst_vma, struct vm_area_struct *src_vma, 1217c78f4636SPeter Xu pgd_t *dst_pgd, pgd_t *src_pgd, unsigned long addr, 1218c78f4636SPeter Xu unsigned long end) 1219c2febafcSKirill A. Shutemov { 1220c78f4636SPeter Xu struct mm_struct *dst_mm = dst_vma->vm_mm; 1221c2febafcSKirill A. Shutemov p4d_t *src_p4d, *dst_p4d; 1222c2febafcSKirill A. Shutemov unsigned long next; 1223c2febafcSKirill A. Shutemov 1224c2febafcSKirill A. Shutemov dst_p4d = p4d_alloc(dst_mm, dst_pgd, addr); 1225c2febafcSKirill A. Shutemov if (!dst_p4d) 1226c2febafcSKirill A. Shutemov return -ENOMEM; 1227c2febafcSKirill A. Shutemov src_p4d = p4d_offset(src_pgd, addr); 1228c2febafcSKirill A. Shutemov do { 1229c2febafcSKirill A. Shutemov next = p4d_addr_end(addr, end); 1230c2febafcSKirill A. Shutemov if (p4d_none_or_clear_bad(src_p4d)) 1231c2febafcSKirill A. Shutemov continue; 1232c78f4636SPeter Xu if (copy_pud_range(dst_vma, src_vma, dst_p4d, src_p4d, 1233c78f4636SPeter Xu addr, next)) 1234c2febafcSKirill A. Shutemov return -ENOMEM; 1235c2febafcSKirill A. Shutemov } while (dst_p4d++, src_p4d++, addr = next, addr != end); 1236c2febafcSKirill A. Shutemov return 0; 1237c2febafcSKirill A. Shutemov } 1238c2febafcSKirill A. Shutemov 1239c56d1b62SPeter Xu /* 1240c56d1b62SPeter Xu * Return true if the vma needs to copy the pgtable during this fork(). Return 1241c56d1b62SPeter Xu * false when we can speed up fork() by allowing lazy page faults later until 1242c56d1b62SPeter Xu * when the child accesses the memory range. 1243c56d1b62SPeter Xu */ 1244bc70fbf2SPeter Xu static bool 1245c56d1b62SPeter Xu vma_needs_copy(struct vm_area_struct *dst_vma, struct vm_area_struct *src_vma) 1246c56d1b62SPeter Xu { 1247c56d1b62SPeter Xu /* 1248c56d1b62SPeter Xu * Always copy pgtables when dst_vma has uffd-wp enabled even if it's 1249c56d1b62SPeter Xu * file-backed (e.g. shmem). Because when uffd-wp is enabled, pgtable 1250c56d1b62SPeter Xu * contains uffd-wp protection information, that's something we can't 1251c56d1b62SPeter Xu * retrieve from page cache, and skip copying will lose those info. 1252c56d1b62SPeter Xu */ 1253c56d1b62SPeter Xu if (userfaultfd_wp(dst_vma)) 1254c56d1b62SPeter Xu return true; 1255c56d1b62SPeter Xu 1256c56d1b62SPeter Xu if (src_vma->vm_flags & (VM_HUGETLB | VM_PFNMAP | VM_MIXEDMAP)) 1257c56d1b62SPeter Xu return true; 1258c56d1b62SPeter Xu 1259c56d1b62SPeter Xu if (src_vma->anon_vma) 1260c56d1b62SPeter Xu return true; 1261c56d1b62SPeter Xu 1262c56d1b62SPeter Xu /* 1263c56d1b62SPeter Xu * Don't copy ptes where a page fault will fill them correctly. Fork 1264c56d1b62SPeter Xu * becomes much lighter when there are big shared or private readonly 1265c56d1b62SPeter Xu * mappings. The tradeoff is that copy_page_range is more efficient 1266c56d1b62SPeter Xu * than faulting. 1267c56d1b62SPeter Xu */ 1268c56d1b62SPeter Xu return false; 1269c56d1b62SPeter Xu } 1270c56d1b62SPeter Xu 1271c78f4636SPeter Xu int 1272c78f4636SPeter Xu copy_page_range(struct vm_area_struct *dst_vma, struct vm_area_struct *src_vma) 12731da177e4SLinus Torvalds { 12741da177e4SLinus Torvalds pgd_t *src_pgd, *dst_pgd; 12751da177e4SLinus Torvalds unsigned long next; 1276c78f4636SPeter Xu unsigned long addr = src_vma->vm_start; 1277c78f4636SPeter Xu unsigned long end = src_vma->vm_end; 1278c78f4636SPeter Xu struct mm_struct *dst_mm = dst_vma->vm_mm; 1279c78f4636SPeter Xu struct mm_struct *src_mm = src_vma->vm_mm; 1280ac46d4f3SJérôme Glisse struct mmu_notifier_range range; 12812ec74c3eSSagi Grimberg bool is_cow; 1282cddb8a5cSAndrea Arcangeli int ret; 12831da177e4SLinus Torvalds 1284c56d1b62SPeter Xu if (!vma_needs_copy(dst_vma, src_vma)) 1285d992895bSNick Piggin return 0; 1286d992895bSNick Piggin 1287c78f4636SPeter Xu if (is_vm_hugetlb_page(src_vma)) 1288bc70fbf2SPeter Xu return copy_hugetlb_page_range(dst_mm, src_mm, dst_vma, src_vma); 12891da177e4SLinus Torvalds 1290c78f4636SPeter Xu if (unlikely(src_vma->vm_flags & VM_PFNMAP)) { 12912ab64037Svenkatesh.pallipadi@intel.com /* 12922ab64037Svenkatesh.pallipadi@intel.com * We do not free on error cases below as remove_vma 12932ab64037Svenkatesh.pallipadi@intel.com * gets called on error from higher level routine 12942ab64037Svenkatesh.pallipadi@intel.com */ 1295c78f4636SPeter Xu ret = track_pfn_copy(src_vma); 12962ab64037Svenkatesh.pallipadi@intel.com if (ret) 12972ab64037Svenkatesh.pallipadi@intel.com return ret; 12982ab64037Svenkatesh.pallipadi@intel.com } 12992ab64037Svenkatesh.pallipadi@intel.com 1300cddb8a5cSAndrea Arcangeli /* 1301cddb8a5cSAndrea Arcangeli * We need to invalidate the secondary MMU mappings only when 1302cddb8a5cSAndrea Arcangeli * there could be a permission downgrade on the ptes of the 1303cddb8a5cSAndrea Arcangeli * parent mm. And a permission downgrade will only happen if 1304cddb8a5cSAndrea Arcangeli * is_cow_mapping() returns true. 1305cddb8a5cSAndrea Arcangeli */ 1306c78f4636SPeter Xu is_cow = is_cow_mapping(src_vma->vm_flags); 1307ac46d4f3SJérôme Glisse 1308ac46d4f3SJérôme Glisse if (is_cow) { 13097269f999SJérôme Glisse mmu_notifier_range_init(&range, MMU_NOTIFY_PROTECTION_PAGE, 1310c78f4636SPeter Xu 0, src_vma, src_mm, addr, end); 1311ac46d4f3SJérôme Glisse mmu_notifier_invalidate_range_start(&range); 131257efa1feSJason Gunthorpe /* 131357efa1feSJason Gunthorpe * Disabling preemption is not needed for the write side, as 131457efa1feSJason Gunthorpe * the read side doesn't spin, but goes to the mmap_lock. 131557efa1feSJason Gunthorpe * 131657efa1feSJason Gunthorpe * Use the raw variant of the seqcount_t write API to avoid 131757efa1feSJason Gunthorpe * lockdep complaining about preemptibility. 131857efa1feSJason Gunthorpe */ 131957efa1feSJason Gunthorpe mmap_assert_write_locked(src_mm); 132057efa1feSJason Gunthorpe raw_write_seqcount_begin(&src_mm->write_protect_seq); 1321ac46d4f3SJérôme Glisse } 1322cddb8a5cSAndrea Arcangeli 1323cddb8a5cSAndrea Arcangeli ret = 0; 13241da177e4SLinus Torvalds dst_pgd = pgd_offset(dst_mm, addr); 13251da177e4SLinus Torvalds src_pgd = pgd_offset(src_mm, addr); 13261da177e4SLinus Torvalds do { 13271da177e4SLinus Torvalds next = pgd_addr_end(addr, end); 13281da177e4SLinus Torvalds if (pgd_none_or_clear_bad(src_pgd)) 13291da177e4SLinus Torvalds continue; 1330c78f4636SPeter Xu if (unlikely(copy_p4d_range(dst_vma, src_vma, dst_pgd, src_pgd, 1331c78f4636SPeter Xu addr, next))) { 1332cddb8a5cSAndrea Arcangeli ret = -ENOMEM; 1333cddb8a5cSAndrea Arcangeli break; 1334cddb8a5cSAndrea Arcangeli } 13351da177e4SLinus Torvalds } while (dst_pgd++, src_pgd++, addr = next, addr != end); 1336cddb8a5cSAndrea Arcangeli 133757efa1feSJason Gunthorpe if (is_cow) { 133857efa1feSJason Gunthorpe raw_write_seqcount_end(&src_mm->write_protect_seq); 1339ac46d4f3SJérôme Glisse mmu_notifier_invalidate_range_end(&range); 134057efa1feSJason Gunthorpe } 1341cddb8a5cSAndrea Arcangeli return ret; 13421da177e4SLinus Torvalds } 13431da177e4SLinus Torvalds 13443506659eSMatthew Wilcox (Oracle) /* 13453506659eSMatthew Wilcox (Oracle) * Parameter block passed down to zap_pte_range in exceptional cases. 13463506659eSMatthew Wilcox (Oracle) */ 13473506659eSMatthew Wilcox (Oracle) struct zap_details { 13483506659eSMatthew Wilcox (Oracle) struct folio *single_folio; /* Locked folio to be unmapped */ 13492e148f1eSPeter Xu bool even_cows; /* Zap COWed private pages too? */ 1350999dad82SPeter Xu zap_flags_t zap_flags; /* Extra flags for zapping */ 13513506659eSMatthew Wilcox (Oracle) }; 13523506659eSMatthew Wilcox (Oracle) 13535abfd71dSPeter Xu /* Whether we should zap all COWed (private) pages too */ 13545abfd71dSPeter Xu static inline bool should_zap_cows(struct zap_details *details) 13555abfd71dSPeter Xu { 13565abfd71dSPeter Xu /* By default, zap all pages */ 13575abfd71dSPeter Xu if (!details) 13585abfd71dSPeter Xu return true; 13595abfd71dSPeter Xu 13605abfd71dSPeter Xu /* Or, we zap COWed pages only if the caller wants to */ 13612e148f1eSPeter Xu return details->even_cows; 13625abfd71dSPeter Xu } 13635abfd71dSPeter Xu 13642e148f1eSPeter Xu /* Decides whether we should zap this page with the page pointer specified */ 1365254ab940SPeter Xu static inline bool should_zap_page(struct zap_details *details, struct page *page) 13663506659eSMatthew Wilcox (Oracle) { 13675abfd71dSPeter Xu /* If we can make a decision without *page.. */ 13685abfd71dSPeter Xu if (should_zap_cows(details)) 1369254ab940SPeter Xu return true; 13703506659eSMatthew Wilcox (Oracle) 13715abfd71dSPeter Xu /* E.g. the caller passes NULL for the case of a zero page */ 13725abfd71dSPeter Xu if (!page) 1373254ab940SPeter Xu return true; 13745abfd71dSPeter Xu 13752e148f1eSPeter Xu /* Otherwise we should only zap non-anon pages */ 13762e148f1eSPeter Xu return !PageAnon(page); 13773506659eSMatthew Wilcox (Oracle) } 13783506659eSMatthew Wilcox (Oracle) 1379999dad82SPeter Xu static inline bool zap_drop_file_uffd_wp(struct zap_details *details) 1380999dad82SPeter Xu { 1381999dad82SPeter Xu if (!details) 1382999dad82SPeter Xu return false; 1383999dad82SPeter Xu 1384999dad82SPeter Xu return details->zap_flags & ZAP_FLAG_DROP_MARKER; 1385999dad82SPeter Xu } 1386999dad82SPeter Xu 1387999dad82SPeter Xu /* 1388999dad82SPeter Xu * This function makes sure that we'll replace the none pte with an uffd-wp 1389999dad82SPeter Xu * swap special pte marker when necessary. Must be with the pgtable lock held. 1390999dad82SPeter Xu */ 1391999dad82SPeter Xu static inline void 1392999dad82SPeter Xu zap_install_uffd_wp_if_needed(struct vm_area_struct *vma, 1393999dad82SPeter Xu unsigned long addr, pte_t *pte, 1394999dad82SPeter Xu struct zap_details *details, pte_t pteval) 1395999dad82SPeter Xu { 1396999dad82SPeter Xu if (zap_drop_file_uffd_wp(details)) 1397999dad82SPeter Xu return; 1398999dad82SPeter Xu 1399999dad82SPeter Xu pte_install_uffd_wp_if_needed(vma, addr, pte, pteval); 1400999dad82SPeter Xu } 1401999dad82SPeter Xu 140251c6f666SRobin Holt static unsigned long zap_pte_range(struct mmu_gather *tlb, 1403b5810039SNick Piggin struct vm_area_struct *vma, pmd_t *pmd, 14041da177e4SLinus Torvalds unsigned long addr, unsigned long end, 140597a89413SPeter Zijlstra struct zap_details *details) 14061da177e4SLinus Torvalds { 1407b5810039SNick Piggin struct mm_struct *mm = tlb->mm; 1408d16dfc55SPeter Zijlstra int force_flush = 0; 1409d559db08SKAMEZAWA Hiroyuki int rss[NR_MM_COUNTERS]; 141097a89413SPeter Zijlstra spinlock_t *ptl; 14115f1a1907SSteven Rostedt pte_t *start_pte; 141297a89413SPeter Zijlstra pte_t *pte; 14138a5f14a2SKirill A. Shutemov swp_entry_t entry; 1414d559db08SKAMEZAWA Hiroyuki 1415ed6a7935SPeter Zijlstra tlb_change_page_size(tlb, PAGE_SIZE); 1416d16dfc55SPeter Zijlstra again: 1417e303297eSPeter Zijlstra init_rss_vec(rss); 14185f1a1907SSteven Rostedt start_pte = pte_offset_map_lock(mm, pmd, addr, &ptl); 14195f1a1907SSteven Rostedt pte = start_pte; 14203ea27719SMel Gorman flush_tlb_batched_pending(mm); 14216606c3e0SZachary Amsden arch_enter_lazy_mmu_mode(); 14221da177e4SLinus Torvalds do { 14231da177e4SLinus Torvalds pte_t ptent = *pte; 14248018db85SPeter Xu struct page *page; 14258018db85SPeter Xu 1426166f61b9STobin C Harding if (pte_none(ptent)) 14271da177e4SLinus Torvalds continue; 142851c6f666SRobin Holt 14297b167b68SMinchan Kim if (need_resched()) 14307b167b68SMinchan Kim break; 14317b167b68SMinchan Kim 14326f5e6b9eSHugh Dickins if (pte_present(ptent)) { 143325b2995aSChristoph Hellwig page = vm_normal_page(vma, addr, ptent); 1434254ab940SPeter Xu if (unlikely(!should_zap_page(details, page))) 14351da177e4SLinus Torvalds continue; 1436b5810039SNick Piggin ptent = ptep_get_and_clear_full(mm, addr, pte, 1437a600388dSZachary Amsden tlb->fullmm); 14381da177e4SLinus Torvalds tlb_remove_tlb_entry(tlb, pte, addr); 1439999dad82SPeter Xu zap_install_uffd_wp_if_needed(vma, addr, pte, details, 1440999dad82SPeter Xu ptent); 14411da177e4SLinus Torvalds if (unlikely(!page)) 14421da177e4SLinus Torvalds continue; 1443eca56ff9SJerome Marchand 1444eca56ff9SJerome Marchand if (!PageAnon(page)) { 14451cf35d47SLinus Torvalds if (pte_dirty(ptent)) { 14461cf35d47SLinus Torvalds force_flush = 1; 14476237bcd9SHugh Dickins set_page_dirty(page); 14481cf35d47SLinus Torvalds } 14494917e5d0SJohannes Weiner if (pte_young(ptent) && 145064363aadSJoe Perches likely(!(vma->vm_flags & VM_SEQ_READ))) 1451bf3f3bc5SNick Piggin mark_page_accessed(page); 14526237bcd9SHugh Dickins } 1453eca56ff9SJerome Marchand rss[mm_counter(page)]--; 1454cea86fe2SHugh Dickins page_remove_rmap(page, vma, false); 14553dc14741SHugh Dickins if (unlikely(page_mapcount(page) < 0)) 14563dc14741SHugh Dickins print_bad_pte(vma, addr, ptent, page); 1457e9d55e15SAneesh Kumar K.V if (unlikely(__tlb_remove_page(tlb, page))) { 14581cf35d47SLinus Torvalds force_flush = 1; 1459ce9ec37bSWill Deacon addr += PAGE_SIZE; 1460d16dfc55SPeter Zijlstra break; 14611cf35d47SLinus Torvalds } 14621da177e4SLinus Torvalds continue; 14631da177e4SLinus Torvalds } 14645042db43SJérôme Glisse 14655042db43SJérôme Glisse entry = pte_to_swp_entry(ptent); 1466b756a3b5SAlistair Popple if (is_device_private_entry(entry) || 1467b756a3b5SAlistair Popple is_device_exclusive_entry(entry)) { 14688018db85SPeter Xu page = pfn_swap_entry_to_page(entry); 1469254ab940SPeter Xu if (unlikely(!should_zap_page(details, page))) 14705042db43SJérôme Glisse continue; 1471999dad82SPeter Xu /* 1472999dad82SPeter Xu * Both device private/exclusive mappings should only 1473999dad82SPeter Xu * work with anonymous page so far, so we don't need to 1474999dad82SPeter Xu * consider uffd-wp bit when zap. For more information, 1475999dad82SPeter Xu * see zap_install_uffd_wp_if_needed(). 1476999dad82SPeter Xu */ 1477999dad82SPeter Xu WARN_ON_ONCE(!vma_is_anonymous(vma)); 14785042db43SJérôme Glisse rss[mm_counter(page)]--; 1479b756a3b5SAlistair Popple if (is_device_private_entry(entry)) 1480cea86fe2SHugh Dickins page_remove_rmap(page, vma, false); 14815042db43SJérôme Glisse put_page(page); 14828018db85SPeter Xu } else if (!non_swap_entry(entry)) { 14835abfd71dSPeter Xu /* Genuine swap entry, hence a private anon page */ 14845abfd71dSPeter Xu if (!should_zap_cows(details)) 14851da177e4SLinus Torvalds continue; 1486b084d435SKAMEZAWA Hiroyuki rss[MM_SWAPENTS]--; 14878018db85SPeter Xu if (unlikely(!free_swap_and_cache(entry))) 14888018db85SPeter Xu print_bad_pte(vma, addr, ptent, NULL); 14895abfd71dSPeter Xu } else if (is_migration_entry(entry)) { 1490af5cdaf8SAlistair Popple page = pfn_swap_entry_to_page(entry); 1491254ab940SPeter Xu if (!should_zap_page(details, page)) 14925abfd71dSPeter Xu continue; 1493eca56ff9SJerome Marchand rss[mm_counter(page)]--; 1494999dad82SPeter Xu } else if (pte_marker_entry_uffd_wp(entry)) { 1495999dad82SPeter Xu /* Only drop the uffd-wp marker if explicitly requested */ 1496999dad82SPeter Xu if (!zap_drop_file_uffd_wp(details)) 1497999dad82SPeter Xu continue; 14989f186f9eSMiaohe Lin } else if (is_hwpoison_entry(entry) || 14999f186f9eSMiaohe Lin is_swapin_error_entry(entry)) { 15005abfd71dSPeter Xu if (!should_zap_cows(details)) 15015abfd71dSPeter Xu continue; 15025abfd71dSPeter Xu } else { 15035abfd71dSPeter Xu /* We should have covered all the swap entry types */ 15045abfd71dSPeter Xu WARN_ON_ONCE(1); 15059f9f1acdSKonstantin Khlebnikov } 15069888a1caSZachary Amsden pte_clear_not_present_full(mm, addr, pte, tlb->fullmm); 1507999dad82SPeter Xu zap_install_uffd_wp_if_needed(vma, addr, pte, details, ptent); 150897a89413SPeter Zijlstra } while (pte++, addr += PAGE_SIZE, addr != end); 1509ae859762SHugh Dickins 1510d559db08SKAMEZAWA Hiroyuki add_mm_rss_vec(mm, rss); 15116606c3e0SZachary Amsden arch_leave_lazy_mmu_mode(); 151251c6f666SRobin Holt 15131cf35d47SLinus Torvalds /* Do the actual TLB flush before dropping ptl */ 1514fb7332a9SWill Deacon if (force_flush) 15151cf35d47SLinus Torvalds tlb_flush_mmu_tlbonly(tlb); 15161cf35d47SLinus Torvalds pte_unmap_unlock(start_pte, ptl); 15171cf35d47SLinus Torvalds 15181cf35d47SLinus Torvalds /* 15191cf35d47SLinus Torvalds * If we forced a TLB flush (either due to running out of 15201cf35d47SLinus Torvalds * batch buffers or because we needed to flush dirty TLB 15211cf35d47SLinus Torvalds * entries before releasing the ptl), free the batched 15221cf35d47SLinus Torvalds * memory too. Restart if we didn't do everything. 15231cf35d47SLinus Torvalds */ 15241cf35d47SLinus Torvalds if (force_flush) { 15251cf35d47SLinus Torvalds force_flush = 0; 1526fa0aafb8SPeter Zijlstra tlb_flush_mmu(tlb); 15277b167b68SMinchan Kim } 15287b167b68SMinchan Kim 15297b167b68SMinchan Kim if (addr != end) { 15307b167b68SMinchan Kim cond_resched(); 1531d16dfc55SPeter Zijlstra goto again; 1532d16dfc55SPeter Zijlstra } 1533d16dfc55SPeter Zijlstra 153451c6f666SRobin Holt return addr; 15351da177e4SLinus Torvalds } 15361da177e4SLinus Torvalds 153751c6f666SRobin Holt static inline unsigned long zap_pmd_range(struct mmu_gather *tlb, 1538b5810039SNick Piggin struct vm_area_struct *vma, pud_t *pud, 15391da177e4SLinus Torvalds unsigned long addr, unsigned long end, 154097a89413SPeter Zijlstra struct zap_details *details) 15411da177e4SLinus Torvalds { 15421da177e4SLinus Torvalds pmd_t *pmd; 15431da177e4SLinus Torvalds unsigned long next; 15441da177e4SLinus Torvalds 15451da177e4SLinus Torvalds pmd = pmd_offset(pud, addr); 15461da177e4SLinus Torvalds do { 15471da177e4SLinus Torvalds next = pmd_addr_end(addr, end); 154884c3fc4eSZi Yan if (is_swap_pmd(*pmd) || pmd_trans_huge(*pmd) || pmd_devmap(*pmd)) { 154953406ed1SHugh Dickins if (next - addr != HPAGE_PMD_SIZE) 1550fd60775aSDavid Rientjes __split_huge_pmd(vma, pmd, addr, false, NULL); 155153406ed1SHugh Dickins else if (zap_huge_pmd(tlb, vma, pmd, addr)) 15521a5a9906SAndrea Arcangeli goto next; 155371e3aac0SAndrea Arcangeli /* fall through */ 15543506659eSMatthew Wilcox (Oracle) } else if (details && details->single_folio && 15553506659eSMatthew Wilcox (Oracle) folio_test_pmd_mappable(details->single_folio) && 155622061a1fSHugh Dickins next - addr == HPAGE_PMD_SIZE && pmd_none(*pmd)) { 155722061a1fSHugh Dickins spinlock_t *ptl = pmd_lock(tlb->mm, pmd); 155822061a1fSHugh Dickins /* 155922061a1fSHugh Dickins * Take and drop THP pmd lock so that we cannot return 156022061a1fSHugh Dickins * prematurely, while zap_huge_pmd() has cleared *pmd, 156122061a1fSHugh Dickins * but not yet decremented compound_mapcount(). 156222061a1fSHugh Dickins */ 156322061a1fSHugh Dickins spin_unlock(ptl); 156471e3aac0SAndrea Arcangeli } 156522061a1fSHugh Dickins 15661a5a9906SAndrea Arcangeli /* 15671a5a9906SAndrea Arcangeli * Here there can be other concurrent MADV_DONTNEED or 15681a5a9906SAndrea Arcangeli * trans huge page faults running, and if the pmd is 15691a5a9906SAndrea Arcangeli * none or trans huge it can change under us. This is 1570c1e8d7c6SMichel Lespinasse * because MADV_DONTNEED holds the mmap_lock in read 15711a5a9906SAndrea Arcangeli * mode. 15721a5a9906SAndrea Arcangeli */ 15731a5a9906SAndrea Arcangeli if (pmd_none_or_trans_huge_or_clear_bad(pmd)) 15741a5a9906SAndrea Arcangeli goto next; 157597a89413SPeter Zijlstra next = zap_pte_range(tlb, vma, pmd, addr, next, details); 15761a5a9906SAndrea Arcangeli next: 157797a89413SPeter Zijlstra cond_resched(); 157897a89413SPeter Zijlstra } while (pmd++, addr = next, addr != end); 157951c6f666SRobin Holt 158051c6f666SRobin Holt return addr; 15811da177e4SLinus Torvalds } 15821da177e4SLinus Torvalds 158351c6f666SRobin Holt static inline unsigned long zap_pud_range(struct mmu_gather *tlb, 1584c2febafcSKirill A. Shutemov struct vm_area_struct *vma, p4d_t *p4d, 15851da177e4SLinus Torvalds unsigned long addr, unsigned long end, 158697a89413SPeter Zijlstra struct zap_details *details) 15871da177e4SLinus Torvalds { 15881da177e4SLinus Torvalds pud_t *pud; 15891da177e4SLinus Torvalds unsigned long next; 15901da177e4SLinus Torvalds 1591c2febafcSKirill A. Shutemov pud = pud_offset(p4d, addr); 15921da177e4SLinus Torvalds do { 15931da177e4SLinus Torvalds next = pud_addr_end(addr, end); 1594a00cc7d9SMatthew Wilcox if (pud_trans_huge(*pud) || pud_devmap(*pud)) { 1595a00cc7d9SMatthew Wilcox if (next - addr != HPAGE_PUD_SIZE) { 159642fc5414SMichel Lespinasse mmap_assert_locked(tlb->mm); 1597a00cc7d9SMatthew Wilcox split_huge_pud(vma, pud, addr); 1598a00cc7d9SMatthew Wilcox } else if (zap_huge_pud(tlb, vma, pud, addr)) 1599a00cc7d9SMatthew Wilcox goto next; 1600a00cc7d9SMatthew Wilcox /* fall through */ 1601a00cc7d9SMatthew Wilcox } 160297a89413SPeter Zijlstra if (pud_none_or_clear_bad(pud)) 16031da177e4SLinus Torvalds continue; 160497a89413SPeter Zijlstra next = zap_pmd_range(tlb, vma, pud, addr, next, details); 1605a00cc7d9SMatthew Wilcox next: 1606a00cc7d9SMatthew Wilcox cond_resched(); 160797a89413SPeter Zijlstra } while (pud++, addr = next, addr != end); 160851c6f666SRobin Holt 160951c6f666SRobin Holt return addr; 16101da177e4SLinus Torvalds } 16111da177e4SLinus Torvalds 1612c2febafcSKirill A. Shutemov static inline unsigned long zap_p4d_range(struct mmu_gather *tlb, 1613c2febafcSKirill A. Shutemov struct vm_area_struct *vma, pgd_t *pgd, 1614c2febafcSKirill A. Shutemov unsigned long addr, unsigned long end, 1615c2febafcSKirill A. Shutemov struct zap_details *details) 1616c2febafcSKirill A. Shutemov { 1617c2febafcSKirill A. Shutemov p4d_t *p4d; 1618c2febafcSKirill A. Shutemov unsigned long next; 1619c2febafcSKirill A. Shutemov 1620c2febafcSKirill A. Shutemov p4d = p4d_offset(pgd, addr); 1621c2febafcSKirill A. Shutemov do { 1622c2febafcSKirill A. Shutemov next = p4d_addr_end(addr, end); 1623c2febafcSKirill A. Shutemov if (p4d_none_or_clear_bad(p4d)) 1624c2febafcSKirill A. Shutemov continue; 1625c2febafcSKirill A. Shutemov next = zap_pud_range(tlb, vma, p4d, addr, next, details); 1626c2febafcSKirill A. Shutemov } while (p4d++, addr = next, addr != end); 1627c2febafcSKirill A. Shutemov 1628c2febafcSKirill A. Shutemov return addr; 1629c2febafcSKirill A. Shutemov } 1630c2febafcSKirill A. Shutemov 1631aac45363SMichal Hocko void unmap_page_range(struct mmu_gather *tlb, 163251c6f666SRobin Holt struct vm_area_struct *vma, 16331da177e4SLinus Torvalds unsigned long addr, unsigned long end, 163497a89413SPeter Zijlstra struct zap_details *details) 16351da177e4SLinus Torvalds { 16361da177e4SLinus Torvalds pgd_t *pgd; 16371da177e4SLinus Torvalds unsigned long next; 16381da177e4SLinus Torvalds 16391da177e4SLinus Torvalds BUG_ON(addr >= end); 16401da177e4SLinus Torvalds tlb_start_vma(tlb, vma); 16411da177e4SLinus Torvalds pgd = pgd_offset(vma->vm_mm, addr); 16421da177e4SLinus Torvalds do { 16431da177e4SLinus Torvalds next = pgd_addr_end(addr, end); 164497a89413SPeter Zijlstra if (pgd_none_or_clear_bad(pgd)) 16451da177e4SLinus Torvalds continue; 1646c2febafcSKirill A. Shutemov next = zap_p4d_range(tlb, vma, pgd, addr, next, details); 164797a89413SPeter Zijlstra } while (pgd++, addr = next, addr != end); 16481da177e4SLinus Torvalds tlb_end_vma(tlb, vma); 16491da177e4SLinus Torvalds } 16501da177e4SLinus Torvalds 1651f5cc4eefSAl Viro 1652f5cc4eefSAl Viro static void unmap_single_vma(struct mmu_gather *tlb, 16531da177e4SLinus Torvalds struct vm_area_struct *vma, unsigned long start_addr, 16544f74d2c8SLinus Torvalds unsigned long end_addr, 16551da177e4SLinus Torvalds struct zap_details *details) 16561da177e4SLinus Torvalds { 1657f5cc4eefSAl Viro unsigned long start = max(vma->vm_start, start_addr); 16581da177e4SLinus Torvalds unsigned long end; 16591da177e4SLinus Torvalds 16601da177e4SLinus Torvalds if (start >= vma->vm_end) 1661f5cc4eefSAl Viro return; 16621da177e4SLinus Torvalds end = min(vma->vm_end, end_addr); 16631da177e4SLinus Torvalds if (end <= vma->vm_start) 1664f5cc4eefSAl Viro return; 16651da177e4SLinus Torvalds 1666cbc91f71SSrikar Dronamraju if (vma->vm_file) 1667cbc91f71SSrikar Dronamraju uprobe_munmap(vma, start, end); 1668cbc91f71SSrikar Dronamraju 1669b3b9c293SKonstantin Khlebnikov if (unlikely(vma->vm_flags & VM_PFNMAP)) 16705180da41SSuresh Siddha untrack_pfn(vma, 0, 0); 16712ab64037Svenkatesh.pallipadi@intel.com 16728b2a1238SAl Viro if (start != end) { 167351c6f666SRobin Holt if (unlikely(is_vm_hugetlb_page(vma))) { 1674a137e1ccSAndi Kleen /* 1675a137e1ccSAndi Kleen * It is undesirable to test vma->vm_file as it 1676a137e1ccSAndi Kleen * should be non-null for valid hugetlb area. 1677a137e1ccSAndi Kleen * However, vm_file will be NULL in the error 16787aa6b4adSDavidlohr Bueso * cleanup path of mmap_region. When 1679a137e1ccSAndi Kleen * hugetlbfs ->mmap method fails, 16807aa6b4adSDavidlohr Bueso * mmap_region() nullifies vma->vm_file 1681a137e1ccSAndi Kleen * before calling this function to clean up. 1682a137e1ccSAndi Kleen * Since no pte has actually been setup, it is 1683a137e1ccSAndi Kleen * safe to do nothing in this case. 1684a137e1ccSAndi Kleen */ 168524669e58SAneesh Kumar K.V if (vma->vm_file) { 168605e90bd0SPeter Xu zap_flags_t zap_flags = details ? 168705e90bd0SPeter Xu details->zap_flags : 0; 168883cde9e8SDavidlohr Bueso i_mmap_lock_write(vma->vm_file->f_mapping); 168905e90bd0SPeter Xu __unmap_hugepage_range_final(tlb, vma, start, end, 169005e90bd0SPeter Xu NULL, zap_flags); 169183cde9e8SDavidlohr Bueso i_mmap_unlock_write(vma->vm_file->f_mapping); 169224669e58SAneesh Kumar K.V } 169351c6f666SRobin Holt } else 1694038c7aa1SAl Viro unmap_page_range(tlb, vma, start, end, details); 169597a89413SPeter Zijlstra } 169651c6f666SRobin Holt } 16971da177e4SLinus Torvalds 1698f5cc4eefSAl Viro /** 1699f5cc4eefSAl Viro * unmap_vmas - unmap a range of memory covered by a list of vma's 1700f5cc4eefSAl Viro * @tlb: address of the caller's struct mmu_gather 1701f5cc4eefSAl Viro * @vma: the starting vma 1702f5cc4eefSAl Viro * @start_addr: virtual address at which to start unmapping 1703f5cc4eefSAl Viro * @end_addr: virtual address at which to end unmapping 1704f5cc4eefSAl Viro * 1705f5cc4eefSAl Viro * Unmap all pages in the vma list. 1706f5cc4eefSAl Viro * 1707f5cc4eefSAl Viro * Only addresses between `start' and `end' will be unmapped. 1708f5cc4eefSAl Viro * 1709f5cc4eefSAl Viro * The VMA list must be sorted in ascending virtual address order. 1710f5cc4eefSAl Viro * 1711f5cc4eefSAl Viro * unmap_vmas() assumes that the caller will flush the whole unmapped address 1712f5cc4eefSAl Viro * range after unmap_vmas() returns. So the only responsibility here is to 1713f5cc4eefSAl Viro * ensure that any thus-far unmapped pages are flushed before unmap_vmas() 1714f5cc4eefSAl Viro * drops the lock and schedules. 1715f5cc4eefSAl Viro */ 1716f5cc4eefSAl Viro void unmap_vmas(struct mmu_gather *tlb, 1717f5cc4eefSAl Viro struct vm_area_struct *vma, unsigned long start_addr, 17184f74d2c8SLinus Torvalds unsigned long end_addr) 1719f5cc4eefSAl Viro { 1720ac46d4f3SJérôme Glisse struct mmu_notifier_range range; 1721999dad82SPeter Xu struct zap_details details = { 1722999dad82SPeter Xu .zap_flags = ZAP_FLAG_DROP_MARKER, 1723999dad82SPeter Xu /* Careful - we need to zap private pages too! */ 1724999dad82SPeter Xu .even_cows = true, 1725999dad82SPeter Xu }; 1726f5cc4eefSAl Viro 17276f4f13e8SJérôme Glisse mmu_notifier_range_init(&range, MMU_NOTIFY_UNMAP, 0, vma, vma->vm_mm, 17286f4f13e8SJérôme Glisse start_addr, end_addr); 1729ac46d4f3SJérôme Glisse mmu_notifier_invalidate_range_start(&range); 1730f5cc4eefSAl Viro for ( ; vma && vma->vm_start < end_addr; vma = vma->vm_next) 1731999dad82SPeter Xu unmap_single_vma(tlb, vma, start_addr, end_addr, &details); 1732ac46d4f3SJérôme Glisse mmu_notifier_invalidate_range_end(&range); 17331da177e4SLinus Torvalds } 17341da177e4SLinus Torvalds 17351da177e4SLinus Torvalds /** 17361da177e4SLinus Torvalds * zap_page_range - remove user pages in a given range 17371da177e4SLinus Torvalds * @vma: vm_area_struct holding the applicable pages 1738eb4546bbSRandy Dunlap * @start: starting address of pages to zap 17391da177e4SLinus Torvalds * @size: number of bytes to zap 1740f5cc4eefSAl Viro * 1741f5cc4eefSAl Viro * Caller must protect the VMA list 17421da177e4SLinus Torvalds */ 17437e027b14SLinus Torvalds void zap_page_range(struct vm_area_struct *vma, unsigned long start, 1744ecf1385dSKirill A. Shutemov unsigned long size) 17451da177e4SLinus Torvalds { 1746ac46d4f3SJérôme Glisse struct mmu_notifier_range range; 1747d16dfc55SPeter Zijlstra struct mmu_gather tlb; 17481da177e4SLinus Torvalds 17491da177e4SLinus Torvalds lru_add_drain(); 17507269f999SJérôme Glisse mmu_notifier_range_init(&range, MMU_NOTIFY_CLEAR, 0, vma, vma->vm_mm, 17516f4f13e8SJérôme Glisse start, start + size); 1752a72afd87SWill Deacon tlb_gather_mmu(&tlb, vma->vm_mm); 1753ac46d4f3SJérôme Glisse update_hiwater_rss(vma->vm_mm); 1754ac46d4f3SJérôme Glisse mmu_notifier_invalidate_range_start(&range); 1755ac46d4f3SJérôme Glisse for ( ; vma && vma->vm_start < range.end; vma = vma->vm_next) 1756ac46d4f3SJérôme Glisse unmap_single_vma(&tlb, vma, start, range.end, NULL); 1757ac46d4f3SJérôme Glisse mmu_notifier_invalidate_range_end(&range); 1758ae8eba8bSWill Deacon tlb_finish_mmu(&tlb); 17591da177e4SLinus Torvalds } 17601da177e4SLinus Torvalds 1761c627f9ccSJack Steiner /** 1762f5cc4eefSAl Viro * zap_page_range_single - remove user pages in a given range 1763f5cc4eefSAl Viro * @vma: vm_area_struct holding the applicable pages 1764f5cc4eefSAl Viro * @address: starting address of pages to zap 1765f5cc4eefSAl Viro * @size: number of bytes to zap 17668a5f14a2SKirill A. Shutemov * @details: details of shared cache invalidation 1767f5cc4eefSAl Viro * 1768f5cc4eefSAl Viro * The range must fit into one VMA. 1769f5cc4eefSAl Viro */ 1770f5cc4eefSAl Viro static void zap_page_range_single(struct vm_area_struct *vma, unsigned long address, 1771f5cc4eefSAl Viro unsigned long size, struct zap_details *details) 1772f5cc4eefSAl Viro { 1773ac46d4f3SJérôme Glisse struct mmu_notifier_range range; 1774f5cc4eefSAl Viro struct mmu_gather tlb; 1775f5cc4eefSAl Viro 1776f5cc4eefSAl Viro lru_add_drain(); 17777269f999SJérôme Glisse mmu_notifier_range_init(&range, MMU_NOTIFY_CLEAR, 0, vma, vma->vm_mm, 17786f4f13e8SJérôme Glisse address, address + size); 1779a72afd87SWill Deacon tlb_gather_mmu(&tlb, vma->vm_mm); 1780ac46d4f3SJérôme Glisse update_hiwater_rss(vma->vm_mm); 1781ac46d4f3SJérôme Glisse mmu_notifier_invalidate_range_start(&range); 1782ac46d4f3SJérôme Glisse unmap_single_vma(&tlb, vma, address, range.end, details); 1783ac46d4f3SJérôme Glisse mmu_notifier_invalidate_range_end(&range); 1784ae8eba8bSWill Deacon tlb_finish_mmu(&tlb); 17851da177e4SLinus Torvalds } 17861da177e4SLinus Torvalds 1787c627f9ccSJack Steiner /** 1788c627f9ccSJack Steiner * zap_vma_ptes - remove ptes mapping the vma 1789c627f9ccSJack Steiner * @vma: vm_area_struct holding ptes to be zapped 1790c627f9ccSJack Steiner * @address: starting address of pages to zap 1791c627f9ccSJack Steiner * @size: number of bytes to zap 1792c627f9ccSJack Steiner * 1793c627f9ccSJack Steiner * This function only unmaps ptes assigned to VM_PFNMAP vmas. 1794c627f9ccSJack Steiner * 1795c627f9ccSJack Steiner * The entire address range must be fully contained within the vma. 1796c627f9ccSJack Steiner * 1797c627f9ccSJack Steiner */ 179827d036e3SLeon Romanovsky void zap_vma_ptes(struct vm_area_struct *vma, unsigned long address, 1799c627f9ccSJack Steiner unsigned long size) 1800c627f9ccSJack Steiner { 180188a35912SMiaohe Lin if (!range_in_vma(vma, address, address + size) || 1802c627f9ccSJack Steiner !(vma->vm_flags & VM_PFNMAP)) 180327d036e3SLeon Romanovsky return; 180427d036e3SLeon Romanovsky 1805f5cc4eefSAl Viro zap_page_range_single(vma, address, size, NULL); 1806c627f9ccSJack Steiner } 1807c627f9ccSJack Steiner EXPORT_SYMBOL_GPL(zap_vma_ptes); 1808c627f9ccSJack Steiner 18098cd3984dSArjun Roy static pmd_t *walk_to_pmd(struct mm_struct *mm, unsigned long addr) 1810c9cfcddfSLinus Torvalds { 1811c2febafcSKirill A. Shutemov pgd_t *pgd; 1812c2febafcSKirill A. Shutemov p4d_t *p4d; 1813c2febafcSKirill A. Shutemov pud_t *pud; 1814c2febafcSKirill A. Shutemov pmd_t *pmd; 1815c2febafcSKirill A. Shutemov 1816c2febafcSKirill A. Shutemov pgd = pgd_offset(mm, addr); 1817c2febafcSKirill A. Shutemov p4d = p4d_alloc(mm, pgd, addr); 1818c2febafcSKirill A. Shutemov if (!p4d) 1819c2febafcSKirill A. Shutemov return NULL; 1820c2febafcSKirill A. Shutemov pud = pud_alloc(mm, p4d, addr); 1821c2febafcSKirill A. Shutemov if (!pud) 1822c2febafcSKirill A. Shutemov return NULL; 1823c2febafcSKirill A. Shutemov pmd = pmd_alloc(mm, pud, addr); 1824c2febafcSKirill A. Shutemov if (!pmd) 1825c2febafcSKirill A. Shutemov return NULL; 1826c2febafcSKirill A. Shutemov 1827f66055abSAndrea Arcangeli VM_BUG_ON(pmd_trans_huge(*pmd)); 18288cd3984dSArjun Roy return pmd; 18298cd3984dSArjun Roy } 18308cd3984dSArjun Roy 18318cd3984dSArjun Roy pte_t *__get_locked_pte(struct mm_struct *mm, unsigned long addr, 18328cd3984dSArjun Roy spinlock_t **ptl) 18338cd3984dSArjun Roy { 18348cd3984dSArjun Roy pmd_t *pmd = walk_to_pmd(mm, addr); 18358cd3984dSArjun Roy 18368cd3984dSArjun Roy if (!pmd) 18378cd3984dSArjun Roy return NULL; 1838c9cfcddfSLinus Torvalds return pte_alloc_map_lock(mm, pmd, addr, ptl); 1839c9cfcddfSLinus Torvalds } 1840c9cfcddfSLinus Torvalds 18418efd6f5bSArjun Roy static int validate_page_before_insert(struct page *page) 18428efd6f5bSArjun Roy { 18438efd6f5bSArjun Roy if (PageAnon(page) || PageSlab(page) || page_has_type(page)) 18448efd6f5bSArjun Roy return -EINVAL; 18458efd6f5bSArjun Roy flush_dcache_page(page); 18468efd6f5bSArjun Roy return 0; 18478efd6f5bSArjun Roy } 18488efd6f5bSArjun Roy 1849cea86fe2SHugh Dickins static int insert_page_into_pte_locked(struct vm_area_struct *vma, pte_t *pte, 18508efd6f5bSArjun Roy unsigned long addr, struct page *page, pgprot_t prot) 18518efd6f5bSArjun Roy { 18528efd6f5bSArjun Roy if (!pte_none(*pte)) 18538efd6f5bSArjun Roy return -EBUSY; 18548efd6f5bSArjun Roy /* Ok, finally just insert the thing.. */ 18558efd6f5bSArjun Roy get_page(page); 1856cea86fe2SHugh Dickins inc_mm_counter_fast(vma->vm_mm, mm_counter_file(page)); 1857cea86fe2SHugh Dickins page_add_file_rmap(page, vma, false); 1858cea86fe2SHugh Dickins set_pte_at(vma->vm_mm, addr, pte, mk_pte(page, prot)); 18598efd6f5bSArjun Roy return 0; 18608efd6f5bSArjun Roy } 18618efd6f5bSArjun Roy 18621da177e4SLinus Torvalds /* 1863238f58d8SLinus Torvalds * This is the old fallback for page remapping. 1864238f58d8SLinus Torvalds * 1865238f58d8SLinus Torvalds * For historical reasons, it only allows reserved pages. Only 1866238f58d8SLinus Torvalds * old drivers should use this, and they needed to mark their 1867238f58d8SLinus Torvalds * pages reserved for the old functions anyway. 1868238f58d8SLinus Torvalds */ 1869423bad60SNick Piggin static int insert_page(struct vm_area_struct *vma, unsigned long addr, 1870423bad60SNick Piggin struct page *page, pgprot_t prot) 1871238f58d8SLinus Torvalds { 1872238f58d8SLinus Torvalds int retval; 1873238f58d8SLinus Torvalds pte_t *pte; 1874238f58d8SLinus Torvalds spinlock_t *ptl; 1875238f58d8SLinus Torvalds 18768efd6f5bSArjun Roy retval = validate_page_before_insert(page); 18778efd6f5bSArjun Roy if (retval) 18785b4e655eSKAMEZAWA Hiroyuki goto out; 1879238f58d8SLinus Torvalds retval = -ENOMEM; 1880cea86fe2SHugh Dickins pte = get_locked_pte(vma->vm_mm, addr, &ptl); 1881238f58d8SLinus Torvalds if (!pte) 18825b4e655eSKAMEZAWA Hiroyuki goto out; 1883cea86fe2SHugh Dickins retval = insert_page_into_pte_locked(vma, pte, addr, page, prot); 1884238f58d8SLinus Torvalds pte_unmap_unlock(pte, ptl); 1885238f58d8SLinus Torvalds out: 1886238f58d8SLinus Torvalds return retval; 1887238f58d8SLinus Torvalds } 1888238f58d8SLinus Torvalds 18898cd3984dSArjun Roy #ifdef pte_index 1890cea86fe2SHugh Dickins static int insert_page_in_batch_locked(struct vm_area_struct *vma, pte_t *pte, 18918cd3984dSArjun Roy unsigned long addr, struct page *page, pgprot_t prot) 18928cd3984dSArjun Roy { 18938cd3984dSArjun Roy int err; 18948cd3984dSArjun Roy 18958cd3984dSArjun Roy if (!page_count(page)) 18968cd3984dSArjun Roy return -EINVAL; 18978cd3984dSArjun Roy err = validate_page_before_insert(page); 18987f70c2a6SArjun Roy if (err) 18997f70c2a6SArjun Roy return err; 1900cea86fe2SHugh Dickins return insert_page_into_pte_locked(vma, pte, addr, page, prot); 19018cd3984dSArjun Roy } 19028cd3984dSArjun Roy 19038cd3984dSArjun Roy /* insert_pages() amortizes the cost of spinlock operations 19048cd3984dSArjun Roy * when inserting pages in a loop. Arch *must* define pte_index. 19058cd3984dSArjun Roy */ 19068cd3984dSArjun Roy static int insert_pages(struct vm_area_struct *vma, unsigned long addr, 19078cd3984dSArjun Roy struct page **pages, unsigned long *num, pgprot_t prot) 19088cd3984dSArjun Roy { 19098cd3984dSArjun Roy pmd_t *pmd = NULL; 19107f70c2a6SArjun Roy pte_t *start_pte, *pte; 19117f70c2a6SArjun Roy spinlock_t *pte_lock; 19128cd3984dSArjun Roy struct mm_struct *const mm = vma->vm_mm; 19138cd3984dSArjun Roy unsigned long curr_page_idx = 0; 19148cd3984dSArjun Roy unsigned long remaining_pages_total = *num; 19158cd3984dSArjun Roy unsigned long pages_to_write_in_pmd; 19168cd3984dSArjun Roy int ret; 19178cd3984dSArjun Roy more: 19188cd3984dSArjun Roy ret = -EFAULT; 19198cd3984dSArjun Roy pmd = walk_to_pmd(mm, addr); 19208cd3984dSArjun Roy if (!pmd) 19218cd3984dSArjun Roy goto out; 19228cd3984dSArjun Roy 19238cd3984dSArjun Roy pages_to_write_in_pmd = min_t(unsigned long, 19248cd3984dSArjun Roy remaining_pages_total, PTRS_PER_PTE - pte_index(addr)); 19258cd3984dSArjun Roy 19268cd3984dSArjun Roy /* Allocate the PTE if necessary; takes PMD lock once only. */ 19278cd3984dSArjun Roy ret = -ENOMEM; 19288cd3984dSArjun Roy if (pte_alloc(mm, pmd)) 19298cd3984dSArjun Roy goto out; 19308cd3984dSArjun Roy 19318cd3984dSArjun Roy while (pages_to_write_in_pmd) { 19328cd3984dSArjun Roy int pte_idx = 0; 19338cd3984dSArjun Roy const int batch_size = min_t(int, pages_to_write_in_pmd, 8); 19348cd3984dSArjun Roy 19357f70c2a6SArjun Roy start_pte = pte_offset_map_lock(mm, pmd, addr, &pte_lock); 19367f70c2a6SArjun Roy for (pte = start_pte; pte_idx < batch_size; ++pte, ++pte_idx) { 1937cea86fe2SHugh Dickins int err = insert_page_in_batch_locked(vma, pte, 19388cd3984dSArjun Roy addr, pages[curr_page_idx], prot); 19398cd3984dSArjun Roy if (unlikely(err)) { 19407f70c2a6SArjun Roy pte_unmap_unlock(start_pte, pte_lock); 19418cd3984dSArjun Roy ret = err; 19428cd3984dSArjun Roy remaining_pages_total -= pte_idx; 19438cd3984dSArjun Roy goto out; 19448cd3984dSArjun Roy } 19458cd3984dSArjun Roy addr += PAGE_SIZE; 19468cd3984dSArjun Roy ++curr_page_idx; 19478cd3984dSArjun Roy } 19487f70c2a6SArjun Roy pte_unmap_unlock(start_pte, pte_lock); 19498cd3984dSArjun Roy pages_to_write_in_pmd -= batch_size; 19508cd3984dSArjun Roy remaining_pages_total -= batch_size; 19518cd3984dSArjun Roy } 19528cd3984dSArjun Roy if (remaining_pages_total) 19538cd3984dSArjun Roy goto more; 19548cd3984dSArjun Roy ret = 0; 19558cd3984dSArjun Roy out: 19568cd3984dSArjun Roy *num = remaining_pages_total; 19578cd3984dSArjun Roy return ret; 19588cd3984dSArjun Roy } 19598cd3984dSArjun Roy #endif /* ifdef pte_index */ 19608cd3984dSArjun Roy 19618cd3984dSArjun Roy /** 19628cd3984dSArjun Roy * vm_insert_pages - insert multiple pages into user vma, batching the pmd lock. 19638cd3984dSArjun Roy * @vma: user vma to map to 19648cd3984dSArjun Roy * @addr: target start user address of these pages 19658cd3984dSArjun Roy * @pages: source kernel pages 19668cd3984dSArjun Roy * @num: in: number of pages to map. out: number of pages that were *not* 19678cd3984dSArjun Roy * mapped. (0 means all pages were successfully mapped). 19688cd3984dSArjun Roy * 19698cd3984dSArjun Roy * Preferred over vm_insert_page() when inserting multiple pages. 19708cd3984dSArjun Roy * 19718cd3984dSArjun Roy * In case of error, we may have mapped a subset of the provided 19728cd3984dSArjun Roy * pages. It is the caller's responsibility to account for this case. 19738cd3984dSArjun Roy * 19748cd3984dSArjun Roy * The same restrictions apply as in vm_insert_page(). 19758cd3984dSArjun Roy */ 19768cd3984dSArjun Roy int vm_insert_pages(struct vm_area_struct *vma, unsigned long addr, 19778cd3984dSArjun Roy struct page **pages, unsigned long *num) 19788cd3984dSArjun Roy { 19798cd3984dSArjun Roy #ifdef pte_index 19808cd3984dSArjun Roy const unsigned long end_addr = addr + (*num * PAGE_SIZE) - 1; 19818cd3984dSArjun Roy 19828cd3984dSArjun Roy if (addr < vma->vm_start || end_addr >= vma->vm_end) 19838cd3984dSArjun Roy return -EFAULT; 19848cd3984dSArjun Roy if (!(vma->vm_flags & VM_MIXEDMAP)) { 1985d8ed45c5SMichel Lespinasse BUG_ON(mmap_read_trylock(vma->vm_mm)); 19868cd3984dSArjun Roy BUG_ON(vma->vm_flags & VM_PFNMAP); 19878cd3984dSArjun Roy vma->vm_flags |= VM_MIXEDMAP; 19888cd3984dSArjun Roy } 19898cd3984dSArjun Roy /* Defer page refcount checking till we're about to map that page. */ 19908cd3984dSArjun Roy return insert_pages(vma, addr, pages, num, vma->vm_page_prot); 19918cd3984dSArjun Roy #else 19928cd3984dSArjun Roy unsigned long idx = 0, pgcount = *num; 199345779b03STom Rix int err = -EINVAL; 19948cd3984dSArjun Roy 19958cd3984dSArjun Roy for (; idx < pgcount; ++idx) { 19968cd3984dSArjun Roy err = vm_insert_page(vma, addr + (PAGE_SIZE * idx), pages[idx]); 19978cd3984dSArjun Roy if (err) 19988cd3984dSArjun Roy break; 19998cd3984dSArjun Roy } 20008cd3984dSArjun Roy *num = pgcount - idx; 20018cd3984dSArjun Roy return err; 20028cd3984dSArjun Roy #endif /* ifdef pte_index */ 20038cd3984dSArjun Roy } 20048cd3984dSArjun Roy EXPORT_SYMBOL(vm_insert_pages); 20058cd3984dSArjun Roy 2006bfa5bf6dSRolf Eike Beer /** 2007bfa5bf6dSRolf Eike Beer * vm_insert_page - insert single page into user vma 2008bfa5bf6dSRolf Eike Beer * @vma: user vma to map to 2009bfa5bf6dSRolf Eike Beer * @addr: target user address of this page 2010bfa5bf6dSRolf Eike Beer * @page: source kernel page 2011bfa5bf6dSRolf Eike Beer * 2012a145dd41SLinus Torvalds * This allows drivers to insert individual pages they've allocated 2013a145dd41SLinus Torvalds * into a user vma. 2014a145dd41SLinus Torvalds * 2015a145dd41SLinus Torvalds * The page has to be a nice clean _individual_ kernel allocation. 2016a145dd41SLinus Torvalds * If you allocate a compound page, you need to have marked it as 2017a145dd41SLinus Torvalds * such (__GFP_COMP), or manually just split the page up yourself 20188dfcc9baSNick Piggin * (see split_page()). 2019a145dd41SLinus Torvalds * 2020a145dd41SLinus Torvalds * NOTE! Traditionally this was done with "remap_pfn_range()" which 2021a145dd41SLinus Torvalds * took an arbitrary page protection parameter. This doesn't allow 2022a145dd41SLinus Torvalds * that. Your vma protection will have to be set up correctly, which 2023a145dd41SLinus Torvalds * means that if you want a shared writable mapping, you'd better 2024a145dd41SLinus Torvalds * ask for a shared writable mapping! 2025a145dd41SLinus Torvalds * 2026a145dd41SLinus Torvalds * The page does not need to be reserved. 20274b6e1e37SKonstantin Khlebnikov * 20284b6e1e37SKonstantin Khlebnikov * Usually this function is called from f_op->mmap() handler 2029c1e8d7c6SMichel Lespinasse * under mm->mmap_lock write-lock, so it can change vma->vm_flags. 20304b6e1e37SKonstantin Khlebnikov * Caller must set VM_MIXEDMAP on vma if it wants to call this 20314b6e1e37SKonstantin Khlebnikov * function from other places, for example from page-fault handler. 2032a862f68aSMike Rapoport * 2033a862f68aSMike Rapoport * Return: %0 on success, negative error code otherwise. 2034a145dd41SLinus Torvalds */ 2035423bad60SNick Piggin int vm_insert_page(struct vm_area_struct *vma, unsigned long addr, 2036423bad60SNick Piggin struct page *page) 2037a145dd41SLinus Torvalds { 2038a145dd41SLinus Torvalds if (addr < vma->vm_start || addr >= vma->vm_end) 2039a145dd41SLinus Torvalds return -EFAULT; 2040a145dd41SLinus Torvalds if (!page_count(page)) 2041a145dd41SLinus Torvalds return -EINVAL; 20424b6e1e37SKonstantin Khlebnikov if (!(vma->vm_flags & VM_MIXEDMAP)) { 2043d8ed45c5SMichel Lespinasse BUG_ON(mmap_read_trylock(vma->vm_mm)); 20444b6e1e37SKonstantin Khlebnikov BUG_ON(vma->vm_flags & VM_PFNMAP); 20454b6e1e37SKonstantin Khlebnikov vma->vm_flags |= VM_MIXEDMAP; 20464b6e1e37SKonstantin Khlebnikov } 2047423bad60SNick Piggin return insert_page(vma, addr, page, vma->vm_page_prot); 2048a145dd41SLinus Torvalds } 2049e3c3374fSLinus Torvalds EXPORT_SYMBOL(vm_insert_page); 2050a145dd41SLinus Torvalds 2051a667d745SSouptick Joarder /* 2052a667d745SSouptick Joarder * __vm_map_pages - maps range of kernel pages into user vma 2053a667d745SSouptick Joarder * @vma: user vma to map to 2054a667d745SSouptick Joarder * @pages: pointer to array of source kernel pages 2055a667d745SSouptick Joarder * @num: number of pages in page array 2056a667d745SSouptick Joarder * @offset: user's requested vm_pgoff 2057a667d745SSouptick Joarder * 2058a667d745SSouptick Joarder * This allows drivers to map range of kernel pages into a user vma. 2059a667d745SSouptick Joarder * 2060a667d745SSouptick Joarder * Return: 0 on success and error code otherwise. 2061a667d745SSouptick Joarder */ 2062a667d745SSouptick Joarder static int __vm_map_pages(struct vm_area_struct *vma, struct page **pages, 2063a667d745SSouptick Joarder unsigned long num, unsigned long offset) 2064a667d745SSouptick Joarder { 2065a667d745SSouptick Joarder unsigned long count = vma_pages(vma); 2066a667d745SSouptick Joarder unsigned long uaddr = vma->vm_start; 2067a667d745SSouptick Joarder int ret, i; 2068a667d745SSouptick Joarder 2069a667d745SSouptick Joarder /* Fail if the user requested offset is beyond the end of the object */ 207096756fcbSMiguel Ojeda if (offset >= num) 2071a667d745SSouptick Joarder return -ENXIO; 2072a667d745SSouptick Joarder 2073a667d745SSouptick Joarder /* Fail if the user requested size exceeds available object size */ 2074a667d745SSouptick Joarder if (count > num - offset) 2075a667d745SSouptick Joarder return -ENXIO; 2076a667d745SSouptick Joarder 2077a667d745SSouptick Joarder for (i = 0; i < count; i++) { 2078a667d745SSouptick Joarder ret = vm_insert_page(vma, uaddr, pages[offset + i]); 2079a667d745SSouptick Joarder if (ret < 0) 2080a667d745SSouptick Joarder return ret; 2081a667d745SSouptick Joarder uaddr += PAGE_SIZE; 2082a667d745SSouptick Joarder } 2083a667d745SSouptick Joarder 2084a667d745SSouptick Joarder return 0; 2085a667d745SSouptick Joarder } 2086a667d745SSouptick Joarder 2087a667d745SSouptick Joarder /** 2088a667d745SSouptick Joarder * vm_map_pages - maps range of kernel pages starts with non zero offset 2089a667d745SSouptick Joarder * @vma: user vma to map to 2090a667d745SSouptick Joarder * @pages: pointer to array of source kernel pages 2091a667d745SSouptick Joarder * @num: number of pages in page array 2092a667d745SSouptick Joarder * 2093a667d745SSouptick Joarder * Maps an object consisting of @num pages, catering for the user's 2094a667d745SSouptick Joarder * requested vm_pgoff 2095a667d745SSouptick Joarder * 2096a667d745SSouptick Joarder * If we fail to insert any page into the vma, the function will return 2097a667d745SSouptick Joarder * immediately leaving any previously inserted pages present. Callers 2098a667d745SSouptick Joarder * from the mmap handler may immediately return the error as their caller 2099a667d745SSouptick Joarder * will destroy the vma, removing any successfully inserted pages. Other 2100a667d745SSouptick Joarder * callers should make their own arrangements for calling unmap_region(). 2101a667d745SSouptick Joarder * 2102a667d745SSouptick Joarder * Context: Process context. Called by mmap handlers. 2103a667d745SSouptick Joarder * Return: 0 on success and error code otherwise. 2104a667d745SSouptick Joarder */ 2105a667d745SSouptick Joarder int vm_map_pages(struct vm_area_struct *vma, struct page **pages, 2106a667d745SSouptick Joarder unsigned long num) 2107a667d745SSouptick Joarder { 2108a667d745SSouptick Joarder return __vm_map_pages(vma, pages, num, vma->vm_pgoff); 2109a667d745SSouptick Joarder } 2110a667d745SSouptick Joarder EXPORT_SYMBOL(vm_map_pages); 2111a667d745SSouptick Joarder 2112a667d745SSouptick Joarder /** 2113a667d745SSouptick Joarder * vm_map_pages_zero - map range of kernel pages starts with zero offset 2114a667d745SSouptick Joarder * @vma: user vma to map to 2115a667d745SSouptick Joarder * @pages: pointer to array of source kernel pages 2116a667d745SSouptick Joarder * @num: number of pages in page array 2117a667d745SSouptick Joarder * 2118a667d745SSouptick Joarder * Similar to vm_map_pages(), except that it explicitly sets the offset 2119a667d745SSouptick Joarder * to 0. This function is intended for the drivers that did not consider 2120a667d745SSouptick Joarder * vm_pgoff. 2121a667d745SSouptick Joarder * 2122a667d745SSouptick Joarder * Context: Process context. Called by mmap handlers. 2123a667d745SSouptick Joarder * Return: 0 on success and error code otherwise. 2124a667d745SSouptick Joarder */ 2125a667d745SSouptick Joarder int vm_map_pages_zero(struct vm_area_struct *vma, struct page **pages, 2126a667d745SSouptick Joarder unsigned long num) 2127a667d745SSouptick Joarder { 2128a667d745SSouptick Joarder return __vm_map_pages(vma, pages, num, 0); 2129a667d745SSouptick Joarder } 2130a667d745SSouptick Joarder EXPORT_SYMBOL(vm_map_pages_zero); 2131a667d745SSouptick Joarder 21329b5a8e00SMatthew Wilcox static vm_fault_t insert_pfn(struct vm_area_struct *vma, unsigned long addr, 2133b2770da6SRoss Zwisler pfn_t pfn, pgprot_t prot, bool mkwrite) 2134423bad60SNick Piggin { 2135423bad60SNick Piggin struct mm_struct *mm = vma->vm_mm; 2136423bad60SNick Piggin pte_t *pte, entry; 2137423bad60SNick Piggin spinlock_t *ptl; 2138423bad60SNick Piggin 2139423bad60SNick Piggin pte = get_locked_pte(mm, addr, &ptl); 2140423bad60SNick Piggin if (!pte) 21419b5a8e00SMatthew Wilcox return VM_FAULT_OOM; 2142b2770da6SRoss Zwisler if (!pte_none(*pte)) { 2143b2770da6SRoss Zwisler if (mkwrite) { 2144b2770da6SRoss Zwisler /* 2145b2770da6SRoss Zwisler * For read faults on private mappings the PFN passed 2146b2770da6SRoss Zwisler * in may not match the PFN we have mapped if the 2147b2770da6SRoss Zwisler * mapped PFN is a writeable COW page. In the mkwrite 2148b2770da6SRoss Zwisler * case we are creating a writable PTE for a shared 2149f2c57d91SJan Kara * mapping and we expect the PFNs to match. If they 2150f2c57d91SJan Kara * don't match, we are likely racing with block 2151f2c57d91SJan Kara * allocation and mapping invalidation so just skip the 2152f2c57d91SJan Kara * update. 2153b2770da6SRoss Zwisler */ 2154f2c57d91SJan Kara if (pte_pfn(*pte) != pfn_t_to_pfn(pfn)) { 2155f2c57d91SJan Kara WARN_ON_ONCE(!is_zero_pfn(pte_pfn(*pte))); 2156423bad60SNick Piggin goto out_unlock; 2157f2c57d91SJan Kara } 2158cae85cb8SJan Kara entry = pte_mkyoung(*pte); 2159cae85cb8SJan Kara entry = maybe_mkwrite(pte_mkdirty(entry), vma); 2160cae85cb8SJan Kara if (ptep_set_access_flags(vma, addr, pte, entry, 1)) 2161cae85cb8SJan Kara update_mmu_cache(vma, addr, pte); 2162cae85cb8SJan Kara } 2163b2770da6SRoss Zwisler goto out_unlock; 2164b2770da6SRoss Zwisler } 2165423bad60SNick Piggin 2166423bad60SNick Piggin /* Ok, finally just insert the thing.. */ 216701c8f1c4SDan Williams if (pfn_t_devmap(pfn)) 216801c8f1c4SDan Williams entry = pte_mkdevmap(pfn_t_pte(pfn, prot)); 216901c8f1c4SDan Williams else 217001c8f1c4SDan Williams entry = pte_mkspecial(pfn_t_pte(pfn, prot)); 2171b2770da6SRoss Zwisler 2172b2770da6SRoss Zwisler if (mkwrite) { 2173b2770da6SRoss Zwisler entry = pte_mkyoung(entry); 2174b2770da6SRoss Zwisler entry = maybe_mkwrite(pte_mkdirty(entry), vma); 2175b2770da6SRoss Zwisler } 2176b2770da6SRoss Zwisler 2177423bad60SNick Piggin set_pte_at(mm, addr, pte, entry); 21784b3073e1SRussell King update_mmu_cache(vma, addr, pte); /* XXX: why not for insert_page? */ 2179423bad60SNick Piggin 2180423bad60SNick Piggin out_unlock: 2181423bad60SNick Piggin pte_unmap_unlock(pte, ptl); 21829b5a8e00SMatthew Wilcox return VM_FAULT_NOPAGE; 2183423bad60SNick Piggin } 2184423bad60SNick Piggin 2185f5e6d1d5SMatthew Wilcox /** 2186f5e6d1d5SMatthew Wilcox * vmf_insert_pfn_prot - insert single pfn into user vma with specified pgprot 2187f5e6d1d5SMatthew Wilcox * @vma: user vma to map to 2188f5e6d1d5SMatthew Wilcox * @addr: target user address of this page 2189f5e6d1d5SMatthew Wilcox * @pfn: source kernel pfn 2190f5e6d1d5SMatthew Wilcox * @pgprot: pgprot flags for the inserted page 2191f5e6d1d5SMatthew Wilcox * 2192a1a0aea5SRandy Dunlap * This is exactly like vmf_insert_pfn(), except that it allows drivers 2193f5e6d1d5SMatthew Wilcox * to override pgprot on a per-page basis. 2194f5e6d1d5SMatthew Wilcox * 2195f5e6d1d5SMatthew Wilcox * This only makes sense for IO mappings, and it makes no sense for 2196f5e6d1d5SMatthew Wilcox * COW mappings. In general, using multiple vmas is preferable; 2197ae2b01f3SMatthew Wilcox * vmf_insert_pfn_prot should only be used if using multiple VMAs is 2198f5e6d1d5SMatthew Wilcox * impractical. 2199f5e6d1d5SMatthew Wilcox * 2200574c5b3dSThomas Hellstrom * See vmf_insert_mixed_prot() for a discussion of the implication of using 2201574c5b3dSThomas Hellstrom * a value of @pgprot different from that of @vma->vm_page_prot. 2202574c5b3dSThomas Hellstrom * 2203ae2b01f3SMatthew Wilcox * Context: Process context. May allocate using %GFP_KERNEL. 2204f5e6d1d5SMatthew Wilcox * Return: vm_fault_t value. 2205f5e6d1d5SMatthew Wilcox */ 2206f5e6d1d5SMatthew Wilcox vm_fault_t vmf_insert_pfn_prot(struct vm_area_struct *vma, unsigned long addr, 2207f5e6d1d5SMatthew Wilcox unsigned long pfn, pgprot_t pgprot) 2208f5e6d1d5SMatthew Wilcox { 22096d958546SMatthew Wilcox /* 22106d958546SMatthew Wilcox * Technically, architectures with pte_special can avoid all these 22116d958546SMatthew Wilcox * restrictions (same for remap_pfn_range). However we would like 22126d958546SMatthew Wilcox * consistency in testing and feature parity among all, so we should 22136d958546SMatthew Wilcox * try to keep these invariants in place for everybody. 22146d958546SMatthew Wilcox */ 22156d958546SMatthew Wilcox BUG_ON(!(vma->vm_flags & (VM_PFNMAP|VM_MIXEDMAP))); 22166d958546SMatthew Wilcox BUG_ON((vma->vm_flags & (VM_PFNMAP|VM_MIXEDMAP)) == 22176d958546SMatthew Wilcox (VM_PFNMAP|VM_MIXEDMAP)); 22186d958546SMatthew Wilcox BUG_ON((vma->vm_flags & VM_PFNMAP) && is_cow_mapping(vma->vm_flags)); 22196d958546SMatthew Wilcox BUG_ON((vma->vm_flags & VM_MIXEDMAP) && pfn_valid(pfn)); 22206d958546SMatthew Wilcox 22216d958546SMatthew Wilcox if (addr < vma->vm_start || addr >= vma->vm_end) 22226d958546SMatthew Wilcox return VM_FAULT_SIGBUS; 22236d958546SMatthew Wilcox 22246d958546SMatthew Wilcox if (!pfn_modify_allowed(pfn, pgprot)) 22256d958546SMatthew Wilcox return VM_FAULT_SIGBUS; 22266d958546SMatthew Wilcox 22276d958546SMatthew Wilcox track_pfn_insert(vma, &pgprot, __pfn_to_pfn_t(pfn, PFN_DEV)); 22286d958546SMatthew Wilcox 22299b5a8e00SMatthew Wilcox return insert_pfn(vma, addr, __pfn_to_pfn_t(pfn, PFN_DEV), pgprot, 22306d958546SMatthew Wilcox false); 2231f5e6d1d5SMatthew Wilcox } 2232f5e6d1d5SMatthew Wilcox EXPORT_SYMBOL(vmf_insert_pfn_prot); 2233e0dc0d8fSNick Piggin 2234ae2b01f3SMatthew Wilcox /** 2235ae2b01f3SMatthew Wilcox * vmf_insert_pfn - insert single pfn into user vma 2236ae2b01f3SMatthew Wilcox * @vma: user vma to map to 2237ae2b01f3SMatthew Wilcox * @addr: target user address of this page 2238ae2b01f3SMatthew Wilcox * @pfn: source kernel pfn 2239ae2b01f3SMatthew Wilcox * 2240ae2b01f3SMatthew Wilcox * Similar to vm_insert_page, this allows drivers to insert individual pages 2241ae2b01f3SMatthew Wilcox * they've allocated into a user vma. Same comments apply. 2242ae2b01f3SMatthew Wilcox * 2243ae2b01f3SMatthew Wilcox * This function should only be called from a vm_ops->fault handler, and 2244ae2b01f3SMatthew Wilcox * in that case the handler should return the result of this function. 2245ae2b01f3SMatthew Wilcox * 2246ae2b01f3SMatthew Wilcox * vma cannot be a COW mapping. 2247ae2b01f3SMatthew Wilcox * 2248ae2b01f3SMatthew Wilcox * As this is called only for pages that do not currently exist, we 2249ae2b01f3SMatthew Wilcox * do not need to flush old virtual caches or the TLB. 2250ae2b01f3SMatthew Wilcox * 2251ae2b01f3SMatthew Wilcox * Context: Process context. May allocate using %GFP_KERNEL. 2252ae2b01f3SMatthew Wilcox * Return: vm_fault_t value. 2253ae2b01f3SMatthew Wilcox */ 2254ae2b01f3SMatthew Wilcox vm_fault_t vmf_insert_pfn(struct vm_area_struct *vma, unsigned long addr, 2255ae2b01f3SMatthew Wilcox unsigned long pfn) 2256ae2b01f3SMatthew Wilcox { 2257ae2b01f3SMatthew Wilcox return vmf_insert_pfn_prot(vma, addr, pfn, vma->vm_page_prot); 2258ae2b01f3SMatthew Wilcox } 2259ae2b01f3SMatthew Wilcox EXPORT_SYMBOL(vmf_insert_pfn); 2260ae2b01f3SMatthew Wilcox 2261785a3fabSDan Williams static bool vm_mixed_ok(struct vm_area_struct *vma, pfn_t pfn) 2262785a3fabSDan Williams { 2263785a3fabSDan Williams /* these checks mirror the abort conditions in vm_normal_page */ 2264785a3fabSDan Williams if (vma->vm_flags & VM_MIXEDMAP) 2265785a3fabSDan Williams return true; 2266785a3fabSDan Williams if (pfn_t_devmap(pfn)) 2267785a3fabSDan Williams return true; 2268785a3fabSDan Williams if (pfn_t_special(pfn)) 2269785a3fabSDan Williams return true; 2270785a3fabSDan Williams if (is_zero_pfn(pfn_t_to_pfn(pfn))) 2271785a3fabSDan Williams return true; 2272785a3fabSDan Williams return false; 2273785a3fabSDan Williams } 2274785a3fabSDan Williams 227579f3aa5bSMatthew Wilcox static vm_fault_t __vm_insert_mixed(struct vm_area_struct *vma, 2276574c5b3dSThomas Hellstrom unsigned long addr, pfn_t pfn, pgprot_t pgprot, 2277574c5b3dSThomas Hellstrom bool mkwrite) 2278423bad60SNick Piggin { 227979f3aa5bSMatthew Wilcox int err; 228087744ab3SDan Williams 2281785a3fabSDan Williams BUG_ON(!vm_mixed_ok(vma, pfn)); 2282423bad60SNick Piggin 2283423bad60SNick Piggin if (addr < vma->vm_start || addr >= vma->vm_end) 228479f3aa5bSMatthew Wilcox return VM_FAULT_SIGBUS; 2285308a047cSBorislav Petkov 2286308a047cSBorislav Petkov track_pfn_insert(vma, &pgprot, pfn); 2287423bad60SNick Piggin 228842e4089cSAndi Kleen if (!pfn_modify_allowed(pfn_t_to_pfn(pfn), pgprot)) 228979f3aa5bSMatthew Wilcox return VM_FAULT_SIGBUS; 229042e4089cSAndi Kleen 2291423bad60SNick Piggin /* 2292423bad60SNick Piggin * If we don't have pte special, then we have to use the pfn_valid() 2293423bad60SNick Piggin * based VM_MIXEDMAP scheme (see vm_normal_page), and thus we *must* 2294423bad60SNick Piggin * refcount the page if pfn_valid is true (hence insert_page rather 229562eede62SHugh Dickins * than insert_pfn). If a zero_pfn were inserted into a VM_MIXEDMAP 229662eede62SHugh Dickins * without pte special, it would there be refcounted as a normal page. 2297423bad60SNick Piggin */ 229800b3a331SLaurent Dufour if (!IS_ENABLED(CONFIG_ARCH_HAS_PTE_SPECIAL) && 229900b3a331SLaurent Dufour !pfn_t_devmap(pfn) && pfn_t_valid(pfn)) { 2300423bad60SNick Piggin struct page *page; 2301423bad60SNick Piggin 230203fc2da6SDan Williams /* 230303fc2da6SDan Williams * At this point we are committed to insert_page() 230403fc2da6SDan Williams * regardless of whether the caller specified flags that 230503fc2da6SDan Williams * result in pfn_t_has_page() == false. 230603fc2da6SDan Williams */ 230703fc2da6SDan Williams page = pfn_to_page(pfn_t_to_pfn(pfn)); 230879f3aa5bSMatthew Wilcox err = insert_page(vma, addr, page, pgprot); 230979f3aa5bSMatthew Wilcox } else { 23109b5a8e00SMatthew Wilcox return insert_pfn(vma, addr, pfn, pgprot, mkwrite); 2311423bad60SNick Piggin } 2312b2770da6SRoss Zwisler 23135d747637SMatthew Wilcox if (err == -ENOMEM) 23145d747637SMatthew Wilcox return VM_FAULT_OOM; 23155d747637SMatthew Wilcox if (err < 0 && err != -EBUSY) 23165d747637SMatthew Wilcox return VM_FAULT_SIGBUS; 23175d747637SMatthew Wilcox 23185d747637SMatthew Wilcox return VM_FAULT_NOPAGE; 2319423bad60SNick Piggin } 232079f3aa5bSMatthew Wilcox 2321574c5b3dSThomas Hellstrom /** 2322574c5b3dSThomas Hellstrom * vmf_insert_mixed_prot - insert single pfn into user vma with specified pgprot 2323574c5b3dSThomas Hellstrom * @vma: user vma to map to 2324574c5b3dSThomas Hellstrom * @addr: target user address of this page 2325574c5b3dSThomas Hellstrom * @pfn: source kernel pfn 2326574c5b3dSThomas Hellstrom * @pgprot: pgprot flags for the inserted page 2327574c5b3dSThomas Hellstrom * 2328a1a0aea5SRandy Dunlap * This is exactly like vmf_insert_mixed(), except that it allows drivers 2329574c5b3dSThomas Hellstrom * to override pgprot on a per-page basis. 2330574c5b3dSThomas Hellstrom * 2331574c5b3dSThomas Hellstrom * Typically this function should be used by drivers to set caching- and 2332574c5b3dSThomas Hellstrom * encryption bits different than those of @vma->vm_page_prot, because 2333574c5b3dSThomas Hellstrom * the caching- or encryption mode may not be known at mmap() time. 2334574c5b3dSThomas Hellstrom * This is ok as long as @vma->vm_page_prot is not used by the core vm 2335574c5b3dSThomas Hellstrom * to set caching and encryption bits for those vmas (except for COW pages). 2336574c5b3dSThomas Hellstrom * This is ensured by core vm only modifying these page table entries using 2337574c5b3dSThomas Hellstrom * functions that don't touch caching- or encryption bits, using pte_modify() 2338574c5b3dSThomas Hellstrom * if needed. (See for example mprotect()). 2339574c5b3dSThomas Hellstrom * Also when new page-table entries are created, this is only done using the 2340574c5b3dSThomas Hellstrom * fault() callback, and never using the value of vma->vm_page_prot, 2341574c5b3dSThomas Hellstrom * except for page-table entries that point to anonymous pages as the result 2342574c5b3dSThomas Hellstrom * of COW. 2343574c5b3dSThomas Hellstrom * 2344574c5b3dSThomas Hellstrom * Context: Process context. May allocate using %GFP_KERNEL. 2345574c5b3dSThomas Hellstrom * Return: vm_fault_t value. 2346574c5b3dSThomas Hellstrom */ 2347574c5b3dSThomas Hellstrom vm_fault_t vmf_insert_mixed_prot(struct vm_area_struct *vma, unsigned long addr, 2348574c5b3dSThomas Hellstrom pfn_t pfn, pgprot_t pgprot) 2349574c5b3dSThomas Hellstrom { 2350574c5b3dSThomas Hellstrom return __vm_insert_mixed(vma, addr, pfn, pgprot, false); 2351574c5b3dSThomas Hellstrom } 23525379e4ddSThomas Hellstrom EXPORT_SYMBOL(vmf_insert_mixed_prot); 2353574c5b3dSThomas Hellstrom 235479f3aa5bSMatthew Wilcox vm_fault_t vmf_insert_mixed(struct vm_area_struct *vma, unsigned long addr, 235579f3aa5bSMatthew Wilcox pfn_t pfn) 235679f3aa5bSMatthew Wilcox { 2357574c5b3dSThomas Hellstrom return __vm_insert_mixed(vma, addr, pfn, vma->vm_page_prot, false); 235879f3aa5bSMatthew Wilcox } 23595d747637SMatthew Wilcox EXPORT_SYMBOL(vmf_insert_mixed); 2360423bad60SNick Piggin 2361ab77dab4SSouptick Joarder /* 2362ab77dab4SSouptick Joarder * If the insertion of PTE failed because someone else already added a 2363ab77dab4SSouptick Joarder * different entry in the mean time, we treat that as success as we assume 2364ab77dab4SSouptick Joarder * the same entry was actually inserted. 2365ab77dab4SSouptick Joarder */ 2366ab77dab4SSouptick Joarder vm_fault_t vmf_insert_mixed_mkwrite(struct vm_area_struct *vma, 2367ab77dab4SSouptick Joarder unsigned long addr, pfn_t pfn) 2368b2770da6SRoss Zwisler { 2369574c5b3dSThomas Hellstrom return __vm_insert_mixed(vma, addr, pfn, vma->vm_page_prot, true); 2370b2770da6SRoss Zwisler } 2371ab77dab4SSouptick Joarder EXPORT_SYMBOL(vmf_insert_mixed_mkwrite); 2372b2770da6SRoss Zwisler 2373a145dd41SLinus Torvalds /* 23741da177e4SLinus Torvalds * maps a range of physical memory into the requested pages. the old 23751da177e4SLinus Torvalds * mappings are removed. any references to nonexistent pages results 23761da177e4SLinus Torvalds * in null mappings (currently treated as "copy-on-access") 23771da177e4SLinus Torvalds */ 23781da177e4SLinus Torvalds static int remap_pte_range(struct mm_struct *mm, pmd_t *pmd, 23791da177e4SLinus Torvalds unsigned long addr, unsigned long end, 23801da177e4SLinus Torvalds unsigned long pfn, pgprot_t prot) 23811da177e4SLinus Torvalds { 238290a3e375SMiaohe Lin pte_t *pte, *mapped_pte; 2383c74df32cSHugh Dickins spinlock_t *ptl; 238442e4089cSAndi Kleen int err = 0; 23851da177e4SLinus Torvalds 238690a3e375SMiaohe Lin mapped_pte = pte = pte_alloc_map_lock(mm, pmd, addr, &ptl); 23871da177e4SLinus Torvalds if (!pte) 23881da177e4SLinus Torvalds return -ENOMEM; 23896606c3e0SZachary Amsden arch_enter_lazy_mmu_mode(); 23901da177e4SLinus Torvalds do { 23911da177e4SLinus Torvalds BUG_ON(!pte_none(*pte)); 239242e4089cSAndi Kleen if (!pfn_modify_allowed(pfn, prot)) { 239342e4089cSAndi Kleen err = -EACCES; 239442e4089cSAndi Kleen break; 239542e4089cSAndi Kleen } 23967e675137SNick Piggin set_pte_at(mm, addr, pte, pte_mkspecial(pfn_pte(pfn, prot))); 23971da177e4SLinus Torvalds pfn++; 23981da177e4SLinus Torvalds } while (pte++, addr += PAGE_SIZE, addr != end); 23996606c3e0SZachary Amsden arch_leave_lazy_mmu_mode(); 240090a3e375SMiaohe Lin pte_unmap_unlock(mapped_pte, ptl); 240142e4089cSAndi Kleen return err; 24021da177e4SLinus Torvalds } 24031da177e4SLinus Torvalds 24041da177e4SLinus Torvalds static inline int remap_pmd_range(struct mm_struct *mm, pud_t *pud, 24051da177e4SLinus Torvalds unsigned long addr, unsigned long end, 24061da177e4SLinus Torvalds unsigned long pfn, pgprot_t prot) 24071da177e4SLinus Torvalds { 24081da177e4SLinus Torvalds pmd_t *pmd; 24091da177e4SLinus Torvalds unsigned long next; 241042e4089cSAndi Kleen int err; 24111da177e4SLinus Torvalds 24121da177e4SLinus Torvalds pfn -= addr >> PAGE_SHIFT; 24131da177e4SLinus Torvalds pmd = pmd_alloc(mm, pud, addr); 24141da177e4SLinus Torvalds if (!pmd) 24151da177e4SLinus Torvalds return -ENOMEM; 2416f66055abSAndrea Arcangeli VM_BUG_ON(pmd_trans_huge(*pmd)); 24171da177e4SLinus Torvalds do { 24181da177e4SLinus Torvalds next = pmd_addr_end(addr, end); 241942e4089cSAndi Kleen err = remap_pte_range(mm, pmd, addr, next, 242042e4089cSAndi Kleen pfn + (addr >> PAGE_SHIFT), prot); 242142e4089cSAndi Kleen if (err) 242242e4089cSAndi Kleen return err; 24231da177e4SLinus Torvalds } while (pmd++, addr = next, addr != end); 24241da177e4SLinus Torvalds return 0; 24251da177e4SLinus Torvalds } 24261da177e4SLinus Torvalds 2427c2febafcSKirill A. Shutemov static inline int remap_pud_range(struct mm_struct *mm, p4d_t *p4d, 24281da177e4SLinus Torvalds unsigned long addr, unsigned long end, 24291da177e4SLinus Torvalds unsigned long pfn, pgprot_t prot) 24301da177e4SLinus Torvalds { 24311da177e4SLinus Torvalds pud_t *pud; 24321da177e4SLinus Torvalds unsigned long next; 243342e4089cSAndi Kleen int err; 24341da177e4SLinus Torvalds 24351da177e4SLinus Torvalds pfn -= addr >> PAGE_SHIFT; 2436c2febafcSKirill A. Shutemov pud = pud_alloc(mm, p4d, addr); 24371da177e4SLinus Torvalds if (!pud) 24381da177e4SLinus Torvalds return -ENOMEM; 24391da177e4SLinus Torvalds do { 24401da177e4SLinus Torvalds next = pud_addr_end(addr, end); 244142e4089cSAndi Kleen err = remap_pmd_range(mm, pud, addr, next, 244242e4089cSAndi Kleen pfn + (addr >> PAGE_SHIFT), prot); 244342e4089cSAndi Kleen if (err) 244442e4089cSAndi Kleen return err; 24451da177e4SLinus Torvalds } while (pud++, addr = next, addr != end); 24461da177e4SLinus Torvalds return 0; 24471da177e4SLinus Torvalds } 24481da177e4SLinus Torvalds 2449c2febafcSKirill A. Shutemov static inline int remap_p4d_range(struct mm_struct *mm, pgd_t *pgd, 2450c2febafcSKirill A. Shutemov unsigned long addr, unsigned long end, 2451c2febafcSKirill A. Shutemov unsigned long pfn, pgprot_t prot) 2452c2febafcSKirill A. Shutemov { 2453c2febafcSKirill A. Shutemov p4d_t *p4d; 2454c2febafcSKirill A. Shutemov unsigned long next; 245542e4089cSAndi Kleen int err; 2456c2febafcSKirill A. Shutemov 2457c2febafcSKirill A. Shutemov pfn -= addr >> PAGE_SHIFT; 2458c2febafcSKirill A. Shutemov p4d = p4d_alloc(mm, pgd, addr); 2459c2febafcSKirill A. Shutemov if (!p4d) 2460c2febafcSKirill A. Shutemov return -ENOMEM; 2461c2febafcSKirill A. Shutemov do { 2462c2febafcSKirill A. Shutemov next = p4d_addr_end(addr, end); 246342e4089cSAndi Kleen err = remap_pud_range(mm, p4d, addr, next, 246442e4089cSAndi Kleen pfn + (addr >> PAGE_SHIFT), prot); 246542e4089cSAndi Kleen if (err) 246642e4089cSAndi Kleen return err; 2467c2febafcSKirill A. Shutemov } while (p4d++, addr = next, addr != end); 2468c2febafcSKirill A. Shutemov return 0; 2469c2febafcSKirill A. Shutemov } 2470c2febafcSKirill A. Shutemov 247174ffa5a3SChristoph Hellwig /* 247274ffa5a3SChristoph Hellwig * Variant of remap_pfn_range that does not call track_pfn_remap. The caller 247374ffa5a3SChristoph Hellwig * must have pre-validated the caching bits of the pgprot_t. 2474bfa5bf6dSRolf Eike Beer */ 247574ffa5a3SChristoph Hellwig int remap_pfn_range_notrack(struct vm_area_struct *vma, unsigned long addr, 24761da177e4SLinus Torvalds unsigned long pfn, unsigned long size, pgprot_t prot) 24771da177e4SLinus Torvalds { 24781da177e4SLinus Torvalds pgd_t *pgd; 24791da177e4SLinus Torvalds unsigned long next; 24802d15cab8SHugh Dickins unsigned long end = addr + PAGE_ALIGN(size); 24811da177e4SLinus Torvalds struct mm_struct *mm = vma->vm_mm; 24821da177e4SLinus Torvalds int err; 24831da177e4SLinus Torvalds 24840c4123e3SAlex Zhang if (WARN_ON_ONCE(!PAGE_ALIGNED(addr))) 24850c4123e3SAlex Zhang return -EINVAL; 24860c4123e3SAlex Zhang 24871da177e4SLinus Torvalds /* 24881da177e4SLinus Torvalds * Physically remapped pages are special. Tell the 24891da177e4SLinus Torvalds * rest of the world about it: 24901da177e4SLinus Torvalds * VM_IO tells people not to look at these pages 24911da177e4SLinus Torvalds * (accesses can have side effects). 24926aab341eSLinus Torvalds * VM_PFNMAP tells the core MM that the base pages are just 24936aab341eSLinus Torvalds * raw PFN mappings, and do not have a "struct page" associated 24946aab341eSLinus Torvalds * with them. 2495314e51b9SKonstantin Khlebnikov * VM_DONTEXPAND 2496314e51b9SKonstantin Khlebnikov * Disable vma merging and expanding with mremap(). 2497314e51b9SKonstantin Khlebnikov * VM_DONTDUMP 2498314e51b9SKonstantin Khlebnikov * Omit vma from core dump, even when VM_IO turned off. 2499fb155c16SLinus Torvalds * 2500fb155c16SLinus Torvalds * There's a horrible special case to handle copy-on-write 2501fb155c16SLinus Torvalds * behaviour that some programs depend on. We mark the "original" 2502fb155c16SLinus Torvalds * un-COW'ed pages by matching them up with "vma->vm_pgoff". 2503b3b9c293SKonstantin Khlebnikov * See vm_normal_page() for details. 25041da177e4SLinus Torvalds */ 2505b3b9c293SKonstantin Khlebnikov if (is_cow_mapping(vma->vm_flags)) { 2506b3b9c293SKonstantin Khlebnikov if (addr != vma->vm_start || end != vma->vm_end) 2507b3b9c293SKonstantin Khlebnikov return -EINVAL; 25086aab341eSLinus Torvalds vma->vm_pgoff = pfn; 2509b3b9c293SKonstantin Khlebnikov } 2510b3b9c293SKonstantin Khlebnikov 2511314e51b9SKonstantin Khlebnikov vma->vm_flags |= VM_IO | VM_PFNMAP | VM_DONTEXPAND | VM_DONTDUMP; 25121da177e4SLinus Torvalds 25131da177e4SLinus Torvalds BUG_ON(addr >= end); 25141da177e4SLinus Torvalds pfn -= addr >> PAGE_SHIFT; 25151da177e4SLinus Torvalds pgd = pgd_offset(mm, addr); 25161da177e4SLinus Torvalds flush_cache_range(vma, addr, end); 25171da177e4SLinus Torvalds do { 25181da177e4SLinus Torvalds next = pgd_addr_end(addr, end); 2519c2febafcSKirill A. Shutemov err = remap_p4d_range(mm, pgd, addr, next, 25201da177e4SLinus Torvalds pfn + (addr >> PAGE_SHIFT), prot); 25211da177e4SLinus Torvalds if (err) 252274ffa5a3SChristoph Hellwig return err; 25231da177e4SLinus Torvalds } while (pgd++, addr = next, addr != end); 25242ab64037Svenkatesh.pallipadi@intel.com 252574ffa5a3SChristoph Hellwig return 0; 252674ffa5a3SChristoph Hellwig } 25272ab64037Svenkatesh.pallipadi@intel.com 252874ffa5a3SChristoph Hellwig /** 252974ffa5a3SChristoph Hellwig * remap_pfn_range - remap kernel memory to userspace 253074ffa5a3SChristoph Hellwig * @vma: user vma to map to 253174ffa5a3SChristoph Hellwig * @addr: target page aligned user address to start at 253274ffa5a3SChristoph Hellwig * @pfn: page frame number of kernel physical memory address 253374ffa5a3SChristoph Hellwig * @size: size of mapping area 253474ffa5a3SChristoph Hellwig * @prot: page protection flags for this mapping 253574ffa5a3SChristoph Hellwig * 253674ffa5a3SChristoph Hellwig * Note: this is only safe if the mm semaphore is held when called. 253774ffa5a3SChristoph Hellwig * 253874ffa5a3SChristoph Hellwig * Return: %0 on success, negative error code otherwise. 253974ffa5a3SChristoph Hellwig */ 254074ffa5a3SChristoph Hellwig int remap_pfn_range(struct vm_area_struct *vma, unsigned long addr, 254174ffa5a3SChristoph Hellwig unsigned long pfn, unsigned long size, pgprot_t prot) 254274ffa5a3SChristoph Hellwig { 254374ffa5a3SChristoph Hellwig int err; 254474ffa5a3SChristoph Hellwig 254574ffa5a3SChristoph Hellwig err = track_pfn_remap(vma, &prot, pfn, addr, PAGE_ALIGN(size)); 254674ffa5a3SChristoph Hellwig if (err) 254774ffa5a3SChristoph Hellwig return -EINVAL; 254874ffa5a3SChristoph Hellwig 254974ffa5a3SChristoph Hellwig err = remap_pfn_range_notrack(vma, addr, pfn, size, prot); 255074ffa5a3SChristoph Hellwig if (err) 255174ffa5a3SChristoph Hellwig untrack_pfn(vma, pfn, PAGE_ALIGN(size)); 25521da177e4SLinus Torvalds return err; 25531da177e4SLinus Torvalds } 25541da177e4SLinus Torvalds EXPORT_SYMBOL(remap_pfn_range); 25551da177e4SLinus Torvalds 2556b4cbb197SLinus Torvalds /** 2557b4cbb197SLinus Torvalds * vm_iomap_memory - remap memory to userspace 2558b4cbb197SLinus Torvalds * @vma: user vma to map to 2559abd69b9eSWang Wenhu * @start: start of the physical memory to be mapped 2560b4cbb197SLinus Torvalds * @len: size of area 2561b4cbb197SLinus Torvalds * 2562b4cbb197SLinus Torvalds * This is a simplified io_remap_pfn_range() for common driver use. The 2563b4cbb197SLinus Torvalds * driver just needs to give us the physical memory range to be mapped, 2564b4cbb197SLinus Torvalds * we'll figure out the rest from the vma information. 2565b4cbb197SLinus Torvalds * 2566b4cbb197SLinus Torvalds * NOTE! Some drivers might want to tweak vma->vm_page_prot first to get 2567b4cbb197SLinus Torvalds * whatever write-combining details or similar. 2568a862f68aSMike Rapoport * 2569a862f68aSMike Rapoport * Return: %0 on success, negative error code otherwise. 2570b4cbb197SLinus Torvalds */ 2571b4cbb197SLinus Torvalds int vm_iomap_memory(struct vm_area_struct *vma, phys_addr_t start, unsigned long len) 2572b4cbb197SLinus Torvalds { 2573b4cbb197SLinus Torvalds unsigned long vm_len, pfn, pages; 2574b4cbb197SLinus Torvalds 2575b4cbb197SLinus Torvalds /* Check that the physical memory area passed in looks valid */ 2576b4cbb197SLinus Torvalds if (start + len < start) 2577b4cbb197SLinus Torvalds return -EINVAL; 2578b4cbb197SLinus Torvalds /* 2579b4cbb197SLinus Torvalds * You *really* shouldn't map things that aren't page-aligned, 2580b4cbb197SLinus Torvalds * but we've historically allowed it because IO memory might 2581b4cbb197SLinus Torvalds * just have smaller alignment. 2582b4cbb197SLinus Torvalds */ 2583b4cbb197SLinus Torvalds len += start & ~PAGE_MASK; 2584b4cbb197SLinus Torvalds pfn = start >> PAGE_SHIFT; 2585b4cbb197SLinus Torvalds pages = (len + ~PAGE_MASK) >> PAGE_SHIFT; 2586b4cbb197SLinus Torvalds if (pfn + pages < pfn) 2587b4cbb197SLinus Torvalds return -EINVAL; 2588b4cbb197SLinus Torvalds 2589b4cbb197SLinus Torvalds /* We start the mapping 'vm_pgoff' pages into the area */ 2590b4cbb197SLinus Torvalds if (vma->vm_pgoff > pages) 2591b4cbb197SLinus Torvalds return -EINVAL; 2592b4cbb197SLinus Torvalds pfn += vma->vm_pgoff; 2593b4cbb197SLinus Torvalds pages -= vma->vm_pgoff; 2594b4cbb197SLinus Torvalds 2595b4cbb197SLinus Torvalds /* Can we fit all of the mapping? */ 2596b4cbb197SLinus Torvalds vm_len = vma->vm_end - vma->vm_start; 2597b4cbb197SLinus Torvalds if (vm_len >> PAGE_SHIFT > pages) 2598b4cbb197SLinus Torvalds return -EINVAL; 2599b4cbb197SLinus Torvalds 2600b4cbb197SLinus Torvalds /* Ok, let it rip */ 2601b4cbb197SLinus Torvalds return io_remap_pfn_range(vma, vma->vm_start, pfn, vm_len, vma->vm_page_prot); 2602b4cbb197SLinus Torvalds } 2603b4cbb197SLinus Torvalds EXPORT_SYMBOL(vm_iomap_memory); 2604b4cbb197SLinus Torvalds 2605aee16b3cSJeremy Fitzhardinge static int apply_to_pte_range(struct mm_struct *mm, pmd_t *pmd, 2606aee16b3cSJeremy Fitzhardinge unsigned long addr, unsigned long end, 2607e80d3909SJoerg Roedel pte_fn_t fn, void *data, bool create, 2608e80d3909SJoerg Roedel pgtbl_mod_mask *mask) 2609aee16b3cSJeremy Fitzhardinge { 26108abb50c7SMiaohe Lin pte_t *pte, *mapped_pte; 2611be1db475SDaniel Axtens int err = 0; 26123f649ab7SKees Cook spinlock_t *ptl; 2613aee16b3cSJeremy Fitzhardinge 2614be1db475SDaniel Axtens if (create) { 26158abb50c7SMiaohe Lin mapped_pte = pte = (mm == &init_mm) ? 2616e80d3909SJoerg Roedel pte_alloc_kernel_track(pmd, addr, mask) : 2617aee16b3cSJeremy Fitzhardinge pte_alloc_map_lock(mm, pmd, addr, &ptl); 2618aee16b3cSJeremy Fitzhardinge if (!pte) 2619aee16b3cSJeremy Fitzhardinge return -ENOMEM; 2620be1db475SDaniel Axtens } else { 26218abb50c7SMiaohe Lin mapped_pte = pte = (mm == &init_mm) ? 2622be1db475SDaniel Axtens pte_offset_kernel(pmd, addr) : 2623be1db475SDaniel Axtens pte_offset_map_lock(mm, pmd, addr, &ptl); 2624be1db475SDaniel Axtens } 2625aee16b3cSJeremy Fitzhardinge 2626aee16b3cSJeremy Fitzhardinge BUG_ON(pmd_huge(*pmd)); 2627aee16b3cSJeremy Fitzhardinge 262838e0edb1SJeremy Fitzhardinge arch_enter_lazy_mmu_mode(); 262938e0edb1SJeremy Fitzhardinge 2630eeb4a05fSChristoph Hellwig if (fn) { 2631aee16b3cSJeremy Fitzhardinge do { 2632be1db475SDaniel Axtens if (create || !pte_none(*pte)) { 26338b1e0f81SAnshuman Khandual err = fn(pte++, addr, data); 2634aee16b3cSJeremy Fitzhardinge if (err) 2635aee16b3cSJeremy Fitzhardinge break; 2636be1db475SDaniel Axtens } 2637c36987e2SDaisuke Nishimura } while (addr += PAGE_SIZE, addr != end); 2638eeb4a05fSChristoph Hellwig } 2639e80d3909SJoerg Roedel *mask |= PGTBL_PTE_MODIFIED; 2640aee16b3cSJeremy Fitzhardinge 264138e0edb1SJeremy Fitzhardinge arch_leave_lazy_mmu_mode(); 264238e0edb1SJeremy Fitzhardinge 2643aee16b3cSJeremy Fitzhardinge if (mm != &init_mm) 26448abb50c7SMiaohe Lin pte_unmap_unlock(mapped_pte, ptl); 2645aee16b3cSJeremy Fitzhardinge return err; 2646aee16b3cSJeremy Fitzhardinge } 2647aee16b3cSJeremy Fitzhardinge 2648aee16b3cSJeremy Fitzhardinge static int apply_to_pmd_range(struct mm_struct *mm, pud_t *pud, 2649aee16b3cSJeremy Fitzhardinge unsigned long addr, unsigned long end, 2650e80d3909SJoerg Roedel pte_fn_t fn, void *data, bool create, 2651e80d3909SJoerg Roedel pgtbl_mod_mask *mask) 2652aee16b3cSJeremy Fitzhardinge { 2653aee16b3cSJeremy Fitzhardinge pmd_t *pmd; 2654aee16b3cSJeremy Fitzhardinge unsigned long next; 2655be1db475SDaniel Axtens int err = 0; 2656aee16b3cSJeremy Fitzhardinge 2657ceb86879SAndi Kleen BUG_ON(pud_huge(*pud)); 2658ceb86879SAndi Kleen 2659be1db475SDaniel Axtens if (create) { 2660e80d3909SJoerg Roedel pmd = pmd_alloc_track(mm, pud, addr, mask); 2661aee16b3cSJeremy Fitzhardinge if (!pmd) 2662aee16b3cSJeremy Fitzhardinge return -ENOMEM; 2663be1db475SDaniel Axtens } else { 2664be1db475SDaniel Axtens pmd = pmd_offset(pud, addr); 2665be1db475SDaniel Axtens } 2666aee16b3cSJeremy Fitzhardinge do { 2667aee16b3cSJeremy Fitzhardinge next = pmd_addr_end(addr, end); 26680c95cba4SNicholas Piggin if (pmd_none(*pmd) && !create) 26690c95cba4SNicholas Piggin continue; 26700c95cba4SNicholas Piggin if (WARN_ON_ONCE(pmd_leaf(*pmd))) 26710c95cba4SNicholas Piggin return -EINVAL; 26720c95cba4SNicholas Piggin if (!pmd_none(*pmd) && WARN_ON_ONCE(pmd_bad(*pmd))) { 26730c95cba4SNicholas Piggin if (!create) 26740c95cba4SNicholas Piggin continue; 26750c95cba4SNicholas Piggin pmd_clear_bad(pmd); 26760c95cba4SNicholas Piggin } 26770c95cba4SNicholas Piggin err = apply_to_pte_range(mm, pmd, addr, next, 26780c95cba4SNicholas Piggin fn, data, create, mask); 2679aee16b3cSJeremy Fitzhardinge if (err) 2680aee16b3cSJeremy Fitzhardinge break; 2681aee16b3cSJeremy Fitzhardinge } while (pmd++, addr = next, addr != end); 26820c95cba4SNicholas Piggin 2683aee16b3cSJeremy Fitzhardinge return err; 2684aee16b3cSJeremy Fitzhardinge } 2685aee16b3cSJeremy Fitzhardinge 2686c2febafcSKirill A. Shutemov static int apply_to_pud_range(struct mm_struct *mm, p4d_t *p4d, 2687aee16b3cSJeremy Fitzhardinge unsigned long addr, unsigned long end, 2688e80d3909SJoerg Roedel pte_fn_t fn, void *data, bool create, 2689e80d3909SJoerg Roedel pgtbl_mod_mask *mask) 2690aee16b3cSJeremy Fitzhardinge { 2691aee16b3cSJeremy Fitzhardinge pud_t *pud; 2692aee16b3cSJeremy Fitzhardinge unsigned long next; 2693be1db475SDaniel Axtens int err = 0; 2694aee16b3cSJeremy Fitzhardinge 2695be1db475SDaniel Axtens if (create) { 2696e80d3909SJoerg Roedel pud = pud_alloc_track(mm, p4d, addr, mask); 2697aee16b3cSJeremy Fitzhardinge if (!pud) 2698aee16b3cSJeremy Fitzhardinge return -ENOMEM; 2699be1db475SDaniel Axtens } else { 2700be1db475SDaniel Axtens pud = pud_offset(p4d, addr); 2701be1db475SDaniel Axtens } 2702aee16b3cSJeremy Fitzhardinge do { 2703aee16b3cSJeremy Fitzhardinge next = pud_addr_end(addr, end); 27040c95cba4SNicholas Piggin if (pud_none(*pud) && !create) 27050c95cba4SNicholas Piggin continue; 27060c95cba4SNicholas Piggin if (WARN_ON_ONCE(pud_leaf(*pud))) 27070c95cba4SNicholas Piggin return -EINVAL; 27080c95cba4SNicholas Piggin if (!pud_none(*pud) && WARN_ON_ONCE(pud_bad(*pud))) { 27090c95cba4SNicholas Piggin if (!create) 27100c95cba4SNicholas Piggin continue; 27110c95cba4SNicholas Piggin pud_clear_bad(pud); 27120c95cba4SNicholas Piggin } 27130c95cba4SNicholas Piggin err = apply_to_pmd_range(mm, pud, addr, next, 27140c95cba4SNicholas Piggin fn, data, create, mask); 2715aee16b3cSJeremy Fitzhardinge if (err) 2716aee16b3cSJeremy Fitzhardinge break; 2717aee16b3cSJeremy Fitzhardinge } while (pud++, addr = next, addr != end); 27180c95cba4SNicholas Piggin 2719aee16b3cSJeremy Fitzhardinge return err; 2720aee16b3cSJeremy Fitzhardinge } 2721aee16b3cSJeremy Fitzhardinge 2722c2febafcSKirill A. Shutemov static int apply_to_p4d_range(struct mm_struct *mm, pgd_t *pgd, 2723c2febafcSKirill A. Shutemov unsigned long addr, unsigned long end, 2724e80d3909SJoerg Roedel pte_fn_t fn, void *data, bool create, 2725e80d3909SJoerg Roedel pgtbl_mod_mask *mask) 2726c2febafcSKirill A. Shutemov { 2727c2febafcSKirill A. Shutemov p4d_t *p4d; 2728c2febafcSKirill A. Shutemov unsigned long next; 2729be1db475SDaniel Axtens int err = 0; 2730c2febafcSKirill A. Shutemov 2731be1db475SDaniel Axtens if (create) { 2732e80d3909SJoerg Roedel p4d = p4d_alloc_track(mm, pgd, addr, mask); 2733c2febafcSKirill A. Shutemov if (!p4d) 2734c2febafcSKirill A. Shutemov return -ENOMEM; 2735be1db475SDaniel Axtens } else { 2736be1db475SDaniel Axtens p4d = p4d_offset(pgd, addr); 2737be1db475SDaniel Axtens } 2738c2febafcSKirill A. Shutemov do { 2739c2febafcSKirill A. Shutemov next = p4d_addr_end(addr, end); 27400c95cba4SNicholas Piggin if (p4d_none(*p4d) && !create) 27410c95cba4SNicholas Piggin continue; 27420c95cba4SNicholas Piggin if (WARN_ON_ONCE(p4d_leaf(*p4d))) 27430c95cba4SNicholas Piggin return -EINVAL; 27440c95cba4SNicholas Piggin if (!p4d_none(*p4d) && WARN_ON_ONCE(p4d_bad(*p4d))) { 27450c95cba4SNicholas Piggin if (!create) 27460c95cba4SNicholas Piggin continue; 27470c95cba4SNicholas Piggin p4d_clear_bad(p4d); 27480c95cba4SNicholas Piggin } 27490c95cba4SNicholas Piggin err = apply_to_pud_range(mm, p4d, addr, next, 27500c95cba4SNicholas Piggin fn, data, create, mask); 2751c2febafcSKirill A. Shutemov if (err) 2752c2febafcSKirill A. Shutemov break; 2753c2febafcSKirill A. Shutemov } while (p4d++, addr = next, addr != end); 27540c95cba4SNicholas Piggin 2755c2febafcSKirill A. Shutemov return err; 2756c2febafcSKirill A. Shutemov } 2757c2febafcSKirill A. Shutemov 2758be1db475SDaniel Axtens static int __apply_to_page_range(struct mm_struct *mm, unsigned long addr, 2759be1db475SDaniel Axtens unsigned long size, pte_fn_t fn, 2760be1db475SDaniel Axtens void *data, bool create) 2761be1db475SDaniel Axtens { 2762be1db475SDaniel Axtens pgd_t *pgd; 2763e80d3909SJoerg Roedel unsigned long start = addr, next; 2764be1db475SDaniel Axtens unsigned long end = addr + size; 2765e80d3909SJoerg Roedel pgtbl_mod_mask mask = 0; 2766be1db475SDaniel Axtens int err = 0; 2767be1db475SDaniel Axtens 2768be1db475SDaniel Axtens if (WARN_ON(addr >= end)) 2769be1db475SDaniel Axtens return -EINVAL; 2770be1db475SDaniel Axtens 2771be1db475SDaniel Axtens pgd = pgd_offset(mm, addr); 2772be1db475SDaniel Axtens do { 2773be1db475SDaniel Axtens next = pgd_addr_end(addr, end); 27740c95cba4SNicholas Piggin if (pgd_none(*pgd) && !create) 2775be1db475SDaniel Axtens continue; 27760c95cba4SNicholas Piggin if (WARN_ON_ONCE(pgd_leaf(*pgd))) 27770c95cba4SNicholas Piggin return -EINVAL; 27780c95cba4SNicholas Piggin if (!pgd_none(*pgd) && WARN_ON_ONCE(pgd_bad(*pgd))) { 27790c95cba4SNicholas Piggin if (!create) 27800c95cba4SNicholas Piggin continue; 27810c95cba4SNicholas Piggin pgd_clear_bad(pgd); 27820c95cba4SNicholas Piggin } 27830c95cba4SNicholas Piggin err = apply_to_p4d_range(mm, pgd, addr, next, 27840c95cba4SNicholas Piggin fn, data, create, &mask); 2785be1db475SDaniel Axtens if (err) 2786be1db475SDaniel Axtens break; 2787be1db475SDaniel Axtens } while (pgd++, addr = next, addr != end); 2788be1db475SDaniel Axtens 2789e80d3909SJoerg Roedel if (mask & ARCH_PAGE_TABLE_SYNC_MASK) 2790e80d3909SJoerg Roedel arch_sync_kernel_mappings(start, start + size); 2791e80d3909SJoerg Roedel 2792be1db475SDaniel Axtens return err; 2793be1db475SDaniel Axtens } 2794be1db475SDaniel Axtens 2795aee16b3cSJeremy Fitzhardinge /* 2796aee16b3cSJeremy Fitzhardinge * Scan a region of virtual memory, filling in page tables as necessary 2797aee16b3cSJeremy Fitzhardinge * and calling a provided function on each leaf page table. 2798aee16b3cSJeremy Fitzhardinge */ 2799aee16b3cSJeremy Fitzhardinge int apply_to_page_range(struct mm_struct *mm, unsigned long addr, 2800aee16b3cSJeremy Fitzhardinge unsigned long size, pte_fn_t fn, void *data) 2801aee16b3cSJeremy Fitzhardinge { 2802be1db475SDaniel Axtens return __apply_to_page_range(mm, addr, size, fn, data, true); 2803aee16b3cSJeremy Fitzhardinge } 2804aee16b3cSJeremy Fitzhardinge EXPORT_SYMBOL_GPL(apply_to_page_range); 2805aee16b3cSJeremy Fitzhardinge 28061da177e4SLinus Torvalds /* 2807be1db475SDaniel Axtens * Scan a region of virtual memory, calling a provided function on 2808be1db475SDaniel Axtens * each leaf page table where it exists. 2809be1db475SDaniel Axtens * 2810be1db475SDaniel Axtens * Unlike apply_to_page_range, this does _not_ fill in page tables 2811be1db475SDaniel Axtens * where they are absent. 2812be1db475SDaniel Axtens */ 2813be1db475SDaniel Axtens int apply_to_existing_page_range(struct mm_struct *mm, unsigned long addr, 2814be1db475SDaniel Axtens unsigned long size, pte_fn_t fn, void *data) 2815be1db475SDaniel Axtens { 2816be1db475SDaniel Axtens return __apply_to_page_range(mm, addr, size, fn, data, false); 2817be1db475SDaniel Axtens } 2818be1db475SDaniel Axtens EXPORT_SYMBOL_GPL(apply_to_existing_page_range); 2819be1db475SDaniel Axtens 2820be1db475SDaniel Axtens /* 28219b4bdd2fSKirill A. Shutemov * handle_pte_fault chooses page fault handler according to an entry which was 28229b4bdd2fSKirill A. Shutemov * read non-atomically. Before making any commitment, on those architectures 28239b4bdd2fSKirill A. Shutemov * or configurations (e.g. i386 with PAE) which might give a mix of unmatched 28249b4bdd2fSKirill A. Shutemov * parts, do_swap_page must check under lock before unmapping the pte and 28259b4bdd2fSKirill A. Shutemov * proceeding (but do_wp_page is only called after already making such a check; 2826a335b2e1SRyota Ozaki * and do_anonymous_page can safely check later on). 28278f4e2101SHugh Dickins */ 28282ca99358SPeter Xu static inline int pte_unmap_same(struct vm_fault *vmf) 28298f4e2101SHugh Dickins { 28308f4e2101SHugh Dickins int same = 1; 2831923717cbSThomas Gleixner #if defined(CONFIG_SMP) || defined(CONFIG_PREEMPTION) 28328f4e2101SHugh Dickins if (sizeof(pte_t) > sizeof(unsigned long)) { 28332ca99358SPeter Xu spinlock_t *ptl = pte_lockptr(vmf->vma->vm_mm, vmf->pmd); 28344c21e2f2SHugh Dickins spin_lock(ptl); 28352ca99358SPeter Xu same = pte_same(*vmf->pte, vmf->orig_pte); 28364c21e2f2SHugh Dickins spin_unlock(ptl); 28378f4e2101SHugh Dickins } 28388f4e2101SHugh Dickins #endif 28392ca99358SPeter Xu pte_unmap(vmf->pte); 28402ca99358SPeter Xu vmf->pte = NULL; 28418f4e2101SHugh Dickins return same; 28428f4e2101SHugh Dickins } 28438f4e2101SHugh Dickins 2844c89357e2SDavid Hildenbrand static inline bool __wp_page_copy_user(struct page *dst, struct page *src, 284583d116c5SJia He struct vm_fault *vmf) 28466aab341eSLinus Torvalds { 284783d116c5SJia He bool ret; 284883d116c5SJia He void *kaddr; 284983d116c5SJia He void __user *uaddr; 2850c3e5ea6eSKirill A. Shutemov bool locked = false; 285183d116c5SJia He struct vm_area_struct *vma = vmf->vma; 285283d116c5SJia He struct mm_struct *mm = vma->vm_mm; 285383d116c5SJia He unsigned long addr = vmf->address; 285483d116c5SJia He 285583d116c5SJia He if (likely(src)) { 285683d116c5SJia He copy_user_highpage(dst, src, addr, vma); 285783d116c5SJia He return true; 285883d116c5SJia He } 285983d116c5SJia He 28606aab341eSLinus Torvalds /* 28616aab341eSLinus Torvalds * If the source page was a PFN mapping, we don't have 28626aab341eSLinus Torvalds * a "struct page" for it. We do a best-effort copy by 28636aab341eSLinus Torvalds * just copying from the original user address. If that 28646aab341eSLinus Torvalds * fails, we just zero-fill it. Live with it. 28656aab341eSLinus Torvalds */ 286683d116c5SJia He kaddr = kmap_atomic(dst); 286783d116c5SJia He uaddr = (void __user *)(addr & PAGE_MASK); 286883d116c5SJia He 286983d116c5SJia He /* 287083d116c5SJia He * On architectures with software "accessed" bits, we would 287183d116c5SJia He * take a double page fault, so mark it accessed here. 287283d116c5SJia He */ 2873c3e5ea6eSKirill A. Shutemov if (arch_faults_on_old_pte() && !pte_young(vmf->orig_pte)) { 287483d116c5SJia He pte_t entry; 287583d116c5SJia He 287683d116c5SJia He vmf->pte = pte_offset_map_lock(mm, vmf->pmd, addr, &vmf->ptl); 2877c3e5ea6eSKirill A. Shutemov locked = true; 287883d116c5SJia He if (!likely(pte_same(*vmf->pte, vmf->orig_pte))) { 287983d116c5SJia He /* 288083d116c5SJia He * Other thread has already handled the fault 28817df67697SBibo Mao * and update local tlb only 288283d116c5SJia He */ 28837df67697SBibo Mao update_mmu_tlb(vma, addr, vmf->pte); 288483d116c5SJia He ret = false; 288583d116c5SJia He goto pte_unlock; 288683d116c5SJia He } 288783d116c5SJia He 288883d116c5SJia He entry = pte_mkyoung(vmf->orig_pte); 288983d116c5SJia He if (ptep_set_access_flags(vma, addr, vmf->pte, entry, 0)) 289083d116c5SJia He update_mmu_cache(vma, addr, vmf->pte); 289183d116c5SJia He } 28925d2a2dbbSLinus Torvalds 28935d2a2dbbSLinus Torvalds /* 28945d2a2dbbSLinus Torvalds * This really shouldn't fail, because the page is there 28955d2a2dbbSLinus Torvalds * in the page tables. But it might just be unreadable, 28965d2a2dbbSLinus Torvalds * in which case we just give up and fill the result with 28975d2a2dbbSLinus Torvalds * zeroes. 28985d2a2dbbSLinus Torvalds */ 289983d116c5SJia He if (__copy_from_user_inatomic(kaddr, uaddr, PAGE_SIZE)) { 2900c3e5ea6eSKirill A. Shutemov if (locked) 2901c3e5ea6eSKirill A. Shutemov goto warn; 2902c3e5ea6eSKirill A. Shutemov 2903c3e5ea6eSKirill A. Shutemov /* Re-validate under PTL if the page is still mapped */ 2904c3e5ea6eSKirill A. Shutemov vmf->pte = pte_offset_map_lock(mm, vmf->pmd, addr, &vmf->ptl); 2905c3e5ea6eSKirill A. Shutemov locked = true; 2906c3e5ea6eSKirill A. Shutemov if (!likely(pte_same(*vmf->pte, vmf->orig_pte))) { 29077df67697SBibo Mao /* The PTE changed under us, update local tlb */ 29087df67697SBibo Mao update_mmu_tlb(vma, addr, vmf->pte); 2909c3e5ea6eSKirill A. Shutemov ret = false; 2910c3e5ea6eSKirill A. Shutemov goto pte_unlock; 2911c3e5ea6eSKirill A. Shutemov } 2912c3e5ea6eSKirill A. Shutemov 2913c3e5ea6eSKirill A. Shutemov /* 2914985ba004SEthon Paul * The same page can be mapped back since last copy attempt. 2915c3e5ea6eSKirill A. Shutemov * Try to copy again under PTL. 2916c3e5ea6eSKirill A. Shutemov */ 2917c3e5ea6eSKirill A. Shutemov if (__copy_from_user_inatomic(kaddr, uaddr, PAGE_SIZE)) { 291883d116c5SJia He /* 291983d116c5SJia He * Give a warn in case there can be some obscure 292083d116c5SJia He * use-case 292183d116c5SJia He */ 2922c3e5ea6eSKirill A. Shutemov warn: 292383d116c5SJia He WARN_ON_ONCE(1); 29243ecb01dfSJan Beulich clear_page(kaddr); 292583d116c5SJia He } 2926c3e5ea6eSKirill A. Shutemov } 292783d116c5SJia He 292883d116c5SJia He ret = true; 292983d116c5SJia He 293083d116c5SJia He pte_unlock: 2931c3e5ea6eSKirill A. Shutemov if (locked) 293283d116c5SJia He pte_unmap_unlock(vmf->pte, vmf->ptl); 29339b04c5feSCong Wang kunmap_atomic(kaddr); 2934c4ec7b0dSDmitriy Monakhov flush_dcache_page(dst); 293583d116c5SJia He 293683d116c5SJia He return ret; 29376aab341eSLinus Torvalds } 29386aab341eSLinus Torvalds 2939c20cd45eSMichal Hocko static gfp_t __get_fault_gfp_mask(struct vm_area_struct *vma) 2940c20cd45eSMichal Hocko { 2941c20cd45eSMichal Hocko struct file *vm_file = vma->vm_file; 2942c20cd45eSMichal Hocko 2943c20cd45eSMichal Hocko if (vm_file) 2944c20cd45eSMichal Hocko return mapping_gfp_mask(vm_file->f_mapping) | __GFP_FS | __GFP_IO; 2945c20cd45eSMichal Hocko 2946c20cd45eSMichal Hocko /* 2947c20cd45eSMichal Hocko * Special mappings (e.g. VDSO) do not have any file so fake 2948c20cd45eSMichal Hocko * a default GFP_KERNEL for them. 2949c20cd45eSMichal Hocko */ 2950c20cd45eSMichal Hocko return GFP_KERNEL; 2951c20cd45eSMichal Hocko } 2952c20cd45eSMichal Hocko 29531da177e4SLinus Torvalds /* 2954fb09a464SKirill A. Shutemov * Notify the address space that the page is about to become writable so that 2955fb09a464SKirill A. Shutemov * it can prohibit this or wait for the page to get into an appropriate state. 2956fb09a464SKirill A. Shutemov * 2957fb09a464SKirill A. Shutemov * We do this without the lock held, so that it can sleep if it needs to. 2958fb09a464SKirill A. Shutemov */ 29592b740303SSouptick Joarder static vm_fault_t do_page_mkwrite(struct vm_fault *vmf) 2960fb09a464SKirill A. Shutemov { 29612b740303SSouptick Joarder vm_fault_t ret; 296238b8cb7fSJan Kara struct page *page = vmf->page; 296338b8cb7fSJan Kara unsigned int old_flags = vmf->flags; 2964fb09a464SKirill A. Shutemov 296538b8cb7fSJan Kara vmf->flags = FAULT_FLAG_WRITE|FAULT_FLAG_MKWRITE; 2966fb09a464SKirill A. Shutemov 2967dc617f29SDarrick J. Wong if (vmf->vma->vm_file && 2968dc617f29SDarrick J. Wong IS_SWAPFILE(vmf->vma->vm_file->f_mapping->host)) 2969dc617f29SDarrick J. Wong return VM_FAULT_SIGBUS; 2970dc617f29SDarrick J. Wong 297111bac800SDave Jiang ret = vmf->vma->vm_ops->page_mkwrite(vmf); 297238b8cb7fSJan Kara /* Restore original flags so that caller is not surprised */ 297338b8cb7fSJan Kara vmf->flags = old_flags; 2974fb09a464SKirill A. Shutemov if (unlikely(ret & (VM_FAULT_ERROR | VM_FAULT_NOPAGE))) 2975fb09a464SKirill A. Shutemov return ret; 2976fb09a464SKirill A. Shutemov if (unlikely(!(ret & VM_FAULT_LOCKED))) { 2977fb09a464SKirill A. Shutemov lock_page(page); 2978fb09a464SKirill A. Shutemov if (!page->mapping) { 2979fb09a464SKirill A. Shutemov unlock_page(page); 2980fb09a464SKirill A. Shutemov return 0; /* retry */ 2981fb09a464SKirill A. Shutemov } 2982fb09a464SKirill A. Shutemov ret |= VM_FAULT_LOCKED; 2983fb09a464SKirill A. Shutemov } else 2984fb09a464SKirill A. Shutemov VM_BUG_ON_PAGE(!PageLocked(page), page); 2985fb09a464SKirill A. Shutemov return ret; 2986fb09a464SKirill A. Shutemov } 2987fb09a464SKirill A. Shutemov 2988fb09a464SKirill A. Shutemov /* 298997ba0c2bSJan Kara * Handle dirtying of a page in shared file mapping on a write fault. 29904e047f89SShachar Raindel * 299197ba0c2bSJan Kara * The function expects the page to be locked and unlocks it. 29924e047f89SShachar Raindel */ 299389b15332SJohannes Weiner static vm_fault_t fault_dirty_shared_page(struct vm_fault *vmf) 29944e047f89SShachar Raindel { 299589b15332SJohannes Weiner struct vm_area_struct *vma = vmf->vma; 29964e047f89SShachar Raindel struct address_space *mapping; 299789b15332SJohannes Weiner struct page *page = vmf->page; 299897ba0c2bSJan Kara bool dirtied; 299997ba0c2bSJan Kara bool page_mkwrite = vma->vm_ops && vma->vm_ops->page_mkwrite; 30004e047f89SShachar Raindel 30014e047f89SShachar Raindel dirtied = set_page_dirty(page); 30024e047f89SShachar Raindel VM_BUG_ON_PAGE(PageAnon(page), page); 300397ba0c2bSJan Kara /* 300497ba0c2bSJan Kara * Take a local copy of the address_space - page.mapping may be zeroed 300597ba0c2bSJan Kara * by truncate after unlock_page(). The address_space itself remains 300697ba0c2bSJan Kara * pinned by vma->vm_file's reference. We rely on unlock_page()'s 300797ba0c2bSJan Kara * release semantics to prevent the compiler from undoing this copying. 300897ba0c2bSJan Kara */ 300997ba0c2bSJan Kara mapping = page_rmapping(page); 30104e047f89SShachar Raindel unlock_page(page); 30114e047f89SShachar Raindel 30124e047f89SShachar Raindel if (!page_mkwrite) 30134e047f89SShachar Raindel file_update_time(vma->vm_file); 301489b15332SJohannes Weiner 301589b15332SJohannes Weiner /* 301689b15332SJohannes Weiner * Throttle page dirtying rate down to writeback speed. 301789b15332SJohannes Weiner * 301889b15332SJohannes Weiner * mapping may be NULL here because some device drivers do not 301989b15332SJohannes Weiner * set page.mapping but still dirty their pages 302089b15332SJohannes Weiner * 3021c1e8d7c6SMichel Lespinasse * Drop the mmap_lock before waiting on IO, if we can. The file 302289b15332SJohannes Weiner * is pinning the mapping, as per above. 302389b15332SJohannes Weiner */ 302489b15332SJohannes Weiner if ((dirtied || page_mkwrite) && mapping) { 302589b15332SJohannes Weiner struct file *fpin; 302689b15332SJohannes Weiner 302789b15332SJohannes Weiner fpin = maybe_unlock_mmap_for_io(vmf, NULL); 302889b15332SJohannes Weiner balance_dirty_pages_ratelimited(mapping); 302989b15332SJohannes Weiner if (fpin) { 303089b15332SJohannes Weiner fput(fpin); 3031d9272525SPeter Xu return VM_FAULT_COMPLETED; 303289b15332SJohannes Weiner } 303389b15332SJohannes Weiner } 303489b15332SJohannes Weiner 303589b15332SJohannes Weiner return 0; 30364e047f89SShachar Raindel } 30374e047f89SShachar Raindel 303897ba0c2bSJan Kara /* 30394e047f89SShachar Raindel * Handle write page faults for pages that can be reused in the current vma 30404e047f89SShachar Raindel * 30414e047f89SShachar Raindel * This can happen either due to the mapping being with the VM_SHARED flag, 30424e047f89SShachar Raindel * or due to us being the last reference standing to the page. In either 30434e047f89SShachar Raindel * case, all we need to do here is to mark the page as writable and update 30444e047f89SShachar Raindel * any related book-keeping. 30454e047f89SShachar Raindel */ 3046997dd98dSJan Kara static inline void wp_page_reuse(struct vm_fault *vmf) 304782b0f8c3SJan Kara __releases(vmf->ptl) 30484e047f89SShachar Raindel { 304982b0f8c3SJan Kara struct vm_area_struct *vma = vmf->vma; 3050a41b70d6SJan Kara struct page *page = vmf->page; 30514e047f89SShachar Raindel pte_t entry; 30526c287605SDavid Hildenbrand 3053c89357e2SDavid Hildenbrand VM_BUG_ON(!(vmf->flags & FAULT_FLAG_WRITE)); 30546c287605SDavid Hildenbrand VM_BUG_ON(PageAnon(page) && !PageAnonExclusive(page)); 30556c287605SDavid Hildenbrand 30564e047f89SShachar Raindel /* 30574e047f89SShachar Raindel * Clear the pages cpupid information as the existing 30584e047f89SShachar Raindel * information potentially belongs to a now completely 30594e047f89SShachar Raindel * unrelated process. 30604e047f89SShachar Raindel */ 30614e047f89SShachar Raindel if (page) 30624e047f89SShachar Raindel page_cpupid_xchg_last(page, (1 << LAST_CPUPID_SHIFT) - 1); 30634e047f89SShachar Raindel 30642994302bSJan Kara flush_cache_page(vma, vmf->address, pte_pfn(vmf->orig_pte)); 30652994302bSJan Kara entry = pte_mkyoung(vmf->orig_pte); 30664e047f89SShachar Raindel entry = maybe_mkwrite(pte_mkdirty(entry), vma); 306782b0f8c3SJan Kara if (ptep_set_access_flags(vma, vmf->address, vmf->pte, entry, 1)) 306882b0f8c3SJan Kara update_mmu_cache(vma, vmf->address, vmf->pte); 306982b0f8c3SJan Kara pte_unmap_unlock(vmf->pte, vmf->ptl); 3070798a6b87SPeter Xu count_vm_event(PGREUSE); 30714e047f89SShachar Raindel } 30724e047f89SShachar Raindel 30734e047f89SShachar Raindel /* 3074c89357e2SDavid Hildenbrand * Handle the case of a page which we actually need to copy to a new page, 3075c89357e2SDavid Hildenbrand * either due to COW or unsharing. 30762f38ab2cSShachar Raindel * 3077c1e8d7c6SMichel Lespinasse * Called with mmap_lock locked and the old page referenced, but 30782f38ab2cSShachar Raindel * without the ptl held. 30792f38ab2cSShachar Raindel * 30802f38ab2cSShachar Raindel * High level logic flow: 30812f38ab2cSShachar Raindel * 30822f38ab2cSShachar Raindel * - Allocate a page, copy the content of the old page to the new one. 30832f38ab2cSShachar Raindel * - Handle book keeping and accounting - cgroups, mmu-notifiers, etc. 30842f38ab2cSShachar Raindel * - Take the PTL. If the pte changed, bail out and release the allocated page 30852f38ab2cSShachar Raindel * - If the pte is still the way we remember it, update the page table and all 30862f38ab2cSShachar Raindel * relevant references. This includes dropping the reference the page-table 30872f38ab2cSShachar Raindel * held to the old page, as well as updating the rmap. 30882f38ab2cSShachar Raindel * - In any case, unlock the PTL and drop the reference we took to the old page. 30892f38ab2cSShachar Raindel */ 30902b740303SSouptick Joarder static vm_fault_t wp_page_copy(struct vm_fault *vmf) 30912f38ab2cSShachar Raindel { 3092c89357e2SDavid Hildenbrand const bool unshare = vmf->flags & FAULT_FLAG_UNSHARE; 309382b0f8c3SJan Kara struct vm_area_struct *vma = vmf->vma; 3094bae473a4SKirill A. Shutemov struct mm_struct *mm = vma->vm_mm; 3095a41b70d6SJan Kara struct page *old_page = vmf->page; 30962f38ab2cSShachar Raindel struct page *new_page = NULL; 30972f38ab2cSShachar Raindel pte_t entry; 30982f38ab2cSShachar Raindel int page_copied = 0; 3099ac46d4f3SJérôme Glisse struct mmu_notifier_range range; 31002f38ab2cSShachar Raindel 3101662ce1dcSYang Yang delayacct_wpcopy_start(); 3102662ce1dcSYang Yang 31032f38ab2cSShachar Raindel if (unlikely(anon_vma_prepare(vma))) 31042f38ab2cSShachar Raindel goto oom; 31052f38ab2cSShachar Raindel 31062994302bSJan Kara if (is_zero_pfn(pte_pfn(vmf->orig_pte))) { 310782b0f8c3SJan Kara new_page = alloc_zeroed_user_highpage_movable(vma, 310882b0f8c3SJan Kara vmf->address); 31092f38ab2cSShachar Raindel if (!new_page) 31102f38ab2cSShachar Raindel goto oom; 31112f38ab2cSShachar Raindel } else { 3112bae473a4SKirill A. Shutemov new_page = alloc_page_vma(GFP_HIGHUSER_MOVABLE, vma, 311382b0f8c3SJan Kara vmf->address); 31142f38ab2cSShachar Raindel if (!new_page) 31152f38ab2cSShachar Raindel goto oom; 311683d116c5SJia He 3117c89357e2SDavid Hildenbrand if (!__wp_page_copy_user(new_page, old_page, vmf)) { 311883d116c5SJia He /* 311983d116c5SJia He * COW failed, if the fault was solved by other, 312083d116c5SJia He * it's fine. If not, userspace would re-fault on 312183d116c5SJia He * the same address and we will handle the fault 312283d116c5SJia He * from the second attempt. 312383d116c5SJia He */ 312483d116c5SJia He put_page(new_page); 312583d116c5SJia He if (old_page) 312683d116c5SJia He put_page(old_page); 3127662ce1dcSYang Yang 3128662ce1dcSYang Yang delayacct_wpcopy_end(); 312983d116c5SJia He return 0; 313083d116c5SJia He } 31312f38ab2cSShachar Raindel } 31322f38ab2cSShachar Raindel 31338f425e4eSMatthew Wilcox (Oracle) if (mem_cgroup_charge(page_folio(new_page), mm, GFP_KERNEL)) 31342f38ab2cSShachar Raindel goto oom_free_new; 31359d82c694SJohannes Weiner cgroup_throttle_swaprate(new_page, GFP_KERNEL); 31362f38ab2cSShachar Raindel 3137eb3c24f3SMel Gorman __SetPageUptodate(new_page); 3138eb3c24f3SMel Gorman 31397269f999SJérôme Glisse mmu_notifier_range_init(&range, MMU_NOTIFY_CLEAR, 0, vma, mm, 31406f4f13e8SJérôme Glisse vmf->address & PAGE_MASK, 3141ac46d4f3SJérôme Glisse (vmf->address & PAGE_MASK) + PAGE_SIZE); 3142ac46d4f3SJérôme Glisse mmu_notifier_invalidate_range_start(&range); 31432f38ab2cSShachar Raindel 31442f38ab2cSShachar Raindel /* 31452f38ab2cSShachar Raindel * Re-check the pte - we dropped the lock 31462f38ab2cSShachar Raindel */ 314782b0f8c3SJan Kara vmf->pte = pte_offset_map_lock(mm, vmf->pmd, vmf->address, &vmf->ptl); 31482994302bSJan Kara if (likely(pte_same(*vmf->pte, vmf->orig_pte))) { 31492f38ab2cSShachar Raindel if (old_page) { 31502f38ab2cSShachar Raindel if (!PageAnon(old_page)) { 3151eca56ff9SJerome Marchand dec_mm_counter_fast(mm, 3152eca56ff9SJerome Marchand mm_counter_file(old_page)); 31532f38ab2cSShachar Raindel inc_mm_counter_fast(mm, MM_ANONPAGES); 31542f38ab2cSShachar Raindel } 31552f38ab2cSShachar Raindel } else { 31562f38ab2cSShachar Raindel inc_mm_counter_fast(mm, MM_ANONPAGES); 31572f38ab2cSShachar Raindel } 31582994302bSJan Kara flush_cache_page(vma, vmf->address, pte_pfn(vmf->orig_pte)); 31592f38ab2cSShachar Raindel entry = mk_pte(new_page, vma->vm_page_prot); 316050c25ee9SThomas Bogendoerfer entry = pte_sw_mkyoung(entry); 3161c89357e2SDavid Hildenbrand if (unlikely(unshare)) { 3162c89357e2SDavid Hildenbrand if (pte_soft_dirty(vmf->orig_pte)) 3163c89357e2SDavid Hildenbrand entry = pte_mksoft_dirty(entry); 3164c89357e2SDavid Hildenbrand if (pte_uffd_wp(vmf->orig_pte)) 3165c89357e2SDavid Hildenbrand entry = pte_mkuffd_wp(entry); 3166c89357e2SDavid Hildenbrand } else { 31672f38ab2cSShachar Raindel entry = maybe_mkwrite(pte_mkdirty(entry), vma); 3168c89357e2SDavid Hildenbrand } 3169111fe718SNicholas Piggin 31702f38ab2cSShachar Raindel /* 31712f38ab2cSShachar Raindel * Clear the pte entry and flush it first, before updating the 3172111fe718SNicholas Piggin * pte with the new entry, to keep TLBs on different CPUs in 3173111fe718SNicholas Piggin * sync. This code used to set the new PTE then flush TLBs, but 3174111fe718SNicholas Piggin * that left a window where the new PTE could be loaded into 3175111fe718SNicholas Piggin * some TLBs while the old PTE remains in others. 31762f38ab2cSShachar Raindel */ 317782b0f8c3SJan Kara ptep_clear_flush_notify(vma, vmf->address, vmf->pte); 317840f2bbf7SDavid Hildenbrand page_add_new_anon_rmap(new_page, vma, vmf->address); 3179b518154eSJoonsoo Kim lru_cache_add_inactive_or_unevictable(new_page, vma); 31802f38ab2cSShachar Raindel /* 31812f38ab2cSShachar Raindel * We call the notify macro here because, when using secondary 31822f38ab2cSShachar Raindel * mmu page tables (such as kvm shadow page tables), we want the 31832f38ab2cSShachar Raindel * new page to be mapped directly into the secondary page table. 31842f38ab2cSShachar Raindel */ 3185c89357e2SDavid Hildenbrand BUG_ON(unshare && pte_write(entry)); 318682b0f8c3SJan Kara set_pte_at_notify(mm, vmf->address, vmf->pte, entry); 318782b0f8c3SJan Kara update_mmu_cache(vma, vmf->address, vmf->pte); 31882f38ab2cSShachar Raindel if (old_page) { 31892f38ab2cSShachar Raindel /* 31902f38ab2cSShachar Raindel * Only after switching the pte to the new page may 31912f38ab2cSShachar Raindel * we remove the mapcount here. Otherwise another 31922f38ab2cSShachar Raindel * process may come and find the rmap count decremented 31932f38ab2cSShachar Raindel * before the pte is switched to the new page, and 31942f38ab2cSShachar Raindel * "reuse" the old page writing into it while our pte 31952f38ab2cSShachar Raindel * here still points into it and can be read by other 31962f38ab2cSShachar Raindel * threads. 31972f38ab2cSShachar Raindel * 31982f38ab2cSShachar Raindel * The critical issue is to order this 31992f38ab2cSShachar Raindel * page_remove_rmap with the ptp_clear_flush above. 32002f38ab2cSShachar Raindel * Those stores are ordered by (if nothing else,) 32012f38ab2cSShachar Raindel * the barrier present in the atomic_add_negative 32022f38ab2cSShachar Raindel * in page_remove_rmap. 32032f38ab2cSShachar Raindel * 32042f38ab2cSShachar Raindel * Then the TLB flush in ptep_clear_flush ensures that 32052f38ab2cSShachar Raindel * no process can access the old page before the 32062f38ab2cSShachar Raindel * decremented mapcount is visible. And the old page 32072f38ab2cSShachar Raindel * cannot be reused until after the decremented 32082f38ab2cSShachar Raindel * mapcount is visible. So transitively, TLBs to 32092f38ab2cSShachar Raindel * old page will be flushed before it can be reused. 32102f38ab2cSShachar Raindel */ 3211cea86fe2SHugh Dickins page_remove_rmap(old_page, vma, false); 32122f38ab2cSShachar Raindel } 32132f38ab2cSShachar Raindel 32142f38ab2cSShachar Raindel /* Free the old page.. */ 32152f38ab2cSShachar Raindel new_page = old_page; 32162f38ab2cSShachar Raindel page_copied = 1; 32172f38ab2cSShachar Raindel } else { 32187df67697SBibo Mao update_mmu_tlb(vma, vmf->address, vmf->pte); 32192f38ab2cSShachar Raindel } 32202f38ab2cSShachar Raindel 32212f38ab2cSShachar Raindel if (new_page) 322209cbfeafSKirill A. Shutemov put_page(new_page); 32232f38ab2cSShachar Raindel 322482b0f8c3SJan Kara pte_unmap_unlock(vmf->pte, vmf->ptl); 32254645b9feSJérôme Glisse /* 32264645b9feSJérôme Glisse * No need to double call mmu_notifier->invalidate_range() callback as 32274645b9feSJérôme Glisse * the above ptep_clear_flush_notify() did already call it. 32284645b9feSJérôme Glisse */ 3229ac46d4f3SJérôme Glisse mmu_notifier_invalidate_range_only_end(&range); 32302f38ab2cSShachar Raindel if (old_page) { 3231f4c4a3f4SHuang Ying if (page_copied) 3232f4c4a3f4SHuang Ying free_swap_cache(old_page); 323309cbfeafSKirill A. Shutemov put_page(old_page); 32342f38ab2cSShachar Raindel } 3235662ce1dcSYang Yang 3236662ce1dcSYang Yang delayacct_wpcopy_end(); 3237c89357e2SDavid Hildenbrand return (page_copied && !unshare) ? VM_FAULT_WRITE : 0; 32382f38ab2cSShachar Raindel oom_free_new: 323909cbfeafSKirill A. Shutemov put_page(new_page); 32402f38ab2cSShachar Raindel oom: 32412f38ab2cSShachar Raindel if (old_page) 324209cbfeafSKirill A. Shutemov put_page(old_page); 3243662ce1dcSYang Yang 3244662ce1dcSYang Yang delayacct_wpcopy_end(); 32452f38ab2cSShachar Raindel return VM_FAULT_OOM; 32462f38ab2cSShachar Raindel } 32472f38ab2cSShachar Raindel 324866a6197cSJan Kara /** 324966a6197cSJan Kara * finish_mkwrite_fault - finish page fault for a shared mapping, making PTE 325066a6197cSJan Kara * writeable once the page is prepared 325166a6197cSJan Kara * 325266a6197cSJan Kara * @vmf: structure describing the fault 325366a6197cSJan Kara * 325466a6197cSJan Kara * This function handles all that is needed to finish a write page fault in a 325566a6197cSJan Kara * shared mapping due to PTE being read-only once the mapped page is prepared. 3256a862f68aSMike Rapoport * It handles locking of PTE and modifying it. 325766a6197cSJan Kara * 325866a6197cSJan Kara * The function expects the page to be locked or other protection against 325966a6197cSJan Kara * concurrent faults / writeback (such as DAX radix tree locks). 3260a862f68aSMike Rapoport * 32612797e79fSLiu Xiang * Return: %0 on success, %VM_FAULT_NOPAGE when PTE got changed before 3262a862f68aSMike Rapoport * we acquired PTE lock. 326366a6197cSJan Kara */ 32642b740303SSouptick Joarder vm_fault_t finish_mkwrite_fault(struct vm_fault *vmf) 326566a6197cSJan Kara { 326666a6197cSJan Kara WARN_ON_ONCE(!(vmf->vma->vm_flags & VM_SHARED)); 326766a6197cSJan Kara vmf->pte = pte_offset_map_lock(vmf->vma->vm_mm, vmf->pmd, vmf->address, 326866a6197cSJan Kara &vmf->ptl); 326966a6197cSJan Kara /* 327066a6197cSJan Kara * We might have raced with another page fault while we released the 327166a6197cSJan Kara * pte_offset_map_lock. 327266a6197cSJan Kara */ 327366a6197cSJan Kara if (!pte_same(*vmf->pte, vmf->orig_pte)) { 32747df67697SBibo Mao update_mmu_tlb(vmf->vma, vmf->address, vmf->pte); 327566a6197cSJan Kara pte_unmap_unlock(vmf->pte, vmf->ptl); 3276a19e2553SJan Kara return VM_FAULT_NOPAGE; 327766a6197cSJan Kara } 327866a6197cSJan Kara wp_page_reuse(vmf); 3279a19e2553SJan Kara return 0; 328066a6197cSJan Kara } 328166a6197cSJan Kara 3282dd906184SBoaz Harrosh /* 3283dd906184SBoaz Harrosh * Handle write page faults for VM_MIXEDMAP or VM_PFNMAP for a VM_SHARED 3284dd906184SBoaz Harrosh * mapping 3285dd906184SBoaz Harrosh */ 32862b740303SSouptick Joarder static vm_fault_t wp_pfn_shared(struct vm_fault *vmf) 3287dd906184SBoaz Harrosh { 328882b0f8c3SJan Kara struct vm_area_struct *vma = vmf->vma; 3289bae473a4SKirill A. Shutemov 3290dd906184SBoaz Harrosh if (vma->vm_ops && vma->vm_ops->pfn_mkwrite) { 32912b740303SSouptick Joarder vm_fault_t ret; 3292dd906184SBoaz Harrosh 329382b0f8c3SJan Kara pte_unmap_unlock(vmf->pte, vmf->ptl); 3294fe82221fSJan Kara vmf->flags |= FAULT_FLAG_MKWRITE; 329511bac800SDave Jiang ret = vma->vm_ops->pfn_mkwrite(vmf); 32962f89dc12SJan Kara if (ret & (VM_FAULT_ERROR | VM_FAULT_NOPAGE)) 3297dd906184SBoaz Harrosh return ret; 329866a6197cSJan Kara return finish_mkwrite_fault(vmf); 3299dd906184SBoaz Harrosh } 3300997dd98dSJan Kara wp_page_reuse(vmf); 3301997dd98dSJan Kara return VM_FAULT_WRITE; 3302dd906184SBoaz Harrosh } 3303dd906184SBoaz Harrosh 33042b740303SSouptick Joarder static vm_fault_t wp_page_shared(struct vm_fault *vmf) 330582b0f8c3SJan Kara __releases(vmf->ptl) 330693e478d4SShachar Raindel { 330782b0f8c3SJan Kara struct vm_area_struct *vma = vmf->vma; 330889b15332SJohannes Weiner vm_fault_t ret = VM_FAULT_WRITE; 330993e478d4SShachar Raindel 3310a41b70d6SJan Kara get_page(vmf->page); 331193e478d4SShachar Raindel 331293e478d4SShachar Raindel if (vma->vm_ops && vma->vm_ops->page_mkwrite) { 33132b740303SSouptick Joarder vm_fault_t tmp; 331493e478d4SShachar Raindel 331582b0f8c3SJan Kara pte_unmap_unlock(vmf->pte, vmf->ptl); 331638b8cb7fSJan Kara tmp = do_page_mkwrite(vmf); 331793e478d4SShachar Raindel if (unlikely(!tmp || (tmp & 331893e478d4SShachar Raindel (VM_FAULT_ERROR | VM_FAULT_NOPAGE)))) { 3319a41b70d6SJan Kara put_page(vmf->page); 332093e478d4SShachar Raindel return tmp; 332193e478d4SShachar Raindel } 332266a6197cSJan Kara tmp = finish_mkwrite_fault(vmf); 3323a19e2553SJan Kara if (unlikely(tmp & (VM_FAULT_ERROR | VM_FAULT_NOPAGE))) { 3324a41b70d6SJan Kara unlock_page(vmf->page); 3325a41b70d6SJan Kara put_page(vmf->page); 332666a6197cSJan Kara return tmp; 332793e478d4SShachar Raindel } 332866a6197cSJan Kara } else { 3329997dd98dSJan Kara wp_page_reuse(vmf); 3330997dd98dSJan Kara lock_page(vmf->page); 333193e478d4SShachar Raindel } 333289b15332SJohannes Weiner ret |= fault_dirty_shared_page(vmf); 3333997dd98dSJan Kara put_page(vmf->page); 333493e478d4SShachar Raindel 333589b15332SJohannes Weiner return ret; 333693e478d4SShachar Raindel } 333793e478d4SShachar Raindel 33382f38ab2cSShachar Raindel /* 3339c89357e2SDavid Hildenbrand * This routine handles present pages, when 3340c89357e2SDavid Hildenbrand * * users try to write to a shared page (FAULT_FLAG_WRITE) 3341c89357e2SDavid Hildenbrand * * GUP wants to take a R/O pin on a possibly shared anonymous page 3342c89357e2SDavid Hildenbrand * (FAULT_FLAG_UNSHARE) 3343c89357e2SDavid Hildenbrand * 3344c89357e2SDavid Hildenbrand * It is done by copying the page to a new address and decrementing the 3345c89357e2SDavid Hildenbrand * shared-page counter for the old page. 33461da177e4SLinus Torvalds * 33471da177e4SLinus Torvalds * Note that this routine assumes that the protection checks have been 33481da177e4SLinus Torvalds * done by the caller (the low-level page fault routine in most cases). 3349c89357e2SDavid Hildenbrand * Thus, with FAULT_FLAG_WRITE, we can safely just mark it writable once we've 3350c89357e2SDavid Hildenbrand * done any necessary COW. 33511da177e4SLinus Torvalds * 3352c89357e2SDavid Hildenbrand * In case of FAULT_FLAG_WRITE, we also mark the page dirty at this point even 3353c89357e2SDavid Hildenbrand * though the page will change only once the write actually happens. This 3354c89357e2SDavid Hildenbrand * avoids a few races, and potentially makes it more efficient. 33551da177e4SLinus Torvalds * 3356c1e8d7c6SMichel Lespinasse * We enter with non-exclusive mmap_lock (to exclude vma changes, 33578f4e2101SHugh Dickins * but allow concurrent faults), with pte both mapped and locked. 3358c1e8d7c6SMichel Lespinasse * We return with mmap_lock still held, but pte unmapped and unlocked. 33591da177e4SLinus Torvalds */ 33602b740303SSouptick Joarder static vm_fault_t do_wp_page(struct vm_fault *vmf) 336182b0f8c3SJan Kara __releases(vmf->ptl) 33621da177e4SLinus Torvalds { 3363c89357e2SDavid Hildenbrand const bool unshare = vmf->flags & FAULT_FLAG_UNSHARE; 336482b0f8c3SJan Kara struct vm_area_struct *vma = vmf->vma; 33651da177e4SLinus Torvalds 3366c89357e2SDavid Hildenbrand VM_BUG_ON(unshare && (vmf->flags & FAULT_FLAG_WRITE)); 3367c89357e2SDavid Hildenbrand VM_BUG_ON(!unshare && !(vmf->flags & FAULT_FLAG_WRITE)); 3368c89357e2SDavid Hildenbrand 3369c89357e2SDavid Hildenbrand if (likely(!unshare)) { 3370292924b2SPeter Xu if (userfaultfd_pte_wp(vma, *vmf->pte)) { 3371529b930bSAndrea Arcangeli pte_unmap_unlock(vmf->pte, vmf->ptl); 3372529b930bSAndrea Arcangeli return handle_userfault(vmf, VM_UFFD_WP); 3373529b930bSAndrea Arcangeli } 3374529b930bSAndrea Arcangeli 33756ce64428SNadav Amit /* 33766ce64428SNadav Amit * Userfaultfd write-protect can defer flushes. Ensure the TLB 33776ce64428SNadav Amit * is flushed in this case before copying. 33786ce64428SNadav Amit */ 33796ce64428SNadav Amit if (unlikely(userfaultfd_wp(vmf->vma) && 33806ce64428SNadav Amit mm_tlb_flush_pending(vmf->vma->vm_mm))) 33816ce64428SNadav Amit flush_tlb_page(vmf->vma, vmf->address); 3382c89357e2SDavid Hildenbrand } 33836ce64428SNadav Amit 3384a41b70d6SJan Kara vmf->page = vm_normal_page(vma, vmf->address, vmf->orig_pte); 3385a41b70d6SJan Kara if (!vmf->page) { 3386c89357e2SDavid Hildenbrand if (unlikely(unshare)) { 3387c89357e2SDavid Hildenbrand /* No anonymous page -> nothing to do. */ 3388c89357e2SDavid Hildenbrand pte_unmap_unlock(vmf->pte, vmf->ptl); 3389c89357e2SDavid Hildenbrand return 0; 3390c89357e2SDavid Hildenbrand } 3391c89357e2SDavid Hildenbrand 3392251b97f5SPeter Zijlstra /* 339364e45507SPeter Feiner * VM_MIXEDMAP !pfn_valid() case, or VM_SOFTDIRTY clear on a 339464e45507SPeter Feiner * VM_PFNMAP VMA. 3395251b97f5SPeter Zijlstra * 3396251b97f5SPeter Zijlstra * We should not cow pages in a shared writeable mapping. 3397dd906184SBoaz Harrosh * Just mark the pages writable and/or call ops->pfn_mkwrite. 3398251b97f5SPeter Zijlstra */ 3399251b97f5SPeter Zijlstra if ((vma->vm_flags & (VM_WRITE|VM_SHARED)) == 3400251b97f5SPeter Zijlstra (VM_WRITE|VM_SHARED)) 34012994302bSJan Kara return wp_pfn_shared(vmf); 34022f38ab2cSShachar Raindel 340382b0f8c3SJan Kara pte_unmap_unlock(vmf->pte, vmf->ptl); 3404a41b70d6SJan Kara return wp_page_copy(vmf); 3405251b97f5SPeter Zijlstra } 34061da177e4SLinus Torvalds 3407d08b3851SPeter Zijlstra /* 3408ee6a6457SPeter Zijlstra * Take out anonymous pages first, anonymous shared vmas are 3409ee6a6457SPeter Zijlstra * not dirty accountable. 3410d08b3851SPeter Zijlstra */ 341152d1e606SKirill Tkhai if (PageAnon(vmf->page)) { 341209854ba9SLinus Torvalds struct page *page = vmf->page; 341309854ba9SLinus Torvalds 341453a05ad9SDavid Hildenbrand /* 34156c287605SDavid Hildenbrand * If the page is exclusive to this process we must reuse the 34166c287605SDavid Hildenbrand * page without further checks. 34176c287605SDavid Hildenbrand */ 34186c287605SDavid Hildenbrand if (PageAnonExclusive(page)) 34196c287605SDavid Hildenbrand goto reuse; 34206c287605SDavid Hildenbrand 34216c287605SDavid Hildenbrand /* 342253a05ad9SDavid Hildenbrand * We have to verify under page lock: these early checks are 342353a05ad9SDavid Hildenbrand * just an optimization to avoid locking the page and freeing 342453a05ad9SDavid Hildenbrand * the swapcache if there is little hope that we can reuse. 342553a05ad9SDavid Hildenbrand * 342653a05ad9SDavid Hildenbrand * PageKsm() doesn't necessarily raise the page refcount. 342753a05ad9SDavid Hildenbrand */ 3428d4c47097SDavid Hildenbrand if (PageKsm(page) || page_count(page) > 3) 3429d4c47097SDavid Hildenbrand goto copy; 3430d4c47097SDavid Hildenbrand if (!PageLRU(page)) 3431d4c47097SDavid Hildenbrand /* 3432d4c47097SDavid Hildenbrand * Note: We cannot easily detect+handle references from 3433d4c47097SDavid Hildenbrand * remote LRU pagevecs or references to PageLRU() pages. 3434d4c47097SDavid Hildenbrand */ 3435d4c47097SDavid Hildenbrand lru_add_drain(); 3436d4c47097SDavid Hildenbrand if (page_count(page) > 1 + PageSwapCache(page)) 343752d1e606SKirill Tkhai goto copy; 343809854ba9SLinus Torvalds if (!trylock_page(page)) 343952d1e606SKirill Tkhai goto copy; 344053a05ad9SDavid Hildenbrand if (PageSwapCache(page)) 344153a05ad9SDavid Hildenbrand try_to_free_swap(page); 344253a05ad9SDavid Hildenbrand if (PageKsm(page) || page_count(page) != 1) { 344309854ba9SLinus Torvalds unlock_page(page); 344409854ba9SLinus Torvalds goto copy; 344552d1e606SKirill Tkhai } 3446c44b6743SRik van Riel /* 344753a05ad9SDavid Hildenbrand * Ok, we've got the only page reference from our mapping 344853a05ad9SDavid Hildenbrand * and the page is locked, it's dark out, and we're wearing 344953a05ad9SDavid Hildenbrand * sunglasses. Hit it. 3450c44b6743SRik van Riel */ 34516c54dc6cSDavid Hildenbrand page_move_anon_rmap(page, vma); 345209854ba9SLinus Torvalds unlock_page(page); 34536c287605SDavid Hildenbrand reuse: 3454c89357e2SDavid Hildenbrand if (unlikely(unshare)) { 3455c89357e2SDavid Hildenbrand pte_unmap_unlock(vmf->pte, vmf->ptl); 3456c89357e2SDavid Hildenbrand return 0; 3457c89357e2SDavid Hildenbrand } 3458be068f29SLinus Torvalds wp_page_reuse(vmf); 3459997dd98dSJan Kara return VM_FAULT_WRITE; 3460c89357e2SDavid Hildenbrand } else if (unshare) { 3461c89357e2SDavid Hildenbrand /* No anonymous page -> nothing to do. */ 3462c89357e2SDavid Hildenbrand pte_unmap_unlock(vmf->pte, vmf->ptl); 3463c89357e2SDavid Hildenbrand return 0; 3464ee6a6457SPeter Zijlstra } else if (unlikely((vma->vm_flags & (VM_WRITE|VM_SHARED)) == 3465d08b3851SPeter Zijlstra (VM_WRITE|VM_SHARED))) { 3466a41b70d6SJan Kara return wp_page_shared(vmf); 34671da177e4SLinus Torvalds } 346852d1e606SKirill Tkhai copy: 34691da177e4SLinus Torvalds /* 34701da177e4SLinus Torvalds * Ok, we need to copy. Oh, well.. 34711da177e4SLinus Torvalds */ 3472a41b70d6SJan Kara get_page(vmf->page); 347328766805SShachar Raindel 347482b0f8c3SJan Kara pte_unmap_unlock(vmf->pte, vmf->ptl); 347594bfe85bSYang Yang #ifdef CONFIG_KSM 347694bfe85bSYang Yang if (PageKsm(vmf->page)) 347794bfe85bSYang Yang count_vm_event(COW_KSM); 347894bfe85bSYang Yang #endif 3479a41b70d6SJan Kara return wp_page_copy(vmf); 34801da177e4SLinus Torvalds } 34811da177e4SLinus Torvalds 348297a89413SPeter Zijlstra static void unmap_mapping_range_vma(struct vm_area_struct *vma, 34831da177e4SLinus Torvalds unsigned long start_addr, unsigned long end_addr, 34841da177e4SLinus Torvalds struct zap_details *details) 34851da177e4SLinus Torvalds { 3486f5cc4eefSAl Viro zap_page_range_single(vma, start_addr, end_addr - start_addr, details); 34871da177e4SLinus Torvalds } 34881da177e4SLinus Torvalds 3489f808c13fSDavidlohr Bueso static inline void unmap_mapping_range_tree(struct rb_root_cached *root, 3490232a6a1cSPeter Xu pgoff_t first_index, 3491232a6a1cSPeter Xu pgoff_t last_index, 34921da177e4SLinus Torvalds struct zap_details *details) 34931da177e4SLinus Torvalds { 34941da177e4SLinus Torvalds struct vm_area_struct *vma; 34951da177e4SLinus Torvalds pgoff_t vba, vea, zba, zea; 34961da177e4SLinus Torvalds 3497232a6a1cSPeter Xu vma_interval_tree_foreach(vma, root, first_index, last_index) { 34981da177e4SLinus Torvalds vba = vma->vm_pgoff; 3499d6e93217SLibin vea = vba + vma_pages(vma) - 1; 3500f9871da9SMiaohe Lin zba = max(first_index, vba); 3501f9871da9SMiaohe Lin zea = min(last_index, vea); 35021da177e4SLinus Torvalds 350397a89413SPeter Zijlstra unmap_mapping_range_vma(vma, 35041da177e4SLinus Torvalds ((zba - vba) << PAGE_SHIFT) + vma->vm_start, 35051da177e4SLinus Torvalds ((zea - vba + 1) << PAGE_SHIFT) + vma->vm_start, 350697a89413SPeter Zijlstra details); 35071da177e4SLinus Torvalds } 35081da177e4SLinus Torvalds } 35091da177e4SLinus Torvalds 35101da177e4SLinus Torvalds /** 35113506659eSMatthew Wilcox (Oracle) * unmap_mapping_folio() - Unmap single folio from processes. 35123506659eSMatthew Wilcox (Oracle) * @folio: The locked folio to be unmapped. 351322061a1fSHugh Dickins * 35143506659eSMatthew Wilcox (Oracle) * Unmap this folio from any userspace process which still has it mmaped. 351522061a1fSHugh Dickins * Typically, for efficiency, the range of nearby pages has already been 351622061a1fSHugh Dickins * unmapped by unmap_mapping_pages() or unmap_mapping_range(). But once 35173506659eSMatthew Wilcox (Oracle) * truncation or invalidation holds the lock on a folio, it may find that 35183506659eSMatthew Wilcox (Oracle) * the page has been remapped again: and then uses unmap_mapping_folio() 351922061a1fSHugh Dickins * to unmap it finally. 352022061a1fSHugh Dickins */ 35213506659eSMatthew Wilcox (Oracle) void unmap_mapping_folio(struct folio *folio) 352222061a1fSHugh Dickins { 35233506659eSMatthew Wilcox (Oracle) struct address_space *mapping = folio->mapping; 352422061a1fSHugh Dickins struct zap_details details = { }; 3525232a6a1cSPeter Xu pgoff_t first_index; 3526232a6a1cSPeter Xu pgoff_t last_index; 352722061a1fSHugh Dickins 35283506659eSMatthew Wilcox (Oracle) VM_BUG_ON(!folio_test_locked(folio)); 352922061a1fSHugh Dickins 35303506659eSMatthew Wilcox (Oracle) first_index = folio->index; 35313506659eSMatthew Wilcox (Oracle) last_index = folio->index + folio_nr_pages(folio) - 1; 3532232a6a1cSPeter Xu 35332e148f1eSPeter Xu details.even_cows = false; 35343506659eSMatthew Wilcox (Oracle) details.single_folio = folio; 3535999dad82SPeter Xu details.zap_flags = ZAP_FLAG_DROP_MARKER; 353622061a1fSHugh Dickins 35372c865995SHugh Dickins i_mmap_lock_read(mapping); 353822061a1fSHugh Dickins if (unlikely(!RB_EMPTY_ROOT(&mapping->i_mmap.rb_root))) 3539232a6a1cSPeter Xu unmap_mapping_range_tree(&mapping->i_mmap, first_index, 3540232a6a1cSPeter Xu last_index, &details); 35412c865995SHugh Dickins i_mmap_unlock_read(mapping); 354222061a1fSHugh Dickins } 354322061a1fSHugh Dickins 354422061a1fSHugh Dickins /** 3545977fbdcdSMatthew Wilcox * unmap_mapping_pages() - Unmap pages from processes. 3546977fbdcdSMatthew Wilcox * @mapping: The address space containing pages to be unmapped. 3547977fbdcdSMatthew Wilcox * @start: Index of first page to be unmapped. 3548977fbdcdSMatthew Wilcox * @nr: Number of pages to be unmapped. 0 to unmap to end of file. 3549977fbdcdSMatthew Wilcox * @even_cows: Whether to unmap even private COWed pages. 3550977fbdcdSMatthew Wilcox * 3551977fbdcdSMatthew Wilcox * Unmap the pages in this address space from any userspace process which 3552977fbdcdSMatthew Wilcox * has them mmaped. Generally, you want to remove COWed pages as well when 3553977fbdcdSMatthew Wilcox * a file is being truncated, but not when invalidating pages from the page 3554977fbdcdSMatthew Wilcox * cache. 3555977fbdcdSMatthew Wilcox */ 3556977fbdcdSMatthew Wilcox void unmap_mapping_pages(struct address_space *mapping, pgoff_t start, 3557977fbdcdSMatthew Wilcox pgoff_t nr, bool even_cows) 3558977fbdcdSMatthew Wilcox { 3559977fbdcdSMatthew Wilcox struct zap_details details = { }; 3560232a6a1cSPeter Xu pgoff_t first_index = start; 3561232a6a1cSPeter Xu pgoff_t last_index = start + nr - 1; 3562977fbdcdSMatthew Wilcox 35632e148f1eSPeter Xu details.even_cows = even_cows; 3564232a6a1cSPeter Xu if (last_index < first_index) 3565232a6a1cSPeter Xu last_index = ULONG_MAX; 3566977fbdcdSMatthew Wilcox 35672c865995SHugh Dickins i_mmap_lock_read(mapping); 3568977fbdcdSMatthew Wilcox if (unlikely(!RB_EMPTY_ROOT(&mapping->i_mmap.rb_root))) 3569232a6a1cSPeter Xu unmap_mapping_range_tree(&mapping->i_mmap, first_index, 3570232a6a1cSPeter Xu last_index, &details); 35712c865995SHugh Dickins i_mmap_unlock_read(mapping); 3572977fbdcdSMatthew Wilcox } 35736e0e99d5SDavid Howells EXPORT_SYMBOL_GPL(unmap_mapping_pages); 3574977fbdcdSMatthew Wilcox 3575977fbdcdSMatthew Wilcox /** 35768a5f14a2SKirill A. Shutemov * unmap_mapping_range - unmap the portion of all mmaps in the specified 3577977fbdcdSMatthew Wilcox * address_space corresponding to the specified byte range in the underlying 35788a5f14a2SKirill A. Shutemov * file. 35798a5f14a2SKirill A. Shutemov * 35803d41088fSMartin Waitz * @mapping: the address space containing mmaps to be unmapped. 35811da177e4SLinus Torvalds * @holebegin: byte in first page to unmap, relative to the start of 35821da177e4SLinus Torvalds * the underlying file. This will be rounded down to a PAGE_SIZE 358325d9e2d1Snpiggin@suse.de * boundary. Note that this is different from truncate_pagecache(), which 35841da177e4SLinus Torvalds * must keep the partial page. In contrast, we must get rid of 35851da177e4SLinus Torvalds * partial pages. 35861da177e4SLinus Torvalds * @holelen: size of prospective hole in bytes. This will be rounded 35871da177e4SLinus Torvalds * up to a PAGE_SIZE boundary. A holelen of zero truncates to the 35881da177e4SLinus Torvalds * end of the file. 35891da177e4SLinus Torvalds * @even_cows: 1 when truncating a file, unmap even private COWed pages; 35901da177e4SLinus Torvalds * but 0 when invalidating pagecache, don't throw away private data. 35911da177e4SLinus Torvalds */ 35921da177e4SLinus Torvalds void unmap_mapping_range(struct address_space *mapping, 35931da177e4SLinus Torvalds loff_t const holebegin, loff_t const holelen, int even_cows) 35941da177e4SLinus Torvalds { 35951da177e4SLinus Torvalds pgoff_t hba = holebegin >> PAGE_SHIFT; 35961da177e4SLinus Torvalds pgoff_t hlen = (holelen + PAGE_SIZE - 1) >> PAGE_SHIFT; 35971da177e4SLinus Torvalds 35981da177e4SLinus Torvalds /* Check for overflow. */ 35991da177e4SLinus Torvalds if (sizeof(holelen) > sizeof(hlen)) { 36001da177e4SLinus Torvalds long long holeend = 36011da177e4SLinus Torvalds (holebegin + holelen + PAGE_SIZE - 1) >> PAGE_SHIFT; 36021da177e4SLinus Torvalds if (holeend & ~(long long)ULONG_MAX) 36031da177e4SLinus Torvalds hlen = ULONG_MAX - hba + 1; 36041da177e4SLinus Torvalds } 36051da177e4SLinus Torvalds 3606977fbdcdSMatthew Wilcox unmap_mapping_pages(mapping, hba, hlen, even_cows); 36071da177e4SLinus Torvalds } 36081da177e4SLinus Torvalds EXPORT_SYMBOL(unmap_mapping_range); 36091da177e4SLinus Torvalds 36101da177e4SLinus Torvalds /* 3611b756a3b5SAlistair Popple * Restore a potential device exclusive pte to a working pte entry 3612b756a3b5SAlistair Popple */ 3613b756a3b5SAlistair Popple static vm_fault_t remove_device_exclusive_entry(struct vm_fault *vmf) 3614b756a3b5SAlistair Popple { 3615b756a3b5SAlistair Popple struct page *page = vmf->page; 3616b756a3b5SAlistair Popple struct vm_area_struct *vma = vmf->vma; 3617b756a3b5SAlistair Popple struct mmu_notifier_range range; 3618b756a3b5SAlistair Popple 3619b756a3b5SAlistair Popple if (!lock_page_or_retry(page, vma->vm_mm, vmf->flags)) 3620b756a3b5SAlistair Popple return VM_FAULT_RETRY; 3621b756a3b5SAlistair Popple mmu_notifier_range_init_owner(&range, MMU_NOTIFY_EXCLUSIVE, 0, vma, 3622b756a3b5SAlistair Popple vma->vm_mm, vmf->address & PAGE_MASK, 3623b756a3b5SAlistair Popple (vmf->address & PAGE_MASK) + PAGE_SIZE, NULL); 3624b756a3b5SAlistair Popple mmu_notifier_invalidate_range_start(&range); 3625b756a3b5SAlistair Popple 3626b756a3b5SAlistair Popple vmf->pte = pte_offset_map_lock(vma->vm_mm, vmf->pmd, vmf->address, 3627b756a3b5SAlistair Popple &vmf->ptl); 3628b756a3b5SAlistair Popple if (likely(pte_same(*vmf->pte, vmf->orig_pte))) 3629b756a3b5SAlistair Popple restore_exclusive_pte(vma, page, vmf->address, vmf->pte); 3630b756a3b5SAlistair Popple 3631b756a3b5SAlistair Popple pte_unmap_unlock(vmf->pte, vmf->ptl); 3632b756a3b5SAlistair Popple unlock_page(page); 3633b756a3b5SAlistair Popple 3634b756a3b5SAlistair Popple mmu_notifier_invalidate_range_end(&range); 3635b756a3b5SAlistair Popple return 0; 3636b756a3b5SAlistair Popple } 3637b756a3b5SAlistair Popple 3638c145e0b4SDavid Hildenbrand static inline bool should_try_to_free_swap(struct page *page, 3639c145e0b4SDavid Hildenbrand struct vm_area_struct *vma, 3640c145e0b4SDavid Hildenbrand unsigned int fault_flags) 3641c145e0b4SDavid Hildenbrand { 3642c145e0b4SDavid Hildenbrand if (!PageSwapCache(page)) 3643c145e0b4SDavid Hildenbrand return false; 3644c145e0b4SDavid Hildenbrand if (mem_cgroup_swap_full(page) || (vma->vm_flags & VM_LOCKED) || 3645c145e0b4SDavid Hildenbrand PageMlocked(page)) 3646c145e0b4SDavid Hildenbrand return true; 3647c145e0b4SDavid Hildenbrand /* 3648c145e0b4SDavid Hildenbrand * If we want to map a page that's in the swapcache writable, we 3649c145e0b4SDavid Hildenbrand * have to detect via the refcount if we're really the exclusive 3650c145e0b4SDavid Hildenbrand * user. Try freeing the swapcache to get rid of the swapcache 3651c145e0b4SDavid Hildenbrand * reference only in case it's likely that we'll be the exlusive user. 3652c145e0b4SDavid Hildenbrand */ 3653c145e0b4SDavid Hildenbrand return (fault_flags & FAULT_FLAG_WRITE) && !PageKsm(page) && 3654c145e0b4SDavid Hildenbrand page_count(page) == 2; 3655c145e0b4SDavid Hildenbrand } 3656c145e0b4SDavid Hildenbrand 36579c28a205SPeter Xu static vm_fault_t pte_marker_clear(struct vm_fault *vmf) 36589c28a205SPeter Xu { 36599c28a205SPeter Xu vmf->pte = pte_offset_map_lock(vmf->vma->vm_mm, vmf->pmd, 36609c28a205SPeter Xu vmf->address, &vmf->ptl); 36619c28a205SPeter Xu /* 36629c28a205SPeter Xu * Be careful so that we will only recover a special uffd-wp pte into a 36639c28a205SPeter Xu * none pte. Otherwise it means the pte could have changed, so retry. 36649c28a205SPeter Xu */ 36659c28a205SPeter Xu if (is_pte_marker(*vmf->pte)) 36669c28a205SPeter Xu pte_clear(vmf->vma->vm_mm, vmf->address, vmf->pte); 36679c28a205SPeter Xu pte_unmap_unlock(vmf->pte, vmf->ptl); 36689c28a205SPeter Xu return 0; 36699c28a205SPeter Xu } 36709c28a205SPeter Xu 36719c28a205SPeter Xu /* 36729c28a205SPeter Xu * This is actually a page-missing access, but with uffd-wp special pte 36739c28a205SPeter Xu * installed. It means this pte was wr-protected before being unmapped. 36749c28a205SPeter Xu */ 36759c28a205SPeter Xu static vm_fault_t pte_marker_handle_uffd_wp(struct vm_fault *vmf) 36769c28a205SPeter Xu { 36779c28a205SPeter Xu /* 36789c28a205SPeter Xu * Just in case there're leftover special ptes even after the region 36799c28a205SPeter Xu * got unregistered - we can simply clear them. We can also do that 36809c28a205SPeter Xu * proactively when e.g. when we do UFFDIO_UNREGISTER upon some uffd-wp 36819c28a205SPeter Xu * ranges, but it should be more efficient to be done lazily here. 36829c28a205SPeter Xu */ 36839c28a205SPeter Xu if (unlikely(!userfaultfd_wp(vmf->vma) || vma_is_anonymous(vmf->vma))) 36849c28a205SPeter Xu return pte_marker_clear(vmf); 36859c28a205SPeter Xu 36869c28a205SPeter Xu /* do_fault() can handle pte markers too like none pte */ 36879c28a205SPeter Xu return do_fault(vmf); 36889c28a205SPeter Xu } 36899c28a205SPeter Xu 36905c041f5dSPeter Xu static vm_fault_t handle_pte_marker(struct vm_fault *vmf) 36915c041f5dSPeter Xu { 36925c041f5dSPeter Xu swp_entry_t entry = pte_to_swp_entry(vmf->orig_pte); 36935c041f5dSPeter Xu unsigned long marker = pte_marker_get(entry); 36945c041f5dSPeter Xu 36955c041f5dSPeter Xu /* 36965c041f5dSPeter Xu * PTE markers should always be with file-backed memories, and the 36975c041f5dSPeter Xu * marker should never be empty. If anything weird happened, the best 36985c041f5dSPeter Xu * thing to do is to kill the process along with its mm. 36995c041f5dSPeter Xu */ 37005c041f5dSPeter Xu if (WARN_ON_ONCE(vma_is_anonymous(vmf->vma) || !marker)) 37015c041f5dSPeter Xu return VM_FAULT_SIGBUS; 37025c041f5dSPeter Xu 37039c28a205SPeter Xu if (pte_marker_entry_uffd_wp(entry)) 37049c28a205SPeter Xu return pte_marker_handle_uffd_wp(vmf); 37059c28a205SPeter Xu 37069c28a205SPeter Xu /* This is an unknown pte marker */ 37079c28a205SPeter Xu return VM_FAULT_SIGBUS; 37085c041f5dSPeter Xu } 37095c041f5dSPeter Xu 3710b756a3b5SAlistair Popple /* 3711c1e8d7c6SMichel Lespinasse * We enter with non-exclusive mmap_lock (to exclude vma changes, 37128f4e2101SHugh Dickins * but allow concurrent faults), and pte mapped but not yet locked. 37139a95f3cfSPaul Cassella * We return with pte unmapped and unlocked. 37149a95f3cfSPaul Cassella * 3715c1e8d7c6SMichel Lespinasse * We return with the mmap_lock locked or unlocked in the same cases 37169a95f3cfSPaul Cassella * as does filemap_fault(). 37171da177e4SLinus Torvalds */ 37182b740303SSouptick Joarder vm_fault_t do_swap_page(struct vm_fault *vmf) 37191da177e4SLinus Torvalds { 372082b0f8c3SJan Kara struct vm_area_struct *vma = vmf->vma; 3721eaf649ebSMinchan Kim struct page *page = NULL, *swapcache; 37222799e775SMiaohe Lin struct swap_info_struct *si = NULL; 372314f9135dSDavid Hildenbrand rmap_t rmap_flags = RMAP_NONE; 37241493a191SDavid Hildenbrand bool exclusive = false; 372565500d23SHugh Dickins swp_entry_t entry; 37261da177e4SLinus Torvalds pte_t pte; 3727d065bd81SMichel Lespinasse int locked; 37282b740303SSouptick Joarder vm_fault_t ret = 0; 3729aae466b0SJoonsoo Kim void *shadow = NULL; 37301da177e4SLinus Torvalds 37312ca99358SPeter Xu if (!pte_unmap_same(vmf)) 37328f4e2101SHugh Dickins goto out; 373365500d23SHugh Dickins 37342994302bSJan Kara entry = pte_to_swp_entry(vmf->orig_pte); 3735d1737fdbSAndi Kleen if (unlikely(non_swap_entry(entry))) { 37360697212aSChristoph Lameter if (is_migration_entry(entry)) { 373782b0f8c3SJan Kara migration_entry_wait(vma->vm_mm, vmf->pmd, 373882b0f8c3SJan Kara vmf->address); 3739b756a3b5SAlistair Popple } else if (is_device_exclusive_entry(entry)) { 3740b756a3b5SAlistair Popple vmf->page = pfn_swap_entry_to_page(entry); 3741b756a3b5SAlistair Popple ret = remove_device_exclusive_entry(vmf); 37425042db43SJérôme Glisse } else if (is_device_private_entry(entry)) { 3743af5cdaf8SAlistair Popple vmf->page = pfn_swap_entry_to_page(entry); 3744897e6365SChristoph Hellwig ret = vmf->page->pgmap->ops->migrate_to_ram(vmf); 3745d1737fdbSAndi Kleen } else if (is_hwpoison_entry(entry)) { 3746d1737fdbSAndi Kleen ret = VM_FAULT_HWPOISON; 37479f186f9eSMiaohe Lin } else if (is_swapin_error_entry(entry)) { 37489f186f9eSMiaohe Lin ret = VM_FAULT_SIGBUS; 37495c041f5dSPeter Xu } else if (is_pte_marker_entry(entry)) { 37505c041f5dSPeter Xu ret = handle_pte_marker(vmf); 3751d1737fdbSAndi Kleen } else { 37522994302bSJan Kara print_bad_pte(vma, vmf->address, vmf->orig_pte, NULL); 3753d99be1a8SHugh Dickins ret = VM_FAULT_SIGBUS; 3754d1737fdbSAndi Kleen } 37550697212aSChristoph Lameter goto out; 37560697212aSChristoph Lameter } 37570bcac06fSMinchan Kim 37582799e775SMiaohe Lin /* Prevent swapoff from happening to us. */ 37592799e775SMiaohe Lin si = get_swap_device(entry); 37602799e775SMiaohe Lin if (unlikely(!si)) 37612799e775SMiaohe Lin goto out; 37620bcac06fSMinchan Kim 3763eaf649ebSMinchan Kim page = lookup_swap_cache(entry, vma, vmf->address); 3764f8020772SMinchan Kim swapcache = page; 3765f8020772SMinchan Kim 37661da177e4SLinus Torvalds if (!page) { 3767a449bf58SQian Cai if (data_race(si->flags & SWP_SYNCHRONOUS_IO) && 3768eb085574SHuang Ying __swap_count(entry) == 1) { 37690bcac06fSMinchan Kim /* skip swapcache */ 3770e9e9b7ecSMinchan Kim page = alloc_page_vma(GFP_HIGHUSER_MOVABLE, vma, 3771e9e9b7ecSMinchan Kim vmf->address); 37720bcac06fSMinchan Kim if (page) { 37730bcac06fSMinchan Kim __SetPageLocked(page); 37740bcac06fSMinchan Kim __SetPageSwapBacked(page); 37754c6355b2SJohannes Weiner 37760add0c77SShakeel Butt if (mem_cgroup_swapin_charge_page(page, 37770add0c77SShakeel Butt vma->vm_mm, GFP_KERNEL, entry)) { 3778545b1b07SMichal Hocko ret = VM_FAULT_OOM; 37794c6355b2SJohannes Weiner goto out_page; 3780545b1b07SMichal Hocko } 37810add0c77SShakeel Butt mem_cgroup_swapin_uncharge_swap(entry); 37824c6355b2SJohannes Weiner 3783aae466b0SJoonsoo Kim shadow = get_shadow_from_swap_cache(entry); 3784aae466b0SJoonsoo Kim if (shadow) 37850995d7e5SMatthew Wilcox (Oracle) workingset_refault(page_folio(page), 37860995d7e5SMatthew Wilcox (Oracle) shadow); 37870076f029SJoonsoo Kim 37886058eaecSJohannes Weiner lru_cache_add(page); 37890add0c77SShakeel Butt 37900add0c77SShakeel Butt /* To provide entry to swap_readpage() */ 37910add0c77SShakeel Butt set_page_private(page, entry.val); 37925169b844SNeilBrown swap_readpage(page, true, NULL); 37930add0c77SShakeel Butt set_page_private(page, 0); 37940bcac06fSMinchan Kim } 3795aa8d22a1SMinchan Kim } else { 3796e9e9b7ecSMinchan Kim page = swapin_readahead(entry, GFP_HIGHUSER_MOVABLE, 3797e9e9b7ecSMinchan Kim vmf); 3798aa8d22a1SMinchan Kim swapcache = page; 37990bcac06fSMinchan Kim } 38000bcac06fSMinchan Kim 3801ec560175SHuang Ying if (!page) { 38021da177e4SLinus Torvalds /* 38038f4e2101SHugh Dickins * Back out if somebody else faulted in this pte 38048f4e2101SHugh Dickins * while we released the pte lock. 38051da177e4SLinus Torvalds */ 380682b0f8c3SJan Kara vmf->pte = pte_offset_map_lock(vma->vm_mm, vmf->pmd, 380782b0f8c3SJan Kara vmf->address, &vmf->ptl); 38082994302bSJan Kara if (likely(pte_same(*vmf->pte, vmf->orig_pte))) 38091da177e4SLinus Torvalds ret = VM_FAULT_OOM; 381065500d23SHugh Dickins goto unlock; 38111da177e4SLinus Torvalds } 38121da177e4SLinus Torvalds 38131da177e4SLinus Torvalds /* Had to read the page from swap area: Major fault */ 38141da177e4SLinus Torvalds ret = VM_FAULT_MAJOR; 3815f8891e5eSChristoph Lameter count_vm_event(PGMAJFAULT); 38162262185cSRoman Gushchin count_memcg_event_mm(vma->vm_mm, PGMAJFAULT); 3817d1737fdbSAndi Kleen } else if (PageHWPoison(page)) { 381871f72525SWu Fengguang /* 381971f72525SWu Fengguang * hwpoisoned dirty swapcache pages are kept for killing 382071f72525SWu Fengguang * owner processes (which may be unknown at hwpoison time) 382171f72525SWu Fengguang */ 3822d1737fdbSAndi Kleen ret = VM_FAULT_HWPOISON; 38234779cb31SAndi Kleen goto out_release; 38241da177e4SLinus Torvalds } 38251da177e4SLinus Torvalds 382682b0f8c3SJan Kara locked = lock_page_or_retry(page, vma->vm_mm, vmf->flags); 3827e709ffd6SRik van Riel 3828d065bd81SMichel Lespinasse if (!locked) { 3829d065bd81SMichel Lespinasse ret |= VM_FAULT_RETRY; 3830d065bd81SMichel Lespinasse goto out_release; 3831d065bd81SMichel Lespinasse } 38321da177e4SLinus Torvalds 383384d60fddSDavid Hildenbrand if (swapcache) { 38344969c119SAndrea Arcangeli /* 383584d60fddSDavid Hildenbrand * Make sure try_to_free_swap or swapoff did not release the 383684d60fddSDavid Hildenbrand * swapcache from under us. The page pin, and pte_same test 383784d60fddSDavid Hildenbrand * below, are not enough to exclude that. Even if it is still 383884d60fddSDavid Hildenbrand * swapcache, we need to check that the page's swap has not 383984d60fddSDavid Hildenbrand * changed. 38404969c119SAndrea Arcangeli */ 384184d60fddSDavid Hildenbrand if (unlikely(!PageSwapCache(page) || 384284d60fddSDavid Hildenbrand page_private(page) != entry.val)) 38434969c119SAndrea Arcangeli goto out_page; 38444969c119SAndrea Arcangeli 384584d60fddSDavid Hildenbrand /* 384684d60fddSDavid Hildenbrand * KSM sometimes has to copy on read faults, for example, if 384784d60fddSDavid Hildenbrand * page->index of !PageKSM() pages would be nonlinear inside the 384884d60fddSDavid Hildenbrand * anon VMA -- PageKSM() is lost on actual swapout. 384984d60fddSDavid Hildenbrand */ 385082b0f8c3SJan Kara page = ksm_might_need_to_copy(page, vma, vmf->address); 38514969c119SAndrea Arcangeli if (unlikely(!page)) { 38525ad64688SHugh Dickins ret = VM_FAULT_OOM; 38534969c119SAndrea Arcangeli page = swapcache; 38544969c119SAndrea Arcangeli goto out_page; 38554969c119SAndrea Arcangeli } 3856c145e0b4SDavid Hildenbrand 3857c145e0b4SDavid Hildenbrand /* 3858c145e0b4SDavid Hildenbrand * If we want to map a page that's in the swapcache writable, we 3859c145e0b4SDavid Hildenbrand * have to detect via the refcount if we're really the exclusive 3860c145e0b4SDavid Hildenbrand * owner. Try removing the extra reference from the local LRU 3861c145e0b4SDavid Hildenbrand * pagevecs if required. 3862c145e0b4SDavid Hildenbrand */ 3863c145e0b4SDavid Hildenbrand if ((vmf->flags & FAULT_FLAG_WRITE) && page == swapcache && 3864c145e0b4SDavid Hildenbrand !PageKsm(page) && !PageLRU(page)) 3865c145e0b4SDavid Hildenbrand lru_add_drain(); 386684d60fddSDavid Hildenbrand } 38675ad64688SHugh Dickins 38689d82c694SJohannes Weiner cgroup_throttle_swaprate(page, GFP_KERNEL); 3869073e587eSKAMEZAWA Hiroyuki 38701da177e4SLinus Torvalds /* 38718f4e2101SHugh Dickins * Back out if somebody else already faulted in this pte. 38721da177e4SLinus Torvalds */ 387382b0f8c3SJan Kara vmf->pte = pte_offset_map_lock(vma->vm_mm, vmf->pmd, vmf->address, 387482b0f8c3SJan Kara &vmf->ptl); 38752994302bSJan Kara if (unlikely(!pte_same(*vmf->pte, vmf->orig_pte))) 3876b8107480SKirill Korotaev goto out_nomap; 3877b8107480SKirill Korotaev 3878b8107480SKirill Korotaev if (unlikely(!PageUptodate(page))) { 3879b8107480SKirill Korotaev ret = VM_FAULT_SIGBUS; 3880b8107480SKirill Korotaev goto out_nomap; 38811da177e4SLinus Torvalds } 38821da177e4SLinus Torvalds 38838c7c6e34SKAMEZAWA Hiroyuki /* 388478fbe906SDavid Hildenbrand * PG_anon_exclusive reuses PG_mappedtodisk for anon pages. A swap pte 388578fbe906SDavid Hildenbrand * must never point at an anonymous page in the swapcache that is 388678fbe906SDavid Hildenbrand * PG_anon_exclusive. Sanity check that this holds and especially, that 388778fbe906SDavid Hildenbrand * no filesystem set PG_mappedtodisk on a page in the swapcache. Sanity 388878fbe906SDavid Hildenbrand * check after taking the PT lock and making sure that nobody 388978fbe906SDavid Hildenbrand * concurrently faulted in this page and set PG_anon_exclusive. 389078fbe906SDavid Hildenbrand */ 389178fbe906SDavid Hildenbrand BUG_ON(!PageAnon(page) && PageMappedToDisk(page)); 389278fbe906SDavid Hildenbrand BUG_ON(PageAnon(page) && PageAnonExclusive(page)); 389378fbe906SDavid Hildenbrand 389478fbe906SDavid Hildenbrand /* 38951493a191SDavid Hildenbrand * Check under PT lock (to protect against concurrent fork() sharing 38961493a191SDavid Hildenbrand * the swap entry concurrently) for certainly exclusive pages. 38971493a191SDavid Hildenbrand */ 38981493a191SDavid Hildenbrand if (!PageKsm(page)) { 38991493a191SDavid Hildenbrand /* 39001493a191SDavid Hildenbrand * Note that pte_swp_exclusive() == false for architectures 39011493a191SDavid Hildenbrand * without __HAVE_ARCH_PTE_SWP_EXCLUSIVE. 39021493a191SDavid Hildenbrand */ 39031493a191SDavid Hildenbrand exclusive = pte_swp_exclusive(vmf->orig_pte); 39041493a191SDavid Hildenbrand if (page != swapcache) { 39051493a191SDavid Hildenbrand /* 39061493a191SDavid Hildenbrand * We have a fresh page that is not exposed to the 39071493a191SDavid Hildenbrand * swapcache -> certainly exclusive. 39081493a191SDavid Hildenbrand */ 39091493a191SDavid Hildenbrand exclusive = true; 39101493a191SDavid Hildenbrand } else if (exclusive && PageWriteback(page) && 3911eacde327SMiaohe Lin data_race(si->flags & SWP_STABLE_WRITES)) { 39121493a191SDavid Hildenbrand /* 39131493a191SDavid Hildenbrand * This is tricky: not all swap backends support 39141493a191SDavid Hildenbrand * concurrent page modifications while under writeback. 39151493a191SDavid Hildenbrand * 39161493a191SDavid Hildenbrand * So if we stumble over such a page in the swapcache 39171493a191SDavid Hildenbrand * we must not set the page exclusive, otherwise we can 39181493a191SDavid Hildenbrand * map it writable without further checks and modify it 39191493a191SDavid Hildenbrand * while still under writeback. 39201493a191SDavid Hildenbrand * 39211493a191SDavid Hildenbrand * For these problematic swap backends, simply drop the 39221493a191SDavid Hildenbrand * exclusive marker: this is perfectly fine as we start 39231493a191SDavid Hildenbrand * writeback only if we fully unmapped the page and 39241493a191SDavid Hildenbrand * there are no unexpected references on the page after 39251493a191SDavid Hildenbrand * unmapping succeeded. After fully unmapped, no 39261493a191SDavid Hildenbrand * further GUP references (FOLL_GET and FOLL_PIN) can 39271493a191SDavid Hildenbrand * appear, so dropping the exclusive marker and mapping 39281493a191SDavid Hildenbrand * it only R/O is fine. 39291493a191SDavid Hildenbrand */ 39301493a191SDavid Hildenbrand exclusive = false; 39311493a191SDavid Hildenbrand } 39321493a191SDavid Hildenbrand } 39331493a191SDavid Hildenbrand 39341493a191SDavid Hildenbrand /* 3935c145e0b4SDavid Hildenbrand * Remove the swap entry and conditionally try to free up the swapcache. 3936c145e0b4SDavid Hildenbrand * We're already holding a reference on the page but haven't mapped it 3937c145e0b4SDavid Hildenbrand * yet. 39388c7c6e34SKAMEZAWA Hiroyuki */ 3939c145e0b4SDavid Hildenbrand swap_free(entry); 3940c145e0b4SDavid Hildenbrand if (should_try_to_free_swap(page, vma, vmf->flags)) 3941c145e0b4SDavid Hildenbrand try_to_free_swap(page); 39421da177e4SLinus Torvalds 3943bae473a4SKirill A. Shutemov inc_mm_counter_fast(vma->vm_mm, MM_ANONPAGES); 3944bae473a4SKirill A. Shutemov dec_mm_counter_fast(vma->vm_mm, MM_SWAPENTS); 39451da177e4SLinus Torvalds pte = mk_pte(page, vma->vm_page_prot); 3946c145e0b4SDavid Hildenbrand 3947c145e0b4SDavid Hildenbrand /* 39481493a191SDavid Hildenbrand * Same logic as in do_wp_page(); however, optimize for pages that are 39491493a191SDavid Hildenbrand * certainly not shared either because we just allocated them without 39501493a191SDavid Hildenbrand * exposing them to the swapcache or because the swap entry indicates 39511493a191SDavid Hildenbrand * exclusivity. 3952c145e0b4SDavid Hildenbrand */ 39531493a191SDavid Hildenbrand if (!PageKsm(page) && (exclusive || page_count(page) == 1)) { 39546c287605SDavid Hildenbrand if (vmf->flags & FAULT_FLAG_WRITE) { 39551da177e4SLinus Torvalds pte = maybe_mkwrite(pte_mkdirty(pte), vma); 395682b0f8c3SJan Kara vmf->flags &= ~FAULT_FLAG_WRITE; 39579a5b489bSAndrea Arcangeli ret |= VM_FAULT_WRITE; 39586c287605SDavid Hildenbrand } 395914f9135dSDavid Hildenbrand rmap_flags |= RMAP_EXCLUSIVE; 39601da177e4SLinus Torvalds } 39611da177e4SLinus Torvalds flush_icache_page(vma, page); 39622994302bSJan Kara if (pte_swp_soft_dirty(vmf->orig_pte)) 3963179ef71cSCyrill Gorcunov pte = pte_mksoft_dirty(pte); 3964f45ec5ffSPeter Xu if (pte_swp_uffd_wp(vmf->orig_pte)) { 3965f45ec5ffSPeter Xu pte = pte_mkuffd_wp(pte); 3966f45ec5ffSPeter Xu pte = pte_wrprotect(pte); 3967f45ec5ffSPeter Xu } 39682994302bSJan Kara vmf->orig_pte = pte; 39690bcac06fSMinchan Kim 39700bcac06fSMinchan Kim /* ksm created a completely new copy */ 39710bcac06fSMinchan Kim if (unlikely(page != swapcache && swapcache)) { 397240f2bbf7SDavid Hildenbrand page_add_new_anon_rmap(page, vma, vmf->address); 3973b518154eSJoonsoo Kim lru_cache_add_inactive_or_unevictable(page, vma); 39740bcac06fSMinchan Kim } else { 3975f1e2db12SDavid Hildenbrand page_add_anon_rmap(page, vma, vmf->address, rmap_flags); 397600501b53SJohannes Weiner } 39771da177e4SLinus Torvalds 39786c287605SDavid Hildenbrand VM_BUG_ON(!PageAnon(page) || (pte_write(pte) && !PageAnonExclusive(page))); 39791eba86c0SPasha Tatashin set_pte_at(vma->vm_mm, vmf->address, vmf->pte, pte); 39801eba86c0SPasha Tatashin arch_do_swap_page(vma->vm_mm, vma, vmf->address, pte, vmf->orig_pte); 39811eba86c0SPasha Tatashin 3982c475a8abSHugh Dickins unlock_page(page); 39830bcac06fSMinchan Kim if (page != swapcache && swapcache) { 39844969c119SAndrea Arcangeli /* 39854969c119SAndrea Arcangeli * Hold the lock to avoid the swap entry to be reused 39864969c119SAndrea Arcangeli * until we take the PT lock for the pte_same() check 39874969c119SAndrea Arcangeli * (to avoid false positives from pte_same). For 39884969c119SAndrea Arcangeli * further safety release the lock after the swap_free 39894969c119SAndrea Arcangeli * so that the swap count won't change under a 39904969c119SAndrea Arcangeli * parallel locked swapcache. 39914969c119SAndrea Arcangeli */ 39924969c119SAndrea Arcangeli unlock_page(swapcache); 399309cbfeafSKirill A. Shutemov put_page(swapcache); 39944969c119SAndrea Arcangeli } 3995c475a8abSHugh Dickins 399682b0f8c3SJan Kara if (vmf->flags & FAULT_FLAG_WRITE) { 39972994302bSJan Kara ret |= do_wp_page(vmf); 399861469f1dSHugh Dickins if (ret & VM_FAULT_ERROR) 399961469f1dSHugh Dickins ret &= VM_FAULT_ERROR; 40001da177e4SLinus Torvalds goto out; 40011da177e4SLinus Torvalds } 40021da177e4SLinus Torvalds 40031da177e4SLinus Torvalds /* No need to invalidate - it was non-present before */ 400482b0f8c3SJan Kara update_mmu_cache(vma, vmf->address, vmf->pte); 400565500d23SHugh Dickins unlock: 400682b0f8c3SJan Kara pte_unmap_unlock(vmf->pte, vmf->ptl); 40071da177e4SLinus Torvalds out: 40082799e775SMiaohe Lin if (si) 40092799e775SMiaohe Lin put_swap_device(si); 40101da177e4SLinus Torvalds return ret; 4011b8107480SKirill Korotaev out_nomap: 401282b0f8c3SJan Kara pte_unmap_unlock(vmf->pte, vmf->ptl); 4013bc43f75cSJohannes Weiner out_page: 4014b8107480SKirill Korotaev unlock_page(page); 40154779cb31SAndi Kleen out_release: 401609cbfeafSKirill A. Shutemov put_page(page); 40170bcac06fSMinchan Kim if (page != swapcache && swapcache) { 40184969c119SAndrea Arcangeli unlock_page(swapcache); 401909cbfeafSKirill A. Shutemov put_page(swapcache); 40204969c119SAndrea Arcangeli } 40212799e775SMiaohe Lin if (si) 40222799e775SMiaohe Lin put_swap_device(si); 402365500d23SHugh Dickins return ret; 40241da177e4SLinus Torvalds } 40251da177e4SLinus Torvalds 40261da177e4SLinus Torvalds /* 4027c1e8d7c6SMichel Lespinasse * We enter with non-exclusive mmap_lock (to exclude vma changes, 40288f4e2101SHugh Dickins * but allow concurrent faults), and pte mapped but not yet locked. 4029c1e8d7c6SMichel Lespinasse * We return with mmap_lock still held, but pte unmapped and unlocked. 40301da177e4SLinus Torvalds */ 40312b740303SSouptick Joarder static vm_fault_t do_anonymous_page(struct vm_fault *vmf) 40321da177e4SLinus Torvalds { 403382b0f8c3SJan Kara struct vm_area_struct *vma = vmf->vma; 40348f4e2101SHugh Dickins struct page *page; 40352b740303SSouptick Joarder vm_fault_t ret = 0; 40361da177e4SLinus Torvalds pte_t entry; 40371da177e4SLinus Torvalds 40386b7339f4SKirill A. Shutemov /* File mapping without ->vm_ops ? */ 40396b7339f4SKirill A. Shutemov if (vma->vm_flags & VM_SHARED) 40406b7339f4SKirill A. Shutemov return VM_FAULT_SIGBUS; 40416b7339f4SKirill A. Shutemov 40427267ec00SKirill A. Shutemov /* 40437267ec00SKirill A. Shutemov * Use pte_alloc() instead of pte_alloc_map(). We can't run 40447267ec00SKirill A. Shutemov * pte_offset_map() on pmds where a huge pmd might be created 40457267ec00SKirill A. Shutemov * from a different thread. 40467267ec00SKirill A. Shutemov * 40473e4e28c5SMichel Lespinasse * pte_alloc_map() is safe to use under mmap_write_lock(mm) or when 40487267ec00SKirill A. Shutemov * parallel threads are excluded by other means. 40497267ec00SKirill A. Shutemov * 40503e4e28c5SMichel Lespinasse * Here we only have mmap_read_lock(mm). 40517267ec00SKirill A. Shutemov */ 40524cf58924SJoel Fernandes (Google) if (pte_alloc(vma->vm_mm, vmf->pmd)) 40537267ec00SKirill A. Shutemov return VM_FAULT_OOM; 40547267ec00SKirill A. Shutemov 4055f9ce0be7SKirill A. Shutemov /* See comment in handle_pte_fault() */ 405682b0f8c3SJan Kara if (unlikely(pmd_trans_unstable(vmf->pmd))) 40577267ec00SKirill A. Shutemov return 0; 40587267ec00SKirill A. Shutemov 405911ac5524SLinus Torvalds /* Use the zero-page for reads */ 406082b0f8c3SJan Kara if (!(vmf->flags & FAULT_FLAG_WRITE) && 4061bae473a4SKirill A. Shutemov !mm_forbids_zeropage(vma->vm_mm)) { 406282b0f8c3SJan Kara entry = pte_mkspecial(pfn_pte(my_zero_pfn(vmf->address), 406362eede62SHugh Dickins vma->vm_page_prot)); 406482b0f8c3SJan Kara vmf->pte = pte_offset_map_lock(vma->vm_mm, vmf->pmd, 406582b0f8c3SJan Kara vmf->address, &vmf->ptl); 40667df67697SBibo Mao if (!pte_none(*vmf->pte)) { 40677df67697SBibo Mao update_mmu_tlb(vma, vmf->address, vmf->pte); 4068a13ea5b7SHugh Dickins goto unlock; 40697df67697SBibo Mao } 40706b31d595SMichal Hocko ret = check_stable_address_space(vma->vm_mm); 40716b31d595SMichal Hocko if (ret) 40726b31d595SMichal Hocko goto unlock; 40736b251fc9SAndrea Arcangeli /* Deliver the page fault to userland, check inside PT lock */ 40746b251fc9SAndrea Arcangeli if (userfaultfd_missing(vma)) { 407582b0f8c3SJan Kara pte_unmap_unlock(vmf->pte, vmf->ptl); 407682b0f8c3SJan Kara return handle_userfault(vmf, VM_UFFD_MISSING); 40776b251fc9SAndrea Arcangeli } 4078a13ea5b7SHugh Dickins goto setpte; 4079a13ea5b7SHugh Dickins } 4080a13ea5b7SHugh Dickins 40811da177e4SLinus Torvalds /* Allocate our own private page. */ 40821da177e4SLinus Torvalds if (unlikely(anon_vma_prepare(vma))) 408365500d23SHugh Dickins goto oom; 408482b0f8c3SJan Kara page = alloc_zeroed_user_highpage_movable(vma, vmf->address); 40851da177e4SLinus Torvalds if (!page) 408665500d23SHugh Dickins goto oom; 4087eb3c24f3SMel Gorman 40888f425e4eSMatthew Wilcox (Oracle) if (mem_cgroup_charge(page_folio(page), vma->vm_mm, GFP_KERNEL)) 4089eb3c24f3SMel Gorman goto oom_free_page; 40909d82c694SJohannes Weiner cgroup_throttle_swaprate(page, GFP_KERNEL); 4091eb3c24f3SMel Gorman 409252f37629SMinchan Kim /* 409352f37629SMinchan Kim * The memory barrier inside __SetPageUptodate makes sure that 4094f4f5329dSWei Yang * preceding stores to the page contents become visible before 409552f37629SMinchan Kim * the set_pte_at() write. 409652f37629SMinchan Kim */ 40970ed361deSNick Piggin __SetPageUptodate(page); 40981da177e4SLinus Torvalds 409965500d23SHugh Dickins entry = mk_pte(page, vma->vm_page_prot); 410050c25ee9SThomas Bogendoerfer entry = pte_sw_mkyoung(entry); 41011ac0cb5dSHugh Dickins if (vma->vm_flags & VM_WRITE) 41021ac0cb5dSHugh Dickins entry = pte_mkwrite(pte_mkdirty(entry)); 41038f4e2101SHugh Dickins 410482b0f8c3SJan Kara vmf->pte = pte_offset_map_lock(vma->vm_mm, vmf->pmd, vmf->address, 410582b0f8c3SJan Kara &vmf->ptl); 41067df67697SBibo Mao if (!pte_none(*vmf->pte)) { 41077df67697SBibo Mao update_mmu_cache(vma, vmf->address, vmf->pte); 41088f4e2101SHugh Dickins goto release; 41097df67697SBibo Mao } 41109ba69294SHugh Dickins 41116b31d595SMichal Hocko ret = check_stable_address_space(vma->vm_mm); 41126b31d595SMichal Hocko if (ret) 41136b31d595SMichal Hocko goto release; 41146b31d595SMichal Hocko 41156b251fc9SAndrea Arcangeli /* Deliver the page fault to userland, check inside PT lock */ 41166b251fc9SAndrea Arcangeli if (userfaultfd_missing(vma)) { 411782b0f8c3SJan Kara pte_unmap_unlock(vmf->pte, vmf->ptl); 411809cbfeafSKirill A. Shutemov put_page(page); 411982b0f8c3SJan Kara return handle_userfault(vmf, VM_UFFD_MISSING); 41206b251fc9SAndrea Arcangeli } 41216b251fc9SAndrea Arcangeli 4122bae473a4SKirill A. Shutemov inc_mm_counter_fast(vma->vm_mm, MM_ANONPAGES); 412340f2bbf7SDavid Hildenbrand page_add_new_anon_rmap(page, vma, vmf->address); 4124b518154eSJoonsoo Kim lru_cache_add_inactive_or_unevictable(page, vma); 4125a13ea5b7SHugh Dickins setpte: 412682b0f8c3SJan Kara set_pte_at(vma->vm_mm, vmf->address, vmf->pte, entry); 41271da177e4SLinus Torvalds 41281da177e4SLinus Torvalds /* No need to invalidate - it was non-present before */ 412982b0f8c3SJan Kara update_mmu_cache(vma, vmf->address, vmf->pte); 413065500d23SHugh Dickins unlock: 413182b0f8c3SJan Kara pte_unmap_unlock(vmf->pte, vmf->ptl); 41326b31d595SMichal Hocko return ret; 41338f4e2101SHugh Dickins release: 413409cbfeafSKirill A. Shutemov put_page(page); 41358f4e2101SHugh Dickins goto unlock; 41368a9f3ccdSBalbir Singh oom_free_page: 413709cbfeafSKirill A. Shutemov put_page(page); 413865500d23SHugh Dickins oom: 41391da177e4SLinus Torvalds return VM_FAULT_OOM; 41401da177e4SLinus Torvalds } 41411da177e4SLinus Torvalds 41429a95f3cfSPaul Cassella /* 4143c1e8d7c6SMichel Lespinasse * The mmap_lock must have been held on entry, and may have been 41449a95f3cfSPaul Cassella * released depending on flags and vma->vm_ops->fault() return value. 41459a95f3cfSPaul Cassella * See filemap_fault() and __lock_page_retry(). 41469a95f3cfSPaul Cassella */ 41472b740303SSouptick Joarder static vm_fault_t __do_fault(struct vm_fault *vmf) 41487eae74afSKirill A. Shutemov { 414982b0f8c3SJan Kara struct vm_area_struct *vma = vmf->vma; 41502b740303SSouptick Joarder vm_fault_t ret; 41517eae74afSKirill A. Shutemov 415263f3655fSMichal Hocko /* 415363f3655fSMichal Hocko * Preallocate pte before we take page_lock because this might lead to 415463f3655fSMichal Hocko * deadlocks for memcg reclaim which waits for pages under writeback: 415563f3655fSMichal Hocko * lock_page(A) 415663f3655fSMichal Hocko * SetPageWriteback(A) 415763f3655fSMichal Hocko * unlock_page(A) 415863f3655fSMichal Hocko * lock_page(B) 415963f3655fSMichal Hocko * lock_page(B) 4160d383807aSYanfei Xu * pte_alloc_one 416163f3655fSMichal Hocko * shrink_page_list 416263f3655fSMichal Hocko * wait_on_page_writeback(A) 416363f3655fSMichal Hocko * SetPageWriteback(B) 416463f3655fSMichal Hocko * unlock_page(B) 416563f3655fSMichal Hocko * # flush A, B to clear the writeback 416663f3655fSMichal Hocko */ 416763f3655fSMichal Hocko if (pmd_none(*vmf->pmd) && !vmf->prealloc_pte) { 4168a7069ee3SYanfei Xu vmf->prealloc_pte = pte_alloc_one(vma->vm_mm); 416963f3655fSMichal Hocko if (!vmf->prealloc_pte) 417063f3655fSMichal Hocko return VM_FAULT_OOM; 417163f3655fSMichal Hocko } 417263f3655fSMichal Hocko 417311bac800SDave Jiang ret = vma->vm_ops->fault(vmf); 41743917048dSJan Kara if (unlikely(ret & (VM_FAULT_ERROR | VM_FAULT_NOPAGE | VM_FAULT_RETRY | 4175b1aa812bSJan Kara VM_FAULT_DONE_COW))) 41767eae74afSKirill A. Shutemov return ret; 41777eae74afSKirill A. Shutemov 4178667240e0SJan Kara if (unlikely(PageHWPoison(vmf->page))) { 41793149c79fSRik van Riel struct page *page = vmf->page; 4180e53ac737SRik van Riel vm_fault_t poisonret = VM_FAULT_HWPOISON; 4181e53ac737SRik van Riel if (ret & VM_FAULT_LOCKED) { 41823149c79fSRik van Riel if (page_mapped(page)) 41833149c79fSRik van Riel unmap_mapping_pages(page_mapping(page), 41843149c79fSRik van Riel page->index, 1, false); 4185e53ac737SRik van Riel /* Retry if a clean page was removed from the cache. */ 41863149c79fSRik van Riel if (invalidate_inode_page(page)) 41873149c79fSRik van Riel poisonret = VM_FAULT_NOPAGE; 41883149c79fSRik van Riel unlock_page(page); 4189e53ac737SRik van Riel } 41903149c79fSRik van Riel put_page(page); 4191936ca80dSJan Kara vmf->page = NULL; 4192e53ac737SRik van Riel return poisonret; 41937eae74afSKirill A. Shutemov } 41947eae74afSKirill A. Shutemov 41957eae74afSKirill A. Shutemov if (unlikely(!(ret & VM_FAULT_LOCKED))) 4196667240e0SJan Kara lock_page(vmf->page); 41977eae74afSKirill A. Shutemov else 4198667240e0SJan Kara VM_BUG_ON_PAGE(!PageLocked(vmf->page), vmf->page); 41997eae74afSKirill A. Shutemov 42007eae74afSKirill A. Shutemov return ret; 42017eae74afSKirill A. Shutemov } 42027eae74afSKirill A. Shutemov 4203396bcc52SMatthew Wilcox (Oracle) #ifdef CONFIG_TRANSPARENT_HUGEPAGE 420482b0f8c3SJan Kara static void deposit_prealloc_pte(struct vm_fault *vmf) 4205953c66c2SAneesh Kumar K.V { 420682b0f8c3SJan Kara struct vm_area_struct *vma = vmf->vma; 4207953c66c2SAneesh Kumar K.V 420882b0f8c3SJan Kara pgtable_trans_huge_deposit(vma->vm_mm, vmf->pmd, vmf->prealloc_pte); 4209953c66c2SAneesh Kumar K.V /* 4210953c66c2SAneesh Kumar K.V * We are going to consume the prealloc table, 4211953c66c2SAneesh Kumar K.V * count that as nr_ptes. 4212953c66c2SAneesh Kumar K.V */ 4213c4812909SKirill A. Shutemov mm_inc_nr_ptes(vma->vm_mm); 42147f2b6ce8STobin C Harding vmf->prealloc_pte = NULL; 4215953c66c2SAneesh Kumar K.V } 4216953c66c2SAneesh Kumar K.V 4217f9ce0be7SKirill A. Shutemov vm_fault_t do_set_pmd(struct vm_fault *vmf, struct page *page) 421810102459SKirill A. Shutemov { 421982b0f8c3SJan Kara struct vm_area_struct *vma = vmf->vma; 422082b0f8c3SJan Kara bool write = vmf->flags & FAULT_FLAG_WRITE; 422182b0f8c3SJan Kara unsigned long haddr = vmf->address & HPAGE_PMD_MASK; 422210102459SKirill A. Shutemov pmd_t entry; 42232b740303SSouptick Joarder int i; 4224d01ac3c3SMatthew Wilcox (Oracle) vm_fault_t ret = VM_FAULT_FALLBACK; 422510102459SKirill A. Shutemov 422610102459SKirill A. Shutemov if (!transhuge_vma_suitable(vma, haddr)) 4227d01ac3c3SMatthew Wilcox (Oracle) return ret; 422810102459SKirill A. Shutemov 422910102459SKirill A. Shutemov page = compound_head(page); 4230d01ac3c3SMatthew Wilcox (Oracle) if (compound_order(page) != HPAGE_PMD_ORDER) 4231d01ac3c3SMatthew Wilcox (Oracle) return ret; 423210102459SKirill A. Shutemov 4233953c66c2SAneesh Kumar K.V /* 4234eac96c3eSYang Shi * Just backoff if any subpage of a THP is corrupted otherwise 4235eac96c3eSYang Shi * the corrupted page may mapped by PMD silently to escape the 4236eac96c3eSYang Shi * check. This kind of THP just can be PTE mapped. Access to 4237eac96c3eSYang Shi * the corrupted subpage should trigger SIGBUS as expected. 4238eac96c3eSYang Shi */ 4239eac96c3eSYang Shi if (unlikely(PageHasHWPoisoned(page))) 4240eac96c3eSYang Shi return ret; 4241eac96c3eSYang Shi 4242eac96c3eSYang Shi /* 4243f0953a1bSIngo Molnar * Archs like ppc64 need additional space to store information 4244953c66c2SAneesh Kumar K.V * related to pte entry. Use the preallocated table for that. 4245953c66c2SAneesh Kumar K.V */ 424682b0f8c3SJan Kara if (arch_needs_pgtable_deposit() && !vmf->prealloc_pte) { 42474cf58924SJoel Fernandes (Google) vmf->prealloc_pte = pte_alloc_one(vma->vm_mm); 424882b0f8c3SJan Kara if (!vmf->prealloc_pte) 4249953c66c2SAneesh Kumar K.V return VM_FAULT_OOM; 4250953c66c2SAneesh Kumar K.V } 4251953c66c2SAneesh Kumar K.V 425282b0f8c3SJan Kara vmf->ptl = pmd_lock(vma->vm_mm, vmf->pmd); 425382b0f8c3SJan Kara if (unlikely(!pmd_none(*vmf->pmd))) 425410102459SKirill A. Shutemov goto out; 425510102459SKirill A. Shutemov 425610102459SKirill A. Shutemov for (i = 0; i < HPAGE_PMD_NR; i++) 425710102459SKirill A. Shutemov flush_icache_page(vma, page + i); 425810102459SKirill A. Shutemov 425910102459SKirill A. Shutemov entry = mk_huge_pmd(page, vma->vm_page_prot); 426010102459SKirill A. Shutemov if (write) 4261f55e1014SLinus Torvalds entry = maybe_pmd_mkwrite(pmd_mkdirty(entry), vma); 426210102459SKirill A. Shutemov 4263fadae295SYang Shi add_mm_counter(vma->vm_mm, mm_counter_file(page), HPAGE_PMD_NR); 4264cea86fe2SHugh Dickins page_add_file_rmap(page, vma, true); 4265cea86fe2SHugh Dickins 4266953c66c2SAneesh Kumar K.V /* 4267953c66c2SAneesh Kumar K.V * deposit and withdraw with pmd lock held 4268953c66c2SAneesh Kumar K.V */ 4269953c66c2SAneesh Kumar K.V if (arch_needs_pgtable_deposit()) 427082b0f8c3SJan Kara deposit_prealloc_pte(vmf); 427110102459SKirill A. Shutemov 427282b0f8c3SJan Kara set_pmd_at(vma->vm_mm, haddr, vmf->pmd, entry); 427310102459SKirill A. Shutemov 427482b0f8c3SJan Kara update_mmu_cache_pmd(vma, haddr, vmf->pmd); 427510102459SKirill A. Shutemov 427610102459SKirill A. Shutemov /* fault is handled */ 427710102459SKirill A. Shutemov ret = 0; 427895ecedcdSKirill A. Shutemov count_vm_event(THP_FILE_MAPPED); 427910102459SKirill A. Shutemov out: 428082b0f8c3SJan Kara spin_unlock(vmf->ptl); 428110102459SKirill A. Shutemov return ret; 428210102459SKirill A. Shutemov } 428310102459SKirill A. Shutemov #else 4284f9ce0be7SKirill A. Shutemov vm_fault_t do_set_pmd(struct vm_fault *vmf, struct page *page) 428510102459SKirill A. Shutemov { 4286f9ce0be7SKirill A. Shutemov return VM_FAULT_FALLBACK; 428710102459SKirill A. Shutemov } 428810102459SKirill A. Shutemov #endif 428910102459SKirill A. Shutemov 42909d3af4b4SWill Deacon void do_set_pte(struct vm_fault *vmf, struct page *page, unsigned long addr) 42913bb97794SKirill A. Shutemov { 429282b0f8c3SJan Kara struct vm_area_struct *vma = vmf->vma; 42939c28a205SPeter Xu bool uffd_wp = pte_marker_uffd_wp(vmf->orig_pte); 429482b0f8c3SJan Kara bool write = vmf->flags & FAULT_FLAG_WRITE; 42959d3af4b4SWill Deacon bool prefault = vmf->address != addr; 42963bb97794SKirill A. Shutemov pte_t entry; 42977267ec00SKirill A. Shutemov 42983bb97794SKirill A. Shutemov flush_icache_page(vma, page); 42993bb97794SKirill A. Shutemov entry = mk_pte(page, vma->vm_page_prot); 430046bdb427SWill Deacon 430146bdb427SWill Deacon if (prefault && arch_wants_old_prefaulted_pte()) 430246bdb427SWill Deacon entry = pte_mkold(entry); 430350c25ee9SThomas Bogendoerfer else 430450c25ee9SThomas Bogendoerfer entry = pte_sw_mkyoung(entry); 430546bdb427SWill Deacon 43063bb97794SKirill A. Shutemov if (write) 43073bb97794SKirill A. Shutemov entry = maybe_mkwrite(pte_mkdirty(entry), vma); 43089c28a205SPeter Xu if (unlikely(uffd_wp)) 43099c28a205SPeter Xu entry = pte_mkuffd_wp(pte_wrprotect(entry)); 4310bae473a4SKirill A. Shutemov /* copy-on-write page */ 4311bae473a4SKirill A. Shutemov if (write && !(vma->vm_flags & VM_SHARED)) { 43123bb97794SKirill A. Shutemov inc_mm_counter_fast(vma->vm_mm, MM_ANONPAGES); 431340f2bbf7SDavid Hildenbrand page_add_new_anon_rmap(page, vma, addr); 4314b518154eSJoonsoo Kim lru_cache_add_inactive_or_unevictable(page, vma); 43153bb97794SKirill A. Shutemov } else { 4316eca56ff9SJerome Marchand inc_mm_counter_fast(vma->vm_mm, mm_counter_file(page)); 4317cea86fe2SHugh Dickins page_add_file_rmap(page, vma, false); 43183bb97794SKirill A. Shutemov } 43199d3af4b4SWill Deacon set_pte_at(vma->vm_mm, addr, vmf->pte, entry); 43203bb97794SKirill A. Shutemov } 43213bb97794SKirill A. Shutemov 4322f46f2adeSPeter Xu static bool vmf_pte_changed(struct vm_fault *vmf) 4323f46f2adeSPeter Xu { 4324f46f2adeSPeter Xu if (vmf->flags & FAULT_FLAG_ORIG_PTE_VALID) 4325f46f2adeSPeter Xu return !pte_same(*vmf->pte, vmf->orig_pte); 4326f46f2adeSPeter Xu 4327f46f2adeSPeter Xu return !pte_none(*vmf->pte); 4328f46f2adeSPeter Xu } 4329f46f2adeSPeter Xu 43309118c0cbSJan Kara /** 43319118c0cbSJan Kara * finish_fault - finish page fault once we have prepared the page to fault 43329118c0cbSJan Kara * 43339118c0cbSJan Kara * @vmf: structure describing the fault 43349118c0cbSJan Kara * 43359118c0cbSJan Kara * This function handles all that is needed to finish a page fault once the 43369118c0cbSJan Kara * page to fault in is prepared. It handles locking of PTEs, inserts PTE for 43379118c0cbSJan Kara * given page, adds reverse page mapping, handles memcg charges and LRU 4338a862f68aSMike Rapoport * addition. 43399118c0cbSJan Kara * 43409118c0cbSJan Kara * The function expects the page to be locked and on success it consumes a 43419118c0cbSJan Kara * reference of a page being mapped (for the PTE which maps it). 4342a862f68aSMike Rapoport * 4343a862f68aSMike Rapoport * Return: %0 on success, %VM_FAULT_ code in case of error. 43449118c0cbSJan Kara */ 43452b740303SSouptick Joarder vm_fault_t finish_fault(struct vm_fault *vmf) 43469118c0cbSJan Kara { 4347f9ce0be7SKirill A. Shutemov struct vm_area_struct *vma = vmf->vma; 43489118c0cbSJan Kara struct page *page; 4349f9ce0be7SKirill A. Shutemov vm_fault_t ret; 43509118c0cbSJan Kara 43519118c0cbSJan Kara /* Did we COW the page? */ 4352f9ce0be7SKirill A. Shutemov if ((vmf->flags & FAULT_FLAG_WRITE) && !(vma->vm_flags & VM_SHARED)) 43539118c0cbSJan Kara page = vmf->cow_page; 43549118c0cbSJan Kara else 43559118c0cbSJan Kara page = vmf->page; 43566b31d595SMichal Hocko 43576b31d595SMichal Hocko /* 43586b31d595SMichal Hocko * check even for read faults because we might have lost our CoWed 43596b31d595SMichal Hocko * page 43606b31d595SMichal Hocko */ 4361f9ce0be7SKirill A. Shutemov if (!(vma->vm_flags & VM_SHARED)) { 4362f9ce0be7SKirill A. Shutemov ret = check_stable_address_space(vma->vm_mm); 4363f9ce0be7SKirill A. Shutemov if (ret) 4364f9ce0be7SKirill A. Shutemov return ret; 4365f9ce0be7SKirill A. Shutemov } 4366f9ce0be7SKirill A. Shutemov 4367f9ce0be7SKirill A. Shutemov if (pmd_none(*vmf->pmd)) { 4368f9ce0be7SKirill A. Shutemov if (PageTransCompound(page)) { 4369f9ce0be7SKirill A. Shutemov ret = do_set_pmd(vmf, page); 4370f9ce0be7SKirill A. Shutemov if (ret != VM_FAULT_FALLBACK) 4371f9ce0be7SKirill A. Shutemov return ret; 4372f9ce0be7SKirill A. Shutemov } 4373f9ce0be7SKirill A. Shutemov 437403c4f204SQi Zheng if (vmf->prealloc_pte) 437503c4f204SQi Zheng pmd_install(vma->vm_mm, vmf->pmd, &vmf->prealloc_pte); 437603c4f204SQi Zheng else if (unlikely(pte_alloc(vma->vm_mm, vmf->pmd))) 4377f9ce0be7SKirill A. Shutemov return VM_FAULT_OOM; 4378f9ce0be7SKirill A. Shutemov } 4379f9ce0be7SKirill A. Shutemov 4380f9ce0be7SKirill A. Shutemov /* See comment in handle_pte_fault() */ 4381f9ce0be7SKirill A. Shutemov if (pmd_devmap_trans_unstable(vmf->pmd)) 4382f9ce0be7SKirill A. Shutemov return 0; 4383f9ce0be7SKirill A. Shutemov 4384f9ce0be7SKirill A. Shutemov vmf->pte = pte_offset_map_lock(vma->vm_mm, vmf->pmd, 4385f9ce0be7SKirill A. Shutemov vmf->address, &vmf->ptl); 4386f9ce0be7SKirill A. Shutemov ret = 0; 4387f9ce0be7SKirill A. Shutemov /* Re-check under ptl */ 4388f46f2adeSPeter Xu if (likely(!vmf_pte_changed(vmf))) 43899d3af4b4SWill Deacon do_set_pte(vmf, page, vmf->address); 4390f9ce0be7SKirill A. Shutemov else 4391f9ce0be7SKirill A. Shutemov ret = VM_FAULT_NOPAGE; 4392f9ce0be7SKirill A. Shutemov 4393f9ce0be7SKirill A. Shutemov update_mmu_tlb(vma, vmf->address, vmf->pte); 43949118c0cbSJan Kara pte_unmap_unlock(vmf->pte, vmf->ptl); 43959118c0cbSJan Kara return ret; 43969118c0cbSJan Kara } 43979118c0cbSJan Kara 43983a91053aSKirill A. Shutemov static unsigned long fault_around_bytes __read_mostly = 43993a91053aSKirill A. Shutemov rounddown_pow_of_two(65536); 4400a9b0f861SKirill A. Shutemov 44011592eef0SKirill A. Shutemov #ifdef CONFIG_DEBUG_FS 4402a9b0f861SKirill A. Shutemov static int fault_around_bytes_get(void *data, u64 *val) 44031592eef0SKirill A. Shutemov { 4404a9b0f861SKirill A. Shutemov *val = fault_around_bytes; 44051592eef0SKirill A. Shutemov return 0; 44061592eef0SKirill A. Shutemov } 44071592eef0SKirill A. Shutemov 4408b4903d6eSAndrey Ryabinin /* 4409da391d64SWilliam Kucharski * fault_around_bytes must be rounded down to the nearest page order as it's 4410da391d64SWilliam Kucharski * what do_fault_around() expects to see. 4411b4903d6eSAndrey Ryabinin */ 4412a9b0f861SKirill A. Shutemov static int fault_around_bytes_set(void *data, u64 val) 44131592eef0SKirill A. Shutemov { 4414a9b0f861SKirill A. Shutemov if (val / PAGE_SIZE > PTRS_PER_PTE) 44151592eef0SKirill A. Shutemov return -EINVAL; 4416b4903d6eSAndrey Ryabinin if (val > PAGE_SIZE) 4417b4903d6eSAndrey Ryabinin fault_around_bytes = rounddown_pow_of_two(val); 4418b4903d6eSAndrey Ryabinin else 4419b4903d6eSAndrey Ryabinin fault_around_bytes = PAGE_SIZE; /* rounddown_pow_of_two(0) is undefined */ 44201592eef0SKirill A. Shutemov return 0; 44211592eef0SKirill A. Shutemov } 44220a1345f8SYevgen Pronenko DEFINE_DEBUGFS_ATTRIBUTE(fault_around_bytes_fops, 4423a9b0f861SKirill A. Shutemov fault_around_bytes_get, fault_around_bytes_set, "%llu\n"); 44241592eef0SKirill A. Shutemov 44251592eef0SKirill A. Shutemov static int __init fault_around_debugfs(void) 44261592eef0SKirill A. Shutemov { 4427d9f7979cSGreg Kroah-Hartman debugfs_create_file_unsafe("fault_around_bytes", 0644, NULL, NULL, 4428a9b0f861SKirill A. Shutemov &fault_around_bytes_fops); 44291592eef0SKirill A. Shutemov return 0; 44301592eef0SKirill A. Shutemov } 44311592eef0SKirill A. Shutemov late_initcall(fault_around_debugfs); 44321592eef0SKirill A. Shutemov #endif 44338c6e50b0SKirill A. Shutemov 44341fdb412bSKirill A. Shutemov /* 44351fdb412bSKirill A. Shutemov * do_fault_around() tries to map few pages around the fault address. The hope 44361fdb412bSKirill A. Shutemov * is that the pages will be needed soon and this will lower the number of 44371fdb412bSKirill A. Shutemov * faults to handle. 44381fdb412bSKirill A. Shutemov * 44391fdb412bSKirill A. Shutemov * It uses vm_ops->map_pages() to map the pages, which skips the page if it's 44401fdb412bSKirill A. Shutemov * not ready to be mapped: not up-to-date, locked, etc. 44411fdb412bSKirill A. Shutemov * 44421fdb412bSKirill A. Shutemov * This function is called with the page table lock taken. In the split ptlock 44431fdb412bSKirill A. Shutemov * case the page table lock only protects only those entries which belong to 44441fdb412bSKirill A. Shutemov * the page table corresponding to the fault address. 44451fdb412bSKirill A. Shutemov * 44461fdb412bSKirill A. Shutemov * This function doesn't cross the VMA boundaries, in order to call map_pages() 44471fdb412bSKirill A. Shutemov * only once. 44481fdb412bSKirill A. Shutemov * 4449da391d64SWilliam Kucharski * fault_around_bytes defines how many bytes we'll try to map. 4450da391d64SWilliam Kucharski * do_fault_around() expects it to be set to a power of two less than or equal 4451da391d64SWilliam Kucharski * to PTRS_PER_PTE. 44521fdb412bSKirill A. Shutemov * 4453da391d64SWilliam Kucharski * The virtual address of the area that we map is naturally aligned to 4454da391d64SWilliam Kucharski * fault_around_bytes rounded down to the machine page size 4455da391d64SWilliam Kucharski * (and therefore to page order). This way it's easier to guarantee 4456da391d64SWilliam Kucharski * that we don't cross page table boundaries. 44571fdb412bSKirill A. Shutemov */ 44582b740303SSouptick Joarder static vm_fault_t do_fault_around(struct vm_fault *vmf) 44598c6e50b0SKirill A. Shutemov { 446082b0f8c3SJan Kara unsigned long address = vmf->address, nr_pages, mask; 44610721ec8bSJan Kara pgoff_t start_pgoff = vmf->pgoff; 4462bae473a4SKirill A. Shutemov pgoff_t end_pgoff; 44632b740303SSouptick Joarder int off; 44648c6e50b0SKirill A. Shutemov 44654db0c3c2SJason Low nr_pages = READ_ONCE(fault_around_bytes) >> PAGE_SHIFT; 4466aecd6f44SKirill A. Shutemov mask = ~(nr_pages * PAGE_SIZE - 1) & PAGE_MASK; 4467aecd6f44SKirill A. Shutemov 4468f9ce0be7SKirill A. Shutemov address = max(address & mask, vmf->vma->vm_start); 4469f9ce0be7SKirill A. Shutemov off = ((vmf->address - address) >> PAGE_SHIFT) & (PTRS_PER_PTE - 1); 4470bae473a4SKirill A. Shutemov start_pgoff -= off; 44718c6e50b0SKirill A. Shutemov 44728c6e50b0SKirill A. Shutemov /* 4473da391d64SWilliam Kucharski * end_pgoff is either the end of the page table, the end of 4474da391d64SWilliam Kucharski * the vma or nr_pages from start_pgoff, depending what is nearest. 44758c6e50b0SKirill A. Shutemov */ 4476bae473a4SKirill A. Shutemov end_pgoff = start_pgoff - 4477f9ce0be7SKirill A. Shutemov ((address >> PAGE_SHIFT) & (PTRS_PER_PTE - 1)) + 44788c6e50b0SKirill A. Shutemov PTRS_PER_PTE - 1; 447982b0f8c3SJan Kara end_pgoff = min3(end_pgoff, vma_pages(vmf->vma) + vmf->vma->vm_pgoff - 1, 4480bae473a4SKirill A. Shutemov start_pgoff + nr_pages - 1); 44818c6e50b0SKirill A. Shutemov 448282b0f8c3SJan Kara if (pmd_none(*vmf->pmd)) { 44834cf58924SJoel Fernandes (Google) vmf->prealloc_pte = pte_alloc_one(vmf->vma->vm_mm); 448482b0f8c3SJan Kara if (!vmf->prealloc_pte) 4485f9ce0be7SKirill A. Shutemov return VM_FAULT_OOM; 44868c6e50b0SKirill A. Shutemov } 44878c6e50b0SKirill A. Shutemov 4488f9ce0be7SKirill A. Shutemov return vmf->vma->vm_ops->map_pages(vmf, start_pgoff, end_pgoff); 44897267ec00SKirill A. Shutemov } 44907267ec00SKirill A. Shutemov 44919c28a205SPeter Xu /* Return true if we should do read fault-around, false otherwise */ 44929c28a205SPeter Xu static inline bool should_fault_around(struct vm_fault *vmf) 44939c28a205SPeter Xu { 44949c28a205SPeter Xu /* No ->map_pages? No way to fault around... */ 44959c28a205SPeter Xu if (!vmf->vma->vm_ops->map_pages) 44969c28a205SPeter Xu return false; 44979c28a205SPeter Xu 44989c28a205SPeter Xu if (uffd_disable_fault_around(vmf->vma)) 44999c28a205SPeter Xu return false; 45009c28a205SPeter Xu 45019c28a205SPeter Xu return fault_around_bytes >> PAGE_SHIFT > 1; 45029c28a205SPeter Xu } 45039c28a205SPeter Xu 45042b740303SSouptick Joarder static vm_fault_t do_read_fault(struct vm_fault *vmf) 4505e655fb29SKirill A. Shutemov { 45062b740303SSouptick Joarder vm_fault_t ret = 0; 45078c6e50b0SKirill A. Shutemov 45088c6e50b0SKirill A. Shutemov /* 45098c6e50b0SKirill A. Shutemov * Let's call ->map_pages() first and use ->fault() as fallback 45108c6e50b0SKirill A. Shutemov * if page by the offset is not ready to be mapped (cold cache or 45118c6e50b0SKirill A. Shutemov * something). 45128c6e50b0SKirill A. Shutemov */ 45139c28a205SPeter Xu if (should_fault_around(vmf)) { 45140721ec8bSJan Kara ret = do_fault_around(vmf); 45157267ec00SKirill A. Shutemov if (ret) 45167267ec00SKirill A. Shutemov return ret; 45178c6e50b0SKirill A. Shutemov } 4518e655fb29SKirill A. Shutemov 4519936ca80dSJan Kara ret = __do_fault(vmf); 4520e655fb29SKirill A. Shutemov if (unlikely(ret & (VM_FAULT_ERROR | VM_FAULT_NOPAGE | VM_FAULT_RETRY))) 4521e655fb29SKirill A. Shutemov return ret; 4522e655fb29SKirill A. Shutemov 45239118c0cbSJan Kara ret |= finish_fault(vmf); 4524936ca80dSJan Kara unlock_page(vmf->page); 45257267ec00SKirill A. Shutemov if (unlikely(ret & (VM_FAULT_ERROR | VM_FAULT_NOPAGE | VM_FAULT_RETRY))) 4526936ca80dSJan Kara put_page(vmf->page); 4527e655fb29SKirill A. Shutemov return ret; 4528e655fb29SKirill A. Shutemov } 4529e655fb29SKirill A. Shutemov 45302b740303SSouptick Joarder static vm_fault_t do_cow_fault(struct vm_fault *vmf) 4531ec47c3b9SKirill A. Shutemov { 453282b0f8c3SJan Kara struct vm_area_struct *vma = vmf->vma; 45332b740303SSouptick Joarder vm_fault_t ret; 4534ec47c3b9SKirill A. Shutemov 4535ec47c3b9SKirill A. Shutemov if (unlikely(anon_vma_prepare(vma))) 4536ec47c3b9SKirill A. Shutemov return VM_FAULT_OOM; 4537ec47c3b9SKirill A. Shutemov 4538936ca80dSJan Kara vmf->cow_page = alloc_page_vma(GFP_HIGHUSER_MOVABLE, vma, vmf->address); 4539936ca80dSJan Kara if (!vmf->cow_page) 4540ec47c3b9SKirill A. Shutemov return VM_FAULT_OOM; 4541ec47c3b9SKirill A. Shutemov 45428f425e4eSMatthew Wilcox (Oracle) if (mem_cgroup_charge(page_folio(vmf->cow_page), vma->vm_mm, 45438f425e4eSMatthew Wilcox (Oracle) GFP_KERNEL)) { 4544936ca80dSJan Kara put_page(vmf->cow_page); 4545ec47c3b9SKirill A. Shutemov return VM_FAULT_OOM; 4546ec47c3b9SKirill A. Shutemov } 45479d82c694SJohannes Weiner cgroup_throttle_swaprate(vmf->cow_page, GFP_KERNEL); 4548ec47c3b9SKirill A. Shutemov 4549936ca80dSJan Kara ret = __do_fault(vmf); 4550ec47c3b9SKirill A. Shutemov if (unlikely(ret & (VM_FAULT_ERROR | VM_FAULT_NOPAGE | VM_FAULT_RETRY))) 4551ec47c3b9SKirill A. Shutemov goto uncharge_out; 45523917048dSJan Kara if (ret & VM_FAULT_DONE_COW) 45533917048dSJan Kara return ret; 4554ec47c3b9SKirill A. Shutemov 4555936ca80dSJan Kara copy_user_highpage(vmf->cow_page, vmf->page, vmf->address, vma); 4556936ca80dSJan Kara __SetPageUptodate(vmf->cow_page); 4557ec47c3b9SKirill A. Shutemov 45589118c0cbSJan Kara ret |= finish_fault(vmf); 4559936ca80dSJan Kara unlock_page(vmf->page); 4560936ca80dSJan Kara put_page(vmf->page); 45617267ec00SKirill A. Shutemov if (unlikely(ret & (VM_FAULT_ERROR | VM_FAULT_NOPAGE | VM_FAULT_RETRY))) 45627267ec00SKirill A. Shutemov goto uncharge_out; 4563ec47c3b9SKirill A. Shutemov return ret; 4564ec47c3b9SKirill A. Shutemov uncharge_out: 4565936ca80dSJan Kara put_page(vmf->cow_page); 4566ec47c3b9SKirill A. Shutemov return ret; 4567ec47c3b9SKirill A. Shutemov } 4568ec47c3b9SKirill A. Shutemov 45692b740303SSouptick Joarder static vm_fault_t do_shared_fault(struct vm_fault *vmf) 45701da177e4SLinus Torvalds { 457182b0f8c3SJan Kara struct vm_area_struct *vma = vmf->vma; 45722b740303SSouptick Joarder vm_fault_t ret, tmp; 45731d65f86dSKAMEZAWA Hiroyuki 4574936ca80dSJan Kara ret = __do_fault(vmf); 45757eae74afSKirill A. Shutemov if (unlikely(ret & (VM_FAULT_ERROR | VM_FAULT_NOPAGE | VM_FAULT_RETRY))) 4576f0c6d4d2SKirill A. Shutemov return ret; 45771da177e4SLinus Torvalds 45781da177e4SLinus Torvalds /* 4579f0c6d4d2SKirill A. Shutemov * Check if the backing address space wants to know that the page is 4580f0c6d4d2SKirill A. Shutemov * about to become writable 45811da177e4SLinus Torvalds */ 4582fb09a464SKirill A. Shutemov if (vma->vm_ops->page_mkwrite) { 4583936ca80dSJan Kara unlock_page(vmf->page); 458438b8cb7fSJan Kara tmp = do_page_mkwrite(vmf); 4585fb09a464SKirill A. Shutemov if (unlikely(!tmp || 4586fb09a464SKirill A. Shutemov (tmp & (VM_FAULT_ERROR | VM_FAULT_NOPAGE)))) { 4587936ca80dSJan Kara put_page(vmf->page); 4588f0c6d4d2SKirill A. Shutemov return tmp; 458969676147SMark Fasheh } 4590d0217ac0SNick Piggin } 4591fb09a464SKirill A. Shutemov 45929118c0cbSJan Kara ret |= finish_fault(vmf); 45937267ec00SKirill A. Shutemov if (unlikely(ret & (VM_FAULT_ERROR | VM_FAULT_NOPAGE | 45947267ec00SKirill A. Shutemov VM_FAULT_RETRY))) { 4595936ca80dSJan Kara unlock_page(vmf->page); 4596936ca80dSJan Kara put_page(vmf->page); 4597f0c6d4d2SKirill A. Shutemov return ret; 45989637a5efSDavid Howells } 4599d00806b1SNick Piggin 460089b15332SJohannes Weiner ret |= fault_dirty_shared_page(vmf); 4601b827e496SNick Piggin return ret; 460254cb8821SNick Piggin } 4603d00806b1SNick Piggin 46049a95f3cfSPaul Cassella /* 4605c1e8d7c6SMichel Lespinasse * We enter with non-exclusive mmap_lock (to exclude vma changes, 46069a95f3cfSPaul Cassella * but allow concurrent faults). 4607c1e8d7c6SMichel Lespinasse * The mmap_lock may have been released depending on flags and our 46089138e47eSMatthew Wilcox (Oracle) * return value. See filemap_fault() and __folio_lock_or_retry(). 4609c1e8d7c6SMichel Lespinasse * If mmap_lock is released, vma may become invalid (for example 4610fc8efd2dSJan Stancek * by other thread calling munmap()). 46119a95f3cfSPaul Cassella */ 46122b740303SSouptick Joarder static vm_fault_t do_fault(struct vm_fault *vmf) 461354cb8821SNick Piggin { 461482b0f8c3SJan Kara struct vm_area_struct *vma = vmf->vma; 4615fc8efd2dSJan Stancek struct mm_struct *vm_mm = vma->vm_mm; 46162b740303SSouptick Joarder vm_fault_t ret; 461754cb8821SNick Piggin 4618ff09d7ecSAneesh Kumar K.V /* 4619ff09d7ecSAneesh Kumar K.V * The VMA was not fully populated on mmap() or missing VM_DONTEXPAND 4620ff09d7ecSAneesh Kumar K.V */ 4621ff09d7ecSAneesh Kumar K.V if (!vma->vm_ops->fault) { 4622ff09d7ecSAneesh Kumar K.V /* 4623ff09d7ecSAneesh Kumar K.V * If we find a migration pmd entry or a none pmd entry, which 4624ff09d7ecSAneesh Kumar K.V * should never happen, return SIGBUS 4625ff09d7ecSAneesh Kumar K.V */ 4626ff09d7ecSAneesh Kumar K.V if (unlikely(!pmd_present(*vmf->pmd))) 4627b0b9b3dfSHugh Dickins ret = VM_FAULT_SIGBUS; 4628ff09d7ecSAneesh Kumar K.V else { 4629ff09d7ecSAneesh Kumar K.V vmf->pte = pte_offset_map_lock(vmf->vma->vm_mm, 4630ff09d7ecSAneesh Kumar K.V vmf->pmd, 4631ff09d7ecSAneesh Kumar K.V vmf->address, 4632ff09d7ecSAneesh Kumar K.V &vmf->ptl); 4633ff09d7ecSAneesh Kumar K.V /* 4634ff09d7ecSAneesh Kumar K.V * Make sure this is not a temporary clearing of pte 4635ff09d7ecSAneesh Kumar K.V * by holding ptl and checking again. A R/M/W update 4636ff09d7ecSAneesh Kumar K.V * of pte involves: take ptl, clearing the pte so that 4637ff09d7ecSAneesh Kumar K.V * we don't have concurrent modification by hardware 4638ff09d7ecSAneesh Kumar K.V * followed by an update. 4639ff09d7ecSAneesh Kumar K.V */ 4640ff09d7ecSAneesh Kumar K.V if (unlikely(pte_none(*vmf->pte))) 4641ff09d7ecSAneesh Kumar K.V ret = VM_FAULT_SIGBUS; 4642ff09d7ecSAneesh Kumar K.V else 4643ff09d7ecSAneesh Kumar K.V ret = VM_FAULT_NOPAGE; 4644ff09d7ecSAneesh Kumar K.V 4645ff09d7ecSAneesh Kumar K.V pte_unmap_unlock(vmf->pte, vmf->ptl); 4646ff09d7ecSAneesh Kumar K.V } 4647ff09d7ecSAneesh Kumar K.V } else if (!(vmf->flags & FAULT_FLAG_WRITE)) 4648b0b9b3dfSHugh Dickins ret = do_read_fault(vmf); 4649b0b9b3dfSHugh Dickins else if (!(vma->vm_flags & VM_SHARED)) 4650b0b9b3dfSHugh Dickins ret = do_cow_fault(vmf); 4651b0b9b3dfSHugh Dickins else 4652b0b9b3dfSHugh Dickins ret = do_shared_fault(vmf); 4653b0b9b3dfSHugh Dickins 4654b0b9b3dfSHugh Dickins /* preallocated pagetable is unused: free it */ 4655b0b9b3dfSHugh Dickins if (vmf->prealloc_pte) { 4656fc8efd2dSJan Stancek pte_free(vm_mm, vmf->prealloc_pte); 46577f2b6ce8STobin C Harding vmf->prealloc_pte = NULL; 4658b0b9b3dfSHugh Dickins } 4659b0b9b3dfSHugh Dickins return ret; 466054cb8821SNick Piggin } 466154cb8821SNick Piggin 4662f4c0d836SYang Shi int numa_migrate_prep(struct page *page, struct vm_area_struct *vma, 4663f4c0d836SYang Shi unsigned long addr, int page_nid, int *flags) 46649532fec1SMel Gorman { 46659532fec1SMel Gorman get_page(page); 46669532fec1SMel Gorman 46679532fec1SMel Gorman count_vm_numa_event(NUMA_HINT_FAULTS); 466804bb2f94SRik van Riel if (page_nid == numa_node_id()) { 46699532fec1SMel Gorman count_vm_numa_event(NUMA_HINT_FAULTS_LOCAL); 467004bb2f94SRik van Riel *flags |= TNF_FAULT_LOCAL; 467104bb2f94SRik van Riel } 46729532fec1SMel Gorman 46739532fec1SMel Gorman return mpol_misplaced(page, vma, addr); 46749532fec1SMel Gorman } 46759532fec1SMel Gorman 46762b740303SSouptick Joarder static vm_fault_t do_numa_page(struct vm_fault *vmf) 4677d10e63f2SMel Gorman { 467882b0f8c3SJan Kara struct vm_area_struct *vma = vmf->vma; 46794daae3b4SMel Gorman struct page *page = NULL; 468098fa15f3SAnshuman Khandual int page_nid = NUMA_NO_NODE; 468190572890SPeter Zijlstra int last_cpupid; 4682cbee9f88SPeter Zijlstra int target_nid; 468304a86453SAneesh Kumar K.V pte_t pte, old_pte; 4684288bc549SAneesh Kumar K.V bool was_writable = pte_savedwrite(vmf->orig_pte); 46856688cc05SPeter Zijlstra int flags = 0; 4686d10e63f2SMel Gorman 4687d10e63f2SMel Gorman /* 4688d10e63f2SMel Gorman * The "pte" at this point cannot be used safely without 4689d10e63f2SMel Gorman * validation through pte_unmap_same(). It's of NUMA type but 4690d10e63f2SMel Gorman * the pfn may be screwed if the read is non atomic. 4691d10e63f2SMel Gorman */ 469282b0f8c3SJan Kara vmf->ptl = pte_lockptr(vma->vm_mm, vmf->pmd); 469382b0f8c3SJan Kara spin_lock(vmf->ptl); 4694cee216a6SAneesh Kumar K.V if (unlikely(!pte_same(*vmf->pte, vmf->orig_pte))) { 469582b0f8c3SJan Kara pte_unmap_unlock(vmf->pte, vmf->ptl); 46964daae3b4SMel Gorman goto out; 46974daae3b4SMel Gorman } 46984daae3b4SMel Gorman 4699b99a342dSHuang Ying /* Get the normal PTE */ 4700b99a342dSHuang Ying old_pte = ptep_get(vmf->pte); 470104a86453SAneesh Kumar K.V pte = pte_modify(old_pte, vma->vm_page_prot); 4702d10e63f2SMel Gorman 470382b0f8c3SJan Kara page = vm_normal_page(vma, vmf->address, pte); 4704*3218f871SAlex Sierra if (!page || is_zone_device_page(page)) 4705b99a342dSHuang Ying goto out_map; 4706d10e63f2SMel Gorman 4707e81c4802SKirill A. Shutemov /* TODO: handle PTE-mapped THP */ 4708b99a342dSHuang Ying if (PageCompound(page)) 4709b99a342dSHuang Ying goto out_map; 4710e81c4802SKirill A. Shutemov 47116688cc05SPeter Zijlstra /* 4712bea66fbdSMel Gorman * Avoid grouping on RO pages in general. RO pages shouldn't hurt as 4713bea66fbdSMel Gorman * much anyway since they can be in shared cache state. This misses 4714bea66fbdSMel Gorman * the case where a mapping is writable but the process never writes 4715bea66fbdSMel Gorman * to it but pte_write gets cleared during protection updates and 4716bea66fbdSMel Gorman * pte_dirty has unpredictable behaviour between PTE scan updates, 4717bea66fbdSMel Gorman * background writeback, dirty balancing and application behaviour. 47186688cc05SPeter Zijlstra */ 4719b99a342dSHuang Ying if (!was_writable) 47206688cc05SPeter Zijlstra flags |= TNF_NO_GROUP; 47216688cc05SPeter Zijlstra 4722dabe1d99SRik van Riel /* 4723dabe1d99SRik van Riel * Flag if the page is shared between multiple address spaces. This 4724dabe1d99SRik van Riel * is later used when determining whether to group tasks together 4725dabe1d99SRik van Riel */ 4726dabe1d99SRik van Riel if (page_mapcount(page) > 1 && (vma->vm_flags & VM_SHARED)) 4727dabe1d99SRik van Riel flags |= TNF_SHARED; 4728dabe1d99SRik van Riel 472990572890SPeter Zijlstra last_cpupid = page_cpupid_last(page); 47308191acbdSMel Gorman page_nid = page_to_nid(page); 473182b0f8c3SJan Kara target_nid = numa_migrate_prep(page, vma, vmf->address, page_nid, 4732bae473a4SKirill A. Shutemov &flags); 473398fa15f3SAnshuman Khandual if (target_nid == NUMA_NO_NODE) { 47344daae3b4SMel Gorman put_page(page); 4735b99a342dSHuang Ying goto out_map; 47364daae3b4SMel Gorman } 4737b99a342dSHuang Ying pte_unmap_unlock(vmf->pte, vmf->ptl); 47384daae3b4SMel Gorman 47394daae3b4SMel Gorman /* Migrate to the requested node */ 4740bf90ac19SWang Qing if (migrate_misplaced_page(page, vma, target_nid)) { 47418191acbdSMel Gorman page_nid = target_nid; 47426688cc05SPeter Zijlstra flags |= TNF_MIGRATED; 4743b99a342dSHuang Ying } else { 4744074c2381SMel Gorman flags |= TNF_MIGRATE_FAIL; 4745b99a342dSHuang Ying vmf->pte = pte_offset_map(vmf->pmd, vmf->address); 4746b99a342dSHuang Ying spin_lock(vmf->ptl); 4747b99a342dSHuang Ying if (unlikely(!pte_same(*vmf->pte, vmf->orig_pte))) { 4748b99a342dSHuang Ying pte_unmap_unlock(vmf->pte, vmf->ptl); 4749b99a342dSHuang Ying goto out; 4750b99a342dSHuang Ying } 4751b99a342dSHuang Ying goto out_map; 4752b99a342dSHuang Ying } 47534daae3b4SMel Gorman 47544daae3b4SMel Gorman out: 475598fa15f3SAnshuman Khandual if (page_nid != NUMA_NO_NODE) 47566688cc05SPeter Zijlstra task_numa_fault(last_cpupid, page_nid, 1, flags); 4757d10e63f2SMel Gorman return 0; 4758b99a342dSHuang Ying out_map: 4759b99a342dSHuang Ying /* 4760b99a342dSHuang Ying * Make it present again, depending on how arch implements 4761b99a342dSHuang Ying * non-accessible ptes, some can allow access by kernel mode. 4762b99a342dSHuang Ying */ 4763b99a342dSHuang Ying old_pte = ptep_modify_prot_start(vma, vmf->address, vmf->pte); 4764b99a342dSHuang Ying pte = pte_modify(old_pte, vma->vm_page_prot); 4765b99a342dSHuang Ying pte = pte_mkyoung(pte); 4766b99a342dSHuang Ying if (was_writable) 4767b99a342dSHuang Ying pte = pte_mkwrite(pte); 4768b99a342dSHuang Ying ptep_modify_prot_commit(vma, vmf->address, vmf->pte, old_pte, pte); 4769b99a342dSHuang Ying update_mmu_cache(vma, vmf->address, vmf->pte); 4770b99a342dSHuang Ying pte_unmap_unlock(vmf->pte, vmf->ptl); 4771b99a342dSHuang Ying goto out; 4772d10e63f2SMel Gorman } 4773d10e63f2SMel Gorman 47742b740303SSouptick Joarder static inline vm_fault_t create_huge_pmd(struct vm_fault *vmf) 4775b96375f7SMatthew Wilcox { 4776f4200391SDave Jiang if (vma_is_anonymous(vmf->vma)) 477782b0f8c3SJan Kara return do_huge_pmd_anonymous_page(vmf); 4778a2d58167SDave Jiang if (vmf->vma->vm_ops->huge_fault) 4779c791ace1SDave Jiang return vmf->vma->vm_ops->huge_fault(vmf, PE_SIZE_PMD); 4780b96375f7SMatthew Wilcox return VM_FAULT_FALLBACK; 4781b96375f7SMatthew Wilcox } 4782b96375f7SMatthew Wilcox 4783183f24aaSGeert Uytterhoeven /* `inline' is required to avoid gcc 4.1.2 build error */ 47845db4f15cSYang Shi static inline vm_fault_t wp_huge_pmd(struct vm_fault *vmf) 4785b96375f7SMatthew Wilcox { 4786c89357e2SDavid Hildenbrand const bool unshare = vmf->flags & FAULT_FLAG_UNSHARE; 4787c89357e2SDavid Hildenbrand 4788529b930bSAndrea Arcangeli if (vma_is_anonymous(vmf->vma)) { 4789c89357e2SDavid Hildenbrand if (likely(!unshare) && 4790c89357e2SDavid Hildenbrand userfaultfd_huge_pmd_wp(vmf->vma, vmf->orig_pmd)) 4791529b930bSAndrea Arcangeli return handle_userfault(vmf, VM_UFFD_WP); 47925db4f15cSYang Shi return do_huge_pmd_wp_page(vmf); 4793529b930bSAndrea Arcangeli } 4794327e9fd4SThomas Hellstrom (VMware) if (vmf->vma->vm_ops->huge_fault) { 4795327e9fd4SThomas Hellstrom (VMware) vm_fault_t ret = vmf->vma->vm_ops->huge_fault(vmf, PE_SIZE_PMD); 4796af9e4d5fSKirill A. Shutemov 4797327e9fd4SThomas Hellstrom (VMware) if (!(ret & VM_FAULT_FALLBACK)) 4798327e9fd4SThomas Hellstrom (VMware) return ret; 4799327e9fd4SThomas Hellstrom (VMware) } 4800327e9fd4SThomas Hellstrom (VMware) 4801327e9fd4SThomas Hellstrom (VMware) /* COW or write-notify handled on pte level: split pmd. */ 480282b0f8c3SJan Kara __split_huge_pmd(vmf->vma, vmf->pmd, vmf->address, false, NULL); 4803af9e4d5fSKirill A. Shutemov 4804b96375f7SMatthew Wilcox return VM_FAULT_FALLBACK; 4805b96375f7SMatthew Wilcox } 4806b96375f7SMatthew Wilcox 48072b740303SSouptick Joarder static vm_fault_t create_huge_pud(struct vm_fault *vmf) 4808a00cc7d9SMatthew Wilcox { 4809327e9fd4SThomas Hellstrom (VMware) #if defined(CONFIG_TRANSPARENT_HUGEPAGE) && \ 4810327e9fd4SThomas Hellstrom (VMware) defined(CONFIG_HAVE_ARCH_TRANSPARENT_HUGEPAGE_PUD) 4811a00cc7d9SMatthew Wilcox /* No support for anonymous transparent PUD pages yet */ 4812a00cc7d9SMatthew Wilcox if (vma_is_anonymous(vmf->vma)) 4813327e9fd4SThomas Hellstrom (VMware) goto split; 4814327e9fd4SThomas Hellstrom (VMware) if (vmf->vma->vm_ops->huge_fault) { 4815327e9fd4SThomas Hellstrom (VMware) vm_fault_t ret = vmf->vma->vm_ops->huge_fault(vmf, PE_SIZE_PUD); 4816327e9fd4SThomas Hellstrom (VMware) 4817327e9fd4SThomas Hellstrom (VMware) if (!(ret & VM_FAULT_FALLBACK)) 4818327e9fd4SThomas Hellstrom (VMware) return ret; 4819327e9fd4SThomas Hellstrom (VMware) } 4820327e9fd4SThomas Hellstrom (VMware) split: 4821327e9fd4SThomas Hellstrom (VMware) /* COW or write-notify not handled on PUD level: split pud.*/ 4822327e9fd4SThomas Hellstrom (VMware) __split_huge_pud(vmf->vma, vmf->pud, vmf->address); 4823a00cc7d9SMatthew Wilcox #endif /* CONFIG_TRANSPARENT_HUGEPAGE */ 4824a00cc7d9SMatthew Wilcox return VM_FAULT_FALLBACK; 4825a00cc7d9SMatthew Wilcox } 4826a00cc7d9SMatthew Wilcox 48272b740303SSouptick Joarder static vm_fault_t wp_huge_pud(struct vm_fault *vmf, pud_t orig_pud) 4828a00cc7d9SMatthew Wilcox { 4829a00cc7d9SMatthew Wilcox #ifdef CONFIG_TRANSPARENT_HUGEPAGE 4830a00cc7d9SMatthew Wilcox /* No support for anonymous transparent PUD pages yet */ 4831a00cc7d9SMatthew Wilcox if (vma_is_anonymous(vmf->vma)) 4832a00cc7d9SMatthew Wilcox return VM_FAULT_FALLBACK; 4833a00cc7d9SMatthew Wilcox if (vmf->vma->vm_ops->huge_fault) 4834c791ace1SDave Jiang return vmf->vma->vm_ops->huge_fault(vmf, PE_SIZE_PUD); 4835a00cc7d9SMatthew Wilcox #endif /* CONFIG_TRANSPARENT_HUGEPAGE */ 4836a00cc7d9SMatthew Wilcox return VM_FAULT_FALLBACK; 4837a00cc7d9SMatthew Wilcox } 4838a00cc7d9SMatthew Wilcox 48391da177e4SLinus Torvalds /* 48401da177e4SLinus Torvalds * These routines also need to handle stuff like marking pages dirty 48411da177e4SLinus Torvalds * and/or accessed for architectures that don't do it in hardware (most 48421da177e4SLinus Torvalds * RISC architectures). The early dirtying is also good on the i386. 48431da177e4SLinus Torvalds * 48441da177e4SLinus Torvalds * There is also a hook called "update_mmu_cache()" that architectures 48451da177e4SLinus Torvalds * with external mmu caches can use to update those (ie the Sparc or 48461da177e4SLinus Torvalds * PowerPC hashed page tables that act as extended TLBs). 48471da177e4SLinus Torvalds * 4848c1e8d7c6SMichel Lespinasse * We enter with non-exclusive mmap_lock (to exclude vma changes, but allow 48497267ec00SKirill A. Shutemov * concurrent faults). 48509a95f3cfSPaul Cassella * 4851c1e8d7c6SMichel Lespinasse * The mmap_lock may have been released depending on flags and our return value. 48529138e47eSMatthew Wilcox (Oracle) * See filemap_fault() and __folio_lock_or_retry(). 48531da177e4SLinus Torvalds */ 48542b740303SSouptick Joarder static vm_fault_t handle_pte_fault(struct vm_fault *vmf) 48551da177e4SLinus Torvalds { 48561da177e4SLinus Torvalds pte_t entry; 48571da177e4SLinus Torvalds 485882b0f8c3SJan Kara if (unlikely(pmd_none(*vmf->pmd))) { 48597267ec00SKirill A. Shutemov /* 48607267ec00SKirill A. Shutemov * Leave __pte_alloc() until later: because vm_ops->fault may 48617267ec00SKirill A. Shutemov * want to allocate huge page, and if we expose page table 48627267ec00SKirill A. Shutemov * for an instant, it will be difficult to retract from 48637267ec00SKirill A. Shutemov * concurrent faults and from rmap lookups. 48647267ec00SKirill A. Shutemov */ 486582b0f8c3SJan Kara vmf->pte = NULL; 4866f46f2adeSPeter Xu vmf->flags &= ~FAULT_FLAG_ORIG_PTE_VALID; 48677267ec00SKirill A. Shutemov } else { 4868f9ce0be7SKirill A. Shutemov /* 4869f9ce0be7SKirill A. Shutemov * If a huge pmd materialized under us just retry later. Use 4870f9ce0be7SKirill A. Shutemov * pmd_trans_unstable() via pmd_devmap_trans_unstable() instead 4871f9ce0be7SKirill A. Shutemov * of pmd_trans_huge() to ensure the pmd didn't become 4872f9ce0be7SKirill A. Shutemov * pmd_trans_huge under us and then back to pmd_none, as a 4873f9ce0be7SKirill A. Shutemov * result of MADV_DONTNEED running immediately after a huge pmd 4874f9ce0be7SKirill A. Shutemov * fault in a different thread of this mm, in turn leading to a 4875f9ce0be7SKirill A. Shutemov * misleading pmd_trans_huge() retval. All we have to ensure is 4876f9ce0be7SKirill A. Shutemov * that it is a regular pmd that we can walk with 4877f9ce0be7SKirill A. Shutemov * pte_offset_map() and we can do that through an atomic read 4878f9ce0be7SKirill A. Shutemov * in C, which is what pmd_trans_unstable() provides. 4879f9ce0be7SKirill A. Shutemov */ 4880d0f0931dSRoss Zwisler if (pmd_devmap_trans_unstable(vmf->pmd)) 48817267ec00SKirill A. Shutemov return 0; 48827267ec00SKirill A. Shutemov /* 48837267ec00SKirill A. Shutemov * A regular pmd is established and it can't morph into a huge 48847267ec00SKirill A. Shutemov * pmd from under us anymore at this point because we hold the 4885c1e8d7c6SMichel Lespinasse * mmap_lock read mode and khugepaged takes it in write mode. 48867267ec00SKirill A. Shutemov * So now it's safe to run pte_offset_map(). 48877267ec00SKirill A. Shutemov */ 488882b0f8c3SJan Kara vmf->pte = pte_offset_map(vmf->pmd, vmf->address); 48892994302bSJan Kara vmf->orig_pte = *vmf->pte; 4890f46f2adeSPeter Xu vmf->flags |= FAULT_FLAG_ORIG_PTE_VALID; 48917267ec00SKirill A. Shutemov 4892e37c6982SChristian Borntraeger /* 4893e37c6982SChristian Borntraeger * some architectures can have larger ptes than wordsize, 48947267ec00SKirill A. Shutemov * e.g.ppc44x-defconfig has CONFIG_PTE_64BIT=y and 4895b03a0fe0SPaul E. McKenney * CONFIG_32BIT=y, so READ_ONCE cannot guarantee atomic 4896b03a0fe0SPaul E. McKenney * accesses. The code below just needs a consistent view 4897b03a0fe0SPaul E. McKenney * for the ifs and we later double check anyway with the 48987267ec00SKirill A. Shutemov * ptl lock held. So here a barrier will do. 4899e37c6982SChristian Borntraeger */ 4900e37c6982SChristian Borntraeger barrier(); 49012994302bSJan Kara if (pte_none(vmf->orig_pte)) { 490282b0f8c3SJan Kara pte_unmap(vmf->pte); 490382b0f8c3SJan Kara vmf->pte = NULL; 49047267ec00SKirill A. Shutemov } 49057267ec00SKirill A. Shutemov } 49067267ec00SKirill A. Shutemov 490782b0f8c3SJan Kara if (!vmf->pte) { 490882b0f8c3SJan Kara if (vma_is_anonymous(vmf->vma)) 490982b0f8c3SJan Kara return do_anonymous_page(vmf); 4910b5330628SOleg Nesterov else 491182b0f8c3SJan Kara return do_fault(vmf); 491265500d23SHugh Dickins } 49137267ec00SKirill A. Shutemov 49142994302bSJan Kara if (!pte_present(vmf->orig_pte)) 49152994302bSJan Kara return do_swap_page(vmf); 49161da177e4SLinus Torvalds 49172994302bSJan Kara if (pte_protnone(vmf->orig_pte) && vma_is_accessible(vmf->vma)) 49182994302bSJan Kara return do_numa_page(vmf); 4919d10e63f2SMel Gorman 492082b0f8c3SJan Kara vmf->ptl = pte_lockptr(vmf->vma->vm_mm, vmf->pmd); 492182b0f8c3SJan Kara spin_lock(vmf->ptl); 49222994302bSJan Kara entry = vmf->orig_pte; 49237df67697SBibo Mao if (unlikely(!pte_same(*vmf->pte, entry))) { 49247df67697SBibo Mao update_mmu_tlb(vmf->vma, vmf->address, vmf->pte); 49258f4e2101SHugh Dickins goto unlock; 49267df67697SBibo Mao } 4927c89357e2SDavid Hildenbrand if (vmf->flags & (FAULT_FLAG_WRITE|FAULT_FLAG_UNSHARE)) { 4928f6f37321SLinus Torvalds if (!pte_write(entry)) 49292994302bSJan Kara return do_wp_page(vmf); 4930c89357e2SDavid Hildenbrand else if (likely(vmf->flags & FAULT_FLAG_WRITE)) 49311da177e4SLinus Torvalds entry = pte_mkdirty(entry); 49321da177e4SLinus Torvalds } 49331da177e4SLinus Torvalds entry = pte_mkyoung(entry); 493482b0f8c3SJan Kara if (ptep_set_access_flags(vmf->vma, vmf->address, vmf->pte, entry, 493582b0f8c3SJan Kara vmf->flags & FAULT_FLAG_WRITE)) { 493682b0f8c3SJan Kara update_mmu_cache(vmf->vma, vmf->address, vmf->pte); 49371a44e149SAndrea Arcangeli } else { 4938b7333b58SYang Shi /* Skip spurious TLB flush for retried page fault */ 4939b7333b58SYang Shi if (vmf->flags & FAULT_FLAG_TRIED) 4940b7333b58SYang Shi goto unlock; 49411a44e149SAndrea Arcangeli /* 49421a44e149SAndrea Arcangeli * This is needed only for protection faults but the arch code 49431a44e149SAndrea Arcangeli * is not yet telling us if this is a protection fault or not. 49441a44e149SAndrea Arcangeli * This still avoids useless tlb flushes for .text page faults 49451a44e149SAndrea Arcangeli * with threads. 49461a44e149SAndrea Arcangeli */ 494782b0f8c3SJan Kara if (vmf->flags & FAULT_FLAG_WRITE) 494882b0f8c3SJan Kara flush_tlb_fix_spurious_fault(vmf->vma, vmf->address); 49491a44e149SAndrea Arcangeli } 49508f4e2101SHugh Dickins unlock: 495182b0f8c3SJan Kara pte_unmap_unlock(vmf->pte, vmf->ptl); 495283c54070SNick Piggin return 0; 49531da177e4SLinus Torvalds } 49541da177e4SLinus Torvalds 49551da177e4SLinus Torvalds /* 49561da177e4SLinus Torvalds * By the time we get here, we already hold the mm semaphore 49579a95f3cfSPaul Cassella * 4958c1e8d7c6SMichel Lespinasse * The mmap_lock may have been released depending on flags and our 49599138e47eSMatthew Wilcox (Oracle) * return value. See filemap_fault() and __folio_lock_or_retry(). 49601da177e4SLinus Torvalds */ 49612b740303SSouptick Joarder static vm_fault_t __handle_mm_fault(struct vm_area_struct *vma, 49622b740303SSouptick Joarder unsigned long address, unsigned int flags) 49631da177e4SLinus Torvalds { 496482b0f8c3SJan Kara struct vm_fault vmf = { 4965bae473a4SKirill A. Shutemov .vma = vma, 49661a29d85eSJan Kara .address = address & PAGE_MASK, 4967824ddc60SNadav Amit .real_address = address, 4968bae473a4SKirill A. Shutemov .flags = flags, 49690721ec8bSJan Kara .pgoff = linear_page_index(vma, address), 4970667240e0SJan Kara .gfp_mask = __get_fault_gfp_mask(vma), 4971bae473a4SKirill A. Shutemov }; 4972dcddffd4SKirill A. Shutemov struct mm_struct *mm = vma->vm_mm; 49731da177e4SLinus Torvalds pgd_t *pgd; 4974c2febafcSKirill A. Shutemov p4d_t *p4d; 49752b740303SSouptick Joarder vm_fault_t ret; 49761da177e4SLinus Torvalds 49771da177e4SLinus Torvalds pgd = pgd_offset(mm, address); 4978c2febafcSKirill A. Shutemov p4d = p4d_alloc(mm, pgd, address); 4979c2febafcSKirill A. Shutemov if (!p4d) 4980c2febafcSKirill A. Shutemov return VM_FAULT_OOM; 4981a00cc7d9SMatthew Wilcox 4982c2febafcSKirill A. Shutemov vmf.pud = pud_alloc(mm, p4d, address); 4983a00cc7d9SMatthew Wilcox if (!vmf.pud) 4984c74df32cSHugh Dickins return VM_FAULT_OOM; 4985625110b5SThomas Hellstrom retry_pud: 49867635d9cbSMichal Hocko if (pud_none(*vmf.pud) && __transparent_hugepage_enabled(vma)) { 4987a00cc7d9SMatthew Wilcox ret = create_huge_pud(&vmf); 4988a00cc7d9SMatthew Wilcox if (!(ret & VM_FAULT_FALLBACK)) 4989a00cc7d9SMatthew Wilcox return ret; 4990a00cc7d9SMatthew Wilcox } else { 4991a00cc7d9SMatthew Wilcox pud_t orig_pud = *vmf.pud; 4992a00cc7d9SMatthew Wilcox 4993a00cc7d9SMatthew Wilcox barrier(); 4994a00cc7d9SMatthew Wilcox if (pud_trans_huge(orig_pud) || pud_devmap(orig_pud)) { 4995a00cc7d9SMatthew Wilcox 4996c89357e2SDavid Hildenbrand /* 4997c89357e2SDavid Hildenbrand * TODO once we support anonymous PUDs: NUMA case and 4998c89357e2SDavid Hildenbrand * FAULT_FLAG_UNSHARE handling. 4999c89357e2SDavid Hildenbrand */ 5000c89357e2SDavid Hildenbrand if ((flags & FAULT_FLAG_WRITE) && !pud_write(orig_pud)) { 5001a00cc7d9SMatthew Wilcox ret = wp_huge_pud(&vmf, orig_pud); 5002a00cc7d9SMatthew Wilcox if (!(ret & VM_FAULT_FALLBACK)) 5003a00cc7d9SMatthew Wilcox return ret; 5004a00cc7d9SMatthew Wilcox } else { 5005a00cc7d9SMatthew Wilcox huge_pud_set_accessed(&vmf, orig_pud); 5006a00cc7d9SMatthew Wilcox return 0; 5007a00cc7d9SMatthew Wilcox } 5008a00cc7d9SMatthew Wilcox } 5009a00cc7d9SMatthew Wilcox } 5010a00cc7d9SMatthew Wilcox 5011a00cc7d9SMatthew Wilcox vmf.pmd = pmd_alloc(mm, vmf.pud, address); 501282b0f8c3SJan Kara if (!vmf.pmd) 5013c74df32cSHugh Dickins return VM_FAULT_OOM; 5014625110b5SThomas Hellstrom 5015625110b5SThomas Hellstrom /* Huge pud page fault raced with pmd_alloc? */ 5016625110b5SThomas Hellstrom if (pud_trans_unstable(vmf.pud)) 5017625110b5SThomas Hellstrom goto retry_pud; 5018625110b5SThomas Hellstrom 50197635d9cbSMichal Hocko if (pmd_none(*vmf.pmd) && __transparent_hugepage_enabled(vma)) { 5020a2d58167SDave Jiang ret = create_huge_pmd(&vmf); 5021c0292554SKirill A. Shutemov if (!(ret & VM_FAULT_FALLBACK)) 5022c0292554SKirill A. Shutemov return ret; 502371e3aac0SAndrea Arcangeli } else { 50245db4f15cSYang Shi vmf.orig_pmd = *vmf.pmd; 50251f1d06c3SDavid Rientjes 502671e3aac0SAndrea Arcangeli barrier(); 50275db4f15cSYang Shi if (unlikely(is_swap_pmd(vmf.orig_pmd))) { 502884c3fc4eSZi Yan VM_BUG_ON(thp_migration_supported() && 50295db4f15cSYang Shi !is_pmd_migration_entry(vmf.orig_pmd)); 50305db4f15cSYang Shi if (is_pmd_migration_entry(vmf.orig_pmd)) 503184c3fc4eSZi Yan pmd_migration_entry_wait(mm, vmf.pmd); 503284c3fc4eSZi Yan return 0; 503384c3fc4eSZi Yan } 50345db4f15cSYang Shi if (pmd_trans_huge(vmf.orig_pmd) || pmd_devmap(vmf.orig_pmd)) { 50355db4f15cSYang Shi if (pmd_protnone(vmf.orig_pmd) && vma_is_accessible(vma)) 50365db4f15cSYang Shi return do_huge_pmd_numa_page(&vmf); 5037d10e63f2SMel Gorman 5038c89357e2SDavid Hildenbrand if ((flags & (FAULT_FLAG_WRITE|FAULT_FLAG_UNSHARE)) && 5039c89357e2SDavid Hildenbrand !pmd_write(vmf.orig_pmd)) { 50405db4f15cSYang Shi ret = wp_huge_pmd(&vmf); 50419845cbbdSKirill A. Shutemov if (!(ret & VM_FAULT_FALLBACK)) 50421f1d06c3SDavid Rientjes return ret; 5043a1dd450bSWill Deacon } else { 50445db4f15cSYang Shi huge_pmd_set_accessed(&vmf); 504571e3aac0SAndrea Arcangeli return 0; 504671e3aac0SAndrea Arcangeli } 504771e3aac0SAndrea Arcangeli } 50489845cbbdSKirill A. Shutemov } 504971e3aac0SAndrea Arcangeli 505082b0f8c3SJan Kara return handle_pte_fault(&vmf); 50511da177e4SLinus Torvalds } 50521da177e4SLinus Torvalds 5053bce617edSPeter Xu /** 5054f0953a1bSIngo Molnar * mm_account_fault - Do page fault accounting 5055bce617edSPeter Xu * 5056bce617edSPeter Xu * @regs: the pt_regs struct pointer. When set to NULL, will skip accounting 5057bce617edSPeter Xu * of perf event counters, but we'll still do the per-task accounting to 5058bce617edSPeter Xu * the task who triggered this page fault. 5059bce617edSPeter Xu * @address: the faulted address. 5060bce617edSPeter Xu * @flags: the fault flags. 5061bce617edSPeter Xu * @ret: the fault retcode. 5062bce617edSPeter Xu * 5063f0953a1bSIngo Molnar * This will take care of most of the page fault accounting. Meanwhile, it 5064bce617edSPeter Xu * will also include the PERF_COUNT_SW_PAGE_FAULTS_[MAJ|MIN] perf counter 5065f0953a1bSIngo Molnar * updates. However, note that the handling of PERF_COUNT_SW_PAGE_FAULTS should 5066bce617edSPeter Xu * still be in per-arch page fault handlers at the entry of page fault. 5067bce617edSPeter Xu */ 5068bce617edSPeter Xu static inline void mm_account_fault(struct pt_regs *regs, 5069bce617edSPeter Xu unsigned long address, unsigned int flags, 5070bce617edSPeter Xu vm_fault_t ret) 5071bce617edSPeter Xu { 5072bce617edSPeter Xu bool major; 5073bce617edSPeter Xu 5074bce617edSPeter Xu /* 5075bce617edSPeter Xu * We don't do accounting for some specific faults: 5076bce617edSPeter Xu * 5077bce617edSPeter Xu * - Unsuccessful faults (e.g. when the address wasn't valid). That 5078bce617edSPeter Xu * includes arch_vma_access_permitted() failing before reaching here. 5079bce617edSPeter Xu * So this is not a "this many hardware page faults" counter. We 5080bce617edSPeter Xu * should use the hw profiling for that. 5081bce617edSPeter Xu * 5082bce617edSPeter Xu * - Incomplete faults (VM_FAULT_RETRY). They will only be counted 5083bce617edSPeter Xu * once they're completed. 5084bce617edSPeter Xu */ 5085bce617edSPeter Xu if (ret & (VM_FAULT_ERROR | VM_FAULT_RETRY)) 5086bce617edSPeter Xu return; 5087bce617edSPeter Xu 5088bce617edSPeter Xu /* 5089bce617edSPeter Xu * We define the fault as a major fault when the final successful fault 5090bce617edSPeter Xu * is VM_FAULT_MAJOR, or if it retried (which implies that we couldn't 5091bce617edSPeter Xu * handle it immediately previously). 5092bce617edSPeter Xu */ 5093bce617edSPeter Xu major = (ret & VM_FAULT_MAJOR) || (flags & FAULT_FLAG_TRIED); 5094bce617edSPeter Xu 5095a2beb5f1SPeter Xu if (major) 5096a2beb5f1SPeter Xu current->maj_flt++; 5097a2beb5f1SPeter Xu else 5098a2beb5f1SPeter Xu current->min_flt++; 5099a2beb5f1SPeter Xu 5100bce617edSPeter Xu /* 5101a2beb5f1SPeter Xu * If the fault is done for GUP, regs will be NULL. We only do the 5102a2beb5f1SPeter Xu * accounting for the per thread fault counters who triggered the 5103a2beb5f1SPeter Xu * fault, and we skip the perf event updates. 5104bce617edSPeter Xu */ 5105bce617edSPeter Xu if (!regs) 5106bce617edSPeter Xu return; 5107bce617edSPeter Xu 5108a2beb5f1SPeter Xu if (major) 5109bce617edSPeter Xu perf_sw_event(PERF_COUNT_SW_PAGE_FAULTS_MAJ, 1, regs, address); 5110a2beb5f1SPeter Xu else 5111bce617edSPeter Xu perf_sw_event(PERF_COUNT_SW_PAGE_FAULTS_MIN, 1, regs, address); 5112bce617edSPeter Xu } 5113bce617edSPeter Xu 51149a95f3cfSPaul Cassella /* 51159a95f3cfSPaul Cassella * By the time we get here, we already hold the mm semaphore 51169a95f3cfSPaul Cassella * 5117c1e8d7c6SMichel Lespinasse * The mmap_lock may have been released depending on flags and our 51189138e47eSMatthew Wilcox (Oracle) * return value. See filemap_fault() and __folio_lock_or_retry(). 51199a95f3cfSPaul Cassella */ 51202b740303SSouptick Joarder vm_fault_t handle_mm_fault(struct vm_area_struct *vma, unsigned long address, 5121bce617edSPeter Xu unsigned int flags, struct pt_regs *regs) 5122519e5247SJohannes Weiner { 51232b740303SSouptick Joarder vm_fault_t ret; 5124519e5247SJohannes Weiner 5125519e5247SJohannes Weiner __set_current_state(TASK_RUNNING); 5126519e5247SJohannes Weiner 5127519e5247SJohannes Weiner count_vm_event(PGFAULT); 51282262185cSRoman Gushchin count_memcg_event_mm(vma->vm_mm, PGFAULT); 5129519e5247SJohannes Weiner 5130519e5247SJohannes Weiner /* do counter updates before entering really critical section. */ 5131519e5247SJohannes Weiner check_sync_rss_stat(current); 5132519e5247SJohannes Weiner 5133de0c799bSLaurent Dufour if (!arch_vma_access_permitted(vma, flags & FAULT_FLAG_WRITE, 5134de0c799bSLaurent Dufour flags & FAULT_FLAG_INSTRUCTION, 5135de0c799bSLaurent Dufour flags & FAULT_FLAG_REMOTE)) 5136de0c799bSLaurent Dufour return VM_FAULT_SIGSEGV; 5137de0c799bSLaurent Dufour 5138519e5247SJohannes Weiner /* 5139519e5247SJohannes Weiner * Enable the memcg OOM handling for faults triggered in user 5140519e5247SJohannes Weiner * space. Kernel faults are handled more gracefully. 5141519e5247SJohannes Weiner */ 5142519e5247SJohannes Weiner if (flags & FAULT_FLAG_USER) 514329ef680aSMichal Hocko mem_cgroup_enter_user_fault(); 5144519e5247SJohannes Weiner 5145bae473a4SKirill A. Shutemov if (unlikely(is_vm_hugetlb_page(vma))) 5146bae473a4SKirill A. Shutemov ret = hugetlb_fault(vma->vm_mm, vma, address, flags); 5147bae473a4SKirill A. Shutemov else 5148dcddffd4SKirill A. Shutemov ret = __handle_mm_fault(vma, address, flags); 5149519e5247SJohannes Weiner 515049426420SJohannes Weiner if (flags & FAULT_FLAG_USER) { 515129ef680aSMichal Hocko mem_cgroup_exit_user_fault(); 515249426420SJohannes Weiner /* 515349426420SJohannes Weiner * The task may have entered a memcg OOM situation but 515449426420SJohannes Weiner * if the allocation error was handled gracefully (no 515549426420SJohannes Weiner * VM_FAULT_OOM), there is no need to kill anything. 515649426420SJohannes Weiner * Just clean up the OOM state peacefully. 515749426420SJohannes Weiner */ 515849426420SJohannes Weiner if (task_in_memcg_oom(current) && !(ret & VM_FAULT_OOM)) 515949426420SJohannes Weiner mem_cgroup_oom_synchronize(false); 516049426420SJohannes Weiner } 51613812c8c8SJohannes Weiner 5162bce617edSPeter Xu mm_account_fault(regs, address, flags, ret); 5163bce617edSPeter Xu 5164519e5247SJohannes Weiner return ret; 5165519e5247SJohannes Weiner } 5166e1d6d01aSJesse Barnes EXPORT_SYMBOL_GPL(handle_mm_fault); 5167519e5247SJohannes Weiner 516890eceff1SKirill A. Shutemov #ifndef __PAGETABLE_P4D_FOLDED 516990eceff1SKirill A. Shutemov /* 517090eceff1SKirill A. Shutemov * Allocate p4d page table. 517190eceff1SKirill A. Shutemov * We've already handled the fast-path in-line. 517290eceff1SKirill A. Shutemov */ 517390eceff1SKirill A. Shutemov int __p4d_alloc(struct mm_struct *mm, pgd_t *pgd, unsigned long address) 517490eceff1SKirill A. Shutemov { 517590eceff1SKirill A. Shutemov p4d_t *new = p4d_alloc_one(mm, address); 517690eceff1SKirill A. Shutemov if (!new) 517790eceff1SKirill A. Shutemov return -ENOMEM; 517890eceff1SKirill A. Shutemov 517990eceff1SKirill A. Shutemov spin_lock(&mm->page_table_lock); 5180ed33b5a6SQi Zheng if (pgd_present(*pgd)) { /* Another has populated it */ 518190eceff1SKirill A. Shutemov p4d_free(mm, new); 5182ed33b5a6SQi Zheng } else { 5183ed33b5a6SQi Zheng smp_wmb(); /* See comment in pmd_install() */ 518490eceff1SKirill A. Shutemov pgd_populate(mm, pgd, new); 5185ed33b5a6SQi Zheng } 518690eceff1SKirill A. Shutemov spin_unlock(&mm->page_table_lock); 518790eceff1SKirill A. Shutemov return 0; 518890eceff1SKirill A. Shutemov } 518990eceff1SKirill A. Shutemov #endif /* __PAGETABLE_P4D_FOLDED */ 519090eceff1SKirill A. Shutemov 51911da177e4SLinus Torvalds #ifndef __PAGETABLE_PUD_FOLDED 51921da177e4SLinus Torvalds /* 51931da177e4SLinus Torvalds * Allocate page upper directory. 5194872fec16SHugh Dickins * We've already handled the fast-path in-line. 51951da177e4SLinus Torvalds */ 5196c2febafcSKirill A. Shutemov int __pud_alloc(struct mm_struct *mm, p4d_t *p4d, unsigned long address) 51971da177e4SLinus Torvalds { 5198c74df32cSHugh Dickins pud_t *new = pud_alloc_one(mm, address); 5199c74df32cSHugh Dickins if (!new) 52001bb3630eSHugh Dickins return -ENOMEM; 52011da177e4SLinus Torvalds 5202872fec16SHugh Dickins spin_lock(&mm->page_table_lock); 5203b4e98d9aSKirill A. Shutemov if (!p4d_present(*p4d)) { 5204b4e98d9aSKirill A. Shutemov mm_inc_nr_puds(mm); 5205ed33b5a6SQi Zheng smp_wmb(); /* See comment in pmd_install() */ 5206c2febafcSKirill A. Shutemov p4d_populate(mm, p4d, new); 5207b4e98d9aSKirill A. Shutemov } else /* Another has populated it */ 5208c2febafcSKirill A. Shutemov pud_free(mm, new); 5209872fec16SHugh Dickins spin_unlock(&mm->page_table_lock); 52101bb3630eSHugh Dickins return 0; 52111da177e4SLinus Torvalds } 52121da177e4SLinus Torvalds #endif /* __PAGETABLE_PUD_FOLDED */ 52131da177e4SLinus Torvalds 52141da177e4SLinus Torvalds #ifndef __PAGETABLE_PMD_FOLDED 52151da177e4SLinus Torvalds /* 52161da177e4SLinus Torvalds * Allocate page middle directory. 5217872fec16SHugh Dickins * We've already handled the fast-path in-line. 52181da177e4SLinus Torvalds */ 52191bb3630eSHugh Dickins int __pmd_alloc(struct mm_struct *mm, pud_t *pud, unsigned long address) 52201da177e4SLinus Torvalds { 5221a00cc7d9SMatthew Wilcox spinlock_t *ptl; 5222c74df32cSHugh Dickins pmd_t *new = pmd_alloc_one(mm, address); 5223c74df32cSHugh Dickins if (!new) 52241bb3630eSHugh Dickins return -ENOMEM; 52251da177e4SLinus Torvalds 5226a00cc7d9SMatthew Wilcox ptl = pud_lock(mm, pud); 5227dc6c9a35SKirill A. Shutemov if (!pud_present(*pud)) { 5228dc6c9a35SKirill A. Shutemov mm_inc_nr_pmds(mm); 5229ed33b5a6SQi Zheng smp_wmb(); /* See comment in pmd_install() */ 52301da177e4SLinus Torvalds pud_populate(mm, pud, new); 5231ed33b5a6SQi Zheng } else { /* Another has populated it */ 52325e541973SBenjamin Herrenschmidt pmd_free(mm, new); 5233ed33b5a6SQi Zheng } 5234a00cc7d9SMatthew Wilcox spin_unlock(ptl); 52351bb3630eSHugh Dickins return 0; 52361da177e4SLinus Torvalds } 52371da177e4SLinus Torvalds #endif /* __PAGETABLE_PMD_FOLDED */ 52381da177e4SLinus Torvalds 52393b6748e2SJohannes Weiner /** 52409fd6dad1SPaolo Bonzini * follow_pte - look up PTE at a user virtual address 52419fd6dad1SPaolo Bonzini * @mm: the mm_struct of the target address space 52429fd6dad1SPaolo Bonzini * @address: user virtual address 52439fd6dad1SPaolo Bonzini * @ptepp: location to store found PTE 52449fd6dad1SPaolo Bonzini * @ptlp: location to store the lock for the PTE 52459fd6dad1SPaolo Bonzini * 52469fd6dad1SPaolo Bonzini * On a successful return, the pointer to the PTE is stored in @ptepp; 52479fd6dad1SPaolo Bonzini * the corresponding lock is taken and its location is stored in @ptlp. 52489fd6dad1SPaolo Bonzini * The contents of the PTE are only stable until @ptlp is released; 52499fd6dad1SPaolo Bonzini * any further use, if any, must be protected against invalidation 52509fd6dad1SPaolo Bonzini * with MMU notifiers. 52519fd6dad1SPaolo Bonzini * 52529fd6dad1SPaolo Bonzini * Only IO mappings and raw PFN mappings are allowed. The mmap semaphore 52539fd6dad1SPaolo Bonzini * should be taken for read. 52549fd6dad1SPaolo Bonzini * 52559fd6dad1SPaolo Bonzini * KVM uses this function. While it is arguably less bad than ``follow_pfn``, 52569fd6dad1SPaolo Bonzini * it is not a good general-purpose API. 52579fd6dad1SPaolo Bonzini * 52589fd6dad1SPaolo Bonzini * Return: zero on success, -ve otherwise. 52599fd6dad1SPaolo Bonzini */ 52609fd6dad1SPaolo Bonzini int follow_pte(struct mm_struct *mm, unsigned long address, 52619fd6dad1SPaolo Bonzini pte_t **ptepp, spinlock_t **ptlp) 52629fd6dad1SPaolo Bonzini { 52630e5e64c0SMuchun Song pgd_t *pgd; 52640e5e64c0SMuchun Song p4d_t *p4d; 52650e5e64c0SMuchun Song pud_t *pud; 52660e5e64c0SMuchun Song pmd_t *pmd; 52670e5e64c0SMuchun Song pte_t *ptep; 52680e5e64c0SMuchun Song 52690e5e64c0SMuchun Song pgd = pgd_offset(mm, address); 52700e5e64c0SMuchun Song if (pgd_none(*pgd) || unlikely(pgd_bad(*pgd))) 52710e5e64c0SMuchun Song goto out; 52720e5e64c0SMuchun Song 52730e5e64c0SMuchun Song p4d = p4d_offset(pgd, address); 52740e5e64c0SMuchun Song if (p4d_none(*p4d) || unlikely(p4d_bad(*p4d))) 52750e5e64c0SMuchun Song goto out; 52760e5e64c0SMuchun Song 52770e5e64c0SMuchun Song pud = pud_offset(p4d, address); 52780e5e64c0SMuchun Song if (pud_none(*pud) || unlikely(pud_bad(*pud))) 52790e5e64c0SMuchun Song goto out; 52800e5e64c0SMuchun Song 52810e5e64c0SMuchun Song pmd = pmd_offset(pud, address); 52820e5e64c0SMuchun Song VM_BUG_ON(pmd_trans_huge(*pmd)); 52830e5e64c0SMuchun Song 52840e5e64c0SMuchun Song if (pmd_none(*pmd) || unlikely(pmd_bad(*pmd))) 52850e5e64c0SMuchun Song goto out; 52860e5e64c0SMuchun Song 52870e5e64c0SMuchun Song ptep = pte_offset_map_lock(mm, pmd, address, ptlp); 52880e5e64c0SMuchun Song if (!pte_present(*ptep)) 52890e5e64c0SMuchun Song goto unlock; 52900e5e64c0SMuchun Song *ptepp = ptep; 52910e5e64c0SMuchun Song return 0; 52920e5e64c0SMuchun Song unlock: 52930e5e64c0SMuchun Song pte_unmap_unlock(ptep, *ptlp); 52940e5e64c0SMuchun Song out: 52950e5e64c0SMuchun Song return -EINVAL; 52969fd6dad1SPaolo Bonzini } 52979fd6dad1SPaolo Bonzini EXPORT_SYMBOL_GPL(follow_pte); 52989fd6dad1SPaolo Bonzini 52999fd6dad1SPaolo Bonzini /** 53003b6748e2SJohannes Weiner * follow_pfn - look up PFN at a user virtual address 53013b6748e2SJohannes Weiner * @vma: memory mapping 53023b6748e2SJohannes Weiner * @address: user virtual address 53033b6748e2SJohannes Weiner * @pfn: location to store found PFN 53043b6748e2SJohannes Weiner * 53053b6748e2SJohannes Weiner * Only IO mappings and raw PFN mappings are allowed. 53063b6748e2SJohannes Weiner * 53079fd6dad1SPaolo Bonzini * This function does not allow the caller to read the permissions 53089fd6dad1SPaolo Bonzini * of the PTE. Do not use it. 53099fd6dad1SPaolo Bonzini * 5310a862f68aSMike Rapoport * Return: zero and the pfn at @pfn on success, -ve otherwise. 53113b6748e2SJohannes Weiner */ 53123b6748e2SJohannes Weiner int follow_pfn(struct vm_area_struct *vma, unsigned long address, 53133b6748e2SJohannes Weiner unsigned long *pfn) 53143b6748e2SJohannes Weiner { 53153b6748e2SJohannes Weiner int ret = -EINVAL; 53163b6748e2SJohannes Weiner spinlock_t *ptl; 53173b6748e2SJohannes Weiner pte_t *ptep; 53183b6748e2SJohannes Weiner 53193b6748e2SJohannes Weiner if (!(vma->vm_flags & (VM_IO | VM_PFNMAP))) 53203b6748e2SJohannes Weiner return ret; 53213b6748e2SJohannes Weiner 53229fd6dad1SPaolo Bonzini ret = follow_pte(vma->vm_mm, address, &ptep, &ptl); 53233b6748e2SJohannes Weiner if (ret) 53243b6748e2SJohannes Weiner return ret; 53253b6748e2SJohannes Weiner *pfn = pte_pfn(*ptep); 53263b6748e2SJohannes Weiner pte_unmap_unlock(ptep, ptl); 53273b6748e2SJohannes Weiner return 0; 53283b6748e2SJohannes Weiner } 53293b6748e2SJohannes Weiner EXPORT_SYMBOL(follow_pfn); 53303b6748e2SJohannes Weiner 533128b2ee20SRik van Riel #ifdef CONFIG_HAVE_IOREMAP_PROT 5332d87fe660Svenkatesh.pallipadi@intel.com int follow_phys(struct vm_area_struct *vma, 533328b2ee20SRik van Riel unsigned long address, unsigned int flags, 5334d87fe660Svenkatesh.pallipadi@intel.com unsigned long *prot, resource_size_t *phys) 533528b2ee20SRik van Riel { 533603668a4dSJohannes Weiner int ret = -EINVAL; 533728b2ee20SRik van Riel pte_t *ptep, pte; 533828b2ee20SRik van Riel spinlock_t *ptl; 533928b2ee20SRik van Riel 5340d87fe660Svenkatesh.pallipadi@intel.com if (!(vma->vm_flags & (VM_IO | VM_PFNMAP))) 5341d87fe660Svenkatesh.pallipadi@intel.com goto out; 534228b2ee20SRik van Riel 53439fd6dad1SPaolo Bonzini if (follow_pte(vma->vm_mm, address, &ptep, &ptl)) 5344d87fe660Svenkatesh.pallipadi@intel.com goto out; 534528b2ee20SRik van Riel pte = *ptep; 534603668a4dSJohannes Weiner 5347f6f37321SLinus Torvalds if ((flags & FOLL_WRITE) && !pte_write(pte)) 534828b2ee20SRik van Riel goto unlock; 534928b2ee20SRik van Riel 535028b2ee20SRik van Riel *prot = pgprot_val(pte_pgprot(pte)); 535103668a4dSJohannes Weiner *phys = (resource_size_t)pte_pfn(pte) << PAGE_SHIFT; 535228b2ee20SRik van Riel 535303668a4dSJohannes Weiner ret = 0; 535428b2ee20SRik van Riel unlock: 535528b2ee20SRik van Riel pte_unmap_unlock(ptep, ptl); 535628b2ee20SRik van Riel out: 5357d87fe660Svenkatesh.pallipadi@intel.com return ret; 535828b2ee20SRik van Riel } 535928b2ee20SRik van Riel 536096667f8aSDaniel Vetter /** 536196667f8aSDaniel Vetter * generic_access_phys - generic implementation for iomem mmap access 536296667f8aSDaniel Vetter * @vma: the vma to access 5363f0953a1bSIngo Molnar * @addr: userspace address, not relative offset within @vma 536496667f8aSDaniel Vetter * @buf: buffer to read/write 536596667f8aSDaniel Vetter * @len: length of transfer 536696667f8aSDaniel Vetter * @write: set to FOLL_WRITE when writing, otherwise reading 536796667f8aSDaniel Vetter * 536896667f8aSDaniel Vetter * This is a generic implementation for &vm_operations_struct.access for an 536996667f8aSDaniel Vetter * iomem mapping. This callback is used by access_process_vm() when the @vma is 537096667f8aSDaniel Vetter * not page based. 537196667f8aSDaniel Vetter */ 537228b2ee20SRik van Riel int generic_access_phys(struct vm_area_struct *vma, unsigned long addr, 537328b2ee20SRik van Riel void *buf, int len, int write) 537428b2ee20SRik van Riel { 537528b2ee20SRik van Riel resource_size_t phys_addr; 537628b2ee20SRik van Riel unsigned long prot = 0; 53772bc7273bSKOSAKI Motohiro void __iomem *maddr; 537896667f8aSDaniel Vetter pte_t *ptep, pte; 537996667f8aSDaniel Vetter spinlock_t *ptl; 538096667f8aSDaniel Vetter int offset = offset_in_page(addr); 538196667f8aSDaniel Vetter int ret = -EINVAL; 538228b2ee20SRik van Riel 538396667f8aSDaniel Vetter if (!(vma->vm_flags & (VM_IO | VM_PFNMAP))) 538496667f8aSDaniel Vetter return -EINVAL; 538596667f8aSDaniel Vetter 538696667f8aSDaniel Vetter retry: 5387e913a8cdSLinus Torvalds if (follow_pte(vma->vm_mm, addr, &ptep, &ptl)) 538896667f8aSDaniel Vetter return -EINVAL; 538996667f8aSDaniel Vetter pte = *ptep; 539096667f8aSDaniel Vetter pte_unmap_unlock(ptep, ptl); 539196667f8aSDaniel Vetter 539296667f8aSDaniel Vetter prot = pgprot_val(pte_pgprot(pte)); 539396667f8aSDaniel Vetter phys_addr = (resource_size_t)pte_pfn(pte) << PAGE_SHIFT; 539496667f8aSDaniel Vetter 539596667f8aSDaniel Vetter if ((write & FOLL_WRITE) && !pte_write(pte)) 539628b2ee20SRik van Riel return -EINVAL; 539728b2ee20SRik van Riel 53989cb12d7bSGrazvydas Ignotas maddr = ioremap_prot(phys_addr, PAGE_ALIGN(len + offset), prot); 539924eee1e4Sjie@chenjie6@huwei.com if (!maddr) 540024eee1e4Sjie@chenjie6@huwei.com return -ENOMEM; 540124eee1e4Sjie@chenjie6@huwei.com 5402e913a8cdSLinus Torvalds if (follow_pte(vma->vm_mm, addr, &ptep, &ptl)) 540396667f8aSDaniel Vetter goto out_unmap; 540496667f8aSDaniel Vetter 540596667f8aSDaniel Vetter if (!pte_same(pte, *ptep)) { 540696667f8aSDaniel Vetter pte_unmap_unlock(ptep, ptl); 540796667f8aSDaniel Vetter iounmap(maddr); 540896667f8aSDaniel Vetter 540996667f8aSDaniel Vetter goto retry; 541096667f8aSDaniel Vetter } 541196667f8aSDaniel Vetter 541228b2ee20SRik van Riel if (write) 541328b2ee20SRik van Riel memcpy_toio(maddr + offset, buf, len); 541428b2ee20SRik van Riel else 541528b2ee20SRik van Riel memcpy_fromio(buf, maddr + offset, len); 541696667f8aSDaniel Vetter ret = len; 541796667f8aSDaniel Vetter pte_unmap_unlock(ptep, ptl); 541896667f8aSDaniel Vetter out_unmap: 541928b2ee20SRik van Riel iounmap(maddr); 542028b2ee20SRik van Riel 542196667f8aSDaniel Vetter return ret; 542228b2ee20SRik van Riel } 54235a73633eSUwe Kleine-König EXPORT_SYMBOL_GPL(generic_access_phys); 542428b2ee20SRik van Riel #endif 542528b2ee20SRik van Riel 54260ec76a11SDavid Howells /* 5427d3f5ffcaSJohn Hubbard * Access another process' address space as given in mm. 54280ec76a11SDavid Howells */ 5429d3f5ffcaSJohn Hubbard int __access_remote_vm(struct mm_struct *mm, unsigned long addr, void *buf, 5430d3f5ffcaSJohn Hubbard int len, unsigned int gup_flags) 54310ec76a11SDavid Howells { 54320ec76a11SDavid Howells struct vm_area_struct *vma; 54330ec76a11SDavid Howells void *old_buf = buf; 5434442486ecSLorenzo Stoakes int write = gup_flags & FOLL_WRITE; 54350ec76a11SDavid Howells 5436d8ed45c5SMichel Lespinasse if (mmap_read_lock_killable(mm)) 54371e426fe2SKonstantin Khlebnikov return 0; 54381e426fe2SKonstantin Khlebnikov 5439183ff22bSSimon Arlott /* ignore errors, just check how much was successfully transferred */ 54400ec76a11SDavid Howells while (len) { 54410ec76a11SDavid Howells int bytes, ret, offset; 54420ec76a11SDavid Howells void *maddr; 544328b2ee20SRik van Riel struct page *page = NULL; 54440ec76a11SDavid Howells 544564019a2eSPeter Xu ret = get_user_pages_remote(mm, addr, 1, 54465b56d49fSLorenzo Stoakes gup_flags, &page, &vma, NULL); 544728b2ee20SRik van Riel if (ret <= 0) { 5448dbffcd03SRik van Riel #ifndef CONFIG_HAVE_IOREMAP_PROT 5449dbffcd03SRik van Riel break; 5450dbffcd03SRik van Riel #else 545128b2ee20SRik van Riel /* 545228b2ee20SRik van Riel * Check if this is a VM_IO | VM_PFNMAP VMA, which 545328b2ee20SRik van Riel * we can access using slightly different code. 545428b2ee20SRik van Riel */ 54553e418f98SLiam Howlett vma = vma_lookup(mm, addr); 54563e418f98SLiam Howlett if (!vma) 54570ec76a11SDavid Howells break; 545828b2ee20SRik van Riel if (vma->vm_ops && vma->vm_ops->access) 545928b2ee20SRik van Riel ret = vma->vm_ops->access(vma, addr, buf, 546028b2ee20SRik van Riel len, write); 546128b2ee20SRik van Riel if (ret <= 0) 546228b2ee20SRik van Riel break; 546328b2ee20SRik van Riel bytes = ret; 5464dbffcd03SRik van Riel #endif 546528b2ee20SRik van Riel } else { 54660ec76a11SDavid Howells bytes = len; 54670ec76a11SDavid Howells offset = addr & (PAGE_SIZE-1); 54680ec76a11SDavid Howells if (bytes > PAGE_SIZE-offset) 54690ec76a11SDavid Howells bytes = PAGE_SIZE-offset; 54700ec76a11SDavid Howells 54710ec76a11SDavid Howells maddr = kmap(page); 54720ec76a11SDavid Howells if (write) { 54730ec76a11SDavid Howells copy_to_user_page(vma, page, addr, 54740ec76a11SDavid Howells maddr + offset, buf, bytes); 54750ec76a11SDavid Howells set_page_dirty_lock(page); 54760ec76a11SDavid Howells } else { 54770ec76a11SDavid Howells copy_from_user_page(vma, page, addr, 54780ec76a11SDavid Howells buf, maddr + offset, bytes); 54790ec76a11SDavid Howells } 54800ec76a11SDavid Howells kunmap(page); 548109cbfeafSKirill A. Shutemov put_page(page); 548228b2ee20SRik van Riel } 54830ec76a11SDavid Howells len -= bytes; 54840ec76a11SDavid Howells buf += bytes; 54850ec76a11SDavid Howells addr += bytes; 54860ec76a11SDavid Howells } 5487d8ed45c5SMichel Lespinasse mmap_read_unlock(mm); 54880ec76a11SDavid Howells 54890ec76a11SDavid Howells return buf - old_buf; 54900ec76a11SDavid Howells } 549103252919SAndi Kleen 54925ddd36b9SStephen Wilson /** 5493ae91dbfcSRandy Dunlap * access_remote_vm - access another process' address space 54945ddd36b9SStephen Wilson * @mm: the mm_struct of the target address space 54955ddd36b9SStephen Wilson * @addr: start address to access 54965ddd36b9SStephen Wilson * @buf: source or destination buffer 54975ddd36b9SStephen Wilson * @len: number of bytes to transfer 54986347e8d5SLorenzo Stoakes * @gup_flags: flags modifying lookup behaviour 54995ddd36b9SStephen Wilson * 55005ddd36b9SStephen Wilson * The caller must hold a reference on @mm. 5501a862f68aSMike Rapoport * 5502a862f68aSMike Rapoport * Return: number of bytes copied from source to destination. 55035ddd36b9SStephen Wilson */ 55045ddd36b9SStephen Wilson int access_remote_vm(struct mm_struct *mm, unsigned long addr, 55056347e8d5SLorenzo Stoakes void *buf, int len, unsigned int gup_flags) 55065ddd36b9SStephen Wilson { 5507d3f5ffcaSJohn Hubbard return __access_remote_vm(mm, addr, buf, len, gup_flags); 55085ddd36b9SStephen Wilson } 55095ddd36b9SStephen Wilson 551003252919SAndi Kleen /* 5511206cb636SStephen Wilson * Access another process' address space. 5512206cb636SStephen Wilson * Source/target buffer must be kernel space, 5513206cb636SStephen Wilson * Do not walk the page table directly, use get_user_pages 5514206cb636SStephen Wilson */ 5515206cb636SStephen Wilson int access_process_vm(struct task_struct *tsk, unsigned long addr, 5516f307ab6dSLorenzo Stoakes void *buf, int len, unsigned int gup_flags) 5517206cb636SStephen Wilson { 5518206cb636SStephen Wilson struct mm_struct *mm; 5519206cb636SStephen Wilson int ret; 5520206cb636SStephen Wilson 5521206cb636SStephen Wilson mm = get_task_mm(tsk); 5522206cb636SStephen Wilson if (!mm) 5523206cb636SStephen Wilson return 0; 5524206cb636SStephen Wilson 5525d3f5ffcaSJohn Hubbard ret = __access_remote_vm(mm, addr, buf, len, gup_flags); 5526442486ecSLorenzo Stoakes 5527206cb636SStephen Wilson mmput(mm); 5528206cb636SStephen Wilson 5529206cb636SStephen Wilson return ret; 5530206cb636SStephen Wilson } 5531fcd35857SCatalin Marinas EXPORT_SYMBOL_GPL(access_process_vm); 5532206cb636SStephen Wilson 553303252919SAndi Kleen /* 553403252919SAndi Kleen * Print the name of a VMA. 553503252919SAndi Kleen */ 553603252919SAndi Kleen void print_vma_addr(char *prefix, unsigned long ip) 553703252919SAndi Kleen { 553803252919SAndi Kleen struct mm_struct *mm = current->mm; 553903252919SAndi Kleen struct vm_area_struct *vma; 554003252919SAndi Kleen 5541e8bff74aSIngo Molnar /* 55420a7f682dSMichal Hocko * we might be running from an atomic context so we cannot sleep 5543e8bff74aSIngo Molnar */ 5544d8ed45c5SMichel Lespinasse if (!mmap_read_trylock(mm)) 5545e8bff74aSIngo Molnar return; 5546e8bff74aSIngo Molnar 554703252919SAndi Kleen vma = find_vma(mm, ip); 554803252919SAndi Kleen if (vma && vma->vm_file) { 554903252919SAndi Kleen struct file *f = vma->vm_file; 55500a7f682dSMichal Hocko char *buf = (char *)__get_free_page(GFP_NOWAIT); 555103252919SAndi Kleen if (buf) { 55522fbc57c5SAndy Shevchenko char *p; 555303252919SAndi Kleen 55549bf39ab2SMiklos Szeredi p = file_path(f, buf, PAGE_SIZE); 555503252919SAndi Kleen if (IS_ERR(p)) 555603252919SAndi Kleen p = "?"; 55572fbc57c5SAndy Shevchenko printk("%s%s[%lx+%lx]", prefix, kbasename(p), 555803252919SAndi Kleen vma->vm_start, 555903252919SAndi Kleen vma->vm_end - vma->vm_start); 556003252919SAndi Kleen free_page((unsigned long)buf); 556103252919SAndi Kleen } 556203252919SAndi Kleen } 5563d8ed45c5SMichel Lespinasse mmap_read_unlock(mm); 556403252919SAndi Kleen } 55653ee1afa3SNick Piggin 5566662bbcb2SMichael S. Tsirkin #if defined(CONFIG_PROVE_LOCKING) || defined(CONFIG_DEBUG_ATOMIC_SLEEP) 55679ec23531SDavid Hildenbrand void __might_fault(const char *file, int line) 55683ee1afa3SNick Piggin { 55699ec23531SDavid Hildenbrand if (pagefault_disabled()) 5570662bbcb2SMichael S. Tsirkin return; 557142a38756SThomas Gleixner __might_sleep(file, line); 55729ec23531SDavid Hildenbrand #if defined(CONFIG_DEBUG_ATOMIC_SLEEP) 5573662bbcb2SMichael S. Tsirkin if (current->mm) 5574da1c55f1SMichel Lespinasse might_lock_read(¤t->mm->mmap_lock); 55759ec23531SDavid Hildenbrand #endif 55763ee1afa3SNick Piggin } 55779ec23531SDavid Hildenbrand EXPORT_SYMBOL(__might_fault); 55783ee1afa3SNick Piggin #endif 557947ad8475SAndrea Arcangeli 558047ad8475SAndrea Arcangeli #if defined(CONFIG_TRANSPARENT_HUGEPAGE) || defined(CONFIG_HUGETLBFS) 5581c6ddfb6cSHuang Ying /* 5582c6ddfb6cSHuang Ying * Process all subpages of the specified huge page with the specified 5583c6ddfb6cSHuang Ying * operation. The target subpage will be processed last to keep its 5584c6ddfb6cSHuang Ying * cache lines hot. 5585c6ddfb6cSHuang Ying */ 5586c6ddfb6cSHuang Ying static inline void process_huge_page( 5587c6ddfb6cSHuang Ying unsigned long addr_hint, unsigned int pages_per_huge_page, 5588c6ddfb6cSHuang Ying void (*process_subpage)(unsigned long addr, int idx, void *arg), 5589c6ddfb6cSHuang Ying void *arg) 5590c6ddfb6cSHuang Ying { 5591c6ddfb6cSHuang Ying int i, n, base, l; 5592c6ddfb6cSHuang Ying unsigned long addr = addr_hint & 5593c6ddfb6cSHuang Ying ~(((unsigned long)pages_per_huge_page << PAGE_SHIFT) - 1); 5594c6ddfb6cSHuang Ying 5595c6ddfb6cSHuang Ying /* Process target subpage last to keep its cache lines hot */ 5596c6ddfb6cSHuang Ying might_sleep(); 5597c6ddfb6cSHuang Ying n = (addr_hint - addr) / PAGE_SIZE; 5598c6ddfb6cSHuang Ying if (2 * n <= pages_per_huge_page) { 5599c6ddfb6cSHuang Ying /* If target subpage in first half of huge page */ 5600c6ddfb6cSHuang Ying base = 0; 5601c6ddfb6cSHuang Ying l = n; 5602c6ddfb6cSHuang Ying /* Process subpages at the end of huge page */ 5603c6ddfb6cSHuang Ying for (i = pages_per_huge_page - 1; i >= 2 * n; i--) { 5604c6ddfb6cSHuang Ying cond_resched(); 5605c6ddfb6cSHuang Ying process_subpage(addr + i * PAGE_SIZE, i, arg); 5606c6ddfb6cSHuang Ying } 5607c6ddfb6cSHuang Ying } else { 5608c6ddfb6cSHuang Ying /* If target subpage in second half of huge page */ 5609c6ddfb6cSHuang Ying base = pages_per_huge_page - 2 * (pages_per_huge_page - n); 5610c6ddfb6cSHuang Ying l = pages_per_huge_page - n; 5611c6ddfb6cSHuang Ying /* Process subpages at the begin of huge page */ 5612c6ddfb6cSHuang Ying for (i = 0; i < base; i++) { 5613c6ddfb6cSHuang Ying cond_resched(); 5614c6ddfb6cSHuang Ying process_subpage(addr + i * PAGE_SIZE, i, arg); 5615c6ddfb6cSHuang Ying } 5616c6ddfb6cSHuang Ying } 5617c6ddfb6cSHuang Ying /* 5618c6ddfb6cSHuang Ying * Process remaining subpages in left-right-left-right pattern 5619c6ddfb6cSHuang Ying * towards the target subpage 5620c6ddfb6cSHuang Ying */ 5621c6ddfb6cSHuang Ying for (i = 0; i < l; i++) { 5622c6ddfb6cSHuang Ying int left_idx = base + i; 5623c6ddfb6cSHuang Ying int right_idx = base + 2 * l - 1 - i; 5624c6ddfb6cSHuang Ying 5625c6ddfb6cSHuang Ying cond_resched(); 5626c6ddfb6cSHuang Ying process_subpage(addr + left_idx * PAGE_SIZE, left_idx, arg); 5627c6ddfb6cSHuang Ying cond_resched(); 5628c6ddfb6cSHuang Ying process_subpage(addr + right_idx * PAGE_SIZE, right_idx, arg); 5629c6ddfb6cSHuang Ying } 5630c6ddfb6cSHuang Ying } 5631c6ddfb6cSHuang Ying 563247ad8475SAndrea Arcangeli static void clear_gigantic_page(struct page *page, 563347ad8475SAndrea Arcangeli unsigned long addr, 563447ad8475SAndrea Arcangeli unsigned int pages_per_huge_page) 563547ad8475SAndrea Arcangeli { 563647ad8475SAndrea Arcangeli int i; 563747ad8475SAndrea Arcangeli struct page *p = page; 563847ad8475SAndrea Arcangeli 563947ad8475SAndrea Arcangeli might_sleep(); 564047ad8475SAndrea Arcangeli for (i = 0; i < pages_per_huge_page; 564147ad8475SAndrea Arcangeli i++, p = mem_map_next(p, page, i)) { 564247ad8475SAndrea Arcangeli cond_resched(); 564347ad8475SAndrea Arcangeli clear_user_highpage(p, addr + i * PAGE_SIZE); 564447ad8475SAndrea Arcangeli } 564547ad8475SAndrea Arcangeli } 5646c6ddfb6cSHuang Ying 5647c6ddfb6cSHuang Ying static void clear_subpage(unsigned long addr, int idx, void *arg) 5648c6ddfb6cSHuang Ying { 5649c6ddfb6cSHuang Ying struct page *page = arg; 5650c6ddfb6cSHuang Ying 5651c6ddfb6cSHuang Ying clear_user_highpage(page + idx, addr); 5652c6ddfb6cSHuang Ying } 5653c6ddfb6cSHuang Ying 565447ad8475SAndrea Arcangeli void clear_huge_page(struct page *page, 5655c79b57e4SHuang Ying unsigned long addr_hint, unsigned int pages_per_huge_page) 565647ad8475SAndrea Arcangeli { 5657c79b57e4SHuang Ying unsigned long addr = addr_hint & 5658c79b57e4SHuang Ying ~(((unsigned long)pages_per_huge_page << PAGE_SHIFT) - 1); 565947ad8475SAndrea Arcangeli 566047ad8475SAndrea Arcangeli if (unlikely(pages_per_huge_page > MAX_ORDER_NR_PAGES)) { 566147ad8475SAndrea Arcangeli clear_gigantic_page(page, addr, pages_per_huge_page); 566247ad8475SAndrea Arcangeli return; 566347ad8475SAndrea Arcangeli } 566447ad8475SAndrea Arcangeli 5665c6ddfb6cSHuang Ying process_huge_page(addr_hint, pages_per_huge_page, clear_subpage, page); 566647ad8475SAndrea Arcangeli } 566747ad8475SAndrea Arcangeli 566847ad8475SAndrea Arcangeli static void copy_user_gigantic_page(struct page *dst, struct page *src, 566947ad8475SAndrea Arcangeli unsigned long addr, 567047ad8475SAndrea Arcangeli struct vm_area_struct *vma, 567147ad8475SAndrea Arcangeli unsigned int pages_per_huge_page) 567247ad8475SAndrea Arcangeli { 567347ad8475SAndrea Arcangeli int i; 567447ad8475SAndrea Arcangeli struct page *dst_base = dst; 567547ad8475SAndrea Arcangeli struct page *src_base = src; 567647ad8475SAndrea Arcangeli 567747ad8475SAndrea Arcangeli for (i = 0; i < pages_per_huge_page; ) { 567847ad8475SAndrea Arcangeli cond_resched(); 567947ad8475SAndrea Arcangeli copy_user_highpage(dst, src, addr + i*PAGE_SIZE, vma); 568047ad8475SAndrea Arcangeli 568147ad8475SAndrea Arcangeli i++; 568247ad8475SAndrea Arcangeli dst = mem_map_next(dst, dst_base, i); 568347ad8475SAndrea Arcangeli src = mem_map_next(src, src_base, i); 568447ad8475SAndrea Arcangeli } 568547ad8475SAndrea Arcangeli } 568647ad8475SAndrea Arcangeli 5687c9f4cd71SHuang Ying struct copy_subpage_arg { 5688c9f4cd71SHuang Ying struct page *dst; 5689c9f4cd71SHuang Ying struct page *src; 5690c9f4cd71SHuang Ying struct vm_area_struct *vma; 5691c9f4cd71SHuang Ying }; 5692c9f4cd71SHuang Ying 5693c9f4cd71SHuang Ying static void copy_subpage(unsigned long addr, int idx, void *arg) 5694c9f4cd71SHuang Ying { 5695c9f4cd71SHuang Ying struct copy_subpage_arg *copy_arg = arg; 5696c9f4cd71SHuang Ying 5697c9f4cd71SHuang Ying copy_user_highpage(copy_arg->dst + idx, copy_arg->src + idx, 5698c9f4cd71SHuang Ying addr, copy_arg->vma); 5699c9f4cd71SHuang Ying } 5700c9f4cd71SHuang Ying 570147ad8475SAndrea Arcangeli void copy_user_huge_page(struct page *dst, struct page *src, 5702c9f4cd71SHuang Ying unsigned long addr_hint, struct vm_area_struct *vma, 570347ad8475SAndrea Arcangeli unsigned int pages_per_huge_page) 570447ad8475SAndrea Arcangeli { 5705c9f4cd71SHuang Ying unsigned long addr = addr_hint & 5706c9f4cd71SHuang Ying ~(((unsigned long)pages_per_huge_page << PAGE_SHIFT) - 1); 5707c9f4cd71SHuang Ying struct copy_subpage_arg arg = { 5708c9f4cd71SHuang Ying .dst = dst, 5709c9f4cd71SHuang Ying .src = src, 5710c9f4cd71SHuang Ying .vma = vma, 5711c9f4cd71SHuang Ying }; 571247ad8475SAndrea Arcangeli 571347ad8475SAndrea Arcangeli if (unlikely(pages_per_huge_page > MAX_ORDER_NR_PAGES)) { 571447ad8475SAndrea Arcangeli copy_user_gigantic_page(dst, src, addr, vma, 571547ad8475SAndrea Arcangeli pages_per_huge_page); 571647ad8475SAndrea Arcangeli return; 571747ad8475SAndrea Arcangeli } 571847ad8475SAndrea Arcangeli 5719c9f4cd71SHuang Ying process_huge_page(addr_hint, pages_per_huge_page, copy_subpage, &arg); 572047ad8475SAndrea Arcangeli } 5721fa4d75c1SMike Kravetz 5722fa4d75c1SMike Kravetz long copy_huge_page_from_user(struct page *dst_page, 5723fa4d75c1SMike Kravetz const void __user *usr_src, 5724810a56b9SMike Kravetz unsigned int pages_per_huge_page, 5725810a56b9SMike Kravetz bool allow_pagefault) 5726fa4d75c1SMike Kravetz { 5727fa4d75c1SMike Kravetz void *page_kaddr; 5728fa4d75c1SMike Kravetz unsigned long i, rc = 0; 5729fa4d75c1SMike Kravetz unsigned long ret_val = pages_per_huge_page * PAGE_SIZE; 57303272cfc2SMike Kravetz struct page *subpage = dst_page; 5731fa4d75c1SMike Kravetz 57323272cfc2SMike Kravetz for (i = 0; i < pages_per_huge_page; 57333272cfc2SMike Kravetz i++, subpage = mem_map_next(subpage, dst_page, i)) { 5734810a56b9SMike Kravetz if (allow_pagefault) 57353272cfc2SMike Kravetz page_kaddr = kmap(subpage); 5736810a56b9SMike Kravetz else 57373272cfc2SMike Kravetz page_kaddr = kmap_atomic(subpage); 5738fa4d75c1SMike Kravetz rc = copy_from_user(page_kaddr, 5739b063e374SAmit Daniel Kachhap usr_src + i * PAGE_SIZE, PAGE_SIZE); 5740810a56b9SMike Kravetz if (allow_pagefault) 57413272cfc2SMike Kravetz kunmap(subpage); 5742810a56b9SMike Kravetz else 5743fa4d75c1SMike Kravetz kunmap_atomic(page_kaddr); 5744fa4d75c1SMike Kravetz 5745fa4d75c1SMike Kravetz ret_val -= (PAGE_SIZE - rc); 5746fa4d75c1SMike Kravetz if (rc) 5747fa4d75c1SMike Kravetz break; 5748fa4d75c1SMike Kravetz 5749e763243cSMuchun Song flush_dcache_page(subpage); 5750e763243cSMuchun Song 5751fa4d75c1SMike Kravetz cond_resched(); 5752fa4d75c1SMike Kravetz } 5753fa4d75c1SMike Kravetz return ret_val; 5754fa4d75c1SMike Kravetz } 575547ad8475SAndrea Arcangeli #endif /* CONFIG_TRANSPARENT_HUGEPAGE || CONFIG_HUGETLBFS */ 575649076ec2SKirill A. Shutemov 575740b64acdSOlof Johansson #if USE_SPLIT_PTE_PTLOCKS && ALLOC_SPLIT_PTLOCKS 5758b35f1819SKirill A. Shutemov 5759b35f1819SKirill A. Shutemov static struct kmem_cache *page_ptl_cachep; 5760b35f1819SKirill A. Shutemov 5761b35f1819SKirill A. Shutemov void __init ptlock_cache_init(void) 5762b35f1819SKirill A. Shutemov { 5763b35f1819SKirill A. Shutemov page_ptl_cachep = kmem_cache_create("page->ptl", sizeof(spinlock_t), 0, 5764b35f1819SKirill A. Shutemov SLAB_PANIC, NULL); 5765b35f1819SKirill A. Shutemov } 5766b35f1819SKirill A. Shutemov 5767539edb58SPeter Zijlstra bool ptlock_alloc(struct page *page) 576849076ec2SKirill A. Shutemov { 576949076ec2SKirill A. Shutemov spinlock_t *ptl; 577049076ec2SKirill A. Shutemov 5771b35f1819SKirill A. Shutemov ptl = kmem_cache_alloc(page_ptl_cachep, GFP_KERNEL); 577249076ec2SKirill A. Shutemov if (!ptl) 577349076ec2SKirill A. Shutemov return false; 5774539edb58SPeter Zijlstra page->ptl = ptl; 577549076ec2SKirill A. Shutemov return true; 577649076ec2SKirill A. Shutemov } 577749076ec2SKirill A. Shutemov 5778539edb58SPeter Zijlstra void ptlock_free(struct page *page) 577949076ec2SKirill A. Shutemov { 5780b35f1819SKirill A. Shutemov kmem_cache_free(page_ptl_cachep, page->ptl); 578149076ec2SKirill A. Shutemov } 578249076ec2SKirill A. Shutemov #endif 5783