mirror of
https://github.com/scylladb/scylladb.git
synced 2026-04-25 19:10:42 +00:00
4f4731c37b
Network device has to be available when network stack is created, but sometimes network device creation should wait for device initialization by another cpu. This patch makes it possible to delay network stack creation until network device is available.
86 lines
3.1 KiB
C++
86 lines
3.1 KiB
C++
/*
|
|
* Copyright (C) 2014 Cloudius Systems, Ltd.
|
|
*/
|
|
|
|
#ifndef POSIX_STACK_HH_
|
|
#define POSIX_STACK_HH_
|
|
|
|
#include "core/reactor.hh"
|
|
#include <boost/program_options.hpp>
|
|
|
|
namespace net {
|
|
|
|
data_source posix_data_source(pollable_fd& fd);
|
|
data_sink posix_data_sink(pollable_fd& fd);
|
|
|
|
class posix_data_source_impl final : public data_source_impl {
|
|
pollable_fd& _fd;
|
|
temporary_buffer<char> _buf;
|
|
size_t _buf_size;
|
|
public:
|
|
explicit posix_data_source_impl(pollable_fd& fd, size_t buf_size = 8192)
|
|
: _fd(fd), _buf(buf_size), _buf_size(buf_size) {}
|
|
virtual future<temporary_buffer<char>> get() override;
|
|
};
|
|
|
|
class posix_data_sink_impl : public data_sink_impl {
|
|
pollable_fd& _fd;
|
|
std::vector<temporary_buffer<char>> _data;
|
|
private:
|
|
future<> do_write(size_t idx);
|
|
public:
|
|
explicit posix_data_sink_impl(pollable_fd& fd) : _fd(fd) {}
|
|
future<> put(std::vector<temporary_buffer<char>> data) override;
|
|
future<> close() override {
|
|
_fd.close();
|
|
return make_ready_future<>();
|
|
}
|
|
};
|
|
|
|
class posix_ap_server_socket_impl : public server_socket_impl {
|
|
struct connection {
|
|
pollable_fd fd;
|
|
socket_address addr;
|
|
connection(pollable_fd xfd, socket_address xaddr) : fd(std::move(xfd)), addr(xaddr) {}
|
|
};
|
|
static thread_local std::unordered_map<::sockaddr_in, promise<connected_socket, socket_address>> sockets;
|
|
static thread_local std::unordered_multimap<::sockaddr_in, connection> conn_q;
|
|
socket_address _sa;
|
|
public:
|
|
explicit posix_ap_server_socket_impl(socket_address sa) : _sa(sa) {}
|
|
virtual future<connected_socket, socket_address> accept();
|
|
static void move_connected_socket(socket_address sa, pollable_fd fd, socket_address addr);
|
|
};
|
|
|
|
class posix_server_socket_impl : public server_socket_impl {
|
|
socket_address _sa;
|
|
pollable_fd _lfd;
|
|
public:
|
|
explicit posix_server_socket_impl(socket_address sa, pollable_fd lfd) : _sa(sa), _lfd(std::move(lfd)) {}
|
|
virtual future<connected_socket, socket_address> accept();
|
|
};
|
|
|
|
class posix_network_stack : public network_stack {
|
|
public:
|
|
posix_network_stack(boost::program_options::variables_map opts) {}
|
|
virtual server_socket listen(socket_address sa, listen_options opts) override;
|
|
virtual net::udp_channel make_udp_channel(ipv4_addr addr) override;
|
|
static future<std::unique_ptr<network_stack>> create(boost::program_options::variables_map opts) {
|
|
return make_ready_future<std::unique_ptr<network_stack>>(std::unique_ptr<network_stack>(new posix_network_stack(opts)));
|
|
}
|
|
virtual bool has_per_core_namespace() override { return false; };
|
|
};
|
|
|
|
class posix_ap_network_stack : public posix_network_stack {
|
|
public:
|
|
posix_ap_network_stack(boost::program_options::variables_map opts) : posix_network_stack(std::move(opts)) {}
|
|
virtual server_socket listen(socket_address sa, listen_options opts) override;
|
|
static future<std::unique_ptr<network_stack>> create(boost::program_options::variables_map opts) {
|
|
return make_ready_future<std::unique_ptr<network_stack>>(std::unique_ptr<network_stack>(new posix_ap_network_stack(opts)));
|
|
}
|
|
};
|
|
|
|
}
|
|
|
|
#endif
|