2012-03-11 01:55:25 +00:00
/*
* Rufus : The Reliable USB Formatting Utility
2012-11-08 01:20:48 +00:00
* Networking functionality ( web file download , check for update , etc . )
2016-02-03 17:41:27 +00:00
* Copyright © 2012 - 2016 Pete Batard < pete @ akeo . ie >
2012-03-11 01:55:25 +00:00
*
* 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
* the Free Software Foundation , either version 3 of the License , or
* ( at your option ) any later version .
*
* This program 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 General Public License for more details .
*
* You should have received a copy of the GNU General Public License
* along with this program . If not , see < http : //www.gnu.org/licenses/>.
*/
/* Memory leaks detection - define _CRTDBG_MAP_ALLOC as preprocessor macro */
# ifdef _CRTDBG_MAP_ALLOC
# include <stdlib.h>
# include <crtdbg.h>
# endif
# include <windows.h>
2012-11-08 01:20:48 +00:00
# include <wininet.h>
2012-03-11 01:55:25 +00:00
# include <stdio.h>
2012-11-08 01:20:48 +00:00
# include <malloc.h>
2012-03-11 01:55:25 +00:00
# include <string.h>
2012-11-08 01:20:48 +00:00
# include <inttypes.h>
2012-03-11 01:55:25 +00:00
# include "rufus.h"
2016-02-24 16:10:54 +00:00
# include "missing.h"
2012-03-11 01:55:25 +00:00
# include "resource.h"
2016-02-20 22:52:32 +00:00
# include "msapi_utf8.h"
2013-10-15 21:58:27 +00:00
# include "localization.h"
2012-11-08 01:20:48 +00:00
2016-02-20 22:52:32 +00:00
# include "settings.h"
2012-11-08 01:20:48 +00:00
/* Maximum download chunk size, in bytes */
2016-02-24 16:10:54 +00:00
# define DOWNLOAD_BUFFER_SIZE 10*KB
2012-11-08 01:20:48 +00:00
/* Default delay between update checks (1 day) */
# define DEFAULT_UPDATE_INTERVAL (24*3600)
2014-08-05 23:57:32 +00:00
DWORD DownloadStatus ;
BOOL PromptOnError = TRUE ;
2013-10-22 21:46:59 +00:00
extern BOOL force_update ;
2012-12-06 01:40:44 +00:00
static DWORD error_code ;
static BOOL update_check_in_progress = FALSE ;
2012-12-09 20:36:29 +00:00
static BOOL force_update_check = FALSE ;
2012-11-08 01:20:48 +00:00
2012-03-11 01:55:25 +00:00
/*
2012-11-08 01:20:48 +00:00
* FormatMessage does not handle internet errors
2016-02-03 17:41:27 +00:00
* https : //msdn.microsoft.com/en-us/library/windows/desktop/aa385465.aspx
2012-03-11 01:55:25 +00:00
*/
2012-11-08 01:20:48 +00:00
const char * WinInetErrorString ( void )
2012-03-11 01:55:25 +00:00
{
2012-11-08 01:20:48 +00:00
static char error_string [ 256 ] ;
DWORD size = sizeof ( error_string ) ;
2012-03-11 01:55:25 +00:00
2013-11-30 17:39:38 +00:00
error_code = HRESULT_CODE ( GetLastError ( ) ) ;
2012-03-11 01:55:25 +00:00
2012-11-08 01:20:48 +00:00
if ( ( error_code < INTERNET_ERROR_BASE ) | | ( error_code > INTERNET_ERROR_LAST ) )
2012-03-11 01:55:25 +00:00
return WindowsErrorString ( ) ;
2013-10-15 21:58:27 +00:00
// TODO: These should be localized on an ad-hoc basis
2012-03-11 01:55:25 +00:00
switch ( error_code ) {
2012-11-08 01:20:48 +00:00
case ERROR_INTERNET_OUT_OF_HANDLES :
2012-03-11 01:55:25 +00:00
return " No more handles could be generated at this time. " ;
2012-11-08 01:20:48 +00:00
case ERROR_INTERNET_TIMEOUT :
2012-03-11 01:55:25 +00:00
return " The request has timed out. " ;
2012-11-08 01:20:48 +00:00
case ERROR_INTERNET_INTERNAL_ERROR :
2012-03-11 01:55:25 +00:00
return " An internal error has occurred. " ;
2012-11-08 01:20:48 +00:00
case ERROR_INTERNET_INVALID_URL :
2012-03-11 01:55:25 +00:00
return " The URL is invalid. " ;
2012-11-08 01:20:48 +00:00
case ERROR_INTERNET_UNRECOGNIZED_SCHEME :
2012-03-11 01:55:25 +00:00
return " The URL scheme could not be recognized or is not supported. " ;
2012-11-08 01:20:48 +00:00
case ERROR_INTERNET_NAME_NOT_RESOLVED :
2012-03-11 01:55:25 +00:00
return " The server name could not be resolved. " ;
2012-11-08 01:20:48 +00:00
case ERROR_INTERNET_PROTOCOL_NOT_FOUND :
return " The requested protocol could not be located. " ;
case ERROR_INTERNET_INVALID_OPTION :
return " A request specified an invalid option value. " ;
case ERROR_INTERNET_BAD_OPTION_LENGTH :
return " The length of an option supplied is incorrect for the type of option specified. " ;
case ERROR_INTERNET_OPTION_NOT_SETTABLE :
2012-03-11 01:55:25 +00:00
return " The request option cannot be set, only queried. " ;
2012-11-08 01:20:48 +00:00
case ERROR_INTERNET_SHUTDOWN :
return " The Win32 Internet function support is being shut down or unloaded. " ;
case ERROR_INTERNET_INCORRECT_USER_NAME :
return " The request to connect and log on to an FTP server could not be completed because the supplied user name is incorrect. " ;
case ERROR_INTERNET_INCORRECT_PASSWORD :
return " The request to connect and log on to an FTP server could not be completed because the supplied password is incorrect. " ;
case ERROR_INTERNET_LOGIN_FAILURE :
return " The request to connect to and log on to an FTP server failed. " ;
case ERROR_INTERNET_INVALID_OPERATION :
return " The requested operation is invalid. " ;
case ERROR_INTERNET_OPERATION_CANCELLED :
2014-05-20 18:28:46 +00:00
return " The operation was cancelled, usually because the handle on which the request was operating was closed before the operation completed. " ;
2012-11-08 01:20:48 +00:00
case ERROR_INTERNET_INCORRECT_HANDLE_TYPE :
2012-03-11 01:55:25 +00:00
return " The type of handle supplied is incorrect for this operation. " ;
2012-11-08 01:20:48 +00:00
case ERROR_INTERNET_INCORRECT_HANDLE_STATE :
2012-03-11 01:55:25 +00:00
return " The requested operation cannot be carried out because the handle supplied is not in the correct state. " ;
2012-11-08 01:20:48 +00:00
case ERROR_INTERNET_NOT_PROXY_REQUEST :
return " The request cannot be made via a proxy. " ;
case ERROR_INTERNET_REGISTRY_VALUE_NOT_FOUND :
return " A required registry value could not be located. " ;
case ERROR_INTERNET_BAD_REGISTRY_PARAMETER :
return " A required registry value was located but is an incorrect type or has an invalid value. " ;
case ERROR_INTERNET_NO_DIRECT_ACCESS :
return " Direct network access cannot be made at this time. " ;
case ERROR_INTERNET_NO_CONTEXT :
return " An asynchronous request could not be made because a zero context value was supplied. " ;
case ERROR_INTERNET_NO_CALLBACK :
return " An asynchronous request could not be made because a callback function has not been set. " ;
case ERROR_INTERNET_REQUEST_PENDING :
return " The required operation could not be completed because one or more requests are pending. " ;
case ERROR_INTERNET_INCORRECT_FORMAT :
return " The format of the request is invalid. " ;
case ERROR_INTERNET_ITEM_NOT_FOUND :
return " The requested item could not be located. " ;
case ERROR_INTERNET_CANNOT_CONNECT :
2012-03-11 01:55:25 +00:00
return " The attempt to connect to the server failed. " ;
2012-11-08 01:20:48 +00:00
case ERROR_INTERNET_CONNECTION_ABORTED :
2012-03-11 01:55:25 +00:00
return " The connection with the server has been terminated. " ;
2012-11-08 01:20:48 +00:00
case ERROR_INTERNET_CONNECTION_RESET :
return " The connection with the server has been reset. " ;
case ERROR_INTERNET_FORCE_RETRY :
return " Calls for the Win32 Internet function to redo the request. " ;
case ERROR_INTERNET_INVALID_PROXY_REQUEST :
return " The request to the proxy was invalid. " ;
case ERROR_INTERNET_HANDLE_EXISTS :
return " The request failed because the handle already exists. " ;
case ERROR_INTERNET_SEC_CERT_DATE_INVALID :
return " SSL certificate date that was received from the server is bad. The certificate is expired. " ;
case ERROR_INTERNET_SEC_CERT_CN_INVALID :
2012-03-11 01:55:25 +00:00
return " SSL certificate common name (host name field) is incorrect. " ;
2012-11-08 01:20:48 +00:00
case ERROR_INTERNET_HTTP_TO_HTTPS_ON_REDIR :
return " The application is moving from a non-SSL to an SSL connection because of a redirect. " ;
case ERROR_INTERNET_HTTPS_TO_HTTP_ON_REDIR :
return " The application is moving from an SSL to an non-SSL connection because of a redirect. " ;
case ERROR_INTERNET_MIXED_SECURITY :
return " Some of the content being viewed may have come from unsecured servers. " ;
case ERROR_INTERNET_CHG_POST_IS_NON_SECURE :
return " The application is posting and attempting to change multiple lines of text on a server that is not secure. " ;
case ERROR_INTERNET_POST_IS_NON_SECURE :
return " The application is posting data to a server that is not secure. " ;
case ERROR_FTP_TRANSFER_IN_PROGRESS :
return " The requested operation cannot be made on the FTP session handle because an operation is already in progress. " ;
case ERROR_FTP_DROPPED :
return " The FTP operation was not completed because the session was aborted. " ;
case ERROR_GOPHER_PROTOCOL_ERROR :
case ERROR_GOPHER_NOT_FILE :
case ERROR_GOPHER_DATA_ERROR :
case ERROR_GOPHER_END_OF_DATA :
case ERROR_GOPHER_INVALID_LOCATOR :
case ERROR_GOPHER_INCORRECT_LOCATOR_TYPE :
case ERROR_GOPHER_NOT_GOPHER_PLUS :
case ERROR_GOPHER_ATTRIBUTE_NOT_FOUND :
case ERROR_GOPHER_UNKNOWN_LOCATOR :
2013-02-01 22:59:46 +00:00
return " Gopher? Really??? What is this, 1994? " ;
2012-11-08 01:20:48 +00:00
case ERROR_HTTP_HEADER_NOT_FOUND :
return " The requested header could not be located. " ;
case ERROR_HTTP_DOWNLEVEL_SERVER :
return " The server did not return any headers. " ;
case ERROR_HTTP_INVALID_SERVER_RESPONSE :
return " The server response could not be parsed. " ;
case ERROR_HTTP_INVALID_HEADER :
return " The supplied header is invalid. " ;
case ERROR_HTTP_INVALID_QUERY_REQUEST :
return " The request made to HttpQueryInfo is invalid. " ;
case ERROR_HTTP_HEADER_ALREADY_EXISTS :
return " The header could not be added because it already exists. " ;
case ERROR_HTTP_REDIRECT_FAILED :
return " The redirection failed because either the scheme changed or all attempts made to redirect failed. " ;
2017-07-20 16:42:53 +00:00
case ERROR_INTERNET_SECURITY_CHANNEL_ERROR :
return " This system's SSL library is too old to be able to access this website. " ;
2012-11-08 01:20:48 +00:00
case ERROR_INTERNET_CLIENT_AUTH_CERT_NEEDED :
2012-03-11 01:55:25 +00:00
return " Client Authentication certificate needed " ;
2012-11-08 01:20:48 +00:00
case ERROR_INTERNET_BAD_AUTO_PROXY_SCRIPT :
2012-03-11 01:55:25 +00:00
return " Bad auto proxy script. " ;
2012-11-08 01:20:48 +00:00
case ERROR_INTERNET_UNABLE_TO_DOWNLOAD_SCRIPT :
2012-03-11 01:55:25 +00:00
return " Unable to download script. " ;
2012-11-08 01:20:48 +00:00
case ERROR_INTERNET_NOT_INITIALIZED :
return " Internet has not be initialized. " ;
case ERROR_INTERNET_UNABLE_TO_CACHE_FILE :
return " Unable to cache the file. " ;
case ERROR_INTERNET_TCPIP_NOT_INSTALLED :
return " TPC/IP not installed. " ;
case ERROR_INTERNET_DISCONNECTED :
return " Internet is disconnected. " ;
case ERROR_INTERNET_SERVER_UNREACHABLE :
return " Server could not be reached. " ;
case ERROR_INTERNET_PROXY_SERVER_UNREACHABLE :
return " Proxy server could not be reached. " ;
case ERROR_INTERNET_FAILED_DUETOSECURITYCHECK :
return " A security check prevented internet connection. " ;
case ERROR_INTERNET_NEED_MSN_SSPI_PKG :
return " This connection requires an MSN Security Support Provider Interface package. " ;
case ERROR_INTERNET_LOGIN_FAILURE_DISPLAY_ENTITY_BODY :
return " Please ask Microsoft about that one! " ;
case ERROR_INTERNET_EXTENDED_ERROR :
InternetGetLastResponseInfoA ( & error_code , error_string , & size ) ;
return error_string ;
2012-03-11 01:55:25 +00:00
default :
2017-08-10 18:43:04 +00:00
static_sprintf ( error_string , " Unknown internet error 0x%08lX " , error_code ) ;
2012-11-08 01:20:48 +00:00
return error_string ;
2012-03-11 01:55:25 +00:00
}
}
2015-10-18 20:31:47 +00:00
/*
2012-03-11 01:55:25 +00:00
* Download a file from an URL
2012-11-08 01:20:48 +00:00
* Mostly taken from http : //support.microsoft.com/kb/234913
2012-12-06 01:40:44 +00:00
* If hProgressDialog is not NULL , this function will send INIT and EXIT messages
* to the dialog in question , with WPARAM being set to nonzero for EXIT on success
* and also attempt to indicate progress using an IDC_PROGRESS control
2012-03-11 01:55:25 +00:00
*/
2014-01-21 17:08:41 +00:00
DWORD DownloadFile ( const char * url , const char * file , HWND hProgressDialog )
2012-03-11 01:55:25 +00:00
{
2012-12-06 01:40:44 +00:00
HWND hProgressBar = NULL ;
2012-11-08 01:20:48 +00:00
BOOL r = FALSE ;
2014-08-05 23:57:32 +00:00
DWORD dwFlags , dwSize , dwDownloaded , dwTotalSize ;
2015-10-18 20:31:47 +00:00
FILE * fd = NULL ;
2012-03-11 01:55:25 +00:00
LONG progress_style ;
2014-03-14 23:08:49 +00:00
const char * accept_types [ ] = { " */* \0 " , NULL } ;
2012-11-08 01:20:48 +00:00
unsigned char buf [ DOWNLOAD_BUFFER_SIZE ] ;
char agent [ 64 ] , hostname [ 64 ] , urlpath [ 128 ] ;
HINTERNET hSession = NULL , hConnection = NULL , hRequest = NULL ;
URL_COMPONENTSA UrlParts = { sizeof ( URL_COMPONENTSA ) , NULL , 1 , ( INTERNET_SCHEME ) 0 ,
hostname , sizeof ( hostname ) , 0 , NULL , 1 , urlpath , sizeof ( urlpath ) , NULL , 1 } ;
2014-01-21 17:08:41 +00:00
size_t last_slash ;
2012-11-08 01:20:48 +00:00
int i ;
2012-03-11 01:55:25 +00:00
2014-08-05 23:57:32 +00:00
DownloadStatus = 0 ;
2012-12-06 01:40:44 +00:00
if ( hProgressDialog ! = NULL ) {
// Use the progress control provided, if any
hProgressBar = GetDlgItem ( hProgressDialog , IDC_PROGRESS ) ;
if ( hProgressBar ! = NULL ) {
progress_style = GetWindowLong ( hProgressBar , GWL_STYLE ) ;
SetWindowLong ( hProgressBar , GWL_STYLE , progress_style & ( ~ PBS_MARQUEE ) ) ;
SendMessage ( hProgressBar , PBM_SETPOS , 0 , 0 ) ;
}
2014-05-19 22:25:00 +00:00
SendMessage ( hProgressDialog , UM_PROGRESS_INIT , 0 , 0 ) ;
2012-12-06 01:40:44 +00:00
}
2012-03-11 01:55:25 +00:00
2015-08-10 22:19:57 +00:00
if ( file = = NULL )
goto out ;
2014-01-21 17:08:41 +00:00
for ( last_slash = safe_strlen ( file ) ; last_slash ! = 0 ; last_slash - - ) {
if ( ( file [ last_slash ] = = ' / ' ) | | ( file [ last_slash ] = = ' \\ ' ) ) {
last_slash + + ;
break ;
}
}
2017-07-24 16:20:17 +00:00
PrintInfo ( 0 , MSG_085 , & file [ last_slash ] ) ;
2014-01-21 17:08:41 +00:00
uprintf ( " Downloading '%s' from %s \n " , & file [ last_slash ] , url ) ;
2012-03-11 01:55:25 +00:00
2014-05-09 21:05:25 +00:00
if ( ( ! InternetCrackUrlA ( url , ( DWORD ) safe_strlen ( url ) , 0 , & UrlParts ) )
| | ( UrlParts . lpszHostName = = NULL ) | | ( UrlParts . lpszUrlPath = = NULL ) ) {
2014-01-05 19:54:19 +00:00
uprintf ( " Unable to decode URL: %s \n " , WinInetErrorString ( ) ) ;
2012-03-11 01:55:25 +00:00
goto out ;
}
2013-01-09 21:54:28 +00:00
hostname [ sizeof ( hostname ) - 1 ] = 0 ;
2012-03-11 01:55:25 +00:00
2012-11-08 01:20:48 +00:00
// Open an Internet session
for ( i = 5 ; ( i > 0 ) & & ( ! InternetGetConnectedState ( & dwFlags , 0 ) ) ; i - - ) {
Sleep ( 1000 ) ;
}
if ( i < = 0 ) {
// http://msdn.microsoft.com/en-us/library/windows/desktop/aa384702.aspx is wrong...
SetLastError ( ERROR_INTERNET_NOT_INITIALIZED ) ;
uprintf ( " Network is unavailable: %s \n " , WinInetErrorString ( ) ) ;
2012-03-11 01:55:25 +00:00
goto out ;
}
2017-08-10 18:43:04 +00:00
static_sprintf ( agent , APPLICATION_NAME " /%d.%d.%d (Windows NT %d.%d%s) " ,
2015-02-08 22:36:57 +00:00
rufus_version [ 0 ] , rufus_version [ 1 ] , rufus_version [ 2 ] ,
2014-11-08 00:23:50 +00:00
nWindowsVersion > > 4 , nWindowsVersion & 0x0F , is_x64 ( ) ? " ; WOW64 " : " " ) ;
2012-11-08 01:20:48 +00:00
hSession = InternetOpenA ( agent , INTERNET_OPEN_TYPE_PRECONFIG , NULL , NULL , 0 ) ;
if ( hSession = = NULL ) {
2014-01-21 17:08:41 +00:00
uprintf ( " Could not open Internet session: %s \n " , WinInetErrorString ( ) ) ;
2012-03-11 01:55:25 +00:00
goto out ;
}
2012-11-08 01:20:48 +00:00
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 \n " , UrlParts . lpszHostName , UrlParts . nPort , WinInetErrorString ( ) ) ;
2012-03-11 01:55:25 +00:00
goto out ;
}
2014-03-14 23:08:49 +00:00
hRequest = HttpOpenRequestA ( hConnection , " GET " , UrlParts . lpszUrlPath , NULL , NULL , accept_types ,
2016-05-31 15:35:48 +00:00
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 ) ;
2012-11-08 01:20:48 +00:00
if ( hRequest = = NULL ) {
2014-01-21 17:08:41 +00:00
uprintf ( " Could not open URL %s: %s \n " , url , WinInetErrorString ( ) ) ;
2012-03-11 01:55:25 +00:00
goto out ;
}
2013-07-08 20:28:44 +00:00
if ( ! HttpSendRequestA ( hRequest , NULL , 0 , NULL , 0 ) ) {
2012-11-08 01:20:48 +00:00
uprintf ( " Unable to send request: %s \n " , WinInetErrorString ( ) ) ;
2012-03-11 01:55:25 +00:00
goto out ;
}
2012-11-08 01:20:48 +00:00
// Get the file size
2014-08-05 23:57:32 +00:00
dwSize = sizeof ( DownloadStatus ) ;
DownloadStatus = 404 ;
HttpQueryInfoA ( hRequest , HTTP_QUERY_STATUS_CODE | HTTP_QUERY_FLAG_NUMBER , ( LPVOID ) & DownloadStatus , & dwSize , NULL ) ;
if ( DownloadStatus ! = 200 ) {
2012-11-08 01:20:48 +00:00
error_code = ERROR_INTERNET_ITEM_NOT_FOUND ;
2015-05-14 23:36:42 +00:00
uprintf ( " Unable to access file: %d \n " , DownloadStatus ) ;
2012-11-08 01:20:48 +00:00
goto out ;
}
dwSize = sizeof ( dwTotalSize ) ;
if ( ! HttpQueryInfoA ( hRequest , HTTP_QUERY_CONTENT_LENGTH | HTTP_QUERY_FLAG_NUMBER , ( LPVOID ) & dwTotalSize , & dwSize , NULL ) ) {
uprintf ( " Unable to retrieve file length: %s \n " , WinInetErrorString ( ) ) ;
2012-03-11 01:55:25 +00:00
goto out ;
}
uprintf ( " File length: %d bytes \n " , dwTotalSize ) ;
2013-07-01 23:45:47 +00:00
fd = fopenU ( file , " wb " ) ;
2012-03-11 01:55:25 +00:00
if ( fd = = NULL ) {
2014-01-21 17:08:41 +00:00
uprintf ( " Unable to create file '%s': %s \n " , & file [ last_slash ] , WinInetErrorString ( ) ) ;
2012-03-11 01:55:25 +00:00
goto out ;
}
// Keep checking for data until there is nothing left.
2012-11-08 01:20:48 +00:00
dwSize = 0 ;
while ( 1 ) {
2012-03-11 01:55:25 +00:00
if ( IS_ERROR ( FormatStatus ) )
goto out ;
2012-11-08 01:20:48 +00:00
if ( ! InternetReadFile ( hRequest , buf , sizeof ( buf ) , & dwDownloaded ) | | ( dwDownloaded = = 0 ) )
2012-03-11 01:55:25 +00:00
break ;
2012-11-08 01:20:48 +00:00
dwSize + = dwDownloaded ;
2012-12-05 01:53:10 +00:00
SendMessage ( hProgressBar , PBM_SETPOS , ( WPARAM ) ( MAX_PROGRESS * ( ( 1.0f * dwSize ) / ( 1.0f * dwTotalSize ) ) ) , 0 ) ;
2015-01-01 23:39:28 +00:00
PrintInfo ( 0 , MSG_241 , ( 100.0f * dwSize ) / ( 1.0f * dwTotalSize ) ) ;
2012-11-08 01:20:48 +00:00
if ( fwrite ( buf , 1 , dwDownloaded , fd ) ! = dwDownloaded ) {
2014-01-21 17:08:41 +00:00
uprintf ( " Error writing file '%s': %s \n " , & file [ last_slash ] , WinInetErrorString ( ) ) ;
2012-03-11 01:55:25 +00:00
goto out ;
}
}
2012-11-08 01:20:48 +00:00
if ( dwSize ! = dwTotalSize ) {
uprintf ( " Could not download complete file - read: %d bytes, expected: %d bytes \n " , dwSize , dwTotalSize ) ;
FormatStatus = ERROR_SEVERITY_ERROR | FAC ( FACILITY_STORAGE ) | ERROR_WRITE_FAULT ;
goto out ;
} else {
r = TRUE ;
2014-01-21 17:08:41 +00:00
uprintf ( " Successfully downloaded '%s' \n " , & file [ last_slash ] ) ;
2012-11-08 01:20:48 +00:00
}
2012-03-11 01:55:25 +00:00
out :
2012-12-05 01:53:10 +00:00
if ( hProgressDialog ! = NULL )
2014-05-19 22:25:00 +00:00
SendMessage ( hProgressDialog , UM_PROGRESS_EXIT , ( WPARAM ) r , 0 ) ;
2012-03-11 01:55:25 +00:00
if ( fd ! = NULL ) fclose ( fd ) ;
if ( ! r ) {
2015-08-27 17:37:04 +00:00
if ( file ! = NULL )
2017-08-30 10:24:47 +00:00
_unlinkU ( file ) ;
2014-08-05 23:57:32 +00:00
if ( PromptOnError ) {
2015-01-01 23:39:28 +00:00
PrintInfo ( 0 , MSG_242 ) ;
2014-08-05 23:57:32 +00:00
SetLastError ( error_code ) ;
2015-10-18 20:31:47 +00:00
MessageBoxExU ( hMainDialog , IS_ERROR ( FormatStatus ) ? StrError ( FormatStatus , FALSE ) : WinInetErrorString ( ) ,
lmprintf ( MSG_044 ) , MB_OK | MB_ICONERROR | MB_IS_RTL , selected_langid ) ;
2014-08-05 23:57:32 +00:00
}
2012-03-11 01:55:25 +00:00
}
2012-11-08 01:20:48 +00:00
if ( hRequest ) InternetCloseHandle ( hRequest ) ;
if ( hConnection ) InternetCloseHandle ( hConnection ) ;
if ( hSession ) InternetCloseHandle ( hSession ) ;
2014-01-21 17:08:41 +00:00
return r ? dwSize : 0 ;
2012-11-08 01:20:48 +00:00
}
2012-12-06 01:40:44 +00:00
/* Threaded download */
static const char * _url , * _file ;
static HWND _hProgressDialog ;
static DWORD WINAPI _DownloadFileThread ( LPVOID param )
2012-11-08 01:20:48 +00:00
{
2014-01-21 17:08:41 +00:00
ExitThread ( DownloadFile ( _url , _file , _hProgressDialog ) ! = 0 ) ;
2012-12-06 01:40:44 +00:00
}
HANDLE DownloadFileThreaded ( const char * url , const char * file , HWND hProgressDialog )
{
_url = url ;
_file = file ;
_hProgressDialog = hProgressDialog ;
return CreateThread ( NULL , 0 , _DownloadFileThread , NULL , 0 , NULL ) ;
}
2012-12-07 00:54:40 +00:00
static __inline uint64_t to_uint64_t ( uint16_t x [ 4 ] ) {
int i ;
uint64_t ret = 0 ;
2015-02-08 22:36:57 +00:00
for ( i = 0 ; i < 3 ; i + + )
2012-12-07 00:54:40 +00:00
ret = ( ret < < 16 ) + x [ i ] ;
return ret ;
}
2012-12-06 01:40:44 +00:00
/*
* Background thread to check for updates
*/
static DWORD WINAPI CheckForUpdatesThread ( LPVOID param )
{
2012-12-09 20:36:29 +00:00
BOOL releases_only , found_new_version = FALSE ;
2012-12-16 00:29:37 +00:00
int status = 0 ;
2017-08-17 12:33:20 +00:00
const char * server_url = RUFUS_NO_SSL_URL " / " ;
2016-02-03 17:41:27 +00:00
int i , j , k , max_channel , verbose = 0 , verpos [ 4 ] ;
2012-12-07 00:54:40 +00:00
static const char * archname [ ] = { " win_x86 " , " win_x64 " } ;
2016-02-03 17:41:27 +00:00
static const char * channel [ ] = { " release " , " beta " , " test " } ; // release channel
2013-07-08 20:28:44 +00:00
const char * accept_types [ ] = { " */* \0 " , NULL } ;
2012-11-08 01:20:48 +00:00
DWORD dwFlags , dwSize , dwDownloaded , dwTotalSize , dwStatus ;
char * buf = NULL ;
char agent [ 64 ] , hostname [ 64 ] , urlpath [ 128 ] , mime [ 32 ] ;
2012-11-22 01:52:33 +00:00
OSVERSIONINFOA os_version = { sizeof ( OSVERSIONINFOA ) , 0 , 0 , 0 , 0 , " " } ;
2012-11-08 01:20:48 +00:00
HINTERNET hSession = NULL , hConnection = NULL , hRequest = NULL ;
URL_COMPONENTSA UrlParts = { sizeof ( URL_COMPONENTSA ) , NULL , 1 , ( INTERNET_SCHEME ) 0 ,
hostname , sizeof ( hostname ) , 0 , NULL , 1 , urlpath , sizeof ( urlpath ) , NULL , 1 } ;
SYSTEMTIME ServerTime , LocalTime ;
FILETIME FileTime ;
2012-12-06 01:40:44 +00:00
int64_t local_time = 0 , reg_time , server_time , update_interval ;
2012-11-08 01:20:48 +00:00
2012-12-06 01:40:44 +00:00
update_check_in_progress = TRUE ;
2015-01-25 00:56:38 +00:00
verbose = ReadSetting32 ( SETTING_VERBOSE_UPDATES ) ;
2014-03-17 20:42:10 +00:00
// Without this the FileDialog will produce error 0x8001010E when compiled for Vista or later
IGNORE_RETVAL ( CoInitializeEx ( NULL , COINIT_APARTMENTTHREADED ) ) ;
2012-12-09 20:36:29 +00:00
// Unless the update was forced, wait a while before performing the update check
if ( ! force_update_check ) {
// It would of course be a lot nicer to use a timer and wake the thread, but my
// development time is limited and this is FASTER to implement.
2012-12-06 01:40:44 +00:00
do {
2012-12-09 20:36:29 +00:00
for ( i = 0 ; ( i < 30 ) & & ( ! force_update_check ) ; i + + )
Sleep ( 500 ) ;
} while ( ( ! force_update_check ) & & ( ( iso_op_in_progress | | format_op_in_progress | | ( dialog_showing > 0 ) ) ) ) ;
if ( ! force_update_check ) {
2015-01-25 00:56:38 +00:00
if ( ( ReadSetting32 ( SETTING_UPDATE_INTERVAL ) = = - 1 ) ) {
vuprintf ( " Check for updates disabled, as per settings. \n " ) ;
2012-12-09 20:36:29 +00:00
goto out ;
}
2015-01-25 00:56:38 +00:00
reg_time = ReadSetting64 ( SETTING_LAST_UPDATE ) ;
update_interval = ( int64_t ) ReadSetting32 ( SETTING_UPDATE_INTERVAL ) ;
2012-12-09 20:36:29 +00:00
if ( update_interval = = 0 ) {
2015-01-25 00:56:38 +00:00
WriteSetting32 ( SETTING_UPDATE_INTERVAL , DEFAULT_UPDATE_INTERVAL ) ;
2012-12-09 20:36:29 +00:00
update_interval = DEFAULT_UPDATE_INTERVAL ;
}
GetSystemTime ( & LocalTime ) ;
if ( ! SystemTimeToFileTime ( & LocalTime , & FileTime ) )
goto out ;
local_time = ( ( ( ( int64_t ) FileTime . dwHighDateTime ) < < 32 ) + FileTime . dwLowDateTime ) / 10000000 ;
vvuprintf ( " Local time: % " PRId64 " \n " , local_time ) ;
if ( local_time < reg_time + update_interval ) {
vuprintf ( " Next update check in % " PRId64 " seconds. \n " , reg_time + update_interval - local_time ) ;
goto out ;
}
2012-12-06 01:40:44 +00:00
}
2012-11-08 01:20:48 +00:00
}
2015-01-01 23:39:28 +00:00
PrintInfoDebug ( 3000 , MSG_243 ) ;
2012-12-16 00:29:37 +00:00
status + + ; // 1
2012-11-08 01:20:48 +00:00
2012-11-22 01:52:33 +00:00
if ( ! GetVersionExA ( & os_version ) ) {
2012-12-16 00:29:37 +00:00
uprintf ( " Could not read Windows version - Check for updates cancelled. \n " ) ;
2012-12-04 01:47:45 +00:00
goto out ;
2012-11-22 01:52:33 +00:00
}
2012-12-04 01:47:45 +00:00
if ( ( ! InternetCrackUrlA ( server_url , ( DWORD ) safe_strlen ( server_url ) , 0 , & UrlParts ) ) | | ( ! InternetGetConnectedState ( & dwFlags , 0 ) ) )
2012-11-08 01:20:48 +00:00
goto out ;
2013-01-09 21:54:28 +00:00
hostname [ sizeof ( hostname ) - 1 ] = 0 ;
2012-11-08 01:20:48 +00:00
2017-08-10 18:43:04 +00:00
static_sprintf ( agent , APPLICATION_NAME " /%d.%d.%d (Windows NT %d.%d%s) " ,
2016-02-08 18:55:52 +00:00
rufus_version [ 0 ] , rufus_version [ 1 ] , rufus_version [ 2 ] ,
nWindowsVersion > > 4 , nWindowsVersion & 0x0F , is_x64 ( ) ? " ; WOW64 " : " " ) ;
2012-11-08 01:20:48 +00:00
hSession = InternetOpenA ( agent , INTERNET_OPEN_TYPE_PRECONFIG , NULL , NULL , 0 ) ;
if ( hSession = = NULL )
goto out ;
hConnection = InternetConnectA ( hSession , UrlParts . lpszHostName , UrlParts . nPort , NULL , NULL , INTERNET_SERVICE_HTTP , 0 , ( DWORD_PTR ) NULL ) ;
if ( hConnection = = NULL )
goto out ;
2012-03-11 01:55:25 +00:00
2012-12-16 00:29:37 +00:00
status + + ; // 2
2015-01-25 00:56:38 +00:00
releases_only = ! ReadSettingBool ( SETTING_INCLUDE_BETAS ) ;
2012-12-07 00:54:40 +00:00
2016-02-03 17:41:27 +00:00
// Test releases get their own distribution channel (and also force beta checks)
# if defined(TEST)
max_channel = ( int ) ARRAYSIZE ( channel ) ;
# else
max_channel = releases_only ? 1 : ( int ) ARRAYSIZE ( channel ) - 1 ;
# endif
for ( k = 0 ; ( k < max_channel ) & & ( ! found_new_version ) ; k + + ) {
2012-12-07 00:54:40 +00:00
uprintf ( " Checking %s channel... \n " , channel [ k ] ) ;
// At this stage we can query the server for various update version files.
// We first try to lookup for "<appname>_<os_arch>_<os_version_major>_<os_version_minor>.ver"
// and then remove each each of the <os_> components until we find our match. For instance, we may first
// look for rufus_win_x64_6.2.ver (Win8 x64) but only get a match for rufus_win_x64_6.ver (Vista x64 or later)
// This allows sunsetting OS versions (eg XP) or providing different downloads for different archs/groups.
2017-08-10 18:43:04 +00:00
static_sprintf ( urlpath , " %s%s%s_%s_%lu.%lu.ver " , APPLICATION_NAME , ( k = = 0 ) ? " " : " _ " ,
2013-11-23 00:46:06 +00:00
( k = = 0 ) ? " " : channel [ k ] , archname [ is_x64 ( ) ? 1 : 0 ] , os_version . dwMajorVersion , os_version . dwMinorVersion ) ;
2012-12-07 00:54:40 +00:00
vuprintf ( " Base update check: %s \n " , urlpath ) ;
for ( i = 0 , j = ( int ) safe_strlen ( urlpath ) - 5 ; ( j > 0 ) & & ( i < ARRAYSIZE ( verpos ) ) ; j - - ) {
if ( ( urlpath [ j ] = = ' . ' ) | | ( urlpath [ j ] = = ' _ ' ) ) {
verpos [ i + + ] = j ;
}
2012-11-22 01:52:33 +00:00
}
2012-12-07 00:54:40 +00:00
if ( i ! = ARRAYSIZE ( verpos ) ) {
uprintf ( " Broken code in CheckForUpdatesThread()! \n " ) ;
2012-11-22 01:52:33 +00:00
goto out ;
2012-12-07 00:54:40 +00:00
}
2012-11-22 01:52:33 +00:00
2012-12-07 00:54:40 +00:00
UrlParts . lpszUrlPath = urlpath ;
UrlParts . dwUrlPathLength = sizeof ( urlpath ) ;
for ( i = 0 ; i < ARRAYSIZE ( verpos ) ; i + + ) {
vvuprintf ( " Trying %s \n " , UrlParts . lpszUrlPath ) ;
2013-07-08 20:28:44 +00:00
hRequest = HttpOpenRequestA ( hConnection , " GET " , UrlParts . lpszUrlPath , NULL , NULL , accept_types ,
2016-05-31 15:35:48 +00:00
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 ) ;
2017-07-20 16:42:53 +00:00
if ( ( hRequest = = NULL ) | | ( ! HttpSendRequestA ( hRequest , NULL , 0 , NULL , 0 ) ) ) {
uprintf ( " Unable to send request: %s " , WinInetErrorString ( ) ) ;
2012-12-07 00:54:40 +00:00
goto out ;
2017-07-20 16:42:53 +00:00
}
2012-12-07 00:54:40 +00:00
// Ensure that we get a text file
dwSize = sizeof ( dwStatus ) ;
dwStatus = 404 ;
HttpQueryInfoA ( hRequest , HTTP_QUERY_STATUS_CODE | HTTP_QUERY_FLAG_NUMBER , ( LPVOID ) & dwStatus , & dwSize , NULL ) ;
2015-10-18 20:31:47 +00:00
if ( dwStatus = = 200 )
2012-12-07 00:54:40 +00:00
break ;
InternetCloseHandle ( hRequest ) ;
hRequest = NULL ;
safe_strcpy ( & urlpath [ verpos [ i ] ] , 5 , " .ver " ) ;
2012-12-06 01:40:44 +00:00
}
2012-12-07 00:54:40 +00:00
if ( dwStatus ! = 200 ) {
vuprintf ( " Could not find a %s version file on server %s " , channel [ k ] , server_url ) ;
if ( ( releases_only ) | | ( k + 1 > = ARRAYSIZE ( channel ) ) )
goto out ;
continue ;
2012-12-06 01:40:44 +00:00
}
2013-06-06 22:40:37 +00:00
vuprintf ( " Found match for %s on server %s " , urlpath , server_url ) ;
2012-11-08 01:20:48 +00:00
2017-06-24 16:23:06 +00:00
// IMPORTANT: You might need to edit your server's MIME conf so that it returns
// 'text/plain' for .ver files. Use 'curl -I' to check that you get something
// like 'Content-Type: text/plain; charset=UTF-8' when fetching your .ver files.
2012-12-07 00:54:40 +00:00
dwSize = sizeof ( mime ) ;
HttpQueryInfoA ( hRequest , HTTP_QUERY_CONTENT_TYPE , ( LPVOID ) & mime , & dwSize , NULL ) ;
2015-04-18 22:14:55 +00:00
if ( strncmp ( mime , " text/plain " , sizeof ( " text/plain " ) - 1 ) ! = 0 )
2012-12-07 00:54:40 +00:00
goto out ;
2012-11-08 01:20:48 +00:00
2012-12-07 00:54:40 +00:00
// We also get a date from Apache, which we'll use to avoid out of sync check,
// in case some set their clock way into the future and back.
// 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 ( ( ! 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 ;
vvuprintf ( " Server time: % " PRId64 " \n " , server_time ) ;
// Always store the server response time - the only clock we trust!
2015-01-25 00:56:38 +00:00
WriteSetting64 ( SETTING_LAST_UPDATE , server_time ) ;
2012-12-07 00:54:40 +00:00
// Might as well let the user know
2012-12-09 20:36:29 +00:00
if ( ! force_update_check ) {
2013-06-06 22:40:37 +00:00
if ( ( local_time > server_time + 600 ) | | ( local_time < server_time - 600 ) ) {
2015-10-18 20:31:47 +00:00
uprintf ( " IMPORTANT: Your local clock is more than 10 minutes in the %s. Unless you fix this, " APPLICATION_NAME " may not be able to check for updates... " ,
2013-06-06 22:40:37 +00:00
( local_time > server_time + 600 ) ? " future " : " past " ) ;
2012-12-07 00:54:40 +00:00
}
}
dwSize = sizeof ( dwTotalSize ) ;
if ( ! HttpQueryInfoA ( hRequest , HTTP_QUERY_CONTENT_LENGTH | HTTP_QUERY_FLAG_NUMBER , ( LPVOID ) & dwTotalSize , & dwSize , NULL ) )
goto out ;
safe_free ( buf ) ;
// Make sure the file is NUL terminated
buf = ( char * ) calloc ( dwTotalSize + 1 , 1 ) ;
if ( buf = = NULL ) goto out ;
// This is a version file - we should be able to gulp it down in one go
if ( ! InternetReadFile ( hRequest , buf , dwTotalSize , & dwDownloaded ) | | ( dwDownloaded ! = dwTotalSize ) )
goto out ;
2012-12-16 00:29:37 +00:00
status + + ;
2012-12-07 00:54:40 +00:00
vuprintf ( " Successfully downloaded version file (%d bytes) \n " , dwTotalSize ) ;
2012-11-08 01:20:48 +00:00
2012-12-07 00:54:40 +00:00
parse_update ( buf , dwTotalSize + 1 ) ;
2012-11-12 01:53:34 +00:00
2012-12-07 00:54:40 +00:00
vuprintf ( " UPDATE DATA: \n " ) ;
2015-02-08 22:36:57 +00:00
vuprintf ( " version: %d.%d.%d (%s) \n " , update . version [ 0 ] , update . version [ 1 ] , update . version [ 2 ] , channel [ k ] ) ;
2012-12-07 00:54:40 +00:00
vuprintf ( " platform_min: %d.%d \n " , update . platform_min [ 0 ] , update . platform_min [ 1 ] ) ;
vuprintf ( " url: %s \n " , update . download_url ) ;
2013-10-22 21:46:59 +00:00
found_new_version = ( ( to_uint64_t ( update . version ) > to_uint64_t ( rufus_version ) ) | | ( force_update ) )
2012-12-07 00:54:40 +00:00
& & ( ( os_version . dwMajorVersion > update . platform_min [ 0 ] )
| | ( ( os_version . dwMajorVersion = = update . platform_min [ 0 ] ) & & ( os_version . dwMinorVersion > = update . platform_min [ 1 ] ) ) ) ;
uprintf ( " N%sew %s version found%c \n " , found_new_version ? " " : " o n " , channel [ k ] , found_new_version ? ' ! ' : ' . ' ) ;
}
2012-11-08 01:20:48 +00:00
out :
safe_free ( buf ) ;
if ( hRequest ) InternetCloseHandle ( hRequest ) ;
if ( hConnection ) InternetCloseHandle ( hConnection ) ;
if ( hSession ) InternetCloseHandle ( hSession ) ;
2012-12-16 00:29:37 +00:00
switch ( status ) {
case 1 :
2015-01-01 23:39:28 +00:00
PrintInfoDebug ( 3000 , MSG_244 ) ;
2012-12-16 00:29:37 +00:00
break ;
case 2 :
2015-01-01 23:39:28 +00:00
PrintInfoDebug ( 3000 , MSG_245 ) ;
2012-12-16 00:29:37 +00:00
break ;
case 3 :
2013-01-18 01:39:24 +00:00
case 4 :
2015-01-01 23:39:28 +00:00
PrintInfo ( 3000 , found_new_version ? MSG_246 : MSG_247 ) ;
2012-12-16 00:29:37 +00:00
default :
break ;
}
2012-12-07 00:54:40 +00:00
// Start the new download after cleanup
if ( found_new_version ) {
// User may have started an operation while we were checking
2012-12-09 20:36:29 +00:00
while ( ( ! force_update_check ) & & ( iso_op_in_progress | | format_op_in_progress | | ( dialog_showing > 0 ) ) ) {
Sleep ( 15000 ) ;
2012-12-07 00:54:40 +00:00
}
DownloadNewVersion ( ) ;
2015-03-09 02:49:11 +00:00
} else if ( force_update_check ) {
PostMessage ( hMainDialog , UM_NO_UPDATE , 0 , 0 ) ;
2012-12-07 00:54:40 +00:00
}
2012-12-09 20:36:29 +00:00
force_update_check = FALSE ;
2012-12-06 01:40:44 +00:00
update_check_in_progress = FALSE ;
2012-11-22 01:52:33 +00:00
ExitThread ( 0 ) ;
}
2012-12-06 01:40:44 +00:00
/*
* Initiate a check for updates . If force is true , ignore the wait period
*/
BOOL CheckForUpdates ( BOOL force )
2012-11-22 01:52:33 +00:00
{
2012-12-09 20:36:29 +00:00
force_update_check = force ;
2012-12-06 01:40:44 +00:00
if ( update_check_in_progress )
return FALSE ;
2012-12-09 20:36:29 +00:00
if ( CreateThread ( NULL , 0 , CheckForUpdatesThread , NULL , 0 , NULL ) = = NULL ) {
2013-06-06 22:40:37 +00:00
uprintf ( " Unable to start update check thread " ) ;
2012-12-04 01:47:45 +00:00
return FALSE ;
2012-11-22 01:52:33 +00:00
}
return TRUE ;
2012-03-11 01:55:25 +00:00
}