[PATCH 5/5] kernelbase/tests: Add test for DeriveCapabilitySidsFromName().

[Paul Gofman]

From: Paul Gofman

---
 dlls/kernelbase/tests/Makefile.in |  1 +
 dlls/kernelbase/tests/security.c  | 99 +++++++++++++++++++++++++++++++
 2 files changed, 100 insertions(+)
 create mode 100644 dlls/kernelbase/tests/security.c

diff --git a/dlls/kernelbase/tests/Makefile.in b/dlls/kernelbase/tests/Makefile.in
index b51eaefcd13..88a86b70094 100644
--- a/dlls/kernelbase/tests/Makefile.in
+++ b/dlls/kernelbase/tests/Makefile.in
@@ -4,6 +4,7 @@ C_SRCS = \
    file.c \
    path.c \
    process.c \
+	security.c \
    sync.c

RC_SRCS = \
diff --git a/dlls/kernelbase/tests/security.c b/dlls/kernelbase/tests/security.c
new file mode 100644
index 00000000000..e5f1ab93b01
--- /dev/null
+++ b/dlls/kernelbase/tests/security.c
@@ -0,0 +1,99 @@
+/*
+ * Copyright (C) 2023 Paul Gofman for CodeWeavers
+ *
+ * This library is free software; you can redistribute it and/or
+ * modify it under the terms of the GNU Lesser General Public
+ * License as published by the Free Software Foundation; either
+ * version 2.1 of the License, or (at your option) any later version.
+ *
+ * This library is distributed in the hope that it will be useful,
+ * but WITHOUT ANY WARRANTY; without even the implied warranty of
+ * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
+ * Lesser General Public License for more details.
+ *
+ * You should have received a copy of the GNU Lesser General Public
+ * License along with this library; if not, write to the Free Software
+ * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
+ */
+
+#include <stdarg.h>
+#include <stdlib.h>
+
+#include <ntstatus.h>
+#define WIN32_NO_STATUS
+#include <windef.h>
+#include <winbase.h>
+#include <winerror.h>
+#include <winternl.h>
+
+#include "wine/test.h"
+
+static BOOL (WINAPI *pDeriveCapabilitySidsFromName)(const WCHAR *, PSID **, DWORD *, PSID **, DWORD *);
+
+static NTSTATUS (WINAPI *pRtlDeriveCapabilitySidsFromName)(UNICODE_STRING *, PSID, PSID);
+
+static void test_DeriveCapabilitySidsFromName(void)
+{
+    BYTE auth_count_sid, auth_count_group_sid;
+    PSID *check_group_sid, *check_sid;
+    DWORD sid_count, group_sid_count;
+    PSID *group_sids, *sids;
+    UNICODE_STRING name_us;
+    NTSTATUS status;
+    DWORD size;
+    BOOL bret;
+
+    if (!pDeriveCapabilitySidsFromName)
+    {
+        win_skip ("DeriveCapabilitySidsFromName is not available.\n");
+        return;
+    }
+
+    if (0)
+    {
+        /* Crashes on Windows. */
+        pDeriveCapabilitySidsFromName(L"test", NULL, &group_sid_count, NULL, &sid_count);
+    }
+
+    sid_count = group_sid_count = 0xdeadbeef;
+    SetLastError(0xdeadbeef);
+    bret = pDeriveCapabilitySidsFromName(L"test", &group_sids, &group_sid_count, &sids, &sid_count);
+    ok(bret && GetLastError() == 0xdeadbeef, "got bret %d, err %lu.\n", bret, GetLastError());
+    ok(group_sid_count == 1, "got %lu.\n", group_sid_count);
+    ok(sid_count == 1, "got %lu.\n", sid_count);
+
+    auth_count_sid = *RtlSubAuthorityCountSid(sids[0]);
+    auth_count_group_sid = *RtlSubAuthorityCountSid(group_sids[0]);
+
+    size = RtlLengthRequiredSid(auth_count_sid);
+    check_sid = malloc( size );
+    size = RtlLengthRequiredSid(auth_count_group_sid);
+    check_group_sid = malloc( size );
+
+    RtlInitUnicodeString(&name_us, L"test");
+    status = pRtlDeriveCapabilitySidsFromName(&name_us, check_group_sid, check_sid);
+    ok(!status, "failed, status %#lx.\n", status);
+    ok(!memcmp(sids[0], check_sid, RtlLengthRequiredSid(auth_count_sid)), "mismatch.\n");
+    ok(!memcmp(group_sids[0], check_group_sid, RtlLengthRequiredSid(auth_count_group_sid)), "mismatch.\n");
+
+    free(check_sid);
+    free(check_group_sid);
+
+    LocalFree(group_sids[0]);
+    LocalFree(group_sids);
+    LocalFree(sids[0]);
+    LocalFree(sids);
+}
+
+START_TEST(security)
+{
+    HMODULE hmod;
+
+    hmod = LoadLibraryA("kernelbase.dll");
+    pDeriveCapabilitySidsFromName = (void *)GetProcAddress(hmod, "DeriveCapabilitySidsFromName");
+
+    hmod = LoadLibraryA("ntdll.dll");
+    pRtlDeriveCapabilitySidsFromName = (void *)GetProcAddress(hmod, "RtlDeriveCapabilitySidsFromName");
+
+    test_DeriveCapabilitySidsFromName();
+}