[4/8] libceph: fix handling of short returns from get_user_pages

Commit Message

Sage Weil March 3, 2011, 8:40 p.m. UTC

None

Patch

diff --git a/net/ceph/pagevec.c b/net/ceph/pagevec.c
index 1a040e6..b6cf888 100644
--- a/net/ceph/pagevec.c
+++ b/net/ceph/pagevec.c
@@ -16,6 +16,7 @@  struct page **ceph_get_direct_page_vector(const char __user *data,
 					  int num_pages, bool write_page)
 {
 	struct page **pages;
+	int got = 0;
 	int rc;
 
 	pages = kmalloc(sizeof(*pages) * num_pages, GFP_NOFS);
@@ -23,15 +24,22 @@  struct page **ceph_get_direct_page_vector(const char __user *data,
 		return ERR_PTR(-ENOMEM);
 
 	down_read(&current->mm->mmap_sem);
-	rc = get_user_pages(current, current->mm, (unsigned long)data,
-			    num_pages, write_page, 0, pages, NULL);
+	while (got < num_pages) {
+		rc = get_user_pages(current, current->mm,
+		    (unsigned long)data + ((unsigned long)got * PAGE_SIZE),
+		    num_pages - got, write_page, 0, pages + got, NULL);
+		if (rc < 0)
+			break;
+		BUG_ON(rc == 0);
+		got += rc;
+	}
 	up_read(&current->mm->mmap_sem);
-	if (rc < num_pages)
+	if (rc < 0)
 		goto fail;
 	return pages;
 
 fail:
-	ceph_put_page_vector(pages, rc > 0 ? rc : 0, false);
+	ceph_put_page_vector(pages, got, false);
 	return ERR_PTR(rc);
 }
 EXPORT_SYMBOL(ceph_get_direct_page_vector);