Improve photo_get_cover_input_media.

1 files changed, 3 insertions, 0 deletions

diff --git a/td/telegram/Photo.cpp b/td/telegram/Photo.cpp
index 26caba33d..0b35612e4 100644
--- a/td/telegram/Photo.cpp
+++ b/td/telegram/Photo.cpp
@@ -687,6 +687,9 @@ SecretInputMedia photo_get_secret_input_media(FileManager *file_manager, const P
 telegram_api::object_ptr<telegram_api::InputMedia> photo_get_cover_input_media(FileManager *file_manager,
                                                                                const Photo &photo, bool force) {
   auto input_media = photo_get_input_media(file_manager, photo, nullptr, 0, false);
+  if (input_media == nullptr || input_media->get_id() != telegram_api::inputMediaPhoto::ID) {
+    return nullptr;
+  }
   auto file_reference = FileManager::extract_file_reference(input_media);
   if (file_reference == FileReferenceView::invalid_file_reference()) {
     if (!force) {