migrate.c 26.6 KB
Newer Older
C
Christoph Lameter 已提交
1 2 3 4 5 6 7 8 9 10 11
/*
 * Memory Migration functionality - linux/mm/migration.c
 *
 * Copyright (C) 2006 Silicon Graphics, Inc., Christoph Lameter
 *
 * Page migration was first developed in the context of the memory hotplug
 * project. The main authors of the migration code are:
 *
 * IWAMOTO Toshihiro <iwamoto@valinux.co.jp>
 * Hirokazu Takahashi <taka@valinux.co.jp>
 * Dave Hansen <haveblue@us.ibm.com>
C
Christoph Lameter 已提交
12
 * Christoph Lameter
C
Christoph Lameter 已提交
13 14 15 16 17
 */

#include <linux/migrate.h>
#include <linux/module.h>
#include <linux/swap.h>
18
#include <linux/swapops.h>
C
Christoph Lameter 已提交
19
#include <linux/pagemap.h>
20
#include <linux/buffer_head.h>
C
Christoph Lameter 已提交
21
#include <linux/mm_inline.h>
22
#include <linux/nsproxy.h>
C
Christoph Lameter 已提交
23
#include <linux/pagevec.h>
24
#include <linux/ksm.h>
C
Christoph Lameter 已提交
25 26 27 28
#include <linux/rmap.h>
#include <linux/topology.h>
#include <linux/cpu.h>
#include <linux/cpuset.h>
29
#include <linux/writeback.h>
30 31
#include <linux/mempolicy.h>
#include <linux/vmalloc.h>
32
#include <linux/security.h>
33
#include <linux/memcontrol.h>
34
#include <linux/syscalls.h>
35
#include <linux/gfp.h>
C
Christoph Lameter 已提交
36 37 38 39 40 41

#include "internal.h"

#define lru_to_page(_head) (list_entry((_head)->prev, struct page, lru))

/*
42 43
 * migrate_prep() needs to be called before we start compiling a list of pages
 * to be migrated using isolate_lru_page().
C
Christoph Lameter 已提交
44 45 46 47 48 49 50 51 52 53 54 55 56 57 58
 */
int migrate_prep(void)
{
	/*
	 * Clear the LRU lists so pages can be isolated.
	 * Note that pages may be moved off the LRU after we have
	 * drained them. Those pages will fail to migrate like other
	 * pages that may be busy.
	 */
	lru_add_drain_all();

	return 0;
}

/*
L
Lee Schermerhorn 已提交
59 60
 * Add isolated pages on the list back to the LRU under page lock
 * to avoid leaking evictable pages back onto unevictable list.
C
Christoph Lameter 已提交
61
 */
62
void putback_lru_pages(struct list_head *l)
C
Christoph Lameter 已提交
63 64 65 66 67
{
	struct page *page;
	struct page *page2;

	list_for_each_entry_safe(page, page2, l, lru) {
68
		list_del(&page->lru);
K
KOSAKI Motohiro 已提交
69
		dec_zone_page_state(page, NR_ISOLATED_ANON +
70
				page_is_file_cache(page));
L
Lee Schermerhorn 已提交
71
		putback_lru_page(page);
C
Christoph Lameter 已提交
72 73 74
	}
}

75 76 77
/*
 * Restore a potential migration pte to a working pte entry
 */
78 79
static int remove_migration_pte(struct page *new, struct vm_area_struct *vma,
				 unsigned long addr, void *old)
80 81 82 83 84 85 86 87 88 89 90
{
	struct mm_struct *mm = vma->vm_mm;
	swp_entry_t entry;
 	pgd_t *pgd;
 	pud_t *pud;
 	pmd_t *pmd;
	pte_t *ptep, pte;
 	spinlock_t *ptl;

 	pgd = pgd_offset(mm, addr);
	if (!pgd_present(*pgd))
91
		goto out;
92 93 94

	pud = pud_offset(pgd, addr);
	if (!pud_present(*pud))
95
		goto out;
96 97 98

	pmd = pmd_offset(pud, addr);
	if (!pmd_present(*pmd))
99
		goto out;
100 101 102 103 104

	ptep = pte_offset_map(pmd, addr);

	if (!is_swap_pte(*ptep)) {
		pte_unmap(ptep);
105
		goto out;
106 107 108 109 110 111
 	}

 	ptl = pte_lockptr(mm, pmd);
 	spin_lock(ptl);
	pte = *ptep;
	if (!is_swap_pte(pte))
112
		goto unlock;
113 114 115

	entry = pte_to_swp_entry(pte);

116 117 118
	if (!is_migration_entry(entry) ||
	    migration_entry_to_page(entry) != old)
		goto unlock;
119 120 121 122 123

	get_page(new);
	pte = pte_mkold(mk_pte(new, vma->vm_page_prot));
	if (is_write_migration_entry(entry))
		pte = pte_mkwrite(pte);
124
	flush_cache_page(vma, addr, pte_pfn(pte));
125
	set_pte_at(mm, addr, ptep, pte);
126 127 128 129 130 131 132

	if (PageAnon(new))
		page_add_anon_rmap(new, vma, addr);
	else
		page_add_file_rmap(new);

	/* No need to invalidate - it was non-present before */
133
	update_mmu_cache(vma, addr, ptep);
134
unlock:
135
	pte_unmap_unlock(ptep, ptl);
136 137
out:
	return SWAP_AGAIN;
138 139
}

140 141 142 143 144 145
/*
 * Get rid of all migration entries and replace them by
 * references to the indicated page.
 */
static void remove_migration_ptes(struct page *old, struct page *new)
{
146
	rmap_walk(new, remove_migration_pte, old);
147 148
}

149 150 151 152 153 154 155 156 157 158 159 160 161 162 163 164 165 166 167 168 169 170 171 172 173 174
/*
 * Something used the pte of a page under migration. We need to
 * get to the page and wait until migration is finished.
 * When we return from this function the fault will be retried.
 *
 * This function is called from do_swap_page().
 */
void migration_entry_wait(struct mm_struct *mm, pmd_t *pmd,
				unsigned long address)
{
	pte_t *ptep, pte;
	spinlock_t *ptl;
	swp_entry_t entry;
	struct page *page;

	ptep = pte_offset_map_lock(mm, pmd, address, &ptl);
	pte = *ptep;
	if (!is_swap_pte(pte))
		goto out;

	entry = pte_to_swp_entry(pte);
	if (!is_migration_entry(entry))
		goto out;

	page = migration_entry_to_page(entry);

N
Nick Piggin 已提交
175 176 177 178 179 180 181 182 183
	/*
	 * Once radix-tree replacement of page migration started, page_count
	 * *must* be zero. And, we don't want to call wait_on_page_locked()
	 * against a page without get_page().
	 * So, we use get_page_unless_zero(), here. Even failed, page fault
	 * will occur again.
	 */
	if (!get_page_unless_zero(page))
		goto out;
184 185 186 187 188 189 190 191
	pte_unmap_unlock(ptep, ptl);
	wait_on_page_locked(page);
	put_page(page);
	return;
out:
	pte_unmap_unlock(ptep, ptl);
}

C
Christoph Lameter 已提交
192
/*
193
 * Replace the page in the mapping.
194 195 196 197
 *
 * The number of remaining references must be:
 * 1 for anonymous pages without a mapping
 * 2 for pages with a mapping
198
 * 3 for pages with a mapping and PagePrivate/PagePrivate2 set.
C
Christoph Lameter 已提交
199
 */
200 201
static int migrate_page_move_mapping(struct address_space *mapping,
		struct page *newpage, struct page *page)
C
Christoph Lameter 已提交
202
{
N
Nick Piggin 已提交
203
	int expected_count;
204
	void **pslot;
C
Christoph Lameter 已提交
205

206
	if (!mapping) {
207
		/* Anonymous page without mapping */
208 209 210 211 212
		if (page_count(page) != 1)
			return -EAGAIN;
		return 0;
	}

N
Nick Piggin 已提交
213
	spin_lock_irq(&mapping->tree_lock);
C
Christoph Lameter 已提交
214

215 216
	pslot = radix_tree_lookup_slot(&mapping->page_tree,
 					page_index(page));
C
Christoph Lameter 已提交
217

218
	expected_count = 2 + page_has_private(page);
N
Nick Piggin 已提交
219
	if (page_count(page) != expected_count ||
220
			(struct page *)radix_tree_deref_slot(pslot) != page) {
N
Nick Piggin 已提交
221
		spin_unlock_irq(&mapping->tree_lock);
222
		return -EAGAIN;
C
Christoph Lameter 已提交
223 224
	}

N
Nick Piggin 已提交
225
	if (!page_freeze_refs(page, expected_count)) {
N
Nick Piggin 已提交
226
		spin_unlock_irq(&mapping->tree_lock);
N
Nick Piggin 已提交
227 228 229
		return -EAGAIN;
	}

C
Christoph Lameter 已提交
230 231 232
	/*
	 * Now we know that no one else is looking at the page.
	 */
233
	get_page(newpage);	/* add cache reference */
C
Christoph Lameter 已提交
234 235 236 237 238
	if (PageSwapCache(page)) {
		SetPageSwapCache(newpage);
		set_page_private(newpage, page_private(page));
	}

239 240
	radix_tree_replace_slot(pslot, newpage);

N
Nick Piggin 已提交
241
	page_unfreeze_refs(page, expected_count);
242 243 244 245
	/*
	 * Drop cache reference from old page.
	 * We know this isn't the last reference.
	 */
C
Christoph Lameter 已提交
246
	__put_page(page);
247

248 249 250 251 252 253 254 255 256 257 258 259
	/*
	 * If moved to a different zone then also account
	 * the page for that zone. Other VM counters will be
	 * taken care of when we establish references to the
	 * new page and drop references to the old page.
	 *
	 * Note that anonymous pages are accounted for
	 * via NR_FILE_PAGES and NR_ANON_PAGES if they
	 * are mapped to swap space.
	 */
	__dec_zone_page_state(page, NR_FILE_PAGES);
	__inc_zone_page_state(newpage, NR_FILE_PAGES);
260 261 262 263
	if (PageSwapBacked(page)) {
		__dec_zone_page_state(page, NR_SHMEM);
		__inc_zone_page_state(newpage, NR_SHMEM);
	}
N
Nick Piggin 已提交
264
	spin_unlock_irq(&mapping->tree_lock);
C
Christoph Lameter 已提交
265 266 267 268 269 270 271

	return 0;
}

/*
 * Copy the page to its new location
 */
272
static void migrate_page_copy(struct page *newpage, struct page *page)
C
Christoph Lameter 已提交
273 274 275 276 277 278 279 280 281
{
	copy_highpage(newpage, page);

	if (PageError(page))
		SetPageError(newpage);
	if (PageReferenced(page))
		SetPageReferenced(newpage);
	if (PageUptodate(page))
		SetPageUptodate(newpage);
L
Lee Schermerhorn 已提交
282 283
	if (TestClearPageActive(page)) {
		VM_BUG_ON(PageUnevictable(page));
C
Christoph Lameter 已提交
284
		SetPageActive(newpage);
285 286
	} else if (TestClearPageUnevictable(page))
		SetPageUnevictable(newpage);
C
Christoph Lameter 已提交
287 288 289 290 291 292 293
	if (PageChecked(page))
		SetPageChecked(newpage);
	if (PageMappedToDisk(page))
		SetPageMappedToDisk(newpage);

	if (PageDirty(page)) {
		clear_page_dirty_for_io(page);
N
Nick Piggin 已提交
294 295 296 297 298 299 300 301
		/*
		 * Want to mark the page and the radix tree as dirty, and
		 * redo the accounting that clear_page_dirty_for_io undid,
		 * but we can't use set_page_dirty because that function
		 * is actually a signal that all of the page has become dirty.
		 * Wheras only part of our page may be dirty.
		 */
		__set_page_dirty_nobuffers(newpage);
C
Christoph Lameter 已提交
302 303
 	}

N
Nick Piggin 已提交
304
	mlock_migrate_page(newpage, page);
305
	ksm_migrate_page(newpage, page);
N
Nick Piggin 已提交
306

C
Christoph Lameter 已提交
307 308 309 310 311 312 313 314 315 316 317 318 319
	ClearPageSwapCache(page);
	ClearPagePrivate(page);
	set_page_private(page, 0);
	page->mapping = NULL;

	/*
	 * If any waiters have accumulated on the new page then
	 * wake them up.
	 */
	if (PageWriteback(newpage))
		end_page_writeback(newpage);
}

320 321 322 323 324
/************************************************************
 *                    Migration functions
 ***********************************************************/

/* Always fail migration. Used for mappings that are not movable */
325 326
int fail_migrate_page(struct address_space *mapping,
			struct page *newpage, struct page *page)
327 328 329 330 331
{
	return -EIO;
}
EXPORT_SYMBOL(fail_migrate_page);

C
Christoph Lameter 已提交
332 333
/*
 * Common logic to directly migrate a single page suitable for
334
 * pages that do not use PagePrivate/PagePrivate2.
C
Christoph Lameter 已提交
335 336 337
 *
 * Pages are locked upon entry and exit.
 */
338 339
int migrate_page(struct address_space *mapping,
		struct page *newpage, struct page *page)
C
Christoph Lameter 已提交
340 341 342 343 344
{
	int rc;

	BUG_ON(PageWriteback(page));	/* Writeback must be complete */

345
	rc = migrate_page_move_mapping(mapping, newpage, page);
C
Christoph Lameter 已提交
346 347 348 349 350 351 352 353 354

	if (rc)
		return rc;

	migrate_page_copy(newpage, page);
	return 0;
}
EXPORT_SYMBOL(migrate_page);

355
#ifdef CONFIG_BLOCK
356 357 358 359 360
/*
 * Migration function for pages with buffers. This function can only be used
 * if the underlying filesystem guarantees that no other references to "page"
 * exist.
 */
361 362
int buffer_migrate_page(struct address_space *mapping,
		struct page *newpage, struct page *page)
363 364 365 366 367
{
	struct buffer_head *bh, *head;
	int rc;

	if (!page_has_buffers(page))
368
		return migrate_page(mapping, newpage, page);
369 370 371

	head = page_buffers(page);

372
	rc = migrate_page_move_mapping(mapping, newpage, page);
373 374 375 376 377 378 379 380 381 382 383 384 385 386 387 388 389 390 391 392 393 394 395 396 397 398 399 400 401 402 403 404 405 406 407 408 409 410 411 412

	if (rc)
		return rc;

	bh = head;
	do {
		get_bh(bh);
		lock_buffer(bh);
		bh = bh->b_this_page;

	} while (bh != head);

	ClearPagePrivate(page);
	set_page_private(newpage, page_private(page));
	set_page_private(page, 0);
	put_page(page);
	get_page(newpage);

	bh = head;
	do {
		set_bh_page(bh, newpage, bh_offset(bh));
		bh = bh->b_this_page;

	} while (bh != head);

	SetPagePrivate(newpage);

	migrate_page_copy(newpage, page);

	bh = head;
	do {
		unlock_buffer(bh);
 		put_bh(bh);
		bh = bh->b_this_page;

	} while (bh != head);

	return 0;
}
EXPORT_SYMBOL(buffer_migrate_page);
413
#endif
414

415 416 417 418
/*
 * Writeback a page to clean the dirty state
 */
static int writeout(struct address_space *mapping, struct page *page)
419
{
420 421 422 423 424 425 426 427 428 429 430 431 432 433 434 435 436 437
	struct writeback_control wbc = {
		.sync_mode = WB_SYNC_NONE,
		.nr_to_write = 1,
		.range_start = 0,
		.range_end = LLONG_MAX,
		.nonblocking = 1,
		.for_reclaim = 1
	};
	int rc;

	if (!mapping->a_ops->writepage)
		/* No write method for the address space */
		return -EINVAL;

	if (!clear_page_dirty_for_io(page))
		/* Someone else already triggered a write */
		return -EAGAIN;

438
	/*
439 440 441 442 443 444
	 * A dirty page may imply that the underlying filesystem has
	 * the page on some queue. So the page must be clean for
	 * migration. Writeout may mean we loose the lock and the
	 * page state is no longer what we checked for earlier.
	 * At this point we know that the migration attempt cannot
	 * be successful.
445
	 */
446
	remove_migration_ptes(page, page);
447

448
	rc = mapping->a_ops->writepage(page, &wbc);
449

450 451 452 453
	if (rc != AOP_WRITEPAGE_ACTIVATE)
		/* unlocked. Relock */
		lock_page(page);

H
Hugh Dickins 已提交
454
	return (rc < 0) ? -EIO : -EAGAIN;
455 456 457 458 459 460 461 462 463 464
}

/*
 * Default handling if a filesystem does not provide a migration function.
 */
static int fallback_migrate_page(struct address_space *mapping,
	struct page *newpage, struct page *page)
{
	if (PageDirty(page))
		return writeout(mapping, page);
465 466 467 468 469

	/*
	 * Buffers may be managed in a filesystem specific way.
	 * We must have no buffers or drop them.
	 */
470
	if (page_has_private(page) &&
471 472 473 474 475 476
	    !try_to_release_page(page, GFP_KERNEL))
		return -EAGAIN;

	return migrate_page(mapping, newpage, page);
}

477 478 479 480 481 482
/*
 * Move a page to a newly allocated page
 * The page is locked and all ptes have been successfully removed.
 *
 * The new page will have replaced the old page if this function
 * is successful.
L
Lee Schermerhorn 已提交
483 484 485 486
 *
 * Return value:
 *   < 0 - error code
 *  == 0 - success
487
 */
488 489
static int move_to_new_page(struct page *newpage, struct page *page,
						int remap_swapcache)
490 491 492 493 494 495 496 497 498
{
	struct address_space *mapping;
	int rc;

	/*
	 * Block others from accessing the page when we get around to
	 * establishing additional references. We are the only one
	 * holding a reference to the new page at this point.
	 */
N
Nick Piggin 已提交
499
	if (!trylock_page(newpage))
500 501 502 503 504
		BUG();

	/* Prepare mapping for the new page.*/
	newpage->index = page->index;
	newpage->mapping = page->mapping;
R
Rik van Riel 已提交
505 506
	if (PageSwapBacked(page))
		SetPageSwapBacked(newpage);
507 508 509 510 511 512 513 514 515 516 517 518 519 520 521 522 523

	mapping = page_mapping(page);
	if (!mapping)
		rc = migrate_page(mapping, newpage, page);
	else if (mapping->a_ops->migratepage)
		/*
		 * Most pages have a mapping and most filesystems
		 * should provide a migration function. Anonymous
		 * pages are part of swap space which also has its
		 * own migration function. This is the most common
		 * path for page migration.
		 */
		rc = mapping->a_ops->migratepage(mapping,
						newpage, page);
	else
		rc = fallback_migrate_page(mapping, newpage, page);

524
	if (rc) {
525
		newpage->mapping = NULL;
526 527 528 529
	} else {
		if (remap_swapcache)
			remove_migration_ptes(page, newpage);
	}
530 531 532 533 534 535 536 537 538 539

	unlock_page(newpage);

	return rc;
}

/*
 * Obtain the lock on page, remove all ptes and migrate the page
 * to the newly allocated page in newpage.
 */
540
static int unmap_and_move(new_page_t get_new_page, unsigned long private,
541
			struct page *page, int force, int offlining)
542 543
{
	int rc = 0;
544 545
	int *result = NULL;
	struct page *newpage = get_new_page(page, private, &result);
546
	int remap_swapcache = 1;
547
	int rcu_locked = 0;
548
	int charge = 0;
549
	struct mem_cgroup *mem = NULL;
550
	struct anon_vma *anon_vma = NULL;
551 552 553

	if (!newpage)
		return -ENOMEM;
554

L
Lee Schermerhorn 已提交
555
	if (page_count(page) == 1) {
556
		/* page was freed from under us. So we are done. */
557
		goto move_newpage;
L
Lee Schermerhorn 已提交
558
	}
559

560
	/* prepare cgroup just returns 0 or -ENOMEM */
561
	rc = -EAGAIN;
562

N
Nick Piggin 已提交
563
	if (!trylock_page(page)) {
564
		if (!force)
565
			goto move_newpage;
566 567 568
		lock_page(page);
	}

569 570 571 572 573 574 575 576 577 578 579 580 581 582
	/*
	 * Only memory hotplug's offline_pages() caller has locked out KSM,
	 * and can safely migrate a KSM page.  The other cases have skipped
	 * PageKsm along with PageReserved - but it is only now when we have
	 * the page lock that we can be certain it will not go KSM beneath us
	 * (KSM will not upgrade a page from PageAnon to PageKsm when it sees
	 * its pagecount raised, but only here do we take the page lock which
	 * serializes that).
	 */
	if (PageKsm(page) && !offlining) {
		rc = -EBUSY;
		goto unlock;
	}

583 584 585 586 587 588 589 590
	/* charge against new page */
	charge = mem_cgroup_prepare_migration(page, &mem);
	if (charge == -ENOMEM) {
		rc = -ENOMEM;
		goto unlock;
	}
	BUG_ON(charge);

591 592
	if (PageWriteback(page)) {
		if (!force)
593
			goto uncharge;
594 595 596
		wait_on_page_writeback(page);
	}
	/*
597 598 599 600
	 * By try_to_unmap(), page->mapcount goes down to 0 here. In this case,
	 * we cannot notice that anon_vma is freed while we migrates a page.
	 * This rcu_read_lock() delays freeing anon_vma pointer until the end
	 * of migration. File cache pages are no problem because of page_lock()
601 602
	 * File Caches may use write_page() or lock_page() in migration, then,
	 * just care Anon page here.
603
	 */
604 605 606
	if (PageAnon(page)) {
		rcu_read_lock();
		rcu_locked = 1;
607

608 609 610 611
		/* Determine how to safely use anon_vma */
		if (!page_mapped(page)) {
			if (!PageSwapCache(page))
				goto rcu_unlock;
612

613 614 615 616 617 618 619 620 621 622 623 624 625 626 627 628 629 630 631 632 633 634
			/*
			 * We cannot be sure that the anon_vma of an unmapped
			 * swapcache page is safe to use because we don't
			 * know in advance if the VMA that this page belonged
			 * to still exists. If the VMA and others sharing the
			 * data have been freed, then the anon_vma could
			 * already be invalid.
			 *
			 * To avoid this possibility, swapcache pages get
			 * migrated but are not remapped when migration
			 * completes
			 */
			remap_swapcache = 0;
		} else {
			/*
			 * Take a reference count on the anon_vma if the
			 * page is mapped so that it is guaranteed to
			 * exist when the page is remapped later
			 */
			anon_vma = page_anon_vma(page);
			atomic_inc(&anon_vma->external_refcount);
		}
635
	}
636

637
	/*
638 639 640 641 642 643 644 645 646 647
	 * Corner case handling:
	 * 1. When a new swap-cache page is read into, it is added to the LRU
	 * and treated as swapcache but it has no rmap yet.
	 * Calling try_to_unmap() against a page->mapping==NULL page will
	 * trigger a BUG.  So handle it here.
	 * 2. An orphaned page (see truncate_complete_page) might have
	 * fs-private metadata. The page can be picked up due to memory
	 * offlining.  Everywhere else except page reclaim, the page is
	 * invisible to the vm, so the page can not be migrated.  So try to
	 * free the metadata, so the page can be freed.
648
	 */
649
	if (!page->mapping) {
650
		if (!PageAnon(page) && page_has_private(page)) {
651 652 653 654 655 656 657 658
			/*
			 * Go direct to try_to_free_buffers() here because
			 * a) that's what try_to_release_page() would do anyway
			 * b) we may be under rcu_read_lock() here, so we can't
			 *    use GFP_KERNEL which is what try_to_release_page()
			 *    needs to be effective.
			 */
			try_to_free_buffers(page);
659
			goto rcu_unlock;
660
		}
661
		goto skip_unmap;
662 663
	}

664
	/* Establish migration ptes or remove ptes */
665
	try_to_unmap(page, TTU_MIGRATION|TTU_IGNORE_MLOCK|TTU_IGNORE_ACCESS);
666

667
skip_unmap:
668
	if (!page_mapped(page))
669
		rc = move_to_new_page(newpage, page, remap_swapcache);
670

671
	if (rc && remap_swapcache)
672
		remove_migration_ptes(page, page);
673
rcu_unlock:
674 675

	/* Drop an anon_vma reference if we took one */
676
	if (anon_vma && atomic_dec_and_lock(&anon_vma->external_refcount, &anon_vma->lock)) {
677 678 679 680 681 682
		int empty = list_empty(&anon_vma->head);
		spin_unlock(&anon_vma->lock);
		if (empty)
			anon_vma_free(anon_vma);
	}

683 684
	if (rcu_locked)
		rcu_read_unlock();
685 686 687
uncharge:
	if (!charge)
		mem_cgroup_end_migration(mem, page, newpage);
688 689
unlock:
	unlock_page(page);
690

691
	if (rc != -EAGAIN) {
692 693 694 695 696 697 698
 		/*
 		 * A page that has been migrated has all references
 		 * removed and will be freed. A page that has not been
 		 * migrated will have kepts its references and be
 		 * restored.
 		 */
 		list_del(&page->lru);
K
KOSAKI Motohiro 已提交
699
		dec_zone_page_state(page, NR_ISOLATED_ANON +
700
				page_is_file_cache(page));
L
Lee Schermerhorn 已提交
701
		putback_lru_page(page);
702
	}
703 704

move_newpage:
L
Lee Schermerhorn 已提交
705

706 707 708 709
	/*
	 * Move the new page to the LRU. If migration was not successful
	 * then this will free the page.
	 */
L
Lee Schermerhorn 已提交
710 711
	putback_lru_page(newpage);

712 713 714 715 716 717
	if (result) {
		if (rc)
			*result = rc;
		else
			*result = page_to_nid(newpage);
	}
718 719 720
	return rc;
}

C
Christoph Lameter 已提交
721 722 723
/*
 * migrate_pages
 *
724 725 726
 * The function takes one list of pages to migrate and a function
 * that determines from the page to be migrated and the private data
 * the target of the move and allocates the page.
C
Christoph Lameter 已提交
727 728 729
 *
 * The function returns after 10 attempts or if no pages
 * are movable anymore because to has become empty
730
 * or no retryable pages exist anymore. All pages will be
G
Gabriel Craciunescu 已提交
731
 * returned to the LRU or freed.
C
Christoph Lameter 已提交
732
 *
733
 * Return: Number of pages not migrated or error code.
C
Christoph Lameter 已提交
734
 */
735
int migrate_pages(struct list_head *from,
736
		new_page_t get_new_page, unsigned long private, int offlining)
C
Christoph Lameter 已提交
737
{
738
	int retry = 1;
C
Christoph Lameter 已提交
739 740 741 742 743 744 745 746 747 748
	int nr_failed = 0;
	int pass = 0;
	struct page *page;
	struct page *page2;
	int swapwrite = current->flags & PF_SWAPWRITE;
	int rc;

	if (!swapwrite)
		current->flags |= PF_SWAPWRITE;

749 750
	for(pass = 0; pass < 10 && retry; pass++) {
		retry = 0;
C
Christoph Lameter 已提交
751

752 753
		list_for_each_entry_safe(page, page2, from, lru) {
			cond_resched();
754

755
			rc = unmap_and_move(get_new_page, private,
756
						page, pass > 2, offlining);
757

758
			switch(rc) {
759 760
			case -ENOMEM:
				goto out;
761
			case -EAGAIN:
762
				retry++;
763 764 765 766
				break;
			case 0:
				break;
			default:
767 768
				/* Permanent failure */
				nr_failed++;
769
				break;
770
			}
C
Christoph Lameter 已提交
771 772
		}
	}
773 774
	rc = 0;
out:
C
Christoph Lameter 已提交
775 776 777
	if (!swapwrite)
		current->flags &= ~PF_SWAPWRITE;

778
	putback_lru_pages(from);
C
Christoph Lameter 已提交
779

780 781
	if (rc)
		return rc;
C
Christoph Lameter 已提交
782

783
	return nr_failed + retry;
C
Christoph Lameter 已提交
784
}
785

786 787 788 789 790 791 792 793 794 795 796 797 798 799 800 801 802 803 804 805 806 807 808 809
#ifdef CONFIG_NUMA
/*
 * Move a list of individual pages
 */
struct page_to_node {
	unsigned long addr;
	struct page *page;
	int node;
	int status;
};

static struct page *new_page_node(struct page *p, unsigned long private,
		int **result)
{
	struct page_to_node *pm = (struct page_to_node *)private;

	while (pm->node != MAX_NUMNODES && pm->page != p)
		pm++;

	if (pm->node == MAX_NUMNODES)
		return NULL;

	*result = &pm->status;

810
	return alloc_pages_exact_node(pm->node,
811
				GFP_HIGHUSER_MOVABLE | GFP_THISNODE, 0);
812 813 814 815 816 817
}

/*
 * Move a set of pages as indicated in the pm array. The addr
 * field must be set to the virtual address of the page to be moved
 * and the node number must contain a valid target node.
818
 * The pm array ends with node = MAX_NUMNODES.
819
 */
820 821 822
static int do_move_page_to_node_array(struct mm_struct *mm,
				      struct page_to_node *pm,
				      int migrate_all)
823 824 825 826 827 828 829 830 831 832 833 834 835 836 837 838
{
	int err;
	struct page_to_node *pp;
	LIST_HEAD(pagelist);

	down_read(&mm->mmap_sem);

	/*
	 * Build a list of pages to migrate
	 */
	for (pp = pm; pp->node != MAX_NUMNODES; pp++) {
		struct vm_area_struct *vma;
		struct page *page;

		err = -EFAULT;
		vma = find_vma(mm, pp->addr);
839
		if (!vma || !vma_migratable(vma))
840 841 842
			goto set_status;

		page = follow_page(vma, pp->addr, FOLL_GET);
843 844 845 846 847

		err = PTR_ERR(page);
		if (IS_ERR(page))
			goto set_status;

848 849 850 851
		err = -ENOENT;
		if (!page)
			goto set_status;

852 853
		/* Use PageReserved to check for zero page */
		if (PageReserved(page) || PageKsm(page))
854 855 856 857 858 859 860 861 862 863 864 865 866 867 868 869
			goto put_and_set;

		pp->page = page;
		err = page_to_nid(page);

		if (err == pp->node)
			/*
			 * Node already in the right place
			 */
			goto put_and_set;

		err = -EACCES;
		if (page_mapcount(page) > 1 &&
				!migrate_all)
			goto put_and_set;

870
		err = isolate_lru_page(page);
871
		if (!err) {
872
			list_add_tail(&page->lru, &pagelist);
873 874 875
			inc_zone_page_state(page, NR_ISOLATED_ANON +
					    page_is_file_cache(page));
		}
876 877 878 879 880 881 882 883 884 885 886
put_and_set:
		/*
		 * Either remove the duplicate refcount from
		 * isolate_lru_page() or drop the page ref if it was
		 * not isolated.
		 */
		put_page(page);
set_status:
		pp->status = err;
	}

887
	err = 0;
888 889
	if (!list_empty(&pagelist))
		err = migrate_pages(&pagelist, new_page_node,
890
				(unsigned long)pm, 0);
891 892 893 894 895

	up_read(&mm->mmap_sem);
	return err;
}

896 897 898 899 900 901 902 903 904 905
/*
 * Migrate an array of page address onto an array of nodes and fill
 * the corresponding array of status.
 */
static int do_pages_move(struct mm_struct *mm, struct task_struct *task,
			 unsigned long nr_pages,
			 const void __user * __user *pages,
			 const int __user *nodes,
			 int __user *status, int flags)
{
906
	struct page_to_node *pm;
907
	nodemask_t task_nodes;
908 909 910
	unsigned long chunk_nr_pages;
	unsigned long chunk_start;
	int err;
911 912 913

	task_nodes = cpuset_mems_allowed(task);

914 915 916
	err = -ENOMEM;
	pm = (struct page_to_node *)__get_free_page(GFP_KERNEL);
	if (!pm)
917
		goto out;
918 919 920

	migrate_prep();

921
	/*
922 923
	 * Store a chunk of page_to_node array in a page,
	 * but keep the last one as a marker
924
	 */
925
	chunk_nr_pages = (PAGE_SIZE / sizeof(struct page_to_node)) - 1;
926

927 928 929 930
	for (chunk_start = 0;
	     chunk_start < nr_pages;
	     chunk_start += chunk_nr_pages) {
		int j;
931

932 933 934 935 936 937
		if (chunk_start + chunk_nr_pages > nr_pages)
			chunk_nr_pages = nr_pages - chunk_start;

		/* fill the chunk pm with addrs and nodes from user-space */
		for (j = 0; j < chunk_nr_pages; j++) {
			const void __user *p;
938 939
			int node;

940 941 942 943 944 945
			err = -EFAULT;
			if (get_user(p, pages + j + chunk_start))
				goto out_pm;
			pm[j].addr = (unsigned long) p;

			if (get_user(node, nodes + j + chunk_start))
946 947 948
				goto out_pm;

			err = -ENODEV;
949 950 951
			if (node < 0 || node >= MAX_NUMNODES)
				goto out_pm;

952 953 954 955 956 957 958
			if (!node_state(node, N_HIGH_MEMORY))
				goto out_pm;

			err = -EACCES;
			if (!node_isset(node, task_nodes))
				goto out_pm;

959 960 961 962 963 964 965 966 967 968 969
			pm[j].node = node;
		}

		/* End marker for this chunk */
		pm[chunk_nr_pages].node = MAX_NUMNODES;

		/* Migrate this chunk */
		err = do_move_page_to_node_array(mm, pm,
						 flags & MPOL_MF_MOVE_ALL);
		if (err < 0)
			goto out_pm;
970 971

		/* Return status information */
972 973
		for (j = 0; j < chunk_nr_pages; j++)
			if (put_user(pm[j].status, status + j + chunk_start)) {
974
				err = -EFAULT;
975 976 977 978
				goto out_pm;
			}
	}
	err = 0;
979 980

out_pm:
981
	free_page((unsigned long)pm);
982 983 984 985
out:
	return err;
}

986
/*
987
 * Determine the nodes of an array of pages and store it in an array of status.
988
 */
989 990
static void do_pages_stat_array(struct mm_struct *mm, unsigned long nr_pages,
				const void __user **pages, int *status)
991
{
992 993
	unsigned long i;

994 995
	down_read(&mm->mmap_sem);

996
	for (i = 0; i < nr_pages; i++) {
997
		unsigned long addr = (unsigned long)(*pages);
998 999
		struct vm_area_struct *vma;
		struct page *page;
1000
		int err = -EFAULT;
1001 1002

		vma = find_vma(mm, addr);
1003 1004 1005
		if (!vma)
			goto set_status;

1006
		page = follow_page(vma, addr, 0);
1007 1008 1009 1010 1011

		err = PTR_ERR(page);
		if (IS_ERR(page))
			goto set_status;

1012 1013
		err = -ENOENT;
		/* Use PageReserved to check for zero page */
1014
		if (!page || PageReserved(page) || PageKsm(page))
1015 1016 1017 1018
			goto set_status;

		err = page_to_nid(page);
set_status:
1019 1020 1021 1022 1023 1024 1025 1026 1027 1028 1029 1030 1031 1032 1033 1034 1035 1036 1037 1038 1039
		*status = err;

		pages++;
		status++;
	}

	up_read(&mm->mmap_sem);
}

/*
 * Determine the nodes of a user array of pages and store it in
 * a user array of status.
 */
static int do_pages_stat(struct mm_struct *mm, unsigned long nr_pages,
			 const void __user * __user *pages,
			 int __user *status)
{
#define DO_PAGES_STAT_CHUNK_NR 16
	const void __user *chunk_pages[DO_PAGES_STAT_CHUNK_NR];
	int chunk_status[DO_PAGES_STAT_CHUNK_NR];

1040 1041
	while (nr_pages) {
		unsigned long chunk_nr;
1042

1043 1044 1045 1046 1047 1048
		chunk_nr = nr_pages;
		if (chunk_nr > DO_PAGES_STAT_CHUNK_NR)
			chunk_nr = DO_PAGES_STAT_CHUNK_NR;

		if (copy_from_user(chunk_pages, pages, chunk_nr * sizeof(*chunk_pages)))
			break;
1049 1050 1051

		do_pages_stat_array(mm, chunk_nr, chunk_pages, chunk_status);

1052 1053
		if (copy_to_user(status, chunk_status, chunk_nr * sizeof(*status)))
			break;
1054

1055 1056 1057 1058 1059
		pages += chunk_nr;
		status += chunk_nr;
		nr_pages -= chunk_nr;
	}
	return nr_pages ? -EFAULT : 0;
1060 1061 1062 1063 1064 1065
}

/*
 * Move a list of pages in the address space of the currently executing
 * process.
 */
1066 1067 1068 1069
SYSCALL_DEFINE6(move_pages, pid_t, pid, unsigned long, nr_pages,
		const void __user * __user *, pages,
		const int __user *, nodes,
		int __user *, status, int, flags)
1070
{
1071
	const struct cred *cred = current_cred(), *tcred;
1072 1073
	struct task_struct *task;
	struct mm_struct *mm;
1074
	int err;
1075 1076 1077 1078 1079 1080 1081 1082 1083 1084

	/* Check flags */
	if (flags & ~(MPOL_MF_MOVE|MPOL_MF_MOVE_ALL))
		return -EINVAL;

	if ((flags & MPOL_MF_MOVE_ALL) && !capable(CAP_SYS_NICE))
		return -EPERM;

	/* Find the mm_struct */
	read_lock(&tasklist_lock);
1085
	task = pid ? find_task_by_vpid(pid) : current;
1086 1087 1088 1089 1090 1091 1092 1093 1094 1095 1096 1097 1098 1099 1100 1101
	if (!task) {
		read_unlock(&tasklist_lock);
		return -ESRCH;
	}
	mm = get_task_mm(task);
	read_unlock(&tasklist_lock);

	if (!mm)
		return -EINVAL;

	/*
	 * Check if this process has the right to modify the specified
	 * process. The right exists if the process has administrative
	 * capabilities, superuser privileges or the same
	 * userid as the target process.
	 */
1102 1103
	rcu_read_lock();
	tcred = __task_cred(task);
1104 1105
	if (cred->euid != tcred->suid && cred->euid != tcred->uid &&
	    cred->uid  != tcred->suid && cred->uid  != tcred->uid &&
1106
	    !capable(CAP_SYS_NICE)) {
1107
		rcu_read_unlock();
1108
		err = -EPERM;
1109
		goto out;
1110
	}
1111
	rcu_read_unlock();
1112

1113 1114
 	err = security_task_movememory(task);
 	if (err)
1115
		goto out;
1116

1117 1118 1119 1120
	if (nodes) {
		err = do_pages_move(mm, task, nr_pages, pages, nodes, status,
				    flags);
	} else {
1121
		err = do_pages_stat(mm, nr_pages, pages, status);
1122 1123 1124 1125 1126 1127 1128
	}

out:
	mmput(mm);
	return err;
}

1129 1130 1131 1132 1133 1134 1135 1136 1137 1138 1139
/*
 * Call migration functions in the vma_ops that may prepare
 * memory in a vm for migration. migration functions may perform
 * the migration for vmas that do not have an underlying page struct.
 */
int migrate_vmas(struct mm_struct *mm, const nodemask_t *to,
	const nodemask_t *from, unsigned long flags)
{
 	struct vm_area_struct *vma;
 	int err = 0;

1140
	for (vma = mm->mmap; vma && !err; vma = vma->vm_next) {
1141 1142 1143 1144 1145 1146 1147 1148
 		if (vma->vm_ops && vma->vm_ops->migrate) {
 			err = vma->vm_ops->migrate(vma, to, from, flags);
 			if (err)
 				break;
 		}
 	}
 	return err;
}
1149
#endif