init repo

ActiveX/ASCOfficeDocxFile2/BinReader/FileDownloader.h

@@ -0,0 +1,395 @@
+/*
+ * (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;         
+
+};