140 lines
4.0 KiB
C++
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
|