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

Reset server time difference if it is more than 15 seconds in the future.

Browse Source
This commit is contained in:
levlam 2023-05-14 23:11:23 +03:00
parent a395337265
commit 5d499d594c
2 changed files with 16 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
td/mtproto/SessionConnection.cpp
View File

@ -225,6 +225,12 @@ Status SessionConnection::on_packet_container(const MsgInfo &info, Slice packet)
return Status::OK(); return Status::OK();
} }
void SessionConnection::reset_server_time_difference(uint64 message_id) {
VLOG(mtproto) << "Reset server time difference";
auth_data_->reset_server_time_difference(static_cast<uint32>(message_id >> 32) - Time::now());
callback_->on_server_time_difference_updated(true);
}
Status SessionConnection::on_packet_rpc_result(const MsgInfo &info, Slice packet) { Status SessionConnection::on_packet_rpc_result(const MsgInfo &info, Slice packet) {
TlParser parser(packet); TlParser parser(packet);
uint64 req_msg_id = parser.fetch_long(); uint64 req_msg_id = parser.fetch_long();
@ -236,6 +242,10 @@ Status SessionConnection::on_packet_rpc_result(const MsgInfo &info, Slice packet
return Status::Error("Receive an update in rpc_result"); return Status::Error("Receive an update in rpc_result");
} }
if (info.message_id < req_msg_id - (static_cast<uint64>(15) << 32)) {
reset_server_time_difference(info.message_id);
}
switch (parser.fetch_int()) { switch (parser.fetch_int()) {
case mtproto_api::rpc_error::ID: { case mtproto_api::rpc_error::ID: {
mtproto_api::rpc_error rpc_error(parser); mtproto_api::rpc_error rpc_error(parser);
@ -331,8 +341,7 @@ Status SessionConnection::on_packet(const MsgInfo &info,
LOG(WARNING) << bad_info << ": MessageId is too high. Session will be closed"; LOG(WARNING) << bad_info << ": MessageId is too high. Session will be closed";
// All this queries will be re-sent by parent // All this queries will be re-sent by parent
to_send_.clear(); to_send_.clear();
auth_data_->reset_server_time_difference(static_cast<uint32>(info.message_id >> 32) - Time::now()); reset_server_time_difference(info.message_id);
callback_->on_server_time_difference_updated(true);
callback_->on_session_failed(Status::Error("MessageId is too high")); callback_->on_session_failed(Status::Error("MessageId is too high"));
return Status::Error("MessageId is too high"); return Status::Error("MessageId is too high");
} }
@ -407,6 +416,9 @@ Status SessionConnection::on_packet(const MsgInfo &info, const mtproto_api::gzip
Status SessionConnection::on_packet(const MsgInfo &info, const mtproto_api::pong &pong) { Status SessionConnection::on_packet(const MsgInfo &info, const mtproto_api::pong &pong) {
VLOG(mtproto) << "PONG"; VLOG(mtproto) << "PONG";
if (info.message_id < static_cast<uint64>(pong.msg_id_) - (static_cast<uint64>(15) << 32)) {
reset_server_time_difference(info.message_id);
}
last_pong_at_ = Time::now_cached(); last_pong_at_ = Time::now_cached();
real_last_pong_at_ = last_pong_at_; real_last_pong_at_ = last_pong_at_;
return callback_->on_pong(); return callback_->on_pong();

2
td/mtproto/SessionConnection.h
View File

@ -221,6 +221,8 @@ class SessionConnection final
}; };
} }
void reset_server_time_difference(uint64 message_id);
static Status parse_message(TlParser &parser, MsgInfo *info, Slice *packet, static Status parse_message(TlParser &parser, MsgInfo *info, Slice *packet,
bool crypto_flag = true) TD_WARN_UNUSED_RESULT; bool crypto_flag = true) TD_WARN_UNUSED_RESULT;
Status parse_packet(TlParser &parser) TD_WARN_UNUSED_RESULT; Status parse_packet(TlParser &parser) TD_WARN_UNUSED_RESULT;