#include <vmm/vmm.h>
#include <vmm/paging.h>
#include <ubixos/spinlock.h>
#include <string.h>


int vmm_allocPageTable( uint32_t pdI, pidType pid ) {

	uint32_t *pageDirectory	= PD_BASE_ADDR;
	uint32_t *pageTable		= 0x0;

	if( ( pdI >= PD_ENTRIES ) || ( ( pageDirectory[pdI] & PAGE_PRESENT ) == PAGE_PRESENT ) ) {

		return( -1 );

	}


	/* Lock The Page Directory So We Dont Collide With Another Thread */
	//spinLock(&pdSpinLock);

	/* Map Page Table Page Into Page Directory */
	if( ( pdI >= PD_INDEX( VMM_USER_START ) ) && ( pdI <= PD_INDEX( VMM_USER_END ) ) ) {

		pageDirectory[pdI] = ( uint32_t ) vmm_findFreePage( pid ) | PAGE_DEFAULT;

	} else {

		pageDirectory[pdI] = ( uint32_t ) vmm_findFreePage( pid ) | KERNEL_PAGE_DEFAULT;

	}

	/* Map Page Table To Virtual Space So We Can Easily Manipulate It */
	pageTable = ( uint32_t * ) ( PT_BASE_ADDR + ( PD_INDEX( PT_BASE_ADDR ) * PAGE_SIZE ) );

	if( ( pageTable[pdI] & PAGE_PRESENT ) == PAGE_PRESENT ) {

		kpanic( "How did this happen" );

	}

	pageTable[pdI] =  pageDirectory[pdI];
  
	/* Reload Page Directory */
	asm(
			"movl %cr3,%eax\n"
			"movl %eax,%cr3\n"
	);

	/* Clean The Page */
	pageTable = ( uint32_t * ) (PT_BASE_ADDR + ( pdI * PAGE_SIZE ) );

	bzero( pageTable, PAGE_SIZE );

	//spinUnlock(&pdSpinLock);

	return( 0x0 );
  
}