tdlight-team/tdlight
tdlight-team/tdlight
1
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add UpdatesManager::is_acceptable_message_media.

Browse Source
This commit is contained in:
levlam 2023-10-14 22:48:16 +03:00
parent 7a6d6cf492
commit 90fd69b202
2 changed files with 62 additions and 55 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

115
td/telegram/UpdatesManager.cpp
View File

@ -807,6 +807,64 @@ bool UpdatesManager::is_acceptable_message_forward_header(
return true; return true;
} }
bool UpdatesManager::is_acceptable_message_media(
const telegram_api::object_ptr<telegram_api::MessageMedia> &media_ptr) const {
if (media_ptr == nullptr) {
return true;
}
switch (media_ptr->get_id()) {
case telegram_api::messageMediaContact::ID: {
auto message_media = static_cast<const telegram_api::messageMediaContact *>(media_ptr.get());
UserId user_id(message_media->user_id_);
return user_id == UserId() || is_acceptable_user(user_id);
}
case telegram_api::messageMediaStory::ID: {
auto message_media = static_cast<const telegram_api::messageMediaStory *>(media_ptr.get());
return is_acceptable_peer(message_media->peer_);
}
case telegram_api::messageMediaPoll::ID:
/*
// the users and chats are always min, so no need to check
auto message_media_poll = static_cast<const telegram_api::messageMediaPoll *>(media_ptr.get());
for (auto recent_voter : message_media_poll->results_->recent_voters_) {
if (!is_acceptable_peer(recent_voter)) {
return false;
}
}
*/
return true;
case telegram_api::messageMediaWebPage::ID:
/*
// the channel is always min, so no need to check
auto message_media_web_page = static_cast<const telegram_api::messageMediaWebPage *>(media_ptr.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_);
}
}
}
}
}
*/
return true;
default:
return true;
}
}
bool UpdatesManager::is_acceptable_message(const telegram_api::Message *message_ptr) const { bool UpdatesManager::is_acceptable_message(const telegram_api::Message *message_ptr) const {
CHECK(message_ptr != nullptr); CHECK(message_ptr != nullptr);
int32 constructor_id = message_ptr->get_id(); int32 constructor_id = message_ptr->get_id();
@ -840,61 +898,8 @@ bool UpdatesManager::is_acceptable_message(const telegram_api::Message *message_
return false; return false;
} }
if (message->media_ != nullptr) { if (!is_acceptable_message_media(message->media_)) {
auto media_id = message->media_->get_id(); return false;
if (media_id == telegram_api::messageMediaContact::ID) {
auto message_media = static_cast<const telegram_api::messageMediaContact *>(message->media_.get());
UserId user_id(message_media->user_id_);
if (user_id != UserId() && !is_acceptable_user(user_id)) {
return false;
}
}
if (media_id == telegram_api::messageMediaStory::ID) {
auto message_media = static_cast<const telegram_api::messageMediaStory *>(message->media_.get());
if (!is_acceptable_peer(message_media->peer_)) {
return false;
}
}
/*
// the users and chats are always min, so no need to check
if (media_id == telegram_api::messageMediaPoll::ID) {
auto message_media_poll = static_cast<const telegram_api::messageMediaPoll *>(message->media_.get());
for (auto recent_voter : message_media_poll->results_->recent_voters_) {
if (!is_acceptable_peer(recent_voter)) {
return false;
}
}
}
*/
/*
// the channel is always min, so no need to check
if (media_id == telegram_api::messageMediaWebPage::ID) {
auto message_media_web_page = static_cast<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);
} }
/* /*

2
td/telegram/UpdatesManager.h
View File

@ -461,6 +461,8 @@ class UpdatesManager final : public Actor {
bool is_acceptable_message_forward_header( bool is_acceptable_message_forward_header(
const telegram_api::object_ptr<telegram_api::messageFwdHeader> &header) const; const telegram_api::object_ptr<telegram_api::messageFwdHeader> &header) const;
bool is_acceptable_message_media(const telegram_api::object_ptr<telegram_api::MessageMedia> &media_ptr) const;
bool is_acceptable_message(const telegram_api::Message *message_ptr) const; bool is_acceptable_message(const telegram_api::Message *message_ptr) const;
bool is_acceptable_update(const telegram_api::Update *update) const; bool is_acceptable_update(const telegram_api::Update *update) const;