xref: /openbmc/linux/mm/mempolicy.c (revision fc3012896337c83a056c496d7cfb0072e1591181)
11da177e4SLinus Torvalds /*
21da177e4SLinus Torvalds  * Simple NUMA memory policy for the Linux kernel.
31da177e4SLinus Torvalds  *
41da177e4SLinus Torvalds  * Copyright 2003,2004 Andi Kleen, SuSE Labs.
58bccd85fSChristoph Lameter  * (C) Copyright 2005 Christoph Lameter, Silicon Graphics, Inc.
61da177e4SLinus Torvalds  * Subject to the GNU Public License, version 2.
71da177e4SLinus Torvalds  *
81da177e4SLinus Torvalds  * NUMA policy allows the user to give hints in which node(s) memory should
91da177e4SLinus Torvalds  * be allocated.
101da177e4SLinus Torvalds  *
111da177e4SLinus Torvalds  * Support four policies per VMA and per process:
121da177e4SLinus Torvalds  *
131da177e4SLinus Torvalds  * The VMA policy has priority over the process policy for a page fault.
141da177e4SLinus Torvalds  *
151da177e4SLinus Torvalds  * interleave     Allocate memory interleaved over a set of nodes,
161da177e4SLinus Torvalds  *                with normal fallback if it fails.
171da177e4SLinus Torvalds  *                For VMA based allocations this interleaves based on the
181da177e4SLinus Torvalds  *                offset into the backing object or offset into the mapping
191da177e4SLinus Torvalds  *                for anonymous memory. For process policy an process counter
201da177e4SLinus Torvalds  *                is used.
218bccd85fSChristoph Lameter  *
221da177e4SLinus Torvalds  * bind           Only allocate memory on a specific set of nodes,
231da177e4SLinus Torvalds  *                no fallback.
248bccd85fSChristoph Lameter  *                FIXME: memory is allocated starting with the first node
258bccd85fSChristoph Lameter  *                to the last. It would be better if bind would truly restrict
268bccd85fSChristoph Lameter  *                the allocation to memory nodes instead
278bccd85fSChristoph Lameter  *
281da177e4SLinus Torvalds  * preferred       Try a specific node first before normal fallback.
291da177e4SLinus Torvalds  *                As a special case node -1 here means do the allocation
301da177e4SLinus Torvalds  *                on the local CPU. This is normally identical to default,
311da177e4SLinus Torvalds  *                but useful to set in a VMA when you have a non default
321da177e4SLinus Torvalds  *                process policy.
338bccd85fSChristoph Lameter  *
341da177e4SLinus Torvalds  * default        Allocate on the local node first, or when on a VMA
351da177e4SLinus Torvalds  *                use the process policy. This is what Linux always did
361da177e4SLinus Torvalds  *		  in a NUMA aware kernel and still does by, ahem, default.
371da177e4SLinus Torvalds  *
381da177e4SLinus Torvalds  * The process policy is applied for most non interrupt memory allocations
391da177e4SLinus Torvalds  * in that process' context. Interrupts ignore the policies and always
401da177e4SLinus Torvalds  * try to allocate on the local CPU. The VMA policy is only applied for memory
411da177e4SLinus Torvalds  * allocations for a VMA in the VM.
421da177e4SLinus Torvalds  *
431da177e4SLinus Torvalds  * Currently there are a few corner cases in swapping where the policy
441da177e4SLinus Torvalds  * is not applied, but the majority should be handled. When process policy
451da177e4SLinus Torvalds  * is used it is not remembered over swap outs/swap ins.
461da177e4SLinus Torvalds  *
471da177e4SLinus Torvalds  * Only the highest zone in the zone hierarchy gets policied. Allocations
481da177e4SLinus Torvalds  * requesting a lower zone just use default policy. This implies that
491da177e4SLinus Torvalds  * on systems with highmem kernel lowmem allocation don't get policied.
501da177e4SLinus Torvalds  * Same with GFP_DMA allocations.
511da177e4SLinus Torvalds  *
521da177e4SLinus Torvalds  * For shmfs/tmpfs/hugetlbfs shared memory the policy is shared between
531da177e4SLinus Torvalds  * all users and remembered even when nobody has memory mapped.
541da177e4SLinus Torvalds  */
551da177e4SLinus Torvalds 
561da177e4SLinus Torvalds /* Notebook:
571da177e4SLinus Torvalds    fix mmap readahead to honour policy and enable policy for any page cache
581da177e4SLinus Torvalds    object
591da177e4SLinus Torvalds    statistics for bigpages
601da177e4SLinus Torvalds    global policy for page cache? currently it uses process policy. Requires
611da177e4SLinus Torvalds    first item above.
621da177e4SLinus Torvalds    handle mremap for shared memory (currently ignored for the policy)
631da177e4SLinus Torvalds    grows down?
641da177e4SLinus Torvalds    make bind policy root only? It can trigger oom much faster and the
651da177e4SLinus Torvalds    kernel is not always grateful with that.
661da177e4SLinus Torvalds    could replace all the switch()es with a mempolicy_ops structure.
671da177e4SLinus Torvalds */
681da177e4SLinus Torvalds 
691da177e4SLinus Torvalds #include <linux/mempolicy.h>
701da177e4SLinus Torvalds #include <linux/mm.h>
711da177e4SLinus Torvalds #include <linux/highmem.h>
721da177e4SLinus Torvalds #include <linux/hugetlb.h>
731da177e4SLinus Torvalds #include <linux/kernel.h>
741da177e4SLinus Torvalds #include <linux/sched.h>
751da177e4SLinus Torvalds #include <linux/mm.h>
761da177e4SLinus Torvalds #include <linux/nodemask.h>
771da177e4SLinus Torvalds #include <linux/cpuset.h>
781da177e4SLinus Torvalds #include <linux/gfp.h>
791da177e4SLinus Torvalds #include <linux/slab.h>
801da177e4SLinus Torvalds #include <linux/string.h>
811da177e4SLinus Torvalds #include <linux/module.h>
821da177e4SLinus Torvalds #include <linux/interrupt.h>
831da177e4SLinus Torvalds #include <linux/init.h>
841da177e4SLinus Torvalds #include <linux/compat.h>
851da177e4SLinus Torvalds #include <linux/mempolicy.h>
86dc9aa5b9SChristoph Lameter #include <linux/swap.h>
871a75a6c8SChristoph Lameter #include <linux/seq_file.h>
881a75a6c8SChristoph Lameter #include <linux/proc_fs.h>
89dc9aa5b9SChristoph Lameter 
901da177e4SLinus Torvalds #include <asm/tlbflush.h>
911da177e4SLinus Torvalds #include <asm/uaccess.h>
921da177e4SLinus Torvalds 
9338e35860SChristoph Lameter /* Internal flags */
94dc9aa5b9SChristoph Lameter #define MPOL_MF_DISCONTIG_OK (MPOL_MF_INTERNAL << 0)	/* Skip checks for continuous vmas */
9538e35860SChristoph Lameter #define MPOL_MF_INVERT (MPOL_MF_INTERNAL << 1)		/* Invert check for nodemask */
961a75a6c8SChristoph Lameter #define MPOL_MF_STATS (MPOL_MF_INTERNAL << 2)		/* Gather statistics */
97dc9aa5b9SChristoph Lameter 
981da177e4SLinus Torvalds static kmem_cache_t *policy_cache;
991da177e4SLinus Torvalds static kmem_cache_t *sn_cache;
1001da177e4SLinus Torvalds 
1011da177e4SLinus Torvalds #define PDprintk(fmt...)
1021da177e4SLinus Torvalds 
1031da177e4SLinus Torvalds /* Highest zone. An specific allocation for a zone below that is not
1041da177e4SLinus Torvalds    policied. */
1054be38e35SChristoph Lameter int policy_zone = ZONE_DMA;
1061da177e4SLinus Torvalds 
107d42c6997SAndi Kleen struct mempolicy default_policy = {
1081da177e4SLinus Torvalds 	.refcnt = ATOMIC_INIT(1), /* never free it */
1091da177e4SLinus Torvalds 	.policy = MPOL_DEFAULT,
1101da177e4SLinus Torvalds };
1111da177e4SLinus Torvalds 
1121da177e4SLinus Torvalds /* Do sanity checking on a policy */
113dfcd3c0dSAndi Kleen static int mpol_check_policy(int mode, nodemask_t *nodes)
1141da177e4SLinus Torvalds {
115dfcd3c0dSAndi Kleen 	int empty = nodes_empty(*nodes);
1161da177e4SLinus Torvalds 
1171da177e4SLinus Torvalds 	switch (mode) {
1181da177e4SLinus Torvalds 	case MPOL_DEFAULT:
1191da177e4SLinus Torvalds 		if (!empty)
1201da177e4SLinus Torvalds 			return -EINVAL;
1211da177e4SLinus Torvalds 		break;
1221da177e4SLinus Torvalds 	case MPOL_BIND:
1231da177e4SLinus Torvalds 	case MPOL_INTERLEAVE:
1241da177e4SLinus Torvalds 		/* Preferred will only use the first bit, but allow
1251da177e4SLinus Torvalds 		   more for now. */
1261da177e4SLinus Torvalds 		if (empty)
1271da177e4SLinus Torvalds 			return -EINVAL;
1281da177e4SLinus Torvalds 		break;
1291da177e4SLinus Torvalds 	}
130dfcd3c0dSAndi Kleen 	return nodes_subset(*nodes, node_online_map) ? 0 : -EINVAL;
1311da177e4SLinus Torvalds }
1321da177e4SLinus Torvalds /* Generate a custom zonelist for the BIND policy. */
133dfcd3c0dSAndi Kleen static struct zonelist *bind_zonelist(nodemask_t *nodes)
1341da177e4SLinus Torvalds {
1351da177e4SLinus Torvalds 	struct zonelist *zl;
1361da177e4SLinus Torvalds 	int num, max, nd;
1371da177e4SLinus Torvalds 
138dfcd3c0dSAndi Kleen 	max = 1 + MAX_NR_ZONES * nodes_weight(*nodes);
1391da177e4SLinus Torvalds 	zl = kmalloc(sizeof(void *) * max, GFP_KERNEL);
1401da177e4SLinus Torvalds 	if (!zl)
1411da177e4SLinus Torvalds 		return NULL;
1421da177e4SLinus Torvalds 	num = 0;
1434be38e35SChristoph Lameter 	for_each_node_mask(nd, *nodes)
1444be38e35SChristoph Lameter 		zl->zones[num++] = &NODE_DATA(nd)->node_zones[policy_zone];
1451da177e4SLinus Torvalds 	zl->zones[num] = NULL;
1461da177e4SLinus Torvalds 	return zl;
1471da177e4SLinus Torvalds }
1481da177e4SLinus Torvalds 
1491da177e4SLinus Torvalds /* Create a new policy */
150dfcd3c0dSAndi Kleen static struct mempolicy *mpol_new(int mode, nodemask_t *nodes)
1511da177e4SLinus Torvalds {
1521da177e4SLinus Torvalds 	struct mempolicy *policy;
1531da177e4SLinus Torvalds 
154dfcd3c0dSAndi Kleen 	PDprintk("setting mode %d nodes[0] %lx\n", mode, nodes_addr(*nodes)[0]);
1551da177e4SLinus Torvalds 	if (mode == MPOL_DEFAULT)
1561da177e4SLinus Torvalds 		return NULL;
1571da177e4SLinus Torvalds 	policy = kmem_cache_alloc(policy_cache, GFP_KERNEL);
1581da177e4SLinus Torvalds 	if (!policy)
1591da177e4SLinus Torvalds 		return ERR_PTR(-ENOMEM);
1601da177e4SLinus Torvalds 	atomic_set(&policy->refcnt, 1);
1611da177e4SLinus Torvalds 	switch (mode) {
1621da177e4SLinus Torvalds 	case MPOL_INTERLEAVE:
163dfcd3c0dSAndi Kleen 		policy->v.nodes = *nodes;
1648f493d79SAndi Kleen 		if (nodes_weight(*nodes) == 0) {
1658f493d79SAndi Kleen 			kmem_cache_free(policy_cache, policy);
1668f493d79SAndi Kleen 			return ERR_PTR(-EINVAL);
1678f493d79SAndi Kleen 		}
1681da177e4SLinus Torvalds 		break;
1691da177e4SLinus Torvalds 	case MPOL_PREFERRED:
170dfcd3c0dSAndi Kleen 		policy->v.preferred_node = first_node(*nodes);
1711da177e4SLinus Torvalds 		if (policy->v.preferred_node >= MAX_NUMNODES)
1721da177e4SLinus Torvalds 			policy->v.preferred_node = -1;
1731da177e4SLinus Torvalds 		break;
1741da177e4SLinus Torvalds 	case MPOL_BIND:
1751da177e4SLinus Torvalds 		policy->v.zonelist = bind_zonelist(nodes);
1761da177e4SLinus Torvalds 		if (policy->v.zonelist == NULL) {
1771da177e4SLinus Torvalds 			kmem_cache_free(policy_cache, policy);
1781da177e4SLinus Torvalds 			return ERR_PTR(-ENOMEM);
1791da177e4SLinus Torvalds 		}
1801da177e4SLinus Torvalds 		break;
1811da177e4SLinus Torvalds 	}
1821da177e4SLinus Torvalds 	policy->policy = mode;
18374cb2155SPaul Jackson 	policy->cpuset_mems_allowed = cpuset_mems_allowed(current);
1841da177e4SLinus Torvalds 	return policy;
1851da177e4SLinus Torvalds }
1861da177e4SLinus Torvalds 
1871a75a6c8SChristoph Lameter static void gather_stats(struct page *, void *);
188*fc301289SChristoph Lameter static void migrate_page_add(struct page *page, struct list_head *pagelist,
189*fc301289SChristoph Lameter 				unsigned long flags);
1901a75a6c8SChristoph Lameter 
19138e35860SChristoph Lameter /* Scan through pages checking if pages follow certain conditions. */
192b5810039SNick Piggin static int check_pte_range(struct vm_area_struct *vma, pmd_t *pmd,
193dc9aa5b9SChristoph Lameter 		unsigned long addr, unsigned long end,
194dc9aa5b9SChristoph Lameter 		const nodemask_t *nodes, unsigned long flags,
19538e35860SChristoph Lameter 		void *private)
1961da177e4SLinus Torvalds {
19791612e0dSHugh Dickins 	pte_t *orig_pte;
19891612e0dSHugh Dickins 	pte_t *pte;
199705e87c0SHugh Dickins 	spinlock_t *ptl;
200941150a3SHugh Dickins 
201705e87c0SHugh Dickins 	orig_pte = pte = pte_offset_map_lock(vma->vm_mm, pmd, addr, &ptl);
20291612e0dSHugh Dickins 	do {
2036aab341eSLinus Torvalds 		struct page *page;
20491612e0dSHugh Dickins 		unsigned int nid;
20591612e0dSHugh Dickins 
20691612e0dSHugh Dickins 		if (!pte_present(*pte))
20791612e0dSHugh Dickins 			continue;
2086aab341eSLinus Torvalds 		page = vm_normal_page(vma, addr, *pte);
2096aab341eSLinus Torvalds 		if (!page)
21091612e0dSHugh Dickins 			continue;
211053837fcSNick Piggin 		/*
212053837fcSNick Piggin 		 * The check for PageReserved here is important to avoid
213053837fcSNick Piggin 		 * handling zero pages and other pages that may have been
214053837fcSNick Piggin 		 * marked special by the system.
215053837fcSNick Piggin 		 *
216053837fcSNick Piggin 		 * If the PageReserved would not be checked here then f.e.
217053837fcSNick Piggin 		 * the location of the zero page could have an influence
218053837fcSNick Piggin 		 * on MPOL_MF_STRICT, zero pages would be counted for
219053837fcSNick Piggin 		 * the per node stats, and there would be useless attempts
220053837fcSNick Piggin 		 * to put zero pages on the migration list.
221053837fcSNick Piggin 		 */
222f4598c8bSChristoph Lameter 		if (PageReserved(page))
223f4598c8bSChristoph Lameter 			continue;
2246aab341eSLinus Torvalds 		nid = page_to_nid(page);
22538e35860SChristoph Lameter 		if (node_isset(nid, *nodes) == !!(flags & MPOL_MF_INVERT))
22638e35860SChristoph Lameter 			continue;
22738e35860SChristoph Lameter 
2281a75a6c8SChristoph Lameter 		if (flags & MPOL_MF_STATS)
2291a75a6c8SChristoph Lameter 			gather_stats(page, private);
230053837fcSNick Piggin 		else if (flags & (MPOL_MF_MOVE | MPOL_MF_MOVE_ALL))
231*fc301289SChristoph Lameter 			migrate_page_add(page, private, flags);
232dc9aa5b9SChristoph Lameter 		else
2331da177e4SLinus Torvalds 			break;
23491612e0dSHugh Dickins 	} while (pte++, addr += PAGE_SIZE, addr != end);
235705e87c0SHugh Dickins 	pte_unmap_unlock(orig_pte, ptl);
23691612e0dSHugh Dickins 	return addr != end;
23791612e0dSHugh Dickins }
23891612e0dSHugh Dickins 
239b5810039SNick Piggin static inline int check_pmd_range(struct vm_area_struct *vma, pud_t *pud,
240dc9aa5b9SChristoph Lameter 		unsigned long addr, unsigned long end,
241dc9aa5b9SChristoph Lameter 		const nodemask_t *nodes, unsigned long flags,
24238e35860SChristoph Lameter 		void *private)
24391612e0dSHugh Dickins {
24491612e0dSHugh Dickins 	pmd_t *pmd;
24591612e0dSHugh Dickins 	unsigned long next;
24691612e0dSHugh Dickins 
24791612e0dSHugh Dickins 	pmd = pmd_offset(pud, addr);
24891612e0dSHugh Dickins 	do {
24991612e0dSHugh Dickins 		next = pmd_addr_end(addr, end);
25091612e0dSHugh Dickins 		if (pmd_none_or_clear_bad(pmd))
25191612e0dSHugh Dickins 			continue;
252dc9aa5b9SChristoph Lameter 		if (check_pte_range(vma, pmd, addr, next, nodes,
25338e35860SChristoph Lameter 				    flags, private))
25491612e0dSHugh Dickins 			return -EIO;
25591612e0dSHugh Dickins 	} while (pmd++, addr = next, addr != end);
25691612e0dSHugh Dickins 	return 0;
25791612e0dSHugh Dickins }
25891612e0dSHugh Dickins 
259b5810039SNick Piggin static inline int check_pud_range(struct vm_area_struct *vma, pgd_t *pgd,
260dc9aa5b9SChristoph Lameter 		unsigned long addr, unsigned long end,
261dc9aa5b9SChristoph Lameter 		const nodemask_t *nodes, unsigned long flags,
26238e35860SChristoph Lameter 		void *private)
26391612e0dSHugh Dickins {
26491612e0dSHugh Dickins 	pud_t *pud;
26591612e0dSHugh Dickins 	unsigned long next;
26691612e0dSHugh Dickins 
26791612e0dSHugh Dickins 	pud = pud_offset(pgd, addr);
26891612e0dSHugh Dickins 	do {
26991612e0dSHugh Dickins 		next = pud_addr_end(addr, end);
27091612e0dSHugh Dickins 		if (pud_none_or_clear_bad(pud))
27191612e0dSHugh Dickins 			continue;
272dc9aa5b9SChristoph Lameter 		if (check_pmd_range(vma, pud, addr, next, nodes,
27338e35860SChristoph Lameter 				    flags, private))
27491612e0dSHugh Dickins 			return -EIO;
27591612e0dSHugh Dickins 	} while (pud++, addr = next, addr != end);
27691612e0dSHugh Dickins 	return 0;
27791612e0dSHugh Dickins }
27891612e0dSHugh Dickins 
279b5810039SNick Piggin static inline int check_pgd_range(struct vm_area_struct *vma,
280dc9aa5b9SChristoph Lameter 		unsigned long addr, unsigned long end,
281dc9aa5b9SChristoph Lameter 		const nodemask_t *nodes, unsigned long flags,
28238e35860SChristoph Lameter 		void *private)
28391612e0dSHugh Dickins {
28491612e0dSHugh Dickins 	pgd_t *pgd;
28591612e0dSHugh Dickins 	unsigned long next;
28691612e0dSHugh Dickins 
287b5810039SNick Piggin 	pgd = pgd_offset(vma->vm_mm, addr);
28891612e0dSHugh Dickins 	do {
28991612e0dSHugh Dickins 		next = pgd_addr_end(addr, end);
29091612e0dSHugh Dickins 		if (pgd_none_or_clear_bad(pgd))
29191612e0dSHugh Dickins 			continue;
292dc9aa5b9SChristoph Lameter 		if (check_pud_range(vma, pgd, addr, next, nodes,
29338e35860SChristoph Lameter 				    flags, private))
29491612e0dSHugh Dickins 			return -EIO;
29591612e0dSHugh Dickins 	} while (pgd++, addr = next, addr != end);
29691612e0dSHugh Dickins 	return 0;
2971da177e4SLinus Torvalds }
2981da177e4SLinus Torvalds 
299dc9aa5b9SChristoph Lameter /* Check if a vma is migratable */
300dc9aa5b9SChristoph Lameter static inline int vma_migratable(struct vm_area_struct *vma)
301dc9aa5b9SChristoph Lameter {
302dc9aa5b9SChristoph Lameter 	if (vma->vm_flags & (
303f4598c8bSChristoph Lameter 		VM_LOCKED|VM_IO|VM_HUGETLB|VM_PFNMAP|VM_RESERVED))
304dc9aa5b9SChristoph Lameter 		return 0;
305dc9aa5b9SChristoph Lameter 	return 1;
306dc9aa5b9SChristoph Lameter }
307dc9aa5b9SChristoph Lameter 
308dc9aa5b9SChristoph Lameter /*
309dc9aa5b9SChristoph Lameter  * Check if all pages in a range are on a set of nodes.
310dc9aa5b9SChristoph Lameter  * If pagelist != NULL then isolate pages from the LRU and
311dc9aa5b9SChristoph Lameter  * put them on the pagelist.
312dc9aa5b9SChristoph Lameter  */
3131da177e4SLinus Torvalds static struct vm_area_struct *
3141da177e4SLinus Torvalds check_range(struct mm_struct *mm, unsigned long start, unsigned long end,
31538e35860SChristoph Lameter 		const nodemask_t *nodes, unsigned long flags, void *private)
3161da177e4SLinus Torvalds {
3171da177e4SLinus Torvalds 	int err;
3181da177e4SLinus Torvalds 	struct vm_area_struct *first, *vma, *prev;
3191da177e4SLinus Torvalds 
320053837fcSNick Piggin 	/* Clear the LRU lists so pages can be isolated */
321053837fcSNick Piggin 	if (flags & (MPOL_MF_MOVE | MPOL_MF_MOVE_ALL))
322053837fcSNick Piggin 		lru_add_drain_all();
323053837fcSNick Piggin 
3241da177e4SLinus Torvalds 	first = find_vma(mm, start);
3251da177e4SLinus Torvalds 	if (!first)
3261da177e4SLinus Torvalds 		return ERR_PTR(-EFAULT);
3271da177e4SLinus Torvalds 	prev = NULL;
3281da177e4SLinus Torvalds 	for (vma = first; vma && vma->vm_start < end; vma = vma->vm_next) {
329dc9aa5b9SChristoph Lameter 		if (!(flags & MPOL_MF_DISCONTIG_OK)) {
3301da177e4SLinus Torvalds 			if (!vma->vm_next && vma->vm_end < end)
3311da177e4SLinus Torvalds 				return ERR_PTR(-EFAULT);
3321da177e4SLinus Torvalds 			if (prev && prev->vm_end < vma->vm_start)
3331da177e4SLinus Torvalds 				return ERR_PTR(-EFAULT);
334dc9aa5b9SChristoph Lameter 		}
335dc9aa5b9SChristoph Lameter 		if (!is_vm_hugetlb_page(vma) &&
336dc9aa5b9SChristoph Lameter 		    ((flags & MPOL_MF_STRICT) ||
337dc9aa5b9SChristoph Lameter 		     ((flags & (MPOL_MF_MOVE | MPOL_MF_MOVE_ALL)) &&
338dc9aa5b9SChristoph Lameter 				vma_migratable(vma)))) {
3395b952b3cSAndi Kleen 			unsigned long endvma = vma->vm_end;
340dc9aa5b9SChristoph Lameter 
3415b952b3cSAndi Kleen 			if (endvma > end)
3425b952b3cSAndi Kleen 				endvma = end;
3435b952b3cSAndi Kleen 			if (vma->vm_start > start)
3445b952b3cSAndi Kleen 				start = vma->vm_start;
345dc9aa5b9SChristoph Lameter 			err = check_pgd_range(vma, start, endvma, nodes,
34638e35860SChristoph Lameter 						flags, private);
3471da177e4SLinus Torvalds 			if (err) {
3481da177e4SLinus Torvalds 				first = ERR_PTR(err);
3491da177e4SLinus Torvalds 				break;
3501da177e4SLinus Torvalds 			}
3511da177e4SLinus Torvalds 		}
3521da177e4SLinus Torvalds 		prev = vma;
3531da177e4SLinus Torvalds 	}
3541da177e4SLinus Torvalds 	return first;
3551da177e4SLinus Torvalds }
3561da177e4SLinus Torvalds 
3571da177e4SLinus Torvalds /* Apply policy to a single VMA */
3581da177e4SLinus Torvalds static int policy_vma(struct vm_area_struct *vma, struct mempolicy *new)
3591da177e4SLinus Torvalds {
3601da177e4SLinus Torvalds 	int err = 0;
3611da177e4SLinus Torvalds 	struct mempolicy *old = vma->vm_policy;
3621da177e4SLinus Torvalds 
3631da177e4SLinus Torvalds 	PDprintk("vma %lx-%lx/%lx vm_ops %p vm_file %p set_policy %p\n",
3641da177e4SLinus Torvalds 		 vma->vm_start, vma->vm_end, vma->vm_pgoff,
3651da177e4SLinus Torvalds 		 vma->vm_ops, vma->vm_file,
3661da177e4SLinus Torvalds 		 vma->vm_ops ? vma->vm_ops->set_policy : NULL);
3671da177e4SLinus Torvalds 
3681da177e4SLinus Torvalds 	if (vma->vm_ops && vma->vm_ops->set_policy)
3691da177e4SLinus Torvalds 		err = vma->vm_ops->set_policy(vma, new);
3701da177e4SLinus Torvalds 	if (!err) {
3711da177e4SLinus Torvalds 		mpol_get(new);
3721da177e4SLinus Torvalds 		vma->vm_policy = new;
3731da177e4SLinus Torvalds 		mpol_free(old);
3741da177e4SLinus Torvalds 	}
3751da177e4SLinus Torvalds 	return err;
3761da177e4SLinus Torvalds }
3771da177e4SLinus Torvalds 
3781da177e4SLinus Torvalds /* Step 2: apply policy to a range and do splits. */
3791da177e4SLinus Torvalds static int mbind_range(struct vm_area_struct *vma, unsigned long start,
3801da177e4SLinus Torvalds 		       unsigned long end, struct mempolicy *new)
3811da177e4SLinus Torvalds {
3821da177e4SLinus Torvalds 	struct vm_area_struct *next;
3831da177e4SLinus Torvalds 	int err;
3841da177e4SLinus Torvalds 
3851da177e4SLinus Torvalds 	err = 0;
3861da177e4SLinus Torvalds 	for (; vma && vma->vm_start < end; vma = next) {
3871da177e4SLinus Torvalds 		next = vma->vm_next;
3881da177e4SLinus Torvalds 		if (vma->vm_start < start)
3891da177e4SLinus Torvalds 			err = split_vma(vma->vm_mm, vma, start, 1);
3901da177e4SLinus Torvalds 		if (!err && vma->vm_end > end)
3911da177e4SLinus Torvalds 			err = split_vma(vma->vm_mm, vma, end, 0);
3921da177e4SLinus Torvalds 		if (!err)
3931da177e4SLinus Torvalds 			err = policy_vma(vma, new);
3941da177e4SLinus Torvalds 		if (err)
3951da177e4SLinus Torvalds 			break;
3961da177e4SLinus Torvalds 	}
3971da177e4SLinus Torvalds 	return err;
3981da177e4SLinus Torvalds }
3991da177e4SLinus Torvalds 
4008bccd85fSChristoph Lameter static int contextualize_policy(int mode, nodemask_t *nodes)
4018bccd85fSChristoph Lameter {
4028bccd85fSChristoph Lameter 	if (!nodes)
4038bccd85fSChristoph Lameter 		return 0;
4048bccd85fSChristoph Lameter 
405cf2a473cSPaul Jackson 	cpuset_update_task_memory_state();
4065966514dSPaul Jackson 	if (!cpuset_nodes_subset_current_mems_allowed(*nodes))
4075966514dSPaul Jackson 		return -EINVAL;
4088bccd85fSChristoph Lameter 	return mpol_check_policy(mode, nodes);
4098bccd85fSChristoph Lameter }
4108bccd85fSChristoph Lameter 
4111da177e4SLinus Torvalds /* Set the process memory policy */
4128bccd85fSChristoph Lameter long do_set_mempolicy(int mode, nodemask_t *nodes)
4131da177e4SLinus Torvalds {
4141da177e4SLinus Torvalds 	struct mempolicy *new;
4151da177e4SLinus Torvalds 
4168bccd85fSChristoph Lameter 	if (contextualize_policy(mode, nodes))
4171da177e4SLinus Torvalds 		return -EINVAL;
4188bccd85fSChristoph Lameter 	new = mpol_new(mode, nodes);
4191da177e4SLinus Torvalds 	if (IS_ERR(new))
4201da177e4SLinus Torvalds 		return PTR_ERR(new);
4211da177e4SLinus Torvalds 	mpol_free(current->mempolicy);
4221da177e4SLinus Torvalds 	current->mempolicy = new;
4231da177e4SLinus Torvalds 	if (new && new->policy == MPOL_INTERLEAVE)
424dfcd3c0dSAndi Kleen 		current->il_next = first_node(new->v.nodes);
4251da177e4SLinus Torvalds 	return 0;
4261da177e4SLinus Torvalds }
4271da177e4SLinus Torvalds 
4281da177e4SLinus Torvalds /* Fill a zone bitmap for a policy */
429dfcd3c0dSAndi Kleen static void get_zonemask(struct mempolicy *p, nodemask_t *nodes)
4301da177e4SLinus Torvalds {
4311da177e4SLinus Torvalds 	int i;
4321da177e4SLinus Torvalds 
433dfcd3c0dSAndi Kleen 	nodes_clear(*nodes);
4341da177e4SLinus Torvalds 	switch (p->policy) {
4351da177e4SLinus Torvalds 	case MPOL_BIND:
4361da177e4SLinus Torvalds 		for (i = 0; p->v.zonelist->zones[i]; i++)
4378bccd85fSChristoph Lameter 			node_set(p->v.zonelist->zones[i]->zone_pgdat->node_id,
4388bccd85fSChristoph Lameter 				*nodes);
4391da177e4SLinus Torvalds 		break;
4401da177e4SLinus Torvalds 	case MPOL_DEFAULT:
4411da177e4SLinus Torvalds 		break;
4421da177e4SLinus Torvalds 	case MPOL_INTERLEAVE:
443dfcd3c0dSAndi Kleen 		*nodes = p->v.nodes;
4441da177e4SLinus Torvalds 		break;
4451da177e4SLinus Torvalds 	case MPOL_PREFERRED:
4461da177e4SLinus Torvalds 		/* or use current node instead of online map? */
4471da177e4SLinus Torvalds 		if (p->v.preferred_node < 0)
448dfcd3c0dSAndi Kleen 			*nodes = node_online_map;
4491da177e4SLinus Torvalds 		else
450dfcd3c0dSAndi Kleen 			node_set(p->v.preferred_node, *nodes);
4511da177e4SLinus Torvalds 		break;
4521da177e4SLinus Torvalds 	default:
4531da177e4SLinus Torvalds 		BUG();
4541da177e4SLinus Torvalds 	}
4551da177e4SLinus Torvalds }
4561da177e4SLinus Torvalds 
4571da177e4SLinus Torvalds static int lookup_node(struct mm_struct *mm, unsigned long addr)
4581da177e4SLinus Torvalds {
4591da177e4SLinus Torvalds 	struct page *p;
4601da177e4SLinus Torvalds 	int err;
4611da177e4SLinus Torvalds 
4621da177e4SLinus Torvalds 	err = get_user_pages(current, mm, addr & PAGE_MASK, 1, 0, 0, &p, NULL);
4631da177e4SLinus Torvalds 	if (err >= 0) {
4641da177e4SLinus Torvalds 		err = page_to_nid(p);
4651da177e4SLinus Torvalds 		put_page(p);
4661da177e4SLinus Torvalds 	}
4671da177e4SLinus Torvalds 	return err;
4681da177e4SLinus Torvalds }
4691da177e4SLinus Torvalds 
4701da177e4SLinus Torvalds /* Retrieve NUMA policy */
4718bccd85fSChristoph Lameter long do_get_mempolicy(int *policy, nodemask_t *nmask,
4721da177e4SLinus Torvalds 			unsigned long addr, unsigned long flags)
4731da177e4SLinus Torvalds {
4748bccd85fSChristoph Lameter 	int err;
4751da177e4SLinus Torvalds 	struct mm_struct *mm = current->mm;
4761da177e4SLinus Torvalds 	struct vm_area_struct *vma = NULL;
4771da177e4SLinus Torvalds 	struct mempolicy *pol = current->mempolicy;
4781da177e4SLinus Torvalds 
479cf2a473cSPaul Jackson 	cpuset_update_task_memory_state();
4801da177e4SLinus Torvalds 	if (flags & ~(unsigned long)(MPOL_F_NODE|MPOL_F_ADDR))
4811da177e4SLinus Torvalds 		return -EINVAL;
4821da177e4SLinus Torvalds 	if (flags & MPOL_F_ADDR) {
4831da177e4SLinus Torvalds 		down_read(&mm->mmap_sem);
4841da177e4SLinus Torvalds 		vma = find_vma_intersection(mm, addr, addr+1);
4851da177e4SLinus Torvalds 		if (!vma) {
4861da177e4SLinus Torvalds 			up_read(&mm->mmap_sem);
4871da177e4SLinus Torvalds 			return -EFAULT;
4881da177e4SLinus Torvalds 		}
4891da177e4SLinus Torvalds 		if (vma->vm_ops && vma->vm_ops->get_policy)
4901da177e4SLinus Torvalds 			pol = vma->vm_ops->get_policy(vma, addr);
4911da177e4SLinus Torvalds 		else
4921da177e4SLinus Torvalds 			pol = vma->vm_policy;
4931da177e4SLinus Torvalds 	} else if (addr)
4941da177e4SLinus Torvalds 		return -EINVAL;
4951da177e4SLinus Torvalds 
4961da177e4SLinus Torvalds 	if (!pol)
4971da177e4SLinus Torvalds 		pol = &default_policy;
4981da177e4SLinus Torvalds 
4991da177e4SLinus Torvalds 	if (flags & MPOL_F_NODE) {
5001da177e4SLinus Torvalds 		if (flags & MPOL_F_ADDR) {
5011da177e4SLinus Torvalds 			err = lookup_node(mm, addr);
5021da177e4SLinus Torvalds 			if (err < 0)
5031da177e4SLinus Torvalds 				goto out;
5048bccd85fSChristoph Lameter 			*policy = err;
5051da177e4SLinus Torvalds 		} else if (pol == current->mempolicy &&
5061da177e4SLinus Torvalds 				pol->policy == MPOL_INTERLEAVE) {
5078bccd85fSChristoph Lameter 			*policy = current->il_next;
5081da177e4SLinus Torvalds 		} else {
5091da177e4SLinus Torvalds 			err = -EINVAL;
5101da177e4SLinus Torvalds 			goto out;
5111da177e4SLinus Torvalds 		}
5121da177e4SLinus Torvalds 	} else
5138bccd85fSChristoph Lameter 		*policy = pol->policy;
5141da177e4SLinus Torvalds 
5151da177e4SLinus Torvalds 	if (vma) {
5161da177e4SLinus Torvalds 		up_read(&current->mm->mmap_sem);
5171da177e4SLinus Torvalds 		vma = NULL;
5181da177e4SLinus Torvalds 	}
5191da177e4SLinus Torvalds 
5201da177e4SLinus Torvalds 	err = 0;
5218bccd85fSChristoph Lameter 	if (nmask)
5228bccd85fSChristoph Lameter 		get_zonemask(pol, nmask);
5231da177e4SLinus Torvalds 
5241da177e4SLinus Torvalds  out:
5251da177e4SLinus Torvalds 	if (vma)
5261da177e4SLinus Torvalds 		up_read(&current->mm->mmap_sem);
5271da177e4SLinus Torvalds 	return err;
5281da177e4SLinus Torvalds }
5291da177e4SLinus Torvalds 
5308bccd85fSChristoph Lameter /*
5316ce3c4c0SChristoph Lameter  * page migration
5326ce3c4c0SChristoph Lameter  */
5336ce3c4c0SChristoph Lameter 
534*fc301289SChristoph Lameter static void migrate_page_add(struct page *page, struct list_head *pagelist,
535*fc301289SChristoph Lameter 				unsigned long flags)
5366ce3c4c0SChristoph Lameter {
5376ce3c4c0SChristoph Lameter 	/*
538*fc301289SChristoph Lameter 	 * Avoid migrating a page that is shared with others.
5396ce3c4c0SChristoph Lameter 	 */
540*fc301289SChristoph Lameter 	if ((flags & MPOL_MF_MOVE_ALL) || page_mapcount(page) == 1) {
541053837fcSNick Piggin 		if (isolate_lru_page(page))
5426ce3c4c0SChristoph Lameter 			list_add(&page->lru, pagelist);
5436ce3c4c0SChristoph Lameter 	}
5446ce3c4c0SChristoph Lameter }
5456ce3c4c0SChristoph Lameter 
5466ce3c4c0SChristoph Lameter static int swap_pages(struct list_head *pagelist)
5476ce3c4c0SChristoph Lameter {
5486ce3c4c0SChristoph Lameter 	LIST_HEAD(moved);
5496ce3c4c0SChristoph Lameter 	LIST_HEAD(failed);
5506ce3c4c0SChristoph Lameter 	int n;
5516ce3c4c0SChristoph Lameter 
5526ce3c4c0SChristoph Lameter 	n = migrate_pages(pagelist, NULL, &moved, &failed);
5536ce3c4c0SChristoph Lameter 	putback_lru_pages(&failed);
5546ce3c4c0SChristoph Lameter 	putback_lru_pages(&moved);
5556ce3c4c0SChristoph Lameter 
5566ce3c4c0SChristoph Lameter 	return n;
5576ce3c4c0SChristoph Lameter }
5586ce3c4c0SChristoph Lameter 
5596ce3c4c0SChristoph Lameter /*
56039743889SChristoph Lameter  * For now migrate_pages simply swaps out the pages from nodes that are in
56139743889SChristoph Lameter  * the source set but not in the target set. In the future, we would
56239743889SChristoph Lameter  * want a function that moves pages between the two nodesets in such
56339743889SChristoph Lameter  * a way as to preserve the physical layout as much as possible.
56439743889SChristoph Lameter  *
56539743889SChristoph Lameter  * Returns the number of page that could not be moved.
56639743889SChristoph Lameter  */
56739743889SChristoph Lameter int do_migrate_pages(struct mm_struct *mm,
56839743889SChristoph Lameter 	const nodemask_t *from_nodes, const nodemask_t *to_nodes, int flags)
56939743889SChristoph Lameter {
57039743889SChristoph Lameter 	LIST_HEAD(pagelist);
57139743889SChristoph Lameter 	int count = 0;
57239743889SChristoph Lameter 	nodemask_t nodes;
57339743889SChristoph Lameter 
57439743889SChristoph Lameter 	nodes_andnot(nodes, *from_nodes, *to_nodes);
57539743889SChristoph Lameter 
57639743889SChristoph Lameter 	down_read(&mm->mmap_sem);
57739743889SChristoph Lameter 	check_range(mm, mm->mmap->vm_start, TASK_SIZE, &nodes,
57839743889SChristoph Lameter 			flags | MPOL_MF_DISCONTIG_OK, &pagelist);
579d4984711SChristoph Lameter 
58039743889SChristoph Lameter 	if (!list_empty(&pagelist)) {
581d4984711SChristoph Lameter 		count = swap_pages(&pagelist);
582d4984711SChristoph Lameter 		putback_lru_pages(&pagelist);
58339743889SChristoph Lameter 	}
584d4984711SChristoph Lameter 
58539743889SChristoph Lameter 	up_read(&mm->mmap_sem);
58639743889SChristoph Lameter 	return count;
58739743889SChristoph Lameter }
58839743889SChristoph Lameter 
5896ce3c4c0SChristoph Lameter long do_mbind(unsigned long start, unsigned long len,
5906ce3c4c0SChristoph Lameter 		unsigned long mode, nodemask_t *nmask, unsigned long flags)
5916ce3c4c0SChristoph Lameter {
5926ce3c4c0SChristoph Lameter 	struct vm_area_struct *vma;
5936ce3c4c0SChristoph Lameter 	struct mm_struct *mm = current->mm;
5946ce3c4c0SChristoph Lameter 	struct mempolicy *new;
5956ce3c4c0SChristoph Lameter 	unsigned long end;
5966ce3c4c0SChristoph Lameter 	int err;
5976ce3c4c0SChristoph Lameter 	LIST_HEAD(pagelist);
5986ce3c4c0SChristoph Lameter 
5996ce3c4c0SChristoph Lameter 	if ((flags & ~(unsigned long)(MPOL_MF_STRICT |
6006ce3c4c0SChristoph Lameter 				      MPOL_MF_MOVE | MPOL_MF_MOVE_ALL))
6016ce3c4c0SChristoph Lameter 	    || mode > MPOL_MAX)
6026ce3c4c0SChristoph Lameter 		return -EINVAL;
6036ce3c4c0SChristoph Lameter 	if ((flags & MPOL_MF_MOVE_ALL) && !capable(CAP_SYS_RESOURCE))
6046ce3c4c0SChristoph Lameter 		return -EPERM;
6056ce3c4c0SChristoph Lameter 
6066ce3c4c0SChristoph Lameter 	if (start & ~PAGE_MASK)
6076ce3c4c0SChristoph Lameter 		return -EINVAL;
6086ce3c4c0SChristoph Lameter 
6096ce3c4c0SChristoph Lameter 	if (mode == MPOL_DEFAULT)
6106ce3c4c0SChristoph Lameter 		flags &= ~MPOL_MF_STRICT;
6116ce3c4c0SChristoph Lameter 
6126ce3c4c0SChristoph Lameter 	len = (len + PAGE_SIZE - 1) & PAGE_MASK;
6136ce3c4c0SChristoph Lameter 	end = start + len;
6146ce3c4c0SChristoph Lameter 
6156ce3c4c0SChristoph Lameter 	if (end < start)
6166ce3c4c0SChristoph Lameter 		return -EINVAL;
6176ce3c4c0SChristoph Lameter 	if (end == start)
6186ce3c4c0SChristoph Lameter 		return 0;
6196ce3c4c0SChristoph Lameter 
6206ce3c4c0SChristoph Lameter 	if (mpol_check_policy(mode, nmask))
6216ce3c4c0SChristoph Lameter 		return -EINVAL;
6226ce3c4c0SChristoph Lameter 
6236ce3c4c0SChristoph Lameter 	new = mpol_new(mode, nmask);
6246ce3c4c0SChristoph Lameter 	if (IS_ERR(new))
6256ce3c4c0SChristoph Lameter 		return PTR_ERR(new);
6266ce3c4c0SChristoph Lameter 
6276ce3c4c0SChristoph Lameter 	/*
6286ce3c4c0SChristoph Lameter 	 * If we are using the default policy then operation
6296ce3c4c0SChristoph Lameter 	 * on discontinuous address spaces is okay after all
6306ce3c4c0SChristoph Lameter 	 */
6316ce3c4c0SChristoph Lameter 	if (!new)
6326ce3c4c0SChristoph Lameter 		flags |= MPOL_MF_DISCONTIG_OK;
6336ce3c4c0SChristoph Lameter 
6346ce3c4c0SChristoph Lameter 	PDprintk("mbind %lx-%lx mode:%ld nodes:%lx\n",start,start+len,
6356ce3c4c0SChristoph Lameter 			mode,nodes_addr(nodes)[0]);
6366ce3c4c0SChristoph Lameter 
6376ce3c4c0SChristoph Lameter 	down_write(&mm->mmap_sem);
6386ce3c4c0SChristoph Lameter 	vma = check_range(mm, start, end, nmask,
6396ce3c4c0SChristoph Lameter 			  flags | MPOL_MF_INVERT, &pagelist);
6406ce3c4c0SChristoph Lameter 
6416ce3c4c0SChristoph Lameter 	err = PTR_ERR(vma);
6426ce3c4c0SChristoph Lameter 	if (!IS_ERR(vma)) {
6436ce3c4c0SChristoph Lameter 		int nr_failed = 0;
6446ce3c4c0SChristoph Lameter 
6456ce3c4c0SChristoph Lameter 		err = mbind_range(vma, start, end, new);
6466ce3c4c0SChristoph Lameter 		if (!list_empty(&pagelist))
6476ce3c4c0SChristoph Lameter 			nr_failed = swap_pages(&pagelist);
6486ce3c4c0SChristoph Lameter 
6496ce3c4c0SChristoph Lameter 		if (!err && nr_failed && (flags & MPOL_MF_STRICT))
6506ce3c4c0SChristoph Lameter 			err = -EIO;
6516ce3c4c0SChristoph Lameter 	}
6526ce3c4c0SChristoph Lameter 	if (!list_empty(&pagelist))
6536ce3c4c0SChristoph Lameter 		putback_lru_pages(&pagelist);
6546ce3c4c0SChristoph Lameter 
6556ce3c4c0SChristoph Lameter 	up_write(&mm->mmap_sem);
6566ce3c4c0SChristoph Lameter 	mpol_free(new);
6576ce3c4c0SChristoph Lameter 	return err;
6586ce3c4c0SChristoph Lameter }
6596ce3c4c0SChristoph Lameter 
66039743889SChristoph Lameter /*
6618bccd85fSChristoph Lameter  * User space interface with variable sized bitmaps for nodelists.
6628bccd85fSChristoph Lameter  */
6638bccd85fSChristoph Lameter 
6648bccd85fSChristoph Lameter /* Copy a node mask from user space. */
66539743889SChristoph Lameter static int get_nodes(nodemask_t *nodes, const unsigned long __user *nmask,
6668bccd85fSChristoph Lameter 		     unsigned long maxnode)
6678bccd85fSChristoph Lameter {
6688bccd85fSChristoph Lameter 	unsigned long k;
6698bccd85fSChristoph Lameter 	unsigned long nlongs;
6708bccd85fSChristoph Lameter 	unsigned long endmask;
6718bccd85fSChristoph Lameter 
6728bccd85fSChristoph Lameter 	--maxnode;
6738bccd85fSChristoph Lameter 	nodes_clear(*nodes);
6748bccd85fSChristoph Lameter 	if (maxnode == 0 || !nmask)
6758bccd85fSChristoph Lameter 		return 0;
6768bccd85fSChristoph Lameter 
6778bccd85fSChristoph Lameter 	nlongs = BITS_TO_LONGS(maxnode);
6788bccd85fSChristoph Lameter 	if ((maxnode % BITS_PER_LONG) == 0)
6798bccd85fSChristoph Lameter 		endmask = ~0UL;
6808bccd85fSChristoph Lameter 	else
6818bccd85fSChristoph Lameter 		endmask = (1UL << (maxnode % BITS_PER_LONG)) - 1;
6828bccd85fSChristoph Lameter 
6838bccd85fSChristoph Lameter 	/* When the user specified more nodes than supported just check
6848bccd85fSChristoph Lameter 	   if the non supported part is all zero. */
6858bccd85fSChristoph Lameter 	if (nlongs > BITS_TO_LONGS(MAX_NUMNODES)) {
6868bccd85fSChristoph Lameter 		if (nlongs > PAGE_SIZE/sizeof(long))
6878bccd85fSChristoph Lameter 			return -EINVAL;
6888bccd85fSChristoph Lameter 		for (k = BITS_TO_LONGS(MAX_NUMNODES); k < nlongs; k++) {
6898bccd85fSChristoph Lameter 			unsigned long t;
6908bccd85fSChristoph Lameter 			if (get_user(t, nmask + k))
6918bccd85fSChristoph Lameter 				return -EFAULT;
6928bccd85fSChristoph Lameter 			if (k == nlongs - 1) {
6938bccd85fSChristoph Lameter 				if (t & endmask)
6948bccd85fSChristoph Lameter 					return -EINVAL;
6958bccd85fSChristoph Lameter 			} else if (t)
6968bccd85fSChristoph Lameter 				return -EINVAL;
6978bccd85fSChristoph Lameter 		}
6988bccd85fSChristoph Lameter 		nlongs = BITS_TO_LONGS(MAX_NUMNODES);
6998bccd85fSChristoph Lameter 		endmask = ~0UL;
7008bccd85fSChristoph Lameter 	}
7018bccd85fSChristoph Lameter 
7028bccd85fSChristoph Lameter 	if (copy_from_user(nodes_addr(*nodes), nmask, nlongs*sizeof(unsigned long)))
7038bccd85fSChristoph Lameter 		return -EFAULT;
7048bccd85fSChristoph Lameter 	nodes_addr(*nodes)[nlongs-1] &= endmask;
7058bccd85fSChristoph Lameter 	return 0;
7068bccd85fSChristoph Lameter }
7078bccd85fSChristoph Lameter 
7088bccd85fSChristoph Lameter /* Copy a kernel node mask to user space */
7098bccd85fSChristoph Lameter static int copy_nodes_to_user(unsigned long __user *mask, unsigned long maxnode,
7108bccd85fSChristoph Lameter 			      nodemask_t *nodes)
7118bccd85fSChristoph Lameter {
7128bccd85fSChristoph Lameter 	unsigned long copy = ALIGN(maxnode-1, 64) / 8;
7138bccd85fSChristoph Lameter 	const int nbytes = BITS_TO_LONGS(MAX_NUMNODES) * sizeof(long);
7148bccd85fSChristoph Lameter 
7158bccd85fSChristoph Lameter 	if (copy > nbytes) {
7168bccd85fSChristoph Lameter 		if (copy > PAGE_SIZE)
7178bccd85fSChristoph Lameter 			return -EINVAL;
7188bccd85fSChristoph Lameter 		if (clear_user((char __user *)mask + nbytes, copy - nbytes))
7198bccd85fSChristoph Lameter 			return -EFAULT;
7208bccd85fSChristoph Lameter 		copy = nbytes;
7218bccd85fSChristoph Lameter 	}
7228bccd85fSChristoph Lameter 	return copy_to_user(mask, nodes_addr(*nodes), copy) ? -EFAULT : 0;
7238bccd85fSChristoph Lameter }
7248bccd85fSChristoph Lameter 
7258bccd85fSChristoph Lameter asmlinkage long sys_mbind(unsigned long start, unsigned long len,
7268bccd85fSChristoph Lameter 			unsigned long mode,
7278bccd85fSChristoph Lameter 			unsigned long __user *nmask, unsigned long maxnode,
7288bccd85fSChristoph Lameter 			unsigned flags)
7298bccd85fSChristoph Lameter {
7308bccd85fSChristoph Lameter 	nodemask_t nodes;
7318bccd85fSChristoph Lameter 	int err;
7328bccd85fSChristoph Lameter 
7338bccd85fSChristoph Lameter 	err = get_nodes(&nodes, nmask, maxnode);
7348bccd85fSChristoph Lameter 	if (err)
7358bccd85fSChristoph Lameter 		return err;
7368bccd85fSChristoph Lameter 	return do_mbind(start, len, mode, &nodes, flags);
7378bccd85fSChristoph Lameter }
7388bccd85fSChristoph Lameter 
7398bccd85fSChristoph Lameter /* Set the process memory policy */
7408bccd85fSChristoph Lameter asmlinkage long sys_set_mempolicy(int mode, unsigned long __user *nmask,
7418bccd85fSChristoph Lameter 		unsigned long maxnode)
7428bccd85fSChristoph Lameter {
7438bccd85fSChristoph Lameter 	int err;
7448bccd85fSChristoph Lameter 	nodemask_t nodes;
7458bccd85fSChristoph Lameter 
7468bccd85fSChristoph Lameter 	if (mode < 0 || mode > MPOL_MAX)
7478bccd85fSChristoph Lameter 		return -EINVAL;
7488bccd85fSChristoph Lameter 	err = get_nodes(&nodes, nmask, maxnode);
7498bccd85fSChristoph Lameter 	if (err)
7508bccd85fSChristoph Lameter 		return err;
7518bccd85fSChristoph Lameter 	return do_set_mempolicy(mode, &nodes);
7528bccd85fSChristoph Lameter }
7538bccd85fSChristoph Lameter 
75439743889SChristoph Lameter asmlinkage long sys_migrate_pages(pid_t pid, unsigned long maxnode,
75539743889SChristoph Lameter 		const unsigned long __user *old_nodes,
75639743889SChristoph Lameter 		const unsigned long __user *new_nodes)
75739743889SChristoph Lameter {
75839743889SChristoph Lameter 	struct mm_struct *mm;
75939743889SChristoph Lameter 	struct task_struct *task;
76039743889SChristoph Lameter 	nodemask_t old;
76139743889SChristoph Lameter 	nodemask_t new;
76239743889SChristoph Lameter 	nodemask_t task_nodes;
76339743889SChristoph Lameter 	int err;
76439743889SChristoph Lameter 
76539743889SChristoph Lameter 	err = get_nodes(&old, old_nodes, maxnode);
76639743889SChristoph Lameter 	if (err)
76739743889SChristoph Lameter 		return err;
76839743889SChristoph Lameter 
76939743889SChristoph Lameter 	err = get_nodes(&new, new_nodes, maxnode);
77039743889SChristoph Lameter 	if (err)
77139743889SChristoph Lameter 		return err;
77239743889SChristoph Lameter 
77339743889SChristoph Lameter 	/* Find the mm_struct */
77439743889SChristoph Lameter 	read_lock(&tasklist_lock);
77539743889SChristoph Lameter 	task = pid ? find_task_by_pid(pid) : current;
77639743889SChristoph Lameter 	if (!task) {
77739743889SChristoph Lameter 		read_unlock(&tasklist_lock);
77839743889SChristoph Lameter 		return -ESRCH;
77939743889SChristoph Lameter 	}
78039743889SChristoph Lameter 	mm = get_task_mm(task);
78139743889SChristoph Lameter 	read_unlock(&tasklist_lock);
78239743889SChristoph Lameter 
78339743889SChristoph Lameter 	if (!mm)
78439743889SChristoph Lameter 		return -EINVAL;
78539743889SChristoph Lameter 
78639743889SChristoph Lameter 	/*
78739743889SChristoph Lameter 	 * Check if this process has the right to modify the specified
78839743889SChristoph Lameter 	 * process. The right exists if the process has administrative
78939743889SChristoph Lameter 	 * capabilities, superuser priviledges or the same
79039743889SChristoph Lameter 	 * userid as the target process.
79139743889SChristoph Lameter 	 */
79239743889SChristoph Lameter 	if ((current->euid != task->suid) && (current->euid != task->uid) &&
79339743889SChristoph Lameter 	    (current->uid != task->suid) && (current->uid != task->uid) &&
79439743889SChristoph Lameter 	    !capable(CAP_SYS_ADMIN)) {
79539743889SChristoph Lameter 		err = -EPERM;
79639743889SChristoph Lameter 		goto out;
79739743889SChristoph Lameter 	}
79839743889SChristoph Lameter 
79939743889SChristoph Lameter 	task_nodes = cpuset_mems_allowed(task);
80039743889SChristoph Lameter 	/* Is the user allowed to access the target nodes? */
80139743889SChristoph Lameter 	if (!nodes_subset(new, task_nodes) && !capable(CAP_SYS_ADMIN)) {
80239743889SChristoph Lameter 		err = -EPERM;
80339743889SChristoph Lameter 		goto out;
80439743889SChristoph Lameter 	}
80539743889SChristoph Lameter 
80639743889SChristoph Lameter 	err = do_migrate_pages(mm, &old, &new, MPOL_MF_MOVE);
80739743889SChristoph Lameter out:
80839743889SChristoph Lameter 	mmput(mm);
80939743889SChristoph Lameter 	return err;
81039743889SChristoph Lameter }
81139743889SChristoph Lameter 
81239743889SChristoph Lameter 
8138bccd85fSChristoph Lameter /* Retrieve NUMA policy */
8148bccd85fSChristoph Lameter asmlinkage long sys_get_mempolicy(int __user *policy,
8158bccd85fSChristoph Lameter 				unsigned long __user *nmask,
8168bccd85fSChristoph Lameter 				unsigned long maxnode,
8178bccd85fSChristoph Lameter 				unsigned long addr, unsigned long flags)
8188bccd85fSChristoph Lameter {
8198bccd85fSChristoph Lameter 	int err, pval;
8208bccd85fSChristoph Lameter 	nodemask_t nodes;
8218bccd85fSChristoph Lameter 
8228bccd85fSChristoph Lameter 	if (nmask != NULL && maxnode < MAX_NUMNODES)
8238bccd85fSChristoph Lameter 		return -EINVAL;
8248bccd85fSChristoph Lameter 
8258bccd85fSChristoph Lameter 	err = do_get_mempolicy(&pval, &nodes, addr, flags);
8268bccd85fSChristoph Lameter 
8278bccd85fSChristoph Lameter 	if (err)
8288bccd85fSChristoph Lameter 		return err;
8298bccd85fSChristoph Lameter 
8308bccd85fSChristoph Lameter 	if (policy && put_user(pval, policy))
8318bccd85fSChristoph Lameter 		return -EFAULT;
8328bccd85fSChristoph Lameter 
8338bccd85fSChristoph Lameter 	if (nmask)
8348bccd85fSChristoph Lameter 		err = copy_nodes_to_user(nmask, maxnode, &nodes);
8358bccd85fSChristoph Lameter 
8368bccd85fSChristoph Lameter 	return err;
8378bccd85fSChristoph Lameter }
8388bccd85fSChristoph Lameter 
8391da177e4SLinus Torvalds #ifdef CONFIG_COMPAT
8401da177e4SLinus Torvalds 
8411da177e4SLinus Torvalds asmlinkage long compat_sys_get_mempolicy(int __user *policy,
8421da177e4SLinus Torvalds 				     compat_ulong_t __user *nmask,
8431da177e4SLinus Torvalds 				     compat_ulong_t maxnode,
8441da177e4SLinus Torvalds 				     compat_ulong_t addr, compat_ulong_t flags)
8451da177e4SLinus Torvalds {
8461da177e4SLinus Torvalds 	long err;
8471da177e4SLinus Torvalds 	unsigned long __user *nm = NULL;
8481da177e4SLinus Torvalds 	unsigned long nr_bits, alloc_size;
8491da177e4SLinus Torvalds 	DECLARE_BITMAP(bm, MAX_NUMNODES);
8501da177e4SLinus Torvalds 
8511da177e4SLinus Torvalds 	nr_bits = min_t(unsigned long, maxnode-1, MAX_NUMNODES);
8521da177e4SLinus Torvalds 	alloc_size = ALIGN(nr_bits, BITS_PER_LONG) / 8;
8531da177e4SLinus Torvalds 
8541da177e4SLinus Torvalds 	if (nmask)
8551da177e4SLinus Torvalds 		nm = compat_alloc_user_space(alloc_size);
8561da177e4SLinus Torvalds 
8571da177e4SLinus Torvalds 	err = sys_get_mempolicy(policy, nm, nr_bits+1, addr, flags);
8581da177e4SLinus Torvalds 
8591da177e4SLinus Torvalds 	if (!err && nmask) {
8601da177e4SLinus Torvalds 		err = copy_from_user(bm, nm, alloc_size);
8611da177e4SLinus Torvalds 		/* ensure entire bitmap is zeroed */
8621da177e4SLinus Torvalds 		err |= clear_user(nmask, ALIGN(maxnode-1, 8) / 8);
8631da177e4SLinus Torvalds 		err |= compat_put_bitmap(nmask, bm, nr_bits);
8641da177e4SLinus Torvalds 	}
8651da177e4SLinus Torvalds 
8661da177e4SLinus Torvalds 	return err;
8671da177e4SLinus Torvalds }
8681da177e4SLinus Torvalds 
8691da177e4SLinus Torvalds asmlinkage long compat_sys_set_mempolicy(int mode, compat_ulong_t __user *nmask,
8701da177e4SLinus Torvalds 				     compat_ulong_t maxnode)
8711da177e4SLinus Torvalds {
8721da177e4SLinus Torvalds 	long err = 0;
8731da177e4SLinus Torvalds 	unsigned long __user *nm = NULL;
8741da177e4SLinus Torvalds 	unsigned long nr_bits, alloc_size;
8751da177e4SLinus Torvalds 	DECLARE_BITMAP(bm, MAX_NUMNODES);
8761da177e4SLinus Torvalds 
8771da177e4SLinus Torvalds 	nr_bits = min_t(unsigned long, maxnode-1, MAX_NUMNODES);
8781da177e4SLinus Torvalds 	alloc_size = ALIGN(nr_bits, BITS_PER_LONG) / 8;
8791da177e4SLinus Torvalds 
8801da177e4SLinus Torvalds 	if (nmask) {
8811da177e4SLinus Torvalds 		err = compat_get_bitmap(bm, nmask, nr_bits);
8821da177e4SLinus Torvalds 		nm = compat_alloc_user_space(alloc_size);
8831da177e4SLinus Torvalds 		err |= copy_to_user(nm, bm, alloc_size);
8841da177e4SLinus Torvalds 	}
8851da177e4SLinus Torvalds 
8861da177e4SLinus Torvalds 	if (err)
8871da177e4SLinus Torvalds 		return -EFAULT;
8881da177e4SLinus Torvalds 
8891da177e4SLinus Torvalds 	return sys_set_mempolicy(mode, nm, nr_bits+1);
8901da177e4SLinus Torvalds }
8911da177e4SLinus Torvalds 
8921da177e4SLinus Torvalds asmlinkage long compat_sys_mbind(compat_ulong_t start, compat_ulong_t len,
8931da177e4SLinus Torvalds 			     compat_ulong_t mode, compat_ulong_t __user *nmask,
8941da177e4SLinus Torvalds 			     compat_ulong_t maxnode, compat_ulong_t flags)
8951da177e4SLinus Torvalds {
8961da177e4SLinus Torvalds 	long err = 0;
8971da177e4SLinus Torvalds 	unsigned long __user *nm = NULL;
8981da177e4SLinus Torvalds 	unsigned long nr_bits, alloc_size;
899dfcd3c0dSAndi Kleen 	nodemask_t bm;
9001da177e4SLinus Torvalds 
9011da177e4SLinus Torvalds 	nr_bits = min_t(unsigned long, maxnode-1, MAX_NUMNODES);
9021da177e4SLinus Torvalds 	alloc_size = ALIGN(nr_bits, BITS_PER_LONG) / 8;
9031da177e4SLinus Torvalds 
9041da177e4SLinus Torvalds 	if (nmask) {
905dfcd3c0dSAndi Kleen 		err = compat_get_bitmap(nodes_addr(bm), nmask, nr_bits);
9061da177e4SLinus Torvalds 		nm = compat_alloc_user_space(alloc_size);
907dfcd3c0dSAndi Kleen 		err |= copy_to_user(nm, nodes_addr(bm), alloc_size);
9081da177e4SLinus Torvalds 	}
9091da177e4SLinus Torvalds 
9101da177e4SLinus Torvalds 	if (err)
9111da177e4SLinus Torvalds 		return -EFAULT;
9121da177e4SLinus Torvalds 
9131da177e4SLinus Torvalds 	return sys_mbind(start, len, mode, nm, nr_bits+1, flags);
9141da177e4SLinus Torvalds }
9151da177e4SLinus Torvalds 
9161da177e4SLinus Torvalds #endif
9171da177e4SLinus Torvalds 
9181da177e4SLinus Torvalds /* Return effective policy for a VMA */
91948fce342SChristoph Lameter static struct mempolicy * get_vma_policy(struct task_struct *task,
92048fce342SChristoph Lameter 		struct vm_area_struct *vma, unsigned long addr)
9211da177e4SLinus Torvalds {
9226e21c8f1SChristoph Lameter 	struct mempolicy *pol = task->mempolicy;
9231da177e4SLinus Torvalds 
9241da177e4SLinus Torvalds 	if (vma) {
9251da177e4SLinus Torvalds 		if (vma->vm_ops && vma->vm_ops->get_policy)
9261da177e4SLinus Torvalds 			pol = vma->vm_ops->get_policy(vma, addr);
9271da177e4SLinus Torvalds 		else if (vma->vm_policy &&
9281da177e4SLinus Torvalds 				vma->vm_policy->policy != MPOL_DEFAULT)
9291da177e4SLinus Torvalds 			pol = vma->vm_policy;
9301da177e4SLinus Torvalds 	}
9311da177e4SLinus Torvalds 	if (!pol)
9321da177e4SLinus Torvalds 		pol = &default_policy;
9331da177e4SLinus Torvalds 	return pol;
9341da177e4SLinus Torvalds }
9351da177e4SLinus Torvalds 
9361da177e4SLinus Torvalds /* Return a zonelist representing a mempolicy */
937dd0fc66fSAl Viro static struct zonelist *zonelist_policy(gfp_t gfp, struct mempolicy *policy)
9381da177e4SLinus Torvalds {
9391da177e4SLinus Torvalds 	int nd;
9401da177e4SLinus Torvalds 
9411da177e4SLinus Torvalds 	switch (policy->policy) {
9421da177e4SLinus Torvalds 	case MPOL_PREFERRED:
9431da177e4SLinus Torvalds 		nd = policy->v.preferred_node;
9441da177e4SLinus Torvalds 		if (nd < 0)
9451da177e4SLinus Torvalds 			nd = numa_node_id();
9461da177e4SLinus Torvalds 		break;
9471da177e4SLinus Torvalds 	case MPOL_BIND:
9481da177e4SLinus Torvalds 		/* Lower zones don't get a policy applied */
9491da177e4SLinus Torvalds 		/* Careful: current->mems_allowed might have moved */
950af4ca457SAl Viro 		if (gfp_zone(gfp) >= policy_zone)
9511da177e4SLinus Torvalds 			if (cpuset_zonelist_valid_mems_allowed(policy->v.zonelist))
9521da177e4SLinus Torvalds 				return policy->v.zonelist;
9531da177e4SLinus Torvalds 		/*FALL THROUGH*/
9541da177e4SLinus Torvalds 	case MPOL_INTERLEAVE: /* should not happen */
9551da177e4SLinus Torvalds 	case MPOL_DEFAULT:
9561da177e4SLinus Torvalds 		nd = numa_node_id();
9571da177e4SLinus Torvalds 		break;
9581da177e4SLinus Torvalds 	default:
9591da177e4SLinus Torvalds 		nd = 0;
9601da177e4SLinus Torvalds 		BUG();
9611da177e4SLinus Torvalds 	}
962af4ca457SAl Viro 	return NODE_DATA(nd)->node_zonelists + gfp_zone(gfp);
9631da177e4SLinus Torvalds }
9641da177e4SLinus Torvalds 
9651da177e4SLinus Torvalds /* Do dynamic interleaving for a process */
9661da177e4SLinus Torvalds static unsigned interleave_nodes(struct mempolicy *policy)
9671da177e4SLinus Torvalds {
9681da177e4SLinus Torvalds 	unsigned nid, next;
9691da177e4SLinus Torvalds 	struct task_struct *me = current;
9701da177e4SLinus Torvalds 
9711da177e4SLinus Torvalds 	nid = me->il_next;
972dfcd3c0dSAndi Kleen 	next = next_node(nid, policy->v.nodes);
9731da177e4SLinus Torvalds 	if (next >= MAX_NUMNODES)
974dfcd3c0dSAndi Kleen 		next = first_node(policy->v.nodes);
9751da177e4SLinus Torvalds 	me->il_next = next;
9761da177e4SLinus Torvalds 	return nid;
9771da177e4SLinus Torvalds }
9781da177e4SLinus Torvalds 
9791da177e4SLinus Torvalds /* Do static interleaving for a VMA with known offset. */
9801da177e4SLinus Torvalds static unsigned offset_il_node(struct mempolicy *pol,
9811da177e4SLinus Torvalds 		struct vm_area_struct *vma, unsigned long off)
9821da177e4SLinus Torvalds {
983dfcd3c0dSAndi Kleen 	unsigned nnodes = nodes_weight(pol->v.nodes);
9841da177e4SLinus Torvalds 	unsigned target = (unsigned)off % nnodes;
9851da177e4SLinus Torvalds 	int c;
9861da177e4SLinus Torvalds 	int nid = -1;
9871da177e4SLinus Torvalds 
9881da177e4SLinus Torvalds 	c = 0;
9891da177e4SLinus Torvalds 	do {
990dfcd3c0dSAndi Kleen 		nid = next_node(nid, pol->v.nodes);
9911da177e4SLinus Torvalds 		c++;
9921da177e4SLinus Torvalds 	} while (c <= target);
9931da177e4SLinus Torvalds 	return nid;
9941da177e4SLinus Torvalds }
9951da177e4SLinus Torvalds 
9965da7ca86SChristoph Lameter /* Determine a node number for interleave */
9975da7ca86SChristoph Lameter static inline unsigned interleave_nid(struct mempolicy *pol,
9985da7ca86SChristoph Lameter 		 struct vm_area_struct *vma, unsigned long addr, int shift)
9995da7ca86SChristoph Lameter {
10005da7ca86SChristoph Lameter 	if (vma) {
10015da7ca86SChristoph Lameter 		unsigned long off;
10025da7ca86SChristoph Lameter 
10035da7ca86SChristoph Lameter 		off = vma->vm_pgoff;
10045da7ca86SChristoph Lameter 		off += (addr - vma->vm_start) >> shift;
10055da7ca86SChristoph Lameter 		return offset_il_node(pol, vma, off);
10065da7ca86SChristoph Lameter 	} else
10075da7ca86SChristoph Lameter 		return interleave_nodes(pol);
10085da7ca86SChristoph Lameter }
10095da7ca86SChristoph Lameter 
10105da7ca86SChristoph Lameter /* Return a zonelist suitable for a huge page allocation. */
10115da7ca86SChristoph Lameter struct zonelist *huge_zonelist(struct vm_area_struct *vma, unsigned long addr)
10125da7ca86SChristoph Lameter {
10135da7ca86SChristoph Lameter 	struct mempolicy *pol = get_vma_policy(current, vma, addr);
10145da7ca86SChristoph Lameter 
10155da7ca86SChristoph Lameter 	if (pol->policy == MPOL_INTERLEAVE) {
10165da7ca86SChristoph Lameter 		unsigned nid;
10175da7ca86SChristoph Lameter 
10185da7ca86SChristoph Lameter 		nid = interleave_nid(pol, vma, addr, HPAGE_SHIFT);
10195da7ca86SChristoph Lameter 		return NODE_DATA(nid)->node_zonelists + gfp_zone(GFP_HIGHUSER);
10205da7ca86SChristoph Lameter 	}
10215da7ca86SChristoph Lameter 	return zonelist_policy(GFP_HIGHUSER, pol);
10225da7ca86SChristoph Lameter }
10235da7ca86SChristoph Lameter 
10241da177e4SLinus Torvalds /* Allocate a page in interleaved policy.
10251da177e4SLinus Torvalds    Own path because it needs to do special accounting. */
1026662f3a0bSAndi Kleen static struct page *alloc_page_interleave(gfp_t gfp, unsigned order,
1027662f3a0bSAndi Kleen 					unsigned nid)
10281da177e4SLinus Torvalds {
10291da177e4SLinus Torvalds 	struct zonelist *zl;
10301da177e4SLinus Torvalds 	struct page *page;
10311da177e4SLinus Torvalds 
1032af4ca457SAl Viro 	zl = NODE_DATA(nid)->node_zonelists + gfp_zone(gfp);
10331da177e4SLinus Torvalds 	page = __alloc_pages(gfp, order, zl);
10341da177e4SLinus Torvalds 	if (page && page_zone(page) == zl->zones[0]) {
1035e7c8d5c9SChristoph Lameter 		zone_pcp(zl->zones[0],get_cpu())->interleave_hit++;
10361da177e4SLinus Torvalds 		put_cpu();
10371da177e4SLinus Torvalds 	}
10381da177e4SLinus Torvalds 	return page;
10391da177e4SLinus Torvalds }
10401da177e4SLinus Torvalds 
10411da177e4SLinus Torvalds /**
10421da177e4SLinus Torvalds  * 	alloc_page_vma	- Allocate a page for a VMA.
10431da177e4SLinus Torvalds  *
10441da177e4SLinus Torvalds  * 	@gfp:
10451da177e4SLinus Torvalds  *      %GFP_USER    user allocation.
10461da177e4SLinus Torvalds  *      %GFP_KERNEL  kernel allocations,
10471da177e4SLinus Torvalds  *      %GFP_HIGHMEM highmem/user allocations,
10481da177e4SLinus Torvalds  *      %GFP_FS      allocation should not call back into a file system.
10491da177e4SLinus Torvalds  *      %GFP_ATOMIC  don't sleep.
10501da177e4SLinus Torvalds  *
10511da177e4SLinus Torvalds  * 	@vma:  Pointer to VMA or NULL if not available.
10521da177e4SLinus Torvalds  *	@addr: Virtual Address of the allocation. Must be inside the VMA.
10531da177e4SLinus Torvalds  *
10541da177e4SLinus Torvalds  * 	This function allocates a page from the kernel page pool and applies
10551da177e4SLinus Torvalds  *	a NUMA policy associated with the VMA or the current process.
10561da177e4SLinus Torvalds  *	When VMA is not NULL caller must hold down_read on the mmap_sem of the
10571da177e4SLinus Torvalds  *	mm_struct of the VMA to prevent it from going away. Should be used for
10581da177e4SLinus Torvalds  *	all allocations for pages that will be mapped into
10591da177e4SLinus Torvalds  * 	user space. Returns NULL when no page can be allocated.
10601da177e4SLinus Torvalds  *
10611da177e4SLinus Torvalds  *	Should be called with the mm_sem of the vma hold.
10621da177e4SLinus Torvalds  */
10631da177e4SLinus Torvalds struct page *
1064dd0fc66fSAl Viro alloc_page_vma(gfp_t gfp, struct vm_area_struct *vma, unsigned long addr)
10651da177e4SLinus Torvalds {
10666e21c8f1SChristoph Lameter 	struct mempolicy *pol = get_vma_policy(current, vma, addr);
10671da177e4SLinus Torvalds 
1068cf2a473cSPaul Jackson 	cpuset_update_task_memory_state();
10691da177e4SLinus Torvalds 
10701da177e4SLinus Torvalds 	if (unlikely(pol->policy == MPOL_INTERLEAVE)) {
10711da177e4SLinus Torvalds 		unsigned nid;
10725da7ca86SChristoph Lameter 
10735da7ca86SChristoph Lameter 		nid = interleave_nid(pol, vma, addr, PAGE_SHIFT);
10741da177e4SLinus Torvalds 		return alloc_page_interleave(gfp, 0, nid);
10751da177e4SLinus Torvalds 	}
10761da177e4SLinus Torvalds 	return __alloc_pages(gfp, 0, zonelist_policy(gfp, pol));
10771da177e4SLinus Torvalds }
10781da177e4SLinus Torvalds 
10791da177e4SLinus Torvalds /**
10801da177e4SLinus Torvalds  * 	alloc_pages_current - Allocate pages.
10811da177e4SLinus Torvalds  *
10821da177e4SLinus Torvalds  *	@gfp:
10831da177e4SLinus Torvalds  *		%GFP_USER   user allocation,
10841da177e4SLinus Torvalds  *      	%GFP_KERNEL kernel allocation,
10851da177e4SLinus Torvalds  *      	%GFP_HIGHMEM highmem allocation,
10861da177e4SLinus Torvalds  *      	%GFP_FS     don't call back into a file system.
10871da177e4SLinus Torvalds  *      	%GFP_ATOMIC don't sleep.
10881da177e4SLinus Torvalds  *	@order: Power of two of allocation size in pages. 0 is a single page.
10891da177e4SLinus Torvalds  *
10901da177e4SLinus Torvalds  *	Allocate a page from the kernel page pool.  When not in
10911da177e4SLinus Torvalds  *	interrupt context and apply the current process NUMA policy.
10921da177e4SLinus Torvalds  *	Returns NULL when no page can be allocated.
10931da177e4SLinus Torvalds  *
1094cf2a473cSPaul Jackson  *	Don't call cpuset_update_task_memory_state() unless
10951da177e4SLinus Torvalds  *	1) it's ok to take cpuset_sem (can WAIT), and
10961da177e4SLinus Torvalds  *	2) allocating for current task (not interrupt).
10971da177e4SLinus Torvalds  */
1098dd0fc66fSAl Viro struct page *alloc_pages_current(gfp_t gfp, unsigned order)
10991da177e4SLinus Torvalds {
11001da177e4SLinus Torvalds 	struct mempolicy *pol = current->mempolicy;
11011da177e4SLinus Torvalds 
11021da177e4SLinus Torvalds 	if ((gfp & __GFP_WAIT) && !in_interrupt())
1103cf2a473cSPaul Jackson 		cpuset_update_task_memory_state();
11041da177e4SLinus Torvalds 	if (!pol || in_interrupt())
11051da177e4SLinus Torvalds 		pol = &default_policy;
11061da177e4SLinus Torvalds 	if (pol->policy == MPOL_INTERLEAVE)
11071da177e4SLinus Torvalds 		return alloc_page_interleave(gfp, order, interleave_nodes(pol));
11081da177e4SLinus Torvalds 	return __alloc_pages(gfp, order, zonelist_policy(gfp, pol));
11091da177e4SLinus Torvalds }
11101da177e4SLinus Torvalds EXPORT_SYMBOL(alloc_pages_current);
11111da177e4SLinus Torvalds 
11124225399aSPaul Jackson /*
11134225399aSPaul Jackson  * If mpol_copy() sees current->cpuset == cpuset_being_rebound, then it
11144225399aSPaul Jackson  * rebinds the mempolicy its copying by calling mpol_rebind_policy()
11154225399aSPaul Jackson  * with the mems_allowed returned by cpuset_mems_allowed().  This
11164225399aSPaul Jackson  * keeps mempolicies cpuset relative after its cpuset moves.  See
11174225399aSPaul Jackson  * further kernel/cpuset.c update_nodemask().
11184225399aSPaul Jackson  */
11194225399aSPaul Jackson void *cpuset_being_rebound;
11204225399aSPaul Jackson 
11211da177e4SLinus Torvalds /* Slow path of a mempolicy copy */
11221da177e4SLinus Torvalds struct mempolicy *__mpol_copy(struct mempolicy *old)
11231da177e4SLinus Torvalds {
11241da177e4SLinus Torvalds 	struct mempolicy *new = kmem_cache_alloc(policy_cache, GFP_KERNEL);
11251da177e4SLinus Torvalds 
11261da177e4SLinus Torvalds 	if (!new)
11271da177e4SLinus Torvalds 		return ERR_PTR(-ENOMEM);
11284225399aSPaul Jackson 	if (current_cpuset_is_being_rebound()) {
11294225399aSPaul Jackson 		nodemask_t mems = cpuset_mems_allowed(current);
11304225399aSPaul Jackson 		mpol_rebind_policy(old, &mems);
11314225399aSPaul Jackson 	}
11321da177e4SLinus Torvalds 	*new = *old;
11331da177e4SLinus Torvalds 	atomic_set(&new->refcnt, 1);
11341da177e4SLinus Torvalds 	if (new->policy == MPOL_BIND) {
11351da177e4SLinus Torvalds 		int sz = ksize(old->v.zonelist);
11361da177e4SLinus Torvalds 		new->v.zonelist = kmalloc(sz, SLAB_KERNEL);
11371da177e4SLinus Torvalds 		if (!new->v.zonelist) {
11381da177e4SLinus Torvalds 			kmem_cache_free(policy_cache, new);
11391da177e4SLinus Torvalds 			return ERR_PTR(-ENOMEM);
11401da177e4SLinus Torvalds 		}
11411da177e4SLinus Torvalds 		memcpy(new->v.zonelist, old->v.zonelist, sz);
11421da177e4SLinus Torvalds 	}
11431da177e4SLinus Torvalds 	return new;
11441da177e4SLinus Torvalds }
11451da177e4SLinus Torvalds 
11461da177e4SLinus Torvalds /* Slow path of a mempolicy comparison */
11471da177e4SLinus Torvalds int __mpol_equal(struct mempolicy *a, struct mempolicy *b)
11481da177e4SLinus Torvalds {
11491da177e4SLinus Torvalds 	if (!a || !b)
11501da177e4SLinus Torvalds 		return 0;
11511da177e4SLinus Torvalds 	if (a->policy != b->policy)
11521da177e4SLinus Torvalds 		return 0;
11531da177e4SLinus Torvalds 	switch (a->policy) {
11541da177e4SLinus Torvalds 	case MPOL_DEFAULT:
11551da177e4SLinus Torvalds 		return 1;
11561da177e4SLinus Torvalds 	case MPOL_INTERLEAVE:
1157dfcd3c0dSAndi Kleen 		return nodes_equal(a->v.nodes, b->v.nodes);
11581da177e4SLinus Torvalds 	case MPOL_PREFERRED:
11591da177e4SLinus Torvalds 		return a->v.preferred_node == b->v.preferred_node;
11601da177e4SLinus Torvalds 	case MPOL_BIND: {
11611da177e4SLinus Torvalds 		int i;
11621da177e4SLinus Torvalds 		for (i = 0; a->v.zonelist->zones[i]; i++)
11631da177e4SLinus Torvalds 			if (a->v.zonelist->zones[i] != b->v.zonelist->zones[i])
11641da177e4SLinus Torvalds 				return 0;
11651da177e4SLinus Torvalds 		return b->v.zonelist->zones[i] == NULL;
11661da177e4SLinus Torvalds 	}
11671da177e4SLinus Torvalds 	default:
11681da177e4SLinus Torvalds 		BUG();
11691da177e4SLinus Torvalds 		return 0;
11701da177e4SLinus Torvalds 	}
11711da177e4SLinus Torvalds }
11721da177e4SLinus Torvalds 
11731da177e4SLinus Torvalds /* Slow path of a mpol destructor. */
11741da177e4SLinus Torvalds void __mpol_free(struct mempolicy *p)
11751da177e4SLinus Torvalds {
11761da177e4SLinus Torvalds 	if (!atomic_dec_and_test(&p->refcnt))
11771da177e4SLinus Torvalds 		return;
11781da177e4SLinus Torvalds 	if (p->policy == MPOL_BIND)
11791da177e4SLinus Torvalds 		kfree(p->v.zonelist);
11801da177e4SLinus Torvalds 	p->policy = MPOL_DEFAULT;
11811da177e4SLinus Torvalds 	kmem_cache_free(policy_cache, p);
11821da177e4SLinus Torvalds }
11831da177e4SLinus Torvalds 
11841da177e4SLinus Torvalds /*
11851da177e4SLinus Torvalds  * Shared memory backing store policy support.
11861da177e4SLinus Torvalds  *
11871da177e4SLinus Torvalds  * Remember policies even when nobody has shared memory mapped.
11881da177e4SLinus Torvalds  * The policies are kept in Red-Black tree linked from the inode.
11891da177e4SLinus Torvalds  * They are protected by the sp->lock spinlock, which should be held
11901da177e4SLinus Torvalds  * for any accesses to the tree.
11911da177e4SLinus Torvalds  */
11921da177e4SLinus Torvalds 
11931da177e4SLinus Torvalds /* lookup first element intersecting start-end */
11941da177e4SLinus Torvalds /* Caller holds sp->lock */
11951da177e4SLinus Torvalds static struct sp_node *
11961da177e4SLinus Torvalds sp_lookup(struct shared_policy *sp, unsigned long start, unsigned long end)
11971da177e4SLinus Torvalds {
11981da177e4SLinus Torvalds 	struct rb_node *n = sp->root.rb_node;
11991da177e4SLinus Torvalds 
12001da177e4SLinus Torvalds 	while (n) {
12011da177e4SLinus Torvalds 		struct sp_node *p = rb_entry(n, struct sp_node, nd);
12021da177e4SLinus Torvalds 
12031da177e4SLinus Torvalds 		if (start >= p->end)
12041da177e4SLinus Torvalds 			n = n->rb_right;
12051da177e4SLinus Torvalds 		else if (end <= p->start)
12061da177e4SLinus Torvalds 			n = n->rb_left;
12071da177e4SLinus Torvalds 		else
12081da177e4SLinus Torvalds 			break;
12091da177e4SLinus Torvalds 	}
12101da177e4SLinus Torvalds 	if (!n)
12111da177e4SLinus Torvalds 		return NULL;
12121da177e4SLinus Torvalds 	for (;;) {
12131da177e4SLinus Torvalds 		struct sp_node *w = NULL;
12141da177e4SLinus Torvalds 		struct rb_node *prev = rb_prev(n);
12151da177e4SLinus Torvalds 		if (!prev)
12161da177e4SLinus Torvalds 			break;
12171da177e4SLinus Torvalds 		w = rb_entry(prev, struct sp_node, nd);
12181da177e4SLinus Torvalds 		if (w->end <= start)
12191da177e4SLinus Torvalds 			break;
12201da177e4SLinus Torvalds 		n = prev;
12211da177e4SLinus Torvalds 	}
12221da177e4SLinus Torvalds 	return rb_entry(n, struct sp_node, nd);
12231da177e4SLinus Torvalds }
12241da177e4SLinus Torvalds 
12251da177e4SLinus Torvalds /* Insert a new shared policy into the list. */
12261da177e4SLinus Torvalds /* Caller holds sp->lock */
12271da177e4SLinus Torvalds static void sp_insert(struct shared_policy *sp, struct sp_node *new)
12281da177e4SLinus Torvalds {
12291da177e4SLinus Torvalds 	struct rb_node **p = &sp->root.rb_node;
12301da177e4SLinus Torvalds 	struct rb_node *parent = NULL;
12311da177e4SLinus Torvalds 	struct sp_node *nd;
12321da177e4SLinus Torvalds 
12331da177e4SLinus Torvalds 	while (*p) {
12341da177e4SLinus Torvalds 		parent = *p;
12351da177e4SLinus Torvalds 		nd = rb_entry(parent, struct sp_node, nd);
12361da177e4SLinus Torvalds 		if (new->start < nd->start)
12371da177e4SLinus Torvalds 			p = &(*p)->rb_left;
12381da177e4SLinus Torvalds 		else if (new->end > nd->end)
12391da177e4SLinus Torvalds 			p = &(*p)->rb_right;
12401da177e4SLinus Torvalds 		else
12411da177e4SLinus Torvalds 			BUG();
12421da177e4SLinus Torvalds 	}
12431da177e4SLinus Torvalds 	rb_link_node(&new->nd, parent, p);
12441da177e4SLinus Torvalds 	rb_insert_color(&new->nd, &sp->root);
12451da177e4SLinus Torvalds 	PDprintk("inserting %lx-%lx: %d\n", new->start, new->end,
12461da177e4SLinus Torvalds 		 new->policy ? new->policy->policy : 0);
12471da177e4SLinus Torvalds }
12481da177e4SLinus Torvalds 
12491da177e4SLinus Torvalds /* Find shared policy intersecting idx */
12501da177e4SLinus Torvalds struct mempolicy *
12511da177e4SLinus Torvalds mpol_shared_policy_lookup(struct shared_policy *sp, unsigned long idx)
12521da177e4SLinus Torvalds {
12531da177e4SLinus Torvalds 	struct mempolicy *pol = NULL;
12541da177e4SLinus Torvalds 	struct sp_node *sn;
12551da177e4SLinus Torvalds 
12561da177e4SLinus Torvalds 	if (!sp->root.rb_node)
12571da177e4SLinus Torvalds 		return NULL;
12581da177e4SLinus Torvalds 	spin_lock(&sp->lock);
12591da177e4SLinus Torvalds 	sn = sp_lookup(sp, idx, idx+1);
12601da177e4SLinus Torvalds 	if (sn) {
12611da177e4SLinus Torvalds 		mpol_get(sn->policy);
12621da177e4SLinus Torvalds 		pol = sn->policy;
12631da177e4SLinus Torvalds 	}
12641da177e4SLinus Torvalds 	spin_unlock(&sp->lock);
12651da177e4SLinus Torvalds 	return pol;
12661da177e4SLinus Torvalds }
12671da177e4SLinus Torvalds 
12681da177e4SLinus Torvalds static void sp_delete(struct shared_policy *sp, struct sp_node *n)
12691da177e4SLinus Torvalds {
12701da177e4SLinus Torvalds 	PDprintk("deleting %lx-l%x\n", n->start, n->end);
12711da177e4SLinus Torvalds 	rb_erase(&n->nd, &sp->root);
12721da177e4SLinus Torvalds 	mpol_free(n->policy);
12731da177e4SLinus Torvalds 	kmem_cache_free(sn_cache, n);
12741da177e4SLinus Torvalds }
12751da177e4SLinus Torvalds 
12761da177e4SLinus Torvalds struct sp_node *
12771da177e4SLinus Torvalds sp_alloc(unsigned long start, unsigned long end, struct mempolicy *pol)
12781da177e4SLinus Torvalds {
12791da177e4SLinus Torvalds 	struct sp_node *n = kmem_cache_alloc(sn_cache, GFP_KERNEL);
12801da177e4SLinus Torvalds 
12811da177e4SLinus Torvalds 	if (!n)
12821da177e4SLinus Torvalds 		return NULL;
12831da177e4SLinus Torvalds 	n->start = start;
12841da177e4SLinus Torvalds 	n->end = end;
12851da177e4SLinus Torvalds 	mpol_get(pol);
12861da177e4SLinus Torvalds 	n->policy = pol;
12871da177e4SLinus Torvalds 	return n;
12881da177e4SLinus Torvalds }
12891da177e4SLinus Torvalds 
12901da177e4SLinus Torvalds /* Replace a policy range. */
12911da177e4SLinus Torvalds static int shared_policy_replace(struct shared_policy *sp, unsigned long start,
12921da177e4SLinus Torvalds 				 unsigned long end, struct sp_node *new)
12931da177e4SLinus Torvalds {
12941da177e4SLinus Torvalds 	struct sp_node *n, *new2 = NULL;
12951da177e4SLinus Torvalds 
12961da177e4SLinus Torvalds restart:
12971da177e4SLinus Torvalds 	spin_lock(&sp->lock);
12981da177e4SLinus Torvalds 	n = sp_lookup(sp, start, end);
12991da177e4SLinus Torvalds 	/* Take care of old policies in the same range. */
13001da177e4SLinus Torvalds 	while (n && n->start < end) {
13011da177e4SLinus Torvalds 		struct rb_node *next = rb_next(&n->nd);
13021da177e4SLinus Torvalds 		if (n->start >= start) {
13031da177e4SLinus Torvalds 			if (n->end <= end)
13041da177e4SLinus Torvalds 				sp_delete(sp, n);
13051da177e4SLinus Torvalds 			else
13061da177e4SLinus Torvalds 				n->start = end;
13071da177e4SLinus Torvalds 		} else {
13081da177e4SLinus Torvalds 			/* Old policy spanning whole new range. */
13091da177e4SLinus Torvalds 			if (n->end > end) {
13101da177e4SLinus Torvalds 				if (!new2) {
13111da177e4SLinus Torvalds 					spin_unlock(&sp->lock);
13121da177e4SLinus Torvalds 					new2 = sp_alloc(end, n->end, n->policy);
13131da177e4SLinus Torvalds 					if (!new2)
13141da177e4SLinus Torvalds 						return -ENOMEM;
13151da177e4SLinus Torvalds 					goto restart;
13161da177e4SLinus Torvalds 				}
13171da177e4SLinus Torvalds 				n->end = start;
13181da177e4SLinus Torvalds 				sp_insert(sp, new2);
13191da177e4SLinus Torvalds 				new2 = NULL;
13201da177e4SLinus Torvalds 				break;
13211da177e4SLinus Torvalds 			} else
13221da177e4SLinus Torvalds 				n->end = start;
13231da177e4SLinus Torvalds 		}
13241da177e4SLinus Torvalds 		if (!next)
13251da177e4SLinus Torvalds 			break;
13261da177e4SLinus Torvalds 		n = rb_entry(next, struct sp_node, nd);
13271da177e4SLinus Torvalds 	}
13281da177e4SLinus Torvalds 	if (new)
13291da177e4SLinus Torvalds 		sp_insert(sp, new);
13301da177e4SLinus Torvalds 	spin_unlock(&sp->lock);
13311da177e4SLinus Torvalds 	if (new2) {
13321da177e4SLinus Torvalds 		mpol_free(new2->policy);
13331da177e4SLinus Torvalds 		kmem_cache_free(sn_cache, new2);
13341da177e4SLinus Torvalds 	}
13351da177e4SLinus Torvalds 	return 0;
13361da177e4SLinus Torvalds }
13371da177e4SLinus Torvalds 
13387339ff83SRobin Holt void mpol_shared_policy_init(struct shared_policy *info, int policy,
13397339ff83SRobin Holt 				nodemask_t *policy_nodes)
13407339ff83SRobin Holt {
13417339ff83SRobin Holt 	info->root = RB_ROOT;
13427339ff83SRobin Holt 	spin_lock_init(&info->lock);
13437339ff83SRobin Holt 
13447339ff83SRobin Holt 	if (policy != MPOL_DEFAULT) {
13457339ff83SRobin Holt 		struct mempolicy *newpol;
13467339ff83SRobin Holt 
13477339ff83SRobin Holt 		/* Falls back to MPOL_DEFAULT on any error */
13487339ff83SRobin Holt 		newpol = mpol_new(policy, policy_nodes);
13497339ff83SRobin Holt 		if (!IS_ERR(newpol)) {
13507339ff83SRobin Holt 			/* Create pseudo-vma that contains just the policy */
13517339ff83SRobin Holt 			struct vm_area_struct pvma;
13527339ff83SRobin Holt 
13537339ff83SRobin Holt 			memset(&pvma, 0, sizeof(struct vm_area_struct));
13547339ff83SRobin Holt 			/* Policy covers entire file */
13557339ff83SRobin Holt 			pvma.vm_end = TASK_SIZE;
13567339ff83SRobin Holt 			mpol_set_shared_policy(info, &pvma, newpol);
13577339ff83SRobin Holt 			mpol_free(newpol);
13587339ff83SRobin Holt 		}
13597339ff83SRobin Holt 	}
13607339ff83SRobin Holt }
13617339ff83SRobin Holt 
13621da177e4SLinus Torvalds int mpol_set_shared_policy(struct shared_policy *info,
13631da177e4SLinus Torvalds 			struct vm_area_struct *vma, struct mempolicy *npol)
13641da177e4SLinus Torvalds {
13651da177e4SLinus Torvalds 	int err;
13661da177e4SLinus Torvalds 	struct sp_node *new = NULL;
13671da177e4SLinus Torvalds 	unsigned long sz = vma_pages(vma);
13681da177e4SLinus Torvalds 
13691da177e4SLinus Torvalds 	PDprintk("set_shared_policy %lx sz %lu %d %lx\n",
13701da177e4SLinus Torvalds 		 vma->vm_pgoff,
13711da177e4SLinus Torvalds 		 sz, npol? npol->policy : -1,
1372dfcd3c0dSAndi Kleen 		npol ? nodes_addr(npol->v.nodes)[0] : -1);
13731da177e4SLinus Torvalds 
13741da177e4SLinus Torvalds 	if (npol) {
13751da177e4SLinus Torvalds 		new = sp_alloc(vma->vm_pgoff, vma->vm_pgoff + sz, npol);
13761da177e4SLinus Torvalds 		if (!new)
13771da177e4SLinus Torvalds 			return -ENOMEM;
13781da177e4SLinus Torvalds 	}
13791da177e4SLinus Torvalds 	err = shared_policy_replace(info, vma->vm_pgoff, vma->vm_pgoff+sz, new);
13801da177e4SLinus Torvalds 	if (err && new)
13811da177e4SLinus Torvalds 		kmem_cache_free(sn_cache, new);
13821da177e4SLinus Torvalds 	return err;
13831da177e4SLinus Torvalds }
13841da177e4SLinus Torvalds 
13851da177e4SLinus Torvalds /* Free a backing policy store on inode delete. */
13861da177e4SLinus Torvalds void mpol_free_shared_policy(struct shared_policy *p)
13871da177e4SLinus Torvalds {
13881da177e4SLinus Torvalds 	struct sp_node *n;
13891da177e4SLinus Torvalds 	struct rb_node *next;
13901da177e4SLinus Torvalds 
13911da177e4SLinus Torvalds 	if (!p->root.rb_node)
13921da177e4SLinus Torvalds 		return;
13931da177e4SLinus Torvalds 	spin_lock(&p->lock);
13941da177e4SLinus Torvalds 	next = rb_first(&p->root);
13951da177e4SLinus Torvalds 	while (next) {
13961da177e4SLinus Torvalds 		n = rb_entry(next, struct sp_node, nd);
13971da177e4SLinus Torvalds 		next = rb_next(&n->nd);
139890c5029eSAndi Kleen 		rb_erase(&n->nd, &p->root);
13991da177e4SLinus Torvalds 		mpol_free(n->policy);
14001da177e4SLinus Torvalds 		kmem_cache_free(sn_cache, n);
14011da177e4SLinus Torvalds 	}
14021da177e4SLinus Torvalds 	spin_unlock(&p->lock);
14031da177e4SLinus Torvalds }
14041da177e4SLinus Torvalds 
14051da177e4SLinus Torvalds /* assumes fs == KERNEL_DS */
14061da177e4SLinus Torvalds void __init numa_policy_init(void)
14071da177e4SLinus Torvalds {
14081da177e4SLinus Torvalds 	policy_cache = kmem_cache_create("numa_policy",
14091da177e4SLinus Torvalds 					 sizeof(struct mempolicy),
14101da177e4SLinus Torvalds 					 0, SLAB_PANIC, NULL, NULL);
14111da177e4SLinus Torvalds 
14121da177e4SLinus Torvalds 	sn_cache = kmem_cache_create("shared_policy_node",
14131da177e4SLinus Torvalds 				     sizeof(struct sp_node),
14141da177e4SLinus Torvalds 				     0, SLAB_PANIC, NULL, NULL);
14151da177e4SLinus Torvalds 
14161da177e4SLinus Torvalds 	/* Set interleaving policy for system init. This way not all
14171da177e4SLinus Torvalds 	   the data structures allocated at system boot end up in node zero. */
14181da177e4SLinus Torvalds 
14198bccd85fSChristoph Lameter 	if (do_set_mempolicy(MPOL_INTERLEAVE, &node_online_map))
14201da177e4SLinus Torvalds 		printk("numa_policy_init: interleaving failed\n");
14211da177e4SLinus Torvalds }
14221da177e4SLinus Torvalds 
14238bccd85fSChristoph Lameter /* Reset policy of current process to default */
14241da177e4SLinus Torvalds void numa_default_policy(void)
14251da177e4SLinus Torvalds {
14268bccd85fSChristoph Lameter 	do_set_mempolicy(MPOL_DEFAULT, NULL);
14271da177e4SLinus Torvalds }
142868860ec1SPaul Jackson 
142968860ec1SPaul Jackson /* Migrate a policy to a different set of nodes */
143074cb2155SPaul Jackson void mpol_rebind_policy(struct mempolicy *pol, const nodemask_t *newmask)
143168860ec1SPaul Jackson {
143274cb2155SPaul Jackson 	nodemask_t *mpolmask;
143368860ec1SPaul Jackson 	nodemask_t tmp;
143468860ec1SPaul Jackson 
143568860ec1SPaul Jackson 	if (!pol)
143668860ec1SPaul Jackson 		return;
143774cb2155SPaul Jackson 	mpolmask = &pol->cpuset_mems_allowed;
143874cb2155SPaul Jackson 	if (nodes_equal(*mpolmask, *newmask))
143974cb2155SPaul Jackson 		return;
144068860ec1SPaul Jackson 
144168860ec1SPaul Jackson 	switch (pol->policy) {
144268860ec1SPaul Jackson 	case MPOL_DEFAULT:
144368860ec1SPaul Jackson 		break;
144468860ec1SPaul Jackson 	case MPOL_INTERLEAVE:
144574cb2155SPaul Jackson 		nodes_remap(tmp, pol->v.nodes, *mpolmask, *newmask);
144668860ec1SPaul Jackson 		pol->v.nodes = tmp;
144774cb2155SPaul Jackson 		*mpolmask = *newmask;
144874cb2155SPaul Jackson 		current->il_next = node_remap(current->il_next,
144974cb2155SPaul Jackson 						*mpolmask, *newmask);
145068860ec1SPaul Jackson 		break;
145168860ec1SPaul Jackson 	case MPOL_PREFERRED:
145268860ec1SPaul Jackson 		pol->v.preferred_node = node_remap(pol->v.preferred_node,
145374cb2155SPaul Jackson 						*mpolmask, *newmask);
145474cb2155SPaul Jackson 		*mpolmask = *newmask;
145568860ec1SPaul Jackson 		break;
145668860ec1SPaul Jackson 	case MPOL_BIND: {
145768860ec1SPaul Jackson 		nodemask_t nodes;
145868860ec1SPaul Jackson 		struct zone **z;
145968860ec1SPaul Jackson 		struct zonelist *zonelist;
146068860ec1SPaul Jackson 
146168860ec1SPaul Jackson 		nodes_clear(nodes);
146268860ec1SPaul Jackson 		for (z = pol->v.zonelist->zones; *z; z++)
146368860ec1SPaul Jackson 			node_set((*z)->zone_pgdat->node_id, nodes);
146474cb2155SPaul Jackson 		nodes_remap(tmp, nodes, *mpolmask, *newmask);
146568860ec1SPaul Jackson 		nodes = tmp;
146668860ec1SPaul Jackson 
146768860ec1SPaul Jackson 		zonelist = bind_zonelist(&nodes);
146868860ec1SPaul Jackson 
146968860ec1SPaul Jackson 		/* If no mem, then zonelist is NULL and we keep old zonelist.
147068860ec1SPaul Jackson 		 * If that old zonelist has no remaining mems_allowed nodes,
147168860ec1SPaul Jackson 		 * then zonelist_policy() will "FALL THROUGH" to MPOL_DEFAULT.
147268860ec1SPaul Jackson 		 */
147368860ec1SPaul Jackson 
147468860ec1SPaul Jackson 		if (zonelist) {
147568860ec1SPaul Jackson 			/* Good - got mem - substitute new zonelist */
147668860ec1SPaul Jackson 			kfree(pol->v.zonelist);
147768860ec1SPaul Jackson 			pol->v.zonelist = zonelist;
147868860ec1SPaul Jackson 		}
147974cb2155SPaul Jackson 		*mpolmask = *newmask;
148068860ec1SPaul Jackson 		break;
148168860ec1SPaul Jackson 	}
148268860ec1SPaul Jackson 	default:
148368860ec1SPaul Jackson 		BUG();
148468860ec1SPaul Jackson 		break;
148568860ec1SPaul Jackson 	}
148668860ec1SPaul Jackson }
148768860ec1SPaul Jackson 
148868860ec1SPaul Jackson /*
148974cb2155SPaul Jackson  * Wrapper for mpol_rebind_policy() that just requires task
149074cb2155SPaul Jackson  * pointer, and updates task mempolicy.
149168860ec1SPaul Jackson  */
149274cb2155SPaul Jackson 
149374cb2155SPaul Jackson void mpol_rebind_task(struct task_struct *tsk, const nodemask_t *new)
149468860ec1SPaul Jackson {
149574cb2155SPaul Jackson 	mpol_rebind_policy(tsk->mempolicy, new);
149668860ec1SPaul Jackson }
14971a75a6c8SChristoph Lameter 
14981a75a6c8SChristoph Lameter /*
14994225399aSPaul Jackson  * Rebind each vma in mm to new nodemask.
15004225399aSPaul Jackson  *
15014225399aSPaul Jackson  * Call holding a reference to mm.  Takes mm->mmap_sem during call.
15024225399aSPaul Jackson  */
15034225399aSPaul Jackson 
15044225399aSPaul Jackson void mpol_rebind_mm(struct mm_struct *mm, nodemask_t *new)
15054225399aSPaul Jackson {
15064225399aSPaul Jackson 	struct vm_area_struct *vma;
15074225399aSPaul Jackson 
15084225399aSPaul Jackson 	down_write(&mm->mmap_sem);
15094225399aSPaul Jackson 	for (vma = mm->mmap; vma; vma = vma->vm_next)
15104225399aSPaul Jackson 		mpol_rebind_policy(vma->vm_policy, new);
15114225399aSPaul Jackson 	up_write(&mm->mmap_sem);
15124225399aSPaul Jackson }
15134225399aSPaul Jackson 
15144225399aSPaul Jackson /*
15151a75a6c8SChristoph Lameter  * Display pages allocated per node and memory policy via /proc.
15161a75a6c8SChristoph Lameter  */
15171a75a6c8SChristoph Lameter 
15181a75a6c8SChristoph Lameter static const char *policy_types[] = { "default", "prefer", "bind",
15191a75a6c8SChristoph Lameter 				      "interleave" };
15201a75a6c8SChristoph Lameter 
15211a75a6c8SChristoph Lameter /*
15221a75a6c8SChristoph Lameter  * Convert a mempolicy into a string.
15231a75a6c8SChristoph Lameter  * Returns the number of characters in buffer (if positive)
15241a75a6c8SChristoph Lameter  * or an error (negative)
15251a75a6c8SChristoph Lameter  */
15261a75a6c8SChristoph Lameter static inline int mpol_to_str(char *buffer, int maxlen, struct mempolicy *pol)
15271a75a6c8SChristoph Lameter {
15281a75a6c8SChristoph Lameter 	char *p = buffer;
15291a75a6c8SChristoph Lameter 	int l;
15301a75a6c8SChristoph Lameter 	nodemask_t nodes;
15311a75a6c8SChristoph Lameter 	int mode = pol ? pol->policy : MPOL_DEFAULT;
15321a75a6c8SChristoph Lameter 
15331a75a6c8SChristoph Lameter 	switch (mode) {
15341a75a6c8SChristoph Lameter 	case MPOL_DEFAULT:
15351a75a6c8SChristoph Lameter 		nodes_clear(nodes);
15361a75a6c8SChristoph Lameter 		break;
15371a75a6c8SChristoph Lameter 
15381a75a6c8SChristoph Lameter 	case MPOL_PREFERRED:
15391a75a6c8SChristoph Lameter 		nodes_clear(nodes);
15401a75a6c8SChristoph Lameter 		node_set(pol->v.preferred_node, nodes);
15411a75a6c8SChristoph Lameter 		break;
15421a75a6c8SChristoph Lameter 
15431a75a6c8SChristoph Lameter 	case MPOL_BIND:
15441a75a6c8SChristoph Lameter 		get_zonemask(pol, &nodes);
15451a75a6c8SChristoph Lameter 		break;
15461a75a6c8SChristoph Lameter 
15471a75a6c8SChristoph Lameter 	case MPOL_INTERLEAVE:
15481a75a6c8SChristoph Lameter 		nodes = pol->v.nodes;
15491a75a6c8SChristoph Lameter 		break;
15501a75a6c8SChristoph Lameter 
15511a75a6c8SChristoph Lameter 	default:
15521a75a6c8SChristoph Lameter 		BUG();
15531a75a6c8SChristoph Lameter 		return -EFAULT;
15541a75a6c8SChristoph Lameter 	}
15551a75a6c8SChristoph Lameter 
15561a75a6c8SChristoph Lameter 	l = strlen(policy_types[mode]);
15571a75a6c8SChristoph Lameter  	if (buffer + maxlen < p + l + 1)
15581a75a6c8SChristoph Lameter  		return -ENOSPC;
15591a75a6c8SChristoph Lameter 
15601a75a6c8SChristoph Lameter 	strcpy(p, policy_types[mode]);
15611a75a6c8SChristoph Lameter 	p += l;
15621a75a6c8SChristoph Lameter 
15631a75a6c8SChristoph Lameter 	if (!nodes_empty(nodes)) {
15641a75a6c8SChristoph Lameter 		if (buffer + maxlen < p + 2)
15651a75a6c8SChristoph Lameter 			return -ENOSPC;
15661a75a6c8SChristoph Lameter 		*p++ = '=';
15671a75a6c8SChristoph Lameter 	 	p += nodelist_scnprintf(p, buffer + maxlen - p, nodes);
15681a75a6c8SChristoph Lameter 	}
15691a75a6c8SChristoph Lameter 	return p - buffer;
15701a75a6c8SChristoph Lameter }
15711a75a6c8SChristoph Lameter 
15721a75a6c8SChristoph Lameter struct numa_maps {
15731a75a6c8SChristoph Lameter 	unsigned long pages;
15741a75a6c8SChristoph Lameter 	unsigned long anon;
15751a75a6c8SChristoph Lameter 	unsigned long mapped;
15761a75a6c8SChristoph Lameter 	unsigned long mapcount_max;
15771a75a6c8SChristoph Lameter 	unsigned long node[MAX_NUMNODES];
15781a75a6c8SChristoph Lameter };
15791a75a6c8SChristoph Lameter 
15801a75a6c8SChristoph Lameter static void gather_stats(struct page *page, void *private)
15811a75a6c8SChristoph Lameter {
15821a75a6c8SChristoph Lameter 	struct numa_maps *md = private;
15831a75a6c8SChristoph Lameter 	int count = page_mapcount(page);
15841a75a6c8SChristoph Lameter 
15851a75a6c8SChristoph Lameter 	if (count)
15861a75a6c8SChristoph Lameter 		md->mapped++;
15871a75a6c8SChristoph Lameter 
15881a75a6c8SChristoph Lameter 	if (count > md->mapcount_max)
15891a75a6c8SChristoph Lameter 		md->mapcount_max = count;
15901a75a6c8SChristoph Lameter 
15911a75a6c8SChristoph Lameter 	md->pages++;
15921a75a6c8SChristoph Lameter 
15931a75a6c8SChristoph Lameter 	if (PageAnon(page))
15941a75a6c8SChristoph Lameter 		md->anon++;
15951a75a6c8SChristoph Lameter 
15961a75a6c8SChristoph Lameter 	md->node[page_to_nid(page)]++;
15971a75a6c8SChristoph Lameter 	cond_resched();
15981a75a6c8SChristoph Lameter }
15991a75a6c8SChristoph Lameter 
16001a75a6c8SChristoph Lameter int show_numa_map(struct seq_file *m, void *v)
16011a75a6c8SChristoph Lameter {
16021a75a6c8SChristoph Lameter 	struct task_struct *task = m->private;
16031a75a6c8SChristoph Lameter 	struct vm_area_struct *vma = v;
16041a75a6c8SChristoph Lameter 	struct numa_maps *md;
16051a75a6c8SChristoph Lameter 	int n;
16061a75a6c8SChristoph Lameter 	char buffer[50];
16071a75a6c8SChristoph Lameter 
16081a75a6c8SChristoph Lameter 	if (!vma->vm_mm)
16091a75a6c8SChristoph Lameter 		return 0;
16101a75a6c8SChristoph Lameter 
16111a75a6c8SChristoph Lameter 	md = kzalloc(sizeof(struct numa_maps), GFP_KERNEL);
16121a75a6c8SChristoph Lameter 	if (!md)
16131a75a6c8SChristoph Lameter 		return 0;
16141a75a6c8SChristoph Lameter 
16151a75a6c8SChristoph Lameter 	check_pgd_range(vma, vma->vm_start, vma->vm_end,
16161a75a6c8SChristoph Lameter 		    &node_online_map, MPOL_MF_STATS, md);
16171a75a6c8SChristoph Lameter 
16181a75a6c8SChristoph Lameter 	if (md->pages) {
16191a75a6c8SChristoph Lameter 		mpol_to_str(buffer, sizeof(buffer),
16201a75a6c8SChristoph Lameter 			    get_vma_policy(task, vma, vma->vm_start));
16211a75a6c8SChristoph Lameter 
16221a75a6c8SChristoph Lameter 		seq_printf(m, "%08lx %s pages=%lu mapped=%lu maxref=%lu",
16231a75a6c8SChristoph Lameter 			   vma->vm_start, buffer, md->pages,
16241a75a6c8SChristoph Lameter 			   md->mapped, md->mapcount_max);
16251a75a6c8SChristoph Lameter 
16261a75a6c8SChristoph Lameter 		if (md->anon)
16271a75a6c8SChristoph Lameter 			seq_printf(m," anon=%lu",md->anon);
16281a75a6c8SChristoph Lameter 
16291a75a6c8SChristoph Lameter 		for_each_online_node(n)
16301a75a6c8SChristoph Lameter 			if (md->node[n])
16311a75a6c8SChristoph Lameter 				seq_printf(m, " N%d=%lu", n, md->node[n]);
16321a75a6c8SChristoph Lameter 
16331a75a6c8SChristoph Lameter 		seq_putc(m, '\n');
16341a75a6c8SChristoph Lameter 	}
16351a75a6c8SChristoph Lameter 	kfree(md);
16361a75a6c8SChristoph Lameter 
16371a75a6c8SChristoph Lameter 	if (m->count < m->size)
16381a75a6c8SChristoph Lameter 		m->version = (vma != get_gate_vma(task)) ? vma->vm_start : 0;
16391a75a6c8SChristoph Lameter 	return 0;
16401a75a6c8SChristoph Lameter }
16411a75a6c8SChristoph Lameter 
1642