/*
 * Copyright (c) 2014-2019 François Tigeot <ftigeot@wolfpond.org>
 * All rights reserved.
 *
 * Redistribution and use in source and binary forms, with or without
 * modification, are permitted provided that the following conditions
 * are met:
 * 1. Redistributions of source code must retain the above copyright
 *    notice unmodified, this list of conditions, and the following
 *    disclaimer.
 * 2. Redistributions in binary form must reproduce the above copyright
 *    notice, this list of conditions and the following disclaimer in the
 *    documentation and/or other materials provided with the distribution.
 *
 * THIS SOFTWARE IS PROVIDED BY THE AUTHOR ``AS IS'' AND ANY EXPRESS OR
 * IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE IMPLIED WARRANTIES
 * OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE ARE DISCLAIMED.
 * IN NO EVENT SHALL THE AUTHOR BE LIABLE FOR ANY DIRECT, INDIRECT,
 * INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL DAMAGES (INCLUDING, BUT
 * NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS OR SERVICES; LOSS OF USE,
 * DATA, OR PROFITS; OR BUSINESS INTERRUPTION) HOWEVER CAUSED AND ON ANY
 * THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT
 * (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF
 * THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.
 */

#include <machine/pmap.h>
#include <vm/pmap.h>
#include <vm/vm.h>

#include <linux/kernel.h>
#include <linux/slab.h>
#include <linux/bug.h>
#include <asm/page.h>
#include <asm/io.h>

struct lock iomap_lock = LOCK_INITIALIZER("dlioml", 0, LK_CANRECURSE);

SLIST_HEAD(iomap_list_head, iomap) iomap_list = SLIST_HEAD_INITIALIZER(iomap_list);

void __iomem *
__ioremap_common(unsigned long phys_addr, unsigned long size, int cache_mode)
{
	struct iomap *imp;

	/* Ensure mappings are page-aligned */
	BUG_ON(phys_addr & PAGE_MASK);
	BUG_ON(size & PAGE_MASK);

	imp = kmalloc(sizeof(struct iomap), M_DRM, M_WAITOK);
	imp->paddr = phys_addr;
	imp->npages = size / PAGE_SIZE;
	imp->pmap_addr = pmap_mapdev_attr(phys_addr, size, cache_mode);
	lockmgr(&iomap_lock, LK_EXCLUSIVE);
	SLIST_INSERT_HEAD(&iomap_list, imp, im_iomaps);
	lockmgr(&iomap_lock, LK_RELEASE);

	return imp->pmap_addr;
}

void iounmap(void __iomem *ptr)
{
	struct iomap *imp, *tmp_imp;
	int found = 0;
	int indx;
	vm_paddr_t paddr_end;

	SLIST_FOREACH_MUTABLE(imp, &iomap_list, im_iomaps, tmp_imp) {
		if (imp->pmap_addr == ptr) {
			found = 1;
			break;
		}
	}

	if (!found) {
		kprintf("iounmap: invalid address %p\n", ptr);
		return;
	}

	paddr_end = imp->paddr + (imp->npages * PAGE_SIZE) - 1;
	/* Is this address space range backed by regular memory ? */
	for (indx = 0; phys_avail[indx].phys_end != 0; ++indx) {
		vm_paddr_t range_start = phys_avail[indx].phys_beg;
		vm_paddr_t size = phys_avail[indx].phys_end -
				  phys_avail[indx].phys_beg;
		vm_paddr_t range_end = range_start + size - 1;

		if ((imp->paddr >= range_start) && (paddr_end <= range_end)) {
			/* Yes, change page caching attributes */
			pmap_change_attr(imp->paddr, imp->npages, PAT_WRITE_BACK);
			break;
		}

	}

	pmap_unmapdev((vm_offset_t)imp->pmap_addr, imp->npages * PAGE_SIZE);

	lockmgr(&iomap_lock, LK_EXCLUSIVE);
	SLIST_REMOVE(&iomap_list, imp, iomap, im_iomaps);
	lockmgr(&iomap_lock, LK_RELEASE);

	kfree(imp);
}

#include <sys/memrange.h>

int
arch_io_reserve_memtype_wc(resource_size_t start, resource_size_t size)
{
	int act;
	struct mem_range_desc mrdesc;

	mrdesc.mr_base = start;
	mrdesc.mr_len = size;
	mrdesc.mr_flags = MDF_WRITECOMBINE;
	act = MEMRANGE_SET_UPDATE;
	strlcpy(mrdesc.mr_owner, "drm", sizeof(mrdesc.mr_owner));
	return mem_range_attr_set(&mrdesc, &act);
}