[linux-2.6-block.git] / drivers / dma / dmatest.c

/*
 * DMA Engine test module
 *
 * Copyright (C) 2007 Atmel Corporation
 *
 * This program is free software; you can redistribute it and/or modify
 * it under the terms of the GNU General Public License version 2 as
 * published by the Free Software Foundation.
 */
#include <linux/delay.h>
#include <linux/dmaengine.h>
#include <linux/init.h>
#include <linux/kthread.h>
#include <linux/module.h>
#include <linux/moduleparam.h>
#include <linux/random.h>
#include <linux/wait.h>

static unsigned int test_buf_size = 16384;
module_param(test_buf_size, uint, S_IRUGO);
MODULE_PARM_DESC(test_buf_size, "Size of the memcpy test buffer");

static char test_channel[20];
module_param_string(channel, test_channel, sizeof(test_channel), S_IRUGO);
MODULE_PARM_DESC(channel, "Bus ID of the channel to test (default: any)");

static char test_device[20];
module_param_string(device, test_device, sizeof(test_device), S_IRUGO);
MODULE_PARM_DESC(device, "Bus ID of the DMA Engine to test (default: any)");

static unsigned int threads_per_chan = 1;
module_param(threads_per_chan, uint, S_IRUGO);
MODULE_PARM_DESC(threads_per_chan,
		"Number of threads to start per channel (default: 1)");

static unsigned int max_channels;
module_param(max_channels, uint, S_IRUGO);
MODULE_PARM_DESC(max_channels,
		"Maximum number of channels to use (default: all)");

/*
 * Initialization patterns. All bytes in the source buffer has bit 7
 * set, all bytes in the destination buffer has bit 7 cleared.
 *
 * Bit 6 is set for all bytes which are to be copied by the DMA
 * engine. Bit 5 is set for all bytes which are to be overwritten by
 * the DMA engine.
 *
 * The remaining bits are the inverse of a counter which increments by
 * one for each byte address.
 */
#define PATTERN_SRC		0x80
#define PATTERN_DST		0x00
#define PATTERN_COPY		0x40
#define PATTERN_OVERWRITE	0x20
#define PATTERN_COUNT_MASK	0x1f

struct dmatest_thread {
	struct list_head	node;
	struct task_struct	*task;
	struct dma_chan		*chan;
	u8			*srcbuf;
	u8			*dstbuf;
};

struct dmatest_chan {
	struct list_head	node;
	struct dma_chan		*chan;
	struct list_head	threads;
};

/*
 * These are protected by dma_list_mutex since they're only used by
 * the DMA filter function callback
 */
static LIST_HEAD(dmatest_channels);
static unsigned int nr_channels;

static bool dmatest_match_channel(struct dma_chan *chan)
{
	if (test_channel[0] == '\0')
		return true;
	return strcmp(dma_chan_name(chan), test_channel) == 0;
}

static bool dmatest_match_device(struct dma_device *device)
{
	if (test_device[0] == '\0')
		return true;
	return strcmp(dev_name(device->dev), test_device) == 0;
}

static unsigned long dmatest_random(void)
{
	unsigned long buf;

	get_random_bytes(&buf, sizeof(buf));
	return buf;
}

static void dmatest_init_srcbuf(u8 *buf, unsigned int start, unsigned int len)
{
	unsigned int i;

	for (i = 0; i < start; i++)
		buf[i] = PATTERN_SRC | (~i & PATTERN_COUNT_MASK);
	for ( ; i < start + len; i++)
		buf[i] = PATTERN_SRC | PATTERN_COPY
			| (~i & PATTERN_COUNT_MASK);;
	for ( ; i < test_buf_size; i++)
		buf[i] = PATTERN_SRC | (~i & PATTERN_COUNT_MASK);
}

static void dmatest_init_dstbuf(u8 *buf, unsigned int start, unsigned int len)
{
	unsigned int i;

	for (i = 0; i < start; i++)
		buf[i] = PATTERN_DST | (~i & PATTERN_COUNT_MASK);
	for ( ; i < start + len; i++)
		buf[i] = PATTERN_DST | PATTERN_OVERWRITE
			| (~i & PATTERN_COUNT_MASK);
	for ( ; i < test_buf_size; i++)
		buf[i] = PATTERN_DST | (~i & PATTERN_COUNT_MASK);
}

static void dmatest_mismatch(u8 actual, u8 pattern, unsigned int index,
		unsigned int counter, bool is_srcbuf)
{
	u8		diff = actual ^ pattern;
	u8		expected = pattern | (~counter & PATTERN_COUNT_MASK);
	const char	*thread_name = current->comm;

	if (is_srcbuf)
		pr_warning("%s: srcbuf[0x%x] overwritten!"
				" Expected %02x, got %02x\n",
				thread_name, index, expected, actual);
	else if ((pattern & PATTERN_COPY)
			&& (diff & (PATTERN_COPY | PATTERN_OVERWRITE)))
		pr_warning("%s: dstbuf[0x%x] not copied!"
				" Expected %02x, got %02x\n",
				thread_name, index, expected, actual);
	else if (diff & PATTERN_SRC)
		pr_warning("%s: dstbuf[0x%x] was copied!"
				" Expected %02x, got %02x\n",
				thread_name, index, expected, actual);
	else
		pr_warning("%s: dstbuf[0x%x] mismatch!"
				" Expected %02x, got %02x\n",
				thread_name, index, expected, actual);
}

static unsigned int dmatest_verify(u8 *buf, unsigned int start,
		unsigned int end, unsigned int counter, u8 pattern,
		bool is_srcbuf)
{
	unsigned int i;
	unsigned int error_count = 0;
	u8 actual;

	for (i = start; i < end; i++) {
		actual = buf[i];
		if (actual != (pattern | (~counter & PATTERN_COUNT_MASK))) {
			if (error_count < 32)
				dmatest_mismatch(actual, pattern, i, counter,
						is_srcbuf);
			error_count++;
		}
		counter++;
	}

	if (error_count > 32)
		pr_warning("%s: %u errors suppressed\n",
			current->comm, error_count - 32);

	return error_count;
}

/*
 * This function repeatedly tests DMA transfers of various lengths and
 * offsets until it is told to exit by kthread_stop(). There may be
 * multiple threads running this function in parallel for a single
 * channel, and there may be multiple channels being tested in
 * parallel.
 *
 * Before each test, the source and destination buffer is initialized
 * with a known pattern. This pattern is different depending on
 * whether it's in an area which is supposed to be copied or
 * overwritten, and different in the source and destination buffers.
 * So if the DMA engine doesn't copy exactly what we tell it to copy,
 * we'll notice.
 */
static int dmatest_func(void *data)
{
	struct dmatest_thread	*thread = data;
	struct dma_chan		*chan;
	const char		*thread_name;
	unsigned int		src_off, dst_off, len;
	unsigned int		error_count;
	unsigned int		failed_tests = 0;
	unsigned int		total_tests = 0;
	dma_cookie_t		cookie;
	enum dma_status		status;
	int			ret;

	thread_name = current->comm;

	ret = -ENOMEM;
	thread->srcbuf = kmalloc(test_buf_size, GFP_KERNEL);
	if (!thread->srcbuf)
		goto err_srcbuf;
	thread->dstbuf = kmalloc(test_buf_size, GFP_KERNEL);
	if (!thread->dstbuf)
		goto err_dstbuf;

	smp_rmb();
	chan = thread->chan;

	while (!kthread_should_stop()) {
		struct dma_device *dev = chan->device;
		struct dma_async_tx_descriptor *tx;
		dma_addr_t dma_src, dma_dest;

		total_tests++;

		len = dmatest_random() % test_buf_size + 1;
		src_off = dmatest_random() % (test_buf_size - len + 1);
		dst_off = dmatest_random() % (test_buf_size - len + 1);

		dmatest_init_srcbuf(thread->srcbuf, src_off, len);
		dmatest_init_dstbuf(thread->dstbuf, dst_off, len);

		dma_src = dma_map_single(dev->dev, thread->srcbuf + src_off,
				len, DMA_TO_DEVICE);
		/* map with DMA_BIDIRECTIONAL to force writeback/invalidate */
		dma_dest = dma_map_single(dev->dev, thread->dstbuf,
				test_buf_size, DMA_BIDIRECTIONAL);

		tx = dev->device_prep_dma_memcpy(chan, dma_dest + dst_off,
				dma_src, len,
				DMA_CTRL_ACK | DMA_COMPL_SKIP_DEST_UNMAP);
		if (!tx) {
			dma_unmap_single(dev->dev, dma_src, len, DMA_TO_DEVICE);
			dma_unmap_single(dev->dev, dma_dest,
					test_buf_size, DMA_BIDIRECTIONAL);
			pr_warning("%s: #%u: prep error with src_off=0x%x "
					"dst_off=0x%x len=0x%x\n",
					thread_name, total_tests - 1,
					src_off, dst_off, len);
			msleep(100);
			failed_tests++;
			continue;
		}
		tx->callback = NULL;
		cookie = tx->tx_submit(tx);

		if (dma_submit_error(cookie)) {
			pr_warning("%s: #%u: submit error %d with src_off=0x%x "
					"dst_off=0x%x len=0x%x\n",
					thread_name, total_tests - 1, cookie,
					src_off, dst_off, len);
			msleep(100);
			failed_tests++;
			continue;
		}
		dma_async_memcpy_issue_pending(chan);

		do {
			msleep(1);
			status = dma_async_memcpy_complete(
					chan, cookie, NULL, NULL);
		} while (status == DMA_IN_PROGRESS);

		if (status == DMA_ERROR) {
			pr_warning("%s: #%u: error during copy\n",
					thread_name, total_tests - 1);
			failed_tests++;
			continue;
		}
		/* Unmap by myself (see DMA_COMPL_SKIP_DEST_UNMAP above) */
		dma_unmap_single(dev->dev, dma_dest,
				test_buf_size, DMA_BIDIRECTIONAL);

		error_count = 0;

		pr_debug("%s: verifying source buffer...\n", thread_name);
		error_count += dmatest_verify(thread->srcbuf, 0, src_off,
				0, PATTERN_SRC, true);
		error_count += dmatest_verify(thread->srcbuf, src_off,
				src_off + len, src_off,
				PATTERN_SRC | PATTERN_COPY, true);
		error_count += dmatest_verify(thread->srcbuf, src_off + len,
				test_buf_size, src_off + len,
				PATTERN_SRC, true);

		pr_debug("%s: verifying dest buffer...\n",
				thread->task->comm);
		error_count += dmatest_verify(thread->dstbuf, 0, dst_off,
				0, PATTERN_DST, false);
		error_count += dmatest_verify(thread->dstbuf, dst_off,
				dst_off + len, src_off,
				PATTERN_SRC | PATTERN_COPY, false);
		error_count += dmatest_verify(thread->dstbuf, dst_off + len,
				test_buf_size, dst_off + len,
				PATTERN_DST, false);

		if (error_count) {
			pr_warning("%s: #%u: %u errors with "
				"src_off=0x%x dst_off=0x%x len=0x%x\n",
				thread_name, total_tests - 1, error_count,
				src_off, dst_off, len);
			failed_tests++;
		} else {
			pr_debug("%s: #%u: No errors with "
				"src_off=0x%x dst_off=0x%x len=0x%x\n",
				thread_name, total_tests - 1,
				src_off, dst_off, len);
		}
	}

	ret = 0;
	kfree(thread->dstbuf);
err_dstbuf:
	kfree(thread->srcbuf);
err_srcbuf:
	pr_notice("%s: terminating after %u tests, %u failures (status %d)\n",
			thread_name, total_tests, failed_tests, ret);
	return ret;
}

static void dmatest_cleanup_channel(struct dmatest_chan *dtc)
{
	struct dmatest_thread	*thread;
	struct dmatest_thread	*_thread;
	int			ret;

	list_for_each_entry_safe(thread, _thread, &dtc->threads, node) {
		ret = kthread_stop(thread->task);
		pr_debug("dmatest: thread %s exited with status %d\n",
				thread->task->comm, ret);
		list_del(&thread->node);
		kfree(thread);
	}
	kfree(dtc);
}

static int dmatest_add_channel(struct dma_chan *chan)
{
	struct dmatest_chan	*dtc;
	struct dmatest_thread	*thread;
	unsigned int		i;

	dtc = kmalloc(sizeof(struct dmatest_chan), GFP_KERNEL);
	if (!dtc) {
		pr_warning("dmatest: No memory for %s\n", dma_chan_name(chan));
		return -ENOMEM;
	}

	dtc->chan = chan;
	INIT_LIST_HEAD(&dtc->threads);

	for (i = 0; i < threads_per_chan; i++) {
		thread = kzalloc(sizeof(struct dmatest_thread), GFP_KERNEL);
		if (!thread) {
			pr_warning("dmatest: No memory for %s-test%u\n",
				   dma_chan_name(chan), i);
			break;
		}
		thread->chan = dtc->chan;
		smp_wmb();
		thread->task = kthread_run(dmatest_func, thread, "%s-test%u",
				dma_chan_name(chan), i);
		if (IS_ERR(thread->task)) {
			pr_warning("dmatest: Failed to run thread %s-test%u\n",
					dma_chan_name(chan), i);
			kfree(thread);
			break;
		}

		/* srcbuf and dstbuf are allocated by the thread itself */

		list_add_tail(&thread->node, &dtc->threads);
	}

	pr_info("dmatest: Started %u threads using %s\n", i, dma_chan_name(chan));

	list_add_tail(&dtc->node, &dmatest_channels);
	nr_channels++;

	return 0;
}

static bool filter(struct dma_chan *chan, void *param)
{
	if (!dmatest_match_channel(chan) || !dmatest_match_device(chan->device))
		return false;
	else
		return true;
}

static int __init dmatest_init(void)
{
	dma_cap_mask_t mask;
	struct dma_chan *chan;
	int err = 0;

	dma_cap_zero(mask);
	dma_cap_set(DMA_MEMCPY, mask);
	for (;;) {
		chan = dma_request_channel(mask, filter, NULL);
		if (chan) {
			err = dmatest_add_channel(chan);
			if (err == 0)
				continue;
			else {
				dma_release_channel(chan);
				break; /* add_channel failed, punt */
			}
		} else
			break; /* no more channels available */
		if (max_channels && nr_channels >= max_channels)
			break; /* we have all we need */
	}

	return err;
}
/* when compiled-in wait for drivers to load first */
late_initcall(dmatest_init);

static void __exit dmatest_exit(void)
{
	struct dmatest_chan *dtc, *_dtc;
	struct dma_chan *chan;

	list_for_each_entry_safe(dtc, _dtc, &dmatest_channels, node) {
		list_del(&dtc->node);
		chan = dtc->chan;
		dmatest_cleanup_channel(dtc);
		pr_debug("dmatest: dropped channel %s\n",
			 dma_chan_name(chan));
		dma_release_channel(chan);
	}
}
module_exit(dmatest_exit);

MODULE_AUTHOR("Haavard Skinnemoen <hskinnemoen@atmel.com>");
MODULE_LICENSE("GPL v2");
Commit	Line	Data
4a776f0a HS	1	/*
	2	* DMA Engine test module
	3	*
	4	* Copyright (C) 2007 Atmel Corporation
	5	*
	6	* This program is free software; you can redistribute it and/or modify
	7	* it under the terms of the GNU General Public License version 2 as
	8	* published by the Free Software Foundation.
	9	*/
	10	#include <linux/delay.h>
	11	#include <linux/dmaengine.h>
	12	#include <linux/init.h>
	13	#include <linux/kthread.h>
	14	#include <linux/module.h>
	15	#include <linux/moduleparam.h>
	16	#include <linux/random.h>
	17	#include <linux/wait.h>
	18
	19	static unsigned int test_buf_size = 16384;
	20	module_param(test_buf_size, uint, S_IRUGO);
	21	MODULE_PARM_DESC(test_buf_size, "Size of the memcpy test buffer");
	22
06190d84	23	static char test_channel[20];
4a776f0a HS	24	module_param_string(channel, test_channel, sizeof(test_channel), S_IRUGO);
	25	MODULE_PARM_DESC(channel, "Bus ID of the channel to test (default: any)");
	26
06190d84	27	static char test_device[20];
4a776f0a HS	28	module_param_string(device, test_device, sizeof(test_device), S_IRUGO);
	29	MODULE_PARM_DESC(device, "Bus ID of the DMA Engine to test (default: any)");
	30
	31	static unsigned int threads_per_chan = 1;
	32	module_param(threads_per_chan, uint, S_IRUGO);
	33	MODULE_PARM_DESC(threads_per_chan,
	34	"Number of threads to start per channel (default: 1)");
	35
	36	static unsigned int max_channels;
	37	module_param(max_channels, uint, S_IRUGO);
33df8ca0	38	MODULE_PARM_DESC(max_channels,
4a776f0a HS	39	"Maximum number of channels to use (default: all)");
	40
	41	/*
	42	* Initialization patterns. All bytes in the source buffer has bit 7
	43	* set, all bytes in the destination buffer has bit 7 cleared.
	44	*
	45	* Bit 6 is set for all bytes which are to be copied by the DMA
	46	* engine. Bit 5 is set for all bytes which are to be overwritten by
	47	* the DMA engine.
	48	*
	49	* The remaining bits are the inverse of a counter which increments by
	50	* one for each byte address.
	51	*/
	52	#define PATTERN_SRC 0x80
	53	#define PATTERN_DST 0x00
	54	#define PATTERN_COPY 0x40
	55	#define PATTERN_OVERWRITE 0x20
	56	#define PATTERN_COUNT_MASK 0x1f
	57
	58	struct dmatest_thread {
	59	struct list_head node;
	60	struct task_struct *task;
	61	struct dma_chan *chan;
	62	u8 *srcbuf;
	63	u8 *dstbuf;
	64	};
	65
	66	struct dmatest_chan {
	67	struct list_head node;
	68	struct dma_chan *chan;
	69	struct list_head threads;
	70	};
	71
	72	/*
	73	* These are protected by dma_list_mutex since they're only used by
33df8ca0	74	* the DMA filter function callback
4a776f0a HS	75	*/
	76	static LIST_HEAD(dmatest_channels);
	77	static unsigned int nr_channels;
	78
	79	static bool dmatest_match_channel(struct dma_chan *chan)
	80	{
	81	if (test_channel[0] == '\0')
	82	return true;
41d5e59c	83	return strcmp(dma_chan_name(chan), test_channel) == 0;
4a776f0a HS	84	}
	85
	86	static bool dmatest_match_device(struct dma_device *device)
	87	{
	88	if (test_device[0] == '\0')
	89	return true;
06190d84	90	return strcmp(dev_name(device->dev), test_device) == 0;
4a776f0a HS	91	}
	92
	93	static unsigned long dmatest_random(void)
	94	{
	95	unsigned long buf;
	96
	97	get_random_bytes(&buf, sizeof(buf));
	98	return buf;
	99	}
	100
	101	static void dmatest_init_srcbuf(u8 *buf, unsigned int start, unsigned int len)
	102	{
	103	unsigned int i;
	104
	105	for (i = 0; i < start; i++)
	106	buf[i] = PATTERN_SRC \| (~i & PATTERN_COUNT_MASK);
	107	for ( ; i < start + len; i++)
	108	buf[i] = PATTERN_SRC \| PATTERN_COPY
	109	\| (~i & PATTERN_COUNT_MASK);;
	110	for ( ; i < test_buf_size; i++)
	111	buf[i] = PATTERN_SRC \| (~i & PATTERN_COUNT_MASK);
	112	}
	113
	114	static void dmatest_init_dstbuf(u8 *buf, unsigned int start, unsigned int len)
	115	{
	116	unsigned int i;
	117
	118	for (i = 0; i < start; i++)
	119	buf[i] = PATTERN_DST \| (~i & PATTERN_COUNT_MASK);
	120	for ( ; i < start + len; i++)
	121	buf[i] = PATTERN_DST \| PATTERN_OVERWRITE
	122	\| (~i & PATTERN_COUNT_MASK);
	123	for ( ; i < test_buf_size; i++)
	124	buf[i] = PATTERN_DST \| (~i & PATTERN_COUNT_MASK);
	125	}
	126
	127	static void dmatest_mismatch(u8 actual, u8 pattern, unsigned int index,
	128	unsigned int counter, bool is_srcbuf)
	129	{
	130	u8 diff = actual ^ pattern;
	131	u8 expected = pattern \| (~counter & PATTERN_COUNT_MASK);
	132	const char *thread_name = current->comm;
	133
	134	if (is_srcbuf)
	135	pr_warning("%s: srcbuf[0x%x] overwritten!"
	136	" Expected %02x, got %02x\n",
	137	thread_name, index, expected, actual);
	138	else if ((pattern & PATTERN_COPY)
	139	&& (diff & (PATTERN_COPY \| PATTERN_OVERWRITE)))
	140	pr_warning("%s: dstbuf[0x%x] not copied!"
	141	" Expected %02x, got %02x\n",
	142	thread_name, index, expected, actual);
	143	else if (diff & PATTERN_SRC)
	144	pr_warning("%s: dstbuf[0x%x] was copied!"
	145	" Expected %02x, got %02x\n",
	146	thread_name, index, expected, actual);
	147	else
	148	pr_warning("%s: dstbuf[0x%x] mismatch!"
	149	" Expected %02x, got %02x\n",
	150	thread_name, index, expected, actual);
	151	}
	152
	153	static unsigned int dmatest_verify(u8 *buf, unsigned int start,
	154	unsigned int end, unsigned int counter, u8 pattern,
155	bool is_srcbuf)
156	{
157	unsigned int i;
158	unsigned int error_count = 0;
159	u8 actual;
160
161	for (i = start; i < end; i++) {
162	actual = buf[i];
163	if (actual != (pattern \| (~counter & PATTERN_COUNT_MASK))) {
164	if (error_count < 32)
165	dmatest_mismatch(actual, pattern, i, counter,
166	is_srcbuf);
167	error_count++;
168	}
169	counter++;
170	}
171
172	if (error_count > 32)
173	pr_warning("%s: %u errors suppressed\n",
174	current->comm, error_count - 32);
175
176	return error_count;
177	}
178
179	/*
180	* This function repeatedly tests DMA transfers of various lengths and
181	* offsets until it is told to exit by kthread_stop(). There may be
182	* multiple threads running this function in parallel for a single
183	* channel, and there may be multiple channels being tested in
184	* parallel.
185	*
186	* Before each test, the source and destination buffer is initialized
187	* with a known pattern. This pattern is different depending on
188	* whether it's in an area which is supposed to be copied or
189	* overwritten, and different in the source and destination buffers.
190	* So if the DMA engine doesn't copy exactly what we tell it to copy,
191	* we'll notice.
192	*/
193	static int dmatest_func(void *data)
194	{
195	struct dmatest_thread *thread = data;
196	struct dma_chan *chan;
197	const char *thread_name;
198	unsigned int src_off, dst_off, len;
199	unsigned int error_count;
200	unsigned int failed_tests = 0;
201	unsigned int total_tests = 0;
202	dma_cookie_t cookie;
203	enum dma_status status;
204	int ret;
205
206	thread_name = current->comm;
207
208	ret = -ENOMEM;
209	thread->srcbuf = kmalloc(test_buf_size, GFP_KERNEL);
210	if (!thread->srcbuf)
211	goto err_srcbuf;
212	thread->dstbuf = kmalloc(test_buf_size, GFP_KERNEL);
213	if (!thread->dstbuf)
214	goto err_dstbuf;
215
216	smp_rmb();
217	chan = thread->chan;
4a776f0a HS	218
4a776f0a HS	219	while (!kthread_should_stop()) {
d86be86e AN	220	struct dma_device *dev = chan->device;
	221	struct dma_async_tx_descriptor *tx;
	222	dma_addr_t dma_src, dma_dest;
	223
4a776f0a HS	224	total_tests++;
	225
	226	len = dmatest_random() % test_buf_size + 1;
	227	src_off = dmatest_random() % (test_buf_size - len + 1);
	228	dst_off = dmatest_random() % (test_buf_size - len + 1);
	229
	230	dmatest_init_srcbuf(thread->srcbuf, src_off, len);
	231	dmatest_init_dstbuf(thread->dstbuf, dst_off, len);
	232
d86be86e AN	233	dma_src = dma_map_single(dev->dev, thread->srcbuf + src_off,
	234	len, DMA_TO_DEVICE);
	235	/* map with DMA_BIDIRECTIONAL to force writeback/invalidate */
	236	dma_dest = dma_map_single(dev->dev, thread->dstbuf,
	237	test_buf_size, DMA_BIDIRECTIONAL);
	238
	239	tx = dev->device_prep_dma_memcpy(chan, dma_dest + dst_off,
	240	dma_src, len,
	241	DMA_CTRL_ACK \| DMA_COMPL_SKIP_DEST_UNMAP);
	242	if (!tx) {
	243	dma_unmap_single(dev->dev, dma_src, len, DMA_TO_DEVICE);
	244	dma_unmap_single(dev->dev, dma_dest,
	245	test_buf_size, DMA_BIDIRECTIONAL);
	246	pr_warning("%s: #%u: prep error with src_off=0x%x "
	247	"dst_off=0x%x len=0x%x\n",
	248	thread_name, total_tests - 1,
	249	src_off, dst_off, len);
	250	msleep(100);
	251	failed_tests++;
	252	continue;
	253	}
	254	tx->callback = NULL;
	255	cookie = tx->tx_submit(tx);
	256
4a776f0a HS	257	if (dma_submit_error(cookie)) {
	258	pr_warning("%s: #%u: submit error %d with src_off=0x%x "
	259	"dst_off=0x%x len=0x%x\n",
	260	thread_name, total_tests - 1, cookie,
	261	src_off, dst_off, len);
	262	msleep(100);
	263	failed_tests++;
	264	continue;
	265	}
	266	dma_async_memcpy_issue_pending(chan);
	267
	268	do {
	269	msleep(1);
	270	status = dma_async_memcpy_complete(
	271	chan, cookie, NULL, NULL);
	272	} while (status == DMA_IN_PROGRESS);
	273
	274	if (status == DMA_ERROR) {
	275	pr_warning("%s: #%u: error during copy\n",
	276	thread_name, total_tests - 1);
	277	failed_tests++;
	278	continue;
	279	}
d86be86e AN	280	/* Unmap by myself (see DMA_COMPL_SKIP_DEST_UNMAP above) */
	281	dma_unmap_single(dev->dev, dma_dest,
	282	test_buf_size, DMA_BIDIRECTIONAL);
4a776f0a HS	283
	284	error_count = 0;
	285
	286	pr_debug("%s: verifying source buffer...\n", thread_name);
	287	error_count += dmatest_verify(thread->srcbuf, 0, src_off,
	288	0, PATTERN_SRC, true);
	289	error_count += dmatest_verify(thread->srcbuf, src_off,
	290	src_off + len, src_off,
	291	PATTERN_SRC \| PATTERN_COPY, true);
	292	error_count += dmatest_verify(thread->srcbuf, src_off + len,
	293	test_buf_size, src_off + len,
	294	PATTERN_SRC, true);
	295
	296	pr_debug("%s: verifying dest buffer...\n",
	297	thread->task->comm);
	298	error_count += dmatest_verify(thread->dstbuf, 0, dst_off,
	299	0, PATTERN_DST, false);
	300	error_count += dmatest_verify(thread->dstbuf, dst_off,
	301	dst_off + len, src_off,
	302	PATTERN_SRC \| PATTERN_COPY, false);
	303	error_count += dmatest_verify(thread->dstbuf, dst_off + len,
	304	test_buf_size, dst_off + len,
	305	PATTERN_DST, false);
	306
	307	if (error_count) {
	308	pr_warning("%s: #%u: %u errors with "
	309	"src_off=0x%x dst_off=0x%x len=0x%x\n",
	310	thread_name, total_tests - 1, error_count,
	311	src_off, dst_off, len);
	312	failed_tests++;
	313	} else {
	314	pr_debug("%s: #%u: No errors with "
	315	"src_off=0x%x dst_off=0x%x len=0x%x\n",
	316	thread_name, total_tests - 1,
	317	src_off, dst_off, len);
	318	}
	319	}
	320
	321	ret = 0;
4a776f0a HS	322	kfree(thread->dstbuf);
	323	err_dstbuf:
	324	kfree(thread->srcbuf);
	325	err_srcbuf:
	326	pr_notice("%s: terminating after %u tests, %u failures (status %d)\n",
	327	thread_name, total_tests, failed_tests, ret);
	328	return ret;
	329	}
	330
	331	static void dmatest_cleanup_channel(struct dmatest_chan *dtc)
	332	{
	333	struct dmatest_thread *thread;
	334	struct dmatest_thread *_thread;
	335	int ret;
	336
	337	list_for_each_entry_safe(thread, _thread, &dtc->threads, node) {
	338	ret = kthread_stop(thread->task);
	339	pr_debug("dmatest: thread %s exited with status %d\n",
	340	thread->task->comm, ret);
	341	list_del(&thread->node);
	342	kfree(thread);
	343	}
	344	kfree(dtc);
	345	}
	346
33df8ca0	347	static int dmatest_add_channel(struct dma_chan *chan)
4a776f0a HS	348	{
	349	struct dmatest_chan *dtc;
	350	struct dmatest_thread *thread;
	351	unsigned int i;
	352
6fdb8bd4	353	dtc = kmalloc(sizeof(struct dmatest_chan), GFP_KERNEL);
4a776f0a	354	if (!dtc) {
41d5e59c	355	pr_warning("dmatest: No memory for %s\n", dma_chan_name(chan));
33df8ca0	356	return -ENOMEM;
4a776f0a HS	357	}
	358
	359	dtc->chan = chan;
	360	INIT_LIST_HEAD(&dtc->threads);
	361
	362	for (i = 0; i < threads_per_chan; i++) {
	363	thread = kzalloc(sizeof(struct dmatest_thread), GFP_KERNEL);
	364	if (!thread) {
	365	pr_warning("dmatest: No memory for %s-test%u\n",
41d5e59c	366	dma_chan_name(chan), i);
4a776f0a HS	367	break;
	368	}
	369	thread->chan = dtc->chan;
	370	smp_wmb();
	371	thread->task = kthread_run(dmatest_func, thread, "%s-test%u",
41d5e59c	372	dma_chan_name(chan), i);
4a776f0a HS	373	if (IS_ERR(thread->task)) {
4a776f0a HS	374	pr_warning("dmatest: Failed to run thread %s-test%u\n",
41d5e59c	375	dma_chan_name(chan), i);
4a776f0a HS	376	kfree(thread);
	377	break;
	378	}
	379
	380	/* srcbuf and dstbuf are allocated by the thread itself */
	381
	382	list_add_tail(&thread->node, &dtc->threads);
	383	}
	384
41d5e59c	385	pr_info("dmatest: Started %u threads using %s\n", i, dma_chan_name(chan));
4a776f0a HS	386
	387	list_add_tail(&dtc->node, &dmatest_channels);
	388	nr_channels++;
	389
33df8ca0	390	return 0;
4a776f0a HS	391	}
4a776f0a HS	392
7dd60251	393	static bool filter(struct dma_chan chan, void param)
4a776f0a	394	{
33df8ca0	395	if (!dmatest_match_channel(chan) \|\| !dmatest_match_device(chan->device))
7dd60251	396	return false;
33df8ca0	397	else
7dd60251	398	return true;
4a776f0a HS	399	}
4a776f0a HS	400
4a776f0a HS	401	static int __init dmatest_init(void)
4a776f0a HS	402	{
33df8ca0 DW	403	dma_cap_mask_t mask;
	404	struct dma_chan *chan;
	405	int err = 0;
	406
	407	dma_cap_zero(mask);
	408	dma_cap_set(DMA_MEMCPY, mask);
	409	for (;;) {
	410	chan = dma_request_channel(mask, filter, NULL);
	411	if (chan) {
	412	err = dmatest_add_channel(chan);
	413	if (err == 0)
	414	continue;
	415	else {
	416	dma_release_channel(chan);
	417	break; /* add_channel failed, punt */
	418	}
	419	} else
	420	break; /* no more channels available */
	421	if (max_channels && nr_channels >= max_channels)
	422	break; /* we have all we need */
	423	}
4a776f0a	424
33df8ca0	425	return err;
4a776f0a	426	}
33df8ca0 DW	427	/* when compiled-in wait for drivers to load first */
33df8ca0 DW	428	late_initcall(dmatest_init);
4a776f0a HS	429
	430	static void __exit dmatest_exit(void)
	431	{
33df8ca0	432	struct dmatest_chan dtc, _dtc;
7cbd4877	433	struct dma_chan *chan;
33df8ca0 DW	434
	435	list_for_each_entry_safe(dtc, _dtc, &dmatest_channels, node) {
	436	list_del(&dtc->node);
7cbd4877	437	chan = dtc->chan;
33df8ca0 DW	438	dmatest_cleanup_channel(dtc);
33df8ca0 DW	439	pr_debug("dmatest: dropped channel %s\n",
7cbd4877 DW	440	dma_chan_name(chan));
7cbd4877 DW	441	dma_release_channel(chan);
33df8ca0	442	}
4a776f0a HS	443	}
	444	module_exit(dmatest_exit);
	445
	446	MODULE_AUTHOR("Haavard Skinnemoen <hskinnemoen@atmel.com>");
	447	MODULE_LICENSE("GPL v2");