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

Delete expired unconfirmed sessions after loading them after restart.

Browse Source
This commit is contained in:
levlam 2023-09-12 14:48:42 +03:00
parent 6a4c78e849
commit 7f2d1cdd9d
1 changed files with 24 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
td/telegram/AccountManager.cpp
View File

@ -664,6 +664,10 @@ class AccountManager::UnconfirmedAuthorization {
class AccountManager::UnconfirmedAuthorizations { class AccountManager::UnconfirmedAuthorizations {
vector<UnconfirmedAuthorization> authorizations_; vector<UnconfirmedAuthorization> authorizations_;
static int32 get_authorization_autoconfirm_period() {
return narrow_cast<int32>(G()->get_option_integer("authorization_autoconfirm_period", 604800));
}
public: public:
bool is_empty() const { bool is_empty() const {
return authorizations_.empty(); return authorizations_.empty();
@ -701,6 +705,19 @@ class AccountManager::UnconfirmedAuthorizations {
return true; return true;
} }
bool delete_expired_authorizations() {
auto up_to_date = G()->unix_time() - get_authorization_autoconfirm_period();
auto it = authorizations_.begin();
while (it != authorizations_.end() && it->get_date() <= up_to_date) {
++it;
}
if (it == authorizations_.begin()) {
return false;
}
authorizations_.erase(authorizations_.begin(), it);
return true;
}
td_api::object_ptr<td_api::unconfirmedSession> get_first_unconfirmed_session_object() const { td_api::object_ptr<td_api::unconfirmedSession> get_first_unconfirmed_session_object() const {
CHECK(!authorizations_.empty()); CHECK(!authorizations_.empty());
return authorizations_[0].get_unconfirmed_session_object(); return authorizations_[0].get_unconfirmed_session_object();
@ -728,6 +745,13 @@ void AccountManager::start_up() {
G()->td_db()->get_binlog_pmc()->get(get_unconfirmed_authorizations_key()); G()->td_db()->get_binlog_pmc()->get(get_unconfirmed_authorizations_key());
if (!unconfirmed_authorizations_log_event_string.empty()) { if (!unconfirmed_authorizations_log_event_string.empty()) {
log_event_parse(unconfirmed_authorizations_, unconfirmed_authorizations_log_event_string).ensure(); log_event_parse(unconfirmed_authorizations_, unconfirmed_authorizations_log_event_string).ensure();
CHECK(unconfirmed_authorizations_ != nullptr);
if (unconfirmed_authorizations_->delete_expired_authorizations()) {
save_unconfirmed_authorizations();
if (unconfirmed_authorizations_->is_empty()) {
unconfirmed_authorizations_ = nullptr;
}
}
if (unconfirmed_authorizations_ != nullptr) { if (unconfirmed_authorizations_ != nullptr) {
send_update_unconfirmed_session(); send_update_unconfirmed_session();
} }