77 lines
2.0 KiB
C++
77 lines
2.0 KiB
C++
//
|
|
// 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)
|
|
//
|
|
#pragma once
|
|
|
|
#include "td/actor/actor.h"
|
|
|
|
#include "td/utils/common.h"
|
|
#include "td/utils/FlatHashMap.h"
|
|
#include "td/utils/logging.h"
|
|
#include "td/utils/port/IPAddress.h"
|
|
#include "td/utils/Promise.h"
|
|
#include "td/utils/Status.h"
|
|
|
|
#include <utility>
|
|
|
|
namespace td {
|
|
|
|
extern int VERBOSITY_NAME(dns_resolver);
|
|
|
|
class GetHostByNameActor final : public Actor {
|
|
public:
|
|
enum class ResolverType { Native, Google };
|
|
|
|
struct Options {
|
|
static constexpr int32 DEFAULT_CACHE_TIME = 60 * 29; // 29 minutes
|
|
static constexpr int32 DEFAULT_ERROR_CACHE_TIME = 60 * 5; // 5 minutes
|
|
|
|
vector<ResolverType> resolver_types{ResolverType::Native};
|
|
int32 scheduler_id{-1};
|
|
int32 ok_timeout{DEFAULT_CACHE_TIME};
|
|
int32 error_timeout{DEFAULT_ERROR_CACHE_TIME};
|
|
};
|
|
|
|
explicit GetHostByNameActor(Options options);
|
|
|
|
void run(std::string host, int port, bool prefer_ipv6, Promise<IPAddress> promise);
|
|
|
|
private:
|
|
void on_query_result(std::string host, bool prefer_ipv6, Result<IPAddress> result);
|
|
|
|
struct Value {
|
|
Result<IPAddress> ip;
|
|
double expires_at;
|
|
|
|
Value(Result<IPAddress> ip, double expires_at) : ip(std::move(ip)), expires_at(expires_at) {
|
|
}
|
|
|
|
Result<IPAddress> get_ip_port(int port) const {
|
|
auto result = ip.clone();
|
|
if (result.is_ok()) {
|
|
result.ok_ref().set_port(port);
|
|
}
|
|
return result;
|
|
}
|
|
};
|
|
FlatHashMap<string, Value> cache_[2];
|
|
|
|
struct Query {
|
|
ActorOwn<> query;
|
|
size_t pos = 0;
|
|
string real_host;
|
|
double begin_time = 0.0;
|
|
std::vector<std::pair<int, Promise<IPAddress>>> promises;
|
|
};
|
|
FlatHashMap<string, unique_ptr<Query>> active_queries_[2];
|
|
|
|
Options options_;
|
|
|
|
void run_query(std::string host, bool prefer_ipv6, Query &query);
|
|
};
|
|
|
|
} // namespace td
|