// // Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2023 // // Distributed under the Boost Software License, Version 1.0. (See accompanying // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt) // #include "td/telegram/UpdatesManager.h" #include "td/telegram/AnimationsManager.h" #include "td/telegram/AttachMenuManager.h" #include "td/telegram/AuthManager.h" #include "td/telegram/AutosaveManager.h" #include "td/telegram/CallbackQueriesManager.h" #include "td/telegram/CallManager.h" #include "td/telegram/ChannelId.h" #include "td/telegram/ChatId.h" #include "td/telegram/ConfigManager.h" #include "td/telegram/ContactsManager.h" #include "td/telegram/DialogAction.h" #include "td/telegram/DialogFilterManager.h" #include "td/telegram/DialogId.h" #include "td/telegram/DialogInviteLink.h" #include "td/telegram/DialogParticipant.h" #include "td/telegram/DownloadManager.h" #include "td/telegram/EmojiStatus.h" #include "td/telegram/FolderId.h" #include "td/telegram/ForumTopicManager.h" #include "td/telegram/Global.h" #include "td/telegram/GroupCallManager.h" #include "td/telegram/InlineQueriesManager.h" #include "td/telegram/LanguagePackManager.h" #include "td/telegram/Location.h" #include "td/telegram/MessageId.h" #include "td/telegram/MessagesManager.h" #include "td/telegram/MessageTtl.h" #include "td/telegram/misc.h" #include "td/telegram/net/DcOptions.h" #include "td/telegram/net/NetQuery.h" #include "td/telegram/NotificationManager.h" #include "td/telegram/NotificationSettingsManager.h" #include "td/telegram/NotificationSettingsScope.h" #include "td/telegram/OptionManager.h" #include "td/telegram/OrderInfo.h" #include "td/telegram/PollId.h" #include "td/telegram/PollManager.h" #include "td/telegram/PrivacyManager.h" #include "td/telegram/PublicDialogType.h" #include "td/telegram/ScheduledServerMessageId.h" #include "td/telegram/SecretChatId.h" #include "td/telegram/SecretChatsManager.h" #include "td/telegram/ServerMessageId.h" #include "td/telegram/SpecialStickerSetType.h" #include "td/telegram/StateManager.h" #include "td/telegram/StickerSetId.h" #include "td/telegram/StickersManager.h" #include "td/telegram/StickerType.h" #include "td/telegram/StoryId.h" #include "td/telegram/StoryManager.h" #include "td/telegram/Td.h" #include "td/telegram/td_api.h" #include "td/telegram/TdDb.h" #include "td/telegram/telegram_api.h" #include "td/telegram/telegram_api.hpp" #include "td/telegram/ThemeManager.h" #include "td/telegram/Usernames.h" #include "td/telegram/WebPagesManager.h" #include "td/actor/MultiPromise.h" #include "td/actor/PromiseFuture.h" #include "td/utils/algorithm.h" #include "td/utils/buffer.h" #include "td/utils/logging.h" #include "td/utils/misc.h" #include "td/utils/Random.h" #include "td/utils/ScopeGuard.h" #include "td/utils/Slice.h" #include "td/utils/SliceBuilder.h" #include "td/utils/Status.h" #include "td/utils/StringBuilder.h" #include "td/utils/Time.h" #include #include namespace td { int VERBOSITY_NAME(get_difference) = VERBOSITY_NAME(INFO); class OnUpdate { UpdatesManager *updates_manager_; tl_object_ptr &update_; mutable Promise promise_; public: OnUpdate(UpdatesManager *updates_manager, tl_object_ptr &update, Promise &&promise) : updates_manager_(updates_manager), update_(update), promise_(std::move(promise)) { } template void operator()(T &obj) const { CHECK(&*update_ == &obj); updates_manager_->on_update(move_tl_object_as(update_), std::move(promise_)); } }; class GetUpdatesStateQuery final : public Td::ResultHandler { Promise> promise_; public: explicit GetUpdatesStateQuery(Promise> &&promise) : promise_(std::move(promise)) { } void send() { send_query(G()->net_query_creator().create(telegram_api::updates_getState())); } void on_result(BufferSlice packet) final { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(result_ptr.move_as_error()); } promise_.set_value(result_ptr.move_as_ok()); } void on_error(Status status) final { promise_.set_error(std::move(status)); } }; class PingServerQuery final : public Td::ResultHandler { Promise> promise_; public: explicit PingServerQuery(Promise> &&promise) : promise_(std::move(promise)) { } void send() { send_query(G()->net_query_creator().create(telegram_api::updates_getState())); } void on_result(BufferSlice packet) final { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(result_ptr.move_as_error()); } promise_.set_value(result_ptr.move_as_ok()); } void on_error(Status status) final { promise_.set_error(std::move(status)); } }; class InitSessionQuery final : public Td::ResultHandler { public: void send() { send_query(G()->net_query_creator().create(telegram_api::help_getCdnConfig())); } void on_result(BufferSlice) final { } void on_error(Status) final { } }; class GetDifferenceQuery final : public Td::ResultHandler { Promise> promise_; public: explicit GetDifferenceQuery(Promise> &&promise) : promise_(std::move(promise)) { } void send(int32 pts, int32 date, int32 qts) { send_query(G()->net_query_creator().create(telegram_api::updates_getDifference(0, pts, 0, 0, date, qts, 0))); } void on_result(BufferSlice packet) final { VLOG(get_difference) << "Receive getDifference result of size " << packet.size(); auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(result_ptr.move_as_error()); } promise_.set_value(result_ptr.move_as_ok()); } void on_error(Status status) final { promise_.set_error(std::move(status)); } }; class ConfirmPtsQtsQuery final : public Td::ResultHandler { public: void send(int32 pts, int32 qts) { int32 flags = telegram_api::updates_getDifference::PTS_LIMIT_MASK | telegram_api::updates_getDifference::QTS_LIMIT_MASK; send_query(G()->net_query_creator().create( telegram_api::updates_getDifference(flags, pts, 1, 0, std::numeric_limits::max(), qts, 1))); } void on_result(BufferSlice packet) final { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(result_ptr.move_as_error()); } } void on_error(Status status) final { if (!G()->is_expected_error(status)) { LOG(ERROR) << "Failed to confirm PTS/QTS: " << status; } } }; class GetPtsUpdateQuery final : public Td::ResultHandler { Promise> promise_; public: explicit GetPtsUpdateQuery(Promise> &&promise) : promise_(std::move(promise)) { } void send(int32 pts) { int32 flags = telegram_api::updates_getDifference::PTS_LIMIT_MASK | telegram_api::updates_getDifference::QTS_LIMIT_MASK; send_query(G()->net_query_creator().create( telegram_api::updates_getDifference(flags, pts, 1, 0, std::numeric_limits::max(), 0, 0))); } void on_result(BufferSlice packet) final { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(result_ptr.move_as_error()); } promise_.set_value(result_ptr.move_as_ok()); } void on_error(Status status) final { promise_.set_error(std::move(status)); } }; UpdatesManager::UpdatesManager(Td *td, ActorShared<> parent) : td_(td), parent_(std::move(parent)) { last_pts_save_time_ = last_qts_save_time_ = Time::now() - 2 * MAX_PTS_SAVE_DELAY; pending_audio_transcription_timeout_.set_callback(on_pending_audio_transcription_timeout_callback); pending_audio_transcription_timeout_.set_callback_data(static_cast(td_)); if (td_->option_manager_->get_option_integer("since_last_open") < 3600) { finished_first_get_difference_ = true; } } void UpdatesManager::tear_down() { parent_.reset(); LOG(DEBUG) << "Have " << being_processed_updates_ << " unprocessed updates to apply"; } void UpdatesManager::start_up() { class StateCallback final : public StateManager::Callback { public: explicit StateCallback(ActorId parent) : parent_(std::move(parent)) { } bool on_online(bool is_online) final { if (is_online) { send_closure(parent_, &UpdatesManager::try_reload_data); } return parent_.is_alive(); } private: ActorId parent_; }; send_closure(G()->state_manager(), &StateManager::add_callback, make_unique(actor_id(this))); next_data_reload_time_ = Time::now() - 1; } void UpdatesManager::hangup_shared() { ref_cnt_--; if (ref_cnt_ == 0) { stop(); } } void UpdatesManager::hangup() { pending_pts_updates_.clear(); postponed_pts_updates_.clear(); postponed_updates_.clear(); pending_seq_updates_.clear(); pending_qts_updates_.clear(); hangup_shared(); } ActorShared UpdatesManager::create_reference() { ref_cnt_++; return actor_shared(this, 1); } void UpdatesManager::check_pts_gap(void *td) { if (G()->close_flag()) { return; } CHECK(td != nullptr); static_cast(td)->updates_manager_.get()->repair_pts_gap(); } void UpdatesManager::repair_pts_gap() { if (running_get_difference_ || !postponed_pts_updates_.empty()) { return; } auto pts = get_pts() + 1; if (pending_pts_updates_.empty() || pending_pts_updates_.begin()->first != pts + 1) { return; } VLOG(get_difference) << "Fetch update with PTS = " << pts; pts_short_gap_++; auto promise = PromiseCreator::lambda([pts](Result> result) { if (result.is_ok()) { send_closure(G()->updates_manager(), &UpdatesManager::on_get_pts_update, pts, result.move_as_ok()); } }); td_->create_handler(std::move(promise))->send(pts - 1); } void UpdatesManager::fill_pts_gap(void *td) { if (G()->close_flag()) { return; } CHECK(td != nullptr); auto updates_manager = static_cast(td)->updates_manager_.get(); auto min_pts = std::numeric_limits::max(); auto min_pts_count = 0; auto max_pts = 0; if (!updates_manager->pending_pts_updates_.empty()) { auto &min_update = updates_manager->pending_pts_updates_.begin()->second; if (min_update.pts < min_pts) { min_pts = min_update.pts; min_pts_count = min_update.pts_count; } max_pts = max(max_pts, updates_manager->pending_pts_updates_.rbegin()->first); } if (!updates_manager->postponed_pts_updates_.empty()) { auto &min_update = updates_manager->postponed_pts_updates_.begin()->second; if (min_update.pts < min_pts) { min_pts = min_update.pts; min_pts_count = min_update.pts_count; } max_pts = max(max_pts, updates_manager->postponed_pts_updates_.rbegin()->first); } string source = PSTRING() << "PTS from " << updates_manager->get_pts() << " to " << min_pts << "(-" << min_pts_count << ")-" << max_pts; updates_manager->pts_gap_++; fill_gap(td, source.c_str()); } void UpdatesManager::fill_seq_gap(void *td) { if (G()->close_flag()) { return; } CHECK(td != nullptr); auto updates_manager = static_cast(td)->updates_manager_.get(); auto min_seq = std::numeric_limits::max(); auto max_seq = 0; if (!updates_manager->pending_seq_updates_.empty()) { min_seq = updates_manager->pending_seq_updates_.begin()->first; max_seq = updates_manager->pending_seq_updates_.rbegin()->second.seq_end; } string source = PSTRING() << "seq from " << updates_manager->seq_ << " to " << min_seq << '-' << max_seq; fill_gap(td, source.c_str()); } void UpdatesManager::fill_qts_gap(void *td) { if (G()->close_flag()) { return; } CHECK(td != nullptr); auto updates_manager = static_cast(td)->updates_manager_.get(); auto min_qts = std::numeric_limits::max(); auto max_qts = 0; if (!updates_manager->pending_qts_updates_.empty()) { min_qts = updates_manager->pending_qts_updates_.begin()->first; max_qts = updates_manager->pending_qts_updates_.rbegin()->first; } string source = PSTRING() << "QTS from " << updates_manager->get_qts() << " to " << min_qts << '-' << max_qts; updates_manager->qts_gap_++; fill_gap(td, source.c_str()); } void UpdatesManager::fill_get_difference_gap(void *td) { fill_gap(td, nullptr); } void UpdatesManager::fill_gap(void *td, const char *source) { if (G()->close_flag()) { return; } CHECK(td != nullptr); if (!static_cast(td)->auth_manager_->is_authorized()) { return; } auto updates_manager = static_cast(td)->updates_manager_.get(); if (source != nullptr && !updates_manager->running_get_difference_) { auto auth_key_id = updates_manager->get_most_unused_auth_key_id(); uint64 update_count = 0; double active_time = 0.0; double delay_time = 0.0; if (auth_key_id != 0) { auto now = Time::now(); auto &info = updates_manager->session_infos_[auth_key_id]; update_count = info.update_count; active_time = now - info.first_update_time; delay_time = now - info.last_update_time; } LOG(WARNING) << "Filling gap in " << source << " by running getDifference. Receive no updates from session with auth key " << auth_key_id << " for " << delay_time << " seconds, active for " << active_time << " seconds and having " << update_count << " received updates"; } updates_manager->get_difference("fill_gap"); } void UpdatesManager::on_pending_audio_transcription_timeout_callback(void *td, int64 transcription_id) { if (G()->close_flag()) { return; } CHECK(td != nullptr); if (!static_cast(td)->auth_manager_->is_authorized()) { return; } auto updates_manager = static_cast(td)->updates_manager_.get(); send_closure_later(updates_manager->actor_id(updates_manager), &UpdatesManager::on_pending_audio_transcription_failed, transcription_id, Status::Error(500, "Timeout expired")); } void UpdatesManager::get_difference(const char *source) { if (G()->close_flag() || !td_->auth_manager_->is_authorized()) { return; } if (get_pts() == -1) { init_state(); return; } if (running_get_difference_) { VLOG(get_difference) << "Skip running getDifference from " << source << " because it is already running"; return; } run_get_difference(false, source); } void UpdatesManager::run_get_difference(bool is_recursive, const char *source) { CHECK(get_pts() != -1); CHECK(td_->auth_manager_->is_authorized()); CHECK(!running_get_difference_); running_get_difference_ = true; int32 pts = get_pts(); int32 date = get_date(); int32 qts = get_qts(); if (pts < 0) { pts = 0; } VLOG(get_difference) << "-----BEGIN GET DIFFERENCE----- from " << source << " with PTS = " << pts << ", QTS = " << qts << ", date = " << date; before_get_difference(false); if (!is_recursive) { min_postponed_update_pts_ = 0; min_postponed_update_qts_ = 0; } auto promise = PromiseCreator::lambda([](Result> result) { if (result.is_ok()) { send_closure(G()->updates_manager(), &UpdatesManager::on_get_difference, result.move_as_ok()); } else { send_closure(G()->updates_manager(), &UpdatesManager::on_failed_get_difference, result.move_as_error()); } }); td_->create_handler(std::move(promise))->send(pts, date, qts); last_confirmed_pts_ = pts; last_confirmed_qts_ = qts; } void UpdatesManager::before_get_difference(bool is_initial) { // may be called many times before after_get_difference is called send_closure(G()->state_manager(), &StateManager::on_synchronized, false); if (can_postpone_updates()) { postponed_pts_updates_.insert(std::make_move_iterator(pending_pts_updates_.begin()), std::make_move_iterator(pending_pts_updates_.end())); } else { for (auto &update : pending_pts_updates_) { update.second.promise.set_value(Unit()); } } drop_all_pending_pts_updates(); send_closure_later(td_->notification_manager_actor_, &NotificationManager::before_get_difference); if (get_difference_start_time_ <= 0) { get_difference_start_time_ = Time::now(); } } Promise<> UpdatesManager::add_pts(int32 pts) { auto id = pts_manager_.add_pts(pts); return create_event_promise(self_closure(this, &UpdatesManager::on_pts_ack, id)); } Promise<> UpdatesManager::add_qts(int32 qts) { auto id = qts_manager_.add_pts(qts); return create_event_promise(self_closure(this, &UpdatesManager::on_qts_ack, id)); } void UpdatesManager::on_pts_ack(PtsManager::PtsId ack_token) { auto old_pts = pts_manager_.db_pts(); auto new_pts = pts_manager_.finish(ack_token); if (old_pts != new_pts) { save_pts(new_pts); } } void UpdatesManager::on_qts_ack(PtsManager::PtsId ack_token) { auto old_qts = qts_manager_.db_pts(); auto new_qts = qts_manager_.finish(ack_token); if (old_qts != new_qts) { save_qts(new_qts); } } void UpdatesManager::save_pts(int32 pts) { if (pts == std::numeric_limits::max()) { G()->td_db()->get_binlog_pmc()->erase("updates.pts"); last_pts_save_time_ -= 2 * MAX_PTS_SAVE_DELAY; pending_pts_ = 0; } else if (!td_->ignore_background_updates()) { auto now = Time::now(); auto delay = last_pts_save_time_ + MAX_PTS_SAVE_DELAY - now; if (delay <= 0 || !td_->auth_manager_->is_bot()) { last_pts_save_time_ = now; pending_pts_ = 0; G()->td_db()->get_binlog_pmc()->set("updates.pts", to_string(pts)); } else { pending_pts_ = pts; if (!has_timeout()) { set_timeout_in(delay); } } } } void UpdatesManager::save_qts(int32 qts) { if (!td_->ignore_background_updates()) { auto now = Time::now(); auto delay = last_qts_save_time_ + MAX_PTS_SAVE_DELAY - now; if (delay <= 0 || !td_->auth_manager_->is_bot()) { last_qts_save_time_ = now; pending_qts_ = 0; G()->td_db()->get_binlog_pmc()->set("updates.qts", to_string(qts)); } else { pending_qts_ = qts; if (!has_timeout()) { set_timeout_in(delay); } } } } void UpdatesManager::timeout_expired() { if (pending_pts_ != 0) { last_pts_save_time_ -= 2 * MAX_PTS_SAVE_DELAY; save_pts(pending_pts_); CHECK(pending_pts_ == 0); } if (pending_qts_ != 0) { last_qts_save_time_ -= 2 * MAX_PTS_SAVE_DELAY; save_qts(pending_qts_); CHECK(pending_qts_ == 0); } } Promise<> UpdatesManager::set_pts(int32 pts, const char *source) { if (pts == std::numeric_limits::max()) { LOG(WARNING) << "Update PTS from " << get_pts() << " to -1 from " << source; save_pts(pts); // drop saved PTS value auto result = add_pts(pts); init_state(); return result; } Promise<> result; auto old_pts = get_pts(); if (pts > old_pts || (0 < pts && pts < old_pts - 1000009)) { // PTS can only go up or drop cardinally if (pts < old_pts - 1000009) { LOG(WARNING) << "PTS decreases from " << old_pts << " to " << pts << " from " << source; } else { LOG(INFO) << "Update PTS from " << old_pts << " to " << pts << " from " << source; pts_diff_ += pts - old_pts; if (pts_diff_ >= 1000000) { LOG(WARNING) << "Fixed " << pts_gap_ << " PTS gaps and " << pts_fixed_short_gap_ << " short gaps by sending " << pts_short_gap_ << " requests"; pts_short_gap_ = 0; pts_fixed_short_gap_ = 0; pts_gap_ = 0; pts_diff_ = 0; } } result = add_pts(pts); if (last_confirmed_pts_ < get_pts() - FORCED_GET_DIFFERENCE_PTS_DIFF && last_confirmed_pts_ != 0) { confirm_pts_qts(get_qts()); } } else if (pts < get_pts() && (pts > 1 || td_->option_manager_->get_option_integer("session_count") <= 1)) { LOG(ERROR) << "Receive wrong PTS = " << pts << " from " << source << ". Current PTS = " << get_pts(); } return result; } void UpdatesManager::set_date(int32 date, bool from_update, string date_source) { if (date > date_) { LOG(INFO) << "Update date to " << date; if (from_update && false) { // date in updates is decreased by the server date--; if (date == date_) { return; } } auto now = G()->unix_time(); if (date_ > now + 1) { LOG(ERROR) << "Receive wrong by " << (date_ - now) << " date = " << date_ << " from " << date_source; date_ = now; if (date_ <= date) { return; } } date_ = date; date_source_ = std::move(date_source); if (!td_->ignore_background_updates()) { G()->td_db()->get_binlog_pmc()->set("updates.date", to_string(date)); } } else if (date < date_) { if (from_update) { date++; if (date == date_) { return; } } LOG(ERROR) << "Receive wrong by " << (date_ - date) << " date = " << date << " from " << date_source << ". Current date = " << date_ << " from " << date_source_; } } bool UpdatesManager::is_acceptable_user(UserId user_id) const { return td_->contacts_manager_->have_user_force(user_id, "is_acceptable_user") && td_->contacts_manager_->have_user(user_id); } bool UpdatesManager::is_acceptable_chat(ChatId chat_id) const { return td_->contacts_manager_->have_chat_force(chat_id); } bool UpdatesManager::is_acceptable_channel(ChannelId channel_id) const { return td_->contacts_manager_->have_channel_force(channel_id); } bool UpdatesManager::is_acceptable_peer(const tl_object_ptr &peer) const { if (peer == nullptr) { return true; } DialogId dialog_id(peer); switch (dialog_id.get_type()) { case DialogType::User: if (!is_acceptable_user(dialog_id.get_user_id())) { return false; } break; case DialogType::Chat: if (!is_acceptable_chat(dialog_id.get_chat_id())) { return false; } break; case DialogType::Channel: if (!is_acceptable_channel(dialog_id.get_channel_id())) { return false; } break; case DialogType::None: return false; case DialogType::SecretChat: default: UNREACHABLE(); return false; } return true; } bool UpdatesManager::is_acceptable_message_entities( const vector> &message_entities) const { for (auto &entity : message_entities) { if (entity->get_id() == telegram_api::messageEntityMentionName::ID) { auto entity_mention_name = static_cast(entity.get()); UserId user_id(entity_mention_name->user_id_); if (!is_acceptable_user(user_id) || td_->contacts_manager_->get_input_user(user_id).is_error()) { return false; } } } return true; } bool UpdatesManager::is_acceptable_reply_markup(const tl_object_ptr &reply_markup) const { if (reply_markup == nullptr || reply_markup->get_id() != telegram_api::replyInlineMarkup::ID) { return true; } for (const auto &row : static_cast(reply_markup.get())->rows_) { for (const auto &button : row->buttons_) { if (button->get_id() == telegram_api::keyboardButtonUserProfile::ID) { auto user_profile_button = static_cast(button.get()); UserId user_id(user_profile_button->user_id_); if (!is_acceptable_user(user_id) || td_->contacts_manager_->get_input_user(user_id).is_error()) { return false; } } } } return true; } bool UpdatesManager::is_acceptable_message_reply_header( const telegram_api::object_ptr &header) const { if (header == nullptr) { return true; } switch (header->get_id()) { case telegram_api::messageReplyHeader::ID: { auto reply_header = static_cast(header.get()); if (!is_acceptable_peer(reply_header->reply_to_peer_id_)) { return false; } return true; } case telegram_api::messageReplyStoryHeader::ID: { auto reply_header = static_cast(header.get()); if (!is_acceptable_user(UserId(reply_header->user_id_))) { return false; } return true; } default: UNREACHABLE(); return true; } } bool UpdatesManager::is_acceptable_message_forward_header( const telegram_api::object_ptr &header) const { if (header == nullptr) { return true; } if (!is_acceptable_peer(header->from_id_)) { return false; } if (!is_acceptable_peer(header->saved_from_peer_)) { return false; } return true; } bool UpdatesManager::is_acceptable_message(const telegram_api::Message *message_ptr) const { CHECK(message_ptr != nullptr); int32 constructor_id = message_ptr->get_id(); switch (constructor_id) { case telegram_api::messageEmpty::ID: return true; case telegram_api::message::ID: { auto message = static_cast(message_ptr); if (!is_acceptable_peer(message->peer_id_)) { return false; } if (!is_acceptable_peer(message->from_id_)) { return false; } if (!is_acceptable_message_reply_header(message->reply_to_)) { return false; } if (!is_acceptable_message_forward_header(message->fwd_from_)) { return false; } if ((message->flags_ & MessagesManager::MESSAGE_FLAG_IS_SENT_VIA_BOT) && !is_acceptable_user(UserId(message->via_bot_id_))) { return false; } if (!is_acceptable_message_entities(message->entities_)) { return false; } if (message->media_ != nullptr) { auto media_id = message->media_->get_id(); if (media_id == telegram_api::messageMediaContact::ID) { auto message_media = static_cast(message->media_.get()); UserId user_id(message_media->user_id_); if (user_id != UserId() && !is_acceptable_user(user_id)) { return false; } } if (media_id == telegram_api::messageMediaStory::ID) { auto message_media = static_cast(message->media_.get()); UserId user_id(message_media->user_id_); if (!is_acceptable_user(user_id)) { return false; } } /* // the users and chats are always min, so no need to check if (media_id == telegram_api::messageMediaPoll::ID) { auto message_media_poll = static_cast(message->media_.get()); for (auto recent_voter : message_media_poll->results_->recent_voters_) { if (!is_acceptable_peer(recent_voter)) { return false; } } } */ /* // the channel is always min, so no need to check if (media_id == telegram_api::messageMediaWebPage::ID) { auto message_media_web_page = static_cast(message->media_.get()); if (message_media_web_page->webpage_->get_id() == telegram_api::webPage::ID) { auto web_page = static_cast(message_media_web_page->webpage_.get()); if (web_page->cached_page_ != nullptr) { const vector> *page_blocks = nullptr; downcast_call(*web_page->cached_page_, [&page_blocks](auto &page) { page_blocks = &page.blocks_; }); CHECK(page_blocks != nullptr); for (auto &page_block : *page_blocks) { if (page_block->get_id() == telegram_api::pageBlockChannel::ID) { auto page_block_channel = static_cast(page_block.get()); auto channel_id = ContactsManager::get_channel_id(page_block_channel->channel_); if (channel_id.is_valid()) { if (!is_acceptable_channel(channel_id)) { return false; } } else { LOG(ERROR) << "Receive wrong channel " << to_string(page_block_channel->channel_); } } } } } } */ } else { CHECK(message->media_ == nullptr); } /* // the dialogs are always min, so no need to check if (message->replies_ != nullptr) { for (auto &peer : message->replies_->recent_repliers_) { if (!is_acceptable_peer(peer)) { return false; } } } */ break; } case telegram_api::messageService::ID: { auto message = static_cast(message_ptr); if (!is_acceptable_peer(message->peer_id_)) { return false; } if (!is_acceptable_peer(message->from_id_)) { return false; } const telegram_api::MessageAction *action = message->action_.get(); CHECK(action != nullptr); switch (action->get_id()) { case telegram_api::messageActionEmpty::ID: case telegram_api::messageActionChatEditTitle::ID: case telegram_api::messageActionChatEditPhoto::ID: case telegram_api::messageActionChatDeletePhoto::ID: case telegram_api::messageActionCustomAction::ID: case telegram_api::messageActionBotAllowed::ID: case telegram_api::messageActionHistoryClear::ID: case telegram_api::messageActionChannelCreate::ID: case telegram_api::messageActionPinMessage::ID: case telegram_api::messageActionGameScore::ID: case telegram_api::messageActionPhoneCall::ID: case telegram_api::messageActionPaymentSent::ID: case telegram_api::messageActionPaymentSentMe::ID: case telegram_api::messageActionScreenshotTaken::ID: case telegram_api::messageActionSecureValuesSent::ID: case telegram_api::messageActionSecureValuesSentMe::ID: case telegram_api::messageActionContactSignUp::ID: case telegram_api::messageActionGroupCall::ID: case telegram_api::messageActionGroupCallScheduled::ID: case telegram_api::messageActionSetChatTheme::ID: case telegram_api::messageActionChatJoinedByRequest::ID: case telegram_api::messageActionWebViewDataSentMe::ID: case telegram_api::messageActionWebViewDataSent::ID: case telegram_api::messageActionGiftPremium::ID: case telegram_api::messageActionTopicCreate::ID: case telegram_api::messageActionTopicEdit::ID: case telegram_api::messageActionSuggestProfilePhoto::ID: case telegram_api::messageActionSetChatWallPaper::ID: case telegram_api::messageActionSetSameChatWallPaper::ID: break; case telegram_api::messageActionChatCreate::ID: { auto chat_create = static_cast(action); for (auto &user : chat_create->users_) { if (!is_acceptable_user(UserId(user))) { return false; } } break; } case telegram_api::messageActionChatAddUser::ID: { auto chat_add_user = static_cast(action); for (auto &user : chat_add_user->users_) { if (!is_acceptable_user(UserId(user))) { return false; } } break; } case telegram_api::messageActionChatJoinedByLink::ID: // inviter_id_ isn't used break; case telegram_api::messageActionChatDeleteUser::ID: { auto chat_delete_user = static_cast(action); if (!is_acceptable_user(UserId(chat_delete_user->user_id_))) { return false; } break; } case telegram_api::messageActionChatMigrateTo::ID: { auto chat_migrate_to = static_cast(action); if (!is_acceptable_channel(ChannelId(chat_migrate_to->channel_id_))) { return false; } break; } case telegram_api::messageActionChannelMigrateFrom::ID: { auto channel_migrate_from = static_cast(action); if (!is_acceptable_chat(ChatId(channel_migrate_from->chat_id_))) { return false; } break; } case telegram_api::messageActionGeoProximityReached::ID: { auto geo_proximity_reached = static_cast(action); if (!is_acceptable_peer(geo_proximity_reached->from_id_)) { return false; } if (!is_acceptable_peer(geo_proximity_reached->to_id_)) { return false; } break; } case telegram_api::messageActionInviteToGroupCall::ID: { auto invite_to_group_call = static_cast(action); for (auto &user : invite_to_group_call->users_) { if (!is_acceptable_user(UserId(user))) { return false; } } break; } case telegram_api::messageActionSetMessagesTTL::ID: { auto set_messages_ttl = static_cast(action); if (set_messages_ttl->auto_setting_from_ != 0 && !is_acceptable_user(UserId(set_messages_ttl->auto_setting_from_))) { return false; } break; } case telegram_api::messageActionRequestedPeer::ID: { auto requested_peer = static_cast(action); if (!td_->auth_manager_->is_bot() && !is_acceptable_peer(requested_peer->peer_)) { return false; } break; } default: UNREACHABLE(); return false; } break; } default: UNREACHABLE(); return false; } return true; } bool UpdatesManager::is_acceptable_update(const telegram_api::Update *update) const { if (update == nullptr) { return true; } int32 id = update->get_id(); const telegram_api::Message *message = nullptr; if (id == telegram_api::updateNewMessage::ID) { message = static_cast(update)->message_.get(); } if (id == telegram_api::updateNewChannelMessage::ID) { message = static_cast(update)->message_.get(); } if (id == telegram_api::updateNewScheduledMessage::ID) { message = static_cast(update)->message_.get(); } if (id == telegram_api::updateEditMessage::ID) { message = static_cast(update)->message_.get(); } if (id == telegram_api::updateEditChannelMessage::ID) { message = static_cast(update)->message_.get(); } if (message != nullptr) { return is_acceptable_message(message); } if (id == telegram_api::updateDraftMessage::ID) { auto update_draft_message = static_cast(update); CHECK(update_draft_message->draft_ != nullptr); if (update_draft_message->draft_->get_id() == telegram_api::draftMessage::ID) { auto draft_message = static_cast(update_draft_message->draft_.get()); return is_acceptable_message_entities(draft_message->entities_); } } return true; } void UpdatesManager::on_get_updates(tl_object_ptr &&updates_ptr, Promise &&promise) { send_closure_later(actor_id(this), &UpdatesManager::on_get_updates_impl, std::move(updates_ptr), std::move(promise)); } void UpdatesManager::on_get_updates_impl(tl_object_ptr updates_ptr, Promise promise) { CHECK(updates_ptr != nullptr); promise = PromiseCreator::lambda( [promise = std::move(promise), update_ids = get_update_ids(updates_ptr.get())](Result result) mutable { if (!G()->close_flag() && result.is_error()) { LOG(ERROR) << "Failed to process updates " << update_ids << ": " << result.error(); } promise.set_value(Unit()); }); auto updates_type = updates_ptr->get_id(); if (updates_type != telegram_api::updateShort::ID) { LOG(INFO) << "Receive " << to_string(updates_ptr); } if (!td_->auth_manager_->is_authorized()) { if (updates_type == telegram_api::updateShort::ID && !G()->close_flag()) { auto &update = static_cast(updates_ptr.get())->update_; auto update_id = update->get_id(); if (update_id == telegram_api::updateLoginToken::ID) { td_->auth_manager_->on_update_login_token(); return promise.set_value(Unit()); } switch (update_id) { case telegram_api::updateServiceNotification::ID: case telegram_api::updateDcOptions::ID: case telegram_api::updateConfig::ID: case telegram_api::updateLangPackTooLong::ID: case telegram_api::updateLangPack::ID: LOG(INFO) << "Apply without authorization " << to_string(updates_ptr); downcast_call(*update, OnUpdate(this, update, std::move(promise))); return; default: break; } } LOG(INFO) << "Ignore received before authorization or after logout " << to_string(updates_ptr); return promise.set_value(Unit()); } switch (updates_type) { case telegram_api::updatesTooLong::ID: get_difference("updatesTooLong"); promise.set_value(Unit()); break; case telegram_api::updateShortMessage::ID: { auto update = move_tl_object_as(updates_ptr); if (update->flags_ & MessagesManager::MESSAGE_FLAG_HAS_REPLY_MARKUP) { LOG(ERROR) << "Receive updateShortMessage with reply_markup"; update->flags_ ^= MessagesManager::MESSAGE_FLAG_HAS_REPLY_MARKUP; } if (update->flags_ & MessagesManager::MESSAGE_FLAG_HAS_MEDIA) { LOG(ERROR) << "Receive updateShortMessage with media"; update->flags_ ^= MessagesManager::MESSAGE_FLAG_HAS_MEDIA; } auto from_id = update->out_ ? td_->contacts_manager_->get_my_id().get() : update->user_id_; update->flags_ |= MessagesManager::MESSAGE_FLAG_HAS_FROM_ID; auto message = make_tl_object( update->flags_, update->out_, update->mentioned_, update->media_unread_, update->silent_, false, false, false, false, false, false, update->id_, make_tl_object(from_id), make_tl_object(update->user_id_), std::move(update->fwd_from_), update->via_bot_id_, std::move(update->reply_to_), update->date_, update->message_, nullptr, nullptr, std::move(update->entities_), 0, 0, nullptr, 0, string(), 0, nullptr, Auto(), update->ttl_period_); on_pending_update( make_tl_object(std::move(message), update->pts_, update->pts_count_), 0, std::move(promise), "telegram_api::updateShortMessage"); break; } case telegram_api::updateShortChatMessage::ID: { auto update = move_tl_object_as(updates_ptr); if (update->flags_ & MessagesManager::MESSAGE_FLAG_HAS_REPLY_MARKUP) { LOG(ERROR) << "Receive updateShortChatMessage with reply_markup"; update->flags_ ^= MessagesManager::MESSAGE_FLAG_HAS_REPLY_MARKUP; } if (update->flags_ & MessagesManager::MESSAGE_FLAG_HAS_MEDIA) { LOG(ERROR) << "Receive updateShortChatMessage with media"; update->flags_ ^= MessagesManager::MESSAGE_FLAG_HAS_MEDIA; } update->flags_ |= MessagesManager::MESSAGE_FLAG_HAS_FROM_ID; auto message = make_tl_object( update->flags_, update->out_, update->mentioned_, update->media_unread_, update->silent_, false, false, false, false, false, false, update->id_, make_tl_object(update->from_id_), make_tl_object(update->chat_id_), std::move(update->fwd_from_), update->via_bot_id_, std::move(update->reply_to_), update->date_, update->message_, nullptr, nullptr, std::move(update->entities_), 0, 0, nullptr, 0, string(), 0, nullptr, Auto(), update->ttl_period_); on_pending_update( make_tl_object(std::move(message), update->pts_, update->pts_count_), 0, std::move(promise), "telegram_api::updateShortChatMessage"); break; } case telegram_api::updateShort::ID: { auto update = move_tl_object_as(updates_ptr); LOG(DEBUG) << "Receive " << oneline(to_string(update)); if (!is_acceptable_update(update->update_.get())) { LOG(ERROR) << "Receive unacceptable short update: " << oneline(to_string(update)); promise.set_value(Unit()); return get_difference("unacceptable short update"); } short_update_date_ = update->date_; downcast_call(*update->update_, OnUpdate(this, update->update_, std::move(promise))); short_update_date_ = 0; break; } case telegram_api::updatesCombined::ID: { auto updates = move_tl_object_as(updates_ptr); td_->contacts_manager_->on_get_users(std::move(updates->users_), "updatesCombined"); td_->contacts_manager_->on_get_chats(std::move(updates->chats_), "updatesCombined"); on_pending_updates(std::move(updates->updates_), updates->seq_start_, updates->seq_, updates->date_, Time::now(), std::move(promise), "telegram_api::updatesCombined"); break; } case telegram_api::updates::ID: { auto updates = move_tl_object_as(updates_ptr); string source_str; const char *source = "updates"; if (updates->updates_.size() == 1 && updates->updates_[0] != nullptr) { source_str = PSTRING() << "update " << updates->updates_[0]->get_id(); source = source_str.c_str(); } td_->contacts_manager_->on_get_users(std::move(updates->users_), source); td_->contacts_manager_->on_get_chats(std::move(updates->chats_), source); on_pending_updates(std::move(updates->updates_), updates->seq_, updates->seq_, updates->date_, Time::now(), std::move(promise), "telegram_api::updates"); break; } case telegram_api::updateShortSentMessage::ID: LOG(ERROR) << "Receive " << oneline(to_string(updates_ptr)); get_difference("updateShortSentMessage"); promise.set_value(Unit()); break; default: UNREACHABLE(); } } void UpdatesManager::on_failed_get_updates_state(Status &&error) { if (G()->close_flag() || !td_->auth_manager_->is_authorized()) { return; } if (error.code() != 401) { LOG(ERROR) << "Receive updates.getState error: " << error; } running_get_difference_ = false; schedule_get_difference("on_failed_get_updates_state"); } void UpdatesManager::on_failed_get_difference(Status &&error) { if (G()->close_flag() || !td_->auth_manager_->is_authorized()) { return; } if (error.code() != 401) { LOG(ERROR) << "Receive updates.getDifference error: " << error; } running_get_difference_ = false; schedule_get_difference("on_failed_get_difference"); if (error.message() == Slice("PERSISTENT_TIMESTAMP_INVALID")) { set_pts(std::numeric_limits::max(), "PERSISTENT_TIMESTAMP_INVALID").set_value(Unit()); } } void UpdatesManager::schedule_get_difference(const char *source) { if (G()->close_flag() || !td_->auth_manager_->is_authorized()) { return; } if (!retry_timeout_.has_timeout()) { LOG(WARNING) << "Schedule getDifference in " << retry_time_ << " seconds with PTS = " << get_pts() << ", QTS = " << get_qts() << ", date = " << get_date() << " from " << source; retry_timeout_.set_callback(std::move(fill_get_difference_gap)); retry_timeout_.set_callback_data(static_cast(td_)); retry_timeout_.set_timeout_in(retry_time_); retry_time_ *= 2; if (retry_time_ > 60) { retry_time_ = Random::fast(60, 80); } } else { VLOG(get_difference) << "Schedule getDifference from " << source; } } void UpdatesManager::on_get_updates_state(tl_object_ptr &&state, const char *source) { CHECK(state != nullptr); VLOG(get_difference) << "Receive " << oneline(to_string(state)) << " from " << source; // TODO use state->unread_count; if (get_pts() == std::numeric_limits::max()) { LOG(WARNING) << "Restore PTS to " << state->pts_; // restoring right PTS CHECK(pending_pts_updates_.empty()); auto real_running_get_difference = running_get_difference_; running_get_difference_ = false; process_postponed_pts_updates(); // drop all updates with old PTS running_get_difference_ = real_running_get_difference; pts_manager_.init(state->pts_); last_confirmed_pts_ = get_pts(); last_pts_save_time_ = Time::now() - 2 * MAX_PTS_SAVE_DELAY; save_pts(state->pts_); } else { string full_source = "on_get_updates_state " + oneline(to_string(state)) + " from " + source; set_pts(state->pts_, full_source.c_str()).set_value(Unit()); set_date(state->date_, false, std::move(full_source)); add_qts(state->qts_).set_value(Unit()); seq_ = state->seq_; } if (running_get_difference_) { // called from getUpdatesState running_get_difference_ = false; after_get_difference(); } } const vector> *UpdatesManager::get_updates( const telegram_api::Updates *updates_ptr) { switch (updates_ptr->get_id()) { case telegram_api::updatesTooLong::ID: case telegram_api::updateShortMessage::ID: case telegram_api::updateShortChatMessage::ID: case telegram_api::updateShort::ID: case telegram_api::updateShortSentMessage::ID: LOG(ERROR) << "Receive " << oneline(to_string(*updates_ptr)) << " instead of updates"; return nullptr; case telegram_api::updatesCombined::ID: return &static_cast(updates_ptr)->updates_; case telegram_api::updates::ID: return &static_cast(updates_ptr)->updates_; default: UNREACHABLE(); return nullptr; } } vector> *UpdatesManager::get_updates(telegram_api::Updates *updates_ptr) { return const_cast> *>( get_updates(static_cast(updates_ptr))); } bool UpdatesManager::are_empty_updates(const telegram_api::Updates *updates_ptr) { switch (updates_ptr->get_id()) { case telegram_api::updatesTooLong::ID: case telegram_api::updateShortSentMessage::ID: return true; case telegram_api::updateShortMessage::ID: case telegram_api::updateShortChatMessage::ID: case telegram_api::updateShort::ID: return false; case telegram_api::updatesCombined::ID: return static_cast(updates_ptr)->updates_.empty(); case telegram_api::updates::ID: return static_cast(updates_ptr)->updates_.empty(); default: UNREACHABLE(); return true; } } vector UpdatesManager::extract_group_invite_privacy_forbidden_updates( tl_object_ptr &updates_ptr) { auto updates = get_updates(updates_ptr.get()); if (updates == nullptr) { LOG(ERROR) << "Can't find updateGroupInvitePrivacyForbidden updates"; return {}; } LOG(INFO) << "Extract updateGroupInvitePrivacyForbidden updates from " << updates->size() << " updates"; vector user_ids; for (auto &update_ptr : *updates) { if (update_ptr->get_id() != telegram_api::updateGroupInvitePrivacyForbidden::ID) { continue; } auto update = telegram_api::move_object_as(update_ptr); UserId user_id(update->user_id_); if (!user_id.is_valid()) { LOG(ERROR) << "Receive " << to_string(update); continue; } user_ids.push_back(user_id); } if (!user_ids.empty()) { td::remove_if(*updates, [](auto &update) { return update == nullptr; }); } return user_ids; } FlatHashSet UpdatesManager::get_sent_messages_random_ids(const telegram_api::Updates *updates_ptr) { FlatHashSet random_ids; auto updates = get_updates(updates_ptr); if (updates != nullptr) { auto new_messages = get_new_messages(updates_ptr); for (auto &update : *updates) { if (update->get_id() == telegram_api::updateMessageID::ID) { auto update_message_id = static_cast(update.get()); int64 random_id = update_message_id->random_id_; if (random_id != 0) { bool found_message = false; for (auto message : new_messages) { MessageId message_id = MessageId::get_message_id(message.first, message.second); if (message.second) { found_message |= message_id.is_valid_scheduled() && message_id.get_scheduled_server_message_id().get() == update_message_id->id_; } else { found_message |= message_id.is_valid() && message_id.get_server_message_id().get() == update_message_id->id_; } } if (found_message && !random_ids.insert(random_id).second) { LOG(ERROR) << "Receive twice updateMessageID for " << random_id; } } } } } return random_ids; } const telegram_api::Message *UpdatesManager::get_message_by_random_id(const telegram_api::Updates *updates_ptr, DialogId dialog_id, int64 random_id) { auto updates = get_updates(updates_ptr); if (updates == nullptr) { return nullptr; } int32 message_id = 0; for (auto &update : *updates) { if (update->get_id() == telegram_api::updateMessageID::ID) { auto update_message_id = static_cast(update.get()); if (update_message_id->random_id_ == random_id) { if (message_id != 0 || update_message_id->id_ == 0) { return nullptr; } message_id = update_message_id->id_; } } } if (message_id == 0) { return nullptr; } const telegram_api::Message *result = nullptr; FullMessageId full_message_id(dialog_id, MessageId(ServerMessageId(message_id))); for (auto &update : *updates) { auto constructor_id = update->get_id(); const tl_object_ptr *message = nullptr; if (constructor_id == telegram_api::updateNewMessage::ID) { message = &static_cast(update.get())->message_; } else if (constructor_id == telegram_api::updateNewChannelMessage::ID) { message = &static_cast(update.get())->message_; } if (message != nullptr && FullMessageId::get_full_message_id(*message, false) == full_message_id) { if (result != nullptr) { return nullptr; } result = message->get(); } } return result; } bool UpdatesManager::is_additional_service_message(const telegram_api::Message *message) { if (message->get_id() != telegram_api::messageService::ID) { return false; } auto action_id = static_cast(message)->action_->get_id(); return action_id == telegram_api::messageActionSetMessagesTTL::ID; } vector> UpdatesManager::get_new_messages( const telegram_api::Updates *updates_ptr) { vector> messages; auto updates = get_updates(updates_ptr); if (updates != nullptr) { for (auto &update : *updates) { const telegram_api::Message *message = nullptr; bool is_scheduled = false; auto constructor_id = update->get_id(); if (constructor_id == telegram_api::updateNewMessage::ID) { message = static_cast(update.get())->message_.get(); } else if (constructor_id == telegram_api::updateNewChannelMessage::ID) { message = static_cast(update.get())->message_.get(); } else if (constructor_id == telegram_api::updateNewScheduledMessage::ID) { message = static_cast(update.get())->message_.get(); is_scheduled = true; } if (message != nullptr && !is_additional_service_message(message)) { messages.emplace_back(message, is_scheduled); } } } return messages; } vector UpdatesManager::get_update_new_group_call_ids(const telegram_api::Updates *updates_ptr) { vector input_group_call_ids; auto updates = get_updates(updates_ptr); if (updates != nullptr) { for (auto &update : *updates) { InputGroupCallId input_group_call_id; if (update->get_id() == telegram_api::updateGroupCall::ID) { auto group_call_ptr = static_cast(update.get())->call_.get(); if (group_call_ptr->get_id() == telegram_api::groupCall::ID) { auto group_call = static_cast(group_call_ptr); input_group_call_id = InputGroupCallId(group_call->id_, group_call->access_hash_); } } if (input_group_call_id.is_valid()) { input_group_call_ids.push_back(input_group_call_id); } } } return input_group_call_ids; } string UpdatesManager::extract_join_group_call_presentation_params(telegram_api::Updates *updates_ptr) { auto updates = get_updates(updates_ptr); for (auto it = updates->begin(); it != updates->end(); ++it) { auto *update = it->get(); if (update->get_id() == telegram_api::updateGroupCallConnection::ID && static_cast(update)->presentation_) { string result = std::move(static_cast(update)->params_->data_); updates->erase(it); return result; } } return string(); } vector UpdatesManager::get_update_notify_settings_dialog_ids(const telegram_api::Updates *updates_ptr) { vector dialog_ids; auto updates = get_updates(updates_ptr); if (updates != nullptr) { dialog_ids.reserve(updates->size()); for (auto &update : *updates) { DialogId dialog_id; if (update->get_id() == telegram_api::updateNotifySettings::ID) { auto notify_peer = static_cast(update.get())->peer_.get(); if (notify_peer->get_id() == telegram_api::notifyPeer::ID) { dialog_id = DialogId(static_cast(notify_peer)->peer_); } } if (dialog_id.is_valid()) { dialog_ids.push_back(dialog_id); } else { LOG(ERROR) << "Receive unexpected " << to_string(update); } } } return dialog_ids; } vector UpdatesManager::get_update_ids(const telegram_api::Updates *updates_ptr) { const vector> *updates = nullptr; auto updates_type = updates_ptr->get_id(); switch (updates_type) { case telegram_api::updatesTooLong::ID: case telegram_api::updateShortMessage::ID: case telegram_api::updateShortChatMessage::ID: case telegram_api::updateShortSentMessage::ID: return {updates_type}; case telegram_api::updateShort::ID: { const auto *update = static_cast(updates_ptr)->update_.get(); if (update != nullptr) { return {update->get_id()}; } return {updates_type}; } case telegram_api::updatesCombined::ID: updates = &static_cast(updates_ptr)->updates_; break; case telegram_api::updates::ID: updates = &static_cast(updates_ptr)->updates_; break; default: UNREACHABLE(); } vector result; result.reserve(updates->size()); for (auto &update : *updates) { if (update != nullptr) { result.push_back(update->get_id()); } } return result; } vector UpdatesManager::get_chat_dialog_ids(const telegram_api::Updates *updates_ptr) { const vector> *chats = nullptr; switch (updates_ptr->get_id()) { case telegram_api::updatesTooLong::ID: case telegram_api::updateShortMessage::ID: case telegram_api::updateShortChatMessage::ID: case telegram_api::updateShort::ID: case telegram_api::updateShortSentMessage::ID: LOG(ERROR) << "Receive " << oneline(to_string(*updates_ptr)) << " instead of updates"; break; case telegram_api::updatesCombined::ID: chats = &static_cast(updates_ptr)->chats_; break; case telegram_api::updates::ID: chats = &static_cast(updates_ptr)->chats_; break; default: UNREACHABLE(); } if (chats == nullptr) { return {}; } vector dialog_ids; dialog_ids.reserve(chats->size()); for (const auto &chat : *chats) { auto dialog_id = ContactsManager::get_dialog_id(chat); if (dialog_id.is_valid()) { dialog_ids.push_back(dialog_id); } else { LOG(ERROR) << "Can't find identifier of " << oneline(to_string(chat)); } } return dialog_ids; } int32 UpdatesManager::get_update_edit_message_pts(const telegram_api::Updates *updates_ptr, FullMessageId full_message_id) { int32 pts = 0; auto updates = get_updates(updates_ptr); if (updates != nullptr) { for (auto &update_ptr : *updates) { int32 update_pts = [&] { switch (update_ptr->get_id()) { case telegram_api::updateEditMessage::ID: { auto update = static_cast(update_ptr.get()); if (FullMessageId::get_full_message_id(update->message_, false) == full_message_id) { return update->pts_; } return 0; } case telegram_api::updateEditChannelMessage::ID: { auto update = static_cast(update_ptr.get()); if (FullMessageId::get_full_message_id(update->message_, false) == full_message_id) { return update->pts_; } return 0; } case telegram_api::updateNewScheduledMessage::ID: { auto update = static_cast(update_ptr.get()); auto new_full_message_id = FullMessageId::get_full_message_id(update->message_, true); if (new_full_message_id.get_dialog_id() == full_message_id.get_dialog_id()) { auto new_message_id = new_full_message_id.get_message_id(); auto old_message_id = full_message_id.get_message_id(); if (new_message_id.is_valid_scheduled() && new_message_id.is_scheduled_server() && old_message_id.is_valid_scheduled() && old_message_id.is_scheduled_server() && old_message_id.get_scheduled_server_message_id() == new_message_id.get_scheduled_server_message_id()) { return -2; } } return 0; } default: return 0; } }(); if (update_pts != 0) { if (pts == 0) { pts = update_pts; } else { pts = -1; } } } } if (pts == -1) { LOG(ERROR) << "Receive multiple edit message updates in " << to_string(*updates_ptr); pts = 0; } else if (pts == 0) { LOG(ERROR) << "Receive no edit message updates for " << full_message_id << " in " << to_string(*updates_ptr); } return pts; } void UpdatesManager::init_state() { if (G()->close_flag() || !td_->auth_manager_->is_authorized()) { return; } bool drop_state = get_pts() == -1 && td_->can_ignore_background_updates() && td_->auth_manager_->is_bot() && td_->option_manager_->get_option_integer("since_last_open") >= 2 * 86400; auto pmc = G()->td_db()->get_binlog_pmc(); if (td_->ignore_background_updates() || drop_state) { // just in case pmc->erase("updates.pts"); pmc->erase("updates.qts"); pmc->erase("updates.date"); } string pts_str = pmc->get("updates.pts"); if (pts_str.empty()) { if (!running_get_difference_) { running_get_difference_ = true; before_get_difference(true); auto promise = PromiseCreator::lambda([](Result> result) { if (result.is_ok()) { send_closure(G()->updates_manager(), &UpdatesManager::on_get_updates_state, result.move_as_ok(), "GetUpdatesStateQuery"); } else { send_closure(G()->updates_manager(), &UpdatesManager::on_failed_get_updates_state, result.move_as_error()); } }); td_->create_handler(std::move(promise))->send(); } return; } pts_manager_.init(to_integer(pts_str)); last_confirmed_pts_ = get_pts(); qts_manager_.init(to_integer(pmc->get("updates.qts"))); last_confirmed_qts_ = get_qts(); date_ = to_integer(pmc->get("updates.date")); date_source_ = "database"; LOG(DEBUG) << "Init: " << get_pts() << " " << get_qts() << " " << date_; get_difference("init_state"); } uint64 UpdatesManager::get_most_unused_auth_key_id() { uint64 min_auth_key_id = 0; double min_time = Time::now(); for (auto &it : session_infos_) { if (it.second.last_update_time < min_time) { min_time = it.second.last_update_time; min_auth_key_id = it.first; } } return min_auth_key_id; } void UpdatesManager::on_update_from_auth_key_id(uint64 auth_key_id) { if (auth_key_id == 0) { return; } auto &info = session_infos_[auth_key_id]; auto now = Time::now(); info.last_update_time = now; if (info.update_count++ == 0) { info.first_update_time = now; while (session_infos_.size() > static_cast(max(narrow_cast(G()->get_option_integer("session_count")), 1))) { auto unused_auth_key_id = get_most_unused_auth_key_id(); LOG(INFO) << "Delete statistics for auth key " << unused_auth_key_id; session_infos_.erase(unused_auth_key_id); } } } void UpdatesManager::ping_server() { if (is_ping_sent_) { return; } is_ping_sent_ = true; auto promise = PromiseCreator::lambda([](Result> result) { auto state = result.is_ok() ? result.move_as_ok() : nullptr; send_closure(G()->updates_manager(), &UpdatesManager::on_server_pong, std::move(state)); }); td_->create_handler(std::move(promise))->send(); } void UpdatesManager::on_server_pong(tl_object_ptr &&state) { LOG(INFO) << "Receive " << oneline(to_string(state)); is_ping_sent_ = false; if (state == nullptr || state->pts_ > get_pts() || state->seq_ > seq_) { get_difference("on server pong"); } } void UpdatesManager::init_sessions(bool is_first) { if (G()->close_flag()) { return; } if (are_sessions_inited_ == is_first || !td_->auth_manager_->is_authorized()) { return; } are_sessions_inited_ = true; auto session_count = td_->option_manager_->get_option_integer("session_count", 1); if (session_count <= 1) { return; } LOG(INFO) << "Init " << session_count << " sessions"; for (int64 i = 0; i < session_count; i++) { td_->create_handler()->send(); } } void UpdatesManager::process_get_difference_updates( vector> &&new_messages, vector> &&new_encrypted_messages, vector> &&other_updates) { VLOG(get_difference) << "In get difference receive " << new_messages.size() << " messages, " << new_encrypted_messages.size() << " encrypted messages and " << other_updates.size() << " other updates"; for (auto &update : other_updates) { auto constructor_id = update->get_id(); if (constructor_id == telegram_api::updateMessageID::ID) { // in getDifference updateMessageID can't be received for scheduled messages LOG(INFO) << "Receive update about sent message " << to_string(update); auto update_message_id = move_tl_object_as(update); td_->messages_manager_->on_update_message_id(update_message_id->random_id_, MessageId(ServerMessageId(update_message_id->id_)), "getDifference"); CHECK(!running_get_difference_); } if (constructor_id == telegram_api::updateEncryption::ID) { on_update(move_tl_object_as(update), Promise()); CHECK(!running_get_difference_); } if (constructor_id == telegram_api::updateFolderPeers::ID) { auto update_folder_peers = move_tl_object_as(update); if (update_folder_peers->pts_count_ != 0) { LOG(ERROR) << "Receive updateFolderPeers with pts_count = " << update_folder_peers->pts_count_; update_folder_peers->pts_count_ = 0; } update_folder_peers->pts_ = 0; on_update(std::move(update_folder_peers), Promise()); CHECK(!running_get_difference_); } if (constructor_id == telegram_api::updateUser::ID) { update = nullptr; } if (constructor_id == telegram_api::updateChat::ID) { update = nullptr; } if (constructor_id == telegram_api::updateChannel::ID) { update = nullptr; } /* // TODO can't apply it here, because dialog may not be created yet // process updateReadHistoryInbox before new messages if (constructor_id == telegram_api::updateReadHistoryInbox::ID) { static_cast(update.get())->still_unread_count_ = -1; process_pts_update(std::move(update)); CHECK(!running_get_difference_); } */ } for (auto &message : new_messages) { // channel messages must not be received in this vector td_->messages_manager_->on_get_message(std::move(message), true, false, false, "get difference"); CHECK(!running_get_difference_); } for (auto &encrypted_message : new_encrypted_messages) { send_closure(td_->secret_chats_manager_, &SecretChatsManager::on_new_message, std::move(encrypted_message), Promise()); } process_updates(std::move(other_updates), true, Promise()); } void UpdatesManager::on_get_difference(tl_object_ptr &&difference_ptr) { if (G()->close_flag()) { return; } VLOG(get_difference) << "----- END GET DIFFERENCE-----"; running_get_difference_ = false; LOG(DEBUG) << "Result of get difference: " << to_string(difference_ptr); CHECK(difference_ptr != nullptr); switch (difference_ptr->get_id()) { case telegram_api::updates_differenceEmpty::ID: { auto difference = move_tl_object_as(difference_ptr); set_date(difference->date_, false, "on_get_difference_empty"); seq_ = difference->seq_; process_pending_qts_updates(); if (!pending_qts_updates_.empty()) { LOG(WARNING) << "Drop " << pending_qts_updates_.size() << " pending QTS updates after receive empty difference"; auto pending_qts_updates = std::move(pending_qts_updates_); pending_qts_updates_.clear(); for (auto &pending_update : pending_qts_updates) { set_promises(pending_update.second.promises); } } process_pending_seq_updates(); if (!pending_seq_updates_.empty()) { LOG(WARNING) << "Drop " << pending_seq_updates_.size() << " pending seq updates after receive empty difference"; auto pending_seq_updates = std::move(pending_seq_updates_); pending_seq_updates_.clear(); for (auto &pending_update : pending_seq_updates) { pending_update.second.promise.set_value(Unit()); } } break; } case telegram_api::updates_difference::ID: { auto difference = move_tl_object_as(difference_ptr); VLOG(get_difference) << "In get difference receive " << difference->users_.size() << " users and " << difference->chats_.size() << " chats"; td_->contacts_manager_->on_get_users(std::move(difference->users_), "updates.difference"); td_->contacts_manager_->on_get_chats(std::move(difference->chats_), "updates.difference"); if (get_difference_retry_count_ <= 5) { for (const auto &message : difference->new_messages_) { if (MessagesManager::is_invalid_poll_message(message.get())) { get_difference_retry_count_++; LOG(ERROR) << "Receive invalid poll message in updates.difference after " << get_difference_retry_count_ << " tries"; return run_get_difference(true, "reget difference"); } } } process_get_difference_updates(std::move(difference->new_messages_), std::move(difference->new_encrypted_messages_), std::move(difference->other_updates_)); if (running_get_difference_) { LOG(ERROR) << "Get difference has run while processing get difference updates"; break; } on_get_updates_state(std::move(difference->state_), "get difference"); break; } case telegram_api::updates_differenceSlice::ID: { auto difference = move_tl_object_as(difference_ptr); bool is_pts_changed = have_update_pts_changed(difference->other_updates_); if (difference->intermediate_state_->pts_ >= get_pts() && get_pts() != std::numeric_limits::max() && difference->intermediate_state_->date_ >= date_ && difference->intermediate_state_->qts_ == get_qts() && !is_pts_changed) { // TODO send new getDifference request and apply difference slice only after that } VLOG(get_difference) << "In get difference receive " << difference->users_.size() << " users and " << difference->chats_.size() << " chats"; td_->contacts_manager_->on_get_users(std::move(difference->users_), "updates.differenceSlice"); td_->contacts_manager_->on_get_chats(std::move(difference->chats_), "updates.differenceSlice"); if (get_difference_retry_count_ <= 5) { for (const auto &message : difference->new_messages_) { if (MessagesManager::is_invalid_poll_message(message.get())) { get_difference_retry_count_++; LOG(ERROR) << "Receive invalid poll message in updates.differenceSlice after " << get_difference_retry_count_ << " tries"; return run_get_difference(true, "reget difference"); } } } process_get_difference_updates(std::move(difference->new_messages_), std::move(difference->new_encrypted_messages_), std::move(difference->other_updates_)); if (running_get_difference_) { if (!is_pts_changed) { LOG(ERROR) << "Get difference has run while processing get difference updates"; } break; } CHECK(!is_pts_changed); auto old_pts = get_pts(); auto old_date = get_date(); auto old_qts = get_qts(); on_get_updates_state(std::move(difference->intermediate_state_), "get difference slice"); process_postponed_pts_updates(); process_pending_qts_updates(); auto new_pts = get_pts(); auto new_date = get_date(); auto new_qts = get_qts(); if (old_pts != std::numeric_limits::max() && new_date == old_date && (new_pts == old_pts || (min_postponed_update_pts_ != 0 && new_pts >= min_postponed_update_pts_)) && (new_qts == old_qts || (min_postponed_update_qts_ != 0 && new_qts >= min_postponed_update_qts_))) { VLOG(get_difference) << "Switch back from getDifference to update processing"; break; } if (new_pts != -1) { // just in case run_get_difference(true, "on updates_differenceSlice"); } break; } case telegram_api::updates_differenceTooLong::ID: { if (td_->option_manager_->get_option_integer("session_count") <= 1) { LOG(ERROR) << "Receive differenceTooLong"; } // TODO auto difference = move_tl_object_as(difference_ptr); set_pts(difference->pts_, "differenceTooLong").set_value(Unit()); get_difference("on updates_differenceTooLong"); break; } default: UNREACHABLE(); } get_difference_retry_count_ = 0; if (!running_get_difference_) { after_get_difference(); } } void UpdatesManager::on_get_pts_update(int32 pts, telegram_api::object_ptr difference_ptr) { if (G()->close_flag()) { return; } if (get_pts() != pts - 1 || running_get_difference_ || !postponed_pts_updates_.empty() || pending_pts_updates_.empty() || pending_pts_updates_.begin()->first != pts + 1) { return; } LOG(DEBUG) << "Receive update with PTS " << pts << ": " << to_string(difference_ptr); switch (difference_ptr->get_id()) { case telegram_api::updates_difference::ID: { auto difference = move_tl_object_as(difference_ptr); difference_ptr = telegram_api::make_object( std::move(difference->new_messages_), std::move(difference->new_encrypted_messages_), std::move(difference->other_updates_), std::move(difference->chats_), std::move(difference->users_), std::move(difference->state_)); } // fallthrough case telegram_api::updates_differenceSlice::ID: { auto difference = move_tl_object_as(difference_ptr); if (have_update_pts_changed(difference->other_updates_)) { return; } td_->contacts_manager_->on_get_users(std::move(difference->users_), "on_get_pts_update"); td_->contacts_manager_->on_get_chats(std::move(difference->chats_), "on_get_pts_update"); for (auto &message : difference->new_messages_) { difference->other_updates_.push_back( telegram_api::make_object(std::move(message), pts, 1)); } telegram_api::object_ptr *update_ptr = nullptr; size_t update_count = 0; for (auto &update : difference->other_updates_) { auto constructor_id = update->get_id(); if (constructor_id == telegram_api::updateMessageID::ID) { // in getDifference updateMessageID can't be received for scheduled messages LOG(INFO) << "Receive update about sent message " << to_string(update); auto update_message_id = move_tl_object_as(update); td_->messages_manager_->on_update_message_id( update_message_id->random_id_, MessageId(ServerMessageId(update_message_id->id_)), "on_get_pts_update"); continue; } update_ptr = &update; update_count++; } if (!difference->new_encrypted_messages_.empty() || update_count != 1) { LOG(ERROR) << "Receive unexpected updates with PTS " << pts << ": " << to_string(difference_ptr); break; } CHECK(update_ptr != nullptr); VLOG(get_difference) << "Repair update with PTS " << pts; pts_fixed_short_gap_++; add_pending_pts_update(std::move(*update_ptr), pts, 1, Time::now(), Promise(), "on_get_pts_update"); break; } case telegram_api::updates_differenceEmpty::ID: case telegram_api::updates_differenceTooLong::ID: { LOG(ERROR) << "Receive " << to_string(difference_ptr); break; default: UNREACHABLE(); } } } void UpdatesManager::confirm_pts_qts(int32 qts) { int32 pts = get_pts(); if (pts < 0) { pts = 0; } td_->create_handler()->send(pts, qts); last_confirmed_pts_ = pts; last_confirmed_qts_ = qts; } void UpdatesManager::after_get_difference() { CHECK(!running_get_difference_); retry_timeout_.cancel_timeout(); retry_time_ = 1; finished_first_get_difference_ = true; td_->option_manager_->set_option_empty("since_last_open"); // cancels qts_gap_timeout_ if needed, can apply some updates received during getDifference, // but missed in getDifference process_pending_qts_updates(); // cancels seq_gap_timeout_ if needed, can apply some updates received during getDifference, // but missed in getDifference process_pending_seq_updates(); if (running_get_difference_) { return; } if (!postponed_updates_.empty()) { auto begin_time = Time::now(); auto chunk_count = postponed_updates_.size(); VLOG(get_difference) << "Begin to apply " << chunk_count << " postponed update chunks"; size_t total_update_count = 0; while (!postponed_updates_.empty()) { auto it = postponed_updates_.begin(); auto updates = std::move(it->second.updates); auto updates_seq_begin = it->second.seq_begin; auto updates_seq_end = it->second.seq_end; auto receive_time = it->second.receive_time; auto promise = std::move(it->second.promise); // ignore it->second.date, because it may be too old postponed_updates_.erase(it); auto update_count = updates.size(); on_pending_updates(std::move(updates), updates_seq_begin, updates_seq_end, 0, receive_time, std::move(promise), "postponed updates"); if (running_get_difference_) { VLOG(get_difference) << "Finish to apply postponed updates with " << postponed_updates_.size() << " updates left after applied " << total_update_count << " updates, because forced to run getDifference"; return; } total_update_count += update_count; } VLOG(get_difference) << "Finished to apply " << total_update_count << " postponed updates"; auto passed_time = Time::now() - begin_time; if (passed_time >= UPDATE_APPLY_WARNING_TIME) { LOG(WARNING) << "Applied " << total_update_count << " updates in " << chunk_count << " chunks in " << passed_time << " seconds after postponing them for " << (Time::now() - get_difference_start_time_) << " seconds"; } } if (!postponed_pts_updates_.empty()) { // must be before td_->messages_manager_->after_get_difference() auto postponed_updates = std::move(postponed_pts_updates_); postponed_pts_updates_.clear(); auto begin_time = Time::now(); auto update_count = postponed_updates.size(); VLOG(get_difference) << "Begin to apply " << postponed_updates.size() << " postponed PTS updates with PTS = " << get_pts(); for (auto &postponed_update : postponed_updates) { auto &update = postponed_update.second; add_pending_pts_update(std::move(update.update), update.pts, update.pts_count, update.receive_time, std::move(update.promise), AFTER_GET_DIFFERENCE_SOURCE); CHECK(!running_get_difference_); } VLOG(get_difference) << "After applying postponed PTS updates have PTS = " << get_pts() << ", max_pts = " << accumulated_pts_ << " and " << pending_pts_updates_.size() << " + " << postponed_pts_updates_.size() << " pending PTS updates"; auto passed_time = Time::now() - begin_time; if (passed_time >= UPDATE_APPLY_WARNING_TIME) { LOG(WARNING) << "Applied " << update_count << " PTS updates in " << passed_time << " seconds after postponing them for " << (Time::now() - get_difference_start_time_) << " seconds"; } } send_closure(td_->download_manager_actor_, &DownloadManager::after_get_difference); td_->inline_queries_manager_->after_get_difference(); td_->messages_manager_->after_get_difference(); send_closure_later(td_->notification_manager_actor_, &NotificationManager::after_get_difference); send_closure(G()->state_manager(), &StateManager::on_synchronized, true); get_difference_start_time_ = 0.0; init_sessions(true); try_reload_data(); } void UpdatesManager::schedule_data_reload() { if (data_reload_timeout_.has_timeout()) { return; } auto timeout = next_data_reload_time_ - Time::now(); LOG(INFO) << "Schedule data reload in " << timeout; data_reload_timeout_.set_callback(std::move(try_reload_data_static)); data_reload_timeout_.set_callback_data(static_cast(td_)); data_reload_timeout_.set_timeout_in(timeout); } void UpdatesManager::try_reload_data_static(void *td) { if (G()->close_flag()) { return; } CHECK(td != nullptr); static_cast(td)->updates_manager_->try_reload_data(); } void UpdatesManager::try_reload_data() { if (!td_->auth_manager_->is_authorized() || td_->auth_manager_->is_bot() || running_get_difference_ || !td_->is_online()) { return; } auto now = Time::now(); if (now < next_data_reload_time_) { schedule_data_reload(); return; } next_data_reload_time_ = now + 365 * 86400; auto promise = PromiseCreator::lambda([actor_id = actor_id(this)](Result>) { send_closure(actor_id, &UpdatesManager::on_data_reloaded); }); LOG(INFO) << "Reload data"; td_->animations_manager_->reload_saved_animations(true); td_->autosave_manager_->reload_autosave_settings(); td_->contacts_manager_->reload_created_public_dialogs(PublicDialogType::HasUsername, std::move(promise)); td_->contacts_manager_->reload_created_public_dialogs(PublicDialogType::IsLocationBased, Auto()); get_default_emoji_statuses(td_, Auto()); td_->notification_settings_manager_->reload_saved_ringtones(Auto()); td_->notification_settings_manager_->send_get_scope_notification_settings_query(NotificationSettingsScope::Private, Auto()); td_->notification_settings_manager_->send_get_scope_notification_settings_query(NotificationSettingsScope::Group, Auto()); td_->notification_settings_manager_->send_get_scope_notification_settings_query(NotificationSettingsScope::Channel, Auto()); td_->stickers_manager_->reload_reactions(); td_->stickers_manager_->reload_recent_reactions(); td_->stickers_manager_->reload_top_reactions(); for (int32 type = 0; type < MAX_STICKER_TYPE; type++) { auto sticker_type = static_cast(type); td_->stickers_manager_->get_installed_sticker_sets(sticker_type, Auto()); td_->stickers_manager_->get_featured_sticker_sets(sticker_type, 0, 1000, Auto()); } td_->stickers_manager_->get_recent_stickers(false, Auto()); td_->stickers_manager_->get_recent_stickers(true, Auto()); td_->stickers_manager_->get_favorite_stickers(Auto()); td_->stickers_manager_->reload_special_sticker_set_by_type(SpecialStickerSetType::animated_emoji()); td_->stickers_manager_->reload_special_sticker_set_by_type(SpecialStickerSetType::animated_emoji_click()); td_->stickers_manager_->reload_special_sticker_set_by_type(SpecialStickerSetType::premium_gifts()); td_->stickers_manager_->reload_special_sticker_set_by_type(SpecialStickerSetType::generic_animations()); td_->stickers_manager_->reload_special_sticker_set_by_type(SpecialStickerSetType::default_statuses()); td_->stickers_manager_->reload_special_sticker_set_by_type(SpecialStickerSetType::default_topic_icons()); td_->stickers_manager_->get_default_dialog_photo_custom_emoji_stickers(false, true, Auto()); td_->stickers_manager_->get_default_dialog_photo_custom_emoji_stickers(true, true, Auto()); td_->story_manager_->reload_active_stories(); td_->story_manager_->reload_all_read_stories(); schedule_data_reload(); } void UpdatesManager::on_data_reloaded() { if (G()->close_flag()) { return; } next_data_reload_time_ = Time::now() + Random::fast(3000, 4200); data_reload_timeout_.cancel_timeout(); schedule_data_reload(); } void UpdatesManager::subscribe_to_transcribed_audio_updates(int64 transcription_id, TranscribedAudioHandler on_update) { CHECK(transcription_id != 0); if (pending_audio_transcriptions_.count(transcription_id) != 0) { on_pending_audio_transcription_failed(transcription_id, Status::Error(500, "Receive duplicate speech recognition identifier")); } bool is_inserted = pending_audio_transcriptions_.emplace(transcription_id, std::move(on_update)).second; CHECK(is_inserted); pending_audio_transcription_timeout_.set_timeout_in(transcription_id, AUDIO_TRANSCRIPTION_TIMEOUT); } void UpdatesManager::on_pending_audio_transcription_failed(int64 transcription_id, Status &&error) { if (G()->close_flag()) { return; } auto it = pending_audio_transcriptions_.find(transcription_id); if (it == pending_audio_transcriptions_.end()) { return; } auto on_update = std::move(it->second); pending_audio_transcriptions_.erase(it); pending_audio_transcription_timeout_.cancel_timeout(transcription_id); on_update(std::move(error)); } void UpdatesManager::on_pending_updates(vector> &&updates, int32 seq_begin, int32 seq_end, int32 date, double receive_time, Promise &&promise, const char *source) { if (get_pts() == -1) { init_state(); } if (!td_->auth_manager_->is_authorized()) { LOG(INFO) << "Ignore updates received before authorization or after logout"; return promise.set_value(Unit()); } // for (auto &update : updates) { // LOG(WARNING) << "Receive update " << to_string(update.get()); // } if (seq_begin < 0 || seq_end < 0 || date < 0 || seq_end < seq_begin) { LOG(ERROR) << "Wrong updates parameters seq_begin = " << seq_begin << ", seq_end = " << seq_end << ", date = " << date << " from " << source; get_difference("on wrong updates in on_pending_updates"); return promise.set_value(Unit()); } for (auto &update : updates) { if (update != nullptr) { switch (update->get_id()) { case telegram_api::updateUserTyping::ID: case telegram_api::updateChatUserTyping::ID: case telegram_api::updateChannelUserTyping::ID: case telegram_api::updateEncryptedChatTyping::ID: case telegram_api::updateLoginToken::ID: case telegram_api::updateDcOptions::ID: case telegram_api::updateConfig::ID: case telegram_api::updateServiceNotification::ID: case telegram_api::updateLangPackTooLong::ID: case telegram_api::updateLangPack::ID: short_update_date_ = date == 0 ? G()->unix_time() : date; LOG(INFO) << "Process short " << oneline(to_string(update)); // don't need promise for short update downcast_call(*update, OnUpdate(this, update, Promise())); short_update_date_ = 0; update = nullptr; break; default: break; } } } bool need_postpone = running_get_difference_ /*|| string(source) != string("postponed updates")*/; bool can_postpone = can_postpone_updates(); if (!need_postpone) { for (auto &update : updates) { if (!is_acceptable_update(update.get())) { CHECK(update != nullptr); int32 id = update->get_id(); const tl_object_ptr *message_ptr = nullptr; int32 pts = 0; if (id == telegram_api::updateNewChannelMessage::ID) { auto update_new_channel_message = static_cast(update.get()); message_ptr = &update_new_channel_message->message_; pts = update_new_channel_message->pts_; } if (id == telegram_api::updateEditChannelMessage::ID) { auto update_edit_channel_message = static_cast(update.get()); message_ptr = &update_edit_channel_message->message_; pts = update_edit_channel_message->pts_; } // for channels we can try to replace unacceptable update with updateChannelTooLong if (message_ptr != nullptr) { auto dialog_id = DialogId::get_message_dialog_id(*message_ptr); if (dialog_id.get_type() == DialogType::Channel) { auto channel_id = dialog_id.get_channel_id(); if (td_->contacts_manager_->have_channel_force(channel_id)) { if (td_->messages_manager_->is_old_channel_update(dialog_id, pts)) { // the update will be ignored anyway, so there is no reason to replace it or force get_difference LOG(INFO) << "Allow an outdated unacceptable update from " << source; continue; } if ((*message_ptr)->get_id() != telegram_api::messageService::ID) { // don't replace service messages, because they can be about bot's kicking LOG(INFO) << "Replace update about new message with updateChannelTooLong in " << dialog_id; update = telegram_api::make_object( telegram_api::updateChannelTooLong::PTS_MASK, channel_id.get(), pts); continue; } } } else { LOG(ERROR) << "Update is not from a channel: " << to_string(update); } } get_difference("on unacceptable updates in on_pending_updates"); return promise.set_value(Unit()); } } } if (date > 0 && updates.size() == 1 && updates[0] != nullptr && updates[0]->get_id() == telegram_api::updateReadHistoryOutbox::ID) { auto update = static_cast(updates[0].get()); DialogId dialog_id(update->peer_); if (dialog_id.get_type() == DialogType::User) { auto user_id = dialog_id.get_user_id(); if (user_id.is_valid()) { td_->contacts_manager_->on_update_user_local_was_online(user_id, date); } } } size_t ordinary_new_message_count = 0; size_t scheduled_new_message_count = 0; size_t update_message_id_count = 0; size_t update_count = 0; for (auto &update : updates) { if (update != nullptr) { auto constructor_id = update->get_id(); if (constructor_id == telegram_api::updateMessageID::ID) { update_message_id_count++; } else { update_count++; if (constructor_id == telegram_api::updateNewMessage::ID || constructor_id == telegram_api::updateNewChannelMessage::ID) { ordinary_new_message_count++; } else if (constructor_id == telegram_api::updateNewScheduledMessage::ID) { scheduled_new_message_count++; } } } } if (update_message_id_count != 0 && ordinary_new_message_count != 0 && scheduled_new_message_count != 0) { LOG(ERROR) << "Receive mixed message types in updates:"; for (auto &update : updates) { LOG(ERROR) << "Update: " << oneline(to_string(update)); } if (!running_get_difference_) { schedule_get_difference("on_get_wrong_updates"); } return promise.set_value(Unit()); } MultiPromiseActorSafe mpas{"OnPendingUpdatesMultiPromiseActor"}; Promise lock; auto use_mpas = need_postpone || update_count != 1; auto get_promise = [&] { if (use_mpas) { return mpas.get_promise(); } else { CHECK(update_count != 0); update_count--; return std::move(promise); } }; if (use_mpas) { being_processed_updates_++; mpas.add_promise([actor_id = create_reference(), promise = std::move(promise)](Result &&result) mutable { send_closure(actor_id, &UpdatesManager::on_pending_updates_processed, std::move(result), std::move(promise)); }); lock = get_promise(); } SCOPE_EXIT { if (!use_mpas && update_count == 1) { promise.set_value(Unit()); } }; for (auto &update : updates) { if (update != nullptr) { LOG(INFO) << "Receive from " << source << " pending " << to_string(update); int32 id = update->get_id(); if (id == telegram_api::updateMessageID::ID) { LOG(INFO) << "Receive from " << source << " " << to_string(update); auto sent_message_update = move_tl_object_as(update); MessageId message_id; if (ordinary_new_message_count != 0) { message_id = MessageId(ServerMessageId(sent_message_update->id_)); } else if (scheduled_new_message_count != 0) { message_id = MessageId(ScheduledServerMessageId(sent_message_update->id_), std::numeric_limits::max()); } if (!td_->messages_manager_->on_update_message_id(sent_message_update->random_id_, message_id, source)) { for (auto &debug_update : updates) { LOG(ERROR) << "Update: " << oneline(to_string(debug_update)); } } update = nullptr; } if (id == telegram_api::updateFolderPeers::ID) { on_update(move_tl_object_as(update), get_promise()); update = nullptr; } if (id == telegram_api::updateEncryption::ID) { on_update(move_tl_object_as(update), get_promise()); update = nullptr; } // check that if getDifference wasn't run than it isn't run still CHECK(need_postpone || !running_get_difference_); } } for (auto &update : updates) { if (update != nullptr) { if (is_pts_update(update.get())) { if (running_get_difference_ && can_postpone) { auto pts = get_update_pts(update.get()); if (pts != 0 && (min_postponed_update_pts_ == 0 || pts < min_postponed_update_pts_)) { min_postponed_update_pts_ = pts; } } downcast_call(*update, OnUpdate(this, update, get_promise())); update = nullptr; } else if (is_qts_update(update.get())) { if (running_get_difference_ && can_postpone) { auto qts = get_update_qts(update.get()); if (qts != 0 && (min_postponed_update_qts_ == 0 || qts < min_postponed_update_qts_)) { min_postponed_update_qts_ = qts; } } downcast_call(*update, OnUpdate(this, update, get_promise())); update = nullptr; } else if (is_channel_pts_update(update.get())) { downcast_call(*update, OnUpdate(this, update, get_promise())); update = nullptr; } } } if (seq_begin == 0 && seq_end == 0) { bool have_updates = false; for (auto &update : updates) { if (update != nullptr) { have_updates = true; break; } } if (!have_updates) { LOG(INFO) << "All updates were processed"; return lock.set_value(Unit()); } } if (!use_mpas && update_count == 1) { // still need to process the only update lock = std::move(promise); // now we can use lock as the last promise update_count = 0; } if (need_postpone || running_get_difference_) { LOG(INFO) << "Postpone " << updates.size() << " updates [" << seq_begin << ", " << seq_end << "] with date = " << date << " from " << source; if (!need_postpone) { LOG(ERROR) << "Run get difference while applying updates from " << source; } if (can_postpone) { postponed_updates_.emplace( seq_begin, PendingSeqUpdates(seq_begin, seq_end, date, receive_time, std::move(updates), std::move(lock))); } else { lock.set_value(Unit()); } return; } if (seq_begin == 0 || seq_begin == seq_ + 1) { LOG(INFO) << "Process " << updates.size() << " updates [" << seq_begin << ", " << seq_end << "] with date = " << date << " from " << source; process_seq_updates(seq_end, date, std::move(updates), std::move(lock)); process_pending_seq_updates(); return; } if (seq_begin <= seq_) { if (seq_ >= (1 << 30) && seq_begin < seq_ - (1 << 30)) { set_seq_gap_timeout(0.001); } if (seq_end > seq_) { LOG(ERROR) << "Receive updates with seq_begin = " << seq_begin << ", seq_end = " << seq_end << ", but seq = " << seq_ << " from " << source; } else { LOG(INFO) << "Receive old updates with seq_begin = " << seq_begin << ", seq_end = " << seq_end << ", but seq = " << seq_ << " from " << source; } return lock.set_value(Unit()); } LOG(INFO) << "Gap in seq has found. Receive " << updates.size() << " updates [" << seq_begin << ", " << seq_end << "] from " << source << ", but seq = " << seq_; LOG_IF(WARNING, pending_seq_updates_.count(seq_begin) > 0) << "Already have pending updates with seq = " << seq_begin << ", but receive it again from " << source; if (can_postpone_updates()) { pending_seq_updates_.emplace( seq_begin, PendingSeqUpdates(seq_begin, seq_end, date, receive_time, std::move(updates), std::move(lock))); } else { lock.set_value(Unit()); } set_seq_gap_timeout(receive_time + MAX_UNFILLED_GAP_TIME - Time::now()); } void UpdatesManager::on_pending_updates_processed(Result result, Promise promise) { being_processed_updates_--; promise.set_result(std::move(result)); } void UpdatesManager::add_pending_qts_update(tl_object_ptr &&update, int32 qts, Promise &&promise) { CHECK(update != nullptr); if (qts <= 1) { LOG(ERROR) << "Receive wrong QTS " << qts << " in " << oneline(to_string(update)); schedule_get_difference("wrong QTS"); promise.set_value(Unit()); return; } int32 old_qts = get_qts(); LOG(INFO) << "Process update with QTS = " << qts << ", current QTS = " << old_qts; if (qts < old_qts - 1000009) { LOG(WARNING) << "Restore QTS after QTS overflow from " << old_qts << " to " << qts << " by " << oneline(to_string(update)); add_qts(qts - 1).set_value(Unit()); CHECK(get_qts() == qts - 1); old_qts = qts - 1; last_confirmed_qts_ = get_qts(); } if (qts <= old_qts) { LOG(INFO) << "Skip already applied update with QTS = " << qts; promise.set_value(Unit()); return; } if (running_get_difference_ || (qts - 1 > old_qts && old_qts > 0)) { LOG(INFO) << "Postpone update with QTS = " << qts; if (!running_get_difference_ && pending_qts_updates_.empty()) { set_qts_gap_timeout(MAX_UNFILLED_GAP_TIME); } if (!can_postpone_updates()) { return promise.set_value(Unit()); } auto &pending_update = pending_qts_updates_[qts]; if (pending_update.update != nullptr) { LOG(WARNING) << "Receive duplicate update with QTS = " << qts; } else { pending_update.receive_time = Time::now(); } pending_update.update = std::move(update); pending_update.promises.push_back(std::move(promise)); return; } process_qts_update(std::move(update), qts, std::move(promise)); process_pending_qts_updates(); } void UpdatesManager::process_updates(vector> &&updates, bool force_apply, Promise &&promise) { int32 update_count = 0; for (auto &update : updates) { if (update != nullptr) { update_count++; } } if (update_count == 0) { return promise.set_value(Unit()); } MultiPromiseActorSafe mpas{"OnProcessUpdatesMultiPromiseActor"}; Promise lock; auto use_mpas = update_count != 1; auto get_promise = [&] { if (use_mpas) { return mpas.get_promise(); } else { CHECK(update_count != 0); update_count--; return std::move(promise); } }; if (use_mpas) { mpas.add_promise(std::move(promise)); lock = get_promise(); } SCOPE_EXIT { if (!use_mpas && update_count == 1) { promise.set_value(Unit()); } }; /* for (auto &update : updates) { if (update != nullptr) { // TODO can't apply it here, because dialog may not be created yet // process updateReadChannelInbox before updateNewChannelMessage auto constructor_id = update->get_id(); if (constructor_id == telegram_api::updateReadChannelInbox::ID) { on_update(move_tl_object_as(update), get_promise()); } } } */ tl_object_ptr update_pts_changed; for (auto &update : updates) { if (update != nullptr) { // process updateNewChannelMessage first auto constructor_id = update->get_id(); if (constructor_id == telegram_api::updateNewChannelMessage::ID) { on_update(move_tl_object_as(update), get_promise()); continue; } // process updateNewScheduledMessage first if (constructor_id == telegram_api::updateNewScheduledMessage::ID) { on_update(move_tl_object_as(update), get_promise()); continue; } // updateGroupCallConnection must be processed before updateGroupCall if (constructor_id == telegram_api::updateGroupCallConnection::ID) { on_update(move_tl_object_as(update), get_promise()); continue; } // updatePtsChanged forces get difference, so process it last if (constructor_id == telegram_api::updatePtsChanged::ID) { update_pts_changed = move_tl_object_as(update); continue; } } } if (force_apply) { for (auto &update : updates) { if (update != nullptr) { if (is_pts_update(update.get())) { auto constructor_id = update->get_id(); if (constructor_id == telegram_api::updateWebPage::ID) { auto update_web_page = move_tl_object_as(update); td_->web_pages_manager_->on_get_web_page(std::move(update_web_page->webpage_), DialogId()); continue; } CHECK(constructor_id != telegram_api::updateFolderPeers::ID); if (constructor_id == telegram_api::updateReadHistoryInbox::ID) { static_cast(update.get())->still_unread_count_ = -1; } process_pts_update(std::move(update)); CHECK(update == nullptr); } else if (is_qts_update(update.get())) { process_qts_update(std::move(update), 0, get_promise()); CHECK(update == nullptr); } else if (update->get_id() == telegram_api::updateChannelTooLong::ID) { td_->messages_manager_->on_update_channel_too_long( move_tl_object_as(update), true); } } } } for (auto &update : updates) { if (update != nullptr) { LOG(INFO) << "Process update " << to_string(update); downcast_call(*update, OnUpdate(this, update, get_promise())); CHECK(!running_get_difference_); } } if (update_pts_changed != nullptr) { on_update(std::move(update_pts_changed), get_promise()); } lock.set_value(Unit()); } void UpdatesManager::process_pts_update(tl_object_ptr &&update) { CHECK(update != nullptr); // TODO need to save all updates that can change result of running queries not associated with PTS (for example // getHistory) and apply the updates to results of the queries if (!check_pts_update(update)) { LOG(ERROR) << "Receive wrong PTS update: " << oneline(to_string(update)); update = nullptr; return; } // must be called only during getDifference CHECK(pending_pts_updates_.empty()); CHECK(accumulated_pts_ == -1); td_->messages_manager_->process_pts_update(std::move(update)); } void UpdatesManager::add_pending_pts_update(tl_object_ptr &&update, int32 new_pts, int32 pts_count, double receive_time, Promise &&promise, const char *source) { // do not try to run getDifference from this function CHECK(update != nullptr); CHECK(source != nullptr); LOG(INFO) << "Receive from " << source << " pending " << to_string(update); if (pts_count < 0 || new_pts <= pts_count) { LOG(ERROR) << "Receive update with wrong PTS = " << new_pts << " or pts_count = " << pts_count << " from " << source << ": " << oneline(to_string(update)); return promise.set_value(Unit()); } // TODO need to save all updates that can change result of running queries not associated with PTS (for example // getHistory) and apply them to result of this queries if (!check_pts_update(update)) { LOG(ERROR) << "Receive wrong PTS update from " << source << ": " << oneline(to_string(update)); return promise.set_value(Unit()); } if (DROP_PTS_UPDATES && Slice(source) != Slice("on_get_pts_update")) { set_pts_gap_timeout(1.0); return promise.set_value(Unit()); } int32 old_pts = get_pts(); if (new_pts < old_pts - 99 && source != AFTER_GET_DIFFERENCE_SOURCE) { bool need_restore_pts = new_pts < old_pts - 19999; auto now = Time::now(); if (old_pts == 2100000000 && new_pts < 1100000000 && pts_count <= 10000 && td_->option_manager_->get_option_integer("session_count") > 1) { set_pts(1, "restore PTS").set_value(Unit()); old_pts = get_pts(); set_pts_gap_timeout(0.001); return promise.set_value(Unit()); } else if (now > last_pts_jump_warning_time_ + 1 && (need_restore_pts || now < last_pts_jump_warning_time_ + 5) && !(old_pts == std::numeric_limits::max() && running_get_difference_)) { LOG(ERROR) << "Restore PTS after delete_first_messages from " << old_pts << " to " << new_pts << " is disabled, pts_count = " << pts_count << ", update is from " << source << ": " << oneline(to_string(update)); last_pts_jump_warning_time_ = now; } if (need_restore_pts) { set_pts_gap_timeout(0.001); } } if (new_pts <= old_pts || (old_pts >= 1 && new_pts - (1 << 30) > old_pts)) { td_->messages_manager_->skip_old_pending_pts_update(std::move(update), new_pts, old_pts, pts_count, source); return promise.set_value(Unit()); } if (source == AFTER_GET_DIFFERENCE_SOURCE && old_pts >= 1 && new_pts - (1 << 21) > old_pts) { LOG(ERROR) << "Ignore update " << oneline(to_string(update)) << ", because current PTS is " << old_pts; return promise.set_value(Unit()); } if (running_get_difference_ || !postponed_pts_updates_.empty()) { LOG(INFO) << "Save pending update received while running getDifference from " << source; postpone_pts_update(std::move(update), new_pts, pts_count, receive_time, std::move(promise)); return; } // is_acceptable_update check was skipped for postponed PTS updates if (source == AFTER_GET_DIFFERENCE_SOURCE && !is_acceptable_update(update.get())) { LOG(INFO) << "Postpone again unacceptable pending update"; postpone_pts_update(std::move(update), new_pts, pts_count, receive_time, std::move(promise)); set_pts_gap_timeout(0.001); return; } if (old_pts > new_pts - pts_count) { LOG(WARNING) << "Have old_pts (= " << old_pts << ") + pts_count (= " << pts_count << ") > new_pts (= " << new_pts << "). Logged in " << td_->option_manager_->get_option_integer("authorization_date") << ". Update from " << source << " = " << oneline(to_string(update)); postpone_pts_update(std::move(update), new_pts, pts_count, receive_time, std::move(promise)); set_pts_gap_timeout(0.001); return; } accumulated_pts_count_ += pts_count; if (new_pts > accumulated_pts_) { accumulated_pts_ = new_pts; } if (old_pts > accumulated_pts_ - accumulated_pts_count_) { LOG(WARNING) << "Have old_pts (= " << old_pts << ") + accumulated_pts_count (= " << accumulated_pts_count_ << ") > accumulated_pts (= " << accumulated_pts_ << "). new_pts = " << new_pts << ", pts_count = " << pts_count << ". Logged in " << td_->option_manager_->get_option_integer("authorization_date") << ". Update from " << source << " = " << oneline(to_string(update)); postpone_pts_update(std::move(update), new_pts, pts_count, receive_time, std::move(promise)); set_pts_gap_timeout(0.001); return; } LOG_IF(INFO, pts_count == 0 && update->get_id() != dummyUpdate::ID) << "Skip useless update " << to_string(update); if (pending_pts_updates_.empty() && old_pts == accumulated_pts_ - accumulated_pts_count_ && !pts_gap_timeout_.has_timeout()) { if (pts_count > 0) { td_->messages_manager_->process_pts_update(std::move(update)); set_pts(accumulated_pts_, "process pending updates fast path") .set_value(Unit()); // TODO can't set until data are really stored on persistent storage accumulated_pts_count_ = 0; accumulated_pts_ = -1; } promise.set_value(Unit()); return; } pending_pts_updates_.emplace( new_pts, PendingPtsUpdate(std::move(update), new_pts, pts_count, receive_time, std::move(promise))); if (old_pts < accumulated_pts_ - accumulated_pts_count_) { if (old_pts == new_pts - pts_count) { // can't apply all updates, but can apply this and probably some other updates process_pending_pts_updates(); } else { set_pts_gap_timeout(receive_time + MAX_UNFILLED_GAP_TIME - Time::now()); } return; } CHECK(old_pts == accumulated_pts_ - accumulated_pts_count_); process_all_pending_pts_updates(); } void UpdatesManager::postpone_pts_update(tl_object_ptr &&update, int32 pts, int32 pts_count, double receive_time, Promise &&promise) { if (!can_postpone_updates()) { return promise.set_value(Unit()); } postponed_pts_updates_.emplace(pts, PendingPtsUpdate(std::move(update), pts, pts_count, receive_time, std::move(promise))); } void UpdatesManager::process_seq_updates(int32 seq_end, int32 date, vector> &&updates, Promise &&promise) { string serialized_updates; if (date && seq_end) { serialized_updates = PSTRING() << "process_seq_updates [seq_ = " << seq_ << ", seq_end = " << seq_end << "]: "; // TODO remove after bugs will be fixed for (auto &update : updates) { if (update != nullptr) { serialized_updates += oneline(to_string(update)); } } } process_updates(std::move(updates), false, std::move(promise)); if (seq_end) { seq_ = seq_end; } if (date && seq_end) { set_date(date, true, std::move(serialized_updates)); } } void UpdatesManager::process_qts_update(tl_object_ptr &&update_ptr, int32 qts, Promise &&promise) { LOG(DEBUG) << "Process " << to_string(update_ptr); if (last_confirmed_qts_ < qts - FORCED_GET_DIFFERENCE_PTS_DIFF && last_confirmed_qts_ != 0) { confirm_pts_qts(qts); } qts_diff_++; if (qts_diff_ >= 1000000) { LOG(WARNING) << "Fixed " << qts_gap_ << " QTS gaps"; qts_gap_ = 0; qts_diff_ = 0; } switch (update_ptr->get_id()) { case telegram_api::updateNewEncryptedMessage::ID: { auto update = move_tl_object_as(update_ptr); send_closure(td_->secret_chats_manager_, &SecretChatsManager::on_new_message, std::move(update->message_), add_qts(qts)); break; } case telegram_api::updateMessagePollVote::ID: { auto update = move_tl_object_as(update_ptr); DialogId dialog_id(update->peer_); td_->poll_manager_->on_get_poll_vote(PollId(update->poll_id_), dialog_id, std::move(update->options_)); add_qts(qts).set_value(Unit()); break; } case telegram_api::updateBotStopped::ID: { auto update = move_tl_object_as(update_ptr); td_->contacts_manager_->on_update_bot_stopped(UserId(update->user_id_), update->date_, update->stopped_); add_qts(qts).set_value(Unit()); break; } case telegram_api::updateChatParticipant::ID: { auto update = move_tl_object_as(update_ptr); td_->contacts_manager_->on_update_chat_participant( ChatId(update->chat_id_), UserId(update->actor_id_), update->date_, DialogInviteLink(std::move(update->invite_), true, "updateChatParticipant"), std::move(update->prev_participant_), std::move(update->new_participant_)); add_qts(qts).set_value(Unit()); break; } case telegram_api::updateChannelParticipant::ID: { auto update = move_tl_object_as(update_ptr); td_->contacts_manager_->on_update_channel_participant( ChannelId(update->channel_id_), UserId(update->actor_id_), update->date_, DialogInviteLink(std::move(update->invite_), true, "updateChannelParticipant"), update->via_chatlist_, std::move(update->prev_participant_), std::move(update->new_participant_)); add_qts(qts).set_value(Unit()); break; } case telegram_api::updateBotChatInviteRequester::ID: { auto update = move_tl_object_as(update_ptr); td_->contacts_manager_->on_update_chat_invite_requester( DialogId(update->peer_), UserId(update->user_id_), std::move(update->about_), update->date_, DialogInviteLink(std::move(update->invite_), true, "updateBotChatInviteRequester")); add_qts(qts).set_value(Unit()); break; } default: UNREACHABLE(); break; } promise.set_value(Unit()); } void UpdatesManager::process_all_pending_pts_updates() { auto begin_time = Time::now(); for (auto &update : pending_pts_updates_) { td_->messages_manager_->process_pts_update(std::move(update.second.update)); update.second.promise.set_value(Unit()); } if (last_pts_gap_time_ != 0) { auto begin_diff = begin_time - last_pts_gap_time_; auto diff = Time::now() - last_pts_gap_time_; last_pts_gap_time_ = 0; if (diff > 0.1) { VLOG(get_difference) << "Gap in PTS from " << accumulated_pts_ - accumulated_pts_count_ << " to " << accumulated_pts_ << " has been filled in " << begin_diff << '-' << diff << " seconds"; } } set_pts(accumulated_pts_, "process_all_pending_pts_updates") .set_value(Unit()); // TODO can't set until updates are stored on persistent storage drop_all_pending_pts_updates(); } void UpdatesManager::drop_all_pending_pts_updates() { accumulated_pts_count_ = 0; accumulated_pts_ = -1; min_pts_gap_timeout_.cancel_timeout(); pts_gap_timeout_.cancel_timeout(); pending_pts_updates_.clear(); } void UpdatesManager::process_postponed_pts_updates() { if (postponed_pts_updates_.empty()) { return; } auto begin_time = Time::now(); auto initial_pts = get_pts(); auto old_pts = initial_pts; int32 skipped_update_count = 0; int32 applied_update_count = 0; auto update_it = postponed_pts_updates_.begin(); while (update_it != postponed_pts_updates_.end()) { auto new_pts = update_it->second.pts; auto pts_count = update_it->second.pts_count; if (new_pts <= old_pts || (old_pts >= 1 && new_pts - (1 << 30) > old_pts)) { skipped_update_count++; td_->messages_manager_->skip_old_pending_pts_update(std::move(update_it->second.update), new_pts, old_pts, pts_count, "process_postponed_pts_updates"); update_it->second.promise.set_value(Unit()); update_it = postponed_pts_updates_.erase(update_it); continue; } if (Time::now() - begin_time >= td::min(UPDATE_APPLY_WARNING_TIME / 2, 0.1)) { // the updates will be applied or skipped later; reget the remaining updates through getDifference break; } auto last_update_it = update_it; for (int32 i = 1; true; i++) { ++last_update_it; if (old_pts == new_pts - pts_count) { // the updates can be applied break; } if (old_pts > new_pts - pts_count || last_update_it == postponed_pts_updates_.end() || i == GAP_TIMEOUT_UPDATE_COUNT) { // the updates can't be applied VLOG(get_difference) << "Can't apply " << i << " next postponed updates with PTS " << update_it->second.pts << '-' << new_pts << ", because their pts_count is " << pts_count << " instead of expected " << new_pts - old_pts; last_update_it = update_it; break; } new_pts = last_update_it->second.pts; pts_count += last_update_it->second.pts_count; } if (last_update_it == update_it) { // the updates will be applied or skipped later break; } CHECK(old_pts == new_pts - pts_count); while (update_it != last_update_it) { if (update_it->second.pts_count > 0) { applied_update_count++; td_->messages_manager_->process_pts_update(std::move(update_it->second.update)); } update_it->second.promise.set_value(Unit()); update_it = postponed_pts_updates_.erase(update_it); } old_pts = new_pts; } if (old_pts != initial_pts) { set_pts(old_pts, "process_postponed_pts_updates") .set_value(Unit()); // TODO can't set until data are really stored on persistent storage } CHECK(!running_get_difference_); if (skipped_update_count + applied_update_count > 0) { VLOG(get_difference) << "PTS has changed from " << initial_pts << " to " << old_pts << " after skipping " << skipped_update_count << ", applying " << applied_update_count << " and keeping " << postponed_pts_updates_.size() << " postponed updates"; } auto passed_time = Time::now() - begin_time; if (passed_time >= UPDATE_APPLY_WARNING_TIME) { LOG(WARNING) << "PTS has changed from " << initial_pts << " to " << old_pts << " after skipping " << skipped_update_count << ", applying " << applied_update_count << " and keeping " << postponed_pts_updates_.size() << " postponed for " << (Time::now() - get_difference_start_time_) << " seconds updates in " << passed_time << " seconds"; } } void UpdatesManager::process_pending_pts_updates() { if (pending_pts_updates_.empty()) { return; } auto begin_time = Time::now(); auto initial_pts = get_pts(); int32 applied_update_count = 0; while (!pending_pts_updates_.empty()) { auto update_it = pending_pts_updates_.begin(); auto &update = update_it->second; if (get_pts() != update.pts - update.pts_count) { // the updates will be applied or skipped later break; } applied_update_count++; if (update.pts_count > 0) { td_->messages_manager_->process_pts_update(std::move(update.update)); set_pts(update.pts, "process_pending_pts_updates") .set_value(Unit()); // TODO can't set until data are really stored on persistent storage if (accumulated_pts_ != -1) { CHECK(update.pts <= accumulated_pts_); CHECK(accumulated_pts_count_ >= update.pts_count); accumulated_pts_count_ -= update.pts_count; } } update.promise.set_value(Unit()); pending_pts_updates_.erase(update_it); } if (applied_update_count > 0) { min_pts_gap_timeout_.cancel_timeout(); pts_gap_timeout_.cancel_timeout(); } if (!pending_pts_updates_.empty()) { // if still have a gap, reset timeout auto update_it = pending_pts_updates_.begin(); double receive_time = update_it->second.receive_time; for (size_t i = 0; i < GAP_TIMEOUT_UPDATE_COUNT; i++) { if (++update_it == pending_pts_updates_.end()) { break; } receive_time = min(receive_time, update_it->second.receive_time); } set_pts_gap_timeout(receive_time + MAX_UNFILLED_GAP_TIME - Time::now()); } auto passed_time = Time::now() - begin_time; if (passed_time >= UPDATE_APPLY_WARNING_TIME) { LOG(WARNING) << "PTS has changed from " << initial_pts << " to " << get_pts() << " after applying " << applied_update_count << " and keeping " << pending_pts_updates_.size() << " pending updates in " << passed_time << " seconds"; } } void UpdatesManager::process_pending_seq_updates() { if (!pending_seq_updates_.empty()) { LOG(DEBUG) << "Trying to process " << pending_seq_updates_.size() << " pending seq updates"; // must not return, because in case of seq overflow there are no pending seq updates } auto begin_time = Time::now(); int32 initial_seq = seq_; int32 applied_update_count = 0; while (!pending_seq_updates_.empty() && !running_get_difference_) { auto update_it = pending_seq_updates_.begin(); auto &update = update_it->second; auto seq_begin = update.seq_begin; if (seq_begin - 1 > seq_ && seq_begin - (1 << 30) <= seq_) { // the updates will be applied later break; } applied_update_count++; auto seq_end = update.seq_end; if (seq_begin - 1 == seq_) { process_seq_updates(seq_end, update.date, std::move(update.updates), std::move(update.promise)); } else { // old update CHECK(seq_begin != 0); if (seq_begin <= seq_ && seq_ < seq_end) { LOG(ERROR) << "Receive updates with seq_begin = " << seq_begin << ", seq_end = " << seq_end << ", but seq = " << seq_; } update.promise.set_value(Unit()); } pending_seq_updates_.erase(update_it); } if (pending_seq_updates_.empty() || applied_update_count > 0) { seq_gap_timeout_.cancel_timeout(); } if (!pending_seq_updates_.empty()) { // if still have a gap, reset timeout auto update_it = pending_seq_updates_.begin(); double receive_time = update_it->second.receive_time; for (size_t i = 0; i < GAP_TIMEOUT_UPDATE_COUNT; i++) { if (++update_it == pending_seq_updates_.end()) { break; } receive_time = min(receive_time, update_it->second.receive_time); } set_seq_gap_timeout(receive_time + MAX_UNFILLED_GAP_TIME - Time::now()); } auto passed_time = Time::now() - begin_time; if (passed_time >= UPDATE_APPLY_WARNING_TIME) { LOG(WARNING) << "Seq has changed from " << initial_seq << " to " << seq_ << " after applying " << applied_update_count << " and keeping " << pending_seq_updates_.size() << " pending updates in " << passed_time << " seconds"; } } void UpdatesManager::process_pending_qts_updates() { if (pending_qts_updates_.empty()) { return; } LOG(DEBUG) << "Process " << pending_qts_updates_.size() << " pending QTS updates"; auto begin_time = Time::now(); auto initial_qts = get_qts(); int32 applied_update_count = 0; while (!pending_qts_updates_.empty()) { CHECK(!running_get_difference_); auto update_it = pending_qts_updates_.begin(); auto qts = update_it->first; auto old_qts = get_qts(); if (qts - 1 > old_qts && qts - (1 << 30) <= old_qts) { // the update will be applied later break; } auto promise = PromiseCreator::lambda( [promises = std::move(update_it->second.promises)](Unit) mutable { set_promises(promises); }); applied_update_count++; if (qts == old_qts + 1) { process_qts_update(std::move(update_it->second.update), qts, std::move(promise)); } else { promise.set_value(Unit()); } pending_qts_updates_.erase(update_it); } if (applied_update_count > 0) { qts_gap_timeout_.cancel_timeout(); } if (!pending_qts_updates_.empty()) { // if still have a gap, reset timeout auto update_it = pending_qts_updates_.begin(); double receive_time = update_it->second.receive_time; for (size_t i = 0; i < GAP_TIMEOUT_UPDATE_COUNT; i++) { if (++update_it == pending_qts_updates_.end()) { break; } receive_time = min(receive_time, update_it->second.receive_time); } set_qts_gap_timeout(receive_time + MAX_UNFILLED_GAP_TIME - Time::now()); } CHECK(!running_get_difference_); auto passed_time = Time::now() - begin_time; if (passed_time >= UPDATE_APPLY_WARNING_TIME) { LOG(WARNING) << "QTS has changed from " << initial_qts << " to " << get_qts() << " after applying " << applied_update_count << " and keeping " << pending_qts_updates_.size() << " pending updates in " << passed_time << " seconds"; } } void UpdatesManager::set_pts_gap_timeout(double timeout) { if (!pts_gap_timeout_.has_timeout() || timeout < pts_gap_timeout_.get_timeout()) { if (timeout > 2 * MIN_UNFILLED_GAP_TIME) { min_pts_gap_timeout_.set_callback(std::move(check_pts_gap)); min_pts_gap_timeout_.set_callback_data(static_cast(td_)); min_pts_gap_timeout_.set_timeout_in(MIN_UNFILLED_GAP_TIME); } pts_gap_timeout_.set_callback(std::move(fill_pts_gap)); pts_gap_timeout_.set_callback_data(static_cast(td_)); pts_gap_timeout_.set_timeout_in(timeout); last_pts_gap_time_ = Time::now(); } } void UpdatesManager::set_seq_gap_timeout(double timeout) { if (!seq_gap_timeout_.has_timeout() || timeout < seq_gap_timeout_.get_timeout()) { seq_gap_timeout_.set_callback(std::move(fill_seq_gap)); seq_gap_timeout_.set_callback_data(static_cast(td_)); seq_gap_timeout_.set_timeout_in(timeout); } } void UpdatesManager::set_qts_gap_timeout(double timeout) { if (!qts_gap_timeout_.has_timeout() || timeout < qts_gap_timeout_.get_timeout()) { qts_gap_timeout_.set_callback(std::move(fill_qts_gap)); qts_gap_timeout_.set_callback_data(static_cast(td_)); qts_gap_timeout_.set_timeout_in(timeout); } } void UpdatesManager::on_pending_update(tl_object_ptr update, int32 seq, Promise &&promise, const char *source) { vector> updates; updates.push_back(std::move(update)); on_pending_updates(std::move(updates), seq, seq, 0, Time::now(), std::move(promise), source); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { int new_pts = update->pts_; int pts_count = update->pts_count_; add_pending_pts_update(std::move(update), new_pts, pts_count, Time::now(), std::move(promise), "updateNewMessage"); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { DialogId dialog_id = DialogId::get_message_dialog_id(update->message_); int new_pts = update->pts_; int pts_count = update->pts_count_; td_->messages_manager_->add_pending_channel_update(dialog_id, std::move(update), new_pts, pts_count, std::move(promise), "updateNewChannelMessage"); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { LOG(ERROR) << "Receive not in getDifference and not in on_pending_updates " << to_string(update); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { int new_pts = update->pts_; int pts_count = update->pts_count_; add_pending_pts_update(std::move(update), new_pts, pts_count, Time::now(), std::move(promise), "updateReadMessagesContents"); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { int new_pts = update->pts_; int pts_count = update->pts_count_; add_pending_pts_update(std::move(update), new_pts, pts_count, Time::now(), std::move(promise), "updateEditMessage"); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { int new_pts = update->pts_; int pts_count = update->pts_count_; if (update->messages_.empty()) { add_pending_pts_update(make_tl_object(), new_pts, pts_count, Time::now(), Promise(), "updateDeleteMessages"); promise.set_value(Unit()); } else { add_pending_pts_update(std::move(update), new_pts, pts_count, Time::now(), std::move(promise), "updateDeleteMessages"); } } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { int new_pts = update->pts_; int pts_count = update->pts_count_; add_pending_pts_update(std::move(update), new_pts, pts_count, Time::now(), std::move(promise), "updateReadHistoryInbox"); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { int new_pts = update->pts_; int pts_count = update->pts_count_; add_pending_pts_update(std::move(update), new_pts, pts_count, Time::now(), std::move(promise), "updateReadHistoryOutbox"); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->messages_manager_->on_update_service_notification(std::move(update), true, Promise()); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->messages_manager_->on_dialog_info_full_invalidated(DialogId(ChatId(update->chat_id_))); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->messages_manager_->on_update_read_channel_inbox(std::move(update)); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->messages_manager_->on_update_read_channel_outbox(std::move(update)); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->messages_manager_->on_update_read_channel_messages_contents(std::move(update)); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->messages_manager_->on_update_channel_too_long(std::move(update), false); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->contacts_manager_->invalidate_channel_full(ChannelId(update->channel_id_), false, "updateChannel"); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { DialogId dialog_id = DialogId::get_message_dialog_id(update->message_); int new_pts = update->pts_; int pts_count = update->pts_count_; td_->messages_manager_->add_pending_channel_update(dialog_id, std::move(update), new_pts, pts_count, std::move(promise), "updateEditChannelMessage"); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { DialogId dialog_id(ChannelId(update->channel_id_)); int new_pts = update->pts_; int pts_count = update->pts_count_; td_->messages_manager_->add_pending_channel_update(dialog_id, std::move(update), new_pts, pts_count, std::move(promise), "updateDeleteChannelMessages"); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { ChannelId channel_id(update->channel_id_); if (!channel_id.is_valid()) { LOG(ERROR) << "Receive invalid " << channel_id; } else { DialogId dialog_id(channel_id); td_->messages_manager_->on_update_message_view_count({dialog_id, MessageId(ServerMessageId(update->id_))}, update->views_); } promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { ChannelId channel_id(update->channel_id_); if (!channel_id.is_valid()) { LOG(ERROR) << "Receive invalid " << channel_id; } else { DialogId dialog_id(channel_id); td_->messages_manager_->on_update_message_forward_count({dialog_id, MessageId(ServerMessageId(update->id_))}, update->forwards_); } promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->messages_manager_->on_update_channel_max_unavailable_message_id( ChannelId(update->channel_id_), MessageId(ServerMessageId(update->available_min_id_)), "updateChannelAvailableMessages"); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { auto last_read_inbox_message_id = MessageId(ServerMessageId(update->read_max_id_)); if (!last_read_inbox_message_id.is_valid()) { LOG(ERROR) << "Receive " << to_string(update); return; } td_->messages_manager_->on_update_read_message_comments(DialogId(ChannelId(update->channel_id_)), MessageId(ServerMessageId(update->top_msg_id_)), MessageId(), last_read_inbox_message_id, MessageId(), -1); if ((update->flags_ & telegram_api::updateReadChannelDiscussionInbox::BROADCAST_ID_MASK) != 0) { td_->messages_manager_->on_update_read_message_comments(DialogId(ChannelId(update->broadcast_id_)), MessageId(ServerMessageId(update->broadcast_post_)), MessageId(), last_read_inbox_message_id, MessageId(), -1); } promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { auto last_read_outbox_message_id = MessageId(ServerMessageId(update->read_max_id_)); if (!last_read_outbox_message_id.is_valid()) { LOG(ERROR) << "Receive " << to_string(update); return; } td_->messages_manager_->on_update_read_message_comments(DialogId(ChannelId(update->channel_id_)), MessageId(ServerMessageId(update->top_msg_id_)), MessageId(), MessageId(), last_read_outbox_message_id, -1); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->forum_topic_manager_->on_update_forum_topic_is_pinned( DialogId(ChannelId(update->channel_id_)), MessageId(ServerMessageId(update->topic_id_)), update->pinned_); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { vector top_thread_message_ids; for (auto &server_message_id : update->order_) { auto message_id = MessageId(ServerMessageId(server_message_id)); if (!message_id.is_valid()) { LOG(ERROR) << "Receive " << to_string(update); break; } top_thread_message_ids.push_back(message_id); } td_->forum_topic_manager_->on_update_pinned_forum_topics(DialogId(ChannelId(update->channel_id_)), std::move(top_thread_message_ids)); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { int new_pts = update->pts_; int pts_count = update->pts_count_; add_pending_pts_update(std::move(update), new_pts, pts_count, Time::now(), std::move(promise), "updatePinnedMessages"); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { DialogId dialog_id(ChannelId(update->channel_id_)); int new_pts = update->pts_; int pts_count = update->pts_count_; td_->messages_manager_->add_pending_channel_update(dialog_id, std::move(update), new_pts, pts_count, std::move(promise), "updatePinnedChannelMessages"); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { switch (update->peer_->get_id()) { case telegram_api::notifyPeer::ID: { auto notify_peer = static_cast(update->peer_.get()); DialogId dialog_id(notify_peer->peer_); if (dialog_id.is_valid()) { td_->messages_manager_->on_update_dialog_notify_settings(dialog_id, std::move(update->notify_settings_), "updateNotifySettings"); } else { LOG(ERROR) << "Receive wrong " << to_string(update); } break; } case telegram_api::notifyUsers::ID: td_->notification_settings_manager_->on_update_scope_notify_settings(NotificationSettingsScope::Private, std::move(update->notify_settings_)); break; case telegram_api::notifyChats::ID: td_->notification_settings_manager_->on_update_scope_notify_settings(NotificationSettingsScope::Group, std::move(update->notify_settings_)); break; case telegram_api::notifyBroadcasts::ID: td_->notification_settings_manager_->on_update_scope_notify_settings(NotificationSettingsScope::Channel, std::move(update->notify_settings_)); break; case telegram_api::notifyForumTopic::ID: { auto notify_peer = static_cast(update->peer_.get()); DialogId dialog_id(notify_peer->peer_); auto top_thread_message_id = MessageId(ServerMessageId(notify_peer->top_msg_id_)); if (dialog_id.is_valid() && top_thread_message_id.is_valid()) { td_->forum_topic_manager_->on_update_forum_topic_notify_settings( dialog_id, top_thread_message_id, std::move(update->notify_settings_), "updateNotifySettings"); } else { LOG(ERROR) << "Receive wrong " << to_string(update); } break; } default: UNREACHABLE(); } promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->messages_manager_->on_get_peer_settings(DialogId(update->peer_), std::move(update->settings_)); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->messages_manager_->on_update_dialog_message_ttl(DialogId(update->peer_), MessageTtl(update->ttl_period_)); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->contacts_manager_->on_update_peer_located(std::move(update->peers_), true); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->web_pages_manager_->on_get_web_page(std::move(update->webpage_), DialogId()); add_pending_pts_update(make_tl_object(), update->pts_, update->pts_count_, Time::now(), Promise(), "updateWebPage"); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->web_pages_manager_->on_get_web_page(std::move(update->webpage_), DialogId()); DialogId dialog_id(ChannelId(update->channel_id_)); td_->messages_manager_->add_pending_channel_update(dialog_id, make_tl_object(), update->pts_, update->pts_count_, Promise(), "updateChannelWebPage"); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->messages_manager_->on_update_message_reactions( {DialogId(update->peer_), MessageId(ServerMessageId(update->msg_id_))}, std::move(update->reactions_), std::move(promise)); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->stickers_manager_->reload_recent_reactions(); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->attach_menu_manager_->reload_attach_menu_bots(std::move(promise)); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->attach_menu_manager_->close_web_view(update->query_id_, std::move(promise)); send_closure(G()->td(), &Td::send_update, td_api::make_object(update->query_id_)); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { for (auto &folder_peer : update->folder_peers_) { DialogId dialog_id(folder_peer->peer_); FolderId folder_id(folder_peer->folder_id_); td_->messages_manager_->on_update_dialog_folder_id(dialog_id, folder_id); } if (update->pts_ > 0) { add_pending_pts_update(make_tl_object(), update->pts_, update->pts_count_, Time::now(), Promise(), "updateFolderPeers"); } promise.set_value(Unit()); } int32 UpdatesManager::get_short_update_date() const { int32 now = G()->unix_time(); if (short_update_date_ > 0) { return min(short_update_date_, now); } return now; } bool UpdatesManager::have_update_pts_changed(const vector> &updates) { for (auto &update : updates) { CHECK(update != nullptr); if (update->get_id() == telegram_api::updatePtsChanged::ID) { return true; } } return false; } bool UpdatesManager::check_pts_update_dialog_id(DialogId dialog_id) { switch (dialog_id.get_type()) { case DialogType::User: case DialogType::Chat: return true; case DialogType::Channel: case DialogType::SecretChat: case DialogType::None: return false; default: UNREACHABLE(); return false; } } bool UpdatesManager::check_pts_update(const tl_object_ptr &update) { CHECK(update != nullptr); switch (update->get_id()) { case dummyUpdate::ID: case updateSentMessage::ID: case telegram_api::updateReadMessagesContents::ID: case telegram_api::updateDeleteMessages::ID: return true; case telegram_api::updateNewMessage::ID: { auto update_new_message = static_cast(update.get()); return check_pts_update_dialog_id(DialogId::get_message_dialog_id(update_new_message->message_)); } case telegram_api::updateReadHistoryInbox::ID: { auto update_read_history_inbox = static_cast(update.get()); return check_pts_update_dialog_id(DialogId(update_read_history_inbox->peer_)); } case telegram_api::updateReadHistoryOutbox::ID: { auto update_read_history_outbox = static_cast(update.get()); return check_pts_update_dialog_id(DialogId(update_read_history_outbox->peer_)); } case telegram_api::updateEditMessage::ID: { auto update_edit_message = static_cast(update.get()); return check_pts_update_dialog_id(DialogId::get_message_dialog_id(update_edit_message->message_)); } case telegram_api::updatePinnedMessages::ID: { auto update_pinned_messages = static_cast(update.get()); return check_pts_update_dialog_id(DialogId(update_pinned_messages->peer_)); } default: return false; } } bool UpdatesManager::is_pts_update(const telegram_api::Update *update) { switch (update->get_id()) { case telegram_api::updateNewMessage::ID: case telegram_api::updateReadMessagesContents::ID: case telegram_api::updateEditMessage::ID: case telegram_api::updateDeleteMessages::ID: case telegram_api::updateReadHistoryInbox::ID: case telegram_api::updateReadHistoryOutbox::ID: case telegram_api::updateWebPage::ID: case telegram_api::updatePinnedMessages::ID: case telegram_api::updateFolderPeers::ID: return true; default: return false; } } int32 UpdatesManager::get_update_pts(const telegram_api::Update *update) { switch (update->get_id()) { case telegram_api::updateNewMessage::ID: return static_cast(update)->pts_; case telegram_api::updateReadMessagesContents::ID: return static_cast(update)->pts_; case telegram_api::updateEditMessage::ID: return static_cast(update)->pts_; case telegram_api::updateDeleteMessages::ID: return static_cast(update)->pts_; case telegram_api::updateReadHistoryInbox::ID: return static_cast(update)->pts_; case telegram_api::updateReadHistoryOutbox::ID: return static_cast(update)->pts_; case telegram_api::updateWebPage::ID: return static_cast(update)->pts_; case telegram_api::updatePinnedMessages::ID: return static_cast(update)->pts_; case telegram_api::updateFolderPeers::ID: return static_cast(update)->pts_; default: return 0; } } bool UpdatesManager::is_qts_update(const telegram_api::Update *update) { switch (update->get_id()) { case telegram_api::updateNewEncryptedMessage::ID: case telegram_api::updateMessagePollVote::ID: case telegram_api::updateBotStopped::ID: case telegram_api::updateChatParticipant::ID: case telegram_api::updateChannelParticipant::ID: case telegram_api::updateBotChatInviteRequester::ID: return true; default: return false; } } int32 UpdatesManager::get_update_qts(const telegram_api::Update *update) { switch (update->get_id()) { case telegram_api::updateNewEncryptedMessage::ID: return static_cast(update)->qts_; case telegram_api::updateMessagePollVote::ID: return static_cast(update)->qts_; case telegram_api::updateBotStopped::ID: return static_cast(update)->qts_; case telegram_api::updateChatParticipant::ID: return static_cast(update)->qts_; case telegram_api::updateChannelParticipant::ID: return static_cast(update)->qts_; case telegram_api::updateBotChatInviteRequester::ID: return static_cast(update)->qts_; default: return 0; } } bool UpdatesManager::is_channel_pts_update(const telegram_api::Update *update) { switch (update->get_id()) { case telegram_api::updateNewChannelMessage::ID: case telegram_api::updateEditChannelMessage::ID: case telegram_api::updateDeleteChannelMessages::ID: case telegram_api::updatePinnedChannelMessages::ID: return true; default: return false; } } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { DialogId dialog_id(UserId(update->user_id_)); td_->messages_manager_->on_dialog_action(dialog_id, MessageId(), dialog_id, DialogAction(std::move(update->action_)), get_short_update_date()); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->messages_manager_->on_dialog_action(DialogId(ChatId(update->chat_id_)), MessageId(), DialogId(update->from_id_), DialogAction(std::move(update->action_)), get_short_update_date()); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->messages_manager_->on_dialog_action(DialogId(ChannelId(update->channel_id_)), MessageId(ServerMessageId(update->top_msg_id_)), DialogId(update->from_id_), DialogAction(std::move(update->action_)), get_short_update_date()); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { SecretChatId secret_chat_id(update->chat_id_); UserId user_id = td_->contacts_manager_->get_secret_chat_user_id(secret_chat_id); td_->messages_manager_->on_dialog_action(DialogId(secret_chat_id), MessageId(), DialogId(user_id), DialogAction::get_typing_action(), get_short_update_date()); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->contacts_manager_->on_update_user_online(UserId(update->user_id_), std::move(update->status_)); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->contacts_manager_->on_update_user_name(UserId(update->user_id_), std::move(update->first_name_), std::move(update->last_name_), Usernames{string(), std::move(update->usernames_)}); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->contacts_manager_->on_update_user_phone_number(UserId(update->user_id_), std::move(update->phone_)); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->contacts_manager_->invalidate_user_full(UserId(update->user_id_)); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->contacts_manager_->on_update_user_emoji_status(UserId(update->user_id_), std::move(update->emoji_status_)); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { get_recent_emoji_statuses(td_, Auto()); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->messages_manager_->on_update_dialog_is_blocked(DialogId(update->peer_id_), update->blocked_); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->contacts_manager_->on_update_bot_commands(DialogId(update->peer_), UserId(update->bot_id_), std::move(update->commands_)); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->contacts_manager_->on_update_bot_menu_button(UserId(update->bot_id_), std::move(update->button_)); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->contacts_manager_->on_get_chat_participants(std::move(update->participants_), true); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->contacts_manager_->on_update_chat_add_user(ChatId(update->chat_id_), UserId(update->inviter_id_), UserId(update->user_id_), update->date_, update->version_); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->contacts_manager_->on_update_chat_edit_administrator(ChatId(update->chat_id_), UserId(update->user_id_), update->is_admin_, update->version_); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->contacts_manager_->on_update_chat_delete_user(ChatId(update->chat_id_), UserId(update->user_id_), update->version_); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { DialogId dialog_id(update->peer_); RestrictedRights permissions(update->default_banned_rights_); auto version = update->version_; switch (dialog_id.get_type()) { case DialogType::Chat: td_->contacts_manager_->on_update_chat_default_permissions(dialog_id.get_chat_id(), permissions, version); break; case DialogType::Channel: LOG_IF(ERROR, version != 0) << "Receive version " << version << " in " << dialog_id; td_->contacts_manager_->on_update_channel_default_permissions(dialog_id.get_channel_id(), permissions); break; case DialogType::None: case DialogType::User: case DialogType::SecretChat: default: LOG(ERROR) << "Receive updateChatDefaultBannedRights in " << dialog_id; break; } promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->messages_manager_->on_update_dialog_draft_message( DialogId(update->peer_), MessageId(ServerMessageId(update->top_msg_id_)), std::move(update->draft_)); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->messages_manager_->on_update_dialog_is_pinned(FolderId(update->folder_id_), DialogId(update->peer_), update->pinned_); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->messages_manager_->on_update_pinned_dialogs(FolderId(update->folder_id_)); // TODO use update->order_ promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->messages_manager_->on_update_dialog_is_marked_as_unread(DialogId(update->peer_), update->unread_); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->dialog_filter_manager_->on_update_dialog_filters(std::move(promise)); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->dialog_filter_manager_->on_update_dialog_filters(std::move(promise)); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->dialog_filter_manager_->on_update_dialog_filters(std::move(promise)); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { send_closure(G()->config_manager(), &ConfigManager::on_dc_options_update, DcOptions(update->dc_options_)); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->inline_queries_manager_->on_new_query(update->query_id_, UserId(update->user_id_), Location(td_, update->geo_), std::move(update->peer_type_), update->query_, update->offset_); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->inline_queries_manager_->on_chosen_result(UserId(update->user_id_), Location(td_, update->geo_), update->query_, update->id_, std::move(update->msg_id_)); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->callback_queries_manager_->on_new_query(update->flags_, update->query_id_, UserId(update->user_id_), DialogId(update->peer_), MessageId(ServerMessageId(update->msg_id_)), std::move(update->data_), update->chat_instance_, std::move(update->game_short_name_)); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->callback_queries_manager_->on_new_inline_query(update->flags_, update->query_id_, UserId(update->user_id_), std::move(update->msg_id_), std::move(update->data_), update->chat_instance_, std::move(update->game_short_name_)); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->stickers_manager_->reload_favorite_stickers(true); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->animations_manager_->reload_saved_animations(true); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { send_closure(td_->config_manager_, &ConfigManager::request_config, false); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { if (td_->option_manager_->get_option_integer("session_count") > 1) { auto old_pts = get_pts(); auto new_pts = 1; if (old_pts != new_pts) { LOG(WARNING) << "PTS changes from " << old_pts << " from updatePtsChanged"; save_pts(new_pts); add_pts(new_pts).set_value(Unit()); get_difference("updatePtsChanged"); } } else { set_pts(std::numeric_limits::max(), "updatePtsChanged").set_value(Unit()); } promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { send_closure(td_->secret_chats_manager_, &SecretChatsManager::on_update_chat, std::move(update)); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { auto qts = update->qts_; add_pending_qts_update(std::move(update), qts, std::move(promise)); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->messages_manager_->read_secret_chat_outbox(SecretChatId(update->chat_id_), update->max_date_, update->date_); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->privacy_manager_->on_update_privacy(std::move(update)); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->stickers_manager_->on_get_messages_sticker_set(StickerSetId(), std::move(update->stickerset_), true, "updateNewStickerSet"); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { auto sticker_type = get_sticker_type(update->masks_, update->emojis_); td_->stickers_manager_->on_update_sticker_sets(sticker_type); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { auto sticker_type = get_sticker_type(update->masks_, update->emojis_); td_->stickers_manager_->on_update_sticker_sets_order(sticker_type, StickersManager::convert_sticker_set_ids(update->order_)); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { auto sticker_type = get_sticker_type(update->masks_, update->emojis_); td_->stickers_manager_->on_update_move_sticker_set_to_top(sticker_type, StickerSetId(update->stickerset_)); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->stickers_manager_->reload_featured_sticker_sets(StickerType::Regular, true); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->stickers_manager_->reload_featured_sticker_sets(StickerType::CustomEmoji, true); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->stickers_manager_->reload_recent_stickers(false, true); td_->stickers_manager_->reload_recent_stickers(true, true); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { UserId user_id(update->user_id_); if (!user_id.is_valid()) { LOG(ERROR) << "Receive shipping query from invalid " << user_id; } else { CHECK(update->shipping_address_ != nullptr); send_closure( G()->td(), &Td::send_update, make_tl_object( update->query_id_, td_->contacts_manager_->get_user_id_object(user_id, "updateNewShippingQuery"), update->payload_.as_slice().str(), get_address_object(get_address(std::move(update->shipping_address_))))); // TODO use convert_address } promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { UserId user_id(update->user_id_); if (!user_id.is_valid()) { LOG(ERROR) << "Receive pre-checkout query from invalid " << user_id; } else if (update->total_amount_ <= 0 || !check_currency_amount(update->total_amount_)) { LOG(ERROR) << "Receive pre-checkout query with invalid total amount " << update->total_amount_; } else { send_closure( G()->td(), &Td::send_update, make_tl_object( update->query_id_, td_->contacts_manager_->get_user_id_object(user_id, "updateNewPreCheckoutQuery"), update->currency_, update->total_amount_, update->payload_.as_slice().str(), update->shipping_option_id_, get_order_info_object(get_order_info(std::move(update->info_))))); } promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { send_closure(G()->td(), &Td::send_update, make_tl_object(update->data_->data_)); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { send_closure(G()->td(), &Td::send_update, make_tl_object(update->query_id_, update->data_->data_, update->timeout_)); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { send_closure(G()->call_manager(), &CallManager::update_call, std::move(update)); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { send_closure(G()->call_manager(), &CallManager::update_call_signaling_data, update->phone_call_id_, update->data_.as_slice().str()); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { if (update->presentation_) { LOG(ERROR) << "Receive unexpected updateGroupCallConnection"; } else { send_closure(G()->group_call_manager(), &GroupCallManager::on_update_group_call_connection, std::move(update->params_->data_)); } promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { DialogId dialog_id(ChatId(update->chat_id_)); if (!td_->messages_manager_->have_dialog_force(dialog_id, "updateGroupCall")) { dialog_id = DialogId(ChannelId(update->chat_id_)); if (!td_->messages_manager_->have_dialog_force(dialog_id, "updateGroupCall")) { dialog_id = DialogId(); } } send_closure(G()->group_call_manager(), &GroupCallManager::on_update_group_call, std::move(update->call_), dialog_id); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { send_closure(G()->group_call_manager(), &GroupCallManager::on_update_group_call_participants, InputGroupCallId(update->call_), std::move(update->participants_), update->version_, false); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->contacts_manager_->on_update_contacts_reset(); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { send_closure(G()->language_pack_manager(), &LanguagePackManager::on_language_pack_too_long, std::move(update->lang_code_)); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { send_closure(G()->language_pack_manager(), &LanguagePackManager::on_update_language_pack, std::move(update->difference_)); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->messages_manager_->on_update_live_location_viewed( {DialogId(update->peer_), MessageId(ServerMessageId(update->msg_id_))}); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->messages_manager_->on_update_message_extended_media( {DialogId(update->peer_), MessageId(ServerMessageId(update->msg_id_))}, std::move(update->extended_media_)); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->poll_manager_->on_get_poll(PollId(update->poll_id_), std::move(update->poll_), std::move(update->results_), "updateMessagePoll"); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { auto qts = update->qts_; add_pending_qts_update(std::move(update), qts, std::move(promise)); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->messages_manager_->on_get_message(std::move(update->message_), true, false, true, "updateNewScheduledMessage"); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { vector message_ids = transform(update->messages_, [](int32 scheduled_server_message_id) { return ScheduledServerMessageId(scheduled_server_message_id); }); td_->messages_manager_->on_update_delete_scheduled_messages(DialogId(update->peer_), std::move(message_ids)); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { LOG(INFO) << "Ignore updateLoginToken after authorization"; promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { auto qts = update->qts_; add_pending_qts_update(std::move(update), qts, std::move(promise)); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { auto qts = update->qts_; add_pending_qts_update(std::move(update), qts, std::move(promise)); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { auto qts = update->qts_; add_pending_qts_update(std::move(update), qts, std::move(promise)); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { auto qts = update->qts_; add_pending_qts_update(std::move(update), qts, std::move(promise)); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->theme_manager_->on_update_theme(std::move(update->theme_), std::move(promise)); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->messages_manager_->on_update_dialog_pending_join_requests(DialogId(update->peer_), update->requests_pending_, std::move(update->recent_requesters_)); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->notification_settings_manager_->reload_saved_ringtones(std::move(promise)); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { auto it = pending_audio_transcriptions_.find(update->transcription_id_); if (it == pending_audio_transcriptions_.end()) { return promise.set_value(Unit()); } // flags_, dialog_id_ and message_id_ must not be used if (!update->pending_) { auto on_update = std::move(it->second); pending_audio_transcriptions_.erase(it); pending_audio_transcription_timeout_.cancel_timeout(update->transcription_id_); on_update(std::move(update)); } else { it->second(std::move(update)); } promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { LOG(ERROR) << "Receive " << to_string(update); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->autosave_manager_->reload_autosave_settings(); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->story_manager_->on_get_story(DialogId(UserId(update->user_id_)), std::move(update->story_)); promise.set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { td_->story_manager_->on_update_read_stories(DialogId(UserId(update->user_id_)), StoryId(update->max_id_)); promise.set_value(Unit()); } // unsupported updates void UpdatesManager::on_update(tl_object_ptr update, Promise &&promise) { promise.set_value(Unit()); } } // namespace td