/*-
 * Copyright (C) 2010 Nathan Whitehorn
 * 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, 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 TOOLS GMBH 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 <sys/cdefs.h>
__FBSDID("$FreeBSD$");

#include <stand.h>
#include <stdint.h>

#define _KERNEL
#include <machine/cpufunc.h>
#include <machine/psl.h>
#include <machine/pte.h>
#include <machine/slb.h>
#include <machine/param.h>

#include "bootstrap.h"
#include "lv1call.h"
#include "ps3.h"

register_t pteg_count, pteg_mask;
uint64_t as_id;
uint64_t virtual_avail;

int
ps3mmu_map(uint64_t va, uint64_t pa)
{
	struct lpte pt;
	int shift;
	uint64_t vsid, ptegidx;
	
	if (pa < 0x8000000) { /* Phys mem? */
		pt.pte_hi = LPTE_BIG;
		pt.pte_lo = LPTE_M;
		shift = 24;
		vsid = 0;
	} else {
		pt.pte_hi = 0;
		pt.pte_lo = LPTE_I | LPTE_G | LPTE_M | LPTE_NOEXEC;
		shift = ADDR_PIDX_SHFT;
		vsid = 1;
	}

	pt.pte_hi |= (vsid << LPTE_VSID_SHIFT) |
            (((uint64_t)(va & ADDR_PIDX) >> ADDR_API_SHFT64) & LPTE_API);
	pt.pte_lo |= pa;
	ptegidx = vsid ^ (((uint64_t)va & ADDR_PIDX) >> shift);

	pt.pte_hi |= LPTE_LOCKED | LPTE_VALID;
	ptegidx &= pteg_mask;

	return (lv1_insert_pte(ptegidx, &pt, LPTE_LOCKED));
}

void *
ps3mmu_mapdev(uint64_t pa, size_t length)
{
	uint64_t spa;
	void *mapstart;
	int err;
	
	mapstart = (void *)(uintptr_t)virtual_avail;

	for (spa = pa; spa < pa + length; spa += PAGE_SIZE) {
		err = ps3mmu_map(virtual_avail, spa);
		virtual_avail += PAGE_SIZE;
		if (err != 0)
			return (NULL);
	}

	return (mapstart);
}

int
ps3mmu_init(int maxmem)
{
	uint64_t ptsize;
	int i;

	i = lv1_setup_address_space(&as_id, &ptsize);
	pteg_count = ptsize / sizeof(struct lpteg);
	pteg_mask = pteg_count - 1;

	for (i = 0; i < maxmem; i += 16*1024*1024)
		ps3mmu_map(i,i);

	virtual_avail = 0x10000000;

	__asm __volatile ("slbia; slbmte %0, %1; slbmte %2,%3" ::
	    "r"((0 << SLBV_VSID_SHIFT) | SLBV_L), "r"(0 | SLBE_VALID),
	    "r"(1 << SLBV_VSID_SHIFT),
	    "r"((1 << SLBE_ESID_SHIFT) | SLBE_VALID | 1));

	mtmsr(mfmsr() | PSL_IR | PSL_DR | PSL_RI | PSL_ME);

	return (0);
}