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

Check that poll has no duplicate options.

Browse Source 
GitOrigin-RevId: 781558fac8edf2f42a57dac0800590095babb20a
This commit is contained in:
levlam 2020-05-23 17:21:56 +03:00
parent 43704b69e0
commit 8b0561f3be
1 changed files with 14 additions and 2 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
td/telegram/PollManager.cpp
View File

@ -1305,8 +1305,20 @@ PollId PollManager::on_get_poll(PollId poll_id, tl_object_ptr<telegram_api::poll
LOG(ERROR) << "Receive poll " << poll_server->id_ << " instead of " << poll_id;
return PollId();
}
if (poll_server != nullptr && poll_server->answers_.empty()) {
LOG(ERROR) << "Receive " << poll_id << " without answers: " << to_string(poll_server);
constexpr size_t MAX_POLL_OPTIONS = 10; // server-side limit
if (poll_server != nullptr &&
(poll_server->answers_.size() <= 1 || poll_server->answers_.size() > 10 * MAX_POLL_OPTIONS)) {
LOG(ERROR) << "Receive " << poll_id << " with wrong number of answers: " << to_string(poll_server);
return PollId();
}
if (poll_server != nullptr) {
std::unordered_set<Slice, SliceHash> option_data;
for (auto &answer : poll_server->answers_) {
option_data.insert(answer->option_.as_slice());
}
if (option_data.size() != poll_server->answers_.size()) {
LOG(ERROR) << "Receive " << poll_id << " with duplicate options: " << to_string(poll_server);
}
return PollId();
}