#include <assert.h>
#include <stdio.h>
#include "lib.h"

#define PTEH  0xFF000000
#define PTEL  0xFF000004
#define TTB   0xFF000008
#define TEA   0xFF00000C
#define MMUCR 0xFF000010
#define PTEA  0xFF000034

#define ITLB_ADDR(entry) (0xF2000000 + (entry<<8))
#define ITLB_DATA(entry) (0xF3000000 + (entry<<8))
#define UTLB_ADDR(entry) (0xF6000000 + (entry<<8))
#define UTLB_DATA1(entry) (0xF7000000 + (entry<<8))
#define UTLB_DATA2(entry) (0xF7800000 + (entry<<8))

/* Bang on the mmio side of the TLBs to make sure the bits
 * respond appropriately (with AT disabled so we don't risk
 * doing a hard crash) */
void test_tlb_mmio()
{
    int entry;
    for( entry=0; entry<64; entry++ ) {
	long_write( UTLB_DATA1(entry), 0 );
	long_write( UTLB_ADDR(entry), 0xFFFFFFFF );
	assert( long_read( UTLB_ADDR(entry) ) == 0xFFFFFFFF );
	assert( long_read( UTLB_DATA1(entry) ) == 0x00000104 );
	long_write( UTLB_ADDR(entry), 0x00000000 );
	assert( long_read( UTLB_ADDR(entry) ) == 0x00000000 );
	assert( long_read( UTLB_DATA1(entry) ) == 0x00000000 );
	long_write( UTLB_DATA1(entry), 0xFFFFFFFF );
	assert( long_read( UTLB_DATA1(entry) ) == 0x1FFFFDFF );
	assert( long_read( UTLB_ADDR(entry) ) == 0x00000300 );
	long_write( UTLB_DATA1(entry), 0x00000000 );
	assert( long_read( UTLB_DATA1(entry) ) == 0x00000000 );
	assert( long_read( UTLB_ADDR(entry) ) == 0x00000000 );
	long_write( UTLB_DATA2(entry), 0xFFFFFFFF );
	assert( long_read( UTLB_DATA2(entry) ) == 0x0000000F );
	long_write( UTLB_DATA2(entry), 0x00000000 );
	assert( long_read( UTLB_DATA2(entry) ) == 0x00000000 );
    }
    
    for( entry=0; entry<4; entry++ ) {
	long_write( ITLB_DATA(entry), 0 );
	long_write( ITLB_ADDR(entry), 0xFFFFFFFF );
	assert( long_read( ITLB_ADDR(entry) ) == 0xFFFFFDFF );
	assert( long_read( ITLB_DATA(entry) ) == 0x00000100 );
	long_write( ITLB_ADDR(entry), 0x00000000 );
	assert( long_read( ITLB_ADDR(entry) ) == 0x00000000 );
	assert( long_read( ITLB_DATA(entry) ) == 0x00000000 );
	long_write( ITLB_DATA(entry), 0xFFFFFFFF );
	assert( long_read( ITLB_DATA(entry) ) == 0x1FFFFDDA );
	assert( long_read( ITLB_ADDR(entry) ) == 0x00000100 );
	long_write( ITLB_DATA(entry), 0x00000000 );
	assert( long_read( ITLB_DATA(entry) ) == 0x00000000 );
	assert( long_read( ITLB_ADDR(entry) ) == 0x00000000 );
	
    }
}

int main( int argc, char *argv[] )
{
    test_tlb_mmio();
    return 0;
}