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

Add push content basic checks.

Browse Source 
GitOrigin-RevId: f6c436f578b5eac8799bb5f84ac2545c490814b3
This commit is contained in:
levlam 2019-03-26 16:23:24 +03:00
parent 89e5b00a95
commit ac5a716ffe
2 changed files with 39 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

37
td/telegram/NotificationManager.cpp
View File

@ -2431,16 +2431,45 @@ Status NotificationManager::process_push_notification_payload(string payload) {
TRY_RESULT(contains_mention_int, get_json_object_int_field(custom, "mention"));
bool contains_mention = contains_mention_int != 0;
process_message_push_notification(dialog_id, sender_user_id, MessageId(server_message_id), random_id,
contains_mention, std::move(loc_key), std::move(loc_args));
if (begins_with(loc_key, "CHANNEL_MESSAGE")) {
if (dialog_id.get_type() != DialogType::Channel) {
return Status::Error("Receive wrong chat type");
}
loc_key = loc_key.substr(8);
}
if (begins_with(loc_key, "CHAT_MESSAGE")) {
if (dialog_id.get_type() != DialogType::Chat) {
return Status::Error("Receive wrong chat type");
}
loc_key = loc_key.substr(5);
}
if (begins_with(loc_key, "CHAT_")) {
auto dialog_type = dialog_id.get_type();
if (dialog_type != DialogType::Chat && dialog_type != DialogType::Channel) {
return Status::Error("Receive wrong chat type");
}
}
if (begins_with(loc_key, "MESSAGE") && !server_message_id.is_valid()) {
return Status::Error("Receive no message ID");
}
if (begins_with(loc_key, "ENCRYPT") || random_id != 0) {
if (dialog_id.get_type() != DialogType::SecretChat) {
return Status::Error("Receive wrong chat type");
}
}
if (server_message_id.is_valid() && dialog_id.get_type() == DialogType::SecretChat) {
return Status::Error("Receive message ID in secret chat push");
}
return Status::OK();
return process_message_push_notification(dialog_id, sender_user_id, MessageId(server_message_id), random_id,
contains_mention, std::move(loc_key), std::move(loc_args));
}
void NotificationManager::process_message_push_notification(DialogId dialog_id, UserId sender_user_id,
Status NotificationManager::process_message_push_notification(DialogId dialog_id, UserId sender_user_id,
MessageId message_id, int64 random_id,
bool contains_mention, string loc_key,
vector<string> loc_args) {
return Status::OK();
}
Result<int64> NotificationManager::get_push_receiver_id(string payload) {

2
td/telegram/NotificationManager.h
View File

@ -255,7 +255,7 @@ class NotificationManager : public Actor {
Status process_push_notification_payload(string payload);
void process_message_push_notification(DialogId dialog_id, UserId sender_user_id, MessageId message_id,
Status process_message_push_notification(DialogId dialog_id, UserId sender_user_id, MessageId message_id,
int64 random_id, bool contains_mention, string loc_key,
vector<string> loc_args);