/**

  * $Id$

  * 

  * Translation cache management. This part is architecture independent.

  *

  * Copyright (c) 2005 Nathan Keynes.

  *

  * This program is free software; you can redistribute it and/or modify

  * it under the terms of the GNU General Public License as published by

  * the Free Software Foundation; either version 2 of the License, or

  * (at your option) any later version.

  *

  * This program is distributed in the hope that it will be useful,

  * but WITHOUT ANY WARRANTY; without even the implied warranty of

  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the

  * GNU General Public License for more details.

  */

 #include <sys/types.h>

 #include <sys/mman.h>

 #include <assert.h>

 #include "dreamcast.h"

 #include "sh4/sh4core.h"

 #include "sh4/xltcache.h"

 #include "x86dasm/x86dasm.h"

 #define XLAT_LUT_PAGE_BITS 12

 #define XLAT_LUT_TOTAL_BITS 28

 #define XLAT_LUT_PAGE(addr) (((addr)>>13) & 0xFFFF)

 #define XLAT_LUT_ENTRY(addr) (((addr)&0x1FFE) >> 1)

 #define XLAT_LUT_PAGES (1<<(XLAT_LUT_TOTAL_BITS-XLAT_LUT_PAGE_BITS))

 #define XLAT_LUT_PAGE_ENTRIES (1<<XLAT_LUT_PAGE_BITS)

 #define XLAT_LUT_PAGE_SIZE (XLAT_LUT_PAGE_ENTRIES * sizeof(void *))

 #define XLAT_LUT_ENTRY_EMPTY (void *)0

 #define XLAT_LUT_ENTRY_USED  (void *)1

 #define NEXT(block) ( (xlat_cache_block_t)&((block)->code[(block)->size]))

 #define BLOCK_FOR_CODE(code) (((xlat_cache_block_t)code)-1)

 #define IS_ENTRY_POINT(ent) (ent > XLAT_LUT_ENTRY_USED)

 #define IS_ENTRY_USED(ent) (ent != XLAT_LUT_ENTRY_EMPTY)

 #define MIN_BLOCK_SIZE 32

 #define MIN_TOTAL_SIZE (sizeof(struct xlat_cache_block)+MIN_BLOCK_SIZE)

 #define BLOCK_INACTIVE 0

 #define BLOCK_ACTIVE 1

 #define BLOCK_USED 2

 xlat_cache_block_t xlat_new_cache;

 xlat_cache_block_t xlat_new_cache_ptr;

 xlat_cache_block_t xlat_new_create_ptr;

 xlat_cache_block_t xlat_temp_cache;

 xlat_cache_block_t xlat_temp_cache_ptr;

 xlat_cache_block_t xlat_old_cache;

 xlat_cache_block_t xlat_old_cache_ptr;

 static void ***xlat_lut;

 static gboolean xlat_initialized = FALSE;

 void xlat_cache_init(void) 

 {

     if( !xlat_initialized ) {

 	xlat_initialized = TRUE;

 	xlat_new_cache = mmap( NULL, XLAT_NEW_CACHE_SIZE, PROT_EXEC|PROT_READ|PROT_WRITE,

 			       MAP_PRIVATE|MAP_ANON, -1, 0 );

 	xlat_temp_cache = mmap( NULL, XLAT_TEMP_CACHE_SIZE, PROT_EXEC|PROT_READ|PROT_WRITE,

 				MAP_PRIVATE|MAP_ANON, -1, 0 );

 	xlat_old_cache = mmap( NULL, XLAT_OLD_CACHE_SIZE, PROT_EXEC|PROT_READ|PROT_WRITE,

 			       MAP_PRIVATE|MAP_ANON, -1, 0 );

 	xlat_new_cache_ptr = xlat_new_cache;

 	xlat_temp_cache_ptr = xlat_temp_cache;

 	xlat_old_cache_ptr = xlat_old_cache;

 	xlat_new_create_ptr = xlat_new_cache;

 	xlat_lut = mmap( NULL, XLAT_LUT_PAGES*sizeof(void *), PROT_READ|PROT_WRITE,

 			 MAP_PRIVATE|MAP_ANON, -1, 0);

 	memset( xlat_lut, 0, XLAT_LUT_PAGES*sizeof(void *) );

     }

     xlat_flush_cache();

 }

 void xlat_print_free( FILE *out )

 {

     fprintf( out, "New space: %d\nTemp space: %d\nOld space: %d\n", 

 	     xlat_new_cache_ptr->size, xlat_temp_cache_ptr->size, xlat_old_cache_ptr->size );

 }

 /**

  * Reset the cache structure to its default state

  */

 void xlat_flush_cache() 

 {

     xlat_cache_block_t tmp;

     int i;

     xlat_new_cache_ptr = xlat_new_cache;

     xlat_new_cache_ptr->active = 0;

     xlat_new_cache_ptr->size = XLAT_NEW_CACHE_SIZE - 2*sizeof(struct xlat_cache_block);

     tmp = NEXT(xlat_new_cache_ptr);

     tmp->active = 1;

     tmp->size = 0;

     xlat_temp_cache_ptr = xlat_temp_cache;

     xlat_temp_cache_ptr->active = 0;

     xlat_temp_cache_ptr->size = XLAT_TEMP_CACHE_SIZE - 2*sizeof(struct xlat_cache_block);

     tmp = NEXT(xlat_temp_cache_ptr);

     tmp->active = 1;

     tmp->size = 0;

     xlat_old_cache_ptr = xlat_old_cache;

     xlat_old_cache_ptr->active = 0;

     xlat_old_cache_ptr->size = XLAT_OLD_CACHE_SIZE - 2*sizeof(struct xlat_cache_block);

     tmp = NEXT(xlat_old_cache_ptr);

     tmp->active = 1;

     tmp->size = 0;

     for( i=0; i<XLAT_LUT_PAGES; i++ ) {

 	if( xlat_lut[i] != NULL ) {

 	    memset( xlat_lut[i], 0, XLAT_LUT_PAGE_SIZE );

 	}

     }

 }

 static void xlat_flush_page_by_lut( void **page )

 {

     int i;

     for( i=0; i<XLAT_LUT_PAGE_ENTRIES; i++ ) {

 	if( IS_ENTRY_POINT(page[i]) ) {

 	    BLOCK_FOR_CODE(page[i])->active = 0;

 	}

 	page[i] = NULL;

     }

 }

 void xlat_invalidate_word( sh4addr_t addr )

 {

     if( xlat_lut ) {

 	void **page = xlat_lut[XLAT_LUT_PAGE(addr)];

 	if( page != NULL ) {

 	    int entry = XLAT_LUT_ENTRY(addr);

 	    if( page[entry] != NULL ) {

 		xlat_flush_page_by_lut(page);

 	    }

 	}

     }

 }

 void xlat_invalidate_long( sh4addr_t addr )

 {

     if( xlat_lut ) {

 	void **page = xlat_lut[XLAT_LUT_PAGE(addr)];

 	if( page != NULL ) {

 	    int entry = XLAT_LUT_ENTRY(addr);

 	    if( page[entry] != NULL || page[entry+1] != NULL ) {

 		xlat_flush_page_by_lut(page);

 	    }

 	}

     }

 }

 void xlat_invalidate_block( sh4addr_t address, size_t size )

 {

     int i;

     int entry_count = size >> 1; // words;

     uint32_t page_no = XLAT_LUT_PAGE(address);

     int entry = XLAT_LUT_ENTRY(address);

     if( xlat_lut ) {

 	do {

 	    void **page = xlat_lut[page_no];

 	    int page_entries = XLAT_LUT_PAGE_ENTRIES - entry;

 	    if( entry_count < page_entries ) {

 		page_entries = entry_count;

 	    }

 	    if( page != NULL ) {

 		if( page_entries == XLAT_LUT_PAGE_ENTRIES ) {

 		    /* Overwriting the entire page anyway */

 		    xlat_flush_page_by_lut(page);

 		} else {

 		    for( i=entry; i<entry+page_entries; i++ ) {

 			if( page[i] != NULL ) {

 			    xlat_flush_page_by_lut(page);

 			    break;

 			}

 		    }

 		}

 		entry_count -= page_entries;

 	    }

 	    page_no ++;

 	    entry_count -= page_entries;

 	    entry = 0;

 	} while( entry_count > 0 );

     }

 }

 void xlat_flush_page( sh4addr_t address )

 {

     void **page = xlat_lut[XLAT_LUT_PAGE(address)];

     if( page != NULL ) {

 	xlat_flush_page_by_lut(page);

     }

 }

 void *xlat_get_code( sh4addr_t address )

 {

     void *result = NULL;

     void **page = xlat_lut[XLAT_LUT_PAGE(address)];

     if( page != NULL ) {

 	result = (void *)(((uintptr_t)(page[XLAT_LUT_ENTRY(address)])) & (~((uintptr_t)0x03)));

     }

     return result;

 }

 xlat_recovery_record_t xlat_get_recovery( void *code, void *native_pc, gboolean recover_after )

 {

     if( code != NULL ) {

 	uintptr_t pc_offset = ((uint8_t *)native_pc) - ((uint8_t *)code);

 	xlat_cache_block_t block = BLOCK_FOR_CODE(code);

 	uint32_t count = block->recover_table_size;

 	xlat_recovery_record_t records = (xlat_recovery_record_t)(&block->code[block->recover_table_offset]);

 	uint32_t posn;

 	if( recover_after ) {

 	    if( records[count-1].xlat_offset < pc_offset ) {

 		return NULL;

 	    }

 	    for( posn=count-1; posn > 0; posn-- ) {

 		if( records[posn-1].xlat_offset < pc_offset ) {

 		    return &records[posn];

 		}

 	    }

 	    return &records[0]; // shouldn't happen

 	} else {

 	    for( posn = 1; posn < count; posn++ ) {

 		if( records[posn].xlat_offset >= pc_offset ) {

 		    return &records[posn-1];

 		}

 	    }

 	    return &records[count-1];

 	}

     }

     return NULL;

 }

 void **xlat_get_lut_entry( sh4addr_t address )

 {

     void **page = xlat_lut[XLAT_LUT_PAGE(address)];

     /* Add the LUT entry for the block */

     if( page == NULL ) {

 	xlat_lut[XLAT_LUT_PAGE(address)] = page =

 	    mmap( NULL, XLAT_LUT_PAGE_SIZE, PROT_READ|PROT_WRITE,

 		  MAP_PRIVATE|MAP_ANON, -1, 0 );

 	memset( page, 0, XLAT_LUT_PAGE_SIZE );

     }

     return &page[XLAT_LUT_ENTRY(address)];

 }

 uint32_t xlat_get_block_size( void *block )

 {

     xlat_cache_block_t xlt = (xlat_cache_block_t)(((char *)block)-sizeof(struct xlat_cache_block));

     return xlt->size;

 }

 uint32_t xlat_get_code_size( void *block )

 {

     xlat_cache_block_t xlt = (xlat_cache_block_t)(((char *)block)-sizeof(struct xlat_cache_block));

     if( xlt->recover_table_offset == 0 ) {

 	return xlt->size;

     } else {

 	return xlt->recover_table_offset;

     }

 }

 /**

  * Cut the specified block so that it has the given size, with the remaining data

  * forming a new free block. If the free block would be less than the minimum size,

  * the cut is not performed.

  * @return the next block after the (possibly cut) block.

  */

 static inline xlat_cache_block_t xlat_cut_block( xlat_cache_block_t block, int cutsize )

 {

     cutsize = (cutsize + 3) & 0xFFFFFFFC; // force word alignment

     assert( cutsize <= block->size );

     if( block->size > cutsize + MIN_TOTAL_SIZE ) {

 	int oldsize = block->size;

 	block->size = cutsize;

 	xlat_cache_block_t next = NEXT(block);

 	next->active = 0;

 	next->size = oldsize - cutsize - sizeof(struct xlat_cache_block);

 	return next;

     } else {

 	return NEXT(block);

     }

 }

 /**

  * Promote a block in temp space (or elsewhere for that matter) to old space.

  *

  * @param block to promote.

  */

 static void xlat_promote_to_old_space( xlat_cache_block_t block )

 {

     int allocation = -sizeof(struct xlat_cache_block);

     int size = block->size;

     xlat_cache_block_t curr = xlat_old_cache_ptr;

     xlat_cache_block_t start_block = curr;

     do {

 	allocation += curr->size + sizeof(struct xlat_cache_block);

 	curr = NEXT(curr);

 	if( allocation > size ) {

 	    break; /* done */

 	}

 	if( curr->size == 0 ) { /* End-of-cache Sentinel */

 	    /* Leave what we just released as free space and start again from the

 	     * top of the cache

 	     */

 	    start_block->active = 0;

 	    start_block->size = allocation;

 	    allocation = -sizeof(struct xlat_cache_block);

 	    start_block = curr = xlat_old_cache;

 	}

     } while(1);

     start_block->active = 1;

     start_block->size = allocation;

     start_block->lut_entry = block->lut_entry;

     start_block->recover_table_offset = block->recover_table_offset;

     start_block->recover_table_size = block->recover_table_size;

     *block->lut_entry = &start_block->code;

     memcpy( start_block->code, block->code, block->size );

     xlat_old_cache_ptr = xlat_cut_block(start_block, size );

     if( xlat_old_cache_ptr->size == 0 ) {

 	xlat_old_cache_ptr = xlat_old_cache;

     }

 }

 /**

  * Similarly to the above method, promotes a block to temp space.

  * TODO: Try to combine these - they're nearly identical

  */

 void xlat_promote_to_temp_space( xlat_cache_block_t block )

 {

     int size = block->size;

     int allocation = -sizeof(struct xlat_cache_block);

     xlat_cache_block_t curr = xlat_temp_cache_ptr;

     xlat_cache_block_t start_block = curr;

     do {

 	if( curr->active == BLOCK_USED ) {

 	    xlat_promote_to_old_space( curr );

 	} else if( curr->active == BLOCK_ACTIVE ) {

 	    // Active but not used, release block

 	   *((uintptr_t *)curr->lut_entry) &= ((uintptr_t)0x03);

 	}

 	allocation += curr->size + sizeof(struct xlat_cache_block);

 	curr = NEXT(curr);

 	if( allocation > size ) {

 	    break; /* done */

 	}

 	if( curr->size == 0 ) { /* End-of-cache Sentinel */

 	    /* Leave what we just released as free space and start again from the

 	     * top of the cache

 	     */

 	    start_block->active = 0;

 	    start_block->size = allocation;

 	    allocation = -sizeof(struct xlat_cache_block);

 	    start_block = curr = xlat_temp_cache;

 	}

     } while(1);

     start_block->active = 1;

     start_block->size = allocation;

     start_block->lut_entry = block->lut_entry;

     start_block->recover_table_offset = block->recover_table_offset;

     start_block->recover_table_size = block->recover_table_size;

     *block->lut_entry = &start_block->code;

     memcpy( start_block->code, block->code, block->size );

     xlat_temp_cache_ptr = xlat_cut_block(start_block, size );

     if( xlat_temp_cache_ptr->size == 0 ) {

 	xlat_temp_cache_ptr = xlat_temp_cache;

     }

 }

 /**

  * Returns the next block in the new cache list that can be written to by the

  * translator. If the next block is active, it is evicted first.

  */

 xlat_cache_block_t xlat_start_block( sh4addr_t address )

 {

     if( xlat_new_cache_ptr->size == 0 ) {

 	xlat_new_cache_ptr = xlat_new_cache;

     }

     if( xlat_new_cache_ptr->active ) {

 	xlat_promote_to_temp_space( xlat_new_cache_ptr );

     }

     xlat_new_create_ptr = xlat_new_cache_ptr;

     xlat_new_create_ptr->active = 1;

     xlat_new_cache_ptr = NEXT(xlat_new_cache_ptr);

     /* Add the LUT entry for the block */

     if( xlat_lut[XLAT_LUT_PAGE(address)] == NULL ) {

 	xlat_lut[XLAT_LUT_PAGE(address)] =

 	    mmap( NULL, XLAT_LUT_PAGE_SIZE, PROT_READ|PROT_WRITE,

 		  MAP_PRIVATE|MAP_ANON, -1, 0 );

 	memset( xlat_lut[XLAT_LUT_PAGE(address)], 0, XLAT_LUT_PAGE_SIZE );

     }

     if( IS_ENTRY_POINT(xlat_lut[XLAT_LUT_PAGE(address)][XLAT_LUT_ENTRY(address)]) ) {

 	xlat_cache_block_t oldblock = BLOCK_FOR_CODE(xlat_lut[XLAT_LUT_PAGE(address)][XLAT_LUT_ENTRY(address)]);

 	oldblock->active = 0;

     }

     xlat_lut[XLAT_LUT_PAGE(address)][XLAT_LUT_ENTRY(address)] = 

 	&xlat_new_create_ptr->code;

     xlat_new_create_ptr->lut_entry = xlat_lut[XLAT_LUT_PAGE(address)] + XLAT_LUT_ENTRY(address);

     return xlat_new_create_ptr;

 }

 xlat_cache_block_t xlat_extend_block( uint32_t newSize )

 {

     while( xlat_new_create_ptr->size < newSize ) {

 	if( xlat_new_cache_ptr->size == 0 ) {

 	    /* Migrate to the front of the cache to keep it contiguous */

 	    xlat_new_create_ptr->active = 0;

 	    sh4ptr_t olddata = xlat_new_create_ptr->code;

 	    int oldsize = xlat_new_create_ptr->size;

 	    int size = oldsize + MIN_BLOCK_SIZE; /* minimum expansion */

 	    void **lut_entry = xlat_new_create_ptr->lut_entry;

 	    int allocation = -sizeof(struct xlat_cache_block);

 	    xlat_new_cache_ptr = xlat_new_cache;

 	    do {

 		if( xlat_new_cache_ptr->active ) {

 		    xlat_promote_to_temp_space( xlat_new_cache_ptr );

 		}

 		allocation += xlat_new_cache_ptr->size + sizeof(struct xlat_cache_block);

 		xlat_new_cache_ptr = NEXT(xlat_new_cache_ptr);

 	    } while( allocation < size );

 	    xlat_new_create_ptr = xlat_new_cache;

 	    xlat_new_create_ptr->active = 1;

 	    xlat_new_create_ptr->size = allocation;

 	    xlat_new_create_ptr->lut_entry = lut_entry;

 	    *lut_entry = &xlat_new_create_ptr->code;

 	    memmove( xlat_new_create_ptr->code, olddata, oldsize );

 	} else {

 	    if( xlat_new_cache_ptr->active ) {

 		xlat_promote_to_temp_space( xlat_new_cache_ptr );

 	    }

 	    xlat_new_create_ptr->size += xlat_new_cache_ptr->size + sizeof(struct xlat_cache_block);

 	    xlat_new_cache_ptr = NEXT(xlat_new_cache_ptr);

 	}

     }

     return xlat_new_create_ptr;

 }

 void xlat_commit_block( uint32_t destsize, uint32_t srcsize )

 {

     void **ptr = xlat_new_create_ptr->lut_entry;

     void **endptr = ptr + (srcsize>>2);

     while( ptr < endptr ) {

 	if( *ptr == NULL ) {

 	    *ptr = XLAT_LUT_ENTRY_USED;

 	}

 	ptr++;

     }

     xlat_new_cache_ptr = xlat_cut_block( xlat_new_create_ptr, destsize );

 }

 void xlat_delete_block( xlat_cache_block_t block ) 

 {

     block->active = 0;

     *block->lut_entry = NULL;

 }

 void xlat_check_cache_integrity( xlat_cache_block_t cache, xlat_cache_block_t ptr, int size )

 {

     int foundptr = 0;

     xlat_cache_block_t tail = 

 	(xlat_cache_block_t)(((char *)cache) + size - sizeof(struct xlat_cache_block));

     assert( tail->active == 1 );

     assert( tail->size == 0 ); 

     while( cache < tail ) {

 	assert( cache->active >= 0 && cache->active <= 2 );

 	assert( cache->size >= 0 && cache->size < size );

 	if( cache == ptr ) {

 	    foundptr = 1;

 	}

 	cache = NEXT(cache);

     }

     assert( cache == tail );

     assert( foundptr == 1 || tail == ptr );

 }

 void xlat_check_integrity( )

 {

     xlat_check_cache_integrity( xlat_new_cache, xlat_new_cache_ptr, XLAT_NEW_CACHE_SIZE );

     xlat_check_cache_integrity( xlat_temp_cache, xlat_temp_cache_ptr, XLAT_TEMP_CACHE_SIZE );

     xlat_check_cache_integrity( xlat_old_cache, xlat_old_cache_ptr, XLAT_OLD_CACHE_SIZE );

 }