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

Fix possible crash in CDN file loading.

This commit is contained in:
John Preston 2020-01-17 17:45:43 +03:00
parent 3643622e1b
commit d24735b217
2 changed files with 8 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
Telegram/SourceFiles/storage/download_manager_mtproto.cpp
View File

@ -487,6 +487,13 @@ void DownloadMtprotoTask::removeSession(int sessionIndex) {
redirect.push_back({ requestId, requestData.offset }); redirect.push_back({ requestId, requestData.offset });
} }
} }
for (auto &[requestData, bytes] : _cdnUncheckedParts) {
if (requestData.sessionIndex == sessionIndex) {
const auto newIndex = _owner->chooseSessionIndex(dcId());
Assert(newIndex < sessionIndex);
requestData.sessionIndex = newIndex;
}
}
for (const auto &[requestId, offset] : redirect) { for (const auto &[requestId, offset] : redirect) {
cancelRequest(requestId); cancelRequest(requestId);
const auto newIndex = _owner->chooseSessionIndex(dcId()); const auto newIndex = _owner->chooseSessionIndex(dcId());

2
Telegram/SourceFiles/storage/download_manager_mtproto.h
View File

@ -163,7 +163,7 @@ protected:
private: private:
struct RequestData { struct RequestData {
int offset = 0; int offset = 0;
int sessionIndex = 0; mutable int sessionIndex = 0;
int requestedInSession = 0; int requestedInSession = 0;
crl::time sent = 0; crl::time sent = 0;