[PATCH v2] mm/gup.c: Simplify and fix check_and_migrate_movable_pages() return codes

From: Alistair Popple
Date: Mon Aug 01 2022 - 20:32:36 EST

When pinning pages with FOLL_LONGTERM check_and_migrate_movable_pages()
is called to migrate pages out of zones which should not contain any
longterm pinned pages.

When migration succeeds all pages will have been unpinned so pinning
needs to be retried. This is indicated by returning zero. When all pages
are in the correct zone the number of pinned pages is returned.

However migration can also fail, in which case pages are unpinned and
-ENOMEM is returned. However if the failure was due to not being unable
to isolate a page zero is returned. This leads to indefinite looping in
__gup_longterm_locked().

Fix this by simplifying the return codes such that zero indicates all
pages were successfully pinned in the correct zone while errors indicate
either pages were migrated and pinning should be retried or that
migration has failed and therefore the pinning operation should fail.

This fixes the indefinite looping on page isolation failure by failing
the pin operation instead of retrying indefinitely.

Signed-off-by: Alistair Popple <apopple@xxxxxxxxxx>

---

Changes for v2:
- Changed error handling to be move conventional using goto as
suggested by Jason.
- Removed coherent_pages check as it isn't necessary.
---
mm/gup.c | 81 ++++++++++++++++++++++++++++-----------------------------
1 file changed, 41 insertions(+), 40 deletions(-)

diff --git a/mm/gup.c b/mm/gup.c
index 364b274..5707c56 100644
--- a/mm/gup.c
+++ b/mm/gup.c
@@ -1901,20 +1901,24 @@ struct page *get_dump_page(unsigned long addr)

#ifdef CONFIG_MIGRATION
/*
- * Check whether all pages are pinnable, if so return number of pages. If some
- * pages are not pinnable, migrate them, and unpin all pages. Return zero if
- * pages were migrated, or if some pages were not successfully isolated.
- * Return negative error if migration fails.
+ * Check whether all pages are pinnable. If some pages are not pinnable migrate
+ * them and unpin all the pages. Returns -EAGAIN if pages were unpinned or zero
+ * if all pages are pinnable and in the right zone. Other errors indicate
+ * migration failure.
*/
static long check_and_migrate_movable_pages(unsigned long nr_pages,
struct page **pages,
unsigned int gup_flags)
{
- unsigned long isolation_error_count = 0, i;
+ unsigned long i;
struct folio *prev_folio = NULL;
LIST_HEAD(movable_page_list);
- bool drain_allow = true, coherent_pages = false;
- int ret = 0;
+ bool drain_allow = true;
+ int ret = -EAGAIN;
+ struct migration_target_control mtc = {
+ .nid = NUMA_NO_NODE,
+ .gfp_mask = GFP_USER | __GFP_NOWARN,
+ };

for (i = 0; i < nr_pages; i++) {
struct folio *folio = page_folio(pages[i]);
@@ -1935,7 +1939,6 @@ static long check_and_migrate_movable_pages(unsigned long nr_pages,
* pages.
*/
pages[i] = 0;
- coherent_pages = true;

/*
* Migration will fail if the page is pinned, so convert
@@ -1946,10 +1949,10 @@ static long check_and_migrate_movable_pages(unsigned long nr_pages,
unpin_user_page(&folio->page);
}

- ret = migrate_device_coherent_page(&folio->page);
- if (ret)
- goto unpin_pages;
-
+ if (migrate_device_coherent_page(&folio->page)) {
+ ret = -EBUSY;
+ goto error;
+ }
continue;
}

@@ -1960,8 +1963,10 @@ static long check_and_migrate_movable_pages(unsigned long nr_pages,
*/
if (folio_test_hugetlb(folio)) {
if (isolate_hugetlb(&folio->page,
- &movable_page_list))
- isolation_error_count++;
+ &movable_page_list)) {
+ ret = -EBUSY;
+ goto error;
+ }
continue;
}

@@ -1971,28 +1976,26 @@ static long check_and_migrate_movable_pages(unsigned long nr_pages,
}

if (folio_isolate_lru(folio)) {
- isolation_error_count++;
- continue;
+ ret = -EBUSY;
+ goto error;
}
+
list_add_tail(&folio->lru, &movable_page_list);
node_stat_mod_folio(folio,
NR_ISOLATED_ANON + folio_is_file_lru(folio),
folio_nr_pages(folio));
}

- if (!list_empty(&movable_page_list) || isolation_error_count
- || coherent_pages)
- goto unpin_pages;
-
/*
- * If list is empty, and no isolation errors, means that all pages are
- * in the correct zone.
+ * All pages are in the correct zone.
*/
- return nr_pages;
+ if (list_empty(&movable_page_list))
+ return 0;

-unpin_pages:
/*
- * pages[i] might be NULL if any device coherent pages were found.
+ * Unpin all pages. If device coherent pages were found
+ * migrate_deivce_coherent_page() will have already dropped the pin and
+ * set pages[i] == NULL.
*/
for (i = 0; i < nr_pages; i++) {
if (!pages[i])
@@ -2004,21 +2007,19 @@ static long check_and_migrate_movable_pages(unsigned long nr_pages,
put_page(pages[i]);
}

- if (!list_empty(&movable_page_list)) {
- struct migration_target_control mtc = {
- .nid = NUMA_NO_NODE,
- .gfp_mask = GFP_USER | __GFP_NOWARN,
- };
-
- ret = migrate_pages(&movable_page_list, alloc_migration_target,
- NULL, (unsigned long)&mtc, MIGRATE_SYNC,
- MR_LONGTERM_PIN, NULL);
- if (ret > 0) /* number of pages not migrated */
- ret = -ENOMEM;
+ if (migrate_pages(&movable_page_list, alloc_migration_target,
+ NULL, (unsigned long)&mtc, MIGRATE_SYNC,
+ MR_LONGTERM_PIN, NULL)) {
+ ret = -ENOMEM;
+ goto error;
}

- if (ret && !list_empty(&movable_page_list))
+ return -EAGAIN;
+
+error:
+ if (!list_empty(&movable_page_list))
putback_movable_pages(&movable_page_list);
+
return ret;
}
#else
@@ -2026,7 +2027,7 @@ static long check_and_migrate_movable_pages(unsigned long nr_pages,
struct page **pages,
unsigned int gup_flags)
{
- return nr_pages;
+ return 0;
}
#endif /* CONFIG_MIGRATION */

@@ -2054,10 +2055,10 @@ static long __gup_longterm_locked(struct mm_struct *mm,
if (rc <= 0)
break;
rc = check_and_migrate_movable_pages(rc, pages, gup_flags);
- } while (!rc);
+ } while (rc == -EAGAIN);
memalloc_pin_restore(flags);

- return rc;
+ return rc ? rc : nr_pages;
}

static bool is_valid_gup_flags(unsigned int gup_flags)

base-commit: 187e7c41445a0f202bb551f08ca7f8158fea1cd7
--
git-series 0.9.1