From 19927ee517977addd56a040dc7549976a3a0439b Mon Sep 17 00:00:00 2001
From: Sebastian Lackner <sebastian@fds-team.de>
Date: Tue, 21 Oct 2014 07:43:12 +0200
Subject: [PATCH] kernel32/tests: Add test for GetProcessDEPPolicy.

---
 dlls/kernel32/tests/virtual.c | 40 +++++++++++++++++++++++++++++++++++
 include/winbase.h             |  3 +++
 2 files changed, 43 insertions(+)

diff --git a/dlls/kernel32/tests/virtual.c b/dlls/kernel32/tests/virtual.c
index 1592b87e289..40f45efb168 100644
--- a/dlls/kernel32/tests/virtual.c
+++ b/dlls/kernel32/tests/virtual.c
@@ -46,6 +46,7 @@ static DWORD (WINAPI *pNtUnmapViewOfSection)(HANDLE, PVOID);
 static struct _TEB * (WINAPI *pNtCurrentTeb)(void);
 static PVOID  (WINAPI *pRtlAddVectoredExceptionHandler)(ULONG, PVECTORED_EXCEPTION_HANDLER);
 static ULONG  (WINAPI *pRtlRemoveVectoredExceptionHandler)(PVOID);
+static BOOL   (WINAPI *pGetProcessDEPPolicy)(HANDLE, LPDWORD, PBOOL);
 
 /* ############################### */
 
@@ -1972,6 +1973,44 @@ static void test_atl_thunk_emulation( ULONG dep_flags )
     base = VirtualAlloc( 0, size, MEM_RESERVE | MEM_COMMIT, PAGE_READWRITE );
     ok( base != NULL, "VirtualAlloc failed %u\n", GetLastError() );
 
+    /* Check result of GetProcessDEPPolicy */
+    if (!pGetProcessDEPPolicy)
+        win_skip( "GetProcessDEPPolicy not supported\n" );
+    else
+    {
+        BOOL (WINAPI *get_dep_policy)(HANDLE, LPDWORD, PBOOL) = (void *)base;
+        BOOL policy_permanent = 0xdeadbeef;
+        DWORD policy_flags = 0xdeadbeef;
+
+        /* GetProcessDEPPolicy crashes on Windows when a NULL pointer is passed.
+         * Moreover this function has a bug on Windows 8, which has the effect that
+         * policy_permanent is set to the content of the CL register instead of 0,
+         * when the policy is not permanent. To detect that we use an assembler
+         * wrapper to call the function. */
+
+        memcpy( base, code_atl2, sizeof(code_atl2) );
+        *(DWORD *)(base + 6) = (DWORD_PTR)pGetProcessDEPPolicy - (DWORD_PTR)(base + 10);
+
+        success = VirtualProtect( base, size, PAGE_EXECUTE_READWRITE, &old_prot );
+        ok( success, "VirtualProtect failed %u\n", GetLastError() );
+
+        success = get_dep_policy( GetCurrentProcess(), &policy_flags, &policy_permanent );
+        todo_wine
+        ok( success, "GetProcessDEPPolicy failed %u\n", GetLastError() );
+
+        ret = 0;
+        if (dep_flags & MEM_EXECUTE_OPTION_DISABLE)
+            ret |= PROCESS_DEP_ENABLE;
+        if (dep_flags & MEM_EXECUTE_OPTION_DISABLE_THUNK_EMULATION)
+            ret |= PROCESS_DEP_DISABLE_ATL_THUNK_EMULATION;
+
+        todo_wine
+        ok( policy_flags == ret, "expected policy flags %d, got %d\n", ret, policy_flags );
+        todo_wine
+        ok( !policy_permanent || broken(policy_permanent == 0x44),
+            "expected policy permanent FALSE, got %d\n", policy_permanent );
+    }
+
     memcpy( base, code_jmp, sizeof(code_jmp) );
     *(DWORD *)(base + 1) = (DWORD_PTR)jmp_test_func - (DWORD_PTR)(base + 5);
 
@@ -3424,6 +3463,7 @@ START_TEST(virtual)
     pVirtualFreeEx = (void *) GetProcAddress(hkernel32, "VirtualFreeEx");
     pGetWriteWatch = (void *) GetProcAddress(hkernel32, "GetWriteWatch");
     pResetWriteWatch = (void *) GetProcAddress(hkernel32, "ResetWriteWatch");
+    pGetProcessDEPPolicy = (void *)GetProcAddress( hkernel32, "GetProcessDEPPolicy" );
     pNtAreMappedFilesTheSame = (void *)GetProcAddress( hntdll, "NtAreMappedFilesTheSame" );
     pNtMapViewOfSection = (void *)GetProcAddress( hntdll, "NtMapViewOfSection" );
     pNtUnmapViewOfSection = (void *)GetProcAddress( hntdll, "NtUnmapViewOfSection" );
diff --git a/include/winbase.h b/include/winbase.h
index edd6ad6ccd1..723da268bf3 100644
--- a/include/winbase.h
+++ b/include/winbase.h
@@ -1449,6 +1449,9 @@ typedef enum _DEP_SYSTEM_POLICY_TYPE {
     OptOut = 3
 } DEP_SYSTEM_POLICY_TYPE;
 
+#define PROCESS_DEP_ENABLE 1
+#define PROCESS_DEP_DISABLE_ATL_THUNK_EMULATION 2
+
 /* Event Logging */
 
 #define EVENTLOG_FULL_INFO          0