procxx
/
kepka
mirror of https://github.com/procxx/kepka.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Add some more checks to file downloader.

This commit is contained in:
John Preston 2017-12-02 12:32:43 +04:00
parent 85b3d3f64d
commit 001be82566
1 changed files with 10 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
Telegram/SourceFiles/storage/file_download.cpp
View File

@ -381,6 +381,7 @@ void FileLoader::cancel(bool fail) {
_file.remove(); _file.remove();
} }
_data = QByteArray(); _data = QByteArray();
removeFromQueue();
if (fail) { if (fail) {
emit failed(this, started); emit failed(this, started);
@ -480,6 +481,8 @@ mtpFileLoader::RequestData mtpFileLoader::prepareRequest(int offset) const {
} }
void mtpFileLoader::makeRequest(int offset) { void mtpFileLoader::makeRequest(int offset) {
Expects(!_finished);
auto requestData = prepareRequest(offset); auto requestData = prepareRequest(offset);
auto send = [this, &requestData] { auto send = [this, &requestData] {
auto offset = requestData.offset; auto offset = requestData.offset;
@ -521,6 +524,7 @@ void mtpFileLoader::requestMoreCdnFileHashes() {
} }
void mtpFileLoader::normalPartLoaded(const MTPupload_File &result, mtpRequestId requestId) { void mtpFileLoader::normalPartLoaded(const MTPupload_File &result, mtpRequestId requestId) {
Expects(!_finished);
Expects(result.type() == mtpc_upload_fileCdnRedirect || result.type() == mtpc_upload_file); Expects(result.type() == mtpc_upload_fileCdnRedirect || result.type() == mtpc_upload_file);
auto offset = finishSentRequestGetOffset(requestId); auto offset = finishSentRequestGetOffset(requestId);
@ -547,6 +551,8 @@ void mtpFileLoader::webPartLoaded(const MTPupload_WebFile &result, mtpRequestId
} }
void mtpFileLoader::cdnPartLoaded(const MTPupload_CdnFile &result, mtpRequestId requestId) { void mtpFileLoader::cdnPartLoaded(const MTPupload_CdnFile &result, mtpRequestId requestId) {
Expects(!_finished);
auto offset = finishSentRequestGetOffset(requestId); auto offset = finishSentRequestGetOffset(requestId);
if (result.type() == mtpc_upload_cdnFileReuploadNeeded) { if (result.type() == mtpc_upload_cdnFileReuploadNeeded) {
auto requestData = RequestData(); auto requestData = RequestData();
@ -616,7 +622,9 @@ void mtpFileLoader::reuploadDone(const MTPVector<MTPCdnFileHash> &result, mtpReq
} }
void mtpFileLoader::getCdnFileHashesDone(const MTPVector<MTPCdnFileHash> &result, mtpRequestId requestId) { void mtpFileLoader::getCdnFileHashesDone(const MTPVector<MTPCdnFileHash> &result, mtpRequestId requestId) {
Expects(!_finished);
Expects(_cdnHashesRequestId == requestId); Expects(_cdnHashesRequestId == requestId);
_cdnHashesRequestId = 0; _cdnHashesRequestId = 0;
auto offset = finishSentRequestGetOffset(requestId); auto offset = finishSentRequestGetOffset(requestId);
@ -683,6 +691,8 @@ int mtpFileLoader::finishSentRequestGetOffset(mtpRequestId requestId) {
} }
void mtpFileLoader::partLoaded(int offset, base::const_byte_span bytes) { void mtpFileLoader::partLoaded(int offset, base::const_byte_span bytes) {
Expects(!_finished);
if (bytes.size()) { if (bytes.size()) {
if (_fileIsOpen) { if (_fileIsOpen) {
auto fsize = _file.size(); auto fsize = _file.size();