From 12ae67a64f3a64e910fd8eb7192a117151e7f6d0 Mon Sep 17 00:00:00 2001
From: levlam <levlam@telegram.org>
Date: Sun, 27 Jun 2021 06:02:58 +0300
Subject: [PATCH] Don't try to schedule getDifference if unauthorized.

---
 td/telegram/UpdatesManager.cpp | 16 ++++++----------
 1 file changed, 6 insertions(+), 10 deletions(-)

diff --git a/td/telegram/UpdatesManager.cpp b/td/telegram/UpdatesManager.cpp
index e20718267..64cc39b0f 100644
--- a/td/telegram/UpdatesManager.cpp
+++ b/td/telegram/UpdatesManager.cpp
@@ -223,7 +223,7 @@ void UpdatesManager::fill_get_difference_gap(void *td) {
 
 void UpdatesManager::fill_gap(void *td, const char *source) {
   CHECK(td != nullptr);
-  if (G()->close_flag()) {
+  if (G()->close_flag() || !static_cast<Td *>(td)->auth_manager_->is_authorized()) {
     return;
   }
   auto updates_manager = static_cast<Td *>(td)->updates_manager_.get();
@@ -236,7 +236,7 @@ void UpdatesManager::fill_gap(void *td, const char *source) {
 }
 
 void UpdatesManager::get_difference(const char *source) {
-  if (G()->close_flag()) {
+  if (G()->close_flag() || !td_->auth_manager_->is_authorized()) {
     return;
   }
   if (get_pts() == -1) {
@@ -244,10 +244,6 @@ void UpdatesManager::get_difference(const char *source) {
     return;
   }
 
-  if (!td_->auth_manager_->is_authorized()) {
-    return;
-  }
-
   if (running_get_difference_) {
     VLOG(get_difference) << "Skip running getDifference from " << source << " because it is already running";
     return;
@@ -881,7 +877,7 @@ void UpdatesManager::on_get_updates(tl_object_ptr<telegram_api::Updates> &&updat
 }
 
 void UpdatesManager::on_failed_get_updates_state(Status &&error) {
-  if (G()->close_flag()) {
+  if (G()->close_flag() || !td_->auth_manager_->is_authorized()) {
     return;
   }
   if (error.code() != 401) {
@@ -893,7 +889,7 @@ void UpdatesManager::on_failed_get_updates_state(Status &&error) {
 }
 
 void UpdatesManager::on_failed_get_difference(Status &&error) {
-  if (G()->close_flag()) {
+  if (G()->close_flag() || !td_->auth_manager_->is_authorized()) {
     return;
   }
   if (error.code() != 401) {
@@ -909,7 +905,7 @@ void UpdatesManager::on_failed_get_difference(Status &&error) {
 }
 
 void UpdatesManager::schedule_get_difference(const char *source) {
-  if (G()->close_flag()) {
+  if (G()->close_flag() || !td_->auth_manager_->is_authorized()) {
     return;
   }
   if (!retry_timeout_.has_timeout()) {
@@ -1151,7 +1147,7 @@ int32 UpdatesManager::get_update_edit_message_pts(const telegram_api::Updates *u
 }
 
 void UpdatesManager::init_state() {
-  if (!td_->auth_manager_->is_authorized()) {
+  if (G()->close_flag() || !td_->auth_manager_->is_authorized()) {
     return;
   }