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

Don't send more than 100 user entities to server.

Browse Source
This commit is contained in:
levlam 2023-01-08 23:07:57 +03:00
parent d0500988ac
commit 1c037ff0e0
1 changed files with 16 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
td/telegram/MessageEntity.cpp
View File

@ -4398,6 +4398,8 @@ vector<tl_object_ptr<telegram_api::MessageEntity>> get_input_message_entities(co
const char *source) { const char *source) {
vector<tl_object_ptr<telegram_api::MessageEntity>> result; vector<tl_object_ptr<telegram_api::MessageEntity>> result;
vector<MessageEntity> splittable_entities; vector<MessageEntity> splittable_entities;
constexpr size_t MAX_USER_ENTITY_COUNT = 100; // server-side limit
size_t user_entity_count = 0;
for (auto &entity : entities) { for (auto &entity : entities) {
if (!is_user_entity(entity.type)) { if (!is_user_entity(entity.type)) {
continue; continue;
@ -4406,6 +4408,15 @@ vector<tl_object_ptr<telegram_api::MessageEntity>> get_input_message_entities(co
splittable_entities.push_back(entity); splittable_entities.push_back(entity);
continue; continue;
} }
if (entity.type == MessageEntity::Type::CustomEmoji) {
result.push_back(make_tl_object<telegram_api::messageEntityCustomEmoji>(entity.offset, entity.length,
entity.custom_emoji_id.get()));
continue;
}
if (user_entity_count >= MAX_USER_ENTITY_COUNT) {
continue;
}
user_entity_count++;
switch (entity.type) { switch (entity.type) {
case MessageEntity::Type::BlockQuote: case MessageEntity::Type::BlockQuote:
result.push_back(make_tl_object<telegram_api::messageEntityBlockquote>(entity.offset, entity.length)); result.push_back(make_tl_object<telegram_api::messageEntityBlockquote>(entity.offset, entity.length));
@ -4430,16 +4441,17 @@ vector<tl_object_ptr<telegram_api::MessageEntity>> get_input_message_entities(co
r_input_user.move_as_ok())); r_input_user.move_as_ok()));
break; break;
} }
case MessageEntity::Type::CustomEmoji:
result.push_back(make_tl_object<telegram_api::messageEntityCustomEmoji>(entity.offset, entity.length,
entity.custom_emoji_id.get()));
break;
default: default:
UNREACHABLE(); UNREACHABLE();
} }
} }
split_entities(splittable_entities, vector<MessageEntity>()); split_entities(splittable_entities, vector<MessageEntity>());
for (auto &entity : splittable_entities) { for (auto &entity : splittable_entities) {
if (user_entity_count >= MAX_USER_ENTITY_COUNT) {
break;
}
user_entity_count++;
switch (entity.type) { switch (entity.type) {
case MessageEntity::Type::Bold: case MessageEntity::Type::Bold:
result.push_back(make_tl_object<telegram_api::messageEntityBold>(entity.offset, entity.length)); result.push_back(make_tl_object<telegram_api::messageEntityBold>(entity.offset, entity.length));