| /* |
| * 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 ); |
| } |