[CRIU] [PATCH v2 11/11] lazy-pages: refactor uffd_handle_page

[Mike Rapoport]

Inline relevant parts of get_page inside uffd_handle_page and call
uffd_{copy,zero}_page after we've got the data.

Signed-off-by: Mike Rapoport <rppt at linux.vnet.ibm.com>
---
 criu/uffd.c | 56 ++++++++++++++++++++++----------------------------------
 1 file changed, 22 insertions(+), 34 deletions(-)

diff --git a/criu/uffd.c b/criu/uffd.c
index ae40826..c2e2082 100644
--- a/criu/uffd.c
+++ b/criu/uffd.c
@@ -502,41 +502,12 @@ out:
 	return -1;
 }
 
-static int get_page(struct lazy_pages_info *lpi, unsigned long addr, void *dest)
-{
-	int ret;
-
-	lpi->pr.reset(&lpi->pr);
-
-	ret = lpi->pr.seek_page(&lpi->pr, addr, true);
-	pr_debug("seek_pagemap_page ret 0x%x\n", ret);
-	if (ret <= 0)
-		return ret;
-
-	if (pagemap_zero(lpi->pr.pe))
-		return 0;
-
-	ret = lpi->pr.read_pages(&lpi->pr, addr, 1, dest);
-	pr_debug("read_pages ret %d\n", ret);
-	if (ret <= 0)
-		return ret;
-
-	return 1;
-}
-
 static int uffd_copy_page(struct lazy_pages_info *lpi, __u64 address,
 			  void *dest)
 {
 	struct uffdio_copy uffdio_copy;
 	int rc;
 
-	if (opts.use_page_server)
-		rc = get_remote_pages(lpi->pid, address, 1, dest);
-	else
-		rc = get_page(lpi, address, dest);
-	if (rc <= 0)
-		return rc;
-
 	uffdio_copy.dst = address;
 	uffdio_copy.src = (unsigned long) dest;
 	uffdio_copy.len = page_size();
@@ -589,13 +560,30 @@ static int uffd_zero_page(struct lazy_pages_info *lpi, __u64 address)
 static int uffd_handle_page(struct lazy_pages_info *lpi, __u64 address,
 			    void *dest)
 {
-	int rc;
+	int ret;
+
+	lpi->pr.reset(&lpi->pr);
+
+	ret = lpi->pr.seek_page(&lpi->pr, address, false);
+	if (ret < 0) {
+		pr_err("%d: no pagemap containing %llx\n", lpi->pid, address);
+		return ret;
+	}
 
-	rc = uffd_copy_page(lpi, address, dest);
-	if (rc == 0)
-		rc = uffd_zero_page(lpi, address);
+	if (pagemap_zero(lpi->pr.pe))
+		return uffd_zero_page(lpi, address);
+
+	if (opts.use_page_server)
+		ret = get_remote_pages(lpi->pid, address, 1, dest);
+	else
+		ret = lpi->pr.read_pages(&lpi->pr, address, 1, dest);
+
+	if (ret <= 0) {
+		pr_err("%d: failed reading pages at %llx\n", lpi->pid, address);
+		return ret;
+	}
 
-	return rc;
+	return uffd_copy_page(lpi, address, dest);
 }
 
 static int handle_remaining_pages(struct lazy_pages_info *lpi, void *dest)
-- 
1.9.1