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

43
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")); TRY_RESULT(contains_mention_int, get_json_object_int_field(custom, "mention"));
bool contains_mention = contains_mention_int != 0; bool contains_mention = contains_mention_int != 0;
process_message_push_notification(dialog_id, sender_user_id, MessageId(server_message_id), random_id, if (begins_with(loc_key, "CHANNEL_MESSAGE")) {
contains_mention, std::move(loc_key), std::move(loc_args)); 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, MessageId message_id, int64 random_id,
bool contains_mention, string loc_key, bool contains_mention, string loc_key,
vector<string> loc_args) { vector<string> loc_args) {
return Status::OK();
} }
Result<int64> NotificationManager::get_push_receiver_id(string payload) { Result<int64> NotificationManager::get_push_receiver_id(string payload) {

6
td/telegram/NotificationManager.h
View File

@ -255,9 +255,9 @@ class NotificationManager : public Actor {
Status process_push_notification_payload(string payload); 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, int64 random_id, bool contains_mention, string loc_key,
vector<string> loc_args); vector<string> loc_args);
void after_get_difference_impl(); void after_get_difference_impl();