From 175da716bbd3e8d4619a47a9fb9048acdd3f849c Mon Sep 17 00:00:00 2001 From: levlam Date: Fri, 25 Jan 2019 17:44:47 +0300 Subject: [PATCH] Do not allow to set invalid file referene from server. GitOrigin-RevId: aabcdd17500097c931a146a61236865f96d759ca --- td/telegram/files/FileLocation.h | 8 ++++++++ 1 file changed, 8 insertions(+) diff --git a/td/telegram/files/FileLocation.h b/td/telegram/files/FileLocation.h index 033f0619f..6f7678396 100644 --- a/td/telegram/files/FileLocation.h +++ b/td/telegram/files/FileLocation.h @@ -513,6 +513,10 @@ class FullRemoteFileLocation { , file_reference_(std::move(file_reference)) , variant_(PhotoRemoteFileLocation{id, access_hash, volume_id, secret, local_id}) { CHECK(is_photo()); + if (file_reference_ == invalid_file_reference()) { + LOG(ERROR) << "Tried to register file with invalid file reference"; + file_reference_.clear(); + } } FullRemoteFileLocation(FileType file_type, int64 id, int64 access_hash, DcId dc_id, std::string file_reference) : file_type_(file_type) @@ -520,6 +524,10 @@ class FullRemoteFileLocation { , file_reference_(std::move(file_reference)) , variant_(CommonRemoteFileLocation{id, access_hash}) { CHECK(is_common()); + if (file_reference_ == invalid_file_reference()) { + LOG(ERROR) << "Tried to register file with invalid file reference"; + file_reference_.clear(); + } } FullRemoteFileLocation(FileType file_type, string url, int64 access_hash) : file_type_(file_type)