396 lines
8.2 KiB
C++
396 lines
8.2 KiB
C++
/*
|
|
* (c) Copyright Ascensio System SIA 2010-2014
|
|
*
|
|
* This program is a free software product. You can redistribute it and/or
|
|
* modify it under the terms of the GNU Affero General Public License (AGPL)
|
|
* version 3 as published by the Free Software Foundation. In accordance with
|
|
* Section 7(a) of the GNU AGPL its Section 15 shall be amended to the effect
|
|
* that Ascensio System SIA expressly excludes the warranty of non-infringement
|
|
* of any third-party rights.
|
|
*
|
|
* This program is distributed WITHOUT ANY WARRANTY; without even the implied
|
|
* warranty of MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. For
|
|
* details, see the GNU AGPL at: http://www.gnu.org/licenses/agpl-3.0.html
|
|
*
|
|
* You can contact Ascensio System SIA at Lubanas st. 125a-25, Riga, Latvia,
|
|
* EU, LV-1021.
|
|
*
|
|
* The interactive user interfaces in modified source and object code versions
|
|
* of the Program must display Appropriate Legal Notices, as required under
|
|
* Section 5 of the GNU AGPL version 3.
|
|
*
|
|
* Pursuant to Section 7(b) of the License you must retain the original Product
|
|
* logo when distributing the program. Pursuant to Section 7(e) we decline to
|
|
* grant you any rights under trademark law for use of our trademarks.
|
|
*
|
|
* All the Product's GUI elements, including illustrations and icon sets, as
|
|
* well as technical writing content are licensed under the terms of the
|
|
* Creative Commons Attribution-ShareAlike 4.0 International. See the License
|
|
* terms at http://creativecommons.org/licenses/by-sa/4.0/legalcode
|
|
*
|
|
*/
|
|
#pragma once
|
|
|
|
#include "../../Common/BaseThread.h"
|
|
|
|
#include <wininet.h>
|
|
#pragma comment(lib, "Wininet")
|
|
|
|
|
|
|
|
|
|
#define MAX_SIZE 256
|
|
|
|
#define DOWNLOAD_FILE_SIZE 32768
|
|
#define MAX_SINGLE_DOWNLOAD_FILE_SIZE 524288
|
|
|
|
|
|
|
|
#define CONTENT_RANGE _T("bytes 0-0/")
|
|
|
|
#define CONTENT_RANGE_SIZE ( 11 - 1 )
|
|
|
|
|
|
|
|
|
|
|
|
|
|
class CFileDownloader : public CBaseThread
|
|
{
|
|
public :
|
|
|
|
CFileDownloader (CString sFileUrl, BOOL bDelete = TRUE) : CBaseThread(0)
|
|
{
|
|
m_pFile = NULL;
|
|
m_sFilePath = _T("");
|
|
m_sFileUrl = sFileUrl;
|
|
m_bComplete = FALSE;
|
|
m_bDelete = bDelete;
|
|
}
|
|
~CFileDownloader ()
|
|
{
|
|
if ( m_pFile )
|
|
{
|
|
::fclose( m_pFile );
|
|
m_pFile = NULL;
|
|
}
|
|
if ( m_sFilePath.GetLength() > 0 && m_bDelete )
|
|
{
|
|
DeleteFileW( m_sFilePath.GetBuffer() );
|
|
m_sFilePath = _T("");
|
|
}
|
|
|
|
}
|
|
|
|
|
|
CString GetFilePath()
|
|
{
|
|
return m_sFilePath;
|
|
}
|
|
BOOL IsFileDownloaded()
|
|
{
|
|
return m_bComplete;
|
|
}
|
|
protected :
|
|
|
|
unsigned int DownloadFile(CString sFileUrl)
|
|
{
|
|
|
|
if ( FALSE == InternetGetConnectedState ( 0, 0 ) )
|
|
return S_FALSE;
|
|
|
|
char sTempPath[MAX_PATH], sTempFile[MAX_PATH];
|
|
if ( 0 == GetTempPathA( MAX_PATH, sTempPath ) )
|
|
return S_FALSE;
|
|
|
|
if ( 0 == GetTempFileNameA( sTempPath, "CSS", 0, sTempFile ) )
|
|
return S_FALSE;
|
|
|
|
m_pFile = ::fopen( sTempFile, "wb" );
|
|
if ( !m_pFile )
|
|
return S_FALSE;
|
|
|
|
m_sFilePath = CString( sTempFile );
|
|
|
|
|
|
HINTERNET hInternetSession = InternetOpen ( _T ("Mozilla/4.0 (compatible; MSIE 5.0; Windows 98)"), INTERNET_OPEN_TYPE_PRECONFIG, NULL, NULL, 0 );
|
|
if ( NULL == hInternetSession )
|
|
return S_FALSE;
|
|
|
|
|
|
CString sHTTPHdr = _T ("Range: bytes=0-0");
|
|
|
|
HINTERNET hInternetOpenURL = InternetOpenUrl ( hInternetSession, sFileUrl, sHTTPHdr, -1, INTERNET_FLAG_RESYNCHRONIZE, 0 );
|
|
if ( NULL != hInternetOpenURL )
|
|
{
|
|
|
|
if ( TRUE == QueryStatusCode ( hInternetOpenURL, TRUE ) )
|
|
{
|
|
|
|
LONGLONG nFileSize = IsAccept_Ranges ( hInternetOpenURL );
|
|
|
|
InternetCloseHandle ( hInternetOpenURL );
|
|
if ( -1 == nFileSize )
|
|
{
|
|
|
|
|
|
InternetCloseHandle ( hInternetSession );
|
|
|
|
|
|
return S_FALSE;
|
|
}
|
|
else
|
|
{
|
|
|
|
LONGLONG nStartByte = 0;
|
|
while ( m_bRunThread )
|
|
{
|
|
|
|
if ( nStartByte == nFileSize - 1 )
|
|
{
|
|
|
|
InternetCloseHandle ( hInternetSession );
|
|
return S_OK;
|
|
}
|
|
LONGLONG nEndByte = nStartByte + DOWNLOAD_FILE_SIZE;
|
|
|
|
if ( nEndByte >= nFileSize )
|
|
nEndByte = nFileSize - 1;
|
|
|
|
|
|
BYTE arrBuffer [ DOWNLOAD_FILE_SIZE ] = { 0 };
|
|
DWORD dwBytesDownload = DownloadFilePath ( hInternetSession, arrBuffer, nStartByte, nEndByte, sFileUrl );
|
|
|
|
nStartByte = nEndByte;
|
|
if ( -1 == dwBytesDownload )
|
|
{
|
|
|
|
|
|
InternetCloseHandle ( hInternetSession );
|
|
|
|
|
|
return S_FALSE;
|
|
}
|
|
|
|
|
|
::fwrite( (BYTE*)arrBuffer, 1, dwBytesDownload, m_pFile );
|
|
::fflush( m_pFile );
|
|
|
|
CheckSuspend ();
|
|
}
|
|
}
|
|
}
|
|
else
|
|
{
|
|
|
|
InternetCloseHandle ( hInternetSession );
|
|
|
|
|
|
return S_FALSE;
|
|
}
|
|
}
|
|
else
|
|
{
|
|
|
|
InternetCloseHandle ( hInternetSession );
|
|
|
|
|
|
return S_FALSE;
|
|
}
|
|
|
|
|
|
InternetCloseHandle ( hInternetSession );
|
|
|
|
return S_OK;
|
|
}
|
|
DWORD DownloadFilePath ( HINTERNET hInternet, LPBYTE pBuffer, LONGLONG nStartByte, LONGLONG nEndByte, CString sFileURL )
|
|
{
|
|
|
|
if ( NULL == hInternet )
|
|
return -1;
|
|
|
|
|
|
if ( nStartByte > nEndByte || !pBuffer )
|
|
return -1;
|
|
|
|
|
|
CString sHTTPHdr = _T (""); sHTTPHdr.Format ( _T ("Range: bytes=%I64d-%I64d"), nStartByte, nEndByte );
|
|
|
|
HINTERNET hInternetOpenURL = InternetOpenUrl ( hInternet, sFileURL, sHTTPHdr, -1, INTERNET_FLAG_RESYNCHRONIZE, 0 );
|
|
if ( NULL == hInternetOpenURL )
|
|
return -1;
|
|
|
|
if ( FALSE == QueryStatusCode ( hInternetOpenURL, TRUE ) )
|
|
{
|
|
|
|
InternetCloseHandle ( hInternetOpenURL );
|
|
return -1;
|
|
}
|
|
|
|
|
|
DWORD dwBytesRead = 0;
|
|
|
|
if ( FALSE == InternetReadFile ( hInternetOpenURL, pBuffer, DOWNLOAD_FILE_SIZE, &dwBytesRead ) )
|
|
{
|
|
|
|
InternetCloseHandle ( hInternetOpenURL );
|
|
return -1;
|
|
}
|
|
|
|
|
|
InternetCloseHandle ( hInternetOpenURL );
|
|
|
|
return dwBytesRead;
|
|
}
|
|
virtual DWORD ThreadProc ()
|
|
{
|
|
m_bComplete = FALSE;
|
|
|
|
CoInitialize ( NULL );
|
|
|
|
if ( S_OK != DownloadFile ( m_sFileUrl ) )
|
|
{
|
|
HRESULT hrResultAll = DownloadFileAll(m_sFileUrl, m_sFilePath);
|
|
|
|
if (S_OK != hrResultAll)
|
|
{
|
|
m_bRunThread = FALSE;
|
|
CoUninitialize ();
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
m_bRunThread = FALSE;
|
|
CoUninitialize ();
|
|
|
|
m_bComplete = TRUE;
|
|
return 0;
|
|
}
|
|
|
|
BOOL QueryStatusCode ( HINTERNET hInternet, BOOL bIsRanges )
|
|
{
|
|
|
|
if ( NULL == hInternet )
|
|
return FALSE;
|
|
|
|
|
|
INT nResult = 0;
|
|
|
|
DWORD dwLengthDataSize = 4;
|
|
|
|
|
|
if ( FALSE == HttpQueryInfo ( hInternet, HTTP_QUERY_STATUS_CODE | HTTP_QUERY_FLAG_NUMBER, &nResult, &dwLengthDataSize, NULL ) )
|
|
return FALSE;
|
|
|
|
|
|
if ( HTTP_STATUS_NOT_FOUND == nResult )
|
|
{
|
|
|
|
return FALSE;
|
|
}
|
|
else if ( ( HTTP_STATUS_OK != nResult && FALSE == bIsRanges ) || ( HTTP_STATUS_PARTIAL_CONTENT != nResult && TRUE == bIsRanges ) )
|
|
{
|
|
|
|
return FALSE;
|
|
}
|
|
|
|
|
|
return TRUE;
|
|
}
|
|
|
|
LONGLONG IsAccept_Ranges ( HINTERNET hInternet )
|
|
{
|
|
|
|
if ( NULL == hInternet )
|
|
return -1;
|
|
|
|
|
|
char arrResult [ MAX_SIZE ] = { 0 };
|
|
|
|
DWORD dwLengthDataSize = sizeof ( arrResult );
|
|
|
|
|
|
if ( FALSE == HttpQueryInfoA ( hInternet, HTTP_QUERY_CONTENT_RANGE, &arrResult, &dwLengthDataSize, NULL ) )
|
|
{
|
|
|
|
DWORD dwLastError = GetLastError ();
|
|
if ( dwLastError == ERROR_HTTP_HEADER_NOT_FOUND )
|
|
{
|
|
|
|
return -1;
|
|
}
|
|
|
|
|
|
return -1;
|
|
}
|
|
|
|
|
|
if ( 0 >= dwLengthDataSize )
|
|
return -1;
|
|
|
|
|
|
CString strResult ( arrResult );
|
|
|
|
|
|
LONGLONG nFileSize = 0;
|
|
|
|
try
|
|
{
|
|
|
|
INT nStartIndex = strResult.Find ( CONTENT_RANGE );
|
|
if ( -1 == nStartIndex )
|
|
return -1;
|
|
|
|
|
|
strResult = strResult.Mid ( nStartIndex + CONTENT_RANGE_SIZE );
|
|
|
|
nFileSize = _wtoi64 ( strResult.GetBuffer () );
|
|
|
|
if ( 0 < nFileSize )
|
|
nFileSize += 1;
|
|
}
|
|
catch ( ... )
|
|
{
|
|
|
|
return -1;
|
|
}
|
|
|
|
|
|
return nFileSize;
|
|
}
|
|
|
|
HRESULT DownloadFileAll(CString sFileURL, CString strFileOutput)
|
|
{
|
|
if ( m_pFile )
|
|
{
|
|
::fclose( m_pFile );
|
|
m_pFile = NULL;
|
|
}
|
|
|
|
return URLDownloadToFile (NULL, sFileURL, strFileOutput, NULL, NULL);
|
|
}
|
|
|
|
public:
|
|
static bool IsNeedDownload(CString FilePath)
|
|
{
|
|
int n1 = FilePath.Find(_T("www."));
|
|
int n2 = FilePath.Find(_T("http://"));
|
|
int n3 = FilePath.Find(_T("ftp://"));
|
|
int n4 = FilePath.Find(_T("https://"));
|
|
|
|
if (((n1 >= 0) && (n1 < 10)) || ((n2 >= 0) && (n2 < 10)) || ((n3 >= 0) && (n3 < 10)) || ((n4 >= 0) && (n4 < 10)))
|
|
return true;
|
|
return false;
|
|
}
|
|
|
|
protected :
|
|
|
|
FILE *m_pFile;
|
|
CString m_sFilePath;
|
|
CString m_sFileUrl;
|
|
|
|
BOOL m_bComplete;
|
|
BOOL m_bDelete;
|
|
|
|
};
|