git.oblomov.eu Git - wine/blob - 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., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
  19  */
  20
  21 #include "config.h"
  22
  23 #include <stdarg.h>
  24
  25 #include "windef.h"
  26 #include "winbase.h"
  27 #include "winuser.h"
  28 #include "winreg.h"
  29 #include "wininet.h"
  30 #include "winnetwk.h"
  31 #include "winnls.h"
  32 #include "wine/debug.h"
  33 #include "winerror.h"
  34 #define NO_SHLWAPI_STREAM
  35 #include "shlwapi.h"
  36
  37 #include "internet.h"
  38
  39 #include "wine/unicode.h"
  40
  41 #include "resource.h"
  42
  43 WINE_DEFAULT_DEBUG_CHANNEL(wininet);
  44
  45 struct WININET_ErrorDlgParams
  46 {
  47     HWND       hWnd;
  48     HINTERNET  hRequest;
  49     DWORD      dwError;
  50     DWORD      dwFlags;
  51     LPVOID*    lppvData;
  52 };
  53
  54 /***********************************************************************
  55  *         WININET_GetProxyServer
  56  *
  57  *  Determine the name of the proxy server the request is using
  58  */
  59 static BOOL WININET_GetProxyServer( HINTERNET hRequest, LPWSTR szBuf, DWORD sz )
  60 {
  61     LPWININETHTTPREQW lpwhr;
  62     LPWININETHTTPSESSIONW lpwhs = NULL;
  63     LPWININETAPPINFOW hIC = NULL;
  64     LPWSTR p;
  65
  66     lpwhr = (LPWININETHTTPREQW) WININET_GetObject( hRequest );
  67     if (NULL == lpwhr)
  68         return FALSE;
  69
  70     lpwhs = (LPWININETHTTPSESSIONW) lpwhr->hdr.lpwhparent;
  71     if (NULL == lpwhs)
  72         return FALSE;
  73
  74     hIC = (LPWININETAPPINFOW) lpwhs->hdr.lpwhparent;
  75     if (NULL == hIC)
  76         return FALSE;
  77
  78     strncpyW(szBuf, hIC->lpszProxy, sz);
  79
  80     /* FIXME: perhaps it would be better to use InternetCrackUrl here */
  81     p = strchrW(szBuf, ':');
  82     if(*p)
  83         *p = 0;
  84
  85     return TRUE;
  86 }
  87
  88 /***********************************************************************
  89  *         WININET_GetAuthRealm
  90  *
  91  *  Determine the name of the (basic) Authentication realm
  92  */
  93 static BOOL WININET_GetAuthRealm( HINTERNET hRequest, LPWSTR szBuf, DWORD sz )
  94 {
  95     LPWSTR p, q;
  96     DWORD index;
  97     static const WCHAR szRealm[] = { 'r','e','a','l','m','=',0 };
  98
  99     /* extract the Realm from the proxy response and show it */
 100     index = 0;
 101     if( !HttpQueryInfoW( hRequest, HTTP_QUERY_PROXY_AUTHENTICATE,
 102                          szBuf, &sz, &index) )
 103         return FALSE;
 104
 105     /*
 106      * FIXME: maybe we should check that we're
 107      * dealing with 'Basic' Authentication
 108      */
 109     p = strchrW( szBuf, ' ' );
 110     if( p && !strncmpW( p+1, szRealm, strlenW(szRealm) ) )
 111     {
 112         /* remove quotes */
 113         p += 7;
 114         if( *p == '"' )
 115         {
 116             p++;
 117             q = strrchrW( p, '"' );
 118             if( q )
 119                 *q = 0;
 120         }
 121     }
 122
 123     strcpyW( szBuf, p );
 124
 125     return TRUE;
 126 }
 127
 128 /***********************************************************************
 129  *         WININET_GetSetPassword
 130  */
 131 static BOOL WININET_GetSetPassword( HWND hdlg, LPCWSTR szServer,
 132                                     LPCWSTR szRealm, BOOL bSet )
 133 {
 134     WCHAR szResource[0x80], szUserPass[0x40];
 135     LPWSTR p;
 136     HWND hUserItem, hPassItem;
 137     DWORD r, dwMagic = 19;
 138     UINT r_len, u_len;
 139     WORD sz;
 140     static const WCHAR szColon[] = { ':',0 };
 141     static const WCHAR szbs[] = { '/', 0 };
 142
 143     hUserItem = GetDlgItem( hdlg, IDC_USERNAME );
 144     hPassItem = GetDlgItem( hdlg, IDC_PASSWORD );
 145
 146     /* now try fetch the username and password */
 147     lstrcpyW( szResource, szServer);
 148     lstrcatW( szResource, szbs);
 149     lstrcatW( szResource, szRealm);
 150
 151     /*
 152      * WNetCachePassword is only concerned with the length
 153      * of the data stored (which we tell it) and it does
 154      * not use strlen() internally so we can add WCHAR data
 155      * instead of ASCII data and get it back the same way.
 156      */
 157     if( bSet )
 158     {
 159         szUserPass[0] = 0;
 160         GetWindowTextW( hUserItem, szUserPass,
 161                         (sizeof szUserPass-1)/sizeof(WCHAR) );
 162         lstrcatW(szUserPass, szColon);
 163         u_len = strlenW( szUserPass );
 164         GetWindowTextW( hPassItem, szUserPass+u_len,
 165                         (sizeof szUserPass)/sizeof(WCHAR)-u_len );
 166
 167         r_len = (strlenW( szResource ) + 1)*sizeof(WCHAR);
 168         u_len = (strlenW( szUserPass ) + 1)*sizeof(WCHAR);
 169         r = WNetCachePassword( (CHAR*)szResource, r_len,
 170                                (CHAR*)szUserPass, u_len, dwMagic, 0 );
 171
 172         return ( r == WN_SUCCESS );
 173     }
 174
 175     sz = sizeof szUserPass;
 176     r_len = (strlenW( szResource ) + 1)*sizeof(WCHAR);
 177     r = WNetGetCachedPassword( (CHAR*)szResource, r_len,
 178                                (CHAR*)szUserPass, &sz, dwMagic );
 179     if( r != WN_SUCCESS )
 180         return FALSE;
 181
 182     p = strchrW( szUserPass, ':' );
 183     if( p )
 184     {
 185         *p = 0;
 186         SetWindowTextW( hUserItem, szUserPass );
 187         SetWindowTextW( hPassItem, p+1 );
 188     }
 189
 190     return TRUE;
 191 }
 192
 193 /***********************************************************************
 194  *         WININET_SetProxyAuthorization
 195  */
 196 static BOOL WININET_SetProxyAuthorization( HINTERNET hRequest,
 197                                          LPWSTR username, LPWSTR password )
 198 {
 199     LPWININETHTTPREQW lpwhr;
 200     LPWININETHTTPSESSIONW lpwhs;
 201     LPWININETAPPINFOW hIC;
 202     LPWSTR p;
 203
 204     lpwhr = (LPWININETHTTPREQW) WININET_GetObject( hRequest );
 205     if( !lpwhr )
 206         return FALSE;
 207
 208     lpwhs = (LPWININETHTTPSESSIONW) lpwhr->hdr.lpwhparent;
 209     if (NULL == lpwhs ||  lpwhs->hdr.htype != WH_HHTTPSESSION)
 210     {
 211         INTERNET_SetLastError(ERROR_INTERNET_INCORRECT_HANDLE_TYPE);
 212         return FALSE;
 213     }
 214
 215     hIC = (LPWININETAPPINFOW) lpwhs->hdr.lpwhparent;
 216
 217     p = HeapAlloc( GetProcessHeap(), 0, (strlenW( username ) + 1)*sizeof(WCHAR) );
 218     if( !p )
 219         return FALSE;
 220
 221     lstrcpyW( p, username );
 222     hIC->lpszProxyUsername = p;
 223
 224     p = HeapAlloc( GetProcessHeap(), 0, (strlenW( password ) + 1)*sizeof(WCHAR) );
 225     if( !p )
 226         return FALSE;
 227
 228     lstrcpyW( p, password );
 229     hIC->lpszProxyPassword = p;
 230
 231     return TRUE;
 232 }
 233
 234 /***********************************************************************
 235  *         WININET_ProxyPasswordDialog
 236  */
 237 static INT_PTR WINAPI WININET_ProxyPasswordDialog(
 238     HWND hdlg, UINT uMsg, WPARAM wParam, LPARAM lParam )
 239 {
 240     HWND hitem;
 241     struct WININET_ErrorDlgParams *params;
 242     WCHAR szRealm[0x80], szServer[0x80];
 243
 244     if( uMsg == WM_INITDIALOG )
 245     {
 246         TRACE("WM_INITDIALOG (%08lx)\n", lParam);
 247
 248         /* save the parameter list */
 249         params = (struct WININET_ErrorDlgParams*) lParam;
 250         SetWindowLongW( hdlg, GWL_USERDATA, lParam );
 251
 252         /* extract the Realm from the proxy response and show it */
 253         if( WININET_GetAuthRealm( params->hRequest,
 254                                   szRealm, sizeof szRealm/sizeof(WCHAR)) )
 255         {
 256             hitem = GetDlgItem( hdlg, IDC_REALM );
 257             SetWindowTextW( hitem, szRealm );
 258         }
 259
 260         /* extract the name of the proxy server */
 261         if( WININET_GetProxyServer( params->hRequest,
 262                                     szServer, sizeof szServer/sizeof(WCHAR)) )
 263         {
 264             hitem = GetDlgItem( hdlg, IDC_PROXY );
 265             SetWindowTextW( hitem, szServer );
 266         }
 267
 268         WININET_GetSetPassword( hdlg, szServer, szRealm, FALSE );
 269
 270         return TRUE;
 271     }
 272
 273     params = (struct WININET_ErrorDlgParams*)
 274                  GetWindowLongW( hdlg, GWL_USERDATA );
 275
 276     switch( uMsg )
 277     {
 278     case WM_COMMAND:
 279         if( wParam == IDOK )
 280         {
 281             WCHAR username[0x20], password[0x20];
 282
 283             username[0] = 0;
 284             hitem = GetDlgItem( hdlg, IDC_USERNAME );
 285             if( hitem )
 286                 GetWindowTextW( hitem, username, sizeof username/sizeof(WCHAR) );
 287
 288             password[0] = 0;
 289             hitem = GetDlgItem( hdlg, IDC_PASSWORD );
 290             if( hitem )
 291                 GetWindowTextW( hitem, password, sizeof password/sizeof(WCHAR) );
 292
 293             hitem = GetDlgItem( hdlg, IDC_SAVEPASSWORD );
 294             if( hitem &&
 295                 SendMessageW( hitem, BM_GETSTATE, 0, 0 ) &&
 296                 WININET_GetAuthRealm( params->hRequest,
 297                                   szRealm, sizeof szRealm/sizeof(WCHAR)) &&
 298                 WININET_GetProxyServer( params->hRequest,
 299                                     szServer, sizeof szServer/sizeof(WCHAR)) )
 300             {
 301                 WININET_GetSetPassword( hdlg, szServer, szRealm, TRUE );
 302             }
 303             WININET_SetProxyAuthorization( params->hRequest, username, password );
 304
 305             EndDialog( hdlg, ERROR_INTERNET_FORCE_RETRY );
 306             return TRUE;
 307         }
 308         if( wParam == IDCANCEL )
 309         {
 310             EndDialog( hdlg, 0 );
 311             return TRUE;
 312         }
 313         break;
 314     }
 315     return FALSE;
 316 }
 317
 318 /***********************************************************************
 319  *         WININET_GetConnectionStatus
 320  */
 321 static INT WININET_GetConnectionStatus( HINTERNET hRequest )
 322 {
 323     WCHAR szStatus[0x20];
 324     DWORD sz, index, dwStatus;
 325
 326     TRACE("%p\n", hRequest );
 327
 328     sz = sizeof szStatus;
 329     index = 0;
 330     if( !HttpQueryInfoW( hRequest, HTTP_QUERY_STATUS_CODE,
 331                     szStatus, &sz, &index))
 332         return -1;
 333     dwStatus = atoiW( szStatus );
 334
 335     TRACE("request %p status = %ld\n", hRequest, dwStatus );
 336
 337     return dwStatus;
 338 }
 339
 340
 341 /***********************************************************************
 342  *         InternetErrorDlg
 343  */
 344 DWORD WINAPI InternetErrorDlg(HWND hWnd, HINTERNET hRequest,
 345                  DWORD dwError, DWORD dwFlags, LPVOID* lppvData)
 346 {
 347     struct WININET_ErrorDlgParams params;
 348     HMODULE hwininet = GetModuleHandleA( "wininet.dll" );
 349     INT dwStatus;
 350
 351     TRACE("%p %p %ld %08lx %p\n", hWnd, hRequest, dwError, dwFlags, lppvData);
 352
 353     params.hWnd = hWnd;
 354     params.hRequest = hRequest;
 355     params.dwError = dwError;
 356     params.dwFlags = dwFlags;
 357     params.lppvData = lppvData;
 358
 359     switch( dwError )
 360     {
 361     case ERROR_SUCCESS:
 362         if( !(dwFlags & FLAGS_ERROR_UI_FILTER_FOR_ERRORS ) )
 363             return 0;
 364         dwStatus = WININET_GetConnectionStatus( hRequest );
 365         if( HTTP_STATUS_PROXY_AUTH_REQ != dwStatus )
 366             return ERROR_SUCCESS;
 367         return DialogBoxParamW( hwininet, MAKEINTRESOURCEW( IDD_PROXYDLG ),
 368                     hWnd, WININET_ProxyPasswordDialog, (LPARAM) &params );
 369
 370     case ERROR_INTERNET_INCORRECT_PASSWORD:
 371         return DialogBoxParamW( hwininet, MAKEINTRESOURCEW( IDD_PROXYDLG ),
 372                     hWnd, WININET_ProxyPasswordDialog, (LPARAM) &params );
 373
 374     case ERROR_INTERNET_HTTP_TO_HTTPS_ON_REDIR:
 375     case ERROR_INTERNET_INVALID_CA:
 376     case ERROR_INTERNET_POST_IS_NON_SECURE:
 377     case ERROR_INTERNET_SEC_CERT_CN_INVALID:
 378     case ERROR_INTERNET_SEC_CERT_DATE_INVALID:
 379         FIXME("Need to display dialog for error %ld\n", dwError);
 380         return ERROR_SUCCESS;
 381     }
 382     return ERROR_INVALID_PARAMETER;
 383 }