dlls/wininet/dialogs.c

   1 /*
   2  * Wininet
   3  *
   4  * Copyright 2003 Mike McCormack for CodeWeavers Inc.
   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 "config.h"
  22 #include "wine/port.h"
  23
  24 #if defined(__MINGW32__) || defined (_MSC_VER)
  25 #include <ws2tcpip.h>
  26 #endif
  27
  28 #include <stdarg.h>
  29
  30 #include "windef.h"
  31 #include "winbase.h"
  32 #include "winuser.h"
  33 #include "winreg.h"
  34 #include "wininet.h"
  35 #include "winnetwk.h"
  36 #include "wine/debug.h"
  37 #include "winerror.h"
  38 #define NO_SHLWAPI_STREAM
  39 #include "shlwapi.h"
  40
  41 #include "internet.h"
  42
  43 #include "wine/unicode.h"
  44
  45 #include "resource.h"
  46
  47 WINE_DEFAULT_DEBUG_CHANNEL(wininet);
  48
  49 struct WININET_ErrorDlgParams
  50 {
  51     HWND       hWnd;
  52     HINTERNET  hRequest;
  53     DWORD      dwError;
  54     DWORD      dwFlags;
  55     LPVOID*    lppvData;
  56 };
  57
  58 /***********************************************************************
  59  *         WININET_GetProxyServer
  60  *
  61  *  Determine the name of the proxy server the request is using
  62  */
  63 static BOOL WININET_GetProxyServer( HINTERNET hRequest, LPWSTR szBuf, DWORD sz )
  64 {
  65     http_request_t *lpwhr;
  66     http_session_t *lpwhs = NULL;
  67     appinfo_t *hIC = NULL;
  68     BOOL ret = FALSE;
  69     LPWSTR p;
  70
  71     lpwhr = (http_request_t*) WININET_GetObject( hRequest );
  72     if (NULL == lpwhr)
  73         return FALSE;
  74
  75     lpwhs = lpwhr->lpHttpSession;
  76     if (NULL == lpwhs)
  77         goto done;
  78
  79     hIC = lpwhs->lpAppInfo;
  80     if (NULL == hIC)
  81         goto done;
  82
  83     lstrcpynW(szBuf, hIC->lpszProxy, sz);
  84
  85     /* FIXME: perhaps it would be better to use InternetCrackUrl here */
  86     p = strchrW(szBuf, ':');
  87     if (p)
  88         *p = 0;
  89
  90     ret = TRUE;
  91
  92 done:
  93     WININET_Release( &lpwhr->hdr );
  94     return ret;
  95 }
  96
  97 /***********************************************************************
  98  *         WININET_GetServer
  99  *
 100  *  Determine the name of the web server
 101  */
 102 static BOOL WININET_GetServer( HINTERNET hRequest, LPWSTR szBuf, DWORD sz )
 103 {
 104     http_request_t *lpwhr;
 105     http_session_t *lpwhs = NULL;
 106     BOOL ret = FALSE;
 107
 108     lpwhr = (http_request_t*) WININET_GetObject( hRequest );
 109     if (NULL == lpwhr)
 110         return FALSE;
 111
 112     lpwhs = lpwhr->lpHttpSession;
 113     if (NULL == lpwhs)
 114         goto done;
 115
 116     lstrcpynW(szBuf, lpwhs->lpszHostName, sz);
 117
 118     ret = TRUE;
 119
 120 done:
 121     WININET_Release( &lpwhr->hdr );
 122     return ret;
 123 }
 124
 125 /***********************************************************************
 126  *         WININET_GetAuthRealm
 127  *
 128  *  Determine the name of the (basic) Authentication realm
 129  */
 130 static BOOL WININET_GetAuthRealm( HINTERNET hRequest, LPWSTR szBuf, DWORD sz, BOOL proxy )
 131 {
 132     LPWSTR p, q;
 133     DWORD index, query;
 134     static const WCHAR szRealm[] = { 'r','e','a','l','m','=',0 };
 135
 136     if (proxy)
 137         query = HTTP_QUERY_PROXY_AUTHENTICATE;
 138     else
 139         query = HTTP_QUERY_WWW_AUTHENTICATE;
 140
 141     /* extract the Realm from the response and show it */
 142     index = 0;
 143     if( !HttpQueryInfoW( hRequest, query, szBuf, &sz, &index) )
 144         return FALSE;
 145
 146     /*
 147      * FIXME: maybe we should check that we're
 148      * dealing with 'Basic' Authentication
 149      */
 150     p = strchrW( szBuf, ' ' );
 151     if( !p || strncmpW( p+1, szRealm, strlenW(szRealm) ) )
 152     {
 153         ERR("response wrong? (%s)\n", debugstr_w(szBuf));
 154         return FALSE;
 155     }
 156
 157     /* remove quotes */
 158     p += 7;
 159     if( *p == '"' )
 160     {
 161         p++;
 162         q = strrchrW( p, '"' );
 163         if( q )
 164             *q = 0;
 165     }
 166     strcpyW( szBuf, p );
 167
 168     return TRUE;
 169 }
 170
 171 /***********************************************************************
 172  *         WININET_GetSetPassword
 173  */
 174 static BOOL WININET_GetSetPassword( HWND hdlg, LPCWSTR szServer,
 175                                     LPCWSTR szRealm, BOOL bSet )
 176 {
 177     WCHAR szResource[0x80], szUserPass[0x40];
 178     LPWSTR p;
 179     HWND hUserItem, hPassItem;
 180     DWORD r, dwMagic = 19;
 181     UINT r_len, u_len;
 182     WORD sz;
 183     static const WCHAR szColon[] = { ':',0 };
 184     static const WCHAR szbs[] = { '/', 0 };
 185
 186     hUserItem = GetDlgItem( hdlg, IDC_USERNAME );
 187     hPassItem = GetDlgItem( hdlg, IDC_PASSWORD );
 188
 189     /* now try fetch the username and password */
 190     lstrcpyW( szResource, szServer);
 191     lstrcatW( szResource, szbs);
 192     lstrcatW( szResource, szRealm);
 193
 194     /*
 195      * WNetCachePassword is only concerned with the length
 196      * of the data stored (which we tell it) and it does
 197      * not use strlen() internally so we can add WCHAR data
 198      * instead of ASCII data and get it back the same way.
 199      */
 200     if( bSet )
 201     {
 202         szUserPass[0] = 0;
 203         GetWindowTextW( hUserItem, szUserPass,
 204                         (sizeof szUserPass-1)/sizeof(WCHAR) );
 205         lstrcatW(szUserPass, szColon);
 206         u_len = strlenW( szUserPass );
 207         GetWindowTextW( hPassItem, szUserPass+u_len,
 208                         (sizeof szUserPass)/sizeof(WCHAR)-u_len );
 209
 210         r_len = (strlenW( szResource ) + 1)*sizeof(WCHAR);
 211         u_len = (strlenW( szUserPass ) + 1)*sizeof(WCHAR);
 212         r = WNetCachePassword( (CHAR*)szResource, r_len,
 213                                (CHAR*)szUserPass, u_len, dwMagic, 0 );
 214
 215         return ( r == WN_SUCCESS );
 216     }
 217
 218     sz = sizeof szUserPass;
 219     r_len = (strlenW( szResource ) + 1)*sizeof(WCHAR);
 220     r = WNetGetCachedPassword( (CHAR*)szResource, r_len,
 221                                (CHAR*)szUserPass, &sz, dwMagic );
 222     if( r != WN_SUCCESS )
 223         return FALSE;
 224
 225     p = strchrW( szUserPass, ':' );
 226     if( p )
 227     {
 228         *p = 0;
 229         SetWindowTextW( hUserItem, szUserPass );
 230         SetWindowTextW( hPassItem, p+1 );
 231     }
 232
 233     return TRUE;
 234 }
 235
 236 /***********************************************************************
 237  *         WININET_SetAuthorization
 238  */
 239 static BOOL WININET_SetAuthorization( HINTERNET hRequest, LPWSTR username,
 240                                       LPWSTR password, BOOL proxy )
 241 {
 242     http_request_t *lpwhr;
 243     http_session_t *lpwhs;
 244     BOOL ret = FALSE;
 245     LPWSTR p, q;
 246
 247     lpwhr = (http_request_t*) WININET_GetObject( hRequest );
 248     if( !lpwhr )
 249         return FALSE;
 250
 251     lpwhs = lpwhr->lpHttpSession;
 252     if (NULL == lpwhs ||  lpwhs->hdr.htype != WH_HHTTPSESSION)
 253     {
 254         INTERNET_SetLastError(ERROR_INTERNET_INCORRECT_HANDLE_TYPE);
 255         goto done;
 256     }
 257
 258     p = heap_strdupW(username);
 259     if( !p )
 260         goto done;
 261
 262     q = heap_strdupW(password);
 263     if( !q )
 264     {
 265         HeapFree(GetProcessHeap(), 0, username);
 266         goto done;
 267     }
 268
 269     if (proxy)
 270     {
 271         appinfo_t *hIC = lpwhs->lpAppInfo;
 272
 273         HeapFree(GetProcessHeap(), 0, hIC->lpszProxyUsername);
 274         hIC->lpszProxyUsername = p;
 275
 276         HeapFree(GetProcessHeap(), 0, hIC->lpszProxyPassword);
 277         hIC->lpszProxyPassword = q;
 278     }
 279     else
 280     {
 281         HeapFree(GetProcessHeap(), 0, lpwhs->lpszUserName);
 282         lpwhs->lpszUserName = p;
 283
 284         HeapFree(GetProcessHeap(), 0, lpwhs->lpszPassword);
 285         lpwhs->lpszPassword = q;
 286     }
 287
 288     ret = TRUE;
 289
 290 done:
 291     WININET_Release( &lpwhr->hdr );
 292     return ret;
 293 }
 294
 295 /***********************************************************************
 296  *         WININET_ProxyPasswordDialog
 297  */
 298 static INT_PTR WINAPI WININET_ProxyPasswordDialog(
 299     HWND hdlg, UINT uMsg, WPARAM wParam, LPARAM lParam )
 300 {
 301     HWND hitem;
 302     struct WININET_ErrorDlgParams *params;
 303     WCHAR szRealm[0x80], szServer[0x80];
 304
 305     if( uMsg == WM_INITDIALOG )
 306     {
 307         TRACE("WM_INITDIALOG (%08lx)\n", lParam);
 308
 309         /* save the parameter list */
 310         params = (struct WININET_ErrorDlgParams*) lParam;
 311         SetWindowLongPtrW( hdlg, GWLP_USERDATA, lParam );
 312
 313         /* extract the Realm from the proxy response and show it */
 314         if( WININET_GetAuthRealm( params->hRequest,
 315                                   szRealm, sizeof szRealm/sizeof(WCHAR), TRUE ) )
 316         {
 317             hitem = GetDlgItem( hdlg, IDC_REALM );
 318             SetWindowTextW( hitem, szRealm );
 319         }
 320
 321         /* extract the name of the proxy server */
 322         if( WININET_GetProxyServer( params->hRequest,
 323                                     szServer, sizeof szServer/sizeof(WCHAR)) )
 324         {
 325             hitem = GetDlgItem( hdlg, IDC_PROXY );
 326             SetWindowTextW( hitem, szServer );
 327         }
 328
 329         WININET_GetSetPassword( hdlg, szServer, szRealm, FALSE );
 330
 331         return TRUE;
 332     }
 333
 334     params = (struct WININET_ErrorDlgParams*)
 335                  GetWindowLongPtrW( hdlg, GWLP_USERDATA );
 336
 337     switch( uMsg )
 338     {
 339     case WM_COMMAND:
 340         if( wParam == IDOK )
 341         {
 342             WCHAR username[0x20], password[0x20];
 343
 344             username[0] = 0;
 345             hitem = GetDlgItem( hdlg, IDC_USERNAME );
 346             if( hitem )
 347                 GetWindowTextW( hitem, username, sizeof username/sizeof(WCHAR) );
 348
 349             password[0] = 0;
 350             hitem = GetDlgItem( hdlg, IDC_PASSWORD );
 351             if( hitem )
 352                 GetWindowTextW( hitem, password, sizeof password/sizeof(WCHAR) );
 353
 354             hitem = GetDlgItem( hdlg, IDC_SAVEPASSWORD );
 355             if( hitem &&
 356                 SendMessageW( hitem, BM_GETSTATE, 0, 0 ) &&
 357                 WININET_GetAuthRealm( params->hRequest,
 358                                   szRealm, sizeof szRealm/sizeof(WCHAR), TRUE ) &&
 359                 WININET_GetProxyServer( params->hRequest,
 360                                     szServer, sizeof szServer/sizeof(WCHAR)) )
 361             {
 362                 WININET_GetSetPassword( hdlg, szServer, szRealm, TRUE );
 363             }
 364             WININET_SetAuthorization( params->hRequest, username, password, TRUE );
 365
 366             EndDialog( hdlg, ERROR_INTERNET_FORCE_RETRY );
 367             return TRUE;
 368         }
 369         if( wParam == IDCANCEL )
 370         {
 371             EndDialog( hdlg, 0 );
 372             return TRUE;
 373         }
 374         break;
 375     }
 376     return FALSE;
 377 }
 378
 379 /***********************************************************************
 380  *         WININET_PasswordDialog
 381  */
 382 static INT_PTR WINAPI WININET_PasswordDialog(
 383     HWND hdlg, UINT uMsg, WPARAM wParam, LPARAM lParam )
 384 {
 385     HWND hitem;
 386     struct WININET_ErrorDlgParams *params;
 387     WCHAR szRealm[0x80], szServer[0x80];
 388
 389     if( uMsg == WM_INITDIALOG )
 390     {
 391         TRACE("WM_INITDIALOG (%08lx)\n", lParam);
 392
 393         /* save the parameter list */
 394         params = (struct WININET_ErrorDlgParams*) lParam;
 395         SetWindowLongPtrW( hdlg, GWLP_USERDATA, lParam );
 396
 397         /* extract the Realm from the response and show it */
 398         if( WININET_GetAuthRealm( params->hRequest,
 399                                   szRealm, sizeof szRealm/sizeof(WCHAR), FALSE ) )
 400         {
 401             hitem = GetDlgItem( hdlg, IDC_REALM );
 402             SetWindowTextW( hitem, szRealm );
 403         }
 404
 405         /* extract the name of the server */
 406         if( WININET_GetServer( params->hRequest,
 407                                szServer, sizeof szServer/sizeof(WCHAR)) )
 408         {
 409             hitem = GetDlgItem( hdlg, IDC_SERVER );
 410             SetWindowTextW( hitem, szServer );
 411         }
 412
 413         WININET_GetSetPassword( hdlg, szServer, szRealm, FALSE );
 414
 415         return TRUE;
 416     }
 417
 418     params = (struct WININET_ErrorDlgParams*)
 419                  GetWindowLongPtrW( hdlg, GWLP_USERDATA );
 420
 421     switch( uMsg )
 422     {
 423     case WM_COMMAND:
 424         if( wParam == IDOK )
 425         {
 426             WCHAR username[0x20], password[0x20];
 427
 428             username[0] = 0;
 429             hitem = GetDlgItem( hdlg, IDC_USERNAME );
 430             if( hitem )
 431                 GetWindowTextW( hitem, username, sizeof username/sizeof(WCHAR) );
 432
 433             password[0] = 0;
 434             hitem = GetDlgItem( hdlg, IDC_PASSWORD );
 435             if( hitem )
 436                 GetWindowTextW( hitem, password, sizeof password/sizeof(WCHAR) );
 437
 438             hitem = GetDlgItem( hdlg, IDC_SAVEPASSWORD );
 439             if( hitem &&
 440                 SendMessageW( hitem, BM_GETSTATE, 0, 0 ) &&
 441                 WININET_GetAuthRealm( params->hRequest,
 442                                   szRealm, sizeof szRealm/sizeof(WCHAR), FALSE ) &&
 443                 WININET_GetServer( params->hRequest,
 444                                    szServer, sizeof szServer/sizeof(WCHAR)) )
 445             {
 446                 WININET_GetSetPassword( hdlg, szServer, szRealm, TRUE );
 447             }
 448             WININET_SetAuthorization( params->hRequest, username, password, FALSE );
 449
 450             EndDialog( hdlg, ERROR_INTERNET_FORCE_RETRY );
 451             return TRUE;
 452         }
 453         if( wParam == IDCANCEL )
 454         {
 455             EndDialog( hdlg, 0 );
 456             return TRUE;
 457         }
 458         break;
 459     }
 460     return FALSE;
 461 }
 462
 463 /***********************************************************************
 464  *         WININET_GetConnectionStatus
 465  */
 466 static INT WININET_GetConnectionStatus( HINTERNET hRequest )
 467 {
 468     WCHAR szStatus[0x20];
 469     DWORD sz, index, dwStatus;
 470
 471     TRACE("%p\n", hRequest );
 472
 473     sz = sizeof szStatus;
 474     index = 0;
 475     if( !HttpQueryInfoW( hRequest, HTTP_QUERY_STATUS_CODE,
 476                     szStatus, &sz, &index))
 477         return -1;
 478     dwStatus = atoiW( szStatus );
 479
 480     TRACE("request %p status = %d\n", hRequest, dwStatus );
 481
 482     return dwStatus;
 483 }
 484
 485
 486 /***********************************************************************
 487  *         InternetErrorDlg
 488  */
 489 DWORD WINAPI InternetErrorDlg(HWND hWnd, HINTERNET hRequest,
 490                  DWORD dwError, DWORD dwFlags, LPVOID* lppvData)
 491 {
 492     struct WININET_ErrorDlgParams params;
 493     HMODULE hwininet = GetModuleHandleA( "wininet.dll" );
 494     INT dwStatus;
 495
 496     TRACE("%p %p %d %08x %p\n", hWnd, hRequest, dwError, dwFlags, lppvData);
 497
 498     params.hWnd = hWnd;
 499     params.hRequest = hRequest;
 500     params.dwError = dwError;
 501     params.dwFlags = dwFlags;
 502     params.lppvData = lppvData;
 503
 504     switch( dwError )
 505     {
 506     case ERROR_SUCCESS:
 507     case ERROR_INTERNET_INCORRECT_PASSWORD:
 508         if( !dwError && !(dwFlags & FLAGS_ERROR_UI_FILTER_FOR_ERRORS ) )
 509             return 0;
 510
 511         dwStatus = WININET_GetConnectionStatus( hRequest );
 512         switch (dwStatus)
 513         {
 514         case HTTP_STATUS_PROXY_AUTH_REQ:
 515             return DialogBoxParamW( hwininet, MAKEINTRESOURCEW( IDD_PROXYDLG ),
 516                                     hWnd, WININET_ProxyPasswordDialog, (LPARAM) &params );
 517         case HTTP_STATUS_DENIED:
 518             return DialogBoxParamW( hwininet, MAKEINTRESOURCEW( IDD_AUTHDLG ),
 519                                     hWnd, WININET_PasswordDialog, (LPARAM) &params );
 520         default:
 521             WARN("unhandled status %u\n", dwStatus);
 522             return 0;
 523         }
 524
 525     case ERROR_INTERNET_HTTP_TO_HTTPS_ON_REDIR:
 526     case ERROR_INTERNET_INVALID_CA:
 527     case ERROR_INTERNET_POST_IS_NON_SECURE:
 528     case ERROR_INTERNET_SEC_CERT_CN_INVALID:
 529     case ERROR_INTERNET_SEC_CERT_DATE_INVALID:
 530         FIXME("Need to display dialog for error %d\n", dwError);
 531         return ERROR_SUCCESS;
 532     }
 533     return ERROR_INVALID_PARAMETER;
 534 }