3045 lines
105 KiB
C++
3045 lines
105 KiB
C++
/*
|
|
This file is part of Telegram Desktop,
|
|
the official desktop version of Telegram messaging app, see https://telegram.org
|
|
|
|
Telegram Desktop is free software: you can redistribute it and/or modify
|
|
it under the terms of the GNU General Public License as published by
|
|
the Free Software Foundation, either version 3 of the License, or
|
|
(at your option) any later version.
|
|
|
|
It is distributed in the hope that it will be useful,
|
|
but WITHOUT ANY WARRANTY; without even the implied warranty of
|
|
MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
|
GNU General Public License for more details.
|
|
|
|
In addition, as a special exception, the copyright holders give permission
|
|
to link the code of portions of this program with the OpenSSL library.
|
|
|
|
Full license: https://github.com/telegramdesktop/tdesktop/blob/master/LICENSE
|
|
Copyright (c) 2014-2016 John Preston, https://desktop.telegram.org
|
|
*/
|
|
#include "stdafx.h"
|
|
|
|
#include "mtproto/connection.h"
|
|
|
|
#include <openssl/bn.h>
|
|
#include <openssl/err.h>
|
|
#include <openssl/aes.h>
|
|
#include <openssl/sha.h>
|
|
#include <openssl/md5.h>
|
|
#include <openssl/rand.h>
|
|
#include "zlib.h"
|
|
|
|
#include "mtproto/rsa_public_key.h"
|
|
|
|
using std::string;
|
|
|
|
namespace MTP {
|
|
namespace internal {
|
|
|
|
namespace {
|
|
|
|
bool parsePQ(const string &pqStr, string &pStr, string &qStr) {
|
|
if (pqStr.length() > 8) return false; // more than 64 bit pq
|
|
|
|
uint64 pq = 0, p, q;
|
|
const uchar *pqChars = (const uchar*)&pqStr[0];
|
|
for (uint32 i = 0, l = pqStr.length(); i < l; ++i) {
|
|
pq <<= 8;
|
|
pq |= (uint64)pqChars[i];
|
|
}
|
|
uint64 pqSqrt = (uint64)sqrtl((long double)pq), ySqr, y;
|
|
while (pqSqrt * pqSqrt > pq) --pqSqrt;
|
|
while (pqSqrt * pqSqrt < pq) ++pqSqrt;
|
|
for (ySqr = pqSqrt * pqSqrt - pq; ; ++pqSqrt, ySqr = pqSqrt * pqSqrt - pq) {
|
|
y = (uint64)sqrtl((long double)ySqr);
|
|
while (y * y > ySqr) --y;
|
|
while (y * y < ySqr) ++y;
|
|
if (!ySqr || y + pqSqrt >= pq) return false;
|
|
if (y * y == ySqr) {
|
|
p = pqSqrt + y;
|
|
q = (pqSqrt > y) ? (pqSqrt - y) : (y - pqSqrt);
|
|
break;
|
|
}
|
|
}
|
|
if (p > q) std::swap(p, q);
|
|
|
|
pStr.resize(4);
|
|
uchar *pChars = (uchar*)&pStr[0];
|
|
for (uint32 i = 0; i < 4; ++i) {
|
|
*(pChars + 3 - i) = (uchar)(p & 0xFF);
|
|
p >>= 8;
|
|
}
|
|
|
|
qStr.resize(4);
|
|
uchar *qChars = (uchar*)&qStr[0];
|
|
for (uint32 i = 0; i < 4; ++i) {
|
|
*(qChars + 3 - i) = (uchar)(q & 0xFF);
|
|
q >>= 8;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
class BigNumCounter {
|
|
public:
|
|
bool count(const void *power, const void *modul, uint32 g, void *gResult, const void *g_a, void *g_aResult) {
|
|
DEBUG_LOG(("BigNum Info: counting g_b = g ^ b % dh_prime and auth_key = g_a ^ b % dh_prime"));
|
|
uint32 g_be = qToBigEndian(g);
|
|
if (
|
|
!BN_bin2bn((const uchar*)power, 64 * sizeof(uint32), &bnPower) ||
|
|
!BN_bin2bn((const uchar*)modul, 64 * sizeof(uint32), &bnModul) ||
|
|
!BN_bin2bn((const uchar*)&g_be, sizeof(uint32), &bn_g) ||
|
|
!BN_bin2bn((const uchar*)g_a, 64 * sizeof(uint32), &bn_g_a)
|
|
) {
|
|
ERR_load_crypto_strings();
|
|
LOG(("BigNum Error: BN_bin2bn failed, error: %1").arg(ERR_error_string(ERR_get_error(), 0)));
|
|
DEBUG_LOG(("BigNum Error: base %1, power %2, modul %3").arg(Logs::mb(&g_be, sizeof(uint32)).str()).arg(Logs::mb(power, 64 * sizeof(uint32)).str()).arg(Logs::mb(modul, 64 * sizeof(uint32)).str()));
|
|
return false;
|
|
}
|
|
|
|
if (!BN_mod_exp(&bnResult, &bn_g, &bnPower, &bnModul, ctx)) {
|
|
ERR_load_crypto_strings();
|
|
LOG(("BigNum Error: BN_mod_exp failed, error: %1").arg(ERR_error_string(ERR_get_error(), 0)));
|
|
DEBUG_LOG(("BigNum Error: base %1, power %2, modul %3").arg(Logs::mb(&g_be, sizeof(uint32)).str()).arg(Logs::mb(power, 64 * sizeof(uint32)).str()).arg(Logs::mb(modul, 64 * sizeof(uint32)).str()));
|
|
return false;
|
|
}
|
|
|
|
uint32 resultLen = BN_num_bytes(&bnResult);
|
|
if (resultLen != 64 * sizeof(uint32)) {
|
|
DEBUG_LOG(("BigNum Error: bad gResult len (%1)").arg(resultLen));
|
|
return false;
|
|
}
|
|
resultLen = BN_bn2bin(&bnResult, (uchar*)gResult);
|
|
if (resultLen != 64 * sizeof(uint32)) {
|
|
DEBUG_LOG(("BigNum Error: bad gResult export len (%1)").arg(resultLen));
|
|
return false;
|
|
}
|
|
|
|
BN_add_word(&bnResult, 1); // check g_b < dh_prime - 1
|
|
if (BN_cmp(&bnResult, &bnModul) >= 0) {
|
|
DEBUG_LOG(("BigNum Error: bad g_b >= dh_prime - 1"));
|
|
return false;
|
|
}
|
|
|
|
if (!BN_mod_exp(&bnResult, &bn_g_a, &bnPower, &bnModul, ctx)) {
|
|
ERR_load_crypto_strings();
|
|
LOG(("BigNum Error: BN_mod_exp failed, error: %1").arg(ERR_error_string(ERR_get_error(), 0)));
|
|
DEBUG_LOG(("BigNum Error: base %1, power %2, modul %3").arg(Logs::mb(&g_be, sizeof(uint32)).str()).arg(Logs::mb(power, 64 * sizeof(uint32)).str()).arg(Logs::mb(modul, 64 * sizeof(uint32)).str()));
|
|
return false;
|
|
}
|
|
|
|
resultLen = BN_num_bytes(&bnResult);
|
|
if (resultLen != 64 * sizeof(uint32)) {
|
|
DEBUG_LOG(("BigNum Error: bad g_aResult len (%1)").arg(resultLen));
|
|
return false;
|
|
}
|
|
resultLen = BN_bn2bin(&bnResult, (uchar*)g_aResult);
|
|
if (resultLen != 64 * sizeof(uint32)) {
|
|
DEBUG_LOG(("BigNum Error: bad g_aResult export len (%1)").arg(resultLen));
|
|
return false;
|
|
}
|
|
|
|
BN_add_word(&bn_g_a, 1); // check g_a < dh_prime - 1
|
|
if (BN_cmp(&bn_g_a, &bnModul) >= 0) {
|
|
DEBUG_LOG(("BigNum Error: bad g_a >= dh_prime - 1"));
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
BigNumCounter() : ctx(BN_CTX_new()) {
|
|
BN_init(&bnPower);
|
|
BN_init(&bnModul);
|
|
BN_init(&bn_g);
|
|
BN_init(&bn_g_a);
|
|
BN_init(&bnResult);
|
|
}
|
|
~BigNumCounter() {
|
|
BN_CTX_free(ctx);
|
|
BN_clear_free(&bnPower);
|
|
BN_clear_free(&bnModul);
|
|
BN_clear_free(&bn_g);
|
|
BN_clear_free(&bn_g_a);
|
|
BN_clear_free(&bnResult);
|
|
}
|
|
|
|
private:
|
|
BIGNUM bnPower, bnModul, bn_g, bn_g_a, bnResult;
|
|
BN_CTX *ctx;
|
|
};
|
|
|
|
// Miller-Rabin primality test
|
|
class BigNumPrimeTest {
|
|
public:
|
|
|
|
bool isPrimeAndGood(const void *pData, uint32 iterCount, int32 g) {
|
|
if (!memcmp(pData, "\xC7\x1C\xAE\xB9\xC6\xB1\xC9\x04\x8E\x6C\x52\x2F\x70\xF1\x3F\x73\x98\x0D\x40\x23\x8E\x3E\x21\xC1\x49\x34\xD0\x37\x56\x3D\x93\x0F\x48\x19\x8A\x0A\xA7\xC1\x40\x58\x22\x94\x93\xD2\x25\x30\xF4\xDB\xFA\x33\x6F\x6E\x0A\xC9\x25\x13\x95\x43\xAE\xD4\x4C\xCE\x7C\x37\x20\xFD\x51\xF6\x94\x58\x70\x5A\xC6\x8C\xD4\xFE\x6B\x6B\x13\xAB\xDC\x97\x46\x51\x29\x69\x32\x84\x54\xF1\x8F\xAF\x8C\x59\x5F\x64\x24\x77\xFE\x96\xBB\x2A\x94\x1D\x5B\xCD\x1D\x4A\xC8\xCC\x49\x88\x07\x08\xFA\x9B\x37\x8E\x3C\x4F\x3A\x90\x60\xBE\xE6\x7C\xF9\xA4\xA4\xA6\x95\x81\x10\x51\x90\x7E\x16\x27\x53\xB5\x6B\x0F\x6B\x41\x0D\xBA\x74\xD8\xA8\x4B\x2A\x14\xB3\x14\x4E\x0E\xF1\x28\x47\x54\xFD\x17\xED\x95\x0D\x59\x65\xB4\xB9\xDD\x46\x58\x2D\xB1\x17\x8D\x16\x9C\x6B\xC4\x65\xB0\xD6\xFF\x9C\xA3\x92\x8F\xEF\x5B\x9A\xE4\xE4\x18\xFC\x15\xE8\x3E\xBE\xA0\xF8\x7F\xA9\xFF\x5E\xED\x70\x05\x0D\xED\x28\x49\xF4\x7B\xF9\x59\xD9\x56\x85\x0C\xE9\x29\x85\x1F\x0D\x81\x15\xF6\x35\xB1\x05\xEE\x2E\x4E\x15\xD0\x4B\x24\x54\xBF\x6F\x4F\xAD\xF0\x34\xB1\x04\x03\x11\x9C\xD8\xE3\xB9\x2F\xCC\x5B", 256)) {
|
|
if (g == 3 || g == 4 || g == 5 || g == 7) {
|
|
return true;
|
|
}
|
|
}
|
|
if (
|
|
!BN_bin2bn((const uchar*)pData, 64 * sizeof(uint32), &bnPrime)
|
|
) {
|
|
ERR_load_crypto_strings();
|
|
LOG(("BigNum PT Error: BN_bin2bn failed, error: %1").arg(ERR_error_string(ERR_get_error(), 0)));
|
|
DEBUG_LOG(("BigNum PT Error: prime %1").arg(Logs::mb(pData, 64 * sizeof(uint32)).str()));
|
|
return false;
|
|
}
|
|
|
|
int32 numBits = BN_num_bits(&bnPrime);
|
|
if (numBits != 2048) {
|
|
LOG(("BigNum PT Error: BN_bin2bn failed, bad dh_prime num bits: %1").arg(numBits));
|
|
return false;
|
|
}
|
|
|
|
if (BN_is_prime_ex(&bnPrime, MTPMillerRabinIterCount, ctx, NULL) == 0) {
|
|
return false;
|
|
}
|
|
|
|
switch (g) {
|
|
case 2: {
|
|
int32 mod8 = BN_mod_word(&bnPrime, 8);
|
|
if (mod8 != 7) {
|
|
LOG(("BigNum PT Error: bad g value: %1, mod8: %2").arg(g).arg(mod8));
|
|
return false;
|
|
}
|
|
} break;
|
|
case 3: {
|
|
int32 mod3 = BN_mod_word(&bnPrime, 3);
|
|
if (mod3 != 2) {
|
|
LOG(("BigNum PT Error: bad g value: %1, mod3: %2").arg(g).arg(mod3));
|
|
return false;
|
|
}
|
|
} break;
|
|
case 4: break;
|
|
case 5: {
|
|
int32 mod5 = BN_mod_word(&bnPrime, 5);
|
|
if (mod5 != 1 && mod5 != 4) {
|
|
LOG(("BigNum PT Error: bad g value: %1, mod5: %2").arg(g).arg(mod5));
|
|
return false;
|
|
}
|
|
} break;
|
|
case 6: {
|
|
int32 mod24 = BN_mod_word(&bnPrime, 24);
|
|
if (mod24 != 19 && mod24 != 23) {
|
|
LOG(("BigNum PT Error: bad g value: %1, mod24: %2").arg(g).arg(mod24));
|
|
return false;
|
|
}
|
|
} break;
|
|
case 7: {
|
|
int32 mod7 = BN_mod_word(&bnPrime, 7);
|
|
if (mod7 != 3 && mod7 != 5 && mod7 != 6) {
|
|
LOG(("BigNum PT Error: bad g value: %1, mod7: %2").arg(g).arg(mod7));
|
|
return false;
|
|
}
|
|
} break;
|
|
default:
|
|
LOG(("BigNum PT Error: bad g value: %1").arg(g));
|
|
return false;
|
|
break;
|
|
}
|
|
|
|
BN_sub_word(&bnPrime, 1); // (p - 1) / 2
|
|
BN_div_word(&bnPrime, 2);
|
|
|
|
if (BN_is_prime_ex(&bnPrime, MTPMillerRabinIterCount, ctx, NULL) == 0) {
|
|
return false;
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
BigNumPrimeTest() : ctx(BN_CTX_new()) {
|
|
BN_init(&bnPrime);
|
|
}
|
|
~BigNumPrimeTest() {
|
|
BN_CTX_free(ctx);
|
|
BN_clear_free(&bnPrime);
|
|
}
|
|
|
|
private:
|
|
BIGNUM bnPrime;
|
|
BN_CTX *ctx;
|
|
};
|
|
|
|
typedef QMap<uint64, RSAPublicKey> RSAPublicKeys;
|
|
RSAPublicKeys InitRSAPublicKeys() {
|
|
DEBUG_LOG(("MTP Info: RSA public keys list creation"));
|
|
|
|
RSAPublicKeys result;
|
|
|
|
int keysCount;
|
|
const char **keys = cPublicRSAKeys(keysCount);
|
|
for (int i = 0; i < keysCount; ++i) {
|
|
RSAPublicKey key(keys[i]);
|
|
if (key.isValid()) {
|
|
result.insert(key.getFingerPrint(), key);
|
|
} else {
|
|
LOG(("MTP Error: could not read this public RSA key:"));
|
|
LOG((keys[i]));
|
|
}
|
|
}
|
|
DEBUG_LOG(("MTP Info: read %1 public RSA keys").arg(result.size()));
|
|
return result;
|
|
}
|
|
|
|
} // namespace
|
|
|
|
uint32 ThreadIdIncrement = 0;
|
|
|
|
Thread::Thread() : QThread(nullptr)
|
|
, _threadId(++ThreadIdIncrement) {
|
|
}
|
|
|
|
uint32 Thread::getThreadId() const {
|
|
return _threadId;
|
|
}
|
|
|
|
Thread::~Thread() {
|
|
}
|
|
|
|
Connection::Connection() : thread(nullptr), data(nullptr) {
|
|
}
|
|
|
|
int32 Connection::start(SessionData *sessionData, int32 dc) {
|
|
t_assert(thread == nullptr && data == nullptr);
|
|
|
|
thread = new Thread();
|
|
data = new ConnectionPrivate(thread, this, sessionData, dc);
|
|
|
|
dc = data->getDC();
|
|
if (!dc) {
|
|
delete data;
|
|
data = nullptr;
|
|
delete thread;
|
|
thread = nullptr;
|
|
return 0;
|
|
}
|
|
|
|
thread->start();
|
|
return dc;
|
|
}
|
|
|
|
void Connection::kill() {
|
|
t_assert(data != nullptr && thread != nullptr);
|
|
data->stop();
|
|
data = nullptr; // will be deleted in thread::finished signal
|
|
thread->quit();
|
|
queueQuittingConnection(this);
|
|
}
|
|
|
|
void Connection::waitTillFinish() {
|
|
t_assert(data == nullptr && thread != nullptr);
|
|
|
|
DEBUG_LOG(("Waiting for connectionThread to finish"));
|
|
thread->wait();
|
|
delete thread;
|
|
thread = nullptr;
|
|
}
|
|
|
|
int32 Connection::state() const {
|
|
t_assert(data != nullptr && thread != nullptr);
|
|
|
|
return data->getState();
|
|
}
|
|
|
|
QString Connection::transport() const {
|
|
t_assert(data != nullptr && thread != nullptr);
|
|
|
|
return data->transport();
|
|
}
|
|
|
|
Connection::~Connection() {
|
|
t_assert(data == nullptr && thread == nullptr);
|
|
}
|
|
|
|
void ConnectionPrivate::createConn(bool createIPv4, bool createIPv6) {
|
|
destroyConn();
|
|
if (createIPv4) {
|
|
QWriteLocker lock(&stateConnMutex);
|
|
_conn4 = AbstractConnection::create(thread());
|
|
connect(_conn4, SIGNAL(error(bool)), this, SLOT(onError4(bool)));
|
|
connect(_conn4, SIGNAL(receivedSome()), this, SLOT(onReceivedSome()));
|
|
}
|
|
if (createIPv6) {
|
|
QWriteLocker lock(&stateConnMutex);
|
|
_conn6 = AbstractConnection::create(thread());
|
|
connect(_conn6, SIGNAL(error(bool)), this, SLOT(onError6(bool)));
|
|
connect(_conn6, SIGNAL(receivedSome()), this, SLOT(onReceivedSome()));
|
|
}
|
|
firstSentAt = 0;
|
|
if (oldConnection) {
|
|
oldConnection = false;
|
|
DEBUG_LOG(("This connection marked as not old!"));
|
|
}
|
|
oldConnectionTimer.start(MTPConnectionOldTimeout);
|
|
}
|
|
|
|
void ConnectionPrivate::destroyConn(AbstractConnection **conn) {
|
|
if (conn) {
|
|
AbstractConnection *toDisconnect = nullptr;
|
|
|
|
{
|
|
QWriteLocker lock(&stateConnMutex);
|
|
if (*conn) {
|
|
toDisconnect = *conn;
|
|
disconnect(*conn, SIGNAL(connected()), nullptr, nullptr);
|
|
disconnect(*conn, SIGNAL(disconnected()), nullptr, nullptr);
|
|
disconnect(*conn, SIGNAL(error(bool)), nullptr, nullptr);
|
|
disconnect(*conn, SIGNAL(receivedData()), nullptr, nullptr);
|
|
disconnect(*conn, SIGNAL(receivedSome()), nullptr, nullptr);
|
|
*conn = nullptr;
|
|
}
|
|
}
|
|
if (toDisconnect) {
|
|
toDisconnect->disconnectFromServer();
|
|
toDisconnect->deleteLater();
|
|
}
|
|
} else {
|
|
destroyConn(&_conn4);
|
|
destroyConn(&_conn6);
|
|
_conn = nullptr;
|
|
}
|
|
}
|
|
|
|
ConnectionPrivate::ConnectionPrivate(QThread *thread, Connection *owner, SessionData *data, uint32 _dc) : QObject(nullptr)
|
|
, _state(DisconnectedState)
|
|
, _needSessionReset(false)
|
|
, dc(_dc)
|
|
, _owner(owner)
|
|
, _conn(nullptr)
|
|
, _conn4(nullptr)
|
|
, _conn6(nullptr)
|
|
, retryTimeout(1)
|
|
, oldConnection(true)
|
|
, _waitForReceived(MTPMinReceiveDelay)
|
|
, _waitForConnected(MTPMinConnectDelay)
|
|
, firstSentAt(-1)
|
|
, _pingId(0)
|
|
, _pingIdToSend(0)
|
|
, _pingSendAt(0)
|
|
, _pingMsgId(0)
|
|
, restarted(false)
|
|
, _finished(false)
|
|
, keyId(0)
|
|
// , sessionDataMutex(QReadWriteLock::Recursive)
|
|
, sessionData(data)
|
|
, myKeyLock(false)
|
|
, authKeyData(0)
|
|
, authKeyStrings(0) {
|
|
oldConnectionTimer.moveToThread(thread);
|
|
_waitForConnectedTimer.moveToThread(thread);
|
|
_waitForReceivedTimer.moveToThread(thread);
|
|
_waitForIPv4Timer.moveToThread(thread);
|
|
_pingSender.moveToThread(thread);
|
|
retryTimer.moveToThread(thread);
|
|
moveToThread(thread);
|
|
|
|
if (!dc) {
|
|
QReadLocker lock(dcOptionsMutex());
|
|
const auto &options(Global::DcOptions());
|
|
if (options.isEmpty()) {
|
|
LOG(("MTP Error: connect failed, no DCs"));
|
|
dc = 0;
|
|
return;
|
|
}
|
|
dc = options.cbegin().value().id;
|
|
DEBUG_LOG(("MTP Info: searching for any DC, %1 selected...").arg(dc));
|
|
}
|
|
|
|
connect(thread, SIGNAL(started()), this, SLOT(socketStart()));
|
|
connect(thread, SIGNAL(finished()), this, SLOT(doFinish()));
|
|
connect(this, SIGNAL(finished(Connection*)), globalSlotCarrier(), SLOT(connectionFinished(Connection*)), Qt::QueuedConnection);
|
|
|
|
connect(&retryTimer, SIGNAL(timeout()), this, SLOT(retryByTimer()));
|
|
connect(&_waitForConnectedTimer, SIGNAL(timeout()), this, SLOT(onWaitConnectedFailed()));
|
|
connect(&_waitForReceivedTimer, SIGNAL(timeout()), this, SLOT(onWaitReceivedFailed()));
|
|
connect(&_waitForIPv4Timer, SIGNAL(timeout()), this, SLOT(onWaitIPv4Failed()));
|
|
connect(&oldConnectionTimer, SIGNAL(timeout()), this, SLOT(onOldConnection()));
|
|
connect(&_pingSender, SIGNAL(timeout()), this, SLOT(onPingSender()));
|
|
connect(sessionData->owner(), SIGNAL(authKeyCreated()), this, SLOT(updateAuthKey()), Qt::QueuedConnection);
|
|
|
|
connect(sessionData->owner(), SIGNAL(needToRestart()), this, SLOT(restartNow()), Qt::QueuedConnection);
|
|
connect(this, SIGNAL(needToReceive()), sessionData->owner(), SLOT(tryToReceive()), Qt::QueuedConnection);
|
|
connect(this, SIGNAL(stateChanged(qint32)), sessionData->owner(), SLOT(onConnectionStateChange(qint32)), Qt::QueuedConnection);
|
|
connect(sessionData->owner(), SIGNAL(needToSend()), this, SLOT(tryToSend()), Qt::QueuedConnection);
|
|
connect(sessionData->owner(), SIGNAL(needToPing()), this, SLOT(onPingSendForce()), Qt::QueuedConnection);
|
|
connect(this, SIGNAL(sessionResetDone()), sessionData->owner(), SLOT(onResetDone()), Qt::QueuedConnection);
|
|
|
|
static bool _registered = false;
|
|
if (!_registered) {
|
|
_registered = true;
|
|
qRegisterMetaType<QVector<quint64> >("QVector<quint64>");
|
|
}
|
|
|
|
connect(this, SIGNAL(needToSendAsync()), sessionData->owner(), SLOT(needToResumeAndSend()), Qt::QueuedConnection);
|
|
connect(this, SIGNAL(sendAnythingAsync(quint64)), sessionData->owner(), SLOT(sendAnything(quint64)), Qt::QueuedConnection);
|
|
connect(this, SIGNAL(sendHttpWaitAsync()), sessionData->owner(), SLOT(sendAnything()), Qt::QueuedConnection);
|
|
connect(this, SIGNAL(sendPongAsync(quint64,quint64)), sessionData->owner(), SLOT(sendPong(quint64,quint64)), Qt::QueuedConnection);
|
|
connect(this, SIGNAL(sendMsgsStateInfoAsync(quint64, QByteArray)), sessionData->owner(), SLOT(sendMsgsStateInfo(quint64,QByteArray)), Qt::QueuedConnection);
|
|
connect(this, SIGNAL(resendAsync(quint64,quint64,bool,bool)), sessionData->owner(), SLOT(resend(quint64,quint64,bool,bool)), Qt::QueuedConnection);
|
|
connect(this, SIGNAL(resendManyAsync(QVector<quint64>,quint64,bool,bool)), sessionData->owner(), SLOT(resendMany(QVector<quint64>,quint64,bool,bool)), Qt::QueuedConnection);
|
|
connect(this, SIGNAL(resendAllAsync()), sessionData->owner(), SLOT(resendAll()));
|
|
}
|
|
|
|
void ConnectionPrivate::onConfigLoaded() {
|
|
socketStart(true);
|
|
}
|
|
|
|
int32 ConnectionPrivate::getDC() const {
|
|
return dc;
|
|
}
|
|
|
|
int32 ConnectionPrivate::getState() const {
|
|
QReadLocker lock(&stateConnMutex);
|
|
int32 result = _state;
|
|
if (_state < 0) {
|
|
if (retryTimer.isActive()) {
|
|
result = int32(getms(true) - retryWillFinish);
|
|
if (result >= 0) {
|
|
result = -1;
|
|
}
|
|
}
|
|
}
|
|
return result;
|
|
}
|
|
|
|
QString ConnectionPrivate::transport() const {
|
|
QReadLocker lock(&stateConnMutex);
|
|
if ((!_conn4 && !_conn6) || (_conn4 && _conn6) || (_state < 0)) {
|
|
return QString();
|
|
}
|
|
QString result = (_conn4 ? _conn4 : _conn6)->transport();
|
|
if (!result.isEmpty() && Global::TryIPv6()) result += (_conn4 ? "/IPv4" : "/IPv6");
|
|
return result;
|
|
}
|
|
|
|
bool ConnectionPrivate::setState(int32 state, int32 ifState) {
|
|
if (ifState != Connection::UpdateAlways) {
|
|
QReadLocker lock(&stateConnMutex);
|
|
if (_state != ifState) return false;
|
|
}
|
|
QWriteLocker lock(&stateConnMutex);
|
|
if (_state == state) return false;
|
|
_state = state;
|
|
if (state < 0) {
|
|
retryTimeout = -state;
|
|
retryTimer.start(retryTimeout);
|
|
retryWillFinish = getms(true) + retryTimeout;
|
|
}
|
|
emit stateChanged(state);
|
|
return true;
|
|
}
|
|
|
|
void ConnectionPrivate::resetSession() { // recreate all msg_id and msg_seqno
|
|
_needSessionReset = false;
|
|
|
|
QWriteLocker locker1(sessionData->haveSentMutex());
|
|
QWriteLocker locker2(sessionData->toResendMutex());
|
|
QWriteLocker locker3(sessionData->toSendMutex());
|
|
QWriteLocker locker4(sessionData->wereAckedMutex());
|
|
mtpRequestMap &haveSent(sessionData->haveSentMap());
|
|
mtpRequestIdsMap &toResend(sessionData->toResendMap());
|
|
mtpPreRequestMap &toSend(sessionData->toSendMap());
|
|
mtpRequestIdsMap &wereAcked(sessionData->wereAckedMap());
|
|
|
|
mtpMsgId newId = msgid();
|
|
mtpRequestMap setSeqNumbers;
|
|
typedef QMap<mtpMsgId, mtpMsgId> Replaces;
|
|
Replaces replaces;
|
|
for (mtpRequestMap::const_iterator i = haveSent.cbegin(), e = haveSent.cend(); i != e; ++i) {
|
|
if (!mtpRequestData::isSentContainer(i.value())) {
|
|
if (!*(mtpMsgId*)(i.value()->constData() + 4)) continue;
|
|
|
|
mtpMsgId id = i.key();
|
|
if (id > newId) {
|
|
while (true) {
|
|
if (toResend.constFind(newId) == toResend.cend() && wereAcked.constFind(newId) == wereAcked.cend() && haveSent.constFind(newId) == haveSent.cend()) {
|
|
break;
|
|
}
|
|
mtpMsgId m = msgid();
|
|
if (m <= newId) break; // wtf
|
|
|
|
newId = m;
|
|
}
|
|
|
|
MTP_LOG(dc, ("Replacing msgId %1 to %2!").arg(id).arg(newId));
|
|
replaces.insert(id, newId);
|
|
id = newId;
|
|
*(mtpMsgId*)(i.value()->data() + 4) = id;
|
|
}
|
|
setSeqNumbers.insert(id, i.value());
|
|
}
|
|
}
|
|
for (mtpRequestIdsMap::const_iterator i = toResend.cbegin(), e = toResend.cend(); i != e; ++i) { // collect all non-container requests
|
|
mtpPreRequestMap::const_iterator j = toSend.constFind(i.value());
|
|
if (j == toSend.cend()) continue;
|
|
|
|
if (!mtpRequestData::isSentContainer(j.value())) {
|
|
if (!*(mtpMsgId*)(j.value()->constData() + 4)) continue;
|
|
|
|
mtpMsgId id = i.key();
|
|
if (id > newId) {
|
|
while (true) {
|
|
if (toResend.constFind(newId) == toResend.cend() && wereAcked.constFind(newId) == wereAcked.cend() && haveSent.constFind(newId) == haveSent.cend()) {
|
|
break;
|
|
}
|
|
mtpMsgId m = msgid();
|
|
if (m <= newId) break; // wtf
|
|
|
|
newId = m;
|
|
}
|
|
|
|
MTP_LOG(dc, ("Replacing msgId %1 to %2!").arg(id).arg(newId));
|
|
replaces.insert(id, newId);
|
|
id = newId;
|
|
*(mtpMsgId*)(j.value()->data() + 4) = id;
|
|
}
|
|
setSeqNumbers.insert(id, j.value());
|
|
}
|
|
}
|
|
|
|
uint64 session = rand_value<uint64>();
|
|
DEBUG_LOG(("MTP Info: creating new session after bad_msg_notification, setting random server_session %1").arg(session));
|
|
sessionData->setSession(session);
|
|
|
|
for (mtpRequestMap::const_iterator i = setSeqNumbers.cbegin(), e = setSeqNumbers.cend(); i != e; ++i) { // generate new seq_numbers
|
|
bool wasNeedAck = (*(i.value()->data() + 6) & 1);
|
|
*(i.value()->data() + 6) = sessionData->nextRequestSeqNumber(wasNeedAck);
|
|
}
|
|
if (!replaces.isEmpty()) {
|
|
for (Replaces::const_iterator i = replaces.cbegin(), e = replaces.cend(); i != e; ++i) { // replace msgIds keys in all data structs
|
|
mtpRequestMap::iterator j = haveSent.find(i.key());
|
|
if (j != haveSent.cend()) {
|
|
mtpRequest req = j.value();
|
|
haveSent.erase(j);
|
|
haveSent.insert(i.value(), req);
|
|
}
|
|
mtpRequestIdsMap::iterator k = toResend.find(i.key());
|
|
if (k != toResend.cend()) {
|
|
mtpRequestId req = k.value();
|
|
toResend.erase(k);
|
|
toResend.insert(i.value(), req);
|
|
}
|
|
k = wereAcked.find(i.key());
|
|
if (k != wereAcked.cend()) {
|
|
mtpRequestId req = k.value();
|
|
wereAcked.erase(k);
|
|
wereAcked.insert(i.value(), req);
|
|
}
|
|
}
|
|
for (mtpRequestMap::const_iterator i = haveSent.cbegin(), e = haveSent.cend(); i != e; ++i) { // replace msgIds in saved containers
|
|
if (mtpRequestData::isSentContainer(i.value())) {
|
|
mtpMsgId *ids = (mtpMsgId *)(i.value()->data() + 8);
|
|
for (uint32 j = 0, l = (i.value()->size() - 8) >> 1; j < l; ++j) {
|
|
Replaces::const_iterator k = replaces.constFind(ids[j]);
|
|
if (k != replaces.cend()) {
|
|
ids[j] = k.value();
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
ackRequestData.clear();
|
|
resendRequestData.clear();
|
|
{
|
|
QWriteLocker locker5(sessionData->stateRequestMutex());
|
|
sessionData->stateRequestMap().clear();
|
|
}
|
|
|
|
emit sessionResetDone();
|
|
}
|
|
|
|
mtpMsgId ConnectionPrivate::prepareToSend(mtpRequest &request, mtpMsgId currentLastId) {
|
|
if (request->size() < 9) return 0;
|
|
mtpMsgId msgId = *(mtpMsgId*)(request->constData() + 4);
|
|
if (msgId) { // resending this request
|
|
QWriteLocker locker(sessionData->toResendMutex());
|
|
mtpRequestIdsMap &toResend(sessionData->toResendMap());
|
|
mtpRequestIdsMap::iterator i = toResend.find(msgId);
|
|
if (i != toResend.cend()) {
|
|
toResend.erase(i);
|
|
}
|
|
} else {
|
|
msgId = *(mtpMsgId*)(request->data() + 4) = currentLastId;
|
|
*(request->data() + 6) = sessionData->nextRequestSeqNumber(mtpRequestData::needAck(request));
|
|
}
|
|
return msgId;
|
|
}
|
|
|
|
mtpMsgId ConnectionPrivate::replaceMsgId(mtpRequest &request, mtpMsgId newId) {
|
|
if (request->size() < 9) return 0;
|
|
|
|
mtpMsgId oldMsgId = *(mtpMsgId*)(request->constData() + 4);
|
|
if (oldMsgId != newId) {
|
|
if (oldMsgId) {
|
|
QWriteLocker locker(sessionData->toResendMutex());
|
|
// haveSentMutex() and wereAckedMutex() were locked in tryToSend()
|
|
|
|
mtpRequestIdsMap &toResend(sessionData->toResendMap());
|
|
mtpRequestIdsMap &wereAcked(sessionData->wereAckedMap());
|
|
mtpRequestMap &haveSent(sessionData->haveSentMap());
|
|
|
|
while (true) {
|
|
if (toResend.constFind(newId) == toResend.cend() && wereAcked.constFind(newId) == wereAcked.cend() && haveSent.constFind(newId) == haveSent.cend()) {
|
|
break;
|
|
}
|
|
mtpMsgId m = msgid();
|
|
if (m <= newId) break; // wtf
|
|
|
|
newId = m;
|
|
}
|
|
|
|
mtpRequestIdsMap::iterator i = toResend.find(oldMsgId);
|
|
if (i != toResend.cend()) {
|
|
mtpRequestId req = i.value();
|
|
toResend.erase(i);
|
|
toResend.insert(newId, req);
|
|
}
|
|
|
|
mtpRequestIdsMap::iterator j = wereAcked.find(oldMsgId);
|
|
if (j != wereAcked.cend()) {
|
|
mtpRequestId req = j.value();
|
|
wereAcked.erase(j);
|
|
wereAcked.insert(newId, req);
|
|
}
|
|
|
|
mtpRequestMap::iterator k = haveSent.find(oldMsgId);
|
|
if (k != haveSent.cend()) {
|
|
mtpRequest req = k.value();
|
|
haveSent.erase(k);
|
|
haveSent.insert(newId, req);
|
|
}
|
|
|
|
for (k = haveSent.begin(); k != haveSent.cend(); ++k) {
|
|
mtpRequest req(k.value());
|
|
if (mtpRequestData::isSentContainer(req)) {
|
|
mtpMsgId *ids = (mtpMsgId *)(req->data() + 8);
|
|
for (uint32 i = 0, l = (req->size() - 8) >> 1; i < l; ++i) {
|
|
if (ids[i] == oldMsgId) {
|
|
ids[i] = newId;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
} else {
|
|
*(request->data() + 6) = sessionData->nextRequestSeqNumber(mtpRequestData::needAck(request));
|
|
}
|
|
*(mtpMsgId*)(request->data() + 4) = newId;
|
|
}
|
|
return newId;
|
|
}
|
|
|
|
mtpMsgId ConnectionPrivate::placeToContainer(mtpRequest &toSendRequest, mtpMsgId &bigMsgId, mtpMsgId *&haveSentArr, mtpRequest &req) {
|
|
mtpMsgId msgId = prepareToSend(req, bigMsgId);
|
|
if (msgId > bigMsgId) msgId = replaceMsgId(req, bigMsgId);
|
|
if (msgId >= bigMsgId) bigMsgId = msgid();
|
|
*(haveSentArr++) = msgId;
|
|
|
|
uint32 from = toSendRequest->size(), len = mtpRequestData::messageSize(req);
|
|
toSendRequest->resize(from + len);
|
|
memcpy(toSendRequest->data() + from, req->constData() + 4, len * sizeof(mtpPrime));
|
|
|
|
return msgId;
|
|
}
|
|
|
|
void ConnectionPrivate::tryToSend() {
|
|
QReadLocker lockFinished(&sessionDataMutex);
|
|
if (!sessionData || !_conn) {
|
|
return;
|
|
}
|
|
|
|
bool needsLayer = !sessionData->layerWasInited();
|
|
int32 state = getState();
|
|
bool prependOnly = (state != ConnectedState);
|
|
mtpRequest pingRequest;
|
|
if (dc == bareDcId(dc)) { // main session
|
|
if (!prependOnly && !_pingIdToSend && !_pingId && _pingSendAt <= getms(true)) {
|
|
_pingIdToSend = rand_value<mtpPingId>();
|
|
}
|
|
}
|
|
if (_pingIdToSend) {
|
|
if (prependOnly || dc != bareDcId(dc)) {
|
|
MTPPing ping(MTPping(MTP_long(_pingIdToSend)));
|
|
uint32 pingSize = ping.innerLength() >> 2; // copy from Session::send
|
|
pingRequest = mtpRequestData::prepare(pingSize);
|
|
ping.write(*pingRequest);
|
|
DEBUG_LOG(("MTP Info: sending ping, ping_id: %1").arg(_pingIdToSend));
|
|
} else {
|
|
MTPPing_delay_disconnect ping(MTP_long(_pingIdToSend), MTP_int(MTPPingDelayDisconnect));
|
|
uint32 pingSize = ping.innerLength() >> 2; // copy from Session::send
|
|
pingRequest = mtpRequestData::prepare(pingSize);
|
|
ping.write(*pingRequest);
|
|
DEBUG_LOG(("MTP Info: sending ping_delay_disconnect, ping_id: %1").arg(_pingIdToSend));
|
|
}
|
|
|
|
pingRequest->msDate = getms(true); // > 0 - can send without container
|
|
_pingSendAt = pingRequest->msDate + (MTPPingSendAfterAuto * 1000ULL);
|
|
pingRequest->requestId = 0; // dont add to haveSent / wereAcked maps
|
|
|
|
if (dc == bareDcId(dc) && !prependOnly) { // main session
|
|
_pingSender.start(MTPPingSendAfter * 1000);
|
|
}
|
|
|
|
_pingId = _pingIdToSend;
|
|
_pingIdToSend = 0;
|
|
} else {
|
|
if (prependOnly) {
|
|
DEBUG_LOG(("MTP Info: dc %1 not sending, waiting for Connected state, state: %2").arg(dc).arg(state));
|
|
return; // just do nothing, if is not connected yet
|
|
} else {
|
|
DEBUG_LOG(("MTP Info: dc %1 trying to send after ping, state: %2").arg(dc).arg(state));
|
|
}
|
|
}
|
|
|
|
mtpRequest ackRequest, resendRequest, stateRequest, httpWaitRequest;
|
|
if (!prependOnly && !ackRequestData.isEmpty()) {
|
|
MTPMsgsAck ack(MTP_msgs_ack(MTP_vector<MTPlong>(ackRequestData)));
|
|
|
|
ackRequest = mtpRequestData::prepare(ack.innerLength() >> 2);
|
|
ack.write(*ackRequest);
|
|
|
|
ackRequest->msDate = getms(true); // > 0 - can send without container
|
|
ackRequest->requestId = 0; // dont add to haveSent / wereAcked maps
|
|
|
|
ackRequestData.clear();
|
|
}
|
|
if (!prependOnly && !resendRequestData.isEmpty()) {
|
|
MTPMsgResendReq resend(MTP_msg_resend_req(MTP_vector<MTPlong>(resendRequestData)));
|
|
|
|
resendRequest = mtpRequestData::prepare(resend.innerLength() >> 2);
|
|
resend.write(*resendRequest);
|
|
|
|
resendRequest->msDate = getms(true); // > 0 - can send without container
|
|
resendRequest->requestId = 0; // dont add to haveSent / wereAcked maps
|
|
|
|
resendRequestData.clear();
|
|
}
|
|
if (!prependOnly) {
|
|
QVector<MTPlong> stateReq;
|
|
{
|
|
QWriteLocker locker(sessionData->stateRequestMutex());
|
|
mtpMsgIdsSet &ids(sessionData->stateRequestMap());
|
|
if (!ids.isEmpty()) {
|
|
stateReq.reserve(ids.size());
|
|
for (mtpMsgIdsSet::const_iterator i = ids.cbegin(), e = ids.cend(); i != e; ++i) {
|
|
stateReq.push_back(MTP_long(i.key()));
|
|
}
|
|
}
|
|
ids.clear();
|
|
}
|
|
if (!stateReq.isEmpty()) {
|
|
MTPMsgsStateReq req(MTP_msgs_state_req(MTP_vector<MTPlong>(stateReq)));
|
|
|
|
stateRequest = mtpRequestData::prepare(req.innerLength() >> 2);
|
|
req.write(*stateRequest);
|
|
|
|
stateRequest->msDate = getms(true); // > 0 - can send without container
|
|
stateRequest->requestId = reqid();// add to haveSent / wereAcked maps, but don't add to requestMap
|
|
}
|
|
if (_conn->usingHttpWait()) {
|
|
MTPHttpWait req(MTP_http_wait(MTP_int(100), MTP_int(30), MTP_int(25000)));
|
|
|
|
httpWaitRequest = mtpRequestData::prepare(req.innerLength() >> 2);
|
|
req.write(*httpWaitRequest);
|
|
|
|
httpWaitRequest->msDate = getms(true); // > 0 - can send without container
|
|
httpWaitRequest->requestId = 0; // dont add to haveSent / wereAcked maps
|
|
}
|
|
}
|
|
|
|
MTPInitConnection<mtpRequest> initWrapperImpl, *initWrapper = &initWrapperImpl;
|
|
int32 initSize = 0, initSizeInInts = 0;
|
|
if (needsLayer) {
|
|
initWrapperImpl = MTPInitConnection<mtpRequest>(MTP_int(ApiId), MTP_string(cApiDeviceModel()), MTP_string(cApiSystemVersion()), MTP_string(cApiAppVersion()), MTP_string(Sandbox::LangSystemISO()), mtpRequest());
|
|
initSizeInInts = (initWrapper->innerLength() >> 2) + 2;
|
|
initSize = initSizeInInts * sizeof(mtpPrime);
|
|
}
|
|
|
|
bool needAnyResponse = false;
|
|
mtpRequest toSendRequest;
|
|
{
|
|
QWriteLocker locker1(sessionData->toSendMutex());
|
|
|
|
mtpPreRequestMap toSendDummy, &toSend(prependOnly ? toSendDummy : sessionData->toSendMap());
|
|
if (prependOnly) locker1.unlock();
|
|
|
|
uint32 toSendCount = toSend.size();
|
|
if (pingRequest) ++toSendCount;
|
|
if (ackRequest) ++toSendCount;
|
|
if (resendRequest) ++toSendCount;
|
|
if (stateRequest) ++toSendCount;
|
|
if (httpWaitRequest) ++toSendCount;
|
|
|
|
if (!toSendCount) return; // nothing to send
|
|
|
|
mtpRequest first = pingRequest ? pingRequest : (ackRequest ? ackRequest : (resendRequest ? resendRequest : (stateRequest ? stateRequest : (httpWaitRequest ? httpWaitRequest : toSend.cbegin().value()))));
|
|
if (toSendCount == 1 && first->msDate > 0) { // if can send without container
|
|
toSendRequest = first;
|
|
if (!prependOnly) {
|
|
toSend.clear();
|
|
locker1.unlock();
|
|
}
|
|
|
|
mtpMsgId msgId = prepareToSend(toSendRequest, msgid());
|
|
if (pingRequest) {
|
|
_pingMsgId = msgId;
|
|
needAnyResponse = true;
|
|
} else if (resendRequest || stateRequest) {
|
|
needAnyResponse = true;
|
|
}
|
|
|
|
if (toSendRequest->requestId) {
|
|
if (mtpRequestData::needAck(toSendRequest)) {
|
|
toSendRequest->msDate = mtpRequestData::isStateRequest(toSendRequest) ? 0 : getms(true);
|
|
|
|
QWriteLocker locker2(sessionData->haveSentMutex());
|
|
mtpRequestMap &haveSent(sessionData->haveSentMap());
|
|
haveSent.insert(msgId, toSendRequest);
|
|
|
|
if (needsLayer && !toSendRequest->needsLayer) needsLayer = false;
|
|
if (toSendRequest->after) {
|
|
int32 toSendSize = toSendRequest.innerLength() >> 2;
|
|
mtpRequest wrappedRequest(mtpRequestData::prepare(toSendSize, toSendSize + 3)); // cons + msg_id
|
|
wrappedRequest->resize(4);
|
|
memcpy(wrappedRequest->data(), toSendRequest->constData(), 4 * sizeof(mtpPrime));
|
|
wrapInvokeAfter(wrappedRequest, toSendRequest, haveSent);
|
|
toSendRequest = wrappedRequest;
|
|
}
|
|
if (needsLayer) {
|
|
int32 noWrapSize = (toSendRequest.innerLength() >> 2), toSendSize = noWrapSize + initSizeInInts;
|
|
mtpRequest wrappedRequest(mtpRequestData::prepare(toSendSize));
|
|
memcpy(wrappedRequest->data(), toSendRequest->constData(), 7 * sizeof(mtpPrime)); // all except length
|
|
wrappedRequest->push_back(mtpc_invokeWithLayer);
|
|
wrappedRequest->push_back(MTP::internal::CurrentLayer);
|
|
initWrapper->write(*wrappedRequest);
|
|
wrappedRequest->resize(wrappedRequest->size() + noWrapSize);
|
|
memcpy(wrappedRequest->data() + wrappedRequest->size() - noWrapSize, toSendRequest->constData() + 8, noWrapSize * sizeof(mtpPrime));
|
|
toSendRequest = wrappedRequest;
|
|
}
|
|
|
|
needAnyResponse = true;
|
|
} else {
|
|
QWriteLocker locker3(sessionData->wereAckedMutex());
|
|
sessionData->wereAckedMap().insert(msgId, toSendRequest->requestId);
|
|
}
|
|
}
|
|
} else { // send in container
|
|
bool willNeedInit = false;
|
|
uint32 containerSize = 1 + 1, idsWrapSize = (toSendCount << 1); // cons + vector size, idsWrapSize - size of "request-like" wrap for msgId vector
|
|
if (pingRequest) containerSize += mtpRequestData::messageSize(pingRequest);
|
|
if (ackRequest) containerSize += mtpRequestData::messageSize(ackRequest);
|
|
if (resendRequest) containerSize += mtpRequestData::messageSize(resendRequest);
|
|
if (stateRequest) containerSize += mtpRequestData::messageSize(stateRequest);
|
|
if (httpWaitRequest) containerSize += mtpRequestData::messageSize(httpWaitRequest);
|
|
for (mtpPreRequestMap::iterator i = toSend.begin(), e = toSend.end(); i != e; ++i) {
|
|
containerSize += mtpRequestData::messageSize(i.value());
|
|
if (needsLayer && i.value()->needsLayer) {
|
|
containerSize += initSizeInInts;
|
|
willNeedInit = true;
|
|
}
|
|
}
|
|
mtpBuffer initSerialized;
|
|
if (willNeedInit) {
|
|
initSerialized.reserve(initSizeInInts);
|
|
initSerialized.push_back(mtpc_invokeWithLayer);
|
|
initSerialized.push_back(MTP::internal::CurrentLayer);
|
|
initWrapper->write(initSerialized);
|
|
}
|
|
toSendRequest = mtpRequestData::prepare(containerSize, containerSize + 3 * toSend.size()); // prepare container + each in invoke after
|
|
toSendRequest->push_back(mtpc_msg_container);
|
|
toSendRequest->push_back(toSendCount);
|
|
|
|
mtpMsgId bigMsgId = msgid(); // check for a valid container
|
|
|
|
QWriteLocker locker2(sessionData->haveSentMutex()); // the fact of this lock is used in replaceMsgId()
|
|
mtpRequestMap &haveSent(sessionData->haveSentMap());
|
|
|
|
QWriteLocker locker3(sessionData->wereAckedMutex()); // the fact of this lock is used in replaceMsgId()
|
|
mtpRequestIdsMap &wereAcked(sessionData->wereAckedMap());
|
|
|
|
mtpRequest haveSentIdsWrap(mtpRequestData::prepare(idsWrapSize)); // prepare "request-like" wrap for msgId vector
|
|
haveSentIdsWrap->requestId = 0;
|
|
haveSentIdsWrap->resize(haveSentIdsWrap->size() + idsWrapSize);
|
|
mtpMsgId *haveSentArr = (mtpMsgId*)(haveSentIdsWrap->data() + 8);
|
|
|
|
if (pingRequest) {
|
|
_pingMsgId = placeToContainer(toSendRequest, bigMsgId, haveSentArr, pingRequest);
|
|
needAnyResponse = true;
|
|
} else if (resendRequest || stateRequest) {
|
|
needAnyResponse = true;
|
|
}
|
|
for (mtpPreRequestMap::iterator i = toSend.begin(), e = toSend.end(); i != e; ++i) {
|
|
mtpRequest &req(i.value());
|
|
mtpMsgId msgId = prepareToSend(req, bigMsgId);
|
|
if (msgId > bigMsgId) msgId = replaceMsgId(req, bigMsgId);
|
|
if (msgId >= bigMsgId) bigMsgId = msgid();
|
|
*(haveSentArr++) = msgId;
|
|
bool added = false;
|
|
if (req->requestId) {
|
|
if (mtpRequestData::needAck(req)) {
|
|
req->msDate = mtpRequestData::isStateRequest(req) ? 0 : getms(true);
|
|
int32 reqNeedsLayer = (needsLayer && req->needsLayer) ? toSendRequest->size() : 0;
|
|
if (req->after) {
|
|
wrapInvokeAfter(toSendRequest, req, haveSent, reqNeedsLayer ? initSizeInInts : 0);
|
|
if (reqNeedsLayer) {
|
|
memcpy(toSendRequest->data() + reqNeedsLayer + 4, initSerialized.constData(), initSize);
|
|
*(toSendRequest->data() + reqNeedsLayer + 3) += initSize;
|
|
}
|
|
added = true;
|
|
} else if (reqNeedsLayer) {
|
|
toSendRequest->resize(reqNeedsLayer + initSizeInInts + mtpRequestData::messageSize(req));
|
|
memcpy(toSendRequest->data() + reqNeedsLayer, req->constData() + 4, 4 * sizeof(mtpPrime));
|
|
memcpy(toSendRequest->data() + reqNeedsLayer + 4, initSerialized.constData(), initSize);
|
|
memcpy(toSendRequest->data() + reqNeedsLayer + 4 + initSizeInInts, req->constData() + 8, req.innerLength());
|
|
*(toSendRequest->data() + reqNeedsLayer + 3) += initSize;
|
|
added = true;
|
|
}
|
|
haveSent.insert(msgId, req);
|
|
|
|
needAnyResponse = true;
|
|
} else {
|
|
wereAcked.insert(msgId, req->requestId);
|
|
}
|
|
}
|
|
if (!added) {
|
|
uint32 from = toSendRequest->size(), len = mtpRequestData::messageSize(req);
|
|
toSendRequest->resize(from + len);
|
|
memcpy(toSendRequest->data() + from, req->constData() + 4, len * sizeof(mtpPrime));
|
|
}
|
|
}
|
|
if (stateRequest) {
|
|
mtpMsgId msgId = placeToContainer(toSendRequest, bigMsgId, haveSentArr, stateRequest);
|
|
stateRequest->msDate = 0; // 0 for state request, do not request state of it
|
|
haveSent.insert(msgId, stateRequest);
|
|
}
|
|
if (resendRequest) placeToContainer(toSendRequest, bigMsgId, haveSentArr, resendRequest);
|
|
if (ackRequest) placeToContainer(toSendRequest, bigMsgId, haveSentArr, ackRequest);
|
|
if (httpWaitRequest) placeToContainer(toSendRequest, bigMsgId, haveSentArr, httpWaitRequest);
|
|
|
|
mtpMsgId contMsgId = prepareToSend(toSendRequest, bigMsgId);
|
|
*(mtpMsgId*)(haveSentIdsWrap->data() + 4) = contMsgId;
|
|
(*haveSentIdsWrap)[6] = 0; // for container, msDate = 0, seqNo = 0
|
|
haveSent.insert(contMsgId, haveSentIdsWrap);
|
|
toSend.clear();
|
|
}
|
|
}
|
|
mtpRequestData::padding(toSendRequest);
|
|
sendRequest(toSendRequest, needAnyResponse, lockFinished);
|
|
}
|
|
|
|
void ConnectionPrivate::retryByTimer() {
|
|
QReadLocker lockFinished(&sessionDataMutex);
|
|
if (!sessionData) return;
|
|
|
|
if (retryTimeout < 3) {
|
|
++retryTimeout;
|
|
} else if (retryTimeout == 3) {
|
|
retryTimeout = 1000;
|
|
} else if (retryTimeout < 64000) {
|
|
retryTimeout *= 2;
|
|
}
|
|
if (keyId == AuthKey::RecreateKeyId) {
|
|
if (sessionData->getKey()) {
|
|
unlockKey();
|
|
|
|
QWriteLocker lock(sessionData->keyMutex());
|
|
sessionData->owner()->destroyKey();
|
|
}
|
|
keyId = 0;
|
|
}
|
|
socketStart();
|
|
}
|
|
|
|
void ConnectionPrivate::restartNow() {
|
|
retryTimeout = 1;
|
|
retryTimer.stop();
|
|
restart();
|
|
}
|
|
|
|
void ConnectionPrivate::socketStart(bool afterConfig) {
|
|
if (_finished) {
|
|
DEBUG_LOG(("MTP Error: socketStart() called for finished connection!"));
|
|
return;
|
|
}
|
|
bool isDldDc = isDldDcId(dc);
|
|
if (isDldDc) { // using media_only addresses only if key for this dc is already created
|
|
QReadLocker lockFinished(&sessionDataMutex);
|
|
if (sessionData) {
|
|
if (!sessionData->getKey()) {
|
|
isDldDc = false;
|
|
}
|
|
}
|
|
|
|
}
|
|
int32 bareDc = bareDcId(dc);
|
|
|
|
static const int IPv4address = 0, IPv6address = 1;
|
|
static const int TcpProtocol = 0, HttpProtocol = 1;
|
|
MTPDdcOption::Flags flags[2][2] = { { 0 } };
|
|
string ip[2][2];
|
|
uint32 port[2][2] = { { 0 } };
|
|
{
|
|
QReadLocker lock(dcOptionsMutex());
|
|
const auto &options(Global::DcOptions());
|
|
int32 shifts[2][2][4] = {
|
|
{ // IPv4
|
|
{ // TCP IPv4
|
|
isDldDc ? (MTPDdcOption::Flag::f_media_only | MTPDdcOption::Flag::f_tcpo_only) : -1,
|
|
qFlags(MTPDdcOption::Flag::f_tcpo_only),
|
|
isDldDc ? qFlags(MTPDdcOption::Flag::f_media_only) : -1,
|
|
0
|
|
}, { // HTTP IPv4
|
|
-1,
|
|
-1,
|
|
isDldDc ? qFlags(MTPDdcOption::Flag::f_media_only) : -1,
|
|
0
|
|
},
|
|
}, { // IPv6
|
|
{ // TCP IPv6
|
|
isDldDc ? (MTPDdcOption::Flag::f_media_only | MTPDdcOption::Flag::f_tcpo_only | MTPDdcOption::Flag::f_ipv6) : -1,
|
|
MTPDdcOption::Flag::f_tcpo_only | MTPDdcOption::Flag::f_ipv6,
|
|
isDldDc ? (MTPDdcOption::Flag::f_media_only | MTPDdcOption::Flag::f_ipv6) : -1,
|
|
qFlags(MTPDdcOption::Flag::f_ipv6)
|
|
}, { // HTTP IPv6
|
|
-1,
|
|
-1,
|
|
isDldDc ? (MTPDdcOption::Flag::f_media_only | MTPDdcOption::Flag::f_ipv6) : -1,
|
|
qFlags(MTPDdcOption::Flag::f_ipv6)
|
|
},
|
|
},
|
|
};
|
|
for (int32 address = 0, acount = sizeof(shifts) / sizeof(shifts[0]); address < acount; ++address) {
|
|
for (int32 protocol = 0, pcount = sizeof(shifts[0]) / sizeof(shifts[0][0]); protocol < pcount; ++protocol) {
|
|
for (int32 shift = 0, scount = sizeof(shifts[0][0]) / sizeof(shifts[0][0][0]); shift < scount; ++shift) {
|
|
int32 mask = shifts[address][protocol][shift];
|
|
if (mask < 0) continue;
|
|
|
|
auto index = options.constFind(shiftDcId(bareDc, mask));
|
|
if (index != options.cend()) {
|
|
ip[address][protocol] = index->ip;
|
|
flags[address][protocol] = index->flags;
|
|
port[address][protocol] = index->port;
|
|
break;
|
|
}
|
|
}
|
|
}
|
|
}
|
|
}
|
|
bool noIPv4 = !port[IPv4address][HttpProtocol], noIPv6 = (!Global::TryIPv6() || !port[IPv6address][HttpProtocol]);
|
|
if (noIPv4 && noIPv6) {
|
|
if (afterConfig) {
|
|
if (noIPv4) LOG(("MTP Error: DC %1 options for IPv4 over HTTP not found right after config load!").arg(dc));
|
|
if (Global::TryIPv6() && noIPv6) LOG(("MTP Error: DC %1 options for IPv6 over HTTP not found right after config load!").arg(dc));
|
|
return restart();
|
|
}
|
|
if (noIPv4) DEBUG_LOG(("MTP Info: DC %1 options for IPv4 over HTTP not found, waiting for config").arg(dc));
|
|
if (Global::TryIPv6() && noIPv6) DEBUG_LOG(("MTP Info: DC %1 options for IPv6 over HTTP not found, waiting for config").arg(dc));
|
|
connect(configLoader(), SIGNAL(loaded()), this, SLOT(onConfigLoaded()));
|
|
configLoader()->load();
|
|
return;
|
|
}
|
|
|
|
if (afterConfig && (_conn4 || _conn6)) return;
|
|
|
|
createConn(!noIPv4, !noIPv6);
|
|
retryTimer.stop();
|
|
_waitForConnectedTimer.stop();
|
|
|
|
setState(ConnectingState);
|
|
_pingId = _pingMsgId = _pingIdToSend = _pingSendAt = 0;
|
|
_pingSender.stop();
|
|
|
|
if (!noIPv4) DEBUG_LOG(("MTP Info: creating IPv4 connection to %1:%2 (tcp) and %3:%4 (http)...").arg(ip[IPv4address][TcpProtocol].c_str()).arg(port[IPv4address][TcpProtocol]).arg(ip[IPv4address][HttpProtocol].c_str()).arg(port[IPv4address][HttpProtocol]));
|
|
if (!noIPv6) DEBUG_LOG(("MTP Info: creating IPv6 connection to [%1]:%2 (tcp) and [%3]:%4 (http)...").arg(ip[IPv6address][TcpProtocol].c_str()).arg(port[IPv6address][TcpProtocol]).arg(ip[IPv4address][HttpProtocol].c_str()).arg(port[IPv4address][HttpProtocol]));
|
|
|
|
_waitForConnectedTimer.start(_waitForConnected);
|
|
if (auto conn = _conn4) {
|
|
connect(conn, SIGNAL(connected()), this, SLOT(onConnected4()));
|
|
connect(conn, SIGNAL(disconnected()), this, SLOT(onDisconnected4()));
|
|
conn->connectTcp(ip[IPv4address][TcpProtocol].c_str(), port[IPv4address][TcpProtocol], flags[IPv4address][TcpProtocol]);
|
|
conn->connectHttp(ip[IPv4address][HttpProtocol].c_str(), port[IPv4address][HttpProtocol], flags[IPv4address][HttpProtocol]);
|
|
}
|
|
if (auto conn = _conn6) {
|
|
connect(conn, SIGNAL(connected()), this, SLOT(onConnected6()));
|
|
connect(conn, SIGNAL(disconnected()), this, SLOT(onDisconnected6()));
|
|
conn->connectTcp(ip[IPv6address][TcpProtocol].c_str(), port[IPv6address][TcpProtocol], flags[IPv6address][TcpProtocol]);
|
|
conn->connectHttp(ip[IPv6address][HttpProtocol].c_str(), port[IPv6address][HttpProtocol], flags[IPv6address][HttpProtocol]);
|
|
}
|
|
}
|
|
|
|
void ConnectionPrivate::restart(bool mayBeBadKey) {
|
|
QReadLocker lockFinished(&sessionDataMutex);
|
|
if (!sessionData) return;
|
|
|
|
DEBUG_LOG(("MTP Info: restarting Connection, maybe bad key = %1").arg(Logs::b(mayBeBadKey)));
|
|
|
|
_waitForReceivedTimer.stop();
|
|
_waitForConnectedTimer.stop();
|
|
|
|
AuthKeyPtr key(sessionData->getKey());
|
|
if (key) {
|
|
if (!sessionData->isCheckedKey()) {
|
|
if (mayBeBadKey) {
|
|
clearMessages();
|
|
keyId = AuthKey::RecreateKeyId;
|
|
// retryTimeout = 1; // no ddos please
|
|
LOG(("MTP Info: key may be bad and was not checked - but won't be destroyed, no log outs because of bad server right now..."));
|
|
}
|
|
} else {
|
|
sessionData->setCheckedKey(false);
|
|
}
|
|
}
|
|
|
|
lockFinished.unlock();
|
|
doDisconnect();
|
|
|
|
lockFinished.relock();
|
|
if (sessionData && _needSessionReset) {
|
|
resetSession();
|
|
}
|
|
restarted = true;
|
|
if (retryTimer.isActive()) return;
|
|
|
|
DEBUG_LOG(("MTP Info: restart timeout: %1ms").arg(retryTimeout));
|
|
setState(-retryTimeout);
|
|
}
|
|
|
|
void ConnectionPrivate::onSentSome(uint64 size) {
|
|
if (!_waitForReceivedTimer.isActive()) {
|
|
uint64 remain = _waitForReceived;
|
|
if (!oldConnection) {
|
|
uint64 remainBySize = size * _waitForReceived / 8192; // 8kb / sec, so 512 kb give 64 sec
|
|
remain = snap(remainBySize, remain, uint64(MTPMaxReceiveDelay));
|
|
if (remain != _waitForReceived) {
|
|
DEBUG_LOG(("Checking connect for request with size %1 bytes, delay will be %2").arg(size).arg(remain));
|
|
}
|
|
}
|
|
if (isUplDcId(dc)) {
|
|
remain *= MTPUploadSessionsCount;
|
|
} else if (isDldDcId(dc)) {
|
|
remain *= MTPDownloadSessionsCount;
|
|
}
|
|
_waitForReceivedTimer.start(remain);
|
|
}
|
|
if (!firstSentAt) firstSentAt = getms(true);
|
|
}
|
|
|
|
void ConnectionPrivate::onReceivedSome() {
|
|
if (oldConnection) {
|
|
oldConnection = false;
|
|
DEBUG_LOG(("This connection marked as not old!"));
|
|
}
|
|
oldConnectionTimer.start(MTPConnectionOldTimeout);
|
|
_waitForReceivedTimer.stop();
|
|
if (firstSentAt > 0) {
|
|
int32 ms = getms(true) - firstSentAt;
|
|
DEBUG_LOG(("MTP Info: response in %1ms, _waitForReceived: %2ms").arg(ms).arg(_waitForReceived));
|
|
|
|
if (ms > 0 && ms * 2 < int32(_waitForReceived)) _waitForReceived = qMax(ms * 2, int32(MTPMinReceiveDelay));
|
|
firstSentAt = -1;
|
|
}
|
|
}
|
|
|
|
void ConnectionPrivate::onOldConnection() {
|
|
oldConnection = true;
|
|
_waitForReceived = MTPMinReceiveDelay;
|
|
DEBUG_LOG(("This connection marked as old! _waitForReceived now %1ms").arg(_waitForReceived));
|
|
}
|
|
|
|
void ConnectionPrivate::onPingSender() {
|
|
if (_pingId) {
|
|
if (_pingSendAt + (MTPPingSendAfter - MTPPingSendAfterAuto - 1) * 1000ULL < getms(true)) {
|
|
LOG(("Could not send ping for MTPPingSendAfter seconds, restarting..."));
|
|
return restart();
|
|
} else {
|
|
_pingSender.start(_pingSendAt + (MTPPingSendAfter - MTPPingSendAfterAuto) * 1000ULL - getms(true));
|
|
}
|
|
} else {
|
|
emit needToSendAsync();
|
|
}
|
|
}
|
|
|
|
void ConnectionPrivate::onPingSendForce() {
|
|
if (!_pingId) {
|
|
_pingSendAt = 0;
|
|
DEBUG_LOG(("Will send ping!"));
|
|
tryToSend();
|
|
}
|
|
}
|
|
|
|
void ConnectionPrivate::onWaitReceivedFailed() {
|
|
if (Global::ConnectionType() != dbictAuto && Global::ConnectionType() != dbictTcpProxy) {
|
|
return;
|
|
}
|
|
|
|
DEBUG_LOG(("MTP Info: bad connection, _waitForReceived: %1ms").arg(_waitForReceived));
|
|
if (_waitForReceived < MTPMaxReceiveDelay) {
|
|
_waitForReceived *= 2;
|
|
}
|
|
doDisconnect();
|
|
restarted = true;
|
|
if (retryTimer.isActive()) return;
|
|
|
|
DEBUG_LOG(("MTP Info: immediate restart!"));
|
|
QTimer::singleShot(0, this, SLOT(socketStart()));
|
|
}
|
|
|
|
void ConnectionPrivate::onWaitConnectedFailed() {
|
|
DEBUG_LOG(("MTP Info: can't connect in %1ms").arg(_waitForConnected));
|
|
if (_waitForConnected < MTPMaxConnectDelay) _waitForConnected *= 2;
|
|
|
|
doDisconnect();
|
|
restarted = true;
|
|
|
|
DEBUG_LOG(("MTP Info: immediate restart!"));
|
|
QTimer::singleShot(0, this, SLOT(socketStart()));
|
|
}
|
|
|
|
void ConnectionPrivate::onWaitIPv4Failed() {
|
|
_conn = _conn6;
|
|
destroyConn(&_conn4);
|
|
|
|
if (_conn) {
|
|
DEBUG_LOG(("MTP Info: can't connect through IPv4, using IPv6 connection."));
|
|
|
|
updateAuthKey();
|
|
} else {
|
|
restart();
|
|
}
|
|
}
|
|
|
|
void ConnectionPrivate::doDisconnect() {
|
|
destroyConn();
|
|
|
|
{
|
|
QReadLocker lockFinished(&sessionDataMutex);
|
|
if (sessionData) {
|
|
unlockKey();
|
|
}
|
|
}
|
|
|
|
clearAuthKeyData();
|
|
|
|
setState(DisconnectedState);
|
|
restarted = false;
|
|
}
|
|
|
|
void ConnectionPrivate::doFinish() {
|
|
doDisconnect();
|
|
_finished = true;
|
|
emit finished(_owner);
|
|
deleteLater();
|
|
}
|
|
|
|
void ConnectionPrivate::handleReceived() {
|
|
QReadLocker lockFinished(&sessionDataMutex);
|
|
if (!sessionData) return;
|
|
|
|
onReceivedSome();
|
|
|
|
ReadLockerAttempt lock(sessionData->keyMutex());
|
|
if (!lock) {
|
|
DEBUG_LOG(("MTP Error: auth_key for dc %1 busy, cant lock").arg(dc));
|
|
clearMessages();
|
|
keyId = 0;
|
|
|
|
lockFinished.unlock();
|
|
return restart();
|
|
}
|
|
|
|
AuthKeyPtr key(sessionData->getKey());
|
|
if (!key || key->keyId() != keyId) {
|
|
DEBUG_LOG(("MTP Error: auth_key id for dc %1 changed").arg(dc));
|
|
|
|
lockFinished.unlock();
|
|
return restart();
|
|
}
|
|
|
|
while (_conn->received().size()) {
|
|
const mtpBuffer &encryptedBuf(_conn->received().front());
|
|
uint32 len = encryptedBuf.size();
|
|
const mtpPrime *encrypted(encryptedBuf.data());
|
|
if (len < 18) { // 2 auth_key_id, 4 msg_key, 2 salt, 2 session, 2 msg_id, 1 seq_no, 1 length, (1 data + 3 padding) min
|
|
LOG(("TCP Error: bad message received, len %1").arg(len * sizeof(mtpPrime)));
|
|
TCP_LOG(("TCP Error: bad message %1").arg(Logs::mb(encrypted, len * sizeof(mtpPrime)).str()));
|
|
|
|
lockFinished.unlock();
|
|
return restart();
|
|
}
|
|
if (keyId != *(uint64*)encrypted) {
|
|
LOG(("TCP Error: bad auth_key_id %1 instead of %2 received").arg(keyId).arg(*(uint64*)encrypted));
|
|
TCP_LOG(("TCP Error: bad message %1").arg(Logs::mb(encrypted, len * sizeof(mtpPrime)).str()));
|
|
|
|
lockFinished.unlock();
|
|
return restart();
|
|
}
|
|
|
|
QByteArray dataBuffer((len - 6) * sizeof(mtpPrime), Qt::Uninitialized);
|
|
mtpPrime *data((mtpPrime*)dataBuffer.data()), *msg = data + 8;
|
|
const mtpPrime *from(msg), *end;
|
|
MTPint128 msgKey(*(MTPint128*)(encrypted + 2));
|
|
|
|
aesIgeDecrypt(encrypted + 6, data, dataBuffer.size(), key, msgKey);
|
|
|
|
uint64 serverSalt = *(uint64*)&data[0], session = *(uint64*)&data[2], msgId = *(uint64*)&data[4];
|
|
uint32 seqNo = *(uint32*)&data[6], msgLen = *(uint32*)&data[7];
|
|
bool needAck = (seqNo & 0x01);
|
|
|
|
if (uint32(dataBuffer.size()) < msgLen + 8 * sizeof(mtpPrime) || (msgLen & 0x03)) {
|
|
LOG(("TCP Error: bad msg_len received %1, data size: %2").arg(msgLen).arg(dataBuffer.size()));
|
|
TCP_LOG(("TCP Error: bad message %1").arg(Logs::mb(encrypted, len * sizeof(mtpPrime)).str()));
|
|
_conn->received().pop_front();
|
|
|
|
lockFinished.unlock();
|
|
return restart();
|
|
}
|
|
uchar sha1Buffer[20];
|
|
if (memcmp(&msgKey, hashSha1(data, msgLen + 8 * sizeof(mtpPrime), sha1Buffer) + 1, sizeof(msgKey))) {
|
|
LOG(("TCP Error: bad SHA1 hash after aesDecrypt in message"));
|
|
TCP_LOG(("TCP Error: bad message %1").arg(Logs::mb(encrypted, len * sizeof(mtpPrime)).str()));
|
|
_conn->received().pop_front();
|
|
|
|
lockFinished.unlock();
|
|
return restart();
|
|
}
|
|
TCP_LOG(("TCP Info: decrypted message %1,%2,%3 is %4 len").arg(msgId).arg(seqNo).arg(Logs::b(needAck)).arg(msgLen + 8 * sizeof(mtpPrime)));
|
|
|
|
uint64 serverSession = sessionData->getSession();
|
|
if (session != serverSession) {
|
|
LOG(("MTP Error: bad server session received"));
|
|
TCP_LOG(("MTP Error: bad server session %1 instead of %2 in message received").arg(session).arg(serverSession));
|
|
_conn->received().pop_front();
|
|
|
|
lockFinished.unlock();
|
|
return restart();
|
|
}
|
|
|
|
_conn->received().pop_front();
|
|
|
|
int32 serverTime((int32)(msgId >> 32)), clientTime(unixtime());
|
|
bool isReply = ((msgId & 0x03) == 1);
|
|
if (!isReply && ((msgId & 0x03) != 3)) {
|
|
LOG(("MTP Error: bad msg_id %1 in message received").arg(msgId));
|
|
|
|
lockFinished.unlock();
|
|
return restart();
|
|
}
|
|
|
|
bool badTime = false;
|
|
uint64 mySalt = sessionData->getSalt();
|
|
if (serverTime > clientTime + 60 || serverTime + 300 < clientTime) {
|
|
DEBUG_LOG(("MTP Info: bad server time from msg_id: %1, my time: %2").arg(serverTime).arg(clientTime));
|
|
badTime = true;
|
|
}
|
|
|
|
bool wasConnected = (getState() == ConnectedState);
|
|
if (serverSalt != mySalt) {
|
|
if (!badTime) {
|
|
DEBUG_LOG(("MTP Info: other salt received... received: %1, my salt: %2, updating...").arg(serverSalt).arg(mySalt));
|
|
sessionData->setSalt(serverSalt);
|
|
if (setState(ConnectedState, ConnectingState)) { // only connected
|
|
if (restarted) {
|
|
emit resendAllAsync();
|
|
restarted = false;
|
|
}
|
|
}
|
|
} else {
|
|
DEBUG_LOG(("MTP Info: other salt received... received: %1, my salt: %2").arg(serverSalt).arg(mySalt));
|
|
}
|
|
} else {
|
|
serverSalt = 0; // dont pass to handle method, so not to lock in setSalt()
|
|
}
|
|
|
|
if (needAck) ackRequestData.push_back(MTP_long(msgId));
|
|
|
|
int32 res = 1; // if no need to handle, then succeed
|
|
end = data + 8 + (msgLen >> 2);
|
|
const mtpPrime *sfrom(data + 4);
|
|
MTP_LOG(dc, ("Recv: ") + mtpTextSerialize(sfrom, end));
|
|
|
|
bool needToHandle = false;
|
|
{
|
|
QWriteLocker lock(sessionData->receivedIdsMutex());
|
|
mtpMsgIdsMap &receivedIds(sessionData->receivedIdsSet());
|
|
needToHandle = receivedIds.insert(msgId, needAck);
|
|
}
|
|
if (needToHandle) {
|
|
res = handleOneReceived(from, end, msgId, serverTime, serverSalt, badTime);
|
|
}
|
|
{
|
|
QWriteLocker lock(sessionData->receivedIdsMutex());
|
|
mtpMsgIdsMap &receivedIds(sessionData->receivedIdsSet());
|
|
uint32 receivedIdsSize = receivedIds.size();
|
|
while (receivedIdsSize-- > MTPIdsBufferSize) {
|
|
receivedIds.erase(receivedIds.begin());
|
|
}
|
|
}
|
|
|
|
// send acks
|
|
uint32 toAckSize = ackRequestData.size();
|
|
if (toAckSize) {
|
|
DEBUG_LOG(("MTP Info: will send %1 acks, ids: %2").arg(toAckSize).arg(Logs::vector(ackRequestData)));
|
|
emit sendAnythingAsync(MTPAckSendWaiting);
|
|
}
|
|
|
|
bool emitSignal = false;
|
|
{
|
|
QReadLocker locker(sessionData->haveReceivedMutex());
|
|
emitSignal = !sessionData->haveReceivedMap().isEmpty();
|
|
if (emitSignal) {
|
|
DEBUG_LOG(("MTP Info: emitting needToReceive() - need to parse in another thread, haveReceivedMap.size() = %1").arg(sessionData->haveReceivedMap().size()));
|
|
}
|
|
}
|
|
|
|
if (emitSignal) {
|
|
emit needToReceive();
|
|
}
|
|
|
|
if (res < 0) {
|
|
_needSessionReset = (res < -1);
|
|
|
|
lockFinished.unlock();
|
|
return restart();
|
|
}
|
|
retryTimeout = 1; // reset restart() timer
|
|
|
|
if (!sessionData->isCheckedKey()) {
|
|
DEBUG_LOG(("MTP Info: marked auth key as checked"));
|
|
sessionData->setCheckedKey(true);
|
|
}
|
|
|
|
if (!wasConnected) {
|
|
if (getState() == ConnectedState) {
|
|
emit needToSendAsync();
|
|
}
|
|
}
|
|
}
|
|
if (_conn->needHttpWait()) {
|
|
emit sendHttpWaitAsync();
|
|
}
|
|
}
|
|
|
|
int32 ConnectionPrivate::handleOneReceived(const mtpPrime *from, const mtpPrime *end, uint64 msgId, int32 serverTime, uint64 serverSalt, bool badTime) {
|
|
mtpTypeId cons = *from;
|
|
try {
|
|
|
|
switch (cons) {
|
|
|
|
case mtpc_gzip_packed: {
|
|
DEBUG_LOG(("Message Info: gzip container"));
|
|
mtpBuffer response = ungzip(++from, end);
|
|
if (!response.size()) {
|
|
return -1;
|
|
}
|
|
return handleOneReceived(response.data(), response.data() + response.size(), msgId, serverTime, serverSalt, badTime);
|
|
}
|
|
|
|
case mtpc_msg_container: {
|
|
if (++from >= end) throw mtpErrorInsufficient();
|
|
|
|
const mtpPrime *otherEnd;
|
|
uint32 msgsCount = (uint32)*(from++);
|
|
DEBUG_LOG(("Message Info: container received, count: %1").arg(msgsCount));
|
|
for (uint32 i = 0; i < msgsCount; ++i) {
|
|
if (from + 4 >= end) throw mtpErrorInsufficient();
|
|
otherEnd = from + 4;
|
|
|
|
MTPlong inMsgId(from, otherEnd);
|
|
bool isReply = ((inMsgId.v & 0x03) == 1);
|
|
if (!isReply && ((inMsgId.v & 0x03) != 3)) {
|
|
LOG(("Message Error: bad msg_id %1 in contained message received").arg(inMsgId.v));
|
|
return -1;
|
|
}
|
|
|
|
MTPint inSeqNo(from, otherEnd);
|
|
MTPint bytes(from, otherEnd);
|
|
if ((bytes.v & 0x03) || bytes.v < 4) {
|
|
LOG(("Message Error: bad length %1 of contained message received").arg(bytes.v));
|
|
return -1;
|
|
}
|
|
|
|
bool needAck = (inSeqNo.v & 0x01);
|
|
if (needAck) ackRequestData.push_back(inMsgId);
|
|
|
|
DEBUG_LOG(("Message Info: message from container, msg_id: %1, needAck: %2").arg(inMsgId.v).arg(Logs::b(needAck)));
|
|
|
|
otherEnd = from + (bytes.v >> 2);
|
|
if (otherEnd > end) throw mtpErrorInsufficient();
|
|
|
|
bool needToHandle = false;
|
|
{
|
|
QWriteLocker lock(sessionData->receivedIdsMutex());
|
|
mtpMsgIdsMap &receivedIds(sessionData->receivedIdsSet());
|
|
needToHandle = receivedIds.insert(inMsgId.v, needAck);
|
|
}
|
|
int32 res = 1; // if no need to handle, then succeed
|
|
if (needToHandle) {
|
|
res = handleOneReceived(from, otherEnd, inMsgId.v, serverTime, serverSalt, badTime);
|
|
badTime = false;
|
|
}
|
|
if (res <= 0) {
|
|
return res;
|
|
}
|
|
|
|
from = otherEnd;
|
|
}
|
|
} return 1;
|
|
|
|
case mtpc_msgs_ack: {
|
|
MTPMsgsAck msg(from, end);
|
|
const auto &ids(msg.c_msgs_ack().vmsg_ids.c_vector().v);
|
|
uint32 idsCount = ids.size();
|
|
|
|
DEBUG_LOG(("Message Info: acks received, ids: %1").arg(Logs::vector(ids)));
|
|
if (!idsCount) return (badTime ? 0 : 1);
|
|
|
|
if (badTime) {
|
|
if (requestsFixTimeSalt(ids, serverTime, serverSalt)) {
|
|
badTime = false;
|
|
} else {
|
|
return 0;
|
|
}
|
|
}
|
|
requestsAcked(ids);
|
|
} return 1;
|
|
|
|
case mtpc_bad_msg_notification: {
|
|
MTPBadMsgNotification msg(from, end);
|
|
const auto &data(msg.c_bad_msg_notification());
|
|
LOG(("Message Info: bad message notification received (error_code %3) for msg_id = %1, seq_no = %2").arg(data.vbad_msg_id.v).arg(data.vbad_msg_seqno.v).arg(data.verror_code.v));
|
|
|
|
mtpMsgId resendId = data.vbad_msg_id.v;
|
|
if (resendId == _pingMsgId) {
|
|
_pingId = 0;
|
|
}
|
|
int32 errorCode = data.verror_code.v;
|
|
if (errorCode == 16 || errorCode == 17 || errorCode == 32 || errorCode == 33 || errorCode == 64) { // can handle
|
|
bool needResend = (errorCode == 16 || errorCode == 17); // bad msg_id
|
|
if (errorCode == 64) { // bad container!
|
|
needResend = true;
|
|
if (cDebug()) {
|
|
mtpRequest request;
|
|
{
|
|
QWriteLocker locker(sessionData->haveSentMutex());
|
|
mtpRequestMap &haveSent(sessionData->haveSentMap());
|
|
|
|
mtpRequestMap::const_iterator i = haveSent.constFind(resendId);
|
|
if (i == haveSent.cend()) {
|
|
LOG(("Message Error: Container not found!"));
|
|
} else {
|
|
request = i.value();
|
|
}
|
|
}
|
|
if (request) {
|
|
if (mtpRequestData::isSentContainer(request)) {
|
|
QStringList lst;
|
|
const mtpMsgId *ids = (const mtpMsgId *)(request->constData() + 8);
|
|
for (uint32 i = 0, l = (request->size() - 8) >> 1; i < l; ++i) {
|
|
lst.push_back(QString::number(ids[i]));
|
|
}
|
|
LOG(("Message Info: bad container received! messages: %1").arg(lst.join(',')));
|
|
} else {
|
|
LOG(("Message Error: bad container received, but request is not a container!"));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
if (!wasSent(resendId)) {
|
|
DEBUG_LOG(("Message Error: such message was not sent recently %1").arg(resendId));
|
|
return (badTime ? 0 : 1);
|
|
}
|
|
|
|
if (needResend) { // bad msg_id
|
|
if (serverSalt) sessionData->setSalt(serverSalt);
|
|
unixtimeSet(serverTime, true);
|
|
|
|
DEBUG_LOG(("Message Info: unixtime updated, now %1, resending in container...").arg(serverTime));
|
|
|
|
resend(resendId, 0, true);
|
|
} else { // must create new session, because msg_id and msg_seqno are inconsistent
|
|
if (badTime) {
|
|
if (serverSalt) sessionData->setSalt(serverSalt);
|
|
unixtimeSet(serverTime, true);
|
|
badTime = false;
|
|
}
|
|
LOG(("Message Info: bad message notification received, msgId %1, error_code %2").arg(data.vbad_msg_id.v).arg(errorCode));
|
|
return -2;
|
|
}
|
|
} else { // fatal (except 48, but it must not get here)
|
|
mtpMsgId resendId = data.vbad_msg_id.v;
|
|
mtpRequestId requestId = wasSent(resendId);
|
|
if (requestId) {
|
|
LOG(("Message Error: bad message notification received, msgId %1, error_code %2, fatal: clearing callbacks").arg(data.vbad_msg_id.v).arg(errorCode));
|
|
clearCallbacksDelayed(RPCCallbackClears(1, RPCCallbackClear(requestId, -errorCode)));
|
|
} else {
|
|
DEBUG_LOG(("Message Error: such message was not sent recently %1").arg(resendId));
|
|
}
|
|
return (badTime ? 0 : 1);
|
|
}
|
|
} return 1;
|
|
|
|
case mtpc_bad_server_salt: {
|
|
MTPBadMsgNotification msg(from, end);
|
|
const auto &data(msg.c_bad_server_salt());
|
|
DEBUG_LOG(("Message Info: bad server salt received (error_code %4) for msg_id = %1, seq_no = %2, new salt: %3").arg(data.vbad_msg_id.v).arg(data.vbad_msg_seqno.v).arg(data.vnew_server_salt.v).arg(data.verror_code.v));
|
|
|
|
mtpMsgId resendId = data.vbad_msg_id.v;
|
|
if (resendId == _pingMsgId) {
|
|
_pingId = 0;
|
|
} else if (!wasSent(resendId)) {
|
|
DEBUG_LOG(("Message Error: such message was not sent recently %1").arg(resendId));
|
|
return (badTime ? 0 : 1);
|
|
}
|
|
|
|
uint64 serverSalt = data.vnew_server_salt.v;
|
|
sessionData->setSalt(serverSalt);
|
|
unixtimeSet(serverTime);
|
|
|
|
if (setState(ConnectedState, ConnectingState)) { // maybe only connected
|
|
if (restarted) {
|
|
emit resendAllAsync();
|
|
restarted = false;
|
|
}
|
|
}
|
|
|
|
badTime = false;
|
|
|
|
DEBUG_LOG(("Message Info: unixtime updated, now %1, server_salt updated, now %2, resending...").arg(serverTime).arg(serverSalt));
|
|
resend(resendId);
|
|
} return 1;
|
|
|
|
case mtpc_msgs_state_req: {
|
|
if (badTime) {
|
|
DEBUG_LOG(("Message Info: skipping with bad time..."));
|
|
return 0;
|
|
}
|
|
MTPMsgsStateReq msg(from, end);
|
|
const auto &ids(msg.c_msgs_state_req().vmsg_ids.c_vector().v);
|
|
uint32 idsCount = ids.size();
|
|
DEBUG_LOG(("Message Info: msgs_state_req received, ids: %1").arg(Logs::vector(ids)));
|
|
if (!idsCount) return 1;
|
|
|
|
QByteArray info(idsCount, Qt::Uninitialized);
|
|
{
|
|
QReadLocker lock(sessionData->receivedIdsMutex());
|
|
const mtpMsgIdsMap &receivedIds(sessionData->receivedIdsSet());
|
|
mtpMsgIdsMap::const_iterator receivedIdsEnd(receivedIds.cend());
|
|
uint64 minRecv = receivedIds.min(), maxRecv = receivedIds.max();
|
|
|
|
QReadLocker locker(sessionData->wereAckedMutex());
|
|
const mtpRequestIdsMap &wereAcked(sessionData->wereAckedMap());
|
|
mtpRequestIdsMap::const_iterator wereAckedEnd(wereAcked.cend());
|
|
|
|
for (uint32 i = 0, l = idsCount; i < l; ++i) {
|
|
char state = 0;
|
|
uint64 reqMsgId = ids[i].v;
|
|
if (reqMsgId < minRecv) {
|
|
state |= 0x01;
|
|
} else if (reqMsgId > maxRecv) {
|
|
state |= 0x03;
|
|
} else {
|
|
mtpMsgIdsMap::const_iterator recv = receivedIds.constFind(reqMsgId);
|
|
if (recv == receivedIdsEnd) {
|
|
state |= 0x02;
|
|
} else {
|
|
state |= 0x04;
|
|
if (wereAcked.constFind(reqMsgId) != wereAckedEnd) {
|
|
state |= 0x80; // we know, that server knows, that we received request
|
|
}
|
|
if (recv.value()) { // need ack, so we sent ack
|
|
state |= 0x08;
|
|
} else {
|
|
state |= 0x10;
|
|
}
|
|
}
|
|
}
|
|
info[i] = state;
|
|
}
|
|
}
|
|
emit sendMsgsStateInfoAsync(msgId, info);
|
|
} return 1;
|
|
|
|
case mtpc_msgs_state_info: {
|
|
MTPMsgsStateInfo msg(from, end);
|
|
const auto &data(msg.c_msgs_state_info());
|
|
|
|
uint64 reqMsgId = data.vreq_msg_id.v;
|
|
const auto &states(data.vinfo.c_string().v);
|
|
|
|
DEBUG_LOG(("Message Info: msg state received, msgId %1, reqMsgId: %2, HEX states %3").arg(msgId).arg(reqMsgId).arg(Logs::mb(states.data(), states.length()).str()));
|
|
mtpRequest requestBuffer;
|
|
{ // find this request in session-shared sent requests map
|
|
QReadLocker locker(sessionData->haveSentMutex());
|
|
const mtpRequestMap &haveSent(sessionData->haveSentMap());
|
|
mtpRequestMap::const_iterator replyTo = haveSent.constFind(reqMsgId);
|
|
if (replyTo == haveSent.cend()) { // do not look in toResend, because we do not resend msgs_state_req requests
|
|
DEBUG_LOG(("Message Error: such message was not sent recently %1").arg(reqMsgId));
|
|
return (badTime ? 0 : 1);
|
|
}
|
|
if (badTime) {
|
|
if (serverSalt) sessionData->setSalt(serverSalt); // requestsFixTimeSalt with no lookup
|
|
unixtimeSet(serverTime, true);
|
|
|
|
DEBUG_LOG(("Message Info: unixtime updated from mtpc_msgs_state_info, now %1").arg(serverTime));
|
|
|
|
badTime = false;
|
|
}
|
|
requestBuffer = replyTo.value();
|
|
}
|
|
QVector<MTPlong> toAckReq(1, MTP_long(reqMsgId)), toAck;
|
|
requestsAcked(toAck, true);
|
|
|
|
if (requestBuffer->size() < 9) {
|
|
LOG(("Message Error: bad request %1 found in requestMap, size: %2").arg(reqMsgId).arg(requestBuffer->size()));
|
|
return -1;
|
|
}
|
|
try {
|
|
const mtpPrime *rFrom = requestBuffer->constData() + 8, *rEnd = requestBuffer->constData() + requestBuffer->size();
|
|
if (mtpTypeId(*rFrom) == mtpc_msgs_state_req) {
|
|
MTPMsgsStateReq request(rFrom, rEnd);
|
|
handleMsgsStates(request.c_msgs_state_req().vmsg_ids.c_vector().v, states, toAck);
|
|
} else {
|
|
MTPMsgResendReq request(rFrom, rEnd);
|
|
handleMsgsStates(request.c_msg_resend_req().vmsg_ids.c_vector().v, states, toAck);
|
|
}
|
|
} catch(Exception &) {
|
|
LOG(("Message Error: could not parse sent msgs_state_req"));
|
|
throw;
|
|
}
|
|
|
|
requestsAcked(toAck);
|
|
} return 1;
|
|
|
|
case mtpc_msgs_all_info: {
|
|
if (badTime) {
|
|
DEBUG_LOG(("Message Info: skipping with bad time..."));
|
|
return 0;
|
|
}
|
|
|
|
MTPMsgsAllInfo msg(from, end);
|
|
const auto &data(msg.c_msgs_all_info());
|
|
const auto &ids(data.vmsg_ids.c_vector().v);
|
|
const auto &states(data.vinfo.c_string().v);
|
|
|
|
QVector<MTPlong> toAck;
|
|
|
|
DEBUG_LOG(("Message Info: msgs all info received, msgId %1, reqMsgIds: %2, states %3").arg(msgId).arg(Logs::vector(ids)).arg(Logs::mb(states.data(), states.length()).str()));
|
|
handleMsgsStates(ids, states, toAck);
|
|
|
|
requestsAcked(toAck);
|
|
} return 1;
|
|
|
|
case mtpc_msg_detailed_info: {
|
|
MTPMsgDetailedInfo msg(from, end);
|
|
const auto &data(msg.c_msg_detailed_info());
|
|
|
|
DEBUG_LOG(("Message Info: msg detailed info, sent msgId %1, answerId %2, status %3, bytes %4").arg(data.vmsg_id.v).arg(data.vanswer_msg_id.v).arg(data.vstatus.v).arg(data.vbytes.v));
|
|
|
|
QVector<MTPlong> ids(1, data.vmsg_id);
|
|
if (badTime) {
|
|
if (requestsFixTimeSalt(ids, serverTime, serverSalt)) {
|
|
badTime = false;
|
|
} else {
|
|
DEBUG_LOG(("Message Info: error, such message was not sent recently %1").arg(data.vmsg_id.v));
|
|
return 0;
|
|
}
|
|
}
|
|
requestsAcked(ids);
|
|
|
|
bool received = false;
|
|
MTPlong resMsgId = data.vanswer_msg_id;
|
|
{
|
|
QReadLocker lock(sessionData->receivedIdsMutex());
|
|
const mtpMsgIdsMap &receivedIds(sessionData->receivedIdsSet());
|
|
received = (receivedIds.find(resMsgId.v) != receivedIds.cend()) && (receivedIds.min() < resMsgId.v);
|
|
}
|
|
if (received) {
|
|
ackRequestData.push_back(resMsgId);
|
|
} else {
|
|
DEBUG_LOG(("Message Info: answer message %1 was not received, requesting...").arg(resMsgId.v));
|
|
resendRequestData.push_back(resMsgId);
|
|
}
|
|
} return 1;
|
|
|
|
case mtpc_msg_new_detailed_info: {
|
|
if (badTime) {
|
|
DEBUG_LOG(("Message Info: skipping msg_new_detailed_info with bad time..."));
|
|
return 0;
|
|
}
|
|
MTPMsgDetailedInfo msg(from, end);
|
|
const auto &data(msg.c_msg_new_detailed_info());
|
|
|
|
DEBUG_LOG(("Message Info: msg new detailed info, answerId %2, status %3, bytes %4").arg(data.vanswer_msg_id.v).arg(data.vstatus.v).arg(data.vbytes.v));
|
|
|
|
bool received = false;
|
|
MTPlong resMsgId = data.vanswer_msg_id;
|
|
{
|
|
QReadLocker lock(sessionData->receivedIdsMutex());
|
|
const mtpMsgIdsMap &receivedIds(sessionData->receivedIdsSet());
|
|
received = (receivedIds.find(resMsgId.v) != receivedIds.cend()) && (receivedIds.min() < resMsgId.v);
|
|
}
|
|
if (received) {
|
|
ackRequestData.push_back(resMsgId);
|
|
} else {
|
|
DEBUG_LOG(("Message Info: answer message %1 was not received, requesting...").arg(resMsgId.v));
|
|
resendRequestData.push_back(resMsgId);
|
|
}
|
|
} return 1;
|
|
|
|
case mtpc_msg_resend_req: {
|
|
MTPMsgResendReq msg(from, end);
|
|
const auto &ids(msg.c_msg_resend_req().vmsg_ids.c_vector().v);
|
|
|
|
uint32 idsCount = ids.size();
|
|
DEBUG_LOG(("Message Info: resend of msgs requested, ids: %1").arg(Logs::vector(ids)));
|
|
if (!idsCount) return (badTime ? 0 : 1);
|
|
|
|
QVector<quint64> toResend(ids.size());
|
|
for (int32 i = 0, l = ids.size(); i < l; ++i) {
|
|
toResend[i] = ids.at(i).v;
|
|
}
|
|
resendMany(toResend, 0, false, true);
|
|
} return 1;
|
|
|
|
case mtpc_rpc_result: {
|
|
if (from + 3 > end) throw mtpErrorInsufficient();
|
|
mtpResponse response;
|
|
|
|
MTPlong reqMsgId(++from, end);
|
|
mtpTypeId typeId = from[0];
|
|
|
|
DEBUG_LOG(("RPC Info: response received for %1, queueing...").arg(reqMsgId.v));
|
|
|
|
QVector<MTPlong> ids(1, reqMsgId);
|
|
if (badTime) {
|
|
if (requestsFixTimeSalt(ids, serverTime, serverSalt)) {
|
|
badTime = false;
|
|
} else {
|
|
DEBUG_LOG(("Message Info: error, such message was not sent recently %1").arg(reqMsgId.v));
|
|
return 0;
|
|
}
|
|
}
|
|
requestsAcked(ids, true);
|
|
|
|
if (typeId == mtpc_gzip_packed) {
|
|
DEBUG_LOG(("RPC Info: gzip container"));
|
|
response = ungzip(++from, end);
|
|
if (!response.size()) {
|
|
return -1;
|
|
}
|
|
typeId = response[0];
|
|
} else {
|
|
response.resize(end - from);
|
|
memcpy(response.data(), from, (end - from) * sizeof(mtpPrime));
|
|
}
|
|
if (!sessionData->layerWasInited()) {
|
|
sessionData->setLayerWasInited(true);
|
|
sessionData->owner()->notifyLayerInited(true);
|
|
}
|
|
|
|
mtpRequestId requestId = wasSent(reqMsgId.v);
|
|
if (requestId && requestId != mtpRequestId(0xFFFFFFFF)) {
|
|
QWriteLocker locker(sessionData->haveReceivedMutex());
|
|
sessionData->haveReceivedMap().insert(requestId, response); // save rpc_result for processing in main mtp thread
|
|
} else {
|
|
DEBUG_LOG(("RPC Info: requestId not found for msgId %1").arg(reqMsgId.v));
|
|
}
|
|
} return 1;
|
|
|
|
case mtpc_new_session_created: {
|
|
const mtpPrime *start = from;
|
|
MTPNewSession msg(from, end);
|
|
const auto &data(msg.c_new_session_created());
|
|
|
|
if (badTime) {
|
|
if (requestsFixTimeSalt(QVector<MTPlong>(1, data.vfirst_msg_id), serverTime, serverSalt)) {
|
|
badTime = false;
|
|
} else {
|
|
DEBUG_LOG(("Message Info: error, such message was not sent recently %1").arg(data.vfirst_msg_id.v));
|
|
return 0;
|
|
}
|
|
}
|
|
|
|
DEBUG_LOG(("Message Info: new server session created, unique_id %1, first_msg_id %2, server_salt %3").arg(data.vunique_id.v).arg(data.vfirst_msg_id.v).arg(data.vserver_salt.v));
|
|
sessionData->setSalt(data.vserver_salt.v);
|
|
|
|
mtpMsgId firstMsgId = data.vfirst_msg_id.v;
|
|
QVector<quint64> toResend;
|
|
{
|
|
QReadLocker locker(sessionData->haveSentMutex());
|
|
const mtpRequestMap &haveSent(sessionData->haveSentMap());
|
|
toResend.reserve(haveSent.size());
|
|
for (mtpRequestMap::const_iterator i = haveSent.cbegin(), e = haveSent.cend(); i != e; ++i) {
|
|
if (i.key() >= firstMsgId) break;
|
|
if (i.value()->requestId) toResend.push_back(i.key());
|
|
}
|
|
}
|
|
resendMany(toResend, 10, true);
|
|
|
|
mtpBuffer update(from - start);
|
|
if (from > start) memcpy(update.data(), start, (from - start) * sizeof(mtpPrime));
|
|
|
|
QWriteLocker locker(sessionData->haveReceivedMutex());
|
|
mtpResponseMap &haveReceived(sessionData->haveReceivedMap());
|
|
mtpRequestId fakeRequestId = sessionData->nextFakeRequestId();
|
|
haveReceived.insert(fakeRequestId, mtpResponse(update)); // notify main process about new session - need to get difference
|
|
} return 1;
|
|
|
|
case mtpc_ping: {
|
|
if (badTime) return 0;
|
|
|
|
MTPPing msg(from, end);
|
|
DEBUG_LOG(("Message Info: ping received, ping_id: %1, sending pong...").arg(msg.vping_id.v));
|
|
|
|
emit sendPongAsync(msgId, msg.vping_id.v);
|
|
} return 1;
|
|
|
|
case mtpc_pong: {
|
|
MTPPong msg(from, end);
|
|
const auto &data(msg.c_pong());
|
|
DEBUG_LOG(("Message Info: pong received, msg_id: %1, ping_id: %2").arg(data.vmsg_id.v).arg(data.vping_id.v));
|
|
|
|
if (!wasSent(data.vmsg_id.v)) {
|
|
DEBUG_LOG(("Message Error: such msg_id %1 ping_id %2 was not sent recently").arg(data.vmsg_id.v).arg(data.vping_id.v));
|
|
return 0;
|
|
}
|
|
if (data.vping_id.v == _pingId) {
|
|
_pingId = 0;
|
|
} else {
|
|
DEBUG_LOG(("Message Info: just pong..."));
|
|
}
|
|
|
|
QVector<MTPlong> ids(1, data.vmsg_id);
|
|
if (badTime) {
|
|
if (requestsFixTimeSalt(ids, serverTime, serverSalt)) {
|
|
badTime = false;
|
|
} else {
|
|
return 0;
|
|
}
|
|
}
|
|
requestsAcked(ids, true);
|
|
} return 1;
|
|
|
|
}
|
|
|
|
} catch (Exception &) {
|
|
return -1;
|
|
}
|
|
|
|
if (badTime) {
|
|
DEBUG_LOG(("Message Error: bad time in updates cons, must create new session"));
|
|
return -2;
|
|
}
|
|
|
|
mtpBuffer update(end - from);
|
|
if (end > from) memcpy(update.data(), from, (end - from) * sizeof(mtpPrime));
|
|
|
|
QWriteLocker locker(sessionData->haveReceivedMutex());
|
|
mtpResponseMap &haveReceived(sessionData->haveReceivedMap());
|
|
mtpRequestId fakeRequestId = sessionData->nextFakeRequestId();
|
|
haveReceived.insert(fakeRequestId, mtpResponse(update)); // notify main process about new updates
|
|
|
|
if (cons != mtpc_updatesTooLong && cons != mtpc_updateShortMessage && cons != mtpc_updateShortChatMessage && cons != mtpc_updateShortSentMessage && cons != mtpc_updateShort && cons != mtpc_updatesCombined && cons != mtpc_updates) {
|
|
LOG(("Message Error: unknown constructor %1").arg(cons)); // maybe new api?..
|
|
}
|
|
|
|
return 1;
|
|
}
|
|
|
|
mtpBuffer ConnectionPrivate::ungzip(const mtpPrime *from, const mtpPrime *end) const {
|
|
MTPstring packed(from, end); // read packed string as serialized mtp string type
|
|
uint32 packedLen = packed.c_string().v.size(), unpackedChunk = packedLen, unpackedLen = 0;
|
|
|
|
mtpBuffer result; // * 4 because of mtpPrime type
|
|
result.resize(0);
|
|
z_stream stream;
|
|
stream.zalloc = 0;
|
|
stream.zfree = 0;
|
|
stream.opaque = 0;
|
|
stream.avail_in = 0;
|
|
stream.next_in = 0;
|
|
int res = inflateInit2(&stream, 16 + MAX_WBITS);
|
|
if (res != Z_OK) {
|
|
LOG(("RPC Error: could not init zlib stream, code: %1").arg(res));
|
|
return result;
|
|
}
|
|
stream.avail_in = packedLen;
|
|
stream.next_in = (Bytef*)&packed._string().v[0];
|
|
|
|
stream.avail_out = 0;
|
|
while (!stream.avail_out) {
|
|
result.resize(result.size() + unpackedChunk);
|
|
stream.avail_out = unpackedChunk * sizeof(mtpPrime);
|
|
stream.next_out = (Bytef*)&result[result.size() - unpackedChunk];
|
|
int res = inflate(&stream, Z_NO_FLUSH);
|
|
if (res != Z_OK && res != Z_STREAM_END) {
|
|
inflateEnd(&stream);
|
|
LOG(("RPC Error: could not unpack gziped data, code: %1").arg(res));
|
|
DEBUG_LOG(("RPC Error: bad gzip: %1").arg(Logs::mb(&packed.c_string().v[0], packedLen).str()));
|
|
return mtpBuffer();
|
|
}
|
|
}
|
|
if (stream.avail_out & 0x03) {
|
|
uint32 badSize = result.size() * sizeof(mtpPrime) - stream.avail_out;
|
|
LOG(("RPC Error: bad length of unpacked data %1").arg(badSize));
|
|
DEBUG_LOG(("RPC Error: bad unpacked data %1").arg(Logs::mb(result.data(), badSize).str()));
|
|
return mtpBuffer();
|
|
}
|
|
result.resize(result.size() - (stream.avail_out >> 2));
|
|
inflateEnd(&stream);
|
|
if (!result.size()) {
|
|
LOG(("RPC Error: bad length of unpacked data 0"));
|
|
}
|
|
return result;
|
|
}
|
|
|
|
bool ConnectionPrivate::requestsFixTimeSalt(const QVector<MTPlong> &ids, int32 serverTime, uint64 serverSalt) {
|
|
uint32 idsCount = ids.size();
|
|
|
|
for (uint32 i = 0; i < idsCount; ++i) {
|
|
if (wasSent(ids[i].v)) {// found such msg_id in recent acked requests or in recent sent requests
|
|
if (serverSalt) sessionData->setSalt(serverSalt);
|
|
unixtimeSet(serverTime, true);
|
|
return true;
|
|
}
|
|
}
|
|
return false;
|
|
}
|
|
|
|
void ConnectionPrivate::requestsAcked(const QVector<MTPlong> &ids, bool byResponse) {
|
|
uint32 idsCount = ids.size();
|
|
|
|
DEBUG_LOG(("Message Info: requests acked, ids %1").arg(Logs::vector(ids)));
|
|
|
|
RPCCallbackClears clearedAcked;
|
|
QVector<MTPlong> toAckMore;
|
|
{
|
|
QWriteLocker locker1(sessionData->wereAckedMutex());
|
|
mtpRequestIdsMap &wereAcked(sessionData->wereAckedMap());
|
|
|
|
{
|
|
QWriteLocker locker2(sessionData->haveSentMutex());
|
|
mtpRequestMap &haveSent(sessionData->haveSentMap());
|
|
|
|
for (uint32 i = 0; i < idsCount; ++i) {
|
|
mtpMsgId msgId = ids[i].v;
|
|
mtpRequestMap::iterator req = haveSent.find(msgId);
|
|
if (req != haveSent.cend()) {
|
|
if (!req.value()->msDate) {
|
|
DEBUG_LOG(("Message Info: container ack received, msgId %1").arg(ids[i].v));
|
|
uint32 inContCount = ((*req)->size() - 8) / 2;
|
|
const mtpMsgId *inContId = (const mtpMsgId *)(req.value()->constData() + 8);
|
|
toAckMore.reserve(toAckMore.size() + inContCount);
|
|
for (uint32 j = 0; j < inContCount; ++j) {
|
|
toAckMore.push_back(MTP_long(*(inContId++)));
|
|
}
|
|
haveSent.erase(req);
|
|
} else {
|
|
mtpRequestId reqId = req.value()->requestId;
|
|
bool moveToAcked = byResponse;
|
|
if (!moveToAcked) { // ignore ACK, if we need a response (if we have a handler)
|
|
moveToAcked = !hasCallbacks(reqId);
|
|
}
|
|
if (moveToAcked) {
|
|
wereAcked.insert(msgId, reqId);
|
|
haveSent.erase(req);
|
|
} else {
|
|
DEBUG_LOG(("Message Info: ignoring ACK for msgId %1 because request %2 requires a response").arg(msgId).arg(reqId));
|
|
}
|
|
}
|
|
} else {
|
|
DEBUG_LOG(("Message Info: msgId %1 was not found in recent sent, while acking requests, searching in resend...").arg(msgId));
|
|
QWriteLocker locker3(sessionData->toResendMutex());
|
|
mtpRequestIdsMap &toResend(sessionData->toResendMap());
|
|
mtpRequestIdsMap::iterator reqIt = toResend.find(msgId);
|
|
if (reqIt != toResend.cend()) {
|
|
mtpRequestId reqId = reqIt.value();
|
|
bool moveToAcked = byResponse;
|
|
if (!moveToAcked) { // ignore ACK, if we need a response (if we have a handler)
|
|
moveToAcked = !hasCallbacks(reqId);
|
|
}
|
|
if (moveToAcked) {
|
|
QWriteLocker locker4(sessionData->toSendMutex());
|
|
mtpPreRequestMap &toSend(sessionData->toSendMap());
|
|
mtpPreRequestMap::iterator req = toSend.find(reqId);
|
|
if (req != toSend.cend()) {
|
|
wereAcked.insert(msgId, req.value()->requestId);
|
|
if (req.value()->requestId != reqId) {
|
|
DEBUG_LOG(("Message Error: for msgId %1 found resent request, requestId %2, contains requestId %3").arg(msgId).arg(reqId).arg(req.value()->requestId));
|
|
} else {
|
|
DEBUG_LOG(("Message Info: acked msgId %1 that was prepared to resend, requestId %2").arg(msgId).arg(reqId));
|
|
}
|
|
toSend.erase(req);
|
|
} else {
|
|
DEBUG_LOG(("Message Info: msgId %1 was found in recent resent, requestId %2 was not found in prepared to send").arg(msgId));
|
|
}
|
|
toResend.erase(reqIt);
|
|
} else {
|
|
DEBUG_LOG(("Message Info: ignoring ACK for msgId %1 because request %2 requires a response").arg(msgId).arg(reqId));
|
|
}
|
|
} else {
|
|
DEBUG_LOG(("Message Info: msgId %1 was not found in recent resent either").arg(msgId));
|
|
}
|
|
}
|
|
}
|
|
}
|
|
|
|
uint32 ackedCount = wereAcked.size();
|
|
if (ackedCount > MTPIdsBufferSize) {
|
|
DEBUG_LOG(("Message Info: removing some old acked sent msgIds %1").arg(ackedCount - MTPIdsBufferSize));
|
|
clearedAcked.reserve(ackedCount - MTPIdsBufferSize);
|
|
while (ackedCount-- > MTPIdsBufferSize) {
|
|
mtpRequestIdsMap::iterator i(wereAcked.begin());
|
|
clearedAcked.push_back(RPCCallbackClear(i.key(), RPCError::TimeoutError));
|
|
wereAcked.erase(i);
|
|
}
|
|
}
|
|
}
|
|
|
|
if (clearedAcked.size()) {
|
|
clearCallbacksDelayed(clearedAcked);
|
|
}
|
|
|
|
if (toAckMore.size()) {
|
|
requestsAcked(toAckMore);
|
|
}
|
|
}
|
|
|
|
void ConnectionPrivate::handleMsgsStates(const QVector<MTPlong> &ids, const string &states, QVector<MTPlong> &acked) {
|
|
uint32 idsCount = ids.size();
|
|
if (!idsCount) {
|
|
DEBUG_LOG(("Message Info: void ids vector in handleMsgsStates()"));
|
|
return;
|
|
}
|
|
|
|
acked.reserve(acked.size() + idsCount);
|
|
|
|
for (uint32 i = 0, count = idsCount; i < count; ++i) {
|
|
char state = states[i];
|
|
uint64 requestMsgId = ids[i].v;
|
|
{
|
|
QReadLocker locker(sessionData->haveSentMutex());
|
|
const mtpRequestMap &haveSent(sessionData->haveSentMap());
|
|
mtpRequestMap::const_iterator haveSentEnd = haveSent.cend();
|
|
if (haveSent.find(requestMsgId) == haveSentEnd) {
|
|
DEBUG_LOG(("Message Info: state was received for msgId %1, but request is not found, looking in resent requests...").arg(requestMsgId));
|
|
QWriteLocker locker2(sessionData->toResendMutex());
|
|
mtpRequestIdsMap &toResend(sessionData->toResendMap());
|
|
mtpRequestIdsMap::iterator reqIt = toResend.find(requestMsgId);
|
|
if (reqIt != toResend.cend()) {
|
|
if ((state & 0x07) != 0x04) { // was received
|
|
DEBUG_LOG(("Message Info: state was received for msgId %1, state %2, already resending in container").arg(requestMsgId).arg((int32)state));
|
|
} else {
|
|
DEBUG_LOG(("Message Info: state was received for msgId %1, state %2, ack, cancelling resend").arg(requestMsgId).arg((int32)state));
|
|
acked.push_back(MTP_long(requestMsgId)); // will remove from resend in requestsAcked
|
|
}
|
|
} else {
|
|
DEBUG_LOG(("Message Info: msgId %1 was not found in recent resent either").arg(requestMsgId));
|
|
}
|
|
continue;
|
|
}
|
|
}
|
|
if ((state & 0x07) != 0x04) { // was received
|
|
DEBUG_LOG(("Message Info: state was received for msgId %1, state %2, resending in container").arg(requestMsgId).arg((int32)state));
|
|
resend(requestMsgId, 10, true);
|
|
} else {
|
|
DEBUG_LOG(("Message Info: state was received for msgId %1, state %2, ack").arg(requestMsgId).arg((int32)state));
|
|
acked.push_back(MTP_long(requestMsgId));
|
|
}
|
|
}
|
|
}
|
|
|
|
void ConnectionPrivate::resend(quint64 msgId, quint64 msCanWait, bool forceContainer, bool sendMsgStateInfo) {
|
|
if (msgId == _pingMsgId) return;
|
|
emit resendAsync(msgId, msCanWait, forceContainer, sendMsgStateInfo);
|
|
}
|
|
|
|
void ConnectionPrivate::resendMany(QVector<quint64> msgIds, quint64 msCanWait, bool forceContainer, bool sendMsgStateInfo) {
|
|
for (int32 i = 0, l = msgIds.size(); i < l; ++i) {
|
|
if (msgIds.at(i) == _pingMsgId) {
|
|
msgIds.remove(i);
|
|
--l;
|
|
}
|
|
}
|
|
emit resendManyAsync(msgIds, msCanWait, forceContainer, sendMsgStateInfo);
|
|
}
|
|
|
|
void ConnectionPrivate::onConnected4() {
|
|
_waitForConnected = MTPMinConnectDelay;
|
|
_waitForConnectedTimer.stop();
|
|
|
|
_waitForIPv4Timer.stop();
|
|
|
|
QReadLocker lockFinished(&sessionDataMutex);
|
|
if (!sessionData) return;
|
|
|
|
disconnect(_conn4, SIGNAL(connected()), this, SLOT(onConnected4()));
|
|
if (!_conn4->isConnected()) {
|
|
LOG(("Connection Error: not connected in onConnected4(), state: %1").arg(_conn4->debugState()));
|
|
|
|
lockFinished.unlock();
|
|
return restart();
|
|
}
|
|
|
|
_conn = _conn4;
|
|
destroyConn(&_conn6);
|
|
|
|
DEBUG_LOG(("MTP Info: connection through IPv4 succeed."));
|
|
|
|
lockFinished.unlock();
|
|
updateAuthKey();
|
|
}
|
|
|
|
void ConnectionPrivate::onConnected6() {
|
|
_waitForConnected = MTPMinConnectDelay;
|
|
_waitForConnectedTimer.stop();
|
|
|
|
QReadLocker lockFinished(&sessionDataMutex);
|
|
if (!sessionData) return;
|
|
|
|
disconnect(_conn6, SIGNAL(connected()), this, SLOT(onConnected6()));
|
|
if (!_conn6->isConnected()) {
|
|
LOG(("Connection Error: not connected in onConnected(), state: %1").arg(_conn6->debugState()));
|
|
|
|
lockFinished.unlock();
|
|
return restart();
|
|
}
|
|
|
|
DEBUG_LOG(("MTP Info: connection through IPv6 succeed, waiting IPv4 for %1ms.").arg(MTPIPv4ConnectionWaitTimeout));
|
|
|
|
_waitForIPv4Timer.start(MTPIPv4ConnectionWaitTimeout);
|
|
}
|
|
|
|
void ConnectionPrivate::onDisconnected4() {
|
|
if (_conn && _conn == _conn6) return; // disconnected the unused
|
|
|
|
if (_conn || !_conn6) {
|
|
destroyConn();
|
|
restart();
|
|
} else {
|
|
destroyConn(&_conn4);
|
|
}
|
|
}
|
|
|
|
void ConnectionPrivate::onDisconnected6() {
|
|
if (_conn && _conn == _conn4) return; // disconnected the unused
|
|
|
|
if (_conn || !_conn4) {
|
|
destroyConn();
|
|
restart();
|
|
} else {
|
|
destroyConn(&_conn6);
|
|
}
|
|
}
|
|
|
|
void ConnectionPrivate::updateAuthKey() {
|
|
QReadLocker lockFinished(&sessionDataMutex);
|
|
if (!sessionData || !_conn) return;
|
|
|
|
DEBUG_LOG(("AuthKey Info: Connection updating key from Session, dc %1").arg(dc));
|
|
uint64 newKeyId = 0;
|
|
{
|
|
ReadLockerAttempt lock(sessionData->keyMutex());
|
|
if (!lock) {
|
|
DEBUG_LOG(("MTP Info: could not lock auth_key for read, waiting signal emit"));
|
|
clearMessages();
|
|
keyId = newKeyId;
|
|
return; // some other connection is getting key
|
|
}
|
|
const AuthKeyPtr &key(sessionData->getKey());
|
|
newKeyId = key ? key->keyId() : 0;
|
|
}
|
|
if (keyId != newKeyId) {
|
|
clearMessages();
|
|
keyId = newKeyId;
|
|
}
|
|
DEBUG_LOG(("AuthKey Info: Connection update key from Session, dc %1 result: %2").arg(dc).arg(Logs::mb(&keyId, sizeof(keyId)).str()));
|
|
if (keyId) {
|
|
return authKeyCreated();
|
|
}
|
|
|
|
DEBUG_LOG(("AuthKey Info: No key in updateAuthKey(), will be creating auth_key"));
|
|
lockKey();
|
|
|
|
const AuthKeyPtr &key(sessionData->getKey());
|
|
if (key) {
|
|
if (keyId != key->keyId()) clearMessages();
|
|
keyId = key->keyId();
|
|
unlockKey();
|
|
return authKeyCreated();
|
|
}
|
|
|
|
authKeyData = new ConnectionPrivate::AuthKeyCreateData();
|
|
authKeyStrings = new ConnectionPrivate::AuthKeyCreateStrings();
|
|
authKeyData->req_num = 0;
|
|
authKeyData->nonce = rand_value<MTPint128>();
|
|
|
|
MTPReq_pq req_pq;
|
|
req_pq.vnonce = authKeyData->nonce;
|
|
|
|
connect(_conn, SIGNAL(receivedData()), this, SLOT(pqAnswered()));
|
|
|
|
DEBUG_LOG(("AuthKey Info: sending Req_pq..."));
|
|
lockFinished.unlock();
|
|
sendRequestNotSecure(req_pq);
|
|
}
|
|
|
|
void ConnectionPrivate::clearMessages() {
|
|
if (keyId && keyId != AuthKey::RecreateKeyId && _conn) {
|
|
_conn->received().clear();
|
|
}
|
|
}
|
|
|
|
void ConnectionPrivate::pqAnswered() {
|
|
disconnect(_conn, SIGNAL(receivedData()), this, SLOT(pqAnswered()));
|
|
DEBUG_LOG(("AuthKey Info: receiving Req_pq answer..."));
|
|
|
|
MTPReq_pq::ResponseType res_pq;
|
|
if (!readResponseNotSecure(res_pq)) {
|
|
return restart();
|
|
}
|
|
|
|
const auto &res_pq_data(res_pq.c_resPQ());
|
|
if (res_pq_data.vnonce != authKeyData->nonce) {
|
|
LOG(("AuthKey Error: received nonce <> sent nonce (in res_pq)!"));
|
|
DEBUG_LOG(("AuthKey Error: received nonce: %1, sent nonce: %2").arg(Logs::mb(&res_pq_data.vnonce, 16).str()).arg(Logs::mb(&authKeyData->nonce, 16).str()));
|
|
return restart();
|
|
}
|
|
|
|
static MTP::internal::RSAPublicKeys RSAKeys = MTP::internal::InitRSAPublicKeys();
|
|
const MTP::internal::RSAPublicKey *rsaKey = nullptr;
|
|
const auto &fingerPrints(res_pq.c_resPQ().vserver_public_key_fingerprints.c_vector().v);
|
|
for (const auto &fingerPrint : fingerPrints) {
|
|
auto it = RSAKeys.constFind(static_cast<uint64>(fingerPrint.v));
|
|
if (it != RSAKeys.cend()) {
|
|
rsaKey = &it.value();
|
|
break;
|
|
}
|
|
}
|
|
if (!rsaKey) {
|
|
QStringList suggested, my;
|
|
for (const auto &fingerPrint : fingerPrints) {
|
|
suggested.push_back(QString("%1").arg(fingerPrint.v));
|
|
}
|
|
for (auto i = RSAKeys.cbegin(), e = RSAKeys.cend(); i != e; ++i) {
|
|
my.push_back(QString("%1").arg(i.key()));
|
|
}
|
|
LOG(("AuthKey Error: could not choose public RSA key, suggested fingerprints: %1, my fingerprints: %2").arg(suggested.join(", ")).arg(my.join(", ")));
|
|
return restart();
|
|
}
|
|
|
|
authKeyData->server_nonce = res_pq_data.vserver_nonce;
|
|
|
|
MTPP_Q_inner_data p_q_inner;
|
|
MTPDp_q_inner_data &p_q_inner_data(p_q_inner._p_q_inner_data());
|
|
p_q_inner_data.vnonce = authKeyData->nonce;
|
|
p_q_inner_data.vserver_nonce = authKeyData->server_nonce;
|
|
p_q_inner_data.vpq = res_pq_data.vpq;
|
|
|
|
const string &pq(res_pq_data.vpq.c_string().v);
|
|
string &p(p_q_inner_data.vp._string().v), &q(p_q_inner_data.vq._string().v);
|
|
|
|
if (!MTP::internal::parsePQ(pq, p, q)) {
|
|
LOG(("AuthKey Error: could not factor pq!"));
|
|
DEBUG_LOG(("AuthKey Error: problematic pq: %1").arg(Logs::mb(&pq[0], pq.length()).str()));
|
|
return restart();
|
|
}
|
|
|
|
authKeyData->new_nonce = rand_value<MTPint256>();
|
|
p_q_inner_data.vnew_nonce = authKeyData->new_nonce;
|
|
|
|
MTPReq_DH_params req_DH_params;
|
|
req_DH_params.vnonce = authKeyData->nonce;
|
|
req_DH_params.vserver_nonce = authKeyData->server_nonce;
|
|
req_DH_params.vpublic_key_fingerprint = MTP_long(rsaKey->getFingerPrint());
|
|
req_DH_params.vp = p_q_inner_data.vp;
|
|
req_DH_params.vq = p_q_inner_data.vq;
|
|
|
|
string &dhEncString(req_DH_params.vencrypted_data._string().v);
|
|
|
|
uint32 p_q_inner_size = p_q_inner.innerLength(), encSize = (p_q_inner_size >> 2) + 6;
|
|
if (encSize >= 65) {
|
|
mtpBuffer tmp;
|
|
tmp.reserve(encSize);
|
|
p_q_inner.write(tmp);
|
|
LOG(("AuthKey Error: too large data for RSA encrypt, size %1").arg(encSize * sizeof(mtpPrime)));
|
|
DEBUG_LOG(("AuthKey Error: bad data for RSA encrypt %1").arg(Logs::mb(&tmp[0], tmp.size() * 4).str()));
|
|
return restart(); // can't be 255-byte string
|
|
}
|
|
|
|
mtpBuffer encBuffer;
|
|
encBuffer.reserve(65); // 260 bytes
|
|
encBuffer.resize(6);
|
|
encBuffer[0] = 0;
|
|
p_q_inner.write(encBuffer);
|
|
|
|
hashSha1(&encBuffer[6], p_q_inner_size, &encBuffer[1]);
|
|
if (encSize < 65) {
|
|
encBuffer.resize(65);
|
|
memset_rand(&encBuffer[encSize], (65 - encSize) * sizeof(mtpPrime));
|
|
}
|
|
|
|
if (!rsaKey->encrypt(reinterpret_cast<const char*>(&encBuffer[0]) + 3, dhEncString)) {
|
|
return restart();
|
|
}
|
|
connect(_conn, SIGNAL(receivedData()), this, SLOT(dhParamsAnswered()));
|
|
|
|
DEBUG_LOG(("AuthKey Info: sending Req_DH_params..."));
|
|
sendRequestNotSecure(req_DH_params);
|
|
}
|
|
|
|
void ConnectionPrivate::dhParamsAnswered() {
|
|
disconnect(_conn, SIGNAL(receivedData()), this, SLOT(dhParamsAnswered()));
|
|
DEBUG_LOG(("AuthKey Info: receiving Req_DH_params answer..."));
|
|
|
|
MTPReq_DH_params::ResponseType res_DH_params;
|
|
if (!readResponseNotSecure(res_DH_params)) {
|
|
return restart();
|
|
}
|
|
|
|
switch (res_DH_params.type()) {
|
|
case mtpc_server_DH_params_ok: {
|
|
const auto &encDH(res_DH_params.c_server_DH_params_ok());
|
|
if (encDH.vnonce != authKeyData->nonce) {
|
|
LOG(("AuthKey Error: received nonce <> sent nonce (in server_DH_params_ok)!"));
|
|
DEBUG_LOG(("AuthKey Error: received nonce: %1, sent nonce: %2").arg(Logs::mb(&encDH.vnonce, 16).str()).arg(Logs::mb(&authKeyData->nonce, 16).str()));
|
|
return restart();
|
|
}
|
|
if (encDH.vserver_nonce != authKeyData->server_nonce) {
|
|
LOG(("AuthKey Error: received server_nonce <> sent server_nonce (in server_DH_params_ok)!"));
|
|
DEBUG_LOG(("AuthKey Error: received server_nonce: %1, sent server_nonce: %2").arg(Logs::mb(&encDH.vserver_nonce, 16).str()).arg(Logs::mb(&authKeyData->server_nonce, 16).str()));
|
|
return restart();
|
|
}
|
|
|
|
const string &encDHStr(encDH.vencrypted_answer.c_string().v);
|
|
uint32 encDHLen = encDHStr.length(), encDHBufLen = encDHLen >> 2;
|
|
if ((encDHLen & 0x03) || encDHBufLen < 6) {
|
|
LOG(("AuthKey Error: bad encrypted data length %1 (in server_DH_params_ok)!").arg(encDHLen));
|
|
DEBUG_LOG(("AuthKey Error: received encrypted data %1").arg(Logs::mb(&encDHStr[0], encDHLen).str()));
|
|
return restart();
|
|
}
|
|
|
|
uint32 nlen = authKeyData->new_nonce.innerLength(), slen = authKeyData->server_nonce.innerLength();
|
|
uchar tmp_aes[1024], sha1ns[20], sha1sn[20], sha1nn[20];
|
|
memcpy(tmp_aes, &authKeyData->new_nonce, nlen);
|
|
memcpy(tmp_aes + nlen, &authKeyData->server_nonce, slen);
|
|
memcpy(tmp_aes + nlen + slen, &authKeyData->new_nonce, nlen);
|
|
memcpy(tmp_aes + nlen + slen + nlen, &authKeyData->new_nonce, nlen);
|
|
hashSha1(tmp_aes, nlen + slen, sha1ns);
|
|
hashSha1(tmp_aes + nlen, nlen + slen, sha1sn);
|
|
hashSha1(tmp_aes + nlen + slen, nlen + nlen, sha1nn);
|
|
|
|
mtpBuffer decBuffer;
|
|
decBuffer.resize(encDHBufLen);
|
|
|
|
memcpy(authKeyData->aesKey, sha1ns, 20);
|
|
memcpy(authKeyData->aesKey + 20, sha1sn, 12);
|
|
memcpy(authKeyData->aesIV, sha1sn + 12, 8);
|
|
memcpy(authKeyData->aesIV + 8, sha1nn, 20);
|
|
memcpy(authKeyData->aesIV + 28, &authKeyData->new_nonce, 4);
|
|
|
|
aesIgeDecrypt(&encDHStr[0], &decBuffer[0], encDHLen, authKeyData->aesKey, authKeyData->aesIV);
|
|
|
|
const mtpPrime *from(&decBuffer[5]), *to(from), *end(from + (encDHBufLen - 5));
|
|
MTPServer_DH_inner_data dh_inner(to, end);
|
|
const auto &dh_inner_data(dh_inner.c_server_DH_inner_data());
|
|
if (dh_inner_data.vnonce != authKeyData->nonce) {
|
|
LOG(("AuthKey Error: received nonce <> sent nonce (in server_DH_inner_data)!"));
|
|
DEBUG_LOG(("AuthKey Error: received nonce: %1, sent nonce: %2").arg(Logs::mb(&dh_inner_data.vnonce, 16).str()).arg(Logs::mb(&authKeyData->nonce, 16).str()));
|
|
return restart();
|
|
}
|
|
if (dh_inner_data.vserver_nonce != authKeyData->server_nonce) {
|
|
LOG(("AuthKey Error: received server_nonce <> sent server_nonce (in server_DH_inner_data)!"));
|
|
DEBUG_LOG(("AuthKey Error: received server_nonce: %1, sent server_nonce: %2").arg(Logs::mb(&dh_inner_data.vserver_nonce, 16).str()).arg(Logs::mb(&authKeyData->server_nonce, 16).str()));
|
|
return restart();
|
|
}
|
|
uchar sha1Buffer[20];
|
|
if (memcmp(&decBuffer[0], hashSha1(&decBuffer[5], (to - from) * sizeof(mtpPrime), sha1Buffer), 20)) {
|
|
LOG(("AuthKey Error: sha1 hash of encrypted part did not match!"));
|
|
DEBUG_LOG(("AuthKey Error: sha1 did not match, server_nonce: %1, new_nonce %2, encrypted data %3").arg(Logs::mb(&authKeyData->server_nonce, 16).str()).arg(Logs::mb(&authKeyData->new_nonce, 16).str()).arg(Logs::mb(&encDHStr[0], encDHLen).str()));
|
|
return restart();
|
|
}
|
|
unixtimeSet(dh_inner_data.vserver_time.v);
|
|
|
|
const string &dhPrime(dh_inner_data.vdh_prime.c_string().v), &g_a(dh_inner_data.vg_a.c_string().v);
|
|
if (dhPrime.length() != 256 || g_a.length() != 256) {
|
|
LOG(("AuthKey Error: bad dh_prime len (%1) or g_a len (%2)").arg(dhPrime.length()).arg(g_a.length()));
|
|
DEBUG_LOG(("AuthKey Error: dh_prime %1, g_a %2").arg(Logs::mb(&dhPrime[0], dhPrime.length()).str()).arg(Logs::mb(&g_a[0], g_a.length()).str()));
|
|
return restart();
|
|
}
|
|
|
|
// check that dhPrime and (dhPrime - 1) / 2 are really prime using openssl BIGNUM methods
|
|
MTP::internal::BigNumPrimeTest bnPrimeTest;
|
|
if (!bnPrimeTest.isPrimeAndGood(&dhPrime[0], MTPMillerRabinIterCount, dh_inner_data.vg.v)) {
|
|
LOG(("AuthKey Error: bad dh_prime primality!").arg(dhPrime.length()).arg(g_a.length()));
|
|
DEBUG_LOG(("AuthKey Error: dh_prime %1").arg(Logs::mb(&dhPrime[0], dhPrime.length()).str()));
|
|
return restart();
|
|
}
|
|
|
|
authKeyStrings->dh_prime = QByteArray(dhPrime.data(), dhPrime.size());
|
|
authKeyData->g = dh_inner_data.vg.v;
|
|
authKeyStrings->g_a = QByteArray(g_a.data(), g_a.size());
|
|
authKeyData->retry_id = MTP_long(0);
|
|
authKeyData->retries = 0;
|
|
} return dhClientParamsSend();
|
|
|
|
case mtpc_server_DH_params_fail: {
|
|
const auto &encDH(res_DH_params.c_server_DH_params_fail());
|
|
if (encDH.vnonce != authKeyData->nonce) {
|
|
LOG(("AuthKey Error: received nonce <> sent nonce (in server_DH_params_fail)!"));
|
|
DEBUG_LOG(("AuthKey Error: received nonce: %1, sent nonce: %2").arg(Logs::mb(&encDH.vnonce, 16).str()).arg(Logs::mb(&authKeyData->nonce, 16).str()));
|
|
return restart();
|
|
}
|
|
if (encDH.vserver_nonce != authKeyData->server_nonce) {
|
|
LOG(("AuthKey Error: received server_nonce <> sent server_nonce (in server_DH_params_fail)!"));
|
|
DEBUG_LOG(("AuthKey Error: received server_nonce: %1, sent server_nonce: %2").arg(Logs::mb(&encDH.vserver_nonce, 16).str()).arg(Logs::mb(&authKeyData->server_nonce, 16).str()));
|
|
return restart();
|
|
}
|
|
uchar sha1Buffer[20];
|
|
if (encDH.vnew_nonce_hash != *(MTPint128*)(hashSha1(&authKeyData->new_nonce, 32, sha1Buffer) + 1)) {
|
|
LOG(("AuthKey Error: received new_nonce_hash did not match!"));
|
|
DEBUG_LOG(("AuthKey Error: received new_nonce_hash: %1, new_nonce: %2").arg(Logs::mb(&encDH.vnew_nonce_hash, 16).str()).arg(Logs::mb(&authKeyData->new_nonce, 32).str()));
|
|
return restart();
|
|
}
|
|
LOG(("AuthKey Error: server_DH_params_fail received!"));
|
|
} return restart();
|
|
|
|
}
|
|
LOG(("AuthKey Error: unknown server_DH_params received, typeId = %1").arg(res_DH_params.type()));
|
|
return restart();
|
|
}
|
|
|
|
void ConnectionPrivate::dhClientParamsSend() {
|
|
if (++authKeyData->retries > 5) {
|
|
LOG(("AuthKey Error: could not create auth_key for %1 retries").arg(authKeyData->retries - 1));
|
|
return restart();
|
|
}
|
|
|
|
MTPClient_DH_Inner_Data client_dh_inner;
|
|
MTPDclient_DH_inner_data &client_dh_inner_data(client_dh_inner._client_DH_inner_data());
|
|
client_dh_inner_data.vnonce = authKeyData->nonce;
|
|
client_dh_inner_data.vserver_nonce = authKeyData->server_nonce;
|
|
client_dh_inner_data.vretry_id = authKeyData->retry_id;
|
|
client_dh_inner_data.vg_b._string().v.resize(256);
|
|
|
|
// gen rand 'b'
|
|
uint32 b[64], *g_b((uint32*)&client_dh_inner_data.vg_b._string().v[0]);
|
|
memset_rand(b, sizeof(b));
|
|
|
|
// count g_b and auth_key using openssl BIGNUM methods
|
|
MTP::internal::BigNumCounter bnCounter;
|
|
if (!bnCounter.count(b, authKeyStrings->dh_prime.constData(), authKeyData->g, g_b, authKeyStrings->g_a.constData(), authKeyData->auth_key)) {
|
|
return dhClientParamsSend();
|
|
}
|
|
|
|
// count auth_key hashes - parts of sha1(auth_key)
|
|
uchar sha1Buffer[20];
|
|
int32 *auth_key_sha = hashSha1(authKeyData->auth_key, 256, sha1Buffer);
|
|
memcpy(&authKeyData->auth_key_aux_hash, auth_key_sha, 8);
|
|
memcpy(&authKeyData->auth_key_hash, auth_key_sha + 3, 8);
|
|
|
|
MTPSet_client_DH_params req_client_DH_params;
|
|
req_client_DH_params.vnonce = authKeyData->nonce;
|
|
req_client_DH_params.vserver_nonce = authKeyData->server_nonce;
|
|
|
|
string &sdhEncString(req_client_DH_params.vencrypted_data._string().v);
|
|
|
|
uint32 client_dh_inner_size = client_dh_inner.innerLength(), encSize = (client_dh_inner_size >> 2) + 5, encFullSize = encSize;
|
|
if (encSize & 0x03) {
|
|
encFullSize += 4 - (encSize & 0x03);
|
|
}
|
|
|
|
mtpBuffer encBuffer;
|
|
encBuffer.reserve(encFullSize);
|
|
encBuffer.resize(5);
|
|
client_dh_inner.write(encBuffer);
|
|
|
|
hashSha1(&encBuffer[5], client_dh_inner_size, &encBuffer[0]);
|
|
if (encSize < encFullSize) {
|
|
encBuffer.resize(encFullSize);
|
|
memset_rand(&encBuffer[encSize], (encFullSize - encSize) * sizeof(mtpPrime));
|
|
}
|
|
|
|
sdhEncString.resize(encFullSize * 4);
|
|
|
|
aesIgeEncrypt(&encBuffer[0], &sdhEncString[0], encFullSize * sizeof(mtpPrime), authKeyData->aesKey, authKeyData->aesIV);
|
|
|
|
connect(_conn, SIGNAL(receivedData()), this, SLOT(dhClientParamsAnswered()));
|
|
|
|
DEBUG_LOG(("AuthKey Info: sending Req_client_DH_params..."));
|
|
sendRequestNotSecure(req_client_DH_params);
|
|
}
|
|
|
|
void ConnectionPrivate::dhClientParamsAnswered() {
|
|
QReadLocker lockFinished(&sessionDataMutex);
|
|
if (!sessionData) return;
|
|
|
|
disconnect(_conn, SIGNAL(receivedData()), this, SLOT(dhClientParamsAnswered()));
|
|
DEBUG_LOG(("AuthKey Info: receiving Req_client_DH_params answer..."));
|
|
|
|
MTPSet_client_DH_params::ResponseType res_client_DH_params;
|
|
if (!readResponseNotSecure(res_client_DH_params)) {
|
|
lockFinished.unlock();
|
|
return restart();
|
|
}
|
|
|
|
switch (res_client_DH_params.type()) {
|
|
case mtpc_dh_gen_ok: {
|
|
const auto &resDH(res_client_DH_params.c_dh_gen_ok());
|
|
if (resDH.vnonce != authKeyData->nonce) {
|
|
LOG(("AuthKey Error: received nonce <> sent nonce (in dh_gen_ok)!"));
|
|
DEBUG_LOG(("AuthKey Error: received nonce: %1, sent nonce: %2").arg(Logs::mb(&resDH.vnonce, 16).str()).arg(Logs::mb(&authKeyData->nonce, 16).str()));
|
|
|
|
lockFinished.unlock();
|
|
return restart();
|
|
}
|
|
if (resDH.vserver_nonce != authKeyData->server_nonce) {
|
|
LOG(("AuthKey Error: received server_nonce <> sent server_nonce (in dh_gen_ok)!"));
|
|
DEBUG_LOG(("AuthKey Error: received server_nonce: %1, sent server_nonce: %2").arg(Logs::mb(&resDH.vserver_nonce, 16).str()).arg(Logs::mb(&authKeyData->server_nonce, 16).str()));
|
|
|
|
lockFinished.unlock();
|
|
return restart();
|
|
}
|
|
authKeyData->new_nonce_buf[32] = 1;
|
|
uchar sha1Buffer[20];
|
|
if (resDH.vnew_nonce_hash1 != *(MTPint128*)(hashSha1(authKeyData->new_nonce_buf, 41, sha1Buffer) + 1)) {
|
|
LOG(("AuthKey Error: received new_nonce_hash1 did not match!"));
|
|
DEBUG_LOG(("AuthKey Error: received new_nonce_hash1: %1, new_nonce_buf: %2").arg(Logs::mb(&resDH.vnew_nonce_hash1, 16).str()).arg(Logs::mb(authKeyData->new_nonce_buf, 41).str()));
|
|
|
|
lockFinished.unlock();
|
|
return restart();
|
|
}
|
|
|
|
uint64 salt1 = authKeyData->new_nonce.l.l, salt2 = authKeyData->server_nonce.l, serverSalt = salt1 ^ salt2;
|
|
sessionData->setSalt(serverSalt);
|
|
|
|
AuthKeyPtr authKey(new AuthKey());
|
|
authKey->setKey(authKeyData->auth_key);
|
|
authKey->setDC(bareDcId(dc));
|
|
|
|
DEBUG_LOG(("AuthKey Info: auth key gen succeed, id: %1, server salt: %2, auth key: %3").arg(authKey->keyId()).arg(serverSalt).arg(Logs::mb(authKeyData->auth_key, 256).str()));
|
|
|
|
sessionData->owner()->notifyKeyCreated(authKey); // slot will call authKeyCreated()
|
|
sessionData->clear();
|
|
unlockKey();
|
|
} return;
|
|
|
|
case mtpc_dh_gen_retry: {
|
|
const auto &resDH(res_client_DH_params.c_dh_gen_retry());
|
|
if (resDH.vnonce != authKeyData->nonce) {
|
|
LOG(("AuthKey Error: received nonce <> sent nonce (in dh_gen_retry)!"));
|
|
DEBUG_LOG(("AuthKey Error: received nonce: %1, sent nonce: %2").arg(Logs::mb(&resDH.vnonce, 16).str()).arg(Logs::mb(&authKeyData->nonce, 16).str()));
|
|
|
|
lockFinished.unlock();
|
|
return restart();
|
|
}
|
|
if (resDH.vserver_nonce != authKeyData->server_nonce) {
|
|
LOG(("AuthKey Error: received server_nonce <> sent server_nonce (in dh_gen_retry)!"));
|
|
DEBUG_LOG(("AuthKey Error: received server_nonce: %1, sent server_nonce: %2").arg(Logs::mb(&resDH.vserver_nonce, 16).str()).arg(Logs::mb(&authKeyData->server_nonce, 16).str()));
|
|
|
|
lockFinished.unlock();
|
|
return restart();
|
|
}
|
|
authKeyData->new_nonce_buf[32] = 2;
|
|
uchar sha1Buffer[20];
|
|
if (resDH.vnew_nonce_hash2 != *(MTPint128*)(hashSha1(authKeyData->new_nonce_buf, 41, sha1Buffer) + 1)) {
|
|
LOG(("AuthKey Error: received new_nonce_hash2 did not match!"));
|
|
DEBUG_LOG(("AuthKey Error: received new_nonce_hash2: %1, new_nonce_buf: %2").arg(Logs::mb(&resDH.vnew_nonce_hash2, 16).str()).arg(Logs::mb(authKeyData->new_nonce_buf, 41).str()));
|
|
|
|
lockFinished.unlock();
|
|
return restart();
|
|
}
|
|
authKeyData->retry_id = authKeyData->auth_key_aux_hash;
|
|
} return dhClientParamsSend();
|
|
|
|
case mtpc_dh_gen_fail: {
|
|
const auto &resDH(res_client_DH_params.c_dh_gen_fail());
|
|
if (resDH.vnonce != authKeyData->nonce) {
|
|
LOG(("AuthKey Error: received nonce <> sent nonce (in dh_gen_fail)!"));
|
|
DEBUG_LOG(("AuthKey Error: received nonce: %1, sent nonce: %2").arg(Logs::mb(&resDH.vnonce, 16).str()).arg(Logs::mb(&authKeyData->nonce, 16).str()));
|
|
|
|
lockFinished.unlock();
|
|
return restart();
|
|
}
|
|
if (resDH.vserver_nonce != authKeyData->server_nonce) {
|
|
LOG(("AuthKey Error: received server_nonce <> sent server_nonce (in dh_gen_fail)!"));
|
|
DEBUG_LOG(("AuthKey Error: received server_nonce: %1, sent server_nonce: %2").arg(Logs::mb(&resDH.vserver_nonce, 16).str()).arg(Logs::mb(&authKeyData->server_nonce, 16).str()));
|
|
|
|
lockFinished.unlock();
|
|
return restart();
|
|
}
|
|
authKeyData->new_nonce_buf[32] = 3;
|
|
uchar sha1Buffer[20];
|
|
if (resDH.vnew_nonce_hash3 != *(MTPint128*)(hashSha1(authKeyData->new_nonce_buf, 41, sha1Buffer) + 1)) {
|
|
LOG(("AuthKey Error: received new_nonce_hash3 did not match!"));
|
|
DEBUG_LOG(("AuthKey Error: received new_nonce_hash3: %1, new_nonce_buf: %2").arg(Logs::mb(&resDH.vnew_nonce_hash3, 16).str()).arg(Logs::mb(authKeyData->new_nonce_buf, 41).str()));
|
|
|
|
lockFinished.unlock();
|
|
return restart();
|
|
}
|
|
LOG(("AuthKey Error: dh_gen_fail received!"));
|
|
}
|
|
|
|
lockFinished.unlock();
|
|
return restart();
|
|
|
|
}
|
|
LOG(("AuthKey Error: unknown set_client_DH_params_answer received, typeId = %1").arg(res_client_DH_params.type()));
|
|
|
|
lockFinished.unlock();
|
|
return restart();
|
|
}
|
|
|
|
void ConnectionPrivate::authKeyCreated() {
|
|
clearAuthKeyData();
|
|
|
|
connect(_conn, SIGNAL(receivedData()), this, SLOT(handleReceived()));
|
|
|
|
if (sessionData->getSalt()) { // else receive salt in bad_server_salt first, then try to send all the requests
|
|
setState(ConnectedState);
|
|
if (restarted) {
|
|
emit resendAllAsync();
|
|
restarted = false;
|
|
}
|
|
}
|
|
|
|
_pingIdToSend = rand_value<uint64>(); // get server_salt
|
|
|
|
emit needToSendAsync();
|
|
}
|
|
|
|
void ConnectionPrivate::clearAuthKeyData() {
|
|
if (authKeyData) {
|
|
#ifdef Q_OS_WIN
|
|
SecureZeroMemory(authKeyData, sizeof(AuthKeyCreateData));
|
|
if (!authKeyStrings->dh_prime.isEmpty()) SecureZeroMemory(authKeyStrings->dh_prime.data(), authKeyStrings->dh_prime.size());
|
|
if (!authKeyStrings->g_a.isEmpty()) SecureZeroMemory(authKeyStrings->g_a.data(), authKeyStrings->g_a.size());
|
|
#else
|
|
memset(authKeyData, 0, sizeof(AuthKeyCreateData));
|
|
if (!authKeyStrings->dh_prime.isEmpty()) memset(authKeyStrings->dh_prime.data(), 0, authKeyStrings->dh_prime.size());
|
|
if (!authKeyStrings->g_a.isEmpty()) memset(authKeyStrings->g_a.data(), 0, authKeyStrings->g_a.size());
|
|
#endif
|
|
delete authKeyData;
|
|
authKeyData = 0;
|
|
delete authKeyStrings;
|
|
authKeyStrings = 0;
|
|
}
|
|
}
|
|
|
|
void ConnectionPrivate::onError4(bool mayBeBadKey) {
|
|
if (_conn && _conn == _conn6) return; // error in the unused
|
|
|
|
if (_conn || !_conn6) {
|
|
destroyConn();
|
|
_waitForConnectedTimer.stop();
|
|
|
|
MTP_LOG(dc, ("Restarting after error in IPv4 connection, maybe bad key: %1...").arg(Logs::b(mayBeBadKey)));
|
|
return restart(mayBeBadKey);
|
|
} else {
|
|
destroyConn(&_conn4);
|
|
}
|
|
}
|
|
|
|
void ConnectionPrivate::onError6(bool mayBeBadKey) {
|
|
if (_conn && _conn == _conn4) return; // error in the unused
|
|
|
|
if (_conn || !_conn4) {
|
|
destroyConn();
|
|
_waitForConnectedTimer.stop();
|
|
|
|
MTP_LOG(dc, ("Restarting after error in IPv6 connection, maybe bad key: %1...").arg(Logs::b(mayBeBadKey)));
|
|
return restart(mayBeBadKey);
|
|
} else {
|
|
destroyConn(&_conn6);
|
|
}
|
|
}
|
|
|
|
void ConnectionPrivate::onReadyData() {
|
|
}
|
|
|
|
template <typename TRequest>
|
|
void ConnectionPrivate::sendRequestNotSecure(const TRequest &request) {
|
|
try {
|
|
mtpBuffer buffer;
|
|
uint32 requestSize = request.innerLength() >> 2;
|
|
|
|
buffer.resize(0);
|
|
buffer.reserve(8 + requestSize);
|
|
buffer.push_back(0); // tcp packet len
|
|
buffer.push_back(0); // tcp packet num
|
|
buffer.push_back(0);
|
|
buffer.push_back(0);
|
|
buffer.push_back(authKeyData->req_num);
|
|
buffer.push_back(unixtime());
|
|
buffer.push_back(requestSize * 4);
|
|
request.write(buffer);
|
|
buffer.push_back(0); // tcp crc32 hash
|
|
++authKeyData->msgs_sent;
|
|
|
|
DEBUG_LOG(("AuthKey Info: sending request, size: %1, num: %2, time: %3").arg(requestSize).arg(authKeyData->req_num).arg(buffer[5]));
|
|
|
|
_conn->sendData(buffer);
|
|
|
|
onSentSome(buffer.size() * sizeof(mtpPrime));
|
|
|
|
} catch (Exception &) {
|
|
return restart();
|
|
}
|
|
}
|
|
|
|
template <typename TResponse>
|
|
bool ConnectionPrivate::readResponseNotSecure(TResponse &response) {
|
|
onReceivedSome();
|
|
|
|
try {
|
|
if (_conn->received().isEmpty()) {
|
|
LOG(("AuthKey Error: trying to read response from empty received list"));
|
|
return false;
|
|
}
|
|
mtpBuffer buffer(_conn->received().front());
|
|
_conn->received().pop_front();
|
|
|
|
const mtpPrime *answer(buffer.constData());
|
|
uint32 len = buffer.size();
|
|
if (len < 5) {
|
|
LOG(("AuthKey Error: bad request answer, len = %1").arg(len * sizeof(mtpPrime)));
|
|
DEBUG_LOG(("AuthKey Error: answer bytes %1").arg(Logs::mb(answer, len * sizeof(mtpPrime)).str()));
|
|
return false;
|
|
}
|
|
if (answer[0] != 0 || answer[1] != 0 || (((uint32)answer[2]) & 0x03) != 1/* || (unixtime() - answer[3] > 300) || (answer[3] - unixtime() > 60)*/) { // didnt sync time yet
|
|
LOG(("AuthKey Error: bad request answer start (%1 %2 %3)").arg(answer[0]).arg(answer[1]).arg(answer[2]));
|
|
DEBUG_LOG(("AuthKey Error: answer bytes %1").arg(Logs::mb(answer, len * sizeof(mtpPrime)).str()));
|
|
return false;
|
|
}
|
|
uint32 answerLen = (uint32)answer[4];
|
|
if (answerLen != (len - 5) * sizeof(mtpPrime)) {
|
|
LOG(("AuthKey Error: bad request answer %1 <> %2").arg(answerLen).arg((len - 5) * sizeof(mtpPrime)));
|
|
DEBUG_LOG(("AuthKey Error: answer bytes %1").arg(Logs::mb(answer, len * sizeof(mtpPrime)).str()));
|
|
return false;
|
|
}
|
|
const mtpPrime *from(answer + 5), *end(from + len - 5);
|
|
response.read(from, end);
|
|
} catch (Exception &) {
|
|
return false;
|
|
}
|
|
return true;
|
|
}
|
|
|
|
bool ConnectionPrivate::sendRequest(mtpRequest &request, bool needAnyResponse, QReadLocker &lockFinished) {
|
|
uint32 fullSize = request->size();
|
|
if (fullSize < 9) return false;
|
|
|
|
uint32 messageSize = mtpRequestData::messageSize(request);
|
|
if (messageSize < 5 || fullSize < messageSize + 4) return false;
|
|
|
|
ReadLockerAttempt lock(sessionData->keyMutex());
|
|
if (!lock) {
|
|
DEBUG_LOG(("MTP Info: could not lock key for read in sendBuffer(), dc %1, restarting...").arg(dc));
|
|
|
|
lockFinished.unlock();
|
|
restart();
|
|
return false;
|
|
}
|
|
|
|
AuthKeyPtr key(sessionData->getKey());
|
|
if (!key || key->keyId() != keyId) {
|
|
DEBUG_LOG(("MTP Error: auth_key id for dc %1 changed").arg(dc));
|
|
|
|
lockFinished.unlock();
|
|
restart();
|
|
return false;
|
|
}
|
|
|
|
uint32 padding = fullSize - 4 - messageSize;
|
|
uint64 session(sessionData->getSession()), salt(sessionData->getSalt());
|
|
|
|
memcpy(request->data() + 0, &salt, 2 * sizeof(mtpPrime));
|
|
memcpy(request->data() + 2, &session, 2 * sizeof(mtpPrime));
|
|
|
|
const mtpPrime *from = request->constData() + 4;
|
|
MTP_LOG(dc, ("Send: ") + mtpTextSerialize(from, from + messageSize));
|
|
|
|
uchar encryptedSHA[20];
|
|
MTPint128 &msgKey(*(MTPint128*)(encryptedSHA + 4));
|
|
hashSha1(request->constData(), (fullSize - padding) * sizeof(mtpPrime), encryptedSHA);
|
|
|
|
mtpBuffer result;
|
|
result.resize(9 + fullSize);
|
|
*((uint64*)&result[2]) = keyId;
|
|
*((MTPint128*)&result[4]) = msgKey;
|
|
|
|
aesIgeEncrypt(request->constData(), &result[8], fullSize * sizeof(mtpPrime), key, msgKey);
|
|
|
|
DEBUG_LOG(("MTP Info: sending request, size: %1, num: %2, time: %3").arg(fullSize + 6).arg((*request)[4]).arg((*request)[5]));
|
|
|
|
_conn->setSentEncrypted();
|
|
_conn->sendData(result);
|
|
|
|
if (needAnyResponse) {
|
|
onSentSome(result.size() * sizeof(mtpPrime));
|
|
}
|
|
|
|
return true;
|
|
}
|
|
|
|
mtpRequestId ConnectionPrivate::wasSent(mtpMsgId msgId) const {
|
|
if (msgId == _pingMsgId) return mtpRequestId(0xFFFFFFFF);
|
|
{
|
|
QReadLocker locker(sessionData->haveSentMutex());
|
|
const mtpRequestMap &haveSent(sessionData->haveSentMap());
|
|
mtpRequestMap::const_iterator i = haveSent.constFind(msgId);
|
|
if (i != haveSent.cend()) return i.value()->requestId ? i.value()->requestId : mtpRequestId(0xFFFFFFFF);
|
|
}
|
|
{
|
|
QReadLocker locker(sessionData->toResendMutex());
|
|
const mtpRequestIdsMap &toResend(sessionData->toResendMap());
|
|
mtpRequestIdsMap::const_iterator i = toResend.constFind(msgId);
|
|
if (i != toResend.cend()) return i.value();
|
|
}
|
|
{
|
|
QReadLocker locker(sessionData->wereAckedMutex());
|
|
const mtpRequestIdsMap &wereAcked(sessionData->wereAckedMap());
|
|
mtpRequestIdsMap::const_iterator i = wereAcked.constFind(msgId);
|
|
if (i != wereAcked.cend()) return i.value();
|
|
}
|
|
return 0;
|
|
}
|
|
|
|
void ConnectionPrivate::lockKey() {
|
|
unlockKey();
|
|
sessionData->keyMutex()->lockForWrite();
|
|
myKeyLock = true;
|
|
}
|
|
|
|
void ConnectionPrivate::unlockKey() {
|
|
if (myKeyLock) {
|
|
myKeyLock = false;
|
|
sessionData->keyMutex()->unlock();
|
|
}
|
|
}
|
|
|
|
ConnectionPrivate::~ConnectionPrivate() {
|
|
t_assert(_finished && _conn == nullptr && _conn4 == nullptr && _conn6 == nullptr);
|
|
}
|
|
|
|
void ConnectionPrivate::stop() {
|
|
QWriteLocker lockFinished(&sessionDataMutex);
|
|
if (sessionData) {
|
|
if (myKeyLock) {
|
|
sessionData->owner()->notifyKeyCreated(AuthKeyPtr()); // release key lock, let someone else create it
|
|
sessionData->keyMutex()->unlock();
|
|
myKeyLock = false;
|
|
}
|
|
sessionData = nullptr;
|
|
}
|
|
}
|
|
|
|
} // namespace internal
|
|
} // namespace MTP
|