mirror of
https://github.com/matrix-construct/construct
synced 2024-08-25 11:02:08 +02:00
100 lines
2.7 KiB
C++
100 lines
2.7 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_ACCEPTOR_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 acceptor API.
|
|
|
|
namespace ircd::net
|
|
{
|
|
std::ostream &operator<<(std::ostream &s, const struct listener::acceptor &);
|
|
std::ostream &operator<<(std::ostream &s, const struct listener_udp::acceptor &);
|
|
}
|
|
|
|
struct ircd::net::listener::acceptor
|
|
:std::enable_shared_from_this<struct ircd::net::listener::acceptor>
|
|
{
|
|
using error_code = boost::system::error_code;
|
|
|
|
static log::log log;
|
|
static conf::item<milliseconds> timeout;
|
|
|
|
net::listener *listener;
|
|
std::string name;
|
|
std::string opts;
|
|
size_t backlog;
|
|
listener::callback cb;
|
|
listener::proffer pcb;
|
|
asio::ssl::context ssl;
|
|
ip::tcp::endpoint ep;
|
|
ip::tcp::acceptor a;
|
|
size_t accepting {0};
|
|
size_t handshaking {0};
|
|
bool interrupting {false};
|
|
bool handle_set {false};
|
|
ctx::dock joining;
|
|
|
|
void configure(const json::object &opts);
|
|
|
|
// Handshake stack
|
|
void check_handshake_error(const error_code &ec, socket &);
|
|
void handshake(const error_code &ec, std::shared_ptr<socket>, std::weak_ptr<acceptor>) noexcept;
|
|
|
|
// Acceptance stack
|
|
bool check_accept_error(const error_code &ec, socket &);
|
|
void accept(const error_code &ec, std::shared_ptr<socket>, std::weak_ptr<acceptor>) noexcept;
|
|
|
|
// Accept next
|
|
bool set_handle();
|
|
|
|
// Acceptor shutdown
|
|
bool interrupt() noexcept;
|
|
void join() noexcept;
|
|
|
|
acceptor(net::listener &,
|
|
const string_view &name,
|
|
const json::object &opts,
|
|
listener::callback,
|
|
listener::proffer);
|
|
|
|
~acceptor() noexcept;
|
|
};
|
|
|
|
struct ircd::net::listener_udp::acceptor
|
|
{
|
|
using error_code = boost::system::error_code;
|
|
|
|
static constexpr log::log &log {listener::acceptor::log};
|
|
static ip::udp::socket::message_flags flags(const flag &);
|
|
|
|
std::string name;
|
|
std::string opts;
|
|
ip::udp::endpoint ep;
|
|
ip::udp::socket a;
|
|
size_t waiting {0};
|
|
ctx::dock joining;
|
|
|
|
// Yield context for datagram.
|
|
datagram &operator()(datagram &);
|
|
|
|
// Acceptor shutdown
|
|
bool interrupt() noexcept;
|
|
void join() noexcept;
|
|
|
|
acceptor(const string_view &name,
|
|
const json::object &opts);
|
|
|
|
~acceptor() noexcept;
|
|
};
|