From c63aaf7cdec70dbb52992771c788fe3f4b834308 Mon Sep 17 00:00:00 2001
From: levlam <levlam@telegram.org>
Date: Wed, 25 Jul 2018 16:49:44 +0300
Subject: [PATCH] Add source to ttl_unregister_message.

GitOrigin-RevId: d25685de6cbb92569f29f22e111b612d76e4fcb9
---
 td/telegram/MessagesManager.cpp | 11 ++++++-----
 td/telegram/MessagesManager.h   |  2 +-
 2 files changed, 7 insertions(+), 6 deletions(-)

diff --git a/td/telegram/MessagesManager.cpp b/td/telegram/MessagesManager.cpp
index 501a8d4da..0fc304923 100644
--- a/td/telegram/MessagesManager.cpp
+++ b/td/telegram/MessagesManager.cpp
@@ -9716,14 +9716,15 @@ void MessagesManager::ttl_register_message(DialogId dialog_id, const Message *me
   ttl_update_timeout(now);
 }
 
-void MessagesManager::ttl_unregister_message(DialogId dialog_id, const Message *message, double now) {
+void MessagesManager::ttl_unregister_message(DialogId dialog_id, const Message *message, double now,
+                                             const char *source) {
   if (message->ttl_expires_at == 0) {
     return;
   }
 
   TtlNode ttl_node(dialog_id, message->message_id);
   auto it = ttl_nodes_.find(ttl_node);
-  CHECK(it != ttl_nodes_.end());
+  CHECK(it != ttl_nodes_.end()) << dialog_id << " " << message->message_id << " " << source << " " << G()->close_flag();
   auto *heap_node = it->as_heap_node();
   if (heap_node->in_heap()) {
     ttl_heap_.erase(heap_node);
@@ -9770,7 +9771,7 @@ void MessagesManager::on_message_ttl_expired(Dialog *d, Message *message) {
   CHECK(message != nullptr);
   CHECK(message->ttl > 0);
   CHECK(d->dialog_id.get_type() != DialogType::SecretChat);
-  ttl_unregister_message(d->dialog_id, message, Time::now());
+  ttl_unregister_message(d->dialog_id, message, Time::now(), "on_message_ttl_expired");
   on_message_ttl_expired_impl(d, message);
   send_update_message_content(d->dialog_id, message->message_id, message->content.get(), message->date,
                               message->is_content_secret, "on_message_ttl_expired");
@@ -11761,7 +11762,7 @@ unique_ptr<MessagesManager::Message> MessagesManager::do_delete_message(Dialog *
     default:
       UNREACHABLE();
   }
-  ttl_unregister_message(d->dialog_id, result.get(), Time::now());
+  ttl_unregister_message(d->dialog_id, result.get(), Time::now(), "do_delete_message");
 
   return result;
 }
@@ -11808,7 +11809,7 @@ void MessagesManager::do_delete_all_dialog_messages(Dialog *d, unique_ptr<Messag
     default:
       UNREACHABLE();
   }
-  ttl_unregister_message(d->dialog_id, m.get(), Time::now());
+  ttl_unregister_message(d->dialog_id, m.get(), Time::now(), "do_delete_all_dialog_messages");
 
   m = nullptr;
 }
diff --git a/td/telegram/MessagesManager.h b/td/telegram/MessagesManager.h
index 0a8c4b513..1e4ac27fa 100644
--- a/td/telegram/MessagesManager.h
+++ b/td/telegram/MessagesManager.h
@@ -2516,7 +2516,7 @@ class MessagesManager : public Actor {
   void ttl_on_view(const Dialog *d, Message *message, double view_date, double now);
   bool ttl_on_open(Dialog *d, Message *message, double now, bool is_local_read);
   void ttl_register_message(DialogId dialog_id, const Message *message, double now);
-  void ttl_unregister_message(DialogId dialog_id, const Message *message, double now);
+  void ttl_unregister_message(DialogId dialog_id, const Message *message, double now, const char *source);
   void ttl_loop(double now);
   void ttl_update_timeout(double now);