// // Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2019 // // 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/telegram_api.hpp" #include "td/telegram/AnimationsManager.h" #include "td/telegram/AuthManager.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/DialogId.h" #include "td/telegram/FolderId.h" #include "td/telegram/Global.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/net/DcOptions.h" #include "td/telegram/net/NetQuery.h" #include "td/telegram/NotificationManager.h" #include "td/telegram/Payments.h" #include "td/telegram/PollId.h" #include "td/telegram/PollManager.h" #include "td/telegram/PrivacyManager.h" #include "td/telegram/ScheduledServerMessageId.h" #include "td/telegram/SecretChatId.h" #include "td/telegram/SecretChatsManager.h" #include "td/telegram/ServerMessageId.h" #include "td/telegram/StateManager.h" #include "td/telegram/StickerSetId.h" #include "td/telegram/StickersManager.h" #include "td/telegram/Td.h" #include "td/telegram/TdDb.h" #include "td/telegram/WebPagesManager.h" #include "td/utils/buffer.h" #include "td/utils/format.h" #include "td/utils/logging.h" #include "td/utils/misc.h" #include "td/utils/Random.h" #include "td/utils/Slice.h" #include "td/utils/Status.h" #include "td/utils/StringBuilder.h" #include namespace td { int VERBOSITY_NAME(get_difference) = VERBOSITY_NAME(INFO); class OnUpdate { UpdatesManager *manager_; tl_object_ptr &update_; bool force_apply_; public: OnUpdate(UpdatesManager *manager, tl_object_ptr &update, bool force_apply) : manager_(manager), update_(update), force_apply_(force_apply) { } template void operator()(T &obj) const { CHECK(&*update_ == &obj); manager_->on_update(move_tl_object_as(update_), force_apply_); } }; class GetUpdatesStateQuery : public Td::ResultHandler { public: void send() { // TODO this call must be first after client is logged in, there must be no API calls before // it succeeds send_query(G()->net_query_creator().create(create_storer(telegram_api::updates_getState()))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } auto state = result_ptr.move_as_ok(); CHECK(state->get_id() == telegram_api::updates_state::ID); td->updates_manager_->on_get_updates_state(std::move(state), "GetUpdatesStateQuery"); } void on_error(uint64 id, Status status) override { if (status.code() != 401) { LOG(ERROR) << "Receive updates.getState error: " << status; } status.ignore(); td->updates_manager_->on_get_updates_state(nullptr, "GetUpdatesStateQuery"); } }; class PingServerQuery : public Td::ResultHandler { public: void send() { send_query(G()->net_query_creator().create(create_storer(telegram_api::updates_getState()))); } void on_result(uint64 id, BufferSlice packet) override { auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } auto state = result_ptr.move_as_ok(); CHECK(state->get_id() == telegram_api::updates_state::ID); td->updates_manager_->on_server_pong(std::move(state)); } void on_error(uint64 id, Status status) override { status.ignore(); td->updates_manager_->on_server_pong(nullptr); } }; class GetDifferenceQuery : public Td::ResultHandler { public: void send() { int32 pts = td->updates_manager_->get_pts(); int32 date = td->updates_manager_->get_date(); int32 qts = td->updates_manager_->get_qts(); if (pts < 0) { pts = 0; } VLOG(get_difference) << tag("pts", pts) << tag("qts", qts) << tag("date", date); send_query( G()->net_query_creator().create(create_storer(telegram_api::updates_getDifference(0, pts, 0, date, qts)))); } void on_result(uint64 id, BufferSlice packet) override { VLOG(get_difference) << "Receive getDifference result of size " << packet.size(); auto result_ptr = fetch_result(packet); if (result_ptr.is_error()) { return on_error(id, result_ptr.move_as_error()); } td->updates_manager_->on_get_difference(result_ptr.move_as_ok()); } void on_error(uint64 id, Status status) override { if (status.code() != 401) { LOG(ERROR) << "Receive updates.getDifference error: " << status; } td->updates_manager_->on_get_difference(nullptr); if (status.message() == CSlice("PERSISTENT_TIMESTAMP_INVALID")) { td->updates_manager_->set_pts(std::numeric_limits::max(), "PERSISTENT_TIMESTAMP_INVALID") .set_value(Unit()); } status.ignore(); } }; const double UpdatesManager::MAX_UNFILLED_GAP_TIME = 1.0; UpdatesManager::UpdatesManager(Td *td, ActorShared<> parent) : td_(td), parent_(std::move(parent)) { pts_manager_.init(-1); } void UpdatesManager::tear_down() { parent_.reset(); } void UpdatesManager::fill_pts_gap(void *td) { fill_gap(td, "pts"); } void UpdatesManager::fill_seq_gap(void *td) { fill_gap(td, "seq"); } void UpdatesManager::fill_get_difference_gap(void *td) { fill_gap(td, "getDifference"); } void UpdatesManager::fill_gap(void *td, const char *source) { CHECK(td != nullptr); auto updates_manager = static_cast(td)->updates_manager_.get(); LOG(WARNING) << "Filling gap in " << source << " by running getDifference"; updates_manager->get_difference("fill_gap"); } void UpdatesManager::get_difference(const char *source) { if (get_pts() == -1) { init_state(); return; } if (!td_->auth_manager_->is_authorized()) { return; } if (running_get_difference_) { VLOG(get_difference) << "Skip running getDifference from " << source << " because it is already running"; return; } running_get_difference_ = true; VLOG(get_difference) << "-----BEGIN GET DIFFERENCE----- from " << source; before_get_difference(false); td_->create_handler()->send(); last_get_difference_pts_ = get_pts(); } 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); td_->messages_manager_->before_get_difference(); if (!is_initial) { send_closure(td_->secret_chats_manager_, &SecretChatsManager::before_get_difference, get_qts()); } send_closure_later(td_->notification_manager_actor_, &NotificationManager::before_get_difference); } Promise<> UpdatesManager::add_pts(int32 pts) { auto id = pts_manager_.add_pts(pts); return PromiseCreator::event(self_closure(this, &UpdatesManager::on_pts_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::save_pts(int32 pts) { if (pts == std::numeric_limits::max()) { G()->td_db()->get_binlog_pmc()->erase("updates.pts"); } else if (!G()->ignore_backgrond_updates()) { G()->td_db()->get_binlog_pmc()->set("updates.pts", to_string(pts)); } } 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; G()->td_db()->get_binlog_pmc()->erase("updates.pts"); auto result = add_pts(std::numeric_limits::max()); init_state(); return result; } Promise<> result; if (pts > get_pts() || (0 < pts && pts < get_pts() - 399999)) { // pts can only go up or drop cardinally if (pts < get_pts() - 399999) { LOG(WARNING) << "Pts decreases from " << get_pts() << " to " << pts << " from " << source; } else { LOG(INFO) << "Update pts from " << get_pts() << " to " << pts << " from " << source; } result = add_pts(pts); if (last_get_difference_pts_ + FORCED_GET_DIFFERENCE_PTS_DIFF < get_pts()) { last_get_difference_pts_ = get_pts(); schedule_get_difference("set_pts"); } } else if (pts < get_pts()) { LOG(ERROR) << "Receive wrong pts = " << pts << " from " << source << ". Current pts = " << get_pts(); } return result; } void UpdatesManager::set_qts(int32 qts) { if (qts > qts_) { LOG(INFO) << "Update qts to " << qts; qts_ = qts; if (!G()->ignore_backgrond_updates()) { G()->td_db()->get_binlog_pmc()->set("updates.qts", to_string(qts)); } } else if (qts < qts_) { LOG(ERROR) << "Receive wrong qts = " << qts << ". Current qts = " << qts_; } } 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 << ". Now = " << now; date_ = now; if (date_ <= date) { return; } } date_ = date; date_source_ = std::move(date_source); if (!G()->ignore_backgrond_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) && 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_dialog(DialogId dialog_id) const { 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_->have_input_user(user_id)) { return false; } } } return true; } bool UpdatesManager::is_acceptable_message_forward_header( const telegram_api::object_ptr &header) const { if (header == nullptr) { return true; } auto flags = header->flags_; if (flags & telegram_api::messageFwdHeader::CHANNEL_ID_MASK) { ChannelId channel_id(header->channel_id_); if (!is_acceptable_channel(channel_id)) { return false; } } if (flags & telegram_api::messageFwdHeader::FROM_ID_MASK) { UserId user_id(header->from_id_); if (!is_acceptable_user(user_id)) { return false; } } if (flags & telegram_api::messageFwdHeader::SAVED_FROM_PEER_MASK) { DialogId dialog_id(header->saved_from_peer_); if (!is_acceptable_dialog(dialog_id)) { 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_dialog(DialogId(message->to_id_))) { return false; } if (message->flags_ & MessagesManager::MESSAGE_FLAG_HAS_FROM_ID) { if (!is_acceptable_user(UserId(message->from_id_))) { 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->flags_ & MessagesManager::MESSAGE_FLAG_HAS_MEDIA) { CHECK(message->media_ != nullptr); auto media_id = message->media_->get_id(); if (media_id == telegram_api::messageMediaContact::ID) { auto message_media_contact = static_cast(message->media_.get()); UserId user_id(message_media_contact->user_id_); if (user_id != UserId() && !is_acceptable_user(user_id)) { return false; } } /* 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); } break; } case telegram_api::messageService::ID: { auto message = static_cast(message_ptr); if (!is_acceptable_dialog(DialogId(message->to_id_))) { return false; } if (message->flags_ & MessagesManager::MESSAGE_FLAG_HAS_FROM_ID) { if (!is_acceptable_user(UserId(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: 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: { auto chat_joined_by_link = static_cast(action); if (!is_acceptable_user(UserId(chat_joined_by_link->inviter_id_))) { return false; } 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; } 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) { CHECK(updates_ptr != nullptr); 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()) { LOG(INFO) << "Ignore updates received before authorization or after logout"; return; } switch (updates_type) { case telegram_api::updatesTooLong::ID: get_difference("updatesTooLong"); 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->flags_ & MessagesManager::MESSAGE_FLAG_IS_OUT ? td_->contacts_manager_->get_my_id().get() : update->user_id_; update->flags_ |= MessagesManager::MESSAGE_FLAG_HAS_FROM_ID; on_pending_update( make_tl_object( make_tl_object( update->flags_, false /*ignored*/, false /*ignored*/, false /*ignored*/, false /*ignored*/, false /*ignored*/, false /*ignored*/, false /*ignored*/, false /*ignored*/, update->id_, from_id, make_tl_object(update->user_id_), std::move(update->fwd_from_), update->via_bot_id_, update->reply_to_msg_id_, update->date_, update->message_, nullptr, nullptr, std::move(update->entities_), 0, 0, "", 0, Auto()), update->pts_, update->pts_count_), 0, "telegram_api::updatesShortMessage"); 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; on_pending_update( make_tl_object( make_tl_object( update->flags_, false /*ignored*/, false /*ignored*/, false /*ignored*/, false /*ignored*/, false /*ignored*/, false /*ignored*/, false /*ignored*/, false /*ignored*/, update->id_, update->from_id_, make_tl_object(update->chat_id_), std::move(update->fwd_from_), update->via_bot_id_, update->reply_to_msg_id_, update->date_, update->message_, nullptr, nullptr, std::move(update->entities_), 0, 0, "", 0, Auto()), update->pts_, update->pts_count_), 0, "telegram_api::updatesShortChatMessage"); 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)); return get_difference("unacceptable short update"); } short_update_date_ = update->date_; if (!downcast_call(*update->update_, OnUpdate(this, update->update_, false))) { LOG(ERROR) << "Can't call on some update"; } 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_, "telegram_api::updatesCombined"); break; } case telegram_api::updates::ID: { auto updates = move_tl_object_as(updates_ptr); td_->contacts_manager_->on_get_users(std::move(updates->users_), "updates"); td_->contacts_manager_->on_get_chats(std::move(updates->chats_), "updates"); on_pending_updates(std::move(updates->updates_), updates->seq_, updates->seq_, updates->date_, "telegram_api::updates"); break; } case telegram_api::updateShortSentMessage::ID: LOG(ERROR) << "Receive " << oneline(to_string(updates_ptr)); get_difference("updateShortSentMessage"); break; default: UNREACHABLE(); } } void UpdatesManager::on_failed_get_difference() { schedule_get_difference("on_failed_get_difference"); } void UpdatesManager::schedule_get_difference(const char *source) { VLOG(get_difference) << "Schedule getDifference from " << source; if (!retry_timeout_.has_timeout()) { 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); } } } void UpdatesManager::on_get_updates_state(tl_object_ptr &&state, const char *source) { if (state == nullptr) { running_get_difference_ = false; on_failed_get_difference(); return; } 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 pts_manager_.init(state->pts_); last_get_difference_pts_ = get_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)); // set_qts(state->qts_); 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; } } std::unordered_set UpdatesManager::get_sent_messages_random_ids(const telegram_api::Updates *updates_ptr) { std::unordered_set random_ids; auto updates = get_updates(updates_ptr); if (updates != nullptr) { for (auto &update : *updates) { if (update->get_id() == telegram_api::updateMessageID::ID) { int64 random_id = static_cast(update.get())->random_id_; if (!random_ids.insert(random_id).second) { LOG(ERROR) << "Receive twice updateMessageID for " << random_id; } } } } return random_ids; } 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) { auto constructor_id = update->get_id(); if (constructor_id == telegram_api::updateNewMessage::ID) { messages.emplace_back(&static_cast(update.get())->message_); } else if (constructor_id == telegram_api::updateNewChannelMessage::ID) { messages.emplace_back(&static_cast(update.get())->message_); } else if (constructor_id == telegram_api::updateNewScheduledMessage::ID) { messages.emplace_back(&static_cast(update.get())->message_); } } } return messages; } 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_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 chat_id = ContactsManager::get_chat_id(chat); if (chat_id.is_valid()) { dialog_ids.push_back(DialogId(chat_id)); continue; } auto channel_id = ContactsManager::get_channel_id(chat); if (channel_id.is_valid()) { dialog_ids.push_back(DialogId(channel_id)); continue; } LOG(ERROR) << "Can't find id of " << oneline(to_string(chat)); } return dialog_ids; } void UpdatesManager::init_state() { if (!td_->auth_manager_->is_authorized()) { return; } auto pmc = G()->td_db()->get_binlog_pmc(); if (G()->ignore_backgrond_updates()) { // 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); td_->create_handler()->send(); } return; } pts_manager_.init(to_integer(pts_str)); last_get_difference_pts_ = get_pts(); qts_ = to_integer(pmc->get("updates.qts")); date_ = to_integer(pmc->get("updates.date")); date_source_ = "database"; LOG(DEBUG) << "Init: " << get_pts() << " " << qts_ << " " << date_; send_closure(td_->secret_chats_manager_, &SecretChatsManager::init_qts, qts_); get_difference("init_state"); } void UpdatesManager::ping_server() { td_->create_handler()->send(); } void UpdatesManager::on_server_pong(tl_object_ptr &&state) { LOG(INFO) << "Receive " << oneline(to_string(state)); if (state == nullptr || state->pts_ > get_pts() || state->seq_ > seq_) { get_difference("on server pong"); } } void UpdatesManager::process_get_difference_updates( vector> &&new_messages, vector> &&new_encrypted_messages, int32 qts, 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 on_update(move_tl_object_as(update), true); CHECK(!running_get_difference_); } if (constructor_id == telegram_api::updateEncryption::ID) { on_update(move_tl_object_as(update), true); CHECK(!running_get_difference_); } if (constructor_id == telegram_api::updateFolderPeers::ID) { on_update(move_tl_object_as(update), true); CHECK(!running_get_difference_); } /* // 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) { on_update(move_tl_object_as(update), true); 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, true, true, "get difference"); CHECK(!running_get_difference_); } for (auto &encrypted_message : new_encrypted_messages) { on_update(make_tl_object(std::move(encrypted_message), 0), true); } send_closure(td_->secret_chats_manager_, &SecretChatsManager::update_qts, qts); process_updates(std::move(other_updates), true); } void UpdatesManager::on_get_difference(tl_object_ptr &&difference_ptr) { VLOG(get_difference) << "----- END GET DIFFERENCE-----"; running_get_difference_ = false; if (difference_ptr == nullptr) { on_failed_get_difference(); return; } LOG(DEBUG) << "Result of get difference: " << to_string(difference_ptr); 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_; 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"); process_get_difference_updates(std::move(difference->new_messages_), std::move(difference->new_encrypted_messages_), difference->state_->qts_, 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); if (difference->intermediate_state_->pts_ >= get_pts() && get_pts() != std::numeric_limits::max() && difference->intermediate_state_->date_ >= date_ && difference->intermediate_state_->qts_ == qts_) { // 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"); process_get_difference_updates(std::move(difference->new_messages_), std::move(difference->new_encrypted_messages_), difference->intermediate_state_->qts_, 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->intermediate_state_), "get difference slice"); get_difference("on updates_differenceSlice"); break; } case telegram_api::updates_differenceTooLong::ID: { 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(); } if (!running_get_difference_) { after_get_difference(); } } void UpdatesManager::after_get_difference() { CHECK(!running_get_difference_); send_closure(td_->secret_chats_manager_, &SecretChatsManager::after_get_difference); retry_timeout_.cancel_timeout(); retry_time_ = 1; process_pending_seq_updates(); // cancels seq_gap_timeout_, may apply some updates coming before getDifference, but // not returned in getDifference if (running_get_difference_) { return; } if (postponed_updates_.size()) { VLOG(get_difference) << "Begin to apply postponed updates"; 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; // ignore it->second.date, because it may be too old postponed_updates_.erase(it); on_pending_updates(std::move(updates), updates_seq_begin, updates_seq_end, 0, "postponed updates"); if (running_get_difference_) { VLOG(get_difference) << "Finish to apply postponed updates because forced to run getDifference"; return; } } VLOG(get_difference) << "Finish to apply postponed updates"; } td_->animations_manager_->after_get_difference(); td_->contacts_manager_->after_get_difference(); td_->inline_queries_manager_->after_get_difference(); td_->messages_manager_->after_get_difference(); td_->stickers_manager_->after_get_difference(); send_closure_later(td_->notification_manager_actor_, &NotificationManager::after_get_difference); send_closure(G()->state_manager(), &StateManager::on_synchronized, true); } void UpdatesManager::on_pending_updates(vector> &&updates, int32 seq_begin, int32 seq_end, int32 date, 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; } // 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; } if (running_get_difference_) { LOG(INFO) << "Postpone " << updates.size() << " updates [" << seq_begin << ", " << seq_end << "] with date = " << date << " from " << source; postponed_updates_.emplace(seq_begin, PendingUpdates(seq_begin, seq_end, date, std::move(updates))); return; } // TODO typings must be processed before NewMessage size_t processed_updates = 0; 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 // don't do that for service messages, because they can be about bot's kicking if (message_ptr != nullptr && (*message_ptr)->get_id() != telegram_api::messageService::ID) { auto dialog_id = td_->messages_manager_->get_message_dialog_id(*message_ptr); if (dialog_id.get_type() == DialogType::Channel) { LOG(INFO) << "Replace update about new message with updateChannelTooLong in " << dialog_id; auto channel_id = dialog_id.get_channel_id(); if (td_->contacts_manager_->have_channel_force(channel_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); } } return get_difference("on unacceptable updates in on_pending_updates"); } } 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; for (auto &update : updates) { if (update != nullptr) { auto constructor_id = update->get_id(); 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 (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)); } schedule_get_difference("on_get_wrong_updates"); return; } 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); bool success = false; if (ordinary_new_message_count != 0) { success = td_->messages_manager_->on_update_message_id( sent_message_update->random_id_, MessageId(ServerMessageId(sent_message_update->id_)), source); } else if (scheduled_new_message_count != 0) { success = td_->messages_manager_->on_update_scheduled_message_id( sent_message_update->random_id_, ScheduledServerMessageId(sent_message_update->id_), source); } if (!success) { for (auto &debug_update : updates) { LOG(ERROR) << "Update: " << oneline(to_string(debug_update)); } } processed_updates++; update = nullptr; } if (id == telegram_api::updateFolderPeers::ID) { on_update(move_tl_object_as(update), false); processed_updates++; update = nullptr; } CHECK(!running_get_difference_); } } for (auto &update : updates) { if (update != nullptr) { int32 id = update->get_id(); if (id == telegram_api::updateNewMessage::ID || id == telegram_api::updateReadMessagesContents::ID || id == telegram_api::updateEditMessage::ID || id == telegram_api::updateDeleteMessages::ID || id == telegram_api::updateReadHistoryInbox::ID || id == telegram_api::updateReadHistoryOutbox::ID || id == telegram_api::updateWebPage::ID) { if (!downcast_call(*update, OnUpdate(this, update, false))) { LOG(ERROR) << "Can't call on some update received from " << source; } processed_updates++; update = nullptr; } } } if (running_get_difference_) { LOG(INFO) << "Postpone " << updates.size() << " updates [" << seq_begin << ", " << seq_end << "] with date = " << date << " from " << source; postponed_updates_.emplace(seq_begin, PendingUpdates(seq_begin, seq_end, date, std::move(updates))); return; } if (processed_updates == updates.size()) { if (seq_begin || seq_end) { LOG(ERROR) << "All updates from " << source << " was processed but seq = " << seq_ << ", seq_begin = " << seq_begin << ", seq_end = " << seq_end; } else { LOG(INFO) << "All updates was processed"; } 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)); process_pending_seq_updates(); return; } if (seq_begin <= seq_) { if (seq_end > seq_) { LOG(ERROR) << "Strange updates from " << source << " coming with seq_begin = " << seq_begin << ", seq_end = " << seq_end << ", but seq = " << seq_; } else { LOG(INFO) << "Old updates from " << source << " coming with seq_begin = " << seq_begin << ", seq_end = " << seq_end << ", but seq = " << seq_; } return; } 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_.find(seq_begin) != pending_seq_updates_.end()) << "Already have pending updates with seq = " << seq_begin << ", but receive it again from " << source; pending_seq_updates_.emplace(seq_begin, PendingUpdates(seq_begin, seq_end, date, std::move(updates))); set_seq_gap_timeout(MAX_UNFILLED_GAP_TIME); } void UpdatesManager::process_updates(vector> &&updates, bool force_apply) { tl_object_ptr update_pts_changed; /* 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), force_apply); } } } */ 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), force_apply); } // process updateNewScheduledMessage first if (constructor_id == telegram_api::updateNewScheduledMessage::ID) { on_update(move_tl_object_as(update), force_apply); } // updatePtsChanged forces get difference, so process it last if (constructor_id == telegram_api::updatePtsChanged::ID) { update_pts_changed = move_tl_object_as(update); } } } for (auto &update : updates) { if (update != nullptr) { LOG(INFO) << "Process update " << to_string(update); if (!downcast_call(*update, OnUpdate(this, update, force_apply))) { LOG(ERROR) << "Can't call on some update"; } CHECK(!running_get_difference_); } } if (update_pts_changed != nullptr) { on_update(std::move(update_pts_changed), force_apply); } } void UpdatesManager::process_seq_updates(int32 seq_end, int32 date, vector> &&updates) { string 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); if (seq_end) { seq_ = seq_end; } if (date && seq_end) { set_date(date, true, std::move(serialized_updates)); } } void UpdatesManager::process_pending_seq_updates() { while (!pending_seq_updates_.empty() && !running_get_difference_) { auto update_it = pending_seq_updates_.begin(); auto seq_begin = update_it->second.seq_begin; if (seq_begin > seq_ + 1) { break; } if (seq_begin == seq_ + 1) { process_seq_updates(update_it->second.seq_end, update_it->second.date, std::move(update_it->second.updates)); } else { // old update CHECK(seq_begin != 0); LOG_IF(ERROR, update_it->second.seq_end > seq_) << "Strange updates coming with seq_begin = " << seq_begin << ", seq_end = " << update_it->second.seq_end << ", but seq = " << seq_; } pending_seq_updates_.erase(update_it); } if (pending_seq_updates_.empty()) { seq_gap_timeout_.cancel_timeout(); } } void UpdatesManager::set_seq_gap_timeout(double timeout) { if (!seq_gap_timeout_.has_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::on_pending_update(tl_object_ptr update, int32 seq, const char *source) { vector> v; v.push_back(std::move(update)); on_pending_updates(std::move(v), seq, seq, 0, source); // TODO can be optimized } void UpdatesManager::on_update(tl_object_ptr update, bool force_apply) { CHECK(update != nullptr); int new_pts = update->pts_; int pts_count = update->pts_count_; td_->messages_manager_->add_pending_update(std::move(update), new_pts, pts_count, force_apply, "on_updateNewMessage"); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { CHECK(update != nullptr); td_->messages_manager_->on_update_new_channel_message(std::move(update)); } void UpdatesManager::on_update(tl_object_ptr update, bool force_apply) { CHECK(update != nullptr); if (!force_apply) { LOG(ERROR) << "Receive updateMessageID not in getDifference"; return; } LOG(INFO) << "Receive update about sent message " << to_string(update); td_->messages_manager_->on_update_message_id(update->random_id_, MessageId(ServerMessageId(update->id_)), "getDifference"); } void UpdatesManager::on_update(tl_object_ptr update, bool force_apply) { CHECK(update != nullptr); int new_pts = update->pts_; int pts_count = update->pts_count_; td_->messages_manager_->add_pending_update(std::move(update), new_pts, pts_count, force_apply, "on_updateReadMessagesContents"); } void UpdatesManager::on_update(tl_object_ptr update, bool force_apply) { CHECK(update != nullptr); int new_pts = update->pts_; int pts_count = update->pts_count_; td_->messages_manager_->add_pending_update(std::move(update), new_pts, pts_count, force_apply, "on_updateEditMessage"); } void UpdatesManager::on_update(tl_object_ptr update, bool force_apply) { CHECK(update != nullptr); int new_pts = update->pts_; int pts_count = update->pts_count_; if (update->messages_.empty()) { td_->messages_manager_->add_pending_update(make_tl_object(), new_pts, pts_count, force_apply, "on_updateDeleteMessages"); } else { td_->messages_manager_->add_pending_update(std::move(update), new_pts, pts_count, force_apply, "on_updateDeleteMessages"); } } void UpdatesManager::on_update(tl_object_ptr update, bool force_apply) { CHECK(update != nullptr); int new_pts = update->pts_; int pts_count = update->pts_count_; if (force_apply) { update->still_unread_count_ = -1; } td_->messages_manager_->add_pending_update(std::move(update), new_pts, pts_count, force_apply, "on_updateReadHistoryInbox"); } void UpdatesManager::on_update(tl_object_ptr update, bool force_apply) { CHECK(update != nullptr); int new_pts = update->pts_; int pts_count = update->pts_count_; td_->messages_manager_->add_pending_update(std::move(update), new_pts, pts_count, force_apply, "on_updateReadHistoryOutbox"); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { CHECK(update != nullptr); td_->messages_manager_->on_update_service_notification(std::move(update), true, Promise()); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { CHECK(update != nullptr); td_->messages_manager_->on_update_read_channel_inbox(std::move(update)); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { CHECK(update != nullptr); td_->messages_manager_->on_update_read_channel_outbox(std::move(update)); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { td_->messages_manager_->on_update_read_channel_messages_contents(std::move(update)); } void UpdatesManager::on_update(tl_object_ptr update, bool force_apply) { td_->messages_manager_->on_update_channel_too_long(std::move(update), force_apply); } void UpdatesManager::on_update(tl_object_ptr update, bool force_apply) { if (!force_apply) { td_->contacts_manager_->invalidate_channel_full(ChannelId(update->channel_id_), false, false); } } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { td_->messages_manager_->on_update_edit_channel_message(std::move(update)); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { ChannelId channel_id(update->channel_id_); if (!channel_id.is_valid()) { LOG(ERROR) << "Receive invalid " << channel_id; return; } DialogId dialog_id(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, "on_updateDeleteChannelMessages"); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { ChannelId channel_id(update->channel_id_); if (!channel_id.is_valid()) { LOG(ERROR) << "Receive invalid " << channel_id; return; } DialogId dialog_id(channel_id); td_->messages_manager_->on_update_message_views({dialog_id, MessageId(ServerMessageId(update->id_))}, update->views_); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { td_->messages_manager_->on_update_channel_max_unavailable_message_id( ChannelId(update->channel_id_), MessageId(ServerMessageId(update->available_min_id_))); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { td_->messages_manager_->on_update_dialog_pinned_message_id(DialogId(UserId(update->user_id_)), MessageId(ServerMessageId(update->id_))); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { td_->contacts_manager_->on_update_chat_pinned_message(ChatId(update->chat_id_), MessageId(ServerMessageId(update->id_)), update->version_); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { td_->messages_manager_->on_update_dialog_pinned_message_id(DialogId(ChannelId(update->channel_id_)), MessageId(ServerMessageId(update->id_))); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { CHECK(update != nullptr); switch (update->peer_->get_id()) { case telegram_api::notifyPeer::ID: { DialogId dialog_id(static_cast(update->peer_.get())->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: return td_->messages_manager_->on_update_scope_notify_settings(NotificationSettingsScope::Private, std::move(update->notify_settings_)); case telegram_api::notifyChats::ID: return td_->messages_manager_->on_update_scope_notify_settings(NotificationSettingsScope::Group, std::move(update->notify_settings_)); case telegram_api::notifyBroadcasts::ID: return td_->messages_manager_->on_update_scope_notify_settings(NotificationSettingsScope::Channel, std::move(update->notify_settings_)); default: UNREACHABLE(); } } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { td_->messages_manager_->on_get_peer_settings(DialogId(update->peer_), std::move(update->settings_)); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { td_->contacts_manager_->on_update_peer_located(std::move(update->peers_), true); } void UpdatesManager::on_update(tl_object_ptr update, bool force_apply) { CHECK(update != nullptr); td_->web_pages_manager_->on_get_web_page(std::move(update->webpage_), DialogId()); td_->messages_manager_->add_pending_update(make_tl_object(), update->pts_, update->pts_count_, force_apply, "on_updateWebPage"); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { CHECK(update != nullptr); td_->web_pages_manager_->on_get_web_page(std::move(update->webpage_), DialogId()); ChannelId channel_id(update->channel_id_); if (!channel_id.is_valid()) { LOG(ERROR) << "Receive invalid " << channel_id; return; } DialogId dialog_id(channel_id); td_->messages_manager_->add_pending_channel_update(dialog_id, make_tl_object(), update->pts_, update->pts_count_, "on_updateChannelWebPage"); } void UpdatesManager::on_update(tl_object_ptr update, bool force_apply) { CHECK(update != nullptr); 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); } td_->messages_manager_->add_pending_update(make_tl_object(), update->pts_, update->pts_count_, force_apply, "on_updateFolderPeers"); } 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; } tl_object_ptr UpdatesManager::convert_send_message_action( tl_object_ptr action) { auto fix_progress = [](int32 progress) { return progress <= 0 || progress > 100 ? 0 : progress; }; switch (action->get_id()) { case telegram_api::sendMessageCancelAction::ID: return make_tl_object(); case telegram_api::sendMessageTypingAction::ID: return make_tl_object(); case telegram_api::sendMessageRecordVideoAction::ID: return make_tl_object(); case telegram_api::sendMessageUploadVideoAction::ID: { auto upload_video_action = move_tl_object_as(action); return make_tl_object(fix_progress(upload_video_action->progress_)); } case telegram_api::sendMessageRecordAudioAction::ID: return make_tl_object(); case telegram_api::sendMessageUploadAudioAction::ID: { auto upload_audio_action = move_tl_object_as(action); return make_tl_object(fix_progress(upload_audio_action->progress_)); } case telegram_api::sendMessageUploadPhotoAction::ID: { auto upload_photo_action = move_tl_object_as(action); return make_tl_object(fix_progress(upload_photo_action->progress_)); } case telegram_api::sendMessageUploadDocumentAction::ID: { auto upload_document_action = move_tl_object_as(action); return make_tl_object(fix_progress(upload_document_action->progress_)); } case telegram_api::sendMessageGeoLocationAction::ID: return make_tl_object(); case telegram_api::sendMessageChooseContactAction::ID: return make_tl_object(); case telegram_api::sendMessageGamePlayAction::ID: return make_tl_object(); case telegram_api::sendMessageRecordRoundAction::ID: return make_tl_object(); case telegram_api::sendMessageUploadRoundAction::ID: { auto upload_round_action = move_tl_object_as(action); return make_tl_object(fix_progress(upload_round_action->progress_)); } default: UNREACHABLE(); return make_tl_object(); } } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { UserId user_id(update->user_id_); if (!td_->contacts_manager_->have_min_user(user_id)) { LOG(DEBUG) << "Ignore user typing of unknown " << user_id; return; } DialogId dialog_id(user_id); if (!td_->messages_manager_->have_dialog(dialog_id)) { LOG(DEBUG) << "Ignore user typing in unknown " << dialog_id; return; } td_->messages_manager_->on_user_dialog_action( dialog_id, user_id, convert_send_message_action(std::move(update->action_)), get_short_update_date()); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { UserId user_id(update->user_id_); if (!td_->contacts_manager_->have_min_user(user_id)) { LOG(DEBUG) << "Ignore user chat typing of unknown " << user_id; return; } ChatId chat_id(update->chat_id_); DialogId dialog_id(chat_id); if (!td_->messages_manager_->have_dialog(dialog_id)) { ChannelId channel_id(update->chat_id_); dialog_id = DialogId(channel_id); if (!td_->messages_manager_->have_dialog(dialog_id)) { LOG(DEBUG) << "Ignore user chat typing in unknown " << dialog_id; return; } } td_->messages_manager_->on_user_dialog_action( dialog_id, user_id, convert_send_message_action(std::move(update->action_)), get_short_update_date()); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { SecretChatId secret_chat_id(update->chat_id_); DialogId dialog_id(secret_chat_id); if (!td_->messages_manager_->have_dialog(dialog_id)) { LOG(DEBUG) << "Ignore secret chat typing in unknown " << dialog_id; return; } UserId user_id = td_->contacts_manager_->get_secret_chat_user_id(secret_chat_id); if (!td_->contacts_manager_->have_user_force(user_id)) { LOG(DEBUG) << "Ignore secret chat typing of unknown " << user_id; return; } td_->messages_manager_->on_user_dialog_action(dialog_id, user_id, make_tl_object(), get_short_update_date()); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { td_->contacts_manager_->on_update_user_online(UserId(update->user_id_), std::move(update->status_)); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { td_->contacts_manager_->on_update_user_name(UserId(update->user_id_), std::move(update->first_name_), std::move(update->last_name_), std::move(update->username_)); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { td_->contacts_manager_->on_update_user_phone_number(UserId(update->user_id_), std::move(update->phone_)); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { // TODO update->previous_, update->date_ td_->contacts_manager_->on_update_user_photo(UserId(update->user_id_), std::move(update->photo_)); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { td_->contacts_manager_->on_update_user_is_blocked(UserId(update->user_id_), update->blocked_); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { td_->contacts_manager_->on_get_chat_participants(std::move(update->participants_), true); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { td_->contacts_manager_->on_update_chat_add_user(ChatId(update->chat_id_), UserId(update->inviter_id_), UserId(update->user_id_), update->date_, update->version_); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { td_->contacts_manager_->on_update_chat_edit_administrator(ChatId(update->chat_id_), UserId(update->user_id_), update->is_admin_, update->version_); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { td_->contacts_manager_->on_update_chat_delete_user(ChatId(update->chat_id_), UserId(update->user_id_), update->version_); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { DialogId dialog_id(update->peer_); RestrictedRights permissions = get_restricted_rights(std::move(update->default_banned_rights_)); auto version = update->version_; switch (dialog_id.get_type()) { case DialogType::None: case DialogType::User: case DialogType::SecretChat: default: LOG(ERROR) << "Receive updateChatDefaultBannedRights in the " << dialog_id; return; case DialogType::Chat: return td_->contacts_manager_->on_update_chat_default_permissions(dialog_id.get_chat_id(), permissions, version); case DialogType::Channel: { LOG_IF(ERROR, version != 0) << "Receive version " << version << " in " << dialog_id; return td_->contacts_manager_->on_update_channel_default_permissions(dialog_id.get_channel_id(), permissions); } } } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { td_->messages_manager_->on_update_dialog_draft_message(DialogId(update->peer_), std::move(update->draft_)); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { FolderId folder_id(update->flags_ & telegram_api::updateDialogPinned::FOLDER_ID_MASK ? update->folder_id_ : 0); td_->messages_manager_->on_update_dialog_is_pinned( folder_id, DialogId(update->peer_), (update->flags_ & telegram_api::updateDialogPinned::PINNED_MASK) != 0); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { FolderId folder_id(update->flags_ & telegram_api::updatePinnedDialogs::FOLDER_ID_MASK ? update->folder_id_ : 0); td_->messages_manager_->on_update_pinned_dialogs(folder_id); // TODO use update->order_ } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { td_->messages_manager_->on_update_dialog_is_marked_as_unread( DialogId(update->peer_), (update->flags_ & telegram_api::updateDialogUnreadMark::UNREAD_MASK) != 0); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { send_closure(G()->config_manager(), &ConfigManager::on_dc_options_update, DcOptions(update->dc_options_)); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { td_->inline_queries_manager_->on_new_query(update->query_id_, UserId(update->user_id_), Location(update->geo_), update->query_, update->offset_); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { td_->inline_queries_manager_->on_chosen_result(UserId(update->user_id_), Location(update->geo_), update->query_, update->id_, std::move(update->msg_id_)); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { 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_)); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { 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_)); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { td_->stickers_manager_->reload_favorite_stickers(true); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { td_->animations_manager_->reload_saved_animations(true); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { send_closure(td_->config_manager_, &ConfigManager::request_config); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { set_pts(std::numeric_limits::max(), "updatePtsChanged").set_value(Unit()); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { send_closure(td_->secret_chats_manager_, &SecretChatsManager::on_update_chat, std::move(update)); } void UpdatesManager::on_update(tl_object_ptr update, bool force_apply) { send_closure(td_->secret_chats_manager_, &SecretChatsManager::on_update_message, std::move(update), force_apply); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { td_->messages_manager_->read_secret_chat_outbox(SecretChatId(update->chat_id_), update->max_date_, update->date_); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { send_closure(td_->privacy_manager_, &PrivacyManager::update_privacy, std::move(update)); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { td_->stickers_manager_->on_get_messages_sticker_set(StickerSetId(), std::move(update->stickerset_), true); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { td_->stickers_manager_->on_update_sticker_sets(); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { bool is_masks = (update->flags_ & telegram_api::updateStickerSetsOrder::MASKS_MASK) != 0; td_->stickers_manager_->on_update_sticker_sets_order(is_masks, StickersManager::convert_sticker_set_ids(update->order_)); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { td_->stickers_manager_->reload_featured_sticker_sets(true); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { td_->stickers_manager_->reload_recent_stickers(false, true); td_->stickers_manager_->reload_recent_stickers(true, true); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { UserId user_id(update->user_id_); if (!user_id.is_valid()) { LOG(ERROR) << "Receive shipping query from invalid " << user_id; return; } 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 } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { UserId user_id(update->user_id_); if (!user_id.is_valid()) { LOG(ERROR) << "Receive pre-checkout query from invalid " << user_id; return; } 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_))))); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { send_closure(G()->td(), &Td::send_update, make_tl_object(update->data_->data_)); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { send_closure(G()->td(), &Td::send_update, make_tl_object(update->query_id_, update->data_->data_, update->timeout_)); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { send_closure(G()->call_manager(), &CallManager::update_call, std::move(update)); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { td_->contacts_manager_->on_update_contacts_reset(); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { send_closure(G()->language_pack_manager(), &LanguagePackManager::on_language_pack_too_long, std::move(update->lang_code_)); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { send_closure(G()->language_pack_manager(), &LanguagePackManager::on_update_language_pack, std::move(update->difference_)); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { td_->poll_manager_->on_get_poll(PollId(update->poll_id_), std::move(update->poll_), std::move(update->results_)); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { td_->messages_manager_->on_get_message(std::move(update->message_), true, false, true, true, true, "updateNewScheduledMessage"); } void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { 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)); } // unsupported updates void UpdatesManager::on_update(tl_object_ptr update, bool /*force_apply*/) { } } // namespace td