procxx
/
kepka
mirror of https://github.com/procxx/kepka.git
1
0
Fork 0
Code Issues Releases Wiki Activity

Send different dns requests for simple config.

This commit is contained in:
John Preston 2018-05-03 11:11:07 +03:00
parent ad1f089802
commit db7041f2dc
2 changed files with 183 additions and 103 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

249
Telegram/SourceFiles/mtproto/special_config_request.cpp
View File

@ -16,6 +16,8 @@ https://github.com/telegramdesktop/tdesktop/blob/master/LEGAL
namespace MTP { namespace MTP {
namespace { namespace {
constexpr auto kSendNextTimeout = TimeMs(1000);
constexpr auto kPublicKey = str_const("\ constexpr auto kPublicKey = str_const("\
-----BEGIN RSA PUBLIC KEY-----\n\ -----BEGIN RSA PUBLIC KEY-----\n\
MIIBCgKCAQEAyr+18Rex2ohtVy8sroGPBwXD3DOoKCSpjDqYoXgCqB7ioln4eDCF\n\ MIIBCgKCAQEAyr+18Rex2ohtVy8sroGPBwXD3DOoKCSpjDqYoXgCqB7ioln4eDCF\n\
@ -27,6 +29,9 @@ Y1hZCxdv6cs5UnW9+PWvS+WIbkh+GaWYxwIDAQAB\n\
-----END RSA PUBLIC KEY-----\ -----END RSA PUBLIC KEY-----\
"); ");
constexpr auto kUserAgent = "Mozilla/5.0 (Windows NT 10.0; Win64; x64) "
"AppleWebKit/537.36 (KHTML, like Gecko) Chrome/66.0.3359.117 Safari/537.36";
bool CheckPhoneByPrefixesRules(const QString &phone, const QString &rules) { bool CheckPhoneByPrefixesRules(const QString &phone, const QString &rules) {
auto result = false; auto result = false;
for (const auto &prefix : rules.split(',')) { for (const auto &prefix : rules.split(',')) {
@ -41,8 +46,85 @@ bool CheckPhoneByPrefixesRules(const QString &phone, const QString &rules) {
return result; return result;
} }
QByteArray ParseDnsResponse(const QByteArray &response) {
// Read and store to "entries" map all the data bytes from the response:
// { ..,
// "Answer": [
// { .., "data": "bytes1", .. },
// { .., "data": "bytes2", .. }
// ],
// .. }
auto entries = QMap<int, QString>();
auto error = QJsonParseError{ 0, QJsonParseError::NoError };
auto document = QJsonDocument::fromJson(response, &error);
if (error.error != QJsonParseError::NoError) {
LOG(("Config Error: Failed to parse dns response JSON, error: %1"
).arg(error.errorString()));
} else if (!document.isObject()) {
LOG(("Config Error: Not an object received in dns response JSON."));
} else {
auto response = document.object();
auto answerIt = response.find(qsl("Answer"));
if (answerIt == response.constEnd()) {
LOG(("Config Error: Could not find Answer "
"in dns response JSON."));
} else if (!(*answerIt).isArray()) {
LOG(("Config Error: Not an array received "
"in Answer in dns response JSON."));
} else {
for (auto elem : (*answerIt).toArray()) {
if (!elem.isObject()) {
LOG(("Config Error: Not an object found "
"in Answer array in dns response JSON."));
} else {
auto object = elem.toObject();
auto dataIt = object.find(qsl("data"));
if (dataIt == object.constEnd()) {
LOG(("Config Error: Could not find data "
"in Answer array entry in dns response JSON."));
} else if (!(*dataIt).isString()) {
LOG(("Config Error: Not a string data found "
"in Answer array entry in dns response JSON."));
} else {
auto data = (*dataIt).toString();
entries.insertMulti(INT_MAX - data.size(), data);
}
}
}
}
}
return QStringList(entries.values()).join(QString()).toLatin1();
}
} // namespace } // namespace
SpecialConfigRequest::Request::Request(not_null<QNetworkReply*> reply)
: reply(reply.get()) {
}
SpecialConfigRequest::Request::Request(Request &&other)
: reply(base::take(other.reply)) {
}
auto SpecialConfigRequest::Request::operator=(Request &&other) -> Request& {
if (reply != other.reply) {
destroy();
reply = base::take(other.reply);
}
return *this;
}
void SpecialConfigRequest::Request::destroy() {
if (const auto value = base::take(reply)) {
value->deleteLater();
value->abort();
}
}
SpecialConfigRequest::Request::~Request() {
destroy();
}
SpecialConfigRequest::SpecialConfigRequest( SpecialConfigRequest::SpecialConfigRequest(
base::lambda<void( base::lambda<void(
DcId dcId, DcId dcId,
@ -52,97 +134,89 @@ SpecialConfigRequest::SpecialConfigRequest(
const QString &phone) const QString &phone)
: _callback(std::move(callback)) : _callback(std::move(callback))
, _phone(phone) { , _phone(phone) {
performAppRequest(); _attempts = {
performDnsRequest(); { Type::App, qsl("software-download.microsoft.com") },
{ Type::Dns, qsl("google.com") },
{ Type::Dns, qsl("www.google.com") },
{ Type::Dns, qsl("google.ru") },
{ Type::Dns, qsl("www.google.ru") },
};
std::random_device rd;
ranges::shuffle(_attempts, std::mt19937(rd()));
sendNextRequest();
} }
void SpecialConfigRequest::performAppRequest() { void SpecialConfigRequest::sendNextRequest() {
auto appUrl = QUrl(); Expects(!_attempts.empty());
appUrl.setScheme(qsl("https"));
appUrl.setHost(qsl("software-download.microsoft.com")); const auto attempt = _attempts.back();
appUrl.setPath(cTestMode() _attempts.pop_back();
? qsl("/test/config.txt") if (!_attempts.empty()) {
: qsl("/prodv2/config.txt")); App::CallDelayed(kSendNextTimeout, this, [=] {
auto appRequest = QNetworkRequest(appUrl); sendNextRequest();
appRequest.setRawHeader("Host", "tcdnb.azureedge.net"); });
_appReply.reset(_manager.get(appRequest)); }
connect(_appReply.get(), &QNetworkReply::finished, this, [=] { performRequest(attempt);
appFinished(); }
void SpecialConfigRequest::performRequest(const Attempt &attempt) {
const auto type = attempt.type;
auto url = QUrl();
url.setScheme(qsl("https"));
url.setHost(attempt.domain);
auto request = QNetworkRequest();
switch (type) {
case Type::App: {
url.setPath(cTestMode()
? qsl("/test/config.txt")
: qsl("/prodv2/config.txt"));
request.setRawHeader("Host", "tcdnb.azureedge.net");
} break;
case Type::Dns: {
url.setPath(qsl("/resolve"));
url.setQuery(
qsl("name=%1.stel.com&type=16").arg(
cTestMode() ? qsl("tap") : qsl("apv2")));
request.setRawHeader("Host", "dns.google.com");
} break;
default: Unexpected("Type in SpecialConfigRequest::performRequest.");
}
request.setUrl(url);
request.setRawHeader("User-Agent", kUserAgent);
const auto reply = _requests.emplace_back(
_manager.get(request)
).reply;
connect(reply, &QNetworkReply::finished, this, [=] {
requestFinished(type, reply);
}); });
} }
void SpecialConfigRequest::performDnsRequest() { void SpecialConfigRequest::requestFinished(
auto dnsUrl = QUrl(); Type type,
dnsUrl.setScheme(qsl("https")); not_null<QNetworkReply*> reply) {
dnsUrl.setHost(qsl("www.google.com")); const auto result = finalizeRequest(reply);
dnsUrl.setPath(qsl("/resolve")); switch (type) {
dnsUrl.setQuery( case Type::App: handleResponse(result); break;
qsl("name=%1.stel.com&type=16").arg( case Type::Dns: handleResponse(ParseDnsResponse(result)); break;
cTestMode() ? qsl("tap") : qsl("apv2"))); default: Unexpected("Type in SpecialConfigRequest::requestFinished.");
auto dnsRequest = QNetworkRequest(QUrl(dnsUrl)); }
dnsRequest.setRawHeader("Host", "dns.google.com");
_dnsReply.reset(_manager.get(dnsRequest));
connect(_dnsReply.get(), &QNetworkReply::finished, this, [this] {
dnsFinished();
});
} }
void SpecialConfigRequest::appFinished() { QByteArray SpecialConfigRequest::finalizeRequest(
if (!_appReply) { not_null<QNetworkReply*> reply) {
return; if (reply->error() != QNetworkReply::NoError) {
LOG(("Config Error: Failed to get response from %1, error: %2 (%3)"
).arg(reply->request().url().toDisplayString()
).arg(reply->errorString()
).arg(reply->error()));
} }
auto result = _appReply->readAll(); const auto result = reply->readAll();
_appReply.release()->deleteLater(); const auto from = ranges::remove(
handleResponse(result); _requests,
} reply,
[](const Request &request) { return request.reply; });
void SpecialConfigRequest::dnsFinished() { _requests.erase(from, end(_requests));
if (!_dnsReply) { return result;
return;
}
if (_dnsReply->error() != QNetworkReply::NoError) {
LOG(("Config Error: Failed to get dns response JSON, error: %1 (%2)").arg(_dnsReply->errorString()).arg(_dnsReply->error()));
}
auto result = _dnsReply->readAll();
_dnsReply.release()->deleteLater();
// Read and store to "entries" map all the data bytes from this response:
// { .., "Answer": [ { .., "data": "bytes1", .. }, { .., "data": "bytes2", .. } ], .. }
auto entries = QMap<int, QString>();
auto error = QJsonParseError { 0, QJsonParseError::NoError };
auto document = QJsonDocument::fromJson(result, &error);
if (error.error != QJsonParseError::NoError) {
LOG(("Config Error: Failed to parse dns response JSON, error: %1").arg(error.errorString()));
} else if (!document.isObject()) {
LOG(("Config Error: Not an object received in dns response JSON."));
} else {
auto response = document.object();
auto answerIt = response.find(qsl("Answer"));
if (answerIt == response.constEnd()) {
LOG(("Config Error: Could not find Answer in dns response JSON."));
} else if (!(*answerIt).isArray()) {
LOG(("Config Error: Not an array received in Answer in dns response JSON."));
} else {
for (auto elem : (*answerIt).toArray()) {
if (!elem.isObject()) {
LOG(("Config Error: Not an object found in Answer array in dns response JSON."));
} else {
auto object = elem.toObject();
auto dataIt = object.find(qsl("data"));
if (dataIt == object.constEnd()) {
LOG(("Config Error: Could not find data in Answer array entry in dns response JSON."));
} else if (!(*dataIt).isString()) {
LOG(("Config Error: Not a string data found in Answer array entry in dns response JSON."));
} else {
auto data = (*dataIt).toString();
entries.insertMulti(INT_MAX - data.size(), data);
}
}
}
}
}
auto text = QStringList(entries.values()).join(QString());
handleResponse(text.toLatin1());
} }
bool SpecialConfigRequest::decryptSimpleConfig(const QByteArray &bytes) { bool SpecialConfigRequest::decryptSimpleConfig(const QByteArray &bytes) {
@ -270,13 +344,4 @@ void SpecialConfigRequest::handleResponse(const QByteArray &bytes) {
} }
} }
SpecialConfigRequest::~SpecialConfigRequest() {
if (_appReply) {
_appReply->abort();
}
if (_dnsReply) {
_dnsReply->abort();
}
}
} // namespace MTP } // namespace MTP

37
Telegram/SourceFiles/mtproto/special_config_request.h
View File

@ -21,13 +21,31 @@ public:
bytes::const_span secret)> callback, bytes::const_span secret)> callback,
const QString &phone); const QString &phone);
~SpecialConfigRequest();
private: private:
void performAppRequest(); enum class Type {
void performDnsRequest(); App,
void appFinished(); Dns,
void dnsFinished(); };
struct Attempt {
Type type;
QString domain;
};
struct Request {
Request(not_null<QNetworkReply*> reply);
Request(Request &&other);
Request &operator=(Request &&other);
~Request();
void destroy();
QPointer<QNetworkReply> reply;
};
void sendNextRequest();
void performRequest(const Attempt &attempt);
void requestFinished(Type type, not_null<QNetworkReply*> reply);
QByteArray finalizeRequest(not_null<QNetworkReply*> reply);
void handleResponse(const QByteArray &bytes); void handleResponse(const QByteArray &bytes);
bool decryptSimpleConfig(const QByteArray &bytes); bool decryptSimpleConfig(const QByteArray &bytes);
@ -40,11 +58,8 @@ private:
MTPhelp_ConfigSimple _simpleConfig; MTPhelp_ConfigSimple _simpleConfig;
QNetworkAccessManager _manager; QNetworkAccessManager _manager;
std::unique_ptr<QNetworkReply> _appReply; std::vector<Attempt> _attempts;
std::unique_ptr<QNetworkReply> _dnsReply; std::vector<Request> _requests;
std::unique_ptr<DcOptions> _localOptions;
std::unique_ptr<Instance> _localInstance;
}; };