[PATCH 3/3] kernel32/tests: Add tests for RtlActivateActivationContextUnsafe() and RtlDeactivateActivationContextUnsafeFast().

[Zhiyi Zhang]

From: Zhiyi Zhang

---
 dlls/kernel32/tests/actctx.c | 81 ++++++++++++++++++++++++++++++++++++
 1 file changed, 81 insertions(+)

diff --git a/dlls/kernel32/tests/actctx.c b/dlls/kernel32/tests/actctx.c
index ca24ed10549..cc490f984e3 100644
--- a/dlls/kernel32/tests/actctx.c
+++ b/dlls/kernel32/tests/actctx.c
@@ -31,6 +31,9 @@

static BOOL   (WINAPI *pQueryActCtxSettingsW)(DWORD,HANDLE,LPCWSTR,LPCWSTR,LPWSTR,SIZE_T,SIZE_T*);

+static PRTL_ACTIVATION_CONTEXT_STACK_FRAME (FASTCALL *pRtlActivateActivationContextUnsafeFast)(PRTL_CALLER_ALLOCATED_ACTIVATION_CONTEXT_STACK_FRAME_EXTENDED,
+                                                                                               PVOID);
+static VOID (FASTCALL *pRtlDeactivateActivationContextUnsafeFast)(PRTL_CALLER_ALLOCATED_ACTIVATION_CONTEXT_STACK_FRAME_EXTENDED);
 static NTSTATUS(NTAPI *pRtlFindActivationContextSectionString)(DWORD,const GUID *,ULONG,PUNICODE_STRING,PACTCTX_SECTION_KEYED_DATA);
 static BOOLEAN (NTAPI *pRtlCreateUnicodeStringFromAsciiz)(PUNICODE_STRING, PCSZ);
 static VOID    (NTAPI *pRtlFreeUnicodeString)(PUNICODE_STRING);
@@ -3096,6 +3099,8 @@ static BOOL init_funcs(void)
     pQueryActCtxSettingsW = (void *)GetProcAddress( hLibrary, "QueryActCtxSettingsW" );

hLibrary = GetModuleHandleA("ntdll.dll");
+    X(RtlActivateActivationContextUnsafeFast);
+    X(RtlDeactivateActivationContextUnsafeFast);
     X(RtlFindActivationContextSectionString);
     X(RtlCreateUnicodeStringFromAsciiz);
     X(RtlFreeUnicodeString);
@@ -4526,6 +4531,81 @@ static void test_RtlQueryInformationActiveActivationContext(void)
     ReleaseActCtx( context );
 }

+static void test_RtlActivateActivationContextUnsafeFast(void)
+{
+    RTL_CALLER_ALLOCATED_ACTIVATION_CONTEXT_STACK_FRAME_EXTENDED frame_extended1 = { 0 }, frame_extended2 = { 0 };
+    HANDLE context1, context2, current_context;
+    RTL_ACTIVATION_CONTEXT_STACK_FRAME *frame;
+    BOOL ret;
+
+    ret = GetCurrentActCtx( &current_context );
+    ok( ret, "GetCurrentActCtx failed.\n" );
+    ok( !current_context, "Got unexpected handle.\n" );
+
+    if (!create_manifest_file( "test1.manifest", manifest1, -1, NULL, NULL ))
+    {
+        skip( "Could not create manifest file 1.\n" );
+        return;
+    }
+    if (!create_manifest_file( "test2.manifest", manifest1_1, -1, NULL, NULL ))
+    {
+        skip( "Could not create manifest file 2.\n" );
+        DeleteFileA( "test1.manifest" );
+        return;
+    }
+    context1 = test_create( "test1.manifest" );
+    ok( context1 != INVALID_HANDLE_VALUE, "Failed to create context, error %lu.\n", GetLastError() );
+    DeleteFileA( "test1.manifest" );
+    context2 = test_create( "test2.manifest" );
+    ok( context2 != INVALID_HANDLE_VALUE, "Failed to create context, error %lu.\n", GetLastError() );
+    DeleteFileA( "test2.manifest" );
+
+    frame_extended1.Size = sizeof(frame_extended1);
+    frame_extended1.Format = 0xdeadbeef;
+    frame = pRtlActivateActivationContextUnsafeFast( &frame_extended1, context1 );
+    /* RtlActivateActivationContextUnsafeFast() on Win7 returns a NTSTATUS */
+    ok( frame == &frame_extended1.Frame || broken( frame == STATUS_SUCCESS ), "Got unexpected frame.\n" );
+    ok( frame_extended1.Size == sizeof(frame_extended1), "Got unexpected Size %#Ix.\n", frame_extended1.Size );
+    ok( frame_extended1.Format == 0xdeadbeef, "Got unexpected Format %#lx.\n", frame_extended1.Format );
+    ok( !frame_extended1.Frame.Previous, "Got unexpected Previous.\n" );
+    ok( frame_extended1.Frame.ActivationContext == context1, "Got unexpected ActivationContext.\n" );
+    ok( frame_extended1.Frame.Flags == 0x20, "Got unexpected Flags %#lx.\n", frame_extended1.Frame.Flags );
+
+    ret = GetCurrentActCtx( &current_context );
+    ok( current_context == context1, "Got unexpected handle.\n" );
+    ReleaseActCtx( current_context );
+
+    frame_extended2.Size = sizeof(frame_extended2);
+    frame_extended2.Format = 0xdeadbeef;
+    frame = pRtlActivateActivationContextUnsafeFast( &frame_extended2, context2 );
+    /* RtlActivateActivationContextUnsafeFast() on Win7 returns a NTSTATUS */
+    ok( frame == &frame_extended2.Frame || broken( frame == STATUS_SUCCESS ), "Got unexpected frame.\n" );
+    ok( frame_extended2.Size == sizeof(frame_extended2), "Got unexpected Size %#Ix.\n", frame_extended2.Size );
+    ok( frame_extended2.Format == 0xdeadbeef, "Got unexpected Format %#lx.\n", frame_extended2.Format );
+    ok( frame_extended2.Frame.Previous == &frame_extended1.Frame, "Got unexpected Previous.\n" );
+    ok( frame_extended2.Frame.ActivationContext == context2, "Got unexpected ActivationContext.\n" );
+    ok( frame_extended2.Frame.Flags == 0x20, "Got unexpected Flags %#lx.\n", frame_extended2.Frame.Flags );
+
+    ret = GetCurrentActCtx( &current_context );
+    ok( current_context == context2, "Got unexpected handle.\n" );
+    ReleaseActCtx( current_context );
+
+    pRtlDeactivateActivationContextUnsafeFast( &frame_extended2 );
+
+    ret = GetCurrentActCtx( &current_context );
+    ok( current_context == context1, "Got unexpected handle.\n" );
+    ReleaseActCtx( current_context );
+
+    pRtlDeactivateActivationContextUnsafeFast( &frame_extended1 );
+
+    ret = GetCurrentActCtx( &current_context );
+    ok( ret, "GetCurrentActCtx failed.\n" );
+    ok( !current_context, "Got unexpected handle.\n" );
+
+    ReleaseActCtx( context2 );
+    ReleaseActCtx( context1 );
+}
+
 START_TEST(actctx)
 {
     int argc;
@@ -4566,5 +4646,6 @@ START_TEST(actctx)
     test_compatibility();
     test_settings();
     test_RtlQueryInformationActiveActivationContext();
+    test_RtlActivateActivationContextUnsafeFast();
     for (int i = 1; i <= 6; i++) run_child_process_two_dll(i);
 }