Create page_pool_alloc_netmem to be the mirror of page_pool_alloc.

This enables drivers that want currently use page_pool_alloc to
transition to netmem by converting the call sites to
page_pool_alloc_netmem.

Signed-off-by: Mina Almasry <almasrym...@google.com>
---
 include/net/page_pool/helpers.h | 25 ++++++++++++++++---------
 1 file changed, 16 insertions(+), 9 deletions(-)

diff --git a/include/net/page_pool/helpers.h b/include/net/page_pool/helpers.h
index 793e6fd78bc5..8e548ff3044c 100644
--- a/include/net/page_pool/helpers.h
+++ b/include/net/page_pool/helpers.h
@@ -116,22 +116,22 @@ static inline struct page 
*page_pool_dev_alloc_frag(struct page_pool *pool,
        return page_pool_alloc_frag(pool, offset, size, gfp);
 }
 
-static inline struct page *page_pool_alloc(struct page_pool *pool,
-                                          unsigned int *offset,
-                                          unsigned int *size, gfp_t gfp)
+static inline netmem_ref page_pool_alloc_netmem(struct page_pool *pool,
+                                               unsigned int *offset,
+                                               unsigned int *size, gfp_t gfp)
 {
        unsigned int max_size = PAGE_SIZE << pool->p.order;
-       struct page *page;
+       netmem_ref netmem;
 
        if ((*size << 1) > max_size) {
                *size = max_size;
                *offset = 0;
-               return page_pool_alloc_pages(pool, gfp);
+               return page_pool_alloc_netmems(pool, gfp);
        }
 
-       page = page_pool_alloc_frag(pool, offset, *size, gfp);
-       if (unlikely(!page))
-               return NULL;
+       netmem = page_pool_alloc_frag_netmem(pool, offset, *size, gfp);
+       if (unlikely(!netmem))
+               return 0;
 
        /* There is very likely not enough space for another fragment, so append
         * the remaining size to the current fragment to avoid truesize
@@ -142,7 +142,14 @@ static inline struct page *page_pool_alloc(struct 
page_pool *pool,
                pool->frag_offset = max_size;
        }
 
-       return page;
+       return netmem;
+}
+
+static inline struct page *page_pool_alloc(struct page_pool *pool,
+                                          unsigned int *offset,
+                                          unsigned int *size, gfp_t gfp)
+{
+       return netmem_to_page(page_pool_alloc_netmem(pool, offset, size, gfp));
 }
 
 /**
-- 
2.47.0.163.g1226f6d8fa-goog


Reply via email to