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
  39
  40 #define TEST_TARGET_NAME  "credtest.winehq.org"
  41 #define TEST_TARGET_NAME2 "credtest2.winehq.org"
  42 static const WCHAR TEST_PASSWORD[] = {'p','4','$','$','w','0','r','d','!',0};
  43
  44 static void test_CredReadA(void)
  45 {
  46     BOOL ret;
  47     PCREDENTIALA cred;
  48
  49     SetLastError(0xdeadbeef);
  50     ret = pCredReadA(TEST_TARGET_NAME, -1, 0, &cred);
  51     ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
  52         "CredReadA should have failed with ERROR_INVALID_PARAMETER instead of %d\n",
  53         GetLastError());
  54
  55     SetLastError(0xdeadbeef);
  56     ret = pCredReadA(TEST_TARGET_NAME, CRED_TYPE_GENERIC, 0xdeadbeef, &cred);
  57     ok(!ret && ( GetLastError() == ERROR_INVALID_FLAGS || GetLastError() == ERROR_INVALID_PARAMETER ),
  58         "CredReadA should have failed with ERROR_INVALID_FLAGS or ERROR_INVALID_PARAMETER instead of %d\n",
  59         GetLastError());
  60
  61     SetLastError(0xdeadbeef);
  62     ret = pCredReadA(NULL, CRED_TYPE_GENERIC, 0, &cred);
  63     ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
  64         "CredReadA should have failed with ERROR_INVALID_PARAMETER instead of %d\n",
  65         GetLastError());
  66 }
  67
  68 static void test_CredWriteA(void)
  69 {
  70     CREDENTIALA new_cred;
  71     BOOL ret;
  72
  73     SetLastError(0xdeadbeef);
  74     ret = pCredWriteA(NULL, 0);
  75     ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
  76         "CredWriteA should have failed with ERROR_INVALID_PARAMETER instead of %d\n",
  77         GetLastError());
  78
  79     new_cred.Flags = 0;
  80     new_cred.Type = CRED_TYPE_GENERIC;
  81     new_cred.TargetName = NULL;
  82     new_cred.Comment = (char *)"Comment";
  83     new_cred.CredentialBlobSize = 0;
  84     new_cred.CredentialBlob = NULL;
  85     new_cred.Persist = CRED_PERSIST_ENTERPRISE;
  86     new_cred.AttributeCount = 0;
  87     new_cred.Attributes = NULL;
  88     new_cred.TargetAlias = NULL;
  89     new_cred.UserName = (char *)"winetest";
  90
  91     SetLastError(0xdeadbeef);
  92     ret = pCredWriteA(&new_cred, 0);
  93     ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
  94         "CredWriteA should have failed with ERROR_INVALID_PARAMETER instead of %d\n",
  95         GetLastError());
  96
  97     new_cred.TargetName = (char *)TEST_TARGET_NAME;
  98     new_cred.Type = CRED_TYPE_DOMAIN_PASSWORD;
  99
 100     SetLastError(0xdeadbeef);
 101     ret = pCredWriteA(&new_cred, 0);
 102     if (ret)
 103     {
 104         ok(GetLastError() == ERROR_SUCCESS ||
 105            GetLastError() == ERROR_IO_PENDING, /* Vista */
 106            "Expected ERROR_IO_PENDING, got %d\n", GetLastError());
 107     }
 108     else
 109     {
 110         ok(GetLastError() == ERROR_BAD_USERNAME ||
 111            GetLastError() == ERROR_NO_SUCH_LOGON_SESSION, /* Vista */
 112            "CredWrite with username without domain should return ERROR_BAD_USERNAME"
 113            "or ERROR_NO_SUCH_LOGON_SESSION not %d\n", GetLastError());
 114     }
 115
 116     new_cred.UserName = NULL;
 117     SetLastError(0xdeadbeef);
 118     ret = pCredWriteA(&new_cred, 0);
 119     ok(!ret && GetLastError() == ERROR_BAD_USERNAME,
 120         "CredWriteA with NULL username should have failed with ERROR_BAD_USERNAME instead of %d\n",
 121         GetLastError());
 122 }
 123
 124 static void test_CredDeleteA(void)
 125 {
 126     BOOL ret;
 127
 128     SetLastError(0xdeadbeef);
 129     ret = pCredDeleteA(TEST_TARGET_NAME, -1, 0);
 130     ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
 131         "CredDeleteA should have failed with ERROR_INVALID_PARAMETER instead of %d\n",
 132         GetLastError());
 133
 134     SetLastError(0xdeadbeef);
 135     ret = pCredDeleteA(TEST_TARGET_NAME, CRED_TYPE_GENERIC, 0xdeadbeef);
 136     ok(!ret && ( GetLastError() == ERROR_INVALID_FLAGS || GetLastError() == ERROR_INVALID_PARAMETER /* Vista */ ),
 137         "CredDeleteA should have failed with ERROR_INVALID_FLAGS or ERROR_INVALID_PARAMETER instead of %d\n",
 138         GetLastError());
 139 }
 140
 141 static void test_CredReadDomainCredentialsA(void)
 142 {
 143     BOOL ret;
 144     char target_name[] = "no_such_target";
 145     CREDENTIAL_TARGET_INFORMATIONA info = {target_name, NULL, target_name, NULL, NULL, NULL, NULL, 0, 0, NULL};
 146     DWORD count;
 147     PCREDENTIAL* creds;
 148
 149     if (!pCredReadDomainCredentialsA)
 150     {
 151         win_skip("CredReadDomainCredentialsA() is not implemented\n");
 152         return;
 153     }
 154
 155     /* these two tests would crash on both native and Wine. Implementations
 156      * does not check for NULL output pointers and try to zero them out early */
 157 if(0)
 158 {
 159     ok(!pCredReadDomainCredentialsA(&info, 0, NULL, &creds) &&
 160             GetLastError() == ERROR_INVALID_PARAMETER, "!\n");
 161     ok(!pCredReadDomainCredentialsA(&info, 0, &count, NULL) &&
 162             GetLastError() == ERROR_INVALID_PARAMETER, "!\n");
 163 }
 164
 165     SetLastError(0xdeadbeef);
 166     ret = pCredReadDomainCredentialsA(NULL, 0, &count, &creds);
 167     ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
 168         "CredReadDomainCredentialsA should have failed with ERROR_INVALID_PARAMETER instead of %d\n",
 169         GetLastError());
 170
 171     SetLastError(0xdeadbeef);
 172     creds = (void*)0x12345;
 173     count = 2;
 174     ret = pCredReadDomainCredentialsA(&info, 0, &count, &creds);
 175     ok(!ret && GetLastError() == ERROR_NOT_FOUND,
 176         "CredReadDomainCredentialsA should have failed with ERROR_NOT_FOUND instead of %d\n",
 177         GetLastError());
 178     ok(count ==0 && creds == NULL, "CredReadDomainCredentialsA must not return any result\n");
 179
 180     info.TargetName = NULL;
 181
 182     SetLastError(0xdeadbeef);
 183     ret = pCredReadDomainCredentialsA(&info, 0, &count, &creds);
 184     ok(!ret, "CredReadDomainCredentialsA should have failed\n");
 185     ok(GetLastError() == ERROR_NOT_FOUND ||
 186         GetLastError() == ERROR_INVALID_PARAMETER, /* Vista, W2K8 */
 187         "Expected ERROR_NOT_FOUND or ERROR_INVALID_PARAMETER instead of %d\n",
 188         GetLastError());
 189
 190     info.DnsServerName = NULL;
 191
 192     SetLastError(0xdeadbeef);
 193     ret = pCredReadDomainCredentialsA(&info, 0, &count, &creds);
 194     ok(!ret, "CredReadDomainCredentialsA should have failed\n");
 195     ok(GetLastError() == ERROR_NOT_FOUND ||
 196         GetLastError() == ERROR_INVALID_PARAMETER, /* Vista, W2K8 */
 197         "Expected ERROR_NOT_FOUND or ERROR_INVALID_PARAMETER instead of %d\n",
 198         GetLastError());
 199 }
 200
 201 static void check_blob(int line, DWORD cred_type, PCREDENTIALA cred)
 202 {
 203     if (cred_type == CRED_TYPE_DOMAIN_PASSWORD)
 204     {
 205         todo_wine
 206         ok_(__FILE__, line)(cred->CredentialBlobSize == 0, "expected CredentialBlobSize of 0 but got %d\n", cred->CredentialBlobSize);
 207         todo_wine
 208         ok_(__FILE__, line)(!cred->CredentialBlob, "expected NULL credentials but got %p\n", cred->CredentialBlob);
 209     }
 210     else
 211     {
 212         DWORD size=sizeof(TEST_PASSWORD);
 213         ok_(__FILE__, line)(cred->CredentialBlobSize == size, "expected CredentialBlobSize of %u but got %u\n", size, cred->CredentialBlobSize);
 214         ok_(__FILE__, line)(cred->CredentialBlob != NULL, "CredentialBlob should be present\n");
 215         if (cred->CredentialBlob)
 216             ok_(__FILE__, line)(!memcmp(cred->CredentialBlob, TEST_PASSWORD, size), "wrong CredentialBlob\n");
 217     }
 218 }
 219
 220 static void test_generic(void)
 221 {
 222     BOOL ret;
 223     DWORD count, i;
 224     PCREDENTIALA *creds;
 225     CREDENTIALA new_cred;
 226     PCREDENTIALA cred;
 227     BOOL found = FALSE;
 228
 229     new_cred.Flags = 0;
 230     new_cred.Type = CRED_TYPE_GENERIC;
 231     new_cred.TargetName = (char *)TEST_TARGET_NAME;
 232     new_cred.Comment = (char *)"Comment";
 233     new_cred.CredentialBlobSize = sizeof(TEST_PASSWORD);
 234     new_cred.CredentialBlob = (LPBYTE)TEST_PASSWORD;
 235     new_cred.Persist = CRED_PERSIST_ENTERPRISE;
 236     new_cred.AttributeCount = 0;
 237     new_cred.Attributes = NULL;
 238     new_cred.TargetAlias = NULL;
 239     new_cred.UserName = (char *)"winetest";
 240
 241     ret = pCredWriteA(&new_cred, 0);
 242     ok(ret || broken(GetLastError() == ERROR_NO_SUCH_LOGON_SESSION),
 243        "CredWriteA failed with error %d\n", GetLastError());
 244     if (!ret)
 245     {
 246         skip("couldn't write generic credentials, skipping tests\n");
 247         return;
 248     }
 249
 250     ret = pCredEnumerateA(NULL, 0, &count, &creds);
 251     ok(ret, "CredEnumerateA failed with error %d\n", GetLastError());
 252
 253     for (i = 0; i < count; i++)
 254     {
 255         if (!strcmp(creds[i]->TargetName, TEST_TARGET_NAME))
 256         {
 257             ok(creds[i]->Type == CRED_TYPE_GENERIC ||
 258                creds[i]->Type == CRED_TYPE_DOMAIN_PASSWORD, /* Vista */
 259                "expected creds[%d]->Type CRED_TYPE_GENERIC or CRED_TYPE_DOMAIN_PASSWORD but got %d\n", i, creds[i]->Type);
 260             ok(!creds[i]->Flags, "expected creds[%d]->Flags 0 but got 0x%x\n", i, creds[i]->Flags);
 261             ok(!strcmp(creds[i]->Comment, "Comment"), "expected creds[%d]->Comment \"Comment\" but got \"%s\"\n", i, creds[i]->Comment);
 262             check_blob(__LINE__, creds[i]->Type, creds[i]);
 263             ok(creds[i]->Persist, "expected creds[%d]->Persist CRED_PERSIST_ENTERPRISE but got %d\n", i, creds[i]->Persist);
 264             ok(!strcmp(creds[i]->UserName, "winetest"), "expected creds[%d]->UserName \"winetest\" but got \"%s\"\n", i, creds[i]->UserName);
 265             found = TRUE;
 266         }
 267     }
 268     pCredFree(creds);
 269     ok(found, "credentials not found\n");
 270
 271     ret = pCredReadA(TEST_TARGET_NAME, CRED_TYPE_GENERIC, 0, &cred);
 272     ok(ret, "CredReadA failed with error %d\n", GetLastError());
 273     pCredFree(cred);
 274
 275     ret = pCredDeleteA(TEST_TARGET_NAME, CRED_TYPE_GENERIC, 0);
 276     ok(ret, "CredDeleteA failed with error %d\n", GetLastError());
 277 }
 278
 279 static void test_domain_password(DWORD cred_type)
 280 {
 281     BOOL ret;
 282     DWORD count, i;
 283     PCREDENTIALA *creds;
 284     CREDENTIALA new_cred;
 285     PCREDENTIALA cred;
 286     BOOL found = FALSE;
 287
 288     new_cred.Flags = 0;
 289     new_cred.Type = cred_type;
 290     new_cred.TargetName = (char *)TEST_TARGET_NAME;
 291     new_cred.Comment = (char *)"Comment";
 292     new_cred.CredentialBlobSize = sizeof(TEST_PASSWORD);
 293     new_cred.CredentialBlob = (LPBYTE)TEST_PASSWORD;
 294     new_cred.Persist = CRED_PERSIST_ENTERPRISE;
 295     new_cred.AttributeCount = 0;
 296     new_cred.Attributes = NULL;
 297     new_cred.TargetAlias = NULL;
 298     new_cred.UserName = (char *)"test\\winetest";
 299     ret = pCredWriteA(&new_cred, 0);
 300     if (!ret && GetLastError() == ERROR_NO_SUCH_LOGON_SESSION)
 301     {
 302         skip("CRED_TYPE_DOMAIN_PASSWORD credentials are not supported "
 303              "or are disabled. Skipping\n");
 304         return;
 305     }
 306     ok(ret, "CredWriteA failed with error %d\n", GetLastError());
 307
 308     ret = pCredEnumerateA(NULL, 0, &count, &creds);
 309     ok(ret, "CredEnumerateA failed with error %d\n", GetLastError());
 310
 311     for (i = 0; i < count; i++)
 312     {
 313         if (!strcmp(creds[i]->TargetName, TEST_TARGET_NAME))
 314         {
 315             ok(creds[i]->Type == cred_type, "expected creds[%d]->Type CRED_TYPE_DOMAIN_PASSWORD but got %d\n", i, creds[i]->Type);
 316             ok(!creds[i]->Flags, "expected creds[%d]->Flags 0 but got 0x%x\n", i, creds[i]->Flags);
 317             ok(!strcmp(creds[i]->Comment, "Comment"), "expected creds[%d]->Comment \"Comment\" but got \"%s\"\n", i, creds[i]->Comment);
 318             check_blob(__LINE__, cred_type, creds[i]);
 319             ok(creds[i]->Persist, "expected creds[%d]->Persist CRED_PERSIST_ENTERPRISE but got %d\n", i, creds[i]->Persist);
 320             ok(!strcmp(creds[i]->UserName, "test\\winetest"), "expected creds[%d]->UserName \"winetest\" but got \"%s\"\n", i, creds[i]->UserName);
 321             found = TRUE;
 322         }
 323     }
 324     pCredFree(creds);
 325     ok(found, "credentials not found\n");
 326
 327     ret = pCredReadA(TEST_TARGET_NAME, cred_type, 0, &cred);
 328     ok(ret, "CredReadA failed with error %d\n", GetLastError());
 329     if (ret)  /* don't check the values of cred, if CredReadA failed. */
 330     {
 331         check_blob(__LINE__, cred_type, cred);
 332         pCredFree(cred);
 333     }
 334
 335     ret = pCredDeleteA(TEST_TARGET_NAME, cred_type, 0);
 336     ok(ret, "CredDeleteA failed with error %d\n", GetLastError());
 337 }
 338
 339 START_TEST(cred)
 340 {
 341     DWORD persists[CRED_TYPE_MAXIMUM];
 342
 343     pCredEnumerateA = (void *)GetProcAddress(GetModuleHandle("advapi32.dll"), "CredEnumerateA");
 344     pCredFree = (void *)GetProcAddress(GetModuleHandle("advapi32.dll"), "CredFree");
 345     pCredGetSessionTypes = (void *)GetProcAddress(GetModuleHandle("advapi32.dll"), "CredGetSessionTypes");
 346     pCredWriteA = (void *)GetProcAddress(GetModuleHandle("advapi32.dll"), "CredWriteA");
 347     pCredDeleteA = (void *)GetProcAddress(GetModuleHandle("advapi32.dll"), "CredDeleteA");
 348     pCredReadA = (void *)GetProcAddress(GetModuleHandle("advapi32.dll"), "CredReadA");
 349     pCredRenameA = (void *)GetProcAddress(GetModuleHandle("advapi32.dll"), "CredRenameA");
 350     pCredReadDomainCredentialsA = (void *)GetProcAddress(GetModuleHandle("advapi32.dll"), "CredReadDomainCredentialsA");
 351
 352     if (!pCredEnumerateA || !pCredFree || !pCredWriteA || !pCredDeleteA ||
 353         !pCredReadA)
 354     {
 355         win_skip("credentials functions not present in advapi32.dll\n");
 356         return;
 357     }
 358
 359     if (pCredGetSessionTypes)
 360     {
 361         BOOL ret;
 362         DWORD i;
 363         ret = pCredGetSessionTypes(CRED_TYPE_MAXIMUM, persists);
 364         ok(ret, "CredGetSessionTypes failed with error %d\n", GetLastError());
 365         ok(persists[0] == CRED_PERSIST_NONE, "persists[0] = %u instead of CRED_PERSIST_NONE\n", persists[0]);
 366         for (i=0; i < CRED_TYPE_MAXIMUM; i++)
 367             ok(persists[i] <= CRED_PERSIST_ENTERPRISE, "bad value for persists[%u]: %u\n", i, persists[i]);
 368     }
 369     else
 370         memset(persists, CRED_PERSIST_ENTERPRISE, sizeof(persists));
 371
 372     test_CredReadA();
 373     test_CredWriteA();
 374     test_CredDeleteA();
 375
 376     test_CredReadDomainCredentialsA();
 377
 378     trace("generic:\n");
 379     if (persists[CRED_TYPE_GENERIC] == CRED_PERSIST_NONE)
 380         skip("CRED_TYPE_GENERIC credentials are not supported or are disabled. Skipping\n");
 381     else
 382         test_generic();
 383
 384         trace("domain password:\n");
 385     if (persists[CRED_TYPE_DOMAIN_PASSWORD] == CRED_PERSIST_NONE)
 386         skip("CRED_TYPE_DOMAIN_PASSWORD credentials are not supported or are disabled. Skipping\n");
 387     else
 388         test_domain_password(CRED_TYPE_DOMAIN_PASSWORD);
 389
 390     trace("domain visible password:\n");
 391     if (persists[CRED_TYPE_DOMAIN_VISIBLE_PASSWORD] == CRED_PERSIST_NONE)
 392         skip("CRED_TYPE_DOMAIN_VISIBLE_PASSWORD credentials are not supported or are disabled. Skipping\n");
 393     else
 394         test_domain_password(CRED_TYPE_DOMAIN_VISIBLE_PASSWORD);
 395 }