1457c8996SThomas Gleixner // SPDX-License-Identifier: GPL-2.0-only 21da177e4SLinus Torvalds /* 31da177e4SLinus Torvalds * Generic hugetlb support. 46d49e352SNadia Yvette Chambers * (C) Nadia Yvette Chambers, April 2004 51da177e4SLinus Torvalds */ 61da177e4SLinus Torvalds #include <linux/list.h> 71da177e4SLinus Torvalds #include <linux/init.h> 81da177e4SLinus Torvalds #include <linux/mm.h> 9e1759c21SAlexey Dobriyan #include <linux/seq_file.h> 101da177e4SLinus Torvalds #include <linux/sysctl.h> 111da177e4SLinus Torvalds #include <linux/highmem.h> 12cddb8a5cSAndrea Arcangeli #include <linux/mmu_notifier.h> 131da177e4SLinus Torvalds #include <linux/nodemask.h> 1463551ae0SDavid Gibson #include <linux/pagemap.h> 155da7ca86SChristoph Lameter #include <linux/mempolicy.h> 163b32123dSGideon Israel Dsouza #include <linux/compiler.h> 17aea47ff3SChristoph Lameter #include <linux/cpuset.h> 183935baa9SDavid Gibson #include <linux/mutex.h> 1997ad1087SMike Rapoport #include <linux/memblock.h> 20a3437870SNishanth Aravamudan #include <linux/sysfs.h> 215a0e3ad6STejun Heo #include <linux/slab.h> 22bbe88753SJoonsoo Kim #include <linux/sched/mm.h> 2363489f8eSMike Kravetz #include <linux/mmdebug.h> 24174cd4b1SIngo Molnar #include <linux/sched/signal.h> 250fe6e20bSNaoya Horiguchi #include <linux/rmap.h> 26c6247f72SMatthew Wilcox #include <linux/string_helpers.h> 27fd6a03edSNaoya Horiguchi #include <linux/swap.h> 28fd6a03edSNaoya Horiguchi #include <linux/swapops.h> 298382d914SDavidlohr Bueso #include <linux/jhash.h> 3098fa15f3SAnshuman Khandual #include <linux/numa.h> 31c77c0a8aSWaiman Long #include <linux/llist.h> 32cf11e85fSRoman Gushchin #include <linux/cma.h> 338cc5fcbbSMina Almasry #include <linux/migrate.h> 34f9317f77SMike Kravetz #include <linux/nospec.h> 35662ce1dcSYang Yang #include <linux/delayacct.h> 36b958d4d0SMuchun Song #include <linux/memory.h> 37d6606683SLinus Torvalds 3863551ae0SDavid Gibson #include <asm/page.h> 39ca15ca40SMike Rapoport #include <asm/pgalloc.h> 4024669e58SAneesh Kumar K.V #include <asm/tlb.h> 4163551ae0SDavid Gibson 4224669e58SAneesh Kumar K.V #include <linux/io.h> 4363551ae0SDavid Gibson #include <linux/hugetlb.h> 449dd540e2SAneesh Kumar K.V #include <linux/hugetlb_cgroup.h> 459a305230SLee Schermerhorn #include <linux/node.h> 46ab5ac90aSMichal Hocko #include <linux/page_owner.h> 477835e98bSNick Piggin #include "internal.h" 48f41f2ed4SMuchun Song #include "hugetlb_vmemmap.h" 491da177e4SLinus Torvalds 50c3f38a38SAneesh Kumar K.V int hugetlb_max_hstate __read_mostly; 51e5ff2159SAndi Kleen unsigned int default_hstate_idx; 52e5ff2159SAndi Kleen struct hstate hstates[HUGE_MAX_HSTATE]; 53cf11e85fSRoman Gushchin 54dbda8feaSBarry Song #ifdef CONFIG_CMA 55cf11e85fSRoman Gushchin static struct cma *hugetlb_cma[MAX_NUMNODES]; 5638e719abSBaolin Wang static unsigned long hugetlb_cma_size_in_node[MAX_NUMNODES] __initdata; 57a01f4390SMike Kravetz static bool hugetlb_cma_page(struct page *page, unsigned int order) 58a01f4390SMike Kravetz { 59a01f4390SMike Kravetz return cma_pages_valid(hugetlb_cma[page_to_nid(page)], page, 60a01f4390SMike Kravetz 1 << order); 61a01f4390SMike Kravetz } 62a01f4390SMike Kravetz #else 63a01f4390SMike Kravetz static bool hugetlb_cma_page(struct page *page, unsigned int order) 64a01f4390SMike Kravetz { 65a01f4390SMike Kravetz return false; 66a01f4390SMike Kravetz } 67dbda8feaSBarry Song #endif 68dbda8feaSBarry Song static unsigned long hugetlb_cma_size __initdata; 69cf11e85fSRoman Gushchin 7053ba51d2SJon Tollefson __initdata LIST_HEAD(huge_boot_pages); 7153ba51d2SJon Tollefson 72e5ff2159SAndi Kleen /* for command line parsing */ 73e5ff2159SAndi Kleen static struct hstate * __initdata parsed_hstate; 74e5ff2159SAndi Kleen static unsigned long __initdata default_hstate_max_huge_pages; 759fee021dSVaishali Thakkar static bool __initdata parsed_valid_hugepagesz = true; 76282f4214SMike Kravetz static bool __initdata parsed_default_hugepagesz; 77b5389086SZhenguo Yao static unsigned int default_hugepages_in_node[MAX_NUMNODES] __initdata; 78e5ff2159SAndi Kleen 793935baa9SDavid Gibson /* 8031caf665SNaoya Horiguchi * Protects updates to hugepage_freelists, hugepage_activelist, nr_huge_pages, 8131caf665SNaoya Horiguchi * free_huge_pages, and surplus_huge_pages. 823935baa9SDavid Gibson */ 83c3f38a38SAneesh Kumar K.V DEFINE_SPINLOCK(hugetlb_lock); 840bd0f9fbSEric Paris 858382d914SDavidlohr Bueso /* 868382d914SDavidlohr Bueso * Serializes faults on the same logical page. This is used to 878382d914SDavidlohr Bueso * prevent spurious OOMs when the hugepage pool is fully utilized. 888382d914SDavidlohr Bueso */ 898382d914SDavidlohr Bueso static int num_fault_mutexes; 90c672c7f2SMike Kravetz struct mutex *hugetlb_fault_mutex_table ____cacheline_aligned_in_smp; 918382d914SDavidlohr Bueso 927ca02d0aSMike Kravetz /* Forward declaration */ 937ca02d0aSMike Kravetz static int hugetlb_acct_memory(struct hstate *h, long delta); 948d9bfb26SMike Kravetz static void hugetlb_vma_lock_free(struct vm_area_struct *vma); 958d9bfb26SMike Kravetz static void hugetlb_vma_lock_alloc(struct vm_area_struct *vma); 96ecfbd733SMike Kravetz static void __hugetlb_vma_unlock_write_free(struct vm_area_struct *vma); 977ca02d0aSMike Kravetz 981d88433bSMiaohe Lin static inline bool subpool_is_free(struct hugepage_subpool *spool) 991d88433bSMiaohe Lin { 1001d88433bSMiaohe Lin if (spool->count) 1011d88433bSMiaohe Lin return false; 1021d88433bSMiaohe Lin if (spool->max_hpages != -1) 1031d88433bSMiaohe Lin return spool->used_hpages == 0; 1041d88433bSMiaohe Lin if (spool->min_hpages != -1) 1051d88433bSMiaohe Lin return spool->rsv_hpages == spool->min_hpages; 1061d88433bSMiaohe Lin 1071d88433bSMiaohe Lin return true; 1081d88433bSMiaohe Lin } 1091d88433bSMiaohe Lin 110db71ef79SMike Kravetz static inline void unlock_or_release_subpool(struct hugepage_subpool *spool, 111db71ef79SMike Kravetz unsigned long irq_flags) 11290481622SDavid Gibson { 113db71ef79SMike Kravetz spin_unlock_irqrestore(&spool->lock, irq_flags); 11490481622SDavid Gibson 11590481622SDavid Gibson /* If no pages are used, and no other handles to the subpool 1167c8de358SEthon Paul * remain, give up any reservations based on minimum size and 1177ca02d0aSMike Kravetz * free the subpool */ 1181d88433bSMiaohe Lin if (subpool_is_free(spool)) { 1197ca02d0aSMike Kravetz if (spool->min_hpages != -1) 1207ca02d0aSMike Kravetz hugetlb_acct_memory(spool->hstate, 1217ca02d0aSMike Kravetz -spool->min_hpages); 12290481622SDavid Gibson kfree(spool); 12390481622SDavid Gibson } 1247ca02d0aSMike Kravetz } 12590481622SDavid Gibson 1267ca02d0aSMike Kravetz struct hugepage_subpool *hugepage_new_subpool(struct hstate *h, long max_hpages, 1277ca02d0aSMike Kravetz long min_hpages) 12890481622SDavid Gibson { 12990481622SDavid Gibson struct hugepage_subpool *spool; 13090481622SDavid Gibson 131c6a91820SMike Kravetz spool = kzalloc(sizeof(*spool), GFP_KERNEL); 13290481622SDavid Gibson if (!spool) 13390481622SDavid Gibson return NULL; 13490481622SDavid Gibson 13590481622SDavid Gibson spin_lock_init(&spool->lock); 13690481622SDavid Gibson spool->count = 1; 1377ca02d0aSMike Kravetz spool->max_hpages = max_hpages; 1387ca02d0aSMike Kravetz spool->hstate = h; 1397ca02d0aSMike Kravetz spool->min_hpages = min_hpages; 1407ca02d0aSMike Kravetz 1417ca02d0aSMike Kravetz if (min_hpages != -1 && hugetlb_acct_memory(h, min_hpages)) { 1427ca02d0aSMike Kravetz kfree(spool); 1437ca02d0aSMike Kravetz return NULL; 1447ca02d0aSMike Kravetz } 1457ca02d0aSMike Kravetz spool->rsv_hpages = min_hpages; 14690481622SDavid Gibson 14790481622SDavid Gibson return spool; 14890481622SDavid Gibson } 14990481622SDavid Gibson 15090481622SDavid Gibson void hugepage_put_subpool(struct hugepage_subpool *spool) 15190481622SDavid Gibson { 152db71ef79SMike Kravetz unsigned long flags; 153db71ef79SMike Kravetz 154db71ef79SMike Kravetz spin_lock_irqsave(&spool->lock, flags); 15590481622SDavid Gibson BUG_ON(!spool->count); 15690481622SDavid Gibson spool->count--; 157db71ef79SMike Kravetz unlock_or_release_subpool(spool, flags); 15890481622SDavid Gibson } 15990481622SDavid Gibson 1601c5ecae3SMike Kravetz /* 1611c5ecae3SMike Kravetz * Subpool accounting for allocating and reserving pages. 1621c5ecae3SMike Kravetz * Return -ENOMEM if there are not enough resources to satisfy the 1639e7ee400SRandy Dunlap * request. Otherwise, return the number of pages by which the 1641c5ecae3SMike Kravetz * global pools must be adjusted (upward). The returned value may 1651c5ecae3SMike Kravetz * only be different than the passed value (delta) in the case where 1667c8de358SEthon Paul * a subpool minimum size must be maintained. 1671c5ecae3SMike Kravetz */ 1681c5ecae3SMike Kravetz static long hugepage_subpool_get_pages(struct hugepage_subpool *spool, 16990481622SDavid Gibson long delta) 17090481622SDavid Gibson { 1711c5ecae3SMike Kravetz long ret = delta; 17290481622SDavid Gibson 17390481622SDavid Gibson if (!spool) 1741c5ecae3SMike Kravetz return ret; 17590481622SDavid Gibson 176db71ef79SMike Kravetz spin_lock_irq(&spool->lock); 17790481622SDavid Gibson 1781c5ecae3SMike Kravetz if (spool->max_hpages != -1) { /* maximum size accounting */ 1791c5ecae3SMike Kravetz if ((spool->used_hpages + delta) <= spool->max_hpages) 1801c5ecae3SMike Kravetz spool->used_hpages += delta; 1811c5ecae3SMike Kravetz else { 1821c5ecae3SMike Kravetz ret = -ENOMEM; 1831c5ecae3SMike Kravetz goto unlock_ret; 1841c5ecae3SMike Kravetz } 1851c5ecae3SMike Kravetz } 1861c5ecae3SMike Kravetz 18709a95e29SMike Kravetz /* minimum size accounting */ 18809a95e29SMike Kravetz if (spool->min_hpages != -1 && spool->rsv_hpages) { 1891c5ecae3SMike Kravetz if (delta > spool->rsv_hpages) { 1901c5ecae3SMike Kravetz /* 1911c5ecae3SMike Kravetz * Asking for more reserves than those already taken on 1921c5ecae3SMike Kravetz * behalf of subpool. Return difference. 1931c5ecae3SMike Kravetz */ 1941c5ecae3SMike Kravetz ret = delta - spool->rsv_hpages; 1951c5ecae3SMike Kravetz spool->rsv_hpages = 0; 1961c5ecae3SMike Kravetz } else { 1971c5ecae3SMike Kravetz ret = 0; /* reserves already accounted for */ 1981c5ecae3SMike Kravetz spool->rsv_hpages -= delta; 1991c5ecae3SMike Kravetz } 2001c5ecae3SMike Kravetz } 2011c5ecae3SMike Kravetz 2021c5ecae3SMike Kravetz unlock_ret: 203db71ef79SMike Kravetz spin_unlock_irq(&spool->lock); 20490481622SDavid Gibson return ret; 20590481622SDavid Gibson } 20690481622SDavid Gibson 2071c5ecae3SMike Kravetz /* 2081c5ecae3SMike Kravetz * Subpool accounting for freeing and unreserving pages. 2091c5ecae3SMike Kravetz * Return the number of global page reservations that must be dropped. 2101c5ecae3SMike Kravetz * The return value may only be different than the passed value (delta) 2111c5ecae3SMike Kravetz * in the case where a subpool minimum size must be maintained. 2121c5ecae3SMike Kravetz */ 2131c5ecae3SMike Kravetz static long hugepage_subpool_put_pages(struct hugepage_subpool *spool, 21490481622SDavid Gibson long delta) 21590481622SDavid Gibson { 2161c5ecae3SMike Kravetz long ret = delta; 217db71ef79SMike Kravetz unsigned long flags; 2181c5ecae3SMike Kravetz 21990481622SDavid Gibson if (!spool) 2201c5ecae3SMike Kravetz return delta; 22190481622SDavid Gibson 222db71ef79SMike Kravetz spin_lock_irqsave(&spool->lock, flags); 2231c5ecae3SMike Kravetz 2241c5ecae3SMike Kravetz if (spool->max_hpages != -1) /* maximum size accounting */ 22590481622SDavid Gibson spool->used_hpages -= delta; 2261c5ecae3SMike Kravetz 22709a95e29SMike Kravetz /* minimum size accounting */ 22809a95e29SMike Kravetz if (spool->min_hpages != -1 && spool->used_hpages < spool->min_hpages) { 2291c5ecae3SMike Kravetz if (spool->rsv_hpages + delta <= spool->min_hpages) 2301c5ecae3SMike Kravetz ret = 0; 2311c5ecae3SMike Kravetz else 2321c5ecae3SMike Kravetz ret = spool->rsv_hpages + delta - spool->min_hpages; 2331c5ecae3SMike Kravetz 2341c5ecae3SMike Kravetz spool->rsv_hpages += delta; 2351c5ecae3SMike Kravetz if (spool->rsv_hpages > spool->min_hpages) 2361c5ecae3SMike Kravetz spool->rsv_hpages = spool->min_hpages; 2371c5ecae3SMike Kravetz } 2381c5ecae3SMike Kravetz 2391c5ecae3SMike Kravetz /* 2401c5ecae3SMike Kravetz * If hugetlbfs_put_super couldn't free spool due to an outstanding 2411c5ecae3SMike Kravetz * quota reference, free it now. 2421c5ecae3SMike Kravetz */ 243db71ef79SMike Kravetz unlock_or_release_subpool(spool, flags); 2441c5ecae3SMike Kravetz 2451c5ecae3SMike Kravetz return ret; 24690481622SDavid Gibson } 24790481622SDavid Gibson 24890481622SDavid Gibson static inline struct hugepage_subpool *subpool_inode(struct inode *inode) 24990481622SDavid Gibson { 25090481622SDavid Gibson return HUGETLBFS_SB(inode->i_sb)->spool; 25190481622SDavid Gibson } 25290481622SDavid Gibson 25390481622SDavid Gibson static inline struct hugepage_subpool *subpool_vma(struct vm_area_struct *vma) 25490481622SDavid Gibson { 255496ad9aaSAl Viro return subpool_inode(file_inode(vma->vm_file)); 25690481622SDavid Gibson } 25790481622SDavid Gibson 2580db9d74eSMina Almasry /* Helper that removes a struct file_region from the resv_map cache and returns 2590db9d74eSMina Almasry * it for use. 2600db9d74eSMina Almasry */ 2610db9d74eSMina Almasry static struct file_region * 2620db9d74eSMina Almasry get_file_region_entry_from_cache(struct resv_map *resv, long from, long to) 2630db9d74eSMina Almasry { 2643259914fSXU pengfei struct file_region *nrg; 2650db9d74eSMina Almasry 2660db9d74eSMina Almasry VM_BUG_ON(resv->region_cache_count <= 0); 2670db9d74eSMina Almasry 2680db9d74eSMina Almasry resv->region_cache_count--; 2690db9d74eSMina Almasry nrg = list_first_entry(&resv->region_cache, struct file_region, link); 2700db9d74eSMina Almasry list_del(&nrg->link); 2710db9d74eSMina Almasry 2720db9d74eSMina Almasry nrg->from = from; 2730db9d74eSMina Almasry nrg->to = to; 2740db9d74eSMina Almasry 2750db9d74eSMina Almasry return nrg; 2760db9d74eSMina Almasry } 2770db9d74eSMina Almasry 278075a61d0SMina Almasry static void copy_hugetlb_cgroup_uncharge_info(struct file_region *nrg, 279075a61d0SMina Almasry struct file_region *rg) 280075a61d0SMina Almasry { 281075a61d0SMina Almasry #ifdef CONFIG_CGROUP_HUGETLB 282075a61d0SMina Almasry nrg->reservation_counter = rg->reservation_counter; 283075a61d0SMina Almasry nrg->css = rg->css; 284075a61d0SMina Almasry if (rg->css) 285075a61d0SMina Almasry css_get(rg->css); 286075a61d0SMina Almasry #endif 287075a61d0SMina Almasry } 288075a61d0SMina Almasry 289075a61d0SMina Almasry /* Helper that records hugetlb_cgroup uncharge info. */ 290075a61d0SMina Almasry static void record_hugetlb_cgroup_uncharge_info(struct hugetlb_cgroup *h_cg, 291075a61d0SMina Almasry struct hstate *h, 292075a61d0SMina Almasry struct resv_map *resv, 293075a61d0SMina Almasry struct file_region *nrg) 294075a61d0SMina Almasry { 295075a61d0SMina Almasry #ifdef CONFIG_CGROUP_HUGETLB 296075a61d0SMina Almasry if (h_cg) { 297075a61d0SMina Almasry nrg->reservation_counter = 298075a61d0SMina Almasry &h_cg->rsvd_hugepage[hstate_index(h)]; 299075a61d0SMina Almasry nrg->css = &h_cg->css; 300d85aecf2SMiaohe Lin /* 301d85aecf2SMiaohe Lin * The caller will hold exactly one h_cg->css reference for the 302d85aecf2SMiaohe Lin * whole contiguous reservation region. But this area might be 303d85aecf2SMiaohe Lin * scattered when there are already some file_regions reside in 304d85aecf2SMiaohe Lin * it. As a result, many file_regions may share only one css 305d85aecf2SMiaohe Lin * reference. In order to ensure that one file_region must hold 306d85aecf2SMiaohe Lin * exactly one h_cg->css reference, we should do css_get for 307d85aecf2SMiaohe Lin * each file_region and leave the reference held by caller 308d85aecf2SMiaohe Lin * untouched. 309d85aecf2SMiaohe Lin */ 310d85aecf2SMiaohe Lin css_get(&h_cg->css); 311075a61d0SMina Almasry if (!resv->pages_per_hpage) 312075a61d0SMina Almasry resv->pages_per_hpage = pages_per_huge_page(h); 313075a61d0SMina Almasry /* pages_per_hpage should be the same for all entries in 314075a61d0SMina Almasry * a resv_map. 315075a61d0SMina Almasry */ 316075a61d0SMina Almasry VM_BUG_ON(resv->pages_per_hpage != pages_per_huge_page(h)); 317075a61d0SMina Almasry } else { 318075a61d0SMina Almasry nrg->reservation_counter = NULL; 319075a61d0SMina Almasry nrg->css = NULL; 320075a61d0SMina Almasry } 321075a61d0SMina Almasry #endif 322075a61d0SMina Almasry } 323075a61d0SMina Almasry 324d85aecf2SMiaohe Lin static void put_uncharge_info(struct file_region *rg) 325d85aecf2SMiaohe Lin { 326d85aecf2SMiaohe Lin #ifdef CONFIG_CGROUP_HUGETLB 327d85aecf2SMiaohe Lin if (rg->css) 328d85aecf2SMiaohe Lin css_put(rg->css); 329d85aecf2SMiaohe Lin #endif 330d85aecf2SMiaohe Lin } 331d85aecf2SMiaohe Lin 332a9b3f867SMina Almasry static bool has_same_uncharge_info(struct file_region *rg, 333a9b3f867SMina Almasry struct file_region *org) 334a9b3f867SMina Almasry { 335a9b3f867SMina Almasry #ifdef CONFIG_CGROUP_HUGETLB 3360739eb43SBaolin Wang return rg->reservation_counter == org->reservation_counter && 337a9b3f867SMina Almasry rg->css == org->css; 338a9b3f867SMina Almasry 339a9b3f867SMina Almasry #else 340a9b3f867SMina Almasry return true; 341a9b3f867SMina Almasry #endif 342a9b3f867SMina Almasry } 343a9b3f867SMina Almasry 344a9b3f867SMina Almasry static void coalesce_file_region(struct resv_map *resv, struct file_region *rg) 345a9b3f867SMina Almasry { 3463259914fSXU pengfei struct file_region *nrg, *prg; 347a9b3f867SMina Almasry 348a9b3f867SMina Almasry prg = list_prev_entry(rg, link); 349a9b3f867SMina Almasry if (&prg->link != &resv->regions && prg->to == rg->from && 350a9b3f867SMina Almasry has_same_uncharge_info(prg, rg)) { 351a9b3f867SMina Almasry prg->to = rg->to; 352a9b3f867SMina Almasry 353a9b3f867SMina Almasry list_del(&rg->link); 354d85aecf2SMiaohe Lin put_uncharge_info(rg); 355a9b3f867SMina Almasry kfree(rg); 356a9b3f867SMina Almasry 3577db5e7b6SWei Yang rg = prg; 358a9b3f867SMina Almasry } 359a9b3f867SMina Almasry 360a9b3f867SMina Almasry nrg = list_next_entry(rg, link); 361a9b3f867SMina Almasry if (&nrg->link != &resv->regions && nrg->from == rg->to && 362a9b3f867SMina Almasry has_same_uncharge_info(nrg, rg)) { 363a9b3f867SMina Almasry nrg->from = rg->from; 364a9b3f867SMina Almasry 365a9b3f867SMina Almasry list_del(&rg->link); 366d85aecf2SMiaohe Lin put_uncharge_info(rg); 367a9b3f867SMina Almasry kfree(rg); 368a9b3f867SMina Almasry } 369a9b3f867SMina Almasry } 370a9b3f867SMina Almasry 3712103cf9cSPeter Xu static inline long 37284448c8eSJakob Koschel hugetlb_resv_map_add(struct resv_map *map, struct list_head *rg, long from, 3732103cf9cSPeter Xu long to, struct hstate *h, struct hugetlb_cgroup *cg, 3742103cf9cSPeter Xu long *regions_needed) 3752103cf9cSPeter Xu { 3762103cf9cSPeter Xu struct file_region *nrg; 3772103cf9cSPeter Xu 3782103cf9cSPeter Xu if (!regions_needed) { 3792103cf9cSPeter Xu nrg = get_file_region_entry_from_cache(map, from, to); 3802103cf9cSPeter Xu record_hugetlb_cgroup_uncharge_info(cg, h, map, nrg); 38184448c8eSJakob Koschel list_add(&nrg->link, rg); 3822103cf9cSPeter Xu coalesce_file_region(map, nrg); 3832103cf9cSPeter Xu } else 3842103cf9cSPeter Xu *regions_needed += 1; 3852103cf9cSPeter Xu 3862103cf9cSPeter Xu return to - from; 3872103cf9cSPeter Xu } 3882103cf9cSPeter Xu 389972a3da3SWei Yang /* 390972a3da3SWei Yang * Must be called with resv->lock held. 391972a3da3SWei Yang * 392972a3da3SWei Yang * Calling this with regions_needed != NULL will count the number of pages 393972a3da3SWei Yang * to be added but will not modify the linked list. And regions_needed will 394972a3da3SWei Yang * indicate the number of file_regions needed in the cache to carry out to add 395972a3da3SWei Yang * the regions for this range. 396d75c6af9SMina Almasry */ 397d75c6af9SMina Almasry static long add_reservation_in_range(struct resv_map *resv, long f, long t, 398075a61d0SMina Almasry struct hugetlb_cgroup *h_cg, 399972a3da3SWei Yang struct hstate *h, long *regions_needed) 400d75c6af9SMina Almasry { 4010db9d74eSMina Almasry long add = 0; 402d75c6af9SMina Almasry struct list_head *head = &resv->regions; 4030db9d74eSMina Almasry long last_accounted_offset = f; 40484448c8eSJakob Koschel struct file_region *iter, *trg = NULL; 40584448c8eSJakob Koschel struct list_head *rg = NULL; 406d75c6af9SMina Almasry 4070db9d74eSMina Almasry if (regions_needed) 4080db9d74eSMina Almasry *regions_needed = 0; 409d75c6af9SMina Almasry 4100db9d74eSMina Almasry /* In this loop, we essentially handle an entry for the range 41184448c8eSJakob Koschel * [last_accounted_offset, iter->from), at every iteration, with some 4120db9d74eSMina Almasry * bounds checking. 4130db9d74eSMina Almasry */ 41484448c8eSJakob Koschel list_for_each_entry_safe(iter, trg, head, link) { 4150db9d74eSMina Almasry /* Skip irrelevant regions that start before our range. */ 41684448c8eSJakob Koschel if (iter->from < f) { 4170db9d74eSMina Almasry /* If this region ends after the last accounted offset, 4180db9d74eSMina Almasry * then we need to update last_accounted_offset. 4190db9d74eSMina Almasry */ 42084448c8eSJakob Koschel if (iter->to > last_accounted_offset) 42184448c8eSJakob Koschel last_accounted_offset = iter->to; 4220db9d74eSMina Almasry continue; 4230db9d74eSMina Almasry } 424d75c6af9SMina Almasry 4250db9d74eSMina Almasry /* When we find a region that starts beyond our range, we've 4260db9d74eSMina Almasry * finished. 4270db9d74eSMina Almasry */ 42884448c8eSJakob Koschel if (iter->from >= t) { 42984448c8eSJakob Koschel rg = iter->link.prev; 430d75c6af9SMina Almasry break; 43184448c8eSJakob Koschel } 432d75c6af9SMina Almasry 43384448c8eSJakob Koschel /* Add an entry for last_accounted_offset -> iter->from, and 4340db9d74eSMina Almasry * update last_accounted_offset. 435d75c6af9SMina Almasry */ 43684448c8eSJakob Koschel if (iter->from > last_accounted_offset) 43784448c8eSJakob Koschel add += hugetlb_resv_map_add(resv, iter->link.prev, 4382103cf9cSPeter Xu last_accounted_offset, 43984448c8eSJakob Koschel iter->from, h, h_cg, 4402103cf9cSPeter Xu regions_needed); 441d75c6af9SMina Almasry 44284448c8eSJakob Koschel last_accounted_offset = iter->to; 4430db9d74eSMina Almasry } 4440db9d74eSMina Almasry 4450db9d74eSMina Almasry /* Handle the case where our range extends beyond 4460db9d74eSMina Almasry * last_accounted_offset. 4470db9d74eSMina Almasry */ 44884448c8eSJakob Koschel if (!rg) 44984448c8eSJakob Koschel rg = head->prev; 4502103cf9cSPeter Xu if (last_accounted_offset < t) 4512103cf9cSPeter Xu add += hugetlb_resv_map_add(resv, rg, last_accounted_offset, 4522103cf9cSPeter Xu t, h, h_cg, regions_needed); 4530db9d74eSMina Almasry 4540db9d74eSMina Almasry return add; 4550db9d74eSMina Almasry } 4560db9d74eSMina Almasry 4570db9d74eSMina Almasry /* Must be called with resv->lock acquired. Will drop lock to allocate entries. 4580db9d74eSMina Almasry */ 4590db9d74eSMina Almasry static int allocate_file_region_entries(struct resv_map *resv, 4600db9d74eSMina Almasry int regions_needed) 4610db9d74eSMina Almasry __must_hold(&resv->lock) 4620db9d74eSMina Almasry { 46334665341SMiaohe Lin LIST_HEAD(allocated_regions); 4640db9d74eSMina Almasry int to_allocate = 0, i = 0; 4650db9d74eSMina Almasry struct file_region *trg = NULL, *rg = NULL; 4660db9d74eSMina Almasry 4670db9d74eSMina Almasry VM_BUG_ON(regions_needed < 0); 4680db9d74eSMina Almasry 4690db9d74eSMina Almasry /* 4700db9d74eSMina Almasry * Check for sufficient descriptors in the cache to accommodate 4710db9d74eSMina Almasry * the number of in progress add operations plus regions_needed. 4720db9d74eSMina Almasry * 4730db9d74eSMina Almasry * This is a while loop because when we drop the lock, some other call 4740db9d74eSMina Almasry * to region_add or region_del may have consumed some region_entries, 4750db9d74eSMina Almasry * so we keep looping here until we finally have enough entries for 4760db9d74eSMina Almasry * (adds_in_progress + regions_needed). 4770db9d74eSMina Almasry */ 4780db9d74eSMina Almasry while (resv->region_cache_count < 4790db9d74eSMina Almasry (resv->adds_in_progress + regions_needed)) { 4800db9d74eSMina Almasry to_allocate = resv->adds_in_progress + regions_needed - 4810db9d74eSMina Almasry resv->region_cache_count; 4820db9d74eSMina Almasry 4830db9d74eSMina Almasry /* At this point, we should have enough entries in the cache 484f0953a1bSIngo Molnar * for all the existing adds_in_progress. We should only be 4850db9d74eSMina Almasry * needing to allocate for regions_needed. 4860db9d74eSMina Almasry */ 4870db9d74eSMina Almasry VM_BUG_ON(resv->region_cache_count < resv->adds_in_progress); 4880db9d74eSMina Almasry 4890db9d74eSMina Almasry spin_unlock(&resv->lock); 4900db9d74eSMina Almasry for (i = 0; i < to_allocate; i++) { 4910db9d74eSMina Almasry trg = kmalloc(sizeof(*trg), GFP_KERNEL); 4920db9d74eSMina Almasry if (!trg) 4930db9d74eSMina Almasry goto out_of_memory; 4940db9d74eSMina Almasry list_add(&trg->link, &allocated_regions); 4950db9d74eSMina Almasry } 4960db9d74eSMina Almasry 4970db9d74eSMina Almasry spin_lock(&resv->lock); 4980db9d74eSMina Almasry 499d3ec7b6eSWei Yang list_splice(&allocated_regions, &resv->region_cache); 500d3ec7b6eSWei Yang resv->region_cache_count += to_allocate; 5010db9d74eSMina Almasry } 5020db9d74eSMina Almasry 5030db9d74eSMina Almasry return 0; 5040db9d74eSMina Almasry 5050db9d74eSMina Almasry out_of_memory: 5060db9d74eSMina Almasry list_for_each_entry_safe(rg, trg, &allocated_regions, link) { 507d75c6af9SMina Almasry list_del(&rg->link); 508d75c6af9SMina Almasry kfree(rg); 509d75c6af9SMina Almasry } 5100db9d74eSMina Almasry return -ENOMEM; 511d75c6af9SMina Almasry } 512d75c6af9SMina Almasry 5131dd308a7SMike Kravetz /* 5141dd308a7SMike Kravetz * Add the huge page range represented by [f, t) to the reserve 5150db9d74eSMina Almasry * map. Regions will be taken from the cache to fill in this range. 5160db9d74eSMina Almasry * Sufficient regions should exist in the cache due to the previous 5170db9d74eSMina Almasry * call to region_chg with the same range, but in some cases the cache will not 5180db9d74eSMina Almasry * have sufficient entries due to races with other code doing region_add or 5190db9d74eSMina Almasry * region_del. The extra needed entries will be allocated. 520cf3ad20bSMike Kravetz * 5210db9d74eSMina Almasry * regions_needed is the out value provided by a previous call to region_chg. 5220db9d74eSMina Almasry * 5230db9d74eSMina Almasry * Return the number of new huge pages added to the map. This number is greater 5240db9d74eSMina Almasry * than or equal to zero. If file_region entries needed to be allocated for 5257c8de358SEthon Paul * this operation and we were not able to allocate, it returns -ENOMEM. 5260db9d74eSMina Almasry * region_add of regions of length 1 never allocate file_regions and cannot 5270db9d74eSMina Almasry * fail; region_chg will always allocate at least 1 entry and a region_add for 5280db9d74eSMina Almasry * 1 page will only require at most 1 entry. 5291dd308a7SMike Kravetz */ 5300db9d74eSMina Almasry static long region_add(struct resv_map *resv, long f, long t, 531075a61d0SMina Almasry long in_regions_needed, struct hstate *h, 532075a61d0SMina Almasry struct hugetlb_cgroup *h_cg) 53396822904SAndy Whitcroft { 5340db9d74eSMina Almasry long add = 0, actual_regions_needed = 0; 53596822904SAndy Whitcroft 5367b24d861SDavidlohr Bueso spin_lock(&resv->lock); 5370db9d74eSMina Almasry retry: 5380db9d74eSMina Almasry 5390db9d74eSMina Almasry /* Count how many regions are actually needed to execute this add. */ 540972a3da3SWei Yang add_reservation_in_range(resv, f, t, NULL, NULL, 541972a3da3SWei Yang &actual_regions_needed); 54296822904SAndy Whitcroft 5435e911373SMike Kravetz /* 5440db9d74eSMina Almasry * Check for sufficient descriptors in the cache to accommodate 5450db9d74eSMina Almasry * this add operation. Note that actual_regions_needed may be greater 5460db9d74eSMina Almasry * than in_regions_needed, as the resv_map may have been modified since 5470db9d74eSMina Almasry * the region_chg call. In this case, we need to make sure that we 5480db9d74eSMina Almasry * allocate extra entries, such that we have enough for all the 5490db9d74eSMina Almasry * existing adds_in_progress, plus the excess needed for this 5500db9d74eSMina Almasry * operation. 5515e911373SMike Kravetz */ 5520db9d74eSMina Almasry if (actual_regions_needed > in_regions_needed && 5530db9d74eSMina Almasry resv->region_cache_count < 5540db9d74eSMina Almasry resv->adds_in_progress + 5550db9d74eSMina Almasry (actual_regions_needed - in_regions_needed)) { 5560db9d74eSMina Almasry /* region_add operation of range 1 should never need to 5570db9d74eSMina Almasry * allocate file_region entries. 5580db9d74eSMina Almasry */ 5590db9d74eSMina Almasry VM_BUG_ON(t - f <= 1); 5605e911373SMike Kravetz 5610db9d74eSMina Almasry if (allocate_file_region_entries( 5620db9d74eSMina Almasry resv, actual_regions_needed - in_regions_needed)) { 5630db9d74eSMina Almasry return -ENOMEM; 5645e911373SMike Kravetz } 5655e911373SMike Kravetz 5660db9d74eSMina Almasry goto retry; 5670db9d74eSMina Almasry } 568cf3ad20bSMike Kravetz 569972a3da3SWei Yang add = add_reservation_in_range(resv, f, t, h_cg, h, NULL); 5700db9d74eSMina Almasry 5710db9d74eSMina Almasry resv->adds_in_progress -= in_regions_needed; 5720db9d74eSMina Almasry 5737b24d861SDavidlohr Bueso spin_unlock(&resv->lock); 574cf3ad20bSMike Kravetz return add; 57596822904SAndy Whitcroft } 57696822904SAndy Whitcroft 5771dd308a7SMike Kravetz /* 5781dd308a7SMike Kravetz * Examine the existing reserve map and determine how many 5791dd308a7SMike Kravetz * huge pages in the specified range [f, t) are NOT currently 5801dd308a7SMike Kravetz * represented. This routine is called before a subsequent 5811dd308a7SMike Kravetz * call to region_add that will actually modify the reserve 5821dd308a7SMike Kravetz * map to add the specified range [f, t). region_chg does 5831dd308a7SMike Kravetz * not change the number of huge pages represented by the 5840db9d74eSMina Almasry * map. A number of new file_region structures is added to the cache as a 5850db9d74eSMina Almasry * placeholder, for the subsequent region_add call to use. At least 1 5860db9d74eSMina Almasry * file_region structure is added. 5870db9d74eSMina Almasry * 5880db9d74eSMina Almasry * out_regions_needed is the number of regions added to the 5890db9d74eSMina Almasry * resv->adds_in_progress. This value needs to be provided to a follow up call 5900db9d74eSMina Almasry * to region_add or region_abort for proper accounting. 5915e911373SMike Kravetz * 5925e911373SMike Kravetz * Returns the number of huge pages that need to be added to the existing 5935e911373SMike Kravetz * reservation map for the range [f, t). This number is greater or equal to 5945e911373SMike Kravetz * zero. -ENOMEM is returned if a new file_region structure or cache entry 5955e911373SMike Kravetz * is needed and can not be allocated. 5961dd308a7SMike Kravetz */ 5970db9d74eSMina Almasry static long region_chg(struct resv_map *resv, long f, long t, 5980db9d74eSMina Almasry long *out_regions_needed) 59996822904SAndy Whitcroft { 60096822904SAndy Whitcroft long chg = 0; 60196822904SAndy Whitcroft 6027b24d861SDavidlohr Bueso spin_lock(&resv->lock); 6035e911373SMike Kravetz 604972a3da3SWei Yang /* Count how many hugepages in this range are NOT represented. */ 605075a61d0SMina Almasry chg = add_reservation_in_range(resv, f, t, NULL, NULL, 606972a3da3SWei Yang out_regions_needed); 6075e911373SMike Kravetz 6080db9d74eSMina Almasry if (*out_regions_needed == 0) 6090db9d74eSMina Almasry *out_regions_needed = 1; 6105e911373SMike Kravetz 6110db9d74eSMina Almasry if (allocate_file_region_entries(resv, *out_regions_needed)) 6125e911373SMike Kravetz return -ENOMEM; 6135e911373SMike Kravetz 6140db9d74eSMina Almasry resv->adds_in_progress += *out_regions_needed; 61596822904SAndy Whitcroft 6167b24d861SDavidlohr Bueso spin_unlock(&resv->lock); 61796822904SAndy Whitcroft return chg; 61896822904SAndy Whitcroft } 61996822904SAndy Whitcroft 6201dd308a7SMike Kravetz /* 6215e911373SMike Kravetz * Abort the in progress add operation. The adds_in_progress field 6225e911373SMike Kravetz * of the resv_map keeps track of the operations in progress between 6235e911373SMike Kravetz * calls to region_chg and region_add. Operations are sometimes 6245e911373SMike Kravetz * aborted after the call to region_chg. In such cases, region_abort 6250db9d74eSMina Almasry * is called to decrement the adds_in_progress counter. regions_needed 6260db9d74eSMina Almasry * is the value returned by the region_chg call, it is used to decrement 6270db9d74eSMina Almasry * the adds_in_progress counter. 6285e911373SMike Kravetz * 6295e911373SMike Kravetz * NOTE: The range arguments [f, t) are not needed or used in this 6305e911373SMike Kravetz * routine. They are kept to make reading the calling code easier as 6315e911373SMike Kravetz * arguments will match the associated region_chg call. 6325e911373SMike Kravetz */ 6330db9d74eSMina Almasry static void region_abort(struct resv_map *resv, long f, long t, 6340db9d74eSMina Almasry long regions_needed) 6355e911373SMike Kravetz { 6365e911373SMike Kravetz spin_lock(&resv->lock); 6375e911373SMike Kravetz VM_BUG_ON(!resv->region_cache_count); 6380db9d74eSMina Almasry resv->adds_in_progress -= regions_needed; 6395e911373SMike Kravetz spin_unlock(&resv->lock); 6405e911373SMike Kravetz } 6415e911373SMike Kravetz 6425e911373SMike Kravetz /* 643feba16e2SMike Kravetz * Delete the specified range [f, t) from the reserve map. If the 644feba16e2SMike Kravetz * t parameter is LONG_MAX, this indicates that ALL regions after f 645feba16e2SMike Kravetz * should be deleted. Locate the regions which intersect [f, t) 646feba16e2SMike Kravetz * and either trim, delete or split the existing regions. 647feba16e2SMike Kravetz * 648feba16e2SMike Kravetz * Returns the number of huge pages deleted from the reserve map. 649feba16e2SMike Kravetz * In the normal case, the return value is zero or more. In the 650feba16e2SMike Kravetz * case where a region must be split, a new region descriptor must 651feba16e2SMike Kravetz * be allocated. If the allocation fails, -ENOMEM will be returned. 652feba16e2SMike Kravetz * NOTE: If the parameter t == LONG_MAX, then we will never split 653feba16e2SMike Kravetz * a region and possibly return -ENOMEM. Callers specifying 654feba16e2SMike Kravetz * t == LONG_MAX do not need to check for -ENOMEM error. 6551dd308a7SMike Kravetz */ 656feba16e2SMike Kravetz static long region_del(struct resv_map *resv, long f, long t) 65796822904SAndy Whitcroft { 6581406ec9bSJoonsoo Kim struct list_head *head = &resv->regions; 65996822904SAndy Whitcroft struct file_region *rg, *trg; 660feba16e2SMike Kravetz struct file_region *nrg = NULL; 661feba16e2SMike Kravetz long del = 0; 66296822904SAndy Whitcroft 663feba16e2SMike Kravetz retry: 6647b24d861SDavidlohr Bueso spin_lock(&resv->lock); 665feba16e2SMike Kravetz list_for_each_entry_safe(rg, trg, head, link) { 666dbe409e4SMike Kravetz /* 667dbe409e4SMike Kravetz * Skip regions before the range to be deleted. file_region 668dbe409e4SMike Kravetz * ranges are normally of the form [from, to). However, there 669dbe409e4SMike Kravetz * may be a "placeholder" entry in the map which is of the form 670dbe409e4SMike Kravetz * (from, to) with from == to. Check for placeholder entries 671dbe409e4SMike Kravetz * at the beginning of the range to be deleted. 672dbe409e4SMike Kravetz */ 673dbe409e4SMike Kravetz if (rg->to <= f && (rg->to != rg->from || rg->to != f)) 674feba16e2SMike Kravetz continue; 675dbe409e4SMike Kravetz 676feba16e2SMike Kravetz if (rg->from >= t) 67796822904SAndy Whitcroft break; 67896822904SAndy Whitcroft 679feba16e2SMike Kravetz if (f > rg->from && t < rg->to) { /* Must split region */ 680feba16e2SMike Kravetz /* 681feba16e2SMike Kravetz * Check for an entry in the cache before dropping 682feba16e2SMike Kravetz * lock and attempting allocation. 683feba16e2SMike Kravetz */ 684feba16e2SMike Kravetz if (!nrg && 685feba16e2SMike Kravetz resv->region_cache_count > resv->adds_in_progress) { 686feba16e2SMike Kravetz nrg = list_first_entry(&resv->region_cache, 687feba16e2SMike Kravetz struct file_region, 688feba16e2SMike Kravetz link); 689feba16e2SMike Kravetz list_del(&nrg->link); 690feba16e2SMike Kravetz resv->region_cache_count--; 69196822904SAndy Whitcroft } 69296822904SAndy Whitcroft 693feba16e2SMike Kravetz if (!nrg) { 694feba16e2SMike Kravetz spin_unlock(&resv->lock); 695feba16e2SMike Kravetz nrg = kmalloc(sizeof(*nrg), GFP_KERNEL); 696feba16e2SMike Kravetz if (!nrg) 697feba16e2SMike Kravetz return -ENOMEM; 698feba16e2SMike Kravetz goto retry; 699feba16e2SMike Kravetz } 700feba16e2SMike Kravetz 701feba16e2SMike Kravetz del += t - f; 70279aa925bSMike Kravetz hugetlb_cgroup_uncharge_file_region( 703d85aecf2SMiaohe Lin resv, rg, t - f, false); 704feba16e2SMike Kravetz 705feba16e2SMike Kravetz /* New entry for end of split region */ 706feba16e2SMike Kravetz nrg->from = t; 707feba16e2SMike Kravetz nrg->to = rg->to; 708075a61d0SMina Almasry 709075a61d0SMina Almasry copy_hugetlb_cgroup_uncharge_info(nrg, rg); 710075a61d0SMina Almasry 711feba16e2SMike Kravetz INIT_LIST_HEAD(&nrg->link); 712feba16e2SMike Kravetz 713feba16e2SMike Kravetz /* Original entry is trimmed */ 714feba16e2SMike Kravetz rg->to = f; 715feba16e2SMike Kravetz 716feba16e2SMike Kravetz list_add(&nrg->link, &rg->link); 717feba16e2SMike Kravetz nrg = NULL; 71896822904SAndy Whitcroft break; 719feba16e2SMike Kravetz } 720feba16e2SMike Kravetz 721feba16e2SMike Kravetz if (f <= rg->from && t >= rg->to) { /* Remove entire region */ 722feba16e2SMike Kravetz del += rg->to - rg->from; 723075a61d0SMina Almasry hugetlb_cgroup_uncharge_file_region(resv, rg, 724d85aecf2SMiaohe Lin rg->to - rg->from, true); 72596822904SAndy Whitcroft list_del(&rg->link); 72696822904SAndy Whitcroft kfree(rg); 727feba16e2SMike Kravetz continue; 72896822904SAndy Whitcroft } 7297b24d861SDavidlohr Bueso 730feba16e2SMike Kravetz if (f <= rg->from) { /* Trim beginning of region */ 731075a61d0SMina Almasry hugetlb_cgroup_uncharge_file_region(resv, rg, 732d85aecf2SMiaohe Lin t - rg->from, false); 733075a61d0SMina Almasry 73479aa925bSMike Kravetz del += t - rg->from; 73579aa925bSMike Kravetz rg->from = t; 73679aa925bSMike Kravetz } else { /* Trim end of region */ 737075a61d0SMina Almasry hugetlb_cgroup_uncharge_file_region(resv, rg, 738d85aecf2SMiaohe Lin rg->to - f, false); 73979aa925bSMike Kravetz 74079aa925bSMike Kravetz del += rg->to - f; 74179aa925bSMike Kravetz rg->to = f; 742feba16e2SMike Kravetz } 743feba16e2SMike Kravetz } 744feba16e2SMike Kravetz 7457b24d861SDavidlohr Bueso spin_unlock(&resv->lock); 746feba16e2SMike Kravetz kfree(nrg); 747feba16e2SMike Kravetz return del; 74896822904SAndy Whitcroft } 74996822904SAndy Whitcroft 7501dd308a7SMike Kravetz /* 751b5cec28dSMike Kravetz * A rare out of memory error was encountered which prevented removal of 752b5cec28dSMike Kravetz * the reserve map region for a page. The huge page itself was free'ed 753b5cec28dSMike Kravetz * and removed from the page cache. This routine will adjust the subpool 754b5cec28dSMike Kravetz * usage count, and the global reserve count if needed. By incrementing 755b5cec28dSMike Kravetz * these counts, the reserve map entry which could not be deleted will 756b5cec28dSMike Kravetz * appear as a "reserved" entry instead of simply dangling with incorrect 757b5cec28dSMike Kravetz * counts. 758b5cec28dSMike Kravetz */ 75972e2936cSzhong jiang void hugetlb_fix_reserve_counts(struct inode *inode) 760b5cec28dSMike Kravetz { 761b5cec28dSMike Kravetz struct hugepage_subpool *spool = subpool_inode(inode); 762b5cec28dSMike Kravetz long rsv_adjust; 763da56388cSMiaohe Lin bool reserved = false; 764b5cec28dSMike Kravetz 765b5cec28dSMike Kravetz rsv_adjust = hugepage_subpool_get_pages(spool, 1); 766da56388cSMiaohe Lin if (rsv_adjust > 0) { 767b5cec28dSMike Kravetz struct hstate *h = hstate_inode(inode); 768b5cec28dSMike Kravetz 769da56388cSMiaohe Lin if (!hugetlb_acct_memory(h, 1)) 770da56388cSMiaohe Lin reserved = true; 771da56388cSMiaohe Lin } else if (!rsv_adjust) { 772da56388cSMiaohe Lin reserved = true; 773b5cec28dSMike Kravetz } 774da56388cSMiaohe Lin 775da56388cSMiaohe Lin if (!reserved) 776da56388cSMiaohe Lin pr_warn("hugetlb: Huge Page Reserved count may go negative.\n"); 777b5cec28dSMike Kravetz } 778b5cec28dSMike Kravetz 779b5cec28dSMike Kravetz /* 7801dd308a7SMike Kravetz * Count and return the number of huge pages in the reserve map 7811dd308a7SMike Kravetz * that intersect with the range [f, t). 7821dd308a7SMike Kravetz */ 7831406ec9bSJoonsoo Kim static long region_count(struct resv_map *resv, long f, long t) 78484afd99bSAndy Whitcroft { 7851406ec9bSJoonsoo Kim struct list_head *head = &resv->regions; 78684afd99bSAndy Whitcroft struct file_region *rg; 78784afd99bSAndy Whitcroft long chg = 0; 78884afd99bSAndy Whitcroft 7897b24d861SDavidlohr Bueso spin_lock(&resv->lock); 79084afd99bSAndy Whitcroft /* Locate each segment we overlap with, and count that overlap. */ 79184afd99bSAndy Whitcroft list_for_each_entry(rg, head, link) { 792f2135a4aSWang Sheng-Hui long seg_from; 793f2135a4aSWang Sheng-Hui long seg_to; 79484afd99bSAndy Whitcroft 79584afd99bSAndy Whitcroft if (rg->to <= f) 79684afd99bSAndy Whitcroft continue; 79784afd99bSAndy Whitcroft if (rg->from >= t) 79884afd99bSAndy Whitcroft break; 79984afd99bSAndy Whitcroft 80084afd99bSAndy Whitcroft seg_from = max(rg->from, f); 80184afd99bSAndy Whitcroft seg_to = min(rg->to, t); 80284afd99bSAndy Whitcroft 80384afd99bSAndy Whitcroft chg += seg_to - seg_from; 80484afd99bSAndy Whitcroft } 8057b24d861SDavidlohr Bueso spin_unlock(&resv->lock); 80684afd99bSAndy Whitcroft 80784afd99bSAndy Whitcroft return chg; 80884afd99bSAndy Whitcroft } 80984afd99bSAndy Whitcroft 81096822904SAndy Whitcroft /* 811e7c4b0bfSAndy Whitcroft * Convert the address within this vma to the page offset within 812e7c4b0bfSAndy Whitcroft * the mapping, in pagecache page units; huge pages here. 813e7c4b0bfSAndy Whitcroft */ 814a5516438SAndi Kleen static pgoff_t vma_hugecache_offset(struct hstate *h, 815a5516438SAndi Kleen struct vm_area_struct *vma, unsigned long address) 816e7c4b0bfSAndy Whitcroft { 817a5516438SAndi Kleen return ((address - vma->vm_start) >> huge_page_shift(h)) + 818a5516438SAndi Kleen (vma->vm_pgoff >> huge_page_order(h)); 819e7c4b0bfSAndy Whitcroft } 820e7c4b0bfSAndy Whitcroft 8210fe6e20bSNaoya Horiguchi pgoff_t linear_hugepage_index(struct vm_area_struct *vma, 8220fe6e20bSNaoya Horiguchi unsigned long address) 8230fe6e20bSNaoya Horiguchi { 8240fe6e20bSNaoya Horiguchi return vma_hugecache_offset(hstate_vma(vma), vma, address); 8250fe6e20bSNaoya Horiguchi } 826dee41079SDan Williams EXPORT_SYMBOL_GPL(linear_hugepage_index); 8270fe6e20bSNaoya Horiguchi 82884afd99bSAndy Whitcroft /* 82908fba699SMel Gorman * Return the size of the pages allocated when backing a VMA. In the majority 83008fba699SMel Gorman * cases this will be same size as used by the page table entries. 83108fba699SMel Gorman */ 83208fba699SMel Gorman unsigned long vma_kernel_pagesize(struct vm_area_struct *vma) 83308fba699SMel Gorman { 83405ea8860SDan Williams if (vma->vm_ops && vma->vm_ops->pagesize) 83505ea8860SDan Williams return vma->vm_ops->pagesize(vma); 83608fba699SMel Gorman return PAGE_SIZE; 83708fba699SMel Gorman } 838f340ca0fSJoerg Roedel EXPORT_SYMBOL_GPL(vma_kernel_pagesize); 83908fba699SMel Gorman 84008fba699SMel Gorman /* 8413340289dSMel Gorman * Return the page size being used by the MMU to back a VMA. In the majority 8423340289dSMel Gorman * of cases, the page size used by the kernel matches the MMU size. On 84309135cc5SDan Williams * architectures where it differs, an architecture-specific 'strong' 84409135cc5SDan Williams * version of this symbol is required. 8453340289dSMel Gorman */ 84609135cc5SDan Williams __weak unsigned long vma_mmu_pagesize(struct vm_area_struct *vma) 8473340289dSMel Gorman { 8483340289dSMel Gorman return vma_kernel_pagesize(vma); 8493340289dSMel Gorman } 8503340289dSMel Gorman 8513340289dSMel Gorman /* 85284afd99bSAndy Whitcroft * Flags for MAP_PRIVATE reservations. These are stored in the bottom 85384afd99bSAndy Whitcroft * bits of the reservation map pointer, which are always clear due to 85484afd99bSAndy Whitcroft * alignment. 85584afd99bSAndy Whitcroft */ 85684afd99bSAndy Whitcroft #define HPAGE_RESV_OWNER (1UL << 0) 85784afd99bSAndy Whitcroft #define HPAGE_RESV_UNMAPPED (1UL << 1) 85804f2cbe3SMel Gorman #define HPAGE_RESV_MASK (HPAGE_RESV_OWNER | HPAGE_RESV_UNMAPPED) 85984afd99bSAndy Whitcroft 860a1e78772SMel Gorman /* 861a1e78772SMel Gorman * These helpers are used to track how many pages are reserved for 862a1e78772SMel Gorman * faults in a MAP_PRIVATE mapping. Only the process that called mmap() 863a1e78772SMel Gorman * is guaranteed to have their future faults succeed. 864a1e78772SMel Gorman * 8658d9bfb26SMike Kravetz * With the exception of hugetlb_dup_vma_private() which is called at fork(), 866a1e78772SMel Gorman * the reserve counters are updated with the hugetlb_lock held. It is safe 867a1e78772SMel Gorman * to reset the VMA at fork() time as it is not in use yet and there is no 868a1e78772SMel Gorman * chance of the global counters getting corrupted as a result of the values. 86984afd99bSAndy Whitcroft * 87084afd99bSAndy Whitcroft * The private mapping reservation is represented in a subtly different 87184afd99bSAndy Whitcroft * manner to a shared mapping. A shared mapping has a region map associated 87284afd99bSAndy Whitcroft * with the underlying file, this region map represents the backing file 87384afd99bSAndy Whitcroft * pages which have ever had a reservation assigned which this persists even 87484afd99bSAndy Whitcroft * after the page is instantiated. A private mapping has a region map 87584afd99bSAndy Whitcroft * associated with the original mmap which is attached to all VMAs which 87684afd99bSAndy Whitcroft * reference it, this region map represents those offsets which have consumed 87784afd99bSAndy Whitcroft * reservation ie. where pages have been instantiated. 878a1e78772SMel Gorman */ 879e7c4b0bfSAndy Whitcroft static unsigned long get_vma_private_data(struct vm_area_struct *vma) 880e7c4b0bfSAndy Whitcroft { 881e7c4b0bfSAndy Whitcroft return (unsigned long)vma->vm_private_data; 882e7c4b0bfSAndy Whitcroft } 883e7c4b0bfSAndy Whitcroft 884e7c4b0bfSAndy Whitcroft static void set_vma_private_data(struct vm_area_struct *vma, 885e7c4b0bfSAndy Whitcroft unsigned long value) 886e7c4b0bfSAndy Whitcroft { 887e7c4b0bfSAndy Whitcroft vma->vm_private_data = (void *)value; 888e7c4b0bfSAndy Whitcroft } 889e7c4b0bfSAndy Whitcroft 890e9fe92aeSMina Almasry static void 891e9fe92aeSMina Almasry resv_map_set_hugetlb_cgroup_uncharge_info(struct resv_map *resv_map, 892e9fe92aeSMina Almasry struct hugetlb_cgroup *h_cg, 893e9fe92aeSMina Almasry struct hstate *h) 894e9fe92aeSMina Almasry { 895e9fe92aeSMina Almasry #ifdef CONFIG_CGROUP_HUGETLB 896e9fe92aeSMina Almasry if (!h_cg || !h) { 897e9fe92aeSMina Almasry resv_map->reservation_counter = NULL; 898e9fe92aeSMina Almasry resv_map->pages_per_hpage = 0; 899e9fe92aeSMina Almasry resv_map->css = NULL; 900e9fe92aeSMina Almasry } else { 901e9fe92aeSMina Almasry resv_map->reservation_counter = 902e9fe92aeSMina Almasry &h_cg->rsvd_hugepage[hstate_index(h)]; 903e9fe92aeSMina Almasry resv_map->pages_per_hpage = pages_per_huge_page(h); 904e9fe92aeSMina Almasry resv_map->css = &h_cg->css; 905e9fe92aeSMina Almasry } 906e9fe92aeSMina Almasry #endif 907e9fe92aeSMina Almasry } 908e9fe92aeSMina Almasry 9099119a41eSJoonsoo Kim struct resv_map *resv_map_alloc(void) 91084afd99bSAndy Whitcroft { 91184afd99bSAndy Whitcroft struct resv_map *resv_map = kmalloc(sizeof(*resv_map), GFP_KERNEL); 9125e911373SMike Kravetz struct file_region *rg = kmalloc(sizeof(*rg), GFP_KERNEL); 9135e911373SMike Kravetz 9145e911373SMike Kravetz if (!resv_map || !rg) { 9155e911373SMike Kravetz kfree(resv_map); 9165e911373SMike Kravetz kfree(rg); 91784afd99bSAndy Whitcroft return NULL; 9185e911373SMike Kravetz } 91984afd99bSAndy Whitcroft 92084afd99bSAndy Whitcroft kref_init(&resv_map->refs); 9217b24d861SDavidlohr Bueso spin_lock_init(&resv_map->lock); 92284afd99bSAndy Whitcroft INIT_LIST_HEAD(&resv_map->regions); 92384afd99bSAndy Whitcroft 9245e911373SMike Kravetz resv_map->adds_in_progress = 0; 925e9fe92aeSMina Almasry /* 926e9fe92aeSMina Almasry * Initialize these to 0. On shared mappings, 0's here indicate these 927e9fe92aeSMina Almasry * fields don't do cgroup accounting. On private mappings, these will be 928e9fe92aeSMina Almasry * re-initialized to the proper values, to indicate that hugetlb cgroup 929e9fe92aeSMina Almasry * reservations are to be un-charged from here. 930e9fe92aeSMina Almasry */ 931e9fe92aeSMina Almasry resv_map_set_hugetlb_cgroup_uncharge_info(resv_map, NULL, NULL); 9325e911373SMike Kravetz 9335e911373SMike Kravetz INIT_LIST_HEAD(&resv_map->region_cache); 9345e911373SMike Kravetz list_add(&rg->link, &resv_map->region_cache); 9355e911373SMike Kravetz resv_map->region_cache_count = 1; 9365e911373SMike Kravetz 93784afd99bSAndy Whitcroft return resv_map; 93884afd99bSAndy Whitcroft } 93984afd99bSAndy Whitcroft 9409119a41eSJoonsoo Kim void resv_map_release(struct kref *ref) 94184afd99bSAndy Whitcroft { 94284afd99bSAndy Whitcroft struct resv_map *resv_map = container_of(ref, struct resv_map, refs); 9435e911373SMike Kravetz struct list_head *head = &resv_map->region_cache; 9445e911373SMike Kravetz struct file_region *rg, *trg; 94584afd99bSAndy Whitcroft 94684afd99bSAndy Whitcroft /* Clear out any active regions before we release the map. */ 947feba16e2SMike Kravetz region_del(resv_map, 0, LONG_MAX); 9485e911373SMike Kravetz 9495e911373SMike Kravetz /* ... and any entries left in the cache */ 9505e911373SMike Kravetz list_for_each_entry_safe(rg, trg, head, link) { 9515e911373SMike Kravetz list_del(&rg->link); 9525e911373SMike Kravetz kfree(rg); 9535e911373SMike Kravetz } 9545e911373SMike Kravetz 9555e911373SMike Kravetz VM_BUG_ON(resv_map->adds_in_progress); 9565e911373SMike Kravetz 95784afd99bSAndy Whitcroft kfree(resv_map); 95884afd99bSAndy Whitcroft } 95984afd99bSAndy Whitcroft 9604e35f483SJoonsoo Kim static inline struct resv_map *inode_resv_map(struct inode *inode) 9614e35f483SJoonsoo Kim { 962f27a5136SMike Kravetz /* 963f27a5136SMike Kravetz * At inode evict time, i_mapping may not point to the original 964f27a5136SMike Kravetz * address space within the inode. This original address space 965f27a5136SMike Kravetz * contains the pointer to the resv_map. So, always use the 966f27a5136SMike Kravetz * address space embedded within the inode. 967f27a5136SMike Kravetz * The VERY common case is inode->mapping == &inode->i_data but, 968f27a5136SMike Kravetz * this may not be true for device special inodes. 969f27a5136SMike Kravetz */ 970f27a5136SMike Kravetz return (struct resv_map *)(&inode->i_data)->private_data; 9714e35f483SJoonsoo Kim } 9724e35f483SJoonsoo Kim 97384afd99bSAndy Whitcroft static struct resv_map *vma_resv_map(struct vm_area_struct *vma) 974a1e78772SMel Gorman { 97581d1b09cSSasha Levin VM_BUG_ON_VMA(!is_vm_hugetlb_page(vma), vma); 9764e35f483SJoonsoo Kim if (vma->vm_flags & VM_MAYSHARE) { 9774e35f483SJoonsoo Kim struct address_space *mapping = vma->vm_file->f_mapping; 9784e35f483SJoonsoo Kim struct inode *inode = mapping->host; 9794e35f483SJoonsoo Kim 9804e35f483SJoonsoo Kim return inode_resv_map(inode); 9814e35f483SJoonsoo Kim 9824e35f483SJoonsoo Kim } else { 98384afd99bSAndy Whitcroft return (struct resv_map *)(get_vma_private_data(vma) & 98484afd99bSAndy Whitcroft ~HPAGE_RESV_MASK); 9854e35f483SJoonsoo Kim } 986a1e78772SMel Gorman } 987a1e78772SMel Gorman 98884afd99bSAndy Whitcroft static void set_vma_resv_map(struct vm_area_struct *vma, struct resv_map *map) 989a1e78772SMel Gorman { 99081d1b09cSSasha Levin VM_BUG_ON_VMA(!is_vm_hugetlb_page(vma), vma); 99181d1b09cSSasha Levin VM_BUG_ON_VMA(vma->vm_flags & VM_MAYSHARE, vma); 992a1e78772SMel Gorman 99384afd99bSAndy Whitcroft set_vma_private_data(vma, (get_vma_private_data(vma) & 99484afd99bSAndy Whitcroft HPAGE_RESV_MASK) | (unsigned long)map); 99504f2cbe3SMel Gorman } 99604f2cbe3SMel Gorman 99704f2cbe3SMel Gorman static void set_vma_resv_flags(struct vm_area_struct *vma, unsigned long flags) 99804f2cbe3SMel Gorman { 99981d1b09cSSasha Levin VM_BUG_ON_VMA(!is_vm_hugetlb_page(vma), vma); 100081d1b09cSSasha Levin VM_BUG_ON_VMA(vma->vm_flags & VM_MAYSHARE, vma); 1001e7c4b0bfSAndy Whitcroft 1002e7c4b0bfSAndy Whitcroft set_vma_private_data(vma, get_vma_private_data(vma) | flags); 100304f2cbe3SMel Gorman } 100404f2cbe3SMel Gorman 100504f2cbe3SMel Gorman static int is_vma_resv_set(struct vm_area_struct *vma, unsigned long flag) 100604f2cbe3SMel Gorman { 100781d1b09cSSasha Levin VM_BUG_ON_VMA(!is_vm_hugetlb_page(vma), vma); 1008e7c4b0bfSAndy Whitcroft 1009e7c4b0bfSAndy Whitcroft return (get_vma_private_data(vma) & flag) != 0; 1010a1e78772SMel Gorman } 1011a1e78772SMel Gorman 10128d9bfb26SMike Kravetz void hugetlb_dup_vma_private(struct vm_area_struct *vma) 1013a1e78772SMel Gorman { 101481d1b09cSSasha Levin VM_BUG_ON_VMA(!is_vm_hugetlb_page(vma), vma); 10158d9bfb26SMike Kravetz /* 10168d9bfb26SMike Kravetz * Clear vm_private_data 1017612b8a31SMike Kravetz * - For shared mappings this is a per-vma semaphore that may be 1018612b8a31SMike Kravetz * allocated in a subsequent call to hugetlb_vm_op_open. 1019612b8a31SMike Kravetz * Before clearing, make sure pointer is not associated with vma 1020612b8a31SMike Kravetz * as this will leak the structure. This is the case when called 1021612b8a31SMike Kravetz * via clear_vma_resv_huge_pages() and hugetlb_vm_op_open has already 1022612b8a31SMike Kravetz * been called to allocate a new structure. 10238d9bfb26SMike Kravetz * - For MAP_PRIVATE mappings, this is the reserve map which does 10248d9bfb26SMike Kravetz * not apply to children. Faults generated by the children are 10258d9bfb26SMike Kravetz * not guaranteed to succeed, even if read-only. 10268d9bfb26SMike Kravetz */ 1027612b8a31SMike Kravetz if (vma->vm_flags & VM_MAYSHARE) { 1028612b8a31SMike Kravetz struct hugetlb_vma_lock *vma_lock = vma->vm_private_data; 1029612b8a31SMike Kravetz 1030612b8a31SMike Kravetz if (vma_lock && vma_lock->vma != vma) 1031612b8a31SMike Kravetz vma->vm_private_data = NULL; 1032612b8a31SMike Kravetz } else 1033612b8a31SMike Kravetz vma->vm_private_data = NULL; 1034a1e78772SMel Gorman } 1035a1e78772SMel Gorman 1036550a7d60SMina Almasry /* 1037550a7d60SMina Almasry * Reset and decrement one ref on hugepage private reservation. 1038550a7d60SMina Almasry * Called with mm->mmap_sem writer semaphore held. 1039550a7d60SMina Almasry * This function should be only used by move_vma() and operate on 1040550a7d60SMina Almasry * same sized vma. It should never come here with last ref on the 1041550a7d60SMina Almasry * reservation. 1042550a7d60SMina Almasry */ 1043550a7d60SMina Almasry void clear_vma_resv_huge_pages(struct vm_area_struct *vma) 1044550a7d60SMina Almasry { 1045550a7d60SMina Almasry /* 1046550a7d60SMina Almasry * Clear the old hugetlb private page reservation. 1047550a7d60SMina Almasry * It has already been transferred to new_vma. 1048550a7d60SMina Almasry * 1049550a7d60SMina Almasry * During a mremap() operation of a hugetlb vma we call move_vma() 1050550a7d60SMina Almasry * which copies vma into new_vma and unmaps vma. After the copy 1051550a7d60SMina Almasry * operation both new_vma and vma share a reference to the resv_map 1052550a7d60SMina Almasry * struct, and at that point vma is about to be unmapped. We don't 1053550a7d60SMina Almasry * want to return the reservation to the pool at unmap of vma because 1054550a7d60SMina Almasry * the reservation still lives on in new_vma, so simply decrement the 1055550a7d60SMina Almasry * ref here and remove the resv_map reference from this vma. 1056550a7d60SMina Almasry */ 1057550a7d60SMina Almasry struct resv_map *reservations = vma_resv_map(vma); 1058550a7d60SMina Almasry 1059afe041c2SBui Quang Minh if (reservations && is_vma_resv_set(vma, HPAGE_RESV_OWNER)) { 1060afe041c2SBui Quang Minh resv_map_put_hugetlb_cgroup_uncharge_info(reservations); 1061550a7d60SMina Almasry kref_put(&reservations->refs, resv_map_release); 1062afe041c2SBui Quang Minh } 1063550a7d60SMina Almasry 10648d9bfb26SMike Kravetz hugetlb_dup_vma_private(vma); 1065550a7d60SMina Almasry } 1066550a7d60SMina Almasry 1067a1e78772SMel Gorman /* Returns true if the VMA has associated reserve pages */ 1068559ec2f8SNicholas Krause static bool vma_has_reserves(struct vm_area_struct *vma, long chg) 1069a1e78772SMel Gorman { 1070af0ed73eSJoonsoo Kim if (vma->vm_flags & VM_NORESERVE) { 1071af0ed73eSJoonsoo Kim /* 1072af0ed73eSJoonsoo Kim * This address is already reserved by other process(chg == 0), 1073af0ed73eSJoonsoo Kim * so, we should decrement reserved count. Without decrementing, 1074af0ed73eSJoonsoo Kim * reserve count remains after releasing inode, because this 1075af0ed73eSJoonsoo Kim * allocated page will go into page cache and is regarded as 1076af0ed73eSJoonsoo Kim * coming from reserved pool in releasing step. Currently, we 1077af0ed73eSJoonsoo Kim * don't have any other solution to deal with this situation 1078af0ed73eSJoonsoo Kim * properly, so add work-around here. 1079af0ed73eSJoonsoo Kim */ 1080af0ed73eSJoonsoo Kim if (vma->vm_flags & VM_MAYSHARE && chg == 0) 1081559ec2f8SNicholas Krause return true; 1082af0ed73eSJoonsoo Kim else 1083559ec2f8SNicholas Krause return false; 1084af0ed73eSJoonsoo Kim } 1085a63884e9SJoonsoo Kim 1086a63884e9SJoonsoo Kim /* Shared mappings always use reserves */ 10871fb1b0e9SMike Kravetz if (vma->vm_flags & VM_MAYSHARE) { 10881fb1b0e9SMike Kravetz /* 10891fb1b0e9SMike Kravetz * We know VM_NORESERVE is not set. Therefore, there SHOULD 10901fb1b0e9SMike Kravetz * be a region map for all pages. The only situation where 10911fb1b0e9SMike Kravetz * there is no region map is if a hole was punched via 10927c8de358SEthon Paul * fallocate. In this case, there really are no reserves to 10931fb1b0e9SMike Kravetz * use. This situation is indicated if chg != 0. 10941fb1b0e9SMike Kravetz */ 10951fb1b0e9SMike Kravetz if (chg) 10961fb1b0e9SMike Kravetz return false; 10971fb1b0e9SMike Kravetz else 1098559ec2f8SNicholas Krause return true; 10991fb1b0e9SMike Kravetz } 1100a63884e9SJoonsoo Kim 1101a63884e9SJoonsoo Kim /* 1102a63884e9SJoonsoo Kim * Only the process that called mmap() has reserves for 1103a63884e9SJoonsoo Kim * private mappings. 1104a63884e9SJoonsoo Kim */ 110567961f9dSMike Kravetz if (is_vma_resv_set(vma, HPAGE_RESV_OWNER)) { 110667961f9dSMike Kravetz /* 110767961f9dSMike Kravetz * Like the shared case above, a hole punch or truncate 110867961f9dSMike Kravetz * could have been performed on the private mapping. 110967961f9dSMike Kravetz * Examine the value of chg to determine if reserves 111067961f9dSMike Kravetz * actually exist or were previously consumed. 111167961f9dSMike Kravetz * Very Subtle - The value of chg comes from a previous 111267961f9dSMike Kravetz * call to vma_needs_reserves(). The reserve map for 111367961f9dSMike Kravetz * private mappings has different (opposite) semantics 111467961f9dSMike Kravetz * than that of shared mappings. vma_needs_reserves() 111567961f9dSMike Kravetz * has already taken this difference in semantics into 111667961f9dSMike Kravetz * account. Therefore, the meaning of chg is the same 111767961f9dSMike Kravetz * as in the shared case above. Code could easily be 111867961f9dSMike Kravetz * combined, but keeping it separate draws attention to 111967961f9dSMike Kravetz * subtle differences. 112067961f9dSMike Kravetz */ 112167961f9dSMike Kravetz if (chg) 112267961f9dSMike Kravetz return false; 112367961f9dSMike Kravetz else 1124559ec2f8SNicholas Krause return true; 112567961f9dSMike Kravetz } 1126a63884e9SJoonsoo Kim 1127559ec2f8SNicholas Krause return false; 1128a1e78772SMel Gorman } 1129a1e78772SMel Gorman 1130a5516438SAndi Kleen static void enqueue_huge_page(struct hstate *h, struct page *page) 11311da177e4SLinus Torvalds { 11321da177e4SLinus Torvalds int nid = page_to_nid(page); 11339487ca60SMike Kravetz 11349487ca60SMike Kravetz lockdep_assert_held(&hugetlb_lock); 1135b65a4edaSMike Kravetz VM_BUG_ON_PAGE(page_count(page), page); 1136b65a4edaSMike Kravetz 11370edaecfaSAneesh Kumar K.V list_move(&page->lru, &h->hugepage_freelists[nid]); 1138a5516438SAndi Kleen h->free_huge_pages++; 1139a5516438SAndi Kleen h->free_huge_pages_node[nid]++; 11406c037149SMike Kravetz SetHPageFreed(page); 11411da177e4SLinus Torvalds } 11421da177e4SLinus Torvalds 114394310cbcSAnshuman Khandual static struct page *dequeue_huge_page_node_exact(struct hstate *h, int nid) 1144bf50bab2SNaoya Horiguchi { 1145bf50bab2SNaoya Horiguchi struct page *page; 11461a08ae36SPavel Tatashin bool pin = !!(current->flags & PF_MEMALLOC_PIN); 1147bf50bab2SNaoya Horiguchi 11489487ca60SMike Kravetz lockdep_assert_held(&hugetlb_lock); 1149bbe88753SJoonsoo Kim list_for_each_entry(page, &h->hugepage_freelists[nid], lru) { 11506077c943SAlex Sierra if (pin && !is_longterm_pinnable_page(page)) 1151bbe88753SJoonsoo Kim continue; 1152bbe88753SJoonsoo Kim 11536664bfc8SWei Yang if (PageHWPoison(page)) 11546664bfc8SWei Yang continue; 1155bbe88753SJoonsoo Kim 11560edaecfaSAneesh Kumar K.V list_move(&page->lru, &h->hugepage_activelist); 1157a9869b83SNaoya Horiguchi set_page_refcounted(page); 11586c037149SMike Kravetz ClearHPageFreed(page); 1159bf50bab2SNaoya Horiguchi h->free_huge_pages--; 1160bf50bab2SNaoya Horiguchi h->free_huge_pages_node[nid]--; 1161bf50bab2SNaoya Horiguchi return page; 1162bf50bab2SNaoya Horiguchi } 1163bf50bab2SNaoya Horiguchi 11646664bfc8SWei Yang return NULL; 11656664bfc8SWei Yang } 11666664bfc8SWei Yang 11673e59fcb0SMichal Hocko static struct page *dequeue_huge_page_nodemask(struct hstate *h, gfp_t gfp_mask, int nid, 11683e59fcb0SMichal Hocko nodemask_t *nmask) 116994310cbcSAnshuman Khandual { 11703e59fcb0SMichal Hocko unsigned int cpuset_mems_cookie; 11713e59fcb0SMichal Hocko struct zonelist *zonelist; 11723e59fcb0SMichal Hocko struct zone *zone; 11733e59fcb0SMichal Hocko struct zoneref *z; 117498fa15f3SAnshuman Khandual int node = NUMA_NO_NODE; 11753e59fcb0SMichal Hocko 11763e59fcb0SMichal Hocko zonelist = node_zonelist(nid, gfp_mask); 11773e59fcb0SMichal Hocko 11783e59fcb0SMichal Hocko retry_cpuset: 11793e59fcb0SMichal Hocko cpuset_mems_cookie = read_mems_allowed_begin(); 11803e59fcb0SMichal Hocko for_each_zone_zonelist_nodemask(zone, z, zonelist, gfp_zone(gfp_mask), nmask) { 118194310cbcSAnshuman Khandual struct page *page; 118294310cbcSAnshuman Khandual 11833e59fcb0SMichal Hocko if (!cpuset_zone_allowed(zone, gfp_mask)) 11843e59fcb0SMichal Hocko continue; 11853e59fcb0SMichal Hocko /* 11863e59fcb0SMichal Hocko * no need to ask again on the same node. Pool is node rather than 11873e59fcb0SMichal Hocko * zone aware 11883e59fcb0SMichal Hocko */ 11893e59fcb0SMichal Hocko if (zone_to_nid(zone) == node) 11903e59fcb0SMichal Hocko continue; 11913e59fcb0SMichal Hocko node = zone_to_nid(zone); 119294310cbcSAnshuman Khandual 119394310cbcSAnshuman Khandual page = dequeue_huge_page_node_exact(h, node); 119494310cbcSAnshuman Khandual if (page) 119594310cbcSAnshuman Khandual return page; 119694310cbcSAnshuman Khandual } 11973e59fcb0SMichal Hocko if (unlikely(read_mems_allowed_retry(cpuset_mems_cookie))) 11983e59fcb0SMichal Hocko goto retry_cpuset; 11993e59fcb0SMichal Hocko 120094310cbcSAnshuman Khandual return NULL; 120194310cbcSAnshuman Khandual } 120294310cbcSAnshuman Khandual 12038346d69dSXin Hao static unsigned long available_huge_pages(struct hstate *h) 12048346d69dSXin Hao { 12058346d69dSXin Hao return h->free_huge_pages - h->resv_huge_pages; 12068346d69dSXin Hao } 12078346d69dSXin Hao 1208a5516438SAndi Kleen static struct page *dequeue_huge_page_vma(struct hstate *h, 1209a5516438SAndi Kleen struct vm_area_struct *vma, 1210af0ed73eSJoonsoo Kim unsigned long address, int avoid_reserve, 1211af0ed73eSJoonsoo Kim long chg) 12121da177e4SLinus Torvalds { 1213cfcaa66fSBen Widawsky struct page *page = NULL; 1214480eccf9SLee Schermerhorn struct mempolicy *mpol; 121504ec6264SVlastimil Babka gfp_t gfp_mask; 12163e59fcb0SMichal Hocko nodemask_t *nodemask; 121704ec6264SVlastimil Babka int nid; 12181da177e4SLinus Torvalds 1219a1e78772SMel Gorman /* 1220a1e78772SMel Gorman * A child process with MAP_PRIVATE mappings created by their parent 1221a1e78772SMel Gorman * have no page reserves. This check ensures that reservations are 1222a1e78772SMel Gorman * not "stolen". The child may still get SIGKILLed 1223a1e78772SMel Gorman */ 12248346d69dSXin Hao if (!vma_has_reserves(vma, chg) && !available_huge_pages(h)) 1225c0ff7453SMiao Xie goto err; 1226a1e78772SMel Gorman 122704f2cbe3SMel Gorman /* If reserves cannot be used, ensure enough pages are in the pool */ 12288346d69dSXin Hao if (avoid_reserve && !available_huge_pages(h)) 12296eab04a8SJustin P. Mattock goto err; 123004f2cbe3SMel Gorman 123104ec6264SVlastimil Babka gfp_mask = htlb_alloc_mask(h); 123204ec6264SVlastimil Babka nid = huge_node(vma, address, gfp_mask, &mpol, &nodemask); 1233cfcaa66fSBen Widawsky 1234cfcaa66fSBen Widawsky if (mpol_is_preferred_many(mpol)) { 12353e59fcb0SMichal Hocko page = dequeue_huge_page_nodemask(h, gfp_mask, nid, nodemask); 1236cfcaa66fSBen Widawsky 1237cfcaa66fSBen Widawsky /* Fallback to all nodes if page==NULL */ 1238cfcaa66fSBen Widawsky nodemask = NULL; 1239cfcaa66fSBen Widawsky } 1240cfcaa66fSBen Widawsky 1241cfcaa66fSBen Widawsky if (!page) 1242cfcaa66fSBen Widawsky page = dequeue_huge_page_nodemask(h, gfp_mask, nid, nodemask); 1243cfcaa66fSBen Widawsky 12443e59fcb0SMichal Hocko if (page && !avoid_reserve && vma_has_reserves(vma, chg)) { 1245d6995da3SMike Kravetz SetHPageRestoreReserve(page); 1246a63884e9SJoonsoo Kim h->resv_huge_pages--; 1247bf50bab2SNaoya Horiguchi } 1248cc9a6c87SMel Gorman 1249cc9a6c87SMel Gorman mpol_cond_put(mpol); 1250cc9a6c87SMel Gorman return page; 1251cc9a6c87SMel Gorman 1252c0ff7453SMiao Xie err: 1253cc9a6c87SMel Gorman return NULL; 12541da177e4SLinus Torvalds } 12551da177e4SLinus Torvalds 12561cac6f2cSLuiz Capitulino /* 12571cac6f2cSLuiz Capitulino * common helper functions for hstate_next_node_to_{alloc|free}. 12581cac6f2cSLuiz Capitulino * We may have allocated or freed a huge page based on a different 12591cac6f2cSLuiz Capitulino * nodes_allowed previously, so h->next_node_to_{alloc|free} might 12601cac6f2cSLuiz Capitulino * be outside of *nodes_allowed. Ensure that we use an allowed 12611cac6f2cSLuiz Capitulino * node for alloc or free. 12621cac6f2cSLuiz Capitulino */ 12631cac6f2cSLuiz Capitulino static int next_node_allowed(int nid, nodemask_t *nodes_allowed) 12641cac6f2cSLuiz Capitulino { 12650edaf86cSAndrew Morton nid = next_node_in(nid, *nodes_allowed); 12661cac6f2cSLuiz Capitulino VM_BUG_ON(nid >= MAX_NUMNODES); 12671cac6f2cSLuiz Capitulino 12681cac6f2cSLuiz Capitulino return nid; 12691cac6f2cSLuiz Capitulino } 12701cac6f2cSLuiz Capitulino 12711cac6f2cSLuiz Capitulino static int get_valid_node_allowed(int nid, nodemask_t *nodes_allowed) 12721cac6f2cSLuiz Capitulino { 12731cac6f2cSLuiz Capitulino if (!node_isset(nid, *nodes_allowed)) 12741cac6f2cSLuiz Capitulino nid = next_node_allowed(nid, nodes_allowed); 12751cac6f2cSLuiz Capitulino return nid; 12761cac6f2cSLuiz Capitulino } 12771cac6f2cSLuiz Capitulino 12781cac6f2cSLuiz Capitulino /* 12791cac6f2cSLuiz Capitulino * returns the previously saved node ["this node"] from which to 12801cac6f2cSLuiz Capitulino * allocate a persistent huge page for the pool and advance the 12811cac6f2cSLuiz Capitulino * next node from which to allocate, handling wrap at end of node 12821cac6f2cSLuiz Capitulino * mask. 12831cac6f2cSLuiz Capitulino */ 12841cac6f2cSLuiz Capitulino static int hstate_next_node_to_alloc(struct hstate *h, 12851cac6f2cSLuiz Capitulino nodemask_t *nodes_allowed) 12861cac6f2cSLuiz Capitulino { 12871cac6f2cSLuiz Capitulino int nid; 12881cac6f2cSLuiz Capitulino 12891cac6f2cSLuiz Capitulino VM_BUG_ON(!nodes_allowed); 12901cac6f2cSLuiz Capitulino 12911cac6f2cSLuiz Capitulino nid = get_valid_node_allowed(h->next_nid_to_alloc, nodes_allowed); 12921cac6f2cSLuiz Capitulino h->next_nid_to_alloc = next_node_allowed(nid, nodes_allowed); 12931cac6f2cSLuiz Capitulino 12941cac6f2cSLuiz Capitulino return nid; 12951cac6f2cSLuiz Capitulino } 12961cac6f2cSLuiz Capitulino 12971cac6f2cSLuiz Capitulino /* 129810c6ec49SMike Kravetz * helper for remove_pool_huge_page() - return the previously saved 12991cac6f2cSLuiz Capitulino * node ["this node"] from which to free a huge page. Advance the 13001cac6f2cSLuiz Capitulino * next node id whether or not we find a free huge page to free so 13011cac6f2cSLuiz Capitulino * that the next attempt to free addresses the next node. 13021cac6f2cSLuiz Capitulino */ 13031cac6f2cSLuiz Capitulino static int hstate_next_node_to_free(struct hstate *h, nodemask_t *nodes_allowed) 13041cac6f2cSLuiz Capitulino { 13051cac6f2cSLuiz Capitulino int nid; 13061cac6f2cSLuiz Capitulino 13071cac6f2cSLuiz Capitulino VM_BUG_ON(!nodes_allowed); 13081cac6f2cSLuiz Capitulino 13091cac6f2cSLuiz Capitulino nid = get_valid_node_allowed(h->next_nid_to_free, nodes_allowed); 13101cac6f2cSLuiz Capitulino h->next_nid_to_free = next_node_allowed(nid, nodes_allowed); 13111cac6f2cSLuiz Capitulino 13121cac6f2cSLuiz Capitulino return nid; 13131cac6f2cSLuiz Capitulino } 13141cac6f2cSLuiz Capitulino 13151cac6f2cSLuiz Capitulino #define for_each_node_mask_to_alloc(hs, nr_nodes, node, mask) \ 13161cac6f2cSLuiz Capitulino for (nr_nodes = nodes_weight(*mask); \ 13171cac6f2cSLuiz Capitulino nr_nodes > 0 && \ 13181cac6f2cSLuiz Capitulino ((node = hstate_next_node_to_alloc(hs, mask)) || 1); \ 13191cac6f2cSLuiz Capitulino nr_nodes--) 13201cac6f2cSLuiz Capitulino 13211cac6f2cSLuiz Capitulino #define for_each_node_mask_to_free(hs, nr_nodes, node, mask) \ 13221cac6f2cSLuiz Capitulino for (nr_nodes = nodes_weight(*mask); \ 13231cac6f2cSLuiz Capitulino nr_nodes > 0 && \ 13241cac6f2cSLuiz Capitulino ((node = hstate_next_node_to_free(hs, mask)) || 1); \ 13251cac6f2cSLuiz Capitulino nr_nodes--) 13261cac6f2cSLuiz Capitulino 13278531fc6fSMike Kravetz /* used to demote non-gigantic_huge pages as well */ 1328911565b8SSidhartha Kumar static void __destroy_compound_gigantic_folio(struct folio *folio, 132934d9e35bSMike Kravetz unsigned int order, bool demote) 1330944d9fecSLuiz Capitulino { 1331944d9fecSLuiz Capitulino int i; 1332944d9fecSLuiz Capitulino int nr_pages = 1 << order; 133314455eabSCheng Li struct page *p; 1334944d9fecSLuiz Capitulino 1335911565b8SSidhartha Kumar atomic_set(folio_mapcount_ptr(folio), 0); 1336911565b8SSidhartha Kumar atomic_set(folio_subpages_mapcount_ptr(folio), 0); 1337911565b8SSidhartha Kumar atomic_set(folio_pincount_ptr(folio), 0); 133847e29d32SJohn Hubbard 133914455eabSCheng Li for (i = 1; i < nr_pages; i++) { 1340911565b8SSidhartha Kumar p = folio_page(folio, i); 1341a01f4390SMike Kravetz p->mapping = NULL; 13421d798ca3SKirill A. Shutemov clear_compound_head(p); 134334d9e35bSMike Kravetz if (!demote) 1344944d9fecSLuiz Capitulino set_page_refcounted(p); 1345944d9fecSLuiz Capitulino } 1346944d9fecSLuiz Capitulino 1347911565b8SSidhartha Kumar folio_set_compound_order(folio, 0); 1348911565b8SSidhartha Kumar __folio_clear_head(folio); 1349944d9fecSLuiz Capitulino } 1350944d9fecSLuiz Capitulino 1351911565b8SSidhartha Kumar static void destroy_compound_hugetlb_folio_for_demote(struct folio *folio, 13528531fc6fSMike Kravetz unsigned int order) 13538531fc6fSMike Kravetz { 1354911565b8SSidhartha Kumar __destroy_compound_gigantic_folio(folio, order, true); 13558531fc6fSMike Kravetz } 13568531fc6fSMike Kravetz 13578531fc6fSMike Kravetz #ifdef CONFIG_ARCH_HAS_GIGANTIC_PAGE 1358911565b8SSidhartha Kumar static void destroy_compound_gigantic_folio(struct folio *folio, 135934d9e35bSMike Kravetz unsigned int order) 136034d9e35bSMike Kravetz { 1361911565b8SSidhartha Kumar __destroy_compound_gigantic_folio(folio, order, false); 136234d9e35bSMike Kravetz } 136334d9e35bSMike Kravetz 1364d00181b9SKirill A. Shutemov static void free_gigantic_page(struct page *page, unsigned int order) 1365944d9fecSLuiz Capitulino { 1366cf11e85fSRoman Gushchin /* 1367cf11e85fSRoman Gushchin * If the page isn't allocated using the cma allocator, 1368cf11e85fSRoman Gushchin * cma_release() returns false. 1369cf11e85fSRoman Gushchin */ 1370dbda8feaSBarry Song #ifdef CONFIG_CMA 1371dbda8feaSBarry Song if (cma_release(hugetlb_cma[page_to_nid(page)], page, 1 << order)) 1372cf11e85fSRoman Gushchin return; 1373dbda8feaSBarry Song #endif 1374cf11e85fSRoman Gushchin 1375944d9fecSLuiz Capitulino free_contig_range(page_to_pfn(page), 1 << order); 1376944d9fecSLuiz Capitulino } 1377944d9fecSLuiz Capitulino 13784eb0716eSAlexandre Ghiti #ifdef CONFIG_CONTIG_ALLOC 1379d9cc948fSMichal Hocko static struct page *alloc_gigantic_page(struct hstate *h, gfp_t gfp_mask, 1380d9cc948fSMichal Hocko int nid, nodemask_t *nodemask) 1381944d9fecSLuiz Capitulino { 138204adbc3fSMiaohe Lin unsigned long nr_pages = pages_per_huge_page(h); 1383953f064aSLi Xinhai if (nid == NUMA_NO_NODE) 1384953f064aSLi Xinhai nid = numa_mem_id(); 1385944d9fecSLuiz Capitulino 1386dbda8feaSBarry Song #ifdef CONFIG_CMA 1387dbda8feaSBarry Song { 1388cf11e85fSRoman Gushchin struct page *page; 1389cf11e85fSRoman Gushchin int node; 1390cf11e85fSRoman Gushchin 1391953f064aSLi Xinhai if (hugetlb_cma[nid]) { 1392953f064aSLi Xinhai page = cma_alloc(hugetlb_cma[nid], nr_pages, 1393953f064aSLi Xinhai huge_page_order(h), true); 1394953f064aSLi Xinhai if (page) 1395953f064aSLi Xinhai return page; 1396953f064aSLi Xinhai } 1397953f064aSLi Xinhai 1398953f064aSLi Xinhai if (!(gfp_mask & __GFP_THISNODE)) { 1399cf11e85fSRoman Gushchin for_each_node_mask(node, *nodemask) { 1400953f064aSLi Xinhai if (node == nid || !hugetlb_cma[node]) 1401cf11e85fSRoman Gushchin continue; 1402cf11e85fSRoman Gushchin 1403cf11e85fSRoman Gushchin page = cma_alloc(hugetlb_cma[node], nr_pages, 1404cf11e85fSRoman Gushchin huge_page_order(h), true); 1405cf11e85fSRoman Gushchin if (page) 1406cf11e85fSRoman Gushchin return page; 1407cf11e85fSRoman Gushchin } 1408cf11e85fSRoman Gushchin } 1409953f064aSLi Xinhai } 1410dbda8feaSBarry Song #endif 1411cf11e85fSRoman Gushchin 14125e27a2dfSAnshuman Khandual return alloc_contig_pages(nr_pages, gfp_mask, nid, nodemask); 1413944d9fecSLuiz Capitulino } 1414944d9fecSLuiz Capitulino 14154eb0716eSAlexandre Ghiti #else /* !CONFIG_CONTIG_ALLOC */ 14164eb0716eSAlexandre Ghiti static struct page *alloc_gigantic_page(struct hstate *h, gfp_t gfp_mask, 14174eb0716eSAlexandre Ghiti int nid, nodemask_t *nodemask) 14184eb0716eSAlexandre Ghiti { 14194eb0716eSAlexandre Ghiti return NULL; 14204eb0716eSAlexandre Ghiti } 14214eb0716eSAlexandre Ghiti #endif /* CONFIG_CONTIG_ALLOC */ 1422944d9fecSLuiz Capitulino 1423e1073d1eSAneesh Kumar K.V #else /* !CONFIG_ARCH_HAS_GIGANTIC_PAGE */ 1424d9cc948fSMichal Hocko static struct page *alloc_gigantic_page(struct hstate *h, gfp_t gfp_mask, 14254eb0716eSAlexandre Ghiti int nid, nodemask_t *nodemask) 14264eb0716eSAlexandre Ghiti { 14274eb0716eSAlexandre Ghiti return NULL; 14284eb0716eSAlexandre Ghiti } 1429d00181b9SKirill A. Shutemov static inline void free_gigantic_page(struct page *page, unsigned int order) { } 1430911565b8SSidhartha Kumar static inline void destroy_compound_gigantic_folio(struct folio *folio, 1431d00181b9SKirill A. Shutemov unsigned int order) { } 1432944d9fecSLuiz Capitulino #endif 1433944d9fecSLuiz Capitulino 14346eb4e88aSMike Kravetz /* 1435cfd5082bSSidhartha Kumar * Remove hugetlb folio from lists, and update dtor so that the folio appears 143634d9e35bSMike Kravetz * as just a compound page. 143734d9e35bSMike Kravetz * 1438cfd5082bSSidhartha Kumar * A reference is held on the folio, except in the case of demote. 14396eb4e88aSMike Kravetz * 14406eb4e88aSMike Kravetz * Must be called with hugetlb lock held. 14416eb4e88aSMike Kravetz */ 1442cfd5082bSSidhartha Kumar static void __remove_hugetlb_folio(struct hstate *h, struct folio *folio, 144334d9e35bSMike Kravetz bool adjust_surplus, 144434d9e35bSMike Kravetz bool demote) 14456eb4e88aSMike Kravetz { 1446cfd5082bSSidhartha Kumar int nid = folio_nid(folio); 14476eb4e88aSMike Kravetz 1448f074732dSSidhartha Kumar VM_BUG_ON_FOLIO(hugetlb_cgroup_from_folio(folio), folio); 1449f074732dSSidhartha Kumar VM_BUG_ON_FOLIO(hugetlb_cgroup_from_folio_rsvd(folio), folio); 14506eb4e88aSMike Kravetz 14519487ca60SMike Kravetz lockdep_assert_held(&hugetlb_lock); 14526eb4e88aSMike Kravetz if (hstate_is_gigantic(h) && !gigantic_page_runtime_supported()) 14536eb4e88aSMike Kravetz return; 14546eb4e88aSMike Kravetz 1455cfd5082bSSidhartha Kumar list_del(&folio->lru); 14566eb4e88aSMike Kravetz 1457cfd5082bSSidhartha Kumar if (folio_test_hugetlb_freed(folio)) { 14586eb4e88aSMike Kravetz h->free_huge_pages--; 14596eb4e88aSMike Kravetz h->free_huge_pages_node[nid]--; 14606eb4e88aSMike Kravetz } 14616eb4e88aSMike Kravetz if (adjust_surplus) { 14626eb4e88aSMike Kravetz h->surplus_huge_pages--; 14636eb4e88aSMike Kravetz h->surplus_huge_pages_node[nid]--; 14646eb4e88aSMike Kravetz } 14656eb4e88aSMike Kravetz 1466e32d20c0SMike Kravetz /* 1467e32d20c0SMike Kravetz * Very subtle 1468e32d20c0SMike Kravetz * 1469e32d20c0SMike Kravetz * For non-gigantic pages set the destructor to the normal compound 1470e32d20c0SMike Kravetz * page dtor. This is needed in case someone takes an additional 1471e32d20c0SMike Kravetz * temporary ref to the page, and freeing is delayed until they drop 1472e32d20c0SMike Kravetz * their reference. 1473e32d20c0SMike Kravetz * 1474e32d20c0SMike Kravetz * For gigantic pages set the destructor to the null dtor. This 1475e32d20c0SMike Kravetz * destructor will never be called. Before freeing the gigantic 1476911565b8SSidhartha Kumar * page destroy_compound_gigantic_folio will turn the folio into a 1477911565b8SSidhartha Kumar * simple group of pages. After this the destructor does not 1478e32d20c0SMike Kravetz * apply. 1479e32d20c0SMike Kravetz * 1480e32d20c0SMike Kravetz * This handles the case where more than one ref is held when and 1481*d6ef19e2SSidhartha Kumar * after update_and_free_hugetlb_folio is called. 148234d9e35bSMike Kravetz * 148334d9e35bSMike Kravetz * In the case of demote we do not ref count the page as it will soon 148434d9e35bSMike Kravetz * be turned into a page of smaller size. 1485e32d20c0SMike Kravetz */ 148634d9e35bSMike Kravetz if (!demote) 1487cfd5082bSSidhartha Kumar folio_ref_unfreeze(folio, 1); 1488e32d20c0SMike Kravetz if (hstate_is_gigantic(h)) 1489cfd5082bSSidhartha Kumar folio_set_compound_dtor(folio, NULL_COMPOUND_DTOR); 1490e32d20c0SMike Kravetz else 1491cfd5082bSSidhartha Kumar folio_set_compound_dtor(folio, COMPOUND_PAGE_DTOR); 14926eb4e88aSMike Kravetz 14936eb4e88aSMike Kravetz h->nr_huge_pages--; 14946eb4e88aSMike Kravetz h->nr_huge_pages_node[nid]--; 14956eb4e88aSMike Kravetz } 14966eb4e88aSMike Kravetz 1497cfd5082bSSidhartha Kumar static void remove_hugetlb_folio(struct hstate *h, struct folio *folio, 149834d9e35bSMike Kravetz bool adjust_surplus) 149934d9e35bSMike Kravetz { 1500cfd5082bSSidhartha Kumar __remove_hugetlb_folio(h, folio, adjust_surplus, false); 150134d9e35bSMike Kravetz } 150234d9e35bSMike Kravetz 1503cfd5082bSSidhartha Kumar static void remove_hugetlb_folio_for_demote(struct hstate *h, struct folio *folio, 15048531fc6fSMike Kravetz bool adjust_surplus) 15058531fc6fSMike Kravetz { 1506cfd5082bSSidhartha Kumar __remove_hugetlb_folio(h, folio, adjust_surplus, true); 15078531fc6fSMike Kravetz } 15088531fc6fSMike Kravetz 1509ad2fa371SMuchun Song static void add_hugetlb_page(struct hstate *h, struct page *page, 1510ad2fa371SMuchun Song bool adjust_surplus) 1511ad2fa371SMuchun Song { 1512ad2fa371SMuchun Song int zeroed; 1513ad2fa371SMuchun Song int nid = page_to_nid(page); 1514ad2fa371SMuchun Song 1515ad2fa371SMuchun Song VM_BUG_ON_PAGE(!HPageVmemmapOptimized(page), page); 1516ad2fa371SMuchun Song 1517ad2fa371SMuchun Song lockdep_assert_held(&hugetlb_lock); 1518ad2fa371SMuchun Song 1519ad2fa371SMuchun Song INIT_LIST_HEAD(&page->lru); 1520ad2fa371SMuchun Song h->nr_huge_pages++; 1521ad2fa371SMuchun Song h->nr_huge_pages_node[nid]++; 1522ad2fa371SMuchun Song 1523ad2fa371SMuchun Song if (adjust_surplus) { 1524ad2fa371SMuchun Song h->surplus_huge_pages++; 1525ad2fa371SMuchun Song h->surplus_huge_pages_node[nid]++; 1526ad2fa371SMuchun Song } 1527ad2fa371SMuchun Song 1528ad2fa371SMuchun Song set_compound_page_dtor(page, HUGETLB_PAGE_DTOR); 1529ad2fa371SMuchun Song set_page_private(page, 0); 1530a9e1eab2SMiaohe Lin /* 1531a9e1eab2SMiaohe Lin * We have to set HPageVmemmapOptimized again as above 1532a9e1eab2SMiaohe Lin * set_page_private(page, 0) cleared it. 1533a9e1eab2SMiaohe Lin */ 1534ad2fa371SMuchun Song SetHPageVmemmapOptimized(page); 1535ad2fa371SMuchun Song 1536ad2fa371SMuchun Song /* 1537b65a4edaSMike Kravetz * This page is about to be managed by the hugetlb allocator and 1538b65a4edaSMike Kravetz * should have no users. Drop our reference, and check for others 1539b65a4edaSMike Kravetz * just in case. 1540ad2fa371SMuchun Song */ 1541ad2fa371SMuchun Song zeroed = put_page_testzero(page); 1542b65a4edaSMike Kravetz if (!zeroed) 1543b65a4edaSMike Kravetz /* 1544b65a4edaSMike Kravetz * It is VERY unlikely soneone else has taken a ref on 1545b65a4edaSMike Kravetz * the page. In this case, we simply return as the 1546b65a4edaSMike Kravetz * hugetlb destructor (free_huge_page) will be called 1547b65a4edaSMike Kravetz * when this other ref is dropped. 1548b65a4edaSMike Kravetz */ 1549b65a4edaSMike Kravetz return; 1550b65a4edaSMike Kravetz 1551ad2fa371SMuchun Song arch_clear_hugepage_flags(page); 1552ad2fa371SMuchun Song enqueue_huge_page(h, page); 1553ad2fa371SMuchun Song } 1554ad2fa371SMuchun Song 1555b65d4adbSMuchun Song static void __update_and_free_page(struct hstate *h, struct page *page) 15566af2acb6SAdam Litke { 15576af2acb6SAdam Litke int i; 1558911565b8SSidhartha Kumar struct folio *folio = page_folio(page); 155914455eabSCheng Li struct page *subpage; 1560a5516438SAndi Kleen 15614eb0716eSAlexandre Ghiti if (hstate_is_gigantic(h) && !gigantic_page_runtime_supported()) 1562944d9fecSLuiz Capitulino return; 156318229df5SAndy Whitcroft 1564161df60eSNaoya Horiguchi /* 1565161df60eSNaoya Horiguchi * If we don't know which subpages are hwpoisoned, we can't free 1566161df60eSNaoya Horiguchi * the hugepage, so it's leaked intentionally. 1567161df60eSNaoya Horiguchi */ 1568161df60eSNaoya Horiguchi if (HPageRawHwpUnreliable(page)) 1569161df60eSNaoya Horiguchi return; 1570161df60eSNaoya Horiguchi 15716213834cSMuchun Song if (hugetlb_vmemmap_restore(h, page)) { 1572ad2fa371SMuchun Song spin_lock_irq(&hugetlb_lock); 1573ad2fa371SMuchun Song /* 1574ad2fa371SMuchun Song * If we cannot allocate vmemmap pages, just refuse to free the 1575ad2fa371SMuchun Song * page and put the page back on the hugetlb free list and treat 1576ad2fa371SMuchun Song * as a surplus page. 1577ad2fa371SMuchun Song */ 1578ad2fa371SMuchun Song add_hugetlb_page(h, page, true); 1579ad2fa371SMuchun Song spin_unlock_irq(&hugetlb_lock); 1580ad2fa371SMuchun Song return; 1581ad2fa371SMuchun Song } 1582ad2fa371SMuchun Song 1583161df60eSNaoya Horiguchi /* 1584161df60eSNaoya Horiguchi * Move PageHWPoison flag from head page to the raw error pages, 1585161df60eSNaoya Horiguchi * which makes any healthy subpages reusable. 1586161df60eSNaoya Horiguchi */ 1587911565b8SSidhartha Kumar if (unlikely(folio_test_hwpoison(folio))) 1588911565b8SSidhartha Kumar hugetlb_clear_page_hwpoison(&folio->page); 1589161df60eSNaoya Horiguchi 159014455eabSCheng Li for (i = 0; i < pages_per_huge_page(h); i++) { 159114455eabSCheng Li subpage = nth_page(page, i); 1592dbfee5aeSMike Kravetz subpage->flags &= ~(1 << PG_locked | 1 << PG_error | 159332f84528SChris Forbes 1 << PG_referenced | 1 << PG_dirty | 1594a7407a27SLuiz Capitulino 1 << PG_active | 1 << PG_private | 1595a7407a27SLuiz Capitulino 1 << PG_writeback); 15966af2acb6SAdam Litke } 1597a01f4390SMike Kravetz 1598a01f4390SMike Kravetz /* 1599a01f4390SMike Kravetz * Non-gigantic pages demoted from CMA allocated gigantic pages 1600a01f4390SMike Kravetz * need to be given back to CMA in free_gigantic_page. 1601a01f4390SMike Kravetz */ 1602a01f4390SMike Kravetz if (hstate_is_gigantic(h) || 1603a01f4390SMike Kravetz hugetlb_cma_page(page, huge_page_order(h))) { 1604911565b8SSidhartha Kumar destroy_compound_gigantic_folio(folio, huge_page_order(h)); 1605944d9fecSLuiz Capitulino free_gigantic_page(page, huge_page_order(h)); 1606944d9fecSLuiz Capitulino } else { 1607a5516438SAndi Kleen __free_pages(page, huge_page_order(h)); 16086af2acb6SAdam Litke } 1609944d9fecSLuiz Capitulino } 16106af2acb6SAdam Litke 1611b65d4adbSMuchun Song /* 1612*d6ef19e2SSidhartha Kumar * As update_and_free_hugetlb_folio() can be called under any context, so we cannot 1613b65d4adbSMuchun Song * use GFP_KERNEL to allocate vmemmap pages. However, we can defer the 1614b65d4adbSMuchun Song * actual freeing in a workqueue to prevent from using GFP_ATOMIC to allocate 1615b65d4adbSMuchun Song * the vmemmap pages. 1616b65d4adbSMuchun Song * 1617b65d4adbSMuchun Song * free_hpage_workfn() locklessly retrieves the linked list of pages to be 1618b65d4adbSMuchun Song * freed and frees them one-by-one. As the page->mapping pointer is going 1619b65d4adbSMuchun Song * to be cleared in free_hpage_workfn() anyway, it is reused as the llist_node 1620b65d4adbSMuchun Song * structure of a lockless linked list of huge pages to be freed. 1621b65d4adbSMuchun Song */ 1622b65d4adbSMuchun Song static LLIST_HEAD(hpage_freelist); 1623b65d4adbSMuchun Song 1624b65d4adbSMuchun Song static void free_hpage_workfn(struct work_struct *work) 1625b65d4adbSMuchun Song { 1626b65d4adbSMuchun Song struct llist_node *node; 1627b65d4adbSMuchun Song 1628b65d4adbSMuchun Song node = llist_del_all(&hpage_freelist); 1629b65d4adbSMuchun Song 1630b65d4adbSMuchun Song while (node) { 1631b65d4adbSMuchun Song struct page *page; 1632b65d4adbSMuchun Song struct hstate *h; 1633b65d4adbSMuchun Song 1634b65d4adbSMuchun Song page = container_of((struct address_space **)node, 1635b65d4adbSMuchun Song struct page, mapping); 1636b65d4adbSMuchun Song node = node->next; 1637b65d4adbSMuchun Song page->mapping = NULL; 1638b65d4adbSMuchun Song /* 1639b65d4adbSMuchun Song * The VM_BUG_ON_PAGE(!PageHuge(page), page) in page_hstate() 1640b65d4adbSMuchun Song * is going to trigger because a previous call to 1641cfd5082bSSidhartha Kumar * remove_hugetlb_folio() will call folio_set_compound_dtor 1642cfd5082bSSidhartha Kumar * (folio, NULL_COMPOUND_DTOR), so do not use page_hstate() 1643cfd5082bSSidhartha Kumar * directly. 1644b65d4adbSMuchun Song */ 1645b65d4adbSMuchun Song h = size_to_hstate(page_size(page)); 1646b65d4adbSMuchun Song 1647b65d4adbSMuchun Song __update_and_free_page(h, page); 1648b65d4adbSMuchun Song 1649b65d4adbSMuchun Song cond_resched(); 1650b65d4adbSMuchun Song } 1651b65d4adbSMuchun Song } 1652b65d4adbSMuchun Song static DECLARE_WORK(free_hpage_work, free_hpage_workfn); 1653b65d4adbSMuchun Song 1654b65d4adbSMuchun Song static inline void flush_free_hpage_work(struct hstate *h) 1655b65d4adbSMuchun Song { 16566213834cSMuchun Song if (hugetlb_vmemmap_optimizable(h)) 1657b65d4adbSMuchun Song flush_work(&free_hpage_work); 1658b65d4adbSMuchun Song } 1659b65d4adbSMuchun Song 1660*d6ef19e2SSidhartha Kumar static void update_and_free_hugetlb_folio(struct hstate *h, struct folio *folio, 1661b65d4adbSMuchun Song bool atomic) 1662b65d4adbSMuchun Song { 1663*d6ef19e2SSidhartha Kumar if (!folio_test_hugetlb_vmemmap_optimized(folio) || !atomic) { 1664*d6ef19e2SSidhartha Kumar __update_and_free_page(h, &folio->page); 1665b65d4adbSMuchun Song return; 1666b65d4adbSMuchun Song } 1667b65d4adbSMuchun Song 1668b65d4adbSMuchun Song /* 1669b65d4adbSMuchun Song * Defer freeing to avoid using GFP_ATOMIC to allocate vmemmap pages. 1670b65d4adbSMuchun Song * 1671b65d4adbSMuchun Song * Only call schedule_work() if hpage_freelist is previously 1672b65d4adbSMuchun Song * empty. Otherwise, schedule_work() had been called but the workfn 1673b65d4adbSMuchun Song * hasn't retrieved the list yet. 1674b65d4adbSMuchun Song */ 1675*d6ef19e2SSidhartha Kumar if (llist_add((struct llist_node *)&folio->mapping, &hpage_freelist)) 1676b65d4adbSMuchun Song schedule_work(&free_hpage_work); 1677b65d4adbSMuchun Song } 1678b65d4adbSMuchun Song 167910c6ec49SMike Kravetz static void update_and_free_pages_bulk(struct hstate *h, struct list_head *list) 168010c6ec49SMike Kravetz { 168110c6ec49SMike Kravetz struct page *page, *t_page; 1682*d6ef19e2SSidhartha Kumar struct folio *folio; 168310c6ec49SMike Kravetz 168410c6ec49SMike Kravetz list_for_each_entry_safe(page, t_page, list, lru) { 1685*d6ef19e2SSidhartha Kumar folio = page_folio(page); 1686*d6ef19e2SSidhartha Kumar update_and_free_hugetlb_folio(h, folio, false); 168710c6ec49SMike Kravetz cond_resched(); 168810c6ec49SMike Kravetz } 168910c6ec49SMike Kravetz } 169010c6ec49SMike Kravetz 1691e5ff2159SAndi Kleen struct hstate *size_to_hstate(unsigned long size) 1692e5ff2159SAndi Kleen { 1693e5ff2159SAndi Kleen struct hstate *h; 1694e5ff2159SAndi Kleen 1695e5ff2159SAndi Kleen for_each_hstate(h) { 1696e5ff2159SAndi Kleen if (huge_page_size(h) == size) 1697e5ff2159SAndi Kleen return h; 1698e5ff2159SAndi Kleen } 1699e5ff2159SAndi Kleen return NULL; 1700e5ff2159SAndi Kleen } 1701e5ff2159SAndi Kleen 1702db71ef79SMike Kravetz void free_huge_page(struct page *page) 170327a85ef1SDavid Gibson { 1704a5516438SAndi Kleen /* 1705a5516438SAndi Kleen * Can't pass hstate in here because it is called from the 1706a5516438SAndi Kleen * compound page destructor. 1707a5516438SAndi Kleen */ 17080356c4b9SSidhartha Kumar struct folio *folio = page_folio(page); 17090356c4b9SSidhartha Kumar struct hstate *h = folio_hstate(folio); 17100356c4b9SSidhartha Kumar int nid = folio_nid(folio); 17110356c4b9SSidhartha Kumar struct hugepage_subpool *spool = hugetlb_folio_subpool(folio); 171207443a85SJoonsoo Kim bool restore_reserve; 1713db71ef79SMike Kravetz unsigned long flags; 171427a85ef1SDavid Gibson 17150356c4b9SSidhartha Kumar VM_BUG_ON_FOLIO(folio_ref_count(folio), folio); 17160356c4b9SSidhartha Kumar VM_BUG_ON_FOLIO(folio_mapcount(folio), folio); 17178ace22bcSYongkai Wu 17180356c4b9SSidhartha Kumar hugetlb_set_folio_subpool(folio, NULL); 17190356c4b9SSidhartha Kumar if (folio_test_anon(folio)) 17200356c4b9SSidhartha Kumar __ClearPageAnonExclusive(&folio->page); 17210356c4b9SSidhartha Kumar folio->mapping = NULL; 17220356c4b9SSidhartha Kumar restore_reserve = folio_test_hugetlb_restore_reserve(folio); 17230356c4b9SSidhartha Kumar folio_clear_hugetlb_restore_reserve(folio); 172427a85ef1SDavid Gibson 17251c5ecae3SMike Kravetz /* 1726d6995da3SMike Kravetz * If HPageRestoreReserve was set on page, page allocation consumed a 17270919e1b6SMike Kravetz * reservation. If the page was associated with a subpool, there 17280919e1b6SMike Kravetz * would have been a page reserved in the subpool before allocation 17290919e1b6SMike Kravetz * via hugepage_subpool_get_pages(). Since we are 'restoring' the 17306c26d310SMiaohe Lin * reservation, do not call hugepage_subpool_put_pages() as this will 17310919e1b6SMike Kravetz * remove the reserved page from the subpool. 17320919e1b6SMike Kravetz */ 17330919e1b6SMike Kravetz if (!restore_reserve) { 17340919e1b6SMike Kravetz /* 17350919e1b6SMike Kravetz * A return code of zero implies that the subpool will be 17360919e1b6SMike Kravetz * under its minimum size if the reservation is not restored 17370919e1b6SMike Kravetz * after page is free. Therefore, force restore_reserve 17380919e1b6SMike Kravetz * operation. 17391c5ecae3SMike Kravetz */ 17401c5ecae3SMike Kravetz if (hugepage_subpool_put_pages(spool, 1) == 0) 17411c5ecae3SMike Kravetz restore_reserve = true; 17420919e1b6SMike Kravetz } 17431c5ecae3SMike Kravetz 1744db71ef79SMike Kravetz spin_lock_irqsave(&hugetlb_lock, flags); 17450356c4b9SSidhartha Kumar folio_clear_hugetlb_migratable(folio); 1746d4ab0316SSidhartha Kumar hugetlb_cgroup_uncharge_folio(hstate_index(h), 1747d4ab0316SSidhartha Kumar pages_per_huge_page(h), folio); 1748d4ab0316SSidhartha Kumar hugetlb_cgroup_uncharge_folio_rsvd(hstate_index(h), 1749d4ab0316SSidhartha Kumar pages_per_huge_page(h), folio); 175007443a85SJoonsoo Kim if (restore_reserve) 175107443a85SJoonsoo Kim h->resv_huge_pages++; 175207443a85SJoonsoo Kim 17530356c4b9SSidhartha Kumar if (folio_test_hugetlb_temporary(folio)) { 1754cfd5082bSSidhartha Kumar remove_hugetlb_folio(h, folio, false); 1755db71ef79SMike Kravetz spin_unlock_irqrestore(&hugetlb_lock, flags); 1756*d6ef19e2SSidhartha Kumar update_and_free_hugetlb_folio(h, folio, true); 1757ab5ac90aSMichal Hocko } else if (h->surplus_huge_pages_node[nid]) { 17580edaecfaSAneesh Kumar K.V /* remove the page from active list */ 1759cfd5082bSSidhartha Kumar remove_hugetlb_folio(h, folio, true); 1760db71ef79SMike Kravetz spin_unlock_irqrestore(&hugetlb_lock, flags); 1761*d6ef19e2SSidhartha Kumar update_and_free_hugetlb_folio(h, folio, true); 17627893d1d5SAdam Litke } else { 17635d3a551cSWill Deacon arch_clear_hugepage_flags(page); 1764a5516438SAndi Kleen enqueue_huge_page(h, page); 1765db71ef79SMike Kravetz spin_unlock_irqrestore(&hugetlb_lock, flags); 176627a85ef1SDavid Gibson } 17671121828aSMike Kravetz } 176827a85ef1SDavid Gibson 1769d3d99fccSOscar Salvador /* 1770d3d99fccSOscar Salvador * Must be called with the hugetlb lock held 1771d3d99fccSOscar Salvador */ 1772d3d99fccSOscar Salvador static void __prep_account_new_huge_page(struct hstate *h, int nid) 1773d3d99fccSOscar Salvador { 1774d3d99fccSOscar Salvador lockdep_assert_held(&hugetlb_lock); 1775d3d99fccSOscar Salvador h->nr_huge_pages++; 1776d3d99fccSOscar Salvador h->nr_huge_pages_node[nid]++; 1777d3d99fccSOscar Salvador } 1778d3d99fccSOscar Salvador 1779de656ed3SSidhartha Kumar static void __prep_new_hugetlb_folio(struct hstate *h, struct folio *folio) 1780b7ba30c6SAndi Kleen { 1781de656ed3SSidhartha Kumar hugetlb_vmemmap_optimize(h, &folio->page); 1782de656ed3SSidhartha Kumar INIT_LIST_HEAD(&folio->lru); 17839fd33058SSidhartha Kumar folio_set_compound_dtor(folio, HUGETLB_PAGE_DTOR); 1784de656ed3SSidhartha Kumar hugetlb_set_folio_subpool(folio, NULL); 1785de656ed3SSidhartha Kumar set_hugetlb_cgroup(folio, NULL); 1786de656ed3SSidhartha Kumar set_hugetlb_cgroup_rsvd(folio, NULL); 1787d3d99fccSOscar Salvador } 1788d3d99fccSOscar Salvador 1789d3d99fccSOscar Salvador static void prep_new_huge_page(struct hstate *h, struct page *page, int nid) 1790d3d99fccSOscar Salvador { 1791de656ed3SSidhartha Kumar struct folio *folio = page_folio(page); 1792de656ed3SSidhartha Kumar 1793de656ed3SSidhartha Kumar __prep_new_hugetlb_folio(h, folio); 1794db71ef79SMike Kravetz spin_lock_irq(&hugetlb_lock); 1795d3d99fccSOscar Salvador __prep_account_new_huge_page(h, nid); 1796db71ef79SMike Kravetz spin_unlock_irq(&hugetlb_lock); 1797b7ba30c6SAndi Kleen } 1798b7ba30c6SAndi Kleen 179934d9e35bSMike Kravetz static bool __prep_compound_gigantic_page(struct page *page, unsigned int order, 180034d9e35bSMike Kravetz bool demote) 180120a0307cSWu Fengguang { 18027118fc29SMike Kravetz int i, j; 180320a0307cSWu Fengguang int nr_pages = 1 << order; 180414455eabSCheng Li struct page *p; 180520a0307cSWu Fengguang 180620a0307cSWu Fengguang /* we rely on prep_new_huge_page to set the destructor */ 180720a0307cSWu Fengguang set_compound_order(page, order); 18087fb0728aSMike Kravetz __ClearPageReserved(page); 1809de09d31dSKirill A. Shutemov __SetPageHead(page); 18102b21624fSMike Kravetz for (i = 0; i < nr_pages; i++) { 181114455eabSCheng Li p = nth_page(page, i); 181214455eabSCheng Li 1813ef5a22beSAndrea Arcangeli /* 1814ef5a22beSAndrea Arcangeli * For gigantic hugepages allocated through bootmem at 1815ef5a22beSAndrea Arcangeli * boot, it's safer to be consistent with the not-gigantic 1816ef5a22beSAndrea Arcangeli * hugepages and clear the PG_reserved bit from all tail pages 18177c8de358SEthon Paul * too. Otherwise drivers using get_user_pages() to access tail 1818ef5a22beSAndrea Arcangeli * pages may get the reference counting wrong if they see 1819ef5a22beSAndrea Arcangeli * PG_reserved set on a tail page (despite the head page not 1820ef5a22beSAndrea Arcangeli * having PG_reserved set). Enforcing this consistency between 1821ef5a22beSAndrea Arcangeli * head and tail pages allows drivers to optimize away a check 1822ef5a22beSAndrea Arcangeli * on the head page when they need know if put_page() is needed 1823ef5a22beSAndrea Arcangeli * after get_user_pages(). 1824ef5a22beSAndrea Arcangeli */ 18257fb0728aSMike Kravetz if (i != 0) /* head page cleared above */ 1826ef5a22beSAndrea Arcangeli __ClearPageReserved(p); 18277118fc29SMike Kravetz /* 18287118fc29SMike Kravetz * Subtle and very unlikely 18297118fc29SMike Kravetz * 18307118fc29SMike Kravetz * Gigantic 'page allocators' such as memblock or cma will 18317118fc29SMike Kravetz * return a set of pages with each page ref counted. We need 18327118fc29SMike Kravetz * to turn this set of pages into a compound page with tail 18337118fc29SMike Kravetz * page ref counts set to zero. Code such as speculative page 18347118fc29SMike Kravetz * cache adding could take a ref on a 'to be' tail page. 18357118fc29SMike Kravetz * We need to respect any increased ref count, and only set 18367118fc29SMike Kravetz * the ref count to zero if count is currently 1. If count 1837416d85edSMike Kravetz * is not 1, we return an error. An error return indicates 1838416d85edSMike Kravetz * the set of pages can not be converted to a gigantic page. 1839416d85edSMike Kravetz * The caller who allocated the pages should then discard the 1840416d85edSMike Kravetz * pages using the appropriate free interface. 184134d9e35bSMike Kravetz * 184234d9e35bSMike Kravetz * In the case of demote, the ref count will be zero. 18437118fc29SMike Kravetz */ 184434d9e35bSMike Kravetz if (!demote) { 18457118fc29SMike Kravetz if (!page_ref_freeze(p, 1)) { 1846416d85edSMike Kravetz pr_warn("HugeTLB page can not be used due to unexpected inflated ref count\n"); 18477118fc29SMike Kravetz goto out_error; 18487118fc29SMike Kravetz } 184934d9e35bSMike Kravetz } else { 185034d9e35bSMike Kravetz VM_BUG_ON_PAGE(page_count(p), p); 185134d9e35bSMike Kravetz } 18522b21624fSMike Kravetz if (i != 0) 18531d798ca3SKirill A. Shutemov set_compound_head(p, page); 185420a0307cSWu Fengguang } 1855b4330afbSMike Kravetz atomic_set(compound_mapcount_ptr(page), -1); 1856cb67f428SHugh Dickins atomic_set(subpages_mapcount_ptr(page), 0); 185747e29d32SJohn Hubbard atomic_set(compound_pincount_ptr(page), 0); 18587118fc29SMike Kravetz return true; 18597118fc29SMike Kravetz 18607118fc29SMike Kravetz out_error: 18612b21624fSMike Kravetz /* undo page modifications made above */ 18622b21624fSMike Kravetz for (j = 0; j < i; j++) { 186314455eabSCheng Li p = nth_page(page, j); 18642b21624fSMike Kravetz if (j != 0) 18657118fc29SMike Kravetz clear_compound_head(p); 18667118fc29SMike Kravetz set_page_refcounted(p); 18677118fc29SMike Kravetz } 18687118fc29SMike Kravetz /* need to clear PG_reserved on remaining tail pages */ 186914455eabSCheng Li for (; j < nr_pages; j++) { 187014455eabSCheng Li p = nth_page(page, j); 18717118fc29SMike Kravetz __ClearPageReserved(p); 187214455eabSCheng Li } 18737118fc29SMike Kravetz set_compound_order(page, 0); 18745232c63fSMatthew Wilcox (Oracle) #ifdef CONFIG_64BIT 18757118fc29SMike Kravetz page[1].compound_nr = 0; 18765232c63fSMatthew Wilcox (Oracle) #endif 18777118fc29SMike Kravetz __ClearPageHead(page); 18787118fc29SMike Kravetz return false; 187920a0307cSWu Fengguang } 188020a0307cSWu Fengguang 188134d9e35bSMike Kravetz static bool prep_compound_gigantic_page(struct page *page, unsigned int order) 188234d9e35bSMike Kravetz { 188334d9e35bSMike Kravetz return __prep_compound_gigantic_page(page, order, false); 188434d9e35bSMike Kravetz } 188534d9e35bSMike Kravetz 18868531fc6fSMike Kravetz static bool prep_compound_gigantic_page_for_demote(struct page *page, 18878531fc6fSMike Kravetz unsigned int order) 18888531fc6fSMike Kravetz { 18898531fc6fSMike Kravetz return __prep_compound_gigantic_page(page, order, true); 18908531fc6fSMike Kravetz } 18918531fc6fSMike Kravetz 18927795912cSAndrew Morton /* 18937795912cSAndrew Morton * PageHuge() only returns true for hugetlbfs pages, but not for normal or 18947795912cSAndrew Morton * transparent huge pages. See the PageTransHuge() documentation for more 18957795912cSAndrew Morton * details. 18967795912cSAndrew Morton */ 189720a0307cSWu Fengguang int PageHuge(struct page *page) 189820a0307cSWu Fengguang { 189920a0307cSWu Fengguang if (!PageCompound(page)) 190020a0307cSWu Fengguang return 0; 190120a0307cSWu Fengguang 190220a0307cSWu Fengguang page = compound_head(page); 1903f1e61557SKirill A. Shutemov return page[1].compound_dtor == HUGETLB_PAGE_DTOR; 190420a0307cSWu Fengguang } 190543131e14SNaoya Horiguchi EXPORT_SYMBOL_GPL(PageHuge); 190643131e14SNaoya Horiguchi 190727c73ae7SAndrea Arcangeli /* 190827c73ae7SAndrea Arcangeli * PageHeadHuge() only returns true for hugetlbfs head page, but not for 190927c73ae7SAndrea Arcangeli * normal or transparent huge pages. 191027c73ae7SAndrea Arcangeli */ 191127c73ae7SAndrea Arcangeli int PageHeadHuge(struct page *page_head) 191227c73ae7SAndrea Arcangeli { 191327c73ae7SAndrea Arcangeli if (!PageHead(page_head)) 191427c73ae7SAndrea Arcangeli return 0; 191527c73ae7SAndrea Arcangeli 1916d4af73e3SVlastimil Babka return page_head[1].compound_dtor == HUGETLB_PAGE_DTOR; 191727c73ae7SAndrea Arcangeli } 19184e936eccSDavid Howells EXPORT_SYMBOL_GPL(PageHeadHuge); 191927c73ae7SAndrea Arcangeli 1920c0d0381aSMike Kravetz /* 1921c0d0381aSMike Kravetz * Find and lock address space (mapping) in write mode. 1922c0d0381aSMike Kravetz * 1923336bf30eSMike Kravetz * Upon entry, the page is locked which means that page_mapping() is 1924336bf30eSMike Kravetz * stable. Due to locking order, we can only trylock_write. If we can 1925336bf30eSMike Kravetz * not get the lock, simply return NULL to caller. 1926c0d0381aSMike Kravetz */ 1927c0d0381aSMike Kravetz struct address_space *hugetlb_page_mapping_lock_write(struct page *hpage) 1928c0d0381aSMike Kravetz { 1929336bf30eSMike Kravetz struct address_space *mapping = page_mapping(hpage); 1930c0d0381aSMike Kravetz 1931c0d0381aSMike Kravetz if (!mapping) 1932c0d0381aSMike Kravetz return mapping; 1933c0d0381aSMike Kravetz 1934c0d0381aSMike Kravetz if (i_mmap_trylock_write(mapping)) 1935c0d0381aSMike Kravetz return mapping; 1936c0d0381aSMike Kravetz 1937c0d0381aSMike Kravetz return NULL; 1938c0d0381aSMike Kravetz } 1939c0d0381aSMike Kravetz 1940fe19bd3dSHugh Dickins pgoff_t hugetlb_basepage_index(struct page *page) 194113d60f4bSZhang Yi { 194213d60f4bSZhang Yi struct page *page_head = compound_head(page); 194313d60f4bSZhang Yi pgoff_t index = page_index(page_head); 194413d60f4bSZhang Yi unsigned long compound_idx; 194513d60f4bSZhang Yi 194613d60f4bSZhang Yi if (compound_order(page_head) >= MAX_ORDER) 194713d60f4bSZhang Yi compound_idx = page_to_pfn(page) - page_to_pfn(page_head); 194813d60f4bSZhang Yi else 194913d60f4bSZhang Yi compound_idx = page - page_head; 195013d60f4bSZhang Yi 195113d60f4bSZhang Yi return (index << compound_order(page_head)) + compound_idx; 195213d60f4bSZhang Yi } 195313d60f4bSZhang Yi 19540c397daeSMichal Hocko static struct page *alloc_buddy_huge_page(struct hstate *h, 1955f60858f9SMike Kravetz gfp_t gfp_mask, int nid, nodemask_t *nmask, 1956f60858f9SMike Kravetz nodemask_t *node_alloc_noretry) 19571da177e4SLinus Torvalds { 1958af0fb9dfSMichal Hocko int order = huge_page_order(h); 19591da177e4SLinus Torvalds struct page *page; 1960f60858f9SMike Kravetz bool alloc_try_hard = true; 19612b21624fSMike Kravetz bool retry = true; 1962f96efd58SJoe Jin 1963f60858f9SMike Kravetz /* 1964f60858f9SMike Kravetz * By default we always try hard to allocate the page with 1965f60858f9SMike Kravetz * __GFP_RETRY_MAYFAIL flag. However, if we are allocating pages in 1966f60858f9SMike Kravetz * a loop (to adjust global huge page counts) and previous allocation 1967f60858f9SMike Kravetz * failed, do not continue to try hard on the same node. Use the 1968f60858f9SMike Kravetz * node_alloc_noretry bitmap to manage this state information. 1969f60858f9SMike Kravetz */ 1970f60858f9SMike Kravetz if (node_alloc_noretry && node_isset(nid, *node_alloc_noretry)) 1971f60858f9SMike Kravetz alloc_try_hard = false; 1972f60858f9SMike Kravetz gfp_mask |= __GFP_COMP|__GFP_NOWARN; 1973f60858f9SMike Kravetz if (alloc_try_hard) 1974f60858f9SMike Kravetz gfp_mask |= __GFP_RETRY_MAYFAIL; 1975af0fb9dfSMichal Hocko if (nid == NUMA_NO_NODE) 1976af0fb9dfSMichal Hocko nid = numa_mem_id(); 19772b21624fSMike Kravetz retry: 197884172f4bSMatthew Wilcox (Oracle) page = __alloc_pages(gfp_mask, order, nid, nmask); 19792b21624fSMike Kravetz 19802b21624fSMike Kravetz /* Freeze head page */ 19812b21624fSMike Kravetz if (page && !page_ref_freeze(page, 1)) { 19822b21624fSMike Kravetz __free_pages(page, order); 19832b21624fSMike Kravetz if (retry) { /* retry once */ 19842b21624fSMike Kravetz retry = false; 19852b21624fSMike Kravetz goto retry; 19862b21624fSMike Kravetz } 19872b21624fSMike Kravetz /* WOW! twice in a row. */ 19882b21624fSMike Kravetz pr_warn("HugeTLB head page unexpected inflated ref count\n"); 19892b21624fSMike Kravetz page = NULL; 19902b21624fSMike Kravetz } 19912b21624fSMike Kravetz 1992af0fb9dfSMichal Hocko if (page) 1993af0fb9dfSMichal Hocko __count_vm_event(HTLB_BUDDY_PGALLOC); 1994af0fb9dfSMichal Hocko else 1995af0fb9dfSMichal Hocko __count_vm_event(HTLB_BUDDY_PGALLOC_FAIL); 199663b4613cSNishanth Aravamudan 1997f60858f9SMike Kravetz /* 1998f60858f9SMike Kravetz * If we did not specify __GFP_RETRY_MAYFAIL, but still got a page this 1999f60858f9SMike Kravetz * indicates an overall state change. Clear bit so that we resume 2000f60858f9SMike Kravetz * normal 'try hard' allocations. 2001f60858f9SMike Kravetz */ 2002f60858f9SMike Kravetz if (node_alloc_noretry && page && !alloc_try_hard) 2003f60858f9SMike Kravetz node_clear(nid, *node_alloc_noretry); 2004f60858f9SMike Kravetz 2005f60858f9SMike Kravetz /* 2006f60858f9SMike Kravetz * If we tried hard to get a page but failed, set bit so that 2007f60858f9SMike Kravetz * subsequent attempts will not try as hard until there is an 2008f60858f9SMike Kravetz * overall state change. 2009f60858f9SMike Kravetz */ 2010f60858f9SMike Kravetz if (node_alloc_noretry && !page && alloc_try_hard) 2011f60858f9SMike Kravetz node_set(nid, *node_alloc_noretry); 2012f60858f9SMike Kravetz 201363b4613cSNishanth Aravamudan return page; 201463b4613cSNishanth Aravamudan } 201563b4613cSNishanth Aravamudan 2016af0fb9dfSMichal Hocko /* 20170c397daeSMichal Hocko * Common helper to allocate a fresh hugetlb page. All specific allocators 20180c397daeSMichal Hocko * should use this function to get new hugetlb pages 20192b21624fSMike Kravetz * 20202b21624fSMike Kravetz * Note that returned page is 'frozen': ref count of head page and all tail 20212b21624fSMike Kravetz * pages is zero. 20220c397daeSMichal Hocko */ 20230c397daeSMichal Hocko static struct page *alloc_fresh_huge_page(struct hstate *h, 2024f60858f9SMike Kravetz gfp_t gfp_mask, int nid, nodemask_t *nmask, 2025f60858f9SMike Kravetz nodemask_t *node_alloc_noretry) 20260c397daeSMichal Hocko { 20270c397daeSMichal Hocko struct page *page; 20287118fc29SMike Kravetz bool retry = false; 20290c397daeSMichal Hocko 20307118fc29SMike Kravetz retry: 20310c397daeSMichal Hocko if (hstate_is_gigantic(h)) 20320c397daeSMichal Hocko page = alloc_gigantic_page(h, gfp_mask, nid, nmask); 20330c397daeSMichal Hocko else 20340c397daeSMichal Hocko page = alloc_buddy_huge_page(h, gfp_mask, 2035f60858f9SMike Kravetz nid, nmask, node_alloc_noretry); 20360c397daeSMichal Hocko if (!page) 20370c397daeSMichal Hocko return NULL; 20380c397daeSMichal Hocko 20397118fc29SMike Kravetz if (hstate_is_gigantic(h)) { 20407118fc29SMike Kravetz if (!prep_compound_gigantic_page(page, huge_page_order(h))) { 20417118fc29SMike Kravetz /* 20427118fc29SMike Kravetz * Rare failure to convert pages to compound page. 20437118fc29SMike Kravetz * Free pages and try again - ONCE! 20447118fc29SMike Kravetz */ 20457118fc29SMike Kravetz free_gigantic_page(page, huge_page_order(h)); 20467118fc29SMike Kravetz if (!retry) { 20477118fc29SMike Kravetz retry = true; 20487118fc29SMike Kravetz goto retry; 20497118fc29SMike Kravetz } 20507118fc29SMike Kravetz return NULL; 20517118fc29SMike Kravetz } 20527118fc29SMike Kravetz } 20530c397daeSMichal Hocko prep_new_huge_page(h, page, page_to_nid(page)); 20540c397daeSMichal Hocko 20550c397daeSMichal Hocko return page; 20560c397daeSMichal Hocko } 20570c397daeSMichal Hocko 20580c397daeSMichal Hocko /* 2059af0fb9dfSMichal Hocko * Allocates a fresh page to the hugetlb allocator pool in the node interleaved 2060af0fb9dfSMichal Hocko * manner. 2061af0fb9dfSMichal Hocko */ 2062f60858f9SMike Kravetz static int alloc_pool_huge_page(struct hstate *h, nodemask_t *nodes_allowed, 2063f60858f9SMike Kravetz nodemask_t *node_alloc_noretry) 2064b2261026SJoonsoo Kim { 2065b2261026SJoonsoo Kim struct page *page; 2066b2261026SJoonsoo Kim int nr_nodes, node; 2067af0fb9dfSMichal Hocko gfp_t gfp_mask = htlb_alloc_mask(h) | __GFP_THISNODE; 2068b2261026SJoonsoo Kim 2069b2261026SJoonsoo Kim for_each_node_mask_to_alloc(h, nr_nodes, node, nodes_allowed) { 2070f60858f9SMike Kravetz page = alloc_fresh_huge_page(h, gfp_mask, node, nodes_allowed, 2071f60858f9SMike Kravetz node_alloc_noretry); 2072af0fb9dfSMichal Hocko if (page) 2073b2261026SJoonsoo Kim break; 2074b2261026SJoonsoo Kim } 2075b2261026SJoonsoo Kim 2076af0fb9dfSMichal Hocko if (!page) 2077af0fb9dfSMichal Hocko return 0; 2078b2261026SJoonsoo Kim 20792b21624fSMike Kravetz free_huge_page(page); /* free it into the hugepage allocator */ 2080af0fb9dfSMichal Hocko 2081af0fb9dfSMichal Hocko return 1; 2082b2261026SJoonsoo Kim } 2083b2261026SJoonsoo Kim 2084e8c5c824SLee Schermerhorn /* 208510c6ec49SMike Kravetz * Remove huge page from pool from next node to free. Attempt to keep 208610c6ec49SMike Kravetz * persistent huge pages more or less balanced over allowed nodes. 208710c6ec49SMike Kravetz * This routine only 'removes' the hugetlb page. The caller must make 208810c6ec49SMike Kravetz * an additional call to free the page to low level allocators. 2089e8c5c824SLee Schermerhorn * Called with hugetlb_lock locked. 2090e8c5c824SLee Schermerhorn */ 209110c6ec49SMike Kravetz static struct page *remove_pool_huge_page(struct hstate *h, 209210c6ec49SMike Kravetz nodemask_t *nodes_allowed, 20936ae11b27SLee Schermerhorn bool acct_surplus) 2094e8c5c824SLee Schermerhorn { 2095b2261026SJoonsoo Kim int nr_nodes, node; 209610c6ec49SMike Kravetz struct page *page = NULL; 2097cfd5082bSSidhartha Kumar struct folio *folio; 2098e8c5c824SLee Schermerhorn 20999487ca60SMike Kravetz lockdep_assert_held(&hugetlb_lock); 2100b2261026SJoonsoo Kim for_each_node_mask_to_free(h, nr_nodes, node, nodes_allowed) { 2101685f3457SLee Schermerhorn /* 2102685f3457SLee Schermerhorn * If we're returning unused surplus pages, only examine 2103685f3457SLee Schermerhorn * nodes with surplus pages. 2104685f3457SLee Schermerhorn */ 2105b2261026SJoonsoo Kim if ((!acct_surplus || h->surplus_huge_pages_node[node]) && 2106b2261026SJoonsoo Kim !list_empty(&h->hugepage_freelists[node])) { 210710c6ec49SMike Kravetz page = list_entry(h->hugepage_freelists[node].next, 2108e8c5c824SLee Schermerhorn struct page, lru); 2109cfd5082bSSidhartha Kumar folio = page_folio(page); 2110cfd5082bSSidhartha Kumar remove_hugetlb_folio(h, folio, acct_surplus); 21119a76db09SLee Schermerhorn break; 2112e8c5c824SLee Schermerhorn } 2113b2261026SJoonsoo Kim } 2114e8c5c824SLee Schermerhorn 211510c6ec49SMike Kravetz return page; 2116e8c5c824SLee Schermerhorn } 2117e8c5c824SLee Schermerhorn 2118c8721bbbSNaoya Horiguchi /* 2119c8721bbbSNaoya Horiguchi * Dissolve a given free hugepage into free buddy pages. This function does 2120faf53defSNaoya Horiguchi * nothing for in-use hugepages and non-hugepages. 2121faf53defSNaoya Horiguchi * This function returns values like below: 2122faf53defSNaoya Horiguchi * 2123ad2fa371SMuchun Song * -ENOMEM: failed to allocate vmemmap pages to free the freed hugepages 2124ad2fa371SMuchun Song * when the system is under memory pressure and the feature of 2125ad2fa371SMuchun Song * freeing unused vmemmap pages associated with each hugetlb page 2126ad2fa371SMuchun Song * is enabled. 2127faf53defSNaoya Horiguchi * -EBUSY: failed to dissolved free hugepages or the hugepage is in-use 2128faf53defSNaoya Horiguchi * (allocated or reserved.) 2129faf53defSNaoya Horiguchi * 0: successfully dissolved free hugepages or the page is not a 2130faf53defSNaoya Horiguchi * hugepage (considered as already dissolved) 2131c8721bbbSNaoya Horiguchi */ 2132c3114a84SAnshuman Khandual int dissolve_free_huge_page(struct page *page) 2133c8721bbbSNaoya Horiguchi { 21346bc9b564SNaoya Horiguchi int rc = -EBUSY; 21351a7cdab5SSidhartha Kumar struct folio *folio = page_folio(page); 2136082d5b6bSGerald Schaefer 21377ffddd49SMuchun Song retry: 2138faf53defSNaoya Horiguchi /* Not to disrupt normal path by vainly holding hugetlb_lock */ 21391a7cdab5SSidhartha Kumar if (!folio_test_hugetlb(folio)) 2140faf53defSNaoya Horiguchi return 0; 2141faf53defSNaoya Horiguchi 2142db71ef79SMike Kravetz spin_lock_irq(&hugetlb_lock); 21431a7cdab5SSidhartha Kumar if (!folio_test_hugetlb(folio)) { 2144faf53defSNaoya Horiguchi rc = 0; 2145faf53defSNaoya Horiguchi goto out; 2146faf53defSNaoya Horiguchi } 2147faf53defSNaoya Horiguchi 21481a7cdab5SSidhartha Kumar if (!folio_ref_count(folio)) { 21491a7cdab5SSidhartha Kumar struct hstate *h = folio_hstate(folio); 21508346d69dSXin Hao if (!available_huge_pages(h)) 2151082d5b6bSGerald Schaefer goto out; 21527ffddd49SMuchun Song 21537ffddd49SMuchun Song /* 21547ffddd49SMuchun Song * We should make sure that the page is already on the free list 21557ffddd49SMuchun Song * when it is dissolved. 21567ffddd49SMuchun Song */ 21571a7cdab5SSidhartha Kumar if (unlikely(!folio_test_hugetlb_freed(folio))) { 2158db71ef79SMike Kravetz spin_unlock_irq(&hugetlb_lock); 21597ffddd49SMuchun Song cond_resched(); 21607ffddd49SMuchun Song 21617ffddd49SMuchun Song /* 21627ffddd49SMuchun Song * Theoretically, we should return -EBUSY when we 21637ffddd49SMuchun Song * encounter this race. In fact, we have a chance 21647ffddd49SMuchun Song * to successfully dissolve the page if we do a 21657ffddd49SMuchun Song * retry. Because the race window is quite small. 21667ffddd49SMuchun Song * If we seize this opportunity, it is an optimization 21677ffddd49SMuchun Song * for increasing the success rate of dissolving page. 21687ffddd49SMuchun Song */ 21697ffddd49SMuchun Song goto retry; 21707ffddd49SMuchun Song } 21717ffddd49SMuchun Song 2172cfd5082bSSidhartha Kumar remove_hugetlb_folio(h, folio, false); 2173ad2fa371SMuchun Song h->max_huge_pages--; 2174ad2fa371SMuchun Song spin_unlock_irq(&hugetlb_lock); 2175ad2fa371SMuchun Song 2176c3114a84SAnshuman Khandual /* 2177*d6ef19e2SSidhartha Kumar * Normally update_and_free_hugtlb_folio will allocate required vmemmmap 2178*d6ef19e2SSidhartha Kumar * before freeing the page. update_and_free_hugtlb_folio will fail to 2179ad2fa371SMuchun Song * free the page if it can not allocate required vmemmap. We 2180ad2fa371SMuchun Song * need to adjust max_huge_pages if the page is not freed. 2181ad2fa371SMuchun Song * Attempt to allocate vmemmmap here so that we can take 2182ad2fa371SMuchun Song * appropriate action on failure. 2183ad2fa371SMuchun Song */ 21841a7cdab5SSidhartha Kumar rc = hugetlb_vmemmap_restore(h, &folio->page); 2185ad2fa371SMuchun Song if (!rc) { 2186*d6ef19e2SSidhartha Kumar update_and_free_hugetlb_folio(h, folio, false); 2187ad2fa371SMuchun Song } else { 2188ad2fa371SMuchun Song spin_lock_irq(&hugetlb_lock); 21891a7cdab5SSidhartha Kumar add_hugetlb_page(h, &folio->page, false); 2190ad2fa371SMuchun Song h->max_huge_pages++; 2191ad2fa371SMuchun Song spin_unlock_irq(&hugetlb_lock); 2192ad2fa371SMuchun Song } 2193ad2fa371SMuchun Song 2194ad2fa371SMuchun Song return rc; 2195c8721bbbSNaoya Horiguchi } 2196082d5b6bSGerald Schaefer out: 2197db71ef79SMike Kravetz spin_unlock_irq(&hugetlb_lock); 2198082d5b6bSGerald Schaefer return rc; 2199c8721bbbSNaoya Horiguchi } 2200c8721bbbSNaoya Horiguchi 2201c8721bbbSNaoya Horiguchi /* 2202c8721bbbSNaoya Horiguchi * Dissolve free hugepages in a given pfn range. Used by memory hotplug to 2203c8721bbbSNaoya Horiguchi * make specified memory blocks removable from the system. 22042247bb33SGerald Schaefer * Note that this will dissolve a free gigantic hugepage completely, if any 22052247bb33SGerald Schaefer * part of it lies within the given range. 2206082d5b6bSGerald Schaefer * Also note that if dissolve_free_huge_page() returns with an error, all 2207082d5b6bSGerald Schaefer * free hugepages that were dissolved before that error are lost. 2208c8721bbbSNaoya Horiguchi */ 2209082d5b6bSGerald Schaefer int dissolve_free_huge_pages(unsigned long start_pfn, unsigned long end_pfn) 2210c8721bbbSNaoya Horiguchi { 2211c8721bbbSNaoya Horiguchi unsigned long pfn; 2212eb03aa00SGerald Schaefer struct page *page; 2213082d5b6bSGerald Schaefer int rc = 0; 2214dc2628f3SMuchun Song unsigned int order; 2215dc2628f3SMuchun Song struct hstate *h; 2216c8721bbbSNaoya Horiguchi 2217d0177639SLi Zhong if (!hugepages_supported()) 2218082d5b6bSGerald Schaefer return rc; 2219d0177639SLi Zhong 2220dc2628f3SMuchun Song order = huge_page_order(&default_hstate); 2221dc2628f3SMuchun Song for_each_hstate(h) 2222dc2628f3SMuchun Song order = min(order, huge_page_order(h)); 2223dc2628f3SMuchun Song 2224dc2628f3SMuchun Song for (pfn = start_pfn; pfn < end_pfn; pfn += 1 << order) { 2225eb03aa00SGerald Schaefer page = pfn_to_page(pfn); 2226eb03aa00SGerald Schaefer rc = dissolve_free_huge_page(page); 2227eb03aa00SGerald Schaefer if (rc) 2228082d5b6bSGerald Schaefer break; 2229eb03aa00SGerald Schaefer } 2230082d5b6bSGerald Schaefer 2231082d5b6bSGerald Schaefer return rc; 2232c8721bbbSNaoya Horiguchi } 2233c8721bbbSNaoya Horiguchi 2234ab5ac90aSMichal Hocko /* 2235ab5ac90aSMichal Hocko * Allocates a fresh surplus page from the page allocator. 2236ab5ac90aSMichal Hocko */ 22370c397daeSMichal Hocko static struct page *alloc_surplus_huge_page(struct hstate *h, gfp_t gfp_mask, 22382b21624fSMike Kravetz int nid, nodemask_t *nmask) 22397893d1d5SAdam Litke { 22409980d744SMichal Hocko struct page *page = NULL; 22417893d1d5SAdam Litke 2242bae7f4aeSLuiz Capitulino if (hstate_is_gigantic(h)) 2243aa888a74SAndi Kleen return NULL; 2244aa888a74SAndi Kleen 2245db71ef79SMike Kravetz spin_lock_irq(&hugetlb_lock); 22469980d744SMichal Hocko if (h->surplus_huge_pages >= h->nr_overcommit_huge_pages) 22479980d744SMichal Hocko goto out_unlock; 2248db71ef79SMike Kravetz spin_unlock_irq(&hugetlb_lock); 2249d1c3fb1fSNishanth Aravamudan 2250f60858f9SMike Kravetz page = alloc_fresh_huge_page(h, gfp_mask, nid, nmask, NULL); 22519980d744SMichal Hocko if (!page) 22520c397daeSMichal Hocko return NULL; 2253d1c3fb1fSNishanth Aravamudan 2254db71ef79SMike Kravetz spin_lock_irq(&hugetlb_lock); 22559980d744SMichal Hocko /* 22569980d744SMichal Hocko * We could have raced with the pool size change. 22579980d744SMichal Hocko * Double check that and simply deallocate the new page 22589980d744SMichal Hocko * if we would end up overcommiting the surpluses. Abuse 22599980d744SMichal Hocko * temporary page to workaround the nasty free_huge_page 22609980d744SMichal Hocko * codeflow 22619980d744SMichal Hocko */ 22629980d744SMichal Hocko if (h->surplus_huge_pages >= h->nr_overcommit_huge_pages) { 22639157c311SMike Kravetz SetHPageTemporary(page); 2264db71ef79SMike Kravetz spin_unlock_irq(&hugetlb_lock); 22652b21624fSMike Kravetz free_huge_page(page); 22662bf753e6SKai Shen return NULL; 2267b65a4edaSMike Kravetz } 2268b65a4edaSMike Kravetz 22699980d744SMichal Hocko h->surplus_huge_pages++; 22704704dea3SMichal Hocko h->surplus_huge_pages_node[page_to_nid(page)]++; 22719980d744SMichal Hocko 22729980d744SMichal Hocko out_unlock: 2273db71ef79SMike Kravetz spin_unlock_irq(&hugetlb_lock); 22747893d1d5SAdam Litke 22757893d1d5SAdam Litke return page; 22767893d1d5SAdam Litke } 22777893d1d5SAdam Litke 2278bbe88753SJoonsoo Kim static struct page *alloc_migrate_huge_page(struct hstate *h, gfp_t gfp_mask, 2279ab5ac90aSMichal Hocko int nid, nodemask_t *nmask) 2280ab5ac90aSMichal Hocko { 2281ab5ac90aSMichal Hocko struct page *page; 2282ab5ac90aSMichal Hocko 2283ab5ac90aSMichal Hocko if (hstate_is_gigantic(h)) 2284ab5ac90aSMichal Hocko return NULL; 2285ab5ac90aSMichal Hocko 2286f60858f9SMike Kravetz page = alloc_fresh_huge_page(h, gfp_mask, nid, nmask, NULL); 2287ab5ac90aSMichal Hocko if (!page) 2288ab5ac90aSMichal Hocko return NULL; 2289ab5ac90aSMichal Hocko 22902b21624fSMike Kravetz /* fresh huge pages are frozen */ 22912b21624fSMike Kravetz set_page_refcounted(page); 22922b21624fSMike Kravetz 2293ab5ac90aSMichal Hocko /* 2294ab5ac90aSMichal Hocko * We do not account these pages as surplus because they are only 2295ab5ac90aSMichal Hocko * temporary and will be released properly on the last reference 2296ab5ac90aSMichal Hocko */ 22979157c311SMike Kravetz SetHPageTemporary(page); 2298ab5ac90aSMichal Hocko 2299ab5ac90aSMichal Hocko return page; 2300ab5ac90aSMichal Hocko } 2301ab5ac90aSMichal Hocko 2302e4e574b7SAdam Litke /* 2303099730d6SDave Hansen * Use the VMA's mpolicy to allocate a huge page from the buddy. 2304099730d6SDave Hansen */ 2305e0ec90eeSDave Hansen static 23060c397daeSMichal Hocko struct page *alloc_buddy_huge_page_with_mpol(struct hstate *h, 2307099730d6SDave Hansen struct vm_area_struct *vma, unsigned long addr) 2308099730d6SDave Hansen { 2309cfcaa66fSBen Widawsky struct page *page = NULL; 2310aaf14e40SMichal Hocko struct mempolicy *mpol; 2311aaf14e40SMichal Hocko gfp_t gfp_mask = htlb_alloc_mask(h); 2312aaf14e40SMichal Hocko int nid; 2313aaf14e40SMichal Hocko nodemask_t *nodemask; 2314aaf14e40SMichal Hocko 2315aaf14e40SMichal Hocko nid = huge_node(vma, addr, gfp_mask, &mpol, &nodemask); 2316cfcaa66fSBen Widawsky if (mpol_is_preferred_many(mpol)) { 2317cfcaa66fSBen Widawsky gfp_t gfp = gfp_mask | __GFP_NOWARN; 2318cfcaa66fSBen Widawsky 2319cfcaa66fSBen Widawsky gfp &= ~(__GFP_DIRECT_RECLAIM | __GFP_NOFAIL); 23202b21624fSMike Kravetz page = alloc_surplus_huge_page(h, gfp, nid, nodemask); 2321cfcaa66fSBen Widawsky 2322cfcaa66fSBen Widawsky /* Fallback to all nodes if page==NULL */ 2323cfcaa66fSBen Widawsky nodemask = NULL; 2324cfcaa66fSBen Widawsky } 2325cfcaa66fSBen Widawsky 2326cfcaa66fSBen Widawsky if (!page) 23272b21624fSMike Kravetz page = alloc_surplus_huge_page(h, gfp_mask, nid, nodemask); 2328aaf14e40SMichal Hocko mpol_cond_put(mpol); 2329aaf14e40SMichal Hocko return page; 2330099730d6SDave Hansen } 2331099730d6SDave Hansen 2332ab5ac90aSMichal Hocko /* page migration callback function */ 23333e59fcb0SMichal Hocko struct page *alloc_huge_page_nodemask(struct hstate *h, int preferred_nid, 2334d92bbc27SJoonsoo Kim nodemask_t *nmask, gfp_t gfp_mask) 23354db9b2efSMichal Hocko { 2336db71ef79SMike Kravetz spin_lock_irq(&hugetlb_lock); 23378346d69dSXin Hao if (available_huge_pages(h)) { 23383e59fcb0SMichal Hocko struct page *page; 23393e59fcb0SMichal Hocko 23403e59fcb0SMichal Hocko page = dequeue_huge_page_nodemask(h, gfp_mask, preferred_nid, nmask); 23413e59fcb0SMichal Hocko if (page) { 2342db71ef79SMike Kravetz spin_unlock_irq(&hugetlb_lock); 23433e59fcb0SMichal Hocko return page; 23444db9b2efSMichal Hocko } 23454db9b2efSMichal Hocko } 2346db71ef79SMike Kravetz spin_unlock_irq(&hugetlb_lock); 23474db9b2efSMichal Hocko 23480c397daeSMichal Hocko return alloc_migrate_huge_page(h, gfp_mask, preferred_nid, nmask); 23494db9b2efSMichal Hocko } 23504db9b2efSMichal Hocko 2351ebd63723SMichal Hocko /* mempolicy aware migration callback */ 2352389c8178SMichal Hocko struct page *alloc_huge_page_vma(struct hstate *h, struct vm_area_struct *vma, 2353389c8178SMichal Hocko unsigned long address) 2354ebd63723SMichal Hocko { 2355ebd63723SMichal Hocko struct mempolicy *mpol; 2356ebd63723SMichal Hocko nodemask_t *nodemask; 2357ebd63723SMichal Hocko struct page *page; 2358ebd63723SMichal Hocko gfp_t gfp_mask; 2359ebd63723SMichal Hocko int node; 2360ebd63723SMichal Hocko 2361ebd63723SMichal Hocko gfp_mask = htlb_alloc_mask(h); 2362ebd63723SMichal Hocko node = huge_node(vma, address, gfp_mask, &mpol, &nodemask); 2363d92bbc27SJoonsoo Kim page = alloc_huge_page_nodemask(h, node, nodemask, gfp_mask); 2364ebd63723SMichal Hocko mpol_cond_put(mpol); 2365ebd63723SMichal Hocko 2366ebd63723SMichal Hocko return page; 2367ebd63723SMichal Hocko } 2368ebd63723SMichal Hocko 2369bf50bab2SNaoya Horiguchi /* 237025985edcSLucas De Marchi * Increase the hugetlb pool such that it can accommodate a reservation 2371e4e574b7SAdam Litke * of size 'delta'. 2372e4e574b7SAdam Litke */ 23730a4f3d1bSLiu Xiang static int gather_surplus_pages(struct hstate *h, long delta) 23741b2a1e7bSJules Irenge __must_hold(&hugetlb_lock) 2375e4e574b7SAdam Litke { 237634665341SMiaohe Lin LIST_HEAD(surplus_list); 2377e4e574b7SAdam Litke struct page *page, *tmp; 23780a4f3d1bSLiu Xiang int ret; 23790a4f3d1bSLiu Xiang long i; 23800a4f3d1bSLiu Xiang long needed, allocated; 238128073b02SHillf Danton bool alloc_ok = true; 2382e4e574b7SAdam Litke 23839487ca60SMike Kravetz lockdep_assert_held(&hugetlb_lock); 2384a5516438SAndi Kleen needed = (h->resv_huge_pages + delta) - h->free_huge_pages; 2385ac09b3a1SAdam Litke if (needed <= 0) { 2386a5516438SAndi Kleen h->resv_huge_pages += delta; 2387e4e574b7SAdam Litke return 0; 2388ac09b3a1SAdam Litke } 2389e4e574b7SAdam Litke 2390e4e574b7SAdam Litke allocated = 0; 2391e4e574b7SAdam Litke 2392e4e574b7SAdam Litke ret = -ENOMEM; 2393e4e574b7SAdam Litke retry: 2394db71ef79SMike Kravetz spin_unlock_irq(&hugetlb_lock); 2395e4e574b7SAdam Litke for (i = 0; i < needed; i++) { 23960c397daeSMichal Hocko page = alloc_surplus_huge_page(h, htlb_alloc_mask(h), 23972b21624fSMike Kravetz NUMA_NO_NODE, NULL); 239828073b02SHillf Danton if (!page) { 239928073b02SHillf Danton alloc_ok = false; 240028073b02SHillf Danton break; 240128073b02SHillf Danton } 2402e4e574b7SAdam Litke list_add(&page->lru, &surplus_list); 240369ed779aSDavid Rientjes cond_resched(); 2404e4e574b7SAdam Litke } 240528073b02SHillf Danton allocated += i; 2406e4e574b7SAdam Litke 2407e4e574b7SAdam Litke /* 2408e4e574b7SAdam Litke * After retaking hugetlb_lock, we need to recalculate 'needed' 2409e4e574b7SAdam Litke * because either resv_huge_pages or free_huge_pages may have changed. 2410e4e574b7SAdam Litke */ 2411db71ef79SMike Kravetz spin_lock_irq(&hugetlb_lock); 2412a5516438SAndi Kleen needed = (h->resv_huge_pages + delta) - 2413a5516438SAndi Kleen (h->free_huge_pages + allocated); 241428073b02SHillf Danton if (needed > 0) { 241528073b02SHillf Danton if (alloc_ok) 2416e4e574b7SAdam Litke goto retry; 241728073b02SHillf Danton /* 241828073b02SHillf Danton * We were not able to allocate enough pages to 241928073b02SHillf Danton * satisfy the entire reservation so we free what 242028073b02SHillf Danton * we've allocated so far. 242128073b02SHillf Danton */ 242228073b02SHillf Danton goto free; 242328073b02SHillf Danton } 2424e4e574b7SAdam Litke /* 2425e4e574b7SAdam Litke * The surplus_list now contains _at_least_ the number of extra pages 242625985edcSLucas De Marchi * needed to accommodate the reservation. Add the appropriate number 2427e4e574b7SAdam Litke * of pages to the hugetlb pool and free the extras back to the buddy 2428ac09b3a1SAdam Litke * allocator. Commit the entire reservation here to prevent another 2429ac09b3a1SAdam Litke * process from stealing the pages as they are added to the pool but 2430ac09b3a1SAdam Litke * before they are reserved. 2431e4e574b7SAdam Litke */ 2432e4e574b7SAdam Litke needed += allocated; 2433a5516438SAndi Kleen h->resv_huge_pages += delta; 2434e4e574b7SAdam Litke ret = 0; 2435a9869b83SNaoya Horiguchi 243619fc3f0aSAdam Litke /* Free the needed pages to the hugetlb pool */ 243719fc3f0aSAdam Litke list_for_each_entry_safe(page, tmp, &surplus_list, lru) { 243819fc3f0aSAdam Litke if ((--needed) < 0) 243919fc3f0aSAdam Litke break; 2440b65a4edaSMike Kravetz /* Add the page to the hugetlb allocator */ 2441a5516438SAndi Kleen enqueue_huge_page(h, page); 244219fc3f0aSAdam Litke } 244328073b02SHillf Danton free: 2444db71ef79SMike Kravetz spin_unlock_irq(&hugetlb_lock); 244519fc3f0aSAdam Litke 2446b65a4edaSMike Kravetz /* 2447b65a4edaSMike Kravetz * Free unnecessary surplus pages to the buddy allocator. 2448b65a4edaSMike Kravetz * Pages have no ref count, call free_huge_page directly. 2449b65a4edaSMike Kravetz */ 2450c0d934baSJoonsoo Kim list_for_each_entry_safe(page, tmp, &surplus_list, lru) 2451b65a4edaSMike Kravetz free_huge_page(page); 2452db71ef79SMike Kravetz spin_lock_irq(&hugetlb_lock); 2453e4e574b7SAdam Litke 2454e4e574b7SAdam Litke return ret; 2455e4e574b7SAdam Litke } 2456e4e574b7SAdam Litke 2457e4e574b7SAdam Litke /* 2458e5bbc8a6SMike Kravetz * This routine has two main purposes: 2459e5bbc8a6SMike Kravetz * 1) Decrement the reservation count (resv_huge_pages) by the value passed 2460e5bbc8a6SMike Kravetz * in unused_resv_pages. This corresponds to the prior adjustments made 2461e5bbc8a6SMike Kravetz * to the associated reservation map. 2462e5bbc8a6SMike Kravetz * 2) Free any unused surplus pages that may have been allocated to satisfy 2463e5bbc8a6SMike Kravetz * the reservation. As many as unused_resv_pages may be freed. 2464e4e574b7SAdam Litke */ 2465a5516438SAndi Kleen static void return_unused_surplus_pages(struct hstate *h, 2466a5516438SAndi Kleen unsigned long unused_resv_pages) 2467e4e574b7SAdam Litke { 2468e4e574b7SAdam Litke unsigned long nr_pages; 246910c6ec49SMike Kravetz struct page *page; 247010c6ec49SMike Kravetz LIST_HEAD(page_list); 247110c6ec49SMike Kravetz 24729487ca60SMike Kravetz lockdep_assert_held(&hugetlb_lock); 247310c6ec49SMike Kravetz /* Uncommit the reservation */ 247410c6ec49SMike Kravetz h->resv_huge_pages -= unused_resv_pages; 2475e4e574b7SAdam Litke 2476c0531714SNaoya Horiguchi if (hstate_is_gigantic(h) && !gigantic_page_runtime_supported()) 2477e5bbc8a6SMike Kravetz goto out; 2478aa888a74SAndi Kleen 2479e5bbc8a6SMike Kravetz /* 2480e5bbc8a6SMike Kravetz * Part (or even all) of the reservation could have been backed 2481e5bbc8a6SMike Kravetz * by pre-allocated pages. Only free surplus pages. 2482e5bbc8a6SMike Kravetz */ 2483a5516438SAndi Kleen nr_pages = min(unused_resv_pages, h->surplus_huge_pages); 2484e4e574b7SAdam Litke 2485685f3457SLee Schermerhorn /* 2486685f3457SLee Schermerhorn * We want to release as many surplus pages as possible, spread 24879b5e5d0fSLee Schermerhorn * evenly across all nodes with memory. Iterate across these nodes 24889b5e5d0fSLee Schermerhorn * until we can no longer free unreserved surplus pages. This occurs 24899b5e5d0fSLee Schermerhorn * when the nodes with surplus pages have no free pages. 249010c6ec49SMike Kravetz * remove_pool_huge_page() will balance the freed pages across the 24919b5e5d0fSLee Schermerhorn * on-line nodes with memory and will handle the hstate accounting. 2492685f3457SLee Schermerhorn */ 2493685f3457SLee Schermerhorn while (nr_pages--) { 249410c6ec49SMike Kravetz page = remove_pool_huge_page(h, &node_states[N_MEMORY], 1); 249510c6ec49SMike Kravetz if (!page) 2496e5bbc8a6SMike Kravetz goto out; 249710c6ec49SMike Kravetz 249810c6ec49SMike Kravetz list_add(&page->lru, &page_list); 2499e4e574b7SAdam Litke } 2500e5bbc8a6SMike Kravetz 2501e5bbc8a6SMike Kravetz out: 2502db71ef79SMike Kravetz spin_unlock_irq(&hugetlb_lock); 250310c6ec49SMike Kravetz update_and_free_pages_bulk(h, &page_list); 2504db71ef79SMike Kravetz spin_lock_irq(&hugetlb_lock); 2505e4e574b7SAdam Litke } 2506e4e574b7SAdam Litke 25075e911373SMike Kravetz 2508c37f9fb1SAndy Whitcroft /* 2509feba16e2SMike Kravetz * vma_needs_reservation, vma_commit_reservation and vma_end_reservation 25105e911373SMike Kravetz * are used by the huge page allocation routines to manage reservations. 2511cf3ad20bSMike Kravetz * 2512cf3ad20bSMike Kravetz * vma_needs_reservation is called to determine if the huge page at addr 2513cf3ad20bSMike Kravetz * within the vma has an associated reservation. If a reservation is 2514cf3ad20bSMike Kravetz * needed, the value 1 is returned. The caller is then responsible for 2515cf3ad20bSMike Kravetz * managing the global reservation and subpool usage counts. After 2516cf3ad20bSMike Kravetz * the huge page has been allocated, vma_commit_reservation is called 2517feba16e2SMike Kravetz * to add the page to the reservation map. If the page allocation fails, 2518feba16e2SMike Kravetz * the reservation must be ended instead of committed. vma_end_reservation 2519feba16e2SMike Kravetz * is called in such cases. 2520cf3ad20bSMike Kravetz * 2521cf3ad20bSMike Kravetz * In the normal case, vma_commit_reservation returns the same value 2522cf3ad20bSMike Kravetz * as the preceding vma_needs_reservation call. The only time this 2523cf3ad20bSMike Kravetz * is not the case is if a reserve map was changed between calls. It 2524cf3ad20bSMike Kravetz * is the responsibility of the caller to notice the difference and 2525cf3ad20bSMike Kravetz * take appropriate action. 252696b96a96SMike Kravetz * 252796b96a96SMike Kravetz * vma_add_reservation is used in error paths where a reservation must 252896b96a96SMike Kravetz * be restored when a newly allocated huge page must be freed. It is 252996b96a96SMike Kravetz * to be called after calling vma_needs_reservation to determine if a 253096b96a96SMike Kravetz * reservation exists. 2531846be085SMike Kravetz * 2532846be085SMike Kravetz * vma_del_reservation is used in error paths where an entry in the reserve 2533846be085SMike Kravetz * map was created during huge page allocation and must be removed. It is to 2534846be085SMike Kravetz * be called after calling vma_needs_reservation to determine if a reservation 2535846be085SMike Kravetz * exists. 2536c37f9fb1SAndy Whitcroft */ 25375e911373SMike Kravetz enum vma_resv_mode { 25385e911373SMike Kravetz VMA_NEEDS_RESV, 25395e911373SMike Kravetz VMA_COMMIT_RESV, 2540feba16e2SMike Kravetz VMA_END_RESV, 254196b96a96SMike Kravetz VMA_ADD_RESV, 2542846be085SMike Kravetz VMA_DEL_RESV, 25435e911373SMike Kravetz }; 2544cf3ad20bSMike Kravetz static long __vma_reservation_common(struct hstate *h, 2545cf3ad20bSMike Kravetz struct vm_area_struct *vma, unsigned long addr, 25465e911373SMike Kravetz enum vma_resv_mode mode) 2547c37f9fb1SAndy Whitcroft { 25484e35f483SJoonsoo Kim struct resv_map *resv; 25494e35f483SJoonsoo Kim pgoff_t idx; 2550cf3ad20bSMike Kravetz long ret; 25510db9d74eSMina Almasry long dummy_out_regions_needed; 2552c37f9fb1SAndy Whitcroft 25534e35f483SJoonsoo Kim resv = vma_resv_map(vma); 25544e35f483SJoonsoo Kim if (!resv) 2555c37f9fb1SAndy Whitcroft return 1; 2556c37f9fb1SAndy Whitcroft 25574e35f483SJoonsoo Kim idx = vma_hugecache_offset(h, vma, addr); 25585e911373SMike Kravetz switch (mode) { 25595e911373SMike Kravetz case VMA_NEEDS_RESV: 25600db9d74eSMina Almasry ret = region_chg(resv, idx, idx + 1, &dummy_out_regions_needed); 25610db9d74eSMina Almasry /* We assume that vma_reservation_* routines always operate on 25620db9d74eSMina Almasry * 1 page, and that adding to resv map a 1 page entry can only 25630db9d74eSMina Almasry * ever require 1 region. 25640db9d74eSMina Almasry */ 25650db9d74eSMina Almasry VM_BUG_ON(dummy_out_regions_needed != 1); 25665e911373SMike Kravetz break; 25675e911373SMike Kravetz case VMA_COMMIT_RESV: 2568075a61d0SMina Almasry ret = region_add(resv, idx, idx + 1, 1, NULL, NULL); 25690db9d74eSMina Almasry /* region_add calls of range 1 should never fail. */ 25700db9d74eSMina Almasry VM_BUG_ON(ret < 0); 25715e911373SMike Kravetz break; 2572feba16e2SMike Kravetz case VMA_END_RESV: 25730db9d74eSMina Almasry region_abort(resv, idx, idx + 1, 1); 25745e911373SMike Kravetz ret = 0; 25755e911373SMike Kravetz break; 257696b96a96SMike Kravetz case VMA_ADD_RESV: 25770db9d74eSMina Almasry if (vma->vm_flags & VM_MAYSHARE) { 2578075a61d0SMina Almasry ret = region_add(resv, idx, idx + 1, 1, NULL, NULL); 25790db9d74eSMina Almasry /* region_add calls of range 1 should never fail. */ 25800db9d74eSMina Almasry VM_BUG_ON(ret < 0); 25810db9d74eSMina Almasry } else { 25820db9d74eSMina Almasry region_abort(resv, idx, idx + 1, 1); 258396b96a96SMike Kravetz ret = region_del(resv, idx, idx + 1); 258496b96a96SMike Kravetz } 258596b96a96SMike Kravetz break; 2586846be085SMike Kravetz case VMA_DEL_RESV: 2587846be085SMike Kravetz if (vma->vm_flags & VM_MAYSHARE) { 2588846be085SMike Kravetz region_abort(resv, idx, idx + 1, 1); 2589846be085SMike Kravetz ret = region_del(resv, idx, idx + 1); 2590846be085SMike Kravetz } else { 2591846be085SMike Kravetz ret = region_add(resv, idx, idx + 1, 1, NULL, NULL); 2592846be085SMike Kravetz /* region_add calls of range 1 should never fail. */ 2593846be085SMike Kravetz VM_BUG_ON(ret < 0); 2594846be085SMike Kravetz } 2595846be085SMike Kravetz break; 25965e911373SMike Kravetz default: 25975e911373SMike Kravetz BUG(); 25985e911373SMike Kravetz } 259984afd99bSAndy Whitcroft 2600846be085SMike Kravetz if (vma->vm_flags & VM_MAYSHARE || mode == VMA_DEL_RESV) 2601cf3ad20bSMike Kravetz return ret; 260267961f9dSMike Kravetz /* 2603bf3d12b9SMiaohe Lin * We know private mapping must have HPAGE_RESV_OWNER set. 2604bf3d12b9SMiaohe Lin * 260567961f9dSMike Kravetz * In most cases, reserves always exist for private mappings. 260667961f9dSMike Kravetz * However, a file associated with mapping could have been 260767961f9dSMike Kravetz * hole punched or truncated after reserves were consumed. 260867961f9dSMike Kravetz * As subsequent fault on such a range will not use reserves. 260967961f9dSMike Kravetz * Subtle - The reserve map for private mappings has the 261067961f9dSMike Kravetz * opposite meaning than that of shared mappings. If NO 261167961f9dSMike Kravetz * entry is in the reserve map, it means a reservation exists. 261267961f9dSMike Kravetz * If an entry exists in the reserve map, it means the 261367961f9dSMike Kravetz * reservation has already been consumed. As a result, the 261467961f9dSMike Kravetz * return value of this routine is the opposite of the 261567961f9dSMike Kravetz * value returned from reserve map manipulation routines above. 261667961f9dSMike Kravetz */ 2617bf3d12b9SMiaohe Lin if (ret > 0) 261867961f9dSMike Kravetz return 0; 2619bf3d12b9SMiaohe Lin if (ret == 0) 262067961f9dSMike Kravetz return 1; 2621bf3d12b9SMiaohe Lin return ret; 262284afd99bSAndy Whitcroft } 2623cf3ad20bSMike Kravetz 2624cf3ad20bSMike Kravetz static long vma_needs_reservation(struct hstate *h, 2625a5516438SAndi Kleen struct vm_area_struct *vma, unsigned long addr) 2626c37f9fb1SAndy Whitcroft { 26275e911373SMike Kravetz return __vma_reservation_common(h, vma, addr, VMA_NEEDS_RESV); 2628cf3ad20bSMike Kravetz } 2629c37f9fb1SAndy Whitcroft 2630cf3ad20bSMike Kravetz static long vma_commit_reservation(struct hstate *h, 2631cf3ad20bSMike Kravetz struct vm_area_struct *vma, unsigned long addr) 2632cf3ad20bSMike Kravetz { 26335e911373SMike Kravetz return __vma_reservation_common(h, vma, addr, VMA_COMMIT_RESV); 26345e911373SMike Kravetz } 26355e911373SMike Kravetz 2636feba16e2SMike Kravetz static void vma_end_reservation(struct hstate *h, 26375e911373SMike Kravetz struct vm_area_struct *vma, unsigned long addr) 26385e911373SMike Kravetz { 2639feba16e2SMike Kravetz (void)__vma_reservation_common(h, vma, addr, VMA_END_RESV); 2640c37f9fb1SAndy Whitcroft } 2641c37f9fb1SAndy Whitcroft 264296b96a96SMike Kravetz static long vma_add_reservation(struct hstate *h, 264396b96a96SMike Kravetz struct vm_area_struct *vma, unsigned long addr) 264496b96a96SMike Kravetz { 264596b96a96SMike Kravetz return __vma_reservation_common(h, vma, addr, VMA_ADD_RESV); 264696b96a96SMike Kravetz } 264796b96a96SMike Kravetz 2648846be085SMike Kravetz static long vma_del_reservation(struct hstate *h, 2649846be085SMike Kravetz struct vm_area_struct *vma, unsigned long addr) 265096b96a96SMike Kravetz { 2651846be085SMike Kravetz return __vma_reservation_common(h, vma, addr, VMA_DEL_RESV); 2652846be085SMike Kravetz } 2653846be085SMike Kravetz 2654846be085SMike Kravetz /* 2655846be085SMike Kravetz * This routine is called to restore reservation information on error paths. 2656846be085SMike Kravetz * It should ONLY be called for pages allocated via alloc_huge_page(), and 2657846be085SMike Kravetz * the hugetlb mutex should remain held when calling this routine. 2658846be085SMike Kravetz * 2659846be085SMike Kravetz * It handles two specific cases: 2660846be085SMike Kravetz * 1) A reservation was in place and the page consumed the reservation. 2661846be085SMike Kravetz * HPageRestoreReserve is set in the page. 2662846be085SMike Kravetz * 2) No reservation was in place for the page, so HPageRestoreReserve is 2663846be085SMike Kravetz * not set. However, alloc_huge_page always updates the reserve map. 2664846be085SMike Kravetz * 2665846be085SMike Kravetz * In case 1, free_huge_page later in the error path will increment the 2666846be085SMike Kravetz * global reserve count. But, free_huge_page does not have enough context 2667846be085SMike Kravetz * to adjust the reservation map. This case deals primarily with private 2668846be085SMike Kravetz * mappings. Adjust the reserve map here to be consistent with global 2669846be085SMike Kravetz * reserve count adjustments to be made by free_huge_page. Make sure the 2670846be085SMike Kravetz * reserve map indicates there is a reservation present. 2671846be085SMike Kravetz * 2672846be085SMike Kravetz * In case 2, simply undo reserve map modifications done by alloc_huge_page. 2673846be085SMike Kravetz */ 2674846be085SMike Kravetz void restore_reserve_on_error(struct hstate *h, struct vm_area_struct *vma, 2675846be085SMike Kravetz unsigned long address, struct page *page) 2676846be085SMike Kravetz { 267796b96a96SMike Kravetz long rc = vma_needs_reservation(h, vma, address); 267896b96a96SMike Kravetz 2679846be085SMike Kravetz if (HPageRestoreReserve(page)) { 2680846be085SMike Kravetz if (unlikely(rc < 0)) 268196b96a96SMike Kravetz /* 268296b96a96SMike Kravetz * Rare out of memory condition in reserve map 2683d6995da3SMike Kravetz * manipulation. Clear HPageRestoreReserve so that 268496b96a96SMike Kravetz * global reserve count will not be incremented 268596b96a96SMike Kravetz * by free_huge_page. This will make it appear 268696b96a96SMike Kravetz * as though the reservation for this page was 268796b96a96SMike Kravetz * consumed. This may prevent the task from 268896b96a96SMike Kravetz * faulting in the page at a later time. This 268996b96a96SMike Kravetz * is better than inconsistent global huge page 269096b96a96SMike Kravetz * accounting of reserve counts. 269196b96a96SMike Kravetz */ 2692d6995da3SMike Kravetz ClearHPageRestoreReserve(page); 2693846be085SMike Kravetz else if (rc) 2694846be085SMike Kravetz (void)vma_add_reservation(h, vma, address); 2695846be085SMike Kravetz else 2696846be085SMike Kravetz vma_end_reservation(h, vma, address); 2697846be085SMike Kravetz } else { 2698846be085SMike Kravetz if (!rc) { 269996b96a96SMike Kravetz /* 2700846be085SMike Kravetz * This indicates there is an entry in the reserve map 2701c7b1850dSMike Kravetz * not added by alloc_huge_page. We know it was added 2702846be085SMike Kravetz * before the alloc_huge_page call, otherwise 2703846be085SMike Kravetz * HPageRestoreReserve would be set on the page. 2704846be085SMike Kravetz * Remove the entry so that a subsequent allocation 2705846be085SMike Kravetz * does not consume a reservation. 270696b96a96SMike Kravetz */ 2707846be085SMike Kravetz rc = vma_del_reservation(h, vma, address); 2708846be085SMike Kravetz if (rc < 0) 2709846be085SMike Kravetz /* 2710846be085SMike Kravetz * VERY rare out of memory condition. Since 2711846be085SMike Kravetz * we can not delete the entry, set 2712846be085SMike Kravetz * HPageRestoreReserve so that the reserve 2713846be085SMike Kravetz * count will be incremented when the page 2714846be085SMike Kravetz * is freed. This reserve will be consumed 2715846be085SMike Kravetz * on a subsequent allocation. 2716846be085SMike Kravetz */ 2717846be085SMike Kravetz SetHPageRestoreReserve(page); 2718846be085SMike Kravetz } else if (rc < 0) { 2719846be085SMike Kravetz /* 2720846be085SMike Kravetz * Rare out of memory condition from 2721846be085SMike Kravetz * vma_needs_reservation call. Memory allocation is 2722846be085SMike Kravetz * only attempted if a new entry is needed. Therefore, 2723846be085SMike Kravetz * this implies there is not an entry in the 2724846be085SMike Kravetz * reserve map. 2725846be085SMike Kravetz * 2726846be085SMike Kravetz * For shared mappings, no entry in the map indicates 2727846be085SMike Kravetz * no reservation. We are done. 2728846be085SMike Kravetz */ 2729846be085SMike Kravetz if (!(vma->vm_flags & VM_MAYSHARE)) 2730846be085SMike Kravetz /* 2731846be085SMike Kravetz * For private mappings, no entry indicates 2732846be085SMike Kravetz * a reservation is present. Since we can 2733846be085SMike Kravetz * not add an entry, set SetHPageRestoreReserve 2734846be085SMike Kravetz * on the page so reserve count will be 2735846be085SMike Kravetz * incremented when freed. This reserve will 2736846be085SMike Kravetz * be consumed on a subsequent allocation. 2737846be085SMike Kravetz */ 2738846be085SMike Kravetz SetHPageRestoreReserve(page); 273996b96a96SMike Kravetz } else 2740846be085SMike Kravetz /* 2741846be085SMike Kravetz * No reservation present, do nothing 2742846be085SMike Kravetz */ 274396b96a96SMike Kravetz vma_end_reservation(h, vma, address); 274496b96a96SMike Kravetz } 274596b96a96SMike Kravetz } 274696b96a96SMike Kravetz 2747369fa227SOscar Salvador /* 2748369fa227SOscar Salvador * alloc_and_dissolve_huge_page - Allocate a new page and dissolve the old one 2749369fa227SOscar Salvador * @h: struct hstate old page belongs to 2750369fa227SOscar Salvador * @old_page: Old page to dissolve 2751ae37c7ffSOscar Salvador * @list: List to isolate the page in case we need to 2752369fa227SOscar Salvador * Returns 0 on success, otherwise negated error. 2753369fa227SOscar Salvador */ 2754ae37c7ffSOscar Salvador static int alloc_and_dissolve_huge_page(struct hstate *h, struct page *old_page, 2755ae37c7ffSOscar Salvador struct list_head *list) 2756369fa227SOscar Salvador { 2757369fa227SOscar Salvador gfp_t gfp_mask = htlb_alloc_mask(h) | __GFP_THISNODE; 2758de656ed3SSidhartha Kumar struct folio *old_folio = page_folio(old_page); 2759de656ed3SSidhartha Kumar int nid = folio_nid(old_folio); 2760369fa227SOscar Salvador struct page *new_page; 2761de656ed3SSidhartha Kumar struct folio *new_folio; 2762369fa227SOscar Salvador int ret = 0; 2763369fa227SOscar Salvador 2764369fa227SOscar Salvador /* 2765369fa227SOscar Salvador * Before dissolving the page, we need to allocate a new one for the 2766f41f2ed4SMuchun Song * pool to remain stable. Here, we allocate the page and 'prep' it 2767f41f2ed4SMuchun Song * by doing everything but actually updating counters and adding to 2768f41f2ed4SMuchun Song * the pool. This simplifies and let us do most of the processing 2769f41f2ed4SMuchun Song * under the lock. 2770369fa227SOscar Salvador */ 2771369fa227SOscar Salvador new_page = alloc_buddy_huge_page(h, gfp_mask, nid, NULL, NULL); 2772369fa227SOscar Salvador if (!new_page) 2773369fa227SOscar Salvador return -ENOMEM; 2774de656ed3SSidhartha Kumar new_folio = page_folio(new_page); 2775de656ed3SSidhartha Kumar __prep_new_hugetlb_folio(h, new_folio); 2776369fa227SOscar Salvador 2777369fa227SOscar Salvador retry: 2778369fa227SOscar Salvador spin_lock_irq(&hugetlb_lock); 2779de656ed3SSidhartha Kumar if (!folio_test_hugetlb(old_folio)) { 2780369fa227SOscar Salvador /* 2781369fa227SOscar Salvador * Freed from under us. Drop new_page too. 2782369fa227SOscar Salvador */ 2783369fa227SOscar Salvador goto free_new; 2784de656ed3SSidhartha Kumar } else if (folio_ref_count(old_folio)) { 2785369fa227SOscar Salvador /* 2786ae37c7ffSOscar Salvador * Someone has grabbed the page, try to isolate it here. 2787ae37c7ffSOscar Salvador * Fail with -EBUSY if not possible. 2788369fa227SOscar Salvador */ 2789ae37c7ffSOscar Salvador spin_unlock_irq(&hugetlb_lock); 27907ce82f4cSMiaohe Lin ret = isolate_hugetlb(old_page, list); 2791ae37c7ffSOscar Salvador spin_lock_irq(&hugetlb_lock); 2792369fa227SOscar Salvador goto free_new; 2793de656ed3SSidhartha Kumar } else if (!folio_test_hugetlb_freed(old_folio)) { 2794369fa227SOscar Salvador /* 2795369fa227SOscar Salvador * Page's refcount is 0 but it has not been enqueued in the 2796369fa227SOscar Salvador * freelist yet. Race window is small, so we can succeed here if 2797369fa227SOscar Salvador * we retry. 2798369fa227SOscar Salvador */ 2799369fa227SOscar Salvador spin_unlock_irq(&hugetlb_lock); 2800369fa227SOscar Salvador cond_resched(); 2801369fa227SOscar Salvador goto retry; 2802369fa227SOscar Salvador } else { 2803369fa227SOscar Salvador /* 2804369fa227SOscar Salvador * Ok, old_page is still a genuine free hugepage. Remove it from 2805369fa227SOscar Salvador * the freelist and decrease the counters. These will be 2806369fa227SOscar Salvador * incremented again when calling __prep_account_new_huge_page() 2807369fa227SOscar Salvador * and enqueue_huge_page() for new_page. The counters will remain 2808369fa227SOscar Salvador * stable since this happens under the lock. 2809369fa227SOscar Salvador */ 2810cfd5082bSSidhartha Kumar remove_hugetlb_folio(h, old_folio, false); 2811369fa227SOscar Salvador 2812369fa227SOscar Salvador /* 2813b65a4edaSMike Kravetz * Ref count on new page is already zero as it was dropped 2814b65a4edaSMike Kravetz * earlier. It can be directly added to the pool free list. 2815369fa227SOscar Salvador */ 2816369fa227SOscar Salvador __prep_account_new_huge_page(h, nid); 2817369fa227SOscar Salvador enqueue_huge_page(h, new_page); 2818369fa227SOscar Salvador 2819369fa227SOscar Salvador /* 2820369fa227SOscar Salvador * Pages have been replaced, we can safely free the old one. 2821369fa227SOscar Salvador */ 2822369fa227SOscar Salvador spin_unlock_irq(&hugetlb_lock); 2823*d6ef19e2SSidhartha Kumar update_and_free_hugetlb_folio(h, old_folio, false); 2824369fa227SOscar Salvador } 2825369fa227SOscar Salvador 2826369fa227SOscar Salvador return ret; 2827369fa227SOscar Salvador 2828369fa227SOscar Salvador free_new: 2829369fa227SOscar Salvador spin_unlock_irq(&hugetlb_lock); 2830b65a4edaSMike Kravetz /* Page has a zero ref count, but needs a ref to be freed */ 2831de656ed3SSidhartha Kumar folio_ref_unfreeze(new_folio, 1); 2832*d6ef19e2SSidhartha Kumar update_and_free_hugetlb_folio(h, new_folio, false); 2833369fa227SOscar Salvador 2834369fa227SOscar Salvador return ret; 2835369fa227SOscar Salvador } 2836369fa227SOscar Salvador 2837ae37c7ffSOscar Salvador int isolate_or_dissolve_huge_page(struct page *page, struct list_head *list) 2838369fa227SOscar Salvador { 2839369fa227SOscar Salvador struct hstate *h; 2840d5e33bd8SSidhartha Kumar struct folio *folio = page_folio(page); 2841ae37c7ffSOscar Salvador int ret = -EBUSY; 2842369fa227SOscar Salvador 2843369fa227SOscar Salvador /* 2844369fa227SOscar Salvador * The page might have been dissolved from under our feet, so make sure 2845369fa227SOscar Salvador * to carefully check the state under the lock. 2846369fa227SOscar Salvador * Return success when racing as if we dissolved the page ourselves. 2847369fa227SOscar Salvador */ 2848369fa227SOscar Salvador spin_lock_irq(&hugetlb_lock); 2849d5e33bd8SSidhartha Kumar if (folio_test_hugetlb(folio)) { 2850d5e33bd8SSidhartha Kumar h = folio_hstate(folio); 2851369fa227SOscar Salvador } else { 2852369fa227SOscar Salvador spin_unlock_irq(&hugetlb_lock); 2853369fa227SOscar Salvador return 0; 2854369fa227SOscar Salvador } 2855369fa227SOscar Salvador spin_unlock_irq(&hugetlb_lock); 2856369fa227SOscar Salvador 2857369fa227SOscar Salvador /* 2858369fa227SOscar Salvador * Fence off gigantic pages as there is a cyclic dependency between 2859369fa227SOscar Salvador * alloc_contig_range and them. Return -ENOMEM as this has the effect 2860369fa227SOscar Salvador * of bailing out right away without further retrying. 2861369fa227SOscar Salvador */ 2862369fa227SOscar Salvador if (hstate_is_gigantic(h)) 2863369fa227SOscar Salvador return -ENOMEM; 2864369fa227SOscar Salvador 2865d5e33bd8SSidhartha Kumar if (folio_ref_count(folio) && !isolate_hugetlb(&folio->page, list)) 2866ae37c7ffSOscar Salvador ret = 0; 2867d5e33bd8SSidhartha Kumar else if (!folio_ref_count(folio)) 2868d5e33bd8SSidhartha Kumar ret = alloc_and_dissolve_huge_page(h, &folio->page, list); 2869ae37c7ffSOscar Salvador 2870ae37c7ffSOscar Salvador return ret; 2871369fa227SOscar Salvador } 2872369fa227SOscar Salvador 287370c3547eSMike Kravetz struct page *alloc_huge_page(struct vm_area_struct *vma, 287404f2cbe3SMel Gorman unsigned long addr, int avoid_reserve) 2875348ea204SAdam Litke { 287690481622SDavid Gibson struct hugepage_subpool *spool = subpool_vma(vma); 2877a5516438SAndi Kleen struct hstate *h = hstate_vma(vma); 2878348ea204SAdam Litke struct page *page; 2879d4ab0316SSidhartha Kumar struct folio *folio; 2880d85f69b0SMike Kravetz long map_chg, map_commit; 2881d85f69b0SMike Kravetz long gbl_chg; 28826d76dcf4SAneesh Kumar K.V int ret, idx; 28836d76dcf4SAneesh Kumar K.V struct hugetlb_cgroup *h_cg; 288408cf9fafSMina Almasry bool deferred_reserve; 28852fc39cecSAdam Litke 28866d76dcf4SAneesh Kumar K.V idx = hstate_index(h); 2887a1e78772SMel Gorman /* 2888d85f69b0SMike Kravetz * Examine the region/reserve map to determine if the process 2889d85f69b0SMike Kravetz * has a reservation for the page to be allocated. A return 2890d85f69b0SMike Kravetz * code of zero indicates a reservation exists (no change). 2891a1e78772SMel Gorman */ 2892d85f69b0SMike Kravetz map_chg = gbl_chg = vma_needs_reservation(h, vma, addr); 2893d85f69b0SMike Kravetz if (map_chg < 0) 289476dcee75SAneesh Kumar K.V return ERR_PTR(-ENOMEM); 2895d85f69b0SMike Kravetz 2896d85f69b0SMike Kravetz /* 2897d85f69b0SMike Kravetz * Processes that did not create the mapping will have no 2898d85f69b0SMike Kravetz * reserves as indicated by the region/reserve map. Check 2899d85f69b0SMike Kravetz * that the allocation will not exceed the subpool limit. 2900d85f69b0SMike Kravetz * Allocations for MAP_NORESERVE mappings also need to be 2901d85f69b0SMike Kravetz * checked against any subpool limit. 2902d85f69b0SMike Kravetz */ 2903d85f69b0SMike Kravetz if (map_chg || avoid_reserve) { 2904d85f69b0SMike Kravetz gbl_chg = hugepage_subpool_get_pages(spool, 1); 2905d85f69b0SMike Kravetz if (gbl_chg < 0) { 2906feba16e2SMike Kravetz vma_end_reservation(h, vma, addr); 290776dcee75SAneesh Kumar K.V return ERR_PTR(-ENOSPC); 29085e911373SMike Kravetz } 290990d8b7e6SAdam Litke 2910d85f69b0SMike Kravetz /* 2911d85f69b0SMike Kravetz * Even though there was no reservation in the region/reserve 2912d85f69b0SMike Kravetz * map, there could be reservations associated with the 2913d85f69b0SMike Kravetz * subpool that can be used. This would be indicated if the 2914d85f69b0SMike Kravetz * return value of hugepage_subpool_get_pages() is zero. 2915d85f69b0SMike Kravetz * However, if avoid_reserve is specified we still avoid even 2916d85f69b0SMike Kravetz * the subpool reservations. 2917d85f69b0SMike Kravetz */ 2918d85f69b0SMike Kravetz if (avoid_reserve) 2919d85f69b0SMike Kravetz gbl_chg = 1; 2920d85f69b0SMike Kravetz } 2921d85f69b0SMike Kravetz 292208cf9fafSMina Almasry /* If this allocation is not consuming a reservation, charge it now. 292308cf9fafSMina Almasry */ 29246501fe5fSMiaohe Lin deferred_reserve = map_chg || avoid_reserve; 292508cf9fafSMina Almasry if (deferred_reserve) { 292608cf9fafSMina Almasry ret = hugetlb_cgroup_charge_cgroup_rsvd( 292708cf9fafSMina Almasry idx, pages_per_huge_page(h), &h_cg); 29288f34af6fSJianyu Zhan if (ret) 29298f34af6fSJianyu Zhan goto out_subpool_put; 293008cf9fafSMina Almasry } 293108cf9fafSMina Almasry 293208cf9fafSMina Almasry ret = hugetlb_cgroup_charge_cgroup(idx, pages_per_huge_page(h), &h_cg); 293308cf9fafSMina Almasry if (ret) 293408cf9fafSMina Almasry goto out_uncharge_cgroup_reservation; 29358f34af6fSJianyu Zhan 2936db71ef79SMike Kravetz spin_lock_irq(&hugetlb_lock); 2937d85f69b0SMike Kravetz /* 2938d85f69b0SMike Kravetz * glb_chg is passed to indicate whether or not a page must be taken 2939d85f69b0SMike Kravetz * from the global free pool (global change). gbl_chg == 0 indicates 2940d85f69b0SMike Kravetz * a reservation exists for the allocation. 2941d85f69b0SMike Kravetz */ 2942d85f69b0SMike Kravetz page = dequeue_huge_page_vma(h, vma, addr, avoid_reserve, gbl_chg); 294381a6fcaeSJoonsoo Kim if (!page) { 2944db71ef79SMike Kravetz spin_unlock_irq(&hugetlb_lock); 29450c397daeSMichal Hocko page = alloc_buddy_huge_page_with_mpol(h, vma, addr); 29468f34af6fSJianyu Zhan if (!page) 29478f34af6fSJianyu Zhan goto out_uncharge_cgroup; 294812df140fSRik van Riel spin_lock_irq(&hugetlb_lock); 2949a88c7695SNaoya Horiguchi if (!avoid_reserve && vma_has_reserves(vma, gbl_chg)) { 2950d6995da3SMike Kravetz SetHPageRestoreReserve(page); 2951a88c7695SNaoya Horiguchi h->resv_huge_pages--; 2952a88c7695SNaoya Horiguchi } 295315a8d68eSWei Yang list_add(&page->lru, &h->hugepage_activelist); 29542b21624fSMike Kravetz set_page_refcounted(page); 295581a6fcaeSJoonsoo Kim /* Fall through */ 2956a1e78772SMel Gorman } 2957d4ab0316SSidhartha Kumar folio = page_folio(page); 295881a6fcaeSJoonsoo Kim hugetlb_cgroup_commit_charge(idx, pages_per_huge_page(h), h_cg, page); 295908cf9fafSMina Almasry /* If allocation is not consuming a reservation, also store the 296008cf9fafSMina Almasry * hugetlb_cgroup pointer on the page. 296108cf9fafSMina Almasry */ 296208cf9fafSMina Almasry if (deferred_reserve) { 296308cf9fafSMina Almasry hugetlb_cgroup_commit_charge_rsvd(idx, pages_per_huge_page(h), 296408cf9fafSMina Almasry h_cg, page); 296508cf9fafSMina Almasry } 296608cf9fafSMina Almasry 2967db71ef79SMike Kravetz spin_unlock_irq(&hugetlb_lock); 2968a1e78772SMel Gorman 2969d6995da3SMike Kravetz hugetlb_set_page_subpool(page, spool); 2970a1e78772SMel Gorman 2971d85f69b0SMike Kravetz map_commit = vma_commit_reservation(h, vma, addr); 2972d85f69b0SMike Kravetz if (unlikely(map_chg > map_commit)) { 297333039678SMike Kravetz /* 297433039678SMike Kravetz * The page was added to the reservation map between 297533039678SMike Kravetz * vma_needs_reservation and vma_commit_reservation. 297633039678SMike Kravetz * This indicates a race with hugetlb_reserve_pages. 297733039678SMike Kravetz * Adjust for the subpool count incremented above AND 297833039678SMike Kravetz * in hugetlb_reserve_pages for the same page. Also, 297933039678SMike Kravetz * the reservation count added in hugetlb_reserve_pages 298033039678SMike Kravetz * no longer applies. 298133039678SMike Kravetz */ 298233039678SMike Kravetz long rsv_adjust; 298333039678SMike Kravetz 298433039678SMike Kravetz rsv_adjust = hugepage_subpool_put_pages(spool, 1); 298533039678SMike Kravetz hugetlb_acct_memory(h, -rsv_adjust); 298679aa925bSMike Kravetz if (deferred_reserve) 2987d4ab0316SSidhartha Kumar hugetlb_cgroup_uncharge_folio_rsvd(hstate_index(h), 2988d4ab0316SSidhartha Kumar pages_per_huge_page(h), folio); 298933039678SMike Kravetz } 29907893d1d5SAdam Litke return page; 29918f34af6fSJianyu Zhan 29928f34af6fSJianyu Zhan out_uncharge_cgroup: 29938f34af6fSJianyu Zhan hugetlb_cgroup_uncharge_cgroup(idx, pages_per_huge_page(h), h_cg); 299408cf9fafSMina Almasry out_uncharge_cgroup_reservation: 299508cf9fafSMina Almasry if (deferred_reserve) 299608cf9fafSMina Almasry hugetlb_cgroup_uncharge_cgroup_rsvd(idx, pages_per_huge_page(h), 299708cf9fafSMina Almasry h_cg); 29988f34af6fSJianyu Zhan out_subpool_put: 2999d85f69b0SMike Kravetz if (map_chg || avoid_reserve) 30008f34af6fSJianyu Zhan hugepage_subpool_put_pages(spool, 1); 3001feba16e2SMike Kravetz vma_end_reservation(h, vma, addr); 30028f34af6fSJianyu Zhan return ERR_PTR(-ENOSPC); 3003b45b5bd6SDavid Gibson } 3004b45b5bd6SDavid Gibson 3005b5389086SZhenguo Yao int alloc_bootmem_huge_page(struct hstate *h, int nid) 3006e24a1307SAneesh Kumar K.V __attribute__ ((weak, alias("__alloc_bootmem_huge_page"))); 3007b5389086SZhenguo Yao int __alloc_bootmem_huge_page(struct hstate *h, int nid) 3008aa888a74SAndi Kleen { 3009b5389086SZhenguo Yao struct huge_bootmem_page *m = NULL; /* initialize for clang */ 3010b2261026SJoonsoo Kim int nr_nodes, node; 3011aa888a74SAndi Kleen 3012b5389086SZhenguo Yao /* do node specific alloc */ 3013b5389086SZhenguo Yao if (nid != NUMA_NO_NODE) { 3014b5389086SZhenguo Yao m = memblock_alloc_try_nid_raw(huge_page_size(h), huge_page_size(h), 3015b5389086SZhenguo Yao 0, MEMBLOCK_ALLOC_ACCESSIBLE, nid); 3016b5389086SZhenguo Yao if (!m) 3017b5389086SZhenguo Yao return 0; 3018b5389086SZhenguo Yao goto found; 3019b5389086SZhenguo Yao } 3020b5389086SZhenguo Yao /* allocate from next node when distributing huge pages */ 3021b2261026SJoonsoo Kim for_each_node_mask_to_alloc(h, nr_nodes, node, &node_states[N_MEMORY]) { 3022b5389086SZhenguo Yao m = memblock_alloc_try_nid_raw( 30238b89a116SGrygorii Strashko huge_page_size(h), huge_page_size(h), 302497ad1087SMike Rapoport 0, MEMBLOCK_ALLOC_ACCESSIBLE, node); 3025aa888a74SAndi Kleen /* 3026aa888a74SAndi Kleen * Use the beginning of the huge page to store the 3027aa888a74SAndi Kleen * huge_bootmem_page struct (until gather_bootmem 3028aa888a74SAndi Kleen * puts them into the mem_map). 3029aa888a74SAndi Kleen */ 3030b5389086SZhenguo Yao if (!m) 3031b5389086SZhenguo Yao return 0; 3032aa888a74SAndi Kleen goto found; 3033aa888a74SAndi Kleen } 3034aa888a74SAndi Kleen 3035aa888a74SAndi Kleen found: 3036aa888a74SAndi Kleen /* Put them into a private list first because mem_map is not up yet */ 3037330d6e48SCannon Matthews INIT_LIST_HEAD(&m->list); 3038aa888a74SAndi Kleen list_add(&m->list, &huge_boot_pages); 3039aa888a74SAndi Kleen m->hstate = h; 3040aa888a74SAndi Kleen return 1; 3041aa888a74SAndi Kleen } 3042aa888a74SAndi Kleen 304348b8d744SMike Kravetz /* 304448b8d744SMike Kravetz * Put bootmem huge pages into the standard lists after mem_map is up. 304548b8d744SMike Kravetz * Note: This only applies to gigantic (order > MAX_ORDER) pages. 304648b8d744SMike Kravetz */ 3047aa888a74SAndi Kleen static void __init gather_bootmem_prealloc(void) 3048aa888a74SAndi Kleen { 3049aa888a74SAndi Kleen struct huge_bootmem_page *m; 3050aa888a74SAndi Kleen 3051aa888a74SAndi Kleen list_for_each_entry(m, &huge_boot_pages, list) { 305240d18ebfSMike Kravetz struct page *page = virt_to_page(m); 3053aa888a74SAndi Kleen struct hstate *h = m->hstate; 3054ee8f248dSBecky Bruce 305548b8d744SMike Kravetz VM_BUG_ON(!hstate_is_gigantic(h)); 3056aa888a74SAndi Kleen WARN_ON(page_count(page) != 1); 30577118fc29SMike Kravetz if (prep_compound_gigantic_page(page, huge_page_order(h))) { 3058ef5a22beSAndrea Arcangeli WARN_ON(PageReserved(page)); 3059aa888a74SAndi Kleen prep_new_huge_page(h, page, page_to_nid(page)); 30602b21624fSMike Kravetz free_huge_page(page); /* add to the hugepage allocator */ 30617118fc29SMike Kravetz } else { 3062416d85edSMike Kravetz /* VERY unlikely inflated ref count on a tail page */ 30637118fc29SMike Kravetz free_gigantic_page(page, huge_page_order(h)); 30647118fc29SMike Kravetz } 3065af0fb9dfSMichal Hocko 3066b0320c7bSRafael Aquini /* 306748b8d744SMike Kravetz * We need to restore the 'stolen' pages to totalram_pages 306848b8d744SMike Kravetz * in order to fix confusing memory reports from free(1) and 306948b8d744SMike Kravetz * other side-effects, like CommitLimit going negative. 3070b0320c7bSRafael Aquini */ 3071c78a7f36SMiaohe Lin adjust_managed_page_count(page, pages_per_huge_page(h)); 3072520495feSCannon Matthews cond_resched(); 3073aa888a74SAndi Kleen } 3074aa888a74SAndi Kleen } 3075b5389086SZhenguo Yao static void __init hugetlb_hstate_alloc_pages_onenode(struct hstate *h, int nid) 3076b5389086SZhenguo Yao { 3077b5389086SZhenguo Yao unsigned long i; 3078b5389086SZhenguo Yao char buf[32]; 3079b5389086SZhenguo Yao 3080b5389086SZhenguo Yao for (i = 0; i < h->max_huge_pages_node[nid]; ++i) { 3081b5389086SZhenguo Yao if (hstate_is_gigantic(h)) { 3082b5389086SZhenguo Yao if (!alloc_bootmem_huge_page(h, nid)) 3083b5389086SZhenguo Yao break; 3084b5389086SZhenguo Yao } else { 3085b5389086SZhenguo Yao struct page *page; 3086b5389086SZhenguo Yao gfp_t gfp_mask = htlb_alloc_mask(h) | __GFP_THISNODE; 3087b5389086SZhenguo Yao 3088b5389086SZhenguo Yao page = alloc_fresh_huge_page(h, gfp_mask, nid, 3089b5389086SZhenguo Yao &node_states[N_MEMORY], NULL); 3090b5389086SZhenguo Yao if (!page) 3091b5389086SZhenguo Yao break; 30922b21624fSMike Kravetz free_huge_page(page); /* free it into the hugepage allocator */ 3093b5389086SZhenguo Yao } 3094b5389086SZhenguo Yao cond_resched(); 3095b5389086SZhenguo Yao } 3096b5389086SZhenguo Yao if (i == h->max_huge_pages_node[nid]) 3097b5389086SZhenguo Yao return; 3098b5389086SZhenguo Yao 3099b5389086SZhenguo Yao string_get_size(huge_page_size(h), 1, STRING_UNITS_2, buf, 32); 3100b5389086SZhenguo Yao pr_warn("HugeTLB: allocating %u of page size %s failed node%d. Only allocated %lu hugepages.\n", 3101b5389086SZhenguo Yao h->max_huge_pages_node[nid], buf, nid, i); 3102b5389086SZhenguo Yao h->max_huge_pages -= (h->max_huge_pages_node[nid] - i); 3103b5389086SZhenguo Yao h->max_huge_pages_node[nid] = i; 3104b5389086SZhenguo Yao } 3105aa888a74SAndi Kleen 31068faa8b07SAndi Kleen static void __init hugetlb_hstate_alloc_pages(struct hstate *h) 31071da177e4SLinus Torvalds { 31081da177e4SLinus Torvalds unsigned long i; 3109f60858f9SMike Kravetz nodemask_t *node_alloc_noretry; 3110b5389086SZhenguo Yao bool node_specific_alloc = false; 3111f60858f9SMike Kravetz 3112b5389086SZhenguo Yao /* skip gigantic hugepages allocation if hugetlb_cma enabled */ 3113b5389086SZhenguo Yao if (hstate_is_gigantic(h) && hugetlb_cma_size) { 3114b5389086SZhenguo Yao pr_warn_once("HugeTLB: hugetlb_cma is enabled, skip boot time allocation\n"); 3115b5389086SZhenguo Yao return; 3116b5389086SZhenguo Yao } 3117b5389086SZhenguo Yao 3118b5389086SZhenguo Yao /* do node specific alloc */ 31190a7a0f6fSPeng Liu for_each_online_node(i) { 3120b5389086SZhenguo Yao if (h->max_huge_pages_node[i] > 0) { 3121b5389086SZhenguo Yao hugetlb_hstate_alloc_pages_onenode(h, i); 3122b5389086SZhenguo Yao node_specific_alloc = true; 3123b5389086SZhenguo Yao } 3124b5389086SZhenguo Yao } 3125b5389086SZhenguo Yao 3126b5389086SZhenguo Yao if (node_specific_alloc) 3127b5389086SZhenguo Yao return; 3128b5389086SZhenguo Yao 3129b5389086SZhenguo Yao /* below will do all node balanced alloc */ 3130f60858f9SMike Kravetz if (!hstate_is_gigantic(h)) { 3131f60858f9SMike Kravetz /* 3132f60858f9SMike Kravetz * Bit mask controlling how hard we retry per-node allocations. 3133f60858f9SMike Kravetz * Ignore errors as lower level routines can deal with 3134f60858f9SMike Kravetz * node_alloc_noretry == NULL. If this kmalloc fails at boot 3135f60858f9SMike Kravetz * time, we are likely in bigger trouble. 3136f60858f9SMike Kravetz */ 3137f60858f9SMike Kravetz node_alloc_noretry = kmalloc(sizeof(*node_alloc_noretry), 3138f60858f9SMike Kravetz GFP_KERNEL); 3139f60858f9SMike Kravetz } else { 3140f60858f9SMike Kravetz /* allocations done at boot time */ 3141f60858f9SMike Kravetz node_alloc_noretry = NULL; 3142f60858f9SMike Kravetz } 3143f60858f9SMike Kravetz 3144f60858f9SMike Kravetz /* bit mask controlling how hard we retry per-node allocations */ 3145f60858f9SMike Kravetz if (node_alloc_noretry) 3146f60858f9SMike Kravetz nodes_clear(*node_alloc_noretry); 31471da177e4SLinus Torvalds 3148e5ff2159SAndi Kleen for (i = 0; i < h->max_huge_pages; ++i) { 3149bae7f4aeSLuiz Capitulino if (hstate_is_gigantic(h)) { 3150b5389086SZhenguo Yao if (!alloc_bootmem_huge_page(h, NUMA_NO_NODE)) 3151aa888a74SAndi Kleen break; 31520c397daeSMichal Hocko } else if (!alloc_pool_huge_page(h, 3153f60858f9SMike Kravetz &node_states[N_MEMORY], 3154f60858f9SMike Kravetz node_alloc_noretry)) 31551da177e4SLinus Torvalds break; 315669ed779aSDavid Rientjes cond_resched(); 31571da177e4SLinus Torvalds } 3158d715cf80SLiam R. Howlett if (i < h->max_huge_pages) { 3159d715cf80SLiam R. Howlett char buf[32]; 3160d715cf80SLiam R. Howlett 3161c6247f72SMatthew Wilcox string_get_size(huge_page_size(h), 1, STRING_UNITS_2, buf, 32); 3162d715cf80SLiam R. Howlett pr_warn("HugeTLB: allocating %lu of page size %s failed. Only allocated %lu hugepages.\n", 3163d715cf80SLiam R. Howlett h->max_huge_pages, buf, i); 31648faa8b07SAndi Kleen h->max_huge_pages = i; 3165e5ff2159SAndi Kleen } 3166f60858f9SMike Kravetz kfree(node_alloc_noretry); 3167d715cf80SLiam R. Howlett } 3168e5ff2159SAndi Kleen 3169e5ff2159SAndi Kleen static void __init hugetlb_init_hstates(void) 3170e5ff2159SAndi Kleen { 317179dfc695SMike Kravetz struct hstate *h, *h2; 3172e5ff2159SAndi Kleen 3173e5ff2159SAndi Kleen for_each_hstate(h) { 31748faa8b07SAndi Kleen /* oversize hugepages were init'ed in early boot */ 3175bae7f4aeSLuiz Capitulino if (!hstate_is_gigantic(h)) 31768faa8b07SAndi Kleen hugetlb_hstate_alloc_pages(h); 317779dfc695SMike Kravetz 317879dfc695SMike Kravetz /* 317979dfc695SMike Kravetz * Set demote order for each hstate. Note that 318079dfc695SMike Kravetz * h->demote_order is initially 0. 318179dfc695SMike Kravetz * - We can not demote gigantic pages if runtime freeing 318279dfc695SMike Kravetz * is not supported, so skip this. 3183a01f4390SMike Kravetz * - If CMA allocation is possible, we can not demote 3184a01f4390SMike Kravetz * HUGETLB_PAGE_ORDER or smaller size pages. 318579dfc695SMike Kravetz */ 318679dfc695SMike Kravetz if (hstate_is_gigantic(h) && !gigantic_page_runtime_supported()) 318779dfc695SMike Kravetz continue; 3188a01f4390SMike Kravetz if (hugetlb_cma_size && h->order <= HUGETLB_PAGE_ORDER) 3189a01f4390SMike Kravetz continue; 319079dfc695SMike Kravetz for_each_hstate(h2) { 319179dfc695SMike Kravetz if (h2 == h) 319279dfc695SMike Kravetz continue; 319379dfc695SMike Kravetz if (h2->order < h->order && 319479dfc695SMike Kravetz h2->order > h->demote_order) 319579dfc695SMike Kravetz h->demote_order = h2->order; 319679dfc695SMike Kravetz } 3197e5ff2159SAndi Kleen } 3198e5ff2159SAndi Kleen } 3199e5ff2159SAndi Kleen 3200e5ff2159SAndi Kleen static void __init report_hugepages(void) 3201e5ff2159SAndi Kleen { 3202e5ff2159SAndi Kleen struct hstate *h; 3203e5ff2159SAndi Kleen 3204e5ff2159SAndi Kleen for_each_hstate(h) { 32054abd32dbSAndi Kleen char buf[32]; 3206c6247f72SMatthew Wilcox 3207c6247f72SMatthew Wilcox string_get_size(huge_page_size(h), 1, STRING_UNITS_2, buf, 32); 32086213834cSMuchun Song pr_info("HugeTLB: registered %s page size, pre-allocated %ld pages\n", 3209c6247f72SMatthew Wilcox buf, h->free_huge_pages); 32106213834cSMuchun Song pr_info("HugeTLB: %d KiB vmemmap can be freed for a %s page\n", 32116213834cSMuchun Song hugetlb_vmemmap_optimizable_size(h) / SZ_1K, buf); 3212e5ff2159SAndi Kleen } 3213e5ff2159SAndi Kleen } 3214e5ff2159SAndi Kleen 32151da177e4SLinus Torvalds #ifdef CONFIG_HIGHMEM 32166ae11b27SLee Schermerhorn static void try_to_free_low(struct hstate *h, unsigned long count, 32176ae11b27SLee Schermerhorn nodemask_t *nodes_allowed) 32181da177e4SLinus Torvalds { 32194415cc8dSChristoph Lameter int i; 32201121828aSMike Kravetz LIST_HEAD(page_list); 32214415cc8dSChristoph Lameter 32229487ca60SMike Kravetz lockdep_assert_held(&hugetlb_lock); 3223bae7f4aeSLuiz Capitulino if (hstate_is_gigantic(h)) 3224aa888a74SAndi Kleen return; 3225aa888a74SAndi Kleen 32261121828aSMike Kravetz /* 32271121828aSMike Kravetz * Collect pages to be freed on a list, and free after dropping lock 32281121828aSMike Kravetz */ 32296ae11b27SLee Schermerhorn for_each_node_mask(i, *nodes_allowed) { 323010c6ec49SMike Kravetz struct page *page, *next; 3231a5516438SAndi Kleen struct list_head *freel = &h->hugepage_freelists[i]; 3232a5516438SAndi Kleen list_for_each_entry_safe(page, next, freel, lru) { 3233a5516438SAndi Kleen if (count >= h->nr_huge_pages) 32341121828aSMike Kravetz goto out; 32351da177e4SLinus Torvalds if (PageHighMem(page)) 32361da177e4SLinus Torvalds continue; 3237cfd5082bSSidhartha Kumar remove_hugetlb_folio(h, page_folio(page), false); 32381121828aSMike Kravetz list_add(&page->lru, &page_list); 32391121828aSMike Kravetz } 32401121828aSMike Kravetz } 32411121828aSMike Kravetz 32421121828aSMike Kravetz out: 3243db71ef79SMike Kravetz spin_unlock_irq(&hugetlb_lock); 324410c6ec49SMike Kravetz update_and_free_pages_bulk(h, &page_list); 3245db71ef79SMike Kravetz spin_lock_irq(&hugetlb_lock); 32461da177e4SLinus Torvalds } 32471da177e4SLinus Torvalds #else 32486ae11b27SLee Schermerhorn static inline void try_to_free_low(struct hstate *h, unsigned long count, 32496ae11b27SLee Schermerhorn nodemask_t *nodes_allowed) 32501da177e4SLinus Torvalds { 32511da177e4SLinus Torvalds } 32521da177e4SLinus Torvalds #endif 32531da177e4SLinus Torvalds 325420a0307cSWu Fengguang /* 325520a0307cSWu Fengguang * Increment or decrement surplus_huge_pages. Keep node-specific counters 325620a0307cSWu Fengguang * balanced by operating on them in a round-robin fashion. 325720a0307cSWu Fengguang * Returns 1 if an adjustment was made. 325820a0307cSWu Fengguang */ 32596ae11b27SLee Schermerhorn static int adjust_pool_surplus(struct hstate *h, nodemask_t *nodes_allowed, 32606ae11b27SLee Schermerhorn int delta) 326120a0307cSWu Fengguang { 3262b2261026SJoonsoo Kim int nr_nodes, node; 326320a0307cSWu Fengguang 32649487ca60SMike Kravetz lockdep_assert_held(&hugetlb_lock); 326520a0307cSWu Fengguang VM_BUG_ON(delta != -1 && delta != 1); 326620a0307cSWu Fengguang 3267e8c5c824SLee Schermerhorn if (delta < 0) { 3268b2261026SJoonsoo Kim for_each_node_mask_to_alloc(h, nr_nodes, node, nodes_allowed) { 3269b2261026SJoonsoo Kim if (h->surplus_huge_pages_node[node]) 3270b2261026SJoonsoo Kim goto found; 3271b2261026SJoonsoo Kim } 3272b2261026SJoonsoo Kim } else { 3273b2261026SJoonsoo Kim for_each_node_mask_to_free(h, nr_nodes, node, nodes_allowed) { 3274b2261026SJoonsoo Kim if (h->surplus_huge_pages_node[node] < 3275b2261026SJoonsoo Kim h->nr_huge_pages_node[node]) 3276b2261026SJoonsoo Kim goto found; 3277e8c5c824SLee Schermerhorn } 32789a76db09SLee Schermerhorn } 3279b2261026SJoonsoo Kim return 0; 328020a0307cSWu Fengguang 3281b2261026SJoonsoo Kim found: 328220a0307cSWu Fengguang h->surplus_huge_pages += delta; 3283b2261026SJoonsoo Kim h->surplus_huge_pages_node[node] += delta; 3284b2261026SJoonsoo Kim return 1; 328520a0307cSWu Fengguang } 328620a0307cSWu Fengguang 3287a5516438SAndi Kleen #define persistent_huge_pages(h) (h->nr_huge_pages - h->surplus_huge_pages) 3288fd875dcaSMike Kravetz static int set_max_huge_pages(struct hstate *h, unsigned long count, int nid, 32896ae11b27SLee Schermerhorn nodemask_t *nodes_allowed) 32901da177e4SLinus Torvalds { 32917893d1d5SAdam Litke unsigned long min_count, ret; 329210c6ec49SMike Kravetz struct page *page; 329310c6ec49SMike Kravetz LIST_HEAD(page_list); 3294f60858f9SMike Kravetz NODEMASK_ALLOC(nodemask_t, node_alloc_noretry, GFP_KERNEL); 3295f60858f9SMike Kravetz 3296f60858f9SMike Kravetz /* 3297f60858f9SMike Kravetz * Bit mask controlling how hard we retry per-node allocations. 3298f60858f9SMike Kravetz * If we can not allocate the bit mask, do not attempt to allocate 3299f60858f9SMike Kravetz * the requested huge pages. 3300f60858f9SMike Kravetz */ 3301f60858f9SMike Kravetz if (node_alloc_noretry) 3302f60858f9SMike Kravetz nodes_clear(*node_alloc_noretry); 3303f60858f9SMike Kravetz else 3304f60858f9SMike Kravetz return -ENOMEM; 33051da177e4SLinus Torvalds 330629383967SMike Kravetz /* 330729383967SMike Kravetz * resize_lock mutex prevents concurrent adjustments to number of 330829383967SMike Kravetz * pages in hstate via the proc/sysfs interfaces. 330929383967SMike Kravetz */ 331029383967SMike Kravetz mutex_lock(&h->resize_lock); 3311b65d4adbSMuchun Song flush_free_hpage_work(h); 3312db71ef79SMike Kravetz spin_lock_irq(&hugetlb_lock); 33134eb0716eSAlexandre Ghiti 33144eb0716eSAlexandre Ghiti /* 3315fd875dcaSMike Kravetz * Check for a node specific request. 3316fd875dcaSMike Kravetz * Changing node specific huge page count may require a corresponding 3317fd875dcaSMike Kravetz * change to the global count. In any case, the passed node mask 3318fd875dcaSMike Kravetz * (nodes_allowed) will restrict alloc/free to the specified node. 3319fd875dcaSMike Kravetz */ 3320fd875dcaSMike Kravetz if (nid != NUMA_NO_NODE) { 3321fd875dcaSMike Kravetz unsigned long old_count = count; 3322fd875dcaSMike Kravetz 3323fd875dcaSMike Kravetz count += h->nr_huge_pages - h->nr_huge_pages_node[nid]; 3324fd875dcaSMike Kravetz /* 3325fd875dcaSMike Kravetz * User may have specified a large count value which caused the 3326fd875dcaSMike Kravetz * above calculation to overflow. In this case, they wanted 3327fd875dcaSMike Kravetz * to allocate as many huge pages as possible. Set count to 3328fd875dcaSMike Kravetz * largest possible value to align with their intention. 3329fd875dcaSMike Kravetz */ 3330fd875dcaSMike Kravetz if (count < old_count) 3331fd875dcaSMike Kravetz count = ULONG_MAX; 3332fd875dcaSMike Kravetz } 3333fd875dcaSMike Kravetz 3334fd875dcaSMike Kravetz /* 33354eb0716eSAlexandre Ghiti * Gigantic pages runtime allocation depend on the capability for large 33364eb0716eSAlexandre Ghiti * page range allocation. 33374eb0716eSAlexandre Ghiti * If the system does not provide this feature, return an error when 33384eb0716eSAlexandre Ghiti * the user tries to allocate gigantic pages but let the user free the 33394eb0716eSAlexandre Ghiti * boottime allocated gigantic pages. 33404eb0716eSAlexandre Ghiti */ 33414eb0716eSAlexandre Ghiti if (hstate_is_gigantic(h) && !IS_ENABLED(CONFIG_CONTIG_ALLOC)) { 33424eb0716eSAlexandre Ghiti if (count > persistent_huge_pages(h)) { 3343db71ef79SMike Kravetz spin_unlock_irq(&hugetlb_lock); 334429383967SMike Kravetz mutex_unlock(&h->resize_lock); 3345f60858f9SMike Kravetz NODEMASK_FREE(node_alloc_noretry); 33464eb0716eSAlexandre Ghiti return -EINVAL; 33474eb0716eSAlexandre Ghiti } 33484eb0716eSAlexandre Ghiti /* Fall through to decrease pool */ 33494eb0716eSAlexandre Ghiti } 3350aa888a74SAndi Kleen 33517893d1d5SAdam Litke /* 33527893d1d5SAdam Litke * Increase the pool size 33537893d1d5SAdam Litke * First take pages out of surplus state. Then make up the 33547893d1d5SAdam Litke * remaining difference by allocating fresh huge pages. 3355d1c3fb1fSNishanth Aravamudan * 33560c397daeSMichal Hocko * We might race with alloc_surplus_huge_page() here and be unable 3357d1c3fb1fSNishanth Aravamudan * to convert a surplus huge page to a normal huge page. That is 3358d1c3fb1fSNishanth Aravamudan * not critical, though, it just means the overall size of the 3359d1c3fb1fSNishanth Aravamudan * pool might be one hugepage larger than it needs to be, but 3360d1c3fb1fSNishanth Aravamudan * within all the constraints specified by the sysctls. 33617893d1d5SAdam Litke */ 3362a5516438SAndi Kleen while (h->surplus_huge_pages && count > persistent_huge_pages(h)) { 33636ae11b27SLee Schermerhorn if (!adjust_pool_surplus(h, nodes_allowed, -1)) 33647893d1d5SAdam Litke break; 33657893d1d5SAdam Litke } 33667893d1d5SAdam Litke 3367a5516438SAndi Kleen while (count > persistent_huge_pages(h)) { 33687893d1d5SAdam Litke /* 33697893d1d5SAdam Litke * If this allocation races such that we no longer need the 33707893d1d5SAdam Litke * page, free_huge_page will handle it by freeing the page 33717893d1d5SAdam Litke * and reducing the surplus. 33727893d1d5SAdam Litke */ 3373db71ef79SMike Kravetz spin_unlock_irq(&hugetlb_lock); 3374649920c6SJia He 3375649920c6SJia He /* yield cpu to avoid soft lockup */ 3376649920c6SJia He cond_resched(); 3377649920c6SJia He 3378f60858f9SMike Kravetz ret = alloc_pool_huge_page(h, nodes_allowed, 3379f60858f9SMike Kravetz node_alloc_noretry); 3380db71ef79SMike Kravetz spin_lock_irq(&hugetlb_lock); 33817893d1d5SAdam Litke if (!ret) 33827893d1d5SAdam Litke goto out; 33837893d1d5SAdam Litke 3384536240f2SMel Gorman /* Bail for signals. Probably ctrl-c from user */ 3385536240f2SMel Gorman if (signal_pending(current)) 3386536240f2SMel Gorman goto out; 33877893d1d5SAdam Litke } 33887893d1d5SAdam Litke 33897893d1d5SAdam Litke /* 33907893d1d5SAdam Litke * Decrease the pool size 33917893d1d5SAdam Litke * First return free pages to the buddy allocator (being careful 33927893d1d5SAdam Litke * to keep enough around to satisfy reservations). Then place 33937893d1d5SAdam Litke * pages into surplus state as needed so the pool will shrink 33947893d1d5SAdam Litke * to the desired size as pages become free. 3395d1c3fb1fSNishanth Aravamudan * 3396d1c3fb1fSNishanth Aravamudan * By placing pages into the surplus state independent of the 3397d1c3fb1fSNishanth Aravamudan * overcommit value, we are allowing the surplus pool size to 3398d1c3fb1fSNishanth Aravamudan * exceed overcommit. There are few sane options here. Since 33990c397daeSMichal Hocko * alloc_surplus_huge_page() is checking the global counter, 3400d1c3fb1fSNishanth Aravamudan * though, we'll note that we're not allowed to exceed surplus 3401d1c3fb1fSNishanth Aravamudan * and won't grow the pool anywhere else. Not until one of the 3402d1c3fb1fSNishanth Aravamudan * sysctls are changed, or the surplus pages go out of use. 34037893d1d5SAdam Litke */ 3404a5516438SAndi Kleen min_count = h->resv_huge_pages + h->nr_huge_pages - h->free_huge_pages; 34056b0c880dSAdam Litke min_count = max(count, min_count); 34066ae11b27SLee Schermerhorn try_to_free_low(h, min_count, nodes_allowed); 340710c6ec49SMike Kravetz 340810c6ec49SMike Kravetz /* 340910c6ec49SMike Kravetz * Collect pages to be removed on list without dropping lock 341010c6ec49SMike Kravetz */ 3411a5516438SAndi Kleen while (min_count < persistent_huge_pages(h)) { 341210c6ec49SMike Kravetz page = remove_pool_huge_page(h, nodes_allowed, 0); 341310c6ec49SMike Kravetz if (!page) 34141da177e4SLinus Torvalds break; 341510c6ec49SMike Kravetz 341610c6ec49SMike Kravetz list_add(&page->lru, &page_list); 34171da177e4SLinus Torvalds } 341810c6ec49SMike Kravetz /* free the pages after dropping lock */ 3419db71ef79SMike Kravetz spin_unlock_irq(&hugetlb_lock); 342010c6ec49SMike Kravetz update_and_free_pages_bulk(h, &page_list); 3421b65d4adbSMuchun Song flush_free_hpage_work(h); 3422db71ef79SMike Kravetz spin_lock_irq(&hugetlb_lock); 342310c6ec49SMike Kravetz 3424a5516438SAndi Kleen while (count < persistent_huge_pages(h)) { 34256ae11b27SLee Schermerhorn if (!adjust_pool_surplus(h, nodes_allowed, 1)) 34267893d1d5SAdam Litke break; 34277893d1d5SAdam Litke } 34287893d1d5SAdam Litke out: 34294eb0716eSAlexandre Ghiti h->max_huge_pages = persistent_huge_pages(h); 3430db71ef79SMike Kravetz spin_unlock_irq(&hugetlb_lock); 343129383967SMike Kravetz mutex_unlock(&h->resize_lock); 34324eb0716eSAlexandre Ghiti 3433f60858f9SMike Kravetz NODEMASK_FREE(node_alloc_noretry); 3434f60858f9SMike Kravetz 34354eb0716eSAlexandre Ghiti return 0; 34361da177e4SLinus Torvalds } 34371da177e4SLinus Torvalds 34388531fc6fSMike Kravetz static int demote_free_huge_page(struct hstate *h, struct page *page) 34398531fc6fSMike Kravetz { 34408531fc6fSMike Kravetz int i, nid = page_to_nid(page); 34418531fc6fSMike Kravetz struct hstate *target_hstate; 3442911565b8SSidhartha Kumar struct folio *folio = page_folio(page); 344331731452SDoug Berger struct page *subpage; 34448531fc6fSMike Kravetz int rc = 0; 34458531fc6fSMike Kravetz 34468531fc6fSMike Kravetz target_hstate = size_to_hstate(PAGE_SIZE << h->demote_order); 34478531fc6fSMike Kravetz 3448cfd5082bSSidhartha Kumar remove_hugetlb_folio_for_demote(h, folio, false); 34498531fc6fSMike Kravetz spin_unlock_irq(&hugetlb_lock); 34508531fc6fSMike Kravetz 34516213834cSMuchun Song rc = hugetlb_vmemmap_restore(h, page); 34528531fc6fSMike Kravetz if (rc) { 34538531fc6fSMike Kravetz /* Allocation of vmemmmap failed, we can not demote page */ 34548531fc6fSMike Kravetz spin_lock_irq(&hugetlb_lock); 34558531fc6fSMike Kravetz set_page_refcounted(page); 34568531fc6fSMike Kravetz add_hugetlb_page(h, page, false); 34578531fc6fSMike Kravetz return rc; 34588531fc6fSMike Kravetz } 34598531fc6fSMike Kravetz 34608531fc6fSMike Kravetz /* 3461911565b8SSidhartha Kumar * Use destroy_compound_hugetlb_folio_for_demote for all huge page 34628531fc6fSMike Kravetz * sizes as it will not ref count pages. 34638531fc6fSMike Kravetz */ 3464911565b8SSidhartha Kumar destroy_compound_hugetlb_folio_for_demote(folio, huge_page_order(h)); 34658531fc6fSMike Kravetz 34668531fc6fSMike Kravetz /* 34678531fc6fSMike Kravetz * Taking target hstate mutex synchronizes with set_max_huge_pages. 34688531fc6fSMike Kravetz * Without the mutex, pages added to target hstate could be marked 34698531fc6fSMike Kravetz * as surplus. 34708531fc6fSMike Kravetz * 34718531fc6fSMike Kravetz * Note that we already hold h->resize_lock. To prevent deadlock, 34728531fc6fSMike Kravetz * use the convention of always taking larger size hstate mutex first. 34738531fc6fSMike Kravetz */ 34748531fc6fSMike Kravetz mutex_lock(&target_hstate->resize_lock); 34758531fc6fSMike Kravetz for (i = 0; i < pages_per_huge_page(h); 34768531fc6fSMike Kravetz i += pages_per_huge_page(target_hstate)) { 347731731452SDoug Berger subpage = nth_page(page, i); 34788531fc6fSMike Kravetz if (hstate_is_gigantic(target_hstate)) 347931731452SDoug Berger prep_compound_gigantic_page_for_demote(subpage, 34808531fc6fSMike Kravetz target_hstate->order); 34818531fc6fSMike Kravetz else 348231731452SDoug Berger prep_compound_page(subpage, target_hstate->order); 348331731452SDoug Berger set_page_private(subpage, 0); 348431731452SDoug Berger prep_new_huge_page(target_hstate, subpage, nid); 34852b21624fSMike Kravetz free_huge_page(subpage); 34868531fc6fSMike Kravetz } 34878531fc6fSMike Kravetz mutex_unlock(&target_hstate->resize_lock); 34888531fc6fSMike Kravetz 34898531fc6fSMike Kravetz spin_lock_irq(&hugetlb_lock); 34908531fc6fSMike Kravetz 34918531fc6fSMike Kravetz /* 34928531fc6fSMike Kravetz * Not absolutely necessary, but for consistency update max_huge_pages 34938531fc6fSMike Kravetz * based on pool changes for the demoted page. 34948531fc6fSMike Kravetz */ 34958531fc6fSMike Kravetz h->max_huge_pages--; 3496a43a83c7SMiaohe Lin target_hstate->max_huge_pages += 3497a43a83c7SMiaohe Lin pages_per_huge_page(h) / pages_per_huge_page(target_hstate); 34988531fc6fSMike Kravetz 34998531fc6fSMike Kravetz return rc; 35008531fc6fSMike Kravetz } 35018531fc6fSMike Kravetz 350279dfc695SMike Kravetz static int demote_pool_huge_page(struct hstate *h, nodemask_t *nodes_allowed) 350379dfc695SMike Kravetz __must_hold(&hugetlb_lock) 350479dfc695SMike Kravetz { 35058531fc6fSMike Kravetz int nr_nodes, node; 35068531fc6fSMike Kravetz struct page *page; 350779dfc695SMike Kravetz 350879dfc695SMike Kravetz lockdep_assert_held(&hugetlb_lock); 350979dfc695SMike Kravetz 351079dfc695SMike Kravetz /* We should never get here if no demote order */ 351179dfc695SMike Kravetz if (!h->demote_order) { 351279dfc695SMike Kravetz pr_warn("HugeTLB: NULL demote order passed to demote_pool_huge_page.\n"); 351379dfc695SMike Kravetz return -EINVAL; /* internal error */ 351479dfc695SMike Kravetz } 351579dfc695SMike Kravetz 35168531fc6fSMike Kravetz for_each_node_mask_to_free(h, nr_nodes, node, nodes_allowed) { 35175a317412SMike Kravetz list_for_each_entry(page, &h->hugepage_freelists[node], lru) { 35185a317412SMike Kravetz if (PageHWPoison(page)) 35195a317412SMike Kravetz continue; 35205a317412SMike Kravetz 35215a317412SMike Kravetz return demote_free_huge_page(h, page); 35228531fc6fSMike Kravetz } 35238531fc6fSMike Kravetz } 35248531fc6fSMike Kravetz 35255a317412SMike Kravetz /* 35265a317412SMike Kravetz * Only way to get here is if all pages on free lists are poisoned. 35275a317412SMike Kravetz * Return -EBUSY so that caller will not retry. 35285a317412SMike Kravetz */ 35295a317412SMike Kravetz return -EBUSY; 353079dfc695SMike Kravetz } 353179dfc695SMike Kravetz 3532a3437870SNishanth Aravamudan #define HSTATE_ATTR_RO(_name) \ 3533a3437870SNishanth Aravamudan static struct kobj_attribute _name##_attr = __ATTR_RO(_name) 3534a3437870SNishanth Aravamudan 353579dfc695SMike Kravetz #define HSTATE_ATTR_WO(_name) \ 353679dfc695SMike Kravetz static struct kobj_attribute _name##_attr = __ATTR_WO(_name) 353779dfc695SMike Kravetz 3538a3437870SNishanth Aravamudan #define HSTATE_ATTR(_name) \ 353998bc26acSMiaohe Lin static struct kobj_attribute _name##_attr = __ATTR_RW(_name) 3540a3437870SNishanth Aravamudan 3541a3437870SNishanth Aravamudan static struct kobject *hugepages_kobj; 3542a3437870SNishanth Aravamudan static struct kobject *hstate_kobjs[HUGE_MAX_HSTATE]; 3543a3437870SNishanth Aravamudan 35449a305230SLee Schermerhorn static struct hstate *kobj_to_node_hstate(struct kobject *kobj, int *nidp); 35459a305230SLee Schermerhorn 35469a305230SLee Schermerhorn static struct hstate *kobj_to_hstate(struct kobject *kobj, int *nidp) 3547a3437870SNishanth Aravamudan { 3548a3437870SNishanth Aravamudan int i; 35499a305230SLee Schermerhorn 3550a3437870SNishanth Aravamudan for (i = 0; i < HUGE_MAX_HSTATE; i++) 35519a305230SLee Schermerhorn if (hstate_kobjs[i] == kobj) { 35529a305230SLee Schermerhorn if (nidp) 35539a305230SLee Schermerhorn *nidp = NUMA_NO_NODE; 3554a3437870SNishanth Aravamudan return &hstates[i]; 35559a305230SLee Schermerhorn } 35569a305230SLee Schermerhorn 35579a305230SLee Schermerhorn return kobj_to_node_hstate(kobj, nidp); 3558a3437870SNishanth Aravamudan } 3559a3437870SNishanth Aravamudan 356006808b08SLee Schermerhorn static ssize_t nr_hugepages_show_common(struct kobject *kobj, 3561a3437870SNishanth Aravamudan struct kobj_attribute *attr, char *buf) 3562a3437870SNishanth Aravamudan { 35639a305230SLee Schermerhorn struct hstate *h; 35649a305230SLee Schermerhorn unsigned long nr_huge_pages; 35659a305230SLee Schermerhorn int nid; 35669a305230SLee Schermerhorn 35679a305230SLee Schermerhorn h = kobj_to_hstate(kobj, &nid); 35689a305230SLee Schermerhorn if (nid == NUMA_NO_NODE) 35699a305230SLee Schermerhorn nr_huge_pages = h->nr_huge_pages; 35709a305230SLee Schermerhorn else 35719a305230SLee Schermerhorn nr_huge_pages = h->nr_huge_pages_node[nid]; 35729a305230SLee Schermerhorn 3573ae7a927dSJoe Perches return sysfs_emit(buf, "%lu\n", nr_huge_pages); 3574a3437870SNishanth Aravamudan } 3575adbe8726SEric B Munson 3576238d3c13SDavid Rientjes static ssize_t __nr_hugepages_store_common(bool obey_mempolicy, 3577238d3c13SDavid Rientjes struct hstate *h, int nid, 3578238d3c13SDavid Rientjes unsigned long count, size_t len) 3579a3437870SNishanth Aravamudan { 3580a3437870SNishanth Aravamudan int err; 35812d0adf7eSOscar Salvador nodemask_t nodes_allowed, *n_mask; 3582a3437870SNishanth Aravamudan 35832d0adf7eSOscar Salvador if (hstate_is_gigantic(h) && !gigantic_page_runtime_supported()) 35842d0adf7eSOscar Salvador return -EINVAL; 3585adbe8726SEric B Munson 35869a305230SLee Schermerhorn if (nid == NUMA_NO_NODE) { 35879a305230SLee Schermerhorn /* 35889a305230SLee Schermerhorn * global hstate attribute 35899a305230SLee Schermerhorn */ 35909a305230SLee Schermerhorn if (!(obey_mempolicy && 35912d0adf7eSOscar Salvador init_nodemask_of_mempolicy(&nodes_allowed))) 35922d0adf7eSOscar Salvador n_mask = &node_states[N_MEMORY]; 35932d0adf7eSOscar Salvador else 35942d0adf7eSOscar Salvador n_mask = &nodes_allowed; 35952d0adf7eSOscar Salvador } else { 35969a305230SLee Schermerhorn /* 3597fd875dcaSMike Kravetz * Node specific request. count adjustment happens in 3598fd875dcaSMike Kravetz * set_max_huge_pages() after acquiring hugetlb_lock. 35999a305230SLee Schermerhorn */ 36002d0adf7eSOscar Salvador init_nodemask_of_node(&nodes_allowed, nid); 36012d0adf7eSOscar Salvador n_mask = &nodes_allowed; 3602fd875dcaSMike Kravetz } 36039a305230SLee Schermerhorn 36042d0adf7eSOscar Salvador err = set_max_huge_pages(h, count, nid, n_mask); 360506808b08SLee Schermerhorn 36064eb0716eSAlexandre Ghiti return err ? err : len; 360706808b08SLee Schermerhorn } 360806808b08SLee Schermerhorn 3609238d3c13SDavid Rientjes static ssize_t nr_hugepages_store_common(bool obey_mempolicy, 3610238d3c13SDavid Rientjes struct kobject *kobj, const char *buf, 3611238d3c13SDavid Rientjes size_t len) 3612238d3c13SDavid Rientjes { 3613238d3c13SDavid Rientjes struct hstate *h; 3614238d3c13SDavid Rientjes unsigned long count; 3615238d3c13SDavid Rientjes int nid; 3616238d3c13SDavid Rientjes int err; 3617238d3c13SDavid Rientjes 3618238d3c13SDavid Rientjes err = kstrtoul(buf, 10, &count); 3619238d3c13SDavid Rientjes if (err) 3620238d3c13SDavid Rientjes return err; 3621238d3c13SDavid Rientjes 3622238d3c13SDavid Rientjes h = kobj_to_hstate(kobj, &nid); 3623238d3c13SDavid Rientjes return __nr_hugepages_store_common(obey_mempolicy, h, nid, count, len); 3624238d3c13SDavid Rientjes } 3625238d3c13SDavid Rientjes 362606808b08SLee Schermerhorn static ssize_t nr_hugepages_show(struct kobject *kobj, 362706808b08SLee Schermerhorn struct kobj_attribute *attr, char *buf) 362806808b08SLee Schermerhorn { 362906808b08SLee Schermerhorn return nr_hugepages_show_common(kobj, attr, buf); 363006808b08SLee Schermerhorn } 363106808b08SLee Schermerhorn 363206808b08SLee Schermerhorn static ssize_t nr_hugepages_store(struct kobject *kobj, 363306808b08SLee Schermerhorn struct kobj_attribute *attr, const char *buf, size_t len) 363406808b08SLee Schermerhorn { 3635238d3c13SDavid Rientjes return nr_hugepages_store_common(false, kobj, buf, len); 3636a3437870SNishanth Aravamudan } 3637a3437870SNishanth Aravamudan HSTATE_ATTR(nr_hugepages); 3638a3437870SNishanth Aravamudan 363906808b08SLee Schermerhorn #ifdef CONFIG_NUMA 364006808b08SLee Schermerhorn 364106808b08SLee Schermerhorn /* 364206808b08SLee Schermerhorn * hstate attribute for optionally mempolicy-based constraint on persistent 364306808b08SLee Schermerhorn * huge page alloc/free. 364406808b08SLee Schermerhorn */ 364506808b08SLee Schermerhorn static ssize_t nr_hugepages_mempolicy_show(struct kobject *kobj, 3646ae7a927dSJoe Perches struct kobj_attribute *attr, 3647ae7a927dSJoe Perches char *buf) 364806808b08SLee Schermerhorn { 364906808b08SLee Schermerhorn return nr_hugepages_show_common(kobj, attr, buf); 365006808b08SLee Schermerhorn } 365106808b08SLee Schermerhorn 365206808b08SLee Schermerhorn static ssize_t nr_hugepages_mempolicy_store(struct kobject *kobj, 365306808b08SLee Schermerhorn struct kobj_attribute *attr, const char *buf, size_t len) 365406808b08SLee Schermerhorn { 3655238d3c13SDavid Rientjes return nr_hugepages_store_common(true, kobj, buf, len); 365606808b08SLee Schermerhorn } 365706808b08SLee Schermerhorn HSTATE_ATTR(nr_hugepages_mempolicy); 365806808b08SLee Schermerhorn #endif 365906808b08SLee Schermerhorn 366006808b08SLee Schermerhorn 3661a3437870SNishanth Aravamudan static ssize_t nr_overcommit_hugepages_show(struct kobject *kobj, 3662a3437870SNishanth Aravamudan struct kobj_attribute *attr, char *buf) 3663a3437870SNishanth Aravamudan { 36649a305230SLee Schermerhorn struct hstate *h = kobj_to_hstate(kobj, NULL); 3665ae7a927dSJoe Perches return sysfs_emit(buf, "%lu\n", h->nr_overcommit_huge_pages); 3666a3437870SNishanth Aravamudan } 3667adbe8726SEric B Munson 3668a3437870SNishanth Aravamudan static ssize_t nr_overcommit_hugepages_store(struct kobject *kobj, 3669a3437870SNishanth Aravamudan struct kobj_attribute *attr, const char *buf, size_t count) 3670a3437870SNishanth Aravamudan { 3671a3437870SNishanth Aravamudan int err; 3672a3437870SNishanth Aravamudan unsigned long input; 36739a305230SLee Schermerhorn struct hstate *h = kobj_to_hstate(kobj, NULL); 3674a3437870SNishanth Aravamudan 3675bae7f4aeSLuiz Capitulino if (hstate_is_gigantic(h)) 3676adbe8726SEric B Munson return -EINVAL; 3677adbe8726SEric B Munson 36783dbb95f7SJingoo Han err = kstrtoul(buf, 10, &input); 3679a3437870SNishanth Aravamudan if (err) 368073ae31e5SEric B Munson return err; 3681a3437870SNishanth Aravamudan 3682db71ef79SMike Kravetz spin_lock_irq(&hugetlb_lock); 3683a3437870SNishanth Aravamudan h->nr_overcommit_huge_pages = input; 3684db71ef79SMike Kravetz spin_unlock_irq(&hugetlb_lock); 3685a3437870SNishanth Aravamudan 3686a3437870SNishanth Aravamudan return count; 3687a3437870SNishanth Aravamudan } 3688a3437870SNishanth Aravamudan HSTATE_ATTR(nr_overcommit_hugepages); 3689a3437870SNishanth Aravamudan 3690a3437870SNishanth Aravamudan static ssize_t free_hugepages_show(struct kobject *kobj, 3691a3437870SNishanth Aravamudan struct kobj_attribute *attr, char *buf) 3692a3437870SNishanth Aravamudan { 36939a305230SLee Schermerhorn struct hstate *h; 36949a305230SLee Schermerhorn unsigned long free_huge_pages; 36959a305230SLee Schermerhorn int nid; 36969a305230SLee Schermerhorn 36979a305230SLee Schermerhorn h = kobj_to_hstate(kobj, &nid); 36989a305230SLee Schermerhorn if (nid == NUMA_NO_NODE) 36999a305230SLee Schermerhorn free_huge_pages = h->free_huge_pages; 37009a305230SLee Schermerhorn else 37019a305230SLee Schermerhorn free_huge_pages = h->free_huge_pages_node[nid]; 37029a305230SLee Schermerhorn 3703ae7a927dSJoe Perches return sysfs_emit(buf, "%lu\n", free_huge_pages); 3704a3437870SNishanth Aravamudan } 3705a3437870SNishanth Aravamudan HSTATE_ATTR_RO(free_hugepages); 3706a3437870SNishanth Aravamudan 3707a3437870SNishanth Aravamudan static ssize_t resv_hugepages_show(struct kobject *kobj, 3708a3437870SNishanth Aravamudan struct kobj_attribute *attr, char *buf) 3709a3437870SNishanth Aravamudan { 37109a305230SLee Schermerhorn struct hstate *h = kobj_to_hstate(kobj, NULL); 3711ae7a927dSJoe Perches return sysfs_emit(buf, "%lu\n", h->resv_huge_pages); 3712a3437870SNishanth Aravamudan } 3713a3437870SNishanth Aravamudan HSTATE_ATTR_RO(resv_hugepages); 3714a3437870SNishanth Aravamudan 3715a3437870SNishanth Aravamudan static ssize_t surplus_hugepages_show(struct kobject *kobj, 3716a3437870SNishanth Aravamudan struct kobj_attribute *attr, char *buf) 3717a3437870SNishanth Aravamudan { 37189a305230SLee Schermerhorn struct hstate *h; 37199a305230SLee Schermerhorn unsigned long surplus_huge_pages; 37209a305230SLee Schermerhorn int nid; 37219a305230SLee Schermerhorn 37229a305230SLee Schermerhorn h = kobj_to_hstate(kobj, &nid); 37239a305230SLee Schermerhorn if (nid == NUMA_NO_NODE) 37249a305230SLee Schermerhorn surplus_huge_pages = h->surplus_huge_pages; 37259a305230SLee Schermerhorn else 37269a305230SLee Schermerhorn surplus_huge_pages = h->surplus_huge_pages_node[nid]; 37279a305230SLee Schermerhorn 3728ae7a927dSJoe Perches return sysfs_emit(buf, "%lu\n", surplus_huge_pages); 3729a3437870SNishanth Aravamudan } 3730a3437870SNishanth Aravamudan HSTATE_ATTR_RO(surplus_hugepages); 3731a3437870SNishanth Aravamudan 373279dfc695SMike Kravetz static ssize_t demote_store(struct kobject *kobj, 373379dfc695SMike Kravetz struct kobj_attribute *attr, const char *buf, size_t len) 373479dfc695SMike Kravetz { 373579dfc695SMike Kravetz unsigned long nr_demote; 373679dfc695SMike Kravetz unsigned long nr_available; 373779dfc695SMike Kravetz nodemask_t nodes_allowed, *n_mask; 373879dfc695SMike Kravetz struct hstate *h; 37398eeda55fSLi zeming int err; 374079dfc695SMike Kravetz int nid; 374179dfc695SMike Kravetz 374279dfc695SMike Kravetz err = kstrtoul(buf, 10, &nr_demote); 374379dfc695SMike Kravetz if (err) 374479dfc695SMike Kravetz return err; 374579dfc695SMike Kravetz h = kobj_to_hstate(kobj, &nid); 374679dfc695SMike Kravetz 374779dfc695SMike Kravetz if (nid != NUMA_NO_NODE) { 374879dfc695SMike Kravetz init_nodemask_of_node(&nodes_allowed, nid); 374979dfc695SMike Kravetz n_mask = &nodes_allowed; 375079dfc695SMike Kravetz } else { 375179dfc695SMike Kravetz n_mask = &node_states[N_MEMORY]; 375279dfc695SMike Kravetz } 375379dfc695SMike Kravetz 375479dfc695SMike Kravetz /* Synchronize with other sysfs operations modifying huge pages */ 375579dfc695SMike Kravetz mutex_lock(&h->resize_lock); 375679dfc695SMike Kravetz spin_lock_irq(&hugetlb_lock); 375779dfc695SMike Kravetz 375879dfc695SMike Kravetz while (nr_demote) { 375979dfc695SMike Kravetz /* 376079dfc695SMike Kravetz * Check for available pages to demote each time thorough the 376179dfc695SMike Kravetz * loop as demote_pool_huge_page will drop hugetlb_lock. 376279dfc695SMike Kravetz */ 376379dfc695SMike Kravetz if (nid != NUMA_NO_NODE) 376479dfc695SMike Kravetz nr_available = h->free_huge_pages_node[nid]; 376579dfc695SMike Kravetz else 376679dfc695SMike Kravetz nr_available = h->free_huge_pages; 376779dfc695SMike Kravetz nr_available -= h->resv_huge_pages; 376879dfc695SMike Kravetz if (!nr_available) 376979dfc695SMike Kravetz break; 377079dfc695SMike Kravetz 377179dfc695SMike Kravetz err = demote_pool_huge_page(h, n_mask); 377279dfc695SMike Kravetz if (err) 377379dfc695SMike Kravetz break; 377479dfc695SMike Kravetz 377579dfc695SMike Kravetz nr_demote--; 377679dfc695SMike Kravetz } 377779dfc695SMike Kravetz 377879dfc695SMike Kravetz spin_unlock_irq(&hugetlb_lock); 377979dfc695SMike Kravetz mutex_unlock(&h->resize_lock); 378079dfc695SMike Kravetz 378179dfc695SMike Kravetz if (err) 378279dfc695SMike Kravetz return err; 378379dfc695SMike Kravetz return len; 378479dfc695SMike Kravetz } 378579dfc695SMike Kravetz HSTATE_ATTR_WO(demote); 378679dfc695SMike Kravetz 378779dfc695SMike Kravetz static ssize_t demote_size_show(struct kobject *kobj, 378879dfc695SMike Kravetz struct kobj_attribute *attr, char *buf) 378979dfc695SMike Kravetz { 379012658abfSMiaohe Lin struct hstate *h = kobj_to_hstate(kobj, NULL); 379179dfc695SMike Kravetz unsigned long demote_size = (PAGE_SIZE << h->demote_order) / SZ_1K; 379279dfc695SMike Kravetz 379379dfc695SMike Kravetz return sysfs_emit(buf, "%lukB\n", demote_size); 379479dfc695SMike Kravetz } 379579dfc695SMike Kravetz 379679dfc695SMike Kravetz static ssize_t demote_size_store(struct kobject *kobj, 379779dfc695SMike Kravetz struct kobj_attribute *attr, 379879dfc695SMike Kravetz const char *buf, size_t count) 379979dfc695SMike Kravetz { 380079dfc695SMike Kravetz struct hstate *h, *demote_hstate; 380179dfc695SMike Kravetz unsigned long demote_size; 380279dfc695SMike Kravetz unsigned int demote_order; 380379dfc695SMike Kravetz 380479dfc695SMike Kravetz demote_size = (unsigned long)memparse(buf, NULL); 380579dfc695SMike Kravetz 380679dfc695SMike Kravetz demote_hstate = size_to_hstate(demote_size); 380779dfc695SMike Kravetz if (!demote_hstate) 380879dfc695SMike Kravetz return -EINVAL; 380979dfc695SMike Kravetz demote_order = demote_hstate->order; 3810a01f4390SMike Kravetz if (demote_order < HUGETLB_PAGE_ORDER) 3811a01f4390SMike Kravetz return -EINVAL; 381279dfc695SMike Kravetz 381379dfc695SMike Kravetz /* demote order must be smaller than hstate order */ 381412658abfSMiaohe Lin h = kobj_to_hstate(kobj, NULL); 381579dfc695SMike Kravetz if (demote_order >= h->order) 381679dfc695SMike Kravetz return -EINVAL; 381779dfc695SMike Kravetz 381879dfc695SMike Kravetz /* resize_lock synchronizes access to demote size and writes */ 381979dfc695SMike Kravetz mutex_lock(&h->resize_lock); 382079dfc695SMike Kravetz h->demote_order = demote_order; 382179dfc695SMike Kravetz mutex_unlock(&h->resize_lock); 382279dfc695SMike Kravetz 382379dfc695SMike Kravetz return count; 382479dfc695SMike Kravetz } 382579dfc695SMike Kravetz HSTATE_ATTR(demote_size); 382679dfc695SMike Kravetz 3827a3437870SNishanth Aravamudan static struct attribute *hstate_attrs[] = { 3828a3437870SNishanth Aravamudan &nr_hugepages_attr.attr, 3829a3437870SNishanth Aravamudan &nr_overcommit_hugepages_attr.attr, 3830a3437870SNishanth Aravamudan &free_hugepages_attr.attr, 3831a3437870SNishanth Aravamudan &resv_hugepages_attr.attr, 3832a3437870SNishanth Aravamudan &surplus_hugepages_attr.attr, 383306808b08SLee Schermerhorn #ifdef CONFIG_NUMA 383406808b08SLee Schermerhorn &nr_hugepages_mempolicy_attr.attr, 383506808b08SLee Schermerhorn #endif 3836a3437870SNishanth Aravamudan NULL, 3837a3437870SNishanth Aravamudan }; 3838a3437870SNishanth Aravamudan 383967e5ed96SArvind Yadav static const struct attribute_group hstate_attr_group = { 3840a3437870SNishanth Aravamudan .attrs = hstate_attrs, 3841a3437870SNishanth Aravamudan }; 3842a3437870SNishanth Aravamudan 384379dfc695SMike Kravetz static struct attribute *hstate_demote_attrs[] = { 384479dfc695SMike Kravetz &demote_size_attr.attr, 384579dfc695SMike Kravetz &demote_attr.attr, 384679dfc695SMike Kravetz NULL, 384779dfc695SMike Kravetz }; 384879dfc695SMike Kravetz 384979dfc695SMike Kravetz static const struct attribute_group hstate_demote_attr_group = { 385079dfc695SMike Kravetz .attrs = hstate_demote_attrs, 385179dfc695SMike Kravetz }; 385279dfc695SMike Kravetz 3853094e9539SJeff Mahoney static int hugetlb_sysfs_add_hstate(struct hstate *h, struct kobject *parent, 38549a305230SLee Schermerhorn struct kobject **hstate_kobjs, 385567e5ed96SArvind Yadav const struct attribute_group *hstate_attr_group) 3856a3437870SNishanth Aravamudan { 3857a3437870SNishanth Aravamudan int retval; 3858972dc4deSAneesh Kumar K.V int hi = hstate_index(h); 3859a3437870SNishanth Aravamudan 38609a305230SLee Schermerhorn hstate_kobjs[hi] = kobject_create_and_add(h->name, parent); 38619a305230SLee Schermerhorn if (!hstate_kobjs[hi]) 3862a3437870SNishanth Aravamudan return -ENOMEM; 3863a3437870SNishanth Aravamudan 38649a305230SLee Schermerhorn retval = sysfs_create_group(hstate_kobjs[hi], hstate_attr_group); 3865cc2205a6SMiaohe Lin if (retval) { 38669a305230SLee Schermerhorn kobject_put(hstate_kobjs[hi]); 3867cc2205a6SMiaohe Lin hstate_kobjs[hi] = NULL; 38683a6bdda0SMiaohe Lin return retval; 3869cc2205a6SMiaohe Lin } 3870a3437870SNishanth Aravamudan 387179dfc695SMike Kravetz if (h->demote_order) { 387201088a60SMiaohe Lin retval = sysfs_create_group(hstate_kobjs[hi], 387301088a60SMiaohe Lin &hstate_demote_attr_group); 387401088a60SMiaohe Lin if (retval) { 387579dfc695SMike Kravetz pr_warn("HugeTLB unable to create demote interfaces for %s\n", h->name); 387601088a60SMiaohe Lin sysfs_remove_group(hstate_kobjs[hi], hstate_attr_group); 387701088a60SMiaohe Lin kobject_put(hstate_kobjs[hi]); 387801088a60SMiaohe Lin hstate_kobjs[hi] = NULL; 387901088a60SMiaohe Lin return retval; 388001088a60SMiaohe Lin } 388179dfc695SMike Kravetz } 388279dfc695SMike Kravetz 388301088a60SMiaohe Lin return 0; 3884a3437870SNishanth Aravamudan } 3885a3437870SNishanth Aravamudan 38869a305230SLee Schermerhorn #ifdef CONFIG_NUMA 3887a4a00b45SMuchun Song static bool hugetlb_sysfs_initialized __ro_after_init; 38889a305230SLee Schermerhorn 38899a305230SLee Schermerhorn /* 38909a305230SLee Schermerhorn * node_hstate/s - associate per node hstate attributes, via their kobjects, 389110fbcf4cSKay Sievers * with node devices in node_devices[] using a parallel array. The array 389210fbcf4cSKay Sievers * index of a node device or _hstate == node id. 389310fbcf4cSKay Sievers * This is here to avoid any static dependency of the node device driver, in 38949a305230SLee Schermerhorn * the base kernel, on the hugetlb module. 38959a305230SLee Schermerhorn */ 38969a305230SLee Schermerhorn struct node_hstate { 38979a305230SLee Schermerhorn struct kobject *hugepages_kobj; 38989a305230SLee Schermerhorn struct kobject *hstate_kobjs[HUGE_MAX_HSTATE]; 38999a305230SLee Schermerhorn }; 3900b4e289a6SAlexander Kuleshov static struct node_hstate node_hstates[MAX_NUMNODES]; 39019a305230SLee Schermerhorn 39029a305230SLee Schermerhorn /* 390310fbcf4cSKay Sievers * A subset of global hstate attributes for node devices 39049a305230SLee Schermerhorn */ 39059a305230SLee Schermerhorn static struct attribute *per_node_hstate_attrs[] = { 39069a305230SLee Schermerhorn &nr_hugepages_attr.attr, 39079a305230SLee Schermerhorn &free_hugepages_attr.attr, 39089a305230SLee Schermerhorn &surplus_hugepages_attr.attr, 39099a305230SLee Schermerhorn NULL, 39109a305230SLee Schermerhorn }; 39119a305230SLee Schermerhorn 391267e5ed96SArvind Yadav static const struct attribute_group per_node_hstate_attr_group = { 39139a305230SLee Schermerhorn .attrs = per_node_hstate_attrs, 39149a305230SLee Schermerhorn }; 39159a305230SLee Schermerhorn 39169a305230SLee Schermerhorn /* 391710fbcf4cSKay Sievers * kobj_to_node_hstate - lookup global hstate for node device hstate attr kobj. 39189a305230SLee Schermerhorn * Returns node id via non-NULL nidp. 39199a305230SLee Schermerhorn */ 39209a305230SLee Schermerhorn static struct hstate *kobj_to_node_hstate(struct kobject *kobj, int *nidp) 39219a305230SLee Schermerhorn { 39229a305230SLee Schermerhorn int nid; 39239a305230SLee Schermerhorn 39249a305230SLee Schermerhorn for (nid = 0; nid < nr_node_ids; nid++) { 39259a305230SLee Schermerhorn struct node_hstate *nhs = &node_hstates[nid]; 39269a305230SLee Schermerhorn int i; 39279a305230SLee Schermerhorn for (i = 0; i < HUGE_MAX_HSTATE; i++) 39289a305230SLee Schermerhorn if (nhs->hstate_kobjs[i] == kobj) { 39299a305230SLee Schermerhorn if (nidp) 39309a305230SLee Schermerhorn *nidp = nid; 39319a305230SLee Schermerhorn return &hstates[i]; 39329a305230SLee Schermerhorn } 39339a305230SLee Schermerhorn } 39349a305230SLee Schermerhorn 39359a305230SLee Schermerhorn BUG(); 39369a305230SLee Schermerhorn return NULL; 39379a305230SLee Schermerhorn } 39389a305230SLee Schermerhorn 39399a305230SLee Schermerhorn /* 394010fbcf4cSKay Sievers * Unregister hstate attributes from a single node device. 39419a305230SLee Schermerhorn * No-op if no hstate attributes attached. 39429a305230SLee Schermerhorn */ 3943a4a00b45SMuchun Song void hugetlb_unregister_node(struct node *node) 39449a305230SLee Schermerhorn { 39459a305230SLee Schermerhorn struct hstate *h; 394610fbcf4cSKay Sievers struct node_hstate *nhs = &node_hstates[node->dev.id]; 39479a305230SLee Schermerhorn 39489a305230SLee Schermerhorn if (!nhs->hugepages_kobj) 39499b5e5d0fSLee Schermerhorn return; /* no hstate attributes */ 39509a305230SLee Schermerhorn 3951972dc4deSAneesh Kumar K.V for_each_hstate(h) { 3952972dc4deSAneesh Kumar K.V int idx = hstate_index(h); 395301088a60SMiaohe Lin struct kobject *hstate_kobj = nhs->hstate_kobjs[idx]; 395401088a60SMiaohe Lin 395501088a60SMiaohe Lin if (!hstate_kobj) 395601088a60SMiaohe Lin continue; 395701088a60SMiaohe Lin if (h->demote_order) 395801088a60SMiaohe Lin sysfs_remove_group(hstate_kobj, &hstate_demote_attr_group); 395901088a60SMiaohe Lin sysfs_remove_group(hstate_kobj, &per_node_hstate_attr_group); 396001088a60SMiaohe Lin kobject_put(hstate_kobj); 3961972dc4deSAneesh Kumar K.V nhs->hstate_kobjs[idx] = NULL; 3962972dc4deSAneesh Kumar K.V } 39639a305230SLee Schermerhorn 39649a305230SLee Schermerhorn kobject_put(nhs->hugepages_kobj); 39659a305230SLee Schermerhorn nhs->hugepages_kobj = NULL; 39669a305230SLee Schermerhorn } 39679a305230SLee Schermerhorn 39689a305230SLee Schermerhorn 39699a305230SLee Schermerhorn /* 397010fbcf4cSKay Sievers * Register hstate attributes for a single node device. 39719a305230SLee Schermerhorn * No-op if attributes already registered. 39729a305230SLee Schermerhorn */ 3973a4a00b45SMuchun Song void hugetlb_register_node(struct node *node) 39749a305230SLee Schermerhorn { 39759a305230SLee Schermerhorn struct hstate *h; 397610fbcf4cSKay Sievers struct node_hstate *nhs = &node_hstates[node->dev.id]; 39779a305230SLee Schermerhorn int err; 39789a305230SLee Schermerhorn 3979a4a00b45SMuchun Song if (!hugetlb_sysfs_initialized) 3980a4a00b45SMuchun Song return; 3981a4a00b45SMuchun Song 39829a305230SLee Schermerhorn if (nhs->hugepages_kobj) 39839a305230SLee Schermerhorn return; /* already allocated */ 39849a305230SLee Schermerhorn 39859a305230SLee Schermerhorn nhs->hugepages_kobj = kobject_create_and_add("hugepages", 398610fbcf4cSKay Sievers &node->dev.kobj); 39879a305230SLee Schermerhorn if (!nhs->hugepages_kobj) 39889a305230SLee Schermerhorn return; 39899a305230SLee Schermerhorn 39909a305230SLee Schermerhorn for_each_hstate(h) { 39919a305230SLee Schermerhorn err = hugetlb_sysfs_add_hstate(h, nhs->hugepages_kobj, 39929a305230SLee Schermerhorn nhs->hstate_kobjs, 39939a305230SLee Schermerhorn &per_node_hstate_attr_group); 39949a305230SLee Schermerhorn if (err) { 3995282f4214SMike Kravetz pr_err("HugeTLB: Unable to add hstate %s for node %d\n", 399610fbcf4cSKay Sievers h->name, node->dev.id); 39979a305230SLee Schermerhorn hugetlb_unregister_node(node); 39989a305230SLee Schermerhorn break; 39999a305230SLee Schermerhorn } 40009a305230SLee Schermerhorn } 40019a305230SLee Schermerhorn } 40029a305230SLee Schermerhorn 40039a305230SLee Schermerhorn /* 40049b5e5d0fSLee Schermerhorn * hugetlb init time: register hstate attributes for all registered node 400510fbcf4cSKay Sievers * devices of nodes that have memory. All on-line nodes should have 400610fbcf4cSKay Sievers * registered their associated device by this time. 40079a305230SLee Schermerhorn */ 40087d9ca000SLuiz Capitulino static void __init hugetlb_register_all_nodes(void) 40099a305230SLee Schermerhorn { 40109a305230SLee Schermerhorn int nid; 40119a305230SLee Schermerhorn 4012a4a00b45SMuchun Song for_each_online_node(nid) 4013b958d4d0SMuchun Song hugetlb_register_node(node_devices[nid]); 40149a305230SLee Schermerhorn } 40159a305230SLee Schermerhorn #else /* !CONFIG_NUMA */ 40169a305230SLee Schermerhorn 40179a305230SLee Schermerhorn static struct hstate *kobj_to_node_hstate(struct kobject *kobj, int *nidp) 40189a305230SLee Schermerhorn { 40199a305230SLee Schermerhorn BUG(); 40209a305230SLee Schermerhorn if (nidp) 40219a305230SLee Schermerhorn *nidp = -1; 40229a305230SLee Schermerhorn return NULL; 40239a305230SLee Schermerhorn } 40249a305230SLee Schermerhorn 40259a305230SLee Schermerhorn static void hugetlb_register_all_nodes(void) { } 40269a305230SLee Schermerhorn 40279a305230SLee Schermerhorn #endif 40289a305230SLee Schermerhorn 4029263b8998SMiaohe Lin #ifdef CONFIG_CMA 4030263b8998SMiaohe Lin static void __init hugetlb_cma_check(void); 4031263b8998SMiaohe Lin #else 4032263b8998SMiaohe Lin static inline __init void hugetlb_cma_check(void) 4033263b8998SMiaohe Lin { 4034263b8998SMiaohe Lin } 4035263b8998SMiaohe Lin #endif 4036263b8998SMiaohe Lin 4037a4a00b45SMuchun Song static void __init hugetlb_sysfs_init(void) 4038a4a00b45SMuchun Song { 4039a4a00b45SMuchun Song struct hstate *h; 4040a4a00b45SMuchun Song int err; 4041a4a00b45SMuchun Song 4042a4a00b45SMuchun Song hugepages_kobj = kobject_create_and_add("hugepages", mm_kobj); 4043a4a00b45SMuchun Song if (!hugepages_kobj) 4044a4a00b45SMuchun Song return; 4045a4a00b45SMuchun Song 4046a4a00b45SMuchun Song for_each_hstate(h) { 4047a4a00b45SMuchun Song err = hugetlb_sysfs_add_hstate(h, hugepages_kobj, 4048a4a00b45SMuchun Song hstate_kobjs, &hstate_attr_group); 4049a4a00b45SMuchun Song if (err) 4050a4a00b45SMuchun Song pr_err("HugeTLB: Unable to add hstate %s", h->name); 4051a4a00b45SMuchun Song } 4052a4a00b45SMuchun Song 4053a4a00b45SMuchun Song #ifdef CONFIG_NUMA 4054a4a00b45SMuchun Song hugetlb_sysfs_initialized = true; 4055a4a00b45SMuchun Song #endif 4056a4a00b45SMuchun Song hugetlb_register_all_nodes(); 4057a4a00b45SMuchun Song } 4058a4a00b45SMuchun Song 4059a3437870SNishanth Aravamudan static int __init hugetlb_init(void) 4060a3437870SNishanth Aravamudan { 40618382d914SDavidlohr Bueso int i; 40628382d914SDavidlohr Bueso 4063d6995da3SMike Kravetz BUILD_BUG_ON(sizeof_field(struct page, private) * BITS_PER_BYTE < 4064d6995da3SMike Kravetz __NR_HPAGEFLAGS); 4065d6995da3SMike Kravetz 4066c2833a5bSMike Kravetz if (!hugepages_supported()) { 4067c2833a5bSMike Kravetz if (hugetlb_max_hstate || default_hstate_max_huge_pages) 4068c2833a5bSMike Kravetz pr_warn("HugeTLB: huge pages not supported, ignoring associated command-line parameters\n"); 40690ef89d25SBenjamin Herrenschmidt return 0; 4070d715cf80SLiam R. Howlett } 4071d715cf80SLiam R. Howlett 4072282f4214SMike Kravetz /* 4073282f4214SMike Kravetz * Make sure HPAGE_SIZE (HUGETLB_PAGE_ORDER) hstate exists. Some 4074282f4214SMike Kravetz * architectures depend on setup being done here. 4075282f4214SMike Kravetz */ 4076a3437870SNishanth Aravamudan hugetlb_add_hstate(HUGETLB_PAGE_ORDER); 4077282f4214SMike Kravetz if (!parsed_default_hugepagesz) { 4078282f4214SMike Kravetz /* 4079282f4214SMike Kravetz * If we did not parse a default huge page size, set 4080282f4214SMike Kravetz * default_hstate_idx to HPAGE_SIZE hstate. And, if the 4081282f4214SMike Kravetz * number of huge pages for this default size was implicitly 4082282f4214SMike Kravetz * specified, set that here as well. 4083282f4214SMike Kravetz * Note that the implicit setting will overwrite an explicit 4084282f4214SMike Kravetz * setting. A warning will be printed in this case. 4085282f4214SMike Kravetz */ 4086282f4214SMike Kravetz default_hstate_idx = hstate_index(size_to_hstate(HPAGE_SIZE)); 4087f8b74815SVaishali Thakkar if (default_hstate_max_huge_pages) { 4088282f4214SMike Kravetz if (default_hstate.max_huge_pages) { 4089282f4214SMike Kravetz char buf[32]; 4090282f4214SMike Kravetz 4091282f4214SMike Kravetz string_get_size(huge_page_size(&default_hstate), 4092282f4214SMike Kravetz 1, STRING_UNITS_2, buf, 32); 4093282f4214SMike Kravetz pr_warn("HugeTLB: Ignoring hugepages=%lu associated with %s page size\n", 4094282f4214SMike Kravetz default_hstate.max_huge_pages, buf); 4095282f4214SMike Kravetz pr_warn("HugeTLB: Using hugepages=%lu for number of default huge pages\n", 4096282f4214SMike Kravetz default_hstate_max_huge_pages); 4097282f4214SMike Kravetz } 4098282f4214SMike Kravetz default_hstate.max_huge_pages = 4099282f4214SMike Kravetz default_hstate_max_huge_pages; 4100b5389086SZhenguo Yao 41010a7a0f6fSPeng Liu for_each_online_node(i) 4102b5389086SZhenguo Yao default_hstate.max_huge_pages_node[i] = 4103b5389086SZhenguo Yao default_hugepages_in_node[i]; 4104282f4214SMike Kravetz } 4105f8b74815SVaishali Thakkar } 4106a3437870SNishanth Aravamudan 4107cf11e85fSRoman Gushchin hugetlb_cma_check(); 4108a3437870SNishanth Aravamudan hugetlb_init_hstates(); 4109aa888a74SAndi Kleen gather_bootmem_prealloc(); 4110a3437870SNishanth Aravamudan report_hugepages(); 4111a3437870SNishanth Aravamudan 4112a3437870SNishanth Aravamudan hugetlb_sysfs_init(); 41137179e7bfSJianguo Wu hugetlb_cgroup_file_init(); 41149a305230SLee Schermerhorn 41158382d914SDavidlohr Bueso #ifdef CONFIG_SMP 41168382d914SDavidlohr Bueso num_fault_mutexes = roundup_pow_of_two(8 * num_possible_cpus()); 41178382d914SDavidlohr Bueso #else 41188382d914SDavidlohr Bueso num_fault_mutexes = 1; 41198382d914SDavidlohr Bueso #endif 4120c672c7f2SMike Kravetz hugetlb_fault_mutex_table = 41216da2ec56SKees Cook kmalloc_array(num_fault_mutexes, sizeof(struct mutex), 41226da2ec56SKees Cook GFP_KERNEL); 4123c672c7f2SMike Kravetz BUG_ON(!hugetlb_fault_mutex_table); 41248382d914SDavidlohr Bueso 41258382d914SDavidlohr Bueso for (i = 0; i < num_fault_mutexes; i++) 4126c672c7f2SMike Kravetz mutex_init(&hugetlb_fault_mutex_table[i]); 4127a3437870SNishanth Aravamudan return 0; 4128a3437870SNishanth Aravamudan } 41293e89e1c5SPaul Gortmaker subsys_initcall(hugetlb_init); 4130a3437870SNishanth Aravamudan 4131ae94da89SMike Kravetz /* Overwritten by architectures with more huge page sizes */ 4132ae94da89SMike Kravetz bool __init __attribute((weak)) arch_hugetlb_valid_size(unsigned long size) 41339fee021dSVaishali Thakkar { 4134ae94da89SMike Kravetz return size == HPAGE_SIZE; 41359fee021dSVaishali Thakkar } 41369fee021dSVaishali Thakkar 4137d00181b9SKirill A. Shutemov void __init hugetlb_add_hstate(unsigned int order) 4138a3437870SNishanth Aravamudan { 4139a3437870SNishanth Aravamudan struct hstate *h; 41408faa8b07SAndi Kleen unsigned long i; 41418faa8b07SAndi Kleen 4142a3437870SNishanth Aravamudan if (size_to_hstate(PAGE_SIZE << order)) { 4143a3437870SNishanth Aravamudan return; 4144a3437870SNishanth Aravamudan } 414547d38344SAneesh Kumar K.V BUG_ON(hugetlb_max_hstate >= HUGE_MAX_HSTATE); 4146a3437870SNishanth Aravamudan BUG_ON(order == 0); 414747d38344SAneesh Kumar K.V h = &hstates[hugetlb_max_hstate++]; 414829383967SMike Kravetz mutex_init(&h->resize_lock); 4149a3437870SNishanth Aravamudan h->order = order; 4150aca78307SMiaohe Lin h->mask = ~(huge_page_size(h) - 1); 41518faa8b07SAndi Kleen for (i = 0; i < MAX_NUMNODES; ++i) 41528faa8b07SAndi Kleen INIT_LIST_HEAD(&h->hugepage_freelists[i]); 41530edaecfaSAneesh Kumar K.V INIT_LIST_HEAD(&h->hugepage_activelist); 415454f18d35SAndrew Morton h->next_nid_to_alloc = first_memory_node; 415554f18d35SAndrew Morton h->next_nid_to_free = first_memory_node; 4156a3437870SNishanth Aravamudan snprintf(h->name, HSTATE_NAME_LEN, "hugepages-%lukB", 4157c2c3a60aSMiaohe Lin huge_page_size(h)/SZ_1K); 41588faa8b07SAndi Kleen 4159a3437870SNishanth Aravamudan parsed_hstate = h; 4160a3437870SNishanth Aravamudan } 4161a3437870SNishanth Aravamudan 4162b5389086SZhenguo Yao bool __init __weak hugetlb_node_alloc_supported(void) 4163b5389086SZhenguo Yao { 4164b5389086SZhenguo Yao return true; 4165b5389086SZhenguo Yao } 4166f87442f4SPeng Liu 4167f87442f4SPeng Liu static void __init hugepages_clear_pages_in_node(void) 4168f87442f4SPeng Liu { 4169f87442f4SPeng Liu if (!hugetlb_max_hstate) { 4170f87442f4SPeng Liu default_hstate_max_huge_pages = 0; 4171f87442f4SPeng Liu memset(default_hugepages_in_node, 0, 417210395680SMiaohe Lin sizeof(default_hugepages_in_node)); 4173f87442f4SPeng Liu } else { 4174f87442f4SPeng Liu parsed_hstate->max_huge_pages = 0; 4175f87442f4SPeng Liu memset(parsed_hstate->max_huge_pages_node, 0, 417610395680SMiaohe Lin sizeof(parsed_hstate->max_huge_pages_node)); 4177f87442f4SPeng Liu } 4178f87442f4SPeng Liu } 4179f87442f4SPeng Liu 4180282f4214SMike Kravetz /* 4181282f4214SMike Kravetz * hugepages command line processing 4182282f4214SMike Kravetz * hugepages normally follows a valid hugepagsz or default_hugepagsz 4183282f4214SMike Kravetz * specification. If not, ignore the hugepages value. hugepages can also 4184282f4214SMike Kravetz * be the first huge page command line option in which case it implicitly 4185282f4214SMike Kravetz * specifies the number of huge pages for the default size. 4186282f4214SMike Kravetz */ 4187282f4214SMike Kravetz static int __init hugepages_setup(char *s) 4188a3437870SNishanth Aravamudan { 4189a3437870SNishanth Aravamudan unsigned long *mhp; 41908faa8b07SAndi Kleen static unsigned long *last_mhp; 4191b5389086SZhenguo Yao int node = NUMA_NO_NODE; 4192b5389086SZhenguo Yao int count; 4193b5389086SZhenguo Yao unsigned long tmp; 4194b5389086SZhenguo Yao char *p = s; 4195a3437870SNishanth Aravamudan 41969fee021dSVaishali Thakkar if (!parsed_valid_hugepagesz) { 4197282f4214SMike Kravetz pr_warn("HugeTLB: hugepages=%s does not follow a valid hugepagesz, ignoring\n", s); 41989fee021dSVaishali Thakkar parsed_valid_hugepagesz = true; 4199f81f6e4bSPeng Liu return 1; 42009fee021dSVaishali Thakkar } 4201282f4214SMike Kravetz 4202a3437870SNishanth Aravamudan /* 4203282f4214SMike Kravetz * !hugetlb_max_hstate means we haven't parsed a hugepagesz= parameter 4204282f4214SMike Kravetz * yet, so this hugepages= parameter goes to the "default hstate". 4205282f4214SMike Kravetz * Otherwise, it goes with the previously parsed hugepagesz or 4206282f4214SMike Kravetz * default_hugepagesz. 4207a3437870SNishanth Aravamudan */ 42089fee021dSVaishali Thakkar else if (!hugetlb_max_hstate) 4209a3437870SNishanth Aravamudan mhp = &default_hstate_max_huge_pages; 4210a3437870SNishanth Aravamudan else 4211a3437870SNishanth Aravamudan mhp = &parsed_hstate->max_huge_pages; 4212a3437870SNishanth Aravamudan 42138faa8b07SAndi Kleen if (mhp == last_mhp) { 4214282f4214SMike Kravetz pr_warn("HugeTLB: hugepages= specified twice without interleaving hugepagesz=, ignoring hugepages=%s\n", s); 4215f81f6e4bSPeng Liu return 1; 42168faa8b07SAndi Kleen } 42178faa8b07SAndi Kleen 4218b5389086SZhenguo Yao while (*p) { 4219b5389086SZhenguo Yao count = 0; 4220b5389086SZhenguo Yao if (sscanf(p, "%lu%n", &tmp, &count) != 1) 4221b5389086SZhenguo Yao goto invalid; 4222b5389086SZhenguo Yao /* Parameter is node format */ 4223b5389086SZhenguo Yao if (p[count] == ':') { 4224b5389086SZhenguo Yao if (!hugetlb_node_alloc_supported()) { 4225b5389086SZhenguo Yao pr_warn("HugeTLB: architecture can't support node specific alloc, ignoring!\n"); 4226f81f6e4bSPeng Liu return 1; 4227b5389086SZhenguo Yao } 42280a7a0f6fSPeng Liu if (tmp >= MAX_NUMNODES || !node_online(tmp)) 4229e79ce983SLiu Yuntao goto invalid; 42300a7a0f6fSPeng Liu node = array_index_nospec(tmp, MAX_NUMNODES); 4231b5389086SZhenguo Yao p += count + 1; 4232b5389086SZhenguo Yao /* Parse hugepages */ 4233b5389086SZhenguo Yao if (sscanf(p, "%lu%n", &tmp, &count) != 1) 4234b5389086SZhenguo Yao goto invalid; 4235b5389086SZhenguo Yao if (!hugetlb_max_hstate) 4236b5389086SZhenguo Yao default_hugepages_in_node[node] = tmp; 4237b5389086SZhenguo Yao else 4238b5389086SZhenguo Yao parsed_hstate->max_huge_pages_node[node] = tmp; 4239b5389086SZhenguo Yao *mhp += tmp; 4240b5389086SZhenguo Yao /* Go to parse next node*/ 4241b5389086SZhenguo Yao if (p[count] == ',') 4242b5389086SZhenguo Yao p += count + 1; 4243b5389086SZhenguo Yao else 4244b5389086SZhenguo Yao break; 4245b5389086SZhenguo Yao } else { 4246b5389086SZhenguo Yao if (p != s) 4247b5389086SZhenguo Yao goto invalid; 4248b5389086SZhenguo Yao *mhp = tmp; 4249b5389086SZhenguo Yao break; 4250b5389086SZhenguo Yao } 4251b5389086SZhenguo Yao } 4252a3437870SNishanth Aravamudan 42538faa8b07SAndi Kleen /* 42548faa8b07SAndi Kleen * Global state is always initialized later in hugetlb_init. 425504adbc3fSMiaohe Lin * But we need to allocate gigantic hstates here early to still 42568faa8b07SAndi Kleen * use the bootmem allocator. 42578faa8b07SAndi Kleen */ 425804adbc3fSMiaohe Lin if (hugetlb_max_hstate && hstate_is_gigantic(parsed_hstate)) 42598faa8b07SAndi Kleen hugetlb_hstate_alloc_pages(parsed_hstate); 42608faa8b07SAndi Kleen 42618faa8b07SAndi Kleen last_mhp = mhp; 42628faa8b07SAndi Kleen 4263a3437870SNishanth Aravamudan return 1; 4264b5389086SZhenguo Yao 4265b5389086SZhenguo Yao invalid: 4266b5389086SZhenguo Yao pr_warn("HugeTLB: Invalid hugepages parameter %s\n", p); 4267f87442f4SPeng Liu hugepages_clear_pages_in_node(); 4268f81f6e4bSPeng Liu return 1; 4269a3437870SNishanth Aravamudan } 4270282f4214SMike Kravetz __setup("hugepages=", hugepages_setup); 4271e11bfbfcSNick Piggin 4272282f4214SMike Kravetz /* 4273282f4214SMike Kravetz * hugepagesz command line processing 4274282f4214SMike Kravetz * A specific huge page size can only be specified once with hugepagesz. 4275282f4214SMike Kravetz * hugepagesz is followed by hugepages on the command line. The global 4276282f4214SMike Kravetz * variable 'parsed_valid_hugepagesz' is used to determine if prior 4277282f4214SMike Kravetz * hugepagesz argument was valid. 4278282f4214SMike Kravetz */ 4279359f2544SMike Kravetz static int __init hugepagesz_setup(char *s) 4280e11bfbfcSNick Piggin { 4281359f2544SMike Kravetz unsigned long size; 4282282f4214SMike Kravetz struct hstate *h; 4283282f4214SMike Kravetz 4284282f4214SMike Kravetz parsed_valid_hugepagesz = false; 4285359f2544SMike Kravetz size = (unsigned long)memparse(s, NULL); 4286359f2544SMike Kravetz 4287359f2544SMike Kravetz if (!arch_hugetlb_valid_size(size)) { 4288282f4214SMike Kravetz pr_err("HugeTLB: unsupported hugepagesz=%s\n", s); 4289f81f6e4bSPeng Liu return 1; 4290359f2544SMike Kravetz } 4291359f2544SMike Kravetz 4292282f4214SMike Kravetz h = size_to_hstate(size); 4293282f4214SMike Kravetz if (h) { 4294282f4214SMike Kravetz /* 4295282f4214SMike Kravetz * hstate for this size already exists. This is normally 4296282f4214SMike Kravetz * an error, but is allowed if the existing hstate is the 4297282f4214SMike Kravetz * default hstate. More specifically, it is only allowed if 4298282f4214SMike Kravetz * the number of huge pages for the default hstate was not 4299282f4214SMike Kravetz * previously specified. 4300282f4214SMike Kravetz */ 4301282f4214SMike Kravetz if (!parsed_default_hugepagesz || h != &default_hstate || 4302282f4214SMike Kravetz default_hstate.max_huge_pages) { 4303282f4214SMike Kravetz pr_warn("HugeTLB: hugepagesz=%s specified twice, ignoring\n", s); 4304f81f6e4bSPeng Liu return 1; 430538237830SMike Kravetz } 430638237830SMike Kravetz 4307282f4214SMike Kravetz /* 4308282f4214SMike Kravetz * No need to call hugetlb_add_hstate() as hstate already 4309282f4214SMike Kravetz * exists. But, do set parsed_hstate so that a following 4310282f4214SMike Kravetz * hugepages= parameter will be applied to this hstate. 4311282f4214SMike Kravetz */ 4312282f4214SMike Kravetz parsed_hstate = h; 4313282f4214SMike Kravetz parsed_valid_hugepagesz = true; 4314e11bfbfcSNick Piggin return 1; 4315e11bfbfcSNick Piggin } 4316282f4214SMike Kravetz 4317359f2544SMike Kravetz hugetlb_add_hstate(ilog2(size) - PAGE_SHIFT); 4318282f4214SMike Kravetz parsed_valid_hugepagesz = true; 4319359f2544SMike Kravetz return 1; 4320359f2544SMike Kravetz } 4321359f2544SMike Kravetz __setup("hugepagesz=", hugepagesz_setup); 4322359f2544SMike Kravetz 4323282f4214SMike Kravetz /* 4324282f4214SMike Kravetz * default_hugepagesz command line input 4325282f4214SMike Kravetz * Only one instance of default_hugepagesz allowed on command line. 4326282f4214SMike Kravetz */ 4327ae94da89SMike Kravetz static int __init default_hugepagesz_setup(char *s) 4328e11bfbfcSNick Piggin { 4329ae94da89SMike Kravetz unsigned long size; 4330b5389086SZhenguo Yao int i; 4331ae94da89SMike Kravetz 4332282f4214SMike Kravetz parsed_valid_hugepagesz = false; 4333282f4214SMike Kravetz if (parsed_default_hugepagesz) { 4334282f4214SMike Kravetz pr_err("HugeTLB: default_hugepagesz previously specified, ignoring %s\n", s); 4335f81f6e4bSPeng Liu return 1; 4336282f4214SMike Kravetz } 4337282f4214SMike Kravetz 4338282f4214SMike Kravetz size = (unsigned long)memparse(s, NULL); 4339282f4214SMike Kravetz 4340282f4214SMike Kravetz if (!arch_hugetlb_valid_size(size)) { 4341282f4214SMike Kravetz pr_err("HugeTLB: unsupported default_hugepagesz=%s\n", s); 4342f81f6e4bSPeng Liu return 1; 4343282f4214SMike Kravetz } 4344282f4214SMike Kravetz 4345282f4214SMike Kravetz hugetlb_add_hstate(ilog2(size) - PAGE_SHIFT); 4346282f4214SMike Kravetz parsed_valid_hugepagesz = true; 4347282f4214SMike Kravetz parsed_default_hugepagesz = true; 4348282f4214SMike Kravetz default_hstate_idx = hstate_index(size_to_hstate(size)); 4349282f4214SMike Kravetz 4350282f4214SMike Kravetz /* 4351282f4214SMike Kravetz * The number of default huge pages (for this size) could have been 4352282f4214SMike Kravetz * specified as the first hugetlb parameter: hugepages=X. If so, 4353282f4214SMike Kravetz * then default_hstate_max_huge_pages is set. If the default huge 4354282f4214SMike Kravetz * page size is gigantic (>= MAX_ORDER), then the pages must be 4355282f4214SMike Kravetz * allocated here from bootmem allocator. 4356282f4214SMike Kravetz */ 4357282f4214SMike Kravetz if (default_hstate_max_huge_pages) { 4358282f4214SMike Kravetz default_hstate.max_huge_pages = default_hstate_max_huge_pages; 43590a7a0f6fSPeng Liu for_each_online_node(i) 4360b5389086SZhenguo Yao default_hstate.max_huge_pages_node[i] = 4361b5389086SZhenguo Yao default_hugepages_in_node[i]; 4362282f4214SMike Kravetz if (hstate_is_gigantic(&default_hstate)) 4363282f4214SMike Kravetz hugetlb_hstate_alloc_pages(&default_hstate); 4364282f4214SMike Kravetz default_hstate_max_huge_pages = 0; 4365282f4214SMike Kravetz } 4366282f4214SMike Kravetz 4367e11bfbfcSNick Piggin return 1; 4368e11bfbfcSNick Piggin } 4369ae94da89SMike Kravetz __setup("default_hugepagesz=", default_hugepagesz_setup); 4370a3437870SNishanth Aravamudan 4371d2226ebdSFeng Tang static nodemask_t *policy_mbind_nodemask(gfp_t gfp) 4372d2226ebdSFeng Tang { 4373d2226ebdSFeng Tang #ifdef CONFIG_NUMA 4374d2226ebdSFeng Tang struct mempolicy *mpol = get_task_policy(current); 4375d2226ebdSFeng Tang 4376d2226ebdSFeng Tang /* 4377d2226ebdSFeng Tang * Only enforce MPOL_BIND policy which overlaps with cpuset policy 4378d2226ebdSFeng Tang * (from policy_nodemask) specifically for hugetlb case 4379d2226ebdSFeng Tang */ 4380d2226ebdSFeng Tang if (mpol->mode == MPOL_BIND && 4381d2226ebdSFeng Tang (apply_policy_zone(mpol, gfp_zone(gfp)) && 4382d2226ebdSFeng Tang cpuset_nodemask_valid_mems_allowed(&mpol->nodes))) 4383d2226ebdSFeng Tang return &mpol->nodes; 4384d2226ebdSFeng Tang #endif 4385d2226ebdSFeng Tang return NULL; 4386d2226ebdSFeng Tang } 4387d2226ebdSFeng Tang 43888ca39e68SMuchun Song static unsigned int allowed_mems_nr(struct hstate *h) 43898a213460SNishanth Aravamudan { 43908a213460SNishanth Aravamudan int node; 43918a213460SNishanth Aravamudan unsigned int nr = 0; 4392d2226ebdSFeng Tang nodemask_t *mbind_nodemask; 43938ca39e68SMuchun Song unsigned int *array = h->free_huge_pages_node; 43948ca39e68SMuchun Song gfp_t gfp_mask = htlb_alloc_mask(h); 43958a213460SNishanth Aravamudan 4396d2226ebdSFeng Tang mbind_nodemask = policy_mbind_nodemask(gfp_mask); 43978ca39e68SMuchun Song for_each_node_mask(node, cpuset_current_mems_allowed) { 4398d2226ebdSFeng Tang if (!mbind_nodemask || node_isset(node, *mbind_nodemask)) 43998a213460SNishanth Aravamudan nr += array[node]; 44008ca39e68SMuchun Song } 44018a213460SNishanth Aravamudan 44028a213460SNishanth Aravamudan return nr; 44038a213460SNishanth Aravamudan } 44048a213460SNishanth Aravamudan 44058a213460SNishanth Aravamudan #ifdef CONFIG_SYSCTL 440617743798SMuchun Song static int proc_hugetlb_doulongvec_minmax(struct ctl_table *table, int write, 440717743798SMuchun Song void *buffer, size_t *length, 440817743798SMuchun Song loff_t *ppos, unsigned long *out) 440917743798SMuchun Song { 441017743798SMuchun Song struct ctl_table dup_table; 441117743798SMuchun Song 441217743798SMuchun Song /* 441317743798SMuchun Song * In order to avoid races with __do_proc_doulongvec_minmax(), we 441417743798SMuchun Song * can duplicate the @table and alter the duplicate of it. 441517743798SMuchun Song */ 441617743798SMuchun Song dup_table = *table; 441717743798SMuchun Song dup_table.data = out; 441817743798SMuchun Song 441917743798SMuchun Song return proc_doulongvec_minmax(&dup_table, write, buffer, length, ppos); 442017743798SMuchun Song } 442117743798SMuchun Song 442206808b08SLee Schermerhorn static int hugetlb_sysctl_handler_common(bool obey_mempolicy, 442306808b08SLee Schermerhorn struct ctl_table *table, int write, 442432927393SChristoph Hellwig void *buffer, size_t *length, loff_t *ppos) 44251da177e4SLinus Torvalds { 4426e5ff2159SAndi Kleen struct hstate *h = &default_hstate; 4427238d3c13SDavid Rientjes unsigned long tmp = h->max_huge_pages; 442808d4a246SMichal Hocko int ret; 4429e5ff2159SAndi Kleen 4430457c1b27SNishanth Aravamudan if (!hugepages_supported()) 443186613628SJan Stancek return -EOPNOTSUPP; 4432457c1b27SNishanth Aravamudan 443317743798SMuchun Song ret = proc_hugetlb_doulongvec_minmax(table, write, buffer, length, ppos, 443417743798SMuchun Song &tmp); 443508d4a246SMichal Hocko if (ret) 443608d4a246SMichal Hocko goto out; 4437e5ff2159SAndi Kleen 4438238d3c13SDavid Rientjes if (write) 4439238d3c13SDavid Rientjes ret = __nr_hugepages_store_common(obey_mempolicy, h, 4440238d3c13SDavid Rientjes NUMA_NO_NODE, tmp, *length); 444108d4a246SMichal Hocko out: 444208d4a246SMichal Hocko return ret; 44431da177e4SLinus Torvalds } 4444396faf03SMel Gorman 444506808b08SLee Schermerhorn int hugetlb_sysctl_handler(struct ctl_table *table, int write, 444632927393SChristoph Hellwig void *buffer, size_t *length, loff_t *ppos) 444706808b08SLee Schermerhorn { 444806808b08SLee Schermerhorn 444906808b08SLee Schermerhorn return hugetlb_sysctl_handler_common(false, table, write, 445006808b08SLee Schermerhorn buffer, length, ppos); 445106808b08SLee Schermerhorn } 445206808b08SLee Schermerhorn 445306808b08SLee Schermerhorn #ifdef CONFIG_NUMA 445406808b08SLee Schermerhorn int hugetlb_mempolicy_sysctl_handler(struct ctl_table *table, int write, 445532927393SChristoph Hellwig void *buffer, size_t *length, loff_t *ppos) 445606808b08SLee Schermerhorn { 445706808b08SLee Schermerhorn return hugetlb_sysctl_handler_common(true, table, write, 445806808b08SLee Schermerhorn buffer, length, ppos); 445906808b08SLee Schermerhorn } 446006808b08SLee Schermerhorn #endif /* CONFIG_NUMA */ 446106808b08SLee Schermerhorn 4462a3d0c6aaSNishanth Aravamudan int hugetlb_overcommit_handler(struct ctl_table *table, int write, 446332927393SChristoph Hellwig void *buffer, size_t *length, loff_t *ppos) 4464a3d0c6aaSNishanth Aravamudan { 4465a5516438SAndi Kleen struct hstate *h = &default_hstate; 4466e5ff2159SAndi Kleen unsigned long tmp; 446708d4a246SMichal Hocko int ret; 4468e5ff2159SAndi Kleen 4469457c1b27SNishanth Aravamudan if (!hugepages_supported()) 447086613628SJan Stancek return -EOPNOTSUPP; 4471457c1b27SNishanth Aravamudan 4472e5ff2159SAndi Kleen tmp = h->nr_overcommit_huge_pages; 4473e5ff2159SAndi Kleen 4474bae7f4aeSLuiz Capitulino if (write && hstate_is_gigantic(h)) 4475adbe8726SEric B Munson return -EINVAL; 4476adbe8726SEric B Munson 447717743798SMuchun Song ret = proc_hugetlb_doulongvec_minmax(table, write, buffer, length, ppos, 447817743798SMuchun Song &tmp); 447908d4a246SMichal Hocko if (ret) 448008d4a246SMichal Hocko goto out; 4481e5ff2159SAndi Kleen 4482e5ff2159SAndi Kleen if (write) { 4483db71ef79SMike Kravetz spin_lock_irq(&hugetlb_lock); 4484e5ff2159SAndi Kleen h->nr_overcommit_huge_pages = tmp; 4485db71ef79SMike Kravetz spin_unlock_irq(&hugetlb_lock); 4486e5ff2159SAndi Kleen } 448708d4a246SMichal Hocko out: 448808d4a246SMichal Hocko return ret; 4489a3d0c6aaSNishanth Aravamudan } 4490a3d0c6aaSNishanth Aravamudan 44911da177e4SLinus Torvalds #endif /* CONFIG_SYSCTL */ 44921da177e4SLinus Torvalds 4493e1759c21SAlexey Dobriyan void hugetlb_report_meminfo(struct seq_file *m) 44941da177e4SLinus Torvalds { 4495fcb2b0c5SRoman Gushchin struct hstate *h; 4496fcb2b0c5SRoman Gushchin unsigned long total = 0; 4497fcb2b0c5SRoman Gushchin 4498457c1b27SNishanth Aravamudan if (!hugepages_supported()) 4499457c1b27SNishanth Aravamudan return; 4500fcb2b0c5SRoman Gushchin 4501fcb2b0c5SRoman Gushchin for_each_hstate(h) { 4502fcb2b0c5SRoman Gushchin unsigned long count = h->nr_huge_pages; 4503fcb2b0c5SRoman Gushchin 4504aca78307SMiaohe Lin total += huge_page_size(h) * count; 4505fcb2b0c5SRoman Gushchin 4506fcb2b0c5SRoman Gushchin if (h == &default_hstate) 4507e1759c21SAlexey Dobriyan seq_printf(m, 45081da177e4SLinus Torvalds "HugePages_Total: %5lu\n" 45091da177e4SLinus Torvalds "HugePages_Free: %5lu\n" 4510b45b5bd6SDavid Gibson "HugePages_Rsvd: %5lu\n" 45117893d1d5SAdam Litke "HugePages_Surp: %5lu\n" 45124f98a2feSRik van Riel "Hugepagesize: %8lu kB\n", 4513fcb2b0c5SRoman Gushchin count, 4514a5516438SAndi Kleen h->free_huge_pages, 4515a5516438SAndi Kleen h->resv_huge_pages, 4516a5516438SAndi Kleen h->surplus_huge_pages, 4517aca78307SMiaohe Lin huge_page_size(h) / SZ_1K); 4518fcb2b0c5SRoman Gushchin } 4519fcb2b0c5SRoman Gushchin 4520aca78307SMiaohe Lin seq_printf(m, "Hugetlb: %8lu kB\n", total / SZ_1K); 45211da177e4SLinus Torvalds } 45221da177e4SLinus Torvalds 45237981593bSJoe Perches int hugetlb_report_node_meminfo(char *buf, int len, int nid) 45241da177e4SLinus Torvalds { 4525a5516438SAndi Kleen struct hstate *h = &default_hstate; 45267981593bSJoe Perches 4527457c1b27SNishanth Aravamudan if (!hugepages_supported()) 4528457c1b27SNishanth Aravamudan return 0; 45297981593bSJoe Perches 45307981593bSJoe Perches return sysfs_emit_at(buf, len, 45311da177e4SLinus Torvalds "Node %d HugePages_Total: %5u\n" 4532a1de0919SNishanth Aravamudan "Node %d HugePages_Free: %5u\n" 4533a1de0919SNishanth Aravamudan "Node %d HugePages_Surp: %5u\n", 4534a5516438SAndi Kleen nid, h->nr_huge_pages_node[nid], 4535a5516438SAndi Kleen nid, h->free_huge_pages_node[nid], 4536a5516438SAndi Kleen nid, h->surplus_huge_pages_node[nid]); 45371da177e4SLinus Torvalds } 45381da177e4SLinus Torvalds 4539dcadcf1cSGang Li void hugetlb_show_meminfo_node(int nid) 4540949f7ec5SDavid Rientjes { 4541949f7ec5SDavid Rientjes struct hstate *h; 4542949f7ec5SDavid Rientjes 4543457c1b27SNishanth Aravamudan if (!hugepages_supported()) 4544457c1b27SNishanth Aravamudan return; 4545457c1b27SNishanth Aravamudan 4546949f7ec5SDavid Rientjes for_each_hstate(h) 4547dcadcf1cSGang Li printk("Node %d hugepages_total=%u hugepages_free=%u hugepages_surp=%u hugepages_size=%lukB\n", 4548949f7ec5SDavid Rientjes nid, 4549949f7ec5SDavid Rientjes h->nr_huge_pages_node[nid], 4550949f7ec5SDavid Rientjes h->free_huge_pages_node[nid], 4551949f7ec5SDavid Rientjes h->surplus_huge_pages_node[nid], 4552aca78307SMiaohe Lin huge_page_size(h) / SZ_1K); 4553949f7ec5SDavid Rientjes } 4554949f7ec5SDavid Rientjes 45555d317b2bSNaoya Horiguchi void hugetlb_report_usage(struct seq_file *m, struct mm_struct *mm) 45565d317b2bSNaoya Horiguchi { 45575d317b2bSNaoya Horiguchi seq_printf(m, "HugetlbPages:\t%8lu kB\n", 45585d317b2bSNaoya Horiguchi atomic_long_read(&mm->hugetlb_usage) << (PAGE_SHIFT - 10)); 45595d317b2bSNaoya Horiguchi } 45605d317b2bSNaoya Horiguchi 45611da177e4SLinus Torvalds /* Return the number pages of memory we physically have, in PAGE_SIZE units. */ 45621da177e4SLinus Torvalds unsigned long hugetlb_total_pages(void) 45631da177e4SLinus Torvalds { 4564d0028588SWanpeng Li struct hstate *h; 4565d0028588SWanpeng Li unsigned long nr_total_pages = 0; 4566d0028588SWanpeng Li 4567d0028588SWanpeng Li for_each_hstate(h) 4568d0028588SWanpeng Li nr_total_pages += h->nr_huge_pages * pages_per_huge_page(h); 4569d0028588SWanpeng Li return nr_total_pages; 45701da177e4SLinus Torvalds } 45711da177e4SLinus Torvalds 4572a5516438SAndi Kleen static int hugetlb_acct_memory(struct hstate *h, long delta) 4573fc1b8a73SMel Gorman { 4574fc1b8a73SMel Gorman int ret = -ENOMEM; 4575fc1b8a73SMel Gorman 45760aa7f354SMiaohe Lin if (!delta) 45770aa7f354SMiaohe Lin return 0; 45780aa7f354SMiaohe Lin 4579db71ef79SMike Kravetz spin_lock_irq(&hugetlb_lock); 4580fc1b8a73SMel Gorman /* 4581fc1b8a73SMel Gorman * When cpuset is configured, it breaks the strict hugetlb page 4582fc1b8a73SMel Gorman * reservation as the accounting is done on a global variable. Such 4583fc1b8a73SMel Gorman * reservation is completely rubbish in the presence of cpuset because 4584fc1b8a73SMel Gorman * the reservation is not checked against page availability for the 4585fc1b8a73SMel Gorman * current cpuset. Application can still potentially OOM'ed by kernel 4586fc1b8a73SMel Gorman * with lack of free htlb page in cpuset that the task is in. 4587fc1b8a73SMel Gorman * Attempt to enforce strict accounting with cpuset is almost 4588fc1b8a73SMel Gorman * impossible (or too ugly) because cpuset is too fluid that 4589fc1b8a73SMel Gorman * task or memory node can be dynamically moved between cpusets. 4590fc1b8a73SMel Gorman * 4591fc1b8a73SMel Gorman * The change of semantics for shared hugetlb mapping with cpuset is 4592fc1b8a73SMel Gorman * undesirable. However, in order to preserve some of the semantics, 4593fc1b8a73SMel Gorman * we fall back to check against current free page availability as 4594fc1b8a73SMel Gorman * a best attempt and hopefully to minimize the impact of changing 4595fc1b8a73SMel Gorman * semantics that cpuset has. 45968ca39e68SMuchun Song * 45978ca39e68SMuchun Song * Apart from cpuset, we also have memory policy mechanism that 45988ca39e68SMuchun Song * also determines from which node the kernel will allocate memory 45998ca39e68SMuchun Song * in a NUMA system. So similar to cpuset, we also should consider 46008ca39e68SMuchun Song * the memory policy of the current task. Similar to the description 46018ca39e68SMuchun Song * above. 4602fc1b8a73SMel Gorman */ 4603fc1b8a73SMel Gorman if (delta > 0) { 4604a5516438SAndi Kleen if (gather_surplus_pages(h, delta) < 0) 4605fc1b8a73SMel Gorman goto out; 4606fc1b8a73SMel Gorman 46078ca39e68SMuchun Song if (delta > allowed_mems_nr(h)) { 4608a5516438SAndi Kleen return_unused_surplus_pages(h, delta); 4609fc1b8a73SMel Gorman goto out; 4610fc1b8a73SMel Gorman } 4611fc1b8a73SMel Gorman } 4612fc1b8a73SMel Gorman 4613fc1b8a73SMel Gorman ret = 0; 4614fc1b8a73SMel Gorman if (delta < 0) 4615a5516438SAndi Kleen return_unused_surplus_pages(h, (unsigned long) -delta); 4616fc1b8a73SMel Gorman 4617fc1b8a73SMel Gorman out: 4618db71ef79SMike Kravetz spin_unlock_irq(&hugetlb_lock); 4619fc1b8a73SMel Gorman return ret; 4620fc1b8a73SMel Gorman } 4621fc1b8a73SMel Gorman 462284afd99bSAndy Whitcroft static void hugetlb_vm_op_open(struct vm_area_struct *vma) 462384afd99bSAndy Whitcroft { 4624f522c3acSJoonsoo Kim struct resv_map *resv = vma_resv_map(vma); 462584afd99bSAndy Whitcroft 462684afd99bSAndy Whitcroft /* 4627612b8a31SMike Kravetz * HPAGE_RESV_OWNER indicates a private mapping. 462884afd99bSAndy Whitcroft * This new VMA should share its siblings reservation map if present. 462984afd99bSAndy Whitcroft * The VMA will only ever have a valid reservation map pointer where 463084afd99bSAndy Whitcroft * it is being copied for another still existing VMA. As that VMA 463125985edcSLucas De Marchi * has a reference to the reservation map it cannot disappear until 463284afd99bSAndy Whitcroft * after this open call completes. It is therefore safe to take a 463384afd99bSAndy Whitcroft * new reference here without additional locking. 463484afd99bSAndy Whitcroft */ 463509a26e83SMike Kravetz if (resv && is_vma_resv_set(vma, HPAGE_RESV_OWNER)) { 463609a26e83SMike Kravetz resv_map_dup_hugetlb_cgroup_uncharge_info(resv); 4637f522c3acSJoonsoo Kim kref_get(&resv->refs); 463884afd99bSAndy Whitcroft } 46398d9bfb26SMike Kravetz 4640131a79b4SMike Kravetz /* 4641131a79b4SMike Kravetz * vma_lock structure for sharable mappings is vma specific. 4642612b8a31SMike Kravetz * Clear old pointer (if copied via vm_area_dup) and allocate 4643612b8a31SMike Kravetz * new structure. Before clearing, make sure vma_lock is not 4644612b8a31SMike Kravetz * for this vma. 4645131a79b4SMike Kravetz */ 4646131a79b4SMike Kravetz if (vma->vm_flags & VM_MAYSHARE) { 4647612b8a31SMike Kravetz struct hugetlb_vma_lock *vma_lock = vma->vm_private_data; 4648612b8a31SMike Kravetz 4649612b8a31SMike Kravetz if (vma_lock) { 4650612b8a31SMike Kravetz if (vma_lock->vma != vma) { 4651131a79b4SMike Kravetz vma->vm_private_data = NULL; 46528d9bfb26SMike Kravetz hugetlb_vma_lock_alloc(vma); 4653612b8a31SMike Kravetz } else 4654612b8a31SMike Kravetz pr_warn("HugeTLB: vma_lock already exists in %s.\n", __func__); 4655612b8a31SMike Kravetz } else 4656612b8a31SMike Kravetz hugetlb_vma_lock_alloc(vma); 465709a26e83SMike Kravetz } 4658131a79b4SMike Kravetz } 465984afd99bSAndy Whitcroft 4660a1e78772SMel Gorman static void hugetlb_vm_op_close(struct vm_area_struct *vma) 4661a1e78772SMel Gorman { 4662a5516438SAndi Kleen struct hstate *h = hstate_vma(vma); 46638d9bfb26SMike Kravetz struct resv_map *resv; 466490481622SDavid Gibson struct hugepage_subpool *spool = subpool_vma(vma); 46654e35f483SJoonsoo Kim unsigned long reserve, start, end; 46661c5ecae3SMike Kravetz long gbl_reserve; 466784afd99bSAndy Whitcroft 46688d9bfb26SMike Kravetz hugetlb_vma_lock_free(vma); 46698d9bfb26SMike Kravetz 46708d9bfb26SMike Kravetz resv = vma_resv_map(vma); 46714e35f483SJoonsoo Kim if (!resv || !is_vma_resv_set(vma, HPAGE_RESV_OWNER)) 46724e35f483SJoonsoo Kim return; 46734e35f483SJoonsoo Kim 4674a5516438SAndi Kleen start = vma_hugecache_offset(h, vma, vma->vm_start); 4675a5516438SAndi Kleen end = vma_hugecache_offset(h, vma, vma->vm_end); 467684afd99bSAndy Whitcroft 46774e35f483SJoonsoo Kim reserve = (end - start) - region_count(resv, start, end); 4678e9fe92aeSMina Almasry hugetlb_cgroup_uncharge_counter(resv, start, end); 46797251ff78SAdam Litke if (reserve) { 46801c5ecae3SMike Kravetz /* 46811c5ecae3SMike Kravetz * Decrement reserve counts. The global reserve count may be 46821c5ecae3SMike Kravetz * adjusted if the subpool has a minimum size. 46831c5ecae3SMike Kravetz */ 46841c5ecae3SMike Kravetz gbl_reserve = hugepage_subpool_put_pages(spool, reserve); 46851c5ecae3SMike Kravetz hugetlb_acct_memory(h, -gbl_reserve); 46867251ff78SAdam Litke } 4687e9fe92aeSMina Almasry 4688e9fe92aeSMina Almasry kref_put(&resv->refs, resv_map_release); 4689a1e78772SMel Gorman } 4690a1e78772SMel Gorman 469131383c68SDan Williams static int hugetlb_vm_op_split(struct vm_area_struct *vma, unsigned long addr) 469231383c68SDan Williams { 469331383c68SDan Williams if (addr & ~(huge_page_mask(hstate_vma(vma)))) 469431383c68SDan Williams return -EINVAL; 469531383c68SDan Williams return 0; 469631383c68SDan Williams } 469731383c68SDan Williams 469805ea8860SDan Williams static unsigned long hugetlb_vm_op_pagesize(struct vm_area_struct *vma) 469905ea8860SDan Williams { 4700aca78307SMiaohe Lin return huge_page_size(hstate_vma(vma)); 470105ea8860SDan Williams } 470205ea8860SDan Williams 47031da177e4SLinus Torvalds /* 47041da177e4SLinus Torvalds * We cannot handle pagefaults against hugetlb pages at all. They cause 47051da177e4SLinus Torvalds * handle_mm_fault() to try to instantiate regular-sized pages in the 47066c26d310SMiaohe Lin * hugepage VMA. do_page_fault() is supposed to trap this, so BUG is we get 47071da177e4SLinus Torvalds * this far. 47081da177e4SLinus Torvalds */ 4709b3ec9f33SSouptick Joarder static vm_fault_t hugetlb_vm_op_fault(struct vm_fault *vmf) 47101da177e4SLinus Torvalds { 47111da177e4SLinus Torvalds BUG(); 4712d0217ac0SNick Piggin return 0; 47131da177e4SLinus Torvalds } 47141da177e4SLinus Torvalds 4715eec3636aSJane Chu /* 4716eec3636aSJane Chu * When a new function is introduced to vm_operations_struct and added 4717eec3636aSJane Chu * to hugetlb_vm_ops, please consider adding the function to shm_vm_ops. 4718eec3636aSJane Chu * This is because under System V memory model, mappings created via 4719eec3636aSJane Chu * shmget/shmat with "huge page" specified are backed by hugetlbfs files, 4720eec3636aSJane Chu * their original vm_ops are overwritten with shm_vm_ops. 4721eec3636aSJane Chu */ 4722f0f37e2fSAlexey Dobriyan const struct vm_operations_struct hugetlb_vm_ops = { 4723d0217ac0SNick Piggin .fault = hugetlb_vm_op_fault, 472484afd99bSAndy Whitcroft .open = hugetlb_vm_op_open, 4725a1e78772SMel Gorman .close = hugetlb_vm_op_close, 4726dd3b614fSDmitry Safonov .may_split = hugetlb_vm_op_split, 472705ea8860SDan Williams .pagesize = hugetlb_vm_op_pagesize, 47281da177e4SLinus Torvalds }; 47291da177e4SLinus Torvalds 47301e8f889bSDavid Gibson static pte_t make_huge_pte(struct vm_area_struct *vma, struct page *page, 47311e8f889bSDavid Gibson int writable) 473263551ae0SDavid Gibson { 473363551ae0SDavid Gibson pte_t entry; 473479c1c594SChristophe Leroy unsigned int shift = huge_page_shift(hstate_vma(vma)); 473563551ae0SDavid Gibson 47361e8f889bSDavid Gibson if (writable) { 4737106c992aSGerald Schaefer entry = huge_pte_mkwrite(huge_pte_mkdirty(mk_huge_pte(page, 4738106c992aSGerald Schaefer vma->vm_page_prot))); 473963551ae0SDavid Gibson } else { 4740106c992aSGerald Schaefer entry = huge_pte_wrprotect(mk_huge_pte(page, 4741106c992aSGerald Schaefer vma->vm_page_prot)); 474263551ae0SDavid Gibson } 474363551ae0SDavid Gibson entry = pte_mkyoung(entry); 474479c1c594SChristophe Leroy entry = arch_make_huge_pte(entry, shift, vma->vm_flags); 474563551ae0SDavid Gibson 474663551ae0SDavid Gibson return entry; 474763551ae0SDavid Gibson } 474863551ae0SDavid Gibson 47491e8f889bSDavid Gibson static void set_huge_ptep_writable(struct vm_area_struct *vma, 47501e8f889bSDavid Gibson unsigned long address, pte_t *ptep) 47511e8f889bSDavid Gibson { 47521e8f889bSDavid Gibson pte_t entry; 47531e8f889bSDavid Gibson 4754106c992aSGerald Schaefer entry = huge_pte_mkwrite(huge_pte_mkdirty(huge_ptep_get(ptep))); 475532f84528SChris Forbes if (huge_ptep_set_access_flags(vma, address, ptep, entry, 1)) 47564b3073e1SRussell King update_mmu_cache(vma, address, ptep); 47571e8f889bSDavid Gibson } 47581e8f889bSDavid Gibson 4759d5ed7444SAneesh Kumar K.V bool is_hugetlb_entry_migration(pte_t pte) 47604a705fefSNaoya Horiguchi { 47614a705fefSNaoya Horiguchi swp_entry_t swp; 47624a705fefSNaoya Horiguchi 47634a705fefSNaoya Horiguchi if (huge_pte_none(pte) || pte_present(pte)) 4764d5ed7444SAneesh Kumar K.V return false; 47654a705fefSNaoya Horiguchi swp = pte_to_swp_entry(pte); 4766d79d176aSBaoquan He if (is_migration_entry(swp)) 4767d5ed7444SAneesh Kumar K.V return true; 47684a705fefSNaoya Horiguchi else 4769d5ed7444SAneesh Kumar K.V return false; 47704a705fefSNaoya Horiguchi } 47714a705fefSNaoya Horiguchi 47723e5c3600SBaoquan He static bool is_hugetlb_entry_hwpoisoned(pte_t pte) 47734a705fefSNaoya Horiguchi { 47744a705fefSNaoya Horiguchi swp_entry_t swp; 47754a705fefSNaoya Horiguchi 47764a705fefSNaoya Horiguchi if (huge_pte_none(pte) || pte_present(pte)) 47773e5c3600SBaoquan He return false; 47784a705fefSNaoya Horiguchi swp = pte_to_swp_entry(pte); 4779d79d176aSBaoquan He if (is_hwpoison_entry(swp)) 47803e5c3600SBaoquan He return true; 47814a705fefSNaoya Horiguchi else 47823e5c3600SBaoquan He return false; 47834a705fefSNaoya Horiguchi } 47841e8f889bSDavid Gibson 47854eae4efaSPeter Xu static void 47864eae4efaSPeter Xu hugetlb_install_page(struct vm_area_struct *vma, pte_t *ptep, unsigned long addr, 47874eae4efaSPeter Xu struct page *new_page) 47884eae4efaSPeter Xu { 47894eae4efaSPeter Xu __SetPageUptodate(new_page); 47904eae4efaSPeter Xu hugepage_add_new_anon_rmap(new_page, vma, addr); 47911eba86c0SPasha Tatashin set_huge_pte_at(vma->vm_mm, addr, ptep, make_huge_pte(vma, new_page, 1)); 47924eae4efaSPeter Xu hugetlb_count_add(pages_per_huge_page(hstate_vma(vma)), vma->vm_mm); 47934eae4efaSPeter Xu SetHPageMigratable(new_page); 47944eae4efaSPeter Xu } 47954eae4efaSPeter Xu 479663551ae0SDavid Gibson int copy_hugetlb_page_range(struct mm_struct *dst, struct mm_struct *src, 4797bc70fbf2SPeter Xu struct vm_area_struct *dst_vma, 4798bc70fbf2SPeter Xu struct vm_area_struct *src_vma) 479963551ae0SDavid Gibson { 48003aa4ed80SMiaohe Lin pte_t *src_pte, *dst_pte, entry; 480163551ae0SDavid Gibson struct page *ptepage; 48021c59827dSHugh Dickins unsigned long addr; 4803bc70fbf2SPeter Xu bool cow = is_cow_mapping(src_vma->vm_flags); 4804bc70fbf2SPeter Xu struct hstate *h = hstate_vma(src_vma); 4805a5516438SAndi Kleen unsigned long sz = huge_page_size(h); 48064eae4efaSPeter Xu unsigned long npages = pages_per_huge_page(h); 4807ac46d4f3SJérôme Glisse struct mmu_notifier_range range; 4808e95a9851SMike Kravetz unsigned long last_addr_mask; 4809e8569dd2SAndreas Sandberg int ret = 0; 48101e8f889bSDavid Gibson 4811ac46d4f3SJérôme Glisse if (cow) { 4812bc70fbf2SPeter Xu mmu_notifier_range_init(&range, MMU_NOTIFY_CLEAR, 0, src_vma, src, 4813bc70fbf2SPeter Xu src_vma->vm_start, 4814bc70fbf2SPeter Xu src_vma->vm_end); 4815ac46d4f3SJérôme Glisse mmu_notifier_invalidate_range_start(&range); 4816623a1ddfSDavid Hildenbrand mmap_assert_write_locked(src); 4817623a1ddfSDavid Hildenbrand raw_write_seqcount_begin(&src->write_protect_seq); 481840549ba8SMike Kravetz } else { 481940549ba8SMike Kravetz /* 482040549ba8SMike Kravetz * For shared mappings the vma lock must be held before 482140549ba8SMike Kravetz * calling huge_pte_offset in the src vma. Otherwise, the 482240549ba8SMike Kravetz * returned ptep could go away if part of a shared pmd and 482340549ba8SMike Kravetz * another thread calls huge_pmd_unshare. 482440549ba8SMike Kravetz */ 482540549ba8SMike Kravetz hugetlb_vma_lock_read(src_vma); 4826ac46d4f3SJérôme Glisse } 4827e8569dd2SAndreas Sandberg 4828e95a9851SMike Kravetz last_addr_mask = hugetlb_mask_last_page(h); 4829bc70fbf2SPeter Xu for (addr = src_vma->vm_start; addr < src_vma->vm_end; addr += sz) { 4830cb900f41SKirill A. Shutemov spinlock_t *src_ptl, *dst_ptl; 48317868a208SPunit Agrawal src_pte = huge_pte_offset(src, addr, sz); 4832e95a9851SMike Kravetz if (!src_pte) { 4833e95a9851SMike Kravetz addr |= last_addr_mask; 4834c74df32cSHugh Dickins continue; 4835e95a9851SMike Kravetz } 4836bc70fbf2SPeter Xu dst_pte = huge_pte_alloc(dst, dst_vma, addr, sz); 4837e8569dd2SAndreas Sandberg if (!dst_pte) { 4838e8569dd2SAndreas Sandberg ret = -ENOMEM; 4839e8569dd2SAndreas Sandberg break; 4840e8569dd2SAndreas Sandberg } 4841c5c99429SLarry Woodman 48425e41540cSMike Kravetz /* 48435e41540cSMike Kravetz * If the pagetables are shared don't copy or take references. 48445e41540cSMike Kravetz * 48453aa4ed80SMiaohe Lin * dst_pte == src_pte is the common case of src/dest sharing. 48465e41540cSMike Kravetz * However, src could have 'unshared' and dst shares with 48473aa4ed80SMiaohe Lin * another vma. So page_count of ptep page is checked instead 48483aa4ed80SMiaohe Lin * to reliably determine whether pte is shared. 48495e41540cSMike Kravetz */ 48503aa4ed80SMiaohe Lin if (page_count(virt_to_page(dst_pte)) > 1) { 4851e95a9851SMike Kravetz addr |= last_addr_mask; 4852c5c99429SLarry Woodman continue; 4853e95a9851SMike Kravetz } 4854c5c99429SLarry Woodman 4855cb900f41SKirill A. Shutemov dst_ptl = huge_pte_lock(h, dst, dst_pte); 4856cb900f41SKirill A. Shutemov src_ptl = huge_pte_lockptr(h, src, src_pte); 4857cb900f41SKirill A. Shutemov spin_lock_nested(src_ptl, SINGLE_DEPTH_NESTING); 48584a705fefSNaoya Horiguchi entry = huge_ptep_get(src_pte); 48594eae4efaSPeter Xu again: 48603aa4ed80SMiaohe Lin if (huge_pte_none(entry)) { 48615e41540cSMike Kravetz /* 48623aa4ed80SMiaohe Lin * Skip if src entry none. 48635e41540cSMike Kravetz */ 48644a705fefSNaoya Horiguchi ; 4865c2cb0dccSNaoya Horiguchi } else if (unlikely(is_hugetlb_entry_hwpoisoned(entry))) { 4866c2cb0dccSNaoya Horiguchi bool uffd_wp = huge_pte_uffd_wp(entry); 4867c2cb0dccSNaoya Horiguchi 4868c2cb0dccSNaoya Horiguchi if (!userfaultfd_wp(dst_vma) && uffd_wp) 4869c2cb0dccSNaoya Horiguchi entry = huge_pte_clear_uffd_wp(entry); 4870c2cb0dccSNaoya Horiguchi set_huge_pte_at(dst, addr, dst_pte, entry); 4871c2cb0dccSNaoya Horiguchi } else if (unlikely(is_hugetlb_entry_migration(entry))) { 48724a705fefSNaoya Horiguchi swp_entry_t swp_entry = pte_to_swp_entry(entry); 4873bc70fbf2SPeter Xu bool uffd_wp = huge_pte_uffd_wp(entry); 48744a705fefSNaoya Horiguchi 48756c287605SDavid Hildenbrand if (!is_readable_migration_entry(swp_entry) && cow) { 48764a705fefSNaoya Horiguchi /* 48774a705fefSNaoya Horiguchi * COW mappings require pages in both 48784a705fefSNaoya Horiguchi * parent and child to be set to read. 48794a705fefSNaoya Horiguchi */ 48804dd845b5SAlistair Popple swp_entry = make_readable_migration_entry( 48814dd845b5SAlistair Popple swp_offset(swp_entry)); 48824a705fefSNaoya Horiguchi entry = swp_entry_to_pte(swp_entry); 4883bc70fbf2SPeter Xu if (userfaultfd_wp(src_vma) && uffd_wp) 4884bc70fbf2SPeter Xu entry = huge_pte_mkuffd_wp(entry); 488518f39629SQi Zheng set_huge_pte_at(src, addr, src_pte, entry); 48864a705fefSNaoya Horiguchi } 4887bc70fbf2SPeter Xu if (!userfaultfd_wp(dst_vma) && uffd_wp) 4888bc70fbf2SPeter Xu entry = huge_pte_clear_uffd_wp(entry); 488918f39629SQi Zheng set_huge_pte_at(dst, addr, dst_pte, entry); 4890bc70fbf2SPeter Xu } else if (unlikely(is_pte_marker(entry))) { 4891bc70fbf2SPeter Xu /* 4892bc70fbf2SPeter Xu * We copy the pte marker only if the dst vma has 4893bc70fbf2SPeter Xu * uffd-wp enabled. 4894bc70fbf2SPeter Xu */ 4895bc70fbf2SPeter Xu if (userfaultfd_wp(dst_vma)) 4896bc70fbf2SPeter Xu set_huge_pte_at(dst, addr, dst_pte, entry); 48974a705fefSNaoya Horiguchi } else { 48984eae4efaSPeter Xu entry = huge_ptep_get(src_pte); 48994eae4efaSPeter Xu ptepage = pte_page(entry); 49004eae4efaSPeter Xu get_page(ptepage); 49014eae4efaSPeter Xu 49024eae4efaSPeter Xu /* 4903fb3d824dSDavid Hildenbrand * Failing to duplicate the anon rmap is a rare case 4904fb3d824dSDavid Hildenbrand * where we see pinned hugetlb pages while they're 4905fb3d824dSDavid Hildenbrand * prone to COW. We need to do the COW earlier during 4906fb3d824dSDavid Hildenbrand * fork. 49074eae4efaSPeter Xu * 49084eae4efaSPeter Xu * When pre-allocating the page or copying data, we 49094eae4efaSPeter Xu * need to be without the pgtable locks since we could 49104eae4efaSPeter Xu * sleep during the process. 49114eae4efaSPeter Xu */ 4912fb3d824dSDavid Hildenbrand if (!PageAnon(ptepage)) { 4913fb3d824dSDavid Hildenbrand page_dup_file_rmap(ptepage, true); 4914bc70fbf2SPeter Xu } else if (page_try_dup_anon_rmap(ptepage, true, 4915bc70fbf2SPeter Xu src_vma)) { 49164eae4efaSPeter Xu pte_t src_pte_old = entry; 49174eae4efaSPeter Xu struct page *new; 49184eae4efaSPeter Xu 49194eae4efaSPeter Xu spin_unlock(src_ptl); 49204eae4efaSPeter Xu spin_unlock(dst_ptl); 49214eae4efaSPeter Xu /* Do not use reserve as it's private owned */ 4922bc70fbf2SPeter Xu new = alloc_huge_page(dst_vma, addr, 1); 49234eae4efaSPeter Xu if (IS_ERR(new)) { 49244eae4efaSPeter Xu put_page(ptepage); 49254eae4efaSPeter Xu ret = PTR_ERR(new); 49264eae4efaSPeter Xu break; 49274eae4efaSPeter Xu } 4928bc70fbf2SPeter Xu copy_user_huge_page(new, ptepage, addr, dst_vma, 49294eae4efaSPeter Xu npages); 49304eae4efaSPeter Xu put_page(ptepage); 49314eae4efaSPeter Xu 49324eae4efaSPeter Xu /* Install the new huge page if src pte stable */ 49334eae4efaSPeter Xu dst_ptl = huge_pte_lock(h, dst, dst_pte); 49344eae4efaSPeter Xu src_ptl = huge_pte_lockptr(h, src, src_pte); 49354eae4efaSPeter Xu spin_lock_nested(src_ptl, SINGLE_DEPTH_NESTING); 49364eae4efaSPeter Xu entry = huge_ptep_get(src_pte); 49374eae4efaSPeter Xu if (!pte_same(src_pte_old, entry)) { 4938bc70fbf2SPeter Xu restore_reserve_on_error(h, dst_vma, addr, 4939846be085SMike Kravetz new); 49404eae4efaSPeter Xu put_page(new); 49413aa4ed80SMiaohe Lin /* huge_ptep of dst_pte won't change as in child */ 49424eae4efaSPeter Xu goto again; 49434eae4efaSPeter Xu } 4944bc70fbf2SPeter Xu hugetlb_install_page(dst_vma, dst_pte, addr, new); 49454eae4efaSPeter Xu spin_unlock(src_ptl); 49464eae4efaSPeter Xu spin_unlock(dst_ptl); 49474eae4efaSPeter Xu continue; 49484eae4efaSPeter Xu } 49494eae4efaSPeter Xu 495034ee645eSJoerg Roedel if (cow) { 49510f10851eSJérôme Glisse /* 49520f10851eSJérôme Glisse * No need to notify as we are downgrading page 49530f10851eSJérôme Glisse * table protection not changing it to point 49540f10851eSJérôme Glisse * to a new page. 49550f10851eSJérôme Glisse * 4956ee65728eSMike Rapoport * See Documentation/mm/mmu_notifier.rst 49570f10851eSJérôme Glisse */ 49587f2e9525SGerald Schaefer huge_ptep_set_wrprotect(src, addr, src_pte); 495984894e1cSPeter Xu entry = huge_pte_wrprotect(entry); 496034ee645eSJoerg Roedel } 49614eae4efaSPeter Xu 496263551ae0SDavid Gibson set_huge_pte_at(dst, addr, dst_pte, entry); 49634eae4efaSPeter Xu hugetlb_count_add(npages, dst); 49641c59827dSHugh Dickins } 4965cb900f41SKirill A. Shutemov spin_unlock(src_ptl); 4966cb900f41SKirill A. Shutemov spin_unlock(dst_ptl); 496763551ae0SDavid Gibson } 496863551ae0SDavid Gibson 4969623a1ddfSDavid Hildenbrand if (cow) { 4970623a1ddfSDavid Hildenbrand raw_write_seqcount_end(&src->write_protect_seq); 4971ac46d4f3SJérôme Glisse mmu_notifier_invalidate_range_end(&range); 497240549ba8SMike Kravetz } else { 497340549ba8SMike Kravetz hugetlb_vma_unlock_read(src_vma); 4974623a1ddfSDavid Hildenbrand } 4975e8569dd2SAndreas Sandberg 4976e8569dd2SAndreas Sandberg return ret; 497763551ae0SDavid Gibson } 497863551ae0SDavid Gibson 4979550a7d60SMina Almasry static void move_huge_pte(struct vm_area_struct *vma, unsigned long old_addr, 4980db110a99SAneesh Kumar K.V unsigned long new_addr, pte_t *src_pte, pte_t *dst_pte) 4981550a7d60SMina Almasry { 4982550a7d60SMina Almasry struct hstate *h = hstate_vma(vma); 4983550a7d60SMina Almasry struct mm_struct *mm = vma->vm_mm; 4984550a7d60SMina Almasry spinlock_t *src_ptl, *dst_ptl; 4985db110a99SAneesh Kumar K.V pte_t pte; 4986550a7d60SMina Almasry 4987550a7d60SMina Almasry dst_ptl = huge_pte_lock(h, mm, dst_pte); 4988550a7d60SMina Almasry src_ptl = huge_pte_lockptr(h, mm, src_pte); 4989550a7d60SMina Almasry 4990550a7d60SMina Almasry /* 4991550a7d60SMina Almasry * We don't have to worry about the ordering of src and dst ptlocks 4992550a7d60SMina Almasry * because exclusive mmap_sem (or the i_mmap_lock) prevents deadlock. 4993550a7d60SMina Almasry */ 4994550a7d60SMina Almasry if (src_ptl != dst_ptl) 4995550a7d60SMina Almasry spin_lock_nested(src_ptl, SINGLE_DEPTH_NESTING); 4996550a7d60SMina Almasry 4997550a7d60SMina Almasry pte = huge_ptep_get_and_clear(mm, old_addr, src_pte); 4998550a7d60SMina Almasry set_huge_pte_at(mm, new_addr, dst_pte, pte); 4999550a7d60SMina Almasry 5000550a7d60SMina Almasry if (src_ptl != dst_ptl) 5001550a7d60SMina Almasry spin_unlock(src_ptl); 5002550a7d60SMina Almasry spin_unlock(dst_ptl); 5003550a7d60SMina Almasry } 5004550a7d60SMina Almasry 5005550a7d60SMina Almasry int move_hugetlb_page_tables(struct vm_area_struct *vma, 5006550a7d60SMina Almasry struct vm_area_struct *new_vma, 5007550a7d60SMina Almasry unsigned long old_addr, unsigned long new_addr, 5008550a7d60SMina Almasry unsigned long len) 5009550a7d60SMina Almasry { 5010550a7d60SMina Almasry struct hstate *h = hstate_vma(vma); 5011550a7d60SMina Almasry struct address_space *mapping = vma->vm_file->f_mapping; 5012550a7d60SMina Almasry unsigned long sz = huge_page_size(h); 5013550a7d60SMina Almasry struct mm_struct *mm = vma->vm_mm; 5014550a7d60SMina Almasry unsigned long old_end = old_addr + len; 5015e95a9851SMike Kravetz unsigned long last_addr_mask; 5016550a7d60SMina Almasry pte_t *src_pte, *dst_pte; 5017550a7d60SMina Almasry struct mmu_notifier_range range; 50183d0b95cdSBaolin Wang bool shared_pmd = false; 5019550a7d60SMina Almasry 5020550a7d60SMina Almasry mmu_notifier_range_init(&range, MMU_NOTIFY_CLEAR, 0, vma, mm, old_addr, 5021550a7d60SMina Almasry old_end); 5022550a7d60SMina Almasry adjust_range_if_pmd_sharing_possible(vma, &range.start, &range.end); 50233d0b95cdSBaolin Wang /* 50243d0b95cdSBaolin Wang * In case of shared PMDs, we should cover the maximum possible 50253d0b95cdSBaolin Wang * range. 50263d0b95cdSBaolin Wang */ 50273d0b95cdSBaolin Wang flush_cache_range(vma, range.start, range.end); 50283d0b95cdSBaolin Wang 5029550a7d60SMina Almasry mmu_notifier_invalidate_range_start(&range); 5030e95a9851SMike Kravetz last_addr_mask = hugetlb_mask_last_page(h); 5031550a7d60SMina Almasry /* Prevent race with file truncation */ 503240549ba8SMike Kravetz hugetlb_vma_lock_write(vma); 5033550a7d60SMina Almasry i_mmap_lock_write(mapping); 5034550a7d60SMina Almasry for (; old_addr < old_end; old_addr += sz, new_addr += sz) { 5035550a7d60SMina Almasry src_pte = huge_pte_offset(mm, old_addr, sz); 5036e95a9851SMike Kravetz if (!src_pte) { 5037e95a9851SMike Kravetz old_addr |= last_addr_mask; 5038e95a9851SMike Kravetz new_addr |= last_addr_mask; 5039550a7d60SMina Almasry continue; 5040e95a9851SMike Kravetz } 5041550a7d60SMina Almasry if (huge_pte_none(huge_ptep_get(src_pte))) 5042550a7d60SMina Almasry continue; 5043550a7d60SMina Almasry 50444ddb4d91SMike Kravetz if (huge_pmd_unshare(mm, vma, old_addr, src_pte)) { 50453d0b95cdSBaolin Wang shared_pmd = true; 50464ddb4d91SMike Kravetz old_addr |= last_addr_mask; 50474ddb4d91SMike Kravetz new_addr |= last_addr_mask; 5048550a7d60SMina Almasry continue; 50493d0b95cdSBaolin Wang } 5050550a7d60SMina Almasry 5051550a7d60SMina Almasry dst_pte = huge_pte_alloc(mm, new_vma, new_addr, sz); 5052550a7d60SMina Almasry if (!dst_pte) 5053550a7d60SMina Almasry break; 5054550a7d60SMina Almasry 5055db110a99SAneesh Kumar K.V move_huge_pte(vma, old_addr, new_addr, src_pte, dst_pte); 5056550a7d60SMina Almasry } 50573d0b95cdSBaolin Wang 50583d0b95cdSBaolin Wang if (shared_pmd) 50593d0b95cdSBaolin Wang flush_tlb_range(vma, range.start, range.end); 50603d0b95cdSBaolin Wang else 5061550a7d60SMina Almasry flush_tlb_range(vma, old_end - len, old_end); 5062550a7d60SMina Almasry mmu_notifier_invalidate_range_end(&range); 506313e4ad2cSNadav Amit i_mmap_unlock_write(mapping); 506440549ba8SMike Kravetz hugetlb_vma_unlock_write(vma); 5065550a7d60SMina Almasry 5066550a7d60SMina Almasry return len + old_addr - old_end; 5067550a7d60SMina Almasry } 5068550a7d60SMina Almasry 506973c54763SPeter Xu static void __unmap_hugepage_range(struct mmu_gather *tlb, struct vm_area_struct *vma, 507024669e58SAneesh Kumar K.V unsigned long start, unsigned long end, 507105e90bd0SPeter Xu struct page *ref_page, zap_flags_t zap_flags) 507263551ae0SDavid Gibson { 507363551ae0SDavid Gibson struct mm_struct *mm = vma->vm_mm; 507463551ae0SDavid Gibson unsigned long address; 5075c7546f8fSDavid Gibson pte_t *ptep; 507663551ae0SDavid Gibson pte_t pte; 5077cb900f41SKirill A. Shutemov spinlock_t *ptl; 507863551ae0SDavid Gibson struct page *page; 5079a5516438SAndi Kleen struct hstate *h = hstate_vma(vma); 5080a5516438SAndi Kleen unsigned long sz = huge_page_size(h); 5081e95a9851SMike Kravetz unsigned long last_addr_mask; 5082a4a118f2SNadav Amit bool force_flush = false; 5083a5516438SAndi Kleen 508463551ae0SDavid Gibson WARN_ON(!is_vm_hugetlb_page(vma)); 5085a5516438SAndi Kleen BUG_ON(start & ~huge_page_mask(h)); 5086a5516438SAndi Kleen BUG_ON(end & ~huge_page_mask(h)); 508763551ae0SDavid Gibson 508807e32661SAneesh Kumar K.V /* 508907e32661SAneesh Kumar K.V * This is a hugetlb vma, all the pte entries should point 509007e32661SAneesh Kumar K.V * to huge page. 509107e32661SAneesh Kumar K.V */ 5092ed6a7935SPeter Zijlstra tlb_change_page_size(tlb, sz); 509324669e58SAneesh Kumar K.V tlb_start_vma(tlb, vma); 5094dff11abeSMike Kravetz 5095e95a9851SMike Kravetz last_addr_mask = hugetlb_mask_last_page(h); 5096569f48b8SHillf Danton address = start; 5097569f48b8SHillf Danton for (; address < end; address += sz) { 50987868a208SPunit Agrawal ptep = huge_pte_offset(mm, address, sz); 5099e95a9851SMike Kravetz if (!ptep) { 5100e95a9851SMike Kravetz address |= last_addr_mask; 5101c7546f8fSDavid Gibson continue; 5102e95a9851SMike Kravetz } 5103c7546f8fSDavid Gibson 5104cb900f41SKirill A. Shutemov ptl = huge_pte_lock(h, mm, ptep); 51054ddb4d91SMike Kravetz if (huge_pmd_unshare(mm, vma, address, ptep)) { 510631d49da5SAneesh Kumar K.V spin_unlock(ptl); 5107a4a118f2SNadav Amit tlb_flush_pmd_range(tlb, address & PUD_MASK, PUD_SIZE); 5108a4a118f2SNadav Amit force_flush = true; 51094ddb4d91SMike Kravetz address |= last_addr_mask; 511031d49da5SAneesh Kumar K.V continue; 511131d49da5SAneesh Kumar K.V } 511239dde65cSChen, Kenneth W 51136629326bSHillf Danton pte = huge_ptep_get(ptep); 511431d49da5SAneesh Kumar K.V if (huge_pte_none(pte)) { 511531d49da5SAneesh Kumar K.V spin_unlock(ptl); 511631d49da5SAneesh Kumar K.V continue; 511731d49da5SAneesh Kumar K.V } 51186629326bSHillf Danton 51196629326bSHillf Danton /* 51209fbc1f63SNaoya Horiguchi * Migrating hugepage or HWPoisoned hugepage is already 51219fbc1f63SNaoya Horiguchi * unmapped and its refcount is dropped, so just clear pte here. 51226629326bSHillf Danton */ 51239fbc1f63SNaoya Horiguchi if (unlikely(!pte_present(pte))) { 512405e90bd0SPeter Xu /* 512505e90bd0SPeter Xu * If the pte was wr-protected by uffd-wp in any of the 512605e90bd0SPeter Xu * swap forms, meanwhile the caller does not want to 512705e90bd0SPeter Xu * drop the uffd-wp bit in this zap, then replace the 512805e90bd0SPeter Xu * pte with a marker. 512905e90bd0SPeter Xu */ 513005e90bd0SPeter Xu if (pte_swp_uffd_wp_any(pte) && 513105e90bd0SPeter Xu !(zap_flags & ZAP_FLAG_DROP_MARKER)) 513205e90bd0SPeter Xu set_huge_pte_at(mm, address, ptep, 513305e90bd0SPeter Xu make_pte_marker(PTE_MARKER_UFFD_WP)); 513405e90bd0SPeter Xu else 51359386fac3SPunit Agrawal huge_pte_clear(mm, address, ptep, sz); 513631d49da5SAneesh Kumar K.V spin_unlock(ptl); 513731d49da5SAneesh Kumar K.V continue; 51388c4894c6SNaoya Horiguchi } 51396629326bSHillf Danton 51406629326bSHillf Danton page = pte_page(pte); 514104f2cbe3SMel Gorman /* 514204f2cbe3SMel Gorman * If a reference page is supplied, it is because a specific 514304f2cbe3SMel Gorman * page is being unmapped, not a range. Ensure the page we 514404f2cbe3SMel Gorman * are about to unmap is the actual page of interest. 514504f2cbe3SMel Gorman */ 514604f2cbe3SMel Gorman if (ref_page) { 514731d49da5SAneesh Kumar K.V if (page != ref_page) { 514831d49da5SAneesh Kumar K.V spin_unlock(ptl); 514931d49da5SAneesh Kumar K.V continue; 515031d49da5SAneesh Kumar K.V } 515104f2cbe3SMel Gorman /* 515204f2cbe3SMel Gorman * Mark the VMA as having unmapped its page so that 515304f2cbe3SMel Gorman * future faults in this VMA will fail rather than 515404f2cbe3SMel Gorman * looking like data was lost 515504f2cbe3SMel Gorman */ 515604f2cbe3SMel Gorman set_vma_resv_flags(vma, HPAGE_RESV_UNMAPPED); 515704f2cbe3SMel Gorman } 515804f2cbe3SMel Gorman 5159c7546f8fSDavid Gibson pte = huge_ptep_get_and_clear(mm, address, ptep); 5160b528e4b6SAneesh Kumar K.V tlb_remove_huge_tlb_entry(h, tlb, ptep, address); 5161106c992aSGerald Schaefer if (huge_pte_dirty(pte)) 51626649a386SKen Chen set_page_dirty(page); 516305e90bd0SPeter Xu /* Leave a uffd-wp pte marker if needed */ 516405e90bd0SPeter Xu if (huge_pte_uffd_wp(pte) && 516505e90bd0SPeter Xu !(zap_flags & ZAP_FLAG_DROP_MARKER)) 516605e90bd0SPeter Xu set_huge_pte_at(mm, address, ptep, 516705e90bd0SPeter Xu make_pte_marker(PTE_MARKER_UFFD_WP)); 51685d317b2bSNaoya Horiguchi hugetlb_count_sub(pages_per_huge_page(h), mm); 5169cea86fe2SHugh Dickins page_remove_rmap(page, vma, true); 517031d49da5SAneesh Kumar K.V 5171cb900f41SKirill A. Shutemov spin_unlock(ptl); 5172e77b0852SAneesh Kumar K.V tlb_remove_page_size(tlb, page, huge_page_size(h)); 517324669e58SAneesh Kumar K.V /* 517431d49da5SAneesh Kumar K.V * Bail out after unmapping reference page if supplied 517524669e58SAneesh Kumar K.V */ 517631d49da5SAneesh Kumar K.V if (ref_page) 517731d49da5SAneesh Kumar K.V break; 5178fe1668aeSChen, Kenneth W } 517924669e58SAneesh Kumar K.V tlb_end_vma(tlb, vma); 5180a4a118f2SNadav Amit 5181a4a118f2SNadav Amit /* 5182a4a118f2SNadav Amit * If we unshared PMDs, the TLB flush was not recorded in mmu_gather. We 5183a4a118f2SNadav Amit * could defer the flush until now, since by holding i_mmap_rwsem we 5184a4a118f2SNadav Amit * guaranteed that the last refernece would not be dropped. But we must 5185a4a118f2SNadav Amit * do the flushing before we return, as otherwise i_mmap_rwsem will be 5186a4a118f2SNadav Amit * dropped and the last reference to the shared PMDs page might be 5187a4a118f2SNadav Amit * dropped as well. 5188a4a118f2SNadav Amit * 5189a4a118f2SNadav Amit * In theory we could defer the freeing of the PMD pages as well, but 5190a4a118f2SNadav Amit * huge_pmd_unshare() relies on the exact page_count for the PMD page to 5191a4a118f2SNadav Amit * detect sharing, so we cannot defer the release of the page either. 5192a4a118f2SNadav Amit * Instead, do flush now. 5193a4a118f2SNadav Amit */ 5194a4a118f2SNadav Amit if (force_flush) 5195a4a118f2SNadav Amit tlb_flush_mmu_tlbonly(tlb); 51961da177e4SLinus Torvalds } 519763551ae0SDavid Gibson 5198d833352aSMel Gorman void __unmap_hugepage_range_final(struct mmu_gather *tlb, 5199d833352aSMel Gorman struct vm_area_struct *vma, unsigned long start, 520005e90bd0SPeter Xu unsigned long end, struct page *ref_page, 520105e90bd0SPeter Xu zap_flags_t zap_flags) 5202d833352aSMel Gorman { 5203131a79b4SMike Kravetz hugetlb_vma_lock_write(vma); 5204131a79b4SMike Kravetz i_mmap_lock_write(vma->vm_file->f_mapping); 5205131a79b4SMike Kravetz 5206369258ceSMike Kravetz /* mmu notification performed in caller */ 520705e90bd0SPeter Xu __unmap_hugepage_range(tlb, vma, start, end, ref_page, zap_flags); 5208d833352aSMel Gorman 520904ada095SMike Kravetz if (zap_flags & ZAP_FLAG_UNMAP) { /* final unmap */ 5210d833352aSMel Gorman /* 521104ada095SMike Kravetz * Unlock and free the vma lock before releasing i_mmap_rwsem. 521204ada095SMike Kravetz * When the vma_lock is freed, this makes the vma ineligible 521304ada095SMike Kravetz * for pmd sharing. And, i_mmap_rwsem is required to set up 521404ada095SMike Kravetz * pmd sharing. This is important as page tables for this 521504ada095SMike Kravetz * unmapped range will be asynchrously deleted. If the page 521604ada095SMike Kravetz * tables are shared, there will be issues when accessed by 521704ada095SMike Kravetz * someone else. 5218d833352aSMel Gorman */ 5219ecfbd733SMike Kravetz __hugetlb_vma_unlock_write_free(vma); 5220131a79b4SMike Kravetz i_mmap_unlock_write(vma->vm_file->f_mapping); 522104ada095SMike Kravetz } else { 522204ada095SMike Kravetz i_mmap_unlock_write(vma->vm_file->f_mapping); 522304ada095SMike Kravetz hugetlb_vma_unlock_write(vma); 522404ada095SMike Kravetz } 5225d833352aSMel Gorman } 5226d833352aSMel Gorman 5227502717f4SChen, Kenneth W void unmap_hugepage_range(struct vm_area_struct *vma, unsigned long start, 522805e90bd0SPeter Xu unsigned long end, struct page *ref_page, 522905e90bd0SPeter Xu zap_flags_t zap_flags) 5230502717f4SChen, Kenneth W { 5231369258ceSMike Kravetz struct mmu_notifier_range range; 523224669e58SAneesh Kumar K.V struct mmu_gather tlb; 5233dff11abeSMike Kravetz 5234369258ceSMike Kravetz mmu_notifier_range_init(&range, MMU_NOTIFY_CLEAR, 0, vma, vma->vm_mm, 5235369258ceSMike Kravetz start, end); 5236369258ceSMike Kravetz adjust_range_if_pmd_sharing_possible(vma, &range.start, &range.end); 5237369258ceSMike Kravetz mmu_notifier_invalidate_range_start(&range); 5238a72afd87SWill Deacon tlb_gather_mmu(&tlb, vma->vm_mm); 5239369258ceSMike Kravetz 524005e90bd0SPeter Xu __unmap_hugepage_range(&tlb, vma, start, end, ref_page, zap_flags); 5241369258ceSMike Kravetz 5242369258ceSMike Kravetz mmu_notifier_invalidate_range_end(&range); 5243ae8eba8bSWill Deacon tlb_finish_mmu(&tlb); 5244502717f4SChen, Kenneth W } 5245502717f4SChen, Kenneth W 524604f2cbe3SMel Gorman /* 524704f2cbe3SMel Gorman * This is called when the original mapper is failing to COW a MAP_PRIVATE 5248578b7725SZhiyuan Dai * mapping it owns the reserve page for. The intention is to unmap the page 524904f2cbe3SMel Gorman * from other VMAs and let the children be SIGKILLed if they are faulting the 525004f2cbe3SMel Gorman * same region. 525104f2cbe3SMel Gorman */ 52522f4612afSDavidlohr Bueso static void unmap_ref_private(struct mm_struct *mm, struct vm_area_struct *vma, 52532a4b3dedSHarvey Harrison struct page *page, unsigned long address) 525404f2cbe3SMel Gorman { 52557526674dSAdam Litke struct hstate *h = hstate_vma(vma); 525604f2cbe3SMel Gorman struct vm_area_struct *iter_vma; 525704f2cbe3SMel Gorman struct address_space *mapping; 525804f2cbe3SMel Gorman pgoff_t pgoff; 525904f2cbe3SMel Gorman 526004f2cbe3SMel Gorman /* 526104f2cbe3SMel Gorman * vm_pgoff is in PAGE_SIZE units, hence the different calculation 526204f2cbe3SMel Gorman * from page cache lookup which is in HPAGE_SIZE units. 526304f2cbe3SMel Gorman */ 52647526674dSAdam Litke address = address & huge_page_mask(h); 526536e4f20aSMichal Hocko pgoff = ((address - vma->vm_start) >> PAGE_SHIFT) + 526636e4f20aSMichal Hocko vma->vm_pgoff; 526793c76a3dSAl Viro mapping = vma->vm_file->f_mapping; 526804f2cbe3SMel Gorman 52694eb2b1dcSMel Gorman /* 52704eb2b1dcSMel Gorman * Take the mapping lock for the duration of the table walk. As 52714eb2b1dcSMel Gorman * this mapping should be shared between all the VMAs, 52724eb2b1dcSMel Gorman * __unmap_hugepage_range() is called as the lock is already held 52734eb2b1dcSMel Gorman */ 527483cde9e8SDavidlohr Bueso i_mmap_lock_write(mapping); 52756b2dbba8SMichel Lespinasse vma_interval_tree_foreach(iter_vma, &mapping->i_mmap, pgoff, pgoff) { 527604f2cbe3SMel Gorman /* Do not unmap the current VMA */ 527704f2cbe3SMel Gorman if (iter_vma == vma) 527804f2cbe3SMel Gorman continue; 527904f2cbe3SMel Gorman 528004f2cbe3SMel Gorman /* 52812f84a899SMel Gorman * Shared VMAs have their own reserves and do not affect 52822f84a899SMel Gorman * MAP_PRIVATE accounting but it is possible that a shared 52832f84a899SMel Gorman * VMA is using the same page so check and skip such VMAs. 52842f84a899SMel Gorman */ 52852f84a899SMel Gorman if (iter_vma->vm_flags & VM_MAYSHARE) 52862f84a899SMel Gorman continue; 52872f84a899SMel Gorman 52882f84a899SMel Gorman /* 528904f2cbe3SMel Gorman * Unmap the page from other VMAs without their own reserves. 529004f2cbe3SMel Gorman * They get marked to be SIGKILLed if they fault in these 529104f2cbe3SMel Gorman * areas. This is because a future no-page fault on this VMA 529204f2cbe3SMel Gorman * could insert a zeroed page instead of the data existing 529304f2cbe3SMel Gorman * from the time of fork. This would look like data corruption 529404f2cbe3SMel Gorman */ 529504f2cbe3SMel Gorman if (!is_vma_resv_set(iter_vma, HPAGE_RESV_OWNER)) 529624669e58SAneesh Kumar K.V unmap_hugepage_range(iter_vma, address, 529705e90bd0SPeter Xu address + huge_page_size(h), page, 0); 529804f2cbe3SMel Gorman } 529983cde9e8SDavidlohr Bueso i_mmap_unlock_write(mapping); 530004f2cbe3SMel Gorman } 530104f2cbe3SMel Gorman 53020fe6e20bSNaoya Horiguchi /* 5303c89357e2SDavid Hildenbrand * hugetlb_wp() should be called with page lock of the original hugepage held. 5304aa6d2e8cSBaolin Wang * Called with hugetlb_fault_mutex_table held and pte_page locked so we 5305ef009b25SMichal Hocko * cannot race with other handlers or page migration. 5306ef009b25SMichal Hocko * Keep the pte_same checks anyway to make transition from the mutex easier. 53070fe6e20bSNaoya Horiguchi */ 5308c89357e2SDavid Hildenbrand static vm_fault_t hugetlb_wp(struct mm_struct *mm, struct vm_area_struct *vma, 5309c89357e2SDavid Hildenbrand unsigned long address, pte_t *ptep, unsigned int flags, 5310cb900f41SKirill A. Shutemov struct page *pagecache_page, spinlock_t *ptl) 53111e8f889bSDavid Gibson { 5312c89357e2SDavid Hildenbrand const bool unshare = flags & FAULT_FLAG_UNSHARE; 53133999f52eSAneesh Kumar K.V pte_t pte; 5314a5516438SAndi Kleen struct hstate *h = hstate_vma(vma); 53151e8f889bSDavid Gibson struct page *old_page, *new_page; 53162b740303SSouptick Joarder int outside_reserve = 0; 53172b740303SSouptick Joarder vm_fault_t ret = 0; 5318974e6d66SHuang Ying unsigned long haddr = address & huge_page_mask(h); 5319ac46d4f3SJérôme Glisse struct mmu_notifier_range range; 53201e8f889bSDavid Gibson 53211d8d1464SDavid Hildenbrand /* 53221d8d1464SDavid Hildenbrand * hugetlb does not support FOLL_FORCE-style write faults that keep the 53231d8d1464SDavid Hildenbrand * PTE mapped R/O such as maybe_mkwrite() would do. 53241d8d1464SDavid Hildenbrand */ 53251d8d1464SDavid Hildenbrand if (WARN_ON_ONCE(!unshare && !(vma->vm_flags & VM_WRITE))) 53261d8d1464SDavid Hildenbrand return VM_FAULT_SIGSEGV; 53271d8d1464SDavid Hildenbrand 53281d8d1464SDavid Hildenbrand /* Let's take out MAP_SHARED mappings first. */ 53291d8d1464SDavid Hildenbrand if (vma->vm_flags & VM_MAYSHARE) { 53301d8d1464SDavid Hildenbrand set_huge_ptep_writable(vma, haddr, ptep); 53311d8d1464SDavid Hildenbrand return 0; 53321d8d1464SDavid Hildenbrand } 53331d8d1464SDavid Hildenbrand 53343999f52eSAneesh Kumar K.V pte = huge_ptep_get(ptep); 53351e8f889bSDavid Gibson old_page = pte_page(pte); 53361e8f889bSDavid Gibson 5337662ce1dcSYang Yang delayacct_wpcopy_start(); 5338662ce1dcSYang Yang 533904f2cbe3SMel Gorman retry_avoidcopy: 5340c89357e2SDavid Hildenbrand /* 5341c89357e2SDavid Hildenbrand * If no-one else is actually using this page, we're the exclusive 5342c89357e2SDavid Hildenbrand * owner and can reuse this page. 5343c89357e2SDavid Hildenbrand */ 534437a2140dSJoonsoo Kim if (page_mapcount(old_page) == 1 && PageAnon(old_page)) { 5345c89357e2SDavid Hildenbrand if (!PageAnonExclusive(old_page)) 53465a49973dSHugh Dickins page_move_anon_rmap(old_page, vma); 5347c89357e2SDavid Hildenbrand if (likely(!unshare)) 53485b7a1d40SHuang Ying set_huge_ptep_writable(vma, haddr, ptep); 5349662ce1dcSYang Yang 5350662ce1dcSYang Yang delayacct_wpcopy_end(); 535183c54070SNick Piggin return 0; 53521e8f889bSDavid Gibson } 53536c287605SDavid Hildenbrand VM_BUG_ON_PAGE(PageAnon(old_page) && PageAnonExclusive(old_page), 53546c287605SDavid Hildenbrand old_page); 53551e8f889bSDavid Gibson 535604f2cbe3SMel Gorman /* 535704f2cbe3SMel Gorman * If the process that created a MAP_PRIVATE mapping is about to 535804f2cbe3SMel Gorman * perform a COW due to a shared page count, attempt to satisfy 535904f2cbe3SMel Gorman * the allocation without using the existing reserves. The pagecache 536004f2cbe3SMel Gorman * page is used to determine if the reserve at this address was 536104f2cbe3SMel Gorman * consumed or not. If reserves were used, a partial faulted mapping 536204f2cbe3SMel Gorman * at the time of fork() could consume its reserves on COW instead 536304f2cbe3SMel Gorman * of the full address range. 536404f2cbe3SMel Gorman */ 53655944d011SJoonsoo Kim if (is_vma_resv_set(vma, HPAGE_RESV_OWNER) && 536604f2cbe3SMel Gorman old_page != pagecache_page) 536704f2cbe3SMel Gorman outside_reserve = 1; 536804f2cbe3SMel Gorman 536909cbfeafSKirill A. Shutemov get_page(old_page); 5370b76c8cfbSLarry Woodman 5371ad4404a2SDavidlohr Bueso /* 5372ad4404a2SDavidlohr Bueso * Drop page table lock as buddy allocator may be called. It will 5373ad4404a2SDavidlohr Bueso * be acquired again before returning to the caller, as expected. 5374ad4404a2SDavidlohr Bueso */ 5375cb900f41SKirill A. Shutemov spin_unlock(ptl); 53765b7a1d40SHuang Ying new_page = alloc_huge_page(vma, haddr, outside_reserve); 53771e8f889bSDavid Gibson 53782fc39cecSAdam Litke if (IS_ERR(new_page)) { 537904f2cbe3SMel Gorman /* 538004f2cbe3SMel Gorman * If a process owning a MAP_PRIVATE mapping fails to COW, 538104f2cbe3SMel Gorman * it is due to references held by a child and an insufficient 538204f2cbe3SMel Gorman * huge page pool. To guarantee the original mappers 538304f2cbe3SMel Gorman * reliability, unmap the page from child processes. The child 538404f2cbe3SMel Gorman * may get SIGKILLed if it later faults. 538504f2cbe3SMel Gorman */ 538604f2cbe3SMel Gorman if (outside_reserve) { 538740549ba8SMike Kravetz struct address_space *mapping = vma->vm_file->f_mapping; 538840549ba8SMike Kravetz pgoff_t idx; 538940549ba8SMike Kravetz u32 hash; 539040549ba8SMike Kravetz 539109cbfeafSKirill A. Shutemov put_page(old_page); 539240549ba8SMike Kravetz /* 539340549ba8SMike Kravetz * Drop hugetlb_fault_mutex and vma_lock before 539440549ba8SMike Kravetz * unmapping. unmapping needs to hold vma_lock 539540549ba8SMike Kravetz * in write mode. Dropping vma_lock in read mode 539640549ba8SMike Kravetz * here is OK as COW mappings do not interact with 539740549ba8SMike Kravetz * PMD sharing. 539840549ba8SMike Kravetz * 539940549ba8SMike Kravetz * Reacquire both after unmap operation. 540040549ba8SMike Kravetz */ 540140549ba8SMike Kravetz idx = vma_hugecache_offset(h, vma, haddr); 540240549ba8SMike Kravetz hash = hugetlb_fault_mutex_hash(mapping, idx); 540340549ba8SMike Kravetz hugetlb_vma_unlock_read(vma); 540440549ba8SMike Kravetz mutex_unlock(&hugetlb_fault_mutex_table[hash]); 540540549ba8SMike Kravetz 54065b7a1d40SHuang Ying unmap_ref_private(mm, vma, old_page, haddr); 540740549ba8SMike Kravetz 540840549ba8SMike Kravetz mutex_lock(&hugetlb_fault_mutex_table[hash]); 540940549ba8SMike Kravetz hugetlb_vma_lock_read(vma); 5410cb900f41SKirill A. Shutemov spin_lock(ptl); 54115b7a1d40SHuang Ying ptep = huge_pte_offset(mm, haddr, huge_page_size(h)); 5412a9af0c5dSNaoya Horiguchi if (likely(ptep && 5413a9af0c5dSNaoya Horiguchi pte_same(huge_ptep_get(ptep), pte))) 541404f2cbe3SMel Gorman goto retry_avoidcopy; 5415a734bcc8SHillf Danton /* 5416cb900f41SKirill A. Shutemov * race occurs while re-acquiring page table 5417cb900f41SKirill A. Shutemov * lock, and our job is done. 5418a734bcc8SHillf Danton */ 5419662ce1dcSYang Yang delayacct_wpcopy_end(); 5420a734bcc8SHillf Danton return 0; 542104f2cbe3SMel Gorman } 542204f2cbe3SMel Gorman 54232b740303SSouptick Joarder ret = vmf_error(PTR_ERR(new_page)); 5424ad4404a2SDavidlohr Bueso goto out_release_old; 54251e8f889bSDavid Gibson } 54261e8f889bSDavid Gibson 54270fe6e20bSNaoya Horiguchi /* 54280fe6e20bSNaoya Horiguchi * When the original hugepage is shared one, it does not have 54290fe6e20bSNaoya Horiguchi * anon_vma prepared. 54300fe6e20bSNaoya Horiguchi */ 543144e2aa93SDean Nelson if (unlikely(anon_vma_prepare(vma))) { 5432ad4404a2SDavidlohr Bueso ret = VM_FAULT_OOM; 5433ad4404a2SDavidlohr Bueso goto out_release_all; 543444e2aa93SDean Nelson } 54350fe6e20bSNaoya Horiguchi 5436974e6d66SHuang Ying copy_user_huge_page(new_page, old_page, address, vma, 543747ad8475SAndrea Arcangeli pages_per_huge_page(h)); 54380ed361deSNick Piggin __SetPageUptodate(new_page); 54391e8f889bSDavid Gibson 54407269f999SJérôme Glisse mmu_notifier_range_init(&range, MMU_NOTIFY_CLEAR, 0, vma, mm, haddr, 54416f4f13e8SJérôme Glisse haddr + huge_page_size(h)); 5442ac46d4f3SJérôme Glisse mmu_notifier_invalidate_range_start(&range); 5443ad4404a2SDavidlohr Bueso 5444b76c8cfbSLarry Woodman /* 5445cb900f41SKirill A. Shutemov * Retake the page table lock to check for racing updates 5446b76c8cfbSLarry Woodman * before the page tables are altered 5447b76c8cfbSLarry Woodman */ 5448cb900f41SKirill A. Shutemov spin_lock(ptl); 54495b7a1d40SHuang Ying ptep = huge_pte_offset(mm, haddr, huge_page_size(h)); 5450a9af0c5dSNaoya Horiguchi if (likely(ptep && pte_same(huge_ptep_get(ptep), pte))) { 5451c89357e2SDavid Hildenbrand /* Break COW or unshare */ 54525b7a1d40SHuang Ying huge_ptep_clear_flush(vma, haddr, ptep); 5453ac46d4f3SJérôme Glisse mmu_notifier_invalidate_range(mm, range.start, range.end); 5454cea86fe2SHugh Dickins page_remove_rmap(old_page, vma, true); 54555b7a1d40SHuang Ying hugepage_add_new_anon_rmap(new_page, vma, haddr); 54561eba86c0SPasha Tatashin set_huge_pte_at(mm, haddr, ptep, 5457c89357e2SDavid Hildenbrand make_huge_pte(vma, new_page, !unshare)); 54588f251a3dSMike Kravetz SetHPageMigratable(new_page); 54591e8f889bSDavid Gibson /* Make the old page be freed below */ 54601e8f889bSDavid Gibson new_page = old_page; 54611e8f889bSDavid Gibson } 5462cb900f41SKirill A. Shutemov spin_unlock(ptl); 5463ac46d4f3SJérôme Glisse mmu_notifier_invalidate_range_end(&range); 5464ad4404a2SDavidlohr Bueso out_release_all: 5465c89357e2SDavid Hildenbrand /* 5466c89357e2SDavid Hildenbrand * No restore in case of successful pagetable update (Break COW or 5467c89357e2SDavid Hildenbrand * unshare) 5468c89357e2SDavid Hildenbrand */ 5469c7b1850dSMike Kravetz if (new_page != old_page) 54705b7a1d40SHuang Ying restore_reserve_on_error(h, vma, haddr, new_page); 547109cbfeafSKirill A. Shutemov put_page(new_page); 5472ad4404a2SDavidlohr Bueso out_release_old: 547309cbfeafSKirill A. Shutemov put_page(old_page); 54748312034fSJoonsoo Kim 5475ad4404a2SDavidlohr Bueso spin_lock(ptl); /* Caller expects lock to be held */ 5476662ce1dcSYang Yang 5477662ce1dcSYang Yang delayacct_wpcopy_end(); 5478ad4404a2SDavidlohr Bueso return ret; 54791e8f889bSDavid Gibson } 54801e8f889bSDavid Gibson 54813ae77f43SHugh Dickins /* 54823ae77f43SHugh Dickins * Return whether there is a pagecache page to back given address within VMA. 54833ae77f43SHugh Dickins * Caller follow_hugetlb_page() holds page_table_lock so we cannot lock_page. 54843ae77f43SHugh Dickins */ 54853ae77f43SHugh Dickins static bool hugetlbfs_pagecache_present(struct hstate *h, 54862a15efc9SHugh Dickins struct vm_area_struct *vma, unsigned long address) 54872a15efc9SHugh Dickins { 54882a15efc9SHugh Dickins struct address_space *mapping; 54892a15efc9SHugh Dickins pgoff_t idx; 54902a15efc9SHugh Dickins struct page *page; 54912a15efc9SHugh Dickins 54922a15efc9SHugh Dickins mapping = vma->vm_file->f_mapping; 54932a15efc9SHugh Dickins idx = vma_hugecache_offset(h, vma, address); 54942a15efc9SHugh Dickins 54952a15efc9SHugh Dickins page = find_get_page(mapping, idx); 54962a15efc9SHugh Dickins if (page) 54972a15efc9SHugh Dickins put_page(page); 54982a15efc9SHugh Dickins return page != NULL; 54992a15efc9SHugh Dickins } 55002a15efc9SHugh Dickins 55017e1813d4SMike Kravetz int hugetlb_add_to_page_cache(struct page *page, struct address_space *mapping, 5502ab76ad54SMike Kravetz pgoff_t idx) 5503ab76ad54SMike Kravetz { 5504d9ef44deSMatthew Wilcox (Oracle) struct folio *folio = page_folio(page); 5505ab76ad54SMike Kravetz struct inode *inode = mapping->host; 5506ab76ad54SMike Kravetz struct hstate *h = hstate_inode(inode); 5507d9ef44deSMatthew Wilcox (Oracle) int err; 5508ab76ad54SMike Kravetz 5509d9ef44deSMatthew Wilcox (Oracle) __folio_set_locked(folio); 5510d9ef44deSMatthew Wilcox (Oracle) err = __filemap_add_folio(mapping, folio, idx, GFP_KERNEL, NULL); 5511d9ef44deSMatthew Wilcox (Oracle) 5512d9ef44deSMatthew Wilcox (Oracle) if (unlikely(err)) { 5513d9ef44deSMatthew Wilcox (Oracle) __folio_clear_locked(folio); 5514ab76ad54SMike Kravetz return err; 5515d9ef44deSMatthew Wilcox (Oracle) } 5516d6995da3SMike Kravetz ClearHPageRestoreReserve(page); 5517ab76ad54SMike Kravetz 551822146c3cSMike Kravetz /* 5519d9ef44deSMatthew Wilcox (Oracle) * mark folio dirty so that it will not be removed from cache/file 552022146c3cSMike Kravetz * by non-hugetlbfs specific code paths. 552122146c3cSMike Kravetz */ 5522d9ef44deSMatthew Wilcox (Oracle) folio_mark_dirty(folio); 552322146c3cSMike Kravetz 5524ab76ad54SMike Kravetz spin_lock(&inode->i_lock); 5525ab76ad54SMike Kravetz inode->i_blocks += blocks_per_huge_page(h); 5526ab76ad54SMike Kravetz spin_unlock(&inode->i_lock); 5527ab76ad54SMike Kravetz return 0; 5528ab76ad54SMike Kravetz } 5529ab76ad54SMike Kravetz 55307677f7fdSAxel Rasmussen static inline vm_fault_t hugetlb_handle_userfault(struct vm_area_struct *vma, 55317677f7fdSAxel Rasmussen struct address_space *mapping, 55327677f7fdSAxel Rasmussen pgoff_t idx, 55337677f7fdSAxel Rasmussen unsigned int flags, 55347677f7fdSAxel Rasmussen unsigned long haddr, 5535824ddc60SNadav Amit unsigned long addr, 55367677f7fdSAxel Rasmussen unsigned long reason) 55377677f7fdSAxel Rasmussen { 55387677f7fdSAxel Rasmussen u32 hash; 55397677f7fdSAxel Rasmussen struct vm_fault vmf = { 55407677f7fdSAxel Rasmussen .vma = vma, 55417677f7fdSAxel Rasmussen .address = haddr, 5542824ddc60SNadav Amit .real_address = addr, 55437677f7fdSAxel Rasmussen .flags = flags, 55447677f7fdSAxel Rasmussen 55457677f7fdSAxel Rasmussen /* 55467677f7fdSAxel Rasmussen * Hard to debug if it ends up being 55477677f7fdSAxel Rasmussen * used by a callee that assumes 55487677f7fdSAxel Rasmussen * something about the other 55497677f7fdSAxel Rasmussen * uninitialized fields... same as in 55507677f7fdSAxel Rasmussen * memory.c 55517677f7fdSAxel Rasmussen */ 55527677f7fdSAxel Rasmussen }; 55537677f7fdSAxel Rasmussen 55547677f7fdSAxel Rasmussen /* 5555958f32ceSLiu Shixin * vma_lock and hugetlb_fault_mutex must be dropped before handling 5556958f32ceSLiu Shixin * userfault. Also mmap_lock could be dropped due to handling 5557958f32ceSLiu Shixin * userfault, any vma operation should be careful from here. 55587677f7fdSAxel Rasmussen */ 555940549ba8SMike Kravetz hugetlb_vma_unlock_read(vma); 55607677f7fdSAxel Rasmussen hash = hugetlb_fault_mutex_hash(mapping, idx); 55617677f7fdSAxel Rasmussen mutex_unlock(&hugetlb_fault_mutex_table[hash]); 5562958f32ceSLiu Shixin return handle_userfault(&vmf, reason); 55637677f7fdSAxel Rasmussen } 55647677f7fdSAxel Rasmussen 55652ea7ff1eSPeter Xu /* 55662ea7ff1eSPeter Xu * Recheck pte with pgtable lock. Returns true if pte didn't change, or 55672ea7ff1eSPeter Xu * false if pte changed or is changing. 55682ea7ff1eSPeter Xu */ 55692ea7ff1eSPeter Xu static bool hugetlb_pte_stable(struct hstate *h, struct mm_struct *mm, 55702ea7ff1eSPeter Xu pte_t *ptep, pte_t old_pte) 55712ea7ff1eSPeter Xu { 55722ea7ff1eSPeter Xu spinlock_t *ptl; 55732ea7ff1eSPeter Xu bool same; 55742ea7ff1eSPeter Xu 55752ea7ff1eSPeter Xu ptl = huge_pte_lock(h, mm, ptep); 55762ea7ff1eSPeter Xu same = pte_same(huge_ptep_get(ptep), old_pte); 55772ea7ff1eSPeter Xu spin_unlock(ptl); 55782ea7ff1eSPeter Xu 55792ea7ff1eSPeter Xu return same; 55802ea7ff1eSPeter Xu } 55812ea7ff1eSPeter Xu 55822b740303SSouptick Joarder static vm_fault_t hugetlb_no_page(struct mm_struct *mm, 55832b740303SSouptick Joarder struct vm_area_struct *vma, 55848382d914SDavidlohr Bueso struct address_space *mapping, pgoff_t idx, 5585c64e912cSPeter Xu unsigned long address, pte_t *ptep, 5586c64e912cSPeter Xu pte_t old_pte, unsigned int flags) 5587ac9b9c66SHugh Dickins { 5588a5516438SAndi Kleen struct hstate *h = hstate_vma(vma); 55892b740303SSouptick Joarder vm_fault_t ret = VM_FAULT_SIGBUS; 5590409eb8c2SHillf Danton int anon_rmap = 0; 55914c887265SAdam Litke unsigned long size; 55924c887265SAdam Litke struct page *page; 55931e8f889bSDavid Gibson pte_t new_pte; 5594cb900f41SKirill A. Shutemov spinlock_t *ptl; 5595285b8dcaSHuang Ying unsigned long haddr = address & huge_page_mask(h); 5596c7b1850dSMike Kravetz bool new_page, new_pagecache_page = false; 5597958f32ceSLiu Shixin u32 hash = hugetlb_fault_mutex_hash(mapping, idx); 55984c887265SAdam Litke 559904f2cbe3SMel Gorman /* 560004f2cbe3SMel Gorman * Currently, we are forced to kill the process in the event the 560104f2cbe3SMel Gorman * original mapper has unmapped pages from the child due to a failed 5602c89357e2SDavid Hildenbrand * COW/unsharing. Warn that such a situation has occurred as it may not 5603c89357e2SDavid Hildenbrand * be obvious. 560404f2cbe3SMel Gorman */ 560504f2cbe3SMel Gorman if (is_vma_resv_set(vma, HPAGE_RESV_UNMAPPED)) { 5606910154d5SGeoffrey Thomas pr_warn_ratelimited("PID %d killed due to inadequate hugepage pool\n", 560704f2cbe3SMel Gorman current->pid); 5608958f32ceSLiu Shixin goto out; 560904f2cbe3SMel Gorman } 561004f2cbe3SMel Gorman 56114c887265SAdam Litke /* 5612188a3972SMike Kravetz * Use page lock to guard against racing truncation 5613188a3972SMike Kravetz * before we get page_table_lock. 56144c887265SAdam Litke */ 5615c7b1850dSMike Kravetz new_page = false; 561687bf91d3SMike Kravetz page = find_lock_page(mapping, idx); 561787bf91d3SMike Kravetz if (!page) { 5618188a3972SMike Kravetz size = i_size_read(mapping->host) >> huge_page_shift(h); 5619188a3972SMike Kravetz if (idx >= size) 5620188a3972SMike Kravetz goto out; 56217677f7fdSAxel Rasmussen /* Check for page in userfault range */ 56222ea7ff1eSPeter Xu if (userfaultfd_missing(vma)) { 56232ea7ff1eSPeter Xu /* 56242ea7ff1eSPeter Xu * Since hugetlb_no_page() was examining pte 56252ea7ff1eSPeter Xu * without pgtable lock, we need to re-test under 56262ea7ff1eSPeter Xu * lock because the pte may not be stable and could 56272ea7ff1eSPeter Xu * have changed from under us. Try to detect 56282ea7ff1eSPeter Xu * either changed or during-changing ptes and retry 56292ea7ff1eSPeter Xu * properly when needed. 56302ea7ff1eSPeter Xu * 56312ea7ff1eSPeter Xu * Note that userfaultfd is actually fine with 56322ea7ff1eSPeter Xu * false positives (e.g. caused by pte changed), 56332ea7ff1eSPeter Xu * but not wrong logical events (e.g. caused by 56342ea7ff1eSPeter Xu * reading a pte during changing). The latter can 56352ea7ff1eSPeter Xu * confuse the userspace, so the strictness is very 56362ea7ff1eSPeter Xu * much preferred. E.g., MISSING event should 56372ea7ff1eSPeter Xu * never happen on the page after UFFDIO_COPY has 56382ea7ff1eSPeter Xu * correctly installed the page and returned. 56392ea7ff1eSPeter Xu */ 56402ea7ff1eSPeter Xu if (!hugetlb_pte_stable(h, mm, ptep, old_pte)) { 56412ea7ff1eSPeter Xu ret = 0; 56422ea7ff1eSPeter Xu goto out; 56432ea7ff1eSPeter Xu } 56442ea7ff1eSPeter Xu 56452ea7ff1eSPeter Xu return hugetlb_handle_userfault(vma, mapping, idx, flags, 56462ea7ff1eSPeter Xu haddr, address, 56477677f7fdSAxel Rasmussen VM_UFFD_MISSING); 56482ea7ff1eSPeter Xu } 56491a1aad8aSMike Kravetz 5650285b8dcaSHuang Ying page = alloc_huge_page(vma, haddr, 0); 56512fc39cecSAdam Litke if (IS_ERR(page)) { 56524643d67eSMike Kravetz /* 56534643d67eSMike Kravetz * Returning error will result in faulting task being 56544643d67eSMike Kravetz * sent SIGBUS. The hugetlb fault mutex prevents two 56554643d67eSMike Kravetz * tasks from racing to fault in the same page which 56564643d67eSMike Kravetz * could result in false unable to allocate errors. 56574643d67eSMike Kravetz * Page migration does not take the fault mutex, but 56584643d67eSMike Kravetz * does a clear then write of pte's under page table 56594643d67eSMike Kravetz * lock. Page fault code could race with migration, 56604643d67eSMike Kravetz * notice the clear pte and try to allocate a page 56614643d67eSMike Kravetz * here. Before returning error, get ptl and make 56624643d67eSMike Kravetz * sure there really is no pte entry. 56634643d67eSMike Kravetz */ 5664f9bf6c03SPeter Xu if (hugetlb_pte_stable(h, mm, ptep, old_pte)) 56652b740303SSouptick Joarder ret = vmf_error(PTR_ERR(page)); 5666f9bf6c03SPeter Xu else 5667f9bf6c03SPeter Xu ret = 0; 56686bda666aSChristoph Lameter goto out; 56696bda666aSChristoph Lameter } 567047ad8475SAndrea Arcangeli clear_huge_page(page, address, pages_per_huge_page(h)); 56710ed361deSNick Piggin __SetPageUptodate(page); 5672cb6acd01SMike Kravetz new_page = true; 5673ac9b9c66SHugh Dickins 5674f83a275dSMel Gorman if (vma->vm_flags & VM_MAYSHARE) { 56757e1813d4SMike Kravetz int err = hugetlb_add_to_page_cache(page, mapping, idx); 56766bda666aSChristoph Lameter if (err) { 56773a5497a2SMiaohe Lin /* 56783a5497a2SMiaohe Lin * err can't be -EEXIST which implies someone 56793a5497a2SMiaohe Lin * else consumed the reservation since hugetlb 56803a5497a2SMiaohe Lin * fault mutex is held when add a hugetlb page 56813a5497a2SMiaohe Lin * to the page cache. So it's safe to call 56823a5497a2SMiaohe Lin * restore_reserve_on_error() here. 56833a5497a2SMiaohe Lin */ 56843a5497a2SMiaohe Lin restore_reserve_on_error(h, vma, haddr, page); 56856bda666aSChristoph Lameter put_page(page); 56866bda666aSChristoph Lameter goto out; 56876bda666aSChristoph Lameter } 5688c7b1850dSMike Kravetz new_pagecache_page = true; 568923be7468SMel Gorman } else { 56906bda666aSChristoph Lameter lock_page(page); 56910fe6e20bSNaoya Horiguchi if (unlikely(anon_vma_prepare(vma))) { 56920fe6e20bSNaoya Horiguchi ret = VM_FAULT_OOM; 56930fe6e20bSNaoya Horiguchi goto backout_unlocked; 569423be7468SMel Gorman } 5695409eb8c2SHillf Danton anon_rmap = 1; 56960fe6e20bSNaoya Horiguchi } 56970fe6e20bSNaoya Horiguchi } else { 569857303d80SAndy Whitcroft /* 5699998b4382SNaoya Horiguchi * If memory error occurs between mmap() and fault, some process 5700998b4382SNaoya Horiguchi * don't have hwpoisoned swap entry for errored virtual address. 5701998b4382SNaoya Horiguchi * So we need to block hugepage fault by PG_hwpoison bit check. 5702fd6a03edSNaoya Horiguchi */ 5703fd6a03edSNaoya Horiguchi if (unlikely(PageHWPoison(page))) { 57040eb98f15SMiaohe Lin ret = VM_FAULT_HWPOISON_LARGE | 5705972dc4deSAneesh Kumar K.V VM_FAULT_SET_HINDEX(hstate_index(h)); 5706fd6a03edSNaoya Horiguchi goto backout_unlocked; 57076bda666aSChristoph Lameter } 57087677f7fdSAxel Rasmussen 57097677f7fdSAxel Rasmussen /* Check for page in userfault range. */ 57107677f7fdSAxel Rasmussen if (userfaultfd_minor(vma)) { 57117677f7fdSAxel Rasmussen unlock_page(page); 57127677f7fdSAxel Rasmussen put_page(page); 57132ea7ff1eSPeter Xu /* See comment in userfaultfd_missing() block above */ 57142ea7ff1eSPeter Xu if (!hugetlb_pte_stable(h, mm, ptep, old_pte)) { 57152ea7ff1eSPeter Xu ret = 0; 57162ea7ff1eSPeter Xu goto out; 57172ea7ff1eSPeter Xu } 57182ea7ff1eSPeter Xu return hugetlb_handle_userfault(vma, mapping, idx, flags, 57192ea7ff1eSPeter Xu haddr, address, 57207677f7fdSAxel Rasmussen VM_UFFD_MINOR); 57217677f7fdSAxel Rasmussen } 5722998b4382SNaoya Horiguchi } 57231e8f889bSDavid Gibson 572457303d80SAndy Whitcroft /* 572557303d80SAndy Whitcroft * If we are going to COW a private mapping later, we examine the 572657303d80SAndy Whitcroft * pending reservations for this page now. This will ensure that 572757303d80SAndy Whitcroft * any allocations necessary to record that reservation occur outside 572857303d80SAndy Whitcroft * the spinlock. 572957303d80SAndy Whitcroft */ 57305e911373SMike Kravetz if ((flags & FAULT_FLAG_WRITE) && !(vma->vm_flags & VM_SHARED)) { 5731285b8dcaSHuang Ying if (vma_needs_reservation(h, vma, haddr) < 0) { 57322b26736cSAndy Whitcroft ret = VM_FAULT_OOM; 57332b26736cSAndy Whitcroft goto backout_unlocked; 57342b26736cSAndy Whitcroft } 57355e911373SMike Kravetz /* Just decrements count, does not deallocate */ 5736285b8dcaSHuang Ying vma_end_reservation(h, vma, haddr); 57375e911373SMike Kravetz } 573857303d80SAndy Whitcroft 57398bea8052SAneesh Kumar K.V ptl = huge_pte_lock(h, mm, ptep); 574083c54070SNick Piggin ret = 0; 5741c64e912cSPeter Xu /* If pte changed from under us, retry */ 5742c64e912cSPeter Xu if (!pte_same(huge_ptep_get(ptep), old_pte)) 57434c887265SAdam Litke goto backout; 57444c887265SAdam Litke 57454781593dSPeter Xu if (anon_rmap) 5746285b8dcaSHuang Ying hugepage_add_new_anon_rmap(page, vma, haddr); 57474781593dSPeter Xu else 5748fb3d824dSDavid Hildenbrand page_dup_file_rmap(page, true); 57491e8f889bSDavid Gibson new_pte = make_huge_pte(vma, page, ((vma->vm_flags & VM_WRITE) 57501e8f889bSDavid Gibson && (vma->vm_flags & VM_SHARED))); 5751c64e912cSPeter Xu /* 5752c64e912cSPeter Xu * If this pte was previously wr-protected, keep it wr-protected even 5753c64e912cSPeter Xu * if populated. 5754c64e912cSPeter Xu */ 5755c64e912cSPeter Xu if (unlikely(pte_marker_uffd_wp(old_pte))) 5756c64e912cSPeter Xu new_pte = huge_pte_wrprotect(huge_pte_mkuffd_wp(new_pte)); 5757285b8dcaSHuang Ying set_huge_pte_at(mm, haddr, ptep, new_pte); 57581e8f889bSDavid Gibson 57595d317b2bSNaoya Horiguchi hugetlb_count_add(pages_per_huge_page(h), mm); 5760788c7df4SHugh Dickins if ((flags & FAULT_FLAG_WRITE) && !(vma->vm_flags & VM_SHARED)) { 57611e8f889bSDavid Gibson /* Optimization, do the COW without a second fault */ 5762c89357e2SDavid Hildenbrand ret = hugetlb_wp(mm, vma, address, ptep, flags, page, ptl); 57631e8f889bSDavid Gibson } 57641e8f889bSDavid Gibson 5765cb900f41SKirill A. Shutemov spin_unlock(ptl); 5766cb6acd01SMike Kravetz 5767cb6acd01SMike Kravetz /* 57688f251a3dSMike Kravetz * Only set HPageMigratable in newly allocated pages. Existing pages 57698f251a3dSMike Kravetz * found in the pagecache may not have HPageMigratableset if they have 57708f251a3dSMike Kravetz * been isolated for migration. 5771cb6acd01SMike Kravetz */ 5772cb6acd01SMike Kravetz if (new_page) 57738f251a3dSMike Kravetz SetHPageMigratable(page); 5774cb6acd01SMike Kravetz 57754c887265SAdam Litke unlock_page(page); 57764c887265SAdam Litke out: 5777958f32ceSLiu Shixin hugetlb_vma_unlock_read(vma); 5778958f32ceSLiu Shixin mutex_unlock(&hugetlb_fault_mutex_table[hash]); 5779ac9b9c66SHugh Dickins return ret; 57804c887265SAdam Litke 57814c887265SAdam Litke backout: 5782cb900f41SKirill A. Shutemov spin_unlock(ptl); 57832b26736cSAndy Whitcroft backout_unlocked: 5784c7b1850dSMike Kravetz if (new_page && !new_pagecache_page) 5785285b8dcaSHuang Ying restore_reserve_on_error(h, vma, haddr, page); 5786fa27759aSMike Kravetz 5787fa27759aSMike Kravetz unlock_page(page); 57884c887265SAdam Litke put_page(page); 57894c887265SAdam Litke goto out; 5790ac9b9c66SHugh Dickins } 5791ac9b9c66SHugh Dickins 57928382d914SDavidlohr Bueso #ifdef CONFIG_SMP 5793188b04a7SWei Yang u32 hugetlb_fault_mutex_hash(struct address_space *mapping, pgoff_t idx) 57948382d914SDavidlohr Bueso { 57958382d914SDavidlohr Bueso unsigned long key[2]; 57968382d914SDavidlohr Bueso u32 hash; 57978382d914SDavidlohr Bueso 57988382d914SDavidlohr Bueso key[0] = (unsigned long) mapping; 57998382d914SDavidlohr Bueso key[1] = idx; 58008382d914SDavidlohr Bueso 580155254636SMike Kravetz hash = jhash2((u32 *)&key, sizeof(key)/(sizeof(u32)), 0); 58028382d914SDavidlohr Bueso 58038382d914SDavidlohr Bueso return hash & (num_fault_mutexes - 1); 58048382d914SDavidlohr Bueso } 58058382d914SDavidlohr Bueso #else 58068382d914SDavidlohr Bueso /* 58076c26d310SMiaohe Lin * For uniprocessor systems we always use a single mutex, so just 58088382d914SDavidlohr Bueso * return 0 and avoid the hashing overhead. 58098382d914SDavidlohr Bueso */ 5810188b04a7SWei Yang u32 hugetlb_fault_mutex_hash(struct address_space *mapping, pgoff_t idx) 58118382d914SDavidlohr Bueso { 58128382d914SDavidlohr Bueso return 0; 58138382d914SDavidlohr Bueso } 58148382d914SDavidlohr Bueso #endif 58158382d914SDavidlohr Bueso 58162b740303SSouptick Joarder vm_fault_t hugetlb_fault(struct mm_struct *mm, struct vm_area_struct *vma, 5817788c7df4SHugh Dickins unsigned long address, unsigned int flags) 581886e5216fSAdam Litke { 58198382d914SDavidlohr Bueso pte_t *ptep, entry; 5820cb900f41SKirill A. Shutemov spinlock_t *ptl; 58212b740303SSouptick Joarder vm_fault_t ret; 58228382d914SDavidlohr Bueso u32 hash; 58238382d914SDavidlohr Bueso pgoff_t idx; 58240fe6e20bSNaoya Horiguchi struct page *page = NULL; 582557303d80SAndy Whitcroft struct page *pagecache_page = NULL; 5826a5516438SAndi Kleen struct hstate *h = hstate_vma(vma); 58278382d914SDavidlohr Bueso struct address_space *mapping; 58280f792cf9SNaoya Horiguchi int need_wait_lock = 0; 5829285b8dcaSHuang Ying unsigned long haddr = address & huge_page_mask(h); 583086e5216fSAdam Litke 5831285b8dcaSHuang Ying ptep = huge_pte_offset(mm, haddr, huge_page_size(h)); 5832fd6a03edSNaoya Horiguchi if (ptep) { 583340549ba8SMike Kravetz /* 583440549ba8SMike Kravetz * Since we hold no locks, ptep could be stale. That is 583540549ba8SMike Kravetz * OK as we are only making decisions based on content and 583640549ba8SMike Kravetz * not actually modifying content here. 583740549ba8SMike Kravetz */ 5838fd6a03edSNaoya Horiguchi entry = huge_ptep_get(ptep); 5839290408d4SNaoya Horiguchi if (unlikely(is_hugetlb_entry_migration(entry))) { 5840ad1ac596SMiaohe Lin migration_entry_wait_huge(vma, ptep); 5841290408d4SNaoya Horiguchi return 0; 5842290408d4SNaoya Horiguchi } else if (unlikely(is_hugetlb_entry_hwpoisoned(entry))) 5843aa50d3a7SAndi Kleen return VM_FAULT_HWPOISON_LARGE | 5844972dc4deSAneesh Kumar K.V VM_FAULT_SET_HINDEX(hstate_index(h)); 5845c0d0381aSMike Kravetz } 5846ddeaab32SMike Kravetz 58473935baa9SDavid Gibson /* 58483935baa9SDavid Gibson * Serialize hugepage allocation and instantiation, so that we don't 58493935baa9SDavid Gibson * get spurious allocation failures if two CPUs race to instantiate 58503935baa9SDavid Gibson * the same page in the page cache. 58513935baa9SDavid Gibson */ 585240549ba8SMike Kravetz mapping = vma->vm_file->f_mapping; 585340549ba8SMike Kravetz idx = vma_hugecache_offset(h, vma, haddr); 5854188b04a7SWei Yang hash = hugetlb_fault_mutex_hash(mapping, idx); 5855c672c7f2SMike Kravetz mutex_lock(&hugetlb_fault_mutex_table[hash]); 58568382d914SDavidlohr Bueso 585740549ba8SMike Kravetz /* 585840549ba8SMike Kravetz * Acquire vma lock before calling huge_pte_alloc and hold 585940549ba8SMike Kravetz * until finished with ptep. This prevents huge_pmd_unshare from 586040549ba8SMike Kravetz * being called elsewhere and making the ptep no longer valid. 586140549ba8SMike Kravetz * 586240549ba8SMike Kravetz * ptep could have already be assigned via huge_pte_offset. That 586340549ba8SMike Kravetz * is OK, as huge_pte_alloc will return the same value unless 586440549ba8SMike Kravetz * something has changed. 586540549ba8SMike Kravetz */ 586640549ba8SMike Kravetz hugetlb_vma_lock_read(vma); 586740549ba8SMike Kravetz ptep = huge_pte_alloc(mm, vma, haddr, huge_page_size(h)); 586840549ba8SMike Kravetz if (!ptep) { 586940549ba8SMike Kravetz hugetlb_vma_unlock_read(vma); 587040549ba8SMike Kravetz mutex_unlock(&hugetlb_fault_mutex_table[hash]); 587140549ba8SMike Kravetz return VM_FAULT_OOM; 587240549ba8SMike Kravetz } 587340549ba8SMike Kravetz 58747f2e9525SGerald Schaefer entry = huge_ptep_get(ptep); 5875c64e912cSPeter Xu /* PTE markers should be handled the same way as none pte */ 5876958f32ceSLiu Shixin if (huge_pte_none_mostly(entry)) 5877958f32ceSLiu Shixin /* 5878958f32ceSLiu Shixin * hugetlb_no_page will drop vma lock and hugetlb fault 5879958f32ceSLiu Shixin * mutex internally, which make us return immediately. 5880958f32ceSLiu Shixin */ 5881958f32ceSLiu Shixin return hugetlb_no_page(mm, vma, mapping, idx, address, ptep, 5882c64e912cSPeter Xu entry, flags); 588386e5216fSAdam Litke 588483c54070SNick Piggin ret = 0; 58851e8f889bSDavid Gibson 588657303d80SAndy Whitcroft /* 58870f792cf9SNaoya Horiguchi * entry could be a migration/hwpoison entry at this point, so this 58880f792cf9SNaoya Horiguchi * check prevents the kernel from going below assuming that we have 58897c8de358SEthon Paul * an active hugepage in pagecache. This goto expects the 2nd page 58907c8de358SEthon Paul * fault, and is_hugetlb_entry_(migration|hwpoisoned) check will 58917c8de358SEthon Paul * properly handle it. 58920f792cf9SNaoya Horiguchi */ 58930f792cf9SNaoya Horiguchi if (!pte_present(entry)) 58940f792cf9SNaoya Horiguchi goto out_mutex; 58950f792cf9SNaoya Horiguchi 58960f792cf9SNaoya Horiguchi /* 5897c89357e2SDavid Hildenbrand * If we are going to COW/unshare the mapping later, we examine the 5898c89357e2SDavid Hildenbrand * pending reservations for this page now. This will ensure that any 589957303d80SAndy Whitcroft * allocations necessary to record that reservation occur outside the 59001d8d1464SDavid Hildenbrand * spinlock. Also lookup the pagecache page now as it is used to 59011d8d1464SDavid Hildenbrand * determine if a reservation has been consumed. 590257303d80SAndy Whitcroft */ 5903c89357e2SDavid Hildenbrand if ((flags & (FAULT_FLAG_WRITE|FAULT_FLAG_UNSHARE)) && 59041d8d1464SDavid Hildenbrand !(vma->vm_flags & VM_MAYSHARE) && !huge_pte_write(entry)) { 5905285b8dcaSHuang Ying if (vma_needs_reservation(h, vma, haddr) < 0) { 59062b26736cSAndy Whitcroft ret = VM_FAULT_OOM; 5907b4d1d99fSDavid Gibson goto out_mutex; 59082b26736cSAndy Whitcroft } 59095e911373SMike Kravetz /* Just decrements count, does not deallocate */ 5910285b8dcaSHuang Ying vma_end_reservation(h, vma, haddr); 591157303d80SAndy Whitcroft 591229be8426SMiaohe Lin pagecache_page = find_lock_page(mapping, idx); 591357303d80SAndy Whitcroft } 591457303d80SAndy Whitcroft 59150f792cf9SNaoya Horiguchi ptl = huge_pte_lock(h, mm, ptep); 59160fe6e20bSNaoya Horiguchi 5917c89357e2SDavid Hildenbrand /* Check for a racing update before calling hugetlb_wp() */ 5918b4d1d99fSDavid Gibson if (unlikely(!pte_same(entry, huge_ptep_get(ptep)))) 5919cb900f41SKirill A. Shutemov goto out_ptl; 5920b4d1d99fSDavid Gibson 5921166f3eccSPeter Xu /* Handle userfault-wp first, before trying to lock more pages */ 5922166f3eccSPeter Xu if (userfaultfd_wp(vma) && huge_pte_uffd_wp(huge_ptep_get(ptep)) && 5923166f3eccSPeter Xu (flags & FAULT_FLAG_WRITE) && !huge_pte_write(entry)) { 5924166f3eccSPeter Xu struct vm_fault vmf = { 5925166f3eccSPeter Xu .vma = vma, 5926166f3eccSPeter Xu .address = haddr, 5927166f3eccSPeter Xu .real_address = address, 5928166f3eccSPeter Xu .flags = flags, 5929166f3eccSPeter Xu }; 5930166f3eccSPeter Xu 5931166f3eccSPeter Xu spin_unlock(ptl); 5932166f3eccSPeter Xu if (pagecache_page) { 5933166f3eccSPeter Xu unlock_page(pagecache_page); 5934166f3eccSPeter Xu put_page(pagecache_page); 5935166f3eccSPeter Xu } 593640549ba8SMike Kravetz hugetlb_vma_unlock_read(vma); 5937166f3eccSPeter Xu mutex_unlock(&hugetlb_fault_mutex_table[hash]); 5938166f3eccSPeter Xu return handle_userfault(&vmf, VM_UFFD_WP); 5939166f3eccSPeter Xu } 5940166f3eccSPeter Xu 59410f792cf9SNaoya Horiguchi /* 5942c89357e2SDavid Hildenbrand * hugetlb_wp() requires page locks of pte_page(entry) and 59430f792cf9SNaoya Horiguchi * pagecache_page, so here we need take the former one 59440f792cf9SNaoya Horiguchi * when page != pagecache_page or !pagecache_page. 59450f792cf9SNaoya Horiguchi */ 59460f792cf9SNaoya Horiguchi page = pte_page(entry); 59470f792cf9SNaoya Horiguchi if (page != pagecache_page) 59480f792cf9SNaoya Horiguchi if (!trylock_page(page)) { 59490f792cf9SNaoya Horiguchi need_wait_lock = 1; 59500f792cf9SNaoya Horiguchi goto out_ptl; 59510f792cf9SNaoya Horiguchi } 59520f792cf9SNaoya Horiguchi 59530f792cf9SNaoya Horiguchi get_page(page); 5954b4d1d99fSDavid Gibson 5955c89357e2SDavid Hildenbrand if (flags & (FAULT_FLAG_WRITE|FAULT_FLAG_UNSHARE)) { 5956106c992aSGerald Schaefer if (!huge_pte_write(entry)) { 5957c89357e2SDavid Hildenbrand ret = hugetlb_wp(mm, vma, address, ptep, flags, 5958cb900f41SKirill A. Shutemov pagecache_page, ptl); 59590f792cf9SNaoya Horiguchi goto out_put_page; 5960c89357e2SDavid Hildenbrand } else if (likely(flags & FAULT_FLAG_WRITE)) { 5961106c992aSGerald Schaefer entry = huge_pte_mkdirty(entry); 5962b4d1d99fSDavid Gibson } 5963c89357e2SDavid Hildenbrand } 5964b4d1d99fSDavid Gibson entry = pte_mkyoung(entry); 5965285b8dcaSHuang Ying if (huge_ptep_set_access_flags(vma, haddr, ptep, entry, 5966788c7df4SHugh Dickins flags & FAULT_FLAG_WRITE)) 5967285b8dcaSHuang Ying update_mmu_cache(vma, haddr, ptep); 59680f792cf9SNaoya Horiguchi out_put_page: 59690f792cf9SNaoya Horiguchi if (page != pagecache_page) 59700f792cf9SNaoya Horiguchi unlock_page(page); 59710f792cf9SNaoya Horiguchi put_page(page); 5972cb900f41SKirill A. Shutemov out_ptl: 5973cb900f41SKirill A. Shutemov spin_unlock(ptl); 597457303d80SAndy Whitcroft 597557303d80SAndy Whitcroft if (pagecache_page) { 597657303d80SAndy Whitcroft unlock_page(pagecache_page); 597757303d80SAndy Whitcroft put_page(pagecache_page); 597857303d80SAndy Whitcroft } 5979b4d1d99fSDavid Gibson out_mutex: 598040549ba8SMike Kravetz hugetlb_vma_unlock_read(vma); 5981c672c7f2SMike Kravetz mutex_unlock(&hugetlb_fault_mutex_table[hash]); 59820f792cf9SNaoya Horiguchi /* 59830f792cf9SNaoya Horiguchi * Generally it's safe to hold refcount during waiting page lock. But 59840f792cf9SNaoya Horiguchi * here we just wait to defer the next page fault to avoid busy loop and 59850f792cf9SNaoya Horiguchi * the page is not used after unlocked before returning from the current 59860f792cf9SNaoya Horiguchi * page fault. So we are safe from accessing freed page, even if we wait 59870f792cf9SNaoya Horiguchi * here without taking refcount. 59880f792cf9SNaoya Horiguchi */ 59890f792cf9SNaoya Horiguchi if (need_wait_lock) 59900f792cf9SNaoya Horiguchi wait_on_page_locked(page); 59911e8f889bSDavid Gibson return ret; 599286e5216fSAdam Litke } 599386e5216fSAdam Litke 5994714c1891SAxel Rasmussen #ifdef CONFIG_USERFAULTFD 59958fb5debcSMike Kravetz /* 59968fb5debcSMike Kravetz * Used by userfaultfd UFFDIO_COPY. Based on mcopy_atomic_pte with 59978fb5debcSMike Kravetz * modifications for huge pages. 59988fb5debcSMike Kravetz */ 59998fb5debcSMike Kravetz int hugetlb_mcopy_atomic_pte(struct mm_struct *dst_mm, 60008fb5debcSMike Kravetz pte_t *dst_pte, 60018fb5debcSMike Kravetz struct vm_area_struct *dst_vma, 60028fb5debcSMike Kravetz unsigned long dst_addr, 60038fb5debcSMike Kravetz unsigned long src_addr, 6004f6191471SAxel Rasmussen enum mcopy_atomic_mode mode, 60056041c691SPeter Xu struct page **pagep, 60066041c691SPeter Xu bool wp_copy) 60078fb5debcSMike Kravetz { 6008f6191471SAxel Rasmussen bool is_continue = (mode == MCOPY_ATOMIC_CONTINUE); 60098cc5fcbbSMina Almasry struct hstate *h = hstate_vma(dst_vma); 60108cc5fcbbSMina Almasry struct address_space *mapping = dst_vma->vm_file->f_mapping; 60118cc5fcbbSMina Almasry pgoff_t idx = vma_hugecache_offset(h, dst_vma, dst_addr); 60121e392147SAndrea Arcangeli unsigned long size; 60131c9e8defSMike Kravetz int vm_shared = dst_vma->vm_flags & VM_SHARED; 60148fb5debcSMike Kravetz pte_t _dst_pte; 60158fb5debcSMike Kravetz spinlock_t *ptl; 60168cc5fcbbSMina Almasry int ret = -ENOMEM; 60178fb5debcSMike Kravetz struct page *page; 6018f6191471SAxel Rasmussen int writable; 6019cc30042dSMina Almasry bool page_in_pagecache = false; 60208fb5debcSMike Kravetz 6021f6191471SAxel Rasmussen if (is_continue) { 6022f6191471SAxel Rasmussen ret = -EFAULT; 6023f6191471SAxel Rasmussen page = find_lock_page(mapping, idx); 6024f6191471SAxel Rasmussen if (!page) 6025f6191471SAxel Rasmussen goto out; 6026cc30042dSMina Almasry page_in_pagecache = true; 6027f6191471SAxel Rasmussen } else if (!*pagep) { 6028d84cf06eSMina Almasry /* If a page already exists, then it's UFFDIO_COPY for 6029d84cf06eSMina Almasry * a non-missing case. Return -EEXIST. 6030d84cf06eSMina Almasry */ 6031d84cf06eSMina Almasry if (vm_shared && 6032d84cf06eSMina Almasry hugetlbfs_pagecache_present(h, dst_vma, dst_addr)) { 6033d84cf06eSMina Almasry ret = -EEXIST; 60348fb5debcSMike Kravetz goto out; 6035d84cf06eSMina Almasry } 6036d84cf06eSMina Almasry 6037d84cf06eSMina Almasry page = alloc_huge_page(dst_vma, dst_addr, 0); 6038d84cf06eSMina Almasry if (IS_ERR(page)) { 6039d84cf06eSMina Almasry ret = -ENOMEM; 6040d84cf06eSMina Almasry goto out; 6041d84cf06eSMina Almasry } 60428fb5debcSMike Kravetz 60438fb5debcSMike Kravetz ret = copy_huge_page_from_user(page, 60448fb5debcSMike Kravetz (const void __user *) src_addr, 6045810a56b9SMike Kravetz pages_per_huge_page(h), false); 60468fb5debcSMike Kravetz 6047c1e8d7c6SMichel Lespinasse /* fallback to copy_from_user outside mmap_lock */ 60488fb5debcSMike Kravetz if (unlikely(ret)) { 60499e368259SAndrea Arcangeli ret = -ENOENT; 60508cc5fcbbSMina Almasry /* Free the allocated page which may have 60518cc5fcbbSMina Almasry * consumed a reservation. 60528cc5fcbbSMina Almasry */ 60538cc5fcbbSMina Almasry restore_reserve_on_error(h, dst_vma, dst_addr, page); 60548cc5fcbbSMina Almasry put_page(page); 60558cc5fcbbSMina Almasry 60568cc5fcbbSMina Almasry /* Allocate a temporary page to hold the copied 60578cc5fcbbSMina Almasry * contents. 60588cc5fcbbSMina Almasry */ 60598cc5fcbbSMina Almasry page = alloc_huge_page_vma(h, dst_vma, dst_addr); 60608cc5fcbbSMina Almasry if (!page) { 60618cc5fcbbSMina Almasry ret = -ENOMEM; 60628cc5fcbbSMina Almasry goto out; 60638cc5fcbbSMina Almasry } 60648fb5debcSMike Kravetz *pagep = page; 60658cc5fcbbSMina Almasry /* Set the outparam pagep and return to the caller to 60668cc5fcbbSMina Almasry * copy the contents outside the lock. Don't free the 60678cc5fcbbSMina Almasry * page. 60688cc5fcbbSMina Almasry */ 60698fb5debcSMike Kravetz goto out; 60708fb5debcSMike Kravetz } 60718fb5debcSMike Kravetz } else { 60728cc5fcbbSMina Almasry if (vm_shared && 60738cc5fcbbSMina Almasry hugetlbfs_pagecache_present(h, dst_vma, dst_addr)) { 60748cc5fcbbSMina Almasry put_page(*pagep); 60758cc5fcbbSMina Almasry ret = -EEXIST; 60768cc5fcbbSMina Almasry *pagep = NULL; 60778cc5fcbbSMina Almasry goto out; 60788cc5fcbbSMina Almasry } 60798cc5fcbbSMina Almasry 60808cc5fcbbSMina Almasry page = alloc_huge_page(dst_vma, dst_addr, 0); 60818cc5fcbbSMina Almasry if (IS_ERR(page)) { 6082da9a298fSMiaohe Lin put_page(*pagep); 60838cc5fcbbSMina Almasry ret = -ENOMEM; 60848cc5fcbbSMina Almasry *pagep = NULL; 60858cc5fcbbSMina Almasry goto out; 60868cc5fcbbSMina Almasry } 608734892366SMuchun Song copy_user_huge_page(page, *pagep, dst_addr, dst_vma, 608834892366SMuchun Song pages_per_huge_page(h)); 60898cc5fcbbSMina Almasry put_page(*pagep); 60908fb5debcSMike Kravetz *pagep = NULL; 60918fb5debcSMike Kravetz } 60928fb5debcSMike Kravetz 60938fb5debcSMike Kravetz /* 60948fb5debcSMike Kravetz * The memory barrier inside __SetPageUptodate makes sure that 60958fb5debcSMike Kravetz * preceding stores to the page contents become visible before 60968fb5debcSMike Kravetz * the set_pte_at() write. 60978fb5debcSMike Kravetz */ 60988fb5debcSMike Kravetz __SetPageUptodate(page); 60998fb5debcSMike Kravetz 6100f6191471SAxel Rasmussen /* Add shared, newly allocated pages to the page cache. */ 6101f6191471SAxel Rasmussen if (vm_shared && !is_continue) { 61021e392147SAndrea Arcangeli size = i_size_read(mapping->host) >> huge_page_shift(h); 61031e392147SAndrea Arcangeli ret = -EFAULT; 61041e392147SAndrea Arcangeli if (idx >= size) 61051e392147SAndrea Arcangeli goto out_release_nounlock; 61061c9e8defSMike Kravetz 61071e392147SAndrea Arcangeli /* 61081e392147SAndrea Arcangeli * Serialization between remove_inode_hugepages() and 61097e1813d4SMike Kravetz * hugetlb_add_to_page_cache() below happens through the 61101e392147SAndrea Arcangeli * hugetlb_fault_mutex_table that here must be hold by 61111e392147SAndrea Arcangeli * the caller. 61121e392147SAndrea Arcangeli */ 61137e1813d4SMike Kravetz ret = hugetlb_add_to_page_cache(page, mapping, idx); 61141c9e8defSMike Kravetz if (ret) 61151c9e8defSMike Kravetz goto out_release_nounlock; 6116cc30042dSMina Almasry page_in_pagecache = true; 61171c9e8defSMike Kravetz } 61181c9e8defSMike Kravetz 6119bcc66543SMiaohe Lin ptl = huge_pte_lock(h, dst_mm, dst_pte); 61208fb5debcSMike Kravetz 61218625147cSJames Houghton ret = -EIO; 61228625147cSJames Houghton if (PageHWPoison(page)) 61238625147cSJames Houghton goto out_release_unlock; 61248625147cSJames Houghton 61251e392147SAndrea Arcangeli /* 61266041c691SPeter Xu * We allow to overwrite a pte marker: consider when both MISSING|WP 61276041c691SPeter Xu * registered, we firstly wr-protect a none pte which has no page cache 61286041c691SPeter Xu * page backing it, then access the page. 61296041c691SPeter Xu */ 6130fa27759aSMike Kravetz ret = -EEXIST; 61316041c691SPeter Xu if (!huge_pte_none_mostly(huge_ptep_get(dst_pte))) 61328fb5debcSMike Kravetz goto out_release_unlock; 61338fb5debcSMike Kravetz 61344781593dSPeter Xu if (page_in_pagecache) 6135fb3d824dSDavid Hildenbrand page_dup_file_rmap(page, true); 61364781593dSPeter Xu else 61378fb5debcSMike Kravetz hugepage_add_new_anon_rmap(page, dst_vma, dst_addr); 61388fb5debcSMike Kravetz 61396041c691SPeter Xu /* 61406041c691SPeter Xu * For either: (1) CONTINUE on a non-shared VMA, or (2) UFFDIO_COPY 61416041c691SPeter Xu * with wp flag set, don't set pte write bit. 61426041c691SPeter Xu */ 61436041c691SPeter Xu if (wp_copy || (is_continue && !vm_shared)) 6144f6191471SAxel Rasmussen writable = 0; 6145f6191471SAxel Rasmussen else 6146f6191471SAxel Rasmussen writable = dst_vma->vm_flags & VM_WRITE; 6147f6191471SAxel Rasmussen 6148f6191471SAxel Rasmussen _dst_pte = make_huge_pte(dst_vma, page, writable); 61496041c691SPeter Xu /* 61506041c691SPeter Xu * Always mark UFFDIO_COPY page dirty; note that this may not be 61516041c691SPeter Xu * extremely important for hugetlbfs for now since swapping is not 61526041c691SPeter Xu * supported, but we should still be clear in that this page cannot be 61536041c691SPeter Xu * thrown away at will, even if write bit not set. 61546041c691SPeter Xu */ 61558fb5debcSMike Kravetz _dst_pte = huge_pte_mkdirty(_dst_pte); 61568fb5debcSMike Kravetz _dst_pte = pte_mkyoung(_dst_pte); 61578fb5debcSMike Kravetz 61586041c691SPeter Xu if (wp_copy) 61596041c691SPeter Xu _dst_pte = huge_pte_mkuffd_wp(_dst_pte); 61606041c691SPeter Xu 61618fb5debcSMike Kravetz set_huge_pte_at(dst_mm, dst_addr, dst_pte, _dst_pte); 61628fb5debcSMike Kravetz 61638fb5debcSMike Kravetz hugetlb_count_add(pages_per_huge_page(h), dst_mm); 61648fb5debcSMike Kravetz 61658fb5debcSMike Kravetz /* No need to invalidate - it was non-present before */ 61668fb5debcSMike Kravetz update_mmu_cache(dst_vma, dst_addr, dst_pte); 61678fb5debcSMike Kravetz 61688fb5debcSMike Kravetz spin_unlock(ptl); 6169f6191471SAxel Rasmussen if (!is_continue) 61708f251a3dSMike Kravetz SetHPageMigratable(page); 6171f6191471SAxel Rasmussen if (vm_shared || is_continue) 61721c9e8defSMike Kravetz unlock_page(page); 61738fb5debcSMike Kravetz ret = 0; 61748fb5debcSMike Kravetz out: 61758fb5debcSMike Kravetz return ret; 61768fb5debcSMike Kravetz out_release_unlock: 61778fb5debcSMike Kravetz spin_unlock(ptl); 6178f6191471SAxel Rasmussen if (vm_shared || is_continue) 61791c9e8defSMike Kravetz unlock_page(page); 61805af10dfdSAndrea Arcangeli out_release_nounlock: 6181cc30042dSMina Almasry if (!page_in_pagecache) 6182846be085SMike Kravetz restore_reserve_on_error(h, dst_vma, dst_addr, page); 61838fb5debcSMike Kravetz put_page(page); 61848fb5debcSMike Kravetz goto out; 61858fb5debcSMike Kravetz } 6186714c1891SAxel Rasmussen #endif /* CONFIG_USERFAULTFD */ 61878fb5debcSMike Kravetz 618882e5d378SJoao Martins static void record_subpages_vmas(struct page *page, struct vm_area_struct *vma, 618982e5d378SJoao Martins int refs, struct page **pages, 619082e5d378SJoao Martins struct vm_area_struct **vmas) 619182e5d378SJoao Martins { 619282e5d378SJoao Martins int nr; 619382e5d378SJoao Martins 619482e5d378SJoao Martins for (nr = 0; nr < refs; nr++) { 619582e5d378SJoao Martins if (likely(pages)) 619614455eabSCheng Li pages[nr] = nth_page(page, nr); 619782e5d378SJoao Martins if (vmas) 619882e5d378SJoao Martins vmas[nr] = vma; 619982e5d378SJoao Martins } 620082e5d378SJoao Martins } 620182e5d378SJoao Martins 620284209e87SDavid Hildenbrand static inline bool __follow_hugetlb_must_fault(struct vm_area_struct *vma, 620384209e87SDavid Hildenbrand unsigned int flags, pte_t *pte, 6204a7f22660SDavid Hildenbrand bool *unshare) 6205a7f22660SDavid Hildenbrand { 6206a7f22660SDavid Hildenbrand pte_t pteval = huge_ptep_get(pte); 6207a7f22660SDavid Hildenbrand 6208a7f22660SDavid Hildenbrand *unshare = false; 6209a7f22660SDavid Hildenbrand if (is_swap_pte(pteval)) 6210a7f22660SDavid Hildenbrand return true; 6211a7f22660SDavid Hildenbrand if (huge_pte_write(pteval)) 6212a7f22660SDavid Hildenbrand return false; 6213a7f22660SDavid Hildenbrand if (flags & FOLL_WRITE) 6214a7f22660SDavid Hildenbrand return true; 621584209e87SDavid Hildenbrand if (gup_must_unshare(vma, flags, pte_page(pteval))) { 6216a7f22660SDavid Hildenbrand *unshare = true; 6217a7f22660SDavid Hildenbrand return true; 6218a7f22660SDavid Hildenbrand } 6219a7f22660SDavid Hildenbrand return false; 6220a7f22660SDavid Hildenbrand } 6221a7f22660SDavid Hildenbrand 622257a196a5SMike Kravetz struct page *hugetlb_follow_page_mask(struct vm_area_struct *vma, 622357a196a5SMike Kravetz unsigned long address, unsigned int flags) 622457a196a5SMike Kravetz { 622557a196a5SMike Kravetz struct hstate *h = hstate_vma(vma); 622657a196a5SMike Kravetz struct mm_struct *mm = vma->vm_mm; 622757a196a5SMike Kravetz unsigned long haddr = address & huge_page_mask(h); 622857a196a5SMike Kravetz struct page *page = NULL; 622957a196a5SMike Kravetz spinlock_t *ptl; 623057a196a5SMike Kravetz pte_t *pte, entry; 623157a196a5SMike Kravetz 623257a196a5SMike Kravetz /* 623357a196a5SMike Kravetz * FOLL_PIN is not supported for follow_page(). Ordinary GUP goes via 623457a196a5SMike Kravetz * follow_hugetlb_page(). 623557a196a5SMike Kravetz */ 623657a196a5SMike Kravetz if (WARN_ON_ONCE(flags & FOLL_PIN)) 623757a196a5SMike Kravetz return NULL; 623857a196a5SMike Kravetz 623957a196a5SMike Kravetz retry: 624057a196a5SMike Kravetz pte = huge_pte_offset(mm, haddr, huge_page_size(h)); 624157a196a5SMike Kravetz if (!pte) 624257a196a5SMike Kravetz return NULL; 624357a196a5SMike Kravetz 624457a196a5SMike Kravetz ptl = huge_pte_lock(h, mm, pte); 624557a196a5SMike Kravetz entry = huge_ptep_get(pte); 624657a196a5SMike Kravetz if (pte_present(entry)) { 624757a196a5SMike Kravetz page = pte_page(entry) + 624857a196a5SMike Kravetz ((address & ~huge_page_mask(h)) >> PAGE_SHIFT); 624957a196a5SMike Kravetz /* 625057a196a5SMike Kravetz * Note that page may be a sub-page, and with vmemmap 625157a196a5SMike Kravetz * optimizations the page struct may be read only. 625257a196a5SMike Kravetz * try_grab_page() will increase the ref count on the 625357a196a5SMike Kravetz * head page, so this will be OK. 625457a196a5SMike Kravetz * 625557a196a5SMike Kravetz * try_grab_page() should always succeed here, because we hold 625657a196a5SMike Kravetz * the ptl lock and have verified pte_present(). 625757a196a5SMike Kravetz */ 625857a196a5SMike Kravetz if (WARN_ON_ONCE(!try_grab_page(page, flags))) { 625957a196a5SMike Kravetz page = NULL; 626057a196a5SMike Kravetz goto out; 626157a196a5SMike Kravetz } 626257a196a5SMike Kravetz } else { 626357a196a5SMike Kravetz if (is_hugetlb_entry_migration(entry)) { 626457a196a5SMike Kravetz spin_unlock(ptl); 626557a196a5SMike Kravetz __migration_entry_wait_huge(pte, ptl); 626657a196a5SMike Kravetz goto retry; 626757a196a5SMike Kravetz } 626857a196a5SMike Kravetz /* 626957a196a5SMike Kravetz * hwpoisoned entry is treated as no_page_table in 627057a196a5SMike Kravetz * follow_page_mask(). 627157a196a5SMike Kravetz */ 627257a196a5SMike Kravetz } 627357a196a5SMike Kravetz out: 627457a196a5SMike Kravetz spin_unlock(ptl); 627557a196a5SMike Kravetz return page; 627657a196a5SMike Kravetz } 627757a196a5SMike Kravetz 627828a35716SMichel Lespinasse long follow_hugetlb_page(struct mm_struct *mm, struct vm_area_struct *vma, 627963551ae0SDavid Gibson struct page **pages, struct vm_area_struct **vmas, 628028a35716SMichel Lespinasse unsigned long *position, unsigned long *nr_pages, 62814f6da934SPeter Xu long i, unsigned int flags, int *locked) 628263551ae0SDavid Gibson { 6283d5d4b0aaSChen, Kenneth W unsigned long pfn_offset; 6284d5d4b0aaSChen, Kenneth W unsigned long vaddr = *position; 628528a35716SMichel Lespinasse unsigned long remainder = *nr_pages; 6286a5516438SAndi Kleen struct hstate *h = hstate_vma(vma); 62870fa5bc40SJoao Martins int err = -EFAULT, refs; 628863551ae0SDavid Gibson 628963551ae0SDavid Gibson while (vaddr < vma->vm_end && remainder) { 629063551ae0SDavid Gibson pte_t *pte; 6291cb900f41SKirill A. Shutemov spinlock_t *ptl = NULL; 6292a7f22660SDavid Hildenbrand bool unshare = false; 62932a15efc9SHugh Dickins int absent; 629463551ae0SDavid Gibson struct page *page; 629563551ae0SDavid Gibson 62964c887265SAdam Litke /* 629702057967SDavid Rientjes * If we have a pending SIGKILL, don't keep faulting pages and 629802057967SDavid Rientjes * potentially allocating memory. 629902057967SDavid Rientjes */ 6300fa45f116SDavidlohr Bueso if (fatal_signal_pending(current)) { 630102057967SDavid Rientjes remainder = 0; 630202057967SDavid Rientjes break; 630302057967SDavid Rientjes } 630402057967SDavid Rientjes 630502057967SDavid Rientjes /* 63064c887265SAdam Litke * Some archs (sparc64, sh*) have multiple pte_ts to 63072a15efc9SHugh Dickins * each hugepage. We have to make sure we get the 63084c887265SAdam Litke * first, for the page indexing below to work. 6309cb900f41SKirill A. Shutemov * 6310cb900f41SKirill A. Shutemov * Note that page table lock is not held when pte is null. 63114c887265SAdam Litke */ 63127868a208SPunit Agrawal pte = huge_pte_offset(mm, vaddr & huge_page_mask(h), 63137868a208SPunit Agrawal huge_page_size(h)); 6314cb900f41SKirill A. Shutemov if (pte) 6315cb900f41SKirill A. Shutemov ptl = huge_pte_lock(h, mm, pte); 63162a15efc9SHugh Dickins absent = !pte || huge_pte_none(huge_ptep_get(pte)); 631763551ae0SDavid Gibson 63182a15efc9SHugh Dickins /* 63192a15efc9SHugh Dickins * When coredumping, it suits get_dump_page if we just return 63203ae77f43SHugh Dickins * an error where there's an empty slot with no huge pagecache 63213ae77f43SHugh Dickins * to back it. This way, we avoid allocating a hugepage, and 63223ae77f43SHugh Dickins * the sparse dumpfile avoids allocating disk blocks, but its 63233ae77f43SHugh Dickins * huge holes still show up with zeroes where they need to be. 63242a15efc9SHugh Dickins */ 63253ae77f43SHugh Dickins if (absent && (flags & FOLL_DUMP) && 63263ae77f43SHugh Dickins !hugetlbfs_pagecache_present(h, vma, vaddr)) { 6327cb900f41SKirill A. Shutemov if (pte) 6328cb900f41SKirill A. Shutemov spin_unlock(ptl); 63292a15efc9SHugh Dickins remainder = 0; 63302a15efc9SHugh Dickins break; 63312a15efc9SHugh Dickins } 63322a15efc9SHugh Dickins 63339cc3a5bdSNaoya Horiguchi /* 63349cc3a5bdSNaoya Horiguchi * We need call hugetlb_fault for both hugepages under migration 63359cc3a5bdSNaoya Horiguchi * (in which case hugetlb_fault waits for the migration,) and 63369cc3a5bdSNaoya Horiguchi * hwpoisoned hugepages (in which case we need to prevent the 63379cc3a5bdSNaoya Horiguchi * caller from accessing to them.) In order to do this, we use 63389cc3a5bdSNaoya Horiguchi * here is_swap_pte instead of is_hugetlb_entry_migration and 63399cc3a5bdSNaoya Horiguchi * is_hugetlb_entry_hwpoisoned. This is because it simply covers 63409cc3a5bdSNaoya Horiguchi * both cases, and because we can't follow correct pages 63419cc3a5bdSNaoya Horiguchi * directly from any kind of swap entries. 63429cc3a5bdSNaoya Horiguchi */ 6343a7f22660SDavid Hildenbrand if (absent || 634484209e87SDavid Hildenbrand __follow_hugetlb_must_fault(vma, flags, pte, &unshare)) { 63452b740303SSouptick Joarder vm_fault_t ret; 634687ffc118SAndrea Arcangeli unsigned int fault_flags = 0; 63474c887265SAdam Litke 6348cb900f41SKirill A. Shutemov if (pte) 6349cb900f41SKirill A. Shutemov spin_unlock(ptl); 635087ffc118SAndrea Arcangeli if (flags & FOLL_WRITE) 635187ffc118SAndrea Arcangeli fault_flags |= FAULT_FLAG_WRITE; 6352a7f22660SDavid Hildenbrand else if (unshare) 6353a7f22660SDavid Hildenbrand fault_flags |= FAULT_FLAG_UNSHARE; 63544f6da934SPeter Xu if (locked) 635571335f37SPeter Xu fault_flags |= FAULT_FLAG_ALLOW_RETRY | 635671335f37SPeter Xu FAULT_FLAG_KILLABLE; 635787ffc118SAndrea Arcangeli if (flags & FOLL_NOWAIT) 635887ffc118SAndrea Arcangeli fault_flags |= FAULT_FLAG_ALLOW_RETRY | 635987ffc118SAndrea Arcangeli FAULT_FLAG_RETRY_NOWAIT; 636087ffc118SAndrea Arcangeli if (flags & FOLL_TRIED) { 63614426e945SPeter Xu /* 63624426e945SPeter Xu * Note: FAULT_FLAG_ALLOW_RETRY and 63634426e945SPeter Xu * FAULT_FLAG_TRIED can co-exist 63644426e945SPeter Xu */ 636587ffc118SAndrea Arcangeli fault_flags |= FAULT_FLAG_TRIED; 636687ffc118SAndrea Arcangeli } 636787ffc118SAndrea Arcangeli ret = hugetlb_fault(mm, vma, vaddr, fault_flags); 636887ffc118SAndrea Arcangeli if (ret & VM_FAULT_ERROR) { 63692be7cfedSDaniel Jordan err = vm_fault_to_errno(ret, flags); 63701c59827dSHugh Dickins remainder = 0; 63711c59827dSHugh Dickins break; 63721c59827dSHugh Dickins } 637387ffc118SAndrea Arcangeli if (ret & VM_FAULT_RETRY) { 63744f6da934SPeter Xu if (locked && 63751ac25013SAndrea Arcangeli !(fault_flags & FAULT_FLAG_RETRY_NOWAIT)) 63764f6da934SPeter Xu *locked = 0; 637787ffc118SAndrea Arcangeli *nr_pages = 0; 637887ffc118SAndrea Arcangeli /* 637987ffc118SAndrea Arcangeli * VM_FAULT_RETRY must not return an 638087ffc118SAndrea Arcangeli * error, it will return zero 638187ffc118SAndrea Arcangeli * instead. 638287ffc118SAndrea Arcangeli * 638387ffc118SAndrea Arcangeli * No need to update "position" as the 638487ffc118SAndrea Arcangeli * caller will not check it after 638587ffc118SAndrea Arcangeli * *nr_pages is set to 0. 638687ffc118SAndrea Arcangeli */ 638787ffc118SAndrea Arcangeli return i; 638887ffc118SAndrea Arcangeli } 638987ffc118SAndrea Arcangeli continue; 639087ffc118SAndrea Arcangeli } 639163551ae0SDavid Gibson 6392a5516438SAndi Kleen pfn_offset = (vaddr & ~huge_page_mask(h)) >> PAGE_SHIFT; 63937f2e9525SGerald Schaefer page = pte_page(huge_ptep_get(pte)); 63948fde12caSLinus Torvalds 6395b6a2619cSDavid Hildenbrand VM_BUG_ON_PAGE((flags & FOLL_PIN) && PageAnon(page) && 6396b6a2619cSDavid Hildenbrand !PageAnonExclusive(page), page); 6397b6a2619cSDavid Hildenbrand 63988fde12caSLinus Torvalds /* 6399acbfb087SZhigang Lu * If subpage information not requested, update counters 6400acbfb087SZhigang Lu * and skip the same_page loop below. 6401acbfb087SZhigang Lu */ 6402acbfb087SZhigang Lu if (!pages && !vmas && !pfn_offset && 6403acbfb087SZhigang Lu (vaddr + huge_page_size(h) < vma->vm_end) && 6404acbfb087SZhigang Lu (remainder >= pages_per_huge_page(h))) { 6405acbfb087SZhigang Lu vaddr += huge_page_size(h); 6406acbfb087SZhigang Lu remainder -= pages_per_huge_page(h); 6407acbfb087SZhigang Lu i += pages_per_huge_page(h); 6408acbfb087SZhigang Lu spin_unlock(ptl); 6409acbfb087SZhigang Lu continue; 6410acbfb087SZhigang Lu } 6411acbfb087SZhigang Lu 6412d08af0a5SJoao Martins /* vaddr may not be aligned to PAGE_SIZE */ 6413d08af0a5SJoao Martins refs = min3(pages_per_huge_page(h) - pfn_offset, remainder, 6414d08af0a5SJoao Martins (vma->vm_end - ALIGN_DOWN(vaddr, PAGE_SIZE)) >> PAGE_SHIFT); 64150fa5bc40SJoao Martins 641682e5d378SJoao Martins if (pages || vmas) 641714455eabSCheng Li record_subpages_vmas(nth_page(page, pfn_offset), 641882e5d378SJoao Martins vma, refs, 641982e5d378SJoao Martins likely(pages) ? pages + i : NULL, 642082e5d378SJoao Martins vmas ? vmas + i : NULL); 642163551ae0SDavid Gibson 642282e5d378SJoao Martins if (pages) { 64230fa5bc40SJoao Martins /* 6424822951d8SMatthew Wilcox (Oracle) * try_grab_folio() should always succeed here, 64250fa5bc40SJoao Martins * because: a) we hold the ptl lock, and b) we've just 64260fa5bc40SJoao Martins * checked that the huge page is present in the page 64270fa5bc40SJoao Martins * tables. If the huge page is present, then the tail 64280fa5bc40SJoao Martins * pages must also be present. The ptl prevents the 64290fa5bc40SJoao Martins * head page and tail pages from being rearranged in 64300fa5bc40SJoao Martins * any way. So this page must be available at this 64310fa5bc40SJoao Martins * point, unless the page refcount overflowed: 64320fa5bc40SJoao Martins */ 6433822951d8SMatthew Wilcox (Oracle) if (WARN_ON_ONCE(!try_grab_folio(pages[i], refs, 64340fa5bc40SJoao Martins flags))) { 64350fa5bc40SJoao Martins spin_unlock(ptl); 64360fa5bc40SJoao Martins remainder = 0; 64370fa5bc40SJoao Martins err = -ENOMEM; 64380fa5bc40SJoao Martins break; 64390fa5bc40SJoao Martins } 6440d5d4b0aaSChen, Kenneth W } 644182e5d378SJoao Martins 644282e5d378SJoao Martins vaddr += (refs << PAGE_SHIFT); 644382e5d378SJoao Martins remainder -= refs; 644482e5d378SJoao Martins i += refs; 644582e5d378SJoao Martins 6446cb900f41SKirill A. Shutemov spin_unlock(ptl); 644763551ae0SDavid Gibson } 644828a35716SMichel Lespinasse *nr_pages = remainder; 644987ffc118SAndrea Arcangeli /* 645087ffc118SAndrea Arcangeli * setting position is actually required only if remainder is 645187ffc118SAndrea Arcangeli * not zero but it's faster not to add a "if (remainder)" 645287ffc118SAndrea Arcangeli * branch. 645387ffc118SAndrea Arcangeli */ 645463551ae0SDavid Gibson *position = vaddr; 645563551ae0SDavid Gibson 64562be7cfedSDaniel Jordan return i ? i : err; 645763551ae0SDavid Gibson } 64588f860591SZhang, Yanmin 64597da4d641SPeter Zijlstra unsigned long hugetlb_change_protection(struct vm_area_struct *vma, 64605a90d5a1SPeter Xu unsigned long address, unsigned long end, 64615a90d5a1SPeter Xu pgprot_t newprot, unsigned long cp_flags) 64628f860591SZhang, Yanmin { 64638f860591SZhang, Yanmin struct mm_struct *mm = vma->vm_mm; 64648f860591SZhang, Yanmin unsigned long start = address; 64658f860591SZhang, Yanmin pte_t *ptep; 64668f860591SZhang, Yanmin pte_t pte; 6467a5516438SAndi Kleen struct hstate *h = hstate_vma(vma); 646860dfaad6SPeter Xu unsigned long pages = 0, psize = huge_page_size(h); 6469dff11abeSMike Kravetz bool shared_pmd = false; 6470ac46d4f3SJérôme Glisse struct mmu_notifier_range range; 6471e95a9851SMike Kravetz unsigned long last_addr_mask; 64725a90d5a1SPeter Xu bool uffd_wp = cp_flags & MM_CP_UFFD_WP; 64735a90d5a1SPeter Xu bool uffd_wp_resolve = cp_flags & MM_CP_UFFD_WP_RESOLVE; 6474dff11abeSMike Kravetz 6475dff11abeSMike Kravetz /* 6476dff11abeSMike Kravetz * In the case of shared PMDs, the area to flush could be beyond 6477ac46d4f3SJérôme Glisse * start/end. Set range.start/range.end to cover the maximum possible 6478dff11abeSMike Kravetz * range if PMD sharing is possible. 6479dff11abeSMike Kravetz */ 64807269f999SJérôme Glisse mmu_notifier_range_init(&range, MMU_NOTIFY_PROTECTION_VMA, 64817269f999SJérôme Glisse 0, vma, mm, start, end); 6482ac46d4f3SJérôme Glisse adjust_range_if_pmd_sharing_possible(vma, &range.start, &range.end); 64838f860591SZhang, Yanmin 64848f860591SZhang, Yanmin BUG_ON(address >= end); 6485ac46d4f3SJérôme Glisse flush_cache_range(vma, range.start, range.end); 64868f860591SZhang, Yanmin 6487ac46d4f3SJérôme Glisse mmu_notifier_invalidate_range_start(&range); 648840549ba8SMike Kravetz hugetlb_vma_lock_write(vma); 648983cde9e8SDavidlohr Bueso i_mmap_lock_write(vma->vm_file->f_mapping); 649040549ba8SMike Kravetz last_addr_mask = hugetlb_mask_last_page(h); 649160dfaad6SPeter Xu for (; address < end; address += psize) { 6492cb900f41SKirill A. Shutemov spinlock_t *ptl; 649360dfaad6SPeter Xu ptep = huge_pte_offset(mm, address, psize); 6494e95a9851SMike Kravetz if (!ptep) { 6495e95a9851SMike Kravetz address |= last_addr_mask; 64968f860591SZhang, Yanmin continue; 6497e95a9851SMike Kravetz } 6498cb900f41SKirill A. Shutemov ptl = huge_pte_lock(h, mm, ptep); 64994ddb4d91SMike Kravetz if (huge_pmd_unshare(mm, vma, address, ptep)) { 650060dfaad6SPeter Xu /* 650160dfaad6SPeter Xu * When uffd-wp is enabled on the vma, unshare 650260dfaad6SPeter Xu * shouldn't happen at all. Warn about it if it 650360dfaad6SPeter Xu * happened due to some reason. 650460dfaad6SPeter Xu */ 650560dfaad6SPeter Xu WARN_ON_ONCE(uffd_wp || uffd_wp_resolve); 65067da4d641SPeter Zijlstra pages++; 6507cb900f41SKirill A. Shutemov spin_unlock(ptl); 6508dff11abeSMike Kravetz shared_pmd = true; 65094ddb4d91SMike Kravetz address |= last_addr_mask; 651039dde65cSChen, Kenneth W continue; 65117da4d641SPeter Zijlstra } 6512a8bda28dSNaoya Horiguchi pte = huge_ptep_get(ptep); 6513a8bda28dSNaoya Horiguchi if (unlikely(is_hugetlb_entry_hwpoisoned(pte))) { 6514a8bda28dSNaoya Horiguchi spin_unlock(ptl); 6515a8bda28dSNaoya Horiguchi continue; 6516a8bda28dSNaoya Horiguchi } 6517a8bda28dSNaoya Horiguchi if (unlikely(is_hugetlb_entry_migration(pte))) { 6518a8bda28dSNaoya Horiguchi swp_entry_t entry = pte_to_swp_entry(pte); 65196c287605SDavid Hildenbrand struct page *page = pfn_swap_entry_to_page(entry); 6520a8bda28dSNaoya Horiguchi 65216c287605SDavid Hildenbrand if (!is_readable_migration_entry(entry)) { 6522a8bda28dSNaoya Horiguchi pte_t newpte; 6523a8bda28dSNaoya Horiguchi 65246c287605SDavid Hildenbrand if (PageAnon(page)) 65256c287605SDavid Hildenbrand entry = make_readable_exclusive_migration_entry( 65266c287605SDavid Hildenbrand swp_offset(entry)); 65276c287605SDavid Hildenbrand else 65284dd845b5SAlistair Popple entry = make_readable_migration_entry( 65294dd845b5SAlistair Popple swp_offset(entry)); 6530a8bda28dSNaoya Horiguchi newpte = swp_entry_to_pte(entry); 65315a90d5a1SPeter Xu if (uffd_wp) 65325a90d5a1SPeter Xu newpte = pte_swp_mkuffd_wp(newpte); 65335a90d5a1SPeter Xu else if (uffd_wp_resolve) 65345a90d5a1SPeter Xu newpte = pte_swp_clear_uffd_wp(newpte); 653518f39629SQi Zheng set_huge_pte_at(mm, address, ptep, newpte); 6536a8bda28dSNaoya Horiguchi pages++; 6537a8bda28dSNaoya Horiguchi } 6538a8bda28dSNaoya Horiguchi spin_unlock(ptl); 6539a8bda28dSNaoya Horiguchi continue; 6540a8bda28dSNaoya Horiguchi } 654160dfaad6SPeter Xu if (unlikely(pte_marker_uffd_wp(pte))) { 654260dfaad6SPeter Xu /* 654360dfaad6SPeter Xu * This is changing a non-present pte into a none pte, 654460dfaad6SPeter Xu * no need for huge_ptep_modify_prot_start/commit(). 654560dfaad6SPeter Xu */ 654660dfaad6SPeter Xu if (uffd_wp_resolve) 654760dfaad6SPeter Xu huge_pte_clear(mm, address, ptep, psize); 654860dfaad6SPeter Xu } 6549a8bda28dSNaoya Horiguchi if (!huge_pte_none(pte)) { 6550023bdd00SAneesh Kumar K.V pte_t old_pte; 655179c1c594SChristophe Leroy unsigned int shift = huge_page_shift(hstate_vma(vma)); 6552023bdd00SAneesh Kumar K.V 6553023bdd00SAneesh Kumar K.V old_pte = huge_ptep_modify_prot_start(vma, address, ptep); 655416785bd7SAnshuman Khandual pte = huge_pte_modify(old_pte, newprot); 655579c1c594SChristophe Leroy pte = arch_make_huge_pte(pte, shift, vma->vm_flags); 65565a90d5a1SPeter Xu if (uffd_wp) 65575a90d5a1SPeter Xu pte = huge_pte_mkuffd_wp(huge_pte_wrprotect(pte)); 65585a90d5a1SPeter Xu else if (uffd_wp_resolve) 65595a90d5a1SPeter Xu pte = huge_pte_clear_uffd_wp(pte); 6560023bdd00SAneesh Kumar K.V huge_ptep_modify_prot_commit(vma, address, ptep, old_pte, pte); 65617da4d641SPeter Zijlstra pages++; 656260dfaad6SPeter Xu } else { 656360dfaad6SPeter Xu /* None pte */ 656460dfaad6SPeter Xu if (unlikely(uffd_wp)) 656560dfaad6SPeter Xu /* Safe to modify directly (none->non-present). */ 656660dfaad6SPeter Xu set_huge_pte_at(mm, address, ptep, 656760dfaad6SPeter Xu make_pte_marker(PTE_MARKER_UFFD_WP)); 65688f860591SZhang, Yanmin } 6569cb900f41SKirill A. Shutemov spin_unlock(ptl); 65708f860591SZhang, Yanmin } 6571d833352aSMel Gorman /* 6572c8c06efaSDavidlohr Bueso * Must flush TLB before releasing i_mmap_rwsem: x86's huge_pmd_unshare 6573d833352aSMel Gorman * may have cleared our pud entry and done put_page on the page table: 6574c8c06efaSDavidlohr Bueso * once we release i_mmap_rwsem, another task can do the final put_page 6575dff11abeSMike Kravetz * and that page table be reused and filled with junk. If we actually 6576dff11abeSMike Kravetz * did unshare a page of pmds, flush the range corresponding to the pud. 6577d833352aSMel Gorman */ 6578dff11abeSMike Kravetz if (shared_pmd) 6579ac46d4f3SJérôme Glisse flush_hugetlb_tlb_range(vma, range.start, range.end); 6580dff11abeSMike Kravetz else 65815491ae7bSAneesh Kumar K.V flush_hugetlb_tlb_range(vma, start, end); 65820f10851eSJérôme Glisse /* 65830f10851eSJérôme Glisse * No need to call mmu_notifier_invalidate_range() we are downgrading 65840f10851eSJérôme Glisse * page table protection not changing it to point to a new page. 65850f10851eSJérôme Glisse * 6586ee65728eSMike Rapoport * See Documentation/mm/mmu_notifier.rst 65870f10851eSJérôme Glisse */ 658883cde9e8SDavidlohr Bueso i_mmap_unlock_write(vma->vm_file->f_mapping); 658940549ba8SMike Kravetz hugetlb_vma_unlock_write(vma); 6590ac46d4f3SJérôme Glisse mmu_notifier_invalidate_range_end(&range); 65917da4d641SPeter Zijlstra 65927da4d641SPeter Zijlstra return pages << h->order; 65938f860591SZhang, Yanmin } 65948f860591SZhang, Yanmin 659533b8f84aSMike Kravetz /* Return true if reservation was successful, false otherwise. */ 659633b8f84aSMike Kravetz bool hugetlb_reserve_pages(struct inode *inode, 6597a1e78772SMel Gorman long from, long to, 65985a6fe125SMel Gorman struct vm_area_struct *vma, 6599ca16d140SKOSAKI Motohiro vm_flags_t vm_flags) 6600e4e574b7SAdam Litke { 660133b8f84aSMike Kravetz long chg, add = -1; 6602a5516438SAndi Kleen struct hstate *h = hstate_inode(inode); 660390481622SDavid Gibson struct hugepage_subpool *spool = subpool_inode(inode); 66049119a41eSJoonsoo Kim struct resv_map *resv_map; 6605075a61d0SMina Almasry struct hugetlb_cgroup *h_cg = NULL; 66060db9d74eSMina Almasry long gbl_reserve, regions_needed = 0; 6607e4e574b7SAdam Litke 660863489f8eSMike Kravetz /* This should never happen */ 660963489f8eSMike Kravetz if (from > to) { 661063489f8eSMike Kravetz VM_WARN(1, "%s called with a negative range\n", __func__); 661133b8f84aSMike Kravetz return false; 661263489f8eSMike Kravetz } 661363489f8eSMike Kravetz 6614a1e78772SMel Gorman /* 66158d9bfb26SMike Kravetz * vma specific semaphore used for pmd sharing synchronization 66168d9bfb26SMike Kravetz */ 66178d9bfb26SMike Kravetz hugetlb_vma_lock_alloc(vma); 66188d9bfb26SMike Kravetz 66198d9bfb26SMike Kravetz /* 662017c9d12eSMel Gorman * Only apply hugepage reservation if asked. At fault time, an 662117c9d12eSMel Gorman * attempt will be made for VM_NORESERVE to allocate a page 662290481622SDavid Gibson * without using reserves 662317c9d12eSMel Gorman */ 6624ca16d140SKOSAKI Motohiro if (vm_flags & VM_NORESERVE) 662533b8f84aSMike Kravetz return true; 662617c9d12eSMel Gorman 662717c9d12eSMel Gorman /* 6628a1e78772SMel Gorman * Shared mappings base their reservation on the number of pages that 6629a1e78772SMel Gorman * are already allocated on behalf of the file. Private mappings need 6630a1e78772SMel Gorman * to reserve the full area even if read-only as mprotect() may be 6631a1e78772SMel Gorman * called to make the mapping read-write. Assume !vma is a shm mapping 6632a1e78772SMel Gorman */ 66339119a41eSJoonsoo Kim if (!vma || vma->vm_flags & VM_MAYSHARE) { 6634f27a5136SMike Kravetz /* 6635f27a5136SMike Kravetz * resv_map can not be NULL as hugetlb_reserve_pages is only 6636f27a5136SMike Kravetz * called for inodes for which resv_maps were created (see 6637f27a5136SMike Kravetz * hugetlbfs_get_inode). 6638f27a5136SMike Kravetz */ 66394e35f483SJoonsoo Kim resv_map = inode_resv_map(inode); 66409119a41eSJoonsoo Kim 66410db9d74eSMina Almasry chg = region_chg(resv_map, from, to, ®ions_needed); 66429119a41eSJoonsoo Kim } else { 6643e9fe92aeSMina Almasry /* Private mapping. */ 66449119a41eSJoonsoo Kim resv_map = resv_map_alloc(); 66455a6fe125SMel Gorman if (!resv_map) 66468d9bfb26SMike Kravetz goto out_err; 66475a6fe125SMel Gorman 664817c9d12eSMel Gorman chg = to - from; 664917c9d12eSMel Gorman 66505a6fe125SMel Gorman set_vma_resv_map(vma, resv_map); 66515a6fe125SMel Gorman set_vma_resv_flags(vma, HPAGE_RESV_OWNER); 66525a6fe125SMel Gorman } 66535a6fe125SMel Gorman 665433b8f84aSMike Kravetz if (chg < 0) 6655c50ac050SDave Hansen goto out_err; 665617c9d12eSMel Gorman 665733b8f84aSMike Kravetz if (hugetlb_cgroup_charge_cgroup_rsvd(hstate_index(h), 665833b8f84aSMike Kravetz chg * pages_per_huge_page(h), &h_cg) < 0) 6659075a61d0SMina Almasry goto out_err; 6660075a61d0SMina Almasry 6661075a61d0SMina Almasry if (vma && !(vma->vm_flags & VM_MAYSHARE) && h_cg) { 6662075a61d0SMina Almasry /* For private mappings, the hugetlb_cgroup uncharge info hangs 6663075a61d0SMina Almasry * of the resv_map. 6664075a61d0SMina Almasry */ 6665075a61d0SMina Almasry resv_map_set_hugetlb_cgroup_uncharge_info(resv_map, h_cg, h); 6666075a61d0SMina Almasry } 6667075a61d0SMina Almasry 66681c5ecae3SMike Kravetz /* 66691c5ecae3SMike Kravetz * There must be enough pages in the subpool for the mapping. If 66701c5ecae3SMike Kravetz * the subpool has a minimum size, there may be some global 66711c5ecae3SMike Kravetz * reservations already in place (gbl_reserve). 66721c5ecae3SMike Kravetz */ 66731c5ecae3SMike Kravetz gbl_reserve = hugepage_subpool_get_pages(spool, chg); 667433b8f84aSMike Kravetz if (gbl_reserve < 0) 6675075a61d0SMina Almasry goto out_uncharge_cgroup; 667617c9d12eSMel Gorman 667717c9d12eSMel Gorman /* 667817c9d12eSMel Gorman * Check enough hugepages are available for the reservation. 667990481622SDavid Gibson * Hand the pages back to the subpool if there are not 668017c9d12eSMel Gorman */ 668133b8f84aSMike Kravetz if (hugetlb_acct_memory(h, gbl_reserve) < 0) 6682075a61d0SMina Almasry goto out_put_pages; 668317c9d12eSMel Gorman 668417c9d12eSMel Gorman /* 668517c9d12eSMel Gorman * Account for the reservations made. Shared mappings record regions 668617c9d12eSMel Gorman * that have reservations as they are shared by multiple VMAs. 668717c9d12eSMel Gorman * When the last VMA disappears, the region map says how much 668817c9d12eSMel Gorman * the reservation was and the page cache tells how much of 668917c9d12eSMel Gorman * the reservation was consumed. Private mappings are per-VMA and 669017c9d12eSMel Gorman * only the consumed reservations are tracked. When the VMA 669117c9d12eSMel Gorman * disappears, the original reservation is the VMA size and the 669217c9d12eSMel Gorman * consumed reservations are stored in the map. Hence, nothing 669317c9d12eSMel Gorman * else has to be done for private mappings here 669417c9d12eSMel Gorman */ 669533039678SMike Kravetz if (!vma || vma->vm_flags & VM_MAYSHARE) { 6696075a61d0SMina Almasry add = region_add(resv_map, from, to, regions_needed, h, h_cg); 669733039678SMike Kravetz 66980db9d74eSMina Almasry if (unlikely(add < 0)) { 66990db9d74eSMina Almasry hugetlb_acct_memory(h, -gbl_reserve); 6700075a61d0SMina Almasry goto out_put_pages; 67010db9d74eSMina Almasry } else if (unlikely(chg > add)) { 670233039678SMike Kravetz /* 670333039678SMike Kravetz * pages in this range were added to the reserve 670433039678SMike Kravetz * map between region_chg and region_add. This 670533039678SMike Kravetz * indicates a race with alloc_huge_page. Adjust 670633039678SMike Kravetz * the subpool and reserve counts modified above 670733039678SMike Kravetz * based on the difference. 670833039678SMike Kravetz */ 670933039678SMike Kravetz long rsv_adjust; 671033039678SMike Kravetz 6711d85aecf2SMiaohe Lin /* 6712d85aecf2SMiaohe Lin * hugetlb_cgroup_uncharge_cgroup_rsvd() will put the 6713d85aecf2SMiaohe Lin * reference to h_cg->css. See comment below for detail. 6714d85aecf2SMiaohe Lin */ 6715075a61d0SMina Almasry hugetlb_cgroup_uncharge_cgroup_rsvd( 6716075a61d0SMina Almasry hstate_index(h), 6717075a61d0SMina Almasry (chg - add) * pages_per_huge_page(h), h_cg); 6718075a61d0SMina Almasry 671933039678SMike Kravetz rsv_adjust = hugepage_subpool_put_pages(spool, 672033039678SMike Kravetz chg - add); 672133039678SMike Kravetz hugetlb_acct_memory(h, -rsv_adjust); 6722d85aecf2SMiaohe Lin } else if (h_cg) { 6723d85aecf2SMiaohe Lin /* 6724d85aecf2SMiaohe Lin * The file_regions will hold their own reference to 6725d85aecf2SMiaohe Lin * h_cg->css. So we should release the reference held 6726d85aecf2SMiaohe Lin * via hugetlb_cgroup_charge_cgroup_rsvd() when we are 6727d85aecf2SMiaohe Lin * done. 6728d85aecf2SMiaohe Lin */ 6729d85aecf2SMiaohe Lin hugetlb_cgroup_put_rsvd_cgroup(h_cg); 673033039678SMike Kravetz } 673133039678SMike Kravetz } 673233b8f84aSMike Kravetz return true; 673333b8f84aSMike Kravetz 6734075a61d0SMina Almasry out_put_pages: 6735075a61d0SMina Almasry /* put back original number of pages, chg */ 6736075a61d0SMina Almasry (void)hugepage_subpool_put_pages(spool, chg); 6737075a61d0SMina Almasry out_uncharge_cgroup: 6738075a61d0SMina Almasry hugetlb_cgroup_uncharge_cgroup_rsvd(hstate_index(h), 6739075a61d0SMina Almasry chg * pages_per_huge_page(h), h_cg); 6740c50ac050SDave Hansen out_err: 67418d9bfb26SMike Kravetz hugetlb_vma_lock_free(vma); 67425e911373SMike Kravetz if (!vma || vma->vm_flags & VM_MAYSHARE) 67430db9d74eSMina Almasry /* Only call region_abort if the region_chg succeeded but the 67440db9d74eSMina Almasry * region_add failed or didn't run. 67450db9d74eSMina Almasry */ 67460db9d74eSMina Almasry if (chg >= 0 && add < 0) 67470db9d74eSMina Almasry region_abort(resv_map, from, to, regions_needed); 6748f031dd27SJoonsoo Kim if (vma && is_vma_resv_set(vma, HPAGE_RESV_OWNER)) 6749f031dd27SJoonsoo Kim kref_put(&resv_map->refs, resv_map_release); 675033b8f84aSMike Kravetz return false; 6751a43a8c39SChen, Kenneth W } 6752a43a8c39SChen, Kenneth W 6753b5cec28dSMike Kravetz long hugetlb_unreserve_pages(struct inode *inode, long start, long end, 6754b5cec28dSMike Kravetz long freed) 6755a43a8c39SChen, Kenneth W { 6756a5516438SAndi Kleen struct hstate *h = hstate_inode(inode); 67574e35f483SJoonsoo Kim struct resv_map *resv_map = inode_resv_map(inode); 67589119a41eSJoonsoo Kim long chg = 0; 675990481622SDavid Gibson struct hugepage_subpool *spool = subpool_inode(inode); 67601c5ecae3SMike Kravetz long gbl_reserve; 676145c682a6SKen Chen 6762f27a5136SMike Kravetz /* 6763f27a5136SMike Kravetz * Since this routine can be called in the evict inode path for all 6764f27a5136SMike Kravetz * hugetlbfs inodes, resv_map could be NULL. 6765f27a5136SMike Kravetz */ 6766b5cec28dSMike Kravetz if (resv_map) { 6767b5cec28dSMike Kravetz chg = region_del(resv_map, start, end); 6768b5cec28dSMike Kravetz /* 6769b5cec28dSMike Kravetz * region_del() can fail in the rare case where a region 6770b5cec28dSMike Kravetz * must be split and another region descriptor can not be 6771b5cec28dSMike Kravetz * allocated. If end == LONG_MAX, it will not fail. 6772b5cec28dSMike Kravetz */ 6773b5cec28dSMike Kravetz if (chg < 0) 6774b5cec28dSMike Kravetz return chg; 6775b5cec28dSMike Kravetz } 6776b5cec28dSMike Kravetz 677745c682a6SKen Chen spin_lock(&inode->i_lock); 6778e4c6f8beSEric Sandeen inode->i_blocks -= (blocks_per_huge_page(h) * freed); 677945c682a6SKen Chen spin_unlock(&inode->i_lock); 678045c682a6SKen Chen 67811c5ecae3SMike Kravetz /* 67821c5ecae3SMike Kravetz * If the subpool has a minimum size, the number of global 67831c5ecae3SMike Kravetz * reservations to be released may be adjusted. 6784dddf31a4SMiaohe Lin * 6785dddf31a4SMiaohe Lin * Note that !resv_map implies freed == 0. So (chg - freed) 6786dddf31a4SMiaohe Lin * won't go negative. 67871c5ecae3SMike Kravetz */ 67881c5ecae3SMike Kravetz gbl_reserve = hugepage_subpool_put_pages(spool, (chg - freed)); 67891c5ecae3SMike Kravetz hugetlb_acct_memory(h, -gbl_reserve); 6790b5cec28dSMike Kravetz 6791b5cec28dSMike Kravetz return 0; 6792a43a8c39SChen, Kenneth W } 679393f70f90SNaoya Horiguchi 67943212b535SSteve Capper #ifdef CONFIG_ARCH_WANT_HUGE_PMD_SHARE 67953212b535SSteve Capper static unsigned long page_table_shareable(struct vm_area_struct *svma, 67963212b535SSteve Capper struct vm_area_struct *vma, 67973212b535SSteve Capper unsigned long addr, pgoff_t idx) 67983212b535SSteve Capper { 67993212b535SSteve Capper unsigned long saddr = ((idx - svma->vm_pgoff) << PAGE_SHIFT) + 68003212b535SSteve Capper svma->vm_start; 68013212b535SSteve Capper unsigned long sbase = saddr & PUD_MASK; 68023212b535SSteve Capper unsigned long s_end = sbase + PUD_SIZE; 68033212b535SSteve Capper 68043212b535SSteve Capper /* Allow segments to share if only one is marked locked */ 6805de60f5f1SEric B Munson unsigned long vm_flags = vma->vm_flags & VM_LOCKED_CLEAR_MASK; 6806de60f5f1SEric B Munson unsigned long svm_flags = svma->vm_flags & VM_LOCKED_CLEAR_MASK; 68073212b535SSteve Capper 68083212b535SSteve Capper /* 68093212b535SSteve Capper * match the virtual addresses, permission and the alignment of the 68103212b535SSteve Capper * page table page. 6811131a79b4SMike Kravetz * 6812131a79b4SMike Kravetz * Also, vma_lock (vm_private_data) is required for sharing. 68133212b535SSteve Capper */ 68143212b535SSteve Capper if (pmd_index(addr) != pmd_index(saddr) || 68153212b535SSteve Capper vm_flags != svm_flags || 6816131a79b4SMike Kravetz !range_in_vma(svma, sbase, s_end) || 6817131a79b4SMike Kravetz !svma->vm_private_data) 68183212b535SSteve Capper return 0; 68193212b535SSteve Capper 68203212b535SSteve Capper return saddr; 68213212b535SSteve Capper } 68223212b535SSteve Capper 6823bbff39ccSMike Kravetz bool want_pmd_share(struct vm_area_struct *vma, unsigned long addr) 68243212b535SSteve Capper { 6825bbff39ccSMike Kravetz unsigned long start = addr & PUD_MASK; 6826bbff39ccSMike Kravetz unsigned long end = start + PUD_SIZE; 6827bbff39ccSMike Kravetz 68288d9bfb26SMike Kravetz #ifdef CONFIG_USERFAULTFD 68298d9bfb26SMike Kravetz if (uffd_disable_huge_pmd_share(vma)) 68308d9bfb26SMike Kravetz return false; 68318d9bfb26SMike Kravetz #endif 68323212b535SSteve Capper /* 68333212b535SSteve Capper * check on proper vm_flags and page table alignment 68343212b535SSteve Capper */ 68358d9bfb26SMike Kravetz if (!(vma->vm_flags & VM_MAYSHARE)) 683631aafb45SNicholas Krause return false; 6837bbff39ccSMike Kravetz if (!vma->vm_private_data) /* vma lock required for sharing */ 68388d9bfb26SMike Kravetz return false; 68398d9bfb26SMike Kravetz if (!range_in_vma(vma, start, end)) 68408d9bfb26SMike Kravetz return false; 68418d9bfb26SMike Kravetz return true; 68428d9bfb26SMike Kravetz } 68438d9bfb26SMike Kravetz 68443212b535SSteve Capper /* 6845017b1660SMike Kravetz * Determine if start,end range within vma could be mapped by shared pmd. 6846017b1660SMike Kravetz * If yes, adjust start and end to cover range associated with possible 6847017b1660SMike Kravetz * shared pmd mappings. 6848017b1660SMike Kravetz */ 6849017b1660SMike Kravetz void adjust_range_if_pmd_sharing_possible(struct vm_area_struct *vma, 6850017b1660SMike Kravetz unsigned long *start, unsigned long *end) 6851017b1660SMike Kravetz { 6852a1ba9da8SLi Xinhai unsigned long v_start = ALIGN(vma->vm_start, PUD_SIZE), 6853a1ba9da8SLi Xinhai v_end = ALIGN_DOWN(vma->vm_end, PUD_SIZE); 6854017b1660SMike Kravetz 6855a1ba9da8SLi Xinhai /* 6856f0953a1bSIngo Molnar * vma needs to span at least one aligned PUD size, and the range 6857f0953a1bSIngo Molnar * must be at least partially within in. 6858a1ba9da8SLi Xinhai */ 6859a1ba9da8SLi Xinhai if (!(vma->vm_flags & VM_MAYSHARE) || !(v_end > v_start) || 6860a1ba9da8SLi Xinhai (*end <= v_start) || (*start >= v_end)) 6861017b1660SMike Kravetz return; 6862017b1660SMike Kravetz 686375802ca6SPeter Xu /* Extend the range to be PUD aligned for a worst case scenario */ 6864a1ba9da8SLi Xinhai if (*start > v_start) 6865a1ba9da8SLi Xinhai *start = ALIGN_DOWN(*start, PUD_SIZE); 6866017b1660SMike Kravetz 6867a1ba9da8SLi Xinhai if (*end < v_end) 6868a1ba9da8SLi Xinhai *end = ALIGN(*end, PUD_SIZE); 6869017b1660SMike Kravetz } 6870017b1660SMike Kravetz 68718d9bfb26SMike Kravetz static bool __vma_shareable_flags_pmd(struct vm_area_struct *vma) 68728d9bfb26SMike Kravetz { 68738d9bfb26SMike Kravetz return vma->vm_flags & (VM_MAYSHARE | VM_SHARED) && 68748d9bfb26SMike Kravetz vma->vm_private_data; 68758d9bfb26SMike Kravetz } 68768d9bfb26SMike Kravetz 68778d9bfb26SMike Kravetz void hugetlb_vma_lock_read(struct vm_area_struct *vma) 68788d9bfb26SMike Kravetz { 68798d9bfb26SMike Kravetz if (__vma_shareable_flags_pmd(vma)) { 68808d9bfb26SMike Kravetz struct hugetlb_vma_lock *vma_lock = vma->vm_private_data; 68818d9bfb26SMike Kravetz 68828d9bfb26SMike Kravetz down_read(&vma_lock->rw_sema); 68838d9bfb26SMike Kravetz } 68848d9bfb26SMike Kravetz } 68858d9bfb26SMike Kravetz 68868d9bfb26SMike Kravetz void hugetlb_vma_unlock_read(struct vm_area_struct *vma) 68878d9bfb26SMike Kravetz { 68888d9bfb26SMike Kravetz if (__vma_shareable_flags_pmd(vma)) { 68898d9bfb26SMike Kravetz struct hugetlb_vma_lock *vma_lock = vma->vm_private_data; 68908d9bfb26SMike Kravetz 68918d9bfb26SMike Kravetz up_read(&vma_lock->rw_sema); 68928d9bfb26SMike Kravetz } 68938d9bfb26SMike Kravetz } 68948d9bfb26SMike Kravetz 68958d9bfb26SMike Kravetz void hugetlb_vma_lock_write(struct vm_area_struct *vma) 68968d9bfb26SMike Kravetz { 68978d9bfb26SMike Kravetz if (__vma_shareable_flags_pmd(vma)) { 68988d9bfb26SMike Kravetz struct hugetlb_vma_lock *vma_lock = vma->vm_private_data; 68998d9bfb26SMike Kravetz 69008d9bfb26SMike Kravetz down_write(&vma_lock->rw_sema); 69018d9bfb26SMike Kravetz } 69028d9bfb26SMike Kravetz } 69038d9bfb26SMike Kravetz 69048d9bfb26SMike Kravetz void hugetlb_vma_unlock_write(struct vm_area_struct *vma) 69058d9bfb26SMike Kravetz { 69068d9bfb26SMike Kravetz if (__vma_shareable_flags_pmd(vma)) { 69078d9bfb26SMike Kravetz struct hugetlb_vma_lock *vma_lock = vma->vm_private_data; 69088d9bfb26SMike Kravetz 69098d9bfb26SMike Kravetz up_write(&vma_lock->rw_sema); 69108d9bfb26SMike Kravetz } 69118d9bfb26SMike Kravetz } 69128d9bfb26SMike Kravetz 69138d9bfb26SMike Kravetz int hugetlb_vma_trylock_write(struct vm_area_struct *vma) 69148d9bfb26SMike Kravetz { 69158d9bfb26SMike Kravetz struct hugetlb_vma_lock *vma_lock = vma->vm_private_data; 69168d9bfb26SMike Kravetz 69178d9bfb26SMike Kravetz if (!__vma_shareable_flags_pmd(vma)) 69188d9bfb26SMike Kravetz return 1; 69198d9bfb26SMike Kravetz 69208d9bfb26SMike Kravetz return down_write_trylock(&vma_lock->rw_sema); 69218d9bfb26SMike Kravetz } 69228d9bfb26SMike Kravetz 69238d9bfb26SMike Kravetz void hugetlb_vma_assert_locked(struct vm_area_struct *vma) 69248d9bfb26SMike Kravetz { 69258d9bfb26SMike Kravetz if (__vma_shareable_flags_pmd(vma)) { 69268d9bfb26SMike Kravetz struct hugetlb_vma_lock *vma_lock = vma->vm_private_data; 69278d9bfb26SMike Kravetz 69288d9bfb26SMike Kravetz lockdep_assert_held(&vma_lock->rw_sema); 69298d9bfb26SMike Kravetz } 69308d9bfb26SMike Kravetz } 69318d9bfb26SMike Kravetz 69328d9bfb26SMike Kravetz void hugetlb_vma_lock_release(struct kref *kref) 69338d9bfb26SMike Kravetz { 69348d9bfb26SMike Kravetz struct hugetlb_vma_lock *vma_lock = container_of(kref, 69358d9bfb26SMike Kravetz struct hugetlb_vma_lock, refs); 69368d9bfb26SMike Kravetz 69378d9bfb26SMike Kravetz kfree(vma_lock); 69388d9bfb26SMike Kravetz } 69398d9bfb26SMike Kravetz 6940acfac378SAndrew Morton static void __hugetlb_vma_unlock_write_put(struct hugetlb_vma_lock *vma_lock) 6941ecfbd733SMike Kravetz { 6942ecfbd733SMike Kravetz struct vm_area_struct *vma = vma_lock->vma; 6943ecfbd733SMike Kravetz 6944ecfbd733SMike Kravetz /* 6945ecfbd733SMike Kravetz * vma_lock structure may or not be released as a result of put, 6946ecfbd733SMike Kravetz * it certainly will no longer be attached to vma so clear pointer. 6947ecfbd733SMike Kravetz * Semaphore synchronizes access to vma_lock->vma field. 6948ecfbd733SMike Kravetz */ 6949ecfbd733SMike Kravetz vma_lock->vma = NULL; 6950ecfbd733SMike Kravetz vma->vm_private_data = NULL; 6951ecfbd733SMike Kravetz up_write(&vma_lock->rw_sema); 6952ecfbd733SMike Kravetz kref_put(&vma_lock->refs, hugetlb_vma_lock_release); 6953ecfbd733SMike Kravetz } 6954ecfbd733SMike Kravetz 6955ecfbd733SMike Kravetz static void __hugetlb_vma_unlock_write_free(struct vm_area_struct *vma) 6956ecfbd733SMike Kravetz { 6957ecfbd733SMike Kravetz if (__vma_shareable_flags_pmd(vma)) { 6958ecfbd733SMike Kravetz struct hugetlb_vma_lock *vma_lock = vma->vm_private_data; 6959ecfbd733SMike Kravetz 6960ecfbd733SMike Kravetz __hugetlb_vma_unlock_write_put(vma_lock); 6961ecfbd733SMike Kravetz } 6962ecfbd733SMike Kravetz } 6963ecfbd733SMike Kravetz 69648d9bfb26SMike Kravetz static void hugetlb_vma_lock_free(struct vm_area_struct *vma) 69658d9bfb26SMike Kravetz { 69668d9bfb26SMike Kravetz /* 6967131a79b4SMike Kravetz * Only present in sharable vmas. 69688d9bfb26SMike Kravetz */ 6969131a79b4SMike Kravetz if (!vma || !__vma_shareable_flags_pmd(vma)) 69708d9bfb26SMike Kravetz return; 69718d9bfb26SMike Kravetz 69728d9bfb26SMike Kravetz if (vma->vm_private_data) { 69738d9bfb26SMike Kravetz struct hugetlb_vma_lock *vma_lock = vma->vm_private_data; 69748d9bfb26SMike Kravetz 6975ecfbd733SMike Kravetz down_write(&vma_lock->rw_sema); 6976ecfbd733SMike Kravetz __hugetlb_vma_unlock_write_put(vma_lock); 69778d9bfb26SMike Kravetz } 69788d9bfb26SMike Kravetz } 69798d9bfb26SMike Kravetz 69808d9bfb26SMike Kravetz static void hugetlb_vma_lock_alloc(struct vm_area_struct *vma) 69818d9bfb26SMike Kravetz { 69828d9bfb26SMike Kravetz struct hugetlb_vma_lock *vma_lock; 69838d9bfb26SMike Kravetz 69848d9bfb26SMike Kravetz /* Only establish in (flags) sharable vmas */ 69858d9bfb26SMike Kravetz if (!vma || !(vma->vm_flags & VM_MAYSHARE)) 69868d9bfb26SMike Kravetz return; 69878d9bfb26SMike Kravetz 69888d9bfb26SMike Kravetz /* Should never get here with non-NULL vm_private_data */ 69898d9bfb26SMike Kravetz if (vma->vm_private_data) 69908d9bfb26SMike Kravetz return; 69918d9bfb26SMike Kravetz 69928d9bfb26SMike Kravetz vma_lock = kmalloc(sizeof(*vma_lock), GFP_KERNEL); 6993bbff39ccSMike Kravetz if (!vma_lock) { 69948d9bfb26SMike Kravetz /* 69958d9bfb26SMike Kravetz * If we can not allocate structure, then vma can not 6996bbff39ccSMike Kravetz * participate in pmd sharing. This is only a possible 6997bbff39ccSMike Kravetz * performance enhancement and memory saving issue. 6998bbff39ccSMike Kravetz * However, the lock is also used to synchronize page 6999bbff39ccSMike Kravetz * faults with truncation. If the lock is not present, 7000bbff39ccSMike Kravetz * unlikely races could leave pages in a file past i_size 7001bbff39ccSMike Kravetz * until the file is removed. Warn in the unlikely case of 7002bbff39ccSMike Kravetz * allocation failure. 70038d9bfb26SMike Kravetz */ 7004bbff39ccSMike Kravetz pr_warn_once("HugeTLB: unable to allocate vma specific lock\n"); 70058d9bfb26SMike Kravetz return; 7006bbff39ccSMike Kravetz } 70078d9bfb26SMike Kravetz 70088d9bfb26SMike Kravetz kref_init(&vma_lock->refs); 70098d9bfb26SMike Kravetz init_rwsem(&vma_lock->rw_sema); 70108d9bfb26SMike Kravetz vma_lock->vma = vma; 70118d9bfb26SMike Kravetz vma->vm_private_data = vma_lock; 70128d9bfb26SMike Kravetz } 70138d9bfb26SMike Kravetz 7014017b1660SMike Kravetz /* 70153212b535SSteve Capper * Search for a shareable pmd page for hugetlb. In any case calls pmd_alloc() 70163212b535SSteve Capper * and returns the corresponding pte. While this is not necessary for the 70173212b535SSteve Capper * !shared pmd case because we can allocate the pmd later as well, it makes the 70183a47c54fSMike Kravetz * code much cleaner. pmd allocation is essential for the shared case because 70193a47c54fSMike Kravetz * pud has to be populated inside the same i_mmap_rwsem section - otherwise 70203a47c54fSMike Kravetz * racing tasks could either miss the sharing (see huge_pte_offset) or select a 70213a47c54fSMike Kravetz * bad pmd for sharing. 70223212b535SSteve Capper */ 7023aec44e0fSPeter Xu pte_t *huge_pmd_share(struct mm_struct *mm, struct vm_area_struct *vma, 7024aec44e0fSPeter Xu unsigned long addr, pud_t *pud) 70253212b535SSteve Capper { 70263212b535SSteve Capper struct address_space *mapping = vma->vm_file->f_mapping; 70273212b535SSteve Capper pgoff_t idx = ((addr - vma->vm_start) >> PAGE_SHIFT) + 70283212b535SSteve Capper vma->vm_pgoff; 70293212b535SSteve Capper struct vm_area_struct *svma; 70303212b535SSteve Capper unsigned long saddr; 70313212b535SSteve Capper pte_t *spte = NULL; 70323212b535SSteve Capper pte_t *pte; 7033cb900f41SKirill A. Shutemov spinlock_t *ptl; 70343212b535SSteve Capper 70353a47c54fSMike Kravetz i_mmap_lock_read(mapping); 70363212b535SSteve Capper vma_interval_tree_foreach(svma, &mapping->i_mmap, idx, idx) { 70373212b535SSteve Capper if (svma == vma) 70383212b535SSteve Capper continue; 70393212b535SSteve Capper 70403212b535SSteve Capper saddr = page_table_shareable(svma, vma, addr, idx); 70413212b535SSteve Capper if (saddr) { 70427868a208SPunit Agrawal spte = huge_pte_offset(svma->vm_mm, saddr, 70437868a208SPunit Agrawal vma_mmu_pagesize(svma)); 70443212b535SSteve Capper if (spte) { 70453212b535SSteve Capper get_page(virt_to_page(spte)); 70463212b535SSteve Capper break; 70473212b535SSteve Capper } 70483212b535SSteve Capper } 70493212b535SSteve Capper } 70503212b535SSteve Capper 70513212b535SSteve Capper if (!spte) 70523212b535SSteve Capper goto out; 70533212b535SSteve Capper 70548bea8052SAneesh Kumar K.V ptl = huge_pte_lock(hstate_vma(vma), mm, spte); 7055dc6c9a35SKirill A. Shutemov if (pud_none(*pud)) { 70563212b535SSteve Capper pud_populate(mm, pud, 70573212b535SSteve Capper (pmd_t *)((unsigned long)spte & PAGE_MASK)); 7058c17b1f42SKirill A. Shutemov mm_inc_nr_pmds(mm); 7059dc6c9a35SKirill A. Shutemov } else { 70603212b535SSteve Capper put_page(virt_to_page(spte)); 7061dc6c9a35SKirill A. Shutemov } 7062cb900f41SKirill A. Shutemov spin_unlock(ptl); 70633212b535SSteve Capper out: 70643212b535SSteve Capper pte = (pte_t *)pmd_alloc(mm, pud, addr); 70653a47c54fSMike Kravetz i_mmap_unlock_read(mapping); 70663212b535SSteve Capper return pte; 70673212b535SSteve Capper } 70683212b535SSteve Capper 70693212b535SSteve Capper /* 70703212b535SSteve Capper * unmap huge page backed by shared pte. 70713212b535SSteve Capper * 70723212b535SSteve Capper * Hugetlb pte page is ref counted at the time of mapping. If pte is shared 70733212b535SSteve Capper * indicated by page_count > 1, unmap is achieved by clearing pud and 70743212b535SSteve Capper * decrementing the ref count. If count == 1, the pte page is not shared. 70753212b535SSteve Capper * 70763a47c54fSMike Kravetz * Called with page table lock held. 70773212b535SSteve Capper * 70783212b535SSteve Capper * returns: 1 successfully unmapped a shared pte page 70793212b535SSteve Capper * 0 the underlying pte page is not shared, or it is the last user 70803212b535SSteve Capper */ 708134ae204fSMike Kravetz int huge_pmd_unshare(struct mm_struct *mm, struct vm_area_struct *vma, 70824ddb4d91SMike Kravetz unsigned long addr, pte_t *ptep) 70833212b535SSteve Capper { 70844ddb4d91SMike Kravetz pgd_t *pgd = pgd_offset(mm, addr); 70854ddb4d91SMike Kravetz p4d_t *p4d = p4d_offset(pgd, addr); 70864ddb4d91SMike Kravetz pud_t *pud = pud_offset(p4d, addr); 70873212b535SSteve Capper 708834ae204fSMike Kravetz i_mmap_assert_write_locked(vma->vm_file->f_mapping); 708940549ba8SMike Kravetz hugetlb_vma_assert_locked(vma); 70903212b535SSteve Capper BUG_ON(page_count(virt_to_page(ptep)) == 0); 70913212b535SSteve Capper if (page_count(virt_to_page(ptep)) == 1) 70923212b535SSteve Capper return 0; 70933212b535SSteve Capper 70943212b535SSteve Capper pud_clear(pud); 70953212b535SSteve Capper put_page(virt_to_page(ptep)); 7096dc6c9a35SKirill A. Shutemov mm_dec_nr_pmds(mm); 70973212b535SSteve Capper return 1; 70983212b535SSteve Capper } 7099c1991e07SPeter Xu 71009e5fc74cSSteve Capper #else /* !CONFIG_ARCH_WANT_HUGE_PMD_SHARE */ 71018d9bfb26SMike Kravetz 710240549ba8SMike Kravetz void hugetlb_vma_lock_read(struct vm_area_struct *vma) 710340549ba8SMike Kravetz { 710440549ba8SMike Kravetz } 710540549ba8SMike Kravetz 710640549ba8SMike Kravetz void hugetlb_vma_unlock_read(struct vm_area_struct *vma) 710740549ba8SMike Kravetz { 710840549ba8SMike Kravetz } 710940549ba8SMike Kravetz 711040549ba8SMike Kravetz void hugetlb_vma_lock_write(struct vm_area_struct *vma) 711140549ba8SMike Kravetz { 711240549ba8SMike Kravetz } 711340549ba8SMike Kravetz 711440549ba8SMike Kravetz void hugetlb_vma_unlock_write(struct vm_area_struct *vma) 711540549ba8SMike Kravetz { 711640549ba8SMike Kravetz } 711740549ba8SMike Kravetz 711840549ba8SMike Kravetz int hugetlb_vma_trylock_write(struct vm_area_struct *vma) 711940549ba8SMike Kravetz { 712040549ba8SMike Kravetz return 1; 712140549ba8SMike Kravetz } 712240549ba8SMike Kravetz 712340549ba8SMike Kravetz void hugetlb_vma_assert_locked(struct vm_area_struct *vma) 712440549ba8SMike Kravetz { 712540549ba8SMike Kravetz } 712640549ba8SMike Kravetz 71278d9bfb26SMike Kravetz void hugetlb_vma_lock_release(struct kref *kref) 71288d9bfb26SMike Kravetz { 71298d9bfb26SMike Kravetz } 71308d9bfb26SMike Kravetz 7131ecfbd733SMike Kravetz static void __hugetlb_vma_unlock_write_free(struct vm_area_struct *vma) 7132ecfbd733SMike Kravetz { 7133ecfbd733SMike Kravetz } 7134ecfbd733SMike Kravetz 71358d9bfb26SMike Kravetz static void hugetlb_vma_lock_free(struct vm_area_struct *vma) 71368d9bfb26SMike Kravetz { 71378d9bfb26SMike Kravetz } 71388d9bfb26SMike Kravetz 71398d9bfb26SMike Kravetz static void hugetlb_vma_lock_alloc(struct vm_area_struct *vma) 71408d9bfb26SMike Kravetz { 71418d9bfb26SMike Kravetz } 71428d9bfb26SMike Kravetz 7143aec44e0fSPeter Xu pte_t *huge_pmd_share(struct mm_struct *mm, struct vm_area_struct *vma, 7144aec44e0fSPeter Xu unsigned long addr, pud_t *pud) 71459e5fc74cSSteve Capper { 71469e5fc74cSSteve Capper return NULL; 71479e5fc74cSSteve Capper } 7148e81f2d22SZhang Zhen 714934ae204fSMike Kravetz int huge_pmd_unshare(struct mm_struct *mm, struct vm_area_struct *vma, 71504ddb4d91SMike Kravetz unsigned long addr, pte_t *ptep) 7151e81f2d22SZhang Zhen { 7152e81f2d22SZhang Zhen return 0; 7153e81f2d22SZhang Zhen } 7154017b1660SMike Kravetz 7155017b1660SMike Kravetz void adjust_range_if_pmd_sharing_possible(struct vm_area_struct *vma, 7156017b1660SMike Kravetz unsigned long *start, unsigned long *end) 7157017b1660SMike Kravetz { 7158017b1660SMike Kravetz } 7159c1991e07SPeter Xu 7160c1991e07SPeter Xu bool want_pmd_share(struct vm_area_struct *vma, unsigned long addr) 7161c1991e07SPeter Xu { 7162c1991e07SPeter Xu return false; 7163c1991e07SPeter Xu } 71643212b535SSteve Capper #endif /* CONFIG_ARCH_WANT_HUGE_PMD_SHARE */ 71653212b535SSteve Capper 71669e5fc74cSSteve Capper #ifdef CONFIG_ARCH_WANT_GENERAL_HUGETLB 7167aec44e0fSPeter Xu pte_t *huge_pte_alloc(struct mm_struct *mm, struct vm_area_struct *vma, 71689e5fc74cSSteve Capper unsigned long addr, unsigned long sz) 71699e5fc74cSSteve Capper { 71709e5fc74cSSteve Capper pgd_t *pgd; 7171c2febafcSKirill A. Shutemov p4d_t *p4d; 71729e5fc74cSSteve Capper pud_t *pud; 71739e5fc74cSSteve Capper pte_t *pte = NULL; 71749e5fc74cSSteve Capper 71759e5fc74cSSteve Capper pgd = pgd_offset(mm, addr); 7176f4f0a3d8SKirill A. Shutemov p4d = p4d_alloc(mm, pgd, addr); 7177f4f0a3d8SKirill A. Shutemov if (!p4d) 7178f4f0a3d8SKirill A. Shutemov return NULL; 7179c2febafcSKirill A. Shutemov pud = pud_alloc(mm, p4d, addr); 71809e5fc74cSSteve Capper if (pud) { 71819e5fc74cSSteve Capper if (sz == PUD_SIZE) { 71829e5fc74cSSteve Capper pte = (pte_t *)pud; 71839e5fc74cSSteve Capper } else { 71849e5fc74cSSteve Capper BUG_ON(sz != PMD_SIZE); 7185c1991e07SPeter Xu if (want_pmd_share(vma, addr) && pud_none(*pud)) 7186aec44e0fSPeter Xu pte = huge_pmd_share(mm, vma, addr, pud); 71879e5fc74cSSteve Capper else 71889e5fc74cSSteve Capper pte = (pte_t *)pmd_alloc(mm, pud, addr); 71899e5fc74cSSteve Capper } 71909e5fc74cSSteve Capper } 71914e666314SMichal Hocko BUG_ON(pte && pte_present(*pte) && !pte_huge(*pte)); 71929e5fc74cSSteve Capper 71939e5fc74cSSteve Capper return pte; 71949e5fc74cSSteve Capper } 71959e5fc74cSSteve Capper 71969b19df29SPunit Agrawal /* 71979b19df29SPunit Agrawal * huge_pte_offset() - Walk the page table to resolve the hugepage 71989b19df29SPunit Agrawal * entry at address @addr 71999b19df29SPunit Agrawal * 72008ac0b81aSLi Xinhai * Return: Pointer to page table entry (PUD or PMD) for 72018ac0b81aSLi Xinhai * address @addr, or NULL if a !p*d_present() entry is encountered and the 72029b19df29SPunit Agrawal * size @sz doesn't match the hugepage size at this level of the page 72039b19df29SPunit Agrawal * table. 72049b19df29SPunit Agrawal */ 72057868a208SPunit Agrawal pte_t *huge_pte_offset(struct mm_struct *mm, 72067868a208SPunit Agrawal unsigned long addr, unsigned long sz) 72079e5fc74cSSteve Capper { 72089e5fc74cSSteve Capper pgd_t *pgd; 7209c2febafcSKirill A. Shutemov p4d_t *p4d; 72108ac0b81aSLi Xinhai pud_t *pud; 72118ac0b81aSLi Xinhai pmd_t *pmd; 72129e5fc74cSSteve Capper 72139e5fc74cSSteve Capper pgd = pgd_offset(mm, addr); 7214c2febafcSKirill A. Shutemov if (!pgd_present(*pgd)) 7215c2febafcSKirill A. Shutemov return NULL; 7216c2febafcSKirill A. Shutemov p4d = p4d_offset(pgd, addr); 7217c2febafcSKirill A. Shutemov if (!p4d_present(*p4d)) 7218c2febafcSKirill A. Shutemov return NULL; 72199b19df29SPunit Agrawal 7220c2febafcSKirill A. Shutemov pud = pud_offset(p4d, addr); 72218ac0b81aSLi Xinhai if (sz == PUD_SIZE) 72228ac0b81aSLi Xinhai /* must be pud huge, non-present or none */ 72239e5fc74cSSteve Capper return (pte_t *)pud; 72248ac0b81aSLi Xinhai if (!pud_present(*pud)) 72258ac0b81aSLi Xinhai return NULL; 72268ac0b81aSLi Xinhai /* must have a valid entry and size to go further */ 72279b19df29SPunit Agrawal 72289e5fc74cSSteve Capper pmd = pmd_offset(pud, addr); 72298ac0b81aSLi Xinhai /* must be pmd huge, non-present or none */ 72309e5fc74cSSteve Capper return (pte_t *)pmd; 72319e5fc74cSSteve Capper } 72329e5fc74cSSteve Capper 7233e95a9851SMike Kravetz /* 7234e95a9851SMike Kravetz * Return a mask that can be used to update an address to the last huge 7235e95a9851SMike Kravetz * page in a page table page mapping size. Used to skip non-present 7236e95a9851SMike Kravetz * page table entries when linearly scanning address ranges. Architectures 7237e95a9851SMike Kravetz * with unique huge page to page table relationships can define their own 7238e95a9851SMike Kravetz * version of this routine. 7239e95a9851SMike Kravetz */ 7240e95a9851SMike Kravetz unsigned long hugetlb_mask_last_page(struct hstate *h) 7241e95a9851SMike Kravetz { 7242e95a9851SMike Kravetz unsigned long hp_size = huge_page_size(h); 7243e95a9851SMike Kravetz 7244e95a9851SMike Kravetz if (hp_size == PUD_SIZE) 7245e95a9851SMike Kravetz return P4D_SIZE - PUD_SIZE; 7246e95a9851SMike Kravetz else if (hp_size == PMD_SIZE) 7247e95a9851SMike Kravetz return PUD_SIZE - PMD_SIZE; 7248e95a9851SMike Kravetz else 7249e95a9851SMike Kravetz return 0UL; 7250e95a9851SMike Kravetz } 7251e95a9851SMike Kravetz 7252e95a9851SMike Kravetz #else 7253e95a9851SMike Kravetz 7254e95a9851SMike Kravetz /* See description above. Architectures can provide their own version. */ 7255e95a9851SMike Kravetz __weak unsigned long hugetlb_mask_last_page(struct hstate *h) 7256e95a9851SMike Kravetz { 72574ddb4d91SMike Kravetz #ifdef CONFIG_ARCH_WANT_HUGE_PMD_SHARE 72584ddb4d91SMike Kravetz if (huge_page_size(h) == PMD_SIZE) 72594ddb4d91SMike Kravetz return PUD_SIZE - PMD_SIZE; 72604ddb4d91SMike Kravetz #endif 7261e95a9851SMike Kravetz return 0UL; 7262e95a9851SMike Kravetz } 7263e95a9851SMike Kravetz 726461f77edaSNaoya Horiguchi #endif /* CONFIG_ARCH_WANT_GENERAL_HUGETLB */ 726561f77edaSNaoya Horiguchi 726661f77edaSNaoya Horiguchi /* 726761f77edaSNaoya Horiguchi * These functions are overwritable if your architecture needs its own 726861f77edaSNaoya Horiguchi * behavior. 726961f77edaSNaoya Horiguchi */ 72707ce82f4cSMiaohe Lin int isolate_hugetlb(struct page *page, struct list_head *list) 727131caf665SNaoya Horiguchi { 72727ce82f4cSMiaohe Lin int ret = 0; 7273bcc54222SNaoya Horiguchi 7274db71ef79SMike Kravetz spin_lock_irq(&hugetlb_lock); 72758f251a3dSMike Kravetz if (!PageHeadHuge(page) || 72768f251a3dSMike Kravetz !HPageMigratable(page) || 72770eb2df2bSMuchun Song !get_page_unless_zero(page)) { 72787ce82f4cSMiaohe Lin ret = -EBUSY; 7279bcc54222SNaoya Horiguchi goto unlock; 7280bcc54222SNaoya Horiguchi } 72818f251a3dSMike Kravetz ClearHPageMigratable(page); 728231caf665SNaoya Horiguchi list_move_tail(&page->lru, list); 7283bcc54222SNaoya Horiguchi unlock: 7284db71ef79SMike Kravetz spin_unlock_irq(&hugetlb_lock); 7285bcc54222SNaoya Horiguchi return ret; 728631caf665SNaoya Horiguchi } 728731caf665SNaoya Horiguchi 7288e591ef7dSNaoya Horiguchi int get_hwpoison_huge_page(struct page *page, bool *hugetlb, bool unpoison) 728925182f05SNaoya Horiguchi { 729025182f05SNaoya Horiguchi int ret = 0; 729125182f05SNaoya Horiguchi 729225182f05SNaoya Horiguchi *hugetlb = false; 729325182f05SNaoya Horiguchi spin_lock_irq(&hugetlb_lock); 729425182f05SNaoya Horiguchi if (PageHeadHuge(page)) { 729525182f05SNaoya Horiguchi *hugetlb = true; 7296b283d983SNaoya Horiguchi if (HPageFreed(page)) 7297b283d983SNaoya Horiguchi ret = 0; 7298e591ef7dSNaoya Horiguchi else if (HPageMigratable(page) || unpoison) 729925182f05SNaoya Horiguchi ret = get_page_unless_zero(page); 73000ed950d1SNaoya Horiguchi else 73010ed950d1SNaoya Horiguchi ret = -EBUSY; 730225182f05SNaoya Horiguchi } 730325182f05SNaoya Horiguchi spin_unlock_irq(&hugetlb_lock); 730425182f05SNaoya Horiguchi return ret; 730525182f05SNaoya Horiguchi } 730625182f05SNaoya Horiguchi 7307e591ef7dSNaoya Horiguchi int get_huge_page_for_hwpoison(unsigned long pfn, int flags, 7308e591ef7dSNaoya Horiguchi bool *migratable_cleared) 7309405ce051SNaoya Horiguchi { 7310405ce051SNaoya Horiguchi int ret; 7311405ce051SNaoya Horiguchi 7312405ce051SNaoya Horiguchi spin_lock_irq(&hugetlb_lock); 7313e591ef7dSNaoya Horiguchi ret = __get_huge_page_for_hwpoison(pfn, flags, migratable_cleared); 7314405ce051SNaoya Horiguchi spin_unlock_irq(&hugetlb_lock); 7315405ce051SNaoya Horiguchi return ret; 7316405ce051SNaoya Horiguchi } 7317405ce051SNaoya Horiguchi 731831caf665SNaoya Horiguchi void putback_active_hugepage(struct page *page) 731931caf665SNaoya Horiguchi { 7320db71ef79SMike Kravetz spin_lock_irq(&hugetlb_lock); 73218f251a3dSMike Kravetz SetHPageMigratable(page); 732231caf665SNaoya Horiguchi list_move_tail(&page->lru, &(page_hstate(page))->hugepage_activelist); 7323db71ef79SMike Kravetz spin_unlock_irq(&hugetlb_lock); 732431caf665SNaoya Horiguchi put_page(page); 732531caf665SNaoya Horiguchi } 7326ab5ac90aSMichal Hocko 7327345c62d1SSidhartha Kumar void move_hugetlb_state(struct folio *old_folio, struct folio *new_folio, int reason) 7328ab5ac90aSMichal Hocko { 7329345c62d1SSidhartha Kumar struct hstate *h = folio_hstate(old_folio); 7330ab5ac90aSMichal Hocko 7331345c62d1SSidhartha Kumar hugetlb_cgroup_migrate(old_folio, new_folio); 7332345c62d1SSidhartha Kumar set_page_owner_migrate_reason(&new_folio->page, reason); 7333ab5ac90aSMichal Hocko 7334ab5ac90aSMichal Hocko /* 7335345c62d1SSidhartha Kumar * transfer temporary state of the new hugetlb folio. This is 7336ab5ac90aSMichal Hocko * reverse to other transitions because the newpage is going to 7337ab5ac90aSMichal Hocko * be final while the old one will be freed so it takes over 7338ab5ac90aSMichal Hocko * the temporary status. 7339ab5ac90aSMichal Hocko * 7340ab5ac90aSMichal Hocko * Also note that we have to transfer the per-node surplus state 7341ab5ac90aSMichal Hocko * here as well otherwise the global surplus count will not match 7342ab5ac90aSMichal Hocko * the per-node's. 7343ab5ac90aSMichal Hocko */ 7344345c62d1SSidhartha Kumar if (folio_test_hugetlb_temporary(new_folio)) { 7345345c62d1SSidhartha Kumar int old_nid = folio_nid(old_folio); 7346345c62d1SSidhartha Kumar int new_nid = folio_nid(new_folio); 7347ab5ac90aSMichal Hocko 7348345c62d1SSidhartha Kumar folio_set_hugetlb_temporary(old_folio); 7349345c62d1SSidhartha Kumar folio_clear_hugetlb_temporary(new_folio); 7350345c62d1SSidhartha Kumar 7351ab5ac90aSMichal Hocko 73525af1ab1dSMiaohe Lin /* 73535af1ab1dSMiaohe Lin * There is no need to transfer the per-node surplus state 73545af1ab1dSMiaohe Lin * when we do not cross the node. 73555af1ab1dSMiaohe Lin */ 73565af1ab1dSMiaohe Lin if (new_nid == old_nid) 73575af1ab1dSMiaohe Lin return; 7358db71ef79SMike Kravetz spin_lock_irq(&hugetlb_lock); 7359ab5ac90aSMichal Hocko if (h->surplus_huge_pages_node[old_nid]) { 7360ab5ac90aSMichal Hocko h->surplus_huge_pages_node[old_nid]--; 7361ab5ac90aSMichal Hocko h->surplus_huge_pages_node[new_nid]++; 7362ab5ac90aSMichal Hocko } 7363db71ef79SMike Kravetz spin_unlock_irq(&hugetlb_lock); 7364ab5ac90aSMichal Hocko } 7365ab5ac90aSMichal Hocko } 7366cf11e85fSRoman Gushchin 73676dfeaff9SPeter Xu /* 73686dfeaff9SPeter Xu * This function will unconditionally remove all the shared pmd pgtable entries 73696dfeaff9SPeter Xu * within the specific vma for a hugetlbfs memory range. 73706dfeaff9SPeter Xu */ 73716dfeaff9SPeter Xu void hugetlb_unshare_all_pmds(struct vm_area_struct *vma) 73726dfeaff9SPeter Xu { 73736dfeaff9SPeter Xu struct hstate *h = hstate_vma(vma); 73746dfeaff9SPeter Xu unsigned long sz = huge_page_size(h); 73756dfeaff9SPeter Xu struct mm_struct *mm = vma->vm_mm; 73766dfeaff9SPeter Xu struct mmu_notifier_range range; 73776dfeaff9SPeter Xu unsigned long address, start, end; 73786dfeaff9SPeter Xu spinlock_t *ptl; 73796dfeaff9SPeter Xu pte_t *ptep; 73806dfeaff9SPeter Xu 73816dfeaff9SPeter Xu if (!(vma->vm_flags & VM_MAYSHARE)) 73826dfeaff9SPeter Xu return; 73836dfeaff9SPeter Xu 73846dfeaff9SPeter Xu start = ALIGN(vma->vm_start, PUD_SIZE); 73856dfeaff9SPeter Xu end = ALIGN_DOWN(vma->vm_end, PUD_SIZE); 73866dfeaff9SPeter Xu 73876dfeaff9SPeter Xu if (start >= end) 73886dfeaff9SPeter Xu return; 73896dfeaff9SPeter Xu 73909c8bbfacSBaolin Wang flush_cache_range(vma, start, end); 73916dfeaff9SPeter Xu /* 73926dfeaff9SPeter Xu * No need to call adjust_range_if_pmd_sharing_possible(), because 73936dfeaff9SPeter Xu * we have already done the PUD_SIZE alignment. 73946dfeaff9SPeter Xu */ 73956dfeaff9SPeter Xu mmu_notifier_range_init(&range, MMU_NOTIFY_CLEAR, 0, vma, mm, 73966dfeaff9SPeter Xu start, end); 73976dfeaff9SPeter Xu mmu_notifier_invalidate_range_start(&range); 739840549ba8SMike Kravetz hugetlb_vma_lock_write(vma); 73996dfeaff9SPeter Xu i_mmap_lock_write(vma->vm_file->f_mapping); 74006dfeaff9SPeter Xu for (address = start; address < end; address += PUD_SIZE) { 74016dfeaff9SPeter Xu ptep = huge_pte_offset(mm, address, sz); 74026dfeaff9SPeter Xu if (!ptep) 74036dfeaff9SPeter Xu continue; 74046dfeaff9SPeter Xu ptl = huge_pte_lock(h, mm, ptep); 74054ddb4d91SMike Kravetz huge_pmd_unshare(mm, vma, address, ptep); 74066dfeaff9SPeter Xu spin_unlock(ptl); 74076dfeaff9SPeter Xu } 74086dfeaff9SPeter Xu flush_hugetlb_tlb_range(vma, start, end); 74096dfeaff9SPeter Xu i_mmap_unlock_write(vma->vm_file->f_mapping); 741040549ba8SMike Kravetz hugetlb_vma_unlock_write(vma); 74116dfeaff9SPeter Xu /* 74126dfeaff9SPeter Xu * No need to call mmu_notifier_invalidate_range(), see 7413ee65728eSMike Rapoport * Documentation/mm/mmu_notifier.rst. 74146dfeaff9SPeter Xu */ 74156dfeaff9SPeter Xu mmu_notifier_invalidate_range_end(&range); 74166dfeaff9SPeter Xu } 74176dfeaff9SPeter Xu 7418cf11e85fSRoman Gushchin #ifdef CONFIG_CMA 7419cf11e85fSRoman Gushchin static bool cma_reserve_called __initdata; 7420cf11e85fSRoman Gushchin 7421cf11e85fSRoman Gushchin static int __init cmdline_parse_hugetlb_cma(char *p) 7422cf11e85fSRoman Gushchin { 742338e719abSBaolin Wang int nid, count = 0; 742438e719abSBaolin Wang unsigned long tmp; 742538e719abSBaolin Wang char *s = p; 742638e719abSBaolin Wang 742738e719abSBaolin Wang while (*s) { 742838e719abSBaolin Wang if (sscanf(s, "%lu%n", &tmp, &count) != 1) 742938e719abSBaolin Wang break; 743038e719abSBaolin Wang 743138e719abSBaolin Wang if (s[count] == ':') { 7432f9317f77SMike Kravetz if (tmp >= MAX_NUMNODES) 743338e719abSBaolin Wang break; 7434f9317f77SMike Kravetz nid = array_index_nospec(tmp, MAX_NUMNODES); 743538e719abSBaolin Wang 743638e719abSBaolin Wang s += count + 1; 743738e719abSBaolin Wang tmp = memparse(s, &s); 743838e719abSBaolin Wang hugetlb_cma_size_in_node[nid] = tmp; 743938e719abSBaolin Wang hugetlb_cma_size += tmp; 744038e719abSBaolin Wang 744138e719abSBaolin Wang /* 744238e719abSBaolin Wang * Skip the separator if have one, otherwise 744338e719abSBaolin Wang * break the parsing. 744438e719abSBaolin Wang */ 744538e719abSBaolin Wang if (*s == ',') 744638e719abSBaolin Wang s++; 744738e719abSBaolin Wang else 744838e719abSBaolin Wang break; 744938e719abSBaolin Wang } else { 7450cf11e85fSRoman Gushchin hugetlb_cma_size = memparse(p, &p); 745138e719abSBaolin Wang break; 745238e719abSBaolin Wang } 745338e719abSBaolin Wang } 745438e719abSBaolin Wang 7455cf11e85fSRoman Gushchin return 0; 7456cf11e85fSRoman Gushchin } 7457cf11e85fSRoman Gushchin 7458cf11e85fSRoman Gushchin early_param("hugetlb_cma", cmdline_parse_hugetlb_cma); 7459cf11e85fSRoman Gushchin 7460cf11e85fSRoman Gushchin void __init hugetlb_cma_reserve(int order) 7461cf11e85fSRoman Gushchin { 7462cf11e85fSRoman Gushchin unsigned long size, reserved, per_node; 746338e719abSBaolin Wang bool node_specific_cma_alloc = false; 7464cf11e85fSRoman Gushchin int nid; 7465cf11e85fSRoman Gushchin 7466cf11e85fSRoman Gushchin cma_reserve_called = true; 7467cf11e85fSRoman Gushchin 7468cf11e85fSRoman Gushchin if (!hugetlb_cma_size) 7469cf11e85fSRoman Gushchin return; 7470cf11e85fSRoman Gushchin 747138e719abSBaolin Wang for (nid = 0; nid < MAX_NUMNODES; nid++) { 747238e719abSBaolin Wang if (hugetlb_cma_size_in_node[nid] == 0) 747338e719abSBaolin Wang continue; 747438e719abSBaolin Wang 747530a51400SPeng Liu if (!node_online(nid)) { 747638e719abSBaolin Wang pr_warn("hugetlb_cma: invalid node %d specified\n", nid); 747738e719abSBaolin Wang hugetlb_cma_size -= hugetlb_cma_size_in_node[nid]; 747838e719abSBaolin Wang hugetlb_cma_size_in_node[nid] = 0; 747938e719abSBaolin Wang continue; 748038e719abSBaolin Wang } 748138e719abSBaolin Wang 748238e719abSBaolin Wang if (hugetlb_cma_size_in_node[nid] < (PAGE_SIZE << order)) { 748338e719abSBaolin Wang pr_warn("hugetlb_cma: cma area of node %d should be at least %lu MiB\n", 748438e719abSBaolin Wang nid, (PAGE_SIZE << order) / SZ_1M); 748538e719abSBaolin Wang hugetlb_cma_size -= hugetlb_cma_size_in_node[nid]; 748638e719abSBaolin Wang hugetlb_cma_size_in_node[nid] = 0; 748738e719abSBaolin Wang } else { 748838e719abSBaolin Wang node_specific_cma_alloc = true; 748938e719abSBaolin Wang } 749038e719abSBaolin Wang } 749138e719abSBaolin Wang 749238e719abSBaolin Wang /* Validate the CMA size again in case some invalid nodes specified. */ 749338e719abSBaolin Wang if (!hugetlb_cma_size) 749438e719abSBaolin Wang return; 749538e719abSBaolin Wang 7496cf11e85fSRoman Gushchin if (hugetlb_cma_size < (PAGE_SIZE << order)) { 7497cf11e85fSRoman Gushchin pr_warn("hugetlb_cma: cma area should be at least %lu MiB\n", 7498cf11e85fSRoman Gushchin (PAGE_SIZE << order) / SZ_1M); 7499a01f4390SMike Kravetz hugetlb_cma_size = 0; 7500cf11e85fSRoman Gushchin return; 7501cf11e85fSRoman Gushchin } 7502cf11e85fSRoman Gushchin 750338e719abSBaolin Wang if (!node_specific_cma_alloc) { 7504cf11e85fSRoman Gushchin /* 7505cf11e85fSRoman Gushchin * If 3 GB area is requested on a machine with 4 numa nodes, 7506cf11e85fSRoman Gushchin * let's allocate 1 GB on first three nodes and ignore the last one. 7507cf11e85fSRoman Gushchin */ 7508cf11e85fSRoman Gushchin per_node = DIV_ROUND_UP(hugetlb_cma_size, nr_online_nodes); 7509cf11e85fSRoman Gushchin pr_info("hugetlb_cma: reserve %lu MiB, up to %lu MiB per node\n", 7510cf11e85fSRoman Gushchin hugetlb_cma_size / SZ_1M, per_node / SZ_1M); 751138e719abSBaolin Wang } 7512cf11e85fSRoman Gushchin 7513cf11e85fSRoman Gushchin reserved = 0; 751430a51400SPeng Liu for_each_online_node(nid) { 7515cf11e85fSRoman Gushchin int res; 75162281f797SBarry Song char name[CMA_MAX_NAME]; 7517cf11e85fSRoman Gushchin 751838e719abSBaolin Wang if (node_specific_cma_alloc) { 751938e719abSBaolin Wang if (hugetlb_cma_size_in_node[nid] == 0) 752038e719abSBaolin Wang continue; 752138e719abSBaolin Wang 752238e719abSBaolin Wang size = hugetlb_cma_size_in_node[nid]; 752338e719abSBaolin Wang } else { 7524cf11e85fSRoman Gushchin size = min(per_node, hugetlb_cma_size - reserved); 752538e719abSBaolin Wang } 752638e719abSBaolin Wang 7527cf11e85fSRoman Gushchin size = round_up(size, PAGE_SIZE << order); 7528cf11e85fSRoman Gushchin 75292281f797SBarry Song snprintf(name, sizeof(name), "hugetlb%d", nid); 7530a01f4390SMike Kravetz /* 7531a01f4390SMike Kravetz * Note that 'order per bit' is based on smallest size that 7532a01f4390SMike Kravetz * may be returned to CMA allocator in the case of 7533a01f4390SMike Kravetz * huge page demotion. 7534a01f4390SMike Kravetz */ 7535a01f4390SMike Kravetz res = cma_declare_contiguous_nid(0, size, 0, 7536a01f4390SMike Kravetz PAGE_SIZE << HUGETLB_PAGE_ORDER, 753729d0f41dSBarry Song 0, false, name, 7538cf11e85fSRoman Gushchin &hugetlb_cma[nid], nid); 7539cf11e85fSRoman Gushchin if (res) { 7540cf11e85fSRoman Gushchin pr_warn("hugetlb_cma: reservation failed: err %d, node %d", 7541cf11e85fSRoman Gushchin res, nid); 7542cf11e85fSRoman Gushchin continue; 7543cf11e85fSRoman Gushchin } 7544cf11e85fSRoman Gushchin 7545cf11e85fSRoman Gushchin reserved += size; 7546cf11e85fSRoman Gushchin pr_info("hugetlb_cma: reserved %lu MiB on node %d\n", 7547cf11e85fSRoman Gushchin size / SZ_1M, nid); 7548cf11e85fSRoman Gushchin 7549cf11e85fSRoman Gushchin if (reserved >= hugetlb_cma_size) 7550cf11e85fSRoman Gushchin break; 7551cf11e85fSRoman Gushchin } 7552a01f4390SMike Kravetz 7553a01f4390SMike Kravetz if (!reserved) 7554a01f4390SMike Kravetz /* 7555a01f4390SMike Kravetz * hugetlb_cma_size is used to determine if allocations from 7556a01f4390SMike Kravetz * cma are possible. Set to zero if no cma regions are set up. 7557a01f4390SMike Kravetz */ 7558a01f4390SMike Kravetz hugetlb_cma_size = 0; 7559cf11e85fSRoman Gushchin } 7560cf11e85fSRoman Gushchin 7561263b8998SMiaohe Lin static void __init hugetlb_cma_check(void) 7562cf11e85fSRoman Gushchin { 7563cf11e85fSRoman Gushchin if (!hugetlb_cma_size || cma_reserve_called) 7564cf11e85fSRoman Gushchin return; 7565cf11e85fSRoman Gushchin 7566cf11e85fSRoman Gushchin pr_warn("hugetlb_cma: the option isn't supported by current arch\n"); 7567cf11e85fSRoman Gushchin } 7568cf11e85fSRoman Gushchin 7569cf11e85fSRoman Gushchin #endif /* CONFIG_CMA */ 7570