Intravision/reactos
1
0
Fork 0
mirror of https://github.com/reactos/reactos.git synced 2024-07-04 03:34:16 +00:00
Code Issues Packages Projects Releases Wiki Activity

Fixed stack allocation for initial process (smss.exe).

Browse source 
svn path=/trunk/; revision=2148
This commit is contained in:
Eric Kohl 2001-08-03 19:52:12 +00:00
parent 27346f997d
commit 4b420ec7f6
1 changed files with 85 additions and 28 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

113
reactos/ntoskrnl/ldr/init.c
View file

@ -45,13 +45,12 @@ NTSTATUS LdrLoadInitialProcess (VOID)
HANDLE FileHandle; HANDLE FileHandle;
HANDLE SectionHandle; HANDLE SectionHandle;
PVOID LdrStartupAddr; PVOID LdrStartupAddr;
PVOID StackBase;
ULONG StackSize;
PIMAGE_NT_HEADERS NTHeaders; PIMAGE_NT_HEADERS NTHeaders;
PPEB Peb; PPEB Peb;
PEPROCESS Process; PEPROCESS Process;
CONTEXT Context; CONTEXT Context;
HANDLE ThreadHandle; HANDLE ThreadHandle;
INITIAL_TEB InitialTeb;
/* /*
* Get the absolute path to smss.exe using the * Get the absolute path to smss.exe using the
@ -153,26 +152,80 @@ NTSTATUS LdrLoadInitialProcess (VOID)
Peb->ImageBaseAddress); Peb->ImageBaseAddress);
NTHeaders = RtlImageNtHeader(Peb->ImageBaseAddress); NTHeaders = RtlImageNtHeader(Peb->ImageBaseAddress);
DPRINT("NTHeaders %x\n", NTHeaders); DPRINT("NTHeaders %x\n", NTHeaders);
StackSize = NTHeaders->OptionalHeader.SizeOfStackReserve; InitialTeb.StackReserve = NTHeaders->OptionalHeader.SizeOfStackReserve;
DPRINT("StackSize %x\n", StackSize); InitialTeb.StackCommit = NTHeaders->OptionalHeader.SizeOfStackCommit;
/* add guard page size */
InitialTeb.StackCommit += PAGESIZE;
DPRINT("StackReserve 0x%lX StackCommit 0x%lX\n",
InitialTeb.StackReserve, InitialTeb.StackCommit);
KeDetachProcess(); KeDetachProcess();
DPRINT("Dereferencing process\n"); DPRINT("Dereferencing process\n");
// ObDereferenceObject(Process); // ObDereferenceObject(Process);
StackBase = (PVOID)NULL; DPRINT("Allocating stack\n");
DPRINT("StackBase %x StackSize %x\n", StackBase, StackSize); InitialTeb.StackAllocate = NULL;
DPRINT("Allocating virtual memory\n"); Status = ZwAllocateVirtualMemory(ProcessHandle,
Status = ZwAllocateVirtualMemory(ProcessHandle, &InitialTeb.StackAllocate,
(PVOID*)&StackBase, 0,
0, &InitialTeb.StackReserve,
&StackSize, MEM_COMMIT, // MEM_RESERVE,
MEM_COMMIT, PAGE_READWRITE);
PAGE_READWRITE); if (!NT_SUCCESS(Status))
if (!NT_SUCCESS(Status)) {
{ DPRINT("Stack allocation failed (Status %x)", Status);
DPRINT("Stack allocation failed (Status %x)", Status); return(Status);
return Status; }
}
DPRINT("StackAllocate: %p ReserveSize: 0x%lX\n",
InitialTeb.StackAllocate, InitialTeb.StackReserve);
InitialTeb.StackBase = (PVOID)((ULONG)InitialTeb.StackAllocate + InitialTeb.StackReserve);
InitialTeb.StackLimit = (PVOID)((ULONG)InitialTeb.StackBase - InitialTeb.StackCommit);
DPRINT("StackBase: %p StackCommit: 0x%lX\n",
InitialTeb.StackBase, InitialTeb.StackCommit);
#if 0
/* Commit stack */
Status = NtAllocateVirtualMemory(ProcessHandle,
&InitialTeb.StackLimit,
0,
&InitialTeb.StackCommit,
MEM_COMMIT,
PAGE_READWRITE);
if (!NT_SUCCESS(Status))
{
/* release the stack space */
NtFreeVirtualMemory(ProcessHandle,
InitialTeb.StackAllocate,
&InitialTeb.StackReserve,
MEM_RELEASE);
DPRINT("Error comitting stack page!\n");
return(Status);
}
DPRINT("StackLimit: %p\nStackCommit: 0x%lX\n",
InitialTeb.StackLimit,
InitialTeb.StackCommit);
/* Protect guard page */
Status = NtProtectVirtualMemory(ProcessHandle,
InitialTeb.StackLimit,
PAGESIZE,
PAGE_GUARD | PAGE_READWRITE,
&OldPageProtection);
if (!NT_SUCCESS(Status))
{
/* release the stack space */
NtFreeVirtualMemory(ProcessHandle,
InitialTeb.StackAllocate,
&InitialTeb.StackReserve,
MEM_RELEASE);
DPRINT("Error protecting guard page!\n");
return(Status);
}
#endif
DPRINT("Attaching to process\n"); DPRINT("Attaching to process\n");
KeAttachProcess(Process); KeAttachProcess(Process);
@ -187,15 +240,15 @@ NTSTATUS LdrLoadInitialProcess (VOID)
* Initialize context to point to LdrStartup * Initialize context to point to LdrStartup
*/ */
memset(&Context,0,sizeof(CONTEXT)); memset(&Context,0,sizeof(CONTEXT));
Context.SegSs = USER_DS;
Context.Esp = (ULONG)StackBase + StackSize - 20;
Context.EFlags = 0x202;
Context.SegCs = USER_CS;
Context.Eip = (ULONG)LdrStartupAddr; Context.Eip = (ULONG)LdrStartupAddr;
Context.SegCs = USER_CS;
Context.SegDs = USER_DS; Context.SegDs = USER_DS;
Context.SegEs = USER_DS; Context.SegEs = USER_DS;
Context.SegFs = TEB_SELECTOR; Context.SegFs = TEB_SELECTOR;
Context.SegGs = USER_DS; Context.SegGs = USER_DS;
Context.SegSs = USER_DS;
Context.EFlags = 0x202;
Context.Esp = (ULONG)InitialTeb.StackBase - 20;
DPRINT("LdrStartupAddr %x\n",LdrStartupAddr); DPRINT("LdrStartupAddr %x\n",LdrStartupAddr);
@ -209,13 +262,17 @@ NTSTATUS LdrLoadInitialProcess (VOID)
ProcessHandle, ProcessHandle,
NULL, NULL,
&Context, &Context,
NULL, &InitialTeb,
FALSE); FALSE);
if (!NT_SUCCESS(Status)) if (!NT_SUCCESS(Status))
{ {
DPRINT("Thread creation failed (Status %x)\n", Status); DPRINT("Thread creation failed (Status %x)\n", Status);
/* FIXME: destroy the stack memory block here */ NtFreeVirtualMemory(ProcessHandle,
InitialTeb.StackAllocate,
&InitialTeb.StackReserve,
MEM_RELEASE);
/* FIXME: unmap the section here */ /* FIXME: unmap the section here */
/* FIXME: destroy the section here */ /* FIXME: destroy the section here */