wine/dlls/kernelbase/tests/security.c

99 lines
3.3 KiB
C

/*
* 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();
}