#include <test/fake.h>
#include <linux/io.h>

static struct fake_register_map_entry *fake_find_reg_action(
		struct fake_device *fd,
		phys_addr_t offset)
{
	struct fake_register_map_entry *reg_entry;
	struct test *test = fake_get_test(fd);

	for (reg_entry = fd->description->register_map;
	     reg_entry->valid_entry;
	     reg_entry++) {
		if (reg_entry->offset == offset)
			break;
	}

	ASSERT_NOT_ERR_OR_NULL(test, reg_entry);

	return reg_entry;
}

static u32 fake_readl(struct fake_device *fd, phys_addr_t offset)
{
	struct fake_register_map_entry *reg_entry;

	reg_entry = fake_find_reg_action(fd, offset);

	if (reg_entry->readl)
		return reg_entry->readl(fd);
	else
		return 0;
}

static void *fake_readl_action(struct mock_action *this,
			       const void **params,
			       int len)
{
	struct fake_device *fd = container_of(this,
					      struct fake_device,
					      readl_action);
	struct test *test = fake_get_test(fd);
	phys_addr_t offset;
	u32 *ret;

	ASSERT_EQ(test, len, 1);
	offset = CONVERT_TO_ACTUAL_TYPE(phys_addr_t, params[0]);
	ret = test_kzalloc(test, sizeof(*ret), GFP_KERNEL);
	*ret = fake_readl(fd, offset);

	return ret;
}

static void fake_writel(struct fake_device *fd, phys_addr_t offset, u32 value)
{
	struct fake_register_map_entry *reg_entry;

	reg_entry = fake_find_reg_action(fd, offset);

	if (reg_entry->writel)
		reg_entry->writel(fd, value);
}

static void *fake_writel_action(struct mock_action *this,
				const void **params,
				int len)
{
	struct fake_device *fd = container_of(this,
					      struct fake_device,
					      writel_action);
	struct test *test = fake_get_test(fd);
	phys_addr_t offset;
	u32 *ret, value;

	ASSERT_EQ(test, len, 2);
	value = CONVERT_TO_ACTUAL_TYPE(u32, params[0]);
	offset = CONVERT_TO_ACTUAL_TYPE(phys_addr_t, params[1]);
	ret = test_kzalloc(test, sizeof(*ret), GFP_KERNEL);
	*ret = 0;
	fake_writel(fd, offset, value);

	return ret;
}

void fake_device_init(struct test *test,
		      const struct fake_device_description *descr,
		      void *priv)
{
	struct fake_device *fd;

	fd = test_kzalloc(test, sizeof(*fd), GFP_KERNEL);

	fd->description = descr;
	fd->test = test;
	fd->priv = priv;
	fd->readl_action.do_action = fake_readl_action;
	fd->writel_action.do_action = fake_writel_action;

	mock_set_default_action(mock_get_global_mock(),
				"kunit_readl",
				kunit_readl,
				&fd->readl_action);
	mock_set_default_action(mock_get_global_mock(),
				"kunit_writel",
				kunit_writel,
				&fd->writel_action);
}