ntdll: Restore stack guard and prevent stack from shrinking.

6b2c9505 · Piotr Caban · Alexandre Julliard · 6990ecaa · 6b2c9505
Commit 6b2c9505 authored Apr 12, 2011 by Piotr Caban Committed by Alexandre Julliard Apr 13, 2011
Hide whitespace changes
Inline Side-by-side

Showing with 6 additions and 2 deletions

virtual.c dlls/ntdll/virtual.c +6 -2

No files found.
--- a/dlls/ntdll/virtual.c
+++ b/dlls/ntdll/virtual.c
@@ -1618,8 +1618,12 @@ BOOL virtual_handle_stack_fault( void *addr )
        if (vprot & VPROT_GUARD)
        {
            VIRTUAL_SetProt( view, page, page_size, vprot & ~VPROT_GUARD );
-            if ((char *)page + page_size == NtCurrentTeb()->Tib.StackLimit)
-                NtCurrentTeb()->Tib.StackLimit = page;
+            NtCurrentTeb()->Tib.StackLimit = page;
+            if ((char *)page >= (char *)NtCurrentTeb()->DeallocationStack + 2*page_size)
+            {
+                vprot = view->prot[((char *)page - page_size - (char *)view->base) >> page_shift];
+                VIRTUAL_SetProt( view, (char *)page - page_size, page_size, vprot | VPROT_GUARD );
+            }
            ret = TRUE;
        }
    }