xref: /kernel/linux/linux-5.10/sound/core/sgbuf.c (revision 8c2ecf20)
1// SPDX-License-Identifier: GPL-2.0-or-later
2/*
3 * Scatter-Gather buffer
4 *
5 *  Copyright (c) by Takashi Iwai <tiwai@suse.de>
6 */
7
8#include <linux/slab.h>
9#include <linux/mm.h>
10#include <linux/vmalloc.h>
11#include <linux/export.h>
12#include <sound/memalloc.h>
13
14
15/* table entries are align to 32 */
16#define SGBUF_TBL_ALIGN		32
17#define sgbuf_align_table(tbl)	ALIGN((tbl), SGBUF_TBL_ALIGN)
18
19int snd_free_sgbuf_pages(struct snd_dma_buffer *dmab)
20{
21	struct snd_sg_buf *sgbuf = dmab->private_data;
22	struct snd_dma_buffer tmpb;
23	int i;
24
25	if (! sgbuf)
26		return -EINVAL;
27
28	vunmap(dmab->area);
29	dmab->area = NULL;
30
31	tmpb.dev.type = SNDRV_DMA_TYPE_DEV;
32	if (dmab->dev.type == SNDRV_DMA_TYPE_DEV_UC_SG)
33		tmpb.dev.type = SNDRV_DMA_TYPE_DEV_UC;
34	tmpb.dev.dev = sgbuf->dev;
35	for (i = 0; i < sgbuf->pages; i++) {
36		if (!(sgbuf->table[i].addr & ~PAGE_MASK))
37			continue; /* continuous pages */
38		tmpb.area = sgbuf->table[i].buf;
39		tmpb.addr = sgbuf->table[i].addr & PAGE_MASK;
40		tmpb.bytes = (sgbuf->table[i].addr & ~PAGE_MASK) << PAGE_SHIFT;
41		snd_dma_free_pages(&tmpb);
42	}
43
44	kfree(sgbuf->table);
45	kfree(sgbuf->page_table);
46	kfree(sgbuf);
47	dmab->private_data = NULL;
48
49	return 0;
50}
51
52#define MAX_ALLOC_PAGES		32
53
54void *snd_malloc_sgbuf_pages(struct device *device,
55			     size_t size, struct snd_dma_buffer *dmab,
56			     size_t *res_size)
57{
58	struct snd_sg_buf *sgbuf;
59	unsigned int i, pages, chunk, maxpages;
60	struct snd_dma_buffer tmpb;
61	struct snd_sg_page *table;
62	struct page **pgtable;
63	int type = SNDRV_DMA_TYPE_DEV;
64	pgprot_t prot = PAGE_KERNEL;
65
66	dmab->area = NULL;
67	dmab->addr = 0;
68	dmab->private_data = sgbuf = kzalloc(sizeof(*sgbuf), GFP_KERNEL);
69	if (! sgbuf)
70		return NULL;
71	if (dmab->dev.type == SNDRV_DMA_TYPE_DEV_UC_SG) {
72		type = SNDRV_DMA_TYPE_DEV_UC;
73#ifdef pgprot_noncached
74		prot = pgprot_noncached(PAGE_KERNEL);
75#endif
76	}
77	sgbuf->dev = device;
78	pages = snd_sgbuf_aligned_pages(size);
79	sgbuf->tblsize = sgbuf_align_table(pages);
80	table = kcalloc(sgbuf->tblsize, sizeof(*table), GFP_KERNEL);
81	if (!table)
82		goto _failed;
83	sgbuf->table = table;
84	pgtable = kcalloc(sgbuf->tblsize, sizeof(*pgtable), GFP_KERNEL);
85	if (!pgtable)
86		goto _failed;
87	sgbuf->page_table = pgtable;
88
89	/* allocate pages */
90	maxpages = MAX_ALLOC_PAGES;
91	while (pages > 0) {
92		chunk = pages;
93		/* don't be too eager to take a huge chunk */
94		if (chunk > maxpages)
95			chunk = maxpages;
96		chunk <<= PAGE_SHIFT;
97		if (snd_dma_alloc_pages_fallback(type, device,
98						 chunk, &tmpb) < 0) {
99			if (!sgbuf->pages)
100				goto _failed;
101			if (!res_size)
102				goto _failed;
103			size = sgbuf->pages * PAGE_SIZE;
104			break;
105		}
106		chunk = tmpb.bytes >> PAGE_SHIFT;
107		for (i = 0; i < chunk; i++) {
108			table->buf = tmpb.area;
109			table->addr = tmpb.addr;
110			if (!i)
111				table->addr |= chunk; /* mark head */
112			table++;
113			*pgtable++ = virt_to_page(tmpb.area);
114			tmpb.area += PAGE_SIZE;
115			tmpb.addr += PAGE_SIZE;
116		}
117		sgbuf->pages += chunk;
118		pages -= chunk;
119		if (chunk < maxpages)
120			maxpages = chunk;
121	}
122
123	sgbuf->size = size;
124	dmab->area = vmap(sgbuf->page_table, sgbuf->pages, VM_MAP, prot);
125	if (! dmab->area)
126		goto _failed;
127	if (res_size)
128		*res_size = sgbuf->size;
129	return dmab->area;
130
131 _failed:
132	snd_free_sgbuf_pages(dmab); /* free the table */
133	return NULL;
134}
135
136/*
137 * compute the max chunk size with continuous pages on sg-buffer
138 */
139unsigned int snd_sgbuf_get_chunk_size(struct snd_dma_buffer *dmab,
140				      unsigned int ofs, unsigned int size)
141{
142	struct snd_sg_buf *sg = dmab->private_data;
143	unsigned int start, end, pg;
144
145	if (!sg)
146		return size;
147
148	start = ofs >> PAGE_SHIFT;
149	end = (ofs + size - 1) >> PAGE_SHIFT;
150	/* check page continuity */
151	pg = sg->table[start].addr >> PAGE_SHIFT;
152	for (;;) {
153		start++;
154		if (start > end)
155			break;
156		pg++;
157		if ((sg->table[start].addr >> PAGE_SHIFT) != pg)
158			return (start << PAGE_SHIFT) - ofs;
159	}
160	/* ok, all on continuous pages */
161	return size;
162}
163EXPORT_SYMBOL(snd_sgbuf_get_chunk_size);
164