mirror of
https://github.com/matrix-construct/construct
synced 2024-11-26 16:52:44 +01:00
94 lines
3.2 KiB
C++
94 lines
3.2 KiB
C++
// Matrix Construct
|
|
//
|
|
// Copyright (C) Matrix Construct Developers, Authors & Contributors
|
|
// Copyright (C) 2016-2018 Jason Volk <jason@zemos.net>
|
|
//
|
|
// Permission to use, copy, modify, and/or distribute this software for any
|
|
// purpose with or without fee is hereby granted, provided that the above
|
|
// copyright notice and this permission notice is present in all copies. The
|
|
// full license for this software is available in the LICENSE file.
|
|
|
|
#pragma once
|
|
#define HAVE_IRCD_NET_RESOLVER_H
|
|
|
|
// This file is not included with the IRCd standard include stack because
|
|
// it requires symbols we can't forward declare without boost headers. It
|
|
// is part of the <ircd/asio.h> stack which can be included in your
|
|
// definition file if you need low level access to this resolver API.
|
|
|
|
/// Internal resolver service
|
|
struct ircd::net::dns::resolver
|
|
{
|
|
struct tag;
|
|
using header = rfc1035::header;
|
|
|
|
static constexpr const size_t &MAX_COUNT{64};
|
|
static conf::item<milliseconds> timeout;
|
|
static conf::item<milliseconds> send_rate;
|
|
|
|
std::vector<ip::udp::endpoint> server; // The list of active servers
|
|
size_t server_next{0}; // Round-robin state to hit servers
|
|
void init_servers();
|
|
|
|
ctx::dock dock;
|
|
std::map<uint16_t, tag> tags; // The active requests
|
|
steady_point send_last; // Time of last send
|
|
using queued = std::pair<uint16_t, std::string>;
|
|
std::deque<queued> sendq; // Queue of frames for rate-limiting
|
|
|
|
ip::udp::socket ns; // A pollable activity object
|
|
ip::udp::endpoint reply_from; // Remote addr of recv
|
|
unique_buffer<mutable_buffer> reply; // Buffer for recv
|
|
|
|
bool handle_error(const error_code &ec) const;
|
|
bool handle_error(const header &, const rfc1035::question &, tag &);
|
|
void handle_reply(const header &, const const_buffer &body, tag &);
|
|
void handle_reply(const header &, const const_buffer &body);
|
|
void handle(const error_code &ec, const size_t &) noexcept;
|
|
void set_handle();
|
|
|
|
void queue_query(const const_buffer &, tag &);
|
|
void send_query(const ip::udp::endpoint &, const const_buffer &, tag &);
|
|
void send_query(const const_buffer &, tag &);
|
|
void submit(const const_buffer &, tag &);
|
|
|
|
template<class... A> tag &set_tag(A&&...);
|
|
const_buffer make_query(const mutable_buffer &buf, const tag &) const;
|
|
void operator()(const hostport &, const opts &, callback);
|
|
|
|
bool check_timeout(const uint16_t &id, tag &, const steady_point &expired);
|
|
void check_timeouts(const milliseconds &timeout);
|
|
void timeout_worker();
|
|
ctx::context timeout_context;
|
|
|
|
void flush(const queued &);
|
|
void sendq_worker();
|
|
ctx::context sendq_context;
|
|
|
|
resolver();
|
|
~resolver() noexcept;
|
|
};
|
|
|
|
struct ircd::net::dns::resolver::tag
|
|
{
|
|
uint16_t id {0};
|
|
hostport hp;
|
|
dns::opts opts; // note: invalid after query sent
|
|
callback cb;
|
|
steady_point last;
|
|
uint8_t tries {0};
|
|
char hostbuf[256];
|
|
|
|
tag(const hostport &, const dns::opts &, callback &&);
|
|
};
|
|
|
|
inline
|
|
ircd::net::dns::resolver::tag::tag(const hostport &hp,
|
|
const dns::opts &opts,
|
|
callback &&cb)
|
|
:hp{hp}
|
|
,opts{opts}
|
|
,cb{std::move(cb)}
|
|
{
|
|
this->hp.host = { hostbuf, copy(hostbuf, hp.host) };
|
|
}
|