X-Git-Url: https://git.ucc.asn.au/?a=blobdiff_plain;f=Kernel%2Farch%2Fx86%2Fmm_virt.c;h=72c40c2a84b7be751ab2ddbfc1b5ef98307ed586;hb=046980a1afd982af07dc97aaadb186e31eebb340;hp=dce4782867df7c8bfac5033c9e9bb03b2c4454d5;hpb=7d881c2e5fef91a6570e46ef69a5d4a5cf0e8b4d;p=tpg%2Facess2.git

diff --git a/Kernel/arch/x86/mm_virt.c b/Kernel/arch/x86/mm_virt.c
index dce47828..72c40c2a 100644
--- a/Kernel/arch/x86/mm_virt.c
+++ b/Kernel/arch/x86/mm_virt.c
@@ -56,7 +56,7 @@
 #define	PF_WRITE	0x2
 #define	PF_USER		0x4
 #define	PF_COW		0x200
-#define	PF_PAGED	0x400
+#define	PF_NOPAGE	0x400
 
 #define INVLPG(addr)	__asm__ __volatile__ ("invlpg (%0)"::"r"(addr))
 
@@ -78,6 +78,7 @@ void	MM_PreinitVirtual(void);
 void	MM_InstallVirtual(void);
 void	MM_PageFault(tVAddr Addr, Uint ErrorCode, tRegs *Regs);
 void	MM_DumpTables(tVAddr Start, tVAddr End);
+tVAddr	MM_ClearUser(void);
 tPAddr	MM_DuplicatePage(tVAddr VAddr);
 
 // === GLOBALS ===
@@ -217,6 +218,9 @@ void MM_PageFault(tVAddr Addr, Uint ErrorCode, tRegs *Regs)
 			);
 		Warning("User Pagefault: Instruction at %04x:%08x accessed %p", Regs->cs, Regs->eip, Addr);
 		__asm__ __volatile__ ("sti");	// Restart IRQs
+		#if 1
+		Error_Backtrace(Regs->eip, Regs->ebp);
+		#endif
 		Threads_SegFault(Addr);
 		return ;
 	}
@@ -306,7 +310,7 @@ void MM_DumpTables(tVAddr Start, tVAddr End)
 					rangeStart, curPos - 1,
 					gaPageTable[rangeStart>>12] & ~0xFFF,
 					(expected & ~0xFFF) - 1,
-					(expected & PF_PAGED ? "p" : "-"),
+					(expected & PF_NOPAGE ? "P" : "-"),
 					(expected & PF_COW ? "C" : "-"),
 					(expected & PF_USER ? "U" : "-"),
 					(expected & PF_WRITE ? "W" : "-")
@@ -327,7 +331,7 @@ void MM_DumpTables(tVAddr Start, tVAddr End)
 			rangeStart, curPos - 1,
 			gaPageTable[rangeStart>>12] & ~0xFFF,
 			(expected & ~0xFFF) - 1,
-			(expected & PF_PAGED ? "p" : "-"),
+			(expected & PF_NOPAGE ? "p" : "-"),
 			(expected & PF_COW ? "C" : "-"),
 			(expected & PF_USER ? "U" : "-"),
 			(expected & PF_WRITE ? "W" : "-")
@@ -541,37 +545,40 @@ tPAddr MM_Clone(void)
 	//LOG("Allocated Directory (%x)", *gpTmpCR3);
 	memsetd( gaTmpDir, 0, 1024 );
 	
-	// Copy Tables
-	for( i = 0; i < 768; i ++)
-	{
-		// Check if table is allocated
-		if( !(gaPageDir[i] & PF_PRESENT) ) {
-			gaTmpDir[i] = 0;
-			page += 1024;
-			continue;
-		}
-		
-		// Allocate new table
-		gaTmpDir[i] = MM_AllocPhys() | (gaPageDir[i] & 7);
-		INVLPG( &gaTmpTable[page] );
-		// Fill
-		for( j = 0; j < 1024; j ++, page++ )
+	if( Threads_GetPID() != 0 )
+	{	
+		// Copy Tables
+		for( i = 0; i < 768; i ++)
 		{
-			if( !(gaPageTable[page] & PF_PRESENT) ) {
-				gaTmpTable[page] = 0;
+			// Check if table is allocated
+			if( !(gaPageDir[i] & PF_PRESENT) ) {
+				gaTmpDir[i] = 0;
+				page += 1024;
 				continue;
 			}
 			
-			// Refrence old page
-			MM_RefPhys( gaPageTable[page] & ~0xFFF );
-			// Add to new table
-			if(gaPageTable[page] & PF_WRITE) {
-				gaTmpTable[page] = (gaPageTable[page] & ~PF_WRITE) | PF_COW;
-				gaPageTable[page] = (gaPageTable[page] & ~PF_WRITE) | PF_COW;
-				INVLPG( page << 12 );
+			// Allocate new table
+			gaTmpDir[i] = MM_AllocPhys() | (gaPageDir[i] & 7);
+			INVLPG( &gaTmpTable[page] );
+			// Fill
+			for( j = 0; j < 1024; j ++, page++ )
+			{
+				if( !(gaPageTable[page] & PF_PRESENT) ) {
+					gaTmpTable[page] = 0;
+					continue;
+				}
+				
+				// Refrence old page
+				MM_RefPhys( gaPageTable[page] & ~0xFFF );
+				// Add to new table
+				if(gaPageTable[page] & PF_WRITE) {
+					gaTmpTable[page] = (gaPageTable[page] & ~PF_WRITE) | PF_COW;
+					gaPageTable[page] = (gaPageTable[page] & ~PF_WRITE) | PF_COW;
+					INVLPG( page << 12 );
+				}
+				else
+					gaTmpTable[page] = gaPageTable[page];
 			}
-			else
-				gaTmpTable[page] = gaPageTable[page];
 		}
 	}
 	
@@ -653,17 +660,32 @@ tPAddr MM_Clone(void)
  */
 tVAddr MM_NewKStack(void)
 {
-	tVAddr	base = KERNEL_STACKS;
+	tVAddr	base;
 	Uint	i;
-	for(;base<KERNEL_STACKS_END;base+=KERNEL_STACK_SIZE)
+	for(base = KERNEL_STACKS; base < KERNEL_STACKS_END; base += KERNEL_STACK_SIZE)
 	{
+		// Check if space is free
 		if(MM_GetPhysAddr(base) != 0)	continue;
-		for(i=0;i<KERNEL_STACK_SIZE;i+=0x1000) {
-			MM_Allocate(base+i);
+		// Allocate
+		//for(i = KERNEL_STACK_SIZE; i -= 0x1000 ; )
+		for(i = 0; i < KERNEL_STACK_SIZE; i += 0x1000 )
+		{
+			if( MM_Allocate(base+i) == 0 )
+			{
+				// On error, print a warning and return error
+				Warning("MM_NewKStack - Out of memory");
+				// - Clean up
+				//for( i += 0x1000 ; i < KERNEL_STACK_SIZE; i += 0x1000 )
+				//	MM_Deallocate(base+i);
+				return 0;
+			}
 		}
+		// Success
+		Log("MM_NewKStack - Allocated %p", base + KERNEL_STACK_SIZE);
 		return base+KERNEL_STACK_SIZE;
 	}
-	Warning("MM_NewKStack - No address space left\n");
+	// No stacks left
+	Warning("MM_NewKStack - No address space left");
 	return 0;
 }
 
@@ -684,6 +706,7 @@ tVAddr MM_NewWorkerStack()
 	__asm__ __volatile__ ("mov %%esp, %0": "=r"(esp));
 	__asm__ __volatile__ ("mov %%ebp, %0": "=r"(ebp));
 	
+	// TODO: Thread safety
 	// Find a free worker stack address
 	for(base = giLastUsedWorker; base < NUM_WORKER_STACKS; base++)
 	{
@@ -731,6 +754,7 @@ tVAddr MM_NewWorkerStack()
 	
 	// Mapping Time!
 	for( addr = 0; addr < WORKER_STACK_SIZE; addr += 0x1000 )
+	//for( addr = WORKER_STACK_SIZE; addr; addr -= 0x1000 )
 	{
 		pages[ addr >> 12 ] = MM_AllocPhys();
 		gaTmpTable[ (base + addr) >> 12 ] = pages[addr>>12] | 3;
@@ -918,7 +942,7 @@ tVAddr MM_MapTemp(tPAddr PAddr)
 			return TEMP_MAP_ADDR + (i << 12);
 		}
 		Mutex_Release( &glTempMappings );
-		Threads_Yield();	// TODO: Less expensive
+		Threads_Yield();	// TODO: Use a sleep queue here instead
 	}
 }