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

Check permissions for all chat types in can_send_message_content.

Browse Source
This commit is contained in:
levlam 2021-04-13 00:49:27 +03:00
parent 21607d9a48
commit 9964d24ec8
1 changed files with 44 additions and 58 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

102
td/telegram/MessagesManager.cpp
View File

@ -23108,73 +23108,50 @@ Status MessagesManager::can_send_message_content(DialogId dialog_id, const Messa
secret_chat_layer = td_->contacts_manager_->get_secret_chat_layer(secret_chat_id); secret_chat_layer = td_->contacts_manager_->get_secret_chat_layer(secret_chat_id);
} }
bool can_send_messages = true;
bool can_send_media = true;
bool can_send_stickers = true;
bool can_send_animations = true;
bool can_send_games = true;
bool can_send_polls = true;
auto content_type = content->get_type(); auto content_type = content->get_type();
switch (dialog_type) { RestrictedRights permissions = [&] {
case DialogType::User: switch (dialog_type) {
if (content_type == MessageContentType::Poll && !is_forward && !td_->auth_manager_->is_bot() && case DialogType::User:
!td_->contacts_manager_->is_user_bot(dialog_id.get_user_id())) { return td_->contacts_manager_->get_user_default_permissions(dialog_id.get_user_id());
return Status::Error(400, "Polls can't be sent to the private chat"); case DialogType::Chat:
} return td_->contacts_manager_->get_chat_permissions(dialog_id.get_chat_id()).get_restricted_rights();
break; case DialogType::Channel:
case DialogType::Chat: return td_->contacts_manager_->get_channel_permissions(dialog_id.get_channel_id()).get_restricted_rights();
// ok case DialogType::SecretChat:
break; return td_->contacts_manager_->get_secret_chat_default_permissions(dialog_id.get_secret_chat_id());
case DialogType::Channel: { case DialogType::None:
auto channel_status = td_->contacts_manager_->get_channel_permissions(dialog_id.get_channel_id()); default:
can_send_messages = channel_status.can_send_messages(); UNREACHABLE();
can_send_media = channel_status.can_send_media(); return td_->contacts_manager_->get_user_default_permissions(UserId());
can_send_stickers = channel_status.can_send_stickers();
can_send_animations = channel_status.can_send_animations();
can_send_games = channel_status.can_send_games();
can_send_polls = channel_status.can_send_polls();
break;
} }
case DialogType::SecretChat: }();
if (content_type == MessageContentType::Game) {
return Status::Error(400, "Games can't be sent to secret chats");
}
if (content_type == MessageContentType::Poll) {
return Status::Error(400, "Polls can't be sent to secret chats");
}
if (content_type == MessageContentType::Dice) {
return Status::Error(400, "Dice can't be sent to secret chats");
}
break;
case DialogType::None:
default:
UNREACHABLE();
}
switch (content_type) { switch (content_type) {
case MessageContentType::Animation: case MessageContentType::Animation:
if (!can_send_animations) { if (!permissions.can_send_animations()) {
return Status::Error(400, "Not enough rights to send animations to the chat"); return Status::Error(400, "Not enough rights to send animations to the chat");
} }
break; break;
case MessageContentType::Audio: case MessageContentType::Audio:
if (!can_send_media) { if (!permissions.can_send_media()) {
return Status::Error(400, "Not enough rights to send audios to the chat"); return Status::Error(400, "Not enough rights to send audios to the chat");
} }
break; break;
case MessageContentType::Contact: case MessageContentType::Contact:
if (!can_send_messages) { if (!permissions.can_send_messages()) {
return Status::Error(400, "Not enough rights to send contacts to the chat"); return Status::Error(400, "Not enough rights to send contacts to the chat");
} }
break; break;
case MessageContentType::Dice: case MessageContentType::Dice:
if (!can_send_stickers) { if (!permissions.can_send_stickers()) {
return Status::Error(400, "Not enough rights to send dice to the chat"); return Status::Error(400, "Not enough rights to send dice to the chat");
} }
if (dialog_type == DialogType::SecretChat) {
return Status::Error(400, "Dice can't be sent to secret chats");
}
break; break;
case MessageContentType::Document: case MessageContentType::Document:
if (!can_send_media) { if (!permissions.can_send_media()) {
return Status::Error(400, "Not enough rights to send documents to the chat"); return Status::Error(400, "Not enough rights to send documents to the chat");
} }
break; break;
@ -23182,8 +23159,10 @@ Status MessagesManager::can_send_message_content(DialogId dialog_id, const Messa
if (is_broadcast_channel(dialog_id)) { if (is_broadcast_channel(dialog_id)) {
// return Status::Error(400, "Games can't be sent to channel chats"); // return Status::Error(400, "Games can't be sent to channel chats");
} }
if (dialog_type == DialogType::SecretChat) {
if (!can_send_games) { return Status::Error(400, "Games can't be sent to secret chats");
}
if (!permissions.can_send_games()) {
return Status::Error(400, "Not enough rights to send games to the chat"); return Status::Error(400, "Not enough rights to send games to the chat");
} }
break; break;
@ -23204,50 +23183,57 @@ Status MessagesManager::can_send_message_content(DialogId dialog_id, const Messa
} }
break; break;
case MessageContentType::LiveLocation: case MessageContentType::LiveLocation:
if (!can_send_messages) { if (!permissions.can_send_messages()) {
return Status::Error(400, "Not enough rights to send live locations to the chat"); return Status::Error(400, "Not enough rights to send live locations to the chat");
} }
break; break;
case MessageContentType::Location: case MessageContentType::Location:
if (!can_send_messages) { if (!permissions.can_send_messages()) {
return Status::Error(400, "Not enough rights to send locations to the chat"); return Status::Error(400, "Not enough rights to send locations to the chat");
} }
break; break;
case MessageContentType::Photo: case MessageContentType::Photo:
if (!can_send_media) { if (!permissions.can_send_media()) {
return Status::Error(400, "Not enough rights to send photos to the chat"); return Status::Error(400, "Not enough rights to send photos to the chat");
} }
break; break;
case MessageContentType::Poll: case MessageContentType::Poll:
if (!can_send_polls) { if (!permissions.can_send_polls()) {
return Status::Error(400, "Not enough rights to send polls to the chat"); return Status::Error(400, "Not enough rights to send polls to the chat");
} }
if (!get_message_content_poll_is_anonymous(td_, content) && is_broadcast_channel(dialog_id)) { if (!get_message_content_poll_is_anonymous(td_, content) && is_broadcast_channel(dialog_id)) {
return Status::Error(400, "Non-anonymous polls can't be sent to channel chats"); return Status::Error(400, "Non-anonymous polls can't be sent to channel chats");
} }
if (dialog_type == DialogType::User && !is_forward && !td_->auth_manager_->is_bot() &&
!td_->contacts_manager_->is_user_bot(dialog_id.get_user_id())) {
return Status::Error(400, "Polls can't be sent to the private chat");
}
if (dialog_type == DialogType::SecretChat) {
return Status::Error(400, "Polls can't be sent to secret chats");
}
break; break;
case MessageContentType::Sticker: case MessageContentType::Sticker:
if (!can_send_stickers) { if (!permissions.can_send_stickers()) {
return Status::Error(400, "Not enough rights to send stickers to the chat"); return Status::Error(400, "Not enough rights to send stickers to the chat");
} }
break; break;
case MessageContentType::Text: case MessageContentType::Text:
if (!can_send_messages) { if (!permissions.can_send_messages()) {
return Status::Error(400, "Not enough rights to send text messages to the chat"); return Status::Error(400, "Not enough rights to send text messages to the chat");
} }
break; break;
case MessageContentType::Venue: case MessageContentType::Venue:
if (!can_send_messages) { if (!permissions.can_send_messages()) {
return Status::Error(400, "Not enough rights to send venues to the chat"); return Status::Error(400, "Not enough rights to send venues to the chat");
} }
break; break;
case MessageContentType::Video: case MessageContentType::Video:
if (!can_send_media) { if (!permissions.can_send_media()) {
return Status::Error(400, "Not enough rights to send videos to the chat"); return Status::Error(400, "Not enough rights to send videos to the chat");
} }
break; break;
case MessageContentType::VideoNote: case MessageContentType::VideoNote:
if (!can_send_media) { if (!permissions.can_send_media()) {
return Status::Error(400, "Not enough rights to send video notes to the chat"); return Status::Error(400, "Not enough rights to send video notes to the chat");
} }
if (secret_chat_layer < SecretChatActor::VIDEO_NOTES_LAYER) { if (secret_chat_layer < SecretChatActor::VIDEO_NOTES_LAYER) {
@ -23256,7 +23242,7 @@ Status MessagesManager::can_send_message_content(DialogId dialog_id, const Messa
} }
break; break;
case MessageContentType::VoiceNote: case MessageContentType::VoiceNote:
if (!can_send_media) { if (!permissions.can_send_media()) {
return Status::Error(400, "Not enough rights to send voice notes to the chat"); return Status::Error(400, "Not enough rights to send voice notes to the chat");
} }
break; break;