xref: /openbmc/linux/mm/hugetlb.c (revision 1a1aad8a)
11da177e4SLinus Torvalds /*
21da177e4SLinus Torvalds  * Generic hugetlb support.
36d49e352SNadia Yvette Chambers  * (C) Nadia Yvette Chambers, April 2004
41da177e4SLinus Torvalds  */
51da177e4SLinus Torvalds #include <linux/list.h>
61da177e4SLinus Torvalds #include <linux/init.h>
71da177e4SLinus Torvalds #include <linux/mm.h>
8e1759c21SAlexey Dobriyan #include <linux/seq_file.h>
91da177e4SLinus Torvalds #include <linux/sysctl.h>
101da177e4SLinus Torvalds #include <linux/highmem.h>
11cddb8a5cSAndrea Arcangeli #include <linux/mmu_notifier.h>
121da177e4SLinus Torvalds #include <linux/nodemask.h>
1363551ae0SDavid Gibson #include <linux/pagemap.h>
145da7ca86SChristoph Lameter #include <linux/mempolicy.h>
153b32123dSGideon Israel Dsouza #include <linux/compiler.h>
16aea47ff3SChristoph Lameter #include <linux/cpuset.h>
173935baa9SDavid Gibson #include <linux/mutex.h>
18aa888a74SAndi Kleen #include <linux/bootmem.h>
19a3437870SNishanth Aravamudan #include <linux/sysfs.h>
205a0e3ad6STejun Heo #include <linux/slab.h>
210fe6e20bSNaoya Horiguchi #include <linux/rmap.h>
22fd6a03edSNaoya Horiguchi #include <linux/swap.h>
23fd6a03edSNaoya Horiguchi #include <linux/swapops.h>
24c8721bbbSNaoya Horiguchi #include <linux/page-isolation.h>
258382d914SDavidlohr Bueso #include <linux/jhash.h>
26d6606683SLinus Torvalds 
2763551ae0SDavid Gibson #include <asm/page.h>
2863551ae0SDavid Gibson #include <asm/pgtable.h>
2924669e58SAneesh Kumar K.V #include <asm/tlb.h>
3063551ae0SDavid Gibson 
3124669e58SAneesh Kumar K.V #include <linux/io.h>
3263551ae0SDavid Gibson #include <linux/hugetlb.h>
339dd540e2SAneesh Kumar K.V #include <linux/hugetlb_cgroup.h>
349a305230SLee Schermerhorn #include <linux/node.h>
351a1aad8aSMike Kravetz #include <linux/userfaultfd_k.h>
367835e98bSNick Piggin #include "internal.h"
371da177e4SLinus Torvalds 
38753162cdSAndrey Ryabinin int hugepages_treat_as_movable;
39a5516438SAndi Kleen 
40c3f38a38SAneesh Kumar K.V int hugetlb_max_hstate __read_mostly;
41e5ff2159SAndi Kleen unsigned int default_hstate_idx;
42e5ff2159SAndi Kleen struct hstate hstates[HUGE_MAX_HSTATE];
43641844f5SNaoya Horiguchi /*
44641844f5SNaoya Horiguchi  * Minimum page order among possible hugepage sizes, set to a proper value
45641844f5SNaoya Horiguchi  * at boot time.
46641844f5SNaoya Horiguchi  */
47641844f5SNaoya Horiguchi static unsigned int minimum_order __read_mostly = UINT_MAX;
48e5ff2159SAndi Kleen 
4953ba51d2SJon Tollefson __initdata LIST_HEAD(huge_boot_pages);
5053ba51d2SJon Tollefson 
51e5ff2159SAndi Kleen /* for command line parsing */
52e5ff2159SAndi Kleen static struct hstate * __initdata parsed_hstate;
53e5ff2159SAndi Kleen static unsigned long __initdata default_hstate_max_huge_pages;
54e11bfbfcSNick Piggin static unsigned long __initdata default_hstate_size;
559fee021dSVaishali Thakkar static bool __initdata parsed_valid_hugepagesz = true;
56e5ff2159SAndi Kleen 
573935baa9SDavid Gibson /*
5831caf665SNaoya Horiguchi  * Protects updates to hugepage_freelists, hugepage_activelist, nr_huge_pages,
5931caf665SNaoya Horiguchi  * free_huge_pages, and surplus_huge_pages.
603935baa9SDavid Gibson  */
61c3f38a38SAneesh Kumar K.V DEFINE_SPINLOCK(hugetlb_lock);
620bd0f9fbSEric Paris 
638382d914SDavidlohr Bueso /*
648382d914SDavidlohr Bueso  * Serializes faults on the same logical page.  This is used to
658382d914SDavidlohr Bueso  * prevent spurious OOMs when the hugepage pool is fully utilized.
668382d914SDavidlohr Bueso  */
678382d914SDavidlohr Bueso static int num_fault_mutexes;
68c672c7f2SMike Kravetz struct mutex *hugetlb_fault_mutex_table ____cacheline_aligned_in_smp;
698382d914SDavidlohr Bueso 
707ca02d0aSMike Kravetz /* Forward declaration */
717ca02d0aSMike Kravetz static int hugetlb_acct_memory(struct hstate *h, long delta);
727ca02d0aSMike Kravetz 
7390481622SDavid Gibson static inline void unlock_or_release_subpool(struct hugepage_subpool *spool)
7490481622SDavid Gibson {
7590481622SDavid Gibson 	bool free = (spool->count == 0) && (spool->used_hpages == 0);
7690481622SDavid Gibson 
7790481622SDavid Gibson 	spin_unlock(&spool->lock);
7890481622SDavid Gibson 
7990481622SDavid Gibson 	/* If no pages are used, and no other handles to the subpool
807ca02d0aSMike Kravetz 	 * remain, give up any reservations mased on minimum size and
817ca02d0aSMike Kravetz 	 * free the subpool */
827ca02d0aSMike Kravetz 	if (free) {
837ca02d0aSMike Kravetz 		if (spool->min_hpages != -1)
847ca02d0aSMike Kravetz 			hugetlb_acct_memory(spool->hstate,
857ca02d0aSMike Kravetz 						-spool->min_hpages);
8690481622SDavid Gibson 		kfree(spool);
8790481622SDavid Gibson 	}
887ca02d0aSMike Kravetz }
8990481622SDavid Gibson 
907ca02d0aSMike Kravetz struct hugepage_subpool *hugepage_new_subpool(struct hstate *h, long max_hpages,
917ca02d0aSMike Kravetz 						long min_hpages)
9290481622SDavid Gibson {
9390481622SDavid Gibson 	struct hugepage_subpool *spool;
9490481622SDavid Gibson 
95c6a91820SMike Kravetz 	spool = kzalloc(sizeof(*spool), GFP_KERNEL);
9690481622SDavid Gibson 	if (!spool)
9790481622SDavid Gibson 		return NULL;
9890481622SDavid Gibson 
9990481622SDavid Gibson 	spin_lock_init(&spool->lock);
10090481622SDavid Gibson 	spool->count = 1;
1017ca02d0aSMike Kravetz 	spool->max_hpages = max_hpages;
1027ca02d0aSMike Kravetz 	spool->hstate = h;
1037ca02d0aSMike Kravetz 	spool->min_hpages = min_hpages;
1047ca02d0aSMike Kravetz 
1057ca02d0aSMike Kravetz 	if (min_hpages != -1 && hugetlb_acct_memory(h, min_hpages)) {
1067ca02d0aSMike Kravetz 		kfree(spool);
1077ca02d0aSMike Kravetz 		return NULL;
1087ca02d0aSMike Kravetz 	}
1097ca02d0aSMike Kravetz 	spool->rsv_hpages = min_hpages;
11090481622SDavid Gibson 
11190481622SDavid Gibson 	return spool;
11290481622SDavid Gibson }
11390481622SDavid Gibson 
11490481622SDavid Gibson void hugepage_put_subpool(struct hugepage_subpool *spool)
11590481622SDavid Gibson {
11690481622SDavid Gibson 	spin_lock(&spool->lock);
11790481622SDavid Gibson 	BUG_ON(!spool->count);
11890481622SDavid Gibson 	spool->count--;
11990481622SDavid Gibson 	unlock_or_release_subpool(spool);
12090481622SDavid Gibson }
12190481622SDavid Gibson 
1221c5ecae3SMike Kravetz /*
1231c5ecae3SMike Kravetz  * Subpool accounting for allocating and reserving pages.
1241c5ecae3SMike Kravetz  * Return -ENOMEM if there are not enough resources to satisfy the
1251c5ecae3SMike Kravetz  * the request.  Otherwise, return the number of pages by which the
1261c5ecae3SMike Kravetz  * global pools must be adjusted (upward).  The returned value may
1271c5ecae3SMike Kravetz  * only be different than the passed value (delta) in the case where
1281c5ecae3SMike Kravetz  * a subpool minimum size must be manitained.
1291c5ecae3SMike Kravetz  */
1301c5ecae3SMike Kravetz static long hugepage_subpool_get_pages(struct hugepage_subpool *spool,
13190481622SDavid Gibson 				      long delta)
13290481622SDavid Gibson {
1331c5ecae3SMike Kravetz 	long ret = delta;
13490481622SDavid Gibson 
13590481622SDavid Gibson 	if (!spool)
1361c5ecae3SMike Kravetz 		return ret;
13790481622SDavid Gibson 
13890481622SDavid Gibson 	spin_lock(&spool->lock);
13990481622SDavid Gibson 
1401c5ecae3SMike Kravetz 	if (spool->max_hpages != -1) {		/* maximum size accounting */
1411c5ecae3SMike Kravetz 		if ((spool->used_hpages + delta) <= spool->max_hpages)
1421c5ecae3SMike Kravetz 			spool->used_hpages += delta;
1431c5ecae3SMike Kravetz 		else {
1441c5ecae3SMike Kravetz 			ret = -ENOMEM;
1451c5ecae3SMike Kravetz 			goto unlock_ret;
1461c5ecae3SMike Kravetz 		}
1471c5ecae3SMike Kravetz 	}
1481c5ecae3SMike Kravetz 
14909a95e29SMike Kravetz 	/* minimum size accounting */
15009a95e29SMike Kravetz 	if (spool->min_hpages != -1 && spool->rsv_hpages) {
1511c5ecae3SMike Kravetz 		if (delta > spool->rsv_hpages) {
1521c5ecae3SMike Kravetz 			/*
1531c5ecae3SMike Kravetz 			 * Asking for more reserves than those already taken on
1541c5ecae3SMike Kravetz 			 * behalf of subpool.  Return difference.
1551c5ecae3SMike Kravetz 			 */
1561c5ecae3SMike Kravetz 			ret = delta - spool->rsv_hpages;
1571c5ecae3SMike Kravetz 			spool->rsv_hpages = 0;
1581c5ecae3SMike Kravetz 		} else {
1591c5ecae3SMike Kravetz 			ret = 0;	/* reserves already accounted for */
1601c5ecae3SMike Kravetz 			spool->rsv_hpages -= delta;
1611c5ecae3SMike Kravetz 		}
1621c5ecae3SMike Kravetz 	}
1631c5ecae3SMike Kravetz 
1641c5ecae3SMike Kravetz unlock_ret:
1651c5ecae3SMike Kravetz 	spin_unlock(&spool->lock);
16690481622SDavid Gibson 	return ret;
16790481622SDavid Gibson }
16890481622SDavid Gibson 
1691c5ecae3SMike Kravetz /*
1701c5ecae3SMike Kravetz  * Subpool accounting for freeing and unreserving pages.
1711c5ecae3SMike Kravetz  * Return the number of global page reservations that must be dropped.
1721c5ecae3SMike Kravetz  * The return value may only be different than the passed value (delta)
1731c5ecae3SMike Kravetz  * in the case where a subpool minimum size must be maintained.
1741c5ecae3SMike Kravetz  */
1751c5ecae3SMike Kravetz static long hugepage_subpool_put_pages(struct hugepage_subpool *spool,
17690481622SDavid Gibson 				       long delta)
17790481622SDavid Gibson {
1781c5ecae3SMike Kravetz 	long ret = delta;
1791c5ecae3SMike Kravetz 
18090481622SDavid Gibson 	if (!spool)
1811c5ecae3SMike Kravetz 		return delta;
18290481622SDavid Gibson 
18390481622SDavid Gibson 	spin_lock(&spool->lock);
1841c5ecae3SMike Kravetz 
1851c5ecae3SMike Kravetz 	if (spool->max_hpages != -1)		/* maximum size accounting */
18690481622SDavid Gibson 		spool->used_hpages -= delta;
1871c5ecae3SMike Kravetz 
18809a95e29SMike Kravetz 	 /* minimum size accounting */
18909a95e29SMike Kravetz 	if (spool->min_hpages != -1 && spool->used_hpages < spool->min_hpages) {
1901c5ecae3SMike Kravetz 		if (spool->rsv_hpages + delta <= spool->min_hpages)
1911c5ecae3SMike Kravetz 			ret = 0;
1921c5ecae3SMike Kravetz 		else
1931c5ecae3SMike Kravetz 			ret = spool->rsv_hpages + delta - spool->min_hpages;
1941c5ecae3SMike Kravetz 
1951c5ecae3SMike Kravetz 		spool->rsv_hpages += delta;
1961c5ecae3SMike Kravetz 		if (spool->rsv_hpages > spool->min_hpages)
1971c5ecae3SMike Kravetz 			spool->rsv_hpages = spool->min_hpages;
1981c5ecae3SMike Kravetz 	}
1991c5ecae3SMike Kravetz 
2001c5ecae3SMike Kravetz 	/*
2011c5ecae3SMike Kravetz 	 * If hugetlbfs_put_super couldn't free spool due to an outstanding
2021c5ecae3SMike Kravetz 	 * quota reference, free it now.
2031c5ecae3SMike Kravetz 	 */
20490481622SDavid Gibson 	unlock_or_release_subpool(spool);
2051c5ecae3SMike Kravetz 
2061c5ecae3SMike Kravetz 	return ret;
20790481622SDavid Gibson }
20890481622SDavid Gibson 
20990481622SDavid Gibson static inline struct hugepage_subpool *subpool_inode(struct inode *inode)
21090481622SDavid Gibson {
21190481622SDavid Gibson 	return HUGETLBFS_SB(inode->i_sb)->spool;
21290481622SDavid Gibson }
21390481622SDavid Gibson 
21490481622SDavid Gibson static inline struct hugepage_subpool *subpool_vma(struct vm_area_struct *vma)
21590481622SDavid Gibson {
216496ad9aaSAl Viro 	return subpool_inode(file_inode(vma->vm_file));
21790481622SDavid Gibson }
21890481622SDavid Gibson 
219e7c4b0bfSAndy Whitcroft /*
22096822904SAndy Whitcroft  * Region tracking -- allows tracking of reservations and instantiated pages
22196822904SAndy Whitcroft  *                    across the pages in a mapping.
22284afd99bSAndy Whitcroft  *
2231dd308a7SMike Kravetz  * The region data structures are embedded into a resv_map and protected
2241dd308a7SMike Kravetz  * by a resv_map's lock.  The set of regions within the resv_map represent
2251dd308a7SMike Kravetz  * reservations for huge pages, or huge pages that have already been
2261dd308a7SMike Kravetz  * instantiated within the map.  The from and to elements are huge page
2271dd308a7SMike Kravetz  * indicies into the associated mapping.  from indicates the starting index
2281dd308a7SMike Kravetz  * of the region.  to represents the first index past the end of  the region.
2291dd308a7SMike Kravetz  *
2301dd308a7SMike Kravetz  * For example, a file region structure with from == 0 and to == 4 represents
2311dd308a7SMike Kravetz  * four huge pages in a mapping.  It is important to note that the to element
2321dd308a7SMike Kravetz  * represents the first element past the end of the region. This is used in
2331dd308a7SMike Kravetz  * arithmetic as 4(to) - 0(from) = 4 huge pages in the region.
2341dd308a7SMike Kravetz  *
2351dd308a7SMike Kravetz  * Interval notation of the form [from, to) will be used to indicate that
2361dd308a7SMike Kravetz  * the endpoint from is inclusive and to is exclusive.
23796822904SAndy Whitcroft  */
23896822904SAndy Whitcroft struct file_region {
23996822904SAndy Whitcroft 	struct list_head link;
24096822904SAndy Whitcroft 	long from;
24196822904SAndy Whitcroft 	long to;
24296822904SAndy Whitcroft };
24396822904SAndy Whitcroft 
2441dd308a7SMike Kravetz /*
2451dd308a7SMike Kravetz  * Add the huge page range represented by [f, t) to the reserve
2465e911373SMike Kravetz  * map.  In the normal case, existing regions will be expanded
2475e911373SMike Kravetz  * to accommodate the specified range.  Sufficient regions should
2485e911373SMike Kravetz  * exist for expansion due to the previous call to region_chg
2495e911373SMike Kravetz  * with the same range.  However, it is possible that region_del
2505e911373SMike Kravetz  * could have been called after region_chg and modifed the map
2515e911373SMike Kravetz  * in such a way that no region exists to be expanded.  In this
2525e911373SMike Kravetz  * case, pull a region descriptor from the cache associated with
2535e911373SMike Kravetz  * the map and use that for the new range.
254cf3ad20bSMike Kravetz  *
255cf3ad20bSMike Kravetz  * Return the number of new huge pages added to the map.  This
256cf3ad20bSMike Kravetz  * number is greater than or equal to zero.
2571dd308a7SMike Kravetz  */
2581406ec9bSJoonsoo Kim static long region_add(struct resv_map *resv, long f, long t)
25996822904SAndy Whitcroft {
2601406ec9bSJoonsoo Kim 	struct list_head *head = &resv->regions;
26196822904SAndy Whitcroft 	struct file_region *rg, *nrg, *trg;
262cf3ad20bSMike Kravetz 	long add = 0;
26396822904SAndy Whitcroft 
2647b24d861SDavidlohr Bueso 	spin_lock(&resv->lock);
26596822904SAndy Whitcroft 	/* Locate the region we are either in or before. */
26696822904SAndy Whitcroft 	list_for_each_entry(rg, head, link)
26796822904SAndy Whitcroft 		if (f <= rg->to)
26896822904SAndy Whitcroft 			break;
26996822904SAndy Whitcroft 
2705e911373SMike Kravetz 	/*
2715e911373SMike Kravetz 	 * If no region exists which can be expanded to include the
2725e911373SMike Kravetz 	 * specified range, the list must have been modified by an
2735e911373SMike Kravetz 	 * interleving call to region_del().  Pull a region descriptor
2745e911373SMike Kravetz 	 * from the cache and use it for this range.
2755e911373SMike Kravetz 	 */
2765e911373SMike Kravetz 	if (&rg->link == head || t < rg->from) {
2775e911373SMike Kravetz 		VM_BUG_ON(resv->region_cache_count <= 0);
2785e911373SMike Kravetz 
2795e911373SMike Kravetz 		resv->region_cache_count--;
2805e911373SMike Kravetz 		nrg = list_first_entry(&resv->region_cache, struct file_region,
2815e911373SMike Kravetz 					link);
2825e911373SMike Kravetz 		list_del(&nrg->link);
2835e911373SMike Kravetz 
2845e911373SMike Kravetz 		nrg->from = f;
2855e911373SMike Kravetz 		nrg->to = t;
2865e911373SMike Kravetz 		list_add(&nrg->link, rg->link.prev);
2875e911373SMike Kravetz 
2885e911373SMike Kravetz 		add += t - f;
2895e911373SMike Kravetz 		goto out_locked;
2905e911373SMike Kravetz 	}
2915e911373SMike Kravetz 
29296822904SAndy Whitcroft 	/* Round our left edge to the current segment if it encloses us. */
29396822904SAndy Whitcroft 	if (f > rg->from)
29496822904SAndy Whitcroft 		f = rg->from;
29596822904SAndy Whitcroft 
29696822904SAndy Whitcroft 	/* Check for and consume any regions we now overlap with. */
29796822904SAndy Whitcroft 	nrg = rg;
29896822904SAndy Whitcroft 	list_for_each_entry_safe(rg, trg, rg->link.prev, link) {
29996822904SAndy Whitcroft 		if (&rg->link == head)
30096822904SAndy Whitcroft 			break;
30196822904SAndy Whitcroft 		if (rg->from > t)
30296822904SAndy Whitcroft 			break;
30396822904SAndy Whitcroft 
30496822904SAndy Whitcroft 		/* If this area reaches higher then extend our area to
30596822904SAndy Whitcroft 		 * include it completely.  If this is not the first area
30696822904SAndy Whitcroft 		 * which we intend to reuse, free it. */
30796822904SAndy Whitcroft 		if (rg->to > t)
30896822904SAndy Whitcroft 			t = rg->to;
30996822904SAndy Whitcroft 		if (rg != nrg) {
310cf3ad20bSMike Kravetz 			/* Decrement return value by the deleted range.
311cf3ad20bSMike Kravetz 			 * Another range will span this area so that by
312cf3ad20bSMike Kravetz 			 * end of routine add will be >= zero
313cf3ad20bSMike Kravetz 			 */
314cf3ad20bSMike Kravetz 			add -= (rg->to - rg->from);
31596822904SAndy Whitcroft 			list_del(&rg->link);
31696822904SAndy Whitcroft 			kfree(rg);
31796822904SAndy Whitcroft 		}
31896822904SAndy Whitcroft 	}
319cf3ad20bSMike Kravetz 
320cf3ad20bSMike Kravetz 	add += (nrg->from - f);		/* Added to beginning of region */
32196822904SAndy Whitcroft 	nrg->from = f;
322cf3ad20bSMike Kravetz 	add += t - nrg->to;		/* Added to end of region */
32396822904SAndy Whitcroft 	nrg->to = t;
324cf3ad20bSMike Kravetz 
3255e911373SMike Kravetz out_locked:
3265e911373SMike Kravetz 	resv->adds_in_progress--;
3277b24d861SDavidlohr Bueso 	spin_unlock(&resv->lock);
328cf3ad20bSMike Kravetz 	VM_BUG_ON(add < 0);
329cf3ad20bSMike Kravetz 	return add;
33096822904SAndy Whitcroft }
33196822904SAndy Whitcroft 
3321dd308a7SMike Kravetz /*
3331dd308a7SMike Kravetz  * Examine the existing reserve map and determine how many
3341dd308a7SMike Kravetz  * huge pages in the specified range [f, t) are NOT currently
3351dd308a7SMike Kravetz  * represented.  This routine is called before a subsequent
3361dd308a7SMike Kravetz  * call to region_add that will actually modify the reserve
3371dd308a7SMike Kravetz  * map to add the specified range [f, t).  region_chg does
3381dd308a7SMike Kravetz  * not change the number of huge pages represented by the
3391dd308a7SMike Kravetz  * map.  However, if the existing regions in the map can not
3401dd308a7SMike Kravetz  * be expanded to represent the new range, a new file_region
3411dd308a7SMike Kravetz  * structure is added to the map as a placeholder.  This is
3421dd308a7SMike Kravetz  * so that the subsequent region_add call will have all the
3431dd308a7SMike Kravetz  * regions it needs and will not fail.
3441dd308a7SMike Kravetz  *
3455e911373SMike Kravetz  * Upon entry, region_chg will also examine the cache of region descriptors
3465e911373SMike Kravetz  * associated with the map.  If there are not enough descriptors cached, one
3475e911373SMike Kravetz  * will be allocated for the in progress add operation.
3485e911373SMike Kravetz  *
3495e911373SMike Kravetz  * Returns the number of huge pages that need to be added to the existing
3505e911373SMike Kravetz  * reservation map for the range [f, t).  This number is greater or equal to
3515e911373SMike Kravetz  * zero.  -ENOMEM is returned if a new file_region structure or cache entry
3525e911373SMike Kravetz  * is needed and can not be allocated.
3531dd308a7SMike Kravetz  */
3541406ec9bSJoonsoo Kim static long region_chg(struct resv_map *resv, long f, long t)
35596822904SAndy Whitcroft {
3561406ec9bSJoonsoo Kim 	struct list_head *head = &resv->regions;
3577b24d861SDavidlohr Bueso 	struct file_region *rg, *nrg = NULL;
35896822904SAndy Whitcroft 	long chg = 0;
35996822904SAndy Whitcroft 
3607b24d861SDavidlohr Bueso retry:
3617b24d861SDavidlohr Bueso 	spin_lock(&resv->lock);
3625e911373SMike Kravetz retry_locked:
3635e911373SMike Kravetz 	resv->adds_in_progress++;
3645e911373SMike Kravetz 
3655e911373SMike Kravetz 	/*
3665e911373SMike Kravetz 	 * Check for sufficient descriptors in the cache to accommodate
3675e911373SMike Kravetz 	 * the number of in progress add operations.
3685e911373SMike Kravetz 	 */
3695e911373SMike Kravetz 	if (resv->adds_in_progress > resv->region_cache_count) {
3705e911373SMike Kravetz 		struct file_region *trg;
3715e911373SMike Kravetz 
3725e911373SMike Kravetz 		VM_BUG_ON(resv->adds_in_progress - resv->region_cache_count > 1);
3735e911373SMike Kravetz 		/* Must drop lock to allocate a new descriptor. */
3745e911373SMike Kravetz 		resv->adds_in_progress--;
3755e911373SMike Kravetz 		spin_unlock(&resv->lock);
3765e911373SMike Kravetz 
3775e911373SMike Kravetz 		trg = kmalloc(sizeof(*trg), GFP_KERNEL);
378dbe409e4SMike Kravetz 		if (!trg) {
379dbe409e4SMike Kravetz 			kfree(nrg);
3805e911373SMike Kravetz 			return -ENOMEM;
381dbe409e4SMike Kravetz 		}
3825e911373SMike Kravetz 
3835e911373SMike Kravetz 		spin_lock(&resv->lock);
3845e911373SMike Kravetz 		list_add(&trg->link, &resv->region_cache);
3855e911373SMike Kravetz 		resv->region_cache_count++;
3865e911373SMike Kravetz 		goto retry_locked;
3875e911373SMike Kravetz 	}
3885e911373SMike Kravetz 
38996822904SAndy Whitcroft 	/* Locate the region we are before or in. */
39096822904SAndy Whitcroft 	list_for_each_entry(rg, head, link)
39196822904SAndy Whitcroft 		if (f <= rg->to)
39296822904SAndy Whitcroft 			break;
39396822904SAndy Whitcroft 
39496822904SAndy Whitcroft 	/* If we are below the current region then a new region is required.
39596822904SAndy Whitcroft 	 * Subtle, allocate a new region at the position but make it zero
39696822904SAndy Whitcroft 	 * size such that we can guarantee to record the reservation. */
39796822904SAndy Whitcroft 	if (&rg->link == head || t < rg->from) {
3987b24d861SDavidlohr Bueso 		if (!nrg) {
3995e911373SMike Kravetz 			resv->adds_in_progress--;
4007b24d861SDavidlohr Bueso 			spin_unlock(&resv->lock);
40196822904SAndy Whitcroft 			nrg = kmalloc(sizeof(*nrg), GFP_KERNEL);
40296822904SAndy Whitcroft 			if (!nrg)
40396822904SAndy Whitcroft 				return -ENOMEM;
4047b24d861SDavidlohr Bueso 
40596822904SAndy Whitcroft 			nrg->from = f;
40696822904SAndy Whitcroft 			nrg->to   = f;
40796822904SAndy Whitcroft 			INIT_LIST_HEAD(&nrg->link);
4087b24d861SDavidlohr Bueso 			goto retry;
4097b24d861SDavidlohr Bueso 		}
41096822904SAndy Whitcroft 
4117b24d861SDavidlohr Bueso 		list_add(&nrg->link, rg->link.prev);
4127b24d861SDavidlohr Bueso 		chg = t - f;
4137b24d861SDavidlohr Bueso 		goto out_nrg;
41496822904SAndy Whitcroft 	}
41596822904SAndy Whitcroft 
41696822904SAndy Whitcroft 	/* Round our left edge to the current segment if it encloses us. */
41796822904SAndy Whitcroft 	if (f > rg->from)
41896822904SAndy Whitcroft 		f = rg->from;
41996822904SAndy Whitcroft 	chg = t - f;
42096822904SAndy Whitcroft 
42196822904SAndy Whitcroft 	/* Check for and consume any regions we now overlap with. */
42296822904SAndy Whitcroft 	list_for_each_entry(rg, rg->link.prev, link) {
42396822904SAndy Whitcroft 		if (&rg->link == head)
42496822904SAndy Whitcroft 			break;
42596822904SAndy Whitcroft 		if (rg->from > t)
4267b24d861SDavidlohr Bueso 			goto out;
42796822904SAndy Whitcroft 
42825985edcSLucas De Marchi 		/* We overlap with this area, if it extends further than
42996822904SAndy Whitcroft 		 * us then we must extend ourselves.  Account for its
43096822904SAndy Whitcroft 		 * existing reservation. */
43196822904SAndy Whitcroft 		if (rg->to > t) {
43296822904SAndy Whitcroft 			chg += rg->to - t;
43396822904SAndy Whitcroft 			t = rg->to;
43496822904SAndy Whitcroft 		}
43596822904SAndy Whitcroft 		chg -= rg->to - rg->from;
43696822904SAndy Whitcroft 	}
4377b24d861SDavidlohr Bueso 
4387b24d861SDavidlohr Bueso out:
4397b24d861SDavidlohr Bueso 	spin_unlock(&resv->lock);
4407b24d861SDavidlohr Bueso 	/*  We already know we raced and no longer need the new region */
4417b24d861SDavidlohr Bueso 	kfree(nrg);
4427b24d861SDavidlohr Bueso 	return chg;
4437b24d861SDavidlohr Bueso out_nrg:
4447b24d861SDavidlohr Bueso 	spin_unlock(&resv->lock);
44596822904SAndy Whitcroft 	return chg;
44696822904SAndy Whitcroft }
44796822904SAndy Whitcroft 
4481dd308a7SMike Kravetz /*
4495e911373SMike Kravetz  * Abort the in progress add operation.  The adds_in_progress field
4505e911373SMike Kravetz  * of the resv_map keeps track of the operations in progress between
4515e911373SMike Kravetz  * calls to region_chg and region_add.  Operations are sometimes
4525e911373SMike Kravetz  * aborted after the call to region_chg.  In such cases, region_abort
4535e911373SMike Kravetz  * is called to decrement the adds_in_progress counter.
4545e911373SMike Kravetz  *
4555e911373SMike Kravetz  * NOTE: The range arguments [f, t) are not needed or used in this
4565e911373SMike Kravetz  * routine.  They are kept to make reading the calling code easier as
4575e911373SMike Kravetz  * arguments will match the associated region_chg call.
4585e911373SMike Kravetz  */
4595e911373SMike Kravetz static void region_abort(struct resv_map *resv, long f, long t)
4605e911373SMike Kravetz {
4615e911373SMike Kravetz 	spin_lock(&resv->lock);
4625e911373SMike Kravetz 	VM_BUG_ON(!resv->region_cache_count);
4635e911373SMike Kravetz 	resv->adds_in_progress--;
4645e911373SMike Kravetz 	spin_unlock(&resv->lock);
4655e911373SMike Kravetz }
4665e911373SMike Kravetz 
4675e911373SMike Kravetz /*
468feba16e2SMike Kravetz  * Delete the specified range [f, t) from the reserve map.  If the
469feba16e2SMike Kravetz  * t parameter is LONG_MAX, this indicates that ALL regions after f
470feba16e2SMike Kravetz  * should be deleted.  Locate the regions which intersect [f, t)
471feba16e2SMike Kravetz  * and either trim, delete or split the existing regions.
472feba16e2SMike Kravetz  *
473feba16e2SMike Kravetz  * Returns the number of huge pages deleted from the reserve map.
474feba16e2SMike Kravetz  * In the normal case, the return value is zero or more.  In the
475feba16e2SMike Kravetz  * case where a region must be split, a new region descriptor must
476feba16e2SMike Kravetz  * be allocated.  If the allocation fails, -ENOMEM will be returned.
477feba16e2SMike Kravetz  * NOTE: If the parameter t == LONG_MAX, then we will never split
478feba16e2SMike Kravetz  * a region and possibly return -ENOMEM.  Callers specifying
479feba16e2SMike Kravetz  * t == LONG_MAX do not need to check for -ENOMEM error.
4801dd308a7SMike Kravetz  */
481feba16e2SMike Kravetz static long region_del(struct resv_map *resv, long f, long t)
48296822904SAndy Whitcroft {
4831406ec9bSJoonsoo Kim 	struct list_head *head = &resv->regions;
48496822904SAndy Whitcroft 	struct file_region *rg, *trg;
485feba16e2SMike Kravetz 	struct file_region *nrg = NULL;
486feba16e2SMike Kravetz 	long del = 0;
48796822904SAndy Whitcroft 
488feba16e2SMike Kravetz retry:
4897b24d861SDavidlohr Bueso 	spin_lock(&resv->lock);
490feba16e2SMike Kravetz 	list_for_each_entry_safe(rg, trg, head, link) {
491dbe409e4SMike Kravetz 		/*
492dbe409e4SMike Kravetz 		 * Skip regions before the range to be deleted.  file_region
493dbe409e4SMike Kravetz 		 * ranges are normally of the form [from, to).  However, there
494dbe409e4SMike Kravetz 		 * may be a "placeholder" entry in the map which is of the form
495dbe409e4SMike Kravetz 		 * (from, to) with from == to.  Check for placeholder entries
496dbe409e4SMike Kravetz 		 * at the beginning of the range to be deleted.
497dbe409e4SMike Kravetz 		 */
498dbe409e4SMike Kravetz 		if (rg->to <= f && (rg->to != rg->from || rg->to != f))
499feba16e2SMike Kravetz 			continue;
500dbe409e4SMike Kravetz 
501feba16e2SMike Kravetz 		if (rg->from >= t)
50296822904SAndy Whitcroft 			break;
50396822904SAndy Whitcroft 
504feba16e2SMike Kravetz 		if (f > rg->from && t < rg->to) { /* Must split region */
505feba16e2SMike Kravetz 			/*
506feba16e2SMike Kravetz 			 * Check for an entry in the cache before dropping
507feba16e2SMike Kravetz 			 * lock and attempting allocation.
508feba16e2SMike Kravetz 			 */
509feba16e2SMike Kravetz 			if (!nrg &&
510feba16e2SMike Kravetz 			    resv->region_cache_count > resv->adds_in_progress) {
511feba16e2SMike Kravetz 				nrg = list_first_entry(&resv->region_cache,
512feba16e2SMike Kravetz 							struct file_region,
513feba16e2SMike Kravetz 							link);
514feba16e2SMike Kravetz 				list_del(&nrg->link);
515feba16e2SMike Kravetz 				resv->region_cache_count--;
51696822904SAndy Whitcroft 			}
51796822904SAndy Whitcroft 
518feba16e2SMike Kravetz 			if (!nrg) {
519feba16e2SMike Kravetz 				spin_unlock(&resv->lock);
520feba16e2SMike Kravetz 				nrg = kmalloc(sizeof(*nrg), GFP_KERNEL);
521feba16e2SMike Kravetz 				if (!nrg)
522feba16e2SMike Kravetz 					return -ENOMEM;
523feba16e2SMike Kravetz 				goto retry;
524feba16e2SMike Kravetz 			}
525feba16e2SMike Kravetz 
526feba16e2SMike Kravetz 			del += t - f;
527feba16e2SMike Kravetz 
528feba16e2SMike Kravetz 			/* New entry for end of split region */
529feba16e2SMike Kravetz 			nrg->from = t;
530feba16e2SMike Kravetz 			nrg->to = rg->to;
531feba16e2SMike Kravetz 			INIT_LIST_HEAD(&nrg->link);
532feba16e2SMike Kravetz 
533feba16e2SMike Kravetz 			/* Original entry is trimmed */
534feba16e2SMike Kravetz 			rg->to = f;
535feba16e2SMike Kravetz 
536feba16e2SMike Kravetz 			list_add(&nrg->link, &rg->link);
537feba16e2SMike Kravetz 			nrg = NULL;
53896822904SAndy Whitcroft 			break;
539feba16e2SMike Kravetz 		}
540feba16e2SMike Kravetz 
541feba16e2SMike Kravetz 		if (f <= rg->from && t >= rg->to) { /* Remove entire region */
542feba16e2SMike Kravetz 			del += rg->to - rg->from;
54396822904SAndy Whitcroft 			list_del(&rg->link);
54496822904SAndy Whitcroft 			kfree(rg);
545feba16e2SMike Kravetz 			continue;
54696822904SAndy Whitcroft 		}
5477b24d861SDavidlohr Bueso 
548feba16e2SMike Kravetz 		if (f <= rg->from) {	/* Trim beginning of region */
549feba16e2SMike Kravetz 			del += t - rg->from;
550feba16e2SMike Kravetz 			rg->from = t;
551feba16e2SMike Kravetz 		} else {		/* Trim end of region */
552feba16e2SMike Kravetz 			del += rg->to - f;
553feba16e2SMike Kravetz 			rg->to = f;
554feba16e2SMike Kravetz 		}
555feba16e2SMike Kravetz 	}
556feba16e2SMike Kravetz 
5577b24d861SDavidlohr Bueso 	spin_unlock(&resv->lock);
558feba16e2SMike Kravetz 	kfree(nrg);
559feba16e2SMike Kravetz 	return del;
56096822904SAndy Whitcroft }
56196822904SAndy Whitcroft 
5621dd308a7SMike Kravetz /*
563b5cec28dSMike Kravetz  * A rare out of memory error was encountered which prevented removal of
564b5cec28dSMike Kravetz  * the reserve map region for a page.  The huge page itself was free'ed
565b5cec28dSMike Kravetz  * and removed from the page cache.  This routine will adjust the subpool
566b5cec28dSMike Kravetz  * usage count, and the global reserve count if needed.  By incrementing
567b5cec28dSMike Kravetz  * these counts, the reserve map entry which could not be deleted will
568b5cec28dSMike Kravetz  * appear as a "reserved" entry instead of simply dangling with incorrect
569b5cec28dSMike Kravetz  * counts.
570b5cec28dSMike Kravetz  */
57172e2936cSzhong jiang void hugetlb_fix_reserve_counts(struct inode *inode)
572b5cec28dSMike Kravetz {
573b5cec28dSMike Kravetz 	struct hugepage_subpool *spool = subpool_inode(inode);
574b5cec28dSMike Kravetz 	long rsv_adjust;
575b5cec28dSMike Kravetz 
576b5cec28dSMike Kravetz 	rsv_adjust = hugepage_subpool_get_pages(spool, 1);
57772e2936cSzhong jiang 	if (rsv_adjust) {
578b5cec28dSMike Kravetz 		struct hstate *h = hstate_inode(inode);
579b5cec28dSMike Kravetz 
580b5cec28dSMike Kravetz 		hugetlb_acct_memory(h, 1);
581b5cec28dSMike Kravetz 	}
582b5cec28dSMike Kravetz }
583b5cec28dSMike Kravetz 
584b5cec28dSMike Kravetz /*
5851dd308a7SMike Kravetz  * Count and return the number of huge pages in the reserve map
5861dd308a7SMike Kravetz  * that intersect with the range [f, t).
5871dd308a7SMike Kravetz  */
5881406ec9bSJoonsoo Kim static long region_count(struct resv_map *resv, long f, long t)
58984afd99bSAndy Whitcroft {
5901406ec9bSJoonsoo Kim 	struct list_head *head = &resv->regions;
59184afd99bSAndy Whitcroft 	struct file_region *rg;
59284afd99bSAndy Whitcroft 	long chg = 0;
59384afd99bSAndy Whitcroft 
5947b24d861SDavidlohr Bueso 	spin_lock(&resv->lock);
59584afd99bSAndy Whitcroft 	/* Locate each segment we overlap with, and count that overlap. */
59684afd99bSAndy Whitcroft 	list_for_each_entry(rg, head, link) {
597f2135a4aSWang Sheng-Hui 		long seg_from;
598f2135a4aSWang Sheng-Hui 		long seg_to;
59984afd99bSAndy Whitcroft 
60084afd99bSAndy Whitcroft 		if (rg->to <= f)
60184afd99bSAndy Whitcroft 			continue;
60284afd99bSAndy Whitcroft 		if (rg->from >= t)
60384afd99bSAndy Whitcroft 			break;
60484afd99bSAndy Whitcroft 
60584afd99bSAndy Whitcroft 		seg_from = max(rg->from, f);
60684afd99bSAndy Whitcroft 		seg_to = min(rg->to, t);
60784afd99bSAndy Whitcroft 
60884afd99bSAndy Whitcroft 		chg += seg_to - seg_from;
60984afd99bSAndy Whitcroft 	}
6107b24d861SDavidlohr Bueso 	spin_unlock(&resv->lock);
61184afd99bSAndy Whitcroft 
61284afd99bSAndy Whitcroft 	return chg;
61384afd99bSAndy Whitcroft }
61484afd99bSAndy Whitcroft 
61596822904SAndy Whitcroft /*
616e7c4b0bfSAndy Whitcroft  * Convert the address within this vma to the page offset within
617e7c4b0bfSAndy Whitcroft  * the mapping, in pagecache page units; huge pages here.
618e7c4b0bfSAndy Whitcroft  */
619a5516438SAndi Kleen static pgoff_t vma_hugecache_offset(struct hstate *h,
620a5516438SAndi Kleen 			struct vm_area_struct *vma, unsigned long address)
621e7c4b0bfSAndy Whitcroft {
622a5516438SAndi Kleen 	return ((address - vma->vm_start) >> huge_page_shift(h)) +
623a5516438SAndi Kleen 			(vma->vm_pgoff >> huge_page_order(h));
624e7c4b0bfSAndy Whitcroft }
625e7c4b0bfSAndy Whitcroft 
6260fe6e20bSNaoya Horiguchi pgoff_t linear_hugepage_index(struct vm_area_struct *vma,
6270fe6e20bSNaoya Horiguchi 				     unsigned long address)
6280fe6e20bSNaoya Horiguchi {
6290fe6e20bSNaoya Horiguchi 	return vma_hugecache_offset(hstate_vma(vma), vma, address);
6300fe6e20bSNaoya Horiguchi }
631dee41079SDan Williams EXPORT_SYMBOL_GPL(linear_hugepage_index);
6320fe6e20bSNaoya Horiguchi 
63384afd99bSAndy Whitcroft /*
63408fba699SMel Gorman  * Return the size of the pages allocated when backing a VMA. In the majority
63508fba699SMel Gorman  * cases this will be same size as used by the page table entries.
63608fba699SMel Gorman  */
63708fba699SMel Gorman unsigned long vma_kernel_pagesize(struct vm_area_struct *vma)
63808fba699SMel Gorman {
63908fba699SMel Gorman 	struct hstate *hstate;
64008fba699SMel Gorman 
64108fba699SMel Gorman 	if (!is_vm_hugetlb_page(vma))
64208fba699SMel Gorman 		return PAGE_SIZE;
64308fba699SMel Gorman 
64408fba699SMel Gorman 	hstate = hstate_vma(vma);
64508fba699SMel Gorman 
6462415cf12SWanpeng Li 	return 1UL << huge_page_shift(hstate);
64708fba699SMel Gorman }
648f340ca0fSJoerg Roedel EXPORT_SYMBOL_GPL(vma_kernel_pagesize);
64908fba699SMel Gorman 
65008fba699SMel Gorman /*
6513340289dSMel Gorman  * Return the page size being used by the MMU to back a VMA. In the majority
6523340289dSMel Gorman  * of cases, the page size used by the kernel matches the MMU size. On
6533340289dSMel Gorman  * architectures where it differs, an architecture-specific version of this
6543340289dSMel Gorman  * function is required.
6553340289dSMel Gorman  */
6563340289dSMel Gorman #ifndef vma_mmu_pagesize
6573340289dSMel Gorman unsigned long vma_mmu_pagesize(struct vm_area_struct *vma)
6583340289dSMel Gorman {
6593340289dSMel Gorman 	return vma_kernel_pagesize(vma);
6603340289dSMel Gorman }
6613340289dSMel Gorman #endif
6623340289dSMel Gorman 
6633340289dSMel Gorman /*
66484afd99bSAndy Whitcroft  * Flags for MAP_PRIVATE reservations.  These are stored in the bottom
66584afd99bSAndy Whitcroft  * bits of the reservation map pointer, which are always clear due to
66684afd99bSAndy Whitcroft  * alignment.
66784afd99bSAndy Whitcroft  */
66884afd99bSAndy Whitcroft #define HPAGE_RESV_OWNER    (1UL << 0)
66984afd99bSAndy Whitcroft #define HPAGE_RESV_UNMAPPED (1UL << 1)
67004f2cbe3SMel Gorman #define HPAGE_RESV_MASK (HPAGE_RESV_OWNER | HPAGE_RESV_UNMAPPED)
67184afd99bSAndy Whitcroft 
672a1e78772SMel Gorman /*
673a1e78772SMel Gorman  * These helpers are used to track how many pages are reserved for
674a1e78772SMel Gorman  * faults in a MAP_PRIVATE mapping. Only the process that called mmap()
675a1e78772SMel Gorman  * is guaranteed to have their future faults succeed.
676a1e78772SMel Gorman  *
677a1e78772SMel Gorman  * With the exception of reset_vma_resv_huge_pages() which is called at fork(),
678a1e78772SMel Gorman  * the reserve counters are updated with the hugetlb_lock held. It is safe
679a1e78772SMel Gorman  * to reset the VMA at fork() time as it is not in use yet and there is no
680a1e78772SMel Gorman  * chance of the global counters getting corrupted as a result of the values.
68184afd99bSAndy Whitcroft  *
68284afd99bSAndy Whitcroft  * The private mapping reservation is represented in a subtly different
68384afd99bSAndy Whitcroft  * manner to a shared mapping.  A shared mapping has a region map associated
68484afd99bSAndy Whitcroft  * with the underlying file, this region map represents the backing file
68584afd99bSAndy Whitcroft  * pages which have ever had a reservation assigned which this persists even
68684afd99bSAndy Whitcroft  * after the page is instantiated.  A private mapping has a region map
68784afd99bSAndy Whitcroft  * associated with the original mmap which is attached to all VMAs which
68884afd99bSAndy Whitcroft  * reference it, this region map represents those offsets which have consumed
68984afd99bSAndy Whitcroft  * reservation ie. where pages have been instantiated.
690a1e78772SMel Gorman  */
691e7c4b0bfSAndy Whitcroft static unsigned long get_vma_private_data(struct vm_area_struct *vma)
692e7c4b0bfSAndy Whitcroft {
693e7c4b0bfSAndy Whitcroft 	return (unsigned long)vma->vm_private_data;
694e7c4b0bfSAndy Whitcroft }
695e7c4b0bfSAndy Whitcroft 
696e7c4b0bfSAndy Whitcroft static void set_vma_private_data(struct vm_area_struct *vma,
697e7c4b0bfSAndy Whitcroft 							unsigned long value)
698e7c4b0bfSAndy Whitcroft {
699e7c4b0bfSAndy Whitcroft 	vma->vm_private_data = (void *)value;
700e7c4b0bfSAndy Whitcroft }
701e7c4b0bfSAndy Whitcroft 
7029119a41eSJoonsoo Kim struct resv_map *resv_map_alloc(void)
70384afd99bSAndy Whitcroft {
70484afd99bSAndy Whitcroft 	struct resv_map *resv_map = kmalloc(sizeof(*resv_map), GFP_KERNEL);
7055e911373SMike Kravetz 	struct file_region *rg = kmalloc(sizeof(*rg), GFP_KERNEL);
7065e911373SMike Kravetz 
7075e911373SMike Kravetz 	if (!resv_map || !rg) {
7085e911373SMike Kravetz 		kfree(resv_map);
7095e911373SMike Kravetz 		kfree(rg);
71084afd99bSAndy Whitcroft 		return NULL;
7115e911373SMike Kravetz 	}
71284afd99bSAndy Whitcroft 
71384afd99bSAndy Whitcroft 	kref_init(&resv_map->refs);
7147b24d861SDavidlohr Bueso 	spin_lock_init(&resv_map->lock);
71584afd99bSAndy Whitcroft 	INIT_LIST_HEAD(&resv_map->regions);
71684afd99bSAndy Whitcroft 
7175e911373SMike Kravetz 	resv_map->adds_in_progress = 0;
7185e911373SMike Kravetz 
7195e911373SMike Kravetz 	INIT_LIST_HEAD(&resv_map->region_cache);
7205e911373SMike Kravetz 	list_add(&rg->link, &resv_map->region_cache);
7215e911373SMike Kravetz 	resv_map->region_cache_count = 1;
7225e911373SMike Kravetz 
72384afd99bSAndy Whitcroft 	return resv_map;
72484afd99bSAndy Whitcroft }
72584afd99bSAndy Whitcroft 
7269119a41eSJoonsoo Kim void resv_map_release(struct kref *ref)
72784afd99bSAndy Whitcroft {
72884afd99bSAndy Whitcroft 	struct resv_map *resv_map = container_of(ref, struct resv_map, refs);
7295e911373SMike Kravetz 	struct list_head *head = &resv_map->region_cache;
7305e911373SMike Kravetz 	struct file_region *rg, *trg;
73184afd99bSAndy Whitcroft 
73284afd99bSAndy Whitcroft 	/* Clear out any active regions before we release the map. */
733feba16e2SMike Kravetz 	region_del(resv_map, 0, LONG_MAX);
7345e911373SMike Kravetz 
7355e911373SMike Kravetz 	/* ... and any entries left in the cache */
7365e911373SMike Kravetz 	list_for_each_entry_safe(rg, trg, head, link) {
7375e911373SMike Kravetz 		list_del(&rg->link);
7385e911373SMike Kravetz 		kfree(rg);
7395e911373SMike Kravetz 	}
7405e911373SMike Kravetz 
7415e911373SMike Kravetz 	VM_BUG_ON(resv_map->adds_in_progress);
7425e911373SMike Kravetz 
74384afd99bSAndy Whitcroft 	kfree(resv_map);
74484afd99bSAndy Whitcroft }
74584afd99bSAndy Whitcroft 
7464e35f483SJoonsoo Kim static inline struct resv_map *inode_resv_map(struct inode *inode)
7474e35f483SJoonsoo Kim {
7484e35f483SJoonsoo Kim 	return inode->i_mapping->private_data;
7494e35f483SJoonsoo Kim }
7504e35f483SJoonsoo Kim 
75184afd99bSAndy Whitcroft static struct resv_map *vma_resv_map(struct vm_area_struct *vma)
752a1e78772SMel Gorman {
75381d1b09cSSasha Levin 	VM_BUG_ON_VMA(!is_vm_hugetlb_page(vma), vma);
7544e35f483SJoonsoo Kim 	if (vma->vm_flags & VM_MAYSHARE) {
7554e35f483SJoonsoo Kim 		struct address_space *mapping = vma->vm_file->f_mapping;
7564e35f483SJoonsoo Kim 		struct inode *inode = mapping->host;
7574e35f483SJoonsoo Kim 
7584e35f483SJoonsoo Kim 		return inode_resv_map(inode);
7594e35f483SJoonsoo Kim 
7604e35f483SJoonsoo Kim 	} else {
76184afd99bSAndy Whitcroft 		return (struct resv_map *)(get_vma_private_data(vma) &
76284afd99bSAndy Whitcroft 							~HPAGE_RESV_MASK);
7634e35f483SJoonsoo Kim 	}
764a1e78772SMel Gorman }
765a1e78772SMel Gorman 
76684afd99bSAndy Whitcroft static void set_vma_resv_map(struct vm_area_struct *vma, struct resv_map *map)
767a1e78772SMel Gorman {
76881d1b09cSSasha Levin 	VM_BUG_ON_VMA(!is_vm_hugetlb_page(vma), vma);
76981d1b09cSSasha Levin 	VM_BUG_ON_VMA(vma->vm_flags & VM_MAYSHARE, vma);
770a1e78772SMel Gorman 
77184afd99bSAndy Whitcroft 	set_vma_private_data(vma, (get_vma_private_data(vma) &
77284afd99bSAndy Whitcroft 				HPAGE_RESV_MASK) | (unsigned long)map);
77304f2cbe3SMel Gorman }
77404f2cbe3SMel Gorman 
77504f2cbe3SMel Gorman static void set_vma_resv_flags(struct vm_area_struct *vma, unsigned long flags)
77604f2cbe3SMel Gorman {
77781d1b09cSSasha Levin 	VM_BUG_ON_VMA(!is_vm_hugetlb_page(vma), vma);
77881d1b09cSSasha Levin 	VM_BUG_ON_VMA(vma->vm_flags & VM_MAYSHARE, vma);
779e7c4b0bfSAndy Whitcroft 
780e7c4b0bfSAndy Whitcroft 	set_vma_private_data(vma, get_vma_private_data(vma) | flags);
78104f2cbe3SMel Gorman }
78204f2cbe3SMel Gorman 
78304f2cbe3SMel Gorman static int is_vma_resv_set(struct vm_area_struct *vma, unsigned long flag)
78404f2cbe3SMel Gorman {
78581d1b09cSSasha Levin 	VM_BUG_ON_VMA(!is_vm_hugetlb_page(vma), vma);
786e7c4b0bfSAndy Whitcroft 
787e7c4b0bfSAndy Whitcroft 	return (get_vma_private_data(vma) & flag) != 0;
788a1e78772SMel Gorman }
789a1e78772SMel Gorman 
79004f2cbe3SMel Gorman /* Reset counters to 0 and clear all HPAGE_RESV_* flags */
791a1e78772SMel Gorman void reset_vma_resv_huge_pages(struct vm_area_struct *vma)
792a1e78772SMel Gorman {
79381d1b09cSSasha Levin 	VM_BUG_ON_VMA(!is_vm_hugetlb_page(vma), vma);
794f83a275dSMel Gorman 	if (!(vma->vm_flags & VM_MAYSHARE))
795a1e78772SMel Gorman 		vma->vm_private_data = (void *)0;
796a1e78772SMel Gorman }
797a1e78772SMel Gorman 
798a1e78772SMel Gorman /* Returns true if the VMA has associated reserve pages */
799559ec2f8SNicholas Krause static bool vma_has_reserves(struct vm_area_struct *vma, long chg)
800a1e78772SMel Gorman {
801af0ed73eSJoonsoo Kim 	if (vma->vm_flags & VM_NORESERVE) {
802af0ed73eSJoonsoo Kim 		/*
803af0ed73eSJoonsoo Kim 		 * This address is already reserved by other process(chg == 0),
804af0ed73eSJoonsoo Kim 		 * so, we should decrement reserved count. Without decrementing,
805af0ed73eSJoonsoo Kim 		 * reserve count remains after releasing inode, because this
806af0ed73eSJoonsoo Kim 		 * allocated page will go into page cache and is regarded as
807af0ed73eSJoonsoo Kim 		 * coming from reserved pool in releasing step.  Currently, we
808af0ed73eSJoonsoo Kim 		 * don't have any other solution to deal with this situation
809af0ed73eSJoonsoo Kim 		 * properly, so add work-around here.
810af0ed73eSJoonsoo Kim 		 */
811af0ed73eSJoonsoo Kim 		if (vma->vm_flags & VM_MAYSHARE && chg == 0)
812559ec2f8SNicholas Krause 			return true;
813af0ed73eSJoonsoo Kim 		else
814559ec2f8SNicholas Krause 			return false;
815af0ed73eSJoonsoo Kim 	}
816a63884e9SJoonsoo Kim 
817a63884e9SJoonsoo Kim 	/* Shared mappings always use reserves */
8181fb1b0e9SMike Kravetz 	if (vma->vm_flags & VM_MAYSHARE) {
8191fb1b0e9SMike Kravetz 		/*
8201fb1b0e9SMike Kravetz 		 * We know VM_NORESERVE is not set.  Therefore, there SHOULD
8211fb1b0e9SMike Kravetz 		 * be a region map for all pages.  The only situation where
8221fb1b0e9SMike Kravetz 		 * there is no region map is if a hole was punched via
8231fb1b0e9SMike Kravetz 		 * fallocate.  In this case, there really are no reverves to
8241fb1b0e9SMike Kravetz 		 * use.  This situation is indicated if chg != 0.
8251fb1b0e9SMike Kravetz 		 */
8261fb1b0e9SMike Kravetz 		if (chg)
8271fb1b0e9SMike Kravetz 			return false;
8281fb1b0e9SMike Kravetz 		else
829559ec2f8SNicholas Krause 			return true;
8301fb1b0e9SMike Kravetz 	}
831a63884e9SJoonsoo Kim 
832a63884e9SJoonsoo Kim 	/*
833a63884e9SJoonsoo Kim 	 * Only the process that called mmap() has reserves for
834a63884e9SJoonsoo Kim 	 * private mappings.
835a63884e9SJoonsoo Kim 	 */
83667961f9dSMike Kravetz 	if (is_vma_resv_set(vma, HPAGE_RESV_OWNER)) {
83767961f9dSMike Kravetz 		/*
83867961f9dSMike Kravetz 		 * Like the shared case above, a hole punch or truncate
83967961f9dSMike Kravetz 		 * could have been performed on the private mapping.
84067961f9dSMike Kravetz 		 * Examine the value of chg to determine if reserves
84167961f9dSMike Kravetz 		 * actually exist or were previously consumed.
84267961f9dSMike Kravetz 		 * Very Subtle - The value of chg comes from a previous
84367961f9dSMike Kravetz 		 * call to vma_needs_reserves().  The reserve map for
84467961f9dSMike Kravetz 		 * private mappings has different (opposite) semantics
84567961f9dSMike Kravetz 		 * than that of shared mappings.  vma_needs_reserves()
84667961f9dSMike Kravetz 		 * has already taken this difference in semantics into
84767961f9dSMike Kravetz 		 * account.  Therefore, the meaning of chg is the same
84867961f9dSMike Kravetz 		 * as in the shared case above.  Code could easily be
84967961f9dSMike Kravetz 		 * combined, but keeping it separate draws attention to
85067961f9dSMike Kravetz 		 * subtle differences.
85167961f9dSMike Kravetz 		 */
85267961f9dSMike Kravetz 		if (chg)
85367961f9dSMike Kravetz 			return false;
85467961f9dSMike Kravetz 		else
855559ec2f8SNicholas Krause 			return true;
85667961f9dSMike Kravetz 	}
857a63884e9SJoonsoo Kim 
858559ec2f8SNicholas Krause 	return false;
859a1e78772SMel Gorman }
860a1e78772SMel Gorman 
861a5516438SAndi Kleen static void enqueue_huge_page(struct hstate *h, struct page *page)
8621da177e4SLinus Torvalds {
8631da177e4SLinus Torvalds 	int nid = page_to_nid(page);
8640edaecfaSAneesh Kumar K.V 	list_move(&page->lru, &h->hugepage_freelists[nid]);
865a5516438SAndi Kleen 	h->free_huge_pages++;
866a5516438SAndi Kleen 	h->free_huge_pages_node[nid]++;
8671da177e4SLinus Torvalds }
8681da177e4SLinus Torvalds 
869bf50bab2SNaoya Horiguchi static struct page *dequeue_huge_page_node(struct hstate *h, int nid)
870bf50bab2SNaoya Horiguchi {
871bf50bab2SNaoya Horiguchi 	struct page *page;
872bf50bab2SNaoya Horiguchi 
873c8721bbbSNaoya Horiguchi 	list_for_each_entry(page, &h->hugepage_freelists[nid], lru)
874c8721bbbSNaoya Horiguchi 		if (!is_migrate_isolate_page(page))
875c8721bbbSNaoya Horiguchi 			break;
876c8721bbbSNaoya Horiguchi 	/*
877c8721bbbSNaoya Horiguchi 	 * if 'non-isolated free hugepage' not found on the list,
878c8721bbbSNaoya Horiguchi 	 * the allocation fails.
879c8721bbbSNaoya Horiguchi 	 */
880c8721bbbSNaoya Horiguchi 	if (&h->hugepage_freelists[nid] == &page->lru)
881bf50bab2SNaoya Horiguchi 		return NULL;
8820edaecfaSAneesh Kumar K.V 	list_move(&page->lru, &h->hugepage_activelist);
883a9869b83SNaoya Horiguchi 	set_page_refcounted(page);
884bf50bab2SNaoya Horiguchi 	h->free_huge_pages--;
885bf50bab2SNaoya Horiguchi 	h->free_huge_pages_node[nid]--;
886bf50bab2SNaoya Horiguchi 	return page;
887bf50bab2SNaoya Horiguchi }
888bf50bab2SNaoya Horiguchi 
88986cdb465SNaoya Horiguchi /* Movability of hugepages depends on migration support. */
89086cdb465SNaoya Horiguchi static inline gfp_t htlb_alloc_mask(struct hstate *h)
89186cdb465SNaoya Horiguchi {
892100873d7SNaoya Horiguchi 	if (hugepages_treat_as_movable || hugepage_migration_supported(h))
89386cdb465SNaoya Horiguchi 		return GFP_HIGHUSER_MOVABLE;
89486cdb465SNaoya Horiguchi 	else
89586cdb465SNaoya Horiguchi 		return GFP_HIGHUSER;
89686cdb465SNaoya Horiguchi }
89786cdb465SNaoya Horiguchi 
898a5516438SAndi Kleen static struct page *dequeue_huge_page_vma(struct hstate *h,
899a5516438SAndi Kleen 				struct vm_area_struct *vma,
900af0ed73eSJoonsoo Kim 				unsigned long address, int avoid_reserve,
901af0ed73eSJoonsoo Kim 				long chg)
9021da177e4SLinus Torvalds {
903b1c12cbcSKonstantin Khlebnikov 	struct page *page = NULL;
904480eccf9SLee Schermerhorn 	struct mempolicy *mpol;
90519770b32SMel Gorman 	nodemask_t *nodemask;
906c0ff7453SMiao Xie 	struct zonelist *zonelist;
907dd1a239fSMel Gorman 	struct zone *zone;
908dd1a239fSMel Gorman 	struct zoneref *z;
909cc9a6c87SMel Gorman 	unsigned int cpuset_mems_cookie;
9101da177e4SLinus Torvalds 
911a1e78772SMel Gorman 	/*
912a1e78772SMel Gorman 	 * A child process with MAP_PRIVATE mappings created by their parent
913a1e78772SMel Gorman 	 * have no page reserves. This check ensures that reservations are
914a1e78772SMel Gorman 	 * not "stolen". The child may still get SIGKILLed
915a1e78772SMel Gorman 	 */
916af0ed73eSJoonsoo Kim 	if (!vma_has_reserves(vma, chg) &&
917a5516438SAndi Kleen 			h->free_huge_pages - h->resv_huge_pages == 0)
918c0ff7453SMiao Xie 		goto err;
919a1e78772SMel Gorman 
92004f2cbe3SMel Gorman 	/* If reserves cannot be used, ensure enough pages are in the pool */
921a5516438SAndi Kleen 	if (avoid_reserve && h->free_huge_pages - h->resv_huge_pages == 0)
9226eab04a8SJustin P. Mattock 		goto err;
92304f2cbe3SMel Gorman 
9249966c4bbSJoonsoo Kim retry_cpuset:
925d26914d1SMel Gorman 	cpuset_mems_cookie = read_mems_allowed_begin();
9269966c4bbSJoonsoo Kim 	zonelist = huge_zonelist(vma, address,
92786cdb465SNaoya Horiguchi 					htlb_alloc_mask(h), &mpol, &nodemask);
9289966c4bbSJoonsoo Kim 
92919770b32SMel Gorman 	for_each_zone_zonelist_nodemask(zone, z, zonelist,
93019770b32SMel Gorman 						MAX_NR_ZONES - 1, nodemask) {
931344736f2SVladimir Davydov 		if (cpuset_zone_allowed(zone, htlb_alloc_mask(h))) {
932bf50bab2SNaoya Horiguchi 			page = dequeue_huge_page_node(h, zone_to_nid(zone));
933bf50bab2SNaoya Horiguchi 			if (page) {
934af0ed73eSJoonsoo Kim 				if (avoid_reserve)
935af0ed73eSJoonsoo Kim 					break;
936af0ed73eSJoonsoo Kim 				if (!vma_has_reserves(vma, chg))
937af0ed73eSJoonsoo Kim 					break;
938af0ed73eSJoonsoo Kim 
93907443a85SJoonsoo Kim 				SetPagePrivate(page);
940a63884e9SJoonsoo Kim 				h->resv_huge_pages--;
9415ab3ee7bSKen Chen 				break;
9421da177e4SLinus Torvalds 			}
9433abf7afdSAndrew Morton 		}
944bf50bab2SNaoya Horiguchi 	}
945cc9a6c87SMel Gorman 
946cc9a6c87SMel Gorman 	mpol_cond_put(mpol);
947d26914d1SMel Gorman 	if (unlikely(!page && read_mems_allowed_retry(cpuset_mems_cookie)))
948cc9a6c87SMel Gorman 		goto retry_cpuset;
949cc9a6c87SMel Gorman 	return page;
950cc9a6c87SMel Gorman 
951c0ff7453SMiao Xie err:
952cc9a6c87SMel Gorman 	return NULL;
9531da177e4SLinus Torvalds }
9541da177e4SLinus Torvalds 
9551cac6f2cSLuiz Capitulino /*
9561cac6f2cSLuiz Capitulino  * common helper functions for hstate_next_node_to_{alloc|free}.
9571cac6f2cSLuiz Capitulino  * We may have allocated or freed a huge page based on a different
9581cac6f2cSLuiz Capitulino  * nodes_allowed previously, so h->next_node_to_{alloc|free} might
9591cac6f2cSLuiz Capitulino  * be outside of *nodes_allowed.  Ensure that we use an allowed
9601cac6f2cSLuiz Capitulino  * node for alloc or free.
9611cac6f2cSLuiz Capitulino  */
9621cac6f2cSLuiz Capitulino static int next_node_allowed(int nid, nodemask_t *nodes_allowed)
9631cac6f2cSLuiz Capitulino {
9640edaf86cSAndrew Morton 	nid = next_node_in(nid, *nodes_allowed);
9651cac6f2cSLuiz Capitulino 	VM_BUG_ON(nid >= MAX_NUMNODES);
9661cac6f2cSLuiz Capitulino 
9671cac6f2cSLuiz Capitulino 	return nid;
9681cac6f2cSLuiz Capitulino }
9691cac6f2cSLuiz Capitulino 
9701cac6f2cSLuiz Capitulino static int get_valid_node_allowed(int nid, nodemask_t *nodes_allowed)
9711cac6f2cSLuiz Capitulino {
9721cac6f2cSLuiz Capitulino 	if (!node_isset(nid, *nodes_allowed))
9731cac6f2cSLuiz Capitulino 		nid = next_node_allowed(nid, nodes_allowed);
9741cac6f2cSLuiz Capitulino 	return nid;
9751cac6f2cSLuiz Capitulino }
9761cac6f2cSLuiz Capitulino 
9771cac6f2cSLuiz Capitulino /*
9781cac6f2cSLuiz Capitulino  * returns the previously saved node ["this node"] from which to
9791cac6f2cSLuiz Capitulino  * allocate a persistent huge page for the pool and advance the
9801cac6f2cSLuiz Capitulino  * next node from which to allocate, handling wrap at end of node
9811cac6f2cSLuiz Capitulino  * mask.
9821cac6f2cSLuiz Capitulino  */
9831cac6f2cSLuiz Capitulino static int hstate_next_node_to_alloc(struct hstate *h,
9841cac6f2cSLuiz Capitulino 					nodemask_t *nodes_allowed)
9851cac6f2cSLuiz Capitulino {
9861cac6f2cSLuiz Capitulino 	int nid;
9871cac6f2cSLuiz Capitulino 
9881cac6f2cSLuiz Capitulino 	VM_BUG_ON(!nodes_allowed);
9891cac6f2cSLuiz Capitulino 
9901cac6f2cSLuiz Capitulino 	nid = get_valid_node_allowed(h->next_nid_to_alloc, nodes_allowed);
9911cac6f2cSLuiz Capitulino 	h->next_nid_to_alloc = next_node_allowed(nid, nodes_allowed);
9921cac6f2cSLuiz Capitulino 
9931cac6f2cSLuiz Capitulino 	return nid;
9941cac6f2cSLuiz Capitulino }
9951cac6f2cSLuiz Capitulino 
9961cac6f2cSLuiz Capitulino /*
9971cac6f2cSLuiz Capitulino  * helper for free_pool_huge_page() - return the previously saved
9981cac6f2cSLuiz Capitulino  * node ["this node"] from which to free a huge page.  Advance the
9991cac6f2cSLuiz Capitulino  * next node id whether or not we find a free huge page to free so
10001cac6f2cSLuiz Capitulino  * that the next attempt to free addresses the next node.
10011cac6f2cSLuiz Capitulino  */
10021cac6f2cSLuiz Capitulino static int hstate_next_node_to_free(struct hstate *h, nodemask_t *nodes_allowed)
10031cac6f2cSLuiz Capitulino {
10041cac6f2cSLuiz Capitulino 	int nid;
10051cac6f2cSLuiz Capitulino 
10061cac6f2cSLuiz Capitulino 	VM_BUG_ON(!nodes_allowed);
10071cac6f2cSLuiz Capitulino 
10081cac6f2cSLuiz Capitulino 	nid = get_valid_node_allowed(h->next_nid_to_free, nodes_allowed);
10091cac6f2cSLuiz Capitulino 	h->next_nid_to_free = next_node_allowed(nid, nodes_allowed);
10101cac6f2cSLuiz Capitulino 
10111cac6f2cSLuiz Capitulino 	return nid;
10121cac6f2cSLuiz Capitulino }
10131cac6f2cSLuiz Capitulino 
10141cac6f2cSLuiz Capitulino #define for_each_node_mask_to_alloc(hs, nr_nodes, node, mask)		\
10151cac6f2cSLuiz Capitulino 	for (nr_nodes = nodes_weight(*mask);				\
10161cac6f2cSLuiz Capitulino 		nr_nodes > 0 &&						\
10171cac6f2cSLuiz Capitulino 		((node = hstate_next_node_to_alloc(hs, mask)) || 1);	\
10181cac6f2cSLuiz Capitulino 		nr_nodes--)
10191cac6f2cSLuiz Capitulino 
10201cac6f2cSLuiz Capitulino #define for_each_node_mask_to_free(hs, nr_nodes, node, mask)		\
10211cac6f2cSLuiz Capitulino 	for (nr_nodes = nodes_weight(*mask);				\
10221cac6f2cSLuiz Capitulino 		nr_nodes > 0 &&						\
10231cac6f2cSLuiz Capitulino 		((node = hstate_next_node_to_free(hs, mask)) || 1);	\
10241cac6f2cSLuiz Capitulino 		nr_nodes--)
10251cac6f2cSLuiz Capitulino 
1026461a7184SYisheng Xie #if defined(CONFIG_ARCH_HAS_GIGANTIC_PAGE) && \
1027d08de8e2SGerald Schaefer 	((defined(CONFIG_MEMORY_ISOLATION) && defined(CONFIG_COMPACTION)) || \
1028d08de8e2SGerald Schaefer 	defined(CONFIG_CMA))
1029944d9fecSLuiz Capitulino static void destroy_compound_gigantic_page(struct page *page,
1030d00181b9SKirill A. Shutemov 					unsigned int order)
1031944d9fecSLuiz Capitulino {
1032944d9fecSLuiz Capitulino 	int i;
1033944d9fecSLuiz Capitulino 	int nr_pages = 1 << order;
1034944d9fecSLuiz Capitulino 	struct page *p = page + 1;
1035944d9fecSLuiz Capitulino 
1036c8cc708aSGerald Schaefer 	atomic_set(compound_mapcount_ptr(page), 0);
1037944d9fecSLuiz Capitulino 	for (i = 1; i < nr_pages; i++, p = mem_map_next(p, page, i)) {
10381d798ca3SKirill A. Shutemov 		clear_compound_head(p);
1039944d9fecSLuiz Capitulino 		set_page_refcounted(p);
1040944d9fecSLuiz Capitulino 	}
1041944d9fecSLuiz Capitulino 
1042944d9fecSLuiz Capitulino 	set_compound_order(page, 0);
1043944d9fecSLuiz Capitulino 	__ClearPageHead(page);
1044944d9fecSLuiz Capitulino }
1045944d9fecSLuiz Capitulino 
1046d00181b9SKirill A. Shutemov static void free_gigantic_page(struct page *page, unsigned int order)
1047944d9fecSLuiz Capitulino {
1048944d9fecSLuiz Capitulino 	free_contig_range(page_to_pfn(page), 1 << order);
1049944d9fecSLuiz Capitulino }
1050944d9fecSLuiz Capitulino 
1051944d9fecSLuiz Capitulino static int __alloc_gigantic_page(unsigned long start_pfn,
1052944d9fecSLuiz Capitulino 				unsigned long nr_pages)
1053944d9fecSLuiz Capitulino {
1054944d9fecSLuiz Capitulino 	unsigned long end_pfn = start_pfn + nr_pages;
1055944d9fecSLuiz Capitulino 	return alloc_contig_range(start_pfn, end_pfn, MIGRATE_MOVABLE);
1056944d9fecSLuiz Capitulino }
1057944d9fecSLuiz Capitulino 
1058f44b2ddaSJoonsoo Kim static bool pfn_range_valid_gigantic(struct zone *z,
1059f44b2ddaSJoonsoo Kim 			unsigned long start_pfn, unsigned long nr_pages)
1060944d9fecSLuiz Capitulino {
1061944d9fecSLuiz Capitulino 	unsigned long i, end_pfn = start_pfn + nr_pages;
1062944d9fecSLuiz Capitulino 	struct page *page;
1063944d9fecSLuiz Capitulino 
1064944d9fecSLuiz Capitulino 	for (i = start_pfn; i < end_pfn; i++) {
1065944d9fecSLuiz Capitulino 		if (!pfn_valid(i))
1066944d9fecSLuiz Capitulino 			return false;
1067944d9fecSLuiz Capitulino 
1068944d9fecSLuiz Capitulino 		page = pfn_to_page(i);
1069944d9fecSLuiz Capitulino 
1070f44b2ddaSJoonsoo Kim 		if (page_zone(page) != z)
1071f44b2ddaSJoonsoo Kim 			return false;
1072f44b2ddaSJoonsoo Kim 
1073944d9fecSLuiz Capitulino 		if (PageReserved(page))
1074944d9fecSLuiz Capitulino 			return false;
1075944d9fecSLuiz Capitulino 
1076944d9fecSLuiz Capitulino 		if (page_count(page) > 0)
1077944d9fecSLuiz Capitulino 			return false;
1078944d9fecSLuiz Capitulino 
1079944d9fecSLuiz Capitulino 		if (PageHuge(page))
1080944d9fecSLuiz Capitulino 			return false;
1081944d9fecSLuiz Capitulino 	}
1082944d9fecSLuiz Capitulino 
1083944d9fecSLuiz Capitulino 	return true;
1084944d9fecSLuiz Capitulino }
1085944d9fecSLuiz Capitulino 
1086944d9fecSLuiz Capitulino static bool zone_spans_last_pfn(const struct zone *zone,
1087944d9fecSLuiz Capitulino 			unsigned long start_pfn, unsigned long nr_pages)
1088944d9fecSLuiz Capitulino {
1089944d9fecSLuiz Capitulino 	unsigned long last_pfn = start_pfn + nr_pages - 1;
1090944d9fecSLuiz Capitulino 	return zone_spans_pfn(zone, last_pfn);
1091944d9fecSLuiz Capitulino }
1092944d9fecSLuiz Capitulino 
1093d00181b9SKirill A. Shutemov static struct page *alloc_gigantic_page(int nid, unsigned int order)
1094944d9fecSLuiz Capitulino {
1095944d9fecSLuiz Capitulino 	unsigned long nr_pages = 1 << order;
1096944d9fecSLuiz Capitulino 	unsigned long ret, pfn, flags;
1097944d9fecSLuiz Capitulino 	struct zone *z;
1098944d9fecSLuiz Capitulino 
1099944d9fecSLuiz Capitulino 	z = NODE_DATA(nid)->node_zones;
1100944d9fecSLuiz Capitulino 	for (; z - NODE_DATA(nid)->node_zones < MAX_NR_ZONES; z++) {
1101944d9fecSLuiz Capitulino 		spin_lock_irqsave(&z->lock, flags);
1102944d9fecSLuiz Capitulino 
1103944d9fecSLuiz Capitulino 		pfn = ALIGN(z->zone_start_pfn, nr_pages);
1104944d9fecSLuiz Capitulino 		while (zone_spans_last_pfn(z, pfn, nr_pages)) {
1105f44b2ddaSJoonsoo Kim 			if (pfn_range_valid_gigantic(z, pfn, nr_pages)) {
1106944d9fecSLuiz Capitulino 				/*
1107944d9fecSLuiz Capitulino 				 * We release the zone lock here because
1108944d9fecSLuiz Capitulino 				 * alloc_contig_range() will also lock the zone
1109944d9fecSLuiz Capitulino 				 * at some point. If there's an allocation
1110944d9fecSLuiz Capitulino 				 * spinning on this lock, it may win the race
1111944d9fecSLuiz Capitulino 				 * and cause alloc_contig_range() to fail...
1112944d9fecSLuiz Capitulino 				 */
1113944d9fecSLuiz Capitulino 				spin_unlock_irqrestore(&z->lock, flags);
1114944d9fecSLuiz Capitulino 				ret = __alloc_gigantic_page(pfn, nr_pages);
1115944d9fecSLuiz Capitulino 				if (!ret)
1116944d9fecSLuiz Capitulino 					return pfn_to_page(pfn);
1117944d9fecSLuiz Capitulino 				spin_lock_irqsave(&z->lock, flags);
1118944d9fecSLuiz Capitulino 			}
1119944d9fecSLuiz Capitulino 			pfn += nr_pages;
1120944d9fecSLuiz Capitulino 		}
1121944d9fecSLuiz Capitulino 
1122944d9fecSLuiz Capitulino 		spin_unlock_irqrestore(&z->lock, flags);
1123944d9fecSLuiz Capitulino 	}
1124944d9fecSLuiz Capitulino 
1125944d9fecSLuiz Capitulino 	return NULL;
1126944d9fecSLuiz Capitulino }
1127944d9fecSLuiz Capitulino 
1128944d9fecSLuiz Capitulino static void prep_new_huge_page(struct hstate *h, struct page *page, int nid);
1129d00181b9SKirill A. Shutemov static void prep_compound_gigantic_page(struct page *page, unsigned int order);
1130944d9fecSLuiz Capitulino 
1131944d9fecSLuiz Capitulino static struct page *alloc_fresh_gigantic_page_node(struct hstate *h, int nid)
1132944d9fecSLuiz Capitulino {
1133944d9fecSLuiz Capitulino 	struct page *page;
1134944d9fecSLuiz Capitulino 
1135944d9fecSLuiz Capitulino 	page = alloc_gigantic_page(nid, huge_page_order(h));
1136944d9fecSLuiz Capitulino 	if (page) {
1137944d9fecSLuiz Capitulino 		prep_compound_gigantic_page(page, huge_page_order(h));
1138944d9fecSLuiz Capitulino 		prep_new_huge_page(h, page, nid);
1139944d9fecSLuiz Capitulino 	}
1140944d9fecSLuiz Capitulino 
1141944d9fecSLuiz Capitulino 	return page;
1142944d9fecSLuiz Capitulino }
1143944d9fecSLuiz Capitulino 
1144944d9fecSLuiz Capitulino static int alloc_fresh_gigantic_page(struct hstate *h,
1145944d9fecSLuiz Capitulino 				nodemask_t *nodes_allowed)
1146944d9fecSLuiz Capitulino {
1147944d9fecSLuiz Capitulino 	struct page *page = NULL;
1148944d9fecSLuiz Capitulino 	int nr_nodes, node;
1149944d9fecSLuiz Capitulino 
1150944d9fecSLuiz Capitulino 	for_each_node_mask_to_alloc(h, nr_nodes, node, nodes_allowed) {
1151944d9fecSLuiz Capitulino 		page = alloc_fresh_gigantic_page_node(h, node);
1152944d9fecSLuiz Capitulino 		if (page)
1153944d9fecSLuiz Capitulino 			return 1;
1154944d9fecSLuiz Capitulino 	}
1155944d9fecSLuiz Capitulino 
1156944d9fecSLuiz Capitulino 	return 0;
1157944d9fecSLuiz Capitulino }
1158944d9fecSLuiz Capitulino 
1159944d9fecSLuiz Capitulino static inline bool gigantic_page_supported(void) { return true; }
1160944d9fecSLuiz Capitulino #else
1161944d9fecSLuiz Capitulino static inline bool gigantic_page_supported(void) { return false; }
1162d00181b9SKirill A. Shutemov static inline void free_gigantic_page(struct page *page, unsigned int order) { }
1163944d9fecSLuiz Capitulino static inline void destroy_compound_gigantic_page(struct page *page,
1164d00181b9SKirill A. Shutemov 						unsigned int order) { }
1165944d9fecSLuiz Capitulino static inline int alloc_fresh_gigantic_page(struct hstate *h,
1166944d9fecSLuiz Capitulino 					nodemask_t *nodes_allowed) { return 0; }
1167944d9fecSLuiz Capitulino #endif
1168944d9fecSLuiz Capitulino 
1169a5516438SAndi Kleen static void update_and_free_page(struct hstate *h, struct page *page)
11706af2acb6SAdam Litke {
11716af2acb6SAdam Litke 	int i;
1172a5516438SAndi Kleen 
1173944d9fecSLuiz Capitulino 	if (hstate_is_gigantic(h) && !gigantic_page_supported())
1174944d9fecSLuiz Capitulino 		return;
117518229df5SAndy Whitcroft 
1176a5516438SAndi Kleen 	h->nr_huge_pages--;
1177a5516438SAndi Kleen 	h->nr_huge_pages_node[page_to_nid(page)]--;
1178a5516438SAndi Kleen 	for (i = 0; i < pages_per_huge_page(h); i++) {
117932f84528SChris Forbes 		page[i].flags &= ~(1 << PG_locked | 1 << PG_error |
118032f84528SChris Forbes 				1 << PG_referenced | 1 << PG_dirty |
1181a7407a27SLuiz Capitulino 				1 << PG_active | 1 << PG_private |
1182a7407a27SLuiz Capitulino 				1 << PG_writeback);
11836af2acb6SAdam Litke 	}
1184309381feSSasha Levin 	VM_BUG_ON_PAGE(hugetlb_cgroup_from_page(page), page);
1185f1e61557SKirill A. Shutemov 	set_compound_page_dtor(page, NULL_COMPOUND_DTOR);
11866af2acb6SAdam Litke 	set_page_refcounted(page);
1187944d9fecSLuiz Capitulino 	if (hstate_is_gigantic(h)) {
1188944d9fecSLuiz Capitulino 		destroy_compound_gigantic_page(page, huge_page_order(h));
1189944d9fecSLuiz Capitulino 		free_gigantic_page(page, huge_page_order(h));
1190944d9fecSLuiz Capitulino 	} else {
1191a5516438SAndi Kleen 		__free_pages(page, huge_page_order(h));
11926af2acb6SAdam Litke 	}
1193944d9fecSLuiz Capitulino }
11946af2acb6SAdam Litke 
1195e5ff2159SAndi Kleen struct hstate *size_to_hstate(unsigned long size)
1196e5ff2159SAndi Kleen {
1197e5ff2159SAndi Kleen 	struct hstate *h;
1198e5ff2159SAndi Kleen 
1199e5ff2159SAndi Kleen 	for_each_hstate(h) {
1200e5ff2159SAndi Kleen 		if (huge_page_size(h) == size)
1201e5ff2159SAndi Kleen 			return h;
1202e5ff2159SAndi Kleen 	}
1203e5ff2159SAndi Kleen 	return NULL;
1204e5ff2159SAndi Kleen }
1205e5ff2159SAndi Kleen 
1206bcc54222SNaoya Horiguchi /*
1207bcc54222SNaoya Horiguchi  * Test to determine whether the hugepage is "active/in-use" (i.e. being linked
1208bcc54222SNaoya Horiguchi  * to hstate->hugepage_activelist.)
1209bcc54222SNaoya Horiguchi  *
1210bcc54222SNaoya Horiguchi  * This function can be called for tail pages, but never returns true for them.
1211bcc54222SNaoya Horiguchi  */
1212bcc54222SNaoya Horiguchi bool page_huge_active(struct page *page)
1213bcc54222SNaoya Horiguchi {
1214bcc54222SNaoya Horiguchi 	VM_BUG_ON_PAGE(!PageHuge(page), page);
1215bcc54222SNaoya Horiguchi 	return PageHead(page) && PagePrivate(&page[1]);
1216bcc54222SNaoya Horiguchi }
1217bcc54222SNaoya Horiguchi 
1218bcc54222SNaoya Horiguchi /* never called for tail page */
1219bcc54222SNaoya Horiguchi static void set_page_huge_active(struct page *page)
1220bcc54222SNaoya Horiguchi {
1221bcc54222SNaoya Horiguchi 	VM_BUG_ON_PAGE(!PageHeadHuge(page), page);
1222bcc54222SNaoya Horiguchi 	SetPagePrivate(&page[1]);
1223bcc54222SNaoya Horiguchi }
1224bcc54222SNaoya Horiguchi 
1225bcc54222SNaoya Horiguchi static void clear_page_huge_active(struct page *page)
1226bcc54222SNaoya Horiguchi {
1227bcc54222SNaoya Horiguchi 	VM_BUG_ON_PAGE(!PageHeadHuge(page), page);
1228bcc54222SNaoya Horiguchi 	ClearPagePrivate(&page[1]);
1229bcc54222SNaoya Horiguchi }
1230bcc54222SNaoya Horiguchi 
12318f1d26d0SAtsushi Kumagai void free_huge_page(struct page *page)
123227a85ef1SDavid Gibson {
1233a5516438SAndi Kleen 	/*
1234a5516438SAndi Kleen 	 * Can't pass hstate in here because it is called from the
1235a5516438SAndi Kleen 	 * compound page destructor.
1236a5516438SAndi Kleen 	 */
1237e5ff2159SAndi Kleen 	struct hstate *h = page_hstate(page);
12387893d1d5SAdam Litke 	int nid = page_to_nid(page);
123990481622SDavid Gibson 	struct hugepage_subpool *spool =
124090481622SDavid Gibson 		(struct hugepage_subpool *)page_private(page);
124107443a85SJoonsoo Kim 	bool restore_reserve;
124227a85ef1SDavid Gibson 
1243e5df70abSAndy Whitcroft 	set_page_private(page, 0);
124423be7468SMel Gorman 	page->mapping = NULL;
1245b4330afbSMike Kravetz 	VM_BUG_ON_PAGE(page_count(page), page);
1246b4330afbSMike Kravetz 	VM_BUG_ON_PAGE(page_mapcount(page), page);
124707443a85SJoonsoo Kim 	restore_reserve = PagePrivate(page);
124816c794b4SJoonsoo Kim 	ClearPagePrivate(page);
124927a85ef1SDavid Gibson 
12501c5ecae3SMike Kravetz 	/*
12511c5ecae3SMike Kravetz 	 * A return code of zero implies that the subpool will be under its
12521c5ecae3SMike Kravetz 	 * minimum size if the reservation is not restored after page is free.
12531c5ecae3SMike Kravetz 	 * Therefore, force restore_reserve operation.
12541c5ecae3SMike Kravetz 	 */
12551c5ecae3SMike Kravetz 	if (hugepage_subpool_put_pages(spool, 1) == 0)
12561c5ecae3SMike Kravetz 		restore_reserve = true;
12571c5ecae3SMike Kravetz 
125827a85ef1SDavid Gibson 	spin_lock(&hugetlb_lock);
1259bcc54222SNaoya Horiguchi 	clear_page_huge_active(page);
12606d76dcf4SAneesh Kumar K.V 	hugetlb_cgroup_uncharge_page(hstate_index(h),
12616d76dcf4SAneesh Kumar K.V 				     pages_per_huge_page(h), page);
126207443a85SJoonsoo Kim 	if (restore_reserve)
126307443a85SJoonsoo Kim 		h->resv_huge_pages++;
126407443a85SJoonsoo Kim 
1265944d9fecSLuiz Capitulino 	if (h->surplus_huge_pages_node[nid]) {
12660edaecfaSAneesh Kumar K.V 		/* remove the page from active list */
12670edaecfaSAneesh Kumar K.V 		list_del(&page->lru);
1268a5516438SAndi Kleen 		update_and_free_page(h, page);
1269a5516438SAndi Kleen 		h->surplus_huge_pages--;
1270a5516438SAndi Kleen 		h->surplus_huge_pages_node[nid]--;
12717893d1d5SAdam Litke 	} else {
12725d3a551cSWill Deacon 		arch_clear_hugepage_flags(page);
1273a5516438SAndi Kleen 		enqueue_huge_page(h, page);
12747893d1d5SAdam Litke 	}
127527a85ef1SDavid Gibson 	spin_unlock(&hugetlb_lock);
127627a85ef1SDavid Gibson }
127727a85ef1SDavid Gibson 
1278a5516438SAndi Kleen static void prep_new_huge_page(struct hstate *h, struct page *page, int nid)
1279b7ba30c6SAndi Kleen {
12800edaecfaSAneesh Kumar K.V 	INIT_LIST_HEAD(&page->lru);
1281f1e61557SKirill A. Shutemov 	set_compound_page_dtor(page, HUGETLB_PAGE_DTOR);
1282b7ba30c6SAndi Kleen 	spin_lock(&hugetlb_lock);
12839dd540e2SAneesh Kumar K.V 	set_hugetlb_cgroup(page, NULL);
1284a5516438SAndi Kleen 	h->nr_huge_pages++;
1285a5516438SAndi Kleen 	h->nr_huge_pages_node[nid]++;
1286b7ba30c6SAndi Kleen 	spin_unlock(&hugetlb_lock);
1287b7ba30c6SAndi Kleen 	put_page(page); /* free it into the hugepage allocator */
1288b7ba30c6SAndi Kleen }
1289b7ba30c6SAndi Kleen 
1290d00181b9SKirill A. Shutemov static void prep_compound_gigantic_page(struct page *page, unsigned int order)
129120a0307cSWu Fengguang {
129220a0307cSWu Fengguang 	int i;
129320a0307cSWu Fengguang 	int nr_pages = 1 << order;
129420a0307cSWu Fengguang 	struct page *p = page + 1;
129520a0307cSWu Fengguang 
129620a0307cSWu Fengguang 	/* we rely on prep_new_huge_page to set the destructor */
129720a0307cSWu Fengguang 	set_compound_order(page, order);
1298ef5a22beSAndrea Arcangeli 	__ClearPageReserved(page);
1299de09d31dSKirill A. Shutemov 	__SetPageHead(page);
130020a0307cSWu Fengguang 	for (i = 1; i < nr_pages; i++, p = mem_map_next(p, page, i)) {
1301ef5a22beSAndrea Arcangeli 		/*
1302ef5a22beSAndrea Arcangeli 		 * For gigantic hugepages allocated through bootmem at
1303ef5a22beSAndrea Arcangeli 		 * boot, it's safer to be consistent with the not-gigantic
1304ef5a22beSAndrea Arcangeli 		 * hugepages and clear the PG_reserved bit from all tail pages
1305ef5a22beSAndrea Arcangeli 		 * too.  Otherwse drivers using get_user_pages() to access tail
1306ef5a22beSAndrea Arcangeli 		 * pages may get the reference counting wrong if they see
1307ef5a22beSAndrea Arcangeli 		 * PG_reserved set on a tail page (despite the head page not
1308ef5a22beSAndrea Arcangeli 		 * having PG_reserved set).  Enforcing this consistency between
1309ef5a22beSAndrea Arcangeli 		 * head and tail pages allows drivers to optimize away a check
1310ef5a22beSAndrea Arcangeli 		 * on the head page when they need know if put_page() is needed
1311ef5a22beSAndrea Arcangeli 		 * after get_user_pages().
1312ef5a22beSAndrea Arcangeli 		 */
1313ef5a22beSAndrea Arcangeli 		__ClearPageReserved(p);
131458a84aa9SYouquan Song 		set_page_count(p, 0);
13151d798ca3SKirill A. Shutemov 		set_compound_head(p, page);
131620a0307cSWu Fengguang 	}
1317b4330afbSMike Kravetz 	atomic_set(compound_mapcount_ptr(page), -1);
131820a0307cSWu Fengguang }
131920a0307cSWu Fengguang 
13207795912cSAndrew Morton /*
13217795912cSAndrew Morton  * PageHuge() only returns true for hugetlbfs pages, but not for normal or
13227795912cSAndrew Morton  * transparent huge pages.  See the PageTransHuge() documentation for more
13237795912cSAndrew Morton  * details.
13247795912cSAndrew Morton  */
132520a0307cSWu Fengguang int PageHuge(struct page *page)
132620a0307cSWu Fengguang {
132720a0307cSWu Fengguang 	if (!PageCompound(page))
132820a0307cSWu Fengguang 		return 0;
132920a0307cSWu Fengguang 
133020a0307cSWu Fengguang 	page = compound_head(page);
1331f1e61557SKirill A. Shutemov 	return page[1].compound_dtor == HUGETLB_PAGE_DTOR;
133220a0307cSWu Fengguang }
133343131e14SNaoya Horiguchi EXPORT_SYMBOL_GPL(PageHuge);
133443131e14SNaoya Horiguchi 
133527c73ae7SAndrea Arcangeli /*
133627c73ae7SAndrea Arcangeli  * PageHeadHuge() only returns true for hugetlbfs head page, but not for
133727c73ae7SAndrea Arcangeli  * normal or transparent huge pages.
133827c73ae7SAndrea Arcangeli  */
133927c73ae7SAndrea Arcangeli int PageHeadHuge(struct page *page_head)
134027c73ae7SAndrea Arcangeli {
134127c73ae7SAndrea Arcangeli 	if (!PageHead(page_head))
134227c73ae7SAndrea Arcangeli 		return 0;
134327c73ae7SAndrea Arcangeli 
1344758f66a2SAndrew Morton 	return get_compound_page_dtor(page_head) == free_huge_page;
134527c73ae7SAndrea Arcangeli }
134627c73ae7SAndrea Arcangeli 
134713d60f4bSZhang Yi pgoff_t __basepage_index(struct page *page)
134813d60f4bSZhang Yi {
134913d60f4bSZhang Yi 	struct page *page_head = compound_head(page);
135013d60f4bSZhang Yi 	pgoff_t index = page_index(page_head);
135113d60f4bSZhang Yi 	unsigned long compound_idx;
135213d60f4bSZhang Yi 
135313d60f4bSZhang Yi 	if (!PageHuge(page_head))
135413d60f4bSZhang Yi 		return page_index(page);
135513d60f4bSZhang Yi 
135613d60f4bSZhang Yi 	if (compound_order(page_head) >= MAX_ORDER)
135713d60f4bSZhang Yi 		compound_idx = page_to_pfn(page) - page_to_pfn(page_head);
135813d60f4bSZhang Yi 	else
135913d60f4bSZhang Yi 		compound_idx = page - page_head;
136013d60f4bSZhang Yi 
136113d60f4bSZhang Yi 	return (index << compound_order(page_head)) + compound_idx;
136213d60f4bSZhang Yi }
136313d60f4bSZhang Yi 
1364a5516438SAndi Kleen static struct page *alloc_fresh_huge_page_node(struct hstate *h, int nid)
13651da177e4SLinus Torvalds {
13661da177e4SLinus Torvalds 	struct page *page;
1367f96efd58SJoe Jin 
136896db800fSVlastimil Babka 	page = __alloc_pages_node(nid,
136986cdb465SNaoya Horiguchi 		htlb_alloc_mask(h)|__GFP_COMP|__GFP_THISNODE|
1370551883aeSNishanth Aravamudan 						__GFP_REPEAT|__GFP_NOWARN,
1371a5516438SAndi Kleen 		huge_page_order(h));
13721da177e4SLinus Torvalds 	if (page) {
1373a5516438SAndi Kleen 		prep_new_huge_page(h, page, nid);
13741da177e4SLinus Torvalds 	}
137563b4613cSNishanth Aravamudan 
137663b4613cSNishanth Aravamudan 	return page;
137763b4613cSNishanth Aravamudan }
137863b4613cSNishanth Aravamudan 
1379b2261026SJoonsoo Kim static int alloc_fresh_huge_page(struct hstate *h, nodemask_t *nodes_allowed)
1380b2261026SJoonsoo Kim {
1381b2261026SJoonsoo Kim 	struct page *page;
1382b2261026SJoonsoo Kim 	int nr_nodes, node;
1383b2261026SJoonsoo Kim 	int ret = 0;
1384b2261026SJoonsoo Kim 
1385b2261026SJoonsoo Kim 	for_each_node_mask_to_alloc(h, nr_nodes, node, nodes_allowed) {
1386b2261026SJoonsoo Kim 		page = alloc_fresh_huge_page_node(h, node);
1387b2261026SJoonsoo Kim 		if (page) {
1388b2261026SJoonsoo Kim 			ret = 1;
1389b2261026SJoonsoo Kim 			break;
1390b2261026SJoonsoo Kim 		}
1391b2261026SJoonsoo Kim 	}
1392b2261026SJoonsoo Kim 
1393b2261026SJoonsoo Kim 	if (ret)
1394b2261026SJoonsoo Kim 		count_vm_event(HTLB_BUDDY_PGALLOC);
1395b2261026SJoonsoo Kim 	else
1396b2261026SJoonsoo Kim 		count_vm_event(HTLB_BUDDY_PGALLOC_FAIL);
1397b2261026SJoonsoo Kim 
1398b2261026SJoonsoo Kim 	return ret;
1399b2261026SJoonsoo Kim }
1400b2261026SJoonsoo Kim 
1401e8c5c824SLee Schermerhorn /*
1402e8c5c824SLee Schermerhorn  * Free huge page from pool from next node to free.
1403e8c5c824SLee Schermerhorn  * Attempt to keep persistent huge pages more or less
1404e8c5c824SLee Schermerhorn  * balanced over allowed nodes.
1405e8c5c824SLee Schermerhorn  * Called with hugetlb_lock locked.
1406e8c5c824SLee Schermerhorn  */
14076ae11b27SLee Schermerhorn static int free_pool_huge_page(struct hstate *h, nodemask_t *nodes_allowed,
14086ae11b27SLee Schermerhorn 							 bool acct_surplus)
1409e8c5c824SLee Schermerhorn {
1410b2261026SJoonsoo Kim 	int nr_nodes, node;
1411e8c5c824SLee Schermerhorn 	int ret = 0;
1412e8c5c824SLee Schermerhorn 
1413b2261026SJoonsoo Kim 	for_each_node_mask_to_free(h, nr_nodes, node, nodes_allowed) {
1414685f3457SLee Schermerhorn 		/*
1415685f3457SLee Schermerhorn 		 * If we're returning unused surplus pages, only examine
1416685f3457SLee Schermerhorn 		 * nodes with surplus pages.
1417685f3457SLee Schermerhorn 		 */
1418b2261026SJoonsoo Kim 		if ((!acct_surplus || h->surplus_huge_pages_node[node]) &&
1419b2261026SJoonsoo Kim 		    !list_empty(&h->hugepage_freelists[node])) {
1420e8c5c824SLee Schermerhorn 			struct page *page =
1421b2261026SJoonsoo Kim 				list_entry(h->hugepage_freelists[node].next,
1422e8c5c824SLee Schermerhorn 					  struct page, lru);
1423e8c5c824SLee Schermerhorn 			list_del(&page->lru);
1424e8c5c824SLee Schermerhorn 			h->free_huge_pages--;
1425b2261026SJoonsoo Kim 			h->free_huge_pages_node[node]--;
1426685f3457SLee Schermerhorn 			if (acct_surplus) {
1427685f3457SLee Schermerhorn 				h->surplus_huge_pages--;
1428b2261026SJoonsoo Kim 				h->surplus_huge_pages_node[node]--;
1429685f3457SLee Schermerhorn 			}
1430e8c5c824SLee Schermerhorn 			update_and_free_page(h, page);
1431e8c5c824SLee Schermerhorn 			ret = 1;
14329a76db09SLee Schermerhorn 			break;
1433e8c5c824SLee Schermerhorn 		}
1434b2261026SJoonsoo Kim 	}
1435e8c5c824SLee Schermerhorn 
1436e8c5c824SLee Schermerhorn 	return ret;
1437e8c5c824SLee Schermerhorn }
1438e8c5c824SLee Schermerhorn 
1439c8721bbbSNaoya Horiguchi /*
1440c8721bbbSNaoya Horiguchi  * Dissolve a given free hugepage into free buddy pages. This function does
1441082d5b6bSGerald Schaefer  * nothing for in-use (including surplus) hugepages. Returns -EBUSY if the
1442082d5b6bSGerald Schaefer  * number of free hugepages would be reduced below the number of reserved
1443082d5b6bSGerald Schaefer  * hugepages.
1444c8721bbbSNaoya Horiguchi  */
1445082d5b6bSGerald Schaefer static int dissolve_free_huge_page(struct page *page)
1446c8721bbbSNaoya Horiguchi {
1447082d5b6bSGerald Schaefer 	int rc = 0;
1448082d5b6bSGerald Schaefer 
1449c8721bbbSNaoya Horiguchi 	spin_lock(&hugetlb_lock);
1450c8721bbbSNaoya Horiguchi 	if (PageHuge(page) && !page_count(page)) {
14512247bb33SGerald Schaefer 		struct page *head = compound_head(page);
14522247bb33SGerald Schaefer 		struct hstate *h = page_hstate(head);
14532247bb33SGerald Schaefer 		int nid = page_to_nid(head);
1454082d5b6bSGerald Schaefer 		if (h->free_huge_pages - h->resv_huge_pages == 0) {
1455082d5b6bSGerald Schaefer 			rc = -EBUSY;
1456082d5b6bSGerald Schaefer 			goto out;
1457082d5b6bSGerald Schaefer 		}
14582247bb33SGerald Schaefer 		list_del(&head->lru);
1459c8721bbbSNaoya Horiguchi 		h->free_huge_pages--;
1460c8721bbbSNaoya Horiguchi 		h->free_huge_pages_node[nid]--;
1461c1470b33Szhong jiang 		h->max_huge_pages--;
14622247bb33SGerald Schaefer 		update_and_free_page(h, head);
1463c8721bbbSNaoya Horiguchi 	}
1464082d5b6bSGerald Schaefer out:
1465c8721bbbSNaoya Horiguchi 	spin_unlock(&hugetlb_lock);
1466082d5b6bSGerald Schaefer 	return rc;
1467c8721bbbSNaoya Horiguchi }
1468c8721bbbSNaoya Horiguchi 
1469c8721bbbSNaoya Horiguchi /*
1470c8721bbbSNaoya Horiguchi  * Dissolve free hugepages in a given pfn range. Used by memory hotplug to
1471c8721bbbSNaoya Horiguchi  * make specified memory blocks removable from the system.
14722247bb33SGerald Schaefer  * Note that this will dissolve a free gigantic hugepage completely, if any
14732247bb33SGerald Schaefer  * part of it lies within the given range.
1474082d5b6bSGerald Schaefer  * Also note that if dissolve_free_huge_page() returns with an error, all
1475082d5b6bSGerald Schaefer  * free hugepages that were dissolved before that error are lost.
1476c8721bbbSNaoya Horiguchi  */
1477082d5b6bSGerald Schaefer int dissolve_free_huge_pages(unsigned long start_pfn, unsigned long end_pfn)
1478c8721bbbSNaoya Horiguchi {
1479c8721bbbSNaoya Horiguchi 	unsigned long pfn;
1480eb03aa00SGerald Schaefer 	struct page *page;
1481082d5b6bSGerald Schaefer 	int rc = 0;
1482c8721bbbSNaoya Horiguchi 
1483d0177639SLi Zhong 	if (!hugepages_supported())
1484082d5b6bSGerald Schaefer 		return rc;
1485d0177639SLi Zhong 
1486eb03aa00SGerald Schaefer 	for (pfn = start_pfn; pfn < end_pfn; pfn += 1 << minimum_order) {
1487eb03aa00SGerald Schaefer 		page = pfn_to_page(pfn);
1488eb03aa00SGerald Schaefer 		if (PageHuge(page) && !page_count(page)) {
1489eb03aa00SGerald Schaefer 			rc = dissolve_free_huge_page(page);
1490eb03aa00SGerald Schaefer 			if (rc)
1491082d5b6bSGerald Schaefer 				break;
1492eb03aa00SGerald Schaefer 		}
1493eb03aa00SGerald Schaefer 	}
1494082d5b6bSGerald Schaefer 
1495082d5b6bSGerald Schaefer 	return rc;
1496c8721bbbSNaoya Horiguchi }
1497c8721bbbSNaoya Horiguchi 
1498099730d6SDave Hansen /*
1499099730d6SDave Hansen  * There are 3 ways this can get called:
1500099730d6SDave Hansen  * 1. With vma+addr: we use the VMA's memory policy
1501099730d6SDave Hansen  * 2. With !vma, but nid=NUMA_NO_NODE:  We try to allocate a huge
1502099730d6SDave Hansen  *    page from any node, and let the buddy allocator itself figure
1503099730d6SDave Hansen  *    it out.
1504099730d6SDave Hansen  * 3. With !vma, but nid!=NUMA_NO_NODE.  We allocate a huge page
1505099730d6SDave Hansen  *    strictly from 'nid'
1506099730d6SDave Hansen  */
1507099730d6SDave Hansen static struct page *__hugetlb_alloc_buddy_huge_page(struct hstate *h,
1508099730d6SDave Hansen 		struct vm_area_struct *vma, unsigned long addr, int nid)
1509099730d6SDave Hansen {
1510099730d6SDave Hansen 	int order = huge_page_order(h);
1511099730d6SDave Hansen 	gfp_t gfp = htlb_alloc_mask(h)|__GFP_COMP|__GFP_REPEAT|__GFP_NOWARN;
1512099730d6SDave Hansen 	unsigned int cpuset_mems_cookie;
1513099730d6SDave Hansen 
1514099730d6SDave Hansen 	/*
1515099730d6SDave Hansen 	 * We need a VMA to get a memory policy.  If we do not
1516e0ec90eeSDave Hansen 	 * have one, we use the 'nid' argument.
1517e0ec90eeSDave Hansen 	 *
1518e0ec90eeSDave Hansen 	 * The mempolicy stuff below has some non-inlined bits
1519e0ec90eeSDave Hansen 	 * and calls ->vm_ops.  That makes it hard to optimize at
1520e0ec90eeSDave Hansen 	 * compile-time, even when NUMA is off and it does
1521e0ec90eeSDave Hansen 	 * nothing.  This helps the compiler optimize it out.
1522099730d6SDave Hansen 	 */
1523e0ec90eeSDave Hansen 	if (!IS_ENABLED(CONFIG_NUMA) || !vma) {
1524099730d6SDave Hansen 		/*
1525099730d6SDave Hansen 		 * If a specific node is requested, make sure to
1526099730d6SDave Hansen 		 * get memory from there, but only when a node
1527099730d6SDave Hansen 		 * is explicitly specified.
1528099730d6SDave Hansen 		 */
1529099730d6SDave Hansen 		if (nid != NUMA_NO_NODE)
1530099730d6SDave Hansen 			gfp |= __GFP_THISNODE;
1531099730d6SDave Hansen 		/*
1532099730d6SDave Hansen 		 * Make sure to call something that can handle
1533099730d6SDave Hansen 		 * nid=NUMA_NO_NODE
1534099730d6SDave Hansen 		 */
1535099730d6SDave Hansen 		return alloc_pages_node(nid, gfp, order);
1536099730d6SDave Hansen 	}
1537099730d6SDave Hansen 
1538099730d6SDave Hansen 	/*
1539099730d6SDave Hansen 	 * OK, so we have a VMA.  Fetch the mempolicy and try to
1540e0ec90eeSDave Hansen 	 * allocate a huge page with it.  We will only reach this
1541e0ec90eeSDave Hansen 	 * when CONFIG_NUMA=y.
1542099730d6SDave Hansen 	 */
1543099730d6SDave Hansen 	do {
1544099730d6SDave Hansen 		struct page *page;
1545099730d6SDave Hansen 		struct mempolicy *mpol;
1546099730d6SDave Hansen 		struct zonelist *zl;
1547099730d6SDave Hansen 		nodemask_t *nodemask;
1548099730d6SDave Hansen 
1549099730d6SDave Hansen 		cpuset_mems_cookie = read_mems_allowed_begin();
1550099730d6SDave Hansen 		zl = huge_zonelist(vma, addr, gfp, &mpol, &nodemask);
1551099730d6SDave Hansen 		mpol_cond_put(mpol);
1552099730d6SDave Hansen 		page = __alloc_pages_nodemask(gfp, order, zl, nodemask);
1553099730d6SDave Hansen 		if (page)
1554099730d6SDave Hansen 			return page;
1555099730d6SDave Hansen 	} while (read_mems_allowed_retry(cpuset_mems_cookie));
1556099730d6SDave Hansen 
1557099730d6SDave Hansen 	return NULL;
1558099730d6SDave Hansen }
1559099730d6SDave Hansen 
1560099730d6SDave Hansen /*
1561099730d6SDave Hansen  * There are two ways to allocate a huge page:
1562099730d6SDave Hansen  * 1. When you have a VMA and an address (like a fault)
1563099730d6SDave Hansen  * 2. When you have no VMA (like when setting /proc/.../nr_hugepages)
1564099730d6SDave Hansen  *
1565099730d6SDave Hansen  * 'vma' and 'addr' are only for (1).  'nid' is always NUMA_NO_NODE in
1566099730d6SDave Hansen  * this case which signifies that the allocation should be done with
1567099730d6SDave Hansen  * respect for the VMA's memory policy.
1568099730d6SDave Hansen  *
1569099730d6SDave Hansen  * For (2), we ignore 'vma' and 'addr' and use 'nid' exclusively. This
1570099730d6SDave Hansen  * implies that memory policies will not be taken in to account.
1571099730d6SDave Hansen  */
1572099730d6SDave Hansen static struct page *__alloc_buddy_huge_page(struct hstate *h,
1573099730d6SDave Hansen 		struct vm_area_struct *vma, unsigned long addr, int nid)
15747893d1d5SAdam Litke {
15757893d1d5SAdam Litke 	struct page *page;
1576bf50bab2SNaoya Horiguchi 	unsigned int r_nid;
15777893d1d5SAdam Litke 
1578bae7f4aeSLuiz Capitulino 	if (hstate_is_gigantic(h))
1579aa888a74SAndi Kleen 		return NULL;
1580aa888a74SAndi Kleen 
1581d1c3fb1fSNishanth Aravamudan 	/*
1582099730d6SDave Hansen 	 * Make sure that anyone specifying 'nid' is not also specifying a VMA.
1583099730d6SDave Hansen 	 * This makes sure the caller is picking _one_ of the modes with which
1584099730d6SDave Hansen 	 * we can call this function, not both.
1585099730d6SDave Hansen 	 */
1586099730d6SDave Hansen 	if (vma || (addr != -1)) {
1587e0ec90eeSDave Hansen 		VM_WARN_ON_ONCE(addr == -1);
1588e0ec90eeSDave Hansen 		VM_WARN_ON_ONCE(nid != NUMA_NO_NODE);
1589099730d6SDave Hansen 	}
1590099730d6SDave Hansen 	/*
1591d1c3fb1fSNishanth Aravamudan 	 * Assume we will successfully allocate the surplus page to
1592d1c3fb1fSNishanth Aravamudan 	 * prevent racing processes from causing the surplus to exceed
1593d1c3fb1fSNishanth Aravamudan 	 * overcommit
1594d1c3fb1fSNishanth Aravamudan 	 *
1595d1c3fb1fSNishanth Aravamudan 	 * This however introduces a different race, where a process B
1596d1c3fb1fSNishanth Aravamudan 	 * tries to grow the static hugepage pool while alloc_pages() is
1597d1c3fb1fSNishanth Aravamudan 	 * called by process A. B will only examine the per-node
1598d1c3fb1fSNishanth Aravamudan 	 * counters in determining if surplus huge pages can be
1599d1c3fb1fSNishanth Aravamudan 	 * converted to normal huge pages in adjust_pool_surplus(). A
1600d1c3fb1fSNishanth Aravamudan 	 * won't be able to increment the per-node counter, until the
1601d1c3fb1fSNishanth Aravamudan 	 * lock is dropped by B, but B doesn't drop hugetlb_lock until
1602d1c3fb1fSNishanth Aravamudan 	 * no more huge pages can be converted from surplus to normal
1603d1c3fb1fSNishanth Aravamudan 	 * state (and doesn't try to convert again). Thus, we have a
1604d1c3fb1fSNishanth Aravamudan 	 * case where a surplus huge page exists, the pool is grown, and
1605d1c3fb1fSNishanth Aravamudan 	 * the surplus huge page still exists after, even though it
1606d1c3fb1fSNishanth Aravamudan 	 * should just have been converted to a normal huge page. This
1607d1c3fb1fSNishanth Aravamudan 	 * does not leak memory, though, as the hugepage will be freed
1608d1c3fb1fSNishanth Aravamudan 	 * once it is out of use. It also does not allow the counters to
1609d1c3fb1fSNishanth Aravamudan 	 * go out of whack in adjust_pool_surplus() as we don't modify
1610d1c3fb1fSNishanth Aravamudan 	 * the node values until we've gotten the hugepage and only the
1611d1c3fb1fSNishanth Aravamudan 	 * per-node value is checked there.
1612d1c3fb1fSNishanth Aravamudan 	 */
1613d1c3fb1fSNishanth Aravamudan 	spin_lock(&hugetlb_lock);
1614a5516438SAndi Kleen 	if (h->surplus_huge_pages >= h->nr_overcommit_huge_pages) {
1615d1c3fb1fSNishanth Aravamudan 		spin_unlock(&hugetlb_lock);
1616d1c3fb1fSNishanth Aravamudan 		return NULL;
1617d1c3fb1fSNishanth Aravamudan 	} else {
1618a5516438SAndi Kleen 		h->nr_huge_pages++;
1619a5516438SAndi Kleen 		h->surplus_huge_pages++;
1620d1c3fb1fSNishanth Aravamudan 	}
1621d1c3fb1fSNishanth Aravamudan 	spin_unlock(&hugetlb_lock);
1622d1c3fb1fSNishanth Aravamudan 
1623099730d6SDave Hansen 	page = __hugetlb_alloc_buddy_huge_page(h, vma, addr, nid);
1624d1c3fb1fSNishanth Aravamudan 
16257893d1d5SAdam Litke 	spin_lock(&hugetlb_lock);
1626d1c3fb1fSNishanth Aravamudan 	if (page) {
16270edaecfaSAneesh Kumar K.V 		INIT_LIST_HEAD(&page->lru);
1628bf50bab2SNaoya Horiguchi 		r_nid = page_to_nid(page);
1629f1e61557SKirill A. Shutemov 		set_compound_page_dtor(page, HUGETLB_PAGE_DTOR);
16309dd540e2SAneesh Kumar K.V 		set_hugetlb_cgroup(page, NULL);
1631d1c3fb1fSNishanth Aravamudan 		/*
1632d1c3fb1fSNishanth Aravamudan 		 * We incremented the global counters already
1633d1c3fb1fSNishanth Aravamudan 		 */
1634bf50bab2SNaoya Horiguchi 		h->nr_huge_pages_node[r_nid]++;
1635bf50bab2SNaoya Horiguchi 		h->surplus_huge_pages_node[r_nid]++;
16363b116300SAdam Litke 		__count_vm_event(HTLB_BUDDY_PGALLOC);
1637d1c3fb1fSNishanth Aravamudan 	} else {
1638a5516438SAndi Kleen 		h->nr_huge_pages--;
1639a5516438SAndi Kleen 		h->surplus_huge_pages--;
16403b116300SAdam Litke 		__count_vm_event(HTLB_BUDDY_PGALLOC_FAIL);
16417893d1d5SAdam Litke 	}
1642d1c3fb1fSNishanth Aravamudan 	spin_unlock(&hugetlb_lock);
16437893d1d5SAdam Litke 
16447893d1d5SAdam Litke 	return page;
16457893d1d5SAdam Litke }
16467893d1d5SAdam Litke 
1647e4e574b7SAdam Litke /*
1648099730d6SDave Hansen  * Allocate a huge page from 'nid'.  Note, 'nid' may be
1649099730d6SDave Hansen  * NUMA_NO_NODE, which means that it may be allocated
1650099730d6SDave Hansen  * anywhere.
1651099730d6SDave Hansen  */
1652e0ec90eeSDave Hansen static
1653099730d6SDave Hansen struct page *__alloc_buddy_huge_page_no_mpol(struct hstate *h, int nid)
1654099730d6SDave Hansen {
1655099730d6SDave Hansen 	unsigned long addr = -1;
1656099730d6SDave Hansen 
1657099730d6SDave Hansen 	return __alloc_buddy_huge_page(h, NULL, addr, nid);
1658099730d6SDave Hansen }
1659099730d6SDave Hansen 
1660099730d6SDave Hansen /*
1661099730d6SDave Hansen  * Use the VMA's mpolicy to allocate a huge page from the buddy.
1662099730d6SDave Hansen  */
1663e0ec90eeSDave Hansen static
1664099730d6SDave Hansen struct page *__alloc_buddy_huge_page_with_mpol(struct hstate *h,
1665099730d6SDave Hansen 		struct vm_area_struct *vma, unsigned long addr)
1666099730d6SDave Hansen {
1667099730d6SDave Hansen 	return __alloc_buddy_huge_page(h, vma, addr, NUMA_NO_NODE);
1668099730d6SDave Hansen }
1669099730d6SDave Hansen 
1670099730d6SDave Hansen /*
1671bf50bab2SNaoya Horiguchi  * This allocation function is useful in the context where vma is irrelevant.
1672bf50bab2SNaoya Horiguchi  * E.g. soft-offlining uses this function because it only cares physical
1673bf50bab2SNaoya Horiguchi  * address of error page.
1674bf50bab2SNaoya Horiguchi  */
1675bf50bab2SNaoya Horiguchi struct page *alloc_huge_page_node(struct hstate *h, int nid)
1676bf50bab2SNaoya Horiguchi {
16774ef91848SJoonsoo Kim 	struct page *page = NULL;
1678bf50bab2SNaoya Horiguchi 
1679bf50bab2SNaoya Horiguchi 	spin_lock(&hugetlb_lock);
16804ef91848SJoonsoo Kim 	if (h->free_huge_pages - h->resv_huge_pages > 0)
1681bf50bab2SNaoya Horiguchi 		page = dequeue_huge_page_node(h, nid);
1682bf50bab2SNaoya Horiguchi 	spin_unlock(&hugetlb_lock);
1683bf50bab2SNaoya Horiguchi 
168494ae8ba7SAneesh Kumar K.V 	if (!page)
1685099730d6SDave Hansen 		page = __alloc_buddy_huge_page_no_mpol(h, nid);
1686bf50bab2SNaoya Horiguchi 
1687bf50bab2SNaoya Horiguchi 	return page;
1688bf50bab2SNaoya Horiguchi }
1689bf50bab2SNaoya Horiguchi 
1690bf50bab2SNaoya Horiguchi /*
169125985edcSLucas De Marchi  * Increase the hugetlb pool such that it can accommodate a reservation
1692e4e574b7SAdam Litke  * of size 'delta'.
1693e4e574b7SAdam Litke  */
1694a5516438SAndi Kleen static int gather_surplus_pages(struct hstate *h, int delta)
1695e4e574b7SAdam Litke {
1696e4e574b7SAdam Litke 	struct list_head surplus_list;
1697e4e574b7SAdam Litke 	struct page *page, *tmp;
1698e4e574b7SAdam Litke 	int ret, i;
1699e4e574b7SAdam Litke 	int needed, allocated;
170028073b02SHillf Danton 	bool alloc_ok = true;
1701e4e574b7SAdam Litke 
1702a5516438SAndi Kleen 	needed = (h->resv_huge_pages + delta) - h->free_huge_pages;
1703ac09b3a1SAdam Litke 	if (needed <= 0) {
1704a5516438SAndi Kleen 		h->resv_huge_pages += delta;
1705e4e574b7SAdam Litke 		return 0;
1706ac09b3a1SAdam Litke 	}
1707e4e574b7SAdam Litke 
1708e4e574b7SAdam Litke 	allocated = 0;
1709e4e574b7SAdam Litke 	INIT_LIST_HEAD(&surplus_list);
1710e4e574b7SAdam Litke 
1711e4e574b7SAdam Litke 	ret = -ENOMEM;
1712e4e574b7SAdam Litke retry:
1713e4e574b7SAdam Litke 	spin_unlock(&hugetlb_lock);
1714e4e574b7SAdam Litke 	for (i = 0; i < needed; i++) {
1715099730d6SDave Hansen 		page = __alloc_buddy_huge_page_no_mpol(h, NUMA_NO_NODE);
171628073b02SHillf Danton 		if (!page) {
171728073b02SHillf Danton 			alloc_ok = false;
171828073b02SHillf Danton 			break;
171928073b02SHillf Danton 		}
1720e4e574b7SAdam Litke 		list_add(&page->lru, &surplus_list);
1721e4e574b7SAdam Litke 	}
172228073b02SHillf Danton 	allocated += i;
1723e4e574b7SAdam Litke 
1724e4e574b7SAdam Litke 	/*
1725e4e574b7SAdam Litke 	 * After retaking hugetlb_lock, we need to recalculate 'needed'
1726e4e574b7SAdam Litke 	 * because either resv_huge_pages or free_huge_pages may have changed.
1727e4e574b7SAdam Litke 	 */
1728e4e574b7SAdam Litke 	spin_lock(&hugetlb_lock);
1729a5516438SAndi Kleen 	needed = (h->resv_huge_pages + delta) -
1730a5516438SAndi Kleen 			(h->free_huge_pages + allocated);
173128073b02SHillf Danton 	if (needed > 0) {
173228073b02SHillf Danton 		if (alloc_ok)
1733e4e574b7SAdam Litke 			goto retry;
173428073b02SHillf Danton 		/*
173528073b02SHillf Danton 		 * We were not able to allocate enough pages to
173628073b02SHillf Danton 		 * satisfy the entire reservation so we free what
173728073b02SHillf Danton 		 * we've allocated so far.
173828073b02SHillf Danton 		 */
173928073b02SHillf Danton 		goto free;
174028073b02SHillf Danton 	}
1741e4e574b7SAdam Litke 	/*
1742e4e574b7SAdam Litke 	 * The surplus_list now contains _at_least_ the number of extra pages
174325985edcSLucas De Marchi 	 * needed to accommodate the reservation.  Add the appropriate number
1744e4e574b7SAdam Litke 	 * of pages to the hugetlb pool and free the extras back to the buddy
1745ac09b3a1SAdam Litke 	 * allocator.  Commit the entire reservation here to prevent another
1746ac09b3a1SAdam Litke 	 * process from stealing the pages as they are added to the pool but
1747ac09b3a1SAdam Litke 	 * before they are reserved.
1748e4e574b7SAdam Litke 	 */
1749e4e574b7SAdam Litke 	needed += allocated;
1750a5516438SAndi Kleen 	h->resv_huge_pages += delta;
1751e4e574b7SAdam Litke 	ret = 0;
1752a9869b83SNaoya Horiguchi 
175319fc3f0aSAdam Litke 	/* Free the needed pages to the hugetlb pool */
175419fc3f0aSAdam Litke 	list_for_each_entry_safe(page, tmp, &surplus_list, lru) {
175519fc3f0aSAdam Litke 		if ((--needed) < 0)
175619fc3f0aSAdam Litke 			break;
1757a9869b83SNaoya Horiguchi 		/*
1758a9869b83SNaoya Horiguchi 		 * This page is now managed by the hugetlb allocator and has
1759a9869b83SNaoya Horiguchi 		 * no users -- drop the buddy allocator's reference.
1760a9869b83SNaoya Horiguchi 		 */
1761a9869b83SNaoya Horiguchi 		put_page_testzero(page);
1762309381feSSasha Levin 		VM_BUG_ON_PAGE(page_count(page), page);
1763a5516438SAndi Kleen 		enqueue_huge_page(h, page);
176419fc3f0aSAdam Litke 	}
176528073b02SHillf Danton free:
1766b0365c8dSHillf Danton 	spin_unlock(&hugetlb_lock);
176719fc3f0aSAdam Litke 
176819fc3f0aSAdam Litke 	/* Free unnecessary surplus pages to the buddy allocator */
1769c0d934baSJoonsoo Kim 	list_for_each_entry_safe(page, tmp, &surplus_list, lru)
1770a9869b83SNaoya Horiguchi 		put_page(page);
177119fc3f0aSAdam Litke 	spin_lock(&hugetlb_lock);
1772e4e574b7SAdam Litke 
1773e4e574b7SAdam Litke 	return ret;
1774e4e574b7SAdam Litke }
1775e4e574b7SAdam Litke 
1776e4e574b7SAdam Litke /*
1777e5bbc8a6SMike Kravetz  * This routine has two main purposes:
1778e5bbc8a6SMike Kravetz  * 1) Decrement the reservation count (resv_huge_pages) by the value passed
1779e5bbc8a6SMike Kravetz  *    in unused_resv_pages.  This corresponds to the prior adjustments made
1780e5bbc8a6SMike Kravetz  *    to the associated reservation map.
1781e5bbc8a6SMike Kravetz  * 2) Free any unused surplus pages that may have been allocated to satisfy
1782e5bbc8a6SMike Kravetz  *    the reservation.  As many as unused_resv_pages may be freed.
1783e5bbc8a6SMike Kravetz  *
1784e5bbc8a6SMike Kravetz  * Called with hugetlb_lock held.  However, the lock could be dropped (and
1785e5bbc8a6SMike Kravetz  * reacquired) during calls to cond_resched_lock.  Whenever dropping the lock,
1786e5bbc8a6SMike Kravetz  * we must make sure nobody else can claim pages we are in the process of
1787e5bbc8a6SMike Kravetz  * freeing.  Do this by ensuring resv_huge_page always is greater than the
1788e5bbc8a6SMike Kravetz  * number of huge pages we plan to free when dropping the lock.
1789e4e574b7SAdam Litke  */
1790a5516438SAndi Kleen static void return_unused_surplus_pages(struct hstate *h,
1791a5516438SAndi Kleen 					unsigned long unused_resv_pages)
1792e4e574b7SAdam Litke {
1793e4e574b7SAdam Litke 	unsigned long nr_pages;
1794e4e574b7SAdam Litke 
1795aa888a74SAndi Kleen 	/* Cannot return gigantic pages currently */
1796bae7f4aeSLuiz Capitulino 	if (hstate_is_gigantic(h))
1797e5bbc8a6SMike Kravetz 		goto out;
1798aa888a74SAndi Kleen 
1799e5bbc8a6SMike Kravetz 	/*
1800e5bbc8a6SMike Kravetz 	 * Part (or even all) of the reservation could have been backed
1801e5bbc8a6SMike Kravetz 	 * by pre-allocated pages. Only free surplus pages.
1802e5bbc8a6SMike Kravetz 	 */
1803a5516438SAndi Kleen 	nr_pages = min(unused_resv_pages, h->surplus_huge_pages);
1804e4e574b7SAdam Litke 
1805685f3457SLee Schermerhorn 	/*
1806685f3457SLee Schermerhorn 	 * We want to release as many surplus pages as possible, spread
18079b5e5d0fSLee Schermerhorn 	 * evenly across all nodes with memory. Iterate across these nodes
18089b5e5d0fSLee Schermerhorn 	 * until we can no longer free unreserved surplus pages. This occurs
18099b5e5d0fSLee Schermerhorn 	 * when the nodes with surplus pages have no free pages.
18109b5e5d0fSLee Schermerhorn 	 * free_pool_huge_page() will balance the the freed pages across the
18119b5e5d0fSLee Schermerhorn 	 * on-line nodes with memory and will handle the hstate accounting.
1812e5bbc8a6SMike Kravetz 	 *
1813e5bbc8a6SMike Kravetz 	 * Note that we decrement resv_huge_pages as we free the pages.  If
1814e5bbc8a6SMike Kravetz 	 * we drop the lock, resv_huge_pages will still be sufficiently large
1815e5bbc8a6SMike Kravetz 	 * to cover subsequent pages we may free.
1816685f3457SLee Schermerhorn 	 */
1817685f3457SLee Schermerhorn 	while (nr_pages--) {
1818e5bbc8a6SMike Kravetz 		h->resv_huge_pages--;
1819e5bbc8a6SMike Kravetz 		unused_resv_pages--;
18208cebfcd0SLai Jiangshan 		if (!free_pool_huge_page(h, &node_states[N_MEMORY], 1))
1821e5bbc8a6SMike Kravetz 			goto out;
18227848a4bfSMizuma, Masayoshi 		cond_resched_lock(&hugetlb_lock);
1823e4e574b7SAdam Litke 	}
1824e5bbc8a6SMike Kravetz 
1825e5bbc8a6SMike Kravetz out:
1826e5bbc8a6SMike Kravetz 	/* Fully uncommit the reservation */
1827e5bbc8a6SMike Kravetz 	h->resv_huge_pages -= unused_resv_pages;
1828e4e574b7SAdam Litke }
1829e4e574b7SAdam Litke 
18305e911373SMike Kravetz 
1831c37f9fb1SAndy Whitcroft /*
1832feba16e2SMike Kravetz  * vma_needs_reservation, vma_commit_reservation and vma_end_reservation
18335e911373SMike Kravetz  * are used by the huge page allocation routines to manage reservations.
1834cf3ad20bSMike Kravetz  *
1835cf3ad20bSMike Kravetz  * vma_needs_reservation is called to determine if the huge page at addr
1836cf3ad20bSMike Kravetz  * within the vma has an associated reservation.  If a reservation is
1837cf3ad20bSMike Kravetz  * needed, the value 1 is returned.  The caller is then responsible for
1838cf3ad20bSMike Kravetz  * managing the global reservation and subpool usage counts.  After
1839cf3ad20bSMike Kravetz  * the huge page has been allocated, vma_commit_reservation is called
1840feba16e2SMike Kravetz  * to add the page to the reservation map.  If the page allocation fails,
1841feba16e2SMike Kravetz  * the reservation must be ended instead of committed.  vma_end_reservation
1842feba16e2SMike Kravetz  * is called in such cases.
1843cf3ad20bSMike Kravetz  *
1844cf3ad20bSMike Kravetz  * In the normal case, vma_commit_reservation returns the same value
1845cf3ad20bSMike Kravetz  * as the preceding vma_needs_reservation call.  The only time this
1846cf3ad20bSMike Kravetz  * is not the case is if a reserve map was changed between calls.  It
1847cf3ad20bSMike Kravetz  * is the responsibility of the caller to notice the difference and
1848cf3ad20bSMike Kravetz  * take appropriate action.
184996b96a96SMike Kravetz  *
185096b96a96SMike Kravetz  * vma_add_reservation is used in error paths where a reservation must
185196b96a96SMike Kravetz  * be restored when a newly allocated huge page must be freed.  It is
185296b96a96SMike Kravetz  * to be called after calling vma_needs_reservation to determine if a
185396b96a96SMike Kravetz  * reservation exists.
1854c37f9fb1SAndy Whitcroft  */
18555e911373SMike Kravetz enum vma_resv_mode {
18565e911373SMike Kravetz 	VMA_NEEDS_RESV,
18575e911373SMike Kravetz 	VMA_COMMIT_RESV,
1858feba16e2SMike Kravetz 	VMA_END_RESV,
185996b96a96SMike Kravetz 	VMA_ADD_RESV,
18605e911373SMike Kravetz };
1861cf3ad20bSMike Kravetz static long __vma_reservation_common(struct hstate *h,
1862cf3ad20bSMike Kravetz 				struct vm_area_struct *vma, unsigned long addr,
18635e911373SMike Kravetz 				enum vma_resv_mode mode)
1864c37f9fb1SAndy Whitcroft {
18654e35f483SJoonsoo Kim 	struct resv_map *resv;
18664e35f483SJoonsoo Kim 	pgoff_t idx;
1867cf3ad20bSMike Kravetz 	long ret;
1868c37f9fb1SAndy Whitcroft 
18694e35f483SJoonsoo Kim 	resv = vma_resv_map(vma);
18704e35f483SJoonsoo Kim 	if (!resv)
1871c37f9fb1SAndy Whitcroft 		return 1;
1872c37f9fb1SAndy Whitcroft 
18734e35f483SJoonsoo Kim 	idx = vma_hugecache_offset(h, vma, addr);
18745e911373SMike Kravetz 	switch (mode) {
18755e911373SMike Kravetz 	case VMA_NEEDS_RESV:
1876cf3ad20bSMike Kravetz 		ret = region_chg(resv, idx, idx + 1);
18775e911373SMike Kravetz 		break;
18785e911373SMike Kravetz 	case VMA_COMMIT_RESV:
18795e911373SMike Kravetz 		ret = region_add(resv, idx, idx + 1);
18805e911373SMike Kravetz 		break;
1881feba16e2SMike Kravetz 	case VMA_END_RESV:
18825e911373SMike Kravetz 		region_abort(resv, idx, idx + 1);
18835e911373SMike Kravetz 		ret = 0;
18845e911373SMike Kravetz 		break;
188596b96a96SMike Kravetz 	case VMA_ADD_RESV:
188696b96a96SMike Kravetz 		if (vma->vm_flags & VM_MAYSHARE)
188796b96a96SMike Kravetz 			ret = region_add(resv, idx, idx + 1);
188896b96a96SMike Kravetz 		else {
188996b96a96SMike Kravetz 			region_abort(resv, idx, idx + 1);
189096b96a96SMike Kravetz 			ret = region_del(resv, idx, idx + 1);
189196b96a96SMike Kravetz 		}
189296b96a96SMike Kravetz 		break;
18935e911373SMike Kravetz 	default:
18945e911373SMike Kravetz 		BUG();
18955e911373SMike Kravetz 	}
189684afd99bSAndy Whitcroft 
18974e35f483SJoonsoo Kim 	if (vma->vm_flags & VM_MAYSHARE)
1898cf3ad20bSMike Kravetz 		return ret;
189967961f9dSMike Kravetz 	else if (is_vma_resv_set(vma, HPAGE_RESV_OWNER) && ret >= 0) {
190067961f9dSMike Kravetz 		/*
190167961f9dSMike Kravetz 		 * In most cases, reserves always exist for private mappings.
190267961f9dSMike Kravetz 		 * However, a file associated with mapping could have been
190367961f9dSMike Kravetz 		 * hole punched or truncated after reserves were consumed.
190467961f9dSMike Kravetz 		 * As subsequent fault on such a range will not use reserves.
190567961f9dSMike Kravetz 		 * Subtle - The reserve map for private mappings has the
190667961f9dSMike Kravetz 		 * opposite meaning than that of shared mappings.  If NO
190767961f9dSMike Kravetz 		 * entry is in the reserve map, it means a reservation exists.
190867961f9dSMike Kravetz 		 * If an entry exists in the reserve map, it means the
190967961f9dSMike Kravetz 		 * reservation has already been consumed.  As a result, the
191067961f9dSMike Kravetz 		 * return value of this routine is the opposite of the
191167961f9dSMike Kravetz 		 * value returned from reserve map manipulation routines above.
191267961f9dSMike Kravetz 		 */
191367961f9dSMike Kravetz 		if (ret)
191467961f9dSMike Kravetz 			return 0;
191567961f9dSMike Kravetz 		else
191667961f9dSMike Kravetz 			return 1;
191767961f9dSMike Kravetz 	}
19184e35f483SJoonsoo Kim 	else
1919cf3ad20bSMike Kravetz 		return ret < 0 ? ret : 0;
192084afd99bSAndy Whitcroft }
1921cf3ad20bSMike Kravetz 
1922cf3ad20bSMike Kravetz static long vma_needs_reservation(struct hstate *h,
1923a5516438SAndi Kleen 			struct vm_area_struct *vma, unsigned long addr)
1924c37f9fb1SAndy Whitcroft {
19255e911373SMike Kravetz 	return __vma_reservation_common(h, vma, addr, VMA_NEEDS_RESV);
1926cf3ad20bSMike Kravetz }
1927c37f9fb1SAndy Whitcroft 
1928cf3ad20bSMike Kravetz static long vma_commit_reservation(struct hstate *h,
1929cf3ad20bSMike Kravetz 			struct vm_area_struct *vma, unsigned long addr)
1930cf3ad20bSMike Kravetz {
19315e911373SMike Kravetz 	return __vma_reservation_common(h, vma, addr, VMA_COMMIT_RESV);
19325e911373SMike Kravetz }
19335e911373SMike Kravetz 
1934feba16e2SMike Kravetz static void vma_end_reservation(struct hstate *h,
19355e911373SMike Kravetz 			struct vm_area_struct *vma, unsigned long addr)
19365e911373SMike Kravetz {
1937feba16e2SMike Kravetz 	(void)__vma_reservation_common(h, vma, addr, VMA_END_RESV);
1938c37f9fb1SAndy Whitcroft }
1939c37f9fb1SAndy Whitcroft 
194096b96a96SMike Kravetz static long vma_add_reservation(struct hstate *h,
194196b96a96SMike Kravetz 			struct vm_area_struct *vma, unsigned long addr)
194296b96a96SMike Kravetz {
194396b96a96SMike Kravetz 	return __vma_reservation_common(h, vma, addr, VMA_ADD_RESV);
194496b96a96SMike Kravetz }
194596b96a96SMike Kravetz 
194696b96a96SMike Kravetz /*
194796b96a96SMike Kravetz  * This routine is called to restore a reservation on error paths.  In the
194896b96a96SMike Kravetz  * specific error paths, a huge page was allocated (via alloc_huge_page)
194996b96a96SMike Kravetz  * and is about to be freed.  If a reservation for the page existed,
195096b96a96SMike Kravetz  * alloc_huge_page would have consumed the reservation and set PagePrivate
195196b96a96SMike Kravetz  * in the newly allocated page.  When the page is freed via free_huge_page,
195296b96a96SMike Kravetz  * the global reservation count will be incremented if PagePrivate is set.
195396b96a96SMike Kravetz  * However, free_huge_page can not adjust the reserve map.  Adjust the
195496b96a96SMike Kravetz  * reserve map here to be consistent with global reserve count adjustments
195596b96a96SMike Kravetz  * to be made by free_huge_page.
195696b96a96SMike Kravetz  */
195796b96a96SMike Kravetz static void restore_reserve_on_error(struct hstate *h,
195896b96a96SMike Kravetz 			struct vm_area_struct *vma, unsigned long address,
195996b96a96SMike Kravetz 			struct page *page)
196096b96a96SMike Kravetz {
196196b96a96SMike Kravetz 	if (unlikely(PagePrivate(page))) {
196296b96a96SMike Kravetz 		long rc = vma_needs_reservation(h, vma, address);
196396b96a96SMike Kravetz 
196496b96a96SMike Kravetz 		if (unlikely(rc < 0)) {
196596b96a96SMike Kravetz 			/*
196696b96a96SMike Kravetz 			 * Rare out of memory condition in reserve map
196796b96a96SMike Kravetz 			 * manipulation.  Clear PagePrivate so that
196896b96a96SMike Kravetz 			 * global reserve count will not be incremented
196996b96a96SMike Kravetz 			 * by free_huge_page.  This will make it appear
197096b96a96SMike Kravetz 			 * as though the reservation for this page was
197196b96a96SMike Kravetz 			 * consumed.  This may prevent the task from
197296b96a96SMike Kravetz 			 * faulting in the page at a later time.  This
197396b96a96SMike Kravetz 			 * is better than inconsistent global huge page
197496b96a96SMike Kravetz 			 * accounting of reserve counts.
197596b96a96SMike Kravetz 			 */
197696b96a96SMike Kravetz 			ClearPagePrivate(page);
197796b96a96SMike Kravetz 		} else if (rc) {
197896b96a96SMike Kravetz 			rc = vma_add_reservation(h, vma, address);
197996b96a96SMike Kravetz 			if (unlikely(rc < 0))
198096b96a96SMike Kravetz 				/*
198196b96a96SMike Kravetz 				 * See above comment about rare out of
198296b96a96SMike Kravetz 				 * memory condition.
198396b96a96SMike Kravetz 				 */
198496b96a96SMike Kravetz 				ClearPagePrivate(page);
198596b96a96SMike Kravetz 		} else
198696b96a96SMike Kravetz 			vma_end_reservation(h, vma, address);
198796b96a96SMike Kravetz 	}
198896b96a96SMike Kravetz }
198996b96a96SMike Kravetz 
199070c3547eSMike Kravetz struct page *alloc_huge_page(struct vm_area_struct *vma,
199104f2cbe3SMel Gorman 				    unsigned long addr, int avoid_reserve)
1992348ea204SAdam Litke {
199390481622SDavid Gibson 	struct hugepage_subpool *spool = subpool_vma(vma);
1994a5516438SAndi Kleen 	struct hstate *h = hstate_vma(vma);
1995348ea204SAdam Litke 	struct page *page;
1996d85f69b0SMike Kravetz 	long map_chg, map_commit;
1997d85f69b0SMike Kravetz 	long gbl_chg;
19986d76dcf4SAneesh Kumar K.V 	int ret, idx;
19996d76dcf4SAneesh Kumar K.V 	struct hugetlb_cgroup *h_cg;
20002fc39cecSAdam Litke 
20016d76dcf4SAneesh Kumar K.V 	idx = hstate_index(h);
2002a1e78772SMel Gorman 	/*
2003d85f69b0SMike Kravetz 	 * Examine the region/reserve map to determine if the process
2004d85f69b0SMike Kravetz 	 * has a reservation for the page to be allocated.  A return
2005d85f69b0SMike Kravetz 	 * code of zero indicates a reservation exists (no change).
2006a1e78772SMel Gorman 	 */
2007d85f69b0SMike Kravetz 	map_chg = gbl_chg = vma_needs_reservation(h, vma, addr);
2008d85f69b0SMike Kravetz 	if (map_chg < 0)
200976dcee75SAneesh Kumar K.V 		return ERR_PTR(-ENOMEM);
2010d85f69b0SMike Kravetz 
2011d85f69b0SMike Kravetz 	/*
2012d85f69b0SMike Kravetz 	 * Processes that did not create the mapping will have no
2013d85f69b0SMike Kravetz 	 * reserves as indicated by the region/reserve map. Check
2014d85f69b0SMike Kravetz 	 * that the allocation will not exceed the subpool limit.
2015d85f69b0SMike Kravetz 	 * Allocations for MAP_NORESERVE mappings also need to be
2016d85f69b0SMike Kravetz 	 * checked against any subpool limit.
2017d85f69b0SMike Kravetz 	 */
2018d85f69b0SMike Kravetz 	if (map_chg || avoid_reserve) {
2019d85f69b0SMike Kravetz 		gbl_chg = hugepage_subpool_get_pages(spool, 1);
2020d85f69b0SMike Kravetz 		if (gbl_chg < 0) {
2021feba16e2SMike Kravetz 			vma_end_reservation(h, vma, addr);
202276dcee75SAneesh Kumar K.V 			return ERR_PTR(-ENOSPC);
20235e911373SMike Kravetz 		}
202490d8b7e6SAdam Litke 
2025d85f69b0SMike Kravetz 		/*
2026d85f69b0SMike Kravetz 		 * Even though there was no reservation in the region/reserve
2027d85f69b0SMike Kravetz 		 * map, there could be reservations associated with the
2028d85f69b0SMike Kravetz 		 * subpool that can be used.  This would be indicated if the
2029d85f69b0SMike Kravetz 		 * return value of hugepage_subpool_get_pages() is zero.
2030d85f69b0SMike Kravetz 		 * However, if avoid_reserve is specified we still avoid even
2031d85f69b0SMike Kravetz 		 * the subpool reservations.
2032d85f69b0SMike Kravetz 		 */
2033d85f69b0SMike Kravetz 		if (avoid_reserve)
2034d85f69b0SMike Kravetz 			gbl_chg = 1;
2035d85f69b0SMike Kravetz 	}
2036d85f69b0SMike Kravetz 
20376d76dcf4SAneesh Kumar K.V 	ret = hugetlb_cgroup_charge_cgroup(idx, pages_per_huge_page(h), &h_cg);
20388f34af6fSJianyu Zhan 	if (ret)
20398f34af6fSJianyu Zhan 		goto out_subpool_put;
20408f34af6fSJianyu Zhan 
2041a1e78772SMel Gorman 	spin_lock(&hugetlb_lock);
2042d85f69b0SMike Kravetz 	/*
2043d85f69b0SMike Kravetz 	 * glb_chg is passed to indicate whether or not a page must be taken
2044d85f69b0SMike Kravetz 	 * from the global free pool (global change).  gbl_chg == 0 indicates
2045d85f69b0SMike Kravetz 	 * a reservation exists for the allocation.
2046d85f69b0SMike Kravetz 	 */
2047d85f69b0SMike Kravetz 	page = dequeue_huge_page_vma(h, vma, addr, avoid_reserve, gbl_chg);
204881a6fcaeSJoonsoo Kim 	if (!page) {
204994ae8ba7SAneesh Kumar K.V 		spin_unlock(&hugetlb_lock);
2050099730d6SDave Hansen 		page = __alloc_buddy_huge_page_with_mpol(h, vma, addr);
20518f34af6fSJianyu Zhan 		if (!page)
20528f34af6fSJianyu Zhan 			goto out_uncharge_cgroup;
2053a88c7695SNaoya Horiguchi 		if (!avoid_reserve && vma_has_reserves(vma, gbl_chg)) {
2054a88c7695SNaoya Horiguchi 			SetPagePrivate(page);
2055a88c7695SNaoya Horiguchi 			h->resv_huge_pages--;
2056a88c7695SNaoya Horiguchi 		}
205779dbb236SAneesh Kumar K.V 		spin_lock(&hugetlb_lock);
205879dbb236SAneesh Kumar K.V 		list_move(&page->lru, &h->hugepage_activelist);
205981a6fcaeSJoonsoo Kim 		/* Fall through */
2060a1e78772SMel Gorman 	}
206181a6fcaeSJoonsoo Kim 	hugetlb_cgroup_commit_charge(idx, pages_per_huge_page(h), h_cg, page);
206281a6fcaeSJoonsoo Kim 	spin_unlock(&hugetlb_lock);
2063a1e78772SMel Gorman 
206490481622SDavid Gibson 	set_page_private(page, (unsigned long)spool);
2065a1e78772SMel Gorman 
2066d85f69b0SMike Kravetz 	map_commit = vma_commit_reservation(h, vma, addr);
2067d85f69b0SMike Kravetz 	if (unlikely(map_chg > map_commit)) {
206833039678SMike Kravetz 		/*
206933039678SMike Kravetz 		 * The page was added to the reservation map between
207033039678SMike Kravetz 		 * vma_needs_reservation and vma_commit_reservation.
207133039678SMike Kravetz 		 * This indicates a race with hugetlb_reserve_pages.
207233039678SMike Kravetz 		 * Adjust for the subpool count incremented above AND
207333039678SMike Kravetz 		 * in hugetlb_reserve_pages for the same page.  Also,
207433039678SMike Kravetz 		 * the reservation count added in hugetlb_reserve_pages
207533039678SMike Kravetz 		 * no longer applies.
207633039678SMike Kravetz 		 */
207733039678SMike Kravetz 		long rsv_adjust;
207833039678SMike Kravetz 
207933039678SMike Kravetz 		rsv_adjust = hugepage_subpool_put_pages(spool, 1);
208033039678SMike Kravetz 		hugetlb_acct_memory(h, -rsv_adjust);
208133039678SMike Kravetz 	}
20827893d1d5SAdam Litke 	return page;
20838f34af6fSJianyu Zhan 
20848f34af6fSJianyu Zhan out_uncharge_cgroup:
20858f34af6fSJianyu Zhan 	hugetlb_cgroup_uncharge_cgroup(idx, pages_per_huge_page(h), h_cg);
20868f34af6fSJianyu Zhan out_subpool_put:
2087d85f69b0SMike Kravetz 	if (map_chg || avoid_reserve)
20888f34af6fSJianyu Zhan 		hugepage_subpool_put_pages(spool, 1);
2089feba16e2SMike Kravetz 	vma_end_reservation(h, vma, addr);
20908f34af6fSJianyu Zhan 	return ERR_PTR(-ENOSPC);
2091b45b5bd6SDavid Gibson }
2092b45b5bd6SDavid Gibson 
209374060e4dSNaoya Horiguchi /*
209474060e4dSNaoya Horiguchi  * alloc_huge_page()'s wrapper which simply returns the page if allocation
209574060e4dSNaoya Horiguchi  * succeeds, otherwise NULL. This function is called from new_vma_page(),
209674060e4dSNaoya Horiguchi  * where no ERR_VALUE is expected to be returned.
209774060e4dSNaoya Horiguchi  */
209874060e4dSNaoya Horiguchi struct page *alloc_huge_page_noerr(struct vm_area_struct *vma,
209974060e4dSNaoya Horiguchi 				unsigned long addr, int avoid_reserve)
210074060e4dSNaoya Horiguchi {
210174060e4dSNaoya Horiguchi 	struct page *page = alloc_huge_page(vma, addr, avoid_reserve);
210274060e4dSNaoya Horiguchi 	if (IS_ERR(page))
210374060e4dSNaoya Horiguchi 		page = NULL;
210474060e4dSNaoya Horiguchi 	return page;
210574060e4dSNaoya Horiguchi }
210674060e4dSNaoya Horiguchi 
210791f47662SCyrill Gorcunov int __weak alloc_bootmem_huge_page(struct hstate *h)
2108aa888a74SAndi Kleen {
2109aa888a74SAndi Kleen 	struct huge_bootmem_page *m;
2110b2261026SJoonsoo Kim 	int nr_nodes, node;
2111aa888a74SAndi Kleen 
2112b2261026SJoonsoo Kim 	for_each_node_mask_to_alloc(h, nr_nodes, node, &node_states[N_MEMORY]) {
2113aa888a74SAndi Kleen 		void *addr;
2114aa888a74SAndi Kleen 
21158b89a116SGrygorii Strashko 		addr = memblock_virt_alloc_try_nid_nopanic(
21168b89a116SGrygorii Strashko 				huge_page_size(h), huge_page_size(h),
21178b89a116SGrygorii Strashko 				0, BOOTMEM_ALLOC_ACCESSIBLE, node);
2118aa888a74SAndi Kleen 		if (addr) {
2119aa888a74SAndi Kleen 			/*
2120aa888a74SAndi Kleen 			 * Use the beginning of the huge page to store the
2121aa888a74SAndi Kleen 			 * huge_bootmem_page struct (until gather_bootmem
2122aa888a74SAndi Kleen 			 * puts them into the mem_map).
2123aa888a74SAndi Kleen 			 */
2124aa888a74SAndi Kleen 			m = addr;
2125aa888a74SAndi Kleen 			goto found;
2126aa888a74SAndi Kleen 		}
2127aa888a74SAndi Kleen 	}
2128aa888a74SAndi Kleen 	return 0;
2129aa888a74SAndi Kleen 
2130aa888a74SAndi Kleen found:
2131df994eadSLuiz Capitulino 	BUG_ON(!IS_ALIGNED(virt_to_phys(m), huge_page_size(h)));
2132aa888a74SAndi Kleen 	/* Put them into a private list first because mem_map is not up yet */
2133aa888a74SAndi Kleen 	list_add(&m->list, &huge_boot_pages);
2134aa888a74SAndi Kleen 	m->hstate = h;
2135aa888a74SAndi Kleen 	return 1;
2136aa888a74SAndi Kleen }
2137aa888a74SAndi Kleen 
2138d00181b9SKirill A. Shutemov static void __init prep_compound_huge_page(struct page *page,
2139d00181b9SKirill A. Shutemov 		unsigned int order)
214018229df5SAndy Whitcroft {
214118229df5SAndy Whitcroft 	if (unlikely(order > (MAX_ORDER - 1)))
214218229df5SAndy Whitcroft 		prep_compound_gigantic_page(page, order);
214318229df5SAndy Whitcroft 	else
214418229df5SAndy Whitcroft 		prep_compound_page(page, order);
214518229df5SAndy Whitcroft }
214618229df5SAndy Whitcroft 
2147aa888a74SAndi Kleen /* Put bootmem huge pages into the standard lists after mem_map is up */
2148aa888a74SAndi Kleen static void __init gather_bootmem_prealloc(void)
2149aa888a74SAndi Kleen {
2150aa888a74SAndi Kleen 	struct huge_bootmem_page *m;
2151aa888a74SAndi Kleen 
2152aa888a74SAndi Kleen 	list_for_each_entry(m, &huge_boot_pages, list) {
2153aa888a74SAndi Kleen 		struct hstate *h = m->hstate;
2154ee8f248dSBecky Bruce 		struct page *page;
2155ee8f248dSBecky Bruce 
2156ee8f248dSBecky Bruce #ifdef CONFIG_HIGHMEM
2157ee8f248dSBecky Bruce 		page = pfn_to_page(m->phys >> PAGE_SHIFT);
21588b89a116SGrygorii Strashko 		memblock_free_late(__pa(m),
2159ee8f248dSBecky Bruce 				   sizeof(struct huge_bootmem_page));
2160ee8f248dSBecky Bruce #else
2161ee8f248dSBecky Bruce 		page = virt_to_page(m);
2162ee8f248dSBecky Bruce #endif
2163aa888a74SAndi Kleen 		WARN_ON(page_count(page) != 1);
216418229df5SAndy Whitcroft 		prep_compound_huge_page(page, h->order);
2165ef5a22beSAndrea Arcangeli 		WARN_ON(PageReserved(page));
2166aa888a74SAndi Kleen 		prep_new_huge_page(h, page, page_to_nid(page));
2167b0320c7bSRafael Aquini 		/*
2168b0320c7bSRafael Aquini 		 * If we had gigantic hugepages allocated at boot time, we need
2169b0320c7bSRafael Aquini 		 * to restore the 'stolen' pages to totalram_pages in order to
2170b0320c7bSRafael Aquini 		 * fix confusing memory reports from free(1) and another
2171b0320c7bSRafael Aquini 		 * side-effects, like CommitLimit going negative.
2172b0320c7bSRafael Aquini 		 */
2173bae7f4aeSLuiz Capitulino 		if (hstate_is_gigantic(h))
21743dcc0571SJiang Liu 			adjust_managed_page_count(page, 1 << h->order);
2175aa888a74SAndi Kleen 	}
2176aa888a74SAndi Kleen }
2177aa888a74SAndi Kleen 
21788faa8b07SAndi Kleen static void __init hugetlb_hstate_alloc_pages(struct hstate *h)
21791da177e4SLinus Torvalds {
21801da177e4SLinus Torvalds 	unsigned long i;
21811da177e4SLinus Torvalds 
2182e5ff2159SAndi Kleen 	for (i = 0; i < h->max_huge_pages; ++i) {
2183bae7f4aeSLuiz Capitulino 		if (hstate_is_gigantic(h)) {
2184aa888a74SAndi Kleen 			if (!alloc_bootmem_huge_page(h))
2185aa888a74SAndi Kleen 				break;
21869b5e5d0fSLee Schermerhorn 		} else if (!alloc_fresh_huge_page(h,
21878cebfcd0SLai Jiangshan 					 &node_states[N_MEMORY]))
21881da177e4SLinus Torvalds 			break;
21891da177e4SLinus Torvalds 	}
21908faa8b07SAndi Kleen 	h->max_huge_pages = i;
2191e5ff2159SAndi Kleen }
2192e5ff2159SAndi Kleen 
2193e5ff2159SAndi Kleen static void __init hugetlb_init_hstates(void)
2194e5ff2159SAndi Kleen {
2195e5ff2159SAndi Kleen 	struct hstate *h;
2196e5ff2159SAndi Kleen 
2197e5ff2159SAndi Kleen 	for_each_hstate(h) {
2198641844f5SNaoya Horiguchi 		if (minimum_order > huge_page_order(h))
2199641844f5SNaoya Horiguchi 			minimum_order = huge_page_order(h);
2200641844f5SNaoya Horiguchi 
22018faa8b07SAndi Kleen 		/* oversize hugepages were init'ed in early boot */
2202bae7f4aeSLuiz Capitulino 		if (!hstate_is_gigantic(h))
22038faa8b07SAndi Kleen 			hugetlb_hstate_alloc_pages(h);
2204e5ff2159SAndi Kleen 	}
2205641844f5SNaoya Horiguchi 	VM_BUG_ON(minimum_order == UINT_MAX);
2206e5ff2159SAndi Kleen }
2207e5ff2159SAndi Kleen 
22084abd32dbSAndi Kleen static char * __init memfmt(char *buf, unsigned long n)
22094abd32dbSAndi Kleen {
22104abd32dbSAndi Kleen 	if (n >= (1UL << 30))
22114abd32dbSAndi Kleen 		sprintf(buf, "%lu GB", n >> 30);
22124abd32dbSAndi Kleen 	else if (n >= (1UL << 20))
22134abd32dbSAndi Kleen 		sprintf(buf, "%lu MB", n >> 20);
22144abd32dbSAndi Kleen 	else
22154abd32dbSAndi Kleen 		sprintf(buf, "%lu KB", n >> 10);
22164abd32dbSAndi Kleen 	return buf;
22174abd32dbSAndi Kleen }
22184abd32dbSAndi Kleen 
2219e5ff2159SAndi Kleen static void __init report_hugepages(void)
2220e5ff2159SAndi Kleen {
2221e5ff2159SAndi Kleen 	struct hstate *h;
2222e5ff2159SAndi Kleen 
2223e5ff2159SAndi Kleen 	for_each_hstate(h) {
22244abd32dbSAndi Kleen 		char buf[32];
2225ffb22af5SAndrew Morton 		pr_info("HugeTLB registered %s page size, pre-allocated %ld pages\n",
22264abd32dbSAndi Kleen 			memfmt(buf, huge_page_size(h)),
22274abd32dbSAndi Kleen 			h->free_huge_pages);
2228e5ff2159SAndi Kleen 	}
2229e5ff2159SAndi Kleen }
2230e5ff2159SAndi Kleen 
22311da177e4SLinus Torvalds #ifdef CONFIG_HIGHMEM
22326ae11b27SLee Schermerhorn static void try_to_free_low(struct hstate *h, unsigned long count,
22336ae11b27SLee Schermerhorn 						nodemask_t *nodes_allowed)
22341da177e4SLinus Torvalds {
22354415cc8dSChristoph Lameter 	int i;
22364415cc8dSChristoph Lameter 
2237bae7f4aeSLuiz Capitulino 	if (hstate_is_gigantic(h))
2238aa888a74SAndi Kleen 		return;
2239aa888a74SAndi Kleen 
22406ae11b27SLee Schermerhorn 	for_each_node_mask(i, *nodes_allowed) {
22411da177e4SLinus Torvalds 		struct page *page, *next;
2242a5516438SAndi Kleen 		struct list_head *freel = &h->hugepage_freelists[i];
2243a5516438SAndi Kleen 		list_for_each_entry_safe(page, next, freel, lru) {
2244a5516438SAndi Kleen 			if (count >= h->nr_huge_pages)
22456b0c880dSAdam Litke 				return;
22461da177e4SLinus Torvalds 			if (PageHighMem(page))
22471da177e4SLinus Torvalds 				continue;
22481da177e4SLinus Torvalds 			list_del(&page->lru);
2249e5ff2159SAndi Kleen 			update_and_free_page(h, page);
2250a5516438SAndi Kleen 			h->free_huge_pages--;
2251a5516438SAndi Kleen 			h->free_huge_pages_node[page_to_nid(page)]--;
22521da177e4SLinus Torvalds 		}
22531da177e4SLinus Torvalds 	}
22541da177e4SLinus Torvalds }
22551da177e4SLinus Torvalds #else
22566ae11b27SLee Schermerhorn static inline void try_to_free_low(struct hstate *h, unsigned long count,
22576ae11b27SLee Schermerhorn 						nodemask_t *nodes_allowed)
22581da177e4SLinus Torvalds {
22591da177e4SLinus Torvalds }
22601da177e4SLinus Torvalds #endif
22611da177e4SLinus Torvalds 
226220a0307cSWu Fengguang /*
226320a0307cSWu Fengguang  * Increment or decrement surplus_huge_pages.  Keep node-specific counters
226420a0307cSWu Fengguang  * balanced by operating on them in a round-robin fashion.
226520a0307cSWu Fengguang  * Returns 1 if an adjustment was made.
226620a0307cSWu Fengguang  */
22676ae11b27SLee Schermerhorn static int adjust_pool_surplus(struct hstate *h, nodemask_t *nodes_allowed,
22686ae11b27SLee Schermerhorn 				int delta)
226920a0307cSWu Fengguang {
2270b2261026SJoonsoo Kim 	int nr_nodes, node;
227120a0307cSWu Fengguang 
227220a0307cSWu Fengguang 	VM_BUG_ON(delta != -1 && delta != 1);
227320a0307cSWu Fengguang 
2274e8c5c824SLee Schermerhorn 	if (delta < 0) {
2275b2261026SJoonsoo Kim 		for_each_node_mask_to_alloc(h, nr_nodes, node, nodes_allowed) {
2276b2261026SJoonsoo Kim 			if (h->surplus_huge_pages_node[node])
2277b2261026SJoonsoo Kim 				goto found;
2278b2261026SJoonsoo Kim 		}
2279b2261026SJoonsoo Kim 	} else {
2280b2261026SJoonsoo Kim 		for_each_node_mask_to_free(h, nr_nodes, node, nodes_allowed) {
2281b2261026SJoonsoo Kim 			if (h->surplus_huge_pages_node[node] <
2282b2261026SJoonsoo Kim 					h->nr_huge_pages_node[node])
2283b2261026SJoonsoo Kim 				goto found;
2284e8c5c824SLee Schermerhorn 		}
22859a76db09SLee Schermerhorn 	}
2286b2261026SJoonsoo Kim 	return 0;
228720a0307cSWu Fengguang 
2288b2261026SJoonsoo Kim found:
228920a0307cSWu Fengguang 	h->surplus_huge_pages += delta;
2290b2261026SJoonsoo Kim 	h->surplus_huge_pages_node[node] += delta;
2291b2261026SJoonsoo Kim 	return 1;
229220a0307cSWu Fengguang }
229320a0307cSWu Fengguang 
2294a5516438SAndi Kleen #define persistent_huge_pages(h) (h->nr_huge_pages - h->surplus_huge_pages)
22956ae11b27SLee Schermerhorn static unsigned long set_max_huge_pages(struct hstate *h, unsigned long count,
22966ae11b27SLee Schermerhorn 						nodemask_t *nodes_allowed)
22971da177e4SLinus Torvalds {
22987893d1d5SAdam Litke 	unsigned long min_count, ret;
22991da177e4SLinus Torvalds 
2300944d9fecSLuiz Capitulino 	if (hstate_is_gigantic(h) && !gigantic_page_supported())
2301aa888a74SAndi Kleen 		return h->max_huge_pages;
2302aa888a74SAndi Kleen 
23037893d1d5SAdam Litke 	/*
23047893d1d5SAdam Litke 	 * Increase the pool size
23057893d1d5SAdam Litke 	 * First take pages out of surplus state.  Then make up the
23067893d1d5SAdam Litke 	 * remaining difference by allocating fresh huge pages.
2307d1c3fb1fSNishanth Aravamudan 	 *
2308d15c7c09SNaoya Horiguchi 	 * We might race with __alloc_buddy_huge_page() here and be unable
2309d1c3fb1fSNishanth Aravamudan 	 * to convert a surplus huge page to a normal huge page. That is
2310d1c3fb1fSNishanth Aravamudan 	 * not critical, though, it just means the overall size of the
2311d1c3fb1fSNishanth Aravamudan 	 * pool might be one hugepage larger than it needs to be, but
2312d1c3fb1fSNishanth Aravamudan 	 * within all the constraints specified by the sysctls.
23137893d1d5SAdam Litke 	 */
23141da177e4SLinus Torvalds 	spin_lock(&hugetlb_lock);
2315a5516438SAndi Kleen 	while (h->surplus_huge_pages && count > persistent_huge_pages(h)) {
23166ae11b27SLee Schermerhorn 		if (!adjust_pool_surplus(h, nodes_allowed, -1))
23177893d1d5SAdam Litke 			break;
23187893d1d5SAdam Litke 	}
23197893d1d5SAdam Litke 
2320a5516438SAndi Kleen 	while (count > persistent_huge_pages(h)) {
23217893d1d5SAdam Litke 		/*
23227893d1d5SAdam Litke 		 * If this allocation races such that we no longer need the
23237893d1d5SAdam Litke 		 * page, free_huge_page will handle it by freeing the page
23247893d1d5SAdam Litke 		 * and reducing the surplus.
23257893d1d5SAdam Litke 		 */
23267893d1d5SAdam Litke 		spin_unlock(&hugetlb_lock);
2327649920c6SJia He 
2328649920c6SJia He 		/* yield cpu to avoid soft lockup */
2329649920c6SJia He 		cond_resched();
2330649920c6SJia He 
2331944d9fecSLuiz Capitulino 		if (hstate_is_gigantic(h))
2332944d9fecSLuiz Capitulino 			ret = alloc_fresh_gigantic_page(h, nodes_allowed);
2333944d9fecSLuiz Capitulino 		else
23346ae11b27SLee Schermerhorn 			ret = alloc_fresh_huge_page(h, nodes_allowed);
23357893d1d5SAdam Litke 		spin_lock(&hugetlb_lock);
23367893d1d5SAdam Litke 		if (!ret)
23377893d1d5SAdam Litke 			goto out;
23387893d1d5SAdam Litke 
2339536240f2SMel Gorman 		/* Bail for signals. Probably ctrl-c from user */
2340536240f2SMel Gorman 		if (signal_pending(current))
2341536240f2SMel Gorman 			goto out;
23427893d1d5SAdam Litke 	}
23437893d1d5SAdam Litke 
23447893d1d5SAdam Litke 	/*
23457893d1d5SAdam Litke 	 * Decrease the pool size
23467893d1d5SAdam Litke 	 * First return free pages to the buddy allocator (being careful
23477893d1d5SAdam Litke 	 * to keep enough around to satisfy reservations).  Then place
23487893d1d5SAdam Litke 	 * pages into surplus state as needed so the pool will shrink
23497893d1d5SAdam Litke 	 * to the desired size as pages become free.
2350d1c3fb1fSNishanth Aravamudan 	 *
2351d1c3fb1fSNishanth Aravamudan 	 * By placing pages into the surplus state independent of the
2352d1c3fb1fSNishanth Aravamudan 	 * overcommit value, we are allowing the surplus pool size to
2353d1c3fb1fSNishanth Aravamudan 	 * exceed overcommit. There are few sane options here. Since
2354d15c7c09SNaoya Horiguchi 	 * __alloc_buddy_huge_page() is checking the global counter,
2355d1c3fb1fSNishanth Aravamudan 	 * though, we'll note that we're not allowed to exceed surplus
2356d1c3fb1fSNishanth Aravamudan 	 * and won't grow the pool anywhere else. Not until one of the
2357d1c3fb1fSNishanth Aravamudan 	 * sysctls are changed, or the surplus pages go out of use.
23587893d1d5SAdam Litke 	 */
2359a5516438SAndi Kleen 	min_count = h->resv_huge_pages + h->nr_huge_pages - h->free_huge_pages;
23606b0c880dSAdam Litke 	min_count = max(count, min_count);
23616ae11b27SLee Schermerhorn 	try_to_free_low(h, min_count, nodes_allowed);
2362a5516438SAndi Kleen 	while (min_count < persistent_huge_pages(h)) {
23636ae11b27SLee Schermerhorn 		if (!free_pool_huge_page(h, nodes_allowed, 0))
23641da177e4SLinus Torvalds 			break;
236555f67141SMizuma, Masayoshi 		cond_resched_lock(&hugetlb_lock);
23661da177e4SLinus Torvalds 	}
2367a5516438SAndi Kleen 	while (count < persistent_huge_pages(h)) {
23686ae11b27SLee Schermerhorn 		if (!adjust_pool_surplus(h, nodes_allowed, 1))
23697893d1d5SAdam Litke 			break;
23707893d1d5SAdam Litke 	}
23717893d1d5SAdam Litke out:
2372a5516438SAndi Kleen 	ret = persistent_huge_pages(h);
23731da177e4SLinus Torvalds 	spin_unlock(&hugetlb_lock);
23747893d1d5SAdam Litke 	return ret;
23751da177e4SLinus Torvalds }
23761da177e4SLinus Torvalds 
2377a3437870SNishanth Aravamudan #define HSTATE_ATTR_RO(_name) \
2378a3437870SNishanth Aravamudan 	static struct kobj_attribute _name##_attr = __ATTR_RO(_name)
2379a3437870SNishanth Aravamudan 
2380a3437870SNishanth Aravamudan #define HSTATE_ATTR(_name) \
2381a3437870SNishanth Aravamudan 	static struct kobj_attribute _name##_attr = \
2382a3437870SNishanth Aravamudan 		__ATTR(_name, 0644, _name##_show, _name##_store)
2383a3437870SNishanth Aravamudan 
2384a3437870SNishanth Aravamudan static struct kobject *hugepages_kobj;
2385a3437870SNishanth Aravamudan static struct kobject *hstate_kobjs[HUGE_MAX_HSTATE];
2386a3437870SNishanth Aravamudan 
23879a305230SLee Schermerhorn static struct hstate *kobj_to_node_hstate(struct kobject *kobj, int *nidp);
23889a305230SLee Schermerhorn 
23899a305230SLee Schermerhorn static struct hstate *kobj_to_hstate(struct kobject *kobj, int *nidp)
2390a3437870SNishanth Aravamudan {
2391a3437870SNishanth Aravamudan 	int i;
23929a305230SLee Schermerhorn 
2393a3437870SNishanth Aravamudan 	for (i = 0; i < HUGE_MAX_HSTATE; i++)
23949a305230SLee Schermerhorn 		if (hstate_kobjs[i] == kobj) {
23959a305230SLee Schermerhorn 			if (nidp)
23969a305230SLee Schermerhorn 				*nidp = NUMA_NO_NODE;
2397a3437870SNishanth Aravamudan 			return &hstates[i];
23989a305230SLee Schermerhorn 		}
23999a305230SLee Schermerhorn 
24009a305230SLee Schermerhorn 	return kobj_to_node_hstate(kobj, nidp);
2401a3437870SNishanth Aravamudan }
2402a3437870SNishanth Aravamudan 
240306808b08SLee Schermerhorn static ssize_t nr_hugepages_show_common(struct kobject *kobj,
2404a3437870SNishanth Aravamudan 					struct kobj_attribute *attr, char *buf)
2405a3437870SNishanth Aravamudan {
24069a305230SLee Schermerhorn 	struct hstate *h;
24079a305230SLee Schermerhorn 	unsigned long nr_huge_pages;
24089a305230SLee Schermerhorn 	int nid;
24099a305230SLee Schermerhorn 
24109a305230SLee Schermerhorn 	h = kobj_to_hstate(kobj, &nid);
24119a305230SLee Schermerhorn 	if (nid == NUMA_NO_NODE)
24129a305230SLee Schermerhorn 		nr_huge_pages = h->nr_huge_pages;
24139a305230SLee Schermerhorn 	else
24149a305230SLee Schermerhorn 		nr_huge_pages = h->nr_huge_pages_node[nid];
24159a305230SLee Schermerhorn 
24169a305230SLee Schermerhorn 	return sprintf(buf, "%lu\n", nr_huge_pages);
2417a3437870SNishanth Aravamudan }
2418adbe8726SEric B Munson 
2419238d3c13SDavid Rientjes static ssize_t __nr_hugepages_store_common(bool obey_mempolicy,
2420238d3c13SDavid Rientjes 					   struct hstate *h, int nid,
2421238d3c13SDavid Rientjes 					   unsigned long count, size_t len)
2422a3437870SNishanth Aravamudan {
2423a3437870SNishanth Aravamudan 	int err;
2424bad44b5bSDavid Rientjes 	NODEMASK_ALLOC(nodemask_t, nodes_allowed, GFP_KERNEL | __GFP_NORETRY);
2425a3437870SNishanth Aravamudan 
2426944d9fecSLuiz Capitulino 	if (hstate_is_gigantic(h) && !gigantic_page_supported()) {
2427adbe8726SEric B Munson 		err = -EINVAL;
2428adbe8726SEric B Munson 		goto out;
2429adbe8726SEric B Munson 	}
2430adbe8726SEric B Munson 
24319a305230SLee Schermerhorn 	if (nid == NUMA_NO_NODE) {
24329a305230SLee Schermerhorn 		/*
24339a305230SLee Schermerhorn 		 * global hstate attribute
24349a305230SLee Schermerhorn 		 */
24359a305230SLee Schermerhorn 		if (!(obey_mempolicy &&
24369a305230SLee Schermerhorn 				init_nodemask_of_mempolicy(nodes_allowed))) {
243706808b08SLee Schermerhorn 			NODEMASK_FREE(nodes_allowed);
24388cebfcd0SLai Jiangshan 			nodes_allowed = &node_states[N_MEMORY];
243906808b08SLee Schermerhorn 		}
24409a305230SLee Schermerhorn 	} else if (nodes_allowed) {
24419a305230SLee Schermerhorn 		/*
24429a305230SLee Schermerhorn 		 * per node hstate attribute: adjust count to global,
24439a305230SLee Schermerhorn 		 * but restrict alloc/free to the specified node.
24449a305230SLee Schermerhorn 		 */
24459a305230SLee Schermerhorn 		count += h->nr_huge_pages - h->nr_huge_pages_node[nid];
24469a305230SLee Schermerhorn 		init_nodemask_of_node(nodes_allowed, nid);
24479a305230SLee Schermerhorn 	} else
24488cebfcd0SLai Jiangshan 		nodes_allowed = &node_states[N_MEMORY];
24499a305230SLee Schermerhorn 
245006808b08SLee Schermerhorn 	h->max_huge_pages = set_max_huge_pages(h, count, nodes_allowed);
2451a3437870SNishanth Aravamudan 
24528cebfcd0SLai Jiangshan 	if (nodes_allowed != &node_states[N_MEMORY])
245306808b08SLee Schermerhorn 		NODEMASK_FREE(nodes_allowed);
245406808b08SLee Schermerhorn 
245506808b08SLee Schermerhorn 	return len;
2456adbe8726SEric B Munson out:
2457adbe8726SEric B Munson 	NODEMASK_FREE(nodes_allowed);
2458adbe8726SEric B Munson 	return err;
245906808b08SLee Schermerhorn }
246006808b08SLee Schermerhorn 
2461238d3c13SDavid Rientjes static ssize_t nr_hugepages_store_common(bool obey_mempolicy,
2462238d3c13SDavid Rientjes 					 struct kobject *kobj, const char *buf,
2463238d3c13SDavid Rientjes 					 size_t len)
2464238d3c13SDavid Rientjes {
2465238d3c13SDavid Rientjes 	struct hstate *h;
2466238d3c13SDavid Rientjes 	unsigned long count;
2467238d3c13SDavid Rientjes 	int nid;
2468238d3c13SDavid Rientjes 	int err;
2469238d3c13SDavid Rientjes 
2470238d3c13SDavid Rientjes 	err = kstrtoul(buf, 10, &count);
2471238d3c13SDavid Rientjes 	if (err)
2472238d3c13SDavid Rientjes 		return err;
2473238d3c13SDavid Rientjes 
2474238d3c13SDavid Rientjes 	h = kobj_to_hstate(kobj, &nid);
2475238d3c13SDavid Rientjes 	return __nr_hugepages_store_common(obey_mempolicy, h, nid, count, len);
2476238d3c13SDavid Rientjes }
2477238d3c13SDavid Rientjes 
247806808b08SLee Schermerhorn static ssize_t nr_hugepages_show(struct kobject *kobj,
247906808b08SLee Schermerhorn 				       struct kobj_attribute *attr, char *buf)
248006808b08SLee Schermerhorn {
248106808b08SLee Schermerhorn 	return nr_hugepages_show_common(kobj, attr, buf);
248206808b08SLee Schermerhorn }
248306808b08SLee Schermerhorn 
248406808b08SLee Schermerhorn static ssize_t nr_hugepages_store(struct kobject *kobj,
248506808b08SLee Schermerhorn 	       struct kobj_attribute *attr, const char *buf, size_t len)
248606808b08SLee Schermerhorn {
2487238d3c13SDavid Rientjes 	return nr_hugepages_store_common(false, kobj, buf, len);
2488a3437870SNishanth Aravamudan }
2489a3437870SNishanth Aravamudan HSTATE_ATTR(nr_hugepages);
2490a3437870SNishanth Aravamudan 
249106808b08SLee Schermerhorn #ifdef CONFIG_NUMA
249206808b08SLee Schermerhorn 
249306808b08SLee Schermerhorn /*
249406808b08SLee Schermerhorn  * hstate attribute for optionally mempolicy-based constraint on persistent
249506808b08SLee Schermerhorn  * huge page alloc/free.
249606808b08SLee Schermerhorn  */
249706808b08SLee Schermerhorn static ssize_t nr_hugepages_mempolicy_show(struct kobject *kobj,
249806808b08SLee Schermerhorn 				       struct kobj_attribute *attr, char *buf)
249906808b08SLee Schermerhorn {
250006808b08SLee Schermerhorn 	return nr_hugepages_show_common(kobj, attr, buf);
250106808b08SLee Schermerhorn }
250206808b08SLee Schermerhorn 
250306808b08SLee Schermerhorn static ssize_t nr_hugepages_mempolicy_store(struct kobject *kobj,
250406808b08SLee Schermerhorn 	       struct kobj_attribute *attr, const char *buf, size_t len)
250506808b08SLee Schermerhorn {
2506238d3c13SDavid Rientjes 	return nr_hugepages_store_common(true, kobj, buf, len);
250706808b08SLee Schermerhorn }
250806808b08SLee Schermerhorn HSTATE_ATTR(nr_hugepages_mempolicy);
250906808b08SLee Schermerhorn #endif
251006808b08SLee Schermerhorn 
251106808b08SLee Schermerhorn 
2512a3437870SNishanth Aravamudan static ssize_t nr_overcommit_hugepages_show(struct kobject *kobj,
2513a3437870SNishanth Aravamudan 					struct kobj_attribute *attr, char *buf)
2514a3437870SNishanth Aravamudan {
25159a305230SLee Schermerhorn 	struct hstate *h = kobj_to_hstate(kobj, NULL);
2516a3437870SNishanth Aravamudan 	return sprintf(buf, "%lu\n", h->nr_overcommit_huge_pages);
2517a3437870SNishanth Aravamudan }
2518adbe8726SEric B Munson 
2519a3437870SNishanth Aravamudan static ssize_t nr_overcommit_hugepages_store(struct kobject *kobj,
2520a3437870SNishanth Aravamudan 		struct kobj_attribute *attr, const char *buf, size_t count)
2521a3437870SNishanth Aravamudan {
2522a3437870SNishanth Aravamudan 	int err;
2523a3437870SNishanth Aravamudan 	unsigned long input;
25249a305230SLee Schermerhorn 	struct hstate *h = kobj_to_hstate(kobj, NULL);
2525a3437870SNishanth Aravamudan 
2526bae7f4aeSLuiz Capitulino 	if (hstate_is_gigantic(h))
2527adbe8726SEric B Munson 		return -EINVAL;
2528adbe8726SEric B Munson 
25293dbb95f7SJingoo Han 	err = kstrtoul(buf, 10, &input);
2530a3437870SNishanth Aravamudan 	if (err)
253173ae31e5SEric B Munson 		return err;
2532a3437870SNishanth Aravamudan 
2533a3437870SNishanth Aravamudan 	spin_lock(&hugetlb_lock);
2534a3437870SNishanth Aravamudan 	h->nr_overcommit_huge_pages = input;
2535a3437870SNishanth Aravamudan 	spin_unlock(&hugetlb_lock);
2536a3437870SNishanth Aravamudan 
2537a3437870SNishanth Aravamudan 	return count;
2538a3437870SNishanth Aravamudan }
2539a3437870SNishanth Aravamudan HSTATE_ATTR(nr_overcommit_hugepages);
2540a3437870SNishanth Aravamudan 
2541a3437870SNishanth Aravamudan static ssize_t free_hugepages_show(struct kobject *kobj,
2542a3437870SNishanth Aravamudan 					struct kobj_attribute *attr, char *buf)
2543a3437870SNishanth Aravamudan {
25449a305230SLee Schermerhorn 	struct hstate *h;
25459a305230SLee Schermerhorn 	unsigned long free_huge_pages;
25469a305230SLee Schermerhorn 	int nid;
25479a305230SLee Schermerhorn 
25489a305230SLee Schermerhorn 	h = kobj_to_hstate(kobj, &nid);
25499a305230SLee Schermerhorn 	if (nid == NUMA_NO_NODE)
25509a305230SLee Schermerhorn 		free_huge_pages = h->free_huge_pages;
25519a305230SLee Schermerhorn 	else
25529a305230SLee Schermerhorn 		free_huge_pages = h->free_huge_pages_node[nid];
25539a305230SLee Schermerhorn 
25549a305230SLee Schermerhorn 	return sprintf(buf, "%lu\n", free_huge_pages);
2555a3437870SNishanth Aravamudan }
2556a3437870SNishanth Aravamudan HSTATE_ATTR_RO(free_hugepages);
2557a3437870SNishanth Aravamudan 
2558a3437870SNishanth Aravamudan static ssize_t resv_hugepages_show(struct kobject *kobj,
2559a3437870SNishanth Aravamudan 					struct kobj_attribute *attr, char *buf)
2560a3437870SNishanth Aravamudan {
25619a305230SLee Schermerhorn 	struct hstate *h = kobj_to_hstate(kobj, NULL);
2562a3437870SNishanth Aravamudan 	return sprintf(buf, "%lu\n", h->resv_huge_pages);
2563a3437870SNishanth Aravamudan }
2564a3437870SNishanth Aravamudan HSTATE_ATTR_RO(resv_hugepages);
2565a3437870SNishanth Aravamudan 
2566a3437870SNishanth Aravamudan static ssize_t surplus_hugepages_show(struct kobject *kobj,
2567a3437870SNishanth Aravamudan 					struct kobj_attribute *attr, char *buf)
2568a3437870SNishanth Aravamudan {
25699a305230SLee Schermerhorn 	struct hstate *h;
25709a305230SLee Schermerhorn 	unsigned long surplus_huge_pages;
25719a305230SLee Schermerhorn 	int nid;
25729a305230SLee Schermerhorn 
25739a305230SLee Schermerhorn 	h = kobj_to_hstate(kobj, &nid);
25749a305230SLee Schermerhorn 	if (nid == NUMA_NO_NODE)
25759a305230SLee Schermerhorn 		surplus_huge_pages = h->surplus_huge_pages;
25769a305230SLee Schermerhorn 	else
25779a305230SLee Schermerhorn 		surplus_huge_pages = h->surplus_huge_pages_node[nid];
25789a305230SLee Schermerhorn 
25799a305230SLee Schermerhorn 	return sprintf(buf, "%lu\n", surplus_huge_pages);
2580a3437870SNishanth Aravamudan }
2581a3437870SNishanth Aravamudan HSTATE_ATTR_RO(surplus_hugepages);
2582a3437870SNishanth Aravamudan 
2583a3437870SNishanth Aravamudan static struct attribute *hstate_attrs[] = {
2584a3437870SNishanth Aravamudan 	&nr_hugepages_attr.attr,
2585a3437870SNishanth Aravamudan 	&nr_overcommit_hugepages_attr.attr,
2586a3437870SNishanth Aravamudan 	&free_hugepages_attr.attr,
2587a3437870SNishanth Aravamudan 	&resv_hugepages_attr.attr,
2588a3437870SNishanth Aravamudan 	&surplus_hugepages_attr.attr,
258906808b08SLee Schermerhorn #ifdef CONFIG_NUMA
259006808b08SLee Schermerhorn 	&nr_hugepages_mempolicy_attr.attr,
259106808b08SLee Schermerhorn #endif
2592a3437870SNishanth Aravamudan 	NULL,
2593a3437870SNishanth Aravamudan };
2594a3437870SNishanth Aravamudan 
2595a3437870SNishanth Aravamudan static struct attribute_group hstate_attr_group = {
2596a3437870SNishanth Aravamudan 	.attrs = hstate_attrs,
2597a3437870SNishanth Aravamudan };
2598a3437870SNishanth Aravamudan 
2599094e9539SJeff Mahoney static int hugetlb_sysfs_add_hstate(struct hstate *h, struct kobject *parent,
26009a305230SLee Schermerhorn 				    struct kobject **hstate_kobjs,
26019a305230SLee Schermerhorn 				    struct attribute_group *hstate_attr_group)
2602a3437870SNishanth Aravamudan {
2603a3437870SNishanth Aravamudan 	int retval;
2604972dc4deSAneesh Kumar K.V 	int hi = hstate_index(h);
2605a3437870SNishanth Aravamudan 
26069a305230SLee Schermerhorn 	hstate_kobjs[hi] = kobject_create_and_add(h->name, parent);
26079a305230SLee Schermerhorn 	if (!hstate_kobjs[hi])
2608a3437870SNishanth Aravamudan 		return -ENOMEM;
2609a3437870SNishanth Aravamudan 
26109a305230SLee Schermerhorn 	retval = sysfs_create_group(hstate_kobjs[hi], hstate_attr_group);
2611a3437870SNishanth Aravamudan 	if (retval)
26129a305230SLee Schermerhorn 		kobject_put(hstate_kobjs[hi]);
2613a3437870SNishanth Aravamudan 
2614a3437870SNishanth Aravamudan 	return retval;
2615a3437870SNishanth Aravamudan }
2616a3437870SNishanth Aravamudan 
2617a3437870SNishanth Aravamudan static void __init hugetlb_sysfs_init(void)
2618a3437870SNishanth Aravamudan {
2619a3437870SNishanth Aravamudan 	struct hstate *h;
2620a3437870SNishanth Aravamudan 	int err;
2621a3437870SNishanth Aravamudan 
2622a3437870SNishanth Aravamudan 	hugepages_kobj = kobject_create_and_add("hugepages", mm_kobj);
2623a3437870SNishanth Aravamudan 	if (!hugepages_kobj)
2624a3437870SNishanth Aravamudan 		return;
2625a3437870SNishanth Aravamudan 
2626a3437870SNishanth Aravamudan 	for_each_hstate(h) {
26279a305230SLee Schermerhorn 		err = hugetlb_sysfs_add_hstate(h, hugepages_kobj,
26289a305230SLee Schermerhorn 					 hstate_kobjs, &hstate_attr_group);
2629a3437870SNishanth Aravamudan 		if (err)
2630ffb22af5SAndrew Morton 			pr_err("Hugetlb: Unable to add hstate %s", h->name);
2631a3437870SNishanth Aravamudan 	}
2632a3437870SNishanth Aravamudan }
2633a3437870SNishanth Aravamudan 
26349a305230SLee Schermerhorn #ifdef CONFIG_NUMA
26359a305230SLee Schermerhorn 
26369a305230SLee Schermerhorn /*
26379a305230SLee Schermerhorn  * node_hstate/s - associate per node hstate attributes, via their kobjects,
263810fbcf4cSKay Sievers  * with node devices in node_devices[] using a parallel array.  The array
263910fbcf4cSKay Sievers  * index of a node device or _hstate == node id.
264010fbcf4cSKay Sievers  * This is here to avoid any static dependency of the node device driver, in
26419a305230SLee Schermerhorn  * the base kernel, on the hugetlb module.
26429a305230SLee Schermerhorn  */
26439a305230SLee Schermerhorn struct node_hstate {
26449a305230SLee Schermerhorn 	struct kobject		*hugepages_kobj;
26459a305230SLee Schermerhorn 	struct kobject		*hstate_kobjs[HUGE_MAX_HSTATE];
26469a305230SLee Schermerhorn };
2647b4e289a6SAlexander Kuleshov static struct node_hstate node_hstates[MAX_NUMNODES];
26489a305230SLee Schermerhorn 
26499a305230SLee Schermerhorn /*
265010fbcf4cSKay Sievers  * A subset of global hstate attributes for node devices
26519a305230SLee Schermerhorn  */
26529a305230SLee Schermerhorn static struct attribute *per_node_hstate_attrs[] = {
26539a305230SLee Schermerhorn 	&nr_hugepages_attr.attr,
26549a305230SLee Schermerhorn 	&free_hugepages_attr.attr,
26559a305230SLee Schermerhorn 	&surplus_hugepages_attr.attr,
26569a305230SLee Schermerhorn 	NULL,
26579a305230SLee Schermerhorn };
26589a305230SLee Schermerhorn 
26599a305230SLee Schermerhorn static struct attribute_group per_node_hstate_attr_group = {
26609a305230SLee Schermerhorn 	.attrs = per_node_hstate_attrs,
26619a305230SLee Schermerhorn };
26629a305230SLee Schermerhorn 
26639a305230SLee Schermerhorn /*
266410fbcf4cSKay Sievers  * kobj_to_node_hstate - lookup global hstate for node device hstate attr kobj.
26659a305230SLee Schermerhorn  * Returns node id via non-NULL nidp.
26669a305230SLee Schermerhorn  */
26679a305230SLee Schermerhorn static struct hstate *kobj_to_node_hstate(struct kobject *kobj, int *nidp)
26689a305230SLee Schermerhorn {
26699a305230SLee Schermerhorn 	int nid;
26709a305230SLee Schermerhorn 
26719a305230SLee Schermerhorn 	for (nid = 0; nid < nr_node_ids; nid++) {
26729a305230SLee Schermerhorn 		struct node_hstate *nhs = &node_hstates[nid];
26739a305230SLee Schermerhorn 		int i;
26749a305230SLee Schermerhorn 		for (i = 0; i < HUGE_MAX_HSTATE; i++)
26759a305230SLee Schermerhorn 			if (nhs->hstate_kobjs[i] == kobj) {
26769a305230SLee Schermerhorn 				if (nidp)
26779a305230SLee Schermerhorn 					*nidp = nid;
26789a305230SLee Schermerhorn 				return &hstates[i];
26799a305230SLee Schermerhorn 			}
26809a305230SLee Schermerhorn 	}
26819a305230SLee Schermerhorn 
26829a305230SLee Schermerhorn 	BUG();
26839a305230SLee Schermerhorn 	return NULL;
26849a305230SLee Schermerhorn }
26859a305230SLee Schermerhorn 
26869a305230SLee Schermerhorn /*
268710fbcf4cSKay Sievers  * Unregister hstate attributes from a single node device.
26889a305230SLee Schermerhorn  * No-op if no hstate attributes attached.
26899a305230SLee Schermerhorn  */
26903cd8b44fSClaudiu Ghioc static void hugetlb_unregister_node(struct node *node)
26919a305230SLee Schermerhorn {
26929a305230SLee Schermerhorn 	struct hstate *h;
269310fbcf4cSKay Sievers 	struct node_hstate *nhs = &node_hstates[node->dev.id];
26949a305230SLee Schermerhorn 
26959a305230SLee Schermerhorn 	if (!nhs->hugepages_kobj)
26969b5e5d0fSLee Schermerhorn 		return;		/* no hstate attributes */
26979a305230SLee Schermerhorn 
2698972dc4deSAneesh Kumar K.V 	for_each_hstate(h) {
2699972dc4deSAneesh Kumar K.V 		int idx = hstate_index(h);
2700972dc4deSAneesh Kumar K.V 		if (nhs->hstate_kobjs[idx]) {
2701972dc4deSAneesh Kumar K.V 			kobject_put(nhs->hstate_kobjs[idx]);
2702972dc4deSAneesh Kumar K.V 			nhs->hstate_kobjs[idx] = NULL;
2703972dc4deSAneesh Kumar K.V 		}
27049a305230SLee Schermerhorn 	}
27059a305230SLee Schermerhorn 
27069a305230SLee Schermerhorn 	kobject_put(nhs->hugepages_kobj);
27079a305230SLee Schermerhorn 	nhs->hugepages_kobj = NULL;
27089a305230SLee Schermerhorn }
27099a305230SLee Schermerhorn 
27109a305230SLee Schermerhorn 
27119a305230SLee Schermerhorn /*
271210fbcf4cSKay Sievers  * Register hstate attributes for a single node device.
27139a305230SLee Schermerhorn  * No-op if attributes already registered.
27149a305230SLee Schermerhorn  */
27153cd8b44fSClaudiu Ghioc static void hugetlb_register_node(struct node *node)
27169a305230SLee Schermerhorn {
27179a305230SLee Schermerhorn 	struct hstate *h;
271810fbcf4cSKay Sievers 	struct node_hstate *nhs = &node_hstates[node->dev.id];
27199a305230SLee Schermerhorn 	int err;
27209a305230SLee Schermerhorn 
27219a305230SLee Schermerhorn 	if (nhs->hugepages_kobj)
27229a305230SLee Schermerhorn 		return;		/* already allocated */
27239a305230SLee Schermerhorn 
27249a305230SLee Schermerhorn 	nhs->hugepages_kobj = kobject_create_and_add("hugepages",
272510fbcf4cSKay Sievers 							&node->dev.kobj);
27269a305230SLee Schermerhorn 	if (!nhs->hugepages_kobj)
27279a305230SLee Schermerhorn 		return;
27289a305230SLee Schermerhorn 
27299a305230SLee Schermerhorn 	for_each_hstate(h) {
27309a305230SLee Schermerhorn 		err = hugetlb_sysfs_add_hstate(h, nhs->hugepages_kobj,
27319a305230SLee Schermerhorn 						nhs->hstate_kobjs,
27329a305230SLee Schermerhorn 						&per_node_hstate_attr_group);
27339a305230SLee Schermerhorn 		if (err) {
2734ffb22af5SAndrew Morton 			pr_err("Hugetlb: Unable to add hstate %s for node %d\n",
273510fbcf4cSKay Sievers 				h->name, node->dev.id);
27369a305230SLee Schermerhorn 			hugetlb_unregister_node(node);
27379a305230SLee Schermerhorn 			break;
27389a305230SLee Schermerhorn 		}
27399a305230SLee Schermerhorn 	}
27409a305230SLee Schermerhorn }
27419a305230SLee Schermerhorn 
27429a305230SLee Schermerhorn /*
27439b5e5d0fSLee Schermerhorn  * hugetlb init time:  register hstate attributes for all registered node
274410fbcf4cSKay Sievers  * devices of nodes that have memory.  All on-line nodes should have
274510fbcf4cSKay Sievers  * registered their associated device by this time.
27469a305230SLee Schermerhorn  */
27477d9ca000SLuiz Capitulino static void __init hugetlb_register_all_nodes(void)
27489a305230SLee Schermerhorn {
27499a305230SLee Schermerhorn 	int nid;
27509a305230SLee Schermerhorn 
27518cebfcd0SLai Jiangshan 	for_each_node_state(nid, N_MEMORY) {
27528732794bSWen Congyang 		struct node *node = node_devices[nid];
275310fbcf4cSKay Sievers 		if (node->dev.id == nid)
27549a305230SLee Schermerhorn 			hugetlb_register_node(node);
27559a305230SLee Schermerhorn 	}
27569a305230SLee Schermerhorn 
27579a305230SLee Schermerhorn 	/*
275810fbcf4cSKay Sievers 	 * Let the node device driver know we're here so it can
27599a305230SLee Schermerhorn 	 * [un]register hstate attributes on node hotplug.
27609a305230SLee Schermerhorn 	 */
27619a305230SLee Schermerhorn 	register_hugetlbfs_with_node(hugetlb_register_node,
27629a305230SLee Schermerhorn 				     hugetlb_unregister_node);
27639a305230SLee Schermerhorn }
27649a305230SLee Schermerhorn #else	/* !CONFIG_NUMA */
27659a305230SLee Schermerhorn 
27669a305230SLee Schermerhorn static struct hstate *kobj_to_node_hstate(struct kobject *kobj, int *nidp)
27679a305230SLee Schermerhorn {
27689a305230SLee Schermerhorn 	BUG();
27699a305230SLee Schermerhorn 	if (nidp)
27709a305230SLee Schermerhorn 		*nidp = -1;
27719a305230SLee Schermerhorn 	return NULL;
27729a305230SLee Schermerhorn }
27739a305230SLee Schermerhorn 
27749a305230SLee Schermerhorn static void hugetlb_register_all_nodes(void) { }
27759a305230SLee Schermerhorn 
27769a305230SLee Schermerhorn #endif
27779a305230SLee Schermerhorn 
2778a3437870SNishanth Aravamudan static int __init hugetlb_init(void)
2779a3437870SNishanth Aravamudan {
27808382d914SDavidlohr Bueso 	int i;
27818382d914SDavidlohr Bueso 
2782457c1b27SNishanth Aravamudan 	if (!hugepages_supported())
27830ef89d25SBenjamin Herrenschmidt 		return 0;
2784a3437870SNishanth Aravamudan 
2785e11bfbfcSNick Piggin 	if (!size_to_hstate(default_hstate_size)) {
2786e11bfbfcSNick Piggin 		default_hstate_size = HPAGE_SIZE;
2787e11bfbfcSNick Piggin 		if (!size_to_hstate(default_hstate_size))
2788a3437870SNishanth Aravamudan 			hugetlb_add_hstate(HUGETLB_PAGE_ORDER);
2789a3437870SNishanth Aravamudan 	}
2790972dc4deSAneesh Kumar K.V 	default_hstate_idx = hstate_index(size_to_hstate(default_hstate_size));
2791f8b74815SVaishali Thakkar 	if (default_hstate_max_huge_pages) {
2792f8b74815SVaishali Thakkar 		if (!default_hstate.max_huge_pages)
2793e11bfbfcSNick Piggin 			default_hstate.max_huge_pages = default_hstate_max_huge_pages;
2794f8b74815SVaishali Thakkar 	}
2795a3437870SNishanth Aravamudan 
2796a3437870SNishanth Aravamudan 	hugetlb_init_hstates();
2797aa888a74SAndi Kleen 	gather_bootmem_prealloc();
2798a3437870SNishanth Aravamudan 	report_hugepages();
2799a3437870SNishanth Aravamudan 
2800a3437870SNishanth Aravamudan 	hugetlb_sysfs_init();
28019a305230SLee Schermerhorn 	hugetlb_register_all_nodes();
28027179e7bfSJianguo Wu 	hugetlb_cgroup_file_init();
28039a305230SLee Schermerhorn 
28048382d914SDavidlohr Bueso #ifdef CONFIG_SMP
28058382d914SDavidlohr Bueso 	num_fault_mutexes = roundup_pow_of_two(8 * num_possible_cpus());
28068382d914SDavidlohr Bueso #else
28078382d914SDavidlohr Bueso 	num_fault_mutexes = 1;
28088382d914SDavidlohr Bueso #endif
2809c672c7f2SMike Kravetz 	hugetlb_fault_mutex_table =
28108382d914SDavidlohr Bueso 		kmalloc(sizeof(struct mutex) * num_fault_mutexes, GFP_KERNEL);
2811c672c7f2SMike Kravetz 	BUG_ON(!hugetlb_fault_mutex_table);
28128382d914SDavidlohr Bueso 
28138382d914SDavidlohr Bueso 	for (i = 0; i < num_fault_mutexes; i++)
2814c672c7f2SMike Kravetz 		mutex_init(&hugetlb_fault_mutex_table[i]);
2815a3437870SNishanth Aravamudan 	return 0;
2816a3437870SNishanth Aravamudan }
28173e89e1c5SPaul Gortmaker subsys_initcall(hugetlb_init);
2818a3437870SNishanth Aravamudan 
2819a3437870SNishanth Aravamudan /* Should be called on processing a hugepagesz=... option */
28209fee021dSVaishali Thakkar void __init hugetlb_bad_size(void)
28219fee021dSVaishali Thakkar {
28229fee021dSVaishali Thakkar 	parsed_valid_hugepagesz = false;
28239fee021dSVaishali Thakkar }
28249fee021dSVaishali Thakkar 
2825d00181b9SKirill A. Shutemov void __init hugetlb_add_hstate(unsigned int order)
2826a3437870SNishanth Aravamudan {
2827a3437870SNishanth Aravamudan 	struct hstate *h;
28288faa8b07SAndi Kleen 	unsigned long i;
28298faa8b07SAndi Kleen 
2830a3437870SNishanth Aravamudan 	if (size_to_hstate(PAGE_SIZE << order)) {
2831598d8091SJoe Perches 		pr_warn("hugepagesz= specified twice, ignoring\n");
2832a3437870SNishanth Aravamudan 		return;
2833a3437870SNishanth Aravamudan 	}
283447d38344SAneesh Kumar K.V 	BUG_ON(hugetlb_max_hstate >= HUGE_MAX_HSTATE);
2835a3437870SNishanth Aravamudan 	BUG_ON(order == 0);
283647d38344SAneesh Kumar K.V 	h = &hstates[hugetlb_max_hstate++];
2837a3437870SNishanth Aravamudan 	h->order = order;
2838a3437870SNishanth Aravamudan 	h->mask = ~((1ULL << (order + PAGE_SHIFT)) - 1);
28398faa8b07SAndi Kleen 	h->nr_huge_pages = 0;
28408faa8b07SAndi Kleen 	h->free_huge_pages = 0;
28418faa8b07SAndi Kleen 	for (i = 0; i < MAX_NUMNODES; ++i)
28428faa8b07SAndi Kleen 		INIT_LIST_HEAD(&h->hugepage_freelists[i]);
28430edaecfaSAneesh Kumar K.V 	INIT_LIST_HEAD(&h->hugepage_activelist);
284454f18d35SAndrew Morton 	h->next_nid_to_alloc = first_memory_node;
284554f18d35SAndrew Morton 	h->next_nid_to_free = first_memory_node;
2846a3437870SNishanth Aravamudan 	snprintf(h->name, HSTATE_NAME_LEN, "hugepages-%lukB",
2847a3437870SNishanth Aravamudan 					huge_page_size(h)/1024);
28488faa8b07SAndi Kleen 
2849a3437870SNishanth Aravamudan 	parsed_hstate = h;
2850a3437870SNishanth Aravamudan }
2851a3437870SNishanth Aravamudan 
2852e11bfbfcSNick Piggin static int __init hugetlb_nrpages_setup(char *s)
2853a3437870SNishanth Aravamudan {
2854a3437870SNishanth Aravamudan 	unsigned long *mhp;
28558faa8b07SAndi Kleen 	static unsigned long *last_mhp;
2856a3437870SNishanth Aravamudan 
28579fee021dSVaishali Thakkar 	if (!parsed_valid_hugepagesz) {
28589fee021dSVaishali Thakkar 		pr_warn("hugepages = %s preceded by "
28599fee021dSVaishali Thakkar 			"an unsupported hugepagesz, ignoring\n", s);
28609fee021dSVaishali Thakkar 		parsed_valid_hugepagesz = true;
28619fee021dSVaishali Thakkar 		return 1;
28629fee021dSVaishali Thakkar 	}
2863a3437870SNishanth Aravamudan 	/*
286447d38344SAneesh Kumar K.V 	 * !hugetlb_max_hstate means we haven't parsed a hugepagesz= parameter yet,
2865a3437870SNishanth Aravamudan 	 * so this hugepages= parameter goes to the "default hstate".
2866a3437870SNishanth Aravamudan 	 */
28679fee021dSVaishali Thakkar 	else if (!hugetlb_max_hstate)
2868a3437870SNishanth Aravamudan 		mhp = &default_hstate_max_huge_pages;
2869a3437870SNishanth Aravamudan 	else
2870a3437870SNishanth Aravamudan 		mhp = &parsed_hstate->max_huge_pages;
2871a3437870SNishanth Aravamudan 
28728faa8b07SAndi Kleen 	if (mhp == last_mhp) {
2873598d8091SJoe Perches 		pr_warn("hugepages= specified twice without interleaving hugepagesz=, ignoring\n");
28748faa8b07SAndi Kleen 		return 1;
28758faa8b07SAndi Kleen 	}
28768faa8b07SAndi Kleen 
2877a3437870SNishanth Aravamudan 	if (sscanf(s, "%lu", mhp) <= 0)
2878a3437870SNishanth Aravamudan 		*mhp = 0;
2879a3437870SNishanth Aravamudan 
28808faa8b07SAndi Kleen 	/*
28818faa8b07SAndi Kleen 	 * Global state is always initialized later in hugetlb_init.
28828faa8b07SAndi Kleen 	 * But we need to allocate >= MAX_ORDER hstates here early to still
28838faa8b07SAndi Kleen 	 * use the bootmem allocator.
28848faa8b07SAndi Kleen 	 */
288547d38344SAneesh Kumar K.V 	if (hugetlb_max_hstate && parsed_hstate->order >= MAX_ORDER)
28868faa8b07SAndi Kleen 		hugetlb_hstate_alloc_pages(parsed_hstate);
28878faa8b07SAndi Kleen 
28888faa8b07SAndi Kleen 	last_mhp = mhp;
28898faa8b07SAndi Kleen 
2890a3437870SNishanth Aravamudan 	return 1;
2891a3437870SNishanth Aravamudan }
2892e11bfbfcSNick Piggin __setup("hugepages=", hugetlb_nrpages_setup);
2893e11bfbfcSNick Piggin 
2894e11bfbfcSNick Piggin static int __init hugetlb_default_setup(char *s)
2895e11bfbfcSNick Piggin {
2896e11bfbfcSNick Piggin 	default_hstate_size = memparse(s, &s);
2897e11bfbfcSNick Piggin 	return 1;
2898e11bfbfcSNick Piggin }
2899e11bfbfcSNick Piggin __setup("default_hugepagesz=", hugetlb_default_setup);
2900a3437870SNishanth Aravamudan 
29018a213460SNishanth Aravamudan static unsigned int cpuset_mems_nr(unsigned int *array)
29028a213460SNishanth Aravamudan {
29038a213460SNishanth Aravamudan 	int node;
29048a213460SNishanth Aravamudan 	unsigned int nr = 0;
29058a213460SNishanth Aravamudan 
29068a213460SNishanth Aravamudan 	for_each_node_mask(node, cpuset_current_mems_allowed)
29078a213460SNishanth Aravamudan 		nr += array[node];
29088a213460SNishanth Aravamudan 
29098a213460SNishanth Aravamudan 	return nr;
29108a213460SNishanth Aravamudan }
29118a213460SNishanth Aravamudan 
29128a213460SNishanth Aravamudan #ifdef CONFIG_SYSCTL
291306808b08SLee Schermerhorn static int hugetlb_sysctl_handler_common(bool obey_mempolicy,
291406808b08SLee Schermerhorn 			 struct ctl_table *table, int write,
291506808b08SLee Schermerhorn 			 void __user *buffer, size_t *length, loff_t *ppos)
29161da177e4SLinus Torvalds {
2917e5ff2159SAndi Kleen 	struct hstate *h = &default_hstate;
2918238d3c13SDavid Rientjes 	unsigned long tmp = h->max_huge_pages;
291908d4a246SMichal Hocko 	int ret;
2920e5ff2159SAndi Kleen 
2921457c1b27SNishanth Aravamudan 	if (!hugepages_supported())
292286613628SJan Stancek 		return -EOPNOTSUPP;
2923457c1b27SNishanth Aravamudan 
2924e5ff2159SAndi Kleen 	table->data = &tmp;
2925e5ff2159SAndi Kleen 	table->maxlen = sizeof(unsigned long);
292608d4a246SMichal Hocko 	ret = proc_doulongvec_minmax(table, write, buffer, length, ppos);
292708d4a246SMichal Hocko 	if (ret)
292808d4a246SMichal Hocko 		goto out;
2929e5ff2159SAndi Kleen 
2930238d3c13SDavid Rientjes 	if (write)
2931238d3c13SDavid Rientjes 		ret = __nr_hugepages_store_common(obey_mempolicy, h,
2932238d3c13SDavid Rientjes 						  NUMA_NO_NODE, tmp, *length);
293308d4a246SMichal Hocko out:
293408d4a246SMichal Hocko 	return ret;
29351da177e4SLinus Torvalds }
2936396faf03SMel Gorman 
293706808b08SLee Schermerhorn int hugetlb_sysctl_handler(struct ctl_table *table, int write,
293806808b08SLee Schermerhorn 			  void __user *buffer, size_t *length, loff_t *ppos)
293906808b08SLee Schermerhorn {
294006808b08SLee Schermerhorn 
294106808b08SLee Schermerhorn 	return hugetlb_sysctl_handler_common(false, table, write,
294206808b08SLee Schermerhorn 							buffer, length, ppos);
294306808b08SLee Schermerhorn }
294406808b08SLee Schermerhorn 
294506808b08SLee Schermerhorn #ifdef CONFIG_NUMA
294606808b08SLee Schermerhorn int hugetlb_mempolicy_sysctl_handler(struct ctl_table *table, int write,
294706808b08SLee Schermerhorn 			  void __user *buffer, size_t *length, loff_t *ppos)
294806808b08SLee Schermerhorn {
294906808b08SLee Schermerhorn 	return hugetlb_sysctl_handler_common(true, table, write,
295006808b08SLee Schermerhorn 							buffer, length, ppos);
295106808b08SLee Schermerhorn }
295206808b08SLee Schermerhorn #endif /* CONFIG_NUMA */
295306808b08SLee Schermerhorn 
2954a3d0c6aaSNishanth Aravamudan int hugetlb_overcommit_handler(struct ctl_table *table, int write,
29558d65af78SAlexey Dobriyan 			void __user *buffer,
2956a3d0c6aaSNishanth Aravamudan 			size_t *length, loff_t *ppos)
2957a3d0c6aaSNishanth Aravamudan {
2958a5516438SAndi Kleen 	struct hstate *h = &default_hstate;
2959e5ff2159SAndi Kleen 	unsigned long tmp;
296008d4a246SMichal Hocko 	int ret;
2961e5ff2159SAndi Kleen 
2962457c1b27SNishanth Aravamudan 	if (!hugepages_supported())
296386613628SJan Stancek 		return -EOPNOTSUPP;
2964457c1b27SNishanth Aravamudan 
2965e5ff2159SAndi Kleen 	tmp = h->nr_overcommit_huge_pages;
2966e5ff2159SAndi Kleen 
2967bae7f4aeSLuiz Capitulino 	if (write && hstate_is_gigantic(h))
2968adbe8726SEric B Munson 		return -EINVAL;
2969adbe8726SEric B Munson 
2970e5ff2159SAndi Kleen 	table->data = &tmp;
2971e5ff2159SAndi Kleen 	table->maxlen = sizeof(unsigned long);
297208d4a246SMichal Hocko 	ret = proc_doulongvec_minmax(table, write, buffer, length, ppos);
297308d4a246SMichal Hocko 	if (ret)
297408d4a246SMichal Hocko 		goto out;
2975e5ff2159SAndi Kleen 
2976e5ff2159SAndi Kleen 	if (write) {
2977064d9efeSNishanth Aravamudan 		spin_lock(&hugetlb_lock);
2978e5ff2159SAndi Kleen 		h->nr_overcommit_huge_pages = tmp;
2979a3d0c6aaSNishanth Aravamudan 		spin_unlock(&hugetlb_lock);
2980e5ff2159SAndi Kleen 	}
298108d4a246SMichal Hocko out:
298208d4a246SMichal Hocko 	return ret;
2983a3d0c6aaSNishanth Aravamudan }
2984a3d0c6aaSNishanth Aravamudan 
29851da177e4SLinus Torvalds #endif /* CONFIG_SYSCTL */
29861da177e4SLinus Torvalds 
2987e1759c21SAlexey Dobriyan void hugetlb_report_meminfo(struct seq_file *m)
29881da177e4SLinus Torvalds {
2989a5516438SAndi Kleen 	struct hstate *h = &default_hstate;
2990457c1b27SNishanth Aravamudan 	if (!hugepages_supported())
2991457c1b27SNishanth Aravamudan 		return;
2992e1759c21SAlexey Dobriyan 	seq_printf(m,
29931da177e4SLinus Torvalds 			"HugePages_Total:   %5lu\n"
29941da177e4SLinus Torvalds 			"HugePages_Free:    %5lu\n"
2995b45b5bd6SDavid Gibson 			"HugePages_Rsvd:    %5lu\n"
29967893d1d5SAdam Litke 			"HugePages_Surp:    %5lu\n"
29974f98a2feSRik van Riel 			"Hugepagesize:   %8lu kB\n",
2998a5516438SAndi Kleen 			h->nr_huge_pages,
2999a5516438SAndi Kleen 			h->free_huge_pages,
3000a5516438SAndi Kleen 			h->resv_huge_pages,
3001a5516438SAndi Kleen 			h->surplus_huge_pages,
3002a5516438SAndi Kleen 			1UL << (huge_page_order(h) + PAGE_SHIFT - 10));
30031da177e4SLinus Torvalds }
30041da177e4SLinus Torvalds 
30051da177e4SLinus Torvalds int hugetlb_report_node_meminfo(int nid, char *buf)
30061da177e4SLinus Torvalds {
3007a5516438SAndi Kleen 	struct hstate *h = &default_hstate;
3008457c1b27SNishanth Aravamudan 	if (!hugepages_supported())
3009457c1b27SNishanth Aravamudan 		return 0;
30101da177e4SLinus Torvalds 	return sprintf(buf,
30111da177e4SLinus Torvalds 		"Node %d HugePages_Total: %5u\n"
3012a1de0919SNishanth Aravamudan 		"Node %d HugePages_Free:  %5u\n"
3013a1de0919SNishanth Aravamudan 		"Node %d HugePages_Surp:  %5u\n",
3014a5516438SAndi Kleen 		nid, h->nr_huge_pages_node[nid],
3015a5516438SAndi Kleen 		nid, h->free_huge_pages_node[nid],
3016a5516438SAndi Kleen 		nid, h->surplus_huge_pages_node[nid]);
30171da177e4SLinus Torvalds }
30181da177e4SLinus Torvalds 
3019949f7ec5SDavid Rientjes void hugetlb_show_meminfo(void)
3020949f7ec5SDavid Rientjes {
3021949f7ec5SDavid Rientjes 	struct hstate *h;
3022949f7ec5SDavid Rientjes 	int nid;
3023949f7ec5SDavid Rientjes 
3024457c1b27SNishanth Aravamudan 	if (!hugepages_supported())
3025457c1b27SNishanth Aravamudan 		return;
3026457c1b27SNishanth Aravamudan 
3027949f7ec5SDavid Rientjes 	for_each_node_state(nid, N_MEMORY)
3028949f7ec5SDavid Rientjes 		for_each_hstate(h)
3029949f7ec5SDavid Rientjes 			pr_info("Node %d hugepages_total=%u hugepages_free=%u hugepages_surp=%u hugepages_size=%lukB\n",
3030949f7ec5SDavid Rientjes 				nid,
3031949f7ec5SDavid Rientjes 				h->nr_huge_pages_node[nid],
3032949f7ec5SDavid Rientjes 				h->free_huge_pages_node[nid],
3033949f7ec5SDavid Rientjes 				h->surplus_huge_pages_node[nid],
3034949f7ec5SDavid Rientjes 				1UL << (huge_page_order(h) + PAGE_SHIFT - 10));
3035949f7ec5SDavid Rientjes }
3036949f7ec5SDavid Rientjes 
30375d317b2bSNaoya Horiguchi void hugetlb_report_usage(struct seq_file *m, struct mm_struct *mm)
30385d317b2bSNaoya Horiguchi {
30395d317b2bSNaoya Horiguchi 	seq_printf(m, "HugetlbPages:\t%8lu kB\n",
30405d317b2bSNaoya Horiguchi 		   atomic_long_read(&mm->hugetlb_usage) << (PAGE_SHIFT - 10));
30415d317b2bSNaoya Horiguchi }
30425d317b2bSNaoya Horiguchi 
30431da177e4SLinus Torvalds /* Return the number pages of memory we physically have, in PAGE_SIZE units. */
30441da177e4SLinus Torvalds unsigned long hugetlb_total_pages(void)
30451da177e4SLinus Torvalds {
3046d0028588SWanpeng Li 	struct hstate *h;
3047d0028588SWanpeng Li 	unsigned long nr_total_pages = 0;
3048d0028588SWanpeng Li 
3049d0028588SWanpeng Li 	for_each_hstate(h)
3050d0028588SWanpeng Li 		nr_total_pages += h->nr_huge_pages * pages_per_huge_page(h);
3051d0028588SWanpeng Li 	return nr_total_pages;
30521da177e4SLinus Torvalds }
30531da177e4SLinus Torvalds 
3054a5516438SAndi Kleen static int hugetlb_acct_memory(struct hstate *h, long delta)
3055fc1b8a73SMel Gorman {
3056fc1b8a73SMel Gorman 	int ret = -ENOMEM;
3057fc1b8a73SMel Gorman 
3058fc1b8a73SMel Gorman 	spin_lock(&hugetlb_lock);
3059fc1b8a73SMel Gorman 	/*
3060fc1b8a73SMel Gorman 	 * When cpuset is configured, it breaks the strict hugetlb page
3061fc1b8a73SMel Gorman 	 * reservation as the accounting is done on a global variable. Such
3062fc1b8a73SMel Gorman 	 * reservation is completely rubbish in the presence of cpuset because
3063fc1b8a73SMel Gorman 	 * the reservation is not checked against page availability for the
3064fc1b8a73SMel Gorman 	 * current cpuset. Application can still potentially OOM'ed by kernel
3065fc1b8a73SMel Gorman 	 * with lack of free htlb page in cpuset that the task is in.
3066fc1b8a73SMel Gorman 	 * Attempt to enforce strict accounting with cpuset is almost
3067fc1b8a73SMel Gorman 	 * impossible (or too ugly) because cpuset is too fluid that
3068fc1b8a73SMel Gorman 	 * task or memory node can be dynamically moved between cpusets.
3069fc1b8a73SMel Gorman 	 *
3070fc1b8a73SMel Gorman 	 * The change of semantics for shared hugetlb mapping with cpuset is
3071fc1b8a73SMel Gorman 	 * undesirable. However, in order to preserve some of the semantics,
3072fc1b8a73SMel Gorman 	 * we fall back to check against current free page availability as
3073fc1b8a73SMel Gorman 	 * a best attempt and hopefully to minimize the impact of changing
3074fc1b8a73SMel Gorman 	 * semantics that cpuset has.
3075fc1b8a73SMel Gorman 	 */
3076fc1b8a73SMel Gorman 	if (delta > 0) {
3077a5516438SAndi Kleen 		if (gather_surplus_pages(h, delta) < 0)
3078fc1b8a73SMel Gorman 			goto out;
3079fc1b8a73SMel Gorman 
3080a5516438SAndi Kleen 		if (delta > cpuset_mems_nr(h->free_huge_pages_node)) {
3081a5516438SAndi Kleen 			return_unused_surplus_pages(h, delta);
3082fc1b8a73SMel Gorman 			goto out;
3083fc1b8a73SMel Gorman 		}
3084fc1b8a73SMel Gorman 	}
3085fc1b8a73SMel Gorman 
3086fc1b8a73SMel Gorman 	ret = 0;
3087fc1b8a73SMel Gorman 	if (delta < 0)
3088a5516438SAndi Kleen 		return_unused_surplus_pages(h, (unsigned long) -delta);
3089fc1b8a73SMel Gorman 
3090fc1b8a73SMel Gorman out:
3091fc1b8a73SMel Gorman 	spin_unlock(&hugetlb_lock);
3092fc1b8a73SMel Gorman 	return ret;
3093fc1b8a73SMel Gorman }
3094fc1b8a73SMel Gorman 
309584afd99bSAndy Whitcroft static void hugetlb_vm_op_open(struct vm_area_struct *vma)
309684afd99bSAndy Whitcroft {
3097f522c3acSJoonsoo Kim 	struct resv_map *resv = vma_resv_map(vma);
309884afd99bSAndy Whitcroft 
309984afd99bSAndy Whitcroft 	/*
310084afd99bSAndy Whitcroft 	 * This new VMA should share its siblings reservation map if present.
310184afd99bSAndy Whitcroft 	 * The VMA will only ever have a valid reservation map pointer where
310284afd99bSAndy Whitcroft 	 * it is being copied for another still existing VMA.  As that VMA
310325985edcSLucas De Marchi 	 * has a reference to the reservation map it cannot disappear until
310484afd99bSAndy Whitcroft 	 * after this open call completes.  It is therefore safe to take a
310584afd99bSAndy Whitcroft 	 * new reference here without additional locking.
310684afd99bSAndy Whitcroft 	 */
31074e35f483SJoonsoo Kim 	if (resv && is_vma_resv_set(vma, HPAGE_RESV_OWNER))
3108f522c3acSJoonsoo Kim 		kref_get(&resv->refs);
310984afd99bSAndy Whitcroft }
311084afd99bSAndy Whitcroft 
3111a1e78772SMel Gorman static void hugetlb_vm_op_close(struct vm_area_struct *vma)
3112a1e78772SMel Gorman {
3113a5516438SAndi Kleen 	struct hstate *h = hstate_vma(vma);
3114f522c3acSJoonsoo Kim 	struct resv_map *resv = vma_resv_map(vma);
311590481622SDavid Gibson 	struct hugepage_subpool *spool = subpool_vma(vma);
31164e35f483SJoonsoo Kim 	unsigned long reserve, start, end;
31171c5ecae3SMike Kravetz 	long gbl_reserve;
311884afd99bSAndy Whitcroft 
31194e35f483SJoonsoo Kim 	if (!resv || !is_vma_resv_set(vma, HPAGE_RESV_OWNER))
31204e35f483SJoonsoo Kim 		return;
31214e35f483SJoonsoo Kim 
3122a5516438SAndi Kleen 	start = vma_hugecache_offset(h, vma, vma->vm_start);
3123a5516438SAndi Kleen 	end = vma_hugecache_offset(h, vma, vma->vm_end);
312484afd99bSAndy Whitcroft 
31254e35f483SJoonsoo Kim 	reserve = (end - start) - region_count(resv, start, end);
312684afd99bSAndy Whitcroft 
3127f031dd27SJoonsoo Kim 	kref_put(&resv->refs, resv_map_release);
312884afd99bSAndy Whitcroft 
31297251ff78SAdam Litke 	if (reserve) {
31301c5ecae3SMike Kravetz 		/*
31311c5ecae3SMike Kravetz 		 * Decrement reserve counts.  The global reserve count may be
31321c5ecae3SMike Kravetz 		 * adjusted if the subpool has a minimum size.
31331c5ecae3SMike Kravetz 		 */
31341c5ecae3SMike Kravetz 		gbl_reserve = hugepage_subpool_put_pages(spool, reserve);
31351c5ecae3SMike Kravetz 		hugetlb_acct_memory(h, -gbl_reserve);
31367251ff78SAdam Litke 	}
3137a1e78772SMel Gorman }
3138a1e78772SMel Gorman 
31391da177e4SLinus Torvalds /*
31401da177e4SLinus Torvalds  * We cannot handle pagefaults against hugetlb pages at all.  They cause
31411da177e4SLinus Torvalds  * handle_mm_fault() to try to instantiate regular-sized pages in the
31421da177e4SLinus Torvalds  * hugegpage VMA.  do_page_fault() is supposed to trap this, so BUG is we get
31431da177e4SLinus Torvalds  * this far.
31441da177e4SLinus Torvalds  */
3145d0217ac0SNick Piggin static int hugetlb_vm_op_fault(struct vm_area_struct *vma, struct vm_fault *vmf)
31461da177e4SLinus Torvalds {
31471da177e4SLinus Torvalds 	BUG();
3148d0217ac0SNick Piggin 	return 0;
31491da177e4SLinus Torvalds }
31501da177e4SLinus Torvalds 
3151f0f37e2fSAlexey Dobriyan const struct vm_operations_struct hugetlb_vm_ops = {
3152d0217ac0SNick Piggin 	.fault = hugetlb_vm_op_fault,
315384afd99bSAndy Whitcroft 	.open = hugetlb_vm_op_open,
3154a1e78772SMel Gorman 	.close = hugetlb_vm_op_close,
31551da177e4SLinus Torvalds };
31561da177e4SLinus Torvalds 
31571e8f889bSDavid Gibson static pte_t make_huge_pte(struct vm_area_struct *vma, struct page *page,
31581e8f889bSDavid Gibson 				int writable)
315963551ae0SDavid Gibson {
316063551ae0SDavid Gibson 	pte_t entry;
316163551ae0SDavid Gibson 
31621e8f889bSDavid Gibson 	if (writable) {
3163106c992aSGerald Schaefer 		entry = huge_pte_mkwrite(huge_pte_mkdirty(mk_huge_pte(page,
3164106c992aSGerald Schaefer 					 vma->vm_page_prot)));
316563551ae0SDavid Gibson 	} else {
3166106c992aSGerald Schaefer 		entry = huge_pte_wrprotect(mk_huge_pte(page,
3167106c992aSGerald Schaefer 					   vma->vm_page_prot));
316863551ae0SDavid Gibson 	}
316963551ae0SDavid Gibson 	entry = pte_mkyoung(entry);
317063551ae0SDavid Gibson 	entry = pte_mkhuge(entry);
3171d9ed9faaSChris Metcalf 	entry = arch_make_huge_pte(entry, vma, page, writable);
317263551ae0SDavid Gibson 
317363551ae0SDavid Gibson 	return entry;
317463551ae0SDavid Gibson }
317563551ae0SDavid Gibson 
31761e8f889bSDavid Gibson static void set_huge_ptep_writable(struct vm_area_struct *vma,
31771e8f889bSDavid Gibson 				   unsigned long address, pte_t *ptep)
31781e8f889bSDavid Gibson {
31791e8f889bSDavid Gibson 	pte_t entry;
31801e8f889bSDavid Gibson 
3181106c992aSGerald Schaefer 	entry = huge_pte_mkwrite(huge_pte_mkdirty(huge_ptep_get(ptep)));
318232f84528SChris Forbes 	if (huge_ptep_set_access_flags(vma, address, ptep, entry, 1))
31834b3073e1SRussell King 		update_mmu_cache(vma, address, ptep);
31841e8f889bSDavid Gibson }
31851e8f889bSDavid Gibson 
31864a705fefSNaoya Horiguchi static int is_hugetlb_entry_migration(pte_t pte)
31874a705fefSNaoya Horiguchi {
31884a705fefSNaoya Horiguchi 	swp_entry_t swp;
31894a705fefSNaoya Horiguchi 
31904a705fefSNaoya Horiguchi 	if (huge_pte_none(pte) || pte_present(pte))
31914a705fefSNaoya Horiguchi 		return 0;
31924a705fefSNaoya Horiguchi 	swp = pte_to_swp_entry(pte);
31934a705fefSNaoya Horiguchi 	if (non_swap_entry(swp) && is_migration_entry(swp))
31944a705fefSNaoya Horiguchi 		return 1;
31954a705fefSNaoya Horiguchi 	else
31964a705fefSNaoya Horiguchi 		return 0;
31974a705fefSNaoya Horiguchi }
31984a705fefSNaoya Horiguchi 
31994a705fefSNaoya Horiguchi static int is_hugetlb_entry_hwpoisoned(pte_t pte)
32004a705fefSNaoya Horiguchi {
32014a705fefSNaoya Horiguchi 	swp_entry_t swp;
32024a705fefSNaoya Horiguchi 
32034a705fefSNaoya Horiguchi 	if (huge_pte_none(pte) || pte_present(pte))
32044a705fefSNaoya Horiguchi 		return 0;
32054a705fefSNaoya Horiguchi 	swp = pte_to_swp_entry(pte);
32064a705fefSNaoya Horiguchi 	if (non_swap_entry(swp) && is_hwpoison_entry(swp))
32074a705fefSNaoya Horiguchi 		return 1;
32084a705fefSNaoya Horiguchi 	else
32094a705fefSNaoya Horiguchi 		return 0;
32104a705fefSNaoya Horiguchi }
32111e8f889bSDavid Gibson 
321263551ae0SDavid Gibson int copy_hugetlb_page_range(struct mm_struct *dst, struct mm_struct *src,
321363551ae0SDavid Gibson 			    struct vm_area_struct *vma)
321463551ae0SDavid Gibson {
321563551ae0SDavid Gibson 	pte_t *src_pte, *dst_pte, entry;
321663551ae0SDavid Gibson 	struct page *ptepage;
32171c59827dSHugh Dickins 	unsigned long addr;
32181e8f889bSDavid Gibson 	int cow;
3219a5516438SAndi Kleen 	struct hstate *h = hstate_vma(vma);
3220a5516438SAndi Kleen 	unsigned long sz = huge_page_size(h);
3221e8569dd2SAndreas Sandberg 	unsigned long mmun_start;	/* For mmu_notifiers */
3222e8569dd2SAndreas Sandberg 	unsigned long mmun_end;		/* For mmu_notifiers */
3223e8569dd2SAndreas Sandberg 	int ret = 0;
32241e8f889bSDavid Gibson 
32251e8f889bSDavid Gibson 	cow = (vma->vm_flags & (VM_SHARED | VM_MAYWRITE)) == VM_MAYWRITE;
322663551ae0SDavid Gibson 
3227e8569dd2SAndreas Sandberg 	mmun_start = vma->vm_start;
3228e8569dd2SAndreas Sandberg 	mmun_end = vma->vm_end;
3229e8569dd2SAndreas Sandberg 	if (cow)
3230e8569dd2SAndreas Sandberg 		mmu_notifier_invalidate_range_start(src, mmun_start, mmun_end);
3231e8569dd2SAndreas Sandberg 
3232a5516438SAndi Kleen 	for (addr = vma->vm_start; addr < vma->vm_end; addr += sz) {
3233cb900f41SKirill A. Shutemov 		spinlock_t *src_ptl, *dst_ptl;
3234c74df32cSHugh Dickins 		src_pte = huge_pte_offset(src, addr);
3235c74df32cSHugh Dickins 		if (!src_pte)
3236c74df32cSHugh Dickins 			continue;
3237a5516438SAndi Kleen 		dst_pte = huge_pte_alloc(dst, addr, sz);
3238e8569dd2SAndreas Sandberg 		if (!dst_pte) {
3239e8569dd2SAndreas Sandberg 			ret = -ENOMEM;
3240e8569dd2SAndreas Sandberg 			break;
3241e8569dd2SAndreas Sandberg 		}
3242c5c99429SLarry Woodman 
3243c5c99429SLarry Woodman 		/* If the pagetables are shared don't copy or take references */
3244c5c99429SLarry Woodman 		if (dst_pte == src_pte)
3245c5c99429SLarry Woodman 			continue;
3246c5c99429SLarry Woodman 
3247cb900f41SKirill A. Shutemov 		dst_ptl = huge_pte_lock(h, dst, dst_pte);
3248cb900f41SKirill A. Shutemov 		src_ptl = huge_pte_lockptr(h, src, src_pte);
3249cb900f41SKirill A. Shutemov 		spin_lock_nested(src_ptl, SINGLE_DEPTH_NESTING);
32504a705fefSNaoya Horiguchi 		entry = huge_ptep_get(src_pte);
32514a705fefSNaoya Horiguchi 		if (huge_pte_none(entry)) { /* skip none entry */
32524a705fefSNaoya Horiguchi 			;
32534a705fefSNaoya Horiguchi 		} else if (unlikely(is_hugetlb_entry_migration(entry) ||
32544a705fefSNaoya Horiguchi 				    is_hugetlb_entry_hwpoisoned(entry))) {
32554a705fefSNaoya Horiguchi 			swp_entry_t swp_entry = pte_to_swp_entry(entry);
32564a705fefSNaoya Horiguchi 
32574a705fefSNaoya Horiguchi 			if (is_write_migration_entry(swp_entry) && cow) {
32584a705fefSNaoya Horiguchi 				/*
32594a705fefSNaoya Horiguchi 				 * COW mappings require pages in both
32604a705fefSNaoya Horiguchi 				 * parent and child to be set to read.
32614a705fefSNaoya Horiguchi 				 */
32624a705fefSNaoya Horiguchi 				make_migration_entry_read(&swp_entry);
32634a705fefSNaoya Horiguchi 				entry = swp_entry_to_pte(swp_entry);
32644a705fefSNaoya Horiguchi 				set_huge_pte_at(src, addr, src_pte, entry);
32654a705fefSNaoya Horiguchi 			}
32664a705fefSNaoya Horiguchi 			set_huge_pte_at(dst, addr, dst_pte, entry);
32674a705fefSNaoya Horiguchi 		} else {
326834ee645eSJoerg Roedel 			if (cow) {
32697f2e9525SGerald Schaefer 				huge_ptep_set_wrprotect(src, addr, src_pte);
327034ee645eSJoerg Roedel 				mmu_notifier_invalidate_range(src, mmun_start,
327134ee645eSJoerg Roedel 								   mmun_end);
327234ee645eSJoerg Roedel 			}
32730253d634SNaoya Horiguchi 			entry = huge_ptep_get(src_pte);
327463551ae0SDavid Gibson 			ptepage = pte_page(entry);
327563551ae0SDavid Gibson 			get_page(ptepage);
327653f9263bSKirill A. Shutemov 			page_dup_rmap(ptepage, true);
327763551ae0SDavid Gibson 			set_huge_pte_at(dst, addr, dst_pte, entry);
32785d317b2bSNaoya Horiguchi 			hugetlb_count_add(pages_per_huge_page(h), dst);
32791c59827dSHugh Dickins 		}
3280cb900f41SKirill A. Shutemov 		spin_unlock(src_ptl);
3281cb900f41SKirill A. Shutemov 		spin_unlock(dst_ptl);
328263551ae0SDavid Gibson 	}
328363551ae0SDavid Gibson 
3284e8569dd2SAndreas Sandberg 	if (cow)
3285e8569dd2SAndreas Sandberg 		mmu_notifier_invalidate_range_end(src, mmun_start, mmun_end);
3286e8569dd2SAndreas Sandberg 
3287e8569dd2SAndreas Sandberg 	return ret;
328863551ae0SDavid Gibson }
328963551ae0SDavid Gibson 
329024669e58SAneesh Kumar K.V void __unmap_hugepage_range(struct mmu_gather *tlb, struct vm_area_struct *vma,
329124669e58SAneesh Kumar K.V 			    unsigned long start, unsigned long end,
329224669e58SAneesh Kumar K.V 			    struct page *ref_page)
329363551ae0SDavid Gibson {
329463551ae0SDavid Gibson 	struct mm_struct *mm = vma->vm_mm;
329563551ae0SDavid Gibson 	unsigned long address;
3296c7546f8fSDavid Gibson 	pte_t *ptep;
329763551ae0SDavid Gibson 	pte_t pte;
3298cb900f41SKirill A. Shutemov 	spinlock_t *ptl;
329963551ae0SDavid Gibson 	struct page *page;
3300a5516438SAndi Kleen 	struct hstate *h = hstate_vma(vma);
3301a5516438SAndi Kleen 	unsigned long sz = huge_page_size(h);
33022ec74c3eSSagi Grimberg 	const unsigned long mmun_start = start;	/* For mmu_notifiers */
33032ec74c3eSSagi Grimberg 	const unsigned long mmun_end   = end;	/* For mmu_notifiers */
3304a5516438SAndi Kleen 
330563551ae0SDavid Gibson 	WARN_ON(!is_vm_hugetlb_page(vma));
3306a5516438SAndi Kleen 	BUG_ON(start & ~huge_page_mask(h));
3307a5516438SAndi Kleen 	BUG_ON(end & ~huge_page_mask(h));
330863551ae0SDavid Gibson 
330907e32661SAneesh Kumar K.V 	/*
331007e32661SAneesh Kumar K.V 	 * This is a hugetlb vma, all the pte entries should point
331107e32661SAneesh Kumar K.V 	 * to huge page.
331207e32661SAneesh Kumar K.V 	 */
331307e32661SAneesh Kumar K.V 	tlb_remove_check_page_size_change(tlb, sz);
331424669e58SAneesh Kumar K.V 	tlb_start_vma(tlb, vma);
33152ec74c3eSSagi Grimberg 	mmu_notifier_invalidate_range_start(mm, mmun_start, mmun_end);
3316569f48b8SHillf Danton 	address = start;
3317569f48b8SHillf Danton 	for (; address < end; address += sz) {
3318c7546f8fSDavid Gibson 		ptep = huge_pte_offset(mm, address);
3319c7546f8fSDavid Gibson 		if (!ptep)
3320c7546f8fSDavid Gibson 			continue;
3321c7546f8fSDavid Gibson 
3322cb900f41SKirill A. Shutemov 		ptl = huge_pte_lock(h, mm, ptep);
332331d49da5SAneesh Kumar K.V 		if (huge_pmd_unshare(mm, &address, ptep)) {
332431d49da5SAneesh Kumar K.V 			spin_unlock(ptl);
332531d49da5SAneesh Kumar K.V 			continue;
332631d49da5SAneesh Kumar K.V 		}
332739dde65cSChen, Kenneth W 
33286629326bSHillf Danton 		pte = huge_ptep_get(ptep);
332931d49da5SAneesh Kumar K.V 		if (huge_pte_none(pte)) {
333031d49da5SAneesh Kumar K.V 			spin_unlock(ptl);
333131d49da5SAneesh Kumar K.V 			continue;
333231d49da5SAneesh Kumar K.V 		}
33336629326bSHillf Danton 
33346629326bSHillf Danton 		/*
33359fbc1f63SNaoya Horiguchi 		 * Migrating hugepage or HWPoisoned hugepage is already
33369fbc1f63SNaoya Horiguchi 		 * unmapped and its refcount is dropped, so just clear pte here.
33376629326bSHillf Danton 		 */
33389fbc1f63SNaoya Horiguchi 		if (unlikely(!pte_present(pte))) {
3339106c992aSGerald Schaefer 			huge_pte_clear(mm, address, ptep);
334031d49da5SAneesh Kumar K.V 			spin_unlock(ptl);
334131d49da5SAneesh Kumar K.V 			continue;
33428c4894c6SNaoya Horiguchi 		}
33436629326bSHillf Danton 
33446629326bSHillf Danton 		page = pte_page(pte);
334504f2cbe3SMel Gorman 		/*
334604f2cbe3SMel Gorman 		 * If a reference page is supplied, it is because a specific
334704f2cbe3SMel Gorman 		 * page is being unmapped, not a range. Ensure the page we
334804f2cbe3SMel Gorman 		 * are about to unmap is the actual page of interest.
334904f2cbe3SMel Gorman 		 */
335004f2cbe3SMel Gorman 		if (ref_page) {
335131d49da5SAneesh Kumar K.V 			if (page != ref_page) {
335231d49da5SAneesh Kumar K.V 				spin_unlock(ptl);
335331d49da5SAneesh Kumar K.V 				continue;
335431d49da5SAneesh Kumar K.V 			}
335504f2cbe3SMel Gorman 			/*
335604f2cbe3SMel Gorman 			 * Mark the VMA as having unmapped its page so that
335704f2cbe3SMel Gorman 			 * future faults in this VMA will fail rather than
335804f2cbe3SMel Gorman 			 * looking like data was lost
335904f2cbe3SMel Gorman 			 */
336004f2cbe3SMel Gorman 			set_vma_resv_flags(vma, HPAGE_RESV_UNMAPPED);
336104f2cbe3SMel Gorman 		}
336204f2cbe3SMel Gorman 
3363c7546f8fSDavid Gibson 		pte = huge_ptep_get_and_clear(mm, address, ptep);
3364b528e4b6SAneesh Kumar K.V 		tlb_remove_huge_tlb_entry(h, tlb, ptep, address);
3365106c992aSGerald Schaefer 		if (huge_pte_dirty(pte))
33666649a386SKen Chen 			set_page_dirty(page);
33679e81130bSHillf Danton 
33685d317b2bSNaoya Horiguchi 		hugetlb_count_sub(pages_per_huge_page(h), mm);
3369d281ee61SKirill A. Shutemov 		page_remove_rmap(page, true);
337031d49da5SAneesh Kumar K.V 
3371cb900f41SKirill A. Shutemov 		spin_unlock(ptl);
3372e77b0852SAneesh Kumar K.V 		tlb_remove_page_size(tlb, page, huge_page_size(h));
337324669e58SAneesh Kumar K.V 		/*
337431d49da5SAneesh Kumar K.V 		 * Bail out after unmapping reference page if supplied
337524669e58SAneesh Kumar K.V 		 */
337631d49da5SAneesh Kumar K.V 		if (ref_page)
337731d49da5SAneesh Kumar K.V 			break;
3378fe1668aeSChen, Kenneth W 	}
33792ec74c3eSSagi Grimberg 	mmu_notifier_invalidate_range_end(mm, mmun_start, mmun_end);
338024669e58SAneesh Kumar K.V 	tlb_end_vma(tlb, vma);
33811da177e4SLinus Torvalds }
338263551ae0SDavid Gibson 
3383d833352aSMel Gorman void __unmap_hugepage_range_final(struct mmu_gather *tlb,
3384d833352aSMel Gorman 			  struct vm_area_struct *vma, unsigned long start,
3385d833352aSMel Gorman 			  unsigned long end, struct page *ref_page)
3386d833352aSMel Gorman {
3387d833352aSMel Gorman 	__unmap_hugepage_range(tlb, vma, start, end, ref_page);
3388d833352aSMel Gorman 
3389d833352aSMel Gorman 	/*
3390d833352aSMel Gorman 	 * Clear this flag so that x86's huge_pmd_share page_table_shareable
3391d833352aSMel Gorman 	 * test will fail on a vma being torn down, and not grab a page table
3392d833352aSMel Gorman 	 * on its way out.  We're lucky that the flag has such an appropriate
3393d833352aSMel Gorman 	 * name, and can in fact be safely cleared here. We could clear it
3394d833352aSMel Gorman 	 * before the __unmap_hugepage_range above, but all that's necessary
3395c8c06efaSDavidlohr Bueso 	 * is to clear it before releasing the i_mmap_rwsem. This works
3396d833352aSMel Gorman 	 * because in the context this is called, the VMA is about to be
3397c8c06efaSDavidlohr Bueso 	 * destroyed and the i_mmap_rwsem is held.
3398d833352aSMel Gorman 	 */
3399d833352aSMel Gorman 	vma->vm_flags &= ~VM_MAYSHARE;
3400d833352aSMel Gorman }
3401d833352aSMel Gorman 
3402502717f4SChen, Kenneth W void unmap_hugepage_range(struct vm_area_struct *vma, unsigned long start,
340304f2cbe3SMel Gorman 			  unsigned long end, struct page *ref_page)
3404502717f4SChen, Kenneth W {
340524669e58SAneesh Kumar K.V 	struct mm_struct *mm;
340624669e58SAneesh Kumar K.V 	struct mmu_gather tlb;
340724669e58SAneesh Kumar K.V 
340824669e58SAneesh Kumar K.V 	mm = vma->vm_mm;
340924669e58SAneesh Kumar K.V 
34102b047252SLinus Torvalds 	tlb_gather_mmu(&tlb, mm, start, end);
341124669e58SAneesh Kumar K.V 	__unmap_hugepage_range(&tlb, vma, start, end, ref_page);
341224669e58SAneesh Kumar K.V 	tlb_finish_mmu(&tlb, start, end);
3413502717f4SChen, Kenneth W }
3414502717f4SChen, Kenneth W 
341504f2cbe3SMel Gorman /*
341604f2cbe3SMel Gorman  * This is called when the original mapper is failing to COW a MAP_PRIVATE
341704f2cbe3SMel Gorman  * mappping it owns the reserve page for. The intention is to unmap the page
341804f2cbe3SMel Gorman  * from other VMAs and let the children be SIGKILLed if they are faulting the
341904f2cbe3SMel Gorman  * same region.
342004f2cbe3SMel Gorman  */
34212f4612afSDavidlohr Bueso static void unmap_ref_private(struct mm_struct *mm, struct vm_area_struct *vma,
34222a4b3dedSHarvey Harrison 			      struct page *page, unsigned long address)
342304f2cbe3SMel Gorman {
34247526674dSAdam Litke 	struct hstate *h = hstate_vma(vma);
342504f2cbe3SMel Gorman 	struct vm_area_struct *iter_vma;
342604f2cbe3SMel Gorman 	struct address_space *mapping;
342704f2cbe3SMel Gorman 	pgoff_t pgoff;
342804f2cbe3SMel Gorman 
342904f2cbe3SMel Gorman 	/*
343004f2cbe3SMel Gorman 	 * vm_pgoff is in PAGE_SIZE units, hence the different calculation
343104f2cbe3SMel Gorman 	 * from page cache lookup which is in HPAGE_SIZE units.
343204f2cbe3SMel Gorman 	 */
34337526674dSAdam Litke 	address = address & huge_page_mask(h);
343436e4f20aSMichal Hocko 	pgoff = ((address - vma->vm_start) >> PAGE_SHIFT) +
343536e4f20aSMichal Hocko 			vma->vm_pgoff;
343693c76a3dSAl Viro 	mapping = vma->vm_file->f_mapping;
343704f2cbe3SMel Gorman 
34384eb2b1dcSMel Gorman 	/*
34394eb2b1dcSMel Gorman 	 * Take the mapping lock for the duration of the table walk. As
34404eb2b1dcSMel Gorman 	 * this mapping should be shared between all the VMAs,
34414eb2b1dcSMel Gorman 	 * __unmap_hugepage_range() is called as the lock is already held
34424eb2b1dcSMel Gorman 	 */
344383cde9e8SDavidlohr Bueso 	i_mmap_lock_write(mapping);
34446b2dbba8SMichel Lespinasse 	vma_interval_tree_foreach(iter_vma, &mapping->i_mmap, pgoff, pgoff) {
344504f2cbe3SMel Gorman 		/* Do not unmap the current VMA */
344604f2cbe3SMel Gorman 		if (iter_vma == vma)
344704f2cbe3SMel Gorman 			continue;
344804f2cbe3SMel Gorman 
344904f2cbe3SMel Gorman 		/*
34502f84a899SMel Gorman 		 * Shared VMAs have their own reserves and do not affect
34512f84a899SMel Gorman 		 * MAP_PRIVATE accounting but it is possible that a shared
34522f84a899SMel Gorman 		 * VMA is using the same page so check and skip such VMAs.
34532f84a899SMel Gorman 		 */
34542f84a899SMel Gorman 		if (iter_vma->vm_flags & VM_MAYSHARE)
34552f84a899SMel Gorman 			continue;
34562f84a899SMel Gorman 
34572f84a899SMel Gorman 		/*
345804f2cbe3SMel Gorman 		 * Unmap the page from other VMAs without their own reserves.
345904f2cbe3SMel Gorman 		 * They get marked to be SIGKILLed if they fault in these
346004f2cbe3SMel Gorman 		 * areas. This is because a future no-page fault on this VMA
346104f2cbe3SMel Gorman 		 * could insert a zeroed page instead of the data existing
346204f2cbe3SMel Gorman 		 * from the time of fork. This would look like data corruption
346304f2cbe3SMel Gorman 		 */
346404f2cbe3SMel Gorman 		if (!is_vma_resv_set(iter_vma, HPAGE_RESV_OWNER))
346524669e58SAneesh Kumar K.V 			unmap_hugepage_range(iter_vma, address,
346624669e58SAneesh Kumar K.V 					     address + huge_page_size(h), page);
346704f2cbe3SMel Gorman 	}
346883cde9e8SDavidlohr Bueso 	i_mmap_unlock_write(mapping);
346904f2cbe3SMel Gorman }
347004f2cbe3SMel Gorman 
34710fe6e20bSNaoya Horiguchi /*
34720fe6e20bSNaoya Horiguchi  * Hugetlb_cow() should be called with page lock of the original hugepage held.
3473ef009b25SMichal Hocko  * Called with hugetlb_instantiation_mutex held and pte_page locked so we
3474ef009b25SMichal Hocko  * cannot race with other handlers or page migration.
3475ef009b25SMichal Hocko  * Keep the pte_same checks anyway to make transition from the mutex easier.
34760fe6e20bSNaoya Horiguchi  */
34771e8f889bSDavid Gibson static int hugetlb_cow(struct mm_struct *mm, struct vm_area_struct *vma,
34783999f52eSAneesh Kumar K.V 		       unsigned long address, pte_t *ptep,
3479cb900f41SKirill A. Shutemov 		       struct page *pagecache_page, spinlock_t *ptl)
34801e8f889bSDavid Gibson {
34813999f52eSAneesh Kumar K.V 	pte_t pte;
3482a5516438SAndi Kleen 	struct hstate *h = hstate_vma(vma);
34831e8f889bSDavid Gibson 	struct page *old_page, *new_page;
3484ad4404a2SDavidlohr Bueso 	int ret = 0, outside_reserve = 0;
34852ec74c3eSSagi Grimberg 	unsigned long mmun_start;	/* For mmu_notifiers */
34862ec74c3eSSagi Grimberg 	unsigned long mmun_end;		/* For mmu_notifiers */
34871e8f889bSDavid Gibson 
34883999f52eSAneesh Kumar K.V 	pte = huge_ptep_get(ptep);
34891e8f889bSDavid Gibson 	old_page = pte_page(pte);
34901e8f889bSDavid Gibson 
349104f2cbe3SMel Gorman retry_avoidcopy:
34921e8f889bSDavid Gibson 	/* If no-one else is actually using this page, avoid the copy
34931e8f889bSDavid Gibson 	 * and just make the page writable */
349437a2140dSJoonsoo Kim 	if (page_mapcount(old_page) == 1 && PageAnon(old_page)) {
34955a49973dSHugh Dickins 		page_move_anon_rmap(old_page, vma);
34961e8f889bSDavid Gibson 		set_huge_ptep_writable(vma, address, ptep);
349783c54070SNick Piggin 		return 0;
34981e8f889bSDavid Gibson 	}
34991e8f889bSDavid Gibson 
350004f2cbe3SMel Gorman 	/*
350104f2cbe3SMel Gorman 	 * If the process that created a MAP_PRIVATE mapping is about to
350204f2cbe3SMel Gorman 	 * perform a COW due to a shared page count, attempt to satisfy
350304f2cbe3SMel Gorman 	 * the allocation without using the existing reserves. The pagecache
350404f2cbe3SMel Gorman 	 * page is used to determine if the reserve at this address was
350504f2cbe3SMel Gorman 	 * consumed or not. If reserves were used, a partial faulted mapping
350604f2cbe3SMel Gorman 	 * at the time of fork() could consume its reserves on COW instead
350704f2cbe3SMel Gorman 	 * of the full address range.
350804f2cbe3SMel Gorman 	 */
35095944d011SJoonsoo Kim 	if (is_vma_resv_set(vma, HPAGE_RESV_OWNER) &&
351004f2cbe3SMel Gorman 			old_page != pagecache_page)
351104f2cbe3SMel Gorman 		outside_reserve = 1;
351204f2cbe3SMel Gorman 
351309cbfeafSKirill A. Shutemov 	get_page(old_page);
3514b76c8cfbSLarry Woodman 
3515ad4404a2SDavidlohr Bueso 	/*
3516ad4404a2SDavidlohr Bueso 	 * Drop page table lock as buddy allocator may be called. It will
3517ad4404a2SDavidlohr Bueso 	 * be acquired again before returning to the caller, as expected.
3518ad4404a2SDavidlohr Bueso 	 */
3519cb900f41SKirill A. Shutemov 	spin_unlock(ptl);
352004f2cbe3SMel Gorman 	new_page = alloc_huge_page(vma, address, outside_reserve);
35211e8f889bSDavid Gibson 
35222fc39cecSAdam Litke 	if (IS_ERR(new_page)) {
352304f2cbe3SMel Gorman 		/*
352404f2cbe3SMel Gorman 		 * If a process owning a MAP_PRIVATE mapping fails to COW,
352504f2cbe3SMel Gorman 		 * it is due to references held by a child and an insufficient
352604f2cbe3SMel Gorman 		 * huge page pool. To guarantee the original mappers
352704f2cbe3SMel Gorman 		 * reliability, unmap the page from child processes. The child
352804f2cbe3SMel Gorman 		 * may get SIGKILLed if it later faults.
352904f2cbe3SMel Gorman 		 */
353004f2cbe3SMel Gorman 		if (outside_reserve) {
353109cbfeafSKirill A. Shutemov 			put_page(old_page);
353204f2cbe3SMel Gorman 			BUG_ON(huge_pte_none(pte));
35332f4612afSDavidlohr Bueso 			unmap_ref_private(mm, vma, old_page, address);
353404f2cbe3SMel Gorman 			BUG_ON(huge_pte_none(pte));
3535cb900f41SKirill A. Shutemov 			spin_lock(ptl);
3536a734bcc8SHillf Danton 			ptep = huge_pte_offset(mm, address & huge_page_mask(h));
3537a9af0c5dSNaoya Horiguchi 			if (likely(ptep &&
3538a9af0c5dSNaoya Horiguchi 				   pte_same(huge_ptep_get(ptep), pte)))
353904f2cbe3SMel Gorman 				goto retry_avoidcopy;
3540a734bcc8SHillf Danton 			/*
3541cb900f41SKirill A. Shutemov 			 * race occurs while re-acquiring page table
3542cb900f41SKirill A. Shutemov 			 * lock, and our job is done.
3543a734bcc8SHillf Danton 			 */
3544a734bcc8SHillf Danton 			return 0;
354504f2cbe3SMel Gorman 		}
354604f2cbe3SMel Gorman 
3547ad4404a2SDavidlohr Bueso 		ret = (PTR_ERR(new_page) == -ENOMEM) ?
3548ad4404a2SDavidlohr Bueso 			VM_FAULT_OOM : VM_FAULT_SIGBUS;
3549ad4404a2SDavidlohr Bueso 		goto out_release_old;
35501e8f889bSDavid Gibson 	}
35511e8f889bSDavid Gibson 
35520fe6e20bSNaoya Horiguchi 	/*
35530fe6e20bSNaoya Horiguchi 	 * When the original hugepage is shared one, it does not have
35540fe6e20bSNaoya Horiguchi 	 * anon_vma prepared.
35550fe6e20bSNaoya Horiguchi 	 */
355644e2aa93SDean Nelson 	if (unlikely(anon_vma_prepare(vma))) {
3557ad4404a2SDavidlohr Bueso 		ret = VM_FAULT_OOM;
3558ad4404a2SDavidlohr Bueso 		goto out_release_all;
355944e2aa93SDean Nelson 	}
35600fe6e20bSNaoya Horiguchi 
356147ad8475SAndrea Arcangeli 	copy_user_huge_page(new_page, old_page, address, vma,
356247ad8475SAndrea Arcangeli 			    pages_per_huge_page(h));
35630ed361deSNick Piggin 	__SetPageUptodate(new_page);
3564bcc54222SNaoya Horiguchi 	set_page_huge_active(new_page);
35651e8f889bSDavid Gibson 
35662ec74c3eSSagi Grimberg 	mmun_start = address & huge_page_mask(h);
35672ec74c3eSSagi Grimberg 	mmun_end = mmun_start + huge_page_size(h);
35682ec74c3eSSagi Grimberg 	mmu_notifier_invalidate_range_start(mm, mmun_start, mmun_end);
3569ad4404a2SDavidlohr Bueso 
3570b76c8cfbSLarry Woodman 	/*
3571cb900f41SKirill A. Shutemov 	 * Retake the page table lock to check for racing updates
3572b76c8cfbSLarry Woodman 	 * before the page tables are altered
3573b76c8cfbSLarry Woodman 	 */
3574cb900f41SKirill A. Shutemov 	spin_lock(ptl);
3575a5516438SAndi Kleen 	ptep = huge_pte_offset(mm, address & huge_page_mask(h));
3576a9af0c5dSNaoya Horiguchi 	if (likely(ptep && pte_same(huge_ptep_get(ptep), pte))) {
357707443a85SJoonsoo Kim 		ClearPagePrivate(new_page);
357807443a85SJoonsoo Kim 
35791e8f889bSDavid Gibson 		/* Break COW */
35808fe627ecSGerald Schaefer 		huge_ptep_clear_flush(vma, address, ptep);
358134ee645eSJoerg Roedel 		mmu_notifier_invalidate_range(mm, mmun_start, mmun_end);
35821e8f889bSDavid Gibson 		set_huge_pte_at(mm, address, ptep,
35831e8f889bSDavid Gibson 				make_huge_pte(vma, new_page, 1));
3584d281ee61SKirill A. Shutemov 		page_remove_rmap(old_page, true);
3585cd67f0d2SNaoya Horiguchi 		hugepage_add_new_anon_rmap(new_page, vma, address);
35861e8f889bSDavid Gibson 		/* Make the old page be freed below */
35871e8f889bSDavid Gibson 		new_page = old_page;
35881e8f889bSDavid Gibson 	}
3589cb900f41SKirill A. Shutemov 	spin_unlock(ptl);
35902ec74c3eSSagi Grimberg 	mmu_notifier_invalidate_range_end(mm, mmun_start, mmun_end);
3591ad4404a2SDavidlohr Bueso out_release_all:
359296b96a96SMike Kravetz 	restore_reserve_on_error(h, vma, address, new_page);
359309cbfeafSKirill A. Shutemov 	put_page(new_page);
3594ad4404a2SDavidlohr Bueso out_release_old:
359509cbfeafSKirill A. Shutemov 	put_page(old_page);
35968312034fSJoonsoo Kim 
3597ad4404a2SDavidlohr Bueso 	spin_lock(ptl); /* Caller expects lock to be held */
3598ad4404a2SDavidlohr Bueso 	return ret;
35991e8f889bSDavid Gibson }
36001e8f889bSDavid Gibson 
360104f2cbe3SMel Gorman /* Return the pagecache page at a given address within a VMA */
3602a5516438SAndi Kleen static struct page *hugetlbfs_pagecache_page(struct hstate *h,
3603a5516438SAndi Kleen 			struct vm_area_struct *vma, unsigned long address)
360404f2cbe3SMel Gorman {
360504f2cbe3SMel Gorman 	struct address_space *mapping;
3606e7c4b0bfSAndy Whitcroft 	pgoff_t idx;
360704f2cbe3SMel Gorman 
360804f2cbe3SMel Gorman 	mapping = vma->vm_file->f_mapping;
3609a5516438SAndi Kleen 	idx = vma_hugecache_offset(h, vma, address);
361004f2cbe3SMel Gorman 
361104f2cbe3SMel Gorman 	return find_lock_page(mapping, idx);
361204f2cbe3SMel Gorman }
361304f2cbe3SMel Gorman 
36143ae77f43SHugh Dickins /*
36153ae77f43SHugh Dickins  * Return whether there is a pagecache page to back given address within VMA.
36163ae77f43SHugh Dickins  * Caller follow_hugetlb_page() holds page_table_lock so we cannot lock_page.
36173ae77f43SHugh Dickins  */
36183ae77f43SHugh Dickins static bool hugetlbfs_pagecache_present(struct hstate *h,
36192a15efc9SHugh Dickins 			struct vm_area_struct *vma, unsigned long address)
36202a15efc9SHugh Dickins {
36212a15efc9SHugh Dickins 	struct address_space *mapping;
36222a15efc9SHugh Dickins 	pgoff_t idx;
36232a15efc9SHugh Dickins 	struct page *page;
36242a15efc9SHugh Dickins 
36252a15efc9SHugh Dickins 	mapping = vma->vm_file->f_mapping;
36262a15efc9SHugh Dickins 	idx = vma_hugecache_offset(h, vma, address);
36272a15efc9SHugh Dickins 
36282a15efc9SHugh Dickins 	page = find_get_page(mapping, idx);
36292a15efc9SHugh Dickins 	if (page)
36302a15efc9SHugh Dickins 		put_page(page);
36312a15efc9SHugh Dickins 	return page != NULL;
36322a15efc9SHugh Dickins }
36332a15efc9SHugh Dickins 
3634ab76ad54SMike Kravetz int huge_add_to_page_cache(struct page *page, struct address_space *mapping,
3635ab76ad54SMike Kravetz 			   pgoff_t idx)
3636ab76ad54SMike Kravetz {
3637ab76ad54SMike Kravetz 	struct inode *inode = mapping->host;
3638ab76ad54SMike Kravetz 	struct hstate *h = hstate_inode(inode);
3639ab76ad54SMike Kravetz 	int err = add_to_page_cache(page, mapping, idx, GFP_KERNEL);
3640ab76ad54SMike Kravetz 
3641ab76ad54SMike Kravetz 	if (err)
3642ab76ad54SMike Kravetz 		return err;
3643ab76ad54SMike Kravetz 	ClearPagePrivate(page);
3644ab76ad54SMike Kravetz 
3645ab76ad54SMike Kravetz 	spin_lock(&inode->i_lock);
3646ab76ad54SMike Kravetz 	inode->i_blocks += blocks_per_huge_page(h);
3647ab76ad54SMike Kravetz 	spin_unlock(&inode->i_lock);
3648ab76ad54SMike Kravetz 	return 0;
3649ab76ad54SMike Kravetz }
3650ab76ad54SMike Kravetz 
3651a1ed3ddaSRobert P. J. Day static int hugetlb_no_page(struct mm_struct *mm, struct vm_area_struct *vma,
36528382d914SDavidlohr Bueso 			   struct address_space *mapping, pgoff_t idx,
3653788c7df4SHugh Dickins 			   unsigned long address, pte_t *ptep, unsigned int flags)
3654ac9b9c66SHugh Dickins {
3655a5516438SAndi Kleen 	struct hstate *h = hstate_vma(vma);
3656ac9b9c66SHugh Dickins 	int ret = VM_FAULT_SIGBUS;
3657409eb8c2SHillf Danton 	int anon_rmap = 0;
36584c887265SAdam Litke 	unsigned long size;
36594c887265SAdam Litke 	struct page *page;
36601e8f889bSDavid Gibson 	pte_t new_pte;
3661cb900f41SKirill A. Shutemov 	spinlock_t *ptl;
36624c887265SAdam Litke 
366304f2cbe3SMel Gorman 	/*
366404f2cbe3SMel Gorman 	 * Currently, we are forced to kill the process in the event the
366504f2cbe3SMel Gorman 	 * original mapper has unmapped pages from the child due to a failed
366625985edcSLucas De Marchi 	 * COW. Warn that such a situation has occurred as it may not be obvious
366704f2cbe3SMel Gorman 	 */
366804f2cbe3SMel Gorman 	if (is_vma_resv_set(vma, HPAGE_RESV_UNMAPPED)) {
3669910154d5SGeoffrey Thomas 		pr_warn_ratelimited("PID %d killed due to inadequate hugepage pool\n",
367004f2cbe3SMel Gorman 			   current->pid);
367104f2cbe3SMel Gorman 		return ret;
367204f2cbe3SMel Gorman 	}
367304f2cbe3SMel Gorman 
36744c887265SAdam Litke 	/*
36754c887265SAdam Litke 	 * Use page lock to guard against racing truncation
36764c887265SAdam Litke 	 * before we get page_table_lock.
36774c887265SAdam Litke 	 */
36786bda666aSChristoph Lameter retry:
36796bda666aSChristoph Lameter 	page = find_lock_page(mapping, idx);
36806bda666aSChristoph Lameter 	if (!page) {
3681a5516438SAndi Kleen 		size = i_size_read(mapping->host) >> huge_page_shift(h);
3682ebed4bfcSHugh Dickins 		if (idx >= size)
3683ebed4bfcSHugh Dickins 			goto out;
36841a1aad8aSMike Kravetz 
36851a1aad8aSMike Kravetz 		/*
36861a1aad8aSMike Kravetz 		 * Check for page in userfault range
36871a1aad8aSMike Kravetz 		 */
36881a1aad8aSMike Kravetz 		if (userfaultfd_missing(vma)) {
36891a1aad8aSMike Kravetz 			u32 hash;
36901a1aad8aSMike Kravetz 			struct vm_fault vmf = {
36911a1aad8aSMike Kravetz 				.vma = vma,
36921a1aad8aSMike Kravetz 				.address = address,
36931a1aad8aSMike Kravetz 				.flags = flags,
36941a1aad8aSMike Kravetz 				/*
36951a1aad8aSMike Kravetz 				 * Hard to debug if it ends up being
36961a1aad8aSMike Kravetz 				 * used by a callee that assumes
36971a1aad8aSMike Kravetz 				 * something about the other
36981a1aad8aSMike Kravetz 				 * uninitialized fields... same as in
36991a1aad8aSMike Kravetz 				 * memory.c
37001a1aad8aSMike Kravetz 				 */
37011a1aad8aSMike Kravetz 			};
37021a1aad8aSMike Kravetz 
37031a1aad8aSMike Kravetz 			/*
37041a1aad8aSMike Kravetz 			 * hugetlb_fault_mutex must be dropped before
37051a1aad8aSMike Kravetz 			 * handling userfault.  Reacquire after handling
37061a1aad8aSMike Kravetz 			 * fault to make calling code simpler.
37071a1aad8aSMike Kravetz 			 */
37081a1aad8aSMike Kravetz 			hash = hugetlb_fault_mutex_hash(h, mm, vma, mapping,
37091a1aad8aSMike Kravetz 							idx, address);
37101a1aad8aSMike Kravetz 			mutex_unlock(&hugetlb_fault_mutex_table[hash]);
37111a1aad8aSMike Kravetz 			ret = handle_userfault(&vmf, VM_UFFD_MISSING);
37121a1aad8aSMike Kravetz 			mutex_lock(&hugetlb_fault_mutex_table[hash]);
37131a1aad8aSMike Kravetz 			goto out;
37141a1aad8aSMike Kravetz 		}
37151a1aad8aSMike Kravetz 
371604f2cbe3SMel Gorman 		page = alloc_huge_page(vma, address, 0);
37172fc39cecSAdam Litke 		if (IS_ERR(page)) {
371876dcee75SAneesh Kumar K.V 			ret = PTR_ERR(page);
371976dcee75SAneesh Kumar K.V 			if (ret == -ENOMEM)
372076dcee75SAneesh Kumar K.V 				ret = VM_FAULT_OOM;
372176dcee75SAneesh Kumar K.V 			else
372276dcee75SAneesh Kumar K.V 				ret = VM_FAULT_SIGBUS;
37236bda666aSChristoph Lameter 			goto out;
37246bda666aSChristoph Lameter 		}
372547ad8475SAndrea Arcangeli 		clear_huge_page(page, address, pages_per_huge_page(h));
37260ed361deSNick Piggin 		__SetPageUptodate(page);
3727bcc54222SNaoya Horiguchi 		set_page_huge_active(page);
3728ac9b9c66SHugh Dickins 
3729f83a275dSMel Gorman 		if (vma->vm_flags & VM_MAYSHARE) {
3730ab76ad54SMike Kravetz 			int err = huge_add_to_page_cache(page, mapping, idx);
37316bda666aSChristoph Lameter 			if (err) {
37326bda666aSChristoph Lameter 				put_page(page);
37336bda666aSChristoph Lameter 				if (err == -EEXIST)
37346bda666aSChristoph Lameter 					goto retry;
37356bda666aSChristoph Lameter 				goto out;
37366bda666aSChristoph Lameter 			}
373723be7468SMel Gorman 		} else {
37386bda666aSChristoph Lameter 			lock_page(page);
37390fe6e20bSNaoya Horiguchi 			if (unlikely(anon_vma_prepare(vma))) {
37400fe6e20bSNaoya Horiguchi 				ret = VM_FAULT_OOM;
37410fe6e20bSNaoya Horiguchi 				goto backout_unlocked;
374223be7468SMel Gorman 			}
3743409eb8c2SHillf Danton 			anon_rmap = 1;
37440fe6e20bSNaoya Horiguchi 		}
37450fe6e20bSNaoya Horiguchi 	} else {
374657303d80SAndy Whitcroft 		/*
3747998b4382SNaoya Horiguchi 		 * If memory error occurs between mmap() and fault, some process
3748998b4382SNaoya Horiguchi 		 * don't have hwpoisoned swap entry for errored virtual address.
3749998b4382SNaoya Horiguchi 		 * So we need to block hugepage fault by PG_hwpoison bit check.
3750fd6a03edSNaoya Horiguchi 		 */
3751fd6a03edSNaoya Horiguchi 		if (unlikely(PageHWPoison(page))) {
3752aa50d3a7SAndi Kleen 			ret = VM_FAULT_HWPOISON |
3753972dc4deSAneesh Kumar K.V 				VM_FAULT_SET_HINDEX(hstate_index(h));
3754fd6a03edSNaoya Horiguchi 			goto backout_unlocked;
37556bda666aSChristoph Lameter 		}
3756998b4382SNaoya Horiguchi 	}
37571e8f889bSDavid Gibson 
375857303d80SAndy Whitcroft 	/*
375957303d80SAndy Whitcroft 	 * If we are going to COW a private mapping later, we examine the
376057303d80SAndy Whitcroft 	 * pending reservations for this page now. This will ensure that
376157303d80SAndy Whitcroft 	 * any allocations necessary to record that reservation occur outside
376257303d80SAndy Whitcroft 	 * the spinlock.
376357303d80SAndy Whitcroft 	 */
37645e911373SMike Kravetz 	if ((flags & FAULT_FLAG_WRITE) && !(vma->vm_flags & VM_SHARED)) {
37652b26736cSAndy Whitcroft 		if (vma_needs_reservation(h, vma, address) < 0) {
37662b26736cSAndy Whitcroft 			ret = VM_FAULT_OOM;
37672b26736cSAndy Whitcroft 			goto backout_unlocked;
37682b26736cSAndy Whitcroft 		}
37695e911373SMike Kravetz 		/* Just decrements count, does not deallocate */
3770feba16e2SMike Kravetz 		vma_end_reservation(h, vma, address);
37715e911373SMike Kravetz 	}
377257303d80SAndy Whitcroft 
37738bea8052SAneesh Kumar K.V 	ptl = huge_pte_lock(h, mm, ptep);
3774a5516438SAndi Kleen 	size = i_size_read(mapping->host) >> huge_page_shift(h);
37754c887265SAdam Litke 	if (idx >= size)
37764c887265SAdam Litke 		goto backout;
37774c887265SAdam Litke 
377883c54070SNick Piggin 	ret = 0;
37797f2e9525SGerald Schaefer 	if (!huge_pte_none(huge_ptep_get(ptep)))
37804c887265SAdam Litke 		goto backout;
37814c887265SAdam Litke 
378207443a85SJoonsoo Kim 	if (anon_rmap) {
378307443a85SJoonsoo Kim 		ClearPagePrivate(page);
3784409eb8c2SHillf Danton 		hugepage_add_new_anon_rmap(page, vma, address);
3785ac714904SChoi Gi-yong 	} else
378653f9263bSKirill A. Shutemov 		page_dup_rmap(page, true);
37871e8f889bSDavid Gibson 	new_pte = make_huge_pte(vma, page, ((vma->vm_flags & VM_WRITE)
37881e8f889bSDavid Gibson 				&& (vma->vm_flags & VM_SHARED)));
37891e8f889bSDavid Gibson 	set_huge_pte_at(mm, address, ptep, new_pte);
37901e8f889bSDavid Gibson 
37915d317b2bSNaoya Horiguchi 	hugetlb_count_add(pages_per_huge_page(h), mm);
3792788c7df4SHugh Dickins 	if ((flags & FAULT_FLAG_WRITE) && !(vma->vm_flags & VM_SHARED)) {
37931e8f889bSDavid Gibson 		/* Optimization, do the COW without a second fault */
37943999f52eSAneesh Kumar K.V 		ret = hugetlb_cow(mm, vma, address, ptep, page, ptl);
37951e8f889bSDavid Gibson 	}
37961e8f889bSDavid Gibson 
3797cb900f41SKirill A. Shutemov 	spin_unlock(ptl);
37984c887265SAdam Litke 	unlock_page(page);
37994c887265SAdam Litke out:
3800ac9b9c66SHugh Dickins 	return ret;
38014c887265SAdam Litke 
38024c887265SAdam Litke backout:
3803cb900f41SKirill A. Shutemov 	spin_unlock(ptl);
38042b26736cSAndy Whitcroft backout_unlocked:
38054c887265SAdam Litke 	unlock_page(page);
380696b96a96SMike Kravetz 	restore_reserve_on_error(h, vma, address, page);
38074c887265SAdam Litke 	put_page(page);
38084c887265SAdam Litke 	goto out;
3809ac9b9c66SHugh Dickins }
3810ac9b9c66SHugh Dickins 
38118382d914SDavidlohr Bueso #ifdef CONFIG_SMP
3812c672c7f2SMike Kravetz u32 hugetlb_fault_mutex_hash(struct hstate *h, struct mm_struct *mm,
38138382d914SDavidlohr Bueso 			    struct vm_area_struct *vma,
38148382d914SDavidlohr Bueso 			    struct address_space *mapping,
38158382d914SDavidlohr Bueso 			    pgoff_t idx, unsigned long address)
38168382d914SDavidlohr Bueso {
38178382d914SDavidlohr Bueso 	unsigned long key[2];
38188382d914SDavidlohr Bueso 	u32 hash;
38198382d914SDavidlohr Bueso 
38208382d914SDavidlohr Bueso 	if (vma->vm_flags & VM_SHARED) {
38218382d914SDavidlohr Bueso 		key[0] = (unsigned long) mapping;
38228382d914SDavidlohr Bueso 		key[1] = idx;
38238382d914SDavidlohr Bueso 	} else {
38248382d914SDavidlohr Bueso 		key[0] = (unsigned long) mm;
38258382d914SDavidlohr Bueso 		key[1] = address >> huge_page_shift(h);
38268382d914SDavidlohr Bueso 	}
38278382d914SDavidlohr Bueso 
38288382d914SDavidlohr Bueso 	hash = jhash2((u32 *)&key, sizeof(key)/sizeof(u32), 0);
38298382d914SDavidlohr Bueso 
38308382d914SDavidlohr Bueso 	return hash & (num_fault_mutexes - 1);
38318382d914SDavidlohr Bueso }
38328382d914SDavidlohr Bueso #else
38338382d914SDavidlohr Bueso /*
38348382d914SDavidlohr Bueso  * For uniprocesor systems we always use a single mutex, so just
38358382d914SDavidlohr Bueso  * return 0 and avoid the hashing overhead.
38368382d914SDavidlohr Bueso  */
3837c672c7f2SMike Kravetz u32 hugetlb_fault_mutex_hash(struct hstate *h, struct mm_struct *mm,
38388382d914SDavidlohr Bueso 			    struct vm_area_struct *vma,
38398382d914SDavidlohr Bueso 			    struct address_space *mapping,
38408382d914SDavidlohr Bueso 			    pgoff_t idx, unsigned long address)
38418382d914SDavidlohr Bueso {
38428382d914SDavidlohr Bueso 	return 0;
38438382d914SDavidlohr Bueso }
38448382d914SDavidlohr Bueso #endif
38458382d914SDavidlohr Bueso 
384686e5216fSAdam Litke int hugetlb_fault(struct mm_struct *mm, struct vm_area_struct *vma,
3847788c7df4SHugh Dickins 			unsigned long address, unsigned int flags)
384886e5216fSAdam Litke {
38498382d914SDavidlohr Bueso 	pte_t *ptep, entry;
3850cb900f41SKirill A. Shutemov 	spinlock_t *ptl;
38511e8f889bSDavid Gibson 	int ret;
38528382d914SDavidlohr Bueso 	u32 hash;
38538382d914SDavidlohr Bueso 	pgoff_t idx;
38540fe6e20bSNaoya Horiguchi 	struct page *page = NULL;
385557303d80SAndy Whitcroft 	struct page *pagecache_page = NULL;
3856a5516438SAndi Kleen 	struct hstate *h = hstate_vma(vma);
38578382d914SDavidlohr Bueso 	struct address_space *mapping;
38580f792cf9SNaoya Horiguchi 	int need_wait_lock = 0;
385986e5216fSAdam Litke 
38601e16a539SKAMEZAWA Hiroyuki 	address &= huge_page_mask(h);
38611e16a539SKAMEZAWA Hiroyuki 
3862fd6a03edSNaoya Horiguchi 	ptep = huge_pte_offset(mm, address);
3863fd6a03edSNaoya Horiguchi 	if (ptep) {
3864fd6a03edSNaoya Horiguchi 		entry = huge_ptep_get(ptep);
3865290408d4SNaoya Horiguchi 		if (unlikely(is_hugetlb_entry_migration(entry))) {
3866cb900f41SKirill A. Shutemov 			migration_entry_wait_huge(vma, mm, ptep);
3867290408d4SNaoya Horiguchi 			return 0;
3868290408d4SNaoya Horiguchi 		} else if (unlikely(is_hugetlb_entry_hwpoisoned(entry)))
3869aa50d3a7SAndi Kleen 			return VM_FAULT_HWPOISON_LARGE |
3870972dc4deSAneesh Kumar K.V 				VM_FAULT_SET_HINDEX(hstate_index(h));
38710d777df5SNaoya Horiguchi 	} else {
3872a5516438SAndi Kleen 		ptep = huge_pte_alloc(mm, address, huge_page_size(h));
387386e5216fSAdam Litke 		if (!ptep)
387486e5216fSAdam Litke 			return VM_FAULT_OOM;
38750d777df5SNaoya Horiguchi 	}
387686e5216fSAdam Litke 
38778382d914SDavidlohr Bueso 	mapping = vma->vm_file->f_mapping;
38788382d914SDavidlohr Bueso 	idx = vma_hugecache_offset(h, vma, address);
38798382d914SDavidlohr Bueso 
38803935baa9SDavid Gibson 	/*
38813935baa9SDavid Gibson 	 * Serialize hugepage allocation and instantiation, so that we don't
38823935baa9SDavid Gibson 	 * get spurious allocation failures if two CPUs race to instantiate
38833935baa9SDavid Gibson 	 * the same page in the page cache.
38843935baa9SDavid Gibson 	 */
3885c672c7f2SMike Kravetz 	hash = hugetlb_fault_mutex_hash(h, mm, vma, mapping, idx, address);
3886c672c7f2SMike Kravetz 	mutex_lock(&hugetlb_fault_mutex_table[hash]);
38878382d914SDavidlohr Bueso 
38887f2e9525SGerald Schaefer 	entry = huge_ptep_get(ptep);
38897f2e9525SGerald Schaefer 	if (huge_pte_none(entry)) {
38908382d914SDavidlohr Bueso 		ret = hugetlb_no_page(mm, vma, mapping, idx, address, ptep, flags);
3891b4d1d99fSDavid Gibson 		goto out_mutex;
38923935baa9SDavid Gibson 	}
389386e5216fSAdam Litke 
389483c54070SNick Piggin 	ret = 0;
38951e8f889bSDavid Gibson 
389657303d80SAndy Whitcroft 	/*
38970f792cf9SNaoya Horiguchi 	 * entry could be a migration/hwpoison entry at this point, so this
38980f792cf9SNaoya Horiguchi 	 * check prevents the kernel from going below assuming that we have
38990f792cf9SNaoya Horiguchi 	 * a active hugepage in pagecache. This goto expects the 2nd page fault,
39000f792cf9SNaoya Horiguchi 	 * and is_hugetlb_entry_(migration|hwpoisoned) check will properly
39010f792cf9SNaoya Horiguchi 	 * handle it.
39020f792cf9SNaoya Horiguchi 	 */
39030f792cf9SNaoya Horiguchi 	if (!pte_present(entry))
39040f792cf9SNaoya Horiguchi 		goto out_mutex;
39050f792cf9SNaoya Horiguchi 
39060f792cf9SNaoya Horiguchi 	/*
390757303d80SAndy Whitcroft 	 * If we are going to COW the mapping later, we examine the pending
390857303d80SAndy Whitcroft 	 * reservations for this page now. This will ensure that any
390957303d80SAndy Whitcroft 	 * allocations necessary to record that reservation occur outside the
391057303d80SAndy Whitcroft 	 * spinlock. For private mappings, we also lookup the pagecache
391157303d80SAndy Whitcroft 	 * page now as it is used to determine if a reservation has been
391257303d80SAndy Whitcroft 	 * consumed.
391357303d80SAndy Whitcroft 	 */
3914106c992aSGerald Schaefer 	if ((flags & FAULT_FLAG_WRITE) && !huge_pte_write(entry)) {
39152b26736cSAndy Whitcroft 		if (vma_needs_reservation(h, vma, address) < 0) {
39162b26736cSAndy Whitcroft 			ret = VM_FAULT_OOM;
3917b4d1d99fSDavid Gibson 			goto out_mutex;
39182b26736cSAndy Whitcroft 		}
39195e911373SMike Kravetz 		/* Just decrements count, does not deallocate */
3920feba16e2SMike Kravetz 		vma_end_reservation(h, vma, address);
392157303d80SAndy Whitcroft 
3922f83a275dSMel Gorman 		if (!(vma->vm_flags & VM_MAYSHARE))
392357303d80SAndy Whitcroft 			pagecache_page = hugetlbfs_pagecache_page(h,
392457303d80SAndy Whitcroft 								vma, address);
392557303d80SAndy Whitcroft 	}
392657303d80SAndy Whitcroft 
39270f792cf9SNaoya Horiguchi 	ptl = huge_pte_lock(h, mm, ptep);
39280fe6e20bSNaoya Horiguchi 
39291e8f889bSDavid Gibson 	/* Check for a racing update before calling hugetlb_cow */
3930b4d1d99fSDavid Gibson 	if (unlikely(!pte_same(entry, huge_ptep_get(ptep))))
3931cb900f41SKirill A. Shutemov 		goto out_ptl;
3932b4d1d99fSDavid Gibson 
39330f792cf9SNaoya Horiguchi 	/*
39340f792cf9SNaoya Horiguchi 	 * hugetlb_cow() requires page locks of pte_page(entry) and
39350f792cf9SNaoya Horiguchi 	 * pagecache_page, so here we need take the former one
39360f792cf9SNaoya Horiguchi 	 * when page != pagecache_page or !pagecache_page.
39370f792cf9SNaoya Horiguchi 	 */
39380f792cf9SNaoya Horiguchi 	page = pte_page(entry);
39390f792cf9SNaoya Horiguchi 	if (page != pagecache_page)
39400f792cf9SNaoya Horiguchi 		if (!trylock_page(page)) {
39410f792cf9SNaoya Horiguchi 			need_wait_lock = 1;
39420f792cf9SNaoya Horiguchi 			goto out_ptl;
39430f792cf9SNaoya Horiguchi 		}
39440f792cf9SNaoya Horiguchi 
39450f792cf9SNaoya Horiguchi 	get_page(page);
3946b4d1d99fSDavid Gibson 
3947788c7df4SHugh Dickins 	if (flags & FAULT_FLAG_WRITE) {
3948106c992aSGerald Schaefer 		if (!huge_pte_write(entry)) {
39493999f52eSAneesh Kumar K.V 			ret = hugetlb_cow(mm, vma, address, ptep,
3950cb900f41SKirill A. Shutemov 					  pagecache_page, ptl);
39510f792cf9SNaoya Horiguchi 			goto out_put_page;
3952b4d1d99fSDavid Gibson 		}
3953106c992aSGerald Schaefer 		entry = huge_pte_mkdirty(entry);
3954b4d1d99fSDavid Gibson 	}
3955b4d1d99fSDavid Gibson 	entry = pte_mkyoung(entry);
3956788c7df4SHugh Dickins 	if (huge_ptep_set_access_flags(vma, address, ptep, entry,
3957788c7df4SHugh Dickins 						flags & FAULT_FLAG_WRITE))
39584b3073e1SRussell King 		update_mmu_cache(vma, address, ptep);
39590f792cf9SNaoya Horiguchi out_put_page:
39600f792cf9SNaoya Horiguchi 	if (page != pagecache_page)
39610f792cf9SNaoya Horiguchi 		unlock_page(page);
39620f792cf9SNaoya Horiguchi 	put_page(page);
3963cb900f41SKirill A. Shutemov out_ptl:
3964cb900f41SKirill A. Shutemov 	spin_unlock(ptl);
396557303d80SAndy Whitcroft 
396657303d80SAndy Whitcroft 	if (pagecache_page) {
396757303d80SAndy Whitcroft 		unlock_page(pagecache_page);
396857303d80SAndy Whitcroft 		put_page(pagecache_page);
396957303d80SAndy Whitcroft 	}
3970b4d1d99fSDavid Gibson out_mutex:
3971c672c7f2SMike Kravetz 	mutex_unlock(&hugetlb_fault_mutex_table[hash]);
39720f792cf9SNaoya Horiguchi 	/*
39730f792cf9SNaoya Horiguchi 	 * Generally it's safe to hold refcount during waiting page lock. But
39740f792cf9SNaoya Horiguchi 	 * here we just wait to defer the next page fault to avoid busy loop and
39750f792cf9SNaoya Horiguchi 	 * the page is not used after unlocked before returning from the current
39760f792cf9SNaoya Horiguchi 	 * page fault. So we are safe from accessing freed page, even if we wait
39770f792cf9SNaoya Horiguchi 	 * here without taking refcount.
39780f792cf9SNaoya Horiguchi 	 */
39790f792cf9SNaoya Horiguchi 	if (need_wait_lock)
39800f792cf9SNaoya Horiguchi 		wait_on_page_locked(page);
39811e8f889bSDavid Gibson 	return ret;
398286e5216fSAdam Litke }
398386e5216fSAdam Litke 
39848fb5debcSMike Kravetz /*
39858fb5debcSMike Kravetz  * Used by userfaultfd UFFDIO_COPY.  Based on mcopy_atomic_pte with
39868fb5debcSMike Kravetz  * modifications for huge pages.
39878fb5debcSMike Kravetz  */
39888fb5debcSMike Kravetz int hugetlb_mcopy_atomic_pte(struct mm_struct *dst_mm,
39898fb5debcSMike Kravetz 			    pte_t *dst_pte,
39908fb5debcSMike Kravetz 			    struct vm_area_struct *dst_vma,
39918fb5debcSMike Kravetz 			    unsigned long dst_addr,
39928fb5debcSMike Kravetz 			    unsigned long src_addr,
39938fb5debcSMike Kravetz 			    struct page **pagep)
39948fb5debcSMike Kravetz {
39958fb5debcSMike Kravetz 	struct hstate *h = hstate_vma(dst_vma);
39968fb5debcSMike Kravetz 	pte_t _dst_pte;
39978fb5debcSMike Kravetz 	spinlock_t *ptl;
39988fb5debcSMike Kravetz 	int ret;
39998fb5debcSMike Kravetz 	struct page *page;
40008fb5debcSMike Kravetz 
40018fb5debcSMike Kravetz 	if (!*pagep) {
40028fb5debcSMike Kravetz 		ret = -ENOMEM;
40038fb5debcSMike Kravetz 		page = alloc_huge_page(dst_vma, dst_addr, 0);
40048fb5debcSMike Kravetz 		if (IS_ERR(page))
40058fb5debcSMike Kravetz 			goto out;
40068fb5debcSMike Kravetz 
40078fb5debcSMike Kravetz 		ret = copy_huge_page_from_user(page,
40088fb5debcSMike Kravetz 						(const void __user *) src_addr,
4009810a56b9SMike Kravetz 						pages_per_huge_page(h), false);
40108fb5debcSMike Kravetz 
40118fb5debcSMike Kravetz 		/* fallback to copy_from_user outside mmap_sem */
40128fb5debcSMike Kravetz 		if (unlikely(ret)) {
40138fb5debcSMike Kravetz 			ret = -EFAULT;
40148fb5debcSMike Kravetz 			*pagep = page;
40158fb5debcSMike Kravetz 			/* don't free the page */
40168fb5debcSMike Kravetz 			goto out;
40178fb5debcSMike Kravetz 		}
40188fb5debcSMike Kravetz 	} else {
40198fb5debcSMike Kravetz 		page = *pagep;
40208fb5debcSMike Kravetz 		*pagep = NULL;
40218fb5debcSMike Kravetz 	}
40228fb5debcSMike Kravetz 
40238fb5debcSMike Kravetz 	/*
40248fb5debcSMike Kravetz 	 * The memory barrier inside __SetPageUptodate makes sure that
40258fb5debcSMike Kravetz 	 * preceding stores to the page contents become visible before
40268fb5debcSMike Kravetz 	 * the set_pte_at() write.
40278fb5debcSMike Kravetz 	 */
40288fb5debcSMike Kravetz 	__SetPageUptodate(page);
40298fb5debcSMike Kravetz 	set_page_huge_active(page);
40308fb5debcSMike Kravetz 
40318fb5debcSMike Kravetz 	ptl = huge_pte_lockptr(h, dst_mm, dst_pte);
40328fb5debcSMike Kravetz 	spin_lock(ptl);
40338fb5debcSMike Kravetz 
40348fb5debcSMike Kravetz 	ret = -EEXIST;
40358fb5debcSMike Kravetz 	if (!huge_pte_none(huge_ptep_get(dst_pte)))
40368fb5debcSMike Kravetz 		goto out_release_unlock;
40378fb5debcSMike Kravetz 
40388fb5debcSMike Kravetz 	ClearPagePrivate(page);
40398fb5debcSMike Kravetz 	hugepage_add_new_anon_rmap(page, dst_vma, dst_addr);
40408fb5debcSMike Kravetz 
40418fb5debcSMike Kravetz 	_dst_pte = make_huge_pte(dst_vma, page, dst_vma->vm_flags & VM_WRITE);
40428fb5debcSMike Kravetz 	if (dst_vma->vm_flags & VM_WRITE)
40438fb5debcSMike Kravetz 		_dst_pte = huge_pte_mkdirty(_dst_pte);
40448fb5debcSMike Kravetz 	_dst_pte = pte_mkyoung(_dst_pte);
40458fb5debcSMike Kravetz 
40468fb5debcSMike Kravetz 	set_huge_pte_at(dst_mm, dst_addr, dst_pte, _dst_pte);
40478fb5debcSMike Kravetz 
40488fb5debcSMike Kravetz 	(void)huge_ptep_set_access_flags(dst_vma, dst_addr, dst_pte, _dst_pte,
40498fb5debcSMike Kravetz 					dst_vma->vm_flags & VM_WRITE);
40508fb5debcSMike Kravetz 	hugetlb_count_add(pages_per_huge_page(h), dst_mm);
40518fb5debcSMike Kravetz 
40528fb5debcSMike Kravetz 	/* No need to invalidate - it was non-present before */
40538fb5debcSMike Kravetz 	update_mmu_cache(dst_vma, dst_addr, dst_pte);
40548fb5debcSMike Kravetz 
40558fb5debcSMike Kravetz 	spin_unlock(ptl);
40568fb5debcSMike Kravetz 	ret = 0;
40578fb5debcSMike Kravetz out:
40588fb5debcSMike Kravetz 	return ret;
40598fb5debcSMike Kravetz out_release_unlock:
40608fb5debcSMike Kravetz 	spin_unlock(ptl);
40618fb5debcSMike Kravetz 	put_page(page);
40628fb5debcSMike Kravetz 	goto out;
40638fb5debcSMike Kravetz }
40648fb5debcSMike Kravetz 
406528a35716SMichel Lespinasse long follow_hugetlb_page(struct mm_struct *mm, struct vm_area_struct *vma,
406663551ae0SDavid Gibson 			 struct page **pages, struct vm_area_struct **vmas,
406728a35716SMichel Lespinasse 			 unsigned long *position, unsigned long *nr_pages,
406828a35716SMichel Lespinasse 			 long i, unsigned int flags)
406963551ae0SDavid Gibson {
4070d5d4b0aaSChen, Kenneth W 	unsigned long pfn_offset;
4071d5d4b0aaSChen, Kenneth W 	unsigned long vaddr = *position;
407228a35716SMichel Lespinasse 	unsigned long remainder = *nr_pages;
4073a5516438SAndi Kleen 	struct hstate *h = hstate_vma(vma);
407463551ae0SDavid Gibson 
407563551ae0SDavid Gibson 	while (vaddr < vma->vm_end && remainder) {
407663551ae0SDavid Gibson 		pte_t *pte;
4077cb900f41SKirill A. Shutemov 		spinlock_t *ptl = NULL;
40782a15efc9SHugh Dickins 		int absent;
407963551ae0SDavid Gibson 		struct page *page;
408063551ae0SDavid Gibson 
40814c887265SAdam Litke 		/*
408202057967SDavid Rientjes 		 * If we have a pending SIGKILL, don't keep faulting pages and
408302057967SDavid Rientjes 		 * potentially allocating memory.
408402057967SDavid Rientjes 		 */
408502057967SDavid Rientjes 		if (unlikely(fatal_signal_pending(current))) {
408602057967SDavid Rientjes 			remainder = 0;
408702057967SDavid Rientjes 			break;
408802057967SDavid Rientjes 		}
408902057967SDavid Rientjes 
409002057967SDavid Rientjes 		/*
40914c887265SAdam Litke 		 * Some archs (sparc64, sh*) have multiple pte_ts to
40922a15efc9SHugh Dickins 		 * each hugepage.  We have to make sure we get the
40934c887265SAdam Litke 		 * first, for the page indexing below to work.
4094cb900f41SKirill A. Shutemov 		 *
4095cb900f41SKirill A. Shutemov 		 * Note that page table lock is not held when pte is null.
40964c887265SAdam Litke 		 */
4097a5516438SAndi Kleen 		pte = huge_pte_offset(mm, vaddr & huge_page_mask(h));
4098cb900f41SKirill A. Shutemov 		if (pte)
4099cb900f41SKirill A. Shutemov 			ptl = huge_pte_lock(h, mm, pte);
41002a15efc9SHugh Dickins 		absent = !pte || huge_pte_none(huge_ptep_get(pte));
410163551ae0SDavid Gibson 
41022a15efc9SHugh Dickins 		/*
41032a15efc9SHugh Dickins 		 * When coredumping, it suits get_dump_page if we just return
41043ae77f43SHugh Dickins 		 * an error where there's an empty slot with no huge pagecache
41053ae77f43SHugh Dickins 		 * to back it.  This way, we avoid allocating a hugepage, and
41063ae77f43SHugh Dickins 		 * the sparse dumpfile avoids allocating disk blocks, but its
41073ae77f43SHugh Dickins 		 * huge holes still show up with zeroes where they need to be.
41082a15efc9SHugh Dickins 		 */
41093ae77f43SHugh Dickins 		if (absent && (flags & FOLL_DUMP) &&
41103ae77f43SHugh Dickins 		    !hugetlbfs_pagecache_present(h, vma, vaddr)) {
4111cb900f41SKirill A. Shutemov 			if (pte)
4112cb900f41SKirill A. Shutemov 				spin_unlock(ptl);
41132a15efc9SHugh Dickins 			remainder = 0;
41142a15efc9SHugh Dickins 			break;
41152a15efc9SHugh Dickins 		}
41162a15efc9SHugh Dickins 
41179cc3a5bdSNaoya Horiguchi 		/*
41189cc3a5bdSNaoya Horiguchi 		 * We need call hugetlb_fault for both hugepages under migration
41199cc3a5bdSNaoya Horiguchi 		 * (in which case hugetlb_fault waits for the migration,) and
41209cc3a5bdSNaoya Horiguchi 		 * hwpoisoned hugepages (in which case we need to prevent the
41219cc3a5bdSNaoya Horiguchi 		 * caller from accessing to them.) In order to do this, we use
41229cc3a5bdSNaoya Horiguchi 		 * here is_swap_pte instead of is_hugetlb_entry_migration and
41239cc3a5bdSNaoya Horiguchi 		 * is_hugetlb_entry_hwpoisoned. This is because it simply covers
41249cc3a5bdSNaoya Horiguchi 		 * both cases, and because we can't follow correct pages
41259cc3a5bdSNaoya Horiguchi 		 * directly from any kind of swap entries.
41269cc3a5bdSNaoya Horiguchi 		 */
41279cc3a5bdSNaoya Horiguchi 		if (absent || is_swap_pte(huge_ptep_get(pte)) ||
4128106c992aSGerald Schaefer 		    ((flags & FOLL_WRITE) &&
4129106c992aSGerald Schaefer 		      !huge_pte_write(huge_ptep_get(pte)))) {
41304c887265SAdam Litke 			int ret;
41314c887265SAdam Litke 
4132cb900f41SKirill A. Shutemov 			if (pte)
4133cb900f41SKirill A. Shutemov 				spin_unlock(ptl);
41342a15efc9SHugh Dickins 			ret = hugetlb_fault(mm, vma, vaddr,
41352a15efc9SHugh Dickins 				(flags & FOLL_WRITE) ? FAULT_FLAG_WRITE : 0);
4136a89182c7SAdam Litke 			if (!(ret & VM_FAULT_ERROR))
41374c887265SAdam Litke 				continue;
41384c887265SAdam Litke 
41391c59827dSHugh Dickins 			remainder = 0;
41401c59827dSHugh Dickins 			break;
41411c59827dSHugh Dickins 		}
414263551ae0SDavid Gibson 
4143a5516438SAndi Kleen 		pfn_offset = (vaddr & ~huge_page_mask(h)) >> PAGE_SHIFT;
41447f2e9525SGerald Schaefer 		page = pte_page(huge_ptep_get(pte));
4145d5d4b0aaSChen, Kenneth W same_page:
4146d6692183SChen, Kenneth W 		if (pages) {
414769d177c2SAndy Whitcroft 			pages[i] = mem_map_offset(page, pfn_offset);
4148ddc58f27SKirill A. Shutemov 			get_page(pages[i]);
4149d6692183SChen, Kenneth W 		}
415063551ae0SDavid Gibson 
415163551ae0SDavid Gibson 		if (vmas)
415263551ae0SDavid Gibson 			vmas[i] = vma;
415363551ae0SDavid Gibson 
415463551ae0SDavid Gibson 		vaddr += PAGE_SIZE;
4155d5d4b0aaSChen, Kenneth W 		++pfn_offset;
415663551ae0SDavid Gibson 		--remainder;
415763551ae0SDavid Gibson 		++i;
4158d5d4b0aaSChen, Kenneth W 		if (vaddr < vma->vm_end && remainder &&
4159a5516438SAndi Kleen 				pfn_offset < pages_per_huge_page(h)) {
4160d5d4b0aaSChen, Kenneth W 			/*
4161d5d4b0aaSChen, Kenneth W 			 * We use pfn_offset to avoid touching the pageframes
4162d5d4b0aaSChen, Kenneth W 			 * of this compound page.
4163d5d4b0aaSChen, Kenneth W 			 */
4164d5d4b0aaSChen, Kenneth W 			goto same_page;
4165d5d4b0aaSChen, Kenneth W 		}
4166cb900f41SKirill A. Shutemov 		spin_unlock(ptl);
416763551ae0SDavid Gibson 	}
416828a35716SMichel Lespinasse 	*nr_pages = remainder;
416963551ae0SDavid Gibson 	*position = vaddr;
417063551ae0SDavid Gibson 
41712a15efc9SHugh Dickins 	return i ? i : -EFAULT;
417263551ae0SDavid Gibson }
41738f860591SZhang, Yanmin 
41745491ae7bSAneesh Kumar K.V #ifndef __HAVE_ARCH_FLUSH_HUGETLB_TLB_RANGE
41755491ae7bSAneesh Kumar K.V /*
41765491ae7bSAneesh Kumar K.V  * ARCHes with special requirements for evicting HUGETLB backing TLB entries can
41775491ae7bSAneesh Kumar K.V  * implement this.
41785491ae7bSAneesh Kumar K.V  */
41795491ae7bSAneesh Kumar K.V #define flush_hugetlb_tlb_range(vma, addr, end)	flush_tlb_range(vma, addr, end)
41805491ae7bSAneesh Kumar K.V #endif
41815491ae7bSAneesh Kumar K.V 
41827da4d641SPeter Zijlstra unsigned long hugetlb_change_protection(struct vm_area_struct *vma,
41838f860591SZhang, Yanmin 		unsigned long address, unsigned long end, pgprot_t newprot)
41848f860591SZhang, Yanmin {
41858f860591SZhang, Yanmin 	struct mm_struct *mm = vma->vm_mm;
41868f860591SZhang, Yanmin 	unsigned long start = address;
41878f860591SZhang, Yanmin 	pte_t *ptep;
41888f860591SZhang, Yanmin 	pte_t pte;
4189a5516438SAndi Kleen 	struct hstate *h = hstate_vma(vma);
41907da4d641SPeter Zijlstra 	unsigned long pages = 0;
41918f860591SZhang, Yanmin 
41928f860591SZhang, Yanmin 	BUG_ON(address >= end);
41938f860591SZhang, Yanmin 	flush_cache_range(vma, address, end);
41948f860591SZhang, Yanmin 
4195a5338093SRik van Riel 	mmu_notifier_invalidate_range_start(mm, start, end);
419683cde9e8SDavidlohr Bueso 	i_mmap_lock_write(vma->vm_file->f_mapping);
4197a5516438SAndi Kleen 	for (; address < end; address += huge_page_size(h)) {
4198cb900f41SKirill A. Shutemov 		spinlock_t *ptl;
41998f860591SZhang, Yanmin 		ptep = huge_pte_offset(mm, address);
42008f860591SZhang, Yanmin 		if (!ptep)
42018f860591SZhang, Yanmin 			continue;
4202cb900f41SKirill A. Shutemov 		ptl = huge_pte_lock(h, mm, ptep);
42037da4d641SPeter Zijlstra 		if (huge_pmd_unshare(mm, &address, ptep)) {
42047da4d641SPeter Zijlstra 			pages++;
4205cb900f41SKirill A. Shutemov 			spin_unlock(ptl);
420639dde65cSChen, Kenneth W 			continue;
42077da4d641SPeter Zijlstra 		}
4208a8bda28dSNaoya Horiguchi 		pte = huge_ptep_get(ptep);
4209a8bda28dSNaoya Horiguchi 		if (unlikely(is_hugetlb_entry_hwpoisoned(pte))) {
4210a8bda28dSNaoya Horiguchi 			spin_unlock(ptl);
4211a8bda28dSNaoya Horiguchi 			continue;
4212a8bda28dSNaoya Horiguchi 		}
4213a8bda28dSNaoya Horiguchi 		if (unlikely(is_hugetlb_entry_migration(pte))) {
4214a8bda28dSNaoya Horiguchi 			swp_entry_t entry = pte_to_swp_entry(pte);
4215a8bda28dSNaoya Horiguchi 
4216a8bda28dSNaoya Horiguchi 			if (is_write_migration_entry(entry)) {
4217a8bda28dSNaoya Horiguchi 				pte_t newpte;
4218a8bda28dSNaoya Horiguchi 
4219a8bda28dSNaoya Horiguchi 				make_migration_entry_read(&entry);
4220a8bda28dSNaoya Horiguchi 				newpte = swp_entry_to_pte(entry);
4221a8bda28dSNaoya Horiguchi 				set_huge_pte_at(mm, address, ptep, newpte);
4222a8bda28dSNaoya Horiguchi 				pages++;
4223a8bda28dSNaoya Horiguchi 			}
4224a8bda28dSNaoya Horiguchi 			spin_unlock(ptl);
4225a8bda28dSNaoya Horiguchi 			continue;
4226a8bda28dSNaoya Horiguchi 		}
4227a8bda28dSNaoya Horiguchi 		if (!huge_pte_none(pte)) {
42288f860591SZhang, Yanmin 			pte = huge_ptep_get_and_clear(mm, address, ptep);
4229106c992aSGerald Schaefer 			pte = pte_mkhuge(huge_pte_modify(pte, newprot));
4230be7517d6STony Lu 			pte = arch_make_huge_pte(pte, vma, NULL, 0);
42318f860591SZhang, Yanmin 			set_huge_pte_at(mm, address, ptep, pte);
42327da4d641SPeter Zijlstra 			pages++;
42338f860591SZhang, Yanmin 		}
4234cb900f41SKirill A. Shutemov 		spin_unlock(ptl);
42358f860591SZhang, Yanmin 	}
4236d833352aSMel Gorman 	/*
4237c8c06efaSDavidlohr Bueso 	 * Must flush TLB before releasing i_mmap_rwsem: x86's huge_pmd_unshare
4238d833352aSMel Gorman 	 * may have cleared our pud entry and done put_page on the page table:
4239c8c06efaSDavidlohr Bueso 	 * once we release i_mmap_rwsem, another task can do the final put_page
4240d833352aSMel Gorman 	 * and that page table be reused and filled with junk.
4241d833352aSMel Gorman 	 */
42425491ae7bSAneesh Kumar K.V 	flush_hugetlb_tlb_range(vma, start, end);
424334ee645eSJoerg Roedel 	mmu_notifier_invalidate_range(mm, start, end);
424483cde9e8SDavidlohr Bueso 	i_mmap_unlock_write(vma->vm_file->f_mapping);
4245a5338093SRik van Riel 	mmu_notifier_invalidate_range_end(mm, start, end);
42467da4d641SPeter Zijlstra 
42477da4d641SPeter Zijlstra 	return pages << h->order;
42488f860591SZhang, Yanmin }
42498f860591SZhang, Yanmin 
4250a1e78772SMel Gorman int hugetlb_reserve_pages(struct inode *inode,
4251a1e78772SMel Gorman 					long from, long to,
42525a6fe125SMel Gorman 					struct vm_area_struct *vma,
4253ca16d140SKOSAKI Motohiro 					vm_flags_t vm_flags)
4254e4e574b7SAdam Litke {
425517c9d12eSMel Gorman 	long ret, chg;
4256a5516438SAndi Kleen 	struct hstate *h = hstate_inode(inode);
425790481622SDavid Gibson 	struct hugepage_subpool *spool = subpool_inode(inode);
42589119a41eSJoonsoo Kim 	struct resv_map *resv_map;
42591c5ecae3SMike Kravetz 	long gbl_reserve;
4260e4e574b7SAdam Litke 
4261a1e78772SMel Gorman 	/*
426217c9d12eSMel Gorman 	 * Only apply hugepage reservation if asked. At fault time, an
426317c9d12eSMel Gorman 	 * attempt will be made for VM_NORESERVE to allocate a page
426490481622SDavid Gibson 	 * without using reserves
426517c9d12eSMel Gorman 	 */
4266ca16d140SKOSAKI Motohiro 	if (vm_flags & VM_NORESERVE)
426717c9d12eSMel Gorman 		return 0;
426817c9d12eSMel Gorman 
426917c9d12eSMel Gorman 	/*
4270a1e78772SMel Gorman 	 * Shared mappings base their reservation on the number of pages that
4271a1e78772SMel Gorman 	 * are already allocated on behalf of the file. Private mappings need
4272a1e78772SMel Gorman 	 * to reserve the full area even if read-only as mprotect() may be
4273a1e78772SMel Gorman 	 * called to make the mapping read-write. Assume !vma is a shm mapping
4274a1e78772SMel Gorman 	 */
42759119a41eSJoonsoo Kim 	if (!vma || vma->vm_flags & VM_MAYSHARE) {
42764e35f483SJoonsoo Kim 		resv_map = inode_resv_map(inode);
42779119a41eSJoonsoo Kim 
42781406ec9bSJoonsoo Kim 		chg = region_chg(resv_map, from, to);
42799119a41eSJoonsoo Kim 
42809119a41eSJoonsoo Kim 	} else {
42819119a41eSJoonsoo Kim 		resv_map = resv_map_alloc();
42825a6fe125SMel Gorman 		if (!resv_map)
42835a6fe125SMel Gorman 			return -ENOMEM;
42845a6fe125SMel Gorman 
428517c9d12eSMel Gorman 		chg = to - from;
428617c9d12eSMel Gorman 
42875a6fe125SMel Gorman 		set_vma_resv_map(vma, resv_map);
42885a6fe125SMel Gorman 		set_vma_resv_flags(vma, HPAGE_RESV_OWNER);
42895a6fe125SMel Gorman 	}
42905a6fe125SMel Gorman 
4291c50ac050SDave Hansen 	if (chg < 0) {
4292c50ac050SDave Hansen 		ret = chg;
4293c50ac050SDave Hansen 		goto out_err;
4294c50ac050SDave Hansen 	}
429517c9d12eSMel Gorman 
42961c5ecae3SMike Kravetz 	/*
42971c5ecae3SMike Kravetz 	 * There must be enough pages in the subpool for the mapping. If
42981c5ecae3SMike Kravetz 	 * the subpool has a minimum size, there may be some global
42991c5ecae3SMike Kravetz 	 * reservations already in place (gbl_reserve).
43001c5ecae3SMike Kravetz 	 */
43011c5ecae3SMike Kravetz 	gbl_reserve = hugepage_subpool_get_pages(spool, chg);
43021c5ecae3SMike Kravetz 	if (gbl_reserve < 0) {
4303c50ac050SDave Hansen 		ret = -ENOSPC;
4304c50ac050SDave Hansen 		goto out_err;
4305c50ac050SDave Hansen 	}
430617c9d12eSMel Gorman 
430717c9d12eSMel Gorman 	/*
430817c9d12eSMel Gorman 	 * Check enough hugepages are available for the reservation.
430990481622SDavid Gibson 	 * Hand the pages back to the subpool if there are not
431017c9d12eSMel Gorman 	 */
43111c5ecae3SMike Kravetz 	ret = hugetlb_acct_memory(h, gbl_reserve);
431217c9d12eSMel Gorman 	if (ret < 0) {
43131c5ecae3SMike Kravetz 		/* put back original number of pages, chg */
43141c5ecae3SMike Kravetz 		(void)hugepage_subpool_put_pages(spool, chg);
4315c50ac050SDave Hansen 		goto out_err;
431617c9d12eSMel Gorman 	}
431717c9d12eSMel Gorman 
431817c9d12eSMel Gorman 	/*
431917c9d12eSMel Gorman 	 * Account for the reservations made. Shared mappings record regions
432017c9d12eSMel Gorman 	 * that have reservations as they are shared by multiple VMAs.
432117c9d12eSMel Gorman 	 * When the last VMA disappears, the region map says how much
432217c9d12eSMel Gorman 	 * the reservation was and the page cache tells how much of
432317c9d12eSMel Gorman 	 * the reservation was consumed. Private mappings are per-VMA and
432417c9d12eSMel Gorman 	 * only the consumed reservations are tracked. When the VMA
432517c9d12eSMel Gorman 	 * disappears, the original reservation is the VMA size and the
432617c9d12eSMel Gorman 	 * consumed reservations are stored in the map. Hence, nothing
432717c9d12eSMel Gorman 	 * else has to be done for private mappings here
432817c9d12eSMel Gorman 	 */
432933039678SMike Kravetz 	if (!vma || vma->vm_flags & VM_MAYSHARE) {
433033039678SMike Kravetz 		long add = region_add(resv_map, from, to);
433133039678SMike Kravetz 
433233039678SMike Kravetz 		if (unlikely(chg > add)) {
433333039678SMike Kravetz 			/*
433433039678SMike Kravetz 			 * pages in this range were added to the reserve
433533039678SMike Kravetz 			 * map between region_chg and region_add.  This
433633039678SMike Kravetz 			 * indicates a race with alloc_huge_page.  Adjust
433733039678SMike Kravetz 			 * the subpool and reserve counts modified above
433833039678SMike Kravetz 			 * based on the difference.
433933039678SMike Kravetz 			 */
434033039678SMike Kravetz 			long rsv_adjust;
434133039678SMike Kravetz 
434233039678SMike Kravetz 			rsv_adjust = hugepage_subpool_put_pages(spool,
434333039678SMike Kravetz 								chg - add);
434433039678SMike Kravetz 			hugetlb_acct_memory(h, -rsv_adjust);
434533039678SMike Kravetz 		}
434633039678SMike Kravetz 	}
4347a43a8c39SChen, Kenneth W 	return 0;
4348c50ac050SDave Hansen out_err:
43495e911373SMike Kravetz 	if (!vma || vma->vm_flags & VM_MAYSHARE)
43505e911373SMike Kravetz 		region_abort(resv_map, from, to);
4351f031dd27SJoonsoo Kim 	if (vma && is_vma_resv_set(vma, HPAGE_RESV_OWNER))
4352f031dd27SJoonsoo Kim 		kref_put(&resv_map->refs, resv_map_release);
4353c50ac050SDave Hansen 	return ret;
4354a43a8c39SChen, Kenneth W }
4355a43a8c39SChen, Kenneth W 
4356b5cec28dSMike Kravetz long hugetlb_unreserve_pages(struct inode *inode, long start, long end,
4357b5cec28dSMike Kravetz 								long freed)
4358a43a8c39SChen, Kenneth W {
4359a5516438SAndi Kleen 	struct hstate *h = hstate_inode(inode);
43604e35f483SJoonsoo Kim 	struct resv_map *resv_map = inode_resv_map(inode);
43619119a41eSJoonsoo Kim 	long chg = 0;
436290481622SDavid Gibson 	struct hugepage_subpool *spool = subpool_inode(inode);
43631c5ecae3SMike Kravetz 	long gbl_reserve;
436445c682a6SKen Chen 
4365b5cec28dSMike Kravetz 	if (resv_map) {
4366b5cec28dSMike Kravetz 		chg = region_del(resv_map, start, end);
4367b5cec28dSMike Kravetz 		/*
4368b5cec28dSMike Kravetz 		 * region_del() can fail in the rare case where a region
4369b5cec28dSMike Kravetz 		 * must be split and another region descriptor can not be
4370b5cec28dSMike Kravetz 		 * allocated.  If end == LONG_MAX, it will not fail.
4371b5cec28dSMike Kravetz 		 */
4372b5cec28dSMike Kravetz 		if (chg < 0)
4373b5cec28dSMike Kravetz 			return chg;
4374b5cec28dSMike Kravetz 	}
4375b5cec28dSMike Kravetz 
437645c682a6SKen Chen 	spin_lock(&inode->i_lock);
4377e4c6f8beSEric Sandeen 	inode->i_blocks -= (blocks_per_huge_page(h) * freed);
437845c682a6SKen Chen 	spin_unlock(&inode->i_lock);
437945c682a6SKen Chen 
43801c5ecae3SMike Kravetz 	/*
43811c5ecae3SMike Kravetz 	 * If the subpool has a minimum size, the number of global
43821c5ecae3SMike Kravetz 	 * reservations to be released may be adjusted.
43831c5ecae3SMike Kravetz 	 */
43841c5ecae3SMike Kravetz 	gbl_reserve = hugepage_subpool_put_pages(spool, (chg - freed));
43851c5ecae3SMike Kravetz 	hugetlb_acct_memory(h, -gbl_reserve);
4386b5cec28dSMike Kravetz 
4387b5cec28dSMike Kravetz 	return 0;
4388a43a8c39SChen, Kenneth W }
438993f70f90SNaoya Horiguchi 
43903212b535SSteve Capper #ifdef CONFIG_ARCH_WANT_HUGE_PMD_SHARE
43913212b535SSteve Capper static unsigned long page_table_shareable(struct vm_area_struct *svma,
43923212b535SSteve Capper 				struct vm_area_struct *vma,
43933212b535SSteve Capper 				unsigned long addr, pgoff_t idx)
43943212b535SSteve Capper {
43953212b535SSteve Capper 	unsigned long saddr = ((idx - svma->vm_pgoff) << PAGE_SHIFT) +
43963212b535SSteve Capper 				svma->vm_start;
43973212b535SSteve Capper 	unsigned long sbase = saddr & PUD_MASK;
43983212b535SSteve Capper 	unsigned long s_end = sbase + PUD_SIZE;
43993212b535SSteve Capper 
44003212b535SSteve Capper 	/* Allow segments to share if only one is marked locked */
4401de60f5f1SEric B Munson 	unsigned long vm_flags = vma->vm_flags & VM_LOCKED_CLEAR_MASK;
4402de60f5f1SEric B Munson 	unsigned long svm_flags = svma->vm_flags & VM_LOCKED_CLEAR_MASK;
44033212b535SSteve Capper 
44043212b535SSteve Capper 	/*
44053212b535SSteve Capper 	 * match the virtual addresses, permission and the alignment of the
44063212b535SSteve Capper 	 * page table page.
44073212b535SSteve Capper 	 */
44083212b535SSteve Capper 	if (pmd_index(addr) != pmd_index(saddr) ||
44093212b535SSteve Capper 	    vm_flags != svm_flags ||
44103212b535SSteve Capper 	    sbase < svma->vm_start || svma->vm_end < s_end)
44113212b535SSteve Capper 		return 0;
44123212b535SSteve Capper 
44133212b535SSteve Capper 	return saddr;
44143212b535SSteve Capper }
44153212b535SSteve Capper 
441631aafb45SNicholas Krause static bool vma_shareable(struct vm_area_struct *vma, unsigned long addr)
44173212b535SSteve Capper {
44183212b535SSteve Capper 	unsigned long base = addr & PUD_MASK;
44193212b535SSteve Capper 	unsigned long end = base + PUD_SIZE;
44203212b535SSteve Capper 
44213212b535SSteve Capper 	/*
44223212b535SSteve Capper 	 * check on proper vm_flags and page table alignment
44233212b535SSteve Capper 	 */
44243212b535SSteve Capper 	if (vma->vm_flags & VM_MAYSHARE &&
44253212b535SSteve Capper 	    vma->vm_start <= base && end <= vma->vm_end)
442631aafb45SNicholas Krause 		return true;
442731aafb45SNicholas Krause 	return false;
44283212b535SSteve Capper }
44293212b535SSteve Capper 
44303212b535SSteve Capper /*
44313212b535SSteve Capper  * Search for a shareable pmd page for hugetlb. In any case calls pmd_alloc()
44323212b535SSteve Capper  * and returns the corresponding pte. While this is not necessary for the
44333212b535SSteve Capper  * !shared pmd case because we can allocate the pmd later as well, it makes the
44343212b535SSteve Capper  * code much cleaner. pmd allocation is essential for the shared case because
4435c8c06efaSDavidlohr Bueso  * pud has to be populated inside the same i_mmap_rwsem section - otherwise
44363212b535SSteve Capper  * racing tasks could either miss the sharing (see huge_pte_offset) or select a
44373212b535SSteve Capper  * bad pmd for sharing.
44383212b535SSteve Capper  */
44393212b535SSteve Capper pte_t *huge_pmd_share(struct mm_struct *mm, unsigned long addr, pud_t *pud)
44403212b535SSteve Capper {
44413212b535SSteve Capper 	struct vm_area_struct *vma = find_vma(mm, addr);
44423212b535SSteve Capper 	struct address_space *mapping = vma->vm_file->f_mapping;
44433212b535SSteve Capper 	pgoff_t idx = ((addr - vma->vm_start) >> PAGE_SHIFT) +
44443212b535SSteve Capper 			vma->vm_pgoff;
44453212b535SSteve Capper 	struct vm_area_struct *svma;
44463212b535SSteve Capper 	unsigned long saddr;
44473212b535SSteve Capper 	pte_t *spte = NULL;
44483212b535SSteve Capper 	pte_t *pte;
4449cb900f41SKirill A. Shutemov 	spinlock_t *ptl;
44503212b535SSteve Capper 
44513212b535SSteve Capper 	if (!vma_shareable(vma, addr))
44523212b535SSteve Capper 		return (pte_t *)pmd_alloc(mm, pud, addr);
44533212b535SSteve Capper 
445483cde9e8SDavidlohr Bueso 	i_mmap_lock_write(mapping);
44553212b535SSteve Capper 	vma_interval_tree_foreach(svma, &mapping->i_mmap, idx, idx) {
44563212b535SSteve Capper 		if (svma == vma)
44573212b535SSteve Capper 			continue;
44583212b535SSteve Capper 
44593212b535SSteve Capper 		saddr = page_table_shareable(svma, vma, addr, idx);
44603212b535SSteve Capper 		if (saddr) {
44613212b535SSteve Capper 			spte = huge_pte_offset(svma->vm_mm, saddr);
44623212b535SSteve Capper 			if (spte) {
44633212b535SSteve Capper 				get_page(virt_to_page(spte));
44643212b535SSteve Capper 				break;
44653212b535SSteve Capper 			}
44663212b535SSteve Capper 		}
44673212b535SSteve Capper 	}
44683212b535SSteve Capper 
44693212b535SSteve Capper 	if (!spte)
44703212b535SSteve Capper 		goto out;
44713212b535SSteve Capper 
44728bea8052SAneesh Kumar K.V 	ptl = huge_pte_lock(hstate_vma(vma), mm, spte);
4473dc6c9a35SKirill A. Shutemov 	if (pud_none(*pud)) {
44743212b535SSteve Capper 		pud_populate(mm, pud,
44753212b535SSteve Capper 				(pmd_t *)((unsigned long)spte & PAGE_MASK));
4476c17b1f42SKirill A. Shutemov 		mm_inc_nr_pmds(mm);
4477dc6c9a35SKirill A. Shutemov 	} else {
44783212b535SSteve Capper 		put_page(virt_to_page(spte));
4479dc6c9a35SKirill A. Shutemov 	}
4480cb900f41SKirill A. Shutemov 	spin_unlock(ptl);
44813212b535SSteve Capper out:
44823212b535SSteve Capper 	pte = (pte_t *)pmd_alloc(mm, pud, addr);
448383cde9e8SDavidlohr Bueso 	i_mmap_unlock_write(mapping);
44843212b535SSteve Capper 	return pte;
44853212b535SSteve Capper }
44863212b535SSteve Capper 
44873212b535SSteve Capper /*
44883212b535SSteve Capper  * unmap huge page backed by shared pte.
44893212b535SSteve Capper  *
44903212b535SSteve Capper  * Hugetlb pte page is ref counted at the time of mapping.  If pte is shared
44913212b535SSteve Capper  * indicated by page_count > 1, unmap is achieved by clearing pud and
44923212b535SSteve Capper  * decrementing the ref count. If count == 1, the pte page is not shared.
44933212b535SSteve Capper  *
4494cb900f41SKirill A. Shutemov  * called with page table lock held.
44953212b535SSteve Capper  *
44963212b535SSteve Capper  * returns: 1 successfully unmapped a shared pte page
44973212b535SSteve Capper  *	    0 the underlying pte page is not shared, or it is the last user
44983212b535SSteve Capper  */
44993212b535SSteve Capper int huge_pmd_unshare(struct mm_struct *mm, unsigned long *addr, pte_t *ptep)
45003212b535SSteve Capper {
45013212b535SSteve Capper 	pgd_t *pgd = pgd_offset(mm, *addr);
45023212b535SSteve Capper 	pud_t *pud = pud_offset(pgd, *addr);
45033212b535SSteve Capper 
45043212b535SSteve Capper 	BUG_ON(page_count(virt_to_page(ptep)) == 0);
45053212b535SSteve Capper 	if (page_count(virt_to_page(ptep)) == 1)
45063212b535SSteve Capper 		return 0;
45073212b535SSteve Capper 
45083212b535SSteve Capper 	pud_clear(pud);
45093212b535SSteve Capper 	put_page(virt_to_page(ptep));
4510dc6c9a35SKirill A. Shutemov 	mm_dec_nr_pmds(mm);
45113212b535SSteve Capper 	*addr = ALIGN(*addr, HPAGE_SIZE * PTRS_PER_PTE) - HPAGE_SIZE;
45123212b535SSteve Capper 	return 1;
45133212b535SSteve Capper }
45149e5fc74cSSteve Capper #define want_pmd_share()	(1)
45159e5fc74cSSteve Capper #else /* !CONFIG_ARCH_WANT_HUGE_PMD_SHARE */
45169e5fc74cSSteve Capper pte_t *huge_pmd_share(struct mm_struct *mm, unsigned long addr, pud_t *pud)
45179e5fc74cSSteve Capper {
45189e5fc74cSSteve Capper 	return NULL;
45199e5fc74cSSteve Capper }
4520e81f2d22SZhang Zhen 
4521e81f2d22SZhang Zhen int huge_pmd_unshare(struct mm_struct *mm, unsigned long *addr, pte_t *ptep)
4522e81f2d22SZhang Zhen {
4523e81f2d22SZhang Zhen 	return 0;
4524e81f2d22SZhang Zhen }
45259e5fc74cSSteve Capper #define want_pmd_share()	(0)
45263212b535SSteve Capper #endif /* CONFIG_ARCH_WANT_HUGE_PMD_SHARE */
45273212b535SSteve Capper 
45289e5fc74cSSteve Capper #ifdef CONFIG_ARCH_WANT_GENERAL_HUGETLB
45299e5fc74cSSteve Capper pte_t *huge_pte_alloc(struct mm_struct *mm,
45309e5fc74cSSteve Capper 			unsigned long addr, unsigned long sz)
45319e5fc74cSSteve Capper {
45329e5fc74cSSteve Capper 	pgd_t *pgd;
45339e5fc74cSSteve Capper 	pud_t *pud;
45349e5fc74cSSteve Capper 	pte_t *pte = NULL;
45359e5fc74cSSteve Capper 
45369e5fc74cSSteve Capper 	pgd = pgd_offset(mm, addr);
45379e5fc74cSSteve Capper 	pud = pud_alloc(mm, pgd, addr);
45389e5fc74cSSteve Capper 	if (pud) {
45399e5fc74cSSteve Capper 		if (sz == PUD_SIZE) {
45409e5fc74cSSteve Capper 			pte = (pte_t *)pud;
45419e5fc74cSSteve Capper 		} else {
45429e5fc74cSSteve Capper 			BUG_ON(sz != PMD_SIZE);
45439e5fc74cSSteve Capper 			if (want_pmd_share() && pud_none(*pud))
45449e5fc74cSSteve Capper 				pte = huge_pmd_share(mm, addr, pud);
45459e5fc74cSSteve Capper 			else
45469e5fc74cSSteve Capper 				pte = (pte_t *)pmd_alloc(mm, pud, addr);
45479e5fc74cSSteve Capper 		}
45489e5fc74cSSteve Capper 	}
45494e666314SMichal Hocko 	BUG_ON(pte && pte_present(*pte) && !pte_huge(*pte));
45509e5fc74cSSteve Capper 
45519e5fc74cSSteve Capper 	return pte;
45529e5fc74cSSteve Capper }
45539e5fc74cSSteve Capper 
45549e5fc74cSSteve Capper pte_t *huge_pte_offset(struct mm_struct *mm, unsigned long addr)
45559e5fc74cSSteve Capper {
45569e5fc74cSSteve Capper 	pgd_t *pgd;
45579e5fc74cSSteve Capper 	pud_t *pud;
45589e5fc74cSSteve Capper 	pmd_t *pmd = NULL;
45599e5fc74cSSteve Capper 
45609e5fc74cSSteve Capper 	pgd = pgd_offset(mm, addr);
45619e5fc74cSSteve Capper 	if (pgd_present(*pgd)) {
45629e5fc74cSSteve Capper 		pud = pud_offset(pgd, addr);
45639e5fc74cSSteve Capper 		if (pud_present(*pud)) {
45649e5fc74cSSteve Capper 			if (pud_huge(*pud))
45659e5fc74cSSteve Capper 				return (pte_t *)pud;
45669e5fc74cSSteve Capper 			pmd = pmd_offset(pud, addr);
45679e5fc74cSSteve Capper 		}
45689e5fc74cSSteve Capper 	}
45699e5fc74cSSteve Capper 	return (pte_t *) pmd;
45709e5fc74cSSteve Capper }
45719e5fc74cSSteve Capper 
457261f77edaSNaoya Horiguchi #endif /* CONFIG_ARCH_WANT_GENERAL_HUGETLB */
457361f77edaSNaoya Horiguchi 
457461f77edaSNaoya Horiguchi /*
457561f77edaSNaoya Horiguchi  * These functions are overwritable if your architecture needs its own
457661f77edaSNaoya Horiguchi  * behavior.
457761f77edaSNaoya Horiguchi  */
457861f77edaSNaoya Horiguchi struct page * __weak
457961f77edaSNaoya Horiguchi follow_huge_addr(struct mm_struct *mm, unsigned long address,
458061f77edaSNaoya Horiguchi 			      int write)
458161f77edaSNaoya Horiguchi {
458261f77edaSNaoya Horiguchi 	return ERR_PTR(-EINVAL);
458361f77edaSNaoya Horiguchi }
458461f77edaSNaoya Horiguchi 
458561f77edaSNaoya Horiguchi struct page * __weak
45869e5fc74cSSteve Capper follow_huge_pmd(struct mm_struct *mm, unsigned long address,
4587e66f17ffSNaoya Horiguchi 		pmd_t *pmd, int flags)
45889e5fc74cSSteve Capper {
4589e66f17ffSNaoya Horiguchi 	struct page *page = NULL;
4590e66f17ffSNaoya Horiguchi 	spinlock_t *ptl;
4591e66f17ffSNaoya Horiguchi retry:
4592e66f17ffSNaoya Horiguchi 	ptl = pmd_lockptr(mm, pmd);
4593e66f17ffSNaoya Horiguchi 	spin_lock(ptl);
4594e66f17ffSNaoya Horiguchi 	/*
4595e66f17ffSNaoya Horiguchi 	 * make sure that the address range covered by this pmd is not
4596e66f17ffSNaoya Horiguchi 	 * unmapped from other threads.
4597e66f17ffSNaoya Horiguchi 	 */
4598e66f17ffSNaoya Horiguchi 	if (!pmd_huge(*pmd))
4599e66f17ffSNaoya Horiguchi 		goto out;
4600e66f17ffSNaoya Horiguchi 	if (pmd_present(*pmd)) {
460197534127SGerald Schaefer 		page = pmd_page(*pmd) + ((address & ~PMD_MASK) >> PAGE_SHIFT);
4602e66f17ffSNaoya Horiguchi 		if (flags & FOLL_GET)
4603e66f17ffSNaoya Horiguchi 			get_page(page);
4604e66f17ffSNaoya Horiguchi 	} else {
4605e66f17ffSNaoya Horiguchi 		if (is_hugetlb_entry_migration(huge_ptep_get((pte_t *)pmd))) {
4606e66f17ffSNaoya Horiguchi 			spin_unlock(ptl);
4607e66f17ffSNaoya Horiguchi 			__migration_entry_wait(mm, (pte_t *)pmd, ptl);
4608e66f17ffSNaoya Horiguchi 			goto retry;
4609e66f17ffSNaoya Horiguchi 		}
4610e66f17ffSNaoya Horiguchi 		/*
4611e66f17ffSNaoya Horiguchi 		 * hwpoisoned entry is treated as no_page_table in
4612e66f17ffSNaoya Horiguchi 		 * follow_page_mask().
4613e66f17ffSNaoya Horiguchi 		 */
4614e66f17ffSNaoya Horiguchi 	}
4615e66f17ffSNaoya Horiguchi out:
4616e66f17ffSNaoya Horiguchi 	spin_unlock(ptl);
46179e5fc74cSSteve Capper 	return page;
46189e5fc74cSSteve Capper }
46199e5fc74cSSteve Capper 
462061f77edaSNaoya Horiguchi struct page * __weak
46219e5fc74cSSteve Capper follow_huge_pud(struct mm_struct *mm, unsigned long address,
4622e66f17ffSNaoya Horiguchi 		pud_t *pud, int flags)
46239e5fc74cSSteve Capper {
4624e66f17ffSNaoya Horiguchi 	if (flags & FOLL_GET)
4625e66f17ffSNaoya Horiguchi 		return NULL;
46269e5fc74cSSteve Capper 
4627e66f17ffSNaoya Horiguchi 	return pte_page(*(pte_t *)pud) + ((address & ~PUD_MASK) >> PAGE_SHIFT);
46289e5fc74cSSteve Capper }
46299e5fc74cSSteve Capper 
4630d5bd9106SAndi Kleen #ifdef CONFIG_MEMORY_FAILURE
4631d5bd9106SAndi Kleen 
463293f70f90SNaoya Horiguchi /*
463393f70f90SNaoya Horiguchi  * This function is called from memory failure code.
463493f70f90SNaoya Horiguchi  */
46356de2b1aaSNaoya Horiguchi int dequeue_hwpoisoned_huge_page(struct page *hpage)
463693f70f90SNaoya Horiguchi {
463793f70f90SNaoya Horiguchi 	struct hstate *h = page_hstate(hpage);
463893f70f90SNaoya Horiguchi 	int nid = page_to_nid(hpage);
46396de2b1aaSNaoya Horiguchi 	int ret = -EBUSY;
464093f70f90SNaoya Horiguchi 
464193f70f90SNaoya Horiguchi 	spin_lock(&hugetlb_lock);
46427e1f049eSNaoya Horiguchi 	/*
46437e1f049eSNaoya Horiguchi 	 * Just checking !page_huge_active is not enough, because that could be
46447e1f049eSNaoya Horiguchi 	 * an isolated/hwpoisoned hugepage (which have >0 refcount).
46457e1f049eSNaoya Horiguchi 	 */
46467e1f049eSNaoya Horiguchi 	if (!page_huge_active(hpage) && !page_count(hpage)) {
464756f2fb14SNaoya Horiguchi 		/*
464856f2fb14SNaoya Horiguchi 		 * Hwpoisoned hugepage isn't linked to activelist or freelist,
464956f2fb14SNaoya Horiguchi 		 * but dangling hpage->lru can trigger list-debug warnings
465056f2fb14SNaoya Horiguchi 		 * (this happens when we call unpoison_memory() on it),
465156f2fb14SNaoya Horiguchi 		 * so let it point to itself with list_del_init().
465256f2fb14SNaoya Horiguchi 		 */
465356f2fb14SNaoya Horiguchi 		list_del_init(&hpage->lru);
46548c6c2ecbSNaoya Horiguchi 		set_page_refcounted(hpage);
465593f70f90SNaoya Horiguchi 		h->free_huge_pages--;
465693f70f90SNaoya Horiguchi 		h->free_huge_pages_node[nid]--;
46576de2b1aaSNaoya Horiguchi 		ret = 0;
465893f70f90SNaoya Horiguchi 	}
46596de2b1aaSNaoya Horiguchi 	spin_unlock(&hugetlb_lock);
46606de2b1aaSNaoya Horiguchi 	return ret;
46616de2b1aaSNaoya Horiguchi }
46626de2b1aaSNaoya Horiguchi #endif
466331caf665SNaoya Horiguchi 
466431caf665SNaoya Horiguchi bool isolate_huge_page(struct page *page, struct list_head *list)
466531caf665SNaoya Horiguchi {
4666bcc54222SNaoya Horiguchi 	bool ret = true;
4667bcc54222SNaoya Horiguchi 
4668309381feSSasha Levin 	VM_BUG_ON_PAGE(!PageHead(page), page);
466931caf665SNaoya Horiguchi 	spin_lock(&hugetlb_lock);
4670bcc54222SNaoya Horiguchi 	if (!page_huge_active(page) || !get_page_unless_zero(page)) {
4671bcc54222SNaoya Horiguchi 		ret = false;
4672bcc54222SNaoya Horiguchi 		goto unlock;
4673bcc54222SNaoya Horiguchi 	}
4674bcc54222SNaoya Horiguchi 	clear_page_huge_active(page);
467531caf665SNaoya Horiguchi 	list_move_tail(&page->lru, list);
4676bcc54222SNaoya Horiguchi unlock:
467731caf665SNaoya Horiguchi 	spin_unlock(&hugetlb_lock);
4678bcc54222SNaoya Horiguchi 	return ret;
467931caf665SNaoya Horiguchi }
468031caf665SNaoya Horiguchi 
468131caf665SNaoya Horiguchi void putback_active_hugepage(struct page *page)
468231caf665SNaoya Horiguchi {
4683309381feSSasha Levin 	VM_BUG_ON_PAGE(!PageHead(page), page);
468431caf665SNaoya Horiguchi 	spin_lock(&hugetlb_lock);
4685bcc54222SNaoya Horiguchi 	set_page_huge_active(page);
468631caf665SNaoya Horiguchi 	list_move_tail(&page->lru, &(page_hstate(page))->hugepage_activelist);
468731caf665SNaoya Horiguchi 	spin_unlock(&hugetlb_lock);
468831caf665SNaoya Horiguchi 	put_page(page);
468931caf665SNaoya Horiguchi }
4690