tdlight-team/tdlight
tdlight-team/tdlight
1
1
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Add checks for file_id/file_node_id overflow.

Browse Source
This commit is contained in:
levlam 2022-04-25 17:33:31 +03:00
parent 61942cf3ea
commit a7bd57a793
1 changed files with 2 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
td/telegram/files/FileManager.cpp
View File

@ -3385,13 +3385,14 @@ FileId FileManager::next_file_id() {
empty_file_ids_.pop_back(); empty_file_ids_.pop_back();
return FileId{res, 0}; return FileId{res, 0};
} }
CHECK(file_id_info_.size() <= static_cast<size_t>(std::numeric_limits<int32>::max()));
FileId res(static_cast<int32>(file_id_info_.size()), 0); FileId res(static_cast<int32>(file_id_info_.size()), 0);
// LOG(ERROR) << "NEXT file_id " << res;
file_id_info_.push_back({}); file_id_info_.push_back({});
return res; return res;
} }
FileManager::FileNodeId FileManager::next_file_node_id() { FileManager::FileNodeId FileManager::next_file_node_id() {
CHECK(file_nodes_.size() <= static_cast<size_t>(std::numeric_limits<FileNodeId>::max()));
auto res = static_cast<FileNodeId>(file_nodes_.size()); auto res = static_cast<FileNodeId>(file_nodes_.size());
file_nodes_.emplace_back(nullptr); file_nodes_.emplace_back(nullptr);
return res; return res;