1 /* 2 * Credentials User Interface Tests 3 * 4 * Copyright 2007 Robert Shearman for CodeWeavers 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 23 #include "windef.h" 24 #include "winbase.h" 25 #include "wincred.h" 26 #include "sspi.h" 27 28 #include "wine/test.h" 29 30 static SECURITY_STATUS (SEC_ENTRY *pSspiEncodeAuthIdentityAsStrings) 31 (PSEC_WINNT_AUTH_IDENTITY_OPAQUE,PCWSTR*,PCWSTR*,PCWSTR*); 32 static void (SEC_ENTRY *pSspiFreeAuthIdentity) 33 (PSEC_WINNT_AUTH_IDENTITY_OPAQUE); 34 static ULONG (SEC_ENTRY *pSspiPromptForCredentialsW) 35 (PCWSTR,void*,ULONG,PCWSTR,PSEC_WINNT_AUTH_IDENTITY_OPAQUE,PSEC_WINNT_AUTH_IDENTITY_OPAQUE*,int*,ULONG); 36 37 static void test_CredUIPromptForCredentials(void) 38 { 39 static const WCHAR wszServerName[] = {'W','i','n','e','T','e','s','t',0}; 40 DWORD ret; 41 WCHAR username[256]; 42 WCHAR password[256]; 43 CREDUI_INFOW credui_info; 44 BOOL save = FALSE; 45 46 credui_info.cbSize = sizeof(credui_info); 47 credui_info.hwndParent = NULL; 48 credui_info.pszMessageText = NULL; 49 credui_info.hbmBanner = NULL; 50 51 ret = CredUIConfirmCredentialsW(NULL, TRUE); 52 ok(ret == ERROR_INVALID_PARAMETER /* 2003 + */ || ret == ERROR_NOT_FOUND /* XP */, 53 "CredUIConfirmCredentials should have returned ERROR_INVALID_PARAMETER or ERROR_NOT_FOUND instead of %d\n", ret); 54 55 ret = CredUIConfirmCredentialsW(wszServerName, TRUE); 56 ok(ret == ERROR_NOT_FOUND, "CredUIConfirmCredentials should have returned ERROR_NOT_FOUND instead of %d\n", ret); 57 58 username[0] = '\0'; 59 password[0] = '\0'; 60 ret = CredUIPromptForCredentialsW(NULL, NULL, NULL, 0, username, 61 sizeof(username)/sizeof(username[0]), 62 password, sizeof(password)/sizeof(password[0]), 63 NULL, CREDUI_FLAGS_ALWAYS_SHOW_UI); 64 ok(ret == ERROR_INVALID_FLAGS, "CredUIPromptForCredentials should have returned ERROR_INVALID_FLAGS instead of %d\n", ret); 65 66 ret = CredUIPromptForCredentialsW(NULL, NULL, NULL, 0, username, 67 sizeof(username)/sizeof(username[0]), 68 password, sizeof(password)/sizeof(password[0]), 69 NULL, CREDUI_FLAGS_ALWAYS_SHOW_UI | CREDUI_FLAGS_GENERIC_CREDENTIALS); 70 ok(ret == ERROR_INVALID_PARAMETER, "CredUIPromptForCredentials should have returned ERROR_INVALID_PARAMETER instead of %d\n", ret); 71 72 ret = CredUIPromptForCredentialsW(NULL, wszServerName, NULL, 0, username, 73 sizeof(username)/sizeof(username[0]), 74 password, sizeof(password)/sizeof(password[0]), 75 NULL, CREDUI_FLAGS_SHOW_SAVE_CHECK_BOX); 76 ok(ret == ERROR_INVALID_PARAMETER, "CredUIPromptForCredentials should have returned ERROR_INVALID_PARAMETER instead of %d\n", ret); 77 78 if (winetest_interactive) 79 { 80 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}; 81 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','|', 82 '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}; 83 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','|', 84 '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}; 85 static const WCHAR wszCaption4[] = {'C','R','E','D','U','I','_','F','L','A','G','S','_','P','E','R','S','I','S','T','|', 86 '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}; 87 88 ret = CredUIPromptForCredentialsW(NULL, wszServerName, NULL, 0, username, 89 sizeof(username)/sizeof(username[0]), 90 password, sizeof(password)/sizeof(password[0]), 91 &save, CREDUI_FLAGS_EXPECT_CONFIRMATION); 92 ok(ret == ERROR_SUCCESS || ret == ERROR_CANCELLED, "CredUIPromptForCredentials failed with error %d\n", ret); 93 if (ret == ERROR_SUCCESS) 94 { 95 ret = CredUIConfirmCredentialsW(wszServerName, FALSE); 96 ok(ret == ERROR_SUCCESS, "CredUIConfirmCredentials failed with error %d\n", ret); 97 } 98 99 credui_info.pszCaptionText = wszCaption1; 100 ret = CredUIPromptForCredentialsW(&credui_info, wszServerName, NULL, 101 ERROR_ACCESS_DENIED, 102 username, sizeof(username)/sizeof(username[0]), 103 password, sizeof(password)/sizeof(password[0]), 104 &save, CREDUI_FLAGS_EXPECT_CONFIRMATION); 105 ok(ret == ERROR_SUCCESS || ret == ERROR_CANCELLED, "CredUIPromptForCredentials failed with error %d\n", ret); 106 if (ret == ERROR_SUCCESS) 107 { 108 ret = CredUIConfirmCredentialsW(wszServerName, FALSE); 109 ok(ret == ERROR_SUCCESS, "CredUIConfirmCredentials failed with error %d\n", ret); 110 } 111 112 credui_info.pszCaptionText = wszCaption2; 113 ret = CredUIPromptForCredentialsW(&credui_info, wszServerName, NULL, 0, 114 username, sizeof(username)/sizeof(username[0]), 115 password, sizeof(password)/sizeof(password[0]), 116 NULL, CREDUI_FLAGS_INCORRECT_PASSWORD|CREDUI_FLAGS_EXPECT_CONFIRMATION); 117 ok(ret == ERROR_SUCCESS || ret == ERROR_CANCELLED, "CredUIPromptForCredentials failed with error %d\n", ret); 118 if (ret == ERROR_SUCCESS) 119 { 120 ret = CredUIConfirmCredentialsW(wszServerName, FALSE); 121 ok(ret == ERROR_SUCCESS, "CredUIConfirmCredentials failed with error %d\n", ret); 122 } 123 124 125 save = TRUE; 126 credui_info.pszCaptionText = wszCaption3; 127 ret = CredUIPromptForCredentialsW(&credui_info, wszServerName, NULL, 0, 128 username, sizeof(username)/sizeof(username[0]), 129 password, sizeof(password)/sizeof(password[0]), 130 &save, CREDUI_FLAGS_DO_NOT_PERSIST|CREDUI_FLAGS_EXPECT_CONFIRMATION); 131 ok(ret == ERROR_SUCCESS || ret == ERROR_CANCELLED, "CredUIPromptForCredentials failed with error %d\n", ret); 132 ok(save, "save flag should have been untouched\n"); 133 134 save = FALSE; 135 credui_info.pszCaptionText = wszCaption4; 136 ret = CredUIPromptForCredentialsW(&credui_info, wszServerName, NULL, 0, 137 username, sizeof(username)/sizeof(username[0]), 138 password, sizeof(password)/sizeof(password[0]), 139 &save, CREDUI_FLAGS_PERSIST|CREDUI_FLAGS_EXPECT_CONFIRMATION); 140 ok(ret == ERROR_SUCCESS || ret == ERROR_CANCELLED, "CredUIPromptForCredentials failed with error %d\n", ret); 141 ok(!save, "save flag should have been untouched\n"); 142 if (ret == ERROR_SUCCESS) 143 { 144 ret = CredUIConfirmCredentialsW(wszServerName, FALSE); 145 ok(ret == ERROR_SUCCESS, "CredUIConfirmCredentials failed with error %d\n", ret); 146 } 147 148 } 149 } 150 151 static void test_SspiPromptForCredentials(void) 152 { 153 static const WCHAR targetW[] = {'S','s','p','i','T','e','s','t',0}; 154 static const WCHAR basicW[] = {'b','a','s','i','c',0}; 155 ULONG ret; 156 SECURITY_STATUS status; 157 CREDUI_INFOW info; 158 PSEC_WINNT_AUTH_IDENTITY_OPAQUE id; 159 const WCHAR *username, *domain, *creds; 160 int save; 161 162 if (!pSspiPromptForCredentialsW || !pSspiFreeAuthIdentity) 163 { 164 win_skip( "SspiPromptForCredentialsW is missing\n" ); 165 return; 166 } 167 168 info.cbSize = sizeof(info); 169 info.hwndParent = NULL; 170 info.pszMessageText = targetW; 171 info.pszCaptionText = basicW; 172 info.hbmBanner = NULL; 173 ret = pSspiPromptForCredentialsW( NULL, &info, 0, basicW, NULL, &id, &save, 0 ); 174 ok( ret == ERROR_INVALID_PARAMETER, "got %u\n", ret ); 175 176 ret = pSspiPromptForCredentialsW( targetW, &info, 0, NULL, NULL, &id, &save, 0 ); 177 ok( ret == ERROR_NO_SUCH_PACKAGE, "got %u\n", ret ); 178 179 if (winetest_interactive) 180 { 181 id = NULL; 182 save = -1; 183 ret = pSspiPromptForCredentialsW( targetW, &info, 0, basicW, NULL, &id, &save, 0 ); 184 ok( ret == ERROR_SUCCESS || ret == ERROR_CANCELLED, "got %u\n", ret ); 185 if (ret == ERROR_SUCCESS) 186 { 187 ok( id != NULL, "id not set\n" ); 188 ok( save == TRUE || save == FALSE, "got %d\n", save ); 189 190 username = creds = NULL; 191 domain = (const WCHAR *)0xdeadbeef; 192 status = pSspiEncodeAuthIdentityAsStrings( id, &username, &domain, &creds ); 193 ok( status == SEC_E_OK, "got %u\n", status ); 194 ok( username != NULL, "username not set\n" ); 195 ok( domain == NULL, "domain not set\n" ); 196 ok( creds != NULL, "creds not set\n" ); 197 pSspiFreeAuthIdentity( id ); 198 } 199 } 200 } 201 202 START_TEST(credui) 203 { 204 HMODULE hcredui = GetModuleHandleA( "credui.dll" ), hsecur32 = LoadLibraryA( "secur32.dll" ); 205 206 if (hcredui) 207 pSspiPromptForCredentialsW = (void *)GetProcAddress( hcredui, "SspiPromptForCredentialsW" ); 208 if (hsecur32) 209 { 210 pSspiEncodeAuthIdentityAsStrings = (void *)GetProcAddress( hsecur32, "SspiEncodeAuthIdentityAsStrings" ); 211 pSspiFreeAuthIdentity = (void *)GetProcAddress( hsecur32, "SspiFreeAuthIdentity" ); 212 } 213 214 test_CredUIPromptForCredentials(); 215 test_SspiPromptForCredentials(); 216 217 FreeLibrary( hsecur32 ); 218 } 219