// SPDX-License-Identifier: GPL-2.0-only
/*
 * Based on arch/arm/kernel/io.c
 *
 * Copyright (C) 2012 ARM Ltd.
 */

#include <linux/export.h>
#include <linux/types.h>
#include <linux/io.h>

/*
 * This generates a memcpy that works on a from/to address which is aligned to
 * bits. Count is in terms of the number of bits sized quantities to copy. It
 * optimizes to use the STR groupings when possible so that it is WC friendly.
 */
#define memcpy_toio_aligned(to, from, count, bits)                        \
	({                                                                \
		volatile u##bits __iomem *_to = to;                       \
		const u##bits *_from = from;                              \
		size_t _count = count;                                    \
		const u##bits *_end_from = _from + ALIGN_DOWN(_count, 8); \
                                                                          \
		for (; _from < _end_from; _from += 8, _to += 8)           \
			__const_memcpy_toio_aligned##bits(_to, _from, 8); \
		if ((_count % 8) >= 4) {                                  \
			__const_memcpy_toio_aligned##bits(_to, _from, 4); \
			_from += 4;                                       \
			_to += 4;                                         \
		}                                                         \
		if ((_count % 4) >= 2) {                                  \
			__const_memcpy_toio_aligned##bits(_to, _from, 2); \
			_from += 2;                                       \
			_to += 2;                                         \
		}                                                         \
		if (_count % 2)                                           \
			__const_memcpy_toio_aligned##bits(_to, _from, 1); \
	})

void __iowrite64_copy_full(void __iomem *to, const void *from, size_t count)
{
	memcpy_toio_aligned(to, from, count, 64);
	dgh();
}
EXPORT_SYMBOL(__iowrite64_copy_full);

void __iowrite32_copy_full(void __iomem *to, const void *from, size_t count)
{
	memcpy_toio_aligned(to, from, count, 32);
	dgh();
}
EXPORT_SYMBOL(__iowrite32_copy_full);