tdlight/td/telegram/net/PublicRsaKeyWatchdog.cpp
2024-01-01 03:07:21 +03:00

140 lines
4.0 KiB
C++

//
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2024
//
// Distributed under the Boost Software License, Version 1.0. (See accompanying
// file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt)
//
#include "td/telegram/net/PublicRsaKeyWatchdog.h"
#include "td/telegram/Global.h"
#include "td/telegram/net/NetQueryCreator.h"
#include "td/telegram/TdDb.h"
#include "td/telegram/telegram_api.h"
#include "td/telegram/Version.h"
#include "td/mtproto/RSA.h"
#include "td/utils/logging.h"
#include "td/utils/Time.h"
namespace td {
PublicRsaKeyWatchdog::PublicRsaKeyWatchdog(ActorShared<> parent) : parent_(std::move(parent)) {
}
void PublicRsaKeyWatchdog::add_public_rsa_key(std::shared_ptr<PublicRsaKeyShared> key) {
class Listener final : public PublicRsaKeyShared::Listener {
public:
explicit Listener(ActorId<PublicRsaKeyWatchdog> parent) : parent_(std::move(parent)) {
}
bool notify() final {
send_event(parent_, Event::yield());
return parent_.is_alive();
}
private:
ActorId<PublicRsaKeyWatchdog> parent_;
};
key->add_listener(make_unique<Listener>(actor_id(this)));
sync_key(key);
keys_.push_back(std::move(key));
loop();
}
void PublicRsaKeyWatchdog::start_up() {
flood_control_.add_limit(1, 1);
flood_control_.add_limit(2, 60);
flood_control_.add_limit(3, 2 * 60);
string version = G()->td_db()->get_binlog_pmc()->get("cdn_config_version");
current_version_ = to_string(MTPROTO_LAYER);
if (version != current_version_) {
G()->td_db()->get_binlog_pmc()->erase("cdn_config" + version);
} else {
sync(BufferSlice(G()->td_db()->get_binlog_pmc()->get("cdn_config" + version)));
}
CHECK(keys_.empty());
}
void PublicRsaKeyWatchdog::loop() {
if (has_query_) {
return;
}
auto now = Time::now();
if (now < flood_control_.get_wakeup_at()) {
set_timeout_at(flood_control_.get_wakeup_at() + 0.01);
return;
}
bool ok = true;
for (auto &key : keys_) {
if (!key->has_keys()) {
ok = false;
}
}
if (ok) {
return;
}
flood_control_.add_event(now);
has_query_ = true;
auto query = G()->net_query_creator().create(telegram_api::help_getCdnConfig());
query->total_timeout_limit_ = 60 * 60 * 24;
G()->net_query_dispatcher().dispatch_with_callback(std::move(query), actor_shared(this));
}
void PublicRsaKeyWatchdog::on_result(NetQueryPtr net_query) {
has_query_ = false;
yield();
if (net_query->is_error()) {
LOG(ERROR) << "Receive error for GetCdnConfig: " << net_query->move_as_error();
loop();
return;
}
auto buf = net_query->move_as_ok();
G()->td_db()->get_binlog_pmc()->set("cdn_config_version", current_version_);
G()->td_db()->get_binlog_pmc()->set("cdn_config" + current_version_, buf.as_slice().str());
sync(std::move(buf));
}
void PublicRsaKeyWatchdog::sync(BufferSlice cdn_config_serialized) {
if (cdn_config_serialized.empty()) {
loop();
return;
}
auto r_keys = fetch_result<telegram_api::help_getCdnConfig>(cdn_config_serialized);
if (r_keys.is_error()) {
LOG(WARNING) << "Failed to deserialize help_getCdnConfig (probably not a problem) " << r_keys.error();
loop();
return;
}
cdn_config_ = r_keys.move_as_ok();
if (keys_.empty()) {
LOG(INFO) << "Load " << to_string(cdn_config_);
} else {
LOG(INFO) << "Receive " << to_string(cdn_config_);
for (auto &key : keys_) {
sync_key(key);
}
}
}
void PublicRsaKeyWatchdog::sync_key(std::shared_ptr<PublicRsaKeyShared> &key) {
if (!cdn_config_) {
return;
}
for (auto &config_key : cdn_config_->public_keys_) {
if (key->dc_id().get_raw_id() == config_key->dc_id_) {
auto r_rsa = mtproto::RSA::from_pem_public_key(config_key->public_key_);
if (r_rsa.is_error()) {
LOG(ERROR) << r_rsa.error();
continue;
}
LOG(INFO) << "Add CDN " << key->dc_id() << " key with fingerprint " << r_rsa.ok().get_fingerprint();
key->add_rsa(r_rsa.move_as_ok());
}
}
}
} // namespace td