This repository has been archived on 2020-05-25. You can view files and clone it, but cannot push or open issues or pull requests.
tdlib-fork/td/telegram/logevent/SecretChatEvent.h
Arseny Smirnov e5c90f239f Secret chats: check that his_layer is monotonic
GitOrigin-RevId: cb6b50e768a90c5afcece185bea2e7209a9939f5
2018-06-14 18:00:10 +03:00

461 lines
13 KiB
C++

//
// Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2018
//
// 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/logevent/LogEvent.h"
#include "td/utils/buffer.h"
#include "td/utils/format.h"
#include "td/utils/logging.h"
#include "td/utils/Storer.h"
#include "td/utils/StringBuilder.h"
#include "td/utils/tl_helpers.h"
#include "td/telegram/secret_api.h"
#include "td/telegram/telegram_api.h"
namespace td {
namespace logevent {
class SecretChatEvent : public LogEventBase<SecretChatEvent> {
public:
// append only enum
enum class Type : int32 {
InboundSecretMessage = 1,
OutboundSecretMessage = 2,
CloseSecretChat = 3,
CreateSecretChat = 4
};
virtual Type get_type() const = 0;
static constexpr LogEvent::HandlerType get_handler_type() {
return LogEvent::HandlerType::SecretChats;
}
static constexpr int32 version() {
return 2;
}
template <class F>
static void downcast_call(Type type, F &&f);
};
// Internal structure
// inputEncryptedFileEmpty#1837c364 = InputEncryptedFile;
// inputEncryptedFileUploaded#64bd0306 id:long parts:int md5_checksum:string key_fingerprint:int = InputEncryptedFile;
// inputEncryptedFile#5a17b5e5 id:long access_hash:long = InputEncryptedFile;
// inputEncryptedFileBigUploaded#2dc173c8 id:long parts:int key_fingerprint:int = InputEncryptedFile;
struct EncryptedInputFile {
static constexpr int32 magic = 0x4328d38a;
enum Type : int32 { Empty = 0, Uploaded = 1, BigUploaded = 2, Location = 3 } type;
int64 id;
int64 access_hash;
int32 parts;
int32 key_fingerprint;
template <class T>
void store(T &storer) const {
using td::store;
store(magic, storer);
store(type, storer);
store(id, storer);
store(access_hash, storer);
store(parts, storer);
store(key_fingerprint, storer);
}
bool empty() const {
return type == Empty;
}
template <class T>
void parse(T &parser) {
using td::parse;
int32 got_magic;
parse(got_magic, parser);
parse(type, parser);
parse(id, parser);
parse(access_hash, parser);
parse(parts, parser);
parse(key_fingerprint, parser);
if (got_magic != magic) {
parser.set_error("EncryptedInputFile magic mismatch");
return;
}
}
static EncryptedInputFile from_input_encrypted_file(const tl_object_ptr<telegram_api::InputEncryptedFile> &from) {
if (!from) {
return EncryptedInputFile{Empty, 0, 0, 0, 0};
}
return from_input_encrypted_file(*from);
}
static EncryptedInputFile from_input_encrypted_file(const telegram_api::InputEncryptedFile &from) {
switch (from.get_id()) {
case telegram_api::inputEncryptedFileEmpty::ID:
return EncryptedInputFile{Empty, 0, 0, 0, 0};
case telegram_api::inputEncryptedFileUploaded::ID: {
auto &uploaded = static_cast<const telegram_api::inputEncryptedFileUploaded &>(from);
return EncryptedInputFile{Uploaded, uploaded.id_, 0, uploaded.parts_, uploaded.key_fingerprint_};
}
case telegram_api::inputEncryptedFileBigUploaded::ID: {
auto &uploaded = static_cast<const telegram_api::inputEncryptedFileBigUploaded &>(from);
return EncryptedInputFile{BigUploaded, uploaded.id_, 0, uploaded.parts_, uploaded.key_fingerprint_};
}
case telegram_api::inputEncryptedFile::ID: {
auto &uploaded = static_cast<const telegram_api::inputEncryptedFile &>(from);
return EncryptedInputFile{Location, uploaded.id_, uploaded.access_hash_, 0, 0};
}
default:
UNREACHABLE();
}
}
tl_object_ptr<telegram_api::InputEncryptedFile> as_input_encrypted_file() const {
switch (type) {
case Empty:
return make_tl_object<telegram_api::inputEncryptedFileEmpty>();
case Uploaded:
return make_tl_object<telegram_api::inputEncryptedFileUploaded>(id, parts, "", key_fingerprint);
case BigUploaded:
return make_tl_object<telegram_api::inputEncryptedFileBigUploaded>(id, parts, key_fingerprint);
case Location:
return make_tl_object<telegram_api::inputEncryptedFile>(id, access_hash);
}
UNREACHABLE();
}
};
inline StringBuilder &operator<<(StringBuilder &sb, const EncryptedInputFile &file) {
return sb << to_string(file.as_input_encrypted_file());
}
// encryptedFile#4a70994c id:long access_hash:long size:int dc_id:int key_fingerprint:int = EncryptedFile;
struct EncryptedFileLocation {
static constexpr int32 magic = 0x473d738a;
int64 id;
int64 access_hash;
int32 size;
int32 dc_id;
int32 key_fingerprint;
tl_object_ptr<telegram_api::encryptedFile> as_encrypted_file() {
return make_tl_object<telegram_api::encryptedFile>(id, access_hash, size, dc_id, key_fingerprint);
}
template <class T>
void store(T &storer) const {
using td::store;
store(magic, storer);
store(id, storer);
store(access_hash, storer);
store(size, storer);
store(dc_id, storer);
store(key_fingerprint, storer);
}
template <class T>
void parse(T &parser) {
using td::parse;
int32 got_magic;
parse(got_magic, parser);
parse(id, parser);
parse(access_hash, parser);
parse(size, parser);
parse(dc_id, parser);
parse(key_fingerprint, parser);
if (got_magic != magic) {
parser.set_error("EncryptedFileLocation magic mismatch");
return;
}
}
};
inline StringBuilder &operator<<(StringBuilder &sb, const EncryptedFileLocation &file) {
return sb << "[" << tag("id", file.id) << tag("access_hash", file.access_hash) << tag("size", file.size)
<< tag("dc_id", file.dc_id) << tag("key_fingerprint", file.key_fingerprint) << "]";
}
// LogEvents
// TODO: Qts and SeqNoState could be just Logevents that are updated during regenerate
class InboundSecretMessage : public LogEventHelper<InboundSecretMessage, SecretChatEvent> {
public:
static constexpr Type type = SecretChatEvent::Type::InboundSecretMessage;
int32 qts = 0;
int32 chat_id = 0;
int32 date = 0;
BufferSlice encrypted_message; // empty when we store event to binlog
Promise<Unit> qts_ack;
bool is_checked = false;
// after decrypted and checked
tl_object_ptr<secret_api::decryptedMessageLayer> decrypted_message_layer;
uint64 auth_key_id = 0;
int32 message_id = 0;
int32 my_in_seq_no = -1;
int32 my_out_seq_no = -1;
int32 his_in_seq_no = -1;
int32 his_layer() const {
return decrypted_message_layer->layer_;
}
EncryptedFileLocation file;
bool has_encrypted_file;
bool is_pending = false;
template <class T>
void store(T &storer) const {
using td::store;
BEGIN_STORE_FLAGS();
STORE_FLAG(has_encrypted_file);
STORE_FLAG(is_pending);
END_STORE_FLAGS();
store(qts, storer);
store(chat_id, storer);
store(date, storer);
// skip encrypted_message
// skip qts_ack
// TODO
decrypted_message_layer->store(storer);
storer.store_long(static_cast<int64>(auth_key_id));
store(message_id, storer);
store(my_in_seq_no, storer);
store(my_out_seq_no, storer);
store(his_in_seq_no, storer);
if (has_encrypted_file) {
store(file, storer);
}
}
template <class T>
void parse(T &parser) {
using td::parse;
BEGIN_PARSE_FLAGS();
PARSE_FLAG(has_encrypted_file);
PARSE_FLAG(is_pending);
END_PARSE_FLAGS();
parse(qts, parser);
parse(chat_id, parser);
parse(date, parser);
// skip encrypted_message
// skip qts_ack
// TODO
decrypted_message_layer = secret_api::decryptedMessageLayer::fetch(parser);
auth_key_id = static_cast<uint64>(parser.fetch_long());
parse(message_id, parser);
parse(my_in_seq_no, parser);
parse(my_out_seq_no, parser);
parse(his_in_seq_no, parser);
if (has_encrypted_file) {
parse(file, parser);
}
is_checked = true;
}
StringBuilder &print(StringBuilder &sb) const override {
return sb << "[Logevent InboundSecretMessage " << tag("id", logevent_id())
<< tag("auth_key_id", format::as_hex(auth_key_id)) << tag("message_id", message_id)
<< tag("my_in_seq_no", my_in_seq_no) << tag("my_out_seq_no", my_out_seq_no)
<< tag("his_in_seq_no", his_in_seq_no) << tag("message", to_string(decrypted_message_layer))
<< tag("is_pending", is_pending) << format::cond(has_encrypted_file, tag("file", file)) << "]";
}
};
class OutboundSecretMessage : public LogEventHelper<OutboundSecretMessage, SecretChatEvent> {
public:
static constexpr Type type = SecretChatEvent::Type::OutboundSecretMessage;
int32 chat_id = 0;
int64 random_id = 0;
BufferSlice encrypted_message;
EncryptedInputFile file;
int32 message_id = 0;
int32 my_in_seq_no = -1;
int32 my_out_seq_no = -1;
int32 his_in_seq_no = -1;
int32 his_layer() const {
return -1;
}
bool is_sent = false;
bool is_service = false;
bool is_rewritable = false;
bool is_external = false;
tl_object_ptr<secret_api::DecryptedMessageAction> action;
uint64 crc = 0; // DEBUG;
// Flags:
// 2. can_fail = !file.empty() // send of other messages can't fail if chat is ok. It is usless to rewrite them with
// empty
// 1. is_service // use messages_sendEncryptedsService
// 3. can_rewrite_with_empty // false for almost all service messages
// TODO: combine this two functions into one macros hell. Or lambda hell.
template <class T>
void store(T &storer) const {
using td::store;
store(chat_id, storer);
store(random_id, storer);
store(encrypted_message, storer);
store(file, storer);
store(message_id, storer);
store(my_in_seq_no, storer);
store(my_out_seq_no, storer);
store(his_in_seq_no, storer);
bool has_action = static_cast<bool>(action);
BEGIN_STORE_FLAGS();
STORE_FLAG(is_sent);
STORE_FLAG(is_service);
STORE_FLAG(has_action);
STORE_FLAG(is_rewritable);
STORE_FLAG(is_external);
END_STORE_FLAGS();
if (has_action) {
CHECK(action);
// TODO
storer.store_int(action->get_id());
action->store(storer);
}
}
template <class T>
void parse(T &parser) {
using td::parse;
parse(chat_id, parser);
parse(random_id, parser);
parse(encrypted_message, parser);
parse(file, parser);
parse(message_id, parser);
parse(my_in_seq_no, parser);
parse(my_out_seq_no, parser);
parse(his_in_seq_no, parser);
bool has_action;
BEGIN_PARSE_FLAGS();
PARSE_FLAG(is_sent);
PARSE_FLAG(is_service);
PARSE_FLAG(has_action);
PARSE_FLAG(is_rewritable);
PARSE_FLAG(is_external);
END_PARSE_FLAGS();
if (has_action) {
// TODO:
action = secret_api::DecryptedMessageAction::fetch(parser);
}
}
StringBuilder &print(StringBuilder &sb) const override {
return sb << "[Logevent OutboundSecretMessage " << tag("id", logevent_id()) << tag("is_sent", is_sent)
<< tag("is_service", is_service) << tag("is_rewritable", is_rewritable) << tag("is_external", is_external)
<< tag("message_id", message_id) << tag("random_id", random_id) << tag("my_in_seq_no", my_in_seq_no)
<< tag("my_out_seq_no", my_out_seq_no) << tag("his_in_seq_no", his_in_seq_no) << tag("file", file)
<< tag("action", to_string(action)) << "]";
}
};
class CloseSecretChat : public LogEventHelper<CloseSecretChat, SecretChatEvent> {
public:
static constexpr Type type = SecretChatEvent::Type::CloseSecretChat;
int32 chat_id = 0;
template <class T>
void store(T &storer) const {
using td::store;
store(chat_id, storer);
}
template <class T>
void parse(T &parser) {
using td::parse;
parse(chat_id, parser);
}
StringBuilder &print(StringBuilder &sb) const override {
return sb << "[Logevent CloseSecretChat " << tag("id", logevent_id()) << tag("chat_id", chat_id) << "]";
}
};
class CreateSecretChat : public LogEventHelper<CreateSecretChat, SecretChatEvent> {
public:
static constexpr Type type = SecretChatEvent::Type::CreateSecretChat;
int32 random_id = 0;
int32 user_id = 0;
int64 user_access_hash = 0;
template <class T>
void store(T &storer) const {
using td::store;
store(random_id, storer);
store(user_id, storer);
store(user_access_hash, storer);
}
template <class T>
void parse(T &parser) {
using td::parse;
parse(random_id, parser);
parse(user_id, parser);
parse(user_access_hash, parser);
}
StringBuilder &print(StringBuilder &sb) const override {
return sb << "[Logevent CreateSecretChat " << tag("id", logevent_id()) << tag("chat_id", random_id) << "]";
}
};
template <class F>
void SecretChatEvent::downcast_call(Type type, F &&f) {
switch (type) {
case Type::InboundSecretMessage:
f(static_cast<InboundSecretMessage *>(nullptr));
break;
case Type::OutboundSecretMessage:
f(static_cast<OutboundSecretMessage *>(nullptr));
break;
case Type::CloseSecretChat:
f(static_cast<CloseSecretChat *>(nullptr));
break;
case Type::CreateSecretChat:
f(static_cast<CreateSecretChat *>(nullptr));
break;
default:
break;
}
}
} // namespace logevent
inline auto create_storer(logevent::SecretChatEvent &event) {
return logevent::detail::StorerImpl<logevent::SecretChatEvent>(event);
}
} // namespace td