1 files changed, 8 insertions, 16 deletions

diff --git a/drivers/xen/tmem.c b/drivers/xen/tmem.c
index 239738f944ba..945fc4327201 100644
--- a/drivers/xen/tmem.c
+++ b/drivers/xen/tmem.c
@@ -129,21 +129,17 @@ static int xen_tmem_new_pool(struct tmem_pool_uuid uuid,
 /* xen generic tmem ops */
 
 static int xen_tmem_put_page(u32 pool_id, struct tmem_oid oid,
-                             u32 index, unsigned long pfn)
+                             u32 index, struct page *page)
 {
-        unsigned long gmfn = xen_pv_domain() ? pfn_to_mfn(pfn) : pfn;
-
         return xen_tmem_op(TMEM_PUT_PAGE, pool_id, oid, index,
-                gmfn, 0, 0, 0);
+                           xen_page_to_gfn(page), 0, 0, 0);
 }
 
 static int xen_tmem_get_page(u32 pool_id, struct tmem_oid oid,
-                             u32 index, unsigned long pfn)
+                             u32 index, struct page *page)
 {
-        unsigned long gmfn = xen_pv_domain() ? pfn_to_mfn(pfn) : pfn;
-
         return xen_tmem_op(TMEM_GET_PAGE, pool_id, oid, index,
-                gmfn, 0, 0, 0);
+                           xen_page_to_gfn(page), 0, 0, 0);
 }
 
 static int xen_tmem_flush_page(u32 pool_id, struct tmem_oid oid, u32 index)
@@ -173,14 +169,13 @@ static void tmem_cleancache_put_page(int pool, struct cleancache_filekey key,
 {
         u32 ind = (u32) index;
         struct tmem_oid oid = *(struct tmem_oid *)&key;
-        unsigned long pfn = page_to_pfn(page);
 
         if (pool < 0)
                 return;
         if (ind != index)
                 return;
         mb(); /* ensure page is quiescent; tmem may address it with an alias */
-        (void)xen_tmem_put_page((u32)pool, oid, ind, pfn);
+        (void)xen_tmem_put_page((u32)pool, oid, ind, page);
 }
 
 static int tmem_cleancache_get_page(int pool, struct cleancache_filekey key,
@@ -188,7 +183,6 @@ static int tmem_cleancache_get_page(int pool, struct cleancache_filekey key,
 {
         u32 ind = (u32) index;
         struct tmem_oid oid = *(struct tmem_oid *)&key;
-        unsigned long pfn = page_to_pfn(page);
         int ret;
 
         /* translate return values to linux semantics */
@@ -196,7 +190,7 @@ static int tmem_cleancache_get_page(int pool, struct cleancache_filekey key,
                 return -1;
         if (ind != index)
                 return -1;
-        ret = xen_tmem_get_page((u32)pool, oid, ind, pfn);
+        ret = xen_tmem_get_page((u32)pool, oid, ind, page);
         if (ret == 1)
                 return 0;
         else
@@ -287,7 +281,6 @@ static int tmem_frontswap_store(unsigned type, pgoff_t offset,
 {
         u64 ind64 = (u64)offset;
         u32 ind = (u32)offset;
-        unsigned long pfn = page_to_pfn(page);
         int pool = tmem_frontswap_poolid;
         int ret;
 
@@ -296,7 +289,7 @@ static int tmem_frontswap_store(unsigned type, pgoff_t offset,
         if (ind64 != ind)
                 return -1;
         mb(); /* ensure page is quiescent; tmem may address it with an alias */
-        ret = xen_tmem_put_page(pool, oswiz(type, ind), iswiz(ind), pfn);
+        ret = xen_tmem_put_page(pool, oswiz(type, ind), iswiz(ind), page);
         /* translate Xen tmem return values to linux semantics */
         if (ret == 1)
                 return 0;
@@ -313,7 +306,6 @@ static int tmem_frontswap_load(unsigned type, pgoff_t offset,
 {
         u64 ind64 = (u64)offset;
         u32 ind = (u32)offset;
-        unsigned long pfn = page_to_pfn(page);
         int pool = tmem_frontswap_poolid;
         int ret;
 
@@ -321,7 +313,7 @@ static int tmem_frontswap_load(unsigned type, pgoff_t offset,
                 return -1;
         if (ind64 != ind)
                 return -1;
-        ret = xen_tmem_get_page(pool, oswiz(type, ind), iswiz(ind), pfn);
+        ret = xen_tmem_get_page(pool, oswiz(type, ind), iswiz(ind), page);
         /* translate Xen tmem return values to linux semantics */
         if (ret == 1)
                 return 0;