Send empty updates on Session creating to initialize it.

Author: levlam

td/mtproto/AuthData.h

@@ -67,16 +67,20 @@ class AuthData {
   void set_main_auth_key(AuthKey auth_key) {
     main_auth_key_ = std::move(auth_key);
   }
+
   void break_main_auth_key() {
     main_auth_key_.break_key();
   }
+
   const AuthKey &get_main_auth_key() const {
     // CHECK(has_main_auth_key());
     return main_auth_key_;
   }
+
   bool has_main_auth_key() const {
     return !main_auth_key_.empty();
   }
+
   bool need_main_auth_key() const {
     return !has_main_auth_key();
   }
@@ -85,12 +89,15 @@ class AuthData {
     CHECK(!auth_key.empty());
     tmp_auth_key_ = std::move(auth_key);
   }
+
   const AuthKey &get_tmp_auth_key() const {
     return tmp_auth_key_;
   }
+
   bool was_tmp_auth_key() const {
     return use_pfs() && !tmp_auth_key_.empty();
   }
+
   bool need_tmp_auth_key(double now, double refresh_margin) const {
     if (!use_pfs()) {
       return false;
@@ -103,12 +110,15 @@ class AuthData {
     }
     return false;
   }
+
   void drop_main_auth_key() {
     main_auth_key_ = AuthKey();
   }
+
   void drop_tmp_auth_key() {
     tmp_auth_key_ = AuthKey();
   }
+
   bool has_tmp_auth_key(double now) const {
     if (!use_pfs()) {
       return false;
@@ -128,6 +138,7 @@ class AuthData {
     }
     return get_main_auth_key();
   }
+
   bool has_auth_key(double now) const {
     if (use_pfs()) {
       return has_tmp_auth_key(now);
@@ -138,6 +149,7 @@ class AuthData {
   bool get_auth_flag() const {
     return main_auth_key_.auth_flag();
   }
+
   void set_auth_flag(bool auth_flag) {
     main_auth_key_.set_auth_flag(auth_flag);
     if (!auth_flag) {
@@ -148,6 +160,7 @@ class AuthData {
   bool get_bind_flag() const {
     return !use_pfs() || tmp_auth_key_.auth_flag();
   }
+
   void on_bind() {
     CHECK(use_pfs());
     tmp_auth_key_.set_auth_flag(true);
@@ -184,6 +197,7 @@ class AuthData {
   void set_session_id(uint64 session_id) {
     session_id_ = session_id;
   }
+
   uint64 get_session_id() const {
     CHECK(session_id_ != 0);
     return session_id_;

td/telegram/net/SessionProxy.cpp

@@ -48,9 +48,11 @@ class SessionCallback final : public Session::Callback {
   void on_failed() final {
     send_closure(parent_, &SessionProxy::on_failed);
   }
+
   void on_closed() final {
     send_closure(parent_, &SessionProxy::on_closed);
   }
+
   void request_raw_connection(unique_ptr<mtproto::AuthData> auth_data,
                               Promise<unique_ptr<mtproto::RawConnection>> promise) final {
     send_closure(G()->connection_creator(), &ConnectionCreator::request_raw_connection, dc_id_, allow_media_only_,
@@ -224,7 +226,7 @@ void SessionProxy::open_session(bool force) {
   CHECK(session_.empty());
   auto dc_id = auth_data_->dc_id();
   string name = PSTRING() << "Session" << get_name().substr(Slice("SessionProxy").size());
-  string hash_string = PSTRING() << name << " " << dc_id.get_raw_id() << " " << allow_media_only_;
+  string hash_string = PSTRING() << name << ' ' << dc_id.get_raw_id() << ' ' << allow_media_only_;
   auto hash = Hash<string>()(hash_string);
   int32 raw_dc_id = dc_id.get_raw_id();
   int32 int_dc_id = raw_dc_id;
@@ -234,6 +236,10 @@ void SessionProxy::open_session(bool force) {
   if (allow_media_only_ && !is_cdn_) {
     int_dc_id = -int_dc_id;
   }
+  if (is_main_ && use_pfs_ && !tmp_auth_key_.empty()) {
+    send_closure_later(G()->td(), &Td::on_update, telegram_api::make_object<telegram_api::updates>(),
+                       tmp_auth_key_.id());
+  }
   session_ = create_actor<Session>(
       name,
       make_unique<SessionCallback>(actor_shared(this, session_generation_), dc_id, allow_media_only_, is_media_, hash),