jscript: Fix typos in comments, add missing ones.
[wine] / dlls / advapi32 / tests / security.c
1 /*
2  * Unit tests for security functions
3  *
4  * Copyright (c) 2004 Mike McCormack
5  *
6  * This library is free software; you can redistribute it and/or
7  * modify it under the terms of the GNU Lesser General Public
8  * License as published by the Free Software Foundation; either
9  * version 2.1 of the License, or (at your option) any later version.
10  *
11  * This library is distributed in the hope that it will be useful,
12  * but WITHOUT ANY WARRANTY; without even the implied warranty of
13  * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
14  * Lesser General Public License for more details.
15  *
16  * You should have received a copy of the GNU Lesser General Public
17  * License along with this library; if not, write to the Free Software
18  * Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
19  */
20
21 #include <stdarg.h>
22 #include <stdio.h>
23
24 #include "ntstatus.h"
25 #define WIN32_NO_STATUS
26 #include "windef.h"
27 #include "winbase.h"
28 #include "winerror.h"
29 #include "aclapi.h"
30 #include "winnt.h"
31 #include "sddl.h"
32 #include "ntsecapi.h"
33 #include "lmcons.h"
34
35 #include "wine/test.h"
36
37 /* copied from Wine winternl.h - not included in the Windows SDK */
38 typedef enum _OBJECT_INFORMATION_CLASS {
39     ObjectBasicInformation,
40     ObjectNameInformation,
41     ObjectTypeInformation,
42     ObjectAllInformation,
43     ObjectDataInformation
44 } OBJECT_INFORMATION_CLASS, *POBJECT_INFORMATION_CLASS;
45
46 typedef struct _OBJECT_BASIC_INFORMATION {
47     ULONG  Attributes;
48     ACCESS_MASK  GrantedAccess;
49     ULONG  HandleCount;
50     ULONG  PointerCount;
51     ULONG  PagedPoolUsage;
52     ULONG  NonPagedPoolUsage;
53     ULONG  Reserved[3];
54     ULONG  NameInformationLength;
55     ULONG  TypeInformationLength;
56     ULONG  SecurityDescriptorLength;
57     LARGE_INTEGER  CreateTime;
58 } OBJECT_BASIC_INFORMATION, *POBJECT_BASIC_INFORMATION;
59
60 #define expect_eq(expr, value, type, format) { type ret = expr; ok((value) == ret, #expr " expected " format "  got " format "\n", (value), (ret)); }
61
62 static BOOL (WINAPI *pAddAccessAllowedAceEx)(PACL, DWORD, DWORD, DWORD, PSID);
63 static BOOL (WINAPI *pAddAccessDeniedAceEx)(PACL, DWORD, DWORD, DWORD, PSID);
64 static BOOL (WINAPI *pAddAuditAccessAceEx)(PACL, DWORD, DWORD, DWORD, PSID, BOOL, BOOL);
65 typedef VOID (WINAPI *fnBuildTrusteeWithSidA)( PTRUSTEEA pTrustee, PSID pSid );
66 typedef VOID (WINAPI *fnBuildTrusteeWithNameA)( PTRUSTEEA pTrustee, LPSTR pName );
67 typedef VOID (WINAPI *fnBuildTrusteeWithObjectsAndNameA)( PTRUSTEEA pTrustee,
68                                                           POBJECTS_AND_NAME_A pObjName,
69                                                           SE_OBJECT_TYPE ObjectType,
70                                                           LPSTR ObjectTypeName,
71                                                           LPSTR InheritedObjectTypeName,
72                                                           LPSTR Name );
73 typedef VOID (WINAPI *fnBuildTrusteeWithObjectsAndSidA)( PTRUSTEEA pTrustee,
74                                                          POBJECTS_AND_SID pObjSid,
75                                                          GUID* pObjectGuid,
76                                                          GUID* pInheritedObjectGuid,
77                                                          PSID pSid );
78 typedef LPSTR (WINAPI *fnGetTrusteeNameA)( PTRUSTEEA pTrustee );
79 typedef BOOL (WINAPI *fnMakeSelfRelativeSD)( PSECURITY_DESCRIPTOR, PSECURITY_DESCRIPTOR, LPDWORD );
80 typedef BOOL (WINAPI *fnConvertSidToStringSidA)( PSID pSid, LPSTR *str );
81 typedef BOOL (WINAPI *fnConvertStringSidToSidA)( LPCSTR str, PSID pSid );
82 static BOOL (WINAPI *pConvertStringSecurityDescriptorToSecurityDescriptorA)(LPCSTR, DWORD,
83                                                                             PSECURITY_DESCRIPTOR*, PULONG );
84 static BOOL (WINAPI *pConvertStringSecurityDescriptorToSecurityDescriptorW)(LPCWSTR, DWORD,
85                                                                             PSECURITY_DESCRIPTOR*, PULONG );
86 static BOOL (WINAPI *pConvertSecurityDescriptorToStringSecurityDescriptorA)(PSECURITY_DESCRIPTOR, DWORD,
87                                                                             SECURITY_INFORMATION, LPSTR *, PULONG );
88 typedef BOOL (WINAPI *fnGetFileSecurityA)(LPCSTR, SECURITY_INFORMATION,
89                                           PSECURITY_DESCRIPTOR, DWORD, LPDWORD);
90 typedef BOOL (WINAPI *fnSetFileSecurityA)(LPCSTR, SECURITY_INFORMATION,
91                                           PSECURITY_DESCRIPTOR);
92 static DWORD (WINAPI *pGetNamedSecurityInfoA)(LPSTR, SE_OBJECT_TYPE, SECURITY_INFORMATION,
93                                               PSID*, PSID*, PACL*, PACL*,
94                                               PSECURITY_DESCRIPTOR*);
95 typedef DWORD (WINAPI *fnRtlAdjustPrivilege)(ULONG,BOOLEAN,BOOLEAN,PBOOLEAN);
96 typedef BOOL (WINAPI *fnCreateWellKnownSid)(WELL_KNOWN_SID_TYPE,PSID,PSID,DWORD*);
97 typedef BOOL (WINAPI *fnDuplicateTokenEx)(HANDLE,DWORD,LPSECURITY_ATTRIBUTES,
98                                         SECURITY_IMPERSONATION_LEVEL,TOKEN_TYPE,PHANDLE);
99
100 typedef NTSTATUS (WINAPI *fnLsaQueryInformationPolicy)(LSA_HANDLE,POLICY_INFORMATION_CLASS,PVOID*);
101 typedef NTSTATUS (WINAPI *fnLsaClose)(LSA_HANDLE);
102 typedef NTSTATUS (WINAPI *fnLsaFreeMemory)(PVOID);
103 typedef NTSTATUS (WINAPI *fnLsaOpenPolicy)(PLSA_UNICODE_STRING,PLSA_OBJECT_ATTRIBUTES,ACCESS_MASK,PLSA_HANDLE);
104 static NTSTATUS (WINAPI *pNtQueryObject)(HANDLE,OBJECT_INFORMATION_CLASS,PVOID,ULONG,PULONG);
105 static DWORD (WINAPI *pSetEntriesInAclW)(ULONG, PEXPLICIT_ACCESSW, PACL, PACL*);
106 static BOOL (WINAPI *pSetSecurityDescriptorControl)(PSECURITY_DESCRIPTOR, SECURITY_DESCRIPTOR_CONTROL,
107                                                     SECURITY_DESCRIPTOR_CONTROL);
108 static DWORD (WINAPI *pGetSecurityInfo)(HANDLE, SE_OBJECT_TYPE, SECURITY_INFORMATION,
109                                         PSID*, PSID*, PACL*, PACL*, PSECURITY_DESCRIPTOR*);
110 static NTSTATUS (WINAPI *pNtAccessCheck)(PSECURITY_DESCRIPTOR, HANDLE, ACCESS_MASK, PGENERIC_MAPPING,
111                                          PPRIVILEGE_SET, PULONG, PULONG, NTSTATUS*);
112
113 static HMODULE hmod;
114 static int     myARGC;
115 static char**  myARGV;
116
117 fnBuildTrusteeWithSidA   pBuildTrusteeWithSidA;
118 fnBuildTrusteeWithNameA  pBuildTrusteeWithNameA;
119 fnBuildTrusteeWithObjectsAndNameA pBuildTrusteeWithObjectsAndNameA;
120 fnBuildTrusteeWithObjectsAndSidA pBuildTrusteeWithObjectsAndSidA;
121 fnGetTrusteeNameA pGetTrusteeNameA;
122 fnMakeSelfRelativeSD pMakeSelfRelativeSD;
123 fnConvertSidToStringSidA pConvertSidToStringSidA;
124 fnConvertStringSidToSidA pConvertStringSidToSidA;
125 fnGetFileSecurityA pGetFileSecurityA;
126 fnSetFileSecurityA pSetFileSecurityA;
127 fnRtlAdjustPrivilege pRtlAdjustPrivilege;
128 fnCreateWellKnownSid pCreateWellKnownSid;
129 fnDuplicateTokenEx pDuplicateTokenEx;
130 fnLsaQueryInformationPolicy pLsaQueryInformationPolicy;
131 fnLsaClose pLsaClose;
132 fnLsaFreeMemory pLsaFreeMemory;
133 fnLsaOpenPolicy pLsaOpenPolicy;
134
135 struct sidRef
136 {
137     SID_IDENTIFIER_AUTHORITY auth;
138     const char *refStr;
139 };
140
141 static void init(void)
142 {
143     HMODULE hntdll;
144
145     hntdll = GetModuleHandleA("ntdll.dll");
146     pNtQueryObject = (void *)GetProcAddress( hntdll, "NtQueryObject" );
147     pNtAccessCheck = (void *)GetProcAddress( hntdll, "NtAccessCheck" );
148
149     hmod = GetModuleHandle("advapi32.dll");
150     pAddAccessAllowedAceEx = (void *)GetProcAddress(hmod, "AddAccessAllowedAceEx");
151     pAddAccessDeniedAceEx = (void *)GetProcAddress(hmod, "AddAccessDeniedAceEx");
152     pAddAuditAccessAceEx = (void *)GetProcAddress(hmod, "AddAuditAccessAceEx");
153     pConvertStringSecurityDescriptorToSecurityDescriptorA =
154         (void *)GetProcAddress(hmod, "ConvertStringSecurityDescriptorToSecurityDescriptorA" );
155     pConvertStringSecurityDescriptorToSecurityDescriptorW =
156         (void *)GetProcAddress(hmod, "ConvertStringSecurityDescriptorToSecurityDescriptorW" );
157     pConvertSecurityDescriptorToStringSecurityDescriptorA =
158         (void *)GetProcAddress(hmod, "ConvertSecurityDescriptorToStringSecurityDescriptorA" );
159     pGetFileSecurityA = (fnGetFileSecurityA)GetProcAddress(hmod, "GetFileSecurityA" );
160     pSetFileSecurityA = (fnSetFileSecurityA)GetProcAddress(hmod, "SetFileSecurityA" );
161     pCreateWellKnownSid = (fnCreateWellKnownSid)GetProcAddress( hmod, "CreateWellKnownSid" );
162     pGetNamedSecurityInfoA = (void *)GetProcAddress(hmod, "GetNamedSecurityInfoA");
163     pMakeSelfRelativeSD = (void *)GetProcAddress(hmod, "MakeSelfRelativeSD");
164     pSetEntriesInAclW = (void *)GetProcAddress(hmod, "SetEntriesInAclW");
165     pSetSecurityDescriptorControl = (void *)GetProcAddress(hmod, "SetSecurityDescriptorControl");
166     pGetSecurityInfo = (void *)GetProcAddress(hmod, "GetSecurityInfo");
167
168     myARGC = winetest_get_mainargs( &myARGV );
169 }
170
171 static void test_str_sid(const char *str_sid)
172 {
173     PSID psid;
174     char *temp;
175
176     if (pConvertStringSidToSidA(str_sid, &psid))
177     {
178         if (pConvertSidToStringSidA(psid, &temp))
179         {
180             trace(" %s: %s\n", str_sid, temp);
181             LocalFree(temp);
182         }
183         LocalFree(psid);
184     }
185     else
186     {
187         if (GetLastError() != ERROR_INVALID_SID)
188             trace(" %s: couldn't be converted, returned %d\n", str_sid, GetLastError());
189         else
190             trace(" %s: couldn't be converted\n", str_sid);
191     }
192 }
193
194 static void test_sid(void)
195 {
196     struct sidRef refs[] = {
197      { { {0x00,0x00,0x33,0x44,0x55,0x66} }, "S-1-860116326-1" },
198      { { {0x00,0x00,0x01,0x02,0x03,0x04} }, "S-1-16909060-1"  },
199      { { {0x00,0x00,0x00,0x01,0x02,0x03} }, "S-1-66051-1"     },
200      { { {0x00,0x00,0x00,0x00,0x01,0x02} }, "S-1-258-1"       },
201      { { {0x00,0x00,0x00,0x00,0x00,0x02} }, "S-1-2-1"         },
202      { { {0x00,0x00,0x00,0x00,0x00,0x0c} }, "S-1-12-1"        },
203     };
204     const char noSubAuthStr[] = "S-1-5";
205     unsigned int i;
206     PSID psid = NULL;
207     SID *pisid;
208     BOOL r;
209     LPSTR str = NULL;
210
211     pConvertSidToStringSidA = (fnConvertSidToStringSidA)
212                     GetProcAddress( hmod, "ConvertSidToStringSidA" );
213     if( !pConvertSidToStringSidA )
214         return;
215     pConvertStringSidToSidA = (fnConvertStringSidToSidA)
216                     GetProcAddress( hmod, "ConvertStringSidToSidA" );
217     if( !pConvertStringSidToSidA )
218         return;
219
220     r = pConvertStringSidToSidA( NULL, NULL );
221     ok( !r, "expected failure with NULL parameters\n" );
222     if( GetLastError() == ERROR_CALL_NOT_IMPLEMENTED )
223         return;
224     ok( GetLastError() == ERROR_INVALID_PARAMETER,
225      "expected GetLastError() is ERROR_INVALID_PARAMETER, got %d\n",
226      GetLastError() );
227
228     r = pConvertStringSidToSidA( refs[0].refStr, NULL );
229     ok( !r && GetLastError() == ERROR_INVALID_PARAMETER,
230      "expected GetLastError() is ERROR_INVALID_PARAMETER, got %d\n",
231      GetLastError() );
232
233     r = pConvertStringSidToSidA( NULL, &str );
234     ok( !r && GetLastError() == ERROR_INVALID_PARAMETER,
235      "expected GetLastError() is ERROR_INVALID_PARAMETER, got %d\n",
236      GetLastError() );
237
238     r = pConvertStringSidToSidA( noSubAuthStr, &psid );
239     ok( !r,
240      "expected failure with no sub authorities\n" );
241     ok( GetLastError() == ERROR_INVALID_SID,
242      "expected GetLastError() is ERROR_INVALID_SID, got %d\n",
243      GetLastError() );
244
245     ok(pConvertStringSidToSidA("S-1-5-21-93476-23408-4576", &psid), "ConvertStringSidToSidA failed\n");
246     pisid = psid;
247     ok(pisid->SubAuthorityCount == 4, "Invalid sub authority count - expected 4, got %d\n", pisid->SubAuthorityCount);
248     ok(pisid->SubAuthority[0] == 21, "Invalid subauthority 0 - expceted 21, got %d\n", pisid->SubAuthority[0]);
249     ok(pisid->SubAuthority[3] == 4576, "Invalid subauthority 0 - expceted 4576, got %d\n", pisid->SubAuthority[3]);
250     LocalFree(str);
251
252     for( i = 0; i < sizeof(refs) / sizeof(refs[0]); i++ )
253     {
254         PISID pisid;
255
256         r = AllocateAndInitializeSid( &refs[i].auth, 1,1,0,0,0,0,0,0,0,
257          &psid );
258         ok( r, "failed to allocate sid\n" );
259         r = pConvertSidToStringSidA( psid, &str );
260         ok( r, "failed to convert sid\n" );
261         if (r)
262         {
263             ok( !strcmp( str, refs[i].refStr ),
264                 "incorrect sid, expected %s, got %s\n", refs[i].refStr, str );
265             LocalFree( str );
266         }
267         if( psid )
268             FreeSid( psid );
269
270         r = pConvertStringSidToSidA( refs[i].refStr, &psid );
271         ok( r, "failed to parse sid string\n" );
272         pisid = psid;
273         ok( pisid &&
274          !memcmp( pisid->IdentifierAuthority.Value, refs[i].auth.Value,
275          sizeof(refs[i].auth) ),
276          "string sid %s didn't parse to expected value\n"
277          "(got 0x%04x%08x, expected 0x%04x%08x)\n",
278          refs[i].refStr,
279          MAKEWORD( pisid->IdentifierAuthority.Value[1],
280          pisid->IdentifierAuthority.Value[0] ),
281          MAKELONG( MAKEWORD( pisid->IdentifierAuthority.Value[5],
282          pisid->IdentifierAuthority.Value[4] ),
283          MAKEWORD( pisid->IdentifierAuthority.Value[3],
284          pisid->IdentifierAuthority.Value[2] ) ),
285          MAKEWORD( refs[i].auth.Value[1], refs[i].auth.Value[0] ),
286          MAKELONG( MAKEWORD( refs[i].auth.Value[5], refs[i].auth.Value[4] ),
287          MAKEWORD( refs[i].auth.Value[3], refs[i].auth.Value[2] ) ) );
288         if( psid )
289             LocalFree( psid );
290     }
291
292     trace("String SIDs:\n");
293     test_str_sid("AO");
294     test_str_sid("RU");
295     test_str_sid("AN");
296     test_str_sid("AU");
297     test_str_sid("BA");
298     test_str_sid("BG");
299     test_str_sid("BO");
300     test_str_sid("BU");
301     test_str_sid("CA");
302     test_str_sid("CG");
303     test_str_sid("CO");
304     test_str_sid("DA");
305     test_str_sid("DC");
306     test_str_sid("DD");
307     test_str_sid("DG");
308     test_str_sid("DU");
309     test_str_sid("EA");
310     test_str_sid("ED");
311     test_str_sid("WD");
312     test_str_sid("PA");
313     test_str_sid("IU");
314     test_str_sid("LA");
315     test_str_sid("LG");
316     test_str_sid("LS");
317     test_str_sid("SY");
318     test_str_sid("NU");
319     test_str_sid("NO");
320     test_str_sid("NS");
321     test_str_sid("PO");
322     test_str_sid("PS");
323     test_str_sid("PU");
324     test_str_sid("RS");
325     test_str_sid("RD");
326     test_str_sid("RE");
327     test_str_sid("RC");
328     test_str_sid("SA");
329     test_str_sid("SO");
330     test_str_sid("SU");
331 }
332
333 static void test_trustee(void)
334 {
335     GUID ObjectType = {0x12345678, 0x1234, 0x5678, {0x11, 0x22, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88}};
336     GUID InheritedObjectType = {0x23456789, 0x2345, 0x6786, {0x2, 0x33, 0x44, 0x55, 0x66, 0x77, 0x88, 0x99}};
337     GUID ZeroGuid;
338     OBJECTS_AND_NAME_ oan;
339     OBJECTS_AND_SID oas;
340     TRUSTEE trustee;
341     PSID psid;
342     char szObjectTypeName[] = "ObjectTypeName";
343     char szInheritedObjectTypeName[] = "InheritedObjectTypeName";
344     char szTrusteeName[] = "szTrusteeName";
345     SID_IDENTIFIER_AUTHORITY auth = { {0x11,0x22,0,0,0, 0} };
346
347     memset( &ZeroGuid, 0x00, sizeof (ZeroGuid) );
348
349     pBuildTrusteeWithSidA = (fnBuildTrusteeWithSidA)
350                     GetProcAddress( hmod, "BuildTrusteeWithSidA" );
351     pBuildTrusteeWithNameA = (fnBuildTrusteeWithNameA)
352                     GetProcAddress( hmod, "BuildTrusteeWithNameA" );
353     pBuildTrusteeWithObjectsAndNameA = (fnBuildTrusteeWithObjectsAndNameA)
354                     GetProcAddress (hmod, "BuildTrusteeWithObjectsAndNameA" );
355     pBuildTrusteeWithObjectsAndSidA = (fnBuildTrusteeWithObjectsAndSidA)
356                     GetProcAddress (hmod, "BuildTrusteeWithObjectsAndSidA" );
357     pGetTrusteeNameA = (fnGetTrusteeNameA)
358                     GetProcAddress (hmod, "GetTrusteeNameA" );
359     if( !pBuildTrusteeWithSidA || !pBuildTrusteeWithNameA ||
360         !pBuildTrusteeWithObjectsAndNameA || !pBuildTrusteeWithObjectsAndSidA ||
361         !pGetTrusteeNameA )
362         return;
363
364     if ( ! AllocateAndInitializeSid( &auth, 1, 42, 0,0,0,0,0,0,0,&psid ) )
365     {
366         trace( "failed to init SID\n" );
367        return;
368     }
369
370     /* test BuildTrusteeWithSidA */
371     memset( &trustee, 0xff, sizeof trustee );
372     pBuildTrusteeWithSidA( &trustee, psid );
373
374     ok( trustee.pMultipleTrustee == NULL, "pMultipleTrustee wrong\n");
375     ok( trustee.MultipleTrusteeOperation == NO_MULTIPLE_TRUSTEE, 
376         "MultipleTrusteeOperation wrong\n");
377     ok( trustee.TrusteeForm == TRUSTEE_IS_SID, "TrusteeForm wrong\n");
378     ok( trustee.TrusteeType == TRUSTEE_IS_UNKNOWN, "TrusteeType wrong\n");
379     ok( trustee.ptstrName == psid, "ptstrName wrong\n" );
380
381     /* test BuildTrusteeWithObjectsAndSidA (test 1) */
382     memset( &trustee, 0xff, sizeof trustee );
383     memset( &oas, 0xff, sizeof(oas) );
384     pBuildTrusteeWithObjectsAndSidA(&trustee, &oas, &ObjectType,
385                                     &InheritedObjectType, psid);
386
387     ok(trustee.pMultipleTrustee == NULL, "pMultipleTrustee wrong\n");
388     ok(trustee.MultipleTrusteeOperation == NO_MULTIPLE_TRUSTEE, "MultipleTrusteeOperation wrong\n");
389     ok(trustee.TrusteeForm == TRUSTEE_IS_OBJECTS_AND_SID, "TrusteeForm wrong\n");
390     ok(trustee.TrusteeType == TRUSTEE_IS_UNKNOWN, "TrusteeType wrong\n");
391     ok(trustee.ptstrName == (LPSTR)&oas, "ptstrName wrong\n");
392  
393     ok(oas.ObjectsPresent == (ACE_OBJECT_TYPE_PRESENT | ACE_INHERITED_OBJECT_TYPE_PRESENT), "ObjectsPresent wrong\n");
394     ok(!memcmp(&oas.ObjectTypeGuid, &ObjectType, sizeof(GUID)), "ObjectTypeGuid wrong\n");
395     ok(!memcmp(&oas.InheritedObjectTypeGuid, &InheritedObjectType, sizeof(GUID)), "InheritedObjectTypeGuid wrong\n");
396     ok(oas.pSid == psid, "pSid wrong\n");
397
398     /* test GetTrusteeNameA */
399     ok(pGetTrusteeNameA(&trustee) == (LPSTR)&oas, "GetTrusteeName returned wrong value\n");
400
401     /* test BuildTrusteeWithObjectsAndSidA (test 2) */
402     memset( &trustee, 0xff, sizeof trustee );
403     memset( &oas, 0xff, sizeof(oas) );
404     pBuildTrusteeWithObjectsAndSidA(&trustee, &oas, NULL,
405                                     &InheritedObjectType, psid);
406
407     ok(trustee.pMultipleTrustee == NULL, "pMultipleTrustee wrong\n");
408     ok(trustee.MultipleTrusteeOperation == NO_MULTIPLE_TRUSTEE, "MultipleTrusteeOperation wrong\n");
409     ok(trustee.TrusteeForm == TRUSTEE_IS_OBJECTS_AND_SID, "TrusteeForm wrong\n");
410     ok(trustee.TrusteeType == TRUSTEE_IS_UNKNOWN, "TrusteeType wrong\n");
411     ok(trustee.ptstrName == (LPSTR)&oas, "ptstrName wrong\n");
412  
413     ok(oas.ObjectsPresent == ACE_INHERITED_OBJECT_TYPE_PRESENT, "ObjectsPresent wrong\n");
414     ok(!memcmp(&oas.ObjectTypeGuid, &ZeroGuid, sizeof(GUID)), "ObjectTypeGuid wrong\n");
415     ok(!memcmp(&oas.InheritedObjectTypeGuid, &InheritedObjectType, sizeof(GUID)), "InheritedObjectTypeGuid wrong\n");
416     ok(oas.pSid == psid, "pSid wrong\n");
417
418     FreeSid( psid );
419
420     /* test BuildTrusteeWithNameA */
421     memset( &trustee, 0xff, sizeof trustee );
422     pBuildTrusteeWithNameA( &trustee, szTrusteeName );
423
424     ok( trustee.pMultipleTrustee == NULL, "pMultipleTrustee wrong\n");
425     ok( trustee.MultipleTrusteeOperation == NO_MULTIPLE_TRUSTEE, 
426         "MultipleTrusteeOperation wrong\n");
427     ok( trustee.TrusteeForm == TRUSTEE_IS_NAME, "TrusteeForm wrong\n");
428     ok( trustee.TrusteeType == TRUSTEE_IS_UNKNOWN, "TrusteeType wrong\n");
429     ok( trustee.ptstrName == szTrusteeName, "ptstrName wrong\n" );
430
431     /* test BuildTrusteeWithObjectsAndNameA (test 1) */
432     memset( &trustee, 0xff, sizeof trustee );
433     memset( &oan, 0xff, sizeof(oan) );
434     pBuildTrusteeWithObjectsAndNameA(&trustee, &oan, SE_KERNEL_OBJECT, szObjectTypeName,
435                                      szInheritedObjectTypeName, szTrusteeName);
436
437     ok(trustee.pMultipleTrustee == NULL, "pMultipleTrustee wrong\n");
438     ok(trustee.MultipleTrusteeOperation == NO_MULTIPLE_TRUSTEE, "MultipleTrusteeOperation wrong\n");
439     ok(trustee.TrusteeForm == TRUSTEE_IS_OBJECTS_AND_NAME, "TrusteeForm wrong\n");
440     ok(trustee.TrusteeType == TRUSTEE_IS_UNKNOWN, "TrusteeType wrong\n");
441     ok(trustee.ptstrName == (LPTSTR)&oan, "ptstrName wrong\n");
442  
443     ok(oan.ObjectsPresent == (ACE_OBJECT_TYPE_PRESENT | ACE_INHERITED_OBJECT_TYPE_PRESENT), "ObjectsPresent wrong\n");
444     ok(oan.ObjectType == SE_KERNEL_OBJECT, "ObjectType wrong\n");
445     ok(oan.InheritedObjectTypeName == szInheritedObjectTypeName, "InheritedObjectTypeName wrong\n");
446     ok(oan.ptstrName == szTrusteeName, "szTrusteeName wrong\n");
447
448     /* test GetTrusteeNameA */
449     ok(pGetTrusteeNameA(&trustee) == (LPSTR)&oan, "GetTrusteeName returned wrong value\n");
450
451     /* test BuildTrusteeWithObjectsAndNameA (test 2) */
452     memset( &trustee, 0xff, sizeof trustee );
453     memset( &oan, 0xff, sizeof(oan) );
454     pBuildTrusteeWithObjectsAndNameA(&trustee, &oan, SE_KERNEL_OBJECT, NULL,
455                                      szInheritedObjectTypeName, szTrusteeName);
456
457     ok(trustee.pMultipleTrustee == NULL, "pMultipleTrustee wrong\n");
458     ok(trustee.MultipleTrusteeOperation == NO_MULTIPLE_TRUSTEE, "MultipleTrusteeOperation wrong\n");
459     ok(trustee.TrusteeForm == TRUSTEE_IS_OBJECTS_AND_NAME, "TrusteeForm wrong\n");
460     ok(trustee.TrusteeType == TRUSTEE_IS_UNKNOWN, "TrusteeType wrong\n");
461     ok(trustee.ptstrName == (LPSTR)&oan, "ptstrName wrong\n");
462  
463     ok(oan.ObjectsPresent == ACE_INHERITED_OBJECT_TYPE_PRESENT, "ObjectsPresent wrong\n");
464     ok(oan.ObjectType == SE_KERNEL_OBJECT, "ObjectType wrong\n");
465     ok(oan.InheritedObjectTypeName == szInheritedObjectTypeName, "InheritedObjectTypeName wrong\n");
466     ok(oan.ptstrName == szTrusteeName, "szTrusteeName wrong\n");
467
468     /* test BuildTrusteeWithObjectsAndNameA (test 3) */
469     memset( &trustee, 0xff, sizeof trustee );
470     memset( &oan, 0xff, sizeof(oan) );
471     pBuildTrusteeWithObjectsAndNameA(&trustee, &oan, SE_KERNEL_OBJECT, szObjectTypeName,
472                                      NULL, szTrusteeName);
473
474     ok(trustee.pMultipleTrustee == NULL, "pMultipleTrustee wrong\n");
475     ok(trustee.MultipleTrusteeOperation == NO_MULTIPLE_TRUSTEE, "MultipleTrusteeOperation wrong\n");
476     ok(trustee.TrusteeForm == TRUSTEE_IS_OBJECTS_AND_NAME, "TrusteeForm wrong\n");
477     ok(trustee.TrusteeType == TRUSTEE_IS_UNKNOWN, "TrusteeType wrong\n");
478     ok(trustee.ptstrName == (LPTSTR)&oan, "ptstrName wrong\n");
479  
480     ok(oan.ObjectsPresent == ACE_OBJECT_TYPE_PRESENT, "ObjectsPresent wrong\n");
481     ok(oan.ObjectType == SE_KERNEL_OBJECT, "ObjectType wrong\n");
482     ok(oan.InheritedObjectTypeName == NULL, "InheritedObjectTypeName wrong\n");
483     ok(oan.ptstrName == szTrusteeName, "szTrusteeName wrong\n");
484 }
485  
486 /* If the first isn't defined, assume none is */
487 #ifndef SE_MIN_WELL_KNOWN_PRIVILEGE
488 #define SE_MIN_WELL_KNOWN_PRIVILEGE       2L
489 #define SE_CREATE_TOKEN_PRIVILEGE         2L
490 #define SE_ASSIGNPRIMARYTOKEN_PRIVILEGE   3L
491 #define SE_LOCK_MEMORY_PRIVILEGE          4L
492 #define SE_INCREASE_QUOTA_PRIVILEGE       5L
493 #define SE_MACHINE_ACCOUNT_PRIVILEGE      6L
494 #define SE_TCB_PRIVILEGE                  7L
495 #define SE_SECURITY_PRIVILEGE             8L
496 #define SE_TAKE_OWNERSHIP_PRIVILEGE       9L
497 #define SE_LOAD_DRIVER_PRIVILEGE         10L
498 #define SE_SYSTEM_PROFILE_PRIVILEGE      11L
499 #define SE_SYSTEMTIME_PRIVILEGE          12L
500 #define SE_PROF_SINGLE_PROCESS_PRIVILEGE 13L
501 #define SE_INC_BASE_PRIORITY_PRIVILEGE   14L
502 #define SE_CREATE_PAGEFILE_PRIVILEGE     15L
503 #define SE_CREATE_PERMANENT_PRIVILEGE    16L
504 #define SE_BACKUP_PRIVILEGE              17L
505 #define SE_RESTORE_PRIVILEGE             18L
506 #define SE_SHUTDOWN_PRIVILEGE            19L
507 #define SE_DEBUG_PRIVILEGE               20L
508 #define SE_AUDIT_PRIVILEGE               21L
509 #define SE_SYSTEM_ENVIRONMENT_PRIVILEGE  22L
510 #define SE_CHANGE_NOTIFY_PRIVILEGE       23L
511 #define SE_REMOTE_SHUTDOWN_PRIVILEGE     24L
512 #define SE_UNDOCK_PRIVILEGE              25L
513 #define SE_SYNC_AGENT_PRIVILEGE          26L
514 #define SE_ENABLE_DELEGATION_PRIVILEGE   27L
515 #define SE_MANAGE_VOLUME_PRIVILEGE       28L
516 #define SE_IMPERSONATE_PRIVILEGE         29L
517 #define SE_CREATE_GLOBAL_PRIVILEGE       30L
518 #define SE_MAX_WELL_KNOWN_PRIVILEGE      SE_CREATE_GLOBAL_PRIVILEGE
519 #endif /* ndef SE_MIN_WELL_KNOWN_PRIVILEGE */
520
521 static void test_allocateLuid(void)
522 {
523     BOOL (WINAPI *pAllocateLocallyUniqueId)(PLUID);
524     LUID luid1, luid2;
525     BOOL ret;
526
527     pAllocateLocallyUniqueId = (void*)GetProcAddress(hmod, "AllocateLocallyUniqueId");
528     if (!pAllocateLocallyUniqueId) return;
529
530     ret = pAllocateLocallyUniqueId(&luid1);
531     if (!ret && GetLastError() == ERROR_CALL_NOT_IMPLEMENTED)
532         return;
533
534     ok(ret,
535      "AllocateLocallyUniqueId failed: %d\n", GetLastError());
536     ret = pAllocateLocallyUniqueId(&luid2);
537     ok( ret,
538      "AllocateLocallyUniqueId failed: %d\n", GetLastError());
539     ok(luid1.LowPart > SE_MAX_WELL_KNOWN_PRIVILEGE || luid1.HighPart != 0,
540      "AllocateLocallyUniqueId returned a well-known LUID\n");
541     ok(luid1.LowPart != luid2.LowPart || luid1.HighPart != luid2.HighPart,
542      "AllocateLocallyUniqueId returned non-unique LUIDs\n");
543     ret = pAllocateLocallyUniqueId(NULL);
544     ok( !ret && GetLastError() == ERROR_NOACCESS,
545      "AllocateLocallyUniqueId(NULL) didn't return ERROR_NOACCESS: %d\n",
546      GetLastError());
547 }
548
549 static void test_lookupPrivilegeName(void)
550 {
551     BOOL (WINAPI *pLookupPrivilegeNameA)(LPCSTR, PLUID, LPSTR, LPDWORD);
552     char buf[MAX_PATH]; /* arbitrary, seems long enough */
553     DWORD cchName = sizeof(buf);
554     LUID luid = { 0, 0 };
555     LONG i;
556     BOOL ret;
557
558     /* check whether it's available first */
559     pLookupPrivilegeNameA = (void*)GetProcAddress(hmod, "LookupPrivilegeNameA");
560     if (!pLookupPrivilegeNameA) return;
561     luid.LowPart = SE_CREATE_TOKEN_PRIVILEGE;
562     ret = pLookupPrivilegeNameA(NULL, &luid, buf, &cchName);
563     if (!ret && GetLastError() == ERROR_CALL_NOT_IMPLEMENTED)
564         return;
565
566     /* check with a short buffer */
567     cchName = 0;
568     luid.LowPart = SE_CREATE_TOKEN_PRIVILEGE;
569     ret = pLookupPrivilegeNameA(NULL, &luid, NULL, &cchName);
570     ok( !ret && GetLastError() == ERROR_INSUFFICIENT_BUFFER,
571      "LookupPrivilegeNameA didn't fail with ERROR_INSUFFICIENT_BUFFER: %d\n",
572      GetLastError());
573     ok(cchName == strlen("SeCreateTokenPrivilege") + 1,
574      "LookupPrivilegeNameA returned an incorrect required length for\n"
575      "SeCreateTokenPrivilege (got %d, expected %d)\n", cchName,
576      lstrlenA("SeCreateTokenPrivilege") + 1);
577     /* check a known value and its returned length on success */
578     cchName = sizeof(buf);
579     ok(pLookupPrivilegeNameA(NULL, &luid, buf, &cchName) &&
580      cchName == strlen("SeCreateTokenPrivilege"),
581      "LookupPrivilegeNameA returned an incorrect output length for\n"
582      "SeCreateTokenPrivilege (got %d, expected %d)\n", cchName,
583      (int)strlen("SeCreateTokenPrivilege"));
584     /* check known values */
585     for (i = SE_MIN_WELL_KNOWN_PRIVILEGE; i <= SE_MAX_WELL_KNOWN_PRIVILEGE; i++)
586     {
587         luid.LowPart = i;
588         cchName = sizeof(buf);
589         ret = pLookupPrivilegeNameA(NULL, &luid, buf, &cchName);
590         ok( ret || GetLastError() == ERROR_NO_SUCH_PRIVILEGE,
591          "LookupPrivilegeNameA(0.%d) failed: %d\n", i, GetLastError());
592     }
593     /* check a bogus LUID */
594     luid.LowPart = 0xdeadbeef;
595     cchName = sizeof(buf);
596     ret = pLookupPrivilegeNameA(NULL, &luid, buf, &cchName);
597     ok( !ret && GetLastError() == ERROR_NO_SUCH_PRIVILEGE,
598      "LookupPrivilegeNameA didn't fail with ERROR_NO_SUCH_PRIVILEGE: %d\n",
599      GetLastError());
600     /* check on a bogus system */
601     luid.LowPart = SE_CREATE_TOKEN_PRIVILEGE;
602     cchName = sizeof(buf);
603     ret = pLookupPrivilegeNameA("b0gu5.Nam3", &luid, buf, &cchName);
604     ok( !ret && (GetLastError() == RPC_S_SERVER_UNAVAILABLE ||
605                  GetLastError() == RPC_S_INVALID_NET_ADDR) /* w2k8 */,
606      "LookupPrivilegeNameA didn't fail with RPC_S_SERVER_UNAVAILABLE or RPC_S_INVALID_NET_ADDR: %d\n",
607      GetLastError());
608 }
609
610 struct NameToLUID
611 {
612     const char *name;
613     DWORD lowPart;
614 };
615
616 static void test_lookupPrivilegeValue(void)
617 {
618     static const struct NameToLUID privs[] = {
619      { "SeCreateTokenPrivilege", SE_CREATE_TOKEN_PRIVILEGE },
620      { "SeAssignPrimaryTokenPrivilege", SE_ASSIGNPRIMARYTOKEN_PRIVILEGE },
621      { "SeLockMemoryPrivilege", SE_LOCK_MEMORY_PRIVILEGE },
622      { "SeIncreaseQuotaPrivilege", SE_INCREASE_QUOTA_PRIVILEGE },
623      { "SeMachineAccountPrivilege", SE_MACHINE_ACCOUNT_PRIVILEGE },
624      { "SeTcbPrivilege", SE_TCB_PRIVILEGE },
625      { "SeSecurityPrivilege", SE_SECURITY_PRIVILEGE },
626      { "SeTakeOwnershipPrivilege", SE_TAKE_OWNERSHIP_PRIVILEGE },
627      { "SeLoadDriverPrivilege", SE_LOAD_DRIVER_PRIVILEGE },
628      { "SeSystemProfilePrivilege", SE_SYSTEM_PROFILE_PRIVILEGE },
629      { "SeSystemtimePrivilege", SE_SYSTEMTIME_PRIVILEGE },
630      { "SeProfileSingleProcessPrivilege", SE_PROF_SINGLE_PROCESS_PRIVILEGE },
631      { "SeIncreaseBasePriorityPrivilege", SE_INC_BASE_PRIORITY_PRIVILEGE },
632      { "SeCreatePagefilePrivilege", SE_CREATE_PAGEFILE_PRIVILEGE },
633      { "SeCreatePermanentPrivilege", SE_CREATE_PERMANENT_PRIVILEGE },
634      { "SeBackupPrivilege", SE_BACKUP_PRIVILEGE },
635      { "SeRestorePrivilege", SE_RESTORE_PRIVILEGE },
636      { "SeShutdownPrivilege", SE_SHUTDOWN_PRIVILEGE },
637      { "SeDebugPrivilege", SE_DEBUG_PRIVILEGE },
638      { "SeAuditPrivilege", SE_AUDIT_PRIVILEGE },
639      { "SeSystemEnvironmentPrivilege", SE_SYSTEM_ENVIRONMENT_PRIVILEGE },
640      { "SeChangeNotifyPrivilege", SE_CHANGE_NOTIFY_PRIVILEGE },
641      { "SeRemoteShutdownPrivilege", SE_REMOTE_SHUTDOWN_PRIVILEGE },
642      { "SeUndockPrivilege", SE_UNDOCK_PRIVILEGE },
643      { "SeSyncAgentPrivilege", SE_SYNC_AGENT_PRIVILEGE },
644      { "SeEnableDelegationPrivilege", SE_ENABLE_DELEGATION_PRIVILEGE },
645      { "SeManageVolumePrivilege", SE_MANAGE_VOLUME_PRIVILEGE },
646      { "SeImpersonatePrivilege", SE_IMPERSONATE_PRIVILEGE },
647      { "SeCreateGlobalPrivilege", SE_CREATE_GLOBAL_PRIVILEGE },
648     };
649     BOOL (WINAPI *pLookupPrivilegeValueA)(LPCSTR, LPCSTR, PLUID);
650     int i;
651     LUID luid;
652     BOOL ret;
653
654     /* check whether it's available first */
655     pLookupPrivilegeValueA = (void*)GetProcAddress(hmod, "LookupPrivilegeValueA");
656     if (!pLookupPrivilegeValueA) return;
657     ret = pLookupPrivilegeValueA(NULL, "SeCreateTokenPrivilege", &luid);
658     if (!ret && GetLastError() == ERROR_CALL_NOT_IMPLEMENTED)
659         return;
660
661     /* check a bogus system name */
662     ret = pLookupPrivilegeValueA("b0gu5.Nam3", "SeCreateTokenPrivilege", &luid);
663     ok( !ret && (GetLastError() == RPC_S_SERVER_UNAVAILABLE ||
664                 GetLastError() == RPC_S_INVALID_NET_ADDR) /* w2k8 */,
665      "LookupPrivilegeValueA didn't fail with RPC_S_SERVER_UNAVAILABLE or RPC_S_INVALID_NET_ADDR: %d\n",
666      GetLastError());
667     /* check a NULL string */
668     ret = pLookupPrivilegeValueA(NULL, 0, &luid);
669     ok( !ret && GetLastError() == ERROR_NO_SUCH_PRIVILEGE,
670      "LookupPrivilegeValueA didn't fail with ERROR_NO_SUCH_PRIVILEGE: %d\n",
671      GetLastError());
672     /* check a bogus privilege name */
673     ret = pLookupPrivilegeValueA(NULL, "SeBogusPrivilege", &luid);
674     ok( !ret && GetLastError() == ERROR_NO_SUCH_PRIVILEGE,
675      "LookupPrivilegeValueA didn't fail with ERROR_NO_SUCH_PRIVILEGE: %d\n",
676      GetLastError());
677     /* check case insensitive */
678     ret = pLookupPrivilegeValueA(NULL, "sEcREATEtOKENpRIVILEGE", &luid);
679     ok( ret,
680      "LookupPrivilegeValueA(NULL, sEcREATEtOKENpRIVILEGE, &luid) failed: %d\n",
681      GetLastError());
682     for (i = 0; i < sizeof(privs) / sizeof(privs[0]); i++)
683     {
684         /* Not all privileges are implemented on all Windows versions, so
685          * don't worry if the call fails
686          */
687         if (pLookupPrivilegeValueA(NULL, privs[i].name, &luid))
688         {
689             ok(luid.LowPart == privs[i].lowPart,
690              "LookupPrivilegeValueA returned an invalid LUID for %s\n",
691              privs[i].name);
692         }
693     }
694 }
695
696 static void test_luid(void)
697 {
698     test_allocateLuid();
699     test_lookupPrivilegeName();
700     test_lookupPrivilegeValue();
701 }
702
703 static void test_FileSecurity(void)
704 {
705     char wintmpdir [MAX_PATH];
706     char path [MAX_PATH];
707     char file [MAX_PATH];
708     BOOL rc;
709     HANDLE fh;
710     DWORD sdSize;
711     DWORD retSize;
712     BYTE *sd;
713     SECURITY_INFORMATION const request = OWNER_SECURITY_INFORMATION
714                                        | GROUP_SECURITY_INFORMATION
715                                        | DACL_SECURITY_INFORMATION;
716
717     if (!pGetFileSecurityA) {
718         win_skip ("GetFileSecurity is not available\n");
719         return;
720     }
721
722     if (!pSetFileSecurityA) {
723         win_skip ("SetFileSecurity is not available\n");
724         return;
725     }
726
727     if (!GetTempPathA (sizeof (wintmpdir), wintmpdir)) {
728         win_skip ("GetTempPathA failed\n");
729         return;
730     }
731
732     /* Create a temporary directory and in it a temporary file */
733     strcat (strcpy (path, wintmpdir), "rary");
734     SetLastError(0xdeadbeef);
735     rc = CreateDirectoryA (path, NULL);
736     ok (rc || GetLastError() == ERROR_ALREADY_EXISTS, "CreateDirectoryA "
737         "failed for '%s' with %d\n", path, GetLastError());
738
739     strcat (strcpy (file, path), "\\ess");
740     SetLastError(0xdeadbeef);
741     fh = CreateFileA (file, GENERIC_WRITE, 0, NULL, CREATE_ALWAYS, 0, NULL);
742     ok (fh != INVALID_HANDLE_VALUE, "CreateFileA "
743         "failed for '%s' with %d\n", file, GetLastError());
744     CloseHandle (fh);
745
746     /* For the temporary file ... */
747
748     /* Get size needed */
749     retSize = 0;
750     SetLastError(0xdeadbeef);
751     rc = pGetFileSecurityA (file, request, NULL, 0, &retSize);
752     if (!rc && (GetLastError() == ERROR_CALL_NOT_IMPLEMENTED)) {
753         win_skip("GetFileSecurityA is not implemented\n");
754         goto cleanup;
755     }
756     ok (!rc, "GetFileSecurityA "
757         "was expected to fail for '%s'\n", file);
758     ok (GetLastError() == ERROR_INSUFFICIENT_BUFFER, "GetFileSecurityA "
759         "returned %d; expected ERROR_INSUFFICIENT_BUFFER\n", GetLastError());
760     ok (retSize > sizeof (SECURITY_DESCRIPTOR), "GetFileSecurityA returned size %d\n", retSize);
761
762     sdSize = retSize;
763     sd = HeapAlloc (GetProcessHeap (), 0, sdSize);
764
765     /* Get security descriptor for real */
766     retSize = -1;
767     SetLastError(0xdeadbeef);
768     rc = pGetFileSecurityA (file, request, sd, sdSize, &retSize);
769     ok (rc, "GetFileSecurityA "
770         "was not expected to fail '%s': %d\n", file, GetLastError());
771     ok (retSize == sdSize ||
772         broken(retSize == 0), /* NT4 */
773         "GetFileSecurityA returned size %d; expected %d\n", retSize, sdSize);
774
775     /* Use it to set security descriptor */
776     SetLastError(0xdeadbeef);
777     rc = pSetFileSecurityA (file, request, sd);
778     ok (rc, "SetFileSecurityA "
779         "was not expected to fail '%s': %d\n", file, GetLastError());
780
781     HeapFree (GetProcessHeap (), 0, sd);
782
783     /* Repeat for the temporary directory ... */
784
785     /* Get size needed */
786     retSize = 0;
787     SetLastError(0xdeadbeef);
788     rc = pGetFileSecurityA (path, request, NULL, 0, &retSize);
789     ok (!rc, "GetFileSecurityA "
790         "was expected to fail for '%s'\n", path);
791     ok (GetLastError() == ERROR_INSUFFICIENT_BUFFER, "GetFileSecurityA "
792         "returned %d; expected ERROR_INSUFFICIENT_BUFFER\n", GetLastError());
793     ok (retSize > sizeof (SECURITY_DESCRIPTOR), "GetFileSecurityA returned size %d\n", retSize);
794
795     sdSize = retSize;
796     sd = HeapAlloc (GetProcessHeap (), 0, sdSize);
797
798     /* Get security descriptor for real */
799     retSize = -1;
800     SetLastError(0xdeadbeef);
801     rc = pGetFileSecurityA (path, request, sd, sdSize, &retSize);
802     ok (rc, "GetFileSecurityA "
803         "was not expected to fail '%s': %d\n", path, GetLastError());
804     ok (retSize == sdSize ||
805         broken(retSize == 0), /* NT4 */
806         "GetFileSecurityA returned size %d; expected %d\n", retSize, sdSize);
807
808     /* Use it to set security descriptor */
809     SetLastError(0xdeadbeef);
810     rc = pSetFileSecurityA (path, request, sd);
811     ok (rc, "SetFileSecurityA "
812         "was not expected to fail '%s': %d\n", path, GetLastError());
813     HeapFree (GetProcessHeap (), 0, sd);
814
815     /* Old test */
816     strcpy (wintmpdir, "\\Should not exist");
817     SetLastError(0xdeadbeef);
818     rc = pGetFileSecurityA (wintmpdir, OWNER_SECURITY_INFORMATION, NULL, 0, &sdSize);
819     ok (!rc, "GetFileSecurityA should fail for not existing directories/files\n");
820     ok (GetLastError() == ERROR_FILE_NOT_FOUND,
821         "last error ERROR_FILE_NOT_FOUND expected, got %d\n", GetLastError());
822
823 cleanup:
824     /* Remove temporary file and directory */
825     DeleteFileA(file);
826     RemoveDirectoryA(path);
827 }
828
829 static void test_AccessCheck(void)
830 {
831     PSID EveryoneSid = NULL, AdminSid = NULL, UsersSid = NULL;
832     PACL Acl = NULL;
833     SECURITY_DESCRIPTOR *SecurityDescriptor = NULL;
834     SID_IDENTIFIER_AUTHORITY SIDAuthWorld = { SECURITY_WORLD_SID_AUTHORITY };
835     SID_IDENTIFIER_AUTHORITY SIDAuthNT = { SECURITY_NT_AUTHORITY };
836     GENERIC_MAPPING Mapping = { KEY_READ, KEY_WRITE, KEY_EXECUTE, KEY_ALL_ACCESS };
837     ACCESS_MASK Access;
838     BOOL AccessStatus;
839     HANDLE Token;
840     HANDLE ProcessToken;
841     BOOL ret;
842     DWORD PrivSetLen;
843     PRIVILEGE_SET *PrivSet;
844     BOOL res;
845     HMODULE NtDllModule;
846     BOOLEAN Enabled;
847     DWORD err;
848     NTSTATUS ntret, ntAccessStatus;
849
850     NtDllModule = GetModuleHandle("ntdll.dll");
851     if (!NtDllModule)
852     {
853         skip("not running on NT, skipping test\n");
854         return;
855     }
856     pRtlAdjustPrivilege = (fnRtlAdjustPrivilege)
857                           GetProcAddress(NtDllModule, "RtlAdjustPrivilege");
858     if (!pRtlAdjustPrivilege)
859     {
860         win_skip("missing RtlAdjustPrivilege, skipping test\n");
861         return;
862     }
863
864     Acl = HeapAlloc(GetProcessHeap(), 0, 256);
865     res = InitializeAcl(Acl, 256, ACL_REVISION);
866     if(!res && GetLastError() == ERROR_CALL_NOT_IMPLEMENTED)
867     {
868         skip("ACLs not implemented - skipping tests\n");
869         HeapFree(GetProcessHeap(), 0, Acl);
870         return;
871     }
872     ok(res, "InitializeAcl failed with error %d\n", GetLastError());
873
874     res = AllocateAndInitializeSid( &SIDAuthWorld, 1, SECURITY_WORLD_RID, 0, 0, 0, 0, 0, 0, 0, &EveryoneSid);
875     ok(res, "AllocateAndInitializeSid failed with error %d\n", GetLastError());
876
877     res = AllocateAndInitializeSid( &SIDAuthNT, 2, SECURITY_BUILTIN_DOMAIN_RID,
878         DOMAIN_ALIAS_RID_ADMINS, 0, 0, 0, 0, 0, 0, &AdminSid);
879     ok(res, "AllocateAndInitializeSid failed with error %d\n", GetLastError());
880
881     res = AllocateAndInitializeSid( &SIDAuthNT, 2, SECURITY_BUILTIN_DOMAIN_RID,
882         DOMAIN_ALIAS_RID_USERS, 0, 0, 0, 0, 0, 0, &UsersSid);
883     ok(res, "AllocateAndInitializeSid failed with error %d\n", GetLastError());
884
885     SecurityDescriptor = HeapAlloc(GetProcessHeap(), 0, SECURITY_DESCRIPTOR_MIN_LENGTH);
886
887     res = InitializeSecurityDescriptor(SecurityDescriptor, SECURITY_DESCRIPTOR_REVISION);
888     ok(res, "InitializeSecurityDescriptor failed with error %d\n", GetLastError());
889
890     res = SetSecurityDescriptorDacl(SecurityDescriptor, TRUE, Acl, FALSE);
891     ok(res, "SetSecurityDescriptorDacl failed with error %d\n", GetLastError());
892
893     PrivSetLen = FIELD_OFFSET(PRIVILEGE_SET, Privilege[16]);
894     PrivSet = HeapAlloc(GetProcessHeap(), HEAP_ZERO_MEMORY, PrivSetLen);
895     PrivSet->PrivilegeCount = 16;
896
897     res = OpenProcessToken(GetCurrentProcess(), TOKEN_DUPLICATE|TOKEN_QUERY, &ProcessToken);
898     ok(res, "OpenProcessToken failed with error %d\n", GetLastError());
899
900     pRtlAdjustPrivilege(SE_SECURITY_PRIVILEGE, FALSE, TRUE, &Enabled);
901
902     res = DuplicateToken(ProcessToken, SecurityImpersonation, &Token);
903     ok(res, "DuplicateToken failed with error %d\n", GetLastError());
904
905     /* SD without owner/group */
906     SetLastError(0xdeadbeef);
907     Access = AccessStatus = 0xdeadbeef;
908     ret = AccessCheck(SecurityDescriptor, Token, KEY_QUERY_VALUE, &Mapping,
909                       PrivSet, &PrivSetLen, &Access, &AccessStatus);
910     err = GetLastError();
911     ok(!ret && err == ERROR_INVALID_SECURITY_DESCR, "AccessCheck should have "
912        "failed with ERROR_INVALID_SECURITY_DESCR, instead of %d\n", err);
913     ok(Access == 0xdeadbeef && AccessStatus == 0xdeadbeef,
914        "Access and/or AccessStatus were changed!\n");
915
916     /* Set owner and group */
917     res = SetSecurityDescriptorOwner(SecurityDescriptor, AdminSid, FALSE);
918     ok(res, "SetSecurityDescriptorOwner failed with error %d\n", GetLastError());
919     res = SetSecurityDescriptorGroup(SecurityDescriptor, UsersSid, TRUE);
920     ok(res, "SetSecurityDescriptorGroup failed with error %d\n", GetLastError());
921
922     /* Generic access mask */
923     SetLastError(0xdeadbeef);
924     Access = AccessStatus = 0xdeadbeef;
925     ret = AccessCheck(SecurityDescriptor, Token, GENERIC_READ, &Mapping,
926                       PrivSet, &PrivSetLen, &Access, &AccessStatus);
927     err = GetLastError();
928     ok(!ret && err == ERROR_GENERIC_NOT_MAPPED, "AccessCheck should have failed "
929        "with ERROR_GENERIC_NOT_MAPPED, instead of %d\n", err);
930     ok(Access == 0xdeadbeef && AccessStatus == 0xdeadbeef,
931        "Access and/or AccessStatus were changed!\n");
932
933     /* Generic access mask - no privilegeset buffer */
934     SetLastError(0xdeadbeef);
935     Access = AccessStatus = 0xdeadbeef;
936     ret = AccessCheck(SecurityDescriptor, Token, GENERIC_READ, &Mapping,
937                       NULL, &PrivSetLen, &Access, &AccessStatus);
938     err = GetLastError();
939     ok(!ret && err == ERROR_NOACCESS, "AccessCheck should have failed "
940        "with ERROR_NOACCESS, instead of %d\n", err);
941     ok(Access == 0xdeadbeef && AccessStatus == 0xdeadbeef,
942        "Access and/or AccessStatus were changed!\n");
943
944     /* Generic access mask - no returnlength */
945     SetLastError(0xdeadbeef);
946     Access = AccessStatus = 0xdeadbeef;
947     ret = AccessCheck(SecurityDescriptor, Token, GENERIC_READ, &Mapping,
948                       PrivSet, NULL, &Access, &AccessStatus);
949     err = GetLastError();
950     ok(!ret && err == ERROR_NOACCESS, "AccessCheck should have failed "
951        "with ERROR_NOACCESS, instead of %d\n", err);
952     ok(Access == 0xdeadbeef && AccessStatus == 0xdeadbeef,
953        "Access and/or AccessStatus were changed!\n");
954
955     /* Generic access mask - no privilegeset buffer, no returnlength */
956     SetLastError(0xdeadbeef);
957     Access = AccessStatus = 0xdeadbeef;
958     ret = AccessCheck(SecurityDescriptor, Token, GENERIC_READ, &Mapping,
959                       NULL, NULL, &Access, &AccessStatus);
960     err = GetLastError();
961     ok(!ret && err == ERROR_NOACCESS, "AccessCheck should have failed "
962        "with ERROR_NOACCESS, instead of %d\n", err);
963     ok(Access == 0xdeadbeef && AccessStatus == 0xdeadbeef,
964        "Access and/or AccessStatus were changed!\n");
965
966     /* sd with no dacl present */
967     Access = AccessStatus = 0xdeadbeef;
968     ret = SetSecurityDescriptorDacl(SecurityDescriptor, FALSE, NULL, FALSE);
969     ok(ret, "SetSecurityDescriptorDacl failed with error %d\n", GetLastError());
970     ret = AccessCheck(SecurityDescriptor, Token, KEY_READ, &Mapping,
971                       PrivSet, &PrivSetLen, &Access, &AccessStatus);
972     ok(ret, "AccessCheck failed with error %d\n", GetLastError());
973     ok(AccessStatus && (Access == KEY_READ),
974         "AccessCheck failed to grant access with error %d\n",
975         GetLastError());
976
977     /* sd with no dacl present - no privilegeset buffer */
978     SetLastError(0xdeadbeef);
979     Access = AccessStatus = 0xdeadbeef;
980     ret = AccessCheck(SecurityDescriptor, Token, GENERIC_READ, &Mapping,
981                       NULL, &PrivSetLen, &Access, &AccessStatus);
982     err = GetLastError();
983     ok(!ret && err == ERROR_NOACCESS, "AccessCheck should have failed "
984        "with ERROR_NOACCESS, instead of %d\n", err);
985     ok(Access == 0xdeadbeef && AccessStatus == 0xdeadbeef,
986        "Access and/or AccessStatus were changed!\n");
987
988     if(pNtAccessCheck)
989     {
990        /* Generic access mask - no privilegeset buffer */
991        SetLastError(0xdeadbeef);
992        Access = ntAccessStatus = 0xdeadbeef;
993        ntret = pNtAccessCheck(SecurityDescriptor, Token, GENERIC_READ, &Mapping,
994                               NULL, &PrivSetLen, &Access, &ntAccessStatus);
995        err = GetLastError();
996        ok(ntret == STATUS_ACCESS_VIOLATION,
997           "NtAccessCheck should have failed with STATUS_ACCESS_VIOLATION, got %x\n", ntret);
998        ok(err == 0xdeadbeef,
999           "NtAccessCheck shouldn't set last error, got %d\n", err);
1000        ok(Access == 0xdeadbeef && ntAccessStatus == 0xdeadbeef,
1001           "Access and/or AccessStatus were changed!\n");
1002
1003       /* Generic access mask - no returnlength */
1004       SetLastError(0xdeadbeef);
1005       Access = ntAccessStatus = 0xdeadbeef;
1006       ntret = pNtAccessCheck(SecurityDescriptor, Token, GENERIC_READ, &Mapping,
1007                              PrivSet, NULL, &Access, &ntAccessStatus);
1008       err = GetLastError();
1009       ok(ntret == STATUS_ACCESS_VIOLATION,
1010          "NtAccessCheck should have failed with STATUS_ACCESS_VIOLATION, got %x\n", ntret);
1011       ok(err == 0xdeadbeef,
1012          "NtAccessCheck shouldn't set last error, got %d\n", err);
1013       ok(Access == 0xdeadbeef && ntAccessStatus == 0xdeadbeef,
1014          "Access and/or AccessStatus were changed!\n");
1015
1016       /* Generic access mask - no privilegeset buffer, no returnlength */
1017       SetLastError(0xdeadbeef);
1018       Access = ntAccessStatus = 0xdeadbeef;
1019       ntret = pNtAccessCheck(SecurityDescriptor, Token, GENERIC_READ, &Mapping,
1020                              NULL, NULL, &Access, &ntAccessStatus);
1021       err = GetLastError();
1022       ok(ntret == STATUS_ACCESS_VIOLATION,
1023          "NtAccessCheck should have failed with STATUS_ACCESS_VIOLATION, got %x\n", ntret);
1024       ok(err == 0xdeadbeef,
1025          "NtAccessCheck shouldn't set last error, got %d\n", err);
1026       ok(Access == 0xdeadbeef && ntAccessStatus == 0xdeadbeef,
1027          "Access and/or AccessStatus were changed!\n");
1028     }
1029     else
1030        win_skip("NtAccessCheck unavailable. Skipping.\n");
1031
1032     /* sd with NULL dacl */
1033     Access = AccessStatus = 0xdeadbeef;
1034     ret = SetSecurityDescriptorDacl(SecurityDescriptor, TRUE, NULL, FALSE);
1035     ok(ret, "SetSecurityDescriptorDacl failed with error %d\n", GetLastError());
1036     ret = AccessCheck(SecurityDescriptor, Token, KEY_READ, &Mapping,
1037                       PrivSet, &PrivSetLen, &Access, &AccessStatus);
1038     ok(ret, "AccessCheck failed with error %d\n", GetLastError());
1039     ok(AccessStatus && (Access == KEY_READ),
1040         "AccessCheck failed to grant access with error %d\n",
1041         GetLastError());
1042
1043     /* sd with blank dacl */
1044     ret = SetSecurityDescriptorDacl(SecurityDescriptor, TRUE, Acl, FALSE);
1045     ok(ret, "SetSecurityDescriptorDacl failed with error %d\n", GetLastError());
1046     ret = AccessCheck(SecurityDescriptor, Token, KEY_READ, &Mapping,
1047                       PrivSet, &PrivSetLen, &Access, &AccessStatus);
1048     ok(ret, "AccessCheck failed with error %d\n", GetLastError());
1049     err = GetLastError();
1050     ok(!AccessStatus && err == ERROR_ACCESS_DENIED, "AccessCheck should have failed "
1051        "with ERROR_ACCESS_DENIED, instead of %d\n", err);
1052     ok(!Access, "Should have failed to grant any access, got 0x%08x\n", Access);
1053
1054     res = AddAccessAllowedAce(Acl, ACL_REVISION, KEY_READ, EveryoneSid);
1055     ok(res, "AddAccessAllowedAce failed with error %d\n", GetLastError());
1056
1057     res = AddAccessDeniedAce(Acl, ACL_REVISION, KEY_SET_VALUE, AdminSid);
1058     ok(res, "AddAccessDeniedAce failed with error %d\n", GetLastError());
1059
1060     /* sd with dacl */
1061     Access = AccessStatus = 0xdeadbeef;
1062     ret = AccessCheck(SecurityDescriptor, Token, KEY_READ, &Mapping,
1063                       PrivSet, &PrivSetLen, &Access, &AccessStatus);
1064     ok(ret, "AccessCheck failed with error %d\n", GetLastError());
1065     ok(AccessStatus && (Access == KEY_READ),
1066         "AccessCheck failed to grant access with error %d\n",
1067         GetLastError());
1068
1069     ret = AccessCheck(SecurityDescriptor, Token, MAXIMUM_ALLOWED, &Mapping,
1070                       PrivSet, &PrivSetLen, &Access, &AccessStatus);
1071     ok(ret, "AccessCheck failed with error %d\n", GetLastError());
1072     ok(AccessStatus,
1073         "AccessCheck failed to grant any access with error %d\n",
1074         GetLastError());
1075     trace("AccessCheck with MAXIMUM_ALLOWED got Access 0x%08x\n", Access);
1076
1077     /* Access denied by SD */
1078     SetLastError(0xdeadbeef);
1079     Access = AccessStatus = 0xdeadbeef;
1080     ret = AccessCheck(SecurityDescriptor, Token, KEY_WRITE, &Mapping,
1081                       PrivSet, &PrivSetLen, &Access, &AccessStatus);
1082     ok(ret, "AccessCheck failed with error %d\n", GetLastError());
1083     err = GetLastError();
1084     ok(!AccessStatus && err == ERROR_ACCESS_DENIED, "AccessCheck should have failed "
1085        "with ERROR_ACCESS_DENIED, instead of %d\n", err);
1086     ok(!Access, "Should have failed to grant any access, got 0x%08x\n", Access);
1087
1088     SetLastError(0);
1089     PrivSet->PrivilegeCount = 16;
1090     ret = AccessCheck(SecurityDescriptor, Token, ACCESS_SYSTEM_SECURITY, &Mapping,
1091                       PrivSet, &PrivSetLen, &Access, &AccessStatus);
1092     ok(ret && !AccessStatus && GetLastError() == ERROR_PRIVILEGE_NOT_HELD,
1093         "AccessCheck should have failed with ERROR_PRIVILEGE_NOT_HELD, instead of %d\n",
1094         GetLastError());
1095
1096     ret = ImpersonateLoggedOnUser(Token);
1097     ok(ret, "ImpersonateLoggedOnUser failed with error %d\n", GetLastError());
1098     ret = pRtlAdjustPrivilege(SE_SECURITY_PRIVILEGE, TRUE, TRUE, &Enabled);
1099     if (!ret)
1100     {
1101         SetLastError(0);
1102         PrivSet->PrivilegeCount = 16;
1103         ret = AccessCheck(SecurityDescriptor, Token, ACCESS_SYSTEM_SECURITY, &Mapping,
1104                           PrivSet, &PrivSetLen, &Access, &AccessStatus);
1105         ok(ret && AccessStatus && GetLastError() == 0,
1106             "AccessCheck should have succeeded, error %d\n",
1107             GetLastError());
1108         ok(Access == ACCESS_SYSTEM_SECURITY,
1109             "Access should be equal to ACCESS_SYSTEM_SECURITY instead of 0x%08x\n",
1110             Access);
1111     }
1112     else
1113         trace("Couldn't get SE_SECURITY_PRIVILEGE (0x%08x), skipping ACCESS_SYSTEM_SECURITY test\n",
1114             ret);
1115     ret = RevertToSelf();
1116     ok(ret, "RevertToSelf failed with error %d\n", GetLastError());
1117
1118     /* test INHERIT_ONLY_ACE */
1119     ret = InitializeAcl(Acl, 256, ACL_REVISION);
1120     ok(ret, "InitializeAcl failed with error %d\n", GetLastError());
1121
1122     /* NT doesn't have AddAccessAllowedAceEx. Skipping this call/test doesn't influence
1123      * the next ones.
1124      */
1125     if (pAddAccessAllowedAceEx)
1126     {
1127         ret = pAddAccessAllowedAceEx(Acl, ACL_REVISION, INHERIT_ONLY_ACE, KEY_READ, EveryoneSid);
1128         ok(ret, "AddAccessAllowedAceEx failed with error %d\n", GetLastError());
1129     }
1130     else
1131         win_skip("AddAccessAllowedAceEx is not available\n");
1132
1133     ret = AccessCheck(SecurityDescriptor, Token, KEY_READ, &Mapping,
1134                       PrivSet, &PrivSetLen, &Access, &AccessStatus);
1135     ok(ret, "AccessCheck failed with error %d\n", GetLastError());
1136     err = GetLastError();
1137     ok(!AccessStatus && err == ERROR_ACCESS_DENIED, "AccessCheck should have failed "
1138        "with ERROR_ACCESS_DENIED, instead of %d\n", err);
1139     ok(!Access, "Should have failed to grant any access, got 0x%08x\n", Access);
1140
1141     CloseHandle(Token);
1142
1143     res = DuplicateToken(ProcessToken, SecurityAnonymous, &Token);
1144     ok(res, "DuplicateToken failed with error %d\n", GetLastError());
1145
1146     SetLastError(0xdeadbeef);
1147     ret = AccessCheck(SecurityDescriptor, Token, MAXIMUM_ALLOWED, &Mapping,
1148                       PrivSet, &PrivSetLen, &Access, &AccessStatus);
1149     err = GetLastError();
1150     ok(!ret && err == ERROR_BAD_IMPERSONATION_LEVEL, "AccessCheck should have failed "
1151        "with ERROR_BAD_IMPERSONATION_LEVEL, instead of %d\n", err);
1152
1153     CloseHandle(Token);
1154
1155     SetLastError(0xdeadbeef);
1156     ret = AccessCheck(SecurityDescriptor, ProcessToken, KEY_READ, &Mapping,
1157                       PrivSet, &PrivSetLen, &Access, &AccessStatus);
1158     err = GetLastError();
1159     ok(!ret && err == ERROR_NO_IMPERSONATION_TOKEN, "AccessCheck should have failed "
1160        "with ERROR_NO_IMPERSONATION_TOKEN, instead of %d\n", err);
1161
1162     CloseHandle(ProcessToken);
1163
1164     if (EveryoneSid)
1165         FreeSid(EveryoneSid);
1166     if (AdminSid)
1167         FreeSid(AdminSid);
1168     if (UsersSid)
1169         FreeSid(UsersSid);
1170     HeapFree(GetProcessHeap(), 0, Acl);
1171     HeapFree(GetProcessHeap(), 0, SecurityDescriptor);
1172     HeapFree(GetProcessHeap(), 0, PrivSet);
1173 }
1174
1175 /* test GetTokenInformation for the various attributes */
1176 static void test_token_attr(void)
1177 {
1178     HANDLE Token, ImpersonationToken;
1179     DWORD Size, Size2;
1180     TOKEN_PRIVILEGES *Privileges;
1181     TOKEN_GROUPS *Groups;
1182     TOKEN_USER *User;
1183     TOKEN_DEFAULT_DACL *Dacl;
1184     BOOL ret;
1185     DWORD i, GLE;
1186     LPSTR SidString;
1187     SECURITY_IMPERSONATION_LEVEL ImpersonationLevel;
1188     ACL *acl;
1189
1190     /* cygwin-like use case */
1191     SetLastError(0xdeadbeef);
1192     ret = OpenProcessToken(GetCurrentProcess(), MAXIMUM_ALLOWED, &Token);
1193     if(!ret && (GetLastError() == ERROR_CALL_NOT_IMPLEMENTED))
1194     {
1195         win_skip("OpenProcessToken is not implemented\n");
1196         return;
1197     }
1198     ok(ret, "OpenProcessToken failed with error %d\n", GetLastError());
1199     if (ret)
1200     {
1201         DWORD buf[256]; /* GetTokenInformation wants a dword-aligned buffer */
1202         Size = sizeof(buf);
1203         ret = GetTokenInformation(Token, TokenUser,(void*)buf, Size, &Size);
1204         ok(ret, "GetTokenInformation failed with error %d\n", GetLastError());
1205         Size = sizeof(ImpersonationLevel);
1206         ret = GetTokenInformation(Token, TokenImpersonationLevel, &ImpersonationLevel, Size, &Size);
1207         GLE = GetLastError();
1208         ok(!ret && (GLE == ERROR_INVALID_PARAMETER), "GetTokenInformation(TokenImpersonationLevel) on primary token should have failed with ERROR_INVALID_PARAMETER instead of %d\n", GLE);
1209         CloseHandle(Token);
1210     }
1211
1212     if(!pConvertSidToStringSidA)
1213     {
1214         win_skip("ConvertSidToStringSidA is not available\n");
1215         return;
1216     }
1217
1218     SetLastError(0xdeadbeef);
1219     ret = OpenProcessToken(GetCurrentProcess(), TOKEN_ALL_ACCESS, &Token);
1220     ok(ret, "OpenProcessToken failed with error %d\n", GetLastError());
1221
1222     /* groups */
1223     ret = GetTokenInformation(Token, TokenGroups, NULL, 0, &Size);
1224     Groups = HeapAlloc(GetProcessHeap(), 0, Size);
1225     ret = GetTokenInformation(Token, TokenGroups, Groups, Size, &Size);
1226     ok(ret, "GetTokenInformation(TokenGroups) failed with error %d\n", GetLastError());
1227     trace("TokenGroups:\n");
1228     for (i = 0; i < Groups->GroupCount; i++)
1229     {
1230         DWORD NameLength = 255;
1231         TCHAR Name[255];
1232         DWORD DomainLength = 255;
1233         TCHAR Domain[255];
1234         SID_NAME_USE SidNameUse;
1235         pConvertSidToStringSidA(Groups->Groups[i].Sid, &SidString);
1236         Name[0] = '\0';
1237         Domain[0] = '\0';
1238         ret = LookupAccountSid(NULL, Groups->Groups[i].Sid, Name, &NameLength, Domain, &DomainLength, &SidNameUse);
1239         if (ret)
1240         {
1241             trace("%s, %s\\%s use: %d attr: 0x%08x\n", SidString, Domain, Name, SidNameUse, Groups->Groups[i].Attributes);
1242             LocalFree(SidString);
1243         }
1244         else trace("attr: 0x%08x LookupAccountSid failed with error %d\n", Groups->Groups[i].Attributes, GetLastError());
1245     }
1246     HeapFree(GetProcessHeap(), 0, Groups);
1247
1248     /* user */
1249     ret = GetTokenInformation(Token, TokenUser, NULL, 0, &Size);
1250     ok(!ret && (GetLastError() == ERROR_INSUFFICIENT_BUFFER),
1251         "GetTokenInformation(TokenUser) failed with error %d\n", GetLastError());
1252     User = HeapAlloc(GetProcessHeap(), 0, Size);
1253     ret = GetTokenInformation(Token, TokenUser, User, Size, &Size);
1254     ok(ret,
1255         "GetTokenInformation(TokenUser) failed with error %d\n", GetLastError());
1256
1257     pConvertSidToStringSidA(User->User.Sid, &SidString);
1258     trace("TokenUser: %s attr: 0x%08x\n", SidString, User->User.Attributes);
1259     LocalFree(SidString);
1260     HeapFree(GetProcessHeap(), 0, User);
1261
1262     /* privileges */
1263     ret = GetTokenInformation(Token, TokenPrivileges, NULL, 0, &Size);
1264     ok(!ret && (GetLastError() == ERROR_INSUFFICIENT_BUFFER),
1265         "GetTokenInformation(TokenPrivileges) failed with error %d\n", GetLastError());
1266     Privileges = HeapAlloc(GetProcessHeap(), 0, Size);
1267     ret = GetTokenInformation(Token, TokenPrivileges, Privileges, Size, &Size);
1268     ok(ret,
1269         "GetTokenInformation(TokenPrivileges) failed with error %d\n", GetLastError());
1270     trace("TokenPrivileges:\n");
1271     for (i = 0; i < Privileges->PrivilegeCount; i++)
1272     {
1273         TCHAR Name[256];
1274         DWORD NameLen = sizeof(Name)/sizeof(Name[0]);
1275         LookupPrivilegeName(NULL, &Privileges->Privileges[i].Luid, Name, &NameLen);
1276         trace("\t%s, 0x%x\n", Name, Privileges->Privileges[i].Attributes);
1277     }
1278     HeapFree(GetProcessHeap(), 0, Privileges);
1279
1280     ret = DuplicateToken(Token, SecurityAnonymous, &ImpersonationToken);
1281     ok(ret, "DuplicateToken failed with error %d\n", GetLastError());
1282
1283     Size = sizeof(ImpersonationLevel);
1284     ret = GetTokenInformation(ImpersonationToken, TokenImpersonationLevel, &ImpersonationLevel, Size, &Size);
1285     ok(ret, "GetTokenInformation(TokenImpersonationLevel) failed with error %d\n", GetLastError());
1286     ok(ImpersonationLevel == SecurityAnonymous, "ImpersonationLevel should have been SecurityAnonymous instead of %d\n", ImpersonationLevel);
1287
1288     CloseHandle(ImpersonationToken);
1289
1290     /* default dacl */
1291     ret = GetTokenInformation(Token, TokenDefaultDacl, NULL, 0, &Size);
1292     ok(!ret && (GetLastError() == ERROR_INSUFFICIENT_BUFFER),
1293         "GetTokenInformation(TokenDefaultDacl) failed with error %u\n", GetLastError());
1294
1295     Dacl = HeapAlloc(GetProcessHeap(), 0, Size);
1296     ret = GetTokenInformation(Token, TokenDefaultDacl, Dacl, Size, &Size);
1297     ok(ret, "GetTokenInformation(TokenDefaultDacl) failed with error %u\n", GetLastError());
1298
1299     SetLastError(0xdeadbeef);
1300     ret = SetTokenInformation(Token, TokenDefaultDacl, NULL, 0);
1301     GLE = GetLastError();
1302     ok(!ret, "SetTokenInformation(TokenDefaultDacl) succeeded\n");
1303     ok(GLE == ERROR_BAD_LENGTH, "expected ERROR_BAD_LENGTH got %u\n", GLE);
1304
1305     SetLastError(0xdeadbeef);
1306     ret = SetTokenInformation(Token, TokenDefaultDacl, NULL, Size);
1307     GLE = GetLastError();
1308     ok(!ret, "SetTokenInformation(TokenDefaultDacl) succeeded\n");
1309     ok(GLE == ERROR_NOACCESS, "expected ERROR_NOACCESS got %u\n", GLE);
1310
1311     acl = Dacl->DefaultDacl;
1312     Dacl->DefaultDacl = NULL;
1313
1314     ret = SetTokenInformation(Token, TokenDefaultDacl, Dacl, Size);
1315     ok(ret, "SetTokenInformation(TokenDefaultDacl) succeeded\n");
1316
1317     Size2 = 0;
1318     Dacl->DefaultDacl = (ACL *)0xdeadbeef;
1319     ret = GetTokenInformation(Token, TokenDefaultDacl, Dacl, Size, &Size2);
1320     ok(ret, "GetTokenInformation(TokenDefaultDacl) failed with error %u\n", GetLastError());
1321     ok(Dacl->DefaultDacl == NULL, "expected NULL, got %p\n", Dacl->DefaultDacl);
1322     ok(Size2 == sizeof(TOKEN_DEFAULT_DACL), "got %u expected sizeof(TOKEN_DEFAULT_DACL)\n", Size2);
1323
1324     Dacl->DefaultDacl = acl;
1325     ret = SetTokenInformation(Token, TokenDefaultDacl, Dacl, Size);
1326     ok(ret, "SetTokenInformation(TokenDefaultDacl) failed with error %u\n", GetLastError());
1327
1328     ret = GetTokenInformation(Token, TokenDefaultDacl, Dacl, Size, &Size2);
1329     ok(ret, "GetTokenInformation(TokenDefaultDacl) failed with error %u\n", GetLastError());
1330
1331     HeapFree(GetProcessHeap(), 0, Dacl);
1332     CloseHandle(Token);
1333 }
1334
1335 typedef union _MAX_SID
1336 {
1337     SID sid;
1338     char max[SECURITY_MAX_SID_SIZE];
1339 } MAX_SID;
1340
1341 static void test_sid_str(PSID * sid)
1342 {
1343     char *str_sid;
1344     BOOL ret = pConvertSidToStringSidA(sid, &str_sid);
1345     ok(ret, "ConvertSidToStringSidA() failed: %d\n", GetLastError());
1346     if (ret)
1347     {
1348         char account[MAX_PATH], domain[MAX_PATH];
1349         SID_NAME_USE use;
1350         DWORD acc_size = MAX_PATH;
1351         DWORD dom_size = MAX_PATH;
1352         ret = LookupAccountSid(NULL, sid, account, &acc_size, domain, &dom_size, &use);
1353         ok(ret || (!ret && (GetLastError() == ERROR_NONE_MAPPED)),
1354            "LookupAccountSid(%s) failed: %d\n", str_sid, GetLastError());
1355         if (ret)
1356             trace(" %s %s\\%s %d\n", str_sid, domain, account, use);
1357         else if (GetLastError() == ERROR_NONE_MAPPED)
1358             trace(" %s couldn't be mapped\n", str_sid);
1359         LocalFree(str_sid);
1360     }
1361 }
1362
1363 struct well_known_sid_value
1364 {
1365     BOOL without_domain;
1366     const char *sid_string;
1367 } well_known_sid_values[] = {
1368 /*  0 */ {TRUE, "S-1-0-0"},  {TRUE, "S-1-1-0"},  {TRUE, "S-1-2-0"},  {TRUE, "S-1-3-0"},
1369 /*  4 */ {TRUE, "S-1-3-1"},  {TRUE, "S-1-3-2"},  {TRUE, "S-1-3-3"},  {TRUE, "S-1-5"},
1370 /*  8 */ {FALSE, "S-1-5-1"}, {TRUE, "S-1-5-2"},  {TRUE, "S-1-5-3"},  {TRUE, "S-1-5-4"},
1371 /* 12 */ {TRUE, "S-1-5-6"},  {TRUE, "S-1-5-7"},  {TRUE, "S-1-5-8"},  {TRUE, "S-1-5-9"},
1372 /* 16 */ {TRUE, "S-1-5-10"}, {TRUE, "S-1-5-11"}, {TRUE, "S-1-5-12"}, {TRUE, "S-1-5-13"},
1373 /* 20 */ {TRUE, "S-1-5-14"}, {FALSE, NULL},      {TRUE, "S-1-5-18"}, {TRUE, "S-1-5-19"},
1374 /* 24 */ {TRUE, "S-1-5-20"}, {TRUE, "S-1-5-32"},
1375 /* 26 */ {FALSE, "S-1-5-32-544"}, {TRUE, "S-1-5-32-545"}, {TRUE, "S-1-5-32-546"},
1376 /* 29 */ {TRUE, "S-1-5-32-547"},  {TRUE, "S-1-5-32-548"}, {TRUE, "S-1-5-32-549"},
1377 /* 32 */ {TRUE, "S-1-5-32-550"},  {TRUE, "S-1-5-32-551"}, {TRUE, "S-1-5-32-552"},
1378 /* 35 */ {TRUE, "S-1-5-32-554"},  {TRUE, "S-1-5-32-555"}, {TRUE, "S-1-5-32-556"},
1379 /* 38 */ {FALSE, "S-1-5-21-12-23-34-45-56-500"}, {FALSE, "S-1-5-21-12-23-34-45-56-501"},
1380 /* 40 */ {FALSE, "S-1-5-21-12-23-34-45-56-502"}, {FALSE, "S-1-5-21-12-23-34-45-56-512"},
1381 /* 42 */ {FALSE, "S-1-5-21-12-23-34-45-56-513"}, {FALSE, "S-1-5-21-12-23-34-45-56-514"},
1382 /* 44 */ {FALSE, "S-1-5-21-12-23-34-45-56-515"}, {FALSE, "S-1-5-21-12-23-34-45-56-516"},
1383 /* 46 */ {FALSE, "S-1-5-21-12-23-34-45-56-517"}, {FALSE, "S-1-5-21-12-23-34-45-56-518"},
1384 /* 48 */ {FALSE, "S-1-5-21-12-23-34-45-56-519"}, {FALSE, "S-1-5-21-12-23-34-45-56-520"},
1385 /* 50 */ {FALSE, "S-1-5-21-12-23-34-45-56-553"},
1386 /* Added in Windows Server 2003 */
1387 /* 51 */ {TRUE, "S-1-5-64-10"},   {TRUE, "S-1-5-64-21"},   {TRUE, "S-1-5-64-14"},
1388 /* 54 */ {TRUE, "S-1-5-15"},      {TRUE, "S-1-5-1000"},    {FALSE, "S-1-5-32-557"},
1389 /* 57 */ {TRUE, "S-1-5-32-558"},  {TRUE, "S-1-5-32-559"},  {TRUE, "S-1-5-32-560"},
1390 /* 60 */ {TRUE, "S-1-5-32-561"}, {TRUE, "S-1-5-32-562"},
1391 /* Added in Windows Vista: */
1392 /* 62 */ {TRUE, "S-1-5-32-568"},
1393 /* 63 */ {TRUE, "S-1-5-17"},      {FALSE, "S-1-5-32-569"}, {TRUE, "S-1-16-0"},
1394 /* 66 */ {TRUE, "S-1-16-4096"},   {TRUE, "S-1-16-8192"},   {TRUE, "S-1-16-12288"},
1395 /* 69 */ {TRUE, "S-1-16-16384"},  {TRUE, "S-1-5-33"},      {TRUE, "S-1-3-4"},
1396 /* 72 */ {FALSE, "S-1-5-21-12-23-34-45-56-571"},  {FALSE, "S-1-5-21-12-23-34-45-56-572"},
1397 /* 74 */ {TRUE, "S-1-5-22"}, {FALSE, "S-1-5-21-12-23-34-45-56-521"}, {TRUE, "S-1-5-32-573"}
1398 };
1399
1400 static void test_CreateWellKnownSid(void)
1401 {
1402     SID_IDENTIFIER_AUTHORITY ident = { SECURITY_NT_AUTHORITY };
1403     PSID domainsid, sid;
1404     DWORD size, error;
1405     BOOL ret;
1406     int i;
1407
1408     if (!pCreateWellKnownSid)
1409     {
1410         win_skip("CreateWellKnownSid not available\n");
1411         return;
1412     }
1413
1414     size = 0;
1415     SetLastError(0xdeadbeef);
1416     ret = pCreateWellKnownSid(WinInteractiveSid, NULL, NULL, &size);
1417     error = GetLastError();
1418     ok(!ret, "CreateWellKnownSid succeeded\n");
1419     ok(error == ERROR_INSUFFICIENT_BUFFER, "expected ERROR_INSUFFICIENT_BUFFER, got %u\n", error);
1420     ok(size, "expected size > 0\n");
1421
1422     SetLastError(0xdeadbeef);
1423     ret = pCreateWellKnownSid(WinInteractiveSid, NULL, NULL, &size);
1424     error = GetLastError();
1425     ok(!ret, "CreateWellKnownSid succeeded\n");
1426     ok(error == ERROR_INVALID_PARAMETER, "expected ERROR_INVALID_PARAMETER, got %u\n", error);
1427
1428     sid = HeapAlloc(GetProcessHeap(), 0, size);
1429     ret = pCreateWellKnownSid(WinInteractiveSid, NULL, sid, &size);
1430     ok(ret, "CreateWellKnownSid failed %u\n", GetLastError());
1431     HeapFree(GetProcessHeap(), 0, sid);
1432
1433     /* a domain sid usually have three subauthorities but we test that CreateWellKnownSid doesn't check it */
1434     AllocateAndInitializeSid(&ident, 6, SECURITY_NT_NON_UNIQUE, 12, 23, 34, 45, 56, 0, 0, &domainsid);
1435
1436     for (i = 0; i < sizeof(well_known_sid_values)/sizeof(well_known_sid_values[0]); i++)
1437     {
1438         struct well_known_sid_value *value = &well_known_sid_values[i];
1439         char sid_buffer[SECURITY_MAX_SID_SIZE];
1440         LPSTR str;
1441         DWORD cb;
1442
1443         if (value->sid_string == NULL)
1444             continue;
1445
1446         if (i > WinAccountRasAndIasServersSid)
1447         {
1448             /* These SIDs aren't implemented by all Windows versions - detect it and break the loop */
1449             cb = sizeof(sid_buffer);
1450             if (!pCreateWellKnownSid(i, domainsid, sid_buffer, &cb))
1451             {
1452                 skip("Well known SIDs starting from %d are not implemented\n", i);
1453                 break;
1454             }
1455         }
1456
1457         cb = sizeof(sid_buffer);
1458         ok(pCreateWellKnownSid(i, value->without_domain ? NULL : domainsid, sid_buffer, &cb), "Couldn't create well known sid %d\n", i);
1459         expect_eq(GetSidLengthRequired(*GetSidSubAuthorityCount(sid_buffer)), cb, DWORD, "%d");
1460         ok(IsValidSid(sid_buffer), "The sid is not valid\n");
1461         ok(pConvertSidToStringSidA(sid_buffer, &str), "Couldn't convert SID to string\n");
1462         ok(strcmp(str, value->sid_string) == 0, "SID mismatch - expected %s, got %s\n",
1463             value->sid_string, str);
1464         LocalFree(str);
1465
1466         if (value->without_domain)
1467         {
1468             char buf2[SECURITY_MAX_SID_SIZE];
1469             cb = sizeof(buf2);
1470             ok(pCreateWellKnownSid(i, domainsid, buf2, &cb), "Couldn't create well known sid %d with optional domain\n", i);
1471             expect_eq(GetSidLengthRequired(*GetSidSubAuthorityCount(sid_buffer)), cb, DWORD, "%d");
1472             ok(memcmp(buf2, sid_buffer, cb) == 0, "SID create with domain is different than without (%d)\n", i);
1473         }
1474     }
1475 }
1476
1477 static void test_LookupAccountSid(void)
1478 {
1479     SID_IDENTIFIER_AUTHORITY SIDAuthNT = { SECURITY_NT_AUTHORITY };
1480     CHAR accountA[MAX_PATH], domainA[MAX_PATH], usernameA[MAX_PATH];
1481     DWORD acc_sizeA, dom_sizeA, user_sizeA;
1482     DWORD real_acc_sizeA, real_dom_sizeA;
1483     WCHAR accountW[MAX_PATH], domainW[MAX_PATH];
1484     DWORD acc_sizeW, dom_sizeW;
1485     DWORD real_acc_sizeW, real_dom_sizeW;
1486     PSID pUsersSid = NULL;
1487     SID_NAME_USE use;
1488     BOOL ret;
1489     DWORD size,cbti = 0;
1490     MAX_SID  max_sid;
1491     CHAR *str_sidA;
1492     int i;
1493     HANDLE hToken;
1494     PTOKEN_USER ptiUser = NULL;
1495
1496     /* native windows crashes if account size, domain size, or name use is NULL */
1497
1498     ret = AllocateAndInitializeSid(&SIDAuthNT, 2, SECURITY_BUILTIN_DOMAIN_RID,
1499         DOMAIN_ALIAS_RID_USERS, 0, 0, 0, 0, 0, 0, &pUsersSid);
1500     ok(ret || (GetLastError() == ERROR_CALL_NOT_IMPLEMENTED),
1501        "AllocateAndInitializeSid failed with error %d\n", GetLastError());
1502
1503     /* not running on NT so give up */
1504     if (!ret && (GetLastError() == ERROR_CALL_NOT_IMPLEMENTED))
1505         return;
1506
1507     real_acc_sizeA = MAX_PATH;
1508     real_dom_sizeA = MAX_PATH;
1509     ret = LookupAccountSidA(NULL, pUsersSid, accountA, &real_acc_sizeA, domainA, &real_dom_sizeA, &use);
1510     ok(ret, "LookupAccountSidA() Expected TRUE, got FALSE\n");
1511
1512     /* try NULL account */
1513     acc_sizeA = MAX_PATH;
1514     dom_sizeA = MAX_PATH;
1515     ret = LookupAccountSidA(NULL, pUsersSid, NULL, &acc_sizeA, domainA, &dom_sizeA, &use);
1516     ok(ret, "LookupAccountSidA() Expected TRUE, got FALSE\n");
1517
1518     /* try NULL domain */
1519     acc_sizeA = MAX_PATH;
1520     dom_sizeA = MAX_PATH;
1521     ret = LookupAccountSidA(NULL, pUsersSid, accountA, &acc_sizeA, NULL, &dom_sizeA, &use);
1522     ok(ret, "LookupAccountSidA() Expected TRUE, got FALSE\n");
1523
1524     /* try a small account buffer */
1525     acc_sizeA = 1;
1526     dom_sizeA = MAX_PATH;
1527     accountA[0] = 0;
1528     ret = LookupAccountSidA(NULL, pUsersSid, accountA, &acc_sizeA, domainA, &dom_sizeA, &use);
1529     ok(!ret, "LookupAccountSidA() Expected FALSE got TRUE\n");
1530     ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER,
1531        "LookupAccountSidA() Expected ERROR_NOT_ENOUGH_MEMORY, got %u\n", GetLastError());
1532
1533     /* try a 0 sized account buffer */
1534     acc_sizeA = 0;
1535     dom_sizeA = MAX_PATH;
1536     accountA[0] = 0;
1537     ret = LookupAccountSidA(NULL, pUsersSid, accountA, &acc_sizeA, domainA, &dom_sizeA, &use);
1538     /* this can fail or succeed depending on OS version but the size will always be returned */
1539     ok(acc_sizeA == real_acc_sizeA + 1,
1540        "LookupAccountSidA() Expected acc_size = %u, got %u\n",
1541        real_acc_sizeA + 1, acc_sizeA);
1542
1543     /* try a 0 sized account buffer */
1544     acc_sizeA = 0;
1545     dom_sizeA = MAX_PATH;
1546     ret = LookupAccountSidA(NULL, pUsersSid, NULL, &acc_sizeA, domainA, &dom_sizeA, &use);
1547     /* this can fail or succeed depending on OS version but the size will always be returned */
1548     ok(acc_sizeA == real_acc_sizeA + 1,
1549        "LookupAccountSid() Expected acc_size = %u, got %u\n",
1550        real_acc_sizeA + 1, acc_sizeA);
1551
1552     /* try a small domain buffer */
1553     dom_sizeA = 1;
1554     acc_sizeA = MAX_PATH;
1555     accountA[0] = 0;
1556     ret = LookupAccountSidA(NULL, pUsersSid, accountA, &acc_sizeA, domainA, &dom_sizeA, &use);
1557     ok(!ret, "LookupAccountSidA() Expected FALSE got TRUE\n");
1558     ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER,
1559        "LookupAccountSidA() Expected ERROR_NOT_ENOUGH_MEMORY, got %u\n", GetLastError());
1560
1561     /* try a 0 sized domain buffer */
1562     dom_sizeA = 0;
1563     acc_sizeA = MAX_PATH;
1564     accountA[0] = 0;
1565     ret = LookupAccountSidA(NULL, pUsersSid, accountA, &acc_sizeA, domainA, &dom_sizeA, &use);
1566     /* this can fail or succeed depending on OS version but the size will always be returned */
1567     ok(dom_sizeA == real_dom_sizeA + 1,
1568        "LookupAccountSidA() Expected dom_size = %u, got %u\n",
1569        real_dom_sizeA + 1, dom_sizeA);
1570
1571     /* try a 0 sized domain buffer */
1572     dom_sizeA = 0;
1573     acc_sizeA = MAX_PATH;
1574     ret = LookupAccountSidA(NULL, pUsersSid, accountA, &acc_sizeA, NULL, &dom_sizeA, &use);
1575     /* this can fail or succeed depending on OS version but the size will always be returned */
1576     ok(dom_sizeA == real_dom_sizeA + 1,
1577        "LookupAccountSidA() Expected dom_size = %u, got %u\n",
1578        real_dom_sizeA + 1, dom_sizeA);
1579
1580     real_acc_sizeW = MAX_PATH;
1581     real_dom_sizeW = MAX_PATH;
1582     ret = LookupAccountSidW(NULL, pUsersSid, accountW, &real_acc_sizeW, domainW, &real_dom_sizeW, &use);
1583     ok(ret, "LookupAccountSidW() Expected TRUE, got FALSE\n");
1584
1585     /* try an invalid system name */
1586     real_acc_sizeA = MAX_PATH;
1587     real_dom_sizeA = MAX_PATH;
1588     ret = LookupAccountSidA("deepthought", pUsersSid, accountA, &real_acc_sizeA, domainA, &real_dom_sizeA, &use);
1589     ok(!ret, "LookupAccountSidA() Expected FALSE got TRUE\n");
1590     ok(GetLastError() == RPC_S_SERVER_UNAVAILABLE || GetLastError() == RPC_S_INVALID_NET_ADDR /* Vista */,
1591        "LookupAccountSidA() Expected RPC_S_SERVER_UNAVAILABLE or RPC_S_INVALID_NET_ADDR, got %u\n", GetLastError());
1592
1593     /* native windows crashes if domainW or accountW is NULL */
1594
1595     /* try a small account buffer */
1596     acc_sizeW = 1;
1597     dom_sizeW = MAX_PATH;
1598     accountW[0] = 0;
1599     ret = LookupAccountSidW(NULL, pUsersSid, accountW, &acc_sizeW, domainW, &dom_sizeW, &use);
1600     ok(!ret, "LookupAccountSidW() Expected FALSE got TRUE\n");
1601     ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER,
1602        "LookupAccountSidW() Expected ERROR_NOT_ENOUGH_MEMORY, got %u\n", GetLastError());
1603
1604     /* try a 0 sized account buffer */
1605     acc_sizeW = 0;
1606     dom_sizeW = MAX_PATH;
1607     accountW[0] = 0;
1608     ret = LookupAccountSidW(NULL, pUsersSid, accountW, &acc_sizeW, domainW, &dom_sizeW, &use);
1609     /* this can fail or succeed depending on OS version but the size will always be returned */
1610     ok(acc_sizeW == real_acc_sizeW + 1,
1611        "LookupAccountSidW() Expected acc_size = %u, got %u\n",
1612        real_acc_sizeW + 1, acc_sizeW);
1613
1614     /* try a 0 sized account buffer */
1615     acc_sizeW = 0;
1616     dom_sizeW = MAX_PATH;
1617     ret = LookupAccountSidW(NULL, pUsersSid, NULL, &acc_sizeW, domainW, &dom_sizeW, &use);
1618     /* this can fail or succeed depending on OS version but the size will always be returned */
1619     ok(acc_sizeW == real_acc_sizeW + 1,
1620        "LookupAccountSidW() Expected acc_size = %u, got %u\n",
1621        real_acc_sizeW + 1, acc_sizeW);
1622
1623     /* try a small domain buffer */
1624     dom_sizeW = 1;
1625     acc_sizeW = MAX_PATH;
1626     accountW[0] = 0;
1627     ret = LookupAccountSidW(NULL, pUsersSid, accountW, &acc_sizeW, domainW, &dom_sizeW, &use);
1628     ok(!ret, "LookupAccountSidW() Expected FALSE got TRUE\n");
1629     ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER,
1630        "LookupAccountSidW() Expected ERROR_NOT_ENOUGH_MEMORY, got %u\n", GetLastError());
1631
1632     /* try a 0 sized domain buffer */
1633     dom_sizeW = 0;
1634     acc_sizeW = MAX_PATH;
1635     accountW[0] = 0;
1636     ret = LookupAccountSidW(NULL, pUsersSid, accountW, &acc_sizeW, domainW, &dom_sizeW, &use);
1637     /* this can fail or succeed depending on OS version but the size will always be returned */
1638     ok(dom_sizeW == real_dom_sizeW + 1,
1639        "LookupAccountSidW() Expected dom_size = %u, got %u\n",
1640        real_dom_sizeW + 1, dom_sizeW);
1641
1642     /* try a 0 sized domain buffer */
1643     dom_sizeW = 0;
1644     acc_sizeW = MAX_PATH;
1645     ret = LookupAccountSidW(NULL, pUsersSid, accountW, &acc_sizeW, NULL, &dom_sizeW, &use);
1646     /* this can fail or succeed depending on OS version but the size will always be returned */
1647     ok(dom_sizeW == real_dom_sizeW + 1,
1648        "LookupAccountSidW() Expected dom_size = %u, got %u\n",
1649        real_dom_sizeW + 1, dom_sizeW);
1650
1651     FreeSid(pUsersSid);
1652
1653     /* Test LookupAccountSid with Sid retrieved from token information.
1654      This assumes this process is running under the account of the current user.*/
1655     ret = OpenProcessToken(GetCurrentProcess(), TOKEN_QUERY|TOKEN_DUPLICATE, &hToken);
1656     ret = GetTokenInformation(hToken, TokenUser, NULL, 0, &cbti);
1657     ptiUser = (PTOKEN_USER) HeapAlloc(GetProcessHeap(), 0, cbti);
1658     if (GetTokenInformation(hToken, TokenUser, ptiUser, cbti, &cbti))
1659     {
1660         acc_sizeA = dom_sizeA = MAX_PATH;
1661         ret = LookupAccountSidA(NULL, ptiUser->User.Sid, accountA, &acc_sizeA, domainA, &dom_sizeA, &use);
1662         ok(ret, "LookupAccountSidA() Expected TRUE, got FALSE\n");
1663         user_sizeA = MAX_PATH;
1664         ret = GetUserNameA(usernameA , &user_sizeA);
1665         ok(ret, "GetUserNameA() Expected TRUE, got FALSE\n");
1666         todo_wine
1667         {
1668             ok(lstrcmpA(usernameA, accountA) == 0, "LookupAccountSidA() Expected account name: %s got: %s\n", usernameA, accountA );
1669         }
1670     }
1671     HeapFree(GetProcessHeap(), 0, ptiUser);
1672
1673     if (pCreateWellKnownSid && pConvertSidToStringSidA)
1674     {
1675         trace("Well Known SIDs:\n");
1676         for (i = 0; i <= 60; i++)
1677         {
1678             size = SECURITY_MAX_SID_SIZE;
1679             if (pCreateWellKnownSid(i, NULL, &max_sid.sid, &size))
1680             {
1681                 if (pConvertSidToStringSidA(&max_sid.sid, &str_sidA))
1682                 {
1683                     acc_sizeA = MAX_PATH;
1684                     dom_sizeA = MAX_PATH;
1685                     if (LookupAccountSidA(NULL, &max_sid.sid, accountA, &acc_sizeA, domainA, &dom_sizeA, &use))
1686                         trace(" %d: %s %s\\%s %d\n", i, str_sidA, domainA, accountA, use);
1687                     LocalFree(str_sidA);
1688                 }
1689             }
1690             else
1691             {
1692                 if (GetLastError() != ERROR_INVALID_PARAMETER)
1693                     trace(" CreateWellKnownSid(%d) failed: %d\n", i, GetLastError());
1694                 else
1695                     trace(" %d: not supported\n", i);
1696             }
1697         }
1698
1699         pLsaQueryInformationPolicy = (fnLsaQueryInformationPolicy)GetProcAddress( hmod, "LsaQueryInformationPolicy");
1700         pLsaOpenPolicy = (fnLsaOpenPolicy)GetProcAddress( hmod, "LsaOpenPolicy");
1701         pLsaFreeMemory = (fnLsaFreeMemory)GetProcAddress( hmod, "LsaFreeMemory");
1702         pLsaClose = (fnLsaClose)GetProcAddress( hmod, "LsaClose");
1703
1704         if (pLsaQueryInformationPolicy && pLsaOpenPolicy && pLsaFreeMemory && pLsaClose)
1705         {
1706             NTSTATUS status;
1707             LSA_HANDLE handle;
1708             LSA_OBJECT_ATTRIBUTES object_attributes;
1709
1710             ZeroMemory(&object_attributes, sizeof(object_attributes));
1711             object_attributes.Length = sizeof(object_attributes);
1712
1713             status = pLsaOpenPolicy( NULL, &object_attributes, POLICY_ALL_ACCESS, &handle);
1714             ok(status == STATUS_SUCCESS || status == STATUS_ACCESS_DENIED,
1715                "LsaOpenPolicy(POLICY_ALL_ACCESS) returned 0x%08x\n", status);
1716
1717             /* try a more restricted access mask if necessary */
1718             if (status == STATUS_ACCESS_DENIED) {
1719                 trace("LsaOpenPolicy(POLICY_ALL_ACCESS) failed, trying POLICY_VIEW_LOCAL_INFORMATION\n");
1720                 status = pLsaOpenPolicy( NULL, &object_attributes, POLICY_VIEW_LOCAL_INFORMATION, &handle);
1721                 ok(status == STATUS_SUCCESS, "LsaOpenPolicy(POLICY_VIEW_LOCAL_INFORMATION) returned 0x%08x\n", status);
1722             }
1723
1724             if (status == STATUS_SUCCESS)
1725             {
1726                 PPOLICY_ACCOUNT_DOMAIN_INFO info;
1727                 status = pLsaQueryInformationPolicy(handle, PolicyAccountDomainInformation, (PVOID*)&info);
1728                 ok(status == STATUS_SUCCESS, "LsaQueryInformationPolicy() failed, returned 0x%08x\n", status);
1729                 if (status == STATUS_SUCCESS)
1730                 {
1731                     ok(info->DomainSid!=0, "LsaQueryInformationPolicy(PolicyAccountDomainInformation) missing SID\n");
1732                     if (info->DomainSid)
1733                     {
1734                         int count = *GetSidSubAuthorityCount(info->DomainSid);
1735                         CopySid(GetSidLengthRequired(count), &max_sid, info->DomainSid);
1736                         test_sid_str((PSID)&max_sid.sid);
1737                         max_sid.sid.SubAuthority[count] = DOMAIN_USER_RID_ADMIN;
1738                         max_sid.sid.SubAuthorityCount = count + 1;
1739                         test_sid_str((PSID)&max_sid.sid);
1740                         max_sid.sid.SubAuthority[count] = DOMAIN_USER_RID_GUEST;
1741                         test_sid_str((PSID)&max_sid.sid);
1742                         max_sid.sid.SubAuthority[count] = DOMAIN_GROUP_RID_ADMINS;
1743                         test_sid_str((PSID)&max_sid.sid);
1744                         max_sid.sid.SubAuthority[count] = DOMAIN_GROUP_RID_USERS;
1745                         test_sid_str((PSID)&max_sid.sid);
1746                         max_sid.sid.SubAuthority[count] = DOMAIN_GROUP_RID_GUESTS;
1747                         test_sid_str((PSID)&max_sid.sid);
1748                         max_sid.sid.SubAuthority[count] = DOMAIN_GROUP_RID_COMPUTERS;
1749                         test_sid_str((PSID)&max_sid.sid);
1750                         max_sid.sid.SubAuthority[count] = DOMAIN_GROUP_RID_CONTROLLERS;
1751                         test_sid_str((PSID)&max_sid.sid);
1752                         max_sid.sid.SubAuthority[count] = DOMAIN_GROUP_RID_CERT_ADMINS;
1753                         test_sid_str((PSID)&max_sid.sid);
1754                         max_sid.sid.SubAuthority[count] = DOMAIN_GROUP_RID_SCHEMA_ADMINS;
1755                         test_sid_str((PSID)&max_sid.sid);
1756                         max_sid.sid.SubAuthority[count] = DOMAIN_GROUP_RID_ENTERPRISE_ADMINS;
1757                         test_sid_str((PSID)&max_sid.sid);
1758                         max_sid.sid.SubAuthority[count] = DOMAIN_GROUP_RID_POLICY_ADMINS;
1759                         test_sid_str((PSID)&max_sid.sid);
1760                         max_sid.sid.SubAuthority[count] = DOMAIN_ALIAS_RID_RAS_SERVERS;
1761                         test_sid_str((PSID)&max_sid.sid);
1762                         max_sid.sid.SubAuthority[count] = 1000; /* first user account */
1763                         test_sid_str((PSID)&max_sid.sid);
1764                     }
1765
1766                     pLsaFreeMemory((LPVOID)info);
1767                 }
1768
1769                 status = pLsaClose(handle);
1770                 ok(status == STATUS_SUCCESS, "LsaClose() failed, returned 0x%08x\n", status);
1771             }
1772         }
1773     }
1774 }
1775
1776 static BOOL get_sid_info(PSID psid, LPSTR *user, LPSTR *dom)
1777 {
1778     static CHAR account[UNLEN + 1];
1779     static CHAR domain[UNLEN + 1];
1780     DWORD size, dom_size;
1781     SID_NAME_USE use;
1782
1783     *user = account;
1784     *dom = domain;
1785
1786     size = dom_size = UNLEN + 1;
1787     account[0] = '\0';
1788     domain[0] = '\0';
1789     SetLastError(0xdeadbeef);
1790     return LookupAccountSidA(NULL, psid, account, &size, domain, &dom_size, &use);
1791 }
1792
1793 static void check_wellknown_name(const char* name, WELL_KNOWN_SID_TYPE result)
1794 {
1795     SID_IDENTIFIER_AUTHORITY ident = { SECURITY_NT_AUTHORITY };
1796     PSID domainsid;
1797     char wk_sid[SECURITY_MAX_SID_SIZE];
1798     DWORD cb;
1799
1800     DWORD sid_size, domain_size;
1801     SID_NAME_USE sid_use;
1802     LPSTR domain, account, sid_domain, wk_domain, wk_account;
1803     PSID psid;
1804     BOOL ret ,ret2;
1805
1806     sid_size = 0;
1807     domain_size = 0;
1808     ret = LookupAccountNameA(NULL, name, NULL, &sid_size, NULL, &domain_size, &sid_use);
1809     psid = HeapAlloc(GetProcessHeap(),0,sid_size);
1810     domain = HeapAlloc(GetProcessHeap(),0,domain_size);
1811     ret = LookupAccountNameA(NULL, name, psid, &sid_size, domain, &domain_size, &sid_use);
1812
1813     if (!result)
1814     {
1815         ok(!ret, " %s Should have failed to lookup account name\n",name);
1816         goto cleanup;
1817     }
1818
1819     AllocateAndInitializeSid(&ident, 6, SECURITY_NT_NON_UNIQUE, 12, 23, 34, 45, 56, 0, 0, &domainsid);
1820     cb = sizeof(wk_sid);
1821     if (!pCreateWellKnownSid(result, domainsid, wk_sid, &cb))
1822     {
1823         win_skip("SID %i is not available on the system\n",result);
1824         goto cleanup;
1825     }
1826
1827     ret2 = get_sid_info(wk_sid, &wk_account, &wk_domain);
1828     if (!ret2 && GetLastError() == ERROR_NONE_MAPPED)
1829     {
1830         win_skip("CreateWellKnownSid() succeeded but the account '%s' is not present (W2K)\n", name);
1831         goto cleanup;
1832     }
1833
1834     get_sid_info(psid, &account, &sid_domain);
1835
1836     ok(ret, "Failed to lookup account name %s\n",name);
1837     ok(sid_size != 0, "sid_size was zero\n");
1838
1839     ok(EqualSid(psid,wk_sid),"(%s) Sids fail to match well known sid!\n",name);
1840
1841     ok(!lstrcmp(account, wk_account), "Expected %s , got %s\n", account, wk_account);
1842     ok(!lstrcmp(domain, wk_domain), "Expected %s, got %s\n", wk_domain, domain);
1843     ok(sid_use == SidTypeWellKnownGroup , "Expected Use (5), got %d\n", sid_use);
1844
1845 cleanup:
1846     HeapFree(GetProcessHeap(),0,psid);
1847     HeapFree(GetProcessHeap(),0,domain);
1848 }
1849
1850 static void test_LookupAccountName(void)
1851 {
1852     DWORD sid_size, domain_size, user_size;
1853     DWORD sid_save, domain_save;
1854     CHAR user_name[UNLEN + 1];
1855     CHAR computer_name[UNLEN + 1];
1856     SID_NAME_USE sid_use;
1857     LPSTR domain, account, sid_dom;
1858     PSID psid;
1859     BOOL ret;
1860
1861     /* native crashes if (assuming all other parameters correct):
1862      *  - peUse is NULL
1863      *  - Sid is NULL and cbSid is > 0
1864      *  - cbSid or cchReferencedDomainName are NULL
1865      *  - ReferencedDomainName is NULL and cchReferencedDomainName is the correct size
1866      */
1867
1868     user_size = UNLEN + 1;
1869     SetLastError(0xdeadbeef);
1870     ret = GetUserNameA(user_name, &user_size);
1871     if (!ret && (GetLastError() == ERROR_NOT_LOGGED_ON))
1872     {
1873         /* Probably on win9x where the user used 'Cancel' instead of properly logging in */
1874         skip("Cannot get the user name (win9x and not logged in properly)\n");
1875         return;
1876     }
1877     ok(ret, "Failed to get user name : %d\n", GetLastError());
1878
1879     /* get sizes */
1880     sid_size = 0;
1881     domain_size = 0;
1882     sid_use = 0xcafebabe;
1883     SetLastError(0xdeadbeef);
1884     ret = LookupAccountNameA(NULL, user_name, NULL, &sid_size, NULL, &domain_size, &sid_use);
1885     if(!ret && (GetLastError() == ERROR_CALL_NOT_IMPLEMENTED))
1886     {
1887         win_skip("LookupAccountNameA is not implemented\n");
1888         return;
1889     }
1890     ok(!ret, "Expected 0, got %d\n", ret);
1891     ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER,
1892        "Expected ERROR_INSUFFICIENT_BUFFER, got %d\n", GetLastError());
1893     ok(sid_size != 0, "Expected non-zero sid size\n");
1894     ok(domain_size != 0, "Expected non-zero domain size\n");
1895     ok(sid_use == 0xcafebabe, "Expected 0xcafebabe, got %d\n", sid_use);
1896
1897     sid_save = sid_size;
1898     domain_save = domain_size;
1899
1900     psid = HeapAlloc(GetProcessHeap(), 0, sid_size);
1901     domain = HeapAlloc(GetProcessHeap(), 0, domain_size);
1902
1903     /* try valid account name */
1904     ret = LookupAccountNameA(NULL, user_name, psid, &sid_size, domain, &domain_size, &sid_use);
1905     get_sid_info(psid, &account, &sid_dom);
1906     ok(ret, "Failed to lookup account name\n");
1907     ok(sid_size == GetLengthSid(psid), "Expected %d, got %d\n", GetLengthSid(psid), sid_size);
1908     todo_wine
1909     {
1910         ok(!lstrcmp(account, user_name), "Expected %s, got %s\n", user_name, account);
1911         ok(!lstrcmp(domain, sid_dom), "Expected %s, got %s\n", sid_dom, domain);
1912     }
1913     ok(domain_size == domain_save - 1, "Expected %d, got %d\n", domain_save - 1, domain_size);
1914     ok(lstrlen(domain) == domain_size, "Expected %d, got %d\n", lstrlen(domain), domain_size);
1915     ok(sid_use == SidTypeUser, "Expected SidTypeUser (%d), got %d\n", SidTypeUser, sid_use);
1916     domain_size = domain_save;
1917     sid_size = sid_save;
1918
1919     if (PRIMARYLANGID(LANGIDFROMLCID(GetThreadLocale())) != LANG_ENGLISH)
1920     {
1921         skip("Non-english locale (test with hardcoded 'Everyone')\n");
1922     }
1923     else
1924     {
1925         ret = LookupAccountNameA(NULL, "Everyone", psid, &sid_size, domain, &domain_size, &sid_use);
1926         get_sid_info(psid, &account, &sid_dom);
1927         ok(ret, "Failed to lookup account name\n");
1928         ok(sid_size != 0, "sid_size was zero\n");
1929         ok(!lstrcmp(account, "Everyone"), "Expected Everyone, got %s\n", account);
1930         ok(!lstrcmp(domain, sid_dom), "Expected %s, got %s\n", sid_dom, domain);
1931         ok(domain_size == 0, "Expected 0, got %d\n", domain_size);
1932         ok(lstrlen(domain) == domain_size, "Expected %d, got %d\n", lstrlen(domain), domain_size);
1933         ok(sid_use == SidTypeWellKnownGroup, "Expected SidTypeWellKnownGroup (%d), got %d\n", SidTypeWellKnownGroup, sid_use);
1934         domain_size = domain_save;
1935     }
1936
1937     /* NULL Sid with zero sid size */
1938     SetLastError(0xdeadbeef);
1939     sid_size = 0;
1940     ret = LookupAccountNameA(NULL, user_name, NULL, &sid_size, domain, &domain_size, &sid_use);
1941     ok(!ret, "Expected 0, got %d\n", ret);
1942     ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER,
1943        "Expected ERROR_INSUFFICIENT_BUFFER, got %d\n", GetLastError());
1944     ok(sid_size == sid_save, "Expected %d, got %d\n", sid_save, sid_size);
1945     ok(domain_size == domain_save, "Expected %d, got %d\n", domain_save, domain_size);
1946
1947     /* try cchReferencedDomainName - 1 */
1948     SetLastError(0xdeadbeef);
1949     domain_size--;
1950     ret = LookupAccountNameA(NULL, user_name, NULL, &sid_size, domain, &domain_size, &sid_use);
1951     ok(!ret, "Expected 0, got %d\n", ret);
1952     ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER,
1953        "Expected ERROR_INSUFFICIENT_BUFFER, got %d\n", GetLastError());
1954     ok(sid_size == sid_save, "Expected %d, got %d\n", sid_save, sid_size);
1955     ok(domain_size == domain_save, "Expected %d, got %d\n", domain_save, domain_size);
1956
1957     /* NULL ReferencedDomainName with zero domain name size */
1958     SetLastError(0xdeadbeef);
1959     domain_size = 0;
1960     ret = LookupAccountNameA(NULL, user_name, psid, &sid_size, NULL, &domain_size, &sid_use);
1961     ok(!ret, "Expected 0, got %d\n", ret);
1962     ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER,
1963        "Expected ERROR_INSUFFICIENT_BUFFER, got %d\n", GetLastError());
1964     ok(sid_size == sid_save, "Expected %d, got %d\n", sid_save, sid_size);
1965     ok(domain_size == domain_save, "Expected %d, got %d\n", domain_save, domain_size);
1966
1967     HeapFree(GetProcessHeap(), 0, psid);
1968     HeapFree(GetProcessHeap(), 0, domain);
1969
1970     /* get sizes for NULL account name */
1971     sid_size = 0;
1972     domain_size = 0;
1973     sid_use = 0xcafebabe;
1974     SetLastError(0xdeadbeef);
1975     ret = LookupAccountNameA(NULL, NULL, NULL, &sid_size, NULL, &domain_size, &sid_use);
1976     if (!ret && GetLastError() == ERROR_NONE_MAPPED)
1977         win_skip("NULL account name doesn't work on NT4\n");
1978     else
1979     {
1980         ok(!ret, "Expected 0, got %d\n", ret);
1981         ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER,
1982            "Expected ERROR_INSUFFICIENT_BUFFER, got %d\n", GetLastError());
1983         ok(sid_size != 0, "Expected non-zero sid size\n");
1984         ok(domain_size != 0, "Expected non-zero domain size\n");
1985         ok(sid_use == 0xcafebabe, "Expected 0xcafebabe, got %d\n", sid_use);
1986
1987         psid = HeapAlloc(GetProcessHeap(), 0, sid_size);
1988         domain = HeapAlloc(GetProcessHeap(), 0, domain_size);
1989
1990         /* try NULL account name */
1991         ret = LookupAccountNameA(NULL, NULL, psid, &sid_size, domain, &domain_size, &sid_use);
1992         get_sid_info(psid, &account, &sid_dom);
1993         ok(ret, "Failed to lookup account name\n");
1994         /* Using a fixed string will not work on different locales */
1995         ok(!lstrcmp(account, domain),
1996            "Got %s for account and %s for domain, these should be the same\n",
1997            account, domain);
1998         ok(sid_use == SidTypeDomain, "Expected SidTypeDomain (%d), got %d\n", SidTypeDomain, sid_use);
1999
2000         HeapFree(GetProcessHeap(), 0, psid);
2001         HeapFree(GetProcessHeap(), 0, domain);
2002     }
2003
2004     /* try an invalid account name */
2005     SetLastError(0xdeadbeef);
2006     sid_size = 0;
2007     domain_size = 0;
2008     ret = LookupAccountNameA(NULL, "oogabooga", NULL, &sid_size, NULL, &domain_size, &sid_use);
2009     ok(!ret, "Expected 0, got %d\n", ret);
2010     ok(GetLastError() == ERROR_NONE_MAPPED ||
2011        broken(GetLastError() == ERROR_TRUSTED_RELATIONSHIP_FAILURE),
2012        "Expected ERROR_NONE_MAPPED, got %d\n", GetLastError());
2013     ok(sid_size == 0, "Expected 0, got %d\n", sid_size);
2014     ok(domain_size == 0, "Expected 0, got %d\n", domain_size);
2015
2016     /* try an invalid system name */
2017     SetLastError(0xdeadbeef);
2018     sid_size = 0;
2019     domain_size = 0;
2020     ret = LookupAccountNameA("deepthought", NULL, NULL, &sid_size, NULL, &domain_size, &sid_use);
2021     ok(!ret, "Expected 0, got %d\n", ret);
2022     ok(GetLastError() == RPC_S_SERVER_UNAVAILABLE || GetLastError() == RPC_S_INVALID_NET_ADDR /* Vista */,
2023        "Expected RPC_S_SERVER_UNAVAILABLE or RPC_S_INVALID_NET_ADDR, got %d\n", GetLastError());
2024     ok(sid_size == 0, "Expected 0, got %d\n", sid_size);
2025     ok(domain_size == 0, "Expected 0, got %d\n", domain_size);
2026
2027     /* try with the computer name as the account name */
2028     domain_size = sizeof(computer_name);
2029     GetComputerNameA(computer_name, &domain_size);
2030     sid_size = 0;
2031     domain_size = 0;
2032     ret = LookupAccountNameA(NULL, computer_name, NULL, &sid_size, NULL, &domain_size, &sid_use);
2033     ok(!ret && (GetLastError() == ERROR_INSUFFICIENT_BUFFER ||
2034        GetLastError() == ERROR_NONE_MAPPED /* in a domain */ ||
2035        broken(GetLastError() == ERROR_TRUSTED_DOMAIN_FAILURE) ||
2036        broken(GetLastError() == ERROR_TRUSTED_RELATIONSHIP_FAILURE)),
2037        "LookupAccountNameA failed: %d\n", GetLastError());
2038     if (GetLastError() == ERROR_INSUFFICIENT_BUFFER)
2039     {
2040         psid = HeapAlloc(GetProcessHeap(), 0, sid_size);
2041         domain = HeapAlloc(GetProcessHeap(), 0, domain_size);
2042         ret = LookupAccountNameA(NULL, computer_name, psid, &sid_size, domain, &domain_size, &sid_use);
2043         ok(ret, "LookupAccountNameA failed: %d\n", GetLastError());
2044         ok(sid_use == SidTypeDomain, "expected SidTypeDomain, got %d\n", sid_use);
2045         HeapFree(GetProcessHeap(), 0, domain);
2046         HeapFree(GetProcessHeap(), 0, psid);
2047     }
2048
2049     /* Well Known names */
2050     if (!pCreateWellKnownSid)
2051     {
2052         win_skip("CreateWellKnownSid not available\n");
2053         return;
2054     }
2055
2056     if (PRIMARYLANGID(LANGIDFROMLCID(GetThreadLocale())) != LANG_ENGLISH)
2057     {
2058         skip("Non-english locale (skipping well known name creation tests)\n");
2059         return;
2060     }
2061
2062     check_wellknown_name("LocalService", WinLocalServiceSid);
2063     check_wellknown_name("Local Service", WinLocalServiceSid);
2064     /* 2 spaces */
2065     check_wellknown_name("Local  Service", 0);
2066     check_wellknown_name("NetworkService", WinNetworkServiceSid);
2067     check_wellknown_name("Network Service", WinNetworkServiceSid);
2068
2069     /* example of some names where the spaces are not optional */
2070     check_wellknown_name("Terminal Server User", WinTerminalServerSid);
2071     check_wellknown_name("TerminalServer User", 0);
2072     check_wellknown_name("TerminalServerUser", 0);
2073     check_wellknown_name("Terminal ServerUser", 0);
2074
2075     check_wellknown_name("enterprise domain controllers",WinEnterpriseControllersSid);
2076     check_wellknown_name("enterprisedomain controllers", 0);
2077     check_wellknown_name("enterprise domaincontrollers", 0);
2078     check_wellknown_name("enterprisedomaincontrollers", 0);
2079
2080     /* case insensitivity */
2081     check_wellknown_name("lOCAlServICE", WinLocalServiceSid);
2082
2083     /* fully qualified account names */
2084     check_wellknown_name("NT AUTHORITY\\LocalService", WinLocalServiceSid);
2085     check_wellknown_name("nt authority\\Network Service", WinNetworkServiceSid);
2086     check_wellknown_name("nt authority test\\Network Service", 0);
2087     check_wellknown_name("Dummy\\Network Service", 0);
2088     check_wellknown_name("ntauthority\\Network Service", 0);
2089 }
2090
2091 static void test_security_descriptor(void)
2092 {
2093     SECURITY_DESCRIPTOR sd;
2094     char buf[8192];
2095     DWORD size;
2096     BOOL isDefault, isPresent, ret;
2097     PACL pacl;
2098     PSID psid;
2099
2100     SetLastError(0xdeadbeef);
2101     ret = InitializeSecurityDescriptor(&sd, SECURITY_DESCRIPTOR_REVISION);
2102     if (ret && GetLastError() == ERROR_CALL_NOT_IMPLEMENTED)
2103     {
2104         win_skip("InitializeSecurityDescriptor is not implemented\n");
2105         return;
2106     }
2107
2108     ok(GetSecurityDescriptorOwner(&sd, &psid, &isDefault), "GetSecurityDescriptorOwner failed\n");
2109     expect_eq(psid, NULL, PSID, "%p");
2110     expect_eq(isDefault, FALSE, BOOL, "%d");
2111     sd.Control |= SE_DACL_PRESENT | SE_SACL_PRESENT;
2112
2113     SetLastError(0xdeadbeef);
2114     size = 5;
2115     expect_eq(MakeSelfRelativeSD(&sd, buf, &size), FALSE, BOOL, "%d");
2116     expect_eq(GetLastError(), ERROR_INSUFFICIENT_BUFFER, DWORD, "%u");
2117     ok(size > 5, "Size not increased\n");
2118     if (size <= 8192)
2119     {
2120         expect_eq(MakeSelfRelativeSD(&sd, buf, &size), TRUE, BOOL, "%d");
2121         ok(GetSecurityDescriptorOwner(&sd, &psid, &isDefault), "GetSecurityDescriptorOwner failed\n");
2122         expect_eq(psid, NULL, PSID, "%p");
2123         expect_eq(isDefault, FALSE, BOOL, "%d");
2124         ok(GetSecurityDescriptorGroup(&sd, &psid, &isDefault), "GetSecurityDescriptorGroup failed\n");
2125         expect_eq(psid, NULL, PSID, "%p");
2126         expect_eq(isDefault, FALSE, BOOL, "%d");
2127         ok(GetSecurityDescriptorDacl(&sd, &isPresent, &pacl, &isDefault), "GetSecurityDescriptorDacl failed\n");
2128         expect_eq(isPresent, TRUE, BOOL, "%d");
2129         expect_eq(psid, NULL, PSID, "%p");
2130         expect_eq(isDefault, FALSE, BOOL, "%d");
2131         ok(GetSecurityDescriptorSacl(&sd, &isPresent, &pacl, &isDefault), "GetSecurityDescriptorSacl failed\n");
2132         expect_eq(isPresent, TRUE, BOOL, "%d");
2133         expect_eq(psid, NULL, PSID, "%p");
2134         expect_eq(isDefault, FALSE, BOOL, "%d");
2135     }
2136 }
2137
2138 #define TEST_GRANTED_ACCESS(a,b) test_granted_access(a,b,0,__LINE__)
2139 #define TEST_GRANTED_ACCESS2(a,b,c) test_granted_access(a,b,c,__LINE__)
2140 static void test_granted_access(HANDLE handle, ACCESS_MASK access,
2141                                 ACCESS_MASK alt, int line)
2142 {
2143     OBJECT_BASIC_INFORMATION obj_info;
2144     NTSTATUS status;
2145
2146     if (!pNtQueryObject)
2147     {
2148         skip_(__FILE__, line)("Not NT platform - skipping tests\n");
2149         return;
2150     }
2151
2152     status = pNtQueryObject( handle, ObjectBasicInformation, &obj_info,
2153                              sizeof(obj_info), NULL );
2154     ok_(__FILE__, line)(!status, "NtQueryObject with err: %08x\n", status);
2155     if (alt)
2156         ok_(__FILE__, line)(obj_info.GrantedAccess == access ||
2157             obj_info.GrantedAccess == alt, "Granted access should be 0x%08x "
2158             "or 0x%08x, instead of 0x%08x\n", access, alt, obj_info.GrantedAccess);
2159     else
2160         ok_(__FILE__, line)(obj_info.GrantedAccess == access, "Granted access should "
2161             "be 0x%08x, instead of 0x%08x\n", access, obj_info.GrantedAccess);
2162 }
2163
2164 #define CHECK_SET_SECURITY(o,i,e) \
2165     do{ \
2166         BOOL res; \
2167         DWORD err; \
2168         SetLastError( 0xdeadbeef ); \
2169         res = SetKernelObjectSecurity( o, i, SecurityDescriptor ); \
2170         err = GetLastError(); \
2171         if (e == ERROR_SUCCESS) \
2172             ok(res, "SetKernelObjectSecurity failed with %d\n", err); \
2173         else \
2174             ok(!res && err == e, "SetKernelObjectSecurity should have failed " \
2175                "with %s, instead of %d\n", #e, err); \
2176     }while(0)
2177
2178 static void test_process_security(void)
2179 {
2180     BOOL res;
2181     PTOKEN_OWNER owner;
2182     PTOKEN_PRIMARY_GROUP group;
2183     PSID AdminSid = NULL, UsersSid = NULL;
2184     PACL Acl = NULL;
2185     SECURITY_DESCRIPTOR *SecurityDescriptor = NULL;
2186     char buffer[MAX_PATH];
2187     PROCESS_INFORMATION info;
2188     STARTUPINFOA startup;
2189     SECURITY_ATTRIBUTES psa;
2190     HANDLE token, event;
2191     DWORD size;
2192
2193     Acl = HeapAlloc(GetProcessHeap(), 0, 256);
2194     res = InitializeAcl(Acl, 256, ACL_REVISION);
2195     if (!res && GetLastError() == ERROR_CALL_NOT_IMPLEMENTED)
2196     {
2197         win_skip("ACLs not implemented - skipping tests\n");
2198         HeapFree(GetProcessHeap(), 0, Acl);
2199         return;
2200     }
2201     ok(res, "InitializeAcl failed with error %d\n", GetLastError());
2202
2203     /* get owner from the token we might be running as a user not admin */
2204     res = OpenProcessToken( GetCurrentProcess(), MAXIMUM_ALLOWED, &token );
2205     ok(res, "OpenProcessToken failed with error %d\n", GetLastError());
2206     if (!res)
2207     {
2208         HeapFree(GetProcessHeap(), 0, Acl);
2209         return;
2210     }
2211
2212     res = GetTokenInformation( token, TokenOwner, NULL, 0, &size );
2213     ok(!res, "Expected failure, got %d\n", res);
2214     ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER,
2215        "Expected ERROR_INSUFFICIENT_BUFFER, got %d\n", GetLastError());
2216
2217     owner = HeapAlloc(GetProcessHeap(), 0, size);
2218     res = GetTokenInformation( token, TokenOwner, owner, size, &size );
2219     ok(res, "GetTokenInformation failed with error %d\n", GetLastError());
2220     AdminSid = ((TOKEN_OWNER*)owner)->Owner;
2221
2222     res = GetTokenInformation( token, TokenPrimaryGroup, NULL, 0, &size );
2223     ok(!res, "Expected failure, got %d\n", res);
2224     ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER,
2225        "Expected ERROR_INSUFFICIENT_BUFFER, got %d\n", GetLastError());
2226
2227     group = HeapAlloc(GetProcessHeap(), 0, size);
2228     res = GetTokenInformation( token, TokenPrimaryGroup, group, size, &size );
2229     ok(res, "GetTokenInformation failed with error %d\n", GetLastError());
2230     UsersSid = ((TOKEN_PRIMARY_GROUP*)group)->PrimaryGroup;
2231
2232     CloseHandle( token );
2233     if (!res)
2234     {
2235         HeapFree(GetProcessHeap(), 0, group);
2236         HeapFree(GetProcessHeap(), 0, owner);
2237         HeapFree(GetProcessHeap(), 0, Acl);
2238         return;
2239     }
2240
2241     res = AddAccessDeniedAce(Acl, ACL_REVISION, PROCESS_VM_READ, AdminSid);
2242     ok(res, "AddAccessDeniedAce failed with error %d\n", GetLastError());
2243     res = AddAccessAllowedAce(Acl, ACL_REVISION, PROCESS_ALL_ACCESS, AdminSid);
2244     ok(res, "AddAccessAllowedAce failed with error %d\n", GetLastError());
2245
2246     SecurityDescriptor = HeapAlloc(GetProcessHeap(), 0, SECURITY_DESCRIPTOR_MIN_LENGTH);
2247     res = InitializeSecurityDescriptor(SecurityDescriptor, SECURITY_DESCRIPTOR_REVISION);
2248     ok(res, "InitializeSecurityDescriptor failed with error %d\n", GetLastError());
2249
2250     event = CreateEvent( NULL, TRUE, TRUE, "test_event" );
2251     ok(event != NULL, "CreateEvent %d\n", GetLastError());
2252
2253     SecurityDescriptor->Revision = 0;
2254     CHECK_SET_SECURITY( event, OWNER_SECURITY_INFORMATION, ERROR_UNKNOWN_REVISION );
2255     SecurityDescriptor->Revision = SECURITY_DESCRIPTOR_REVISION;
2256
2257     CHECK_SET_SECURITY( event, OWNER_SECURITY_INFORMATION, ERROR_INVALID_SECURITY_DESCR );
2258     CHECK_SET_SECURITY( event, GROUP_SECURITY_INFORMATION, ERROR_INVALID_SECURITY_DESCR );
2259     CHECK_SET_SECURITY( event, SACL_SECURITY_INFORMATION, ERROR_ACCESS_DENIED );
2260     CHECK_SET_SECURITY( event, DACL_SECURITY_INFORMATION, ERROR_SUCCESS );
2261     /* NULL DACL is valid and means default DACL from token */
2262     SecurityDescriptor->Control |= SE_DACL_PRESENT;
2263     CHECK_SET_SECURITY( event, DACL_SECURITY_INFORMATION, ERROR_SUCCESS );
2264
2265     /* Set owner and group and dacl */
2266     res = SetSecurityDescriptorOwner(SecurityDescriptor, AdminSid, FALSE);
2267     ok(res, "SetSecurityDescriptorOwner failed with error %d\n", GetLastError());
2268     CHECK_SET_SECURITY( event, OWNER_SECURITY_INFORMATION, ERROR_SUCCESS );
2269     res = SetSecurityDescriptorGroup(SecurityDescriptor, UsersSid, FALSE);
2270     ok(res, "SetSecurityDescriptorGroup failed with error %d\n", GetLastError());
2271     CHECK_SET_SECURITY( event, GROUP_SECURITY_INFORMATION, ERROR_SUCCESS );
2272     res = SetSecurityDescriptorDacl(SecurityDescriptor, TRUE, Acl, FALSE);
2273     ok(res, "SetSecurityDescriptorDacl failed with error %d\n", GetLastError());
2274     CHECK_SET_SECURITY( event, DACL_SECURITY_INFORMATION, ERROR_SUCCESS );
2275
2276     sprintf(buffer, "%s tests/security.c test", myARGV[0]);
2277     memset(&startup, 0, sizeof(startup));
2278     startup.cb = sizeof(startup);
2279     startup.dwFlags = STARTF_USESHOWWINDOW;
2280     startup.wShowWindow = SW_SHOWNORMAL;
2281
2282     psa.nLength = sizeof(psa);
2283     psa.lpSecurityDescriptor = SecurityDescriptor;
2284     psa.bInheritHandle = TRUE;
2285
2286     /* Doesn't matter what ACL say we should get full access for ourselves */
2287     ok(CreateProcessA( NULL, buffer, &psa, NULL, FALSE, 0, NULL, NULL, &startup, &info ),
2288         "CreateProcess with err:%d\n", GetLastError());
2289     TEST_GRANTED_ACCESS2( info.hProcess, PROCESS_ALL_ACCESS,
2290                           STANDARD_RIGHTS_ALL | SPECIFIC_RIGHTS_ALL );
2291     winetest_wait_child_process( info.hProcess );
2292
2293     CloseHandle( info.hProcess );
2294     CloseHandle( info.hThread );
2295     CloseHandle( event );
2296     HeapFree(GetProcessHeap(), 0, group);
2297     HeapFree(GetProcessHeap(), 0, owner);
2298     HeapFree(GetProcessHeap(), 0, Acl);
2299     HeapFree(GetProcessHeap(), 0, SecurityDescriptor);
2300 }
2301
2302 static void test_process_security_child(void)
2303 {
2304     HANDLE handle, handle1;
2305     BOOL ret;
2306     DWORD err;
2307
2308     handle = OpenProcess( PROCESS_TERMINATE, FALSE, GetCurrentProcessId() );
2309     ok(handle != NULL, "OpenProcess(PROCESS_TERMINATE) with err:%d\n", GetLastError());
2310     TEST_GRANTED_ACCESS( handle, PROCESS_TERMINATE );
2311
2312     ok(DuplicateHandle( GetCurrentProcess(), handle, GetCurrentProcess(),
2313                         &handle1, 0, TRUE, DUPLICATE_SAME_ACCESS ),
2314        "duplicating handle err:%d\n", GetLastError());
2315     TEST_GRANTED_ACCESS( handle1, PROCESS_TERMINATE );
2316
2317     CloseHandle( handle1 );
2318
2319     SetLastError( 0xdeadbeef );
2320     ret = DuplicateHandle( GetCurrentProcess(), handle, GetCurrentProcess(),
2321                            &handle1, PROCESS_ALL_ACCESS, TRUE, 0 );
2322     err = GetLastError();
2323     todo_wine
2324     ok(!ret && err == ERROR_ACCESS_DENIED, "duplicating handle should have failed "
2325        "with STATUS_ACCESS_DENIED, instead of err:%d\n", err);
2326
2327     CloseHandle( handle );
2328
2329     /* These two should fail - they are denied by ACL */
2330     handle = OpenProcess( PROCESS_VM_READ, FALSE, GetCurrentProcessId() );
2331     todo_wine
2332     ok(handle == NULL, "OpenProcess(PROCESS_VM_READ) should have failed\n");
2333     handle = OpenProcess( PROCESS_ALL_ACCESS, FALSE, GetCurrentProcessId() );
2334     todo_wine
2335     ok(handle == NULL, "OpenProcess(PROCESS_ALL_ACCESS) should have failed\n");
2336
2337     /* Documented privilege elevation */
2338     ok(DuplicateHandle( GetCurrentProcess(), GetCurrentProcess(), GetCurrentProcess(),
2339                         &handle, 0, TRUE, DUPLICATE_SAME_ACCESS ),
2340        "duplicating handle err:%d\n", GetLastError());
2341     TEST_GRANTED_ACCESS2( handle, PROCESS_ALL_ACCESS,
2342                           STANDARD_RIGHTS_ALL | SPECIFIC_RIGHTS_ALL );
2343
2344     CloseHandle( handle );
2345
2346     /* Same only explicitly asking for all access rights */
2347     ok(DuplicateHandle( GetCurrentProcess(), GetCurrentProcess(), GetCurrentProcess(),
2348                         &handle, PROCESS_ALL_ACCESS, TRUE, 0 ),
2349        "duplicating handle err:%d\n", GetLastError());
2350     TEST_GRANTED_ACCESS2( handle, PROCESS_ALL_ACCESS,
2351                           PROCESS_ALL_ACCESS | PROCESS_QUERY_LIMITED_INFORMATION );
2352     ok(DuplicateHandle( GetCurrentProcess(), handle, GetCurrentProcess(),
2353                         &handle1, PROCESS_VM_READ, TRUE, 0 ),
2354        "duplicating handle err:%d\n", GetLastError());
2355     TEST_GRANTED_ACCESS( handle1, PROCESS_VM_READ );
2356     CloseHandle( handle1 );
2357     CloseHandle( handle );
2358 }
2359
2360 static void test_impersonation_level(void)
2361 {
2362     HANDLE Token, ProcessToken;
2363     HANDLE Token2;
2364     DWORD Size;
2365     TOKEN_PRIVILEGES *Privileges;
2366     TOKEN_USER *User;
2367     PRIVILEGE_SET *PrivilegeSet;
2368     BOOL AccessGranted;
2369     BOOL ret;
2370     HKEY hkey;
2371     DWORD error;
2372
2373     pDuplicateTokenEx = (fnDuplicateTokenEx) GetProcAddress(hmod, "DuplicateTokenEx");
2374     if( !pDuplicateTokenEx ) {
2375         win_skip("DuplicateTokenEx is not available\n");
2376         return;
2377     }
2378     SetLastError(0xdeadbeef);
2379     ret = ImpersonateSelf(SecurityAnonymous);
2380     if(!ret && (GetLastError() == ERROR_CALL_NOT_IMPLEMENTED))
2381     {
2382         win_skip("ImpersonateSelf is not implemented\n");
2383         return;
2384     }
2385     ok(ret, "ImpersonateSelf(SecurityAnonymous) failed with error %d\n", GetLastError());
2386     ret = OpenThreadToken(GetCurrentThread(), TOKEN_QUERY | TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY_SOURCE | TOKEN_IMPERSONATE | TOKEN_ADJUST_DEFAULT, TRUE, &Token);
2387     ok(!ret, "OpenThreadToken should have failed\n");
2388     error = GetLastError();
2389     ok(error == ERROR_CANT_OPEN_ANONYMOUS, "OpenThreadToken on anonymous token should have returned ERROR_CANT_OPEN_ANONYMOUS instead of %d\n", error);
2390     /* can't perform access check when opening object against an anonymous impersonation token */
2391     todo_wine {
2392     error = RegOpenKeyEx(HKEY_CURRENT_USER, "Software", 0, KEY_READ, &hkey);
2393     ok(error == ERROR_INVALID_HANDLE, "RegOpenKeyEx should have failed with ERROR_INVALID_HANDLE instead of %d\n", error);
2394     }
2395     RevertToSelf();
2396
2397     ret = OpenProcessToken(GetCurrentProcess(), TOKEN_DUPLICATE, &ProcessToken);
2398     ok(ret, "OpenProcessToken failed with error %d\n", GetLastError());
2399
2400     ret = pDuplicateTokenEx(ProcessToken,
2401         TOKEN_QUERY | TOKEN_DUPLICATE | TOKEN_IMPERSONATE, NULL,
2402         SecurityAnonymous, TokenImpersonation, &Token);
2403     ok(ret, "DuplicateTokenEx failed with error %d\n", GetLastError());
2404     /* can't increase the impersonation level */
2405     ret = DuplicateToken(Token, SecurityIdentification, &Token2);
2406     error = GetLastError();
2407     ok(!ret && error == ERROR_BAD_IMPERSONATION_LEVEL,
2408         "Duplicating a token and increasing the impersonation level should have failed with ERROR_BAD_IMPERSONATION_LEVEL instead of %d\n", error);
2409     /* we can query anything from an anonymous token, including the user */
2410     ret = GetTokenInformation(Token, TokenUser, NULL, 0, &Size);
2411     error = GetLastError();
2412     ok(!ret && error == ERROR_INSUFFICIENT_BUFFER, "GetTokenInformation(TokenUser) should have failed with ERROR_INSUFFICIENT_BUFFER instead of %d\n", error);
2413     User = HeapAlloc(GetProcessHeap(), 0, Size);
2414     ret = GetTokenInformation(Token, TokenUser, User, Size, &Size);
2415     ok(ret, "GetTokenInformation(TokenUser) failed with error %d\n", GetLastError());
2416     HeapFree(GetProcessHeap(), 0, User);
2417
2418     /* PrivilegeCheck fails with SecurityAnonymous level */
2419     ret = GetTokenInformation(Token, TokenPrivileges, NULL, 0, &Size);
2420     error = GetLastError();
2421     ok(!ret && error == ERROR_INSUFFICIENT_BUFFER, "GetTokenInformation(TokenPrivileges) should have failed with ERROR_INSUFFICIENT_BUFFER instead of %d\n", error);
2422     Privileges = HeapAlloc(GetProcessHeap(), 0, Size);
2423     ret = GetTokenInformation(Token, TokenPrivileges, Privileges, Size, &Size);
2424     ok(ret, "GetTokenInformation(TokenPrivileges) failed with error %d\n", GetLastError());
2425
2426     PrivilegeSet = HeapAlloc(GetProcessHeap(), 0, FIELD_OFFSET(PRIVILEGE_SET, Privilege[Privileges->PrivilegeCount]));
2427     PrivilegeSet->PrivilegeCount = Privileges->PrivilegeCount;
2428     memcpy(PrivilegeSet->Privilege, Privileges->Privileges, PrivilegeSet->PrivilegeCount * sizeof(PrivilegeSet->Privilege[0]));
2429     PrivilegeSet->Control = PRIVILEGE_SET_ALL_NECESSARY;
2430     HeapFree(GetProcessHeap(), 0, Privileges);
2431
2432     ret = PrivilegeCheck(Token, PrivilegeSet, &AccessGranted);
2433     error = GetLastError();
2434     ok(!ret && error == ERROR_BAD_IMPERSONATION_LEVEL, "PrivilegeCheck for SecurityAnonymous token should have failed with ERROR_BAD_IMPERSONATION_LEVEL instead of %d\n", error);
2435
2436     CloseHandle(Token);
2437
2438     ret = ImpersonateSelf(SecurityIdentification);
2439     ok(ret, "ImpersonateSelf(SecurityIdentification) failed with error %d\n", GetLastError());
2440     ret = OpenThreadToken(GetCurrentThread(), TOKEN_QUERY | TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY_SOURCE | TOKEN_IMPERSONATE | TOKEN_ADJUST_DEFAULT, TRUE, &Token);
2441     ok(ret, "OpenThreadToken failed with error %d\n", GetLastError());
2442
2443     /* can't perform access check when opening object against an identification impersonation token */
2444     error = RegOpenKeyEx(HKEY_CURRENT_USER, "Software", 0, KEY_READ, &hkey);
2445     todo_wine {
2446     ok(error == ERROR_INVALID_HANDLE, "RegOpenKeyEx should have failed with ERROR_INVALID_HANDLE instead of %d\n", error);
2447     }
2448     ret = PrivilegeCheck(Token, PrivilegeSet, &AccessGranted);
2449     ok(ret, "PrivilegeCheck for SecurityIdentification failed with error %d\n", GetLastError());
2450     CloseHandle(Token);
2451     RevertToSelf();
2452
2453     ret = ImpersonateSelf(SecurityImpersonation);
2454     ok(ret, "ImpersonateSelf(SecurityImpersonation) failed with error %d\n", GetLastError());
2455     ret = OpenThreadToken(GetCurrentThread(), TOKEN_QUERY | TOKEN_ADJUST_PRIVILEGES | TOKEN_QUERY_SOURCE | TOKEN_IMPERSONATE | TOKEN_ADJUST_DEFAULT, TRUE, &Token);
2456     ok(ret, "OpenThreadToken failed with error %d\n", GetLastError());
2457     error = RegOpenKeyEx(HKEY_CURRENT_USER, "Software", 0, KEY_READ, &hkey);
2458     ok(error == ERROR_SUCCESS, "RegOpenKeyEx should have succeeded instead of failing with %d\n", error);
2459     RegCloseKey(hkey);
2460     ret = PrivilegeCheck(Token, PrivilegeSet, &AccessGranted);
2461     ok(ret, "PrivilegeCheck for SecurityImpersonation failed with error %d\n", GetLastError());
2462     RevertToSelf();
2463
2464     CloseHandle(Token);
2465     CloseHandle(ProcessToken);
2466
2467     HeapFree(GetProcessHeap(), 0, PrivilegeSet);
2468 }
2469
2470 static void test_SetEntriesInAcl(void)
2471 {
2472     DWORD res;
2473     PSID EveryoneSid = NULL, UsersSid = NULL;
2474     PACL OldAcl = NULL, NewAcl;
2475     SID_IDENTIFIER_AUTHORITY SIDAuthWorld = { SECURITY_WORLD_SID_AUTHORITY };
2476     SID_IDENTIFIER_AUTHORITY SIDAuthNT = { SECURITY_NT_AUTHORITY };
2477     EXPLICIT_ACCESSW ExplicitAccess;
2478     static const WCHAR wszEveryone[] = {'E','v','e','r','y','o','n','e',0};
2479
2480     if (!pSetEntriesInAclW)
2481     {
2482         win_skip("SetEntriesInAclW is not available\n");
2483         return;
2484     }
2485
2486     NewAcl = (PACL)0xdeadbeef;
2487     res = pSetEntriesInAclW(0, NULL, NULL, &NewAcl);
2488     if(res == ERROR_CALL_NOT_IMPLEMENTED)
2489     {
2490         win_skip("SetEntriesInAclW is not implemented\n");
2491         return;
2492     }
2493     ok(res == ERROR_SUCCESS, "SetEntriesInAclW failed: %u\n", res);
2494     ok(NewAcl == NULL ||
2495         broken(NewAcl != NULL), /* NT4 */
2496         "NewAcl=%p, expected NULL\n", NewAcl);
2497     LocalFree(NewAcl);
2498
2499     OldAcl = HeapAlloc(GetProcessHeap(), 0, 256);
2500     res = InitializeAcl(OldAcl, 256, ACL_REVISION);
2501     if(!res && GetLastError() == ERROR_CALL_NOT_IMPLEMENTED)
2502     {
2503         win_skip("ACLs not implemented - skipping tests\n");
2504         HeapFree(GetProcessHeap(), 0, OldAcl);
2505         return;
2506     }
2507     ok(res, "InitializeAcl failed with error %d\n", GetLastError());
2508
2509     res = AllocateAndInitializeSid( &SIDAuthWorld, 1, SECURITY_WORLD_RID, 0, 0, 0, 0, 0, 0, 0, &EveryoneSid);
2510     ok(res, "AllocateAndInitializeSid failed with error %d\n", GetLastError());
2511
2512     res = AllocateAndInitializeSid( &SIDAuthNT, 2, SECURITY_BUILTIN_DOMAIN_RID,
2513         DOMAIN_ALIAS_RID_USERS, 0, 0, 0, 0, 0, 0, &UsersSid);
2514     ok(res, "AllocateAndInitializeSid failed with error %d\n", GetLastError());
2515
2516     res = AddAccessAllowedAce(OldAcl, ACL_REVISION, KEY_READ, UsersSid);
2517     ok(res, "AddAccessAllowedAce failed with error %d\n", GetLastError());
2518
2519     ExplicitAccess.grfAccessPermissions = KEY_WRITE;
2520     ExplicitAccess.grfAccessMode = GRANT_ACCESS;
2521     ExplicitAccess.grfInheritance = NO_INHERITANCE;
2522     ExplicitAccess.Trustee.pMultipleTrustee = NULL;
2523     ExplicitAccess.Trustee.MultipleTrusteeOperation = NO_MULTIPLE_TRUSTEE;
2524     ExplicitAccess.Trustee.TrusteeForm = TRUSTEE_IS_SID;
2525     ExplicitAccess.Trustee.TrusteeType = TRUSTEE_IS_UNKNOWN;
2526     ExplicitAccess.Trustee.ptstrName = EveryoneSid;
2527     res = pSetEntriesInAclW(1, &ExplicitAccess, OldAcl, &NewAcl);
2528     ok(res == ERROR_SUCCESS, "SetEntriesInAclW failed: %u\n", res);
2529     ok(NewAcl != NULL, "returned acl was NULL\n");
2530     LocalFree(NewAcl);
2531
2532     if (PRIMARYLANGID(LANGIDFROMLCID(GetThreadLocale())) != LANG_ENGLISH)
2533     {
2534         skip("Non-english locale (test with hardcoded 'Everyone')\n");
2535     }
2536     else
2537     {
2538         ExplicitAccess.Trustee.TrusteeForm = TRUSTEE_IS_USER;
2539         ExplicitAccess.Trustee.ptstrName = (LPWSTR)wszEveryone;
2540         res = pSetEntriesInAclW(1, &ExplicitAccess, OldAcl, &NewAcl);
2541         ok(res == ERROR_SUCCESS, "SetEntriesInAclW failed: %u\n", res);
2542         ok(NewAcl != NULL, "returned acl was NULL\n");
2543         LocalFree(NewAcl);
2544
2545         ExplicitAccess.Trustee.TrusteeForm = TRUSTEE_BAD_FORM;
2546         res = pSetEntriesInAclW(1, &ExplicitAccess, OldAcl, &NewAcl);
2547         ok(res == ERROR_INVALID_PARAMETER ||
2548             broken(res == ERROR_NOT_SUPPORTED), /* NT4 */
2549             "SetEntriesInAclW failed: %u\n", res);
2550         ok(NewAcl == NULL ||
2551             broken(NewAcl != NULL), /* NT4 */
2552             "returned acl wasn't NULL: %p\n", NewAcl);
2553
2554         ExplicitAccess.Trustee.TrusteeForm = TRUSTEE_IS_USER;
2555         ExplicitAccess.Trustee.MultipleTrusteeOperation = TRUSTEE_IS_IMPERSONATE;
2556         res = pSetEntriesInAclW(1, &ExplicitAccess, OldAcl, &NewAcl);
2557         ok(res == ERROR_INVALID_PARAMETER ||
2558             broken(res == ERROR_NOT_SUPPORTED), /* NT4 */
2559             "SetEntriesInAclW failed: %u\n", res);
2560         ok(NewAcl == NULL ||
2561             broken(NewAcl != NULL), /* NT4 */
2562             "returned acl wasn't NULL: %p\n", NewAcl);
2563
2564         ExplicitAccess.Trustee.MultipleTrusteeOperation = NO_MULTIPLE_TRUSTEE;
2565         ExplicitAccess.grfAccessMode = SET_ACCESS;
2566         res = pSetEntriesInAclW(1, &ExplicitAccess, OldAcl, &NewAcl);
2567         ok(res == ERROR_SUCCESS, "SetEntriesInAclW failed: %u\n", res);
2568         ok(NewAcl != NULL, "returned acl was NULL\n");
2569         LocalFree(NewAcl);
2570     }
2571
2572     ExplicitAccess.grfAccessMode = REVOKE_ACCESS;
2573     ExplicitAccess.Trustee.TrusteeForm = TRUSTEE_IS_SID;
2574     ExplicitAccess.Trustee.ptstrName = UsersSid;
2575     res = pSetEntriesInAclW(1, &ExplicitAccess, OldAcl, &NewAcl);
2576     ok(res == ERROR_SUCCESS, "SetEntriesInAclW failed: %u\n", res);
2577     ok(NewAcl != NULL, "returned acl was NULL\n");
2578     LocalFree(NewAcl);
2579
2580     LocalFree(UsersSid);
2581     LocalFree(EveryoneSid);
2582     HeapFree(GetProcessHeap(), 0, OldAcl);
2583 }
2584
2585 static void test_GetNamedSecurityInfoA(void)
2586 {
2587     PSECURITY_DESCRIPTOR pSecDesc;
2588     DWORD revision;
2589     SECURITY_DESCRIPTOR_CONTROL control;
2590     PSID owner;
2591     PSID group;
2592     BOOL owner_defaulted;
2593     BOOL group_defaulted;
2594     DWORD error;
2595     BOOL ret;
2596     CHAR windows_dir[MAX_PATH];
2597
2598     if (!pGetNamedSecurityInfoA)
2599     {
2600         win_skip("GetNamedSecurityInfoA is not available\n");
2601         return;
2602     }
2603
2604     ret = GetWindowsDirectoryA(windows_dir, MAX_PATH);
2605     ok(ret, "GetWindowsDirectory failed with error %d\n", GetLastError());
2606
2607     SetLastError(0xdeadbeef);
2608     error = pGetNamedSecurityInfoA(windows_dir, SE_FILE_OBJECT,
2609         OWNER_SECURITY_INFORMATION|GROUP_SECURITY_INFORMATION|DACL_SECURITY_INFORMATION,
2610         NULL, NULL, NULL, NULL, &pSecDesc);
2611     if (error != ERROR_SUCCESS && (GetLastError() == ERROR_CALL_NOT_IMPLEMENTED))
2612     {
2613         win_skip("GetNamedSecurityInfoA is not implemented\n");
2614         return;
2615     }
2616     ok(!error, "GetNamedSecurityInfo failed with error %d\n", error);
2617
2618     ret = GetSecurityDescriptorControl(pSecDesc, &control, &revision);
2619     ok(ret, "GetSecurityDescriptorControl failed with error %d\n", GetLastError());
2620     ok((control & (SE_SELF_RELATIVE|SE_DACL_PRESENT)) == (SE_SELF_RELATIVE|SE_DACL_PRESENT) ||
2621         broken((control & (SE_SELF_RELATIVE|SE_DACL_PRESENT)) == SE_DACL_PRESENT), /* NT4 */
2622         "control (0x%x) doesn't have (SE_SELF_RELATIVE|SE_DACL_PRESENT) flags set\n", control);
2623     ok(revision == SECURITY_DESCRIPTOR_REVISION1, "revision was %d instead of 1\n", revision);
2624     ret = GetSecurityDescriptorOwner(pSecDesc, &owner, &owner_defaulted);
2625     ok(ret, "GetSecurityDescriptorOwner failed with error %d\n", GetLastError());
2626     ok(owner != NULL, "owner should not be NULL\n");
2627     ret = GetSecurityDescriptorGroup(pSecDesc, &group, &group_defaulted);
2628     ok(ret, "GetSecurityDescriptorGroup failed with error %d\n", GetLastError());
2629     ok(group != NULL, "group should not be NULL\n");
2630     LocalFree(pSecDesc);
2631 }
2632
2633 static void test_ConvertStringSecurityDescriptor(void)
2634 {
2635     BOOL ret;
2636     PSECURITY_DESCRIPTOR pSD;
2637     static const WCHAR Blank[] = { 0 };
2638
2639     if (!pConvertStringSecurityDescriptorToSecurityDescriptorA)
2640     {
2641         win_skip("ConvertStringSecurityDescriptorToSecurityDescriptor is not available\n");
2642         return;
2643     }
2644
2645     SetLastError(0xdeadbeef);
2646     ret = pConvertStringSecurityDescriptorToSecurityDescriptorA(
2647         "D:(A;;GA;;;WD)", 0xdeadbeef, &pSD, NULL);
2648     ok(!ret && GetLastError() == ERROR_UNKNOWN_REVISION,
2649         "ConvertStringSecurityDescriptorToSecurityDescriptor should have failed with ERROR_UNKNOWN_REVISION instead of %d\n",
2650         GetLastError());
2651
2652     /* test ACE string type */
2653     SetLastError(0xdeadbeef);
2654     ret = pConvertStringSecurityDescriptorToSecurityDescriptorA(
2655         "D:(A;;GA;;;WD)", SDDL_REVISION_1, &pSD, NULL);
2656     ok(ret, "ConvertStringSecurityDescriptorToSecurityDescriptor failed with error %d\n", GetLastError());
2657     LocalFree(pSD);
2658
2659     SetLastError(0xdeadbeef);
2660     ret = pConvertStringSecurityDescriptorToSecurityDescriptorA(
2661         "D:(D;;GA;;;WD)", SDDL_REVISION_1, &pSD, NULL);
2662     ok(ret, "ConvertStringSecurityDescriptorToSecurityDescriptor failed with error %d\n", GetLastError());
2663     LocalFree(pSD);
2664
2665     SetLastError(0xdeadbeef);
2666     ret = pConvertStringSecurityDescriptorToSecurityDescriptorA(
2667         "ERROR:(D;;GA;;;WD)", SDDL_REVISION_1, &pSD, NULL);
2668     ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
2669         "ConvertStringSecurityDescriptorToSecurityDescriptor should have failed with ERROR_INVALID_PARAMETER instead of %d\n",
2670         GetLastError());
2671
2672     /* test ACE string access rights */
2673     SetLastError(0xdeadbeef);
2674     ret = pConvertStringSecurityDescriptorToSecurityDescriptorA(
2675         "D:(A;;GA;;;WD)", SDDL_REVISION_1, &pSD, NULL);
2676     ok(ret, "ConvertStringSecurityDescriptorToSecurityDescriptor failed with error %d\n", GetLastError());
2677     LocalFree(pSD);
2678     SetLastError(0xdeadbeef);
2679     ret = pConvertStringSecurityDescriptorToSecurityDescriptorA(
2680         "D:(A;;GRGWGX;;;WD)", SDDL_REVISION_1, &pSD, NULL);
2681     ok(ret, "ConvertStringSecurityDescriptorToSecurityDescriptor failed with error %d\n", GetLastError());
2682     LocalFree(pSD);
2683     SetLastError(0xdeadbeef);
2684     ret = pConvertStringSecurityDescriptorToSecurityDescriptorA(
2685         "D:(A;;RCSDWDWO;;;WD)", SDDL_REVISION_1, &pSD, NULL);
2686     ok(ret, "ConvertStringSecurityDescriptorToSecurityDescriptor failed with error %d\n", GetLastError());
2687     LocalFree(pSD);
2688     SetLastError(0xdeadbeef);
2689     ret = pConvertStringSecurityDescriptorToSecurityDescriptorA(
2690         "D:(A;;RPWPCCDCLCSWLODTCR;;;WD)", SDDL_REVISION_1, &pSD, NULL);
2691     ok(ret, "ConvertStringSecurityDescriptorToSecurityDescriptor failed with error %d\n", GetLastError());
2692     LocalFree(pSD);
2693     SetLastError(0xdeadbeef);
2694     ret = pConvertStringSecurityDescriptorToSecurityDescriptorA(
2695         "D:(A;;FAFRFWFX;;;WD)", SDDL_REVISION_1, &pSD, NULL);
2696     ok(ret, "ConvertStringSecurityDescriptorToSecurityDescriptor failed with error %d\n", GetLastError());
2697     LocalFree(pSD);
2698     SetLastError(0xdeadbeef);
2699     ret = pConvertStringSecurityDescriptorToSecurityDescriptorA(
2700         "D:(A;;KAKRKWKX;;;WD)", SDDL_REVISION_1, &pSD, NULL);
2701     ok(ret, "ConvertStringSecurityDescriptorToSecurityDescriptor failed with error %d\n", GetLastError());
2702     LocalFree(pSD);
2703     SetLastError(0xdeadbeef);
2704     ret = pConvertStringSecurityDescriptorToSecurityDescriptorA(
2705         "D:(A;;0xFFFFFFFF;;;WD)", SDDL_REVISION_1, &pSD, NULL);
2706     ok(ret, "ConvertStringSecurityDescriptorToSecurityDescriptor failed with error %d\n", GetLastError());
2707     LocalFree(pSD);
2708     SetLastError(0xdeadbeef);
2709     ret = pConvertStringSecurityDescriptorToSecurityDescriptorA(
2710         "S:(AU;;0xFFFFFFFF;;;WD)", SDDL_REVISION_1, &pSD, NULL);
2711     ok(ret, "ConvertStringSecurityDescriptorToSecurityDescriptor failed with error %d\n", GetLastError());
2712     LocalFree(pSD);
2713
2714     /* test ACE string access right error case */
2715     SetLastError(0xdeadbeef);
2716     ret = pConvertStringSecurityDescriptorToSecurityDescriptorA(
2717         "D:(A;;ROB;;;WD)", SDDL_REVISION_1, &pSD, NULL);
2718     ok(!ret && GetLastError() == ERROR_INVALID_ACL,
2719         "ConvertStringSecurityDescriptorToSecurityDescriptor should have failed with ERROR_INVALID_ACL instead of %d\n",
2720         GetLastError());
2721
2722     /* test behaviour with NULL parameters */
2723     SetLastError(0xdeadbeef);
2724     ret = pConvertStringSecurityDescriptorToSecurityDescriptorA(
2725         NULL, 0xdeadbeef, &pSD, NULL);
2726     todo_wine
2727     ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
2728         "ConvertStringSecurityDescriptorToSecurityDescriptor should have failed with ERROR_INVALID_PARAMETER instead of %d\n",
2729         GetLastError());
2730
2731     SetLastError(0xdeadbeef);
2732     ret = pConvertStringSecurityDescriptorToSecurityDescriptorW(
2733         NULL, 0xdeadbeef, &pSD, NULL);
2734     ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
2735         "ConvertStringSecurityDescriptorToSecurityDescriptor should have failed with ERROR_INVALID_PARAMETER instead of %d\n",
2736         GetLastError());
2737
2738     SetLastError(0xdeadbeef);
2739     ret = pConvertStringSecurityDescriptorToSecurityDescriptorA(
2740         "D:(A;;ROB;;;WD)", 0xdeadbeef, NULL, NULL);
2741     ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
2742         "ConvertStringSecurityDescriptorToSecurityDescriptor should have failed with ERROR_INVALID_PARAMETER instead of %d\n",
2743         GetLastError());
2744
2745     SetLastError(0xdeadbeef);
2746     ret = pConvertStringSecurityDescriptorToSecurityDescriptorA(
2747         "D:(A;;ROB;;;WD)", SDDL_REVISION_1, NULL, NULL);
2748     ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
2749         "ConvertStringSecurityDescriptorToSecurityDescriptor should have failed with ERROR_INVALID_PARAMETER instead of %d\n",
2750         GetLastError());
2751
2752     /* test behaviour with empty strings */
2753     SetLastError(0xdeadbeef);
2754     ret = pConvertStringSecurityDescriptorToSecurityDescriptorA(
2755         "", SDDL_REVISION_1, &pSD, NULL);
2756     ok(ret, "ConvertStringSecurityDescriptorToSecurityDescriptor failed with error %d\n", GetLastError());
2757
2758     SetLastError(0xdeadbeef);
2759     ret = pConvertStringSecurityDescriptorToSecurityDescriptorW(
2760         Blank, SDDL_REVISION_1, &pSD, NULL);
2761     ok(ret, "ConvertStringSecurityDescriptorToSecurityDescriptor failed with error %d\n", GetLastError());
2762
2763     /* test ACE string SID */
2764     SetLastError(0xdeadbeef);
2765     ret = pConvertStringSecurityDescriptorToSecurityDescriptorA(
2766         "D:(D;;GA;;;S-1-0-0)", SDDL_REVISION_1, &pSD, NULL);
2767     ok(ret, "ConvertStringSecurityDescriptorToSecurityDescriptor failed with error %d\n", GetLastError());
2768     LocalFree(pSD);
2769
2770     SetLastError(0xdeadbeef);
2771     ret = pConvertStringSecurityDescriptorToSecurityDescriptorA(
2772         "D:(D;;GA;;;Nonexistent account)", SDDL_REVISION_1, &pSD, NULL);
2773     ok(!ret, "Expected failure, got %d\n", ret);
2774     ok(GetLastError() == ERROR_INVALID_ACL || GetLastError() == ERROR_INVALID_SID,
2775        "Expected ERROR_INVALID_ACL or ERROR_INVALID_SID, got %d\n", GetLastError());
2776 }
2777
2778 static void test_ConvertSecurityDescriptorToString(void)
2779 {
2780     SECURITY_DESCRIPTOR desc;
2781     SECURITY_INFORMATION sec_info = OWNER_SECURITY_INFORMATION|GROUP_SECURITY_INFORMATION|DACL_SECURITY_INFORMATION|SACL_SECURITY_INFORMATION;
2782     LPSTR string;
2783     DWORD size;
2784     PSID psid, psid2;
2785     PACL pacl;
2786     char sid_buf[256];
2787     char acl_buf[8192];
2788     ULONG len;
2789
2790     if (!pConvertSecurityDescriptorToStringSecurityDescriptorA)
2791     {
2792         win_skip("ConvertSecurityDescriptorToStringSecurityDescriptor is not available\n");
2793         return;
2794     }
2795     if (!pCreateWellKnownSid)
2796     {
2797         win_skip("CreateWellKnownSid is not available\n");
2798         return;
2799     }
2800
2801 /* It seems Windows XP adds an extra character to the length of the string for each ACE in an ACL. We
2802  * don't replicate this feature so we only test len >= strlen+1. */
2803 #define CHECK_RESULT_AND_FREE(exp_str) \
2804     ok(strcmp(string, (exp_str)) == 0, "String mismatch (expected \"%s\", got \"%s\")\n", (exp_str), string); \
2805     ok(len >= (lstrlen(exp_str) + 1), "Length mismatch (expected %d, got %d)\n", lstrlen(exp_str) + 1, len); \
2806     LocalFree(string);
2807
2808 #define CHECK_ONE_OF_AND_FREE(exp_str1, exp_str2) \
2809     ok(strcmp(string, (exp_str1)) == 0 || strcmp(string, (exp_str2)) == 0, "String mismatch (expected\n\"%s\" or\n\"%s\", got\n\"%s\")\n", (exp_str1), (exp_str2), string); \
2810     ok(len >= (strlen(string) + 1), "Length mismatch (expected %d, got %d)\n", lstrlen(string) + 1, len); \
2811     LocalFree(string);
2812
2813     InitializeSecurityDescriptor(&desc, SECURITY_DESCRIPTOR_REVISION);
2814     ok(pConvertSecurityDescriptorToStringSecurityDescriptorA(&desc, SDDL_REVISION_1, sec_info, &string, &len), "Conversion failed\n");
2815     CHECK_RESULT_AND_FREE("");
2816
2817     size = 4096;
2818     pCreateWellKnownSid(WinLocalSid, NULL, sid_buf, &size);
2819     SetSecurityDescriptorOwner(&desc, sid_buf, FALSE);
2820     ok(pConvertSecurityDescriptorToStringSecurityDescriptorA(&desc, SDDL_REVISION_1, sec_info, &string, &len), "Conversion failed\n");
2821     CHECK_RESULT_AND_FREE("O:S-1-2-0");
2822
2823     SetSecurityDescriptorOwner(&desc, sid_buf, TRUE);
2824     ok(pConvertSecurityDescriptorToStringSecurityDescriptorA(&desc, SDDL_REVISION_1, sec_info, &string, &len), "Conversion failed\n");
2825     CHECK_RESULT_AND_FREE("O:S-1-2-0");
2826
2827     size = sizeof(sid_buf);
2828     pCreateWellKnownSid(WinLocalSystemSid, NULL, sid_buf, &size);
2829     SetSecurityDescriptorOwner(&desc, sid_buf, TRUE);
2830     ok(pConvertSecurityDescriptorToStringSecurityDescriptorA(&desc, SDDL_REVISION_1, sec_info, &string, &len), "Conversion failed\n");
2831     CHECK_RESULT_AND_FREE("O:SY");
2832
2833     pConvertStringSidToSidA("S-1-5-21-93476-23408-4576", &psid);
2834     SetSecurityDescriptorGroup(&desc, psid, TRUE);
2835     ok(pConvertSecurityDescriptorToStringSecurityDescriptorA(&desc, SDDL_REVISION_1, sec_info, &string, &len), "Conversion failed\n");
2836     CHECK_RESULT_AND_FREE("O:SYG:S-1-5-21-93476-23408-4576");
2837
2838     ok(pConvertSecurityDescriptorToStringSecurityDescriptorA(&desc, SDDL_REVISION_1, GROUP_SECURITY_INFORMATION, &string, &len), "Conversion failed\n");
2839     CHECK_RESULT_AND_FREE("G:S-1-5-21-93476-23408-4576");
2840
2841     pacl = (PACL)acl_buf;
2842     InitializeAcl(pacl, sizeof(acl_buf), ACL_REVISION);
2843     SetSecurityDescriptorDacl(&desc, TRUE, pacl, TRUE);
2844     ok(pConvertSecurityDescriptorToStringSecurityDescriptorA(&desc, SDDL_REVISION_1, sec_info, &string, &len), "Conversion failed\n");
2845     CHECK_RESULT_AND_FREE("O:SYG:S-1-5-21-93476-23408-4576D:");
2846
2847     SetSecurityDescriptorDacl(&desc, TRUE, pacl, FALSE);
2848     ok(pConvertSecurityDescriptorToStringSecurityDescriptorA(&desc, SDDL_REVISION_1, sec_info, &string, &len), "Conversion failed\n");
2849     CHECK_RESULT_AND_FREE("O:SYG:S-1-5-21-93476-23408-4576D:");
2850
2851     pConvertStringSidToSidA("S-1-5-6", &psid2);
2852     pAddAccessAllowedAceEx(pacl, ACL_REVISION, NO_PROPAGATE_INHERIT_ACE, 0xf0000000, psid2);
2853     ok(pConvertSecurityDescriptorToStringSecurityDescriptorA(&desc, SDDL_REVISION_1, sec_info, &string, &len), "Conversion failed\n");
2854     CHECK_RESULT_AND_FREE("O:SYG:S-1-5-21-93476-23408-4576D:(A;NP;GAGXGWGR;;;SU)");
2855
2856     pAddAccessAllowedAceEx(pacl, ACL_REVISION, INHERIT_ONLY_ACE|INHERITED_ACE, 0x00000003, psid2);
2857     ok(pConvertSecurityDescriptorToStringSecurityDescriptorA(&desc, SDDL_REVISION_1, sec_info, &string, &len), "Conversion failed\n");
2858     CHECK_RESULT_AND_FREE("O:SYG:S-1-5-21-93476-23408-4576D:(A;NP;GAGXGWGR;;;SU)(A;IOID;CCDC;;;SU)");
2859
2860     pAddAccessDeniedAceEx(pacl, ACL_REVISION, OBJECT_INHERIT_ACE|CONTAINER_INHERIT_ACE, 0xffffffff, psid);
2861     ok(pConvertSecurityDescriptorToStringSecurityDescriptorA(&desc, SDDL_REVISION_1, sec_info, &string, &len), "Conversion failed\n");
2862     CHECK_RESULT_AND_FREE("O:SYG:S-1-5-21-93476-23408-4576D:(A;NP;GAGXGWGR;;;SU)(A;IOID;CCDC;;;SU)(D;OICI;0xffffffff;;;S-1-5-21-93476-23408-4576)");
2863
2864
2865     pacl = (PACL)acl_buf;
2866     InitializeAcl(pacl, sizeof(acl_buf), ACL_REVISION);
2867     SetSecurityDescriptorSacl(&desc, TRUE, pacl, FALSE);
2868     ok(pConvertSecurityDescriptorToStringSecurityDescriptorA(&desc, SDDL_REVISION_1, sec_info, &string, &len), "Conversion failed\n");
2869     CHECK_RESULT_AND_FREE("O:SYG:S-1-5-21-93476-23408-4576D:S:");
2870
2871     /* fails in win2k */
2872     SetSecurityDescriptorDacl(&desc, TRUE, NULL, FALSE);
2873     pAddAuditAccessAceEx(pacl, ACL_REVISION, VALID_INHERIT_FLAGS, KEY_READ|KEY_WRITE, psid2, TRUE, TRUE);
2874     if (pConvertSecurityDescriptorToStringSecurityDescriptorA(&desc, SDDL_REVISION_1, sec_info, &string, &len))
2875     {
2876         CHECK_ONE_OF_AND_FREE("O:SYG:S-1-5-21-93476-23408-4576D:S:(AU;OICINPIOIDSAFA;CCDCLCSWRPRC;;;SU)", /* XP */
2877             "O:SYG:S-1-5-21-93476-23408-4576D:NO_ACCESS_CONTROLS:(AU;OICINPIOIDSAFA;CCDCLCSWRPRC;;;SU)" /* Vista */);
2878     }
2879
2880     /* fails in win2k */
2881     pAddAuditAccessAceEx(pacl, ACL_REVISION, NO_PROPAGATE_INHERIT_ACE, FILE_GENERIC_READ|FILE_GENERIC_WRITE, psid2, TRUE, FALSE);
2882     if (pConvertSecurityDescriptorToStringSecurityDescriptorA(&desc, SDDL_REVISION_1, sec_info, &string, &len))
2883     {
2884         CHECK_ONE_OF_AND_FREE("O:SYG:S-1-5-21-93476-23408-4576D:S:(AU;OICINPIOIDSAFA;CCDCLCSWRPRC;;;SU)(AU;NPSA;0x12019f;;;SU)", /* XP */
2885             "O:SYG:S-1-5-21-93476-23408-4576D:NO_ACCESS_CONTROLS:(AU;OICINPIOIDSAFA;CCDCLCSWRPRC;;;SU)(AU;NPSA;0x12019f;;;SU)" /* Vista */);
2886     }
2887 }
2888
2889 static void test_SetSecurityDescriptorControl (PSECURITY_DESCRIPTOR sec)
2890 {
2891     SECURITY_DESCRIPTOR_CONTROL ref;
2892     SECURITY_DESCRIPTOR_CONTROL test;
2893
2894     SECURITY_DESCRIPTOR_CONTROL const mutable
2895         = SE_DACL_AUTO_INHERIT_REQ | SE_SACL_AUTO_INHERIT_REQ
2896         | SE_DACL_AUTO_INHERITED   | SE_SACL_AUTO_INHERITED
2897         | SE_DACL_PROTECTED        | SE_SACL_PROTECTED
2898         | 0x00000040               | 0x00000080        /* not defined in winnt.h */
2899         ;
2900     SECURITY_DESCRIPTOR_CONTROL const immutable
2901         = SE_OWNER_DEFAULTED       | SE_GROUP_DEFAULTED
2902         | SE_DACL_PRESENT          | SE_DACL_DEFAULTED
2903         | SE_SACL_PRESENT          | SE_SACL_DEFAULTED
2904         | SE_RM_CONTROL_VALID      | SE_SELF_RELATIVE
2905         ;
2906
2907     int     bit;
2908     DWORD   dwRevision;
2909     LPCSTR  fmt = "Expected error %s, got %u\n";
2910
2911     GetSecurityDescriptorControl (sec, &ref, &dwRevision);
2912
2913     /* The mutable bits are mutable regardless of the truth of
2914        SE_DACL_PRESENT and/or SE_SACL_PRESENT */
2915
2916     /* Check call barfs if any bit-of-interest is immutable */
2917     for (bit = 0; bit < 16; ++bit)
2918     {
2919         SECURITY_DESCRIPTOR_CONTROL const bitOfInterest = 1 << bit;
2920         SECURITY_DESCRIPTOR_CONTROL setOrClear = ref & bitOfInterest;
2921
2922         SECURITY_DESCRIPTOR_CONTROL ctrl;
2923
2924         DWORD   dwExpect  = (bitOfInterest & immutable)
2925                           ?  ERROR_INVALID_PARAMETER  :  0xbebecaca;
2926         LPCSTR  strExpect = (bitOfInterest & immutable)
2927                           ? "ERROR_INVALID_PARAMETER" : "0xbebecaca";
2928
2929         ctrl = (bitOfInterest & mutable) ? ref + bitOfInterest : ref;
2930         setOrClear ^= bitOfInterest;
2931         SetLastError (0xbebecaca);
2932         pSetSecurityDescriptorControl (sec, bitOfInterest, setOrClear);
2933         ok (GetLastError () == dwExpect, fmt, strExpect, GetLastError ());
2934         GetSecurityDescriptorControl(sec, &test, &dwRevision);
2935         expect_eq(test, ctrl, int, "%x");
2936
2937         ctrl = ref;
2938         setOrClear ^= bitOfInterest;
2939         SetLastError (0xbebecaca);
2940         pSetSecurityDescriptorControl (sec, bitOfInterest, setOrClear);
2941         ok (GetLastError () == dwExpect, fmt, strExpect, GetLastError ());
2942         GetSecurityDescriptorControl (sec, &test, &dwRevision);
2943         expect_eq(test, ref, int, "%x");
2944     }
2945
2946     /* Check call barfs if any bit-to-set is immutable
2947        even when not a bit-of-interest */
2948     for (bit = 0; bit < 16; ++bit)
2949     {
2950         SECURITY_DESCRIPTOR_CONTROL const bitsOfInterest = mutable;
2951         SECURITY_DESCRIPTOR_CONTROL setOrClear = ref & bitsOfInterest;
2952
2953         SECURITY_DESCRIPTOR_CONTROL ctrl;
2954
2955         DWORD   dwExpect  = ((1 << bit) & immutable)
2956                           ?  ERROR_INVALID_PARAMETER  :  0xbebecaca;
2957         LPCSTR  strExpect = ((1 << bit) & immutable)
2958                           ? "ERROR_INVALID_PARAMETER" : "0xbebecaca";
2959
2960         ctrl = ((1 << bit) & immutable) ? test : ref | mutable;
2961         setOrClear ^= bitsOfInterest;
2962         SetLastError (0xbebecaca);
2963         pSetSecurityDescriptorControl (sec, bitsOfInterest, setOrClear | (1 << bit));
2964         ok (GetLastError () == dwExpect, fmt, strExpect, GetLastError ());
2965         GetSecurityDescriptorControl(sec, &test, &dwRevision);
2966         expect_eq(test, ctrl, int, "%x");
2967
2968         ctrl = ((1 << bit) & immutable) ? test : ref | (1 << bit);
2969         setOrClear ^= bitsOfInterest;
2970         SetLastError (0xbebecaca);
2971         pSetSecurityDescriptorControl (sec, bitsOfInterest, setOrClear | (1 << bit));
2972         ok (GetLastError () == dwExpect, fmt, strExpect, GetLastError ());
2973         GetSecurityDescriptorControl(sec, &test, &dwRevision);
2974         expect_eq(test, ctrl, int, "%x");
2975     }
2976 }
2977
2978 static void test_PrivateObjectSecurity(void)
2979 {
2980     SECURITY_INFORMATION sec_info = OWNER_SECURITY_INFORMATION|GROUP_SECURITY_INFORMATION|DACL_SECURITY_INFORMATION|SACL_SECURITY_INFORMATION;
2981     SECURITY_DESCRIPTOR_CONTROL ctrl;
2982     PSECURITY_DESCRIPTOR sec;
2983     DWORD dwDescSize;
2984     DWORD dwRevision;
2985     DWORD retSize;
2986     LPSTR string;
2987     ULONG len;
2988     PSECURITY_DESCRIPTOR buf;
2989
2990     if (!pConvertStringSecurityDescriptorToSecurityDescriptorA)
2991     {
2992         win_skip("ConvertStringSecurityDescriptorToSecurityDescriptor is not available\n");
2993         return;
2994     }
2995
2996     ok(pConvertStringSecurityDescriptorToSecurityDescriptorA(
2997         "O:SY"
2998         "G:S-1-5-21-93476-23408-4576"
2999         "D:(A;NP;GAGXGWGR;;;SU)(A;IOID;CCDC;;;SU)"
3000           "(D;OICI;0xffffffff;;;S-1-5-21-93476-23408-4576)"
3001         "S:(AU;OICINPIOIDSAFA;CCDCLCSWRPRC;;;SU)(AU;NPSA;0x12019f;;;SU)",
3002         SDDL_REVISION_1, &sec, &dwDescSize), "Creating descriptor failed\n");
3003
3004     test_SetSecurityDescriptorControl(sec);
3005
3006     LocalFree(sec);
3007
3008     ok(pConvertStringSecurityDescriptorToSecurityDescriptorA(
3009         "O:SY"
3010         "G:S-1-5-21-93476-23408-4576",
3011         SDDL_REVISION_1, &sec, &dwDescSize), "Creating descriptor failed\n");
3012
3013     test_SetSecurityDescriptorControl(sec);
3014
3015     LocalFree(sec);
3016
3017     ok(pConvertStringSecurityDescriptorToSecurityDescriptorA(
3018         "O:SY"
3019         "G:S-1-5-21-93476-23408-4576"
3020         "D:(A;NP;GAGXGWGR;;;SU)(A;IOID;CCDC;;;SU)(D;OICI;0xffffffff;;;S-1-5-21-93476-23408-4576)"
3021         "S:(AU;OICINPIOIDSAFA;CCDCLCSWRPRC;;;SU)(AU;NPSA;0x12019f;;;SU)", SDDL_REVISION_1, &sec, &dwDescSize), "Creating descriptor failed\n");
3022     buf = HeapAlloc(GetProcessHeap(), 0, dwDescSize);
3023     pSetSecurityDescriptorControl(sec, SE_DACL_PROTECTED, SE_DACL_PROTECTED);
3024     GetSecurityDescriptorControl(sec, &ctrl, &dwRevision);
3025     expect_eq(ctrl, 0x9014, int, "%x");
3026
3027     ok(GetPrivateObjectSecurity(sec, GROUP_SECURITY_INFORMATION, buf, dwDescSize, &retSize),
3028         "GetPrivateObjectSecurity failed (err=%u)\n", GetLastError());
3029     ok(retSize <= dwDescSize, "Buffer too small (%d vs %d)\n", retSize, dwDescSize);
3030     ok(pConvertSecurityDescriptorToStringSecurityDescriptorA(buf, SDDL_REVISION_1, sec_info, &string, &len), "Conversion failed\n");
3031     CHECK_RESULT_AND_FREE("G:S-1-5-21-93476-23408-4576");
3032     GetSecurityDescriptorControl(buf, &ctrl, &dwRevision);
3033     expect_eq(ctrl, 0x8000, int, "%x");
3034
3035     ok(GetPrivateObjectSecurity(sec, GROUP_SECURITY_INFORMATION|DACL_SECURITY_INFORMATION, buf, dwDescSize, &retSize),
3036         "GetPrivateObjectSecurity failed (err=%u)\n", GetLastError());
3037     ok(retSize <= dwDescSize, "Buffer too small (%d vs %d)\n", retSize, dwDescSize);
3038     ok(pConvertSecurityDescriptorToStringSecurityDescriptorA(buf, SDDL_REVISION_1, sec_info, &string, &len), "Conversion failed err=%u\n", GetLastError());
3039     CHECK_RESULT_AND_FREE("G:S-1-5-21-93476-23408-4576D:(A;NP;GAGXGWGR;;;SU)(A;IOID;CCDC;;;SU)(D;OICI;0xffffffff;;;S-1-5-21-93476-23408-4576)");
3040     GetSecurityDescriptorControl(buf, &ctrl, &dwRevision);
3041     expect_eq(ctrl, 0x8004, int, "%x");
3042
3043     ok(GetPrivateObjectSecurity(sec, sec_info, buf, dwDescSize, &retSize),
3044         "GetPrivateObjectSecurity failed (err=%u)\n", GetLastError());
3045     ok(retSize == dwDescSize, "Buffer too small (%d vs %d)\n", retSize, dwDescSize);
3046     ok(pConvertSecurityDescriptorToStringSecurityDescriptorA(buf, SDDL_REVISION_1, sec_info, &string, &len), "Conversion failed\n");
3047     CHECK_RESULT_AND_FREE("O:SY"
3048         "G:S-1-5-21-93476-23408-4576"
3049         "D:(A;NP;GAGXGWGR;;;SU)(A;IOID;CCDC;;;SU)(D;OICI;0xffffffff;;;S-1-5-21-93476-23408-4576)"
3050         "S:(AU;OICINPIOIDSAFA;CCDCLCSWRPRC;;;SU)(AU;NPSA;0x12019f;;;SU)");
3051     GetSecurityDescriptorControl(buf, &ctrl, &dwRevision);
3052     expect_eq(ctrl, 0x8014, int, "%x");
3053
3054     SetLastError(0xdeadbeef);
3055     ok(GetPrivateObjectSecurity(sec, sec_info, buf, 5, &retSize) == FALSE, "GetPrivateObjectSecurity should have failed\n");
3056     ok(GetLastError() == ERROR_INSUFFICIENT_BUFFER, "Expected error ERROR_INSUFFICIENT_BUFFER, got %u\n", GetLastError());
3057
3058     LocalFree(sec);
3059     HeapFree(GetProcessHeap(), 0, buf);
3060 }
3061 #undef CHECK_RESULT_AND_FREE
3062 #undef CHECK_ONE_OF_AND_FREE
3063
3064 static void test_acls(void)
3065 {
3066     char buffer[256];
3067     PACL pAcl = (PACL)buffer;
3068     BOOL ret;
3069
3070     SetLastError(0xdeadbeef);
3071     ret = InitializeAcl(pAcl, sizeof(ACL) - 1, ACL_REVISION);
3072     if (!ret && GetLastError() == ERROR_CALL_NOT_IMPLEMENTED)
3073     {
3074         win_skip("InitializeAcl is not implemented\n");
3075         return;
3076     }
3077
3078     ok(!ret && GetLastError() == ERROR_INSUFFICIENT_BUFFER, "InitializeAcl with too small a buffer should have failed with ERROR_INSUFFICIENT_BUFFER instead of %d\n", GetLastError());
3079
3080     SetLastError(0xdeadbeef);
3081     ret = InitializeAcl(pAcl, 0xffffffff, ACL_REVISION);
3082     ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER, "InitializeAcl with too large a buffer should have failed with ERROR_INVALID_PARAMETER instead of %d\n", GetLastError());
3083
3084     SetLastError(0xdeadbeef);
3085     ret = InitializeAcl(pAcl, sizeof(buffer), ACL_REVISION1);
3086     ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER, "InitializeAcl(ACL_REVISION1) should have failed with ERROR_INVALID_PARAMETER instead of %d\n", GetLastError());
3087
3088     ret = InitializeAcl(pAcl, sizeof(buffer), ACL_REVISION2);
3089     ok(ret, "InitializeAcl(ACL_REVISION2) failed with error %d\n", GetLastError());
3090
3091     ret = IsValidAcl(pAcl);
3092     ok(ret, "IsValidAcl failed with error %d\n", GetLastError());
3093
3094     ret = InitializeAcl(pAcl, sizeof(buffer), ACL_REVISION3);
3095     ok(ret, "InitializeAcl(ACL_REVISION3) failed with error %d\n", GetLastError());
3096
3097     ret = IsValidAcl(pAcl);
3098     ok(ret, "IsValidAcl failed with error %d\n", GetLastError());
3099
3100     SetLastError(0xdeadbeef);
3101     ret = InitializeAcl(pAcl, sizeof(buffer), ACL_REVISION4);
3102     if (GetLastError() != ERROR_INVALID_PARAMETER)
3103     {
3104         ok(ret, "InitializeAcl(ACL_REVISION4) failed with error %d\n", GetLastError());
3105
3106         ret = IsValidAcl(pAcl);
3107         ok(ret, "IsValidAcl failed with error %d\n", GetLastError());
3108     }
3109     else
3110         win_skip("ACL_REVISION4 is not implemented on NT4\n");
3111
3112     SetLastError(0xdeadbeef);
3113     ret = InitializeAcl(pAcl, sizeof(buffer), -1);
3114     ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER, "InitializeAcl(-1) failed with error %d\n", GetLastError());
3115 }
3116
3117 static void test_GetSecurityInfo(void)
3118 {
3119     HANDLE obj;
3120     PSECURITY_DESCRIPTOR sd;
3121     PSID owner, group;
3122     PACL dacl;
3123     DWORD ret;
3124
3125     if (!pGetSecurityInfo)
3126     {
3127         win_skip("GetSecurityInfo is not available\n");
3128         return;
3129     }
3130
3131     /* Create something.  Files have lots of associated security info.  */
3132     obj = CreateFile(myARGV[0], GENERIC_READ, FILE_SHARE_READ, NULL,
3133                      OPEN_EXISTING, FILE_ATTRIBUTE_NORMAL, NULL);
3134     if (obj == INVALID_HANDLE_VALUE)
3135     {
3136         skip("Couldn't create an object for GetSecurityInfo test\n");
3137         return;
3138     }
3139
3140     ret = pGetSecurityInfo(obj, SE_FILE_OBJECT,
3141                           OWNER_SECURITY_INFORMATION | GROUP_SECURITY_INFORMATION | DACL_SECURITY_INFORMATION,
3142                           &owner, &group, &dacl, NULL, &sd);
3143     if (ret == ERROR_CALL_NOT_IMPLEMENTED)
3144     {
3145         win_skip("GetSecurityInfo is not implemented\n");
3146         CloseHandle(obj);
3147         return;
3148     }
3149     ok(ret == ERROR_SUCCESS, "GetSecurityInfo returned %d\n", ret);
3150     ok(sd != NULL, "GetSecurityInfo\n");
3151     ok(owner != NULL, "GetSecurityInfo\n");
3152     ok(group != NULL, "GetSecurityInfo\n");
3153     ok(dacl != NULL, "GetSecurityInfo\n");
3154     ok(IsValidAcl(dacl), "GetSecurityInfo\n");
3155
3156     LocalFree(sd);
3157
3158     if (!pCreateWellKnownSid)
3159     {
3160         win_skip("NULL parameter test would crash on NT4\n");
3161         CloseHandle(obj);
3162         return;
3163     }
3164
3165     /* If we don't ask for the security descriptor, Windows will still give us
3166        the other stuff, leaving us no way to free it.  */
3167     ret = pGetSecurityInfo(obj, SE_FILE_OBJECT,
3168                           OWNER_SECURITY_INFORMATION | GROUP_SECURITY_INFORMATION | DACL_SECURITY_INFORMATION,
3169                           &owner, &group, &dacl, NULL, NULL);
3170     ok(ret == ERROR_SUCCESS, "GetSecurityInfo returned %d\n", ret);
3171     ok(owner != NULL, "GetSecurityInfo\n");
3172     ok(group != NULL, "GetSecurityInfo\n");
3173     ok(dacl != NULL, "GetSecurityInfo\n");
3174     ok(IsValidAcl(dacl), "GetSecurityInfo\n");
3175
3176     CloseHandle(obj);
3177 }
3178
3179 START_TEST(security)
3180 {
3181     init();
3182     if (!hmod) return;
3183
3184     if (myARGC >= 3)
3185     {
3186         test_process_security_child();
3187         return;
3188     }
3189     test_sid();
3190     test_trustee();
3191     test_luid();
3192     test_CreateWellKnownSid();
3193     test_FileSecurity();
3194     test_AccessCheck();
3195     test_token_attr();
3196     test_LookupAccountSid();
3197     test_LookupAccountName();
3198     test_security_descriptor();
3199     test_process_security();
3200     test_impersonation_level();
3201     test_SetEntriesInAcl();
3202     test_GetNamedSecurityInfoA();
3203     test_ConvertStringSecurityDescriptor();
3204     test_ConvertSecurityDescriptorToString();
3205     test_PrivateObjectSecurity();
3206     test_acls();
3207     test_GetSecurityInfo();
3208 }