X-Git-Url: https://git.ucc.asn.au/?a=blobdiff_plain;f=KernelLand%2FKernel%2Fbinary.c;h=208ab95b734385940999cb12c8417f7f68c3673a;hb=34e8f9c025a5fe5ebc90ada5690685821827a5e7;hp=5213bf2d50cd203a4e44df153509f478f74e4d75;hpb=48743e39650eb1ef988380e9d95f27fd40d3a9ce;p=tpg%2Facess2.git

diff --git a/KernelLand/Kernel/binary.c b/KernelLand/Kernel/binary.c
index 5213bf2d..208ab95b 100644
--- a/KernelLand/Kernel/binary.c
+++ b/KernelLand/Kernel/binary.c
@@ -194,9 +194,10 @@ int Proc_SysSpawn(const char *Binary, const char **ArgV, const char **EnvP, int
 		Proc_Execve(Binary, ArgV, EnvP, size);
 		for(;;);
 	}
-	if( ret < 0 )
+	if( ret == -1 )
 	{
 		VFS_FreeSavedHandles(nFD, handles);
+		free(cachebuf);
 	}
 	
 	return ret;
@@ -406,7 +407,10 @@ tVAddr Binary_MapIn(tBinary *Binary, const char *Path, tVAddr LoadMin, tVAddr Lo
 			for( i = 0; i < Binary->NumSections; i ++ )
 			{
 				tVAddr	addr = Binary->LoadSections[i].Virtual - Binary->Base + base;
-				if( Binary_int_CheckMemFree( addr, Binary->LoadSections[i].MemSize ) )
+				size_t	size = Binary->LoadSections[i].MemSize;
+				if( addr + size > LoadMax )
+					break;
+				if( Binary_int_CheckMemFree( addr, size ) )
 					break;
 			}
 			// If space was found, break
@@ -425,13 +429,16 @@ tVAddr Binary_MapIn(tBinary *Binary, const char *Path, tVAddr LoadMin, tVAddr Lo
 	}
 	
 	// Map Executable In
-	fd = VFS_OpenInode(Binary->MountID, Binary->Inode, VFS_OPENFLAG_READ);
+	if( Binary->MountID )
+		fd = VFS_OpenInode(Binary->MountID, Binary->Inode, VFS_OPENFLAG_READ);
+	else
+		fd = VFS_Open(Path, VFS_OPENFLAG_READ);
 	for( i = 0; i < Binary->NumSections; i ++ )
 	{
 		tBinarySection	*sect = &Binary->LoadSections[i];
 		Uint	protflags, mapflags;
 		tVAddr	addr = sect->Virtual - Binary->Base + base;
-		LOG("%i - %p to offset 0x%llx (%x)", i, addr, sect->Offset, sect->Flags);
+		LOG("%i - %p, 0x%x bytes from offset 0x%llx (%x)", i, addr, sect->FileSize, sect->Offset, sect->Flags);
 
 		protflags = MMAP_PROT_READ;
 		mapflags = MMAP_MAP_FIXED;
@@ -508,7 +515,14 @@ tBinary *Binary_DoLoad(tMount MountID, tInode Inode, const char *Path)
 	ENTER("iMountID XInode sPath", MountID, Inode, Path);
 	
 	// Open File
-	fp = VFS_OpenInode(MountID, Inode, VFS_OPENFLAG_READ);
+	if( MountID )
+	{
+		fp = VFS_OpenInode(MountID, Inode, VFS_OPENFLAG_READ);
+	}
+	else
+	{
+		fp = VFS_Open(Path, VFS_OPENFLAG_READ);
+	}
 	if(fp == -1) {
 		LOG("Unable to load file, access denied");
 		LEAVE('n');
@@ -697,6 +711,7 @@ void *Binary_LoadKernel(const char *File)
 		int fd = VFS_Open(File, VFS_OPENFLAG_READ);
 		tFInfo	info;
 		if(fd == -1) {
+			LOG("Opening failed");
 			LEAVE('n');
 			return NULL;
 		}
@@ -704,6 +719,7 @@ void *Binary_LoadKernel(const char *File)
 		mount_id = info.mount;
 		inode = info.inode;
 		VFS_Close(fd);
+		LOG("Mount %i, Inode %lli", mount_id, inode);
 	}
 	
 	// Check if the binary has already been loaded
@@ -714,6 +730,7 @@ void *Binary_LoadKernel(const char *File)
 			pKBinary = pKBinary->Next )
 		{
 			if(pKBinary->Info == pBinary) {
+				LOG("Already loaded");
 				LEAVE('p', pKBinary->Base);
 				return pKBinary->Base;
 			}
@@ -728,6 +745,7 @@ void *Binary_LoadKernel(const char *File)
 		return NULL;
 	}
 	
+	LOG("Loaded as %p", pBinary);
 	// --------------
 	// Now pBinary is valid (either freshly loaded or only user mapped)
 	// So, map it into kernel space
@@ -736,19 +754,14 @@ void *Binary_LoadKernel(const char *File)
 	// Reference Executable (Makes sure that it isn't unloaded)
 	pBinary->ReferenceCount ++;
 
-	Binary_MapIn(pBinary, File, KLIB_LOWEST, KLIB_HIGHEST);
-
-	// Relocate Library
-	if( !Binary_Relocate( (void*)base ) )
-	{
-		Log_Warning("Binary", "Relocation of '%s' failed, unloading", File);
-		Binary_Unload( (void*)base );
-		Binary_Dereference( pBinary );
+	base = Binary_MapIn(pBinary, File, KLIB_LOWEST, KLIB_HIGHEST);
+	if( base == 0 ) {
 		LEAVE('n');
 		return 0;
 	}
-	
-	// Add to list (relocator must look at itself manually, not via Binary_GetSymbol)
+
+	// Add to list
+	// TODO: Could this cause race conditions if a binary isn't fully loaded when used
 	pKBinary = malloc(sizeof(*pKBinary));
 	pKBinary->Base = (void*)base;
 	pKBinary->Info = pBinary;
@@ -756,7 +769,7 @@ void *Binary_LoadKernel(const char *File)
 	pKBinary->Next = glLoadedKernelLibs;
 	glLoadedKernelLibs = pKBinary;
 	SHORTREL( &glKBinListLock );
-	
+
 	LEAVE('p', base);
 	return (void*)base;
 }
@@ -772,7 +785,7 @@ Uint Binary_Relocate(void *Base)
 	Uint32	ident = *(Uint32*) Base;
 	tBinaryType	*bt = gRegBinTypes;
 	
-	for(; bt; bt = bt->Next)
+	for( ; bt; bt = bt->Next)
 	{
 		if( (ident & bt->Mask) == (Uint)bt->Ident )
 			return bt->Relocate( (void*)Base);
@@ -810,9 +823,12 @@ Uint Binary_GetSymbolEx(const char *Name, Uint *Value)
 	tKernelBin	*pKBin;
 	 int	numKSyms = ((Uint)&gKernelSymbolsEnd-(Uint)&gKernelSymbols)/sizeof(tKernelSymbol);
 	
+	LOG("numKSyms = %i", numKSyms);
+
 	// Scan Kernel
 	for( i = 0; i < numKSyms; i++ )
 	{
+		LOG("KSym %s = %p", gKernelSymbols[i].Name, gKernelSymbols[i].Value);
 		if(strcmp(Name, gKernelSymbols[i].Name) == 0) {
 			*Value = gKernelSymbols[i].Value;
 			return 1;
@@ -862,15 +878,23 @@ Uint Binary_FindSymbol(void *Base, const char *Name, Uint *Val)
  */
 int Binary_int_CheckMemFree( tVAddr _start, size_t _len )
 {
+	ENTER("p_start x_len", _start, _len);
+
 	_len += _start & (PAGE_SIZE-1);
 	_len = (_len + PAGE_SIZE - 1) & ~(PAGE_SIZE-1);
 	_start &= ~(PAGE_SIZE-1);
+	LOG("_start = %p, _len = 0x%x", _start, _len);
 	for( ; _len > PAGE_SIZE; _len -= PAGE_SIZE, _start += PAGE_SIZE ) {
-		if( MM_GetPhysAddr(_start) != 0 )
+		if( MM_GetPhysAddr( (void*)_start ) != 0 ) {
+			LEAVE('i', 1);
 			return 1;
+		}
 	}
-	if( _len == PAGE_SIZE && MM_GetPhysAddr(_start) != 0 )
+	if( _len == PAGE_SIZE && MM_GetPhysAddr( (void*)_start ) != 0 ) {
+		LEAVE('i', 1);
 		return 1;
+	}
+	LEAVE('i', 0);
 	return 0;
 }