d3drm: Fix leakage of pData2 on error.
[wine] / dlls / advapi32 / tests / cred.c
1 /*
2  * Credential Function Tests
3  *
4  * Copyright 2007 Robert Shearman
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 "windef.h"
25 #include "winbase.h"
26 #include "wincred.h"
27
28 #include "wine/test.h"
29
30 static BOOL (WINAPI *pCredDeleteA)(LPCSTR,DWORD,DWORD);
31 static BOOL (WINAPI *pCredEnumerateA)(LPCSTR,DWORD,DWORD *,PCREDENTIALA **);
32 static VOID (WINAPI *pCredFree)(PVOID);
33 static BOOL (WINAPI *pCredGetSessionTypes)(DWORD,LPDWORD);
34 static BOOL (WINAPI *pCredReadA)(LPCSTR,DWORD,DWORD,PCREDENTIALA *);
35 static BOOL (WINAPI *pCredRenameA)(LPCSTR,LPCSTR,DWORD,DWORD);
36 static BOOL (WINAPI *pCredWriteA)(PCREDENTIALA,DWORD);
37 static BOOL (WINAPI *pCredReadDomainCredentialsA)(PCREDENTIAL_TARGET_INFORMATIONA,DWORD,DWORD*,PCREDENTIALA**);
38 static BOOL (WINAPI *pCredMarshalCredentialA)(CRED_MARSHAL_TYPE,PVOID,LPSTR *);
39 static BOOL (WINAPI *pCredUnmarshalCredentialA)(LPCSTR,PCRED_MARSHAL_TYPE,PVOID);
40
41 #define TEST_TARGET_NAME  "credtest.winehq.org"
42 #define TEST_TARGET_NAME2 "credtest2.winehq.org"
43 static const WCHAR TEST_PASSWORD[] = {'p','4','$','$','w','0','r','d','!',0};
44
45 static void test_CredReadA(void)
46 {
47     BOOL ret;
48     PCREDENTIALA cred;
49
50     SetLastError(0xdeadbeef);
51     ret = pCredReadA(TEST_TARGET_NAME, -1, 0, &cred);
52     ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
53         "CredReadA should have failed with ERROR_INVALID_PARAMETER instead of %d\n",
54         GetLastError());
55
56     SetLastError(0xdeadbeef);
57     ret = pCredReadA(TEST_TARGET_NAME, CRED_TYPE_GENERIC, 0xdeadbeef, &cred);
58     ok(!ret && ( GetLastError() == ERROR_INVALID_FLAGS || GetLastError() == ERROR_INVALID_PARAMETER ),
59         "CredReadA should have failed with ERROR_INVALID_FLAGS or ERROR_INVALID_PARAMETER instead of %d\n",
60         GetLastError());
61
62     SetLastError(0xdeadbeef);
63     ret = pCredReadA(NULL, CRED_TYPE_GENERIC, 0, &cred);
64     ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
65         "CredReadA should have failed with ERROR_INVALID_PARAMETER instead of %d\n",
66         GetLastError());
67 }
68
69 static void test_CredWriteA(void)
70 {
71     CREDENTIALA new_cred;
72     BOOL ret;
73
74     SetLastError(0xdeadbeef);
75     ret = pCredWriteA(NULL, 0);
76     ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
77         "CredWriteA should have failed with ERROR_INVALID_PARAMETER instead of %d\n",
78         GetLastError());
79
80     new_cred.Flags = 0;
81     new_cred.Type = CRED_TYPE_GENERIC;
82     new_cred.TargetName = NULL;
83     new_cred.Comment = (char *)"Comment";
84     new_cred.CredentialBlobSize = 0;
85     new_cred.CredentialBlob = NULL;
86     new_cred.Persist = CRED_PERSIST_ENTERPRISE;
87     new_cred.AttributeCount = 0;
88     new_cred.Attributes = NULL;
89     new_cred.TargetAlias = NULL;
90     new_cred.UserName = (char *)"winetest";
91
92     SetLastError(0xdeadbeef);
93     ret = pCredWriteA(&new_cred, 0);
94     ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
95         "CredWriteA should have failed with ERROR_INVALID_PARAMETER instead of %d\n",
96         GetLastError());
97
98     new_cred.TargetName = (char *)TEST_TARGET_NAME;
99     new_cred.Type = CRED_TYPE_DOMAIN_PASSWORD;
100
101     SetLastError(0xdeadbeef);
102     ret = pCredWriteA(&new_cred, 0);
103     if (ret)
104     {
105         ok(GetLastError() == ERROR_SUCCESS ||
106            GetLastError() == ERROR_IO_PENDING, /* Vista */
107            "Expected ERROR_IO_PENDING, got %d\n", GetLastError());
108     }
109     else
110     {
111         ok(GetLastError() == ERROR_BAD_USERNAME ||
112            GetLastError() == ERROR_NO_SUCH_LOGON_SESSION, /* Vista */
113            "CredWrite with username without domain should return ERROR_BAD_USERNAME"
114            "or ERROR_NO_SUCH_LOGON_SESSION not %d\n", GetLastError());
115     }
116
117     new_cred.UserName = NULL;
118     SetLastError(0xdeadbeef);
119     ret = pCredWriteA(&new_cred, 0);
120     ok(!ret && GetLastError() == ERROR_BAD_USERNAME,
121         "CredWriteA with NULL username should have failed with ERROR_BAD_USERNAME instead of %d\n",
122         GetLastError());
123 }
124
125 static void test_CredDeleteA(void)
126 {
127     BOOL ret;
128
129     SetLastError(0xdeadbeef);
130     ret = pCredDeleteA(TEST_TARGET_NAME, -1, 0);
131     ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
132         "CredDeleteA should have failed with ERROR_INVALID_PARAMETER instead of %d\n",
133         GetLastError());
134
135     SetLastError(0xdeadbeef);
136     ret = pCredDeleteA(TEST_TARGET_NAME, CRED_TYPE_GENERIC, 0xdeadbeef);
137     ok(!ret && ( GetLastError() == ERROR_INVALID_FLAGS || GetLastError() == ERROR_INVALID_PARAMETER /* Vista */ ),
138         "CredDeleteA should have failed with ERROR_INVALID_FLAGS or ERROR_INVALID_PARAMETER instead of %d\n",
139         GetLastError());
140 }
141
142 static void test_CredReadDomainCredentialsA(void)
143 {
144     BOOL ret;
145     char target_name[] = "no_such_target";
146     CREDENTIAL_TARGET_INFORMATIONA info = {target_name, NULL, target_name, NULL, NULL, NULL, NULL, 0, 0, NULL};
147     DWORD count;
148     PCREDENTIAL* creds;
149
150     if (!pCredReadDomainCredentialsA)
151     {
152         win_skip("CredReadDomainCredentialsA() is not implemented\n");
153         return;
154     }
155
156     /* these two tests would crash on both native and Wine. Implementations
157      * does not check for NULL output pointers and try to zero them out early */
158 if(0)
159 {
160     ret = pCredReadDomainCredentialsA(&info, 0, NULL, &creds);
161     ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER, "!\n");
162     ret = pCredReadDomainCredentialsA(&info, 0, &count, NULL);
163     ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER, "!\n");
164 }
165
166     SetLastError(0xdeadbeef);
167     ret = pCredReadDomainCredentialsA(NULL, 0, &count, &creds);
168     ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
169         "CredReadDomainCredentialsA should have failed with ERROR_INVALID_PARAMETER instead of %d\n",
170         GetLastError());
171
172     SetLastError(0xdeadbeef);
173     creds = (void*)0x12345;
174     count = 2;
175     ret = pCredReadDomainCredentialsA(&info, 0, &count, &creds);
176     ok(!ret && GetLastError() == ERROR_NOT_FOUND,
177         "CredReadDomainCredentialsA should have failed with ERROR_NOT_FOUND instead of %d\n",
178         GetLastError());
179     ok(count ==0 && creds == NULL, "CredReadDomainCredentialsA must not return any result\n");
180
181     info.TargetName = NULL;
182
183     SetLastError(0xdeadbeef);
184     ret = pCredReadDomainCredentialsA(&info, 0, &count, &creds);
185     ok(!ret, "CredReadDomainCredentialsA should have failed\n");
186     ok(GetLastError() == ERROR_NOT_FOUND ||
187         GetLastError() == ERROR_INVALID_PARAMETER, /* Vista, W2K8 */
188         "Expected ERROR_NOT_FOUND or ERROR_INVALID_PARAMETER instead of %d\n",
189         GetLastError());
190
191     info.DnsServerName = NULL;
192
193     SetLastError(0xdeadbeef);
194     ret = pCredReadDomainCredentialsA(&info, 0, &count, &creds);
195     ok(!ret, "CredReadDomainCredentialsA should have failed\n");
196     ok(GetLastError() == ERROR_NOT_FOUND ||
197         GetLastError() == ERROR_INVALID_PARAMETER, /* Vista, W2K8 */
198         "Expected ERROR_NOT_FOUND or ERROR_INVALID_PARAMETER instead of %d\n",
199         GetLastError());
200 }
201
202 static void check_blob(int line, DWORD cred_type, PCREDENTIALA cred)
203 {
204     if (cred_type == CRED_TYPE_DOMAIN_PASSWORD)
205     {
206         todo_wine
207         ok_(__FILE__, line)(cred->CredentialBlobSize == 0, "expected CredentialBlobSize of 0 but got %d\n", cred->CredentialBlobSize);
208         todo_wine
209         ok_(__FILE__, line)(!cred->CredentialBlob, "expected NULL credentials but got %p\n", cred->CredentialBlob);
210     }
211     else
212     {
213         DWORD size=sizeof(TEST_PASSWORD);
214         ok_(__FILE__, line)(cred->CredentialBlobSize == size, "expected CredentialBlobSize of %u but got %u\n", size, cred->CredentialBlobSize);
215         ok_(__FILE__, line)(cred->CredentialBlob != NULL, "CredentialBlob should be present\n");
216         if (cred->CredentialBlob)
217             ok_(__FILE__, line)(!memcmp(cred->CredentialBlob, TEST_PASSWORD, size), "wrong CredentialBlob\n");
218     }
219 }
220
221 static void test_generic(void)
222 {
223     BOOL ret;
224     DWORD count, i;
225     PCREDENTIALA *creds;
226     CREDENTIALA new_cred;
227     PCREDENTIALA cred;
228     BOOL found = FALSE;
229
230     new_cred.Flags = 0;
231     new_cred.Type = CRED_TYPE_GENERIC;
232     new_cred.TargetName = (char *)TEST_TARGET_NAME;
233     new_cred.Comment = (char *)"Comment";
234     new_cred.CredentialBlobSize = sizeof(TEST_PASSWORD);
235     new_cred.CredentialBlob = (LPBYTE)TEST_PASSWORD;
236     new_cred.Persist = CRED_PERSIST_ENTERPRISE;
237     new_cred.AttributeCount = 0;
238     new_cred.Attributes = NULL;
239     new_cred.TargetAlias = NULL;
240     new_cred.UserName = (char *)"winetest";
241
242     ret = pCredWriteA(&new_cred, 0);
243     ok(ret || broken(GetLastError() == ERROR_NO_SUCH_LOGON_SESSION),
244        "CredWriteA failed with error %d\n", GetLastError());
245     if (!ret)
246     {
247         skip("couldn't write generic credentials, skipping tests\n");
248         return;
249     }
250
251     ret = pCredEnumerateA(NULL, 0, &count, &creds);
252     ok(ret, "CredEnumerateA failed with error %d\n", GetLastError());
253
254     for (i = 0; i < count; i++)
255     {
256         if (!strcmp(creds[i]->TargetName, TEST_TARGET_NAME))
257         {
258             ok(creds[i]->Type == CRED_TYPE_GENERIC ||
259                creds[i]->Type == CRED_TYPE_DOMAIN_PASSWORD, /* Vista */
260                "expected creds[%d]->Type CRED_TYPE_GENERIC or CRED_TYPE_DOMAIN_PASSWORD but got %d\n", i, creds[i]->Type);
261             ok(!creds[i]->Flags, "expected creds[%d]->Flags 0 but got 0x%x\n", i, creds[i]->Flags);
262             ok(!strcmp(creds[i]->Comment, "Comment"), "expected creds[%d]->Comment \"Comment\" but got \"%s\"\n", i, creds[i]->Comment);
263             check_blob(__LINE__, creds[i]->Type, creds[i]);
264             ok(creds[i]->Persist, "expected creds[%d]->Persist CRED_PERSIST_ENTERPRISE but got %d\n", i, creds[i]->Persist);
265             ok(!strcmp(creds[i]->UserName, "winetest"), "expected creds[%d]->UserName \"winetest\" but got \"%s\"\n", i, creds[i]->UserName);
266             found = TRUE;
267         }
268     }
269     pCredFree(creds);
270     ok(found, "credentials not found\n");
271
272     ret = pCredReadA(TEST_TARGET_NAME, CRED_TYPE_GENERIC, 0, &cred);
273     ok(ret, "CredReadA failed with error %d\n", GetLastError());
274     pCredFree(cred);
275
276     ret = pCredDeleteA(TEST_TARGET_NAME, CRED_TYPE_GENERIC, 0);
277     ok(ret, "CredDeleteA failed with error %d\n", GetLastError());
278 }
279
280 static void test_domain_password(DWORD cred_type)
281 {
282     BOOL ret;
283     DWORD count, i;
284     PCREDENTIALA *creds;
285     CREDENTIALA new_cred;
286     PCREDENTIALA cred;
287     BOOL found = FALSE;
288
289     new_cred.Flags = 0;
290     new_cred.Type = cred_type;
291     new_cred.TargetName = (char *)TEST_TARGET_NAME;
292     new_cred.Comment = (char *)"Comment";
293     new_cred.CredentialBlobSize = sizeof(TEST_PASSWORD);
294     new_cred.CredentialBlob = (LPBYTE)TEST_PASSWORD;
295     new_cred.Persist = CRED_PERSIST_ENTERPRISE;
296     new_cred.AttributeCount = 0;
297     new_cred.Attributes = NULL;
298     new_cred.TargetAlias = NULL;
299     new_cred.UserName = (char *)"test\\winetest";
300     ret = pCredWriteA(&new_cred, 0);
301     if (!ret && GetLastError() == ERROR_NO_SUCH_LOGON_SESSION)
302     {
303         skip("CRED_TYPE_DOMAIN_PASSWORD credentials are not supported "
304              "or are disabled. Skipping\n");
305         return;
306     }
307     ok(ret, "CredWriteA failed with error %d\n", GetLastError());
308
309     ret = pCredEnumerateA(NULL, 0, &count, &creds);
310     ok(ret, "CredEnumerateA failed with error %d\n", GetLastError());
311
312     for (i = 0; i < count; i++)
313     {
314         if (!strcmp(creds[i]->TargetName, TEST_TARGET_NAME))
315         {
316             ok(creds[i]->Type == cred_type, "expected creds[%d]->Type CRED_TYPE_DOMAIN_PASSWORD but got %d\n", i, creds[i]->Type);
317             ok(!creds[i]->Flags, "expected creds[%d]->Flags 0 but got 0x%x\n", i, creds[i]->Flags);
318             ok(!strcmp(creds[i]->Comment, "Comment"), "expected creds[%d]->Comment \"Comment\" but got \"%s\"\n", i, creds[i]->Comment);
319             check_blob(__LINE__, cred_type, creds[i]);
320             ok(creds[i]->Persist, "expected creds[%d]->Persist CRED_PERSIST_ENTERPRISE but got %d\n", i, creds[i]->Persist);
321             ok(!strcmp(creds[i]->UserName, "test\\winetest"), "expected creds[%d]->UserName \"winetest\" but got \"%s\"\n", i, creds[i]->UserName);
322             found = TRUE;
323         }
324     }
325     pCredFree(creds);
326     ok(found, "credentials not found\n");
327
328     ret = pCredReadA(TEST_TARGET_NAME, cred_type, 0, &cred);
329     ok(ret, "CredReadA failed with error %d\n", GetLastError());
330     if (ret)  /* don't check the values of cred, if CredReadA failed. */
331     {
332         check_blob(__LINE__, cred_type, cred);
333         pCredFree(cred);
334     }
335
336     ret = pCredDeleteA(TEST_TARGET_NAME, cred_type, 0);
337     ok(ret, "CredDeleteA failed with error %d\n", GetLastError());
338 }
339
340 static void test_CredMarshalCredentialA(void)
341 {
342     static WCHAR emptyW[] = {0};
343     static WCHAR tW[] = {'t',0};
344     static WCHAR teW[] = {'t','e',0};
345     static WCHAR tesW[] = {'t','e','s',0};
346     static WCHAR testW[] = {'t','e','s','t',0};
347     static WCHAR test1W[] = {'t','e','s','t','1',0};
348     CERT_CREDENTIAL_INFO cert;
349     USERNAME_TARGET_CREDENTIAL_INFO username;
350     DWORD error;
351     char *str;
352     BOOL ret;
353
354     SetLastError( 0xdeadbeef );
355     ret = pCredMarshalCredentialA( 0, NULL, NULL );
356     error = GetLastError();
357     ok( !ret, "unexpected success\n" );
358     ok( error == ERROR_INVALID_PARAMETER, "got %u\n", error );
359
360     memset( cert.rgbHashOfCert, 0, sizeof(cert.rgbHashOfCert) );
361     cert.cbSize = sizeof(cert);
362     SetLastError( 0xdeadbeef );
363     ret = pCredMarshalCredentialA( 0, &cert, NULL );
364     error = GetLastError();
365     ok( !ret, "unexpected success\n" );
366     ok( error == ERROR_INVALID_PARAMETER, "got %u\n", error );
367
368     str = (char *)0xdeadbeef;
369     SetLastError( 0xdeadbeef );
370     ret = pCredMarshalCredentialA( 0, &cert, &str );
371     error = GetLastError();
372     ok( !ret, "unexpected success\n" );
373     ok( error == ERROR_INVALID_PARAMETER, "got %u\n", error );
374     ok( str == (char *)0xdeadbeef, "got %p\n", str );
375
376     SetLastError( 0xdeadbeef );
377     ret = pCredMarshalCredentialA( CertCredential, NULL, NULL );
378     error = GetLastError();
379     ok( !ret, "unexpected success\n" );
380     ok( error == ERROR_INVALID_PARAMETER, "got %u\n", error );
381
382     if (0) { /* crash */
383     SetLastError( 0xdeadbeef );
384     ret = pCredMarshalCredentialA( CertCredential, &cert, NULL );
385     error = GetLastError();
386     ok( !ret, "unexpected success\n" );
387     ok( error == ERROR_INVALID_PARAMETER, "got %u\n", error );
388     }
389
390     cert.cbSize = 0;
391     str = (char *)0xdeadbeef;
392     SetLastError( 0xdeadbeef );
393     ret = pCredMarshalCredentialA( CertCredential, &cert, &str );
394     error = GetLastError();
395     ok( !ret, "unexpected success\n" );
396     ok( error == ERROR_INVALID_PARAMETER, "got %u\n", error );
397     ok( str == (char *)0xdeadbeef, "got %p\n", str );
398
399     cert.cbSize = sizeof(cert) + 4;
400     str = NULL;
401     ret = pCredMarshalCredentialA( CertCredential, &cert, &str );
402     ok( ret, "unexpected failure %u\n", GetLastError() );
403     ok( str != NULL, "str not set\n" );
404     ok( !lstrcmpA( str, "@@BAAAAAAAAAAAAAAAAAAAAAAAAAAA" ), "got %s\n", str );
405     pCredFree( str );
406
407     cert.cbSize = sizeof(cert);
408     cert.rgbHashOfCert[0] = 2;
409     str = NULL;
410     ret = pCredMarshalCredentialA( CertCredential, &cert, &str );
411     ok( ret, "unexpected failure %u\n", GetLastError() );
412     ok( str != NULL, "str not set\n" );
413     ok( !lstrcmpA( str, "@@BCAAAAAAAAAAAAAAAAAAAAAAAAAA" ), "got %s\n", str );
414     pCredFree( str );
415
416     cert.rgbHashOfCert[0] = 255;
417     str = NULL;
418     ret = pCredMarshalCredentialA( CertCredential, &cert, &str );
419     ok( ret, "unexpected failure %u\n", GetLastError() );
420     ok( str != NULL, "str not set\n" );
421     ok( !lstrcmpA( str, "@@B-DAAAAAAAAAAAAAAAAAAAAAAAAA" ), "got %s\n", str );
422     pCredFree( str );
423
424     cert.rgbHashOfCert[0] = 1;
425     cert.rgbHashOfCert[1] = 1;
426     str = NULL;
427     ret = pCredMarshalCredentialA( CertCredential, &cert, &str );
428     ok( ret, "unexpected failure %u\n", GetLastError() );
429     ok( str != NULL, "str not set\n" );
430     ok( !lstrcmpA( str, "@@BBEAAAAAAAAAAAAAAAAAAAAAAAAA" ), "got %s\n", str );
431     pCredFree( str );
432
433     cert.rgbHashOfCert[0] = 1;
434     cert.rgbHashOfCert[1] = 1;
435     cert.rgbHashOfCert[2] = 1;
436     str = NULL;
437     ret = pCredMarshalCredentialA( CertCredential, &cert, &str );
438     ok( ret, "unexpected failure %u\n", GetLastError() );
439     ok( str != NULL, "str not set\n" );
440     ok( !lstrcmpA( str, "@@BBEQAAAAAAAAAAAAAAAAAAAAAAAA" ), "got %s\n", str );
441     pCredFree( str );
442
443     memset( cert.rgbHashOfCert, 0, sizeof(cert.rgbHashOfCert) );
444     cert.rgbHashOfCert[0] = 'W';
445     cert.rgbHashOfCert[1] = 'i';
446     cert.rgbHashOfCert[2] = 'n';
447     cert.rgbHashOfCert[3] = 'e';
448     str = NULL;
449     ret = pCredMarshalCredentialA( CertCredential, &cert, &str );
450     ok( ret, "unexpected failure %u\n", GetLastError() );
451     ok( str != NULL, "str not set\n" );
452     ok( !lstrcmpA( str, "@@BXlmblBAAAAAAAAAAAAAAAAAAAAA" ), "got %s\n", str );
453     pCredFree( str );
454
455     memset( cert.rgbHashOfCert, 0xff, sizeof(cert.rgbHashOfCert) );
456     str = NULL;
457     ret = pCredMarshalCredentialA( CertCredential, &cert, &str );
458     ok( ret, "unexpected failure %u\n", GetLastError() );
459     ok( str != NULL, "str not set\n" );
460     ok( !lstrcmpA( str, "@@B--------------------------P" ), "got %s\n", str );
461     pCredFree( str );
462
463     username.UserName = NULL;
464     str = (char *)0xdeadbeef;
465     SetLastError( 0xdeadbeef );
466     ret = pCredMarshalCredentialA( UsernameTargetCredential, &username, &str );
467     ok( !ret, "unexpected success\n" );
468     ok( error == ERROR_INVALID_PARAMETER, "got %u\n", error );
469     ok( str == (char *)0xdeadbeef, "got %p\n", str );
470
471     username.UserName = emptyW;
472     str = (char *)0xdeadbeef;
473     SetLastError( 0xdeadbeef );
474     ret = pCredMarshalCredentialA( UsernameTargetCredential, &username, &str );
475     ok( !ret, "unexpected success\n" );
476     ok( error == ERROR_INVALID_PARAMETER, "got %u\n", error );
477     ok( str == (char *)0xdeadbeef, "got %p\n", str );
478
479     username.UserName = tW;
480     str = NULL;
481     ret = pCredMarshalCredentialA( UsernameTargetCredential, &username, &str );
482     ok( ret, "unexpected failure %u\n", GetLastError() );
483     ok( str != NULL, "str not set\n" );
484     ok( !lstrcmpA( str, "@@CCAAAAA0BA" ), "got %s\n", str );
485     pCredFree( str );
486
487     username.UserName = teW;
488     str = NULL;
489     ret = pCredMarshalCredentialA( UsernameTargetCredential, &username, &str );
490     ok( ret, "unexpected failure %u\n", GetLastError() );
491     ok( str != NULL, "str not set\n" );
492     ok( !lstrcmpA( str, "@@CEAAAAA0BQZAA" ), "got %s\n", str );
493     pCredFree( str );
494
495     username.UserName = tesW;
496     str = NULL;
497     ret = pCredMarshalCredentialA( UsernameTargetCredential, &username, &str );
498     ok( ret, "unexpected failure %u\n", GetLastError() );
499     ok( str != NULL, "str not set\n" );
500     ok( !lstrcmpA( str, "@@CGAAAAA0BQZAMHA" ), "got %s\n", str );
501     pCredFree( str );
502
503     username.UserName = testW;
504     str = NULL;
505     ret = pCredMarshalCredentialA( UsernameTargetCredential, &username, &str );
506     ok( ret, "unexpected failure %u\n", GetLastError() );
507     ok( str != NULL, "str not set\n" );
508     ok( !lstrcmpA( str, "@@CIAAAAA0BQZAMHA0BA" ), "got %s\n", str );
509     pCredFree( str );
510
511     username.UserName = test1W;
512     str = NULL;
513     ret = pCredMarshalCredentialA( UsernameTargetCredential, &username, &str );
514     ok( ret, "unexpected failure %u\n", GetLastError() );
515     ok( str != NULL, "str not set\n" );
516     ok( !lstrcmpA( str, "@@CKAAAAA0BQZAMHA0BQMAA" ), "got %s\n", str );
517     pCredFree( str );
518 }
519
520 static void test_CredUnmarshalCredentialA(void)
521 {
522     static WCHAR tW[] = {'t',0};
523     static WCHAR testW[] = {'t','e','s','t',0};
524     CERT_CREDENTIAL_INFO *cert;
525     USERNAME_TARGET_CREDENTIAL_INFO *username;
526     CRED_MARSHAL_TYPE type;
527     unsigned int i;
528     DWORD error;
529     BOOL ret;
530
531     SetLastError( 0xdeadbeef );
532     ret = pCredUnmarshalCredentialA( NULL, NULL, NULL );
533     error = GetLastError();
534     ok( !ret, "unexpected success\n" );
535     ok( error == ERROR_INVALID_PARAMETER, "got %u\n", error );
536
537     cert = NULL;
538     SetLastError( 0xdeadbeef );
539     ret = pCredUnmarshalCredentialA( NULL, NULL, (void **)&cert );
540     error = GetLastError();
541     ok( !ret, "unexpected success\n" );
542     ok( error == ERROR_INVALID_PARAMETER, "got %u\n", error );
543
544     type = 0;
545     cert = NULL;
546     SetLastError( 0xdeadbeef );
547     ret = pCredUnmarshalCredentialA( NULL, &type, (void **)&cert );
548     error = GetLastError();
549     ok( !ret, "unexpected success\n" );
550     ok( error == ERROR_INVALID_PARAMETER, "got %u\n", error );
551
552     type = 0;
553     cert = NULL;
554     SetLastError( 0xdeadbeef );
555     ret = pCredUnmarshalCredentialA( "", &type, (void **)&cert );
556     error = GetLastError();
557     ok( !ret, "unexpected success\n" );
558     ok( error == ERROR_INVALID_PARAMETER, "got %u\n", error );
559
560     if (0) { /* crash */
561     SetLastError( 0xdeadbeef );
562     ret = pCredUnmarshalCredentialA( "@@BAAAAAAAAAAAAAAAAAAAAAAAAAAA", &type, NULL );
563     error = GetLastError();
564     ok( !ret, "unexpected success\n" );
565     ok( error == ERROR_INVALID_PARAMETER, "got %u\n", error );
566
567     SetLastError( 0xdeadbeef );
568     ret = pCredUnmarshalCredentialA( "@@BAAAAAAAAAAAAAAAAAAAAAAAAAAA", NULL, (void **)&cert );
569     error = GetLastError();
570     ok( !ret, "unexpected success\n" );
571     ok( error == ERROR_INVALID_PARAMETER, "got %u\n", error );
572     }
573
574     type = 0;
575     cert = NULL;
576     ret = pCredUnmarshalCredentialA( "@@BAAAAAAAAAAAAAAAAAAAAAAAAAAA", &type, (void **)&cert );
577     ok( ret, "unexpected failure %u\n", GetLastError() );
578     ok( type == CertCredential, "got %u\n", type );
579     ok( cert != NULL, "cert is NULL\n" );
580     ok( cert->cbSize == sizeof(*cert), "wrong size %u\n", cert->cbSize );
581     for (i = 0; i < sizeof(cert->rgbHashOfCert); i++) ok( !cert->rgbHashOfCert[i], "wrong data\n" );
582     pCredFree( cert );
583
584     type = 0;
585     cert = NULL;
586     ret = pCredUnmarshalCredentialA( "@@BXlmblBAAAAAAAAAAAAAAAAAAAAA", &type, (void **)&cert );
587     ok( ret, "unexpected failure %u\n", GetLastError() );
588     ok( type == CertCredential, "got %u\n", type );
589     ok( cert != NULL, "cert is NULL\n" );
590     ok( cert->cbSize == sizeof(*cert), "wrong size %u\n", cert->cbSize );
591     ok( cert->rgbHashOfCert[0] == 'W', "wrong data)\n" );
592     ok( cert->rgbHashOfCert[1] == 'i', "wrong data\n" );
593     ok( cert->rgbHashOfCert[2] == 'n', "wrong data\n" );
594     ok( cert->rgbHashOfCert[3] == 'e', "wrong data\n" );
595     for (i = 4; i < sizeof(cert->rgbHashOfCert); i++) ok( !cert->rgbHashOfCert[i], "wrong data\n" );
596     pCredFree( cert );
597
598     SetLastError( 0xdeadbeef );
599     ret = pCredUnmarshalCredentialA( "@@CAAAAAA", &type, (void **)&username );
600     error = GetLastError();
601     ok( !ret, "unexpected success\n" );
602     ok( error == ERROR_INVALID_PARAMETER, "got %u\n", error );
603
604     SetLastError( 0xdeadbeef );
605     ret = pCredUnmarshalCredentialA( "@@CAAAAAA0BA", &type, (void **)&username );
606     error = GetLastError();
607     ok( !ret, "unexpected success\n" );
608     ok( error == ERROR_INVALID_PARAMETER, "got %u\n", error );
609
610     type = 0;
611     username = NULL;
612     ret = pCredUnmarshalCredentialA( "@@CCAAAAA0BA", &type, (void **)&username );
613     ok( ret, "unexpected failure %u\n", GetLastError() );
614     ok( type == UsernameTargetCredential, "got %u\n", type );
615     ok( username != NULL, "username is NULL\n" );
616     ok( username->UserName != NULL, "UserName is NULL\n" );
617     ok( !lstrcmpW( username->UserName, tW ), "got %s\n", wine_dbgstr_w(username->UserName) );
618     pCredFree( username );
619
620     type = 0;
621     username = NULL;
622     ret = pCredUnmarshalCredentialA( "@@CIAAAAA0BQZAMHA0BA", &type, (void **)&username );
623     ok( ret, "unexpected failure %u\n", GetLastError() );
624     ok( type == UsernameTargetCredential, "got %u\n", type );
625     ok( username != NULL, "username is NULL\n" );
626     ok( username->UserName != NULL, "UserName is NULL\n" );
627     ok( !lstrcmpW( username->UserName, testW ), "got %s\n", wine_dbgstr_w(username->UserName) );
628     pCredFree( username );
629 }
630
631 START_TEST(cred)
632 {
633     DWORD persists[CRED_TYPE_MAXIMUM];
634
635     pCredEnumerateA = (void *)GetProcAddress(GetModuleHandle("advapi32.dll"), "CredEnumerateA");
636     pCredFree = (void *)GetProcAddress(GetModuleHandle("advapi32.dll"), "CredFree");
637     pCredGetSessionTypes = (void *)GetProcAddress(GetModuleHandle("advapi32.dll"), "CredGetSessionTypes");
638     pCredWriteA = (void *)GetProcAddress(GetModuleHandle("advapi32.dll"), "CredWriteA");
639     pCredDeleteA = (void *)GetProcAddress(GetModuleHandle("advapi32.dll"), "CredDeleteA");
640     pCredReadA = (void *)GetProcAddress(GetModuleHandle("advapi32.dll"), "CredReadA");
641     pCredRenameA = (void *)GetProcAddress(GetModuleHandle("advapi32.dll"), "CredRenameA");
642     pCredReadDomainCredentialsA = (void *)GetProcAddress(GetModuleHandle("advapi32.dll"), "CredReadDomainCredentialsA");
643     pCredMarshalCredentialA = (void *)GetProcAddress(GetModuleHandle("advapi32.dll"), "CredMarshalCredentialA");
644     pCredUnmarshalCredentialA = (void *)GetProcAddress(GetModuleHandle("advapi32.dll"), "CredUnmarshalCredentialA");
645
646     if (!pCredEnumerateA || !pCredFree || !pCredWriteA || !pCredDeleteA || !pCredReadA)
647     {
648         win_skip("credentials functions not present in advapi32.dll\n");
649         return;
650     }
651
652     if (pCredGetSessionTypes)
653     {
654         BOOL ret;
655         DWORD i;
656         ret = pCredGetSessionTypes(CRED_TYPE_MAXIMUM, persists);
657         ok(ret, "CredGetSessionTypes failed with error %d\n", GetLastError());
658         ok(persists[0] == CRED_PERSIST_NONE, "persists[0] = %u instead of CRED_PERSIST_NONE\n", persists[0]);
659         for (i=0; i < CRED_TYPE_MAXIMUM; i++)
660             ok(persists[i] <= CRED_PERSIST_ENTERPRISE, "bad value for persists[%u]: %u\n", i, persists[i]);
661     }
662     else
663         memset(persists, CRED_PERSIST_ENTERPRISE, sizeof(persists));
664
665     test_CredReadA();
666     test_CredWriteA();
667     test_CredDeleteA();
668
669     test_CredReadDomainCredentialsA();
670
671     trace("generic:\n");
672     if (persists[CRED_TYPE_GENERIC] == CRED_PERSIST_NONE)
673         skip("CRED_TYPE_GENERIC credentials are not supported or are disabled. Skipping\n");
674     else
675         test_generic();
676
677         trace("domain password:\n");
678     if (persists[CRED_TYPE_DOMAIN_PASSWORD] == CRED_PERSIST_NONE)
679         skip("CRED_TYPE_DOMAIN_PASSWORD credentials are not supported or are disabled. Skipping\n");
680     else
681         test_domain_password(CRED_TYPE_DOMAIN_PASSWORD);
682
683     trace("domain visible password:\n");
684     if (persists[CRED_TYPE_DOMAIN_VISIBLE_PASSWORD] == CRED_PERSIST_NONE)
685         skip("CRED_TYPE_DOMAIN_VISIBLE_PASSWORD credentials are not supported or are disabled. Skipping\n");
686     else
687         test_domain_password(CRED_TYPE_DOMAIN_VISIBLE_PASSWORD);
688
689     test_CredMarshalCredentialA();
690     test_CredUnmarshalCredentialA();
691 }