[mirror_ubuntu-jammy-kernel.git] / sound / core / memalloc.c

/*
 *  Copyright (c) by Jaroslav Kysela <perex@perex.cz>
 *                   Takashi Iwai <tiwai@suse.de>
 * 
 *  Generic memory allocators
 *
 *
 *   This program is free software; you can redistribute it and/or modify
 *   it under the terms of the GNU General Public License as published by
 *   the Free Software Foundation; either version 2 of the License, or
 *   (at your option) any later version.
 *
 *   This program is distributed in the hope that it will be useful,
 *   but WITHOUT ANY WARRANTY; without even the implied warranty of
 *   MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 *   GNU General Public License for more details.
 *
 *   You should have received a copy of the GNU General Public License
 *   along with this program; if not, write to the Free Software
 *   Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307 USA
 *
 */

#include <linux/module.h>
#include <linux/proc_fs.h>
#include <linux/init.h>
#include <linux/pci.h>
#include <linux/slab.h>
#include <linux/mm.h>
#include <linux/seq_file.h>
#include <asm/uaccess.h>
#include <linux/dma-mapping.h>
#include <linux/genalloc.h>
#include <linux/moduleparam.h>
#include <linux/mutex.h>
#include <sound/memalloc.h>


MODULE_AUTHOR("Takashi Iwai <tiwai@suse.de>, Jaroslav Kysela <perex@perex.cz>");
MODULE_DESCRIPTION("Memory allocator for ALSA system.");
MODULE_LICENSE("GPL");


/*
 */

static DEFINE_MUTEX(list_mutex);
static LIST_HEAD(mem_list_head);

/* buffer preservation list */
struct snd_mem_list {
	struct snd_dma_buffer buffer;
	unsigned int id;
	struct list_head list;
};

/* id for pre-allocated buffers */
#define SNDRV_DMA_DEVICE_UNUSED (unsigned int)-1

/*
 *
 *  Generic memory allocators
 *
 */

static long snd_allocated_pages; /* holding the number of allocated pages */

static inline void inc_snd_pages(int order)
{
	snd_allocated_pages += 1 << order;
}

static inline void dec_snd_pages(int order)
{
	snd_allocated_pages -= 1 << order;
}

/**
 * snd_malloc_pages - allocate pages with the given size
 * @size: the size to allocate in bytes
 * @gfp_flags: the allocation conditions, GFP_XXX
 *
 * Allocates the physically contiguous pages with the given size.
 *
 * Return: The pointer of the buffer, or %NULL if no enough memory.
 */
void *snd_malloc_pages(size_t size, gfp_t gfp_flags)
{
	int pg;
	void *res;

	if (WARN_ON(!size))
		return NULL;
	if (WARN_ON(!gfp_flags))
		return NULL;
	gfp_flags |= __GFP_COMP;	/* compound page lets parts be mapped */
	pg = get_order(size);
	if ((res = (void *) __get_free_pages(gfp_flags, pg)) != NULL)
		inc_snd_pages(pg);
	return res;
}

/**
 * snd_free_pages - release the pages
 * @ptr: the buffer pointer to release
 * @size: the allocated buffer size
 *
 * Releases the buffer allocated via snd_malloc_pages().
 */
void snd_free_pages(void *ptr, size_t size)
{
	int pg;

	if (ptr == NULL)
		return;
	pg = get_order(size);
	dec_snd_pages(pg);
	free_pages((unsigned long) ptr, pg);
}

/*
 *
 *  Bus-specific memory allocators
 *
 */

#ifdef CONFIG_HAS_DMA
/* allocate the coherent DMA pages */
static void *snd_malloc_dev_pages(struct device *dev, size_t size, dma_addr_t *dma)
{
	int pg;
	void *res;
	gfp_t gfp_flags;

	if (WARN_ON(!dma))
		return NULL;
	pg = get_order(size);
	gfp_flags = GFP_KERNEL
		| __GFP_COMP	/* compound page lets parts be mapped */
		| __GFP_NORETRY /* don't trigger OOM-killer */
		| __GFP_NOWARN; /* no stack trace print - this call is non-critical */
	res = dma_alloc_coherent(dev, PAGE_SIZE << pg, dma, gfp_flags);
	if (res != NULL)
		inc_snd_pages(pg);

	return res;
}

/* free the coherent DMA pages */
static void snd_free_dev_pages(struct device *dev, size_t size, void *ptr,
			       dma_addr_t dma)
{
	int pg;

	if (ptr == NULL)
		return;
	pg = get_order(size);
	dec_snd_pages(pg);
	dma_free_coherent(dev, PAGE_SIZE << pg, ptr, dma);
}

#ifdef CONFIG_GENERIC_ALLOCATOR
/**
 * snd_malloc_dev_iram - allocate memory from on-chip internal ram
 * @dmab: buffer allocation record to store the allocated data
 * @size: number of bytes to allocate from the iram
 *
 * This function requires iram phandle provided via of_node
 */
static void snd_malloc_dev_iram(struct snd_dma_buffer *dmab, size_t size)
{
	struct device *dev = dmab->dev.dev;
	struct gen_pool *pool = NULL;

	dmab->area = NULL;
	dmab->addr = 0;

	if (dev->of_node)
		pool = of_get_named_gen_pool(dev->of_node, "iram", 0);

	if (!pool)
		return;

	/* Assign the pool into private_data field */
	dmab->private_data = pool;

	dmab->area = gen_pool_dma_alloc(pool, size, &dmab->addr);
}

/**
 * snd_free_dev_iram - free allocated specific memory from on-chip internal ram
 * @dmab: buffer allocation record to store the allocated data
 */
static void snd_free_dev_iram(struct snd_dma_buffer *dmab)
{
	struct gen_pool *pool = dmab->private_data;

	if (pool && dmab->area)
		gen_pool_free(pool, (unsigned long)dmab->area, dmab->bytes);
}
#endif /* CONFIG_GENERIC_ALLOCATOR */
#endif /* CONFIG_HAS_DMA */

/*
 *
 *  ALSA generic memory management
 *
 */


/**
 * snd_dma_alloc_pages - allocate the buffer area according to the given type
 * @type: the DMA buffer type
 * @device: the device pointer
 * @size: the buffer size to allocate
 * @dmab: buffer allocation record to store the allocated data
 *
 * Calls the memory-allocator function for the corresponding
 * buffer type.
 *
 * Return: Zero if the buffer with the given size is allocated successfully,
 * otherwise a negative value on error.
 */
int snd_dma_alloc_pages(int type, struct device *device, size_t size,
			struct snd_dma_buffer *dmab)
{
	if (WARN_ON(!size))
		return -ENXIO;
	if (WARN_ON(!dmab))
		return -ENXIO;

	dmab->dev.type = type;
	dmab->dev.dev = device;
	dmab->bytes = 0;
	switch (type) {
	case SNDRV_DMA_TYPE_CONTINUOUS:
		dmab->area = snd_malloc_pages(size,
					(__force gfp_t)(unsigned long)device);
		dmab->addr = 0;
		break;
#ifdef CONFIG_HAS_DMA
#ifdef CONFIG_GENERIC_ALLOCATOR
	case SNDRV_DMA_TYPE_DEV_IRAM:
		snd_malloc_dev_iram(dmab, size);
		if (dmab->area)
			break;
		/* Internal memory might have limited size and no enough space,
		 * so if we fail to malloc, try to fetch memory traditionally.
		 */
		dmab->dev.type = SNDRV_DMA_TYPE_DEV;
#endif /* CONFIG_GENERIC_ALLOCATOR */
	case SNDRV_DMA_TYPE_DEV:
		dmab->area = snd_malloc_dev_pages(device, size, &dmab->addr);
		break;
#endif
#ifdef CONFIG_SND_DMA_SGBUF
	case SNDRV_DMA_TYPE_DEV_SG:
		snd_malloc_sgbuf_pages(device, size, dmab, NULL);
		break;
#endif
	default:
		printk(KERN_ERR "snd-malloc: invalid device type %d\n", type);
		dmab->area = NULL;
		dmab->addr = 0;
		return -ENXIO;
	}
	if (! dmab->area)
		return -ENOMEM;
	dmab->bytes = size;
	return 0;
}

/**
 * snd_dma_alloc_pages_fallback - allocate the buffer area according to the given type with fallback
 * @type: the DMA buffer type
 * @device: the device pointer
 * @size: the buffer size to allocate
 * @dmab: buffer allocation record to store the allocated data
 *
 * Calls the memory-allocator function for the corresponding
 * buffer type.  When no space is left, this function reduces the size and
 * tries to allocate again.  The size actually allocated is stored in
 * res_size argument.
 *
 * Return: Zero if the buffer with the given size is allocated successfully,
 * otherwise a negative value on error.
 */
int snd_dma_alloc_pages_fallback(int type, struct device *device, size_t size,
				 struct snd_dma_buffer *dmab)
{
	int err;

	while ((err = snd_dma_alloc_pages(type, device, size, dmab)) < 0) {
		size_t aligned_size;
		if (err != -ENOMEM)
			return err;
		if (size <= PAGE_SIZE)
			return -ENOMEM;
		aligned_size = PAGE_SIZE << get_order(size);
		if (size != aligned_size)
			size = aligned_size;
		else
			size >>= 1;
	}
	if (! dmab->area)
		return -ENOMEM;
	return 0;
}


/**
 * snd_dma_free_pages - release the allocated buffer
 * @dmab: the buffer allocation record to release
 *
 * Releases the allocated buffer via snd_dma_alloc_pages().
 */
void snd_dma_free_pages(struct snd_dma_buffer *dmab)
{
	switch (dmab->dev.type) {
	case SNDRV_DMA_TYPE_CONTINUOUS:
		snd_free_pages(dmab->area, dmab->bytes);
		break;
#ifdef CONFIG_HAS_DMA
#ifdef CONFIG_GENERIC_ALLOCATOR
	case SNDRV_DMA_TYPE_DEV_IRAM:
		snd_free_dev_iram(dmab);
		break;
#endif /* CONFIG_GENERIC_ALLOCATOR */
	case SNDRV_DMA_TYPE_DEV:
		snd_free_dev_pages(dmab->dev.dev, dmab->bytes, dmab->area, dmab->addr);
		break;
#endif
#ifdef CONFIG_SND_DMA_SGBUF
	case SNDRV_DMA_TYPE_DEV_SG:
		snd_free_sgbuf_pages(dmab);
		break;
#endif
	default:
		printk(KERN_ERR "snd-malloc: invalid device type %d\n", dmab->dev.type);
	}
}


/**
 * snd_dma_get_reserved - get the reserved buffer for the given device
 * @dmab: the buffer allocation record to store
 * @id: the buffer id
 *
 * Looks for the reserved-buffer list and re-uses if the same buffer
 * is found in the list.  When the buffer is found, it's removed from the free list.
 *
 * Return: The size of buffer if the buffer is found, or zero if not found.
 */
size_t snd_dma_get_reserved_buf(struct snd_dma_buffer *dmab, unsigned int id)
{
	struct snd_mem_list *mem;

	if (WARN_ON(!dmab))
		return 0;

	mutex_lock(&list_mutex);
	list_for_each_entry(mem, &mem_list_head, list) {
		if (mem->id == id &&
		    (mem->buffer.dev.dev == NULL || dmab->dev.dev == NULL ||
		     ! memcmp(&mem->buffer.dev, &dmab->dev, sizeof(dmab->dev)))) {
			struct device *dev = dmab->dev.dev;
			list_del(&mem->list);
			*dmab = mem->buffer;
			if (dmab->dev.dev == NULL)
				dmab->dev.dev = dev;
			kfree(mem);
			mutex_unlock(&list_mutex);
			return dmab->bytes;
		}
	}
	mutex_unlock(&list_mutex);
	return 0;
}

/**
 * snd_dma_reserve_buf - reserve the buffer
 * @dmab: the buffer to reserve
 * @id: the buffer id
 *
 * Reserves the given buffer as a reserved buffer.
 *
 * Return: Zero if successful, or a negative code on error.
 */
int snd_dma_reserve_buf(struct snd_dma_buffer *dmab, unsigned int id)
{
	struct snd_mem_list *mem;

	if (WARN_ON(!dmab))
		return -EINVAL;
	mem = kmalloc(sizeof(*mem), GFP_KERNEL);
	if (! mem)
		return -ENOMEM;
	mutex_lock(&list_mutex);
	mem->buffer = *dmab;
	mem->id = id;
	list_add_tail(&mem->list, &mem_list_head);
	mutex_unlock(&list_mutex);
	return 0;
}

/*
 * purge all reserved buffers
 */
static void free_all_reserved_pages(void)
{
	struct list_head *p;
	struct snd_mem_list *mem;

	mutex_lock(&list_mutex);
	while (! list_empty(&mem_list_head)) {
		p = mem_list_head.next;
		mem = list_entry(p, struct snd_mem_list, list);
		list_del(p);
		snd_dma_free_pages(&mem->buffer);
		kfree(mem);
	}
	mutex_unlock(&list_mutex);
}


#ifdef CONFIG_PROC_FS
/*
 * proc file interface
 */
#define SND_MEM_PROC_FILE	"driver/snd-page-alloc"
static struct proc_dir_entry *snd_mem_proc;

static int snd_mem_proc_read(struct seq_file *seq, void *offset)
{
	long pages = snd_allocated_pages >> (PAGE_SHIFT-12);
	struct snd_mem_list *mem;
	int devno;
	static char *types[] = { "UNKNOWN", "CONT", "DEV", "DEV-SG" };

	mutex_lock(&list_mutex);
	seq_printf(seq, "pages  : %li bytes (%li pages per %likB)\n",
		   pages * PAGE_SIZE, pages, PAGE_SIZE / 1024);
	devno = 0;
	list_for_each_entry(mem, &mem_list_head, list) {
		devno++;
		seq_printf(seq, "buffer %d : ID %08x : type %s\n",
			   devno, mem->id, types[mem->buffer.dev.type]);
		seq_printf(seq, "  addr = 0x%lx, size = %d bytes\n",
			   (unsigned long)mem->buffer.addr,
			   (int)mem->buffer.bytes);
	}
	mutex_unlock(&list_mutex);
	return 0;
}

static int snd_mem_proc_open(struct inode *inode, struct file *file)
{
	return single_open(file, snd_mem_proc_read, NULL);
}

/* FIXME: for pci only - other bus? */
#ifdef CONFIG_PCI
#define gettoken(bufp) strsep(bufp, " \t\n")

static ssize_t snd_mem_proc_write(struct file *file, const char __user * buffer,
				  size_t count, loff_t * ppos)
{
	char buf[128];
	char *token, *p;

	if (count > sizeof(buf) - 1)
		return -EINVAL;
	if (copy_from_user(buf, buffer, count))
		return -EFAULT;
	buf[count] = '\0';

	p = buf;
	token = gettoken(&p);
	if (! token || *token == '#')
		return count;
	if (strcmp(token, "add") == 0) {
		char *endp;
		int vendor, device, size, buffers;
		long mask;
		int i, alloced;
		struct pci_dev *pci;

		if ((token = gettoken(&p)) == NULL ||
		    (vendor = simple_strtol(token, NULL, 0)) <= 0 ||
		    (token = gettoken(&p)) == NULL ||
		    (device = simple_strtol(token, NULL, 0)) <= 0 ||
		    (token = gettoken(&p)) == NULL ||
		    (mask = simple_strtol(token, NULL, 0)) < 0 ||
		    (token = gettoken(&p)) == NULL ||
		    (size = memparse(token, &endp)) < 64*1024 ||
		    size > 16*1024*1024 /* too big */ ||
		    (token = gettoken(&p)) == NULL ||
		    (buffers = simple_strtol(token, NULL, 0)) <= 0 ||
		    buffers > 4) {
			printk(KERN_ERR "snd-page-alloc: invalid proc write format\n");
			return count;
		}
		vendor &= 0xffff;
		device &= 0xffff;

		alloced = 0;
		pci = NULL;
		while ((pci = pci_get_device(vendor, device, pci)) != NULL) {
			if (mask > 0 && mask < 0xffffffff) {
				if (pci_set_dma_mask(pci, mask) < 0 ||
				    pci_set_consistent_dma_mask(pci, mask) < 0) {
					printk(KERN_ERR "snd-page-alloc: cannot set DMA mask %lx for pci %04x:%04x\n", mask, vendor, device);
					pci_dev_put(pci);
					return count;
				}
			}
			for (i = 0; i < buffers; i++) {
				struct snd_dma_buffer dmab;
				memset(&dmab, 0, sizeof(dmab));
				if (snd_dma_alloc_pages(SNDRV_DMA_TYPE_DEV, snd_dma_pci_data(pci),
							size, &dmab) < 0) {
					printk(KERN_ERR "snd-page-alloc: cannot allocate buffer pages (size = %d)\n", size);
					pci_dev_put(pci);
					return count;
				}
				snd_dma_reserve_buf(&dmab, snd_dma_pci_buf_id(pci));
			}
			alloced++;
		}
		if (! alloced) {
			for (i = 0; i < buffers; i++) {
				struct snd_dma_buffer dmab;
				memset(&dmab, 0, sizeof(dmab));
				/* FIXME: We can allocate only in ZONE_DMA
				 * without a device pointer!
				 */
				if (snd_dma_alloc_pages(SNDRV_DMA_TYPE_DEV, NULL,
							size, &dmab) < 0) {
					printk(KERN_ERR "snd-page-alloc: cannot allocate buffer pages (size = %d)\n", size);
					break;
				}
				snd_dma_reserve_buf(&dmab, (unsigned int)((vendor << 16) | device));
			}
		}
	} else if (strcmp(token, "erase") == 0)
		/* FIXME: need for releasing each buffer chunk? */
		free_all_reserved_pages();
	else
		printk(KERN_ERR "snd-page-alloc: invalid proc cmd\n");
	return count;
}
#endif /* CONFIG_PCI */

static const struct file_operations snd_mem_proc_fops = {
	.owner		= THIS_MODULE,
	.open		= snd_mem_proc_open,
	.read		= seq_read,
#ifdef CONFIG_PCI
	.write		= snd_mem_proc_write,
#endif
	.llseek		= seq_lseek,
	.release	= single_release,
};

#endif /* CONFIG_PROC_FS */

/*
 * module entry
 */

static int __init snd_mem_init(void)
{
#ifdef CONFIG_PROC_FS
	snd_mem_proc = proc_create(SND_MEM_PROC_FILE, 0644, NULL,
				   &snd_mem_proc_fops);
#endif
	return 0;
}

static void __exit snd_mem_exit(void)
{
	remove_proc_entry(SND_MEM_PROC_FILE, NULL);
	free_all_reserved_pages();
	if (snd_allocated_pages > 0)
		printk(KERN_ERR "snd-malloc: Memory leak?  pages not freed = %li\n", snd_allocated_pages);
}


module_init(snd_mem_init)
module_exit(snd_mem_exit)


/*
 * exports
 */
EXPORT_SYMBOL(snd_dma_alloc_pages);
EXPORT_SYMBOL(snd_dma_alloc_pages_fallback);
EXPORT_SYMBOL(snd_dma_free_pages);

EXPORT_SYMBOL(snd_dma_get_reserved_buf);
EXPORT_SYMBOL(snd_dma_reserve_buf);

EXPORT_SYMBOL(snd_malloc_pages);
EXPORT_SYMBOL(snd_free_pages);
Commit	Line	Data
1da177e4	1	/*
c1017a4c	2	* Copyright (c) by Jaroslav Kysela <perex@perex.cz>
1da177e4 LT	3	* Takashi Iwai <tiwai@suse.de>
	4	*
	5	* Generic memory allocators
	6	*
	7	*
	8	* This program is free software; you can redistribute it and/or modify
	9	* it under the terms of the GNU General Public License as published by
	10	* the Free Software Foundation; either version 2 of the License, or
	11	* (at your option) any later version.
	12	*
	13	* This program is distributed in the hope that it will be useful,
	14	* but WITHOUT ANY WARRANTY; without even the implied warranty of
	15	* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
	16	* GNU General Public License for more details.
	17	*
	18	* You should have received a copy of the GNU General Public License
	19	* along with this program; if not, write to the Free Software
	20	* Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
	21	*
	22	*/
	23
1da177e4 LT	24	#include <linux/module.h>
	25	#include <linux/proc_fs.h>
	26	#include <linux/init.h>
	27	#include <linux/pci.h>
	28	#include <linux/slab.h>
	29	#include <linux/mm.h>
ccec6e2c	30	#include <linux/seq_file.h>
b6a96915	31	#include <asm/uaccess.h>
1da177e4	32	#include <linux/dma-mapping.h>
05503214	33	#include <linux/genalloc.h>
1da177e4	34	#include <linux/moduleparam.h>
1a60d4c5	35	#include <linux/mutex.h>
1da177e4	36	#include <sound/memalloc.h>
1da177e4 LT	37
1da177e4 LT	38
c1017a4c	39	MODULE_AUTHOR("Takashi Iwai <tiwai@suse.de>, Jaroslav Kysela <perex@perex.cz>");
1da177e4 LT	40	MODULE_DESCRIPTION("Memory allocator for ALSA system.");
	41	MODULE_LICENSE("GPL");
	42
	43
1da177e4 LT	44	/*
	45	*/
	46
1a60d4c5	47	static DEFINE_MUTEX(list_mutex);
1da177e4 LT	48	static LIST_HEAD(mem_list_head);
	49
	50	/* buffer preservation list */
	51	struct snd_mem_list {
	52	struct snd_dma_buffer buffer;
	53	unsigned int id;
	54	struct list_head list;
	55	};
	56
	57	/* id for pre-allocated buffers */
	58	#define SNDRV_DMA_DEVICE_UNUSED (unsigned int)-1
	59
1da177e4 LT	60	/*
	61	*
	62	* Generic memory allocators
	63	*
	64	*/
	65
	66	static long snd_allocated_pages; /* holding the number of allocated pages */
	67
	68	static inline void inc_snd_pages(int order)
	69	{
	70	snd_allocated_pages += 1 << order;
	71	}
	72
	73	static inline void dec_snd_pages(int order)
	74	{
	75	snd_allocated_pages -= 1 << order;
	76	}
	77
1da177e4 LT	78	/**
	79	* snd_malloc_pages - allocate pages with the given size
	80	* @size: the size to allocate in bytes
	81	* @gfp_flags: the allocation conditions, GFP_XXX
	82	*
	83	* Allocates the physically contiguous pages with the given size.
	84	*
eb7c06e8	85	* Return: The pointer of the buffer, or %NULL if no enough memory.
1da177e4	86	*/
1ef64e67	87	void *snd_malloc_pages(size_t size, gfp_t gfp_flags)
1da177e4 LT	88	{
	89	int pg;
	90	void *res;
	91
7eaa943c TI	92	if (WARN_ON(!size))
	93	return NULL;
	94	if (WARN_ON(!gfp_flags))
	95	return NULL;
f3d48f03	96	gfp_flags \|= __GFP_COMP; /* compound page lets parts be mapped */
1da177e4	97	pg = get_order(size);
2ba8c15c	98	if ((res = (void *) __get_free_pages(gfp_flags, pg)) != NULL)
1da177e4	99	inc_snd_pages(pg);
1da177e4 LT	100	return res;
	101	}
	102
	103	/**
	104	* snd_free_pages - release the pages
	105	* @ptr: the buffer pointer to release
	106	* @size: the allocated buffer size
	107	*
	108	* Releases the buffer allocated via snd_malloc_pages().
	109	*/
	110	void snd_free_pages(void *ptr, size_t size)
	111	{
	112	int pg;
	113
	114	if (ptr == NULL)
	115	return;
	116	pg = get_order(size);
	117	dec_snd_pages(pg);
1da177e4 LT	118	free_pages((unsigned long) ptr, pg);
	119	}
	120
	121	/*
	122	*
	123	* Bus-specific memory allocators
	124	*
	125	*/
	126
8f11551b	127	#ifdef CONFIG_HAS_DMA
1da177e4 LT	128	/* allocate the coherent DMA pages */
	129	static void snd_malloc_dev_pages(struct device dev, size_t size, dma_addr_t *dma)
	130	{
	131	int pg;
	132	void *res;
1ef64e67	133	gfp_t gfp_flags;
1da177e4	134
7eaa943c TI	135	if (WARN_ON(!dma))
7eaa943c TI	136	return NULL;
1da177e4 LT	137	pg = get_order(size);
1da177e4 LT	138	gfp_flags = GFP_KERNEL
f3d48f03	139	\| __GFP_COMP /* compound page lets parts be mapped */
1da177e4 LT	140	\| __GFP_NORETRY /* don't trigger OOM-killer */
	141	\| __GFP_NOWARN; /* no stack trace print - this call is non-critical */
	142	res = dma_alloc_coherent(dev, PAGE_SIZE << pg, dma, gfp_flags);
2ba8c15c	143	if (res != NULL)
1da177e4	144	inc_snd_pages(pg);
1da177e4 LT	145
	146	return res;
	147	}
	148
	149	/* free the coherent DMA pages */
	150	static void snd_free_dev_pages(struct device dev, size_t size, void ptr,
	151	dma_addr_t dma)
	152	{
	153	int pg;
	154
	155	if (ptr == NULL)
	156	return;
	157	pg = get_order(size);
	158	dec_snd_pages(pg);
1da177e4 LT	159	dma_free_coherent(dev, PAGE_SIZE << pg, ptr, dma);
1da177e4 LT	160	}
05503214	161
63437313	162	#ifdef CONFIG_GENERIC_ALLOCATOR
05503214 NC	163	/**
	164	* snd_malloc_dev_iram - allocate memory from on-chip internal ram
	165	* @dmab: buffer allocation record to store the allocated data
	166	* @size: number of bytes to allocate from the iram
	167	*
	168	* This function requires iram phandle provided via of_node
	169	*/
9f694bc7	170	static void snd_malloc_dev_iram(struct snd_dma_buffer *dmab, size_t size)
05503214 NC	171	{
	172	struct device *dev = dmab->dev.dev;
	173	struct gen_pool *pool = NULL;
	174
a40a3937 TI	175	dmab->area = NULL;
	176	dmab->addr = 0;
	177
05503214 NC	178	if (dev->of_node)
	179	pool = of_get_named_gen_pool(dev->of_node, "iram", 0);
	180
	181	if (!pool)
	182	return;
	183
	184	/* Assign the pool into private_data field */
	185	dmab->private_data = pool;
	186
07968fe4	187	dmab->area = gen_pool_dma_alloc(pool, size, &dmab->addr);
05503214 NC	188	}
	189
	190	/**
	191	* snd_free_dev_iram - free allocated specific memory from on-chip internal ram
	192	* @dmab: buffer allocation record to store the allocated data
	193	*/
9f694bc7	194	static void snd_free_dev_iram(struct snd_dma_buffer *dmab)
05503214 NC	195	{
	196	struct gen_pool *pool = dmab->private_data;
	197
	198	if (pool && dmab->area)
	199	gen_pool_free(pool, (unsigned long)dmab->area, dmab->bytes);
	200	}
63437313	201	#endif /* CONFIG_GENERIC_ALLOCATOR */
8f11551b	202	#endif /* CONFIG_HAS_DMA */
1da177e4	203
1da177e4 LT	204	/*
	205	*
	206	* ALSA generic memory management
	207	*
	208	*/
	209
	210
	211	/**
	212	* snd_dma_alloc_pages - allocate the buffer area according to the given type
	213	* @type: the DMA buffer type
	214	* @device: the device pointer
	215	* @size: the buffer size to allocate
	216	* @dmab: buffer allocation record to store the allocated data
	217	*
	218	* Calls the memory-allocator function for the corresponding
	219	* buffer type.
eb7c06e8 YB	220	*
	221	* Return: Zero if the buffer with the given size is allocated successfully,
	222	* otherwise a negative value on error.
1da177e4 LT	223	*/
	224	int snd_dma_alloc_pages(int type, struct device *device, size_t size,
	225	struct snd_dma_buffer *dmab)
	226	{
7eaa943c TI	227	if (WARN_ON(!size))
	228	return -ENXIO;
	229	if (WARN_ON(!dmab))
	230	return -ENXIO;
1da177e4 LT	231
	232	dmab->dev.type = type;
	233	dmab->dev.dev = device;
	234	dmab->bytes = 0;
	235	switch (type) {
	236	case SNDRV_DMA_TYPE_CONTINUOUS:
fea952e5 CL	237	dmab->area = snd_malloc_pages(size,
fea952e5 CL	238	(__force gfp_t)(unsigned long)device);
1da177e4 LT	239	dmab->addr = 0;
1da177e4 LT	240	break;
8f11551b	241	#ifdef CONFIG_HAS_DMA
a5606f85	242	#ifdef CONFIG_GENERIC_ALLOCATOR
05503214 NC	243	case SNDRV_DMA_TYPE_DEV_IRAM:
	244	snd_malloc_dev_iram(dmab, size);
	245	if (dmab->area)
	246	break;
	247	/* Internal memory might have limited size and no enough space,
	248	* so if we fail to malloc, try to fetch memory traditionally.
	249	*/
	250	dmab->dev.type = SNDRV_DMA_TYPE_DEV;
a5606f85	251	#endif /* CONFIG_GENERIC_ALLOCATOR */
1da177e4 LT	252	case SNDRV_DMA_TYPE_DEV:
	253	dmab->area = snd_malloc_dev_pages(device, size, &dmab->addr);
	254	break;
cc6a8acd TI	255	#endif
cc6a8acd TI	256	#ifdef CONFIG_SND_DMA_SGBUF
1da177e4 LT	257	case SNDRV_DMA_TYPE_DEV_SG:
	258	snd_malloc_sgbuf_pages(device, size, dmab, NULL);
	259	break;
8f11551b	260	#endif
1da177e4 LT	261	default:
	262	printk(KERN_ERR "snd-malloc: invalid device type %d\n", type);
	263	dmab->area = NULL;
	264	dmab->addr = 0;
	265	return -ENXIO;
	266	}
	267	if (! dmab->area)
	268	return -ENOMEM;
	269	dmab->bytes = size;
	270	return 0;
	271	}
	272
	273	/**
	274	* snd_dma_alloc_pages_fallback - allocate the buffer area according to the given type with fallback
	275	* @type: the DMA buffer type
	276	* @device: the device pointer
	277	* @size: the buffer size to allocate
	278	* @dmab: buffer allocation record to store the allocated data
	279	*
	280	* Calls the memory-allocator function for the corresponding
	281	* buffer type. When no space is left, this function reduces the size and
	282	* tries to allocate again. The size actually allocated is stored in
	283	* res_size argument.
eb7c06e8 YB	284	*
	285	* Return: Zero if the buffer with the given size is allocated successfully,
	286	* otherwise a negative value on error.
1da177e4 LT	287	*/
	288	int snd_dma_alloc_pages_fallback(int type, struct device *device, size_t size,
	289	struct snd_dma_buffer *dmab)
	290	{
	291	int err;
	292
1da177e4	293	while ((err = snd_dma_alloc_pages(type, device, size, dmab)) < 0) {
4e184f8f	294	size_t aligned_size;
1da177e4 LT	295	if (err != -ENOMEM)
1da177e4 LT	296	return err;
1da177e4 LT	297	if (size <= PAGE_SIZE)
1da177e4 LT	298	return -ENOMEM;
4e184f8f TI	299	aligned_size = PAGE_SIZE << get_order(size);
	300	if (size != aligned_size)
	301	size = aligned_size;
	302	else
	303	size >>= 1;
1da177e4 LT	304	}
	305	if (! dmab->area)
	306	return -ENOMEM;
	307	return 0;
	308	}
	309
	310
	311	/**
	312	* snd_dma_free_pages - release the allocated buffer
	313	* @dmab: the buffer allocation record to release
	314	*
	315	* Releases the allocated buffer via snd_dma_alloc_pages().
	316	*/
	317	void snd_dma_free_pages(struct snd_dma_buffer *dmab)
	318	{
	319	switch (dmab->dev.type) {
	320	case SNDRV_DMA_TYPE_CONTINUOUS:
	321	snd_free_pages(dmab->area, dmab->bytes);
	322	break;
8f11551b	323	#ifdef CONFIG_HAS_DMA
a5606f85	324	#ifdef CONFIG_GENERIC_ALLOCATOR
05503214 NC	325	case SNDRV_DMA_TYPE_DEV_IRAM:
	326	snd_free_dev_iram(dmab);
	327	break;
a5606f85	328	#endif /* CONFIG_GENERIC_ALLOCATOR */
1da177e4 LT	329	case SNDRV_DMA_TYPE_DEV:
	330	snd_free_dev_pages(dmab->dev.dev, dmab->bytes, dmab->area, dmab->addr);
	331	break;
cc6a8acd TI	332	#endif
cc6a8acd TI	333	#ifdef CONFIG_SND_DMA_SGBUF
1da177e4 LT	334	case SNDRV_DMA_TYPE_DEV_SG:
	335	snd_free_sgbuf_pages(dmab);
	336	break;
8f11551b	337	#endif
1da177e4 LT	338	default:
	339	printk(KERN_ERR "snd-malloc: invalid device type %d\n", dmab->dev.type);
	340	}
	341	}
	342
	343
	344	/**
	345	* snd_dma_get_reserved - get the reserved buffer for the given device
	346	* @dmab: the buffer allocation record to store
	347	* @id: the buffer id
	348	*
	349	* Looks for the reserved-buffer list and re-uses if the same buffer
	350	* is found in the list. When the buffer is found, it's removed from the free list.
	351	*
eb7c06e8	352	* Return: The size of buffer if the buffer is found, or zero if not found.
1da177e4 LT	353	*/
	354	size_t snd_dma_get_reserved_buf(struct snd_dma_buffer *dmab, unsigned int id)
	355	{
1da177e4 LT	356	struct snd_mem_list *mem;
1da177e4 LT	357
7eaa943c TI	358	if (WARN_ON(!dmab))
7eaa943c TI	359	return 0;
1da177e4	360
1a60d4c5	361	mutex_lock(&list_mutex);
9244b2c3	362	list_for_each_entry(mem, &mem_list_head, list) {
1da177e4	363	if (mem->id == id &&
b6a96915 TI	364	(mem->buffer.dev.dev == NULL \|\| dmab->dev.dev == NULL \|\|
	365	! memcmp(&mem->buffer.dev, &dmab->dev, sizeof(dmab->dev)))) {
	366	struct device *dev = dmab->dev.dev;
9244b2c3	367	list_del(&mem->list);
1da177e4	368	*dmab = mem->buffer;
b6a96915 TI	369	if (dmab->dev.dev == NULL)
b6a96915 TI	370	dmab->dev.dev = dev;
1da177e4	371	kfree(mem);
1a60d4c5	372	mutex_unlock(&list_mutex);
1da177e4 LT	373	return dmab->bytes;
	374	}
	375	}
1a60d4c5	376	mutex_unlock(&list_mutex);
1da177e4 LT	377	return 0;
	378	}
	379
	380	/**
	381	* snd_dma_reserve_buf - reserve the buffer
	382	* @dmab: the buffer to reserve
	383	* @id: the buffer id
	384	*
	385	* Reserves the given buffer as a reserved buffer.
eb7c06e8 YB	386	*
eb7c06e8 YB	387	* Return: Zero if successful, or a negative code on error.
1da177e4 LT	388	*/
	389	int snd_dma_reserve_buf(struct snd_dma_buffer *dmab, unsigned int id)
	390	{
	391	struct snd_mem_list *mem;
	392
7eaa943c TI	393	if (WARN_ON(!dmab))
7eaa943c TI	394	return -EINVAL;
1da177e4 LT	395	mem = kmalloc(sizeof(*mem), GFP_KERNEL);
	396	if (! mem)
	397	return -ENOMEM;
1a60d4c5	398	mutex_lock(&list_mutex);
1da177e4 LT	399	mem->buffer = *dmab;
	400	mem->id = id;
	401	list_add_tail(&mem->list, &mem_list_head);
1a60d4c5	402	mutex_unlock(&list_mutex);
1da177e4 LT	403	return 0;
	404	}
	405
	406	/*
	407	* purge all reserved buffers
	408	*/
	409	static void free_all_reserved_pages(void)
	410	{
	411	struct list_head *p;
	412	struct snd_mem_list *mem;
	413
1a60d4c5	414	mutex_lock(&list_mutex);
1da177e4 LT	415	while (! list_empty(&mem_list_head)) {
	416	p = mem_list_head.next;
	417	mem = list_entry(p, struct snd_mem_list, list);
	418	list_del(p);
	419	snd_dma_free_pages(&mem->buffer);
	420	kfree(mem);
	421	}
1a60d4c5	422	mutex_unlock(&list_mutex);
1da177e4 LT	423	}
	424
	425
1da177e4 LT	426	#ifdef CONFIG_PROC_FS
	427	/*
	428	* proc file interface
	429	*/
b6a96915	430	#define SND_MEM_PROC_FILE "driver/snd-page-alloc"
a53fc188	431	static struct proc_dir_entry *snd_mem_proc;
b6a96915	432
ccec6e2c	433	static int snd_mem_proc_read(struct seq_file seq, void offset)
1da177e4	434	{
1da177e4	435	long pages = snd_allocated_pages >> (PAGE_SHIFT-12);
1da177e4 LT	436	struct snd_mem_list *mem;
1da177e4 LT	437	int devno;
759ee81b	438	static char *types[] = { "UNKNOWN", "CONT", "DEV", "DEV-SG" };
1da177e4	439
1a60d4c5	440	mutex_lock(&list_mutex);
ccec6e2c TI	441	seq_printf(seq, "pages : %li bytes (%li pages per %likB)\n",
ccec6e2c TI	442	pages * PAGE_SIZE, pages, PAGE_SIZE / 1024);
1da177e4	443	devno = 0;
9244b2c3	444	list_for_each_entry(mem, &mem_list_head, list) {
1da177e4	445	devno++;
ccec6e2c TI	446	seq_printf(seq, "buffer %d : ID %08x : type %s\n",
	447	devno, mem->id, types[mem->buffer.dev.type]);
	448	seq_printf(seq, " addr = 0x%lx, size = %d bytes\n",
	449	(unsigned long)mem->buffer.addr,
	450	(int)mem->buffer.bytes);
1da177e4	451	}
1a60d4c5	452	mutex_unlock(&list_mutex);
ccec6e2c TI	453	return 0;
	454	}
	455
	456	static int snd_mem_proc_open(struct inode inode, struct file file)
	457	{
	458	return single_open(file, snd_mem_proc_read, NULL);
1da177e4	459	}
b6a96915 TI	460
	461	/* FIXME: for pci only - other bus? */
	462	#ifdef CONFIG_PCI
	463	#define gettoken(bufp) strsep(bufp, " \t\n")
	464
ccec6e2c TI	465	static ssize_t snd_mem_proc_write(struct file file, const char __user buffer,
ccec6e2c TI	466	size_t count, loff_t * ppos)
b6a96915 TI	467	{
	468	char buf[128];
	469	char token, p;
	470
ccec6e2c TI	471	if (count > sizeof(buf) - 1)
ccec6e2c TI	472	return -EINVAL;
b6a96915 TI	473	if (copy_from_user(buf, buffer, count))
b6a96915 TI	474	return -EFAULT;
ccec6e2c	475	buf[count] = '\0';
b6a96915 TI	476
	477	p = buf;
	478	token = gettoken(&p);
	479	if (! token \|\| *token == '#')
ccec6e2c	480	return count;
b6a96915 TI	481	if (strcmp(token, "add") == 0) {
	482	char *endp;
	483	int vendor, device, size, buffers;
	484	long mask;
	485	int i, alloced;
	486	struct pci_dev *pci;
	487
	488	if ((token = gettoken(&p)) == NULL \|\|
	489	(vendor = simple_strtol(token, NULL, 0)) <= 0 \|\|
	490	(token = gettoken(&p)) == NULL \|\|
	491	(device = simple_strtol(token, NULL, 0)) <= 0 \|\|
	492	(token = gettoken(&p)) == NULL \|\|
	493	(mask = simple_strtol(token, NULL, 0)) < 0 \|\|
	494	(token = gettoken(&p)) == NULL \|\|
	495	(size = memparse(token, &endp)) < 64*1024 \|\|
	496	size > 1610241024 /* too big */ \|\|
	497	(token = gettoken(&p)) == NULL \|\|
	498	(buffers = simple_strtol(token, NULL, 0)) <= 0 \|\|
	499	buffers > 4) {
	500	printk(KERN_ERR "snd-page-alloc: invalid proc write format\n");
ccec6e2c	501	return count;
b6a96915 TI	502	}
	503	vendor &= 0xffff;
	504	device &= 0xffff;
	505
	506	alloced = 0;
	507	pci = NULL;
0dd119f7	508	while ((pci = pci_get_device(vendor, device, pci)) != NULL) {
b6a96915 TI	509	if (mask > 0 && mask < 0xffffffff) {
	510	if (pci_set_dma_mask(pci, mask) < 0 \|\|
	511	pci_set_consistent_dma_mask(pci, mask) < 0) {
	512	printk(KERN_ERR "snd-page-alloc: cannot set DMA mask %lx for pci %04x:%04x\n", mask, vendor, device);
df1deb67	513	pci_dev_put(pci);
ccec6e2c	514	return count;
b6a96915 TI	515	}
	516	}
	517	for (i = 0; i < buffers; i++) {
	518	struct snd_dma_buffer dmab;
	519	memset(&dmab, 0, sizeof(dmab));
	520	if (snd_dma_alloc_pages(SNDRV_DMA_TYPE_DEV, snd_dma_pci_data(pci),
	521	size, &dmab) < 0) {
	522	printk(KERN_ERR "snd-page-alloc: cannot allocate buffer pages (size = %d)\n", size);
0dd119f7	523	pci_dev_put(pci);
ccec6e2c	524	return count;
b6a96915 TI	525	}
	526	snd_dma_reserve_buf(&dmab, snd_dma_pci_buf_id(pci));
	527	}
	528	alloced++;
	529	}
	530	if (! alloced) {
	531	for (i = 0; i < buffers; i++) {
	532	struct snd_dma_buffer dmab;
	533	memset(&dmab, 0, sizeof(dmab));
	534	/* FIXME: We can allocate only in ZONE_DMA
	535	* without a device pointer!
	536	*/
	537	if (snd_dma_alloc_pages(SNDRV_DMA_TYPE_DEV, NULL,
	538	size, &dmab) < 0) {
	539	printk(KERN_ERR "snd-page-alloc: cannot allocate buffer pages (size = %d)\n", size);
	540	break;
	541	}
	542	snd_dma_reserve_buf(&dmab, (unsigned int)((vendor << 16) \| device));
	543	}
	544	}
	545	} else if (strcmp(token, "erase") == 0)
	546	/* FIXME: need for releasing each buffer chunk? */
	547	free_all_reserved_pages();
	548	else
	549	printk(KERN_ERR "snd-page-alloc: invalid proc cmd\n");
ccec6e2c	550	return count;
b6a96915 TI	551	}
b6a96915 TI	552	#endif /* CONFIG_PCI */
ccec6e2c TI	553
	554	static const struct file_operations snd_mem_proc_fops = {
	555	.owner = THIS_MODULE,
	556	.open = snd_mem_proc_open,
	557	.read = seq_read,
	558	#ifdef CONFIG_PCI
	559	.write = snd_mem_proc_write,
	560	#endif
	561	.llseek = seq_lseek,
	562	.release = single_release,
	563	};
	564
1da177e4 LT	565	#endif /* CONFIG_PROC_FS */
	566
	567	/*
	568	* module entry
	569	*/
	570
	571	static int __init snd_mem_init(void)
	572	{
	573	#ifdef CONFIG_PROC_FS
7bf4e6d3 DL	574	snd_mem_proc = proc_create(SND_MEM_PROC_FILE, 0644, NULL,
7bf4e6d3 DL	575	&snd_mem_proc_fops);
1da177e4	576	#endif
1da177e4 LT	577	return 0;
	578	}
	579
	580	static void __exit snd_mem_exit(void)
	581	{
e0be4d32	582	remove_proc_entry(SND_MEM_PROC_FILE, NULL);
1da177e4 LT	583	free_all_reserved_pages();
	584	if (snd_allocated_pages > 0)
	585	printk(KERN_ERR "snd-malloc: Memory leak? pages not freed = %li\n", snd_allocated_pages);
	586	}
	587
	588
	589	module_init(snd_mem_init)
	590	module_exit(snd_mem_exit)
	591
	592
	593	/*
	594	* exports
	595	*/
	596	EXPORT_SYMBOL(snd_dma_alloc_pages);
	597	EXPORT_SYMBOL(snd_dma_alloc_pages_fallback);
	598	EXPORT_SYMBOL(snd_dma_free_pages);
	599
	600	EXPORT_SYMBOL(snd_dma_get_reserved_buf);
	601	EXPORT_SYMBOL(snd_dma_reserve_buf);
	602
	603	EXPORT_SYMBOL(snd_malloc_pages);
	604	EXPORT_SYMBOL(snd_free_pages);