X-Git-Url: https://git.ucc.asn.au/?a=blobdiff_plain;f=Kernel%2Farch%2Fx86%2Fmain.c;h=3ea4f3f32a4ee7d89fdf587d31bf2e0ca1bdb824;hb=4b80e9762374558077e28e321d75029645529a45;hp=8c094424db342e49c1d6ba5719bea14a17e41d11;hpb=0664d1e9ca50ce1d98fcc2ca66951e28e2fbbbbc;p=tpg%2Facess2.git

diff --git a/Kernel/arch/x86/main.c b/Kernel/arch/x86/main.c
index 8c094424..3ea4f3f3 100644
--- a/Kernel/arch/x86/main.c
+++ b/Kernel/arch/x86/main.c
@@ -12,6 +12,8 @@
 
 #define	VGA_ERRORS	0
 
+#define MAX_ARGSTR_POS	(0x400000-0x2000)
+
 // === IMPORTS ===
 extern void	Heap_Install(void);
 extern void	Desctab_Install(void);
@@ -96,23 +98,26 @@ int kmain(Uint MbMagic, void *MbInfoPtr)
 	for( i = 0; i < mbInfo->ModuleCount; i ++ )
 	{
 		 int	ofs;
-		// TODO: Handle this better by using MM_MapHW/MM_MapTemp
-		// Adjust into higher half
-		//mods[i].Start  += KERNEL_BASE;
-		//mods[i].End    += KERNEL_BASE;
-		mods[i].String += KERNEL_BASE;
 		
+		// Always HW map the module data
 		gaArch_BootModules[i].Size = mods[i].End - mods[i].Start;
 		
 		ofs = mods[i].Start&0xFFF;
 		gaArch_BootModules[i].Base = (void*)( MM_MapHWPages(mods[i].Start,
 			(gaArch_BootModules[i].Size+ofs+0xFFF) / 0x1000
 			) + ofs );
-		//gaArch_BootModules[i].ArgString = MM_MapHW(mods[i].String, 1)
-		// + (mods[i].String&0xFFF);
 		
-		//gaArch_BootModules[i].Base = (void *)mods[i].Start;
-		gaArch_BootModules[i].ArgString = (char *)mods[i].String + KERNEL_BASE;
+		// Only map the string if needed
+		if( (tVAddr)mods[i].String > MAX_ARGSTR_POS )
+		{
+			// Assumes the string is < 4096 bytes long)
+			gaArch_BootModules[i].ArgString = (void*)(
+				MM_MapHWPages((tVAddr)mods[i].String, 2)
+				+ ((tVAddr)mods[i].String&0xFFF)
+				);
+		}
+		else
+			gaArch_BootModules[i].ArgString = (char *)mods[i].String + KERNEL_BASE;
 	}
 	
 	// Pass on to Independent Loader
@@ -145,6 +150,9 @@ void Arch_LoadBootModules(void)
 			(tVAddr)gaArch_BootModules[i].Base,
 			(gaArch_BootModules[i].Size + ((Uint)gaArch_BootModules[i].Base&0xFFF) + 0xFFF) >> 12
 			);
+		
+		if( (tVAddr) gaArch_BootModules[i].ArgString > MAX_ARGSTR_POS )
+			MM_UnmapHWPages( (tVAddr)gaArch_BootModules[i].ArgString, 2 );
 	}
 	Log_Log("Arch", "Boot modules loaded");
 	free( gaArch_BootModules );