| /* | |
| * Unit tests for security functions | |
| * | |
| * Copyright (c) 2004 Mike McCormack | |
| * | |
| * 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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA | |
| */ | |
| #include <stdio.h> | |
| #include "wine/test.h" | |
| #include "windef.h" | |
| #include "winbase.h" | |
| #include "winerror.h" | |
| #include "aclapi.h" | |
| #include "winnt.h" | |
| typedef BOOL (WINAPI *fnBuildTrusteeWithSidA)( TRUSTEE *trustee, PSID psid ); | |
| typedef BOOL (WINAPI *fnBuildTrusteeWithNameA)( TRUSTEE *trustee, LPSTR str ); | |
| typedef BOOL (WINAPI *fnConvertSidToStringSidA)( PSID pSid, LPSTR *str ); | |
| typedef BOOL (WINAPI *fnConvertStringSidToSidA)( LPCSTR str, PSID pSid ); | |
| typedef BOOL (WINAPI *fnGetFileSecurityA)(LPCSTR, SECURITY_INFORMATION, | |
| PSECURITY_DESCRIPTOR, DWORD, LPDWORD); | |
| typedef DWORD (WINAPI *fnRtlAdjustPrivilege)(ULONG,BOOLEAN,BOOLEAN,PBOOLEAN); | |
| static HMODULE hmod; | |
| fnBuildTrusteeWithSidA pBuildTrusteeWithSidA; | |
| fnBuildTrusteeWithNameA pBuildTrusteeWithNameA; | |
| fnConvertSidToStringSidA pConvertSidToStringSidA; | |
| fnConvertStringSidToSidA pConvertStringSidToSidA; | |
| fnGetFileSecurityA pGetFileSecurityA; | |
| fnRtlAdjustPrivilege pRtlAdjustPrivilege; | |
| struct sidRef | |
| { | |
| SID_IDENTIFIER_AUTHORITY auth; | |
| const char *refStr; | |
| }; | |
| static void init(void) | |
| { | |
| hmod = GetModuleHandle("advapi32.dll"); | |
| } | |
| static void test_sid(void) | |
| { | |
| struct sidRef refs[] = { | |
| { { {0x00,0x00,0x33,0x44,0x55,0x66} }, "S-1-860116326-1" }, | |
| { { {0x00,0x00,0x01,0x02,0x03,0x04} }, "S-1-16909060-1" }, | |
| { { {0x00,0x00,0x00,0x01,0x02,0x03} }, "S-1-66051-1" }, | |
| { { {0x00,0x00,0x00,0x00,0x01,0x02} }, "S-1-258-1" }, | |
| { { {0x00,0x00,0x00,0x00,0x00,0x02} }, "S-1-2-1" }, | |
| { { {0x00,0x00,0x00,0x00,0x00,0x0c} }, "S-1-12-1" }, | |
| }; | |
| const char noSubAuthStr[] = "S-1-5"; | |
| unsigned int i; | |
| PSID psid = NULL; | |
| BOOL r; | |
| LPSTR str = NULL; | |
| pConvertSidToStringSidA = (fnConvertSidToStringSidA) | |
| GetProcAddress( hmod, "ConvertSidToStringSidA" ); | |
| if( !pConvertSidToStringSidA ) | |
| return; | |
| pConvertStringSidToSidA = (fnConvertStringSidToSidA) | |
| GetProcAddress( hmod, "ConvertStringSidToSidA" ); | |
| if( !pConvertStringSidToSidA ) | |
| return; | |
| r = pConvertStringSidToSidA( NULL, NULL ); | |
| ok( !r, "expected failure with NULL parameters\n" ); | |
| if( GetLastError() == ERROR_CALL_NOT_IMPLEMENTED ) | |
| return; | |
| ok( GetLastError() == ERROR_INVALID_PARAMETER, | |
| "expected GetLastError() is ERROR_INVALID_PARAMETER, got %ld\n", | |
| GetLastError() ); | |
| r = pConvertStringSidToSidA( refs[0].refStr, NULL ); | |
| ok( !r && GetLastError() == ERROR_INVALID_PARAMETER, | |
| "expected GetLastError() is ERROR_INVALID_PARAMETER, got %ld\n", | |
| GetLastError() ); | |
| r = pConvertStringSidToSidA( NULL, &str ); | |
| ok( !r && GetLastError() == ERROR_INVALID_PARAMETER, | |
| "expected GetLastError() is ERROR_INVALID_PARAMETER, got %ld\n", | |
| GetLastError() ); | |
| r = pConvertStringSidToSidA( noSubAuthStr, &psid ); | |
| ok( !r, | |
| "expected failure with no sub authorities\n" ); | |
| ok( GetLastError() == ERROR_INVALID_SID, | |
| "expected GetLastError() is ERROR_INVALID_SID, got %ld\n", | |
| GetLastError() ); | |
| for( i = 0; i < sizeof(refs) / sizeof(refs[0]); i++ ) | |
| { | |
| PISID pisid; | |
| r = AllocateAndInitializeSid( &refs[i].auth, 1,1,0,0,0,0,0,0,0, | |
| &psid ); | |
| ok( r, "failed to allocate sid\n" ); | |
| r = pConvertSidToStringSidA( psid, &str ); | |
| ok( r, "failed to convert sid\n" ); | |
| if (r) | |
| { | |
| ok( !strcmp( str, refs[i].refStr ), | |
| "incorrect sid, expected %s, got %s\n", refs[i].refStr, str ); | |
| LocalFree( str ); | |
| } | |
| if( psid ) | |
| FreeSid( psid ); | |
| r = pConvertStringSidToSidA( refs[i].refStr, &psid ); | |
| ok( r, "failed to parse sid string\n" ); | |
| pisid = (PISID)psid; | |
| ok( pisid && | |
| !memcmp( pisid->IdentifierAuthority.Value, refs[i].auth.Value, | |
| sizeof(refs[i].auth) ), | |
| "string sid %s didn't parse to expected value\n" | |
| "(got 0x%04x%08lx, expected 0x%04x%08lx)\n", | |
| refs[i].refStr, | |
| MAKEWORD( pisid->IdentifierAuthority.Value[1], | |
| pisid->IdentifierAuthority.Value[0] ), | |
| MAKELONG( MAKEWORD( pisid->IdentifierAuthority.Value[5], | |
| pisid->IdentifierAuthority.Value[4] ), | |
| MAKEWORD( pisid->IdentifierAuthority.Value[3], | |
| pisid->IdentifierAuthority.Value[2] ) ), | |
| MAKEWORD( refs[i].auth.Value[1], refs[i].auth.Value[0] ), | |
| MAKELONG( MAKEWORD( refs[i].auth.Value[5], refs[i].auth.Value[4] ), | |
| MAKEWORD( refs[i].auth.Value[3], refs[i].auth.Value[2] ) ) ); | |
| if( psid ) | |
| LocalFree( psid ); | |
| } | |
| } | |
| static void test_trustee(void) | |
| { | |
| TRUSTEE trustee; | |
| PSID psid; | |
| char str[] = "2jjj"; | |
| SID_IDENTIFIER_AUTHORITY auth = { {0x11,0x22,0,0,0, 0} }; | |
| pBuildTrusteeWithSidA = (fnBuildTrusteeWithSidA) | |
| GetProcAddress( hmod, "BuildTrusteeWithSidA" ); | |
| pBuildTrusteeWithNameA = (fnBuildTrusteeWithNameA) | |
| GetProcAddress( hmod, "BuildTrusteeWithNameA" ); | |
| if( !pBuildTrusteeWithSidA || !pBuildTrusteeWithNameA) | |
| return; | |
| if ( ! AllocateAndInitializeSid( &auth, 1, 42, 0,0,0,0,0,0,0,&psid ) ) | |
| { | |
| trace( "failed to init SID\n" ); | |
| return; | |
| } | |
| memset( &trustee, 0xff, sizeof trustee ); | |
| pBuildTrusteeWithSidA( &trustee, psid ); | |
| ok( trustee.pMultipleTrustee == NULL, "pMultipleTrustee wrong\n"); | |
| ok( trustee.MultipleTrusteeOperation == NO_MULTIPLE_TRUSTEE, | |
| "MultipleTrusteeOperation wrong\n"); | |
| ok( trustee.TrusteeForm == TRUSTEE_IS_SID, "TrusteeForm wrong\n"); | |
| ok( trustee.TrusteeType == TRUSTEE_IS_UNKNOWN, "TrusteeType wrong\n"); | |
| ok( trustee.ptstrName == (LPSTR) psid, "ptstrName wrong\n" ); | |
| FreeSid( psid ); | |
| /* test BuildTrusteeWithNameA */ | |
| memset( &trustee, 0xff, sizeof trustee ); | |
| pBuildTrusteeWithNameA( &trustee, str ); | |
| ok( trustee.pMultipleTrustee == NULL, "pMultipleTrustee wrong\n"); | |
| ok( trustee.MultipleTrusteeOperation == NO_MULTIPLE_TRUSTEE, | |
| "MultipleTrusteeOperation wrong\n"); | |
| ok( trustee.TrusteeForm == TRUSTEE_IS_NAME, "TrusteeForm wrong\n"); | |
| ok( trustee.TrusteeType == TRUSTEE_IS_UNKNOWN, "TrusteeType wrong\n"); | |
| ok( trustee.ptstrName == str, "ptstrName wrong\n" ); | |
| } | |
| /* If the first isn't defined, assume none is */ | |
| #ifndef SE_MIN_WELL_KNOWN_PRIVILEGE | |
| #define SE_MIN_WELL_KNOWN_PRIVILEGE 2L | |
| #define SE_CREATE_TOKEN_PRIVILEGE 2L | |
| #define SE_ASSIGNPRIMARYTOKEN_PRIVILEGE 3L | |
| #define SE_LOCK_MEMORY_PRIVILEGE 4L | |
| #define SE_INCREASE_QUOTA_PRIVILEGE 5L | |
| #define SE_MACHINE_ACCOUNT_PRIVILEGE 6L | |
| #define SE_TCB_PRIVILEGE 7L | |
| #define SE_SECURITY_PRIVILEGE 8L | |
| #define SE_TAKE_OWNERSHIP_PRIVILEGE 9L | |
| #define SE_LOAD_DRIVER_PRIVILEGE 10L | |
| #define SE_SYSTEM_PROFILE_PRIVILEGE 11L | |
| #define SE_SYSTEMTIME_PRIVILEGE 12L | |
| #define SE_PROF_SINGLE_PROCESS_PRIVILEGE 13L | |
| #define SE_INC_BASE_PRIORITY_PRIVILEGE 14L | |
| #define SE_CREATE_PAGEFILE_PRIVILEGE 15L | |
| #define SE_CREATE_PERMANENT_PRIVILEGE 16L | |
| #define SE_BACKUP_PRIVILEGE 17L | |
| #define SE_RESTORE_PRIVILEGE 18L | |
| #define SE_SHUTDOWN_PRIVILEGE 19L | |
| #define SE_DEBUG_PRIVILEGE 20L | |
| #define SE_AUDIT_PRIVILEGE 21L | |
| #define SE_SYSTEM_ENVIRONMENT_PRIVILEGE 22L | |
| #define SE_CHANGE_NOTIFY_PRIVILLEGE 23L | |
| #define SE_REMOTE_SHUTDOWN_PRIVILEGE 24L | |
| #define SE_UNDOCK_PRIVILEGE 25L | |
| #define SE_SYNC_AGENT_PRIVILEGE 26L | |
| #define SE_ENABLE_DELEGATION_PRIVILEGE 27L | |
| #define SE_MANAGE_VOLUME_PRIVILEGE 28L | |
| #define SE_IMPERSONATE_PRIVILEGE 29L | |
| #define SE_CREATE_GLOBAL_PRIVILEGE 30L | |
| #define SE_MAX_WELL_KNOWN_PRIVILEGE SE_CREATE_GLOBAL_PRIVILEGE | |
| #endif /* ndef SE_MIN_WELL_KNOWN_PRIVILEGE */ | |
| static void test_allocateLuid(void) | |
| { | |
| BOOL (WINAPI *pAllocateLocallyUniqueId)(PLUID); | |
| LUID luid1, luid2; | |
| BOOL ret; | |
| pAllocateLocallyUniqueId = (void*)GetProcAddress(hmod, "AllocateLocallyUniqueId"); | |
| if (!pAllocateLocallyUniqueId) return; | |
| ret = pAllocateLocallyUniqueId(&luid1); | |
| if (!ret && GetLastError() == ERROR_CALL_NOT_IMPLEMENTED) | |
| return; | |
| ok(ret, | |
| "AllocateLocallyUniqueId failed: %ld\n", GetLastError()); | |
| ret = pAllocateLocallyUniqueId(&luid2); | |
| ok( ret, | |
| "AllocateLocallyUniqueId failed: %ld\n", GetLastError()); | |
| ok(luid1.LowPart > SE_MAX_WELL_KNOWN_PRIVILEGE || luid1.HighPart != 0, | |
| "AllocateLocallyUniqueId returned a well-known LUID\n"); | |
| ok(luid1.LowPart != luid2.LowPart || luid1.HighPart != luid2.HighPart, | |
| "AllocateLocallyUniqueId returned non-unique LUIDs\n"); | |
| ret = pAllocateLocallyUniqueId(NULL); | |
| ok( !ret && GetLastError() == ERROR_NOACCESS, | |
| "AllocateLocallyUniqueId(NULL) didn't return ERROR_NOACCESS: %ld\n", | |
| GetLastError()); | |
| } | |
| static void test_lookupPrivilegeName(void) | |
| { | |
| BOOL (WINAPI *pLookupPrivilegeNameA)(LPCSTR, PLUID, LPSTR, LPDWORD); | |
| char buf[MAX_PATH]; /* arbitrary, seems long enough */ | |
| DWORD cchName = sizeof(buf); | |
| LUID luid = { 0, 0 }; | |
| LONG i; | |
| BOOL ret; | |
| /* check whether it's available first */ | |
| pLookupPrivilegeNameA = (void*)GetProcAddress(hmod, "LookupPrivilegeNameA"); | |
| if (!pLookupPrivilegeNameA) return; | |
| luid.LowPart = SE_CREATE_TOKEN_PRIVILEGE; | |
| ret = pLookupPrivilegeNameA(NULL, &luid, buf, &cchName); | |
| if (!ret && GetLastError() == ERROR_CALL_NOT_IMPLEMENTED) | |
| return; | |
| /* check with a short buffer */ | |
| cchName = 0; | |
| luid.LowPart = SE_CREATE_TOKEN_PRIVILEGE; | |
| ret = pLookupPrivilegeNameA(NULL, &luid, NULL, &cchName); | |
| ok( !ret && GetLastError() == ERROR_INSUFFICIENT_BUFFER, | |
| "LookupPrivilegeNameA didn't fail with ERROR_INSUFFICIENT_BUFFER: %ld\n", | |
| GetLastError()); | |
| ok(cchName == strlen("SeCreateTokenPrivilege") + 1, | |
| "LookupPrivilegeNameA returned an incorrect required length for\n" | |
| "SeCreateTokenPrivilege (got %ld, expected %d)\n", cchName, | |
| strlen("SeCreateTokenPrivilege") + 1); | |
| /* check a known value and its returned length on success */ | |
| cchName = sizeof(buf); | |
| ok(pLookupPrivilegeNameA(NULL, &luid, buf, &cchName) && | |
| cchName == strlen("SeCreateTokenPrivilege"), | |
| "LookupPrivilegeNameA returned an incorrect output length for\n" | |
| "SeCreateTokenPrivilege (got %ld, expected %d)\n", cchName, | |
| (int)strlen("SeCreateTokenPrivilege")); | |
| /* check known values */ | |
| for (i = SE_MIN_WELL_KNOWN_PRIVILEGE; i < SE_MAX_WELL_KNOWN_PRIVILEGE; i++) | |
| { | |
| luid.LowPart = i; | |
| cchName = sizeof(buf); | |
| ret = pLookupPrivilegeNameA(NULL, &luid, buf, &cchName); | |
| ok( ret || GetLastError() == ERROR_NO_SUCH_PRIVILEGE, | |
| "LookupPrivilegeNameA(0.%ld) failed: %ld\n", i, GetLastError()); | |
| } | |
| /* check a bogus LUID */ | |
| luid.LowPart = 0xdeadbeef; | |
| cchName = sizeof(buf); | |
| ret = pLookupPrivilegeNameA(NULL, &luid, buf, &cchName); | |
| ok( !ret && GetLastError() == ERROR_NO_SUCH_PRIVILEGE, | |
| "LookupPrivilegeNameA didn't fail with ERROR_NO_SUCH_PRIVILEGE: %ld\n", | |
| GetLastError()); | |
| /* check on a bogus system */ | |
| luid.LowPart = SE_CREATE_TOKEN_PRIVILEGE; | |
| cchName = sizeof(buf); | |
| ret = pLookupPrivilegeNameA("b0gu5.Nam3", &luid, buf, &cchName); | |
| ok( !ret && GetLastError() == RPC_S_SERVER_UNAVAILABLE, | |
| "LookupPrivilegeNameA didn't fail with RPC_S_SERVER_UNAVAILABLE: %ld\n", | |
| GetLastError()); | |
| } | |
| struct NameToLUID | |
| { | |
| const char *name; | |
| DWORD lowPart; | |
| }; | |
| static void test_lookupPrivilegeValue(void) | |
| { | |
| static const struct NameToLUID privs[] = { | |
| { "SeCreateTokenPrivilege", SE_CREATE_TOKEN_PRIVILEGE }, | |
| { "SeAssignPrimaryTokenPrivilege", SE_ASSIGNPRIMARYTOKEN_PRIVILEGE }, | |
| { "SeLockMemoryPrivilege", SE_LOCK_MEMORY_PRIVILEGE }, | |
| { "SeIncreaseQuotaPrivilege", SE_INCREASE_QUOTA_PRIVILEGE }, | |
| { "SeMachineAccountPrivilege", SE_MACHINE_ACCOUNT_PRIVILEGE }, | |
| { "SeTcbPrivilege", SE_TCB_PRIVILEGE }, | |
| { "SeSecurityPrivilege", SE_SECURITY_PRIVILEGE }, | |
| { "SeTakeOwnershipPrivilege", SE_TAKE_OWNERSHIP_PRIVILEGE }, | |
| { "SeLoadDriverPrivilege", SE_LOAD_DRIVER_PRIVILEGE }, | |
| { "SeSystemProfilePrivilege", SE_SYSTEM_PROFILE_PRIVILEGE }, | |
| { "SeSystemtimePrivilege", SE_SYSTEMTIME_PRIVILEGE }, | |
| { "SeProfileSingleProcessPrivilege", SE_PROF_SINGLE_PROCESS_PRIVILEGE }, | |
| { "SeIncreaseBasePriorityPrivilege", SE_INC_BASE_PRIORITY_PRIVILEGE }, | |
| { "SeCreatePagefilePrivilege", SE_CREATE_PAGEFILE_PRIVILEGE }, | |
| { "SeCreatePermanentPrivilege", SE_CREATE_PERMANENT_PRIVILEGE }, | |
| { "SeBackupPrivilege", SE_BACKUP_PRIVILEGE }, | |
| { "SeRestorePrivilege", SE_RESTORE_PRIVILEGE }, | |
| { "SeShutdownPrivilege", SE_SHUTDOWN_PRIVILEGE }, | |
| { "SeDebugPrivilege", SE_DEBUG_PRIVILEGE }, | |
| { "SeAuditPrivilege", SE_AUDIT_PRIVILEGE }, | |
| { "SeSystemEnvironmentPrivilege", SE_SYSTEM_ENVIRONMENT_PRIVILEGE }, | |
| { "SeChangeNotifyPrivilege", SE_CHANGE_NOTIFY_PRIVILLEGE }, | |
| { "SeRemoteShutdownPrivilege", SE_REMOTE_SHUTDOWN_PRIVILEGE }, | |
| { "SeUndockPrivilege", SE_UNDOCK_PRIVILEGE }, | |
| { "SeSyncAgentPrivilege", SE_SYNC_AGENT_PRIVILEGE }, | |
| { "SeEnableDelegationPrivilege", SE_ENABLE_DELEGATION_PRIVILEGE }, | |
| { "SeManageVolumePrivilege", SE_MANAGE_VOLUME_PRIVILEGE }, | |
| { "SeImpersonatePrivilege", SE_IMPERSONATE_PRIVILEGE }, | |
| { "SeCreateGlobalPrivilege", SE_CREATE_GLOBAL_PRIVILEGE }, | |
| }; | |
| BOOL (WINAPI *pLookupPrivilegeValueA)(LPCSTR, LPCSTR, PLUID); | |
| int i; | |
| LUID luid; | |
| BOOL ret; | |
| /* check whether it's available first */ | |
| pLookupPrivilegeValueA = (void*)GetProcAddress(hmod, "LookupPrivilegeValueA"); | |
| if (!pLookupPrivilegeValueA) return; | |
| ret = pLookupPrivilegeValueA(NULL, "SeCreateTokenPrivilege", &luid); | |
| if (!ret && GetLastError() == ERROR_CALL_NOT_IMPLEMENTED) | |
| return; | |
| /* check a bogus system name */ | |
| ret = pLookupPrivilegeValueA("b0gu5.Nam3", "SeCreateTokenPrivilege", &luid); | |
| ok( !ret && GetLastError() == RPC_S_SERVER_UNAVAILABLE, | |
| "LookupPrivilegeValueA didn't fail with RPC_S_SERVER_UNAVAILABLE: %ld\n", | |
| GetLastError()); | |
| /* check a NULL string */ | |
| ret = pLookupPrivilegeValueA(NULL, 0, &luid); | |
| ok( !ret && GetLastError() == ERROR_NO_SUCH_PRIVILEGE, | |
| "LookupPrivilegeValueA didn't fail with ERROR_NO_SUCH_PRIVILEGE: %ld\n", | |
| GetLastError()); | |
| /* check a bogus privilege name */ | |
| ret = pLookupPrivilegeValueA(NULL, "SeBogusPrivilege", &luid); | |
| ok( !ret && GetLastError() == ERROR_NO_SUCH_PRIVILEGE, | |
| "LookupPrivilegeValueA didn't fail with ERROR_NO_SUCH_PRIVILEGE: %ld\n", | |
| GetLastError()); | |
| /* check case insensitive */ | |
| ret = pLookupPrivilegeValueA(NULL, "sEcREATEtOKENpRIVILEGE", &luid); | |
| ok( ret, | |
| "LookupPrivilegeValueA(NULL, sEcREATEtOKENpRIVILEGE, &luid) failed: %ld\n", | |
| GetLastError()); | |
| for (i = 0; i < sizeof(privs) / sizeof(privs[0]); i++) | |
| { | |
| /* Not all privileges are implemented on all Windows versions, so | |
| * don't worry if the call fails | |
| */ | |
| if (pLookupPrivilegeValueA(NULL, privs[i].name, &luid)) | |
| { | |
| ok(luid.LowPart == privs[i].lowPart, | |
| "LookupPrivilegeValueA returned an invalid LUID for %s\n", | |
| privs[i].name); | |
| } | |
| } | |
| } | |
| static void test_luid(void) | |
| { | |
| test_allocateLuid(); | |
| test_lookupPrivilegeName(); | |
| test_lookupPrivilegeValue(); | |
| } | |
| static void test_FileSecurity(void) | |
| { | |
| char directory[MAX_PATH]; | |
| DWORD retval, outSize; | |
| BOOL result; | |
| BYTE buffer[0x40]; | |
| pGetFileSecurityA = (fnGetFileSecurityA) | |
| GetProcAddress( hmod, "GetFileSecurityA" ); | |
| if( !pGetFileSecurityA ) | |
| return; | |
| retval = GetTempPathA(sizeof(directory), directory); | |
| if (!retval) { | |
| trace("GetTempPathA failed\n"); | |
| return; | |
| } | |
| strcpy(directory, "\\Should not exist"); | |
| SetLastError(NO_ERROR); | |
| result = pGetFileSecurityA( directory,OWNER_SECURITY_INFORMATION,buffer,0x40,&outSize); | |
| ok(!result, "GetFileSecurityA should fail for not existing directories/files\n"); | |
| ok( (GetLastError() == ERROR_FILE_NOT_FOUND ) || | |
| (GetLastError() == ERROR_CALL_NOT_IMPLEMENTED) , | |
| "last error ERROR_FILE_NOT_FOUND / ERROR_CALL_NOT_IMPLEMENTED (98) " | |
| "expected, got %ld\n", GetLastError()); | |
| } | |
| static void test_AccessCheck(void) | |
| { | |
| PSID EveryoneSid = NULL, AdminSid = NULL, UsersSid = NULL; | |
| PACL Acl = NULL; | |
| SECURITY_DESCRIPTOR *SecurityDescriptor = NULL; | |
| SID_IDENTIFIER_AUTHORITY SIDAuthWorld = { SECURITY_WORLD_SID_AUTHORITY }; | |
| SID_IDENTIFIER_AUTHORITY SIDAuthNT = { SECURITY_NT_AUTHORITY }; | |
| GENERIC_MAPPING Mapping = { KEY_READ, KEY_WRITE, KEY_EXECUTE, KEY_ALL_ACCESS }; | |
| ACCESS_MASK Access; | |
| BOOL AccessStatus; | |
| HANDLE Token; | |
| BOOL ret; | |
| DWORD PrivSetLen; | |
| PRIVILEGE_SET *PrivSet; | |
| BOOL res; | |
| HMODULE NtDllModule; | |
| BOOLEAN Enabled; | |
| NtDllModule = GetModuleHandle("ntdll.dll"); | |
| if (!NtDllModule) | |
| { | |
| trace("not running on NT, skipping test\n"); | |
| return; | |
| } | |
| pRtlAdjustPrivilege = (fnRtlAdjustPrivilege) | |
| GetProcAddress(NtDllModule, "RtlAdjustPrivilege"); | |
| if (!pRtlAdjustPrivilege) return; | |
| Acl = HeapAlloc(GetProcessHeap(), 0, 256); | |
| res = InitializeAcl(Acl, 256, ACL_REVISION); | |
| if(!res && GetLastError() == ERROR_CALL_NOT_IMPLEMENTED) | |
| { | |
| trace("ACLs not implemented - skipping tests\n"); | |
| return; | |
| } | |
| ok(res, "InitializeAcl failed with error %ld\n", GetLastError()); | |
| res = AllocateAndInitializeSid( &SIDAuthWorld, 1, SECURITY_WORLD_RID, 0, 0, 0, 0, 0, 0, 0, &EveryoneSid); | |
| ok(res, "AllocateAndInitializeSid failed with error %ld\n", GetLastError()); | |
| res = AllocateAndInitializeSid( &SIDAuthNT, 2, SECURITY_BUILTIN_DOMAIN_RID, | |
| DOMAIN_ALIAS_RID_ADMINS, 0, 0, 0, 0, 0, 0, &AdminSid); | |
| ok(res, "AllocateAndInitializeSid failed with error %ld\n", GetLastError()); | |
| res = AllocateAndInitializeSid( &SIDAuthNT, 2, SECURITY_BUILTIN_DOMAIN_RID, | |
| DOMAIN_ALIAS_RID_USERS, 0, 0, 0, 0, 0, 0, &UsersSid); | |
| ok(res, "AllocateAndInitializeSid failed with error %ld\n", GetLastError()); | |
| res = AddAccessAllowedAce(Acl, ACL_REVISION, KEY_READ, EveryoneSid); | |
| ok(res, "AddAccessAllowedAceEx failed with error %ld\n", GetLastError()); | |
| res = AddAccessAllowedAce(Acl, ACL_REVISION, KEY_ALL_ACCESS, AdminSid); | |
| ok(res, "AddAccessAllowedAceEx failed with error %ld\n", GetLastError()); | |
| SecurityDescriptor = HeapAlloc(GetProcessHeap(), 0, SECURITY_DESCRIPTOR_MIN_LENGTH); | |
| res = InitializeSecurityDescriptor(SecurityDescriptor, SECURITY_DESCRIPTOR_REVISION); | |
| ok(res, "InitializeSecurityDescriptor failed with error %ld\n", GetLastError()); | |
| res = SetSecurityDescriptorDacl(SecurityDescriptor, TRUE, Acl, FALSE); | |
| ok(res, "SetSecurityDescriptorDacl failed with error %ld\n", GetLastError()); | |
| res = SetSecurityDescriptorOwner(SecurityDescriptor, AdminSid, FALSE); | |
| ok(res, "SetSecurityDescriptorOwner failed with error %ld\n", GetLastError()); | |
| res = SetSecurityDescriptorGroup(SecurityDescriptor, UsersSid, TRUE); | |
| ok(res, "SetSecurityDescriptorGroup failed with error %ld\n", GetLastError()); | |
| PrivSetLen = FIELD_OFFSET(PRIVILEGE_SET, Privilege[16]); | |
| PrivSet = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, PrivSetLen); | |
| PrivSet->PrivilegeCount = 16; | |
| ImpersonateSelf(SecurityImpersonation); | |
| pRtlAdjustPrivilege(SE_SECURITY_PRIVILEGE, FALSE, TRUE, &Enabled); | |
| ret = OpenThreadToken(GetCurrentThread(), | |
| TOKEN_QUERY, TRUE, &Token); | |
| ok(ret, "OpenThreadToken failed with error %ld\n", GetLastError()); | |
| ret = AccessCheck(SecurityDescriptor, Token, KEY_READ, &Mapping, | |
| PrivSet, &PrivSetLen, &Access, &AccessStatus); | |
| ok(ret, "AccessCheck failed with error %ld\n", GetLastError()); | |
| ok(AccessStatus && (Access == KEY_READ), | |
| "AccessCheck failed to grant access with error %ld\n", | |
| GetLastError()); | |
| ret = AccessCheck(SecurityDescriptor, Token, MAXIMUM_ALLOWED, &Mapping, | |
| PrivSet, &PrivSetLen, &Access, &AccessStatus); | |
| ok(ret, "AccessCheck failed with error %ld\n", GetLastError()); | |
| ok(AccessStatus, | |
| "AccessCheck failed to grant any access with error %ld\n", | |
| GetLastError()); | |
| trace("AccessCheck with MAXIMUM_ALLOWED got Access 0x%08lx\n", Access); | |
| SetLastError(0); | |
| PrivSet->PrivilegeCount = 16; | |
| ret = AccessCheck(SecurityDescriptor, Token, ACCESS_SYSTEM_SECURITY, &Mapping, | |
| PrivSet, &PrivSetLen, &Access, &AccessStatus); | |
| ok(ret && !AccessStatus && GetLastError() == ERROR_PRIVILEGE_NOT_HELD, | |
| "AccessCheck should have failed with ERROR_PRIVILEGE_NOT_HELD, instead of %ld\n", | |
| GetLastError()); | |
| ret = pRtlAdjustPrivilege(SE_SECURITY_PRIVILEGE, TRUE, TRUE, &Enabled); | |
| if (!ret) | |
| { | |
| SetLastError(0); | |
| PrivSet->PrivilegeCount = 16; | |
| ret = AccessCheck(SecurityDescriptor, Token, ACCESS_SYSTEM_SECURITY, &Mapping, | |
| PrivSet, &PrivSetLen, &Access, &AccessStatus); | |
| ok(ret && AccessStatus && GetLastError() == 0, | |
| "AccessCheck should have succeeded, error %ld\n", | |
| GetLastError()); | |
| ok(Access == ACCESS_SYSTEM_SECURITY, | |
| "Access should be equal to ACCESS_SYSTEM_SECURITY instead of 0x%08lx\n", | |
| Access); | |
| } | |
| else | |
| trace("Couldn't get SE_SECURITY_PRIVILEGE (0x%08x), skipping ACCESS_SYSTEM_SECURITY test\n", | |
| ret); | |
| RevertToSelf(); | |
| if (EveryoneSid) | |
| FreeSid(EveryoneSid); | |
| if (AdminSid) | |
| FreeSid(AdminSid); | |
| if (UsersSid) | |
| FreeSid(UsersSid); | |
| HeapFree(GetProcessHeap(), 0, Acl); | |
| HeapFree(GetProcessHeap(), 0, SecurityDescriptor); | |
| HeapFree(GetProcessHeap(), 0, PrivSet); | |
| } | |
| START_TEST(security) | |
| { | |
| init(); | |
| if (!hmod) return; | |
| test_sid(); | |
| test_trustee(); | |
| test_luid(); | |
| test_FileSecurity(); | |
| test_AccessCheck(); | |
| } |