[reactos] 02/02: [ADVAPI32_APITEST] If TEB SubProcessTag is set, query ADVAPI32 to validate its correctness

[Pierre Schweitzer]

https://git.reactos.org/?p=reactos.git;a=commitdiff;h=dc65db7c1318675dfbfcbd...

commit dc65db7c1318675dfbfcbdf1e1f55fd2f865c290
Author:     Pierre Schweitzer [e-mail]
AuthorDate: Tue Dec 11 23:39:00 2018 +0100
Commit:     Pierre Schweitzer [e-mail]
CommitDate: Tue Dec 11 23:41:01 2018 +0100

[ADVAPI32_APITEST] If TEB SubProcessTag is set, query ADVAPI32 to validate its correctness

This is based on: https://wj32.org/wp/2010/03/30/howto-use-i_querytaginformation/
    and properly works in W2K3 SP2.
    Thanks to Joachim for his tests.
---
 modules/rostests/apitests/advapi32/ServiceEnv.c | 33 +++++++++++++++++++++++++
 1 file changed, 33 insertions(+)

diff --git a/modules/rostests/apitests/advapi32/ServiceEnv.c b/modules/rostests/apitests/advapi32/ServiceEnv.c
index fd3520452f..f71d25541d 100644
--- a/modules/rostests/apitests/advapi32/ServiceEnv.c
+++ b/modules/rostests/apitests/advapi32/ServiceEnv.c
@@ -106,6 +106,39 @@ service_main(DWORD dwArgc, LPWSTR* lpszArgv)

Teb = NtCurrentTeb();
     service_ok(Teb->SubProcessTag != 0, "SubProcessTag is not defined!\n");
+    if (Teb->SubProcessTag != 0)
+    {
+        ULONG (NTAPI *_I_QueryTagInformation)(PVOID, DWORD, PVOID) = (PVOID)GetProcAddress(GetModuleHandle("advapi32.dll"), "I_QueryTagInformation");
+        if (_I_QueryTagInformation != NULL)
+        {
+            /* IN/OUT parameter structure for I_QueryTagInformation() function
+             * See: https://wj32.org/wp/2010/03/30/howto-use-i_querytaginformation/
+             */
+            struct
+            {
+                ULONG ProcessId;
+                PVOID ServiceTag;
+                ULONG Unknown;
+                PWSTR Buffer;
+            } ServiceQuery;
+
+            /* Set our input parameters */
+            ServiceQuery.ProcessId = GetCurrentProcessId();
+            ServiceQuery.ServiceTag = Teb->SubProcessTag;
+            ServiceQuery.Unknown = 0;
+            ServiceQuery.Buffer = NULL;
+            /* Call ADVAPI32 to query the correctness of our tag */
+            _I_QueryTagInformation(NULL, 1, &ServiceQuery);
+
+            /* If buffer is not NULL, call succeed */
+            if (ServiceQuery.Buffer != NULL)
+            {
+                /* It should match our service name */
+                service_ok(wcscmp(lpszArgv[0], ServiceQuery.Buffer) == 0, "Mismatching info: %S - %S\n", lpszArgv[0], ServiceQuery.Buffer);
+                LocalFree(ServiceQuery.Buffer);
+            }
+        }
+    }

/* Work is done */
     report_service_status(SERVICE_STOPPED, NO_ERROR, 0);