379 lines
11 KiB
C
379 lines
11 KiB
C
/*
|
|
* Wininet
|
|
*
|
|
* Copyright 2003 Mike McCormack for CodeWeavers Inc.
|
|
*
|
|
* 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., 59 Temple Place, Suite 330, Boston, MA 02111-1307 USA
|
|
*/
|
|
|
|
#include "config.h"
|
|
|
|
#include <stdarg.h>
|
|
|
|
#include "windef.h"
|
|
#include "winbase.h"
|
|
#include "winuser.h"
|
|
#include "winreg.h"
|
|
#include "wininet.h"
|
|
#include "winnetwk.h"
|
|
#include "winnls.h"
|
|
#include "wine/debug.h"
|
|
#include "winerror.h"
|
|
#define NO_SHLWAPI_STREAM
|
|
#include "shlwapi.h"
|
|
|
|
#include "internet.h"
|
|
|
|
#include "wine/unicode.h"
|
|
|
|
#include "resource.h"
|
|
|
|
WINE_DEFAULT_DEBUG_CHANNEL(wininet);
|
|
|
|
struct WININET_ErrorDlgParams
|
|
{
|
|
HWND hWnd;
|
|
HINTERNET hRequest;
|
|
DWORD dwError;
|
|
DWORD dwFlags;
|
|
LPVOID* lppvData;
|
|
};
|
|
|
|
/***********************************************************************
|
|
* WININET_GetProxyServer
|
|
*
|
|
* Determine the name of the proxy server the request is using
|
|
*/
|
|
static BOOL WININET_GetProxyServer( HINTERNET hRequest, LPWSTR szBuf, DWORD sz )
|
|
{
|
|
LPWININETHTTPREQW lpwhr = (LPWININETHTTPREQW) hRequest;
|
|
LPWININETHTTPSESSIONW lpwhs = NULL;
|
|
LPWININETAPPINFOW hIC = NULL;
|
|
LPWSTR p;
|
|
|
|
if (NULL == lpwhr)
|
|
return FALSE;
|
|
|
|
lpwhs = (LPWININETHTTPSESSIONW) lpwhr->hdr.lpwhparent;
|
|
if (NULL == lpwhs)
|
|
return FALSE;
|
|
|
|
hIC = (LPWININETAPPINFOW) lpwhs->hdr.lpwhparent;
|
|
if (NULL == hIC)
|
|
return FALSE;
|
|
|
|
strncpyW(szBuf, hIC->lpszProxy, sz);
|
|
|
|
/* FIXME: perhaps it would be better to use InternetCrackUrl here */
|
|
p = strchrW(szBuf, ':');
|
|
if(*p)
|
|
*p = 0;
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
/***********************************************************************
|
|
* WININET_GetAuthRealm
|
|
*
|
|
* Determine the name of the (basic) Authentication realm
|
|
*/
|
|
static BOOL WININET_GetAuthRealm( HINTERNET hRequest, LPWSTR szBuf, DWORD sz )
|
|
{
|
|
LPWSTR p, q;
|
|
DWORD index;
|
|
const WCHAR szRealm[] = { 'r','e','a','l','m','=',0 };
|
|
|
|
/* extract the Realm from the proxy response and show it */
|
|
index = 0;
|
|
if( !HttpQueryInfoW( hRequest, HTTP_QUERY_PROXY_AUTHENTICATE,
|
|
szBuf, &sz, &index) )
|
|
return FALSE;
|
|
|
|
/*
|
|
* FIXME: maybe we should check that we're
|
|
* dealing with 'Basic' Authentication
|
|
*/
|
|
p = strchrW( szBuf, ' ' );
|
|
if( p && !strncmpW( p+1, szRealm, strlenW(szRealm) ) )
|
|
{
|
|
/* remove quotes */
|
|
p += 7;
|
|
if( *p == '"' )
|
|
{
|
|
p++;
|
|
q = strrchrW( p, '"' );
|
|
if( q )
|
|
*q = 0;
|
|
}
|
|
}
|
|
|
|
strcpyW( szBuf, p );
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
/***********************************************************************
|
|
* WININET_GetSetPassword
|
|
*/
|
|
static BOOL WININET_GetSetPassword( HWND hdlg, LPCWSTR szServer,
|
|
LPCWSTR szRealm, BOOL bSet )
|
|
{
|
|
WCHAR szResource[0x80], szUserPass[0x40];
|
|
LPWSTR p;
|
|
HWND hUserItem, hPassItem;
|
|
DWORD r, dwMagic = 19;
|
|
UINT r_len, u_len;
|
|
WORD sz;
|
|
const WCHAR szColon[] = { ':',0 }, szbs[] = { '/', 0 };
|
|
|
|
hUserItem = GetDlgItem( hdlg, IDC_USERNAME );
|
|
hPassItem = GetDlgItem( hdlg, IDC_PASSWORD );
|
|
|
|
/* now try fetch the username and password */
|
|
lstrcpyW( szResource, szServer);
|
|
lstrcatW( szResource, szbs);
|
|
lstrcatW( szResource, szRealm);
|
|
|
|
/*
|
|
* WNetCachePassword is only concerned with the length
|
|
* of the data stored (which we tell it) and it does
|
|
* not use strlen() internally so we can add WCHAR data
|
|
* instead of ASCII data and get it back the same way.
|
|
*/
|
|
if( bSet )
|
|
{
|
|
szUserPass[0] = 0;
|
|
GetWindowTextW( hUserItem, szUserPass,
|
|
(sizeof szUserPass-1)/sizeof(WCHAR) );
|
|
lstrcatW(szUserPass, szColon);
|
|
u_len = strlenW( szUserPass );
|
|
GetWindowTextW( hPassItem, szUserPass+u_len,
|
|
(sizeof szUserPass)/sizeof(WCHAR)-u_len );
|
|
|
|
r_len = (strlenW( szResource ) + 1)*sizeof(WCHAR);
|
|
u_len = (strlenW( szUserPass ) + 1)*sizeof(WCHAR);
|
|
r = WNetCachePassword( (CHAR*)szResource, r_len,
|
|
(CHAR*)szUserPass, u_len, dwMagic, 0 );
|
|
|
|
return ( r == WN_SUCCESS );
|
|
}
|
|
|
|
sz = sizeof szUserPass;
|
|
r_len = (strlenW( szResource ) + 1)*sizeof(WCHAR);
|
|
r = WNetGetCachedPassword( (CHAR*)szResource, r_len,
|
|
(CHAR*)szUserPass, &sz, dwMagic );
|
|
if( r != WN_SUCCESS )
|
|
return FALSE;
|
|
|
|
p = strchrW( szUserPass, ':' );
|
|
if( p )
|
|
{
|
|
*p = 0;
|
|
SetWindowTextW( hUserItem, szUserPass );
|
|
SetWindowTextW( hPassItem, p+1 );
|
|
}
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
/***********************************************************************
|
|
* WININET_SetProxyAuthorization
|
|
*/
|
|
static BOOL WININET_SetProxyAuthorization( HINTERNET hRequest,
|
|
LPWSTR username, LPWSTR password )
|
|
{
|
|
LPWININETHTTPREQW lpwhr = (LPWININETHTTPREQW) hRequest;
|
|
LPWININETHTTPSESSIONW lpwhs;
|
|
LPWININETAPPINFOW hIC;
|
|
LPWSTR p;
|
|
|
|
lpwhs = (LPWININETHTTPSESSIONW) lpwhr->hdr.lpwhparent;
|
|
if (NULL == lpwhs || lpwhs->hdr.htype != WH_HHTTPSESSION)
|
|
{
|
|
INTERNET_SetLastError(ERROR_INTERNET_INCORRECT_HANDLE_TYPE);
|
|
return FALSE;
|
|
}
|
|
|
|
hIC = (LPWININETAPPINFOW) lpwhs->hdr.lpwhparent;
|
|
|
|
p = HeapAlloc( GetProcessHeap(), 0, (strlenW( username ) + 1)*sizeof(WCHAR) );
|
|
if( !p )
|
|
return FALSE;
|
|
|
|
lstrcpyW( p, username );
|
|
hIC->lpszProxyUsername = p;
|
|
|
|
p = HeapAlloc( GetProcessHeap(), 0, (strlenW( password ) + 1)*sizeof(WCHAR) );
|
|
if( !p )
|
|
return FALSE;
|
|
|
|
lstrcpyW( p, password );
|
|
hIC->lpszProxyPassword = p;
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
/***********************************************************************
|
|
* WININET_ProxyPasswordDialog
|
|
*/
|
|
static INT_PTR WINAPI WININET_ProxyPasswordDialog(
|
|
HWND hdlg, UINT uMsg, WPARAM wParam, LPARAM lParam )
|
|
{
|
|
HWND hitem;
|
|
struct WININET_ErrorDlgParams *params;
|
|
WCHAR szRealm[0x80], szServer[0x80];
|
|
|
|
if( uMsg == WM_INITDIALOG )
|
|
{
|
|
TRACE("WM_INITDIALOG (%08lx)\n", lParam);
|
|
|
|
/* save the parameter list */
|
|
params = (struct WININET_ErrorDlgParams*) lParam;
|
|
SetWindowLongW( hdlg, GWL_USERDATA, lParam );
|
|
|
|
/* extract the Realm from the proxy response and show it */
|
|
if( WININET_GetAuthRealm( params->hRequest,
|
|
szRealm, sizeof szRealm/sizeof(WCHAR)) )
|
|
{
|
|
hitem = GetDlgItem( hdlg, IDC_REALM );
|
|
SetWindowTextW( hitem, szRealm );
|
|
}
|
|
|
|
/* extract the name of the proxy server */
|
|
if( WININET_GetProxyServer( params->hRequest,
|
|
szServer, sizeof szServer/sizeof(WCHAR)) )
|
|
{
|
|
hitem = GetDlgItem( hdlg, IDC_PROXY );
|
|
SetWindowTextW( hitem, szServer );
|
|
}
|
|
|
|
WININET_GetSetPassword( hdlg, szServer, szRealm, FALSE );
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
params = (struct WININET_ErrorDlgParams*)
|
|
GetWindowLongW( hdlg, GWL_USERDATA );
|
|
|
|
switch( uMsg )
|
|
{
|
|
case WM_COMMAND:
|
|
if( wParam == IDOK )
|
|
{
|
|
LPWININETHTTPREQW lpwhr = (LPWININETHTTPREQW) params->hRequest;
|
|
WCHAR username[0x20], password[0x20];
|
|
|
|
username[0] = 0;
|
|
hitem = GetDlgItem( hdlg, IDC_USERNAME );
|
|
if( hitem )
|
|
GetWindowTextW( hitem, username, sizeof username/sizeof(WCHAR) );
|
|
|
|
password[0] = 0;
|
|
hitem = GetDlgItem( hdlg, IDC_PASSWORD );
|
|
if( hitem )
|
|
GetWindowTextW( hitem, password, sizeof password/sizeof(WCHAR) );
|
|
|
|
hitem = GetDlgItem( hdlg, IDC_SAVEPASSWORD );
|
|
if( hitem &&
|
|
SendMessageW( hitem, BM_GETSTATE, 0, 0 ) &&
|
|
WININET_GetAuthRealm( params->hRequest,
|
|
szRealm, sizeof szRealm/sizeof(WCHAR)) &&
|
|
WININET_GetProxyServer( params->hRequest,
|
|
szServer, sizeof szServer/sizeof(WCHAR)) )
|
|
{
|
|
WININET_GetSetPassword( hdlg, szServer, szRealm, TRUE );
|
|
}
|
|
WININET_SetProxyAuthorization( lpwhr, username, password );
|
|
|
|
EndDialog( hdlg, ERROR_INTERNET_FORCE_RETRY );
|
|
return TRUE;
|
|
}
|
|
if( wParam == IDCANCEL )
|
|
{
|
|
EndDialog( hdlg, 0 );
|
|
return TRUE;
|
|
}
|
|
break;
|
|
}
|
|
return FALSE;
|
|
}
|
|
|
|
/***********************************************************************
|
|
* WININET_GetConnectionStatus
|
|
*/
|
|
static INT WININET_GetConnectionStatus( HINTERNET hRequest )
|
|
{
|
|
WCHAR szStatus[0x20];
|
|
DWORD sz, index, dwStatus;
|
|
|
|
TRACE("%p\n", hRequest );
|
|
|
|
sz = sizeof szStatus;
|
|
index = 0;
|
|
if( !HttpQueryInfoW( hRequest, HTTP_QUERY_STATUS_CODE,
|
|
szStatus, &sz, &index))
|
|
return -1;
|
|
dwStatus = atoiW( szStatus );
|
|
|
|
TRACE("request %p status = %ld\n", hRequest, dwStatus );
|
|
|
|
return dwStatus;
|
|
}
|
|
|
|
|
|
/***********************************************************************
|
|
* InternetErrorDlg
|
|
*/
|
|
DWORD WINAPI InternetErrorDlg(HWND hWnd, HINTERNET hRequest,
|
|
DWORD dwError, DWORD dwFlags, LPVOID* lppvData)
|
|
{
|
|
struct WININET_ErrorDlgParams params;
|
|
HMODULE hwininet = GetModuleHandleA( "wininet.dll" );
|
|
INT dwStatus;
|
|
|
|
TRACE("%p %p %ld %08lx %p\n", hWnd, hRequest, dwError, dwFlags, lppvData);
|
|
|
|
params.hWnd = hWnd;
|
|
params.hRequest = hRequest;
|
|
params.dwError = dwError;
|
|
params.dwFlags = dwFlags;
|
|
params.lppvData = lppvData;
|
|
|
|
switch( dwError )
|
|
{
|
|
case ERROR_SUCCESS:
|
|
if( !(dwFlags & FLAGS_ERROR_UI_FILTER_FOR_ERRORS ) )
|
|
return 0;
|
|
dwStatus = WININET_GetConnectionStatus( hRequest );
|
|
if( HTTP_STATUS_PROXY_AUTH_REQ != dwStatus )
|
|
return ERROR_SUCCESS;
|
|
return DialogBoxParamW( hwininet, MAKEINTRESOURCEW( IDD_PROXYDLG ),
|
|
hWnd, WININET_ProxyPasswordDialog, (LPARAM) ¶ms );
|
|
|
|
case ERROR_INTERNET_INCORRECT_PASSWORD:
|
|
return DialogBoxParamW( hwininet, MAKEINTRESOURCEW( IDD_PROXYDLG ),
|
|
hWnd, WININET_ProxyPasswordDialog, (LPARAM) ¶ms );
|
|
|
|
case ERROR_INTERNET_HTTP_TO_HTTPS_ON_REDIR:
|
|
case ERROR_INTERNET_INVALID_CA:
|
|
case ERROR_INTERNET_POST_IS_NON_SECURE:
|
|
case ERROR_INTERNET_SEC_CERT_CN_INVALID:
|
|
case ERROR_INTERNET_SEC_CERT_DATE_INVALID:
|
|
FIXME("Need to display dialog for error %ld\n", dwError);
|
|
return ERROR_SUCCESS;
|
|
}
|
|
return ERROR_INVALID_PARAMETER;
|
|
}
|