mirror of
https://github.com/reactos/reactos.git
synced 2024-08-26 05:08:39 +00:00
325 lines
12 KiB
C
325 lines
12 KiB
C
/*
|
|
* Credential Function Tests
|
|
*
|
|
* Copyright 2007 Robert Shearman
|
|
*
|
|
* This library is free software; you can redistribute it and/or
|
|
* modify it under the terms of the GNU Lesser General Public
|
|
* License as published by the Free Software Foundation; either
|
|
* version 2.1 of the License, or (at your option) any later version.
|
|
*
|
|
* This library is distributed in the hope that it will be useful,
|
|
* but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
|
|
* Lesser General Public License for more details.
|
|
*
|
|
* You should have received a copy of the GNU Lesser General Public
|
|
* License along with this library; if not, write to the Free Software
|
|
* Foundation, Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301, USA
|
|
*/
|
|
|
|
#include <stdarg.h>
|
|
#include <stdio.h>
|
|
|
|
#include "windef.h"
|
|
#include "winbase.h"
|
|
#include "wincred.h"
|
|
|
|
#include "wine/test.h"
|
|
|
|
static BOOL (WINAPI *pCredDeleteA)(LPCSTR,DWORD,DWORD);
|
|
static BOOL (WINAPI *pCredEnumerateA)(LPCSTR,DWORD,DWORD *,PCREDENTIALA **);
|
|
static VOID (WINAPI *pCredFree)(PVOID);
|
|
static BOOL (WINAPI *pCredGetSessionTypes)(DWORD,LPDWORD);
|
|
static BOOL (WINAPI *pCredReadA)(LPCSTR,DWORD,DWORD,PCREDENTIALA *);
|
|
static BOOL (WINAPI *pCredRenameA)(LPCSTR,LPCSTR,DWORD,DWORD);
|
|
static BOOL (WINAPI *pCredWriteA)(PCREDENTIALA,DWORD);
|
|
|
|
#define TEST_TARGET_NAME "credtest.winehq.org"
|
|
#define TEST_TARGET_NAME2 "credtest2.winehq.org"
|
|
static const WCHAR TEST_PASSWORD[] = {'p','4','$','$','w','0','r','d','!',0};
|
|
|
|
static void test_CredReadA(void)
|
|
{
|
|
BOOL ret;
|
|
PCREDENTIALA cred;
|
|
|
|
SetLastError(0xdeadbeef);
|
|
ret = pCredReadA(TEST_TARGET_NAME, -1, 0, &cred);
|
|
ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
|
|
"CredReadA should have failed with ERROR_INVALID_PARAMETER instead of %d\n",
|
|
GetLastError());
|
|
|
|
SetLastError(0xdeadbeef);
|
|
ret = pCredReadA(TEST_TARGET_NAME, CRED_TYPE_GENERIC, 0xdeadbeef, &cred);
|
|
ok(!ret && ( GetLastError() == ERROR_INVALID_FLAGS || GetLastError() == ERROR_INVALID_PARAMETER ),
|
|
"CredReadA should have failed with ERROR_INVALID_FLAGS or ERROR_INVALID_PARAMETER instead of %d\n",
|
|
GetLastError());
|
|
|
|
SetLastError(0xdeadbeef);
|
|
ret = pCredReadA(NULL, CRED_TYPE_GENERIC, 0, &cred);
|
|
ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
|
|
"CredReadA should have failed with ERROR_INVALID_PARAMETER instead of %d\n",
|
|
GetLastError());
|
|
}
|
|
|
|
static void test_CredWriteA(void)
|
|
{
|
|
CREDENTIALA new_cred;
|
|
BOOL ret;
|
|
|
|
SetLastError(0xdeadbeef);
|
|
ret = pCredWriteA(NULL, 0);
|
|
ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
|
|
"CredWriteA should have failed with ERROR_INVALID_PARAMETER instead of %d\n",
|
|
GetLastError());
|
|
|
|
new_cred.Flags = 0;
|
|
new_cred.Type = CRED_TYPE_GENERIC;
|
|
new_cred.TargetName = NULL;
|
|
new_cred.Comment = (char *)"Comment";
|
|
new_cred.CredentialBlobSize = 0;
|
|
new_cred.CredentialBlob = NULL;
|
|
new_cred.Persist = CRED_PERSIST_ENTERPRISE;
|
|
new_cred.AttributeCount = 0;
|
|
new_cred.Attributes = NULL;
|
|
new_cred.TargetAlias = NULL;
|
|
new_cred.UserName = (char *)"winetest";
|
|
|
|
SetLastError(0xdeadbeef);
|
|
ret = pCredWriteA(&new_cred, 0);
|
|
ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
|
|
"CredWriteA should have failed with ERROR_INVALID_PARAMETER instead of %d\n",
|
|
GetLastError());
|
|
|
|
new_cred.TargetName = (char *)TEST_TARGET_NAME;
|
|
new_cred.Type = CRED_TYPE_DOMAIN_PASSWORD;
|
|
|
|
SetLastError(0xdeadbeef);
|
|
ret = pCredWriteA(&new_cred, 0);
|
|
if (ret)
|
|
{
|
|
/* Vista */
|
|
ok(GetLastError() == ERROR_IO_PENDING,
|
|
"Expected ERROR_IO_PENDING, got %d\n", GetLastError());
|
|
}
|
|
else
|
|
{
|
|
ok(GetLastError() == ERROR_BAD_USERNAME ||
|
|
GetLastError() == ERROR_NO_SUCH_LOGON_SESSION, /* Vista */
|
|
"CredWrite with username without domain should return ERROR_BAD_USERNAME"
|
|
"or ERROR_NO_SUCH_LOGON_SESSION not %d\n", GetLastError());
|
|
}
|
|
|
|
new_cred.UserName = NULL;
|
|
SetLastError(0xdeadbeef);
|
|
ret = pCredWriteA(&new_cred, 0);
|
|
ok(!ret && GetLastError() == ERROR_BAD_USERNAME,
|
|
"CredWriteA with NULL username should have failed with ERROR_BAD_USERNAME instead of %d\n",
|
|
GetLastError());
|
|
}
|
|
|
|
static void test_CredDeleteA(void)
|
|
{
|
|
BOOL ret;
|
|
|
|
SetLastError(0xdeadbeef);
|
|
ret = pCredDeleteA(TEST_TARGET_NAME, -1, 0);
|
|
ok(!ret && GetLastError() == ERROR_INVALID_PARAMETER,
|
|
"CredDeleteA should have failed with ERROR_INVALID_PARAMETER instead of %d\n",
|
|
GetLastError());
|
|
|
|
SetLastError(0xdeadbeef);
|
|
ret = pCredDeleteA(TEST_TARGET_NAME, CRED_TYPE_GENERIC, 0xdeadbeef);
|
|
ok(!ret && ( GetLastError() == ERROR_INVALID_FLAGS || GetLastError() == ERROR_INVALID_PARAMETER /* Vista */ ),
|
|
"CredDeleteA should have failed with ERROR_INVALID_FLAGS or ERROR_INVALID_PARAMETER instead of %d\n",
|
|
GetLastError());
|
|
}
|
|
|
|
static void check_blob(int line, DWORD cred_type, PCREDENTIALA cred)
|
|
{
|
|
if (cred_type == CRED_TYPE_DOMAIN_PASSWORD)
|
|
{
|
|
todo_wine
|
|
ok_(__FILE__, line)(cred->CredentialBlobSize == 0, "expected CredentialBlobSize of 0 but got %d\n", cred->CredentialBlobSize);
|
|
todo_wine
|
|
ok_(__FILE__, line)(!cred->CredentialBlob, "expected NULL credentials but got %p\n", cred->CredentialBlob);
|
|
}
|
|
else
|
|
{
|
|
DWORD size=sizeof(TEST_PASSWORD);
|
|
ok_(__FILE__, line)(cred->CredentialBlobSize == size, "expected CredentialBlobSize of %u but got %u\n", size, cred->CredentialBlobSize);
|
|
ok_(__FILE__, line)(cred->CredentialBlob != NULL, "CredentialBlob should be present\n");
|
|
if (cred->CredentialBlob)
|
|
ok_(__FILE__, line)(!memcmp(cred->CredentialBlob, TEST_PASSWORD, size), "wrong CredentialBlob\n");
|
|
}
|
|
}
|
|
|
|
static void test_generic(void)
|
|
{
|
|
BOOL ret;
|
|
DWORD count, i;
|
|
PCREDENTIALA *creds;
|
|
CREDENTIALA new_cred;
|
|
PCREDENTIALA cred;
|
|
BOOL found = FALSE;
|
|
|
|
new_cred.Flags = 0;
|
|
new_cred.Type = CRED_TYPE_GENERIC;
|
|
new_cred.TargetName = (char *)TEST_TARGET_NAME;
|
|
new_cred.Comment = (char *)"Comment";
|
|
new_cred.CredentialBlobSize = sizeof(TEST_PASSWORD);
|
|
new_cred.CredentialBlob = (LPBYTE)TEST_PASSWORD;
|
|
new_cred.Persist = CRED_PERSIST_ENTERPRISE;
|
|
new_cred.AttributeCount = 0;
|
|
new_cred.Attributes = NULL;
|
|
new_cred.TargetAlias = NULL;
|
|
new_cred.UserName = (char *)"winetest";
|
|
|
|
ret = pCredWriteA(&new_cred, 0);
|
|
ok(ret, "CredWriteA failed with error %d\n", GetLastError());
|
|
|
|
ret = pCredEnumerateA(NULL, 0, &count, &creds);
|
|
ok(ret, "CredEnumerateA failed with error %d\n", GetLastError());
|
|
|
|
for (i = 0; i < count; i++)
|
|
{
|
|
if (!strcmp(creds[i]->TargetName, TEST_TARGET_NAME))
|
|
{
|
|
ok(creds[i]->Type == CRED_TYPE_GENERIC ||
|
|
creds[i]->Type == CRED_TYPE_DOMAIN_PASSWORD, /* Vista */
|
|
"expected creds[%d]->Type CRED_TYPE_GENERIC or CRED_TYPE_DOMAIN_PASSWORD but got %d\n", i, creds[i]->Type);
|
|
ok(!creds[i]->Flags, "expected creds[%d]->Flags 0 but got 0x%x\n", i, creds[i]->Flags);
|
|
ok(!strcmp(creds[i]->Comment, "Comment"), "expected creds[%d]->Comment \"Comment\" but got \"%s\"\n", i, creds[i]->Comment);
|
|
check_blob(__LINE__, creds[i]->Type, creds[i]);
|
|
ok(creds[i]->Persist, "expected creds[%d]->Persist CRED_PERSIST_ENTERPRISE but got %d\n", i, creds[i]->Persist);
|
|
ok(!strcmp(creds[i]->UserName, "winetest"), "expected creds[%d]->UserName \"winetest\" but got \"%s\"\n", i, creds[i]->UserName);
|
|
found = TRUE;
|
|
}
|
|
}
|
|
pCredFree(creds);
|
|
ok(found, "credentials not found\n");
|
|
|
|
ret = pCredReadA(TEST_TARGET_NAME, CRED_TYPE_GENERIC, 0, &cred);
|
|
ok(ret, "CredReadA failed with error %d\n", GetLastError());
|
|
pCredFree(cred);
|
|
|
|
ret = pCredDeleteA(TEST_TARGET_NAME, CRED_TYPE_GENERIC, 0);
|
|
ok(ret, "CredDeleteA failed with error %d\n", GetLastError());
|
|
}
|
|
|
|
static void test_domain_password(DWORD cred_type)
|
|
{
|
|
BOOL ret;
|
|
DWORD count, i;
|
|
PCREDENTIALA *creds;
|
|
CREDENTIALA new_cred;
|
|
PCREDENTIALA cred;
|
|
BOOL found = FALSE;
|
|
|
|
new_cred.Flags = 0;
|
|
new_cred.Type = cred_type;
|
|
new_cred.TargetName = (char *)TEST_TARGET_NAME;
|
|
new_cred.Comment = (char *)"Comment";
|
|
new_cred.CredentialBlobSize = sizeof(TEST_PASSWORD);
|
|
new_cred.CredentialBlob = (LPBYTE)TEST_PASSWORD;
|
|
new_cred.Persist = CRED_PERSIST_ENTERPRISE;
|
|
new_cred.AttributeCount = 0;
|
|
new_cred.Attributes = NULL;
|
|
new_cred.TargetAlias = NULL;
|
|
new_cred.UserName = (char *)"test\\winetest";
|
|
ret = pCredWriteA(&new_cred, 0);
|
|
if (!ret && GetLastError() == ERROR_NO_SUCH_LOGON_SESSION)
|
|
{
|
|
skip("CRED_TYPE_DOMAIN_PASSWORD credentials are not supported "
|
|
"or are disabled. Skipping\n");
|
|
return;
|
|
}
|
|
ok(ret, "CredWriteA failed with error %d\n", GetLastError());
|
|
|
|
ret = pCredEnumerateA(NULL, 0, &count, &creds);
|
|
ok(ret, "CredEnumerateA failed with error %d\n", GetLastError());
|
|
|
|
for (i = 0; i < count; i++)
|
|
{
|
|
if (!strcmp(creds[i]->TargetName, TEST_TARGET_NAME))
|
|
{
|
|
ok(creds[i]->Type == cred_type, "expected creds[%d]->Type CRED_TYPE_DOMAIN_PASSWORD but got %d\n", i, creds[i]->Type);
|
|
ok(!creds[i]->Flags, "expected creds[%d]->Flags 0 but got 0x%x\n", i, creds[i]->Flags);
|
|
ok(!strcmp(creds[i]->Comment, "Comment"), "expected creds[%d]->Comment \"Comment\" but got \"%s\"\n", i, creds[i]->Comment);
|
|
check_blob(__LINE__, cred_type, creds[i]);
|
|
ok(creds[i]->Persist, "expected creds[%d]->Persist CRED_PERSIST_ENTERPRISE but got %d\n", i, creds[i]->Persist);
|
|
ok(!strcmp(creds[i]->UserName, "test\\winetest"), "expected creds[%d]->UserName \"winetest\" but got \"%s\"\n", i, creds[i]->UserName);
|
|
found = TRUE;
|
|
}
|
|
}
|
|
pCredFree(creds);
|
|
ok(found, "credentials not found\n");
|
|
|
|
ret = pCredReadA(TEST_TARGET_NAME, cred_type, 0, &cred);
|
|
ok(ret, "CredReadA failed with error %d\n", GetLastError());
|
|
if (ret) /* don't check the values of cred, if CredReadA failed. */
|
|
{
|
|
check_blob(__LINE__, cred_type, cred);
|
|
pCredFree(cred);
|
|
}
|
|
|
|
ret = pCredDeleteA(TEST_TARGET_NAME, cred_type, 0);
|
|
ok(ret, "CredDeleteA failed with error %d\n", GetLastError());
|
|
}
|
|
|
|
START_TEST(cred)
|
|
{
|
|
DWORD persists[CRED_TYPE_MAXIMUM];
|
|
|
|
pCredEnumerateA = (void *)GetProcAddress(GetModuleHandle("advapi32.dll"), "CredEnumerateA");
|
|
pCredFree = (void *)GetProcAddress(GetModuleHandle("advapi32.dll"), "CredFree");
|
|
pCredGetSessionTypes = (void *)GetProcAddress(GetModuleHandle("advapi32.dll"), "CredGetSessionTypes");
|
|
pCredWriteA = (void *)GetProcAddress(GetModuleHandle("advapi32.dll"), "CredWriteA");
|
|
pCredDeleteA = (void *)GetProcAddress(GetModuleHandle("advapi32.dll"), "CredDeleteA");
|
|
pCredReadA = (void *)GetProcAddress(GetModuleHandle("advapi32.dll"), "CredReadA");
|
|
pCredRenameA = (void *)GetProcAddress(GetModuleHandle("advapi32.dll"), "CredRenameA");
|
|
|
|
if (!pCredEnumerateA || !pCredFree || !pCredWriteA || !pCredDeleteA ||
|
|
!pCredReadA)
|
|
{
|
|
skip("credentials functions not present in advapi32.dll\n");
|
|
return;
|
|
}
|
|
|
|
if (pCredGetSessionTypes)
|
|
{
|
|
BOOL ret;
|
|
DWORD i;
|
|
ret = pCredGetSessionTypes(CRED_TYPE_MAXIMUM, persists);
|
|
ok(ret, "CredGetSessionTypes failed with error %d\n", GetLastError());
|
|
ok(persists[0] == CRED_PERSIST_NONE, "persists[0] = %u instead of CRED_PERSIST_NONE\n", persists[0]);
|
|
for (i=0; i < CRED_TYPE_MAXIMUM; i++)
|
|
ok(persists[i] <= CRED_PERSIST_ENTERPRISE, "bad value for persists[%u]: %u\n", i, persists[i]);
|
|
}
|
|
else
|
|
memset(persists, CRED_PERSIST_ENTERPRISE, sizeof(persists));
|
|
|
|
test_CredReadA();
|
|
test_CredWriteA();
|
|
test_CredDeleteA();
|
|
|
|
trace("generic:\n");
|
|
if (persists[CRED_TYPE_GENERIC] == CRED_PERSIST_NONE)
|
|
skip("CRED_TYPE_GENERIC credentials are not supported or are disabled. Skipping\n");
|
|
else
|
|
test_generic();
|
|
|
|
trace("domain password:\n");
|
|
if (persists[CRED_TYPE_DOMAIN_PASSWORD] == CRED_PERSIST_NONE)
|
|
skip("CRED_TYPE_DOMAIN_PASSWORD credentials are not supported or are disabled. Skipping\n");
|
|
else
|
|
test_domain_password(CRED_TYPE_DOMAIN_PASSWORD);
|
|
|
|
trace("domain visible password:\n");
|
|
if (persists[CRED_TYPE_DOMAIN_VISIBLE_PASSWORD] == CRED_PERSIST_NONE)
|
|
skip("CRED_TYPE_DOMAIN_VISIBLE_PASSWORD credentials are not supported or are disabled. Skipping\n");
|
|
else
|
|
test_domain_password(CRED_TYPE_DOMAIN_VISIBLE_PASSWORD);
|
|
}
|