lists.openwall.net   lists  /  announce  owl-users  owl-dev  john-users  john-dev  passwdqc-users  yescrypt  popa3d-users  /  oss-security  kernel-hardening  musl  sabotage  tlsify  passwords  /  crypt-dev  xvendor  /  Bugtraq  Full-Disclosure  linux-kernel  linux-netdev  linux-ext4  linux-hardening  linux-cve-announce  PHC 
Open Source and information security mailing list archives
 
Hash Suite: Windows password security audit tool. GUI, reports in PDF.
[<prev] [next>] [<thread-prev] [thread-next>] [day] [month] [year] [list]
Message-ID: <1626255285-5079-3-git-send-email-linyunsheng@huawei.com>
Date:   Wed, 14 Jul 2021 17:34:43 +0800
From:   Yunsheng Lin <linyunsheng@...wei.com>
To:     <davem@...emloft.net>, <kuba@...nel.org>
CC:     <alexander.duyck@...il.com>, <linux@...linux.org.uk>,
        <mw@...ihalf.com>, <linuxarm@...neuler.org>,
        <yisen.zhuang@...wei.com>, <salil.mehta@...wei.com>,
        <thomas.petazzoni@...tlin.com>, <hawk@...nel.org>,
        <ilias.apalodimas@...aro.org>, <ast@...nel.org>,
        <daniel@...earbox.net>, <john.fastabend@...il.com>,
        <akpm@...ux-foundation.org>, <peterz@...radead.org>,
        <will@...nel.org>, <willy@...radead.org>, <vbabka@...e.cz>,
        <fenghua.yu@...el.com>, <guro@...com>, <peterx@...hat.com>,
        <feng.tang@...el.com>, <jgg@...pe.ca>, <mcroce@...rosoft.com>,
        <hughd@...gle.com>, <jonathan.lemon@...il.com>, <alobakin@...me>,
        <willemb@...gle.com>, <wenxu@...oud.cn>, <cong.wang@...edance.com>,
        <haokexin@...il.com>, <nogikh@...gle.com>, <elver@...gle.com>,
        <yhs@...com>, <kpsingh@...nel.org>, <andrii@...nel.org>,
        <kafai@...com>, <songliubraving@...com>, <netdev@...r.kernel.org>,
        <linux-kernel@...r.kernel.org>, <bpf@...r.kernel.org>
Subject: [PATCH rfc v5 2/4] page_pool: add interface to manipulate frag count in page pool

As suggested by Alexander, "A DMA mapping should be page
aligned anyway so the lower 12 bits would be reserved 0",
so it might make more sense to repurpose the lower 12 bits
of the dma address to store the frag count for frag page
support in page pool for 32 bit systems with 64 bit dma,
which should be rare those days.

For normal system, the dma_addr[1] in 'struct page' is not
used, so we can reuse one of the dma_addr for storing frag
count, which means how many frags this page might be splited
to.

The PAGE_POOL_DMA_USE_PP_FRAG_COUNT macro is added to decide
where to store the frag count, as the "sizeof(dma_addr_t) >
sizeof(unsigned long)" is false for most systems those days,
so hopefully the compiler will optimize out the unused code
for those systems.

The newly added page_pool_set_frag_count() should be called
before the page is passed to any user. Otherwise, call the
newly added page_pool_atomic_sub_frag_count_return().

Signed-off-by: Yunsheng Lin <linyunsheng@...wei.com>
---
 include/linux/mm_types.h |  8 +++++--
 include/net/page_pool.h  | 54 ++++++++++++++++++++++++++++++++++++++++++------
 net/core/page_pool.c     | 10 +++++++++
 3 files changed, 64 insertions(+), 8 deletions(-)

diff --git a/include/linux/mm_types.h b/include/linux/mm_types.h
index d33d97c..82bcbb0 100644
--- a/include/linux/mm_types.h
+++ b/include/linux/mm_types.h
@@ -103,11 +103,15 @@ struct page {
 			unsigned long pp_magic;
 			struct page_pool *pp;
 			unsigned long _pp_mapping_pad;
+			atomic_long_t pp_frag_count;
 			/**
 			 * @dma_addr: might require a 64-bit value on
-			 * 32-bit architectures.
+			 * 32-bit architectures, if so, store the lower 32
+			 * bits in pp_frag_count, and a DMA mapping should
+			 * be page aligned, so the frag count can be stored
+			 * in lower 12 bits for 4K page size.
 			 */
-			unsigned long dma_addr[2];
+			unsigned long dma_addr;
 		};
 		struct {	/* slab, slob and slub */
 			union {
diff --git a/include/net/page_pool.h b/include/net/page_pool.h
index 8d7744d..ef449c2 100644
--- a/include/net/page_pool.h
+++ b/include/net/page_pool.h
@@ -198,19 +198,61 @@ static inline void page_pool_recycle_direct(struct page_pool *pool,
 	page_pool_put_full_page(pool, page, true);
 }
 
+#define PAGE_POOL_DMA_USE_PP_FRAG_COUNT	\
+			(sizeof(dma_addr_t) > sizeof(unsigned long))
+
 static inline dma_addr_t page_pool_get_dma_addr(struct page *page)
 {
-	dma_addr_t ret = page->dma_addr[0];
-	if (sizeof(dma_addr_t) > sizeof(unsigned long))
-		ret |= (dma_addr_t)page->dma_addr[1] << 16 << 16;
+	dma_addr_t ret = page->dma_addr;
+
+	if (PAGE_POOL_DMA_USE_PP_FRAG_COUNT) {
+		ret <<= 32;
+		ret |= atomic_long_read(&page->pp_frag_count) & PAGE_MASK;
+	}
+
 	return ret;
 }
 
 static inline void page_pool_set_dma_addr(struct page *page, dma_addr_t addr)
 {
-	page->dma_addr[0] = addr;
-	if (sizeof(dma_addr_t) > sizeof(unsigned long))
-		page->dma_addr[1] = upper_32_bits(addr);
+	if (PAGE_POOL_DMA_USE_PP_FRAG_COUNT) {
+		atomic_long_set(&page->pp_frag_count, addr & PAGE_MASK);
+		addr >>= 32;
+	}
+
+	page->dma_addr = addr;
+}
+
+static inline long page_pool_atomic_sub_frag_count_return(struct page *page,
+							  long nr)
+{
+	long frag_count = atomic_long_read(&page->pp_frag_count);
+	long ret;
+
+	if (PAGE_POOL_DMA_USE_PP_FRAG_COUNT) {
+		if ((frag_count & ~PAGE_MASK) == nr)
+			return 0;
+
+		ret = atomic_long_sub_return(nr, &page->pp_frag_count);
+		WARN_ON((ret & PAGE_MASK) != (frag_count & PAGE_MASK));
+		ret &= ~PAGE_MASK;
+	} else {
+		if (frag_count == nr)
+			return 0;
+
+		ret = atomic_long_sub_return(nr, &page->pp_frag_count);
+		WARN_ON(ret < 0);
+	}
+
+	return ret;
+}
+
+static inline void page_pool_set_frag_count(struct page *page, long nr)
+{
+	if (PAGE_POOL_DMA_USE_PP_FRAG_COUNT)
+		nr |= atomic_long_read(&page->pp_frag_count) & PAGE_MASK;
+
+	atomic_long_set(&page->pp_frag_count, nr);
 }
 
 static inline bool is_page_pool_compiled_in(void)
diff --git a/net/core/page_pool.c b/net/core/page_pool.c
index 78838c6..0082f33 100644
--- a/net/core/page_pool.c
+++ b/net/core/page_pool.c
@@ -198,6 +198,16 @@ static bool page_pool_dma_map(struct page_pool *pool, struct page *page)
 	if (dma_mapping_error(pool->p.dev, dma))
 		return false;
 
+	if (PAGE_POOL_DMA_USE_PP_FRAG_COUNT &&
+	    WARN_ON(pool->p.flags & PP_FLAG_PAGE_FRAG &&
+		    dma & ~PAGE_MASK)) {
+		dma_unmap_page_attrs(pool->p.dev, dma,
+				     PAGE_SIZE << pool->p.order,
+				     pool->p.dma_dir,
+				     DMA_ATTR_SKIP_CPU_SYNC);
+		return false;
+	}
+
 	page_pool_set_dma_addr(page, dma);
 
 	if (pool->p.flags & PP_FLAG_DMA_SYNC_DEV)
-- 
2.7.4

Powered by blists - more mailing lists

Powered by Openwall GNU/*/Linux Powered by OpenVZ