/*
 * This file is subject to the terms and conditions of the GNU General Public
 * License.  See the file "COPYING" in the main directory of this archive
 * for more details.
 *
 * (C) Copyright 1995 1996 Linus Torvalds
 * (C) Copyright 2001, 2002 Ralf Baechle
 */
#include <linux/export.h>
#include <asm/addrspace.h>
#include <asm/byteorder.h>
#include <linux/ioport.h>
#include <linux/sched.h>
#include <linux/slab.h>
#include <linux/vmalloc.h>
#include <linux/mm_types.h>
#include <linux/io.h>
#include <asm/cacheflush.h>
#include <asm/tlbflush.h>
#include <ioremap.h>

#define IS_LOW512(addr) (!((phys_addr_t)(addr) & (phys_addr_t) ~0x1fffffffULL))
#define IS_KSEG1(addr) (((unsigned long)(addr) & ~0x1fffffffUL) == CKSEG1)

static int __ioremap_check_ram(unsigned long start_pfn, unsigned long nr_pages,
                               void *arg)
{
        unsigned long i;

        for (i = 0; i < nr_pages; i++) {
                if (pfn_valid(start_pfn + i) &&
                    !PageReserved(pfn_to_page(start_pfn + i)))
                        return 1;
        }

        return 0;
}

/*
 * ioremap_prot     -   map bus memory into CPU space
 * @phys_addr:    bus address of the memory
 * @size:      size of the resource to map
 *
 * ioremap_prot gives the caller control over cache coherency attributes (CCA)
 */
void __iomem *ioremap_prot(phys_addr_t phys_addr, unsigned long size,
                           pgprot_t prot)
{
        unsigned long flags = pgprot_val(prot) & _CACHE_MASK;
        unsigned long offset, pfn, last_pfn;
        struct vm_struct *area;
        phys_addr_t last_addr;
        unsigned long vaddr;
        void __iomem *cpu_addr;

        cpu_addr = plat_ioremap(phys_addr, size, flags);
        if (cpu_addr)
                return cpu_addr;

        phys_addr = fixup_bigphys_addr(phys_addr, size);

        /* Don't allow wraparound or zero size */
        last_addr = phys_addr + size - 1;
        if (!size || last_addr < phys_addr)
                return NULL;

        /*
         * Map uncached objects in the low 512mb of address space using KSEG1,
         * otherwise map using page tables.
         */
        if (IS_LOW512(phys_addr) && IS_LOW512(last_addr) &&
            flags == _CACHE_UNCACHED)
                return (void __iomem *) CKSEG1ADDR(phys_addr);

        /* Early remaps should use the unmapped regions til' VM is available */
        if (WARN_ON_ONCE(!slab_is_available()))
                return NULL;

        /*
         * Don't allow anybody to remap RAM that may be allocated by the page
         * allocator, since that could lead to races & data clobbering.
         */
        pfn = PFN_DOWN(phys_addr);
        last_pfn = PFN_DOWN(last_addr);
        if (walk_system_ram_range(pfn, last_pfn - pfn + 1, NULL,
                                  __ioremap_check_ram) == 1) {
                WARN_ONCE(1, "ioremap on RAM at %pa - %pa\n",
                          &phys_addr, &last_addr);
                return NULL;
        }

        /*
         * Mappings have to be page-aligned
         */
        offset = phys_addr & ~PAGE_MASK;
        phys_addr &= PAGE_MASK;
        size = PAGE_ALIGN(last_addr + 1) - phys_addr;

        /*
         * Ok, go for it..
         */
        area = get_vm_area(size, VM_IOREMAP);
        if (!area)
                return NULL;
        vaddr = (unsigned long)area->addr;

        flags |= _PAGE_GLOBAL | _PAGE_PRESENT | __READABLE | __WRITEABLE;
        if (ioremap_page_range(vaddr, vaddr + size, phys_addr,
                        __pgprot(flags))) {
                free_vm_area(area);
                return NULL;
        }

        return (void __iomem *)(vaddr + offset);
}
EXPORT_SYMBOL(ioremap_prot);

void iounmap(const volatile void __iomem *addr)
{
        if (!plat_iounmap(addr) && !IS_KSEG1(addr))
                vunmap((void *)((unsigned long)addr & PAGE_MASK));
}
EXPORT_SYMBOL(iounmap);