[RFC PATCH 4/6] udmabuf: use vmap_range_folios

From: Huan Yang
Date: Thu Mar 27 2025 - 05:31:19 EST


This patch change udmabuf from use pfn based into folio based vmap.

Signed-off-by: Huan Yang <link@xxxxxxxx>
---
drivers/dma-buf/udmabuf.c | 25 ++++++++++++-------------
1 file changed, 12 insertions(+), 13 deletions(-)

diff --git a/drivers/dma-buf/udmabuf.c b/drivers/dma-buf/udmabuf.c
index fbe4b59b4c97..f50e11f0f034 100644
--- a/drivers/dma-buf/udmabuf.c
+++ b/drivers/dma-buf/udmabuf.c
@@ -152,29 +152,28 @@ static void *udmabuf_vmap_pfn(unsigned long *pfns, unsigned int count,
static int vmap_udmabuf(struct dma_buf *buf, struct iosys_map *map)
{
struct udmabuf *ubuf = buf->priv;
- unsigned long *pfns;
+ struct vmap_folios_struct folios;
+ struct vmap_folios_range *ranges;
void *vaddr;
pgoff_t pg;

dma_resv_assert_held(buf->resv);

- /**
- * HVO may free tail pages, so just use pfn to map each folio
- * into vmalloc area.
- */
- pfns = kvmalloc_array(ubuf->pagecount, sizeof(*pfns), GFP_KERNEL);
- if (!pfns)
+ ranges = kvmalloc_array(ubuf->pagecount, sizeof(*ranges), GFP_KERNEL);
+ if (!ranges)
return -ENOMEM;

for (pg = 0; pg < ubuf->pagecount; pg++) {
- unsigned long pfn = folio_pfn(ubuf->folios[pg]);
-
- pfn += ubuf->offsets[pg] >> PAGE_SHIFT;
- pfns[pg] = pfn;
+ ranges[pg].folio = ubuf->folios[pg];
+ ranges[pg].pgoff = ubuf->offsets[pg] >> PAGE_SHIFT;
+ ranges[pg].nr_pages = 1;
}

- vaddr = udmabuf_vmap_pfn(pfns, ubuf->pagecount, PAGE_KERNEL);
- kvfree(pfns);
+ folios._range_folios = ranges;
+ folios.range = true;
+
+ vaddr = vmap_range_folios(&folios, ubuf->pagecount, 0, PAGE_KERNEL);
+ kvfree(ranges);
if (!vaddr)
return -EINVAL;

--
2.48.1