[ros-diffs] [jimtabor] 51773: [ApiTest] - Add a test for stack corruption when in window procedure call out.

Author: jimtabor
Date: Sun May 15 21:15:38 2011
New Revision: 51773

URL: http://svn.reactos.org/svn/reactos?rev=51773&view=rev
Log:
[ApiTest]
- Add a test for stack corruption when in window procedure call out.

Added:
    trunk/rostests/apitests/user32/WndProc.c   (with props)
Modified:
    trunk/rostests/apitests/user32/CMakeLists.txt
    trunk/rostests/apitests/user32/testlist.c
    trunk/rostests/apitests/user32/user32_apitest.rbuild

Modified: trunk/rostests/apitests/user32/CMakeLists.txt
URL:
http://svn.reactos.org/svn/reactos/trunk/rostests/apitests/user32/CMakeList…
==============================================================================
--- trunk/rostests/apitests/user32/CMakeLists.txt [iso-8859-1] (original)
+++ trunk/rostests/apitests/user32/CMakeLists.txt [iso-8859-1] Sun May 15 21:15:38 2011
@@ -14,6 +14,7 @@
     ScrollDC.c
     ScrollWindowEx.c
     SetCursorPos.c
+    WndProc.c
     testlist.c
     user32_apitest.rc)
 

Added: trunk/rostests/apitests/user32/WndProc.c
URL:
http://svn.reactos.org/svn/reactos/trunk/rostests/apitests/user32/WndProc.c…
==============================================================================
--- trunk/rostests/apitests/user32/WndProc.c (added)
+++ trunk/rostests/apitests/user32/WndProc.c [iso-8859-1] Sun May 15 21:15:38 2011
@@ -1,0 +1,100 @@
+/*
+ * PROJECT:         ReactOS api tests
+ * LICENSE:         GPL - See COPYING in the top level directory
+ * PURPOSE:         Test for mismatch with function prototype in window procedure
callback.
+ * PROGRAMMERS:
+ */
+
+#include <stdio.h>
+#include <wine/test.h>
+#include <windows.h>
+
+/* Used wine Redraw test for proof in principle. */
+
+/* Global variables to trigger exit from loop */
+static int redrawComplete, WMPAINT_count;
+
+/*
+   Force stack corruption when calling from assumed window procedure callback.
+   Adding (6 and) more will force exception faults and terminate the test program.
+   The test is with five and this is safe for windows.
+
+   But,,,, ReactOS compiled with GCC can handle this,,,,,,
+ */
+static LRESULT WINAPI redraw_window_procA(HWND hwnd, UINT msg, WPARAM wparam, LPARAM
lparam, DWORD ExtraData,DWORD ExtraData1,DWORD ExtraData2,DWORD ExtraData3)
+{
+    switch (msg)
+    {
+    case WM_PAINT:
+        trace("doing WM_PAINT %d\n", WMPAINT_count);
+        WMPAINT_count++;
+        if (WMPAINT_count > 10 && redrawComplete == 0)
+        {
+            PAINTSTRUCT ps;
+            BeginPaint(hwnd, &ps);
+            EndPaint(hwnd, &ps);  
+            return 1;
+        }
+   /*
+       This will force one stack corruption "ret" fault with normal window
+       procedure callback.
+    */
+#ifdef __MINGW32__
+        asm ("movl $0, %eax\n\t"
+             "leave\n\t"
+             "ret");
+#else
+//#ifdef _MSC_VER
+        __asm
+          {
+             mov eax, 0
+             leave
+             ret
+          }
+#endif
+    }
+    return DefWindowProc(hwnd, msg, wparam, lparam);
+}
+
+static void test_wndproc(void)
+{
+   WNDCLASSA cls;
+   HWND hwndMain;
+
+   cls.style = CS_DBLCLKS;
+   cls.lpfnWndProc = (WNDPROC)redraw_window_procA;
+   cls.cbClsExtra = 0;
+   cls.cbWndExtra = 0;  
+   cls.hInstance = GetModuleHandleA(0);
+   cls.hIcon = 0;
+   cls.hCursor = LoadCursorA(0, IDC_ARROW);
+   cls.hbrBackground = GetStockObject(WHITE_BRUSH);
+   cls.lpszMenuName = NULL;
+   cls.lpszClassName = "RedrawWindowClass";
+
+   if (!RegisterClassA(&cls))
+   {
+       trace("Register failed %d\n", (int)GetLastError());
+       return;
+   }
+
+   hwndMain = CreateWindowA("RedrawWindowClass", "Main Window",
WS_OVERLAPPEDWINDOW,
+                            CW_USEDEFAULT, 0, 100, 100, NULL, NULL, 0, NULL);
+
+   ok( WMPAINT_count == 0, "Multiple unexpected WM_PAINT calls %d\n",
WMPAINT_count);
+   ShowWindow(hwndMain, SW_SHOW);
+   ok( WMPAINT_count == 0, "Multiple unexpected WM_PAINT calls %d\n",
WMPAINT_count);
+   RedrawWindow(hwndMain, NULL,NULL,RDW_UPDATENOW | RDW_ALLCHILDREN);
+   ok( WMPAINT_count == 1 || broken(WMPAINT_count == 0), /* sometimes on win9x */
+       "Multiple unexpected WM_PAINT calls %d\n", WMPAINT_count);
+   redrawComplete = TRUE;
+   ok( WMPAINT_count < 10, "RedrawWindow (RDW_UPDATENOW) never completed
(%d)\n", WMPAINT_count);
+
+   /* clean up */
+   DestroyWindow( hwndMain);
+}
+
+START_TEST(WndProc)
+{
+   test_wndproc();
+}

Propchange: trunk/rostests/apitests/user32/WndProc.c
------------------------------------------------------------------------------
    svn:eol-style = native

Modified: trunk/rostests/apitests/user32/testlist.c
URL:
http://svn.reactos.org/svn/reactos/trunk/rostests/apitests/user32/testlist.…
==============================================================================
--- trunk/rostests/apitests/user32/testlist.c [iso-8859-1] (original)
+++ trunk/rostests/apitests/user32/testlist.c [iso-8859-1] Sun May 15 21:15:38 2011
@@ -15,6 +15,7 @@
 extern void func_DeferWindowPos(void);
 extern void func_GetKeyState(void);
 extern void func_SetCursorPos(void);
+extern void func_WndProc(void);
 
 const struct test winetest_testlist[] =
 {
@@ -28,6 +29,7 @@
     { "DeferWindowPos", func_DeferWindowPos },
     { "GetKeyState", func_GetKeyState },
     { "SetCursorPos", func_SetCursorPos },
+    { "WndProc", func_WndProc },
     { 0, 0 }
 };
 

Modified: trunk/rostests/apitests/user32/user32_apitest.rbuild
URL:
http://svn.reactos.org/svn/reactos/trunk/rostests/apitests/user32/user32_ap…
==============================================================================
--- trunk/rostests/apitests/user32/user32_apitest.rbuild [iso-8859-1] (original)
+++ trunk/rostests/apitests/user32/user32_apitest.rbuild [iso-8859-1] Sun May 15 21:15:38
2011
@@ -20,6 +20,7 @@
 	<file>GetPeekMessage.c</file>
 	<file>DeferWindowPos.c</file>
 	<file>SetCursorPos.c</file>
+	<file>WndProc.c</file>
 
 </module>
 </group>