103 lines
3.0 KiB
C++
103 lines
3.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)
|
|
//
|
|
#pragma once
|
|
|
|
#include "td/telegram/net/DcId.h"
|
|
#include "td/telegram/net/NetQuery.h"
|
|
|
|
#include "td/actor/actor.h"
|
|
|
|
#include "td/utils/common.h"
|
|
#include "td/utils/Promise.h"
|
|
#include "td/utils/ScopeGuard.h"
|
|
#include "td/utils/Status.h"
|
|
|
|
#include <array>
|
|
#include <atomic>
|
|
#include <functional>
|
|
#include <memory>
|
|
#include <mutex>
|
|
|
|
namespace td {
|
|
|
|
class DcAuthManager;
|
|
class MultiSequenceDispatcher;
|
|
class NetQueryDelayer;
|
|
class NetQueryVerifier;
|
|
class PublicRsaKeyWatchdog;
|
|
class SessionMultiProxy;
|
|
|
|
// Not just dispatcher.
|
|
class NetQueryDispatcher {
|
|
public:
|
|
explicit NetQueryDispatcher(const std::function<ActorShared<>()> &create_reference);
|
|
NetQueryDispatcher(const NetQueryDispatcher &) = delete;
|
|
NetQueryDispatcher &operator=(const NetQueryDispatcher &) = delete;
|
|
NetQueryDispatcher(NetQueryDispatcher &&) = delete;
|
|
NetQueryDispatcher &operator=(NetQueryDispatcher &&) = delete;
|
|
~NetQueryDispatcher();
|
|
|
|
void dispatch(NetQueryPtr net_query);
|
|
void dispatch_with_callback(NetQueryPtr net_query, ActorShared<NetQueryCallback> callback);
|
|
void stop();
|
|
|
|
void update_session_count();
|
|
void destroy_auth_keys(Promise<> promise);
|
|
void update_use_pfs();
|
|
void update_mtproto_header();
|
|
|
|
DcId get_main_dc_id() const {
|
|
return DcId::internal(main_dc_id_.load(std::memory_order_relaxed));
|
|
}
|
|
|
|
void set_main_dc_id(int32 new_main_dc_id);
|
|
void check_authorization_is_ok();
|
|
|
|
void set_verification_token(int64 verification_id, string &&token, Promise<Unit> &&promise);
|
|
|
|
private:
|
|
std::atomic<bool> stop_flag_{false};
|
|
bool need_destroy_auth_key_{false};
|
|
ActorOwn<NetQueryDelayer> delayer_;
|
|
ActorOwn<NetQueryVerifier> verifier_;
|
|
ActorOwn<DcAuthManager> dc_auth_manager_;
|
|
ActorOwn<MultiSequenceDispatcher> sequence_dispatcher_;
|
|
struct Dc {
|
|
DcId id_;
|
|
std::atomic<bool> is_valid_{false};
|
|
std::atomic<bool> is_inited_{false}; // TODO: cache in scheduler local storage :D
|
|
|
|
ActorOwn<SessionMultiProxy> main_session_;
|
|
ActorOwn<SessionMultiProxy> download_session_;
|
|
ActorOwn<SessionMultiProxy> download_small_session_;
|
|
ActorOwn<SessionMultiProxy> upload_session_;
|
|
};
|
|
std::array<Dc, DcId::MAX_RAW_DC_ID> dcs_;
|
|
#if TD_EMSCRIPTEN // FIXME
|
|
std::atomic<int32> main_dc_id_{2};
|
|
#else
|
|
std::atomic<int32> main_dc_id_{1};
|
|
#endif
|
|
ActorOwn<PublicRsaKeyWatchdog> public_rsa_key_watchdog_;
|
|
std::mutex mutex_;
|
|
std::shared_ptr<Guard> td_guard_;
|
|
|
|
Status wait_dc_init(DcId dc_id, bool force);
|
|
bool is_dc_inited(int32 raw_dc_id);
|
|
|
|
static int32 get_main_session_scheduler_id();
|
|
static int32 get_session_count();
|
|
static bool get_use_pfs();
|
|
|
|
static void complete_net_query(NetQueryPtr net_query);
|
|
bool check_stop_flag(NetQueryPtr &net_query) const;
|
|
|
|
void try_fix_migrate(NetQueryPtr &net_query);
|
|
};
|
|
|
|
} // namespace td
|