mirror of
https://github.com/reactos/reactos.git
synced 2024-11-05 22:26:39 +00:00
218 lines
10 KiB
C
218 lines
10 KiB
C
/*
|
|
* Credentials User Interface Tests
|
|
*
|
|
* Copyright 2007 Robert Shearman for CodeWeavers
|
|
*
|
|
* 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 "windef.h"
|
|
#include "winbase.h"
|
|
#include "wincred.h"
|
|
#include "sspi.h"
|
|
|
|
#include "wine/test.h"
|
|
|
|
static SECURITY_STATUS (SEC_ENTRY *pSspiEncodeAuthIdentityAsStrings)
|
|
(PSEC_WINNT_AUTH_IDENTITY_OPAQUE,PCWSTR*,PCWSTR*,PCWSTR*);
|
|
static void (SEC_ENTRY *pSspiFreeAuthIdentity)
|
|
(PSEC_WINNT_AUTH_IDENTITY_OPAQUE);
|
|
static ULONG (SEC_ENTRY *pSspiPromptForCredentialsW)
|
|
(PCWSTR,void*,ULONG,PCWSTR,PSEC_WINNT_AUTH_IDENTITY_OPAQUE,PSEC_WINNT_AUTH_IDENTITY_OPAQUE*,int*,ULONG);
|
|
|
|
static void test_CredUIPromptForCredentials(void)
|
|
{
|
|
static const WCHAR wszServerName[] = {'W','i','n','e','T','e','s','t',0};
|
|
DWORD ret;
|
|
WCHAR username[256];
|
|
WCHAR password[256];
|
|
CREDUI_INFOW credui_info;
|
|
BOOL save = FALSE;
|
|
|
|
credui_info.cbSize = sizeof(credui_info);
|
|
credui_info.hwndParent = NULL;
|
|
credui_info.pszMessageText = NULL;
|
|
credui_info.hbmBanner = NULL;
|
|
|
|
ret = CredUIConfirmCredentialsW(NULL, TRUE);
|
|
ok(ret == ERROR_INVALID_PARAMETER /* 2003 + */ || ret == ERROR_NOT_FOUND /* XP */,
|
|
"CredUIConfirmCredentials should have returned ERROR_INVALID_PARAMETER or ERROR_NOT_FOUND instead of %d\n", ret);
|
|
|
|
ret = CredUIConfirmCredentialsW(wszServerName, TRUE);
|
|
ok(ret == ERROR_NOT_FOUND, "CredUIConfirmCredentials should have returned ERROR_NOT_FOUND instead of %d\n", ret);
|
|
|
|
username[0] = '\0';
|
|
password[0] = '\0';
|
|
ret = CredUIPromptForCredentialsW(NULL, NULL, NULL, 0, username,
|
|
sizeof(username)/sizeof(username[0]),
|
|
password, sizeof(password)/sizeof(password[0]),
|
|
NULL, CREDUI_FLAGS_ALWAYS_SHOW_UI);
|
|
ok(ret == ERROR_INVALID_FLAGS, "CredUIPromptForCredentials should have returned ERROR_INVALID_FLAGS instead of %d\n", ret);
|
|
|
|
ret = CredUIPromptForCredentialsW(NULL, NULL, NULL, 0, username,
|
|
sizeof(username)/sizeof(username[0]),
|
|
password, sizeof(password)/sizeof(password[0]),
|
|
NULL, CREDUI_FLAGS_ALWAYS_SHOW_UI | CREDUI_FLAGS_GENERIC_CREDENTIALS);
|
|
ok(ret == ERROR_INVALID_PARAMETER, "CredUIPromptForCredentials should have returned ERROR_INVALID_PARAMETER instead of %d\n", ret);
|
|
|
|
ret = CredUIPromptForCredentialsW(NULL, wszServerName, NULL, 0, username,
|
|
sizeof(username)/sizeof(username[0]),
|
|
password, sizeof(password)/sizeof(password[0]),
|
|
NULL, CREDUI_FLAGS_SHOW_SAVE_CHECK_BOX);
|
|
ok(ret == ERROR_INVALID_PARAMETER, "CredUIPromptForCredentials should have returned ERROR_INVALID_PARAMETER instead of %d\n", ret);
|
|
|
|
if (winetest_interactive)
|
|
{
|
|
static const WCHAR wszCaption1[] = {'C','R','E','D','U','I','_','F','L','A','G','S','_','E','X','P','E','C','T','_','C','O','N','F','I','R','M','A','T','I','O','N',0};
|
|
static const WCHAR wszCaption2[] = {'C','R','E','D','U','I','_','F','L','A','G','S','_','I','N','C','O','R','R','E','C','T','_','P','A','S','S','W','O','R','D','|',
|
|
'C','R','E','D','U','I','_','F','L','A','G','S','_','E','X','P','E','C','T','_','C','O','N','F','I','R','M','A','T','I','O','N',0};
|
|
static const WCHAR wszCaption3[] = {'C','R','E','D','U','I','_','F','L','A','G','S','_','D','O','_','N','O','T','_','P','E','R','S','I','S','T','|',
|
|
'C','R','E','D','U','I','_','F','L','A','G','S','_','E','X','P','E','C','T','_','C','O','N','F','I','R','M','A','T','I','O','N',0};
|
|
static const WCHAR wszCaption4[] = {'C','R','E','D','U','I','_','F','L','A','G','S','_','P','E','R','S','I','S','T','|',
|
|
'C','R','E','D','U','I','_','F','L','A','G','S','_','E','X','P','E','C','T','_','C','O','N','F','I','R','M','A','T','I','O','N',0};
|
|
|
|
ret = CredUIPromptForCredentialsW(NULL, wszServerName, NULL, 0, username,
|
|
sizeof(username)/sizeof(username[0]),
|
|
password, sizeof(password)/sizeof(password[0]),
|
|
&save, CREDUI_FLAGS_EXPECT_CONFIRMATION);
|
|
ok(ret == ERROR_SUCCESS || ret == ERROR_CANCELLED, "CredUIPromptForCredentials failed with error %d\n", ret);
|
|
if (ret == ERROR_SUCCESS)
|
|
{
|
|
ret = CredUIConfirmCredentialsW(wszServerName, FALSE);
|
|
ok(ret == ERROR_SUCCESS, "CredUIConfirmCredentials failed with error %d\n", ret);
|
|
}
|
|
|
|
credui_info.pszCaptionText = wszCaption1;
|
|
ret = CredUIPromptForCredentialsW(&credui_info, wszServerName, NULL,
|
|
ERROR_ACCESS_DENIED,
|
|
username, sizeof(username)/sizeof(username[0]),
|
|
password, sizeof(password)/sizeof(password[0]),
|
|
&save, CREDUI_FLAGS_EXPECT_CONFIRMATION);
|
|
ok(ret == ERROR_SUCCESS || ret == ERROR_CANCELLED, "CredUIPromptForCredentials failed with error %d\n", ret);
|
|
if (ret == ERROR_SUCCESS)
|
|
{
|
|
ret = CredUIConfirmCredentialsW(wszServerName, FALSE);
|
|
ok(ret == ERROR_SUCCESS, "CredUIConfirmCredentials failed with error %d\n", ret);
|
|
}
|
|
|
|
credui_info.pszCaptionText = wszCaption2;
|
|
ret = CredUIPromptForCredentialsW(&credui_info, wszServerName, NULL, 0,
|
|
username, sizeof(username)/sizeof(username[0]),
|
|
password, sizeof(password)/sizeof(password[0]),
|
|
NULL, CREDUI_FLAGS_INCORRECT_PASSWORD|CREDUI_FLAGS_EXPECT_CONFIRMATION);
|
|
ok(ret == ERROR_SUCCESS || ret == ERROR_CANCELLED, "CredUIPromptForCredentials failed with error %d\n", ret);
|
|
if (ret == ERROR_SUCCESS)
|
|
{
|
|
ret = CredUIConfirmCredentialsW(wszServerName, FALSE);
|
|
ok(ret == ERROR_SUCCESS, "CredUIConfirmCredentials failed with error %d\n", ret);
|
|
}
|
|
|
|
|
|
save = TRUE;
|
|
credui_info.pszCaptionText = wszCaption3;
|
|
ret = CredUIPromptForCredentialsW(&credui_info, wszServerName, NULL, 0,
|
|
username, sizeof(username)/sizeof(username[0]),
|
|
password, sizeof(password)/sizeof(password[0]),
|
|
&save, CREDUI_FLAGS_DO_NOT_PERSIST|CREDUI_FLAGS_EXPECT_CONFIRMATION);
|
|
ok(ret == ERROR_SUCCESS || ret == ERROR_CANCELLED, "CredUIPromptForCredentials failed with error %d\n", ret);
|
|
ok(save, "save flag should have been untouched\n");
|
|
|
|
save = FALSE;
|
|
credui_info.pszCaptionText = wszCaption4;
|
|
ret = CredUIPromptForCredentialsW(&credui_info, wszServerName, NULL, 0,
|
|
username, sizeof(username)/sizeof(username[0]),
|
|
password, sizeof(password)/sizeof(password[0]),
|
|
&save, CREDUI_FLAGS_PERSIST|CREDUI_FLAGS_EXPECT_CONFIRMATION);
|
|
ok(ret == ERROR_SUCCESS || ret == ERROR_CANCELLED, "CredUIPromptForCredentials failed with error %d\n", ret);
|
|
ok(!save, "save flag should have been untouched\n");
|
|
if (ret == ERROR_SUCCESS)
|
|
{
|
|
ret = CredUIConfirmCredentialsW(wszServerName, FALSE);
|
|
ok(ret == ERROR_SUCCESS, "CredUIConfirmCredentials failed with error %d\n", ret);
|
|
}
|
|
|
|
}
|
|
}
|
|
|
|
static void test_SspiPromptForCredentials(void)
|
|
{
|
|
static const WCHAR targetW[] = {'S','s','p','i','T','e','s','t',0};
|
|
static const WCHAR basicW[] = {'b','a','s','i','c',0};
|
|
ULONG ret;
|
|
SECURITY_STATUS status;
|
|
CREDUI_INFOW info;
|
|
PSEC_WINNT_AUTH_IDENTITY_OPAQUE id;
|
|
const WCHAR *username, *domain, *creds;
|
|
int save;
|
|
|
|
if (!pSspiPromptForCredentialsW || !pSspiFreeAuthIdentity)
|
|
{
|
|
win_skip( "SspiPromptForCredentialsW is missing\n" );
|
|
return;
|
|
}
|
|
|
|
info.cbSize = sizeof(info);
|
|
info.hwndParent = NULL;
|
|
info.pszMessageText = targetW;
|
|
info.pszCaptionText = basicW;
|
|
info.hbmBanner = NULL;
|
|
ret = pSspiPromptForCredentialsW( NULL, &info, 0, basicW, NULL, &id, &save, 0 );
|
|
ok( ret == ERROR_INVALID_PARAMETER, "got %u\n", ret );
|
|
|
|
ret = pSspiPromptForCredentialsW( targetW, &info, 0, NULL, NULL, &id, &save, 0 );
|
|
ok( ret == ERROR_NO_SUCH_PACKAGE, "got %u\n", ret );
|
|
|
|
if (winetest_interactive)
|
|
{
|
|
id = NULL;
|
|
save = -1;
|
|
ret = pSspiPromptForCredentialsW( targetW, &info, 0, basicW, NULL, &id, &save, 0 );
|
|
ok( ret == ERROR_SUCCESS || ret == ERROR_CANCELLED, "got %u\n", ret );
|
|
if (ret == ERROR_SUCCESS)
|
|
{
|
|
ok( id != NULL, "id not set\n" );
|
|
ok( save == TRUE || save == FALSE, "got %d\n", save );
|
|
|
|
username = creds = NULL;
|
|
domain = (const WCHAR *)0xdeadbeef;
|
|
status = pSspiEncodeAuthIdentityAsStrings( id, &username, &domain, &creds );
|
|
ok( status == SEC_E_OK, "got %u\n", status );
|
|
ok( username != NULL, "username not set\n" );
|
|
ok( domain == NULL, "domain not set\n" );
|
|
ok( creds != NULL, "creds not set\n" );
|
|
pSspiFreeAuthIdentity( id );
|
|
}
|
|
}
|
|
}
|
|
|
|
START_TEST(credui)
|
|
{
|
|
HMODULE hcredui = GetModuleHandleA( "credui.dll" ), hsecur32 = LoadLibraryA( "secur32.dll" );
|
|
|
|
if (hcredui)
|
|
pSspiPromptForCredentialsW = (void *)GetProcAddress( hcredui, "SspiPromptForCredentialsW" );
|
|
if (hsecur32)
|
|
{
|
|
pSspiEncodeAuthIdentityAsStrings = (void *)GetProcAddress( hsecur32, "SspiEncodeAuthIdentityAsStrings" );
|
|
pSspiFreeAuthIdentity = (void *)GetProcAddress( hsecur32, "SspiFreeAuthIdentity" );
|
|
}
|
|
|
|
test_CredUIPromptForCredentials();
|
|
test_SspiPromptForCredentials();
|
|
|
|
FreeLibrary( hsecur32 );
|
|
}
|