1935 lines
79 KiB
C++
1935 lines
79 KiB
C++
//
|
|
// 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 <limits>
|
|
|
|
namespace td {
|
|
|
|
int VERBOSITY_NAME(get_difference) = VERBOSITY_NAME(INFO);
|
|
|
|
class OnUpdate {
|
|
UpdatesManager *manager_;
|
|
tl_object_ptr<telegram_api::Update> &update_;
|
|
bool force_apply_;
|
|
|
|
public:
|
|
OnUpdate(UpdatesManager *manager, tl_object_ptr<telegram_api::Update> &update, bool force_apply)
|
|
: manager_(manager), update_(update), force_apply_(force_apply) {
|
|
}
|
|
|
|
template <class T>
|
|
void operator()(T &obj) const {
|
|
CHECK(&*update_ == &obj);
|
|
manager_->on_update(move_tl_object_as<T>(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<telegram_api::updates_getState>(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<telegram_api::updates_getState>(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<telegram_api::updates_getDifference>(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<int32>::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 *>(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<GetDifferenceQuery>()->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<int32>::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<int32>::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<int32>::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<tl_object_ptr<telegram_api::MessageEntity>> &message_entities) const {
|
|
for (auto &entity : message_entities) {
|
|
if (entity->get_id() == telegram_api::messageEntityMentionName::ID) {
|
|
auto entity_mention_name = static_cast<const telegram_api::messageEntityMentionName *>(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<telegram_api::messageFwdHeader> &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<const telegram_api::message *>(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<const telegram_api::messageMediaContact *>(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<const telegram_api::messageMediaWebPage *>(message->media_.get());
|
|
if (message_media_web_page->webpage_->get_id() == telegram_api::webPage::ID) {
|
|
auto web_page = static_cast<const telegram_api::webPage *>(message_media_web_page->webpage_.get());
|
|
if (web_page->cached_page_ != nullptr) {
|
|
const vector<tl_object_ptr<telegram_api::PageBlock>> *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<const telegram_api::pageBlockChannel *>(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<const telegram_api::messageService *>(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<const telegram_api::messageActionChatCreate *>(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<const telegram_api::messageActionChatAddUser *>(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<const telegram_api::messageActionChatJoinedByLink *>(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<const telegram_api::messageActionChatDeleteUser *>(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<const telegram_api::messageActionChatMigrateTo *>(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<const telegram_api::messageActionChannelMigrateFrom *>(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<const telegram_api::updateNewMessage *>(update)->message_.get();
|
|
}
|
|
if (id == telegram_api::updateNewChannelMessage::ID) {
|
|
message = static_cast<const telegram_api::updateNewChannelMessage *>(update)->message_.get();
|
|
}
|
|
if (id == telegram_api::updateEditMessage::ID) {
|
|
message = static_cast<const telegram_api::updateEditMessage *>(update)->message_.get();
|
|
}
|
|
if (id == telegram_api::updateEditChannelMessage::ID) {
|
|
message = static_cast<const telegram_api::updateEditChannelMessage *>(update)->message_.get();
|
|
}
|
|
if (message != nullptr) {
|
|
return is_acceptable_message(message);
|
|
}
|
|
|
|
if (id == telegram_api::updateDraftMessage::ID) {
|
|
auto update_draft_message = static_cast<const telegram_api::updateDraftMessage *>(update);
|
|
CHECK(update_draft_message->draft_ != nullptr);
|
|
if (update_draft_message->draft_->get_id() == telegram_api::draftMessage::ID) {
|
|
auto draft_message = static_cast<const telegram_api::draftMessage *>(update_draft_message->draft_.get());
|
|
return is_acceptable_message_entities(draft_message->entities_);
|
|
}
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
void UpdatesManager::on_get_updates(tl_object_ptr<telegram_api::Updates> &&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<telegram_api::updateShortMessage>(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<telegram_api::updateNewMessage>(
|
|
make_tl_object<telegram_api::message>(
|
|
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<telegram_api::peerUser>(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<telegram_api::updateShortChatMessage>(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<telegram_api::updateNewMessage>(
|
|
make_tl_object<telegram_api::message>(
|
|
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<telegram_api::peerChat>(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<telegram_api::updateShort>(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<telegram_api::updatesCombined>(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<telegram_api::updates>(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<void *>(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<telegram_api::updates_state> &&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<int32>::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<tl_object_ptr<telegram_api::Update>> *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<const telegram_api::updatesCombined *>(updates_ptr)->updates_;
|
|
case telegram_api::updates::ID:
|
|
return &static_cast<const telegram_api::updates *>(updates_ptr)->updates_;
|
|
default:
|
|
UNREACHABLE();
|
|
return nullptr;
|
|
}
|
|
}
|
|
|
|
std::unordered_set<int64> UpdatesManager::get_sent_messages_random_ids(const telegram_api::Updates *updates_ptr) {
|
|
std::unordered_set<int64> 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<const telegram_api::updateMessageID *>(update.get())->random_id_;
|
|
if (!random_ids.insert(random_id).second) {
|
|
LOG(ERROR) << "Receive twice updateMessageID for " << random_id;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
return random_ids;
|
|
}
|
|
|
|
vector<const tl_object_ptr<telegram_api::Message> *> UpdatesManager::get_new_messages(
|
|
const telegram_api::Updates *updates_ptr) {
|
|
vector<const tl_object_ptr<telegram_api::Message> *> 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<const telegram_api::updateNewMessage *>(update.get())->message_);
|
|
} else if (constructor_id == telegram_api::updateNewChannelMessage::ID) {
|
|
messages.emplace_back(&static_cast<const telegram_api::updateNewChannelMessage *>(update.get())->message_);
|
|
}
|
|
}
|
|
}
|
|
return messages;
|
|
}
|
|
|
|
vector<DialogId> UpdatesManager::get_update_notify_settings_dialog_ids(const telegram_api::Updates *updates_ptr) {
|
|
vector<DialogId> 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<const telegram_api::updateNotifySettings *>(update.get())->peer_.get();
|
|
if (notify_peer->get_id() == telegram_api::notifyPeer::ID) {
|
|
dialog_id = DialogId(static_cast<const telegram_api::notifyPeer *>(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<DialogId> UpdatesManager::get_chat_dialog_ids(const telegram_api::Updates *updates_ptr) {
|
|
const vector<tl_object_ptr<telegram_api::Chat>> *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<const telegram_api::updatesCombined *>(updates_ptr)->chats_;
|
|
break;
|
|
}
|
|
case telegram_api::updates::ID: {
|
|
chats = &static_cast<const telegram_api::updates *>(updates_ptr)->chats_;
|
|
break;
|
|
}
|
|
default:
|
|
UNREACHABLE();
|
|
}
|
|
|
|
if (chats == nullptr) {
|
|
return {};
|
|
}
|
|
|
|
vector<DialogId> 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<GetUpdatesStateQuery>()->send();
|
|
}
|
|
return;
|
|
}
|
|
pts_manager_.init(to_integer<int32>(pts_str));
|
|
last_get_difference_pts_ = get_pts();
|
|
qts_ = to_integer<int32>(pmc->get("updates.qts"));
|
|
date_ = to_integer<int32>(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<PingServerQuery>()->send();
|
|
}
|
|
|
|
void UpdatesManager::on_server_pong(tl_object_ptr<telegram_api::updates_state> &&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<tl_object_ptr<telegram_api::Message>> &&new_messages,
|
|
vector<tl_object_ptr<telegram_api::EncryptedMessage>> &&new_encrypted_messages, int32 qts,
|
|
vector<tl_object_ptr<telegram_api::Update>> &&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) {
|
|
on_update(move_tl_object_as<telegram_api::updateMessageID>(update), true);
|
|
CHECK(!running_get_difference_);
|
|
}
|
|
|
|
if (constructor_id == telegram_api::updateEncryption::ID) {
|
|
on_update(move_tl_object_as<telegram_api::updateEncryption>(update), true);
|
|
CHECK(!running_get_difference_);
|
|
}
|
|
|
|
if (constructor_id == telegram_api::updateFolderPeers::ID) {
|
|
on_update(move_tl_object_as<telegram_api::updateFolderPeers>(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<telegram_api::updateReadHistoryInbox>(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<telegram_api::updateNewEncryptedMessage>(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<telegram_api::updates_Difference> &&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<telegram_api::updates_differenceEmpty>(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<telegram_api::updates_difference>(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<telegram_api::updates_differenceSlice>(difference_ptr);
|
|
if (difference->intermediate_state_->pts_ >= get_pts() && get_pts() != std::numeric_limits<int32>::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<telegram_api::updates_differenceTooLong>(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<tl_object_ptr<telegram_api::Update>> &&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<telegram_api::Message> *message_ptr = nullptr;
|
|
int32 pts = 0;
|
|
if (id == telegram_api::updateNewChannelMessage::ID) {
|
|
auto update_new_channel_message = static_cast<const telegram_api::updateNewChannelMessage *>(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<const telegram_api::updateEditChannelMessage *>(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>(
|
|
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<const telegram_api::updateReadHistoryOutbox *>(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);
|
|
}
|
|
}
|
|
}
|
|
|
|
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<telegram_api::updateMessageID>(update);
|
|
if (!td_->messages_manager_->on_update_message_id(
|
|
sent_message_update->random_id_, MessageId(ServerMessageId(sent_message_update->id_)), source)) {
|
|
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<telegram_api::updateFolderPeers>(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<tl_object_ptr<telegram_api::Update>> &&updates, bool force_apply) {
|
|
tl_object_ptr<telegram_api::updatePtsChanged> 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<telegram_api::updateReadChannelInbox>(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<telegram_api::updateNewChannelMessage>(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<telegram_api::updatePtsChanged>(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<tl_object_ptr<telegram_api::Update>> &&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<void *>(td_));
|
|
seq_gap_timeout_.set_timeout_in(timeout);
|
|
}
|
|
}
|
|
|
|
void UpdatesManager::on_pending_update(tl_object_ptr<telegram_api::Update> update, int32 seq, const char *source) {
|
|
vector<tl_object_ptr<telegram_api::Update>> 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<telegram_api::updateNewMessage> 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<telegram_api::updateNewChannelMessage> 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<telegram_api::updateMessageID> 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<telegram_api::updateReadMessagesContents> 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<telegram_api::updateEditMessage> 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<telegram_api::updateDeleteMessages> 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<dummyUpdate>(), 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<telegram_api::updateReadHistoryInbox> 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_updateReadHistoryInbox");
|
|
}
|
|
|
|
void UpdatesManager::on_update(tl_object_ptr<telegram_api::updateReadHistoryOutbox> 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<telegram_api::updateServiceNotification> update, bool /*force_apply*/) {
|
|
CHECK(update != nullptr);
|
|
td_->messages_manager_->on_update_service_notification(std::move(update), true, Promise<Unit>());
|
|
}
|
|
|
|
void UpdatesManager::on_update(tl_object_ptr<telegram_api::updateReadChannelInbox> 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<telegram_api::updateReadChannelOutbox> 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<telegram_api::updateChannelReadMessagesContents> update,
|
|
bool /*force_apply*/) {
|
|
td_->messages_manager_->on_update_read_channel_messages_contents(std::move(update));
|
|
}
|
|
|
|
void UpdatesManager::on_update(tl_object_ptr<telegram_api::updateChannelTooLong> update, bool force_apply) {
|
|
td_->messages_manager_->on_update_channel_too_long(std::move(update), force_apply);
|
|
}
|
|
|
|
void UpdatesManager::on_update(tl_object_ptr<telegram_api::updateChannel> 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<telegram_api::updateEditChannelMessage> update, bool /*force_apply*/) {
|
|
td_->messages_manager_->on_update_edit_channel_message(std::move(update));
|
|
}
|
|
|
|
void UpdatesManager::on_update(tl_object_ptr<telegram_api::updateDeleteChannelMessages> 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<telegram_api::updateChannelMessageViews> 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<telegram_api::updateChannelAvailableMessages> 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<telegram_api::updateUserPinnedMessage> 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<telegram_api::updateChatPinnedMessage> 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<telegram_api::updateChannelPinnedMessage> 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<telegram_api::updateNotifySettings> update, bool /*force_apply*/) {
|
|
CHECK(update != nullptr);
|
|
switch (update->peer_->get_id()) {
|
|
case telegram_api::notifyPeer::ID: {
|
|
DialogId dialog_id(static_cast<const telegram_api::notifyPeer *>(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<telegram_api::updatePeerSettings> 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<telegram_api::updatePeerLocated> update, bool /*force_apply*/) {
|
|
td_->contacts_manager_->on_update_peer_located(std::move(update->peers_), true);
|
|
}
|
|
|
|
void UpdatesManager::on_update(tl_object_ptr<telegram_api::updateWebPage> 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<dummyUpdate>(), update->pts_, update->pts_count_,
|
|
force_apply, "on_updateWebPage");
|
|
}
|
|
|
|
void UpdatesManager::on_update(tl_object_ptr<telegram_api::updateChannelWebPage> 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<dummyUpdate>(), update->pts_,
|
|
update->pts_count_, "on_updateChannelWebPage");
|
|
}
|
|
|
|
void UpdatesManager::on_update(tl_object_ptr<telegram_api::updateFolderPeers> 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<dummyUpdate>(), 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<td_api::ChatAction> UpdatesManager::convert_send_message_action(
|
|
tl_object_ptr<telegram_api::SendMessageAction> 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<td_api::chatActionCancel>();
|
|
case telegram_api::sendMessageTypingAction::ID:
|
|
return make_tl_object<td_api::chatActionTyping>();
|
|
case telegram_api::sendMessageRecordVideoAction::ID:
|
|
return make_tl_object<td_api::chatActionRecordingVideo>();
|
|
case telegram_api::sendMessageUploadVideoAction::ID: {
|
|
auto upload_video_action = move_tl_object_as<telegram_api::sendMessageUploadVideoAction>(action);
|
|
return make_tl_object<td_api::chatActionUploadingVideo>(fix_progress(upload_video_action->progress_));
|
|
}
|
|
case telegram_api::sendMessageRecordAudioAction::ID:
|
|
return make_tl_object<td_api::chatActionRecordingVoiceNote>();
|
|
case telegram_api::sendMessageUploadAudioAction::ID: {
|
|
auto upload_audio_action = move_tl_object_as<telegram_api::sendMessageUploadAudioAction>(action);
|
|
return make_tl_object<td_api::chatActionUploadingVoiceNote>(fix_progress(upload_audio_action->progress_));
|
|
}
|
|
case telegram_api::sendMessageUploadPhotoAction::ID: {
|
|
auto upload_photo_action = move_tl_object_as<telegram_api::sendMessageUploadPhotoAction>(action);
|
|
return make_tl_object<td_api::chatActionUploadingPhoto>(fix_progress(upload_photo_action->progress_));
|
|
}
|
|
case telegram_api::sendMessageUploadDocumentAction::ID: {
|
|
auto upload_document_action = move_tl_object_as<telegram_api::sendMessageUploadDocumentAction>(action);
|
|
return make_tl_object<td_api::chatActionUploadingDocument>(fix_progress(upload_document_action->progress_));
|
|
}
|
|
case telegram_api::sendMessageGeoLocationAction::ID:
|
|
return make_tl_object<td_api::chatActionChoosingLocation>();
|
|
case telegram_api::sendMessageChooseContactAction::ID:
|
|
return make_tl_object<td_api::chatActionChoosingContact>();
|
|
case telegram_api::sendMessageGamePlayAction::ID:
|
|
return make_tl_object<td_api::chatActionStartPlayingGame>();
|
|
case telegram_api::sendMessageRecordRoundAction::ID:
|
|
return make_tl_object<td_api::chatActionRecordingVideoNote>();
|
|
case telegram_api::sendMessageUploadRoundAction::ID: {
|
|
auto upload_round_action = move_tl_object_as<telegram_api::sendMessageUploadRoundAction>(action);
|
|
return make_tl_object<td_api::chatActionUploadingVideoNote>(fix_progress(upload_round_action->progress_));
|
|
}
|
|
default:
|
|
UNREACHABLE();
|
|
return make_tl_object<td_api::chatActionTyping>();
|
|
}
|
|
}
|
|
|
|
void UpdatesManager::on_update(tl_object_ptr<telegram_api::updateUserTyping> 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<telegram_api::updateChatUserTyping> 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<telegram_api::updateEncryptedChatTyping> 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<td_api::chatActionTyping>(),
|
|
get_short_update_date());
|
|
}
|
|
|
|
void UpdatesManager::on_update(tl_object_ptr<telegram_api::updateUserStatus> 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<telegram_api::updateUserName> 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<telegram_api::updateUserPhone> 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<telegram_api::updateUserPhoto> 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<telegram_api::updateUserBlocked> 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<telegram_api::updateChatParticipants> update, bool /*force_apply*/) {
|
|
td_->contacts_manager_->on_get_chat_participants(std::move(update->participants_), true);
|
|
}
|
|
|
|
void UpdatesManager::on_update(tl_object_ptr<telegram_api::updateChatParticipantAdd> 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<telegram_api::updateChatParticipantAdmin> 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<telegram_api::updateChatParticipantDelete> 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<telegram_api::updateChatDefaultBannedRights> 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<telegram_api::updateDraftMessage> 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<telegram_api::updateDialogPinned> 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<telegram_api::updatePinnedDialogs> 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<telegram_api::updateDialogUnreadMark> 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<telegram_api::updateDcOptions> 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<telegram_api::updateBotInlineQuery> 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<telegram_api::updateBotInlineSend> 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<telegram_api::updateBotCallbackQuery> 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<telegram_api::updateInlineBotCallbackQuery> 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<telegram_api::updateFavedStickers> update, bool /*force_apply*/) {
|
|
td_->stickers_manager_->reload_favorite_stickers(true);
|
|
}
|
|
|
|
void UpdatesManager::on_update(tl_object_ptr<telegram_api::updateSavedGifs> update, bool /*force_apply*/) {
|
|
td_->animations_manager_->reload_saved_animations(true);
|
|
}
|
|
|
|
void UpdatesManager::on_update(tl_object_ptr<telegram_api::updateConfig> update, bool /*force_apply*/) {
|
|
send_closure(td_->config_manager_, &ConfigManager::request_config);
|
|
}
|
|
|
|
void UpdatesManager::on_update(tl_object_ptr<telegram_api::updatePtsChanged> update, bool /*force_apply*/) {
|
|
set_pts(std::numeric_limits<int32>::max(), "updatePtsChanged").set_value(Unit());
|
|
}
|
|
|
|
void UpdatesManager::on_update(tl_object_ptr<telegram_api::updateEncryption> update, bool /*force_apply*/) {
|
|
send_closure(td_->secret_chats_manager_, &SecretChatsManager::on_update_chat, std::move(update));
|
|
}
|
|
|
|
void UpdatesManager::on_update(tl_object_ptr<telegram_api::updateNewEncryptedMessage> 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<telegram_api::updateEncryptedMessagesRead> 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<telegram_api::updatePrivacy> update, bool /*force_apply*/) {
|
|
send_closure(td_->privacy_manager_, &PrivacyManager::update_privacy, std::move(update));
|
|
}
|
|
|
|
void UpdatesManager::on_update(tl_object_ptr<telegram_api::updateNewStickerSet> 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<telegram_api::updateStickerSets> update, bool /*force_apply*/) {
|
|
td_->stickers_manager_->on_update_sticker_sets();
|
|
}
|
|
|
|
void UpdatesManager::on_update(tl_object_ptr<telegram_api::updateStickerSetsOrder> 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<telegram_api::updateReadFeaturedStickers> update, bool /*force_apply*/) {
|
|
td_->stickers_manager_->reload_featured_sticker_sets(true);
|
|
}
|
|
|
|
void UpdatesManager::on_update(tl_object_ptr<telegram_api::updateRecentStickers> 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<telegram_api::updateBotShippingQuery> 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<td_api::updateNewShippingQuery>(
|
|
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<telegram_api::updateBotPrecheckoutQuery> 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<td_api::updateNewPreCheckoutQuery>(
|
|
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<telegram_api::updateBotWebhookJSON> update, bool /*force_apply*/) {
|
|
send_closure(G()->td(), &Td::send_update, make_tl_object<td_api::updateNewCustomEvent>(update->data_->data_));
|
|
}
|
|
|
|
void UpdatesManager::on_update(tl_object_ptr<telegram_api::updateBotWebhookJSONQuery> update, bool /*force_apply*/) {
|
|
send_closure(G()->td(), &Td::send_update,
|
|
make_tl_object<td_api::updateNewCustomQuery>(update->query_id_, update->data_->data_, update->timeout_));
|
|
}
|
|
|
|
void UpdatesManager::on_update(tl_object_ptr<telegram_api::updatePhoneCall> update, bool /*force_apply*/) {
|
|
send_closure(G()->call_manager(), &CallManager::update_call, std::move(update));
|
|
}
|
|
|
|
void UpdatesManager::on_update(tl_object_ptr<telegram_api::updateContactsReset> update, bool /*force_apply*/) {
|
|
td_->contacts_manager_->on_update_contacts_reset();
|
|
}
|
|
|
|
void UpdatesManager::on_update(tl_object_ptr<telegram_api::updateLangPackTooLong> 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<telegram_api::updateLangPack> 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<telegram_api::updateMessagePoll> 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<telegram_api::updateNewScheduledMessage> 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<telegram_api::updateDeleteScheduledMessages> update,
|
|
bool /*force_apply*/) {
|
|
vector<ScheduledServerMessageId> 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<telegram_api::updateTheme> update, bool /*force_apply*/) {
|
|
}
|
|
|
|
} // namespace td
|