/* Public domain. */

#ifndef _LINUX_DMA_BUF_MAP_H
#define _LINUX_DMA_BUF_MAP_H

#include <linux/io.h>
#include <linux/string.h>

struct dma_buf_map {
	union {
		void *vaddr_iomem;
		void *vaddr;
	};
	bool is_iomem;
};

static inline void
dma_buf_map_incr(struct dma_buf_map *dbm, size_t n)
{
	if (dbm->is_iomem)
		dbm->vaddr_iomem += n;
	else
		dbm->vaddr += n;
}

static inline void
dma_buf_map_memcpy_to(struct dma_buf_map *dbm, const void *src, size_t len)
{
	if (dbm->is_iomem)
		memcpy_toio(dbm->vaddr_iomem, src, len);
	else
		memcpy(dbm->vaddr, src, len);
}

static inline bool
dma_buf_map_is_null(const struct dma_buf_map *dbm)
{
	if (dbm->is_iomem)
		return (dbm->vaddr_iomem == NULL);
	else
		return (dbm->vaddr == NULL);
}

static inline bool
dma_buf_map_is_set(const struct dma_buf_map *dbm)
{
	if (dbm->is_iomem)
		return (dbm->vaddr_iomem != NULL);
	else
		return (dbm->vaddr != NULL);
}

static inline bool
dma_buf_map_is_equal(
    const struct dma_buf_map *dbm_a, const struct dma_buf_map *dbm_b)
{
	if (dbm_a->is_iomem != dbm_b->is_iomem)
		return (false);

	if (dbm_a->is_iomem)
		return (dbm_a->vaddr_iomem == dbm_b->vaddr_iomem);
	else
		return (dbm_a->vaddr == dbm_b->vaddr);
}

static inline void
dma_buf_map_clear(struct dma_buf_map *dbm)
{
	if (dbm->is_iomem) {
		dbm->vaddr_iomem = NULL;
		dbm->is_iomem = false;
	} else {
		dbm->vaddr = NULL;
	}
}

static inline void
dma_buf_map_set_vaddr_iomem(struct dma_buf_map *dbm, void *addr)
{
	dbm->vaddr_iomem = addr;
	dbm->is_iomem = true;
}

static inline void
dma_buf_map_set_vaddr(struct dma_buf_map *dbm, void *addr)
{
	dbm->vaddr = addr;
	dbm->is_iomem = false;
}

#endif