2 * Credentials User Interface
4 * Copyright 2006 Robert Shearman (for CodeWeavers)
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.
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.
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
30 #include "credui_resources.h"
32 #include "wine/debug.h"
33 #include "wine/unicode.h"
34 #include "wine/list.h"
36 WINE_DEFAULT_DEBUG_CHANNEL(credui);
38 struct pending_credentials
47 static HINSTANCE hinstCredUI;
49 static struct list pending_credentials_list = LIST_INIT(pending_credentials_list);
51 static CRITICAL_SECTION csPendingCredentials;
52 static CRITICAL_SECTION_DEBUG critsect_debug =
54 0, 0, &csPendingCredentials,
55 { &critsect_debug.ProcessLocksList, &critsect_debug.ProcessLocksList },
56 0, 0, { (DWORD_PTR)(__FILE__ ": csPendingCredentials") }
58 static CRITICAL_SECTION csPendingCredentials = { &critsect_debug, -1, 0, 0, 0, 0 };
61 BOOL WINAPI DllMain(HINSTANCE hinstDLL, DWORD fdwReason, LPVOID lpvReserved)
63 TRACE("(0x%p, %d, %p)\n",hinstDLL,fdwReason,lpvReserved);
65 if (fdwReason == DLL_WINE_PREATTACH) return FALSE; /* prefer native version */
67 if (fdwReason == DLL_PROCESS_ATTACH)
69 DisableThreadLibraryCalls(hinstDLL);
70 hinstCredUI = hinstDLL;
73 else if (fdwReason == DLL_PROCESS_DETACH)
75 struct pending_credentials *entry, *cursor2;
76 LIST_FOR_EACH_ENTRY_SAFE(entry, cursor2, &pending_credentials_list, struct pending_credentials, entry)
78 list_remove(&entry->entry);
80 HeapFree(GetProcessHeap(), 0, entry->pszTargetName);
81 HeapFree(GetProcessHeap(), 0, entry->pszUsername);
82 ZeroMemory(entry->pszPassword, (strlenW(entry->pszPassword) + 1) * sizeof(WCHAR));
83 HeapFree(GetProcessHeap(), 0, entry->pszPassword);
84 HeapFree(GetProcessHeap(), 0, entry);
91 static DWORD save_credentials(PCWSTR pszTargetName, PCWSTR pszUsername,
92 PCWSTR pszPassword, BOOL generic)
96 TRACE("saving servername %s with username %s\n", debugstr_w(pszTargetName), debugstr_w(pszUsername));
99 cred.Type = generic ? CRED_TYPE_GENERIC : CRED_TYPE_DOMAIN_PASSWORD;
100 cred.TargetName = (LPWSTR)pszTargetName;
102 cred.CredentialBlobSize = strlenW(pszPassword) * sizeof(WCHAR);
103 cred.CredentialBlob = (LPBYTE)pszPassword;
104 cred.Persist = CRED_PERSIST_ENTERPRISE;
105 cred.AttributeCount = 0;
106 cred.Attributes = NULL;
107 cred.TargetAlias = NULL;
108 cred.UserName = (LPWSTR)pszUsername;
110 if (CredWriteW(&cred, 0))
111 return ERROR_SUCCESS;
114 DWORD ret = GetLastError();
115 ERR("CredWriteW failed with error %d\n", ret);
120 struct cred_dialog_params
122 PCWSTR pszTargetName;
123 PCWSTR pszMessageText;
124 PCWSTR pszCaptionText;
127 ULONG ulUsernameMaxChars;
129 ULONG ulPasswordMaxChars;
134 static void CredDialogFillUsernameCombo(HWND hwndUsername, struct cred_dialog_params *params)
138 PCREDENTIALW *credentials;
140 if (!CredEnumerateW(NULL, 0, &count, &credentials))
143 for (i = 0; i < count; i++)
145 COMBOBOXEXITEMW comboitem;
147 BOOL duplicate = FALSE;
149 if (params->dwFlags & CREDUI_FLAGS_GENERIC_CREDENTIALS)
151 if ((credentials[i]->Type != CRED_TYPE_GENERIC) || !credentials[i]->UserName)
156 if (credentials[i]->Type == CRED_TYPE_GENERIC)
160 /* don't add another item with the same name if we've already added it */
161 for (j = 0; j < i; j++)
162 if (!strcmpW(credentials[i]->UserName, credentials[j]->UserName))
171 comboitem.mask = CBEIF_TEXT;
172 comboitem.iItem = -1;
173 comboitem.pszText = credentials[i]->UserName;
174 SendMessageW(hwndUsername, CBEM_INSERTITEMW, 0, (LPARAM)&comboitem);
177 CredFree(credentials);
180 static BOOL CredDialogInit(HWND hwndDlg, struct cred_dialog_params *params)
182 HWND hwndUsername = GetDlgItem(hwndDlg, IDC_USERNAME);
183 HWND hwndPassword = GetDlgItem(hwndDlg, IDC_PASSWORD);
185 SetWindowLongPtrW(hwndDlg, DWLP_USER, (LONG_PTR)params);
187 if (params->hbmBanner)
188 SendMessageW(GetDlgItem(hwndDlg, IDB_BANNER), STM_SETIMAGE,
189 IMAGE_BITMAP, (LPARAM)params->hbmBanner);
191 if (params->pszMessageText)
192 SetDlgItemTextW(hwndDlg, IDC_MESSAGE, params->pszMessageText);
197 LoadStringW(hinstCredUI, IDS_MESSAGEFORMAT, format, sizeof(format)/sizeof(format[0]));
198 snprintfW(message, sizeof(message)/sizeof(message[0]), format, params->pszTargetName);
199 SetDlgItemTextW(hwndDlg, IDC_MESSAGE, message);
201 SetWindowTextW(hwndUsername, params->pszUsername);
202 SetWindowTextW(hwndPassword, params->pszPassword);
204 CredDialogFillUsernameCombo(hwndUsername, params);
206 if (params->pszUsername[0])
207 SetFocus(hwndPassword);
209 SetFocus(hwndUsername);
211 if (params->pszCaptionText)
212 SetWindowTextW(hwndDlg, params->pszCaptionText);
217 LoadStringW(hinstCredUI, IDS_TITLEFORMAT, format, sizeof(format)/sizeof(format[0]));
218 snprintfW(title, sizeof(title)/sizeof(title[0]), format, params->pszTargetName);
219 SetWindowTextW(hwndDlg, title);
222 if (params->dwFlags & (CREDUI_FLAGS_DO_NOT_PERSIST|CREDUI_FLAGS_PERSIST))
223 ShowWindow(GetDlgItem(hwndDlg, IDC_SAVE), SW_HIDE);
224 else if (params->fSave)
225 CheckDlgButton(hwndDlg, IDC_SAVE, BST_CHECKED);
230 static void CredDialogCommandOk(HWND hwndDlg, struct cred_dialog_params *params)
232 HWND hwndUsername = GetDlgItem(hwndDlg, IDC_USERNAME);
237 len = GetWindowTextLengthW(hwndUsername);
238 user = HeapAlloc(GetProcessHeap(), 0, (len + 1) * sizeof(WCHAR));
239 GetWindowTextW(hwndUsername, user, len + 1);
243 HeapFree(GetProcessHeap(), 0, user);
247 if (!strchrW(user, '\\') && !strchrW(user, '@'))
249 INT len_target = strlenW(params->pszTargetName);
250 memcpy(params->pszUsername, params->pszTargetName,
251 min(len_target, params->ulUsernameMaxChars) * sizeof(WCHAR));
252 if (len_target + 1 < params->ulUsernameMaxChars)
253 params->pszUsername[len_target] = '\\';
254 if (len_target + 2 < params->ulUsernameMaxChars)
255 params->pszUsername[len_target + 1] = '\0';
257 else if (params->ulUsernameMaxChars > 0)
258 params->pszUsername[0] = '\0';
260 len2 = strlenW(params->pszUsername);
261 memcpy(params->pszUsername + len2, user, min(len, params->ulUsernameMaxChars - len2) * sizeof(WCHAR));
262 if (params->ulUsernameMaxChars)
263 params->pszUsername[len2 + min(len, params->ulUsernameMaxChars - len2 - 1)] = '\0';
265 HeapFree(GetProcessHeap(), 0, user);
267 GetDlgItemTextW(hwndDlg, IDC_PASSWORD, params->pszPassword,
268 params->ulPasswordMaxChars);
270 EndDialog(hwndDlg, IDOK);
273 static INT_PTR CALLBACK CredDialogProc(HWND hwndDlg, UINT uMsg, WPARAM wParam,
280 struct cred_dialog_params *params = (struct cred_dialog_params *)lParam;
282 return CredDialogInit(hwndDlg, params);
287 case MAKELONG(IDOK, BN_CLICKED):
289 struct cred_dialog_params *params =
290 (struct cred_dialog_params *)GetWindowLongPtrW(hwndDlg, DWLP_USER);
291 CredDialogCommandOk(hwndDlg, params);
294 case MAKELONG(IDCANCEL, BN_CLICKED):
295 EndDialog(hwndDlg, IDCANCEL);
297 case MAKELONG(IDC_PASSWORD, EN_SETFOCUS):
298 /* don't allow another window to steal focus while the
299 * user is typing their password */
300 LockSetForegroundWindow(LSFW_LOCK);
302 case MAKELONG(IDC_PASSWORD, EN_KILLFOCUS):
303 /* the user is no longer typing their password, so allow
304 * other windows to become foreground ones */
305 LockSetForegroundWindow(LSFW_UNLOCK);
314 /******************************************************************************
315 * CredUIPromptForCredentialsW [CREDUI.@]
317 DWORD WINAPI CredUIPromptForCredentialsW(PCREDUI_INFOW pUIInfo,
318 PCWSTR pszTargetName,
319 PCtxtHandle Reserved,
322 ULONG ulUsernameMaxChars,
324 ULONG ulPasswordMaxChars, PBOOL pfSave,
328 struct cred_dialog_params params;
329 DWORD result = ERROR_SUCCESS;
331 TRACE("(%p, %s, %p, %d, %s, %d, %p, %d, %p, 0x%08x)\n", pUIInfo,
332 debugstr_w(pszTargetName), Reserved, dwAuthError, debugstr_w(pszUsername),
333 ulUsernameMaxChars, pszPassword, ulPasswordMaxChars, pfSave, dwFlags);
335 if ((dwFlags & (CREDUI_FLAGS_ALWAYS_SHOW_UI|CREDUI_FLAGS_GENERIC_CREDENTIALS)) == CREDUI_FLAGS_ALWAYS_SHOW_UI)
336 return ERROR_INVALID_FLAGS;
339 return ERROR_INVALID_PARAMETER;
341 if ((dwFlags & CREDUI_FLAGS_SHOW_SAVE_CHECK_BOX) && !pfSave)
342 return ERROR_INVALID_PARAMETER;
344 params.pszTargetName = pszTargetName;
347 params.pszMessageText = pUIInfo->pszMessageText;
348 params.pszCaptionText = pUIInfo->pszCaptionText;
349 params.hbmBanner = pUIInfo->hbmBanner;
353 params.pszMessageText = NULL;
354 params.pszCaptionText = NULL;
355 params.hbmBanner = NULL;
357 params.pszUsername = pszUsername;
358 params.ulUsernameMaxChars = ulUsernameMaxChars;
359 params.pszPassword = pszPassword;
360 params.ulPasswordMaxChars = ulPasswordMaxChars;
361 params.fSave = pfSave ? *pfSave : FALSE;
362 params.dwFlags = dwFlags;
364 ret = DialogBoxParamW(hinstCredUI, MAKEINTRESOURCEW(IDD_CREDDIALOG),
365 pUIInfo ? pUIInfo->hwndParent : NULL,
366 CredDialogProc, (LPARAM)¶ms);
368 return GetLastError();
372 TRACE("dialog cancelled\n");
373 return ERROR_CANCELLED;
377 *pfSave = params.fSave;
381 if (dwFlags & CREDUI_FLAGS_EXPECT_CONFIRMATION)
384 struct pending_credentials *entry;
387 EnterCriticalSection(&csPendingCredentials);
389 /* find existing pending credentials for the same target and overwrite */
390 /* FIXME: is this correct? */
391 LIST_FOR_EACH_ENTRY(entry, &pending_credentials_list, struct pending_credentials, entry)
392 if (!strcmpW(pszTargetName, entry->pszTargetName))
395 HeapFree(GetProcessHeap(), 0, entry->pszUsername);
396 ZeroMemory(entry->pszPassword, (strlenW(entry->pszPassword) + 1) * sizeof(WCHAR));
397 HeapFree(GetProcessHeap(), 0, entry->pszPassword);
402 entry = HeapAlloc(GetProcessHeap(), 0, sizeof(*entry));
403 list_init(&entry->entry);
404 len = strlenW(pszTargetName);
405 entry->pszTargetName = HeapAlloc(GetProcessHeap(), 0, (len + 1)*sizeof(WCHAR));
406 memcpy(entry->pszTargetName, pszTargetName, (len + 1)*sizeof(WCHAR));
407 list_add_tail(&entry->entry, &pending_credentials_list);
410 len = strlenW(params.pszUsername);
411 entry->pszUsername = HeapAlloc(GetProcessHeap(), 0, (len + 1)*sizeof(WCHAR));
412 memcpy(entry->pszUsername, params.pszUsername, (len + 1)*sizeof(WCHAR));
413 len = strlenW(params.pszPassword);
414 entry->pszPassword = HeapAlloc(GetProcessHeap(), 0, (len + 1)*sizeof(WCHAR));
415 memcpy(entry->pszPassword, params.pszPassword, (len + 1)*sizeof(WCHAR));
416 entry->generic = dwFlags & CREDUI_FLAGS_GENERIC_CREDENTIALS ? TRUE : FALSE;
418 LeaveCriticalSection(&csPendingCredentials);
421 result = save_credentials(pszTargetName, pszUsername, pszPassword,
422 dwFlags & CREDUI_FLAGS_GENERIC_CREDENTIALS ? TRUE : FALSE);
428 /******************************************************************************
429 * CredUIConfirmCredentialsW [CREDUI.@]
431 DWORD WINAPI CredUIConfirmCredentialsW(PCWSTR pszTargetName, BOOL bConfirm)
433 struct pending_credentials *entry;
434 DWORD result = ERROR_NOT_FOUND;
436 TRACE("(%s, %s)\n", debugstr_w(pszTargetName), bConfirm ? "TRUE" : "FALSE");
439 return ERROR_INVALID_PARAMETER;
441 EnterCriticalSection(&csPendingCredentials);
443 LIST_FOR_EACH_ENTRY(entry, &pending_credentials_list, struct pending_credentials, entry)
445 if (!strcmpW(pszTargetName, entry->pszTargetName))
448 result = save_credentials(entry->pszTargetName, entry->pszUsername,
449 entry->pszPassword, entry->generic);
451 result = ERROR_SUCCESS;
453 list_remove(&entry->entry);
455 HeapFree(GetProcessHeap(), 0, entry->pszTargetName);
456 HeapFree(GetProcessHeap(), 0, entry->pszUsername);
457 ZeroMemory(entry->pszPassword, (strlenW(entry->pszPassword) + 1) * sizeof(WCHAR));
458 HeapFree(GetProcessHeap(), 0, entry->pszPassword);
459 HeapFree(GetProcessHeap(), 0, entry);
465 LeaveCriticalSection(&csPendingCredentials);
470 /******************************************************************************
471 * CredUIParseUserNameW [CREDUI.@]
473 DWORD WINAPI CredUIParseUserNameW(PCWSTR pszUserName, PWSTR pszUser,
474 ULONG ulMaxUserChars, PWSTR pszDomain,
475 ULONG ulMaxDomainChars)
479 TRACE("(%s, %p, %d, %p, %d)\n", debugstr_w(pszUserName), pszUser,
480 ulMaxUserChars, pszDomain, ulMaxDomainChars);
482 if (!pszUserName || !pszUser || !ulMaxUserChars || !pszDomain ||
484 return ERROR_INVALID_PARAMETER;
486 /* FIXME: handle marshaled credentials */
488 p = strchrW(pszUserName, '\\');
491 if (p - pszUserName > ulMaxDomainChars - 1)
492 return ERROR_INSUFFICIENT_BUFFER;
493 if (strlenW(p + 1) > ulMaxUserChars - 1)
494 return ERROR_INSUFFICIENT_BUFFER;
495 strcpyW(pszUser, p + 1);
496 memcpy(pszDomain, pszUserName, (p - pszUserName)*sizeof(WCHAR));
497 pszDomain[p - pszUserName] = '\0';
499 return ERROR_SUCCESS;
502 p = strrchrW(pszUserName, '@');
505 if (p + 1 - pszUserName > ulMaxUserChars - 1)
506 return ERROR_INSUFFICIENT_BUFFER;
507 if (strlenW(p + 1) > ulMaxDomainChars - 1)
508 return ERROR_INSUFFICIENT_BUFFER;
509 strcpyW(pszDomain, p + 1);
510 memcpy(pszUser, pszUserName, (p - pszUserName)*sizeof(WCHAR));
511 pszUser[p - pszUserName] = '\0';
513 return ERROR_SUCCESS;
516 if (strlenW(pszUserName) > ulMaxUserChars - 1)
517 return ERROR_INSUFFICIENT_BUFFER;
518 strcpyW(pszUser, pszUserName);
521 return ERROR_SUCCESS;
524 /******************************************************************************
525 * CredUIStoreSSOCredA [CREDUI.@]
527 DWORD WINAPI CredUIStoreSSOCredA(PCSTR pszRealm, PCSTR pszUsername,
528 PCSTR pszPassword, BOOL bPersist)
530 FIXME("(%s, %s, %p, %d)\n", debugstr_a(pszRealm), debugstr_a(pszUsername),
531 pszPassword, bPersist);
532 return ERROR_SUCCESS;
535 /******************************************************************************
536 * CredUIStoreSSOCredW [CREDUI.@]
538 DWORD WINAPI CredUIStoreSSOCredW(PCWSTR pszRealm, PCWSTR pszUsername,
539 PCWSTR pszPassword, BOOL bPersist)
541 FIXME("(%s, %s, %p, %d)\n", debugstr_w(pszRealm), debugstr_w(pszUsername),
542 pszPassword, bPersist);
543 return ERROR_SUCCESS;
546 /******************************************************************************
547 * CredUIReadSSOCredA [CREDUI.@]
549 DWORD WINAPI CredUIReadSSOCredA(PCSTR pszRealm, PSTR *ppszUsername)
551 FIXME("(%s, %p)\n", debugstr_a(pszRealm), ppszUsername);
553 *ppszUsername = NULL;
554 return ERROR_NOT_FOUND;
557 /******************************************************************************
558 * CredUIReadSSOCredW [CREDUI.@]
560 DWORD WINAPI CredUIReadSSOCredW(PCWSTR pszRealm, PWSTR *ppszUsername)
562 FIXME("(%s, %p)\n", debugstr_w(pszRealm), ppszUsername);
564 *ppszUsername = NULL;
565 return ERROR_NOT_FOUND;