// // Copyright Aliaksei Levin (levlam@telegram.org), Arseny Smirnov (arseny30@gmail.com) 2014-2023 // // Distributed under the Boost Software License, Version 1.0. (See accompanying // file LICENSE_1_0.txt or copy at http://www.boost.org/LICENSE_1_0.txt) // #include "td/utils/BigNum.h" #include "td/utils/common.h" #include "td/utils/format.h" #include "td/utils/logging.h" #include "td/utils/misc.h" #include "td/utils/port/detail/Iocp.h" #include "td/utils/port/IPAddress.h" #include "td/utils/port/sleep.h" #include "td/utils/port/SocketFd.h" #include "td/utils/port/thread.h" #include "td/utils/Random.h" #include "td/utils/Slice.h" #include "td/utils/SliceBuilder.h" #include "td/utils/Status.h" #include "td/utils/Time.h" #include <map> static td::BigNumContext context; static bool is_quadratic_residue(const td::BigNum &a) { // 2^255 - 19 td::BigNum mod = td::BigNum::from_hex("7fffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffed").move_as_ok(); // (mod - 1) / 2 = 2^254 - 10 td::BigNum pow = td::BigNum::from_hex("3ffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffff6").move_as_ok(); td::BigNum r; td::BigNum::mod_exp(r, a, pow, mod, context); td::BigNum one = td::BigNum::from_decimal("1").move_as_ok(); td::BigNum::mod_add(r, r, one, mod, context); td::string result = r.to_decimal(); CHECK(result == "0" || result == "1" || result == "2"); return result == "2"; } struct TlsInfo { td::vector<size_t> extension_list; td::vector<size_t> encrypted_application_data_length; }; td::Result<TlsInfo> test_tls(const td::string &url) { td::IPAddress address; TRY_STATUS(address.init_host_port(url, 443)); TRY_RESULT(socket, td::SocketFd::open(address)); td::string request; auto add_string = [&](td::Slice data) { request.append(data.data(), data.size()); }; auto add_random = [&](size_t length) { while (length-- > 0) { request += static_cast<char>(td::Random::secure_int32()); } }; auto add_length = [&](size_t length) { request += static_cast<char>(length / 256); request += static_cast<char>(length % 256); }; auto add_key = [&] { td::string key(32, '\0'); td::BigNum mod = td::BigNum::from_hex("7fffffffffffffffffffffffffffffffffffffffffffffffffffffffffffffed").move_as_ok(); while (true) { td::Random::secure_bytes(key); key[31] = static_cast<char>(key[31] & 127); td::BigNum x = td::BigNum::from_le_binary(key); if (!is_quadratic_residue(x)) { continue; } td::BigNum y = x.clone(); td::BigNum coef = td::BigNum::from_decimal("486662").move_as_ok(); td::BigNum::mod_add(y, y, coef, mod, context); td::BigNum::mod_mul(y, y, x, mod, context); td::BigNum one = td::BigNum::from_decimal("1").move_as_ok(); td::BigNum::mod_add(y, y, one, mod, context); td::BigNum::mod_mul(y, y, x, mod, context); // y = x^3 + 486662 * x^2 + x if (is_quadratic_residue(y)) { break; } } request += key; }; const size_t MAX_GREASE = 7; char greases[MAX_GREASE]; td::Random::secure_bytes(td::MutableSlice{greases, MAX_GREASE}); for (auto &grease : greases) { grease = static_cast<char>((grease & 0xF0) + 0x0A); } for (size_t i = 1; i < MAX_GREASE; i += 2) { if (greases[i] == greases[i - 1]) { greases[i] = static_cast<char>(0x10 ^ greases[i]); } } auto add_grease = [&](size_t num) { auto c = greases[num]; request += c; request += c; }; add_string("\x16\x03\x01\x02\x00\x01\x00\x01\xfc\x03\x03"); add_random(32); add_string("\x20"); add_random(32); add_string("\x00\x20"); add_grease(0); add_string( "\x13\x01\x13\x02\x13\x03\xc0\x2b\xc0\x2f\xc0\x2c\xc0\x30\xcc\xa9\xcc\xa8\xc0\x13\xc0\x14\x00\x9c\x00\x9d\x00" "\x2f\x00\x35\x01\x00\x01\x93"); add_grease(2); add_string("\x00\x00\x00\x00"); add_length(url.size() + 5); add_length(url.size() + 3); add_string("\x00"); add_length(url.size()); add_string(url); add_string("\x00\x17\x00\x00\xff\x01\x00\x01\x00\x00\x0a\x00\x0a\x00\x08"); add_grease(4); add_string( "\x00\x1d\x00\x17\x00\x18\x00\x0b\x00\x02\x01\x00\x00\x23\x00\x00\x00\x10\x00\x0e\x00\x0c\x02\x68\x32\x08\x68" "\x74\x74\x70\x2f\x31\x2e\x31\x00\x05\x00\x05\x01\x00\x00\x00\x00\x00\x0d\x00\x12\x00\x10\x04\x03\x08\x04\x04" "\x01\x05\x03\x08\x05\x05\x01\x08\x06\x06\x01\x00\x12\x00\x00\x00\x33\x00\x2b\x00\x29"); add_grease(4); add_string("\x00\x01\x00\x00\x1d\x00\x20"); add_key(); add_string("\x00\x2d\x00\x02\x01\x01\x00\x2b\x00\x0b\x0a"); add_grease(6); add_string("\x03\x04\x03\x03\x03\x02\x03\x01\x00\x1b\x00\x03\x02\x00\x02"); add_grease(3); add_string("\x00\x01\x00\x00\x15"); auto padding = 515 - static_cast<int>(request.size()); CHECK(padding >= 0); add_length(padding); request.resize(517); // LOG(ERROR) << td::format::as_hex_dump<0>(td::Slice(request)); TRY_STATUS(socket.write(request)); TlsInfo info; auto end_time = td::Time::now() + 3; td::string result; size_t pos = 0; size_t server_hello_length = 0; size_t encrypted_application_data_length_sum = 0; while (td::Time::now() < end_time) { static char buf[20000]; TRY_RESULT(res, socket.read(td::MutableSlice{buf, sizeof(buf)})); if (res > 0) { auto read_length = [&]() -> size_t { CHECK(result.size() >= 2 + pos); pos += 2; return static_cast<unsigned char>(result[pos - 2]) * 256 + static_cast<unsigned char>(result[pos - 1]); }; result += td::Slice(buf, res).str(); while (true) { #define CHECK_LENGTH(length) \ if (pos + (length) > result.size()) { \ break; \ } #define EXPECT_STR(pos, str, error) \ if (!begins_with(td::Slice(result).substr(pos), str)) { \ return td::Status::Error(error); \ } if (pos == 0) { CHECK_LENGTH(3); EXPECT_STR(0, "\x16\x03\x03", "Non-TLS response or TLS <= 1.1"); pos += 3; } if (pos == 3) { CHECK_LENGTH(2); server_hello_length = read_length(); if (server_hello_length <= 39) { return td::Status::Error("Receive too short server hello"); } } if (server_hello_length > 0) { if (pos == 5) { CHECK_LENGTH(server_hello_length); EXPECT_STR(5, "\x02\x00", "Non-TLS response 2"); EXPECT_STR(9, "\x03\x03", "Non-TLS response 3"); auto random_id = td::Slice(result.c_str() + 11, 32); if (random_id == "\xcf\x21\xad\x74\xe5\x9a\x61\x11\xbe\x1d\x8c\x02\x1e\x65\xb8\x91\xc2\xa2\x11\x16\x7a\xbb\x8c\x5e\x07" "\x9e\x09\xe2\xc8\xa8\x33\x9c") { return td::Status::Error("TLS 1.3 servers returning HelloRetryRequest are not supprted"); } if (result[43] == '\x00') { return td::Status::Error("TLS <= 1.2: empty session_id"); } EXPECT_STR(43, "\x20", "Non-TLS response 4"); if (server_hello_length <= 75) { return td::Status::Error("Receive too short server hello 2"); } EXPECT_STR(44, request.substr(44, 32), "TLS <= 1.2: expected mirrored session_id"); EXPECT_STR(76, "\x13\x01\x00", "TLS <= 1.2: expected 0x1301 as a chosen cipher"); pos += 74; size_t extensions_length = read_length(); if (extensions_length + 76 != server_hello_length) { return td::Status::Error("Receive wrong extensions length"); } while (pos < 5 + server_hello_length - 4) { info.extension_list.push_back(read_length()); size_t extension_length = read_length(); if (pos + extension_length > 5 + server_hello_length) { return td::Status::Error("Receive wrong extension length"); } pos += extension_length; } if (pos != 5 + server_hello_length) { return td::Status::Error("Receive wrong extensions list"); } } if (pos == 5 + server_hello_length) { CHECK_LENGTH(6); EXPECT_STR(pos, "\x14\x03\x03\x00\x01\x01", "Expected dummy ChangeCipherSpec"); pos += 6; } if (pos == 11 + server_hello_length + encrypted_application_data_length_sum) { if (pos == result.size()) { return info; } CHECK_LENGTH(3); EXPECT_STR(pos, "\x17\x03\x03", "Expected encrypted application data"); pos += 3; } if (pos == 14 + server_hello_length + encrypted_application_data_length_sum) { CHECK_LENGTH(2); size_t encrypted_application_data_length = read_length(); info.encrypted_application_data_length.push_back(encrypted_application_data_length); if (encrypted_application_data_length == 0) { return td::Status::Error("Receive empty encrypted application data"); } } if (pos == 16 + server_hello_length + encrypted_application_data_length_sum) { CHECK_LENGTH(info.encrypted_application_data_length.back()); pos += info.encrypted_application_data_length.back(); encrypted_application_data_length_sum += info.encrypted_application_data_length.back() + 5; } } } } else { td::usleep_for(10000); } } // LOG(ERROR) << url << ":" << td::format::as_hex_dump<0>(td::Slice(result)); return td::Status::Error("Failed to get response in 3 seconds"); } int main(int argc, char *argv[]) { SET_VERBOSITY_LEVEL(VERBOSITY_NAME(WARNING)); #if TD_PORT_WINDOWS auto iocp = td::make_unique<td::detail::Iocp>(); iocp->init(); auto iocp_thread = td::thread([&iocp] { iocp->loop(); }); td::detail::Iocp::Guard iocp_guard(iocp.get()); #endif td::vector<td::string> urls; for (int i = 1; i < argc; i++) { urls.emplace_back(argv[i]); } for (auto &url : urls) { const int MAX_TRIES = 100; td::vector<std::map<size_t, int>> length_count; td::vector<size_t> extension_list; for (int i = 0; i < MAX_TRIES; i++) { auto r_tls_info = test_tls(url); if (r_tls_info.is_error()) { LOG(ERROR) << url << ": " << r_tls_info.error(); break; } else { auto tls_info = r_tls_info.move_as_ok(); if (length_count.size() < tls_info.encrypted_application_data_length.size()) { length_count.resize(tls_info.encrypted_application_data_length.size()); } for (size_t t = 0; t < tls_info.encrypted_application_data_length.size(); t++) { length_count[t][tls_info.encrypted_application_data_length[t]]++; } if (i == 0) { extension_list = tls_info.extension_list; } else { if (extension_list != tls_info.extension_list) { LOG(ERROR) << url << ": TLS 1.3.0 extension list has changed from " << extension_list << " to " << tls_info.extension_list; break; } } } if (i == MAX_TRIES - 1) { if (extension_list != td::vector<size_t>{51, 43} && extension_list != td::vector<size_t>{43, 51}) { LOG(ERROR) << url << ": TLS 1.3.0 unsupported extension list " << extension_list; } else { td::string length_distribution = "|"; for (size_t t = 0; t < length_count.size(); t++) { for (auto it : length_count[t]) { length_distribution += PSTRING() << it.first << " : " << static_cast<int>(it.second * 100.0 / MAX_TRIES) << "%|"; } if (t + 1 != length_count.size()) { length_distribution += " + |"; } } LOG(ERROR) << url << ": TLS 1.3.0 with extensions " << extension_list << " and " << (length_count.size() != 1 ? "unsupported " : "") << "encrypted application data length distribution " << length_distribution; } } } } #if TD_PORT_WINDOWS iocp->interrupt_loop(); iocp_thread.join(); #endif }