95e09be799
POSIX stack does not allow one to bind more than one socket to given port. Native stack on the other hand does. The way services are set up depends on that. For instance, on native stack one might want to start the service on all cores, but on POSIX stack only on one of them.
86 lines
3.0 KiB
C++
86 lines
3.0 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 std::unique_ptr<network_stack> create(boost::program_options::variables_map opts) {
|
|
return 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 std::unique_ptr<network_stack> create(boost::program_options::variables_map opts) {
|
|
return std::unique_ptr<network_stack>(new posix_ap_network_stack(opts));
|
|
}
|
|
};
|
|
|
|
}
|
|
|
|
#endif
|