mirror of https://github.com/procxx/kepka.git
105 lines
2.3 KiB
C++
105 lines
2.3 KiB
C++
/*
|
|
This file is part of Telegram Desktop,
|
|
the official desktop application for the Telegram messaging service.
|
|
|
|
For license and copyright information please follow this link:
|
|
https://github.com/telegramdesktop/tdesktop/blob/master/LEGAL
|
|
*/
|
|
#include "mtproto/dcenter.h"
|
|
|
|
#include "mtproto/facade.h"
|
|
#include "mtproto/mtproto_auth_key.h"
|
|
#include "mtproto/dc_options.h"
|
|
#include "mtproto/mtp_instance.h"
|
|
#include "mtproto/special_config_request.h"
|
|
#include "storage/localstorage.h"
|
|
|
|
namespace MTP {
|
|
namespace internal {
|
|
namespace {
|
|
|
|
constexpr auto kEnumerateDcTimeout = 8000; // 8 seconds timeout for help_getConfig to work (then move to other dc)
|
|
constexpr auto kSpecialRequestTimeoutMs = 6000; // 4 seconds timeout for it to work in a specially requested dc.
|
|
|
|
} // namespace
|
|
|
|
Dcenter::Dcenter(DcId dcId, AuthKeyPtr &&key)
|
|
: _id(dcId)
|
|
, _key(std::move(key)) {
|
|
}
|
|
|
|
DcId Dcenter::id() const {
|
|
return _id;
|
|
}
|
|
|
|
AuthKeyPtr Dcenter::getKey() const {
|
|
QReadLocker lock(&_mutex);
|
|
return _key;
|
|
}
|
|
|
|
void Dcenter::destroyCdnKey(uint64 keyId) {
|
|
destroyKey(keyId);
|
|
}
|
|
|
|
bool Dcenter::destroyConfirmedForgottenKey(uint64 keyId) {
|
|
return destroyKey(keyId);
|
|
}
|
|
|
|
bool Dcenter::destroyKey(uint64 keyId) {
|
|
Expects(!_creatingKey || !_key);
|
|
|
|
QWriteLocker lock(&_mutex);
|
|
if (_key->keyId() != keyId) {
|
|
return false;
|
|
}
|
|
_key = nullptr;
|
|
_connectionInited = false;
|
|
lock.unlock();
|
|
|
|
emit authKeyChanged();
|
|
return true;
|
|
}
|
|
|
|
bool Dcenter::connectionInited() const {
|
|
QReadLocker lock(&_mutex);
|
|
return _connectionInited;
|
|
}
|
|
|
|
void Dcenter::setConnectionInited(bool connectionInited) {
|
|
QWriteLocker lock(&_mutex);
|
|
_connectionInited = connectionInited;
|
|
}
|
|
|
|
bool Dcenter::acquireKeyCreation() {
|
|
QReadLocker lock(&_mutex);
|
|
if (_key != nullptr) {
|
|
return false;
|
|
}
|
|
auto expected = false;
|
|
return _creatingKey.compare_exchange_strong(expected, true);
|
|
}
|
|
|
|
void Dcenter::releaseKeyCreationOnFail() {
|
|
Expects(_creatingKey);
|
|
Expects(_key == nullptr);
|
|
|
|
_creatingKey = false;
|
|
}
|
|
|
|
void Dcenter::releaseKeyCreationOnDone(AuthKeyPtr &&key) {
|
|
Expects(_creatingKey);
|
|
Expects(_key == nullptr);
|
|
|
|
QWriteLocker lock(&_mutex);
|
|
DEBUG_LOG(("AuthKey Info: Dcenter::releaseKeyCreationOnDone(%1), emitting authKeyChanged, dc %2").arg(key ? key->keyId() : 0).arg(_id));
|
|
_key = std::move(key);
|
|
_connectionInited = false;
|
|
_creatingKey = false;
|
|
lock.unlock();
|
|
|
|
emit authKeyChanged();
|
|
}
|
|
|
|
} // namespace internal
|
|
} // namespace MTP
|