@ -1,7 +1,7 @@
/*
* Rufus : The Reliable USB Formatting Utility
* Networking functionality ( web file download , check for update , etc . )
* Copyright © 2012 - 2021 Pete Batard < pete @ akeo . ie >
* Copyright © 2012 - 2022 Pete Batard < pete @ akeo . ie >
*
* This program is free software : you can redistribute it and / or modify
* it under the terms of the GNU General Public License as published by
@ -69,8 +69,6 @@ const char* WinInetErrorString(void)
{
static char error_string [ 256 ] ;
DWORD size = sizeof ( error_string ) ;
PF_TYPE_DECL ( WINAPI , BOOL , InternetGetLastResponseInfoA , ( LPDWORD , LPSTR , LPDWORD ) ) ;
PF_INIT ( InternetGetLastResponseInfoA , WinInet ) ;
error_code = HRESULT_CODE ( GetLastError ( ) ) ;
@ -221,10 +219,8 @@ const char* WinInetErrorString(void)
case ERROR_INTERNET_LOGIN_FAILURE_DISPLAY_ENTITY_BODY :
return " Please ask Microsoft about that one! " ;
case ERROR_INTERNET_EXTENDED_ERROR :
if ( pfInternetGetLastResponseInfoA ! = NULL ) {
pfInternetGetLastResponseInfoA ( & error_code , error_string , & size ) ;
return error_string ;
}
InternetGetLastResponseInfoA ( & error_code , error_string , & size ) ;
return error_string ;
/ / fall through
default :
static_sprintf ( error_string , " Unknown internet error 0x%08lX " , error_code ) ;
@ -277,13 +273,6 @@ static HINTERNET GetInternetSession(BOOL bRetry)
HRESULT hr = S_FALSE ;
INetworkListManager * pNetworkListManager ;
PF_TYPE_DECL ( WINAPI , HINTERNET , InternetOpenA , ( LPCSTR , DWORD , LPCSTR , LPCSTR , DWORD ) ) ;
PF_TYPE_DECL ( WINAPI , BOOL , InternetSetOptionA , ( HINTERNET , DWORD , LPVOID , DWORD ) ) ;
PF_TYPE_DECL ( WINAPI , BOOL , InternetGetConnectedState , ( LPDWORD , DWORD ) ) ;
PF_INIT_OR_OUT ( InternetOpenA , WinInet ) ;
PF_INIT_OR_OUT ( InternetSetOptionA , WinInet ) ;
PF_INIT ( InternetGetConnectedState , WinInet ) ;
/ / Create a NetworkListManager Instance to check the network connection
IGNORE_RETVAL ( CoInitializeEx ( NULL , COINIT_APARTMENTTHREADED | COINIT_DISABLE_OLE1DDE ) ) ;
hr = CoCreateInstance ( & CLSID_NetworkListManager , NULL , CLSCTX_ALL ,
@ -294,8 +283,8 @@ static HINTERNET GetInternetSession(BOOL bRetry)
/ / INetworkListManager may fail with ERROR_SERVICE_DEPENDENCY_FAIL if the DHCP service
/ / is not running , in which case we must fall back to using InternetGetConnectedState ( ) .
/ / See https : / / github . com / pbatard / rufus / issues / 1801.
if ( ( hr = = HRESULT_FROM_WIN32 ( ERROR_SERVICE_DEPENDENCY_FAIL ) ) & & ( pfInternetGetConnectedState ! = NUL L ) ) {
InternetConnection = pf InternetGetConnectedState( & dwFlags , 0 ) ? VARIANT_TRUE : VARIANT_FALSE ;
if ( hr = = HRESULT_FROM_WIN32 ( ERROR_SERVICE_DEPENDENCY_FAIL ) ) {
InternetConnection = InternetGetConnectedState ( & dwFlags , 0 ) ? VARIANT_TRUE : VARIANT_FALSE ;
break ;
}
if ( hr = = S_OK | | ! bRetry )
@ -310,15 +299,15 @@ static HINTERNET GetInternetSession(BOOL bRetry)
static_sprintf ( agent , APPLICATION_NAME " /%d.%d.%d (Windows NT %d.%d%s) " ,
rufus_version [ 0 ] , rufus_version [ 1 ] , rufus_version [ 2 ] ,
nWindowsVersion > > 4 , nWindowsVersion & 0x0F , is_x64 ( ) ? " ; WOW64 " : " " ) ;
hSession = pf InternetOpenA( agent , INTERNET_OPEN_TYPE_PRECONFIG , NULL , NULL , 0 ) ;
hSession = InternetOpenA ( agent , INTERNET_OPEN_TYPE_PRECONFIG , NULL , NULL , 0 ) ;
/ / Set the timeouts
pf InternetSetOptionA( hSession , INTERNET_OPTION_CONNECT_TIMEOUT , ( LPVOID ) & dwTimeout , sizeof ( dwTimeout ) ) ;
pf InternetSetOptionA( hSession , INTERNET_OPTION_SEND_TIMEOUT , ( LPVOID ) & dwTimeout , sizeof ( dwTimeout ) ) ;
pf InternetSetOptionA( hSession , INTERNET_OPTION_RECEIVE_TIMEOUT , ( LPVOID ) & dwTimeout , sizeof ( dwTimeout ) ) ;
InternetSetOptionA ( hSession , INTERNET_OPTION_CONNECT_TIMEOUT , ( LPVOID ) & dwTimeout , sizeof ( dwTimeout ) ) ;
InternetSetOptionA ( hSession , INTERNET_OPTION_SEND_TIMEOUT , ( LPVOID ) & dwTimeout , sizeof ( dwTimeout ) ) ;
InternetSetOptionA ( hSession , INTERNET_OPTION_RECEIVE_TIMEOUT , ( LPVOID ) & dwTimeout , sizeof ( dwTimeout ) ) ;
/ / Enable gzip and deflate decoding schemes
pf InternetSetOptionA( hSession , INTERNET_OPTION_HTTP_DECODING , ( LPVOID ) & decodingSupport , sizeof ( decodingSupport ) ) ;
InternetSetOptionA ( hSession , INTERNET_OPTION_HTTP_DECODING , ( LPVOID ) & decodingSupport , sizeof ( decodingSupport ) ) ;
/ / Enable HTTP / 2 protocol support
pf InternetSetOptionA( hSession , INTERNET_OPTION_ENABLE_HTTP_PROTOCOL , ( LPVOID ) & dwProtocolSupport , sizeof ( dwProtocolSupport ) ) ;
InternetSetOptionA ( hSession , INTERNET_OPTION_ENABLE_HTTP_PROTOCOL , ( LPVOID ) & dwProtocolSupport , sizeof ( dwProtocolSupport ) ) ;
out :
return hSession ;
@ -348,22 +337,6 @@ uint64_t DownloadToFileOrBuffer(const char* url, const char* file, BYTE** buffer
hostname , sizeof ( hostname ) , 0 , NULL , 1 , urlpath , sizeof ( urlpath ) , NULL , 1 } ;
uint64_t size = 0 , total_size = 0 ;
/ / Can ' t link with wininet . lib because of sideloading issues
PF_TYPE_DECL ( WINAPI , BOOL , InternetCrackUrlA , ( LPCSTR , DWORD , DWORD , LPURL_COMPONENTSA ) ) ;
PF_TYPE_DECL ( WINAPI , HINTERNET , InternetConnectA , ( HINTERNET , LPCSTR , INTERNET_PORT , LPCSTR , LPCSTR , DWORD , DWORD , DWORD_PTR ) ) ;
PF_TYPE_DECL ( WINAPI , BOOL , InternetReadFile , ( HINTERNET , LPVOID , DWORD , LPDWORD ) ) ;
PF_TYPE_DECL ( WINAPI , BOOL , InternetCloseHandle , ( HINTERNET ) ) ;
PF_TYPE_DECL ( WINAPI , HINTERNET , HttpOpenRequestA , ( HINTERNET , LPCSTR , LPCSTR , LPCSTR , LPCSTR , LPCSTR * , DWORD , DWORD_PTR ) ) ;
PF_TYPE_DECL ( WINAPI , BOOL , HttpSendRequestA , ( HINTERNET , LPCSTR , DWORD , LPVOID , DWORD ) ) ;
PF_TYPE_DECL ( WINAPI , BOOL , HttpQueryInfoA , ( HINTERNET , DWORD , LPVOID , LPDWORD , LPDWORD ) ) ;
PF_INIT_OR_OUT ( InternetCrackUrlA , WinInet ) ;
PF_INIT_OR_OUT ( InternetConnectA , WinInet ) ;
PF_INIT_OR_OUT ( InternetReadFile , WinInet ) ;
PF_INIT_OR_OUT ( InternetCloseHandle , WinInet ) ;
PF_INIT_OR_OUT ( HttpOpenRequestA , WinInet ) ;
PF_INIT_OR_OUT ( HttpSendRequestA , WinInet ) ;
PF_INIT_OR_OUT ( HttpQueryInfoA , WinInet ) ;
FormatStatus = 0 ;
DownloadStatus = 404 ;
if ( hProgressDialog ! = NULL )
@ -380,7 +353,7 @@ uint64_t DownloadToFileOrBuffer(const char* url, const char* file, BYTE** buffer
uprintf ( " Downloading %s " , url ) ;
}
if ( ( ! pf InternetCrackUrlA( url , ( DWORD ) safe_strlen ( url ) , 0 , & UrlParts ) )
if ( ( ! InternetCrackUrlA ( url , ( DWORD ) safe_strlen ( url ) , 0 , & UrlParts ) )
| | ( UrlParts . lpszHostName = = NULL ) | | ( UrlParts . lpszUrlPath = = NULL ) ) {
uprintf ( " Unable to decode URL: %s " , WinInetErrorString ( ) ) ;
goto out ;
@ -393,13 +366,13 @@ uint64_t DownloadToFileOrBuffer(const char* url, const char* file, BYTE** buffer
goto out ;
}
hConnection = pf InternetConnectA( hSession , UrlParts . lpszHostName , UrlParts . nPort , NULL , NULL , INTERNET_SERVICE_HTTP , 0 , ( DWORD_PTR ) NULL ) ;
hConnection = InternetConnectA ( hSession , UrlParts . lpszHostName , UrlParts . nPort , NULL , NULL , INTERNET_SERVICE_HTTP , 0 , ( DWORD_PTR ) NULL ) ;
if ( hConnection = = NULL ) {
uprintf ( " Could not connect to server %s:%d: %s " , UrlParts . lpszHostName , UrlParts . nPort , WinInetErrorString ( ) ) ;
goto out ;
}
hRequest = pf HttpOpenRequestA( hConnection , " GET " , UrlParts . lpszUrlPath , NULL , NULL , accept_types ,
hRequest = HttpOpenRequestA ( hConnection , " GET " , UrlParts . lpszUrlPath , NULL , NULL , accept_types ,
INTERNET_FLAG_IGNORE_REDIRECT_TO_HTTP | INTERNET_FLAG_IGNORE_REDIRECT_TO_HTTPS |
INTERNET_FLAG_NO_COOKIES | INTERNET_FLAG_NO_UI | INTERNET_FLAG_NO_CACHE_WRITE | INTERNET_FLAG_HYPERLINK |
( ( UrlParts . nScheme = = INTERNET_SCHEME_HTTPS ) ? INTERNET_FLAG_SECURE : 0 ) , ( DWORD_PTR ) NULL ) ;
@ -408,14 +381,14 @@ uint64_t DownloadToFileOrBuffer(const char* url, const char* file, BYTE** buffer
goto out ;
}
if ( ! pf HttpSendRequestA( hRequest , request_headers , - 1L , NULL , 0 ) ) {
if ( ! HttpSendRequestA ( hRequest , request_headers , - 1L , NULL , 0 ) ) {
uprintf ( " Unable to send request: %s " , WinInetErrorString ( ) ) ;
goto out ;
}
/ / Get the file size
dwSize = sizeof ( DownloadStatus ) ;
pf HttpQueryInfoA( hRequest , HTTP_QUERY_STATUS_CODE | HTTP_QUERY_FLAG_NUMBER , ( LPVOID ) & DownloadStatus , & dwSize , NULL ) ;
HttpQueryInfoA ( hRequest , HTTP_QUERY_STATUS_CODE | HTTP_QUERY_FLAG_NUMBER , ( LPVOID ) & DownloadStatus , & dwSize , NULL ) ;
if ( DownloadStatus ! = 200 ) {
error_code = ERROR_INTERNET_ITEM_NOT_FOUND ;
SetLastError ( ERROR_SEVERITY_ERROR | FAC ( FACILITY_HTTP ) | error_code ) ;
@ -423,7 +396,7 @@ uint64_t DownloadToFileOrBuffer(const char* url, const char* file, BYTE** buffer
goto out ;
}
dwSize = sizeof ( strsize ) ;
if ( ! pf HttpQueryInfoA( hRequest , HTTP_QUERY_CONTENT_LENGTH , ( LPVOID ) strsize , & dwSize , NULL ) ) {
if ( ! HttpQueryInfoA ( hRequest , HTTP_QUERY_CONTENT_LENGTH , ( LPVOID ) strsize , & dwSize , NULL ) ) {
uprintf ( " Unable to retrieve file length: %s " , WinInetErrorString ( ) ) ;
goto out ;
}
@ -462,7 +435,7 @@ uint64_t DownloadToFileOrBuffer(const char* url, const char* file, BYTE** buffer
/ / User may have cancelled the download
if ( IS_ERROR ( FormatStatus ) )
goto out ;
if ( ! pf InternetReadFile( hRequest , buf , sizeof ( buf ) , & dwDownloaded ) | | ( dwDownloaded = = 0 ) )
if ( ! InternetReadFile ( hRequest , buf , sizeof ( buf ) , & dwDownloaded ) | | ( dwDownloaded = = 0 ) )
break ;
if ( hProgressDialog ! = NULL )
UpdateProgressWithInfo ( OP_NOOP , MSG_241 , size , total_size ) ;
@ -507,11 +480,11 @@ out:
safe_free ( * buffer ) ;
}
if ( hRequest )
pf InternetCloseHandle( hRequest ) ;
InternetCloseHandle ( hRequest ) ;
if ( hConnection )
pf InternetCloseHandle( hConnection ) ;
InternetCloseHandle ( hConnection ) ;
if ( hSession )
pf InternetCloseHandle( hSession ) ;
InternetCloseHandle ( hSession ) ;
SetLastError ( error_code ) ;
return r ? size : 0 ;
@ -639,22 +612,6 @@ static DWORD WINAPI CheckForUpdatesThread(LPVOID param)
FILETIME FileTime ;
int64_t local_time = 0 , reg_time , server_time , update_interval ;
/ / Can ' t link with wininet . lib because of sideloading issues
PF_TYPE_DECL ( WINAPI , BOOL , InternetCrackUrlA , ( LPCSTR , DWORD , DWORD , LPURL_COMPONENTSA ) ) ;
PF_TYPE_DECL ( WINAPI , HINTERNET , InternetConnectA , ( HINTERNET , LPCSTR , INTERNET_PORT , LPCSTR , LPCSTR , DWORD , DWORD , DWORD_PTR ) ) ;
PF_TYPE_DECL ( WINAPI , BOOL , InternetReadFile , ( HINTERNET , LPVOID , DWORD , LPDWORD ) ) ;
PF_TYPE_DECL ( WINAPI , BOOL , InternetCloseHandle , ( HINTERNET ) ) ;
PF_TYPE_DECL ( WINAPI , HINTERNET , HttpOpenRequestA , ( HINTERNET , LPCSTR , LPCSTR , LPCSTR , LPCSTR , LPCSTR * , DWORD , DWORD_PTR ) ) ;
PF_TYPE_DECL ( WINAPI , BOOL , HttpSendRequestA , ( HINTERNET , LPCSTR , DWORD , LPVOID , DWORD ) ) ;
PF_TYPE_DECL ( WINAPI , BOOL , HttpQueryInfoA , ( HINTERNET , DWORD , LPVOID , LPDWORD , LPDWORD ) ) ;
PF_INIT_OR_OUT ( InternetCrackUrlA , WinInet ) ;
PF_INIT_OR_OUT ( InternetConnectA , WinInet ) ;
PF_INIT_OR_OUT ( InternetReadFile , WinInet ) ;
PF_INIT_OR_OUT ( InternetCloseHandle , WinInet ) ;
PF_INIT_OR_OUT ( HttpOpenRequestA , WinInet ) ;
PF_INIT_OR_OUT ( HttpSendRequestA , WinInet ) ;
PF_INIT_OR_OUT ( HttpQueryInfoA , WinInet ) ;
verbose = ReadSetting32 ( SETTING_VERBOSE_UPDATES ) ;
/ / Without this the FileDialog will produce error 0x8001010E when compiled for Vista or later
IGNORE_RETVAL ( CoInitializeEx ( NULL , COINIT_APARTMENTTHREADED | COINIT_DISABLE_OLE1DDE ) ) ;
@ -697,7 +654,7 @@ static DWORD WINAPI CheckForUpdatesThread(LPVOID param)
goto out ;
}
if ( ! pf InternetCrackUrlA( server_url , ( DWORD ) safe_strlen ( server_url ) , 0 , & UrlParts ) )
if ( ! InternetCrackUrlA ( server_url , ( DWORD ) safe_strlen ( server_url ) , 0 , & UrlParts ) )
goto out ;
hostname [ sizeof ( hostname ) - 1 ] = 0 ;
@ -707,7 +664,7 @@ static DWORD WINAPI CheckForUpdatesThread(LPVOID param)
hSession = GetInternetSession ( FALSE ) ;
if ( hSession = = NULL )
goto out ;
hConnection = pf InternetConnectA( hSession , UrlParts . lpszHostName , UrlParts . nPort , NULL , NULL , INTERNET_SERVICE_HTTP , 0 , ( DWORD_PTR ) NULL ) ;
hConnection = InternetConnectA ( hSession , UrlParts . lpszHostName , UrlParts . nPort , NULL , NULL , INTERNET_SERVICE_HTTP , 0 , ( DWORD_PTR ) NULL ) ;
if ( hConnection = = NULL )
goto out ;
@ -747,11 +704,11 @@ static DWORD WINAPI CheckForUpdatesThread(LPVOID param)
UrlParts . dwUrlPathLength = sizeof ( urlpath ) ;
for ( i = 0 ; i < ARRAYSIZE ( verpos ) ; i + + ) {
vvuprintf ( " Trying %s " , UrlParts . lpszUrlPath ) ;
hRequest = pf HttpOpenRequestA( hConnection , " GET " , UrlParts . lpszUrlPath , NULL , NULL , accept_types ,
hRequest = HttpOpenRequestA ( hConnection , " GET " , UrlParts . lpszUrlPath , NULL , NULL , accept_types ,
INTERNET_FLAG_IGNORE_REDIRECT_TO_HTTP | INTERNET_FLAG_IGNORE_REDIRECT_TO_HTTPS |
INTERNET_FLAG_NO_COOKIES | INTERNET_FLAG_NO_UI | INTERNET_FLAG_NO_CACHE_WRITE | INTERNET_FLAG_HYPERLINK |
( ( UrlParts . nScheme = = INTERNET_SCHEME_HTTPS ) ? INTERNET_FLAG_SECURE : 0 ) , ( DWORD_PTR ) NULL ) ;
if ( ( hRequest = = NULL ) | | ( ! pf HttpSendRequestA( hRequest , request_headers , - 1L , NULL , 0 ) ) ) {
if ( ( hRequest = = NULL ) | | ( ! HttpSendRequestA ( hRequest , request_headers , - 1L , NULL , 0 ) ) ) {
uprintf ( " Unable to send request: %s " , WinInetErrorString ( ) ) ;
goto out ;
}
@ -759,10 +716,10 @@ static DWORD WINAPI CheckForUpdatesThread(LPVOID param)
/ / Ensure that we get a text file
dwSize = sizeof ( dwStatus ) ;
dwStatus = 404 ;
pf HttpQueryInfoA( hRequest , HTTP_QUERY_STATUS_CODE | HTTP_QUERY_FLAG_NUMBER , ( LPVOID ) & dwStatus , & dwSize , NULL ) ;
HttpQueryInfoA ( hRequest , HTTP_QUERY_STATUS_CODE | HTTP_QUERY_FLAG_NUMBER , ( LPVOID ) & dwStatus , & dwSize , NULL ) ;
if ( dwStatus = = 200 )
break ;
pf InternetCloseHandle( hRequest ) ;
InternetCloseHandle ( hRequest ) ;
hRequest = NULL ;
safe_strcpy ( & urlpath [ verpos [ i ] ] , 5 , " .ver " ) ;
}
@ -779,7 +736,7 @@ static DWORD WINAPI CheckForUpdatesThread(LPVOID param)
/ / On the other hand , if local clock is set way back in the past , we will never check .
dwSize = sizeof ( ServerTime ) ;
/ / If we can ' t get a date we can trust , don ' t bother . . .
if ( ( ! pf HttpQueryInfoA( hRequest , HTTP_QUERY_DATE | HTTP_QUERY_FLAG_SYSTEMTIME , ( LPVOID ) & ServerTime , & dwSize , NULL ) )
if ( ( ! HttpQueryInfoA ( hRequest , HTTP_QUERY_DATE | HTTP_QUERY_FLAG_SYSTEMTIME , ( LPVOID ) & ServerTime , & dwSize , NULL ) )
| | ( ! SystemTimeToFileTime ( & ServerTime , & FileTime ) ) )
goto out ;
server_time = ( ( ( ( int64_t ) FileTime . dwHighDateTime ) < < 32 ) + FileTime . dwLowDateTime ) / 10000000 ;
@ -795,7 +752,7 @@ static DWORD WINAPI CheckForUpdatesThread(LPVOID param)
}
dwSize = sizeof ( dwTotalSize ) ;
if ( ! pf HttpQueryInfoA( hRequest , HTTP_QUERY_CONTENT_LENGTH | HTTP_QUERY_FLAG_NUMBER , ( LPVOID ) & dwTotalSize , & dwSize , NULL ) )
if ( ! HttpQueryInfoA ( hRequest , HTTP_QUERY_CONTENT_LENGTH | HTTP_QUERY_FLAG_NUMBER , ( LPVOID ) & dwTotalSize , & dwSize , NULL ) )
goto out ;
/ / Make sure the file is NUL terminated
@ -803,7 +760,7 @@ static DWORD WINAPI CheckForUpdatesThread(LPVOID param)
if ( buf = = NULL )
goto out ;
/ / This is a version file - we should be able to gulp it down in one go
if ( ! pf InternetReadFile( hRequest , buf , dwTotalSize , & dwDownloaded ) | | ( dwDownloaded ! = dwTotalSize ) )
if ( ! InternetReadFile ( hRequest , buf , dwTotalSize , & dwDownloaded ) | | ( dwDownloaded ! = dwTotalSize ) )
goto out ;
vuprintf ( " Successfully downloaded version file (%d bytes) " , dwTotalSize ) ;
@ -834,11 +791,11 @@ out:
safe_free ( buf ) ;
safe_free ( sig ) ;
if ( hRequest )
pf InternetCloseHandle( hRequest ) ;
InternetCloseHandle ( hRequest ) ;
if ( hConnection )
pf InternetCloseHandle( hConnection ) ;
InternetCloseHandle ( hConnection ) ;
if ( hSession )
pf InternetCloseHandle( hSession ) ;
InternetCloseHandle ( hSession ) ;
switch ( status ) {
case 1 :
PrintInfoDebug ( 3000 , MSG_244 ) ;
@ -1094,26 +1051,13 @@ BOOL IsDownloadable(const char* url)
URL_COMPONENTSA UrlParts = { sizeof ( URL_COMPONENTSA ) , NULL , 1 , ( INTERNET_SCHEME ) 0 ,
hostname , sizeof ( hostname ) , 0 , NULL , 1 , urlpath , sizeof ( urlpath ) , NULL , 1 } ;
PF_TYPE_DECL ( WINAPI , BOOL , InternetCrackUrlA , ( LPCSTR , DWORD , DWORD , LPURL_COMPONENTSA ) ) ;
PF_TYPE_DECL ( WINAPI , HINTERNET , InternetConnectA , ( HINTERNET , LPCSTR , INTERNET_PORT , LPCSTR , LPCSTR , DWORD , DWORD , DWORD_PTR ) ) ;
PF_TYPE_DECL ( WINAPI , BOOL , InternetCloseHandle , ( HINTERNET ) ) ;
PF_TYPE_DECL ( WINAPI , HINTERNET , HttpOpenRequestA , ( HINTERNET , LPCSTR , LPCSTR , LPCSTR , LPCSTR , LPCSTR * , DWORD , DWORD_PTR ) ) ;
PF_TYPE_DECL ( WINAPI , BOOL , HttpSendRequestA , ( HINTERNET , LPCSTR , DWORD , LPVOID , DWORD ) ) ;
PF_TYPE_DECL ( WINAPI , BOOL , HttpQueryInfoA , ( HINTERNET , DWORD , LPVOID , LPDWORD , LPDWORD ) ) ;
PF_INIT_OR_OUT ( InternetCrackUrlA , WinInet ) ;
PF_INIT_OR_OUT ( InternetConnectA , WinInet ) ;
PF_INIT_OR_OUT ( InternetCloseHandle , WinInet ) ;
PF_INIT_OR_OUT ( HttpOpenRequestA , WinInet ) ;
PF_INIT_OR_OUT ( HttpSendRequestA , WinInet ) ;
PF_INIT_OR_OUT ( HttpQueryInfoA , WinInet ) ;
if ( url = = NULL )
return FALSE ;
FormatStatus = 0 ;
DownloadStatus = 404 ;
if ( ( ! pf InternetCrackUrlA( url , ( DWORD ) safe_strlen ( url ) , 0 , & UrlParts ) )
if ( ( ! InternetCrackUrlA ( url , ( DWORD ) safe_strlen ( url ) , 0 , & UrlParts ) )
| | ( UrlParts . lpszHostName = = NULL ) | | ( UrlParts . lpszUrlPath = = NULL ) )
goto out ;
hostname [ sizeof ( hostname ) - 1 ] = 0 ;
@ -1123,35 +1067,35 @@ BOOL IsDownloadable(const char* url)
if ( hSession = = NULL )
goto out ;
hConnection = pf InternetConnectA( hSession , UrlParts . lpszHostName , UrlParts . nPort , NULL , NULL , INTERNET_SERVICE_HTTP , 0 , ( DWORD_PTR ) NULL ) ;
hConnection = InternetConnectA ( hSession , UrlParts . lpszHostName , UrlParts . nPort , NULL , NULL , INTERNET_SERVICE_HTTP , 0 , ( DWORD_PTR ) NULL ) ;
if ( hConnection = = NULL )
goto out ;
hRequest = pf HttpOpenRequestA( hConnection , " GET " , UrlParts . lpszUrlPath , NULL , NULL , accept_types ,
hRequest = HttpOpenRequestA ( hConnection , " GET " , UrlParts . lpszUrlPath , NULL , NULL , accept_types ,
INTERNET_FLAG_IGNORE_REDIRECT_TO_HTTP | INTERNET_FLAG_IGNORE_REDIRECT_TO_HTTPS |
INTERNET_FLAG_NO_COOKIES | INTERNET_FLAG_NO_UI | INTERNET_FLAG_NO_CACHE_WRITE | INTERNET_FLAG_HYPERLINK |
( ( UrlParts . nScheme = = INTERNET_SCHEME_HTTPS ) ? INTERNET_FLAG_SECURE : 0 ) , ( DWORD_PTR ) NULL ) ;
if ( hRequest = = NULL )
goto out ;
if ( ! pf HttpSendRequestA( hRequest , request_headers , - 1L , NULL , 0 ) )
if ( ! HttpSendRequestA ( hRequest , request_headers , - 1L , NULL , 0 ) )
goto out ;
/ / Get the file size
dwSize = sizeof ( DownloadStatus ) ;
pf HttpQueryInfoA( hRequest , HTTP_QUERY_STATUS_CODE | HTTP_QUERY_FLAG_NUMBER , ( LPVOID ) & DownloadStatus , & dwSize , NULL ) ;
HttpQueryInfoA ( hRequest , HTTP_QUERY_STATUS_CODE | HTTP_QUERY_FLAG_NUMBER , ( LPVOID ) & DownloadStatus , & dwSize , NULL ) ;
if ( DownloadStatus ! = 200 )
goto out ;
dwSize = sizeof ( dwTotalSize ) ;
pf HttpQueryInfoA( hRequest , HTTP_QUERY_CONTENT_LENGTH | HTTP_QUERY_FLAG_NUMBER , ( LPVOID ) & dwTotalSize , & dwSize , NULL ) ;
HttpQueryInfoA ( hRequest , HTTP_QUERY_CONTENT_LENGTH | HTTP_QUERY_FLAG_NUMBER , ( LPVOID ) & dwTotalSize , & dwSize , NULL ) ;
out :
if ( hRequest )
pf InternetCloseHandle( hRequest ) ;
InternetCloseHandle ( hRequest ) ;
if ( hConnection )
pf InternetCloseHandle( hConnection ) ;
InternetCloseHandle ( hConnection ) ;
if ( hSession )
pf InternetCloseHandle( hSession ) ;
InternetCloseHandle ( hSession ) ;
return ( dwTotalSize > 0 ) ;
}
Pete Batard
3 years ago