// // Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2023 // // 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/ChannelId.h" #include "td/telegram/ChatId.h" #include "td/telegram/DialogId.h" #include "td/telegram/FullMessageId.h" #include "td/telegram/InputGroupCallId.h" #include "td/telegram/MessageId.h" #include "td/telegram/PtsManager.h" #include "td/telegram/telegram_api.h" #include "td/telegram/UserId.h" #include "td/actor/actor.h" #include "td/actor/MultiTimeout.h" #include "td/actor/Timeout.h" #include "td/utils/common.h" #include "td/utils/FlatHashMap.h" #include "td/utils/FlatHashSet.h" #include "td/utils/logging.h" #include "td/utils/Promise.h" #include "td/utils/Status.h" #include "td/utils/tl_storers.h" #include "td/utils/TlStorerToString.h" #include #include #include namespace td { extern int VERBOSITY_NAME(get_difference); class Td; class dummyUpdate final : public telegram_api::Update { public: static constexpr int32 ID = 1234567891; int32 get_id() const final { return ID; } void store(TlStorerUnsafe &s) const final { UNREACHABLE(); } void store(TlStorerCalcLength &s) const final { UNREACHABLE(); } void store(TlStorerToString &s, const char *field_name) const final { s.store_class_begin(field_name, "dummyUpdate"); s.store_class_end(); } }; class updateSentMessage final : public telegram_api::Update { public: int64 random_id_; MessageId message_id_; int32 date_; int32 ttl_period_; updateSentMessage(int64 random_id, MessageId message_id, int32 date, int32 ttl_period) : random_id_(random_id), message_id_(message_id), date_(date), ttl_period_(ttl_period) { } static constexpr int32 ID = 1234567890; int32 get_id() const final { return ID; } void store(TlStorerUnsafe &s) const final { UNREACHABLE(); } void store(TlStorerCalcLength &s) const final { UNREACHABLE(); } void store(TlStorerToString &s, const char *field_name) const final { s.store_class_begin(field_name, "updateSentMessage"); s.store_field("random_id", random_id_); s.store_field("message_id", message_id_.get()); s.store_field("date", date_); s.store_field("ttl_period", ttl_period_); s.store_class_end(); } }; class UpdatesManager final : public Actor { public: UpdatesManager(Td *td, ActorShared<> parent); void on_get_updates(tl_object_ptr &&updates_ptr, Promise &&promise); void add_pending_pts_update(tl_object_ptr &&update, int32 new_pts, int32 pts_count, double receive_time, Promise &&promise, const char *source); static FlatHashSet get_sent_messages_random_ids(const telegram_api::Updates *updates_ptr); static const telegram_api::Message *get_message_by_random_id(const telegram_api::Updates *updates_ptr, DialogId dialog_id, int64 random_id); // [Message, is_scheduled] static vector> get_new_messages( const telegram_api::Updates *updates_ptr); static vector get_update_new_group_call_ids(const telegram_api::Updates *updates_ptr); static string extract_join_group_call_presentation_params(telegram_api::Updates *updates_ptr); static vector get_update_notify_settings_dialog_ids(const telegram_api::Updates *updates_ptr); static vector get_chat_dialog_ids(const telegram_api::Updates *updates_ptr); static int32 get_update_edit_message_pts(const telegram_api::Updates *updates_ptr, FullMessageId full_message_id); using TranscribedAudioHandler = std::function>)>; void subscribe_to_transcribed_audio_updates(int64 transcription_id, TranscribedAudioHandler on_update); void get_difference(const char *source); void schedule_get_difference(const char *source); void on_update_from_auth_key_id(uint64 auth_key_id); void ping_server(); bool running_get_difference() const { return running_get_difference_; } void timeout_expired() final; private: static constexpr int32 FORCED_GET_DIFFERENCE_PTS_DIFF = 100000; static constexpr int32 GAP_TIMEOUT_UPDATE_COUNT = 20; static constexpr double MAX_UNFILLED_GAP_TIME = 0.7; static constexpr double MAX_PTS_SAVE_DELAY = 0.05; static constexpr double UPDATE_APPLY_WARNING_TIME = 0.25; static constexpr bool DROP_PTS_UPDATES = false; static constexpr const char *AFTER_GET_DIFFERENCE_SOURCE = "after get difference"; static constexpr int32 AUDIO_TRANSCRIPTION_TIMEOUT = 60; friend class OnUpdate; class PendingPtsUpdate { public: tl_object_ptr update; int32 pts; int32 pts_count; double receive_time; Promise promise; PendingPtsUpdate(tl_object_ptr &&update, int32 pts, int32 pts_count, double receive_time, Promise &&promise) : update(std::move(update)) , pts(pts) , pts_count(pts_count) , receive_time(receive_time) , promise(std::move(promise)) { } }; class PendingSeqUpdates { public: int32 seq_begin; int32 seq_end; int32 date; double receive_time; vector> updates; Promise promise; PendingSeqUpdates(int32 seq_begin, int32 seq_end, int32 date, double receive_time, vector> &&updates, Promise &&promise) : seq_begin(seq_begin) , seq_end(seq_end) , date(date) , receive_time(receive_time) , updates(std::move(updates)) , promise(std::move(promise)) { } }; class PendingQtsUpdate { public: double receive_time = 0.0; tl_object_ptr update; vector> promises; }; Td *td_; ActorShared<> parent_; int32 ref_cnt_ = 1; PtsManager pts_manager_; PtsManager qts_manager_; int32 date_ = 0; int32 seq_ = 0; string date_source_ = "nowhere"; double last_pts_save_time_ = 0; double last_qts_save_time_ = 0; int32 pending_pts_ = 0; int32 pending_qts_ = 0; int64 being_processed_updates_ = 0; int32 short_update_date_ = 0; int32 accumulated_pts_count_ = 0; int32 accumulated_pts_ = -1; double last_pts_jump_warning_time_ = 0; double last_pts_gap_time_ = 0; std::multimap pending_pts_updates_; std::multimap postponed_pts_updates_; std::multimap postponed_updates_; // updates received during getDifference std::multimap pending_seq_updates_; // updates with too big seq std::map pending_qts_updates_; // updates with too big QTS Timeout pts_gap_timeout_; Timeout seq_gap_timeout_; Timeout qts_gap_timeout_; int32 retry_time_ = 1; Timeout retry_timeout_; double next_data_reload_time_ = 0.0; Timeout data_reload_timeout_; bool is_ping_sent_ = false; bool running_get_difference_ = false; bool finished_first_get_difference_ = false; int32 last_get_difference_pts_ = 0; int32 last_get_difference_qts_ = 0; int32 min_postponed_update_pts_ = 0; int32 min_postponed_update_qts_ = 0; double get_difference_start_time_ = 0; // time from which we started to get difference without success FlatHashMap pending_audio_transcriptions_; MultiTimeout pending_audio_transcription_timeout_{"PendingAudioTranscriptionTimeout"}; struct SessionInfo { uint64 update_count = 0; double first_update_time = 0.0; double last_update_time = 0.0; }; FlatHashMap session_infos_; void start_up() final; void tear_down() final; void hangup_shared() final; void hangup() final; ActorShared create_reference(); int32 get_pts() const { return pts_manager_.mem_pts(); } int32 get_qts() const { return qts_manager_.mem_pts(); } int32 get_date() const { return date_; } Promise<> set_pts(int32 pts, const char *source) TD_WARN_UNUSED_RESULT; Promise<> add_pts(int32 pts); void on_pts_ack(PtsManager::PtsId ack_token); void save_pts(int32 pts); Promise<> add_qts(int32 qts); void on_qts_ack(PtsManager::PtsId ack_token); void save_qts(int32 qts); bool can_postpone_updates() const { return finished_first_get_difference_; } void set_date(int32 date, bool from_update, string date_source); int32 get_short_update_date() const; void init_state(); void on_get_updates_state(tl_object_ptr &&state, const char *source); void on_server_pong(tl_object_ptr &&state); void on_get_difference(tl_object_ptr &&difference_ptr); void process_get_difference_updates(vector> &&new_messages, vector> &&new_encrypted_messages, vector> &&other_updates); void on_pending_update(tl_object_ptr update, int32 seq, Promise &&promise, const char *source); void add_pending_qts_update(tl_object_ptr &&update, int32 qts, Promise &&promise); void on_pending_updates(vector> &&updates, int32 seq_begin, int32 seq_end, int32 date, double receive_time, Promise &&promise, const char *source); void on_pending_updates_processed(Result result, Promise promise); void process_updates(vector> &&updates, bool force_apply, Promise &&promise); void postpone_pts_update(tl_object_ptr &&update, int32 pts, int32 pts_count, double receive_time, Promise &&promise); void process_pts_update(tl_object_ptr &&update); void process_seq_updates(int32 seq_end, int32 date, vector> &&updates, Promise &&promise); void process_qts_update(tl_object_ptr &&update_ptr, int32 qts, Promise &&promise); void process_all_pending_pts_updates(); void drop_all_pending_pts_updates(); void process_postponed_pts_updates(); void process_pending_pts_updates(); void process_pending_seq_updates(); void process_pending_qts_updates(); static void fill_pts_gap(void *td); static void fill_seq_gap(void *td); static void fill_qts_gap(void *td); static void fill_get_difference_gap(void *td); static void fill_gap(void *td, const char *source); static void on_pending_audio_transcription_timeout_callback(void *td, int64 transcription_id); void set_pts_gap_timeout(double timeout); void set_seq_gap_timeout(double timeout); void set_qts_gap_timeout(double timeout); void run_get_difference(bool is_recursive, const char *source); void on_failed_get_updates_state(Status &&error); void on_failed_get_difference(Status &&error); void before_get_difference(bool is_initial); void after_get_difference(); void schedule_data_reload(); static void try_reload_data_static(void *td); void try_reload_data(); uint64 get_most_unused_auth_key_id(); static vector get_update_ids(const telegram_api::Updates *updates_ptr); static bool have_update_pts_changed(const vector> &updates); static bool check_pts_update_dialog_id(DialogId dialog_id); static bool check_pts_update(const tl_object_ptr &update); static bool is_pts_update(const telegram_api::Update *update); static int32 get_update_pts(const telegram_api::Update *update); static bool is_qts_update(const telegram_api::Update *update); static int32 get_update_qts(const telegram_api::Update *update); static bool is_channel_pts_update(const telegram_api::Update *update); static bool is_additional_service_message(const telegram_api::Message *message); static const vector> *get_updates(const telegram_api::Updates *updates_ptr); static vector> *get_updates(telegram_api::Updates *updates_ptr); void on_pending_audio_transcription_failed(int64 transcription_id, Status &&error); bool is_acceptable_user(UserId user_id) const; bool is_acceptable_chat(ChatId chat_id) const; bool is_acceptable_channel(ChannelId channel_id) const; bool is_acceptable_peer(const tl_object_ptr &peer) const; bool is_acceptable_message_entities(const vector> &message_entities) const; bool is_acceptable_reply_markup(const tl_object_ptr &reply_markup) const; bool is_acceptable_message_reply_header( const telegram_api::object_ptr &header) const; bool is_acceptable_message_forward_header( const telegram_api::object_ptr &header) const; bool is_acceptable_message(const telegram_api::Message *message_ptr) const; bool is_acceptable_update(const telegram_api::Update *update) const; void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); void on_update(tl_object_ptr update, Promise &&promise); // unsupported updates }; } // namespace td