importer.c

#include <linux/dma-buf.h>
#include <linux/module.h>
#include <linux/slab.h>
#include <kunit/test.h>
#include "importer.h"

extern struct dma_buf *dmabuf_exported;

static int importer_test_sg(struct dma_buf *dmabuf)
{
	struct dma_buf_attachment *attachment;
	struct sg_table *table;
	struct device *dev;
	unsigned int reg_addr, reg_size;

	if (!dmabuf)
		return -EINVAL;

	dev = kzalloc(sizeof(*dev), GFP_KERNEL);
	if (!dev)
		return -EINVAL;

	dev_set_name(dev, "importer");

	attachment = dma_buf_attach(dmabuf, dev);
	if (IS_ERR(attachment)) {
		print_err("dma_buf_attach() failed\n");
		return PTR_ERR(attachment);
	}

	table = dma_buf_map_attachment(attachment, DMA_BIDIRECTIONAL);
	if (IS_ERR(table)) {
		print_err("dma_buf_map_attachment() failed\n");
		dma_buf_detach(dmabuf, attachment);
	}

	reg_addr = sg_dma_address(table->sgl);
	reg_size = sg_dma_len(table->sgl);
	print_info("reg_addr = 0x%08x, reg_size = 0x%08x\n", reg_addr, reg_size);

	dma_buf_unmap_attachment(attachment, table, DMA_BIDIRECTIONAL);
	dma_buf_detach(dmabuf, attachment);
	kfree(dev);

	return 0;
}

static int importer_test_vmap(struct dma_buf *dmabuf)
{
	struct iosys_map map;

	if (!dmabuf) {
		print_err("dmabuf_exported is null\n");
		return -EINVAL;
	}

	dma_buf_vmap(dmabuf, &map);
	print_info("read from dmabuf vmap: %s\n", (char *)map.vaddr);
	dma_buf_vunmap(dmabuf, &map);

    return 0;
}

static void kunit_importer_test_sg(struct kunit *test)
{
	KUNIT_EXPECT_EQ(test, importer_test_sg(dmabuf_exported), 0);
}

static void kunit_importer_test_vmap(struct kunit *test)
{
	KUNIT_EXPECT_EQ(test, importer_test_vmap(dmabuf_exported), 0);
}

static struct kunit_case kunit_importer_test_cases[] = {
	KUNIT_CASE(kunit_importer_test_sg),
	KUNIT_CASE(kunit_importer_test_vmap),
	{},
};

static struct kunit_suite kunit_importer_test_suite = {
	.name = "kunit_importer_test",
	.test_cases = kunit_importer_test_cases,
};

static int __init importer_init(void)
{
	int ret;

	print_info("start\n");

	ret = kunit_run_tests(&kunit_importer_test_suite);
	if (ret)
		print_err("Kunit tests failed: %d\n", ret);
	else
		print_info("Kunit tests passed\n");

	print_info("end\n");
	return 0;
}

static void __exit importer_exit(void)
{
	print_info("start\n");
	print_info("end\n");
}

module_init(importer_init);
module_exit(importer_exit);


MODULE_AUTHOR("Jihyuk Park");
MODULE_IMPORT_NS(DMA_BUF);
MODULE_DESCRIPTION("DMA-BUF example for IMPORTER");
MODULE_LICENSE("GPL v2");