dmaengine: dmatest: move test data alloc & free into functions

This patch starts to take advantage of the `dmatest_data` struct by moving
the common allocation & free-ing bits into functions.

Signed-off-by: Alexandru Ardelean <alexandru.ardelean@analog.com>
Signed-off-by: Vinod Koul <vkoul@kernel.org>
This commit is contained in:
Alexandru Ardelean 2019-02-12 17:11:39 +02:00 committed by Vinod Koul
parent 41d00bb7a6
commit 3b6679f91e
1 changed files with 55 additions and 55 deletions

View File

@ -486,6 +486,53 @@ static unsigned long long dmatest_KBs(s64 runtime, unsigned long long len)
return FIXPT_TO_INT(dmatest_persec(runtime, len >> 10));
}
static void __dmatest_free_test_data(struct dmatest_data *d, unsigned int cnt)
{
unsigned int i;
for (i = 0; i < cnt; i++)
kfree(d->raw[i]);
kfree(d->aligned);
kfree(d->raw);
}
static void dmatest_free_test_data(struct dmatest_data *d)
{
__dmatest_free_test_data(d, d->cnt);
}
static int dmatest_alloc_test_data(struct dmatest_data *d,
unsigned int buf_size, u8 align)
{
unsigned int i = 0;
d->raw = kcalloc(d->cnt + 1, sizeof(u8 *), GFP_KERNEL);
if (!d->raw)
return -ENOMEM;
d->aligned = kcalloc(d->cnt + 1, sizeof(u8 *), GFP_KERNEL);
if (!d->aligned)
goto err;
for (i = 0; i < d->cnt; i++) {
d->raw[i] = kmalloc(buf_size + align, GFP_KERNEL);
if (!d->raw[i])
goto err;
/* align to alignment restriction */
if (align)
d->aligned[i] = PTR_ALIGN(d->raw[i], align);
else
d->aligned[i] = d->raw[i];
}
return 0;
err:
__dmatest_free_test_data(d, i);
return -ENOMEM;
}
/*
* This function repeatedly tests DMA transfers of various lengths and
* offsets for a given operation type until it is told to exit by
@ -588,55 +635,17 @@ static int dmatest_func(void *data)
goto err_free_coefs;
}
src->aligned = kcalloc(src->cnt + 1, sizeof(u8 *), GFP_KERNEL);
if (!src->aligned)
if (dmatest_alloc_test_data(src, buf_size, align) < 0)
goto err_free_coefs;
src->raw = kcalloc(src->cnt + 1, sizeof(u8 *), GFP_KERNEL);
if (!src->raw)
goto err_usrcs;
for (i = 0; i < src->cnt; i++) {
src->raw[i] = kmalloc(buf_size + align,
GFP_KERNEL);
if (!src->raw[i])
goto err_srcbuf;
/* align srcs to alignment restriction */
if (align)
src->aligned[i] = PTR_ALIGN(src->raw[i], align);
else
src->aligned[i] = src->raw[i];
}
src->aligned[i] = NULL;
dst->aligned = kcalloc(dst->cnt + 1, sizeof(u8 *), GFP_KERNEL);
if (!dst->aligned)
goto err_dsts;
dst->raw = kcalloc(dst->cnt + 1, sizeof(u8 *), GFP_KERNEL);
if (!dst->raw)
goto err_udsts;
for (i = 0; i < dst->cnt; i++) {
dst->raw[i] = kmalloc(buf_size + align,
GFP_KERNEL);
if (!dst->raw[i])
goto err_dstbuf;
/* align dsts to alignment restriction */
if (align)
dst->aligned[i] = PTR_ALIGN(dst->raw[i], align);
else
dst->aligned[i] = dst->raw[i];
}
dst->aligned[i] = NULL;
if (dmatest_alloc_test_data(dst, buf_size, align) < 0)
goto err_src;
set_user_nice(current, 10);
srcs = kcalloc(src->cnt, sizeof(dma_addr_t), GFP_KERNEL);
if (!srcs)
goto err_dstbuf;
goto err_dst;
dma_pq = kcalloc(dst->cnt, sizeof(dma_addr_t), GFP_KERNEL);
if (!dma_pq)
@ -865,19 +874,10 @@ static int dmatest_func(void *data)
kfree(dma_pq);
err_srcs_array:
kfree(srcs);
err_dstbuf:
for (i = 0; dst->raw[i]; i++)
kfree(dst->raw[i]);
kfree(dst->raw);
err_udsts:
kfree(dst->aligned);
err_dsts:
err_srcbuf:
for (i = 0; src->raw[i]; i++)
kfree(src->raw[i]);
kfree(src->raw);
err_usrcs:
kfree(src->aligned);
err_dst:
dmatest_free_test_data(dst);
err_src:
dmatest_free_test_data(src);
err_free_coefs:
kfree(pq_coefs);
err_thread_type: