Skip to content

O
onlyoffice_core
Commit d475ef7e authored by Oleg.Korshul's avatar Oleg.Korshul Committed by Alexander Trofimov
Browse files
Options

utf8 

git-svn-id: svn://fileserver/activex/AVS/Sources/TeamlabOffice/trunk/ServerComponents@61241 954022d7-b5bf-4e40-9824-e11837661b57
parent 812fd20e
Hide whitespace changes
Inline Side-by-side
Showing with 80 additions and 80 deletions
Common/FileDownloader.h
View file @ d475ef7e
#pragma once #pragma once
#ifdef _WIN32 #ifdef _WIN32
#include "../../Common/BaseThread.h" #include "../../Common/BaseThread.h"
...@@ -8,21 +8,21 @@ ...@@ -8,21 +8,21 @@
//------------------------------------------------------------------------------------------------------ //------------------------------------------------------------------------------------------------------
// // Константа для максимального числа символов в строке
#define MAX_SIZE 256 #define MAX_SIZE 256
// // Константа для максимального числа загружаемых байт
#define DOWNLOAD_FILE_SIZE 32768 #define DOWNLOAD_FILE_SIZE 32768
#define MAX_SINGLE_DOWNLOAD_FILE_SIZE 524288 #define MAX_SINGLE_DOWNLOAD_FILE_SIZE 524288
// // Константа для получения размера файла
#define CONTENT_RANGE _T("bytes 0-0/") #define CONTENT_RANGE _T("bytes 0-0/")
// CONTENT_RANGE // Константа для колличества символов у CONTENT_RANGE
#define CONTENT_RANGE_SIZE ( 11/*sizeof ( CONTENT_RANGE )*/ - 1 ) #define CONTENT_RANGE_SIZE ( 11/*sizeof ( CONTENT_RANGE )*/ - 1 )
//------------------------------------------------------------------------------------------------------ //------------------------------------------------------------------------------------------------------
// // Класс для закачки файла из сети на локальный диск
//------------------------------------------------------------------------------------------------------ //------------------------------------------------------------------------------------------------------
class CFileDownloader : public CBaseThread class CFileDownloader : public CBaseThread
...@@ -65,7 +65,7 @@ protected : ...@@ -65,7 +65,7 @@ protected :
unsigned int DownloadFile(CString sFileUrl) unsigned int DownloadFile(CString sFileUrl)
{ {
// // Проверяем состояние соединения
if ( FALSE == InternetGetConnectedState ( 0, 0 ) ) if ( FALSE == InternetGetConnectedState ( 0, 0 ) )
return S_FALSE; return S_FALSE;
...@@ -82,132 +82,132 @@ protected : ...@@ -82,132 +82,132 @@ protected :
m_sFilePath = CString( sTempFile ); m_sFilePath = CString( sTempFile );
// // Открываем сессию
HINTERNET hInternetSession = InternetOpen ( _T ("Mozilla/4.0 (compatible; MSIE 5.0; Windows 98)"), INTERNET_OPEN_TYPE_PRECONFIG, NULL, NULL, 0 ); HINTERNET hInternetSession = InternetOpen ( _T ("Mozilla/4.0 (compatible; MSIE 5.0; Windows 98)"), INTERNET_OPEN_TYPE_PRECONFIG, NULL, NULL, 0 );
if ( NULL == hInternetSession ) if ( NULL == hInternetSession )
return S_FALSE; return S_FALSE;
// ( 0 ( ) ) // Заголовок запроса ( пока содержит 0 байт ( необходимо для проверки ) )
CString sHTTPHdr = _T ("Range: bytes=0-0"); CString sHTTPHdr = _T ("Range: bytes=0-0");
// , // Открываем ссылку для проверки на ее существование, а также на возможность чтения частями
HINTERNET hInternetOpenURL = InternetOpenUrl ( hInternetSession, sFileUrl, sHTTPHdr, -1, INTERNET_FLAG_RESYNCHRONIZE, 0 ); HINTERNET hInternetOpenURL = InternetOpenUrl ( hInternetSession, sFileUrl, sHTTPHdr, -1, INTERNET_FLAG_RESYNCHRONIZE, 0 );
if ( NULL != hInternetOpenURL ) if ( NULL != hInternetOpenURL )
{ {
// , // Открытие произошло, проверяем ответ
if ( TRUE == QueryStatusCode ( hInternetOpenURL, TRUE ) ) if ( TRUE == QueryStatusCode ( hInternetOpenURL, TRUE ) )
{ {
// , // Запрос прошел удачно, проверяем возможность чтения частями и получаем размер данных
LONGLONG nFileSize = IsAccept_Ranges ( hInternetOpenURL ); LONGLONG nFileSize = IsAccept_Ranges ( hInternetOpenURL );
// // Закрываем хендл
InternetCloseHandle ( hInternetOpenURL ); InternetCloseHandle ( hInternetOpenURL );
if ( -1 == nFileSize ) if ( -1 == nFileSize )
{ {
// // Чтение частями недоступно
// // Закрываем хендл соединения
InternetCloseHandle ( hInternetSession ); InternetCloseHandle ( hInternetSession );
// ( DownloadAll) // Закрываем файл (сделается на DownloadAll)
// // Попробуем записать его целиком
return S_FALSE; return S_FALSE;
} }
else else
{ {
// // Чтение частями доступно
LONGLONG nStartByte = 0; LONGLONG nStartByte = 0;
while ( m_bRunThread ) while ( m_bRunThread )
{ {
// - // Если закачали весь файл - то выходим
if ( nStartByte == nFileSize - 1 ) if ( nStartByte == nFileSize - 1 )
{ {
// // Закрываем хендл соединения
InternetCloseHandle ( hInternetSession ); InternetCloseHandle ( hInternetSession );
return S_OK; return S_OK;
} }
LONGLONG nEndByte = nStartByte + DOWNLOAD_FILE_SIZE; LONGLONG nEndByte = nStartByte + DOWNLOAD_FILE_SIZE;
// , ( 1 , , .. 0 ) // Если файл заканчивается, то загружаем меньшее колличество байт ( на 1 меньше, чем размер, т.к. начинается с 0 )
if ( nEndByte >= nFileSize ) if ( nEndByte >= nFileSize )
nEndByte = nFileSize - 1; nEndByte = nFileSize - 1;
// // Буффер для закачки
BYTE arrBuffer [ DOWNLOAD_FILE_SIZE ] = { 0 }; BYTE arrBuffer [ DOWNLOAD_FILE_SIZE ] = { 0 };
DWORD dwBytesDownload = DownloadFilePath ( hInternetSession, arrBuffer, nStartByte, nEndByte, sFileUrl ); DWORD dwBytesDownload = DownloadFilePath ( hInternetSession, arrBuffer, nStartByte, nEndByte, sFileUrl );
nStartByte = nEndByte; nStartByte = nEndByte;
if ( -1 == dwBytesDownload ) if ( -1 == dwBytesDownload )
{ {
// - !!!! // Ничего не прочиталось - это плохо!!!!
// // Закрываем хендл соединения
InternetCloseHandle ( hInternetSession ); InternetCloseHandle ( hInternetSession );
// ( DownloadAll) // Закрываем файл (сделается на DownloadAll)
// // Попробуем записать его целиком
return S_FALSE; return S_FALSE;
} }
// // Пишем в файл
::fwrite( (BYTE*)arrBuffer, 1, dwBytesDownload, m_pFile ); ::fwrite( (BYTE*)arrBuffer, 1, dwBytesDownload, m_pFile );
::fflush( m_pFile ); ::fflush( m_pFile );
// // Проверка на приостановку
CheckSuspend (); CheckSuspend ();
} }
} }
} }
else else
{ {
// // Закрываем хендл соединения
InternetCloseHandle ( hInternetSession ); InternetCloseHandle ( hInternetSession );
// ( DownloadAll) // Закрываем файл (сделается на DownloadAll)
// // Попробуем записать его целиком
return S_FALSE; return S_FALSE;
} }
} }
else else
{ {
// // Закрываем хендл соединения
InternetCloseHandle ( hInternetSession ); InternetCloseHandle ( hInternetSession );
// ( DownloadAll) // Закрываем файл (сделается на DownloadAll)
// // Попробуем записать его целиком
return S_FALSE; return S_FALSE;
} }
// // Закрываем хендл соединения
InternetCloseHandle ( hInternetSession ); InternetCloseHandle ( hInternetSession );
return S_OK; return S_OK;
} }
DWORD DownloadFilePath ( HINTERNET hInternet, LPBYTE pBuffer, LONGLONG nStartByte, LONGLONG nEndByte, CString sFileURL ) DWORD DownloadFilePath ( HINTERNET hInternet, LPBYTE pBuffer, LONGLONG nStartByte, LONGLONG nEndByte, CString sFileURL )
{ {
// // Неоткрытая сессия
if ( NULL == hInternet ) if ( NULL == hInternet )
return -1; return -1;
// // Пришли непонятные параметры
if ( nStartByte > nEndByte || !pBuffer ) if ( nStartByte > nEndByte || !pBuffer )
return -1; return -1;
// ( nEndByte - nStartByte ) // Заголовок запроса ( содержит nEndByte - nStartByte байт )
CString sHTTPHdr = _T (""); sHTTPHdr.Format ( _T ("Range: bytes=%lld-%lld"), nStartByte, nEndByte ); CString sHTTPHdr = _T (""); sHTTPHdr.Format ( _T ("Range: bytes=%lld-%lld"), nStartByte, nEndByte );
// // Открываем ссылку для закачки
HINTERNET hInternetOpenURL = InternetOpenUrl ( hInternet, sFileURL, sHTTPHdr, -1, INTERNET_FLAG_RESYNCHRONIZE, 0 ); HINTERNET hInternetOpenURL = InternetOpenUrl ( hInternet, sFileURL, sHTTPHdr, -1, INTERNET_FLAG_RESYNCHRONIZE, 0 );
if ( NULL == hInternetOpenURL ) if ( NULL == hInternetOpenURL )
return -1; return -1;
// , // Открытие произошло, проверяем ответ
if ( FALSE == QueryStatusCode ( hInternetOpenURL, TRUE ) ) if ( FALSE == QueryStatusCode ( hInternetOpenURL, TRUE ) )
{ {
// // Закрываем хендл соединения
InternetCloseHandle ( hInternetOpenURL ); InternetCloseHandle ( hInternetOpenURL );
return -1; return -1;
} }
// // Какое колличество байт прочитано
DWORD dwBytesRead = 0; DWORD dwBytesRead = 0;
// // Читаем файл
if ( FALSE == InternetReadFile ( hInternetOpenURL, pBuffer, DOWNLOAD_FILE_SIZE, &dwBytesRead ) ) if ( FALSE == InternetReadFile ( hInternetOpenURL, pBuffer, DOWNLOAD_FILE_SIZE, &dwBytesRead ) )
{ {
// // Закрываем хендл соединения
InternetCloseHandle ( hInternetOpenURL ); InternetCloseHandle ( hInternetOpenURL );
return -1; return -1;
} }
// // Закрываем хендл соединения
InternetCloseHandle ( hInternetOpenURL ); InternetCloseHandle ( hInternetOpenURL );
return dwBytesRead; return dwBytesRead;
...@@ -239,93 +239,93 @@ protected : ...@@ -239,93 +239,93 @@ protected :
BOOL QueryStatusCode ( HINTERNET hInternet, BOOL bIsRanges ) BOOL QueryStatusCode ( HINTERNET hInternet, BOOL bIsRanges )
{ {
// - // Зачем проверять у неоткрытой сессии что-то
if ( NULL == hInternet ) if ( NULL == hInternet )
return FALSE; return FALSE;
// // Результат ответа
INT nResult = 0; INT nResult = 0;
// ( = 4 ) // Размер данных ответа ( должно быть = 4 )
DWORD dwLengthDataSize = 4; DWORD dwLengthDataSize = 4;
// , - FALSE // Делаем запрос, если не проходит - то возвращаем FALSE
if ( FALSE == HttpQueryInfo ( hInternet, HTTP_QUERY_STATUS_CODE | HTTP_QUERY_FLAG_NUMBER, &nResult, &dwLengthDataSize, NULL ) ) if ( FALSE == HttpQueryInfo ( hInternet, HTTP_QUERY_STATUS_CODE | HTTP_QUERY_FLAG_NUMBER, &nResult, &dwLengthDataSize, NULL ) )
return FALSE; return FALSE;
// , // Запрос прошел, теперь проверяем код ответа
if ( HTTP_STATUS_NOT_FOUND == nResult ) if ( HTTP_STATUS_NOT_FOUND == nResult )
{ {
// , - // Объект не найден, плохая ссылка или что-то еще
return FALSE; return FALSE;
} }
else if ( ( HTTP_STATUS_OK != nResult && FALSE == bIsRanges ) || ( HTTP_STATUS_PARTIAL_CONTENT != nResult && TRUE == bIsRanges ) ) else if ( ( HTTP_STATUS_OK != nResult && FALSE == bIsRanges ) || ( HTTP_STATUS_PARTIAL_CONTENT != nResult && TRUE == bIsRanges ) )
{ {
// - // Запрос не прошел по какой-то причине
return FALSE; return FALSE;
} }
// , // Все отлично, запрос прошел
return TRUE; return TRUE;
} }
// , -1 , // Проверяет, доступно ли для ресурса чтение частями и возвращает -1 если не доступно и размер данных, если доступно
LONGLONG IsAccept_Ranges ( HINTERNET hInternet ) LONGLONG IsAccept_Ranges ( HINTERNET hInternet )
{ {
// - // Зачем проверять у неоткрытой сессии что-то
if ( NULL == hInternet ) if ( NULL == hInternet )
return -1; return -1;
// // Результат ответа
char arrResult [ MAX_SIZE ] = { 0 }; char arrResult [ MAX_SIZE ] = { 0 };
// // Размер данных ответа
DWORD dwLengthDataSize = sizeof ( arrResult ); DWORD dwLengthDataSize = sizeof ( arrResult );
// , - FALSE // Делаем запрос, если не проходит - то возвращаем FALSE
if ( FALSE == HttpQueryInfoA ( hInternet, HTTP_QUERY_CONTENT_RANGE, &arrResult, &dwLengthDataSize, NULL ) ) if ( FALSE == HttpQueryInfoA ( hInternet, HTTP_QUERY_CONTENT_RANGE, &arrResult, &dwLengthDataSize, NULL ) )
{ {
// // Получаем последнюю ошибку
DWORD dwLastError = GetLastError (); DWORD dwLastError = GetLastError ();
if ( dwLastError == ERROR_HTTP_HEADER_NOT_FOUND ) if ( dwLastError == ERROR_HTTP_HEADER_NOT_FOUND )
{ {
// , // Не пришел заголовок, значит ресурс не поддерживает чтение частями
return -1; return -1;
} }
// - - FALSE // Возникла какая-то другая ошибка - возвращаем FALSE
return -1; return -1;
} }
// 0, // Если размер 0, то заголовка нет
if ( 0 >= dwLengthDataSize ) if ( 0 >= dwLengthDataSize )
return -1; return -1;
// CString // Приведем к CString
CString strResult ( arrResult ); CString strResult ( arrResult );
// // Содержит размер данных
LONGLONG nFileSize = 0; LONGLONG nFileSize = 0;
try try
{ {
// // Ищем индекс размера данных в строке
INT nStartIndex = strResult.Find ( CONTENT_RANGE ); INT nStartIndex = strResult.Find ( CONTENT_RANGE );
if ( -1 == nStartIndex ) if ( -1 == nStartIndex )
return -1; return -1;
// // Оставляем в строке только размер данных
strResult = strResult.Mid ( nStartIndex + CONTENT_RANGE_SIZE ); strResult = strResult.Mid ( nStartIndex + CONTENT_RANGE_SIZE );
// , LONGLONG // Теперь получим размер данных, переводя стринг в LONGLONG
nFileSize = _wtoi64 ( strResult.GetBuffer () ); nFileSize = _wtoi64 ( strResult.GetBuffer () );
// .. 0 ( 1 ) // Т.к. реально нумерация с 0 ( поэтому добавляем еще 1 байт )
if ( 0 < nFileSize ) if ( 0 < nFileSize )
nFileSize += 1; nFileSize += 1;
} }
catch ( ... ) catch ( ... )
{ {
// // не нашли возвращаем ошибку
return -1; return -1;
} }
// , , // Все отлично, ресурс поддерживает чтение частями, возвращаем размер
return nFileSize; return nFileSize;
} }
...@@ -336,7 +336,7 @@ protected : ...@@ -336,7 +336,7 @@ protected :
::fclose( m_pFile ); ::fclose( m_pFile );
m_pFile = NULL; m_pFile = NULL;
} }
// // Скачиваем файл
return URLDownloadToFile (NULL, sFileURL, strFileOutput, NULL, NULL); return URLDownloadToFile (NULL, sFileURL, strFileOutput, NULL, NULL);
} }
...@@ -355,12 +355,12 @@ public: ...@@ -355,12 +355,12 @@ public:
protected : protected :
FILE *m_pFile; // FILE *m_pFile; // Хэндл на временный файл
CString m_sFilePath; // CString m_sFilePath; // Путь к сохраненному файлу на диске
CString m_sFileUrl; // CString m_sFileUrl; // Ссылка на скачивание файла
BOOL m_bComplete; // BOOL m_bComplete; // Закачался файл или нет
BOOL m_bDelete; // BOOL m_bDelete; // Удалять ли файл в деструкторе
}; };
#else #else
...@@ -489,11 +489,11 @@ public: ...@@ -489,11 +489,11 @@ public:
} }
protected : protected :
CString m_sFilePath; // CString m_sFilePath; // Путь к сохраненному файлу на диске
CString m_sFileUrl; // CString m_sFileUrl; // Ссылка на скачивание файла
BOOL m_bComplete; // BOOL m_bComplete; // Закачался файл или нет
BOOL m_bDelete; // BOOL m_bDelete; // Удалять ли файл в деструкторе
}; };
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment
GitLab Nexedi Edition | About GitLab | About Nexedi | 沪ICP备2021021310号-2 | 沪ICP备2021021310号-7