From 206becbb9fd308b3a86fbe0e1570f468cd495136 Mon Sep 17 00:00:00 2001 From: levlam Date: Sat, 23 Jul 2022 14:52:42 +0300 Subject: [PATCH] Safely traverse map. --- td/telegram/DownloadManager.cpp | 12 +++++++++++- 1 file changed, 11 insertions(+), 1 deletion(-) diff --git a/td/telegram/DownloadManager.cpp b/td/telegram/DownloadManager.cpp index 202a0272a..e6937b641 100644 --- a/td/telegram/DownloadManager.cpp +++ b/td/telegram/DownloadManager.cpp @@ -93,8 +93,18 @@ class DownloadManagerImpl final : public DownloadManager { Status toggle_all_is_paused(bool is_paused) final { TRY_STATUS(check_is_active()); + vector to_toggle; for (auto &it : files_) { - toggle_is_paused(*it.second, is_paused); + FileInfo &file_info = *it.second; + if (!is_completed(file_info) && is_paused != file_info.is_paused) { + to_toggle.push_back(file_info.file_id); + } + } + for (auto file_id : to_toggle) { + auto r_file_info_ptr = get_file_info(file_id); + if (r_file_info_ptr.is_ok()) { + toggle_is_paused(*r_file_info_ptr.ok(), is_paused); + } } return Status::OK();